status-im
/
c-kzg-4844
mirror of https://github.com/status-im/c-kzg-4844.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Compute kzg proof in swigless python

This commit is contained in:
Ramana Kumar 2022-10-02 13:21:13 +01:00
parent c6c4f7d5f6
commit 0ed8ef7b79
No known key found for this signature in database
GPG Key ID: ED471C788B900433
2 changed files with 37 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
min-bindings/python/ckzg.c
View File

@ -352,6 +352,30 @@ static PyObject* g1_lincomb_wrap(PyObject *self, PyObject *args) {
return PyCapsule_New(k, "G1", free_G1);
}
static PyObject* compute_kzg_proof_wrap(PyObject *self, PyObject *args) {
PyObject *p, *x, *s;
if (!PyArg_UnpackTuple(args, "compute_kzg_proof", 3, 3, &p, &x, &s) ||
!PyCapsule_IsValid(p, "PolynomialEvalForm") ||
!PyCapsule_IsValid(x, "BLSFieldElement") ||
!PyCapsule_IsValid(s, "KZGSettings"))
return PyErr_Format(PyExc_ValueError, "expected polynomial, field element, trusted setup");
KZGProof *k = (KZGProof*)malloc(sizeof(KZGProof));
if (k == NULL) return PyErr_NoMemory();
if (compute_kzg_proof(k,
PyCapsule_GetPointer(p, "PolynomialEvalForm"),
PyCapsule_GetPointer(x, "BLSFieldElement"),
PyCapsule_GetPointer(s, "KZGSettings")) != C_KZG_OK) {
free(k);
return PyErr_Format(PyExc_RuntimeError, "compute_kzg_proof failed");
}
return PyCapsule_New(k, "G1", free_G1);
}
static PyMethodDef ckzgmethods[] = {
{"bytes_from_g1", bytes_from_g1_wrap, METH_VARARGS, "Convert a group element to 48 bytes"},
{"int_from_bls_field", int_from_bls_field, METH_VARARGS, "Convert a field element to a 256-bit int"},
@ -362,6 +386,7 @@ static PyMethodDef ckzgmethods[] = {
{"compute_powers", compute_powers_wrap, METH_VARARGS, "Create a list of powers of a field element"},
{"vector_lincomb", vector_lincomb_wrap, METH_VARARGS, "Multiply a matrix of field elements with a vector"},
{"g1_lincomb", g1_lincomb_wrap, METH_VARARGS, "Linear combination of group elements with field elements"},
{"compute_kzg_proof", compute_kzg_proof_wrap, METH_VARARGS, "Compute KZG proof for polynomial at point"},
{NULL, NULL, 0, NULL}
};

12
min-bindings/python/tests.py
View File

@ -52,4 +52,16 @@ aggregated_poly_commitment = ckzg.g1_lincomb(kzg_commitments, r_powers)
# Compute proof
values_sedes = ssz.List(ssz.uint256, MAX_BLOBS_PER_BLOCK)
encoded_polynomial = ssz.encode([ckzg.int_from_bls_field(fr) for fr in values], values_sedes)
encoded_polynomial_length = ssz.encode(len(values), ssz.uint64)
encoded_commitment = ssz.encode(ckzg.bytes_from_g1(aggregated_poly_commitment), ssz.bytes48)
hashed_polynomial_and_commitment = ssz.hash.hashlib.sha256(
encoded_polynomial + encoded_polynomial_length + encoded_commitment).digest()
x = ckzg.bytes_to_bls_field(hashed_polynomial_and_commitment)
proof = ckzg.compute_kzg_proof(aggregated_poly, x, ts)
print('Tests passed')