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

Replace CKZGRet with error in go bindings (#252)

This commit is contained in:
Justin Traglia 2023-03-29 19:33:15 -05:00 committed by GitHub
parent a7f4890a7e
commit f942864765
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 65 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
bindings/go/main.go
View File

@ -7,6 +7,8 @@ package cgokzg4844
import "C" import "C"
import ( import (
"errors"
"fmt"
"unsafe" "unsafe"
// So its functions are available during compilation. // So its functions are available during compilation.
@ -22,7 +24,6 @@ const (
) )
type ( type (
CKZGRet uint
Bytes32 [32]byte Bytes32 [32]byte
Bytes48 [48]byte Bytes48 [48]byte
KZGCommitment Bytes48 KZGCommitment Bytes48
@ -30,17 +31,34 @@ type (
Blob [BytesPerBlob]byte Blob [BytesPerBlob]byte
) )
const ( var (
C_KZG_OK CKZGRet = C.C_KZG_OK loaded = false
C_KZG_BADARGS CKZGRet = C.C_KZG_BADARGS settings = C.KZGSettings{}
C_KZG_ERROR CKZGRet = C.C_KZG_ERROR ErrBadArgs = errors.New("bad arguments")
C_KZG_MALLOC CKZGRet = C.C_KZG_MALLOC ErrError = errors.New("unexpected error")
ErrMalloc = errors.New("malloc failed")
errorMap = map[C.C_KZG_RET]error{
C.C_KZG_OK: nil,
C.C_KZG_BADARGS: ErrBadArgs,
C.C_KZG_ERROR: ErrError,
C.C_KZG_MALLOC: ErrMalloc,
}
) )
var ( ///////////////////////////////////////////////////////////////////////////////
loaded = false // Helper functions
settings = C.KZGSettings{} ///////////////////////////////////////////////////////////////////////////////
)
// makeErrorFromRet translates an (integral) return value, as reported
// by the C library, into a proper Go error. If there is no error, this
// will return nil.
func makeErrorFromRet(ret C.C_KZG_RET) error {
err, ok := errorMap[ret]
if !ok {
panic(fmt.Sprintf("unexpected return value: %v", ret))
}
return err
}
/////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////
// Public functions // Public functions
@ -56,7 +74,7 @@ LoadTrustedSetup is the binding for:
const uint8_t *g2_bytes, const uint8_t *g2_bytes,
size_t n2); size_t n2);
*/ */
func LoadTrustedSetup(g1Bytes, g2Bytes []byte) CKZGRet { func LoadTrustedSetup(g1Bytes, g2Bytes []byte) error {
if loaded { if loaded {
panic("trusted setup is already loaded") panic("trusted setup is already loaded")
} }
@ -74,10 +92,10 @@ func LoadTrustedSetup(g1Bytes, g2Bytes []byte) CKZGRet {
(C.size_t)(numG1Elements), (C.size_t)(numG1Elements),
*(**C.uint8_t)(unsafe.Pointer(&g2Bytes)), *(**C.uint8_t)(unsafe.Pointer(&g2Bytes)),
(C.size_t)(numG2Elements)) (C.size_t)(numG2Elements))
if CKZGRet(ret) == C_KZG_OK { if ret == C.C_KZG_OK {
loaded = true loaded = true
} }
return CKZGRet(ret) return makeErrorFromRet(ret)
} }
/* /*
@ -87,7 +105,7 @@ LoadTrustedSetupFile is the binding for:
KZGSettings *out, KZGSettings *out,
FILE *in); FILE *in);
*/ */
func LoadTrustedSetupFile(trustedSetupFile string) CKZGRet { func LoadTrustedSetupFile(trustedSetupFile string) error {
if loaded { if loaded {
panic("trusted setup is already loaded") panic("trusted setup is already loaded")
} }
@ -101,10 +119,10 @@ func LoadTrustedSetupFile(trustedSetupFile string) CKZGRet {
} }
ret := C.load_trusted_setup_file(&settings, fp) ret := C.load_trusted_setup_file(&settings, fp)
C.fclose(fp) C.fclose(fp)
if CKZGRet(ret) == C_KZG_OK { if ret == C.C_KZG_OK {
loaded = true loaded = true
} }
return CKZGRet(ret) return makeErrorFromRet(ret)
} }
/* /*
@ -129,7 +147,7 @@ BlobToKZGCommitment is the binding for:
const Blob *blob, const Blob *blob,
const KZGSettings *s); const KZGSettings *s);
*/ */
func BlobToKZGCommitment(blob Blob) (KZGCommitment, CKZGRet) { func BlobToKZGCommitment(blob Blob) (KZGCommitment, error) {
if !loaded { if !loaded {
panic("trusted setup isn't loaded") panic("trusted setup isn't loaded")
} }
@ -138,7 +156,7 @@ func BlobToKZGCommitment(blob Blob) (KZGCommitment, CKZGRet) {
(*C.KZGCommitment)(unsafe.Pointer(&commitment)), (*C.KZGCommitment)(unsafe.Pointer(&commitment)),
(*C.Blob)(unsafe.Pointer(&blob)), (*C.Blob)(unsafe.Pointer(&blob)),
&settings) &settings)
return commitment, CKZGRet(ret) return commitment, makeErrorFromRet(ret)
} }
/* /*
@ -151,7 +169,7 @@ ComputeKZGProof is the binding for:
const Bytes32 *z_bytes, const Bytes32 *z_bytes,
const KZGSettings *s); const KZGSettings *s);
*/ */
func ComputeKZGProof(blob Blob, zBytes Bytes32) (KZGProof, Bytes32, CKZGRet) { func ComputeKZGProof(blob Blob, zBytes Bytes32) (KZGProof, Bytes32, error) {
if !loaded { if !loaded {
panic("trusted setup isn't loaded") panic("trusted setup isn't loaded")
} }
@ -163,7 +181,7 @@ func ComputeKZGProof(blob Blob, zBytes Bytes32) (KZGProof, Bytes32, CKZGRet) {
(*C.Blob)(unsafe.Pointer(&blob)), (*C.Blob)(unsafe.Pointer(&blob)),
(*C.Bytes32)(unsafe.Pointer(&zBytes)), (*C.Bytes32)(unsafe.Pointer(&zBytes)),
&settings) &settings)
return proof, y, CKZGRet(ret) return proof, y, makeErrorFromRet(ret)
} }
/* /*
@ -175,7 +193,7 @@ ComputeBlobKZGProof is the binding for:
const Bytes48 *commitment_bytes, const Bytes48 *commitment_bytes,
const KZGSettings *s); const KZGSettings *s);
*/ */
func ComputeBlobKZGProof(blob Blob, commitmentBytes Bytes48) (KZGProof, CKZGRet) { func ComputeBlobKZGProof(blob Blob, commitmentBytes Bytes48) (KZGProof, error) {
if !loaded { if !loaded {
panic("trusted setup isn't loaded") panic("trusted setup isn't loaded")
} }
@ -185,7 +203,7 @@ func ComputeBlobKZGProof(blob Blob, commitmentBytes Bytes48) (KZGProof, CKZGRet)
(*C.Blob)(unsafe.Pointer(&blob)), (*C.Blob)(unsafe.Pointer(&blob)),
(*C.Bytes48)(unsafe.Pointer(&commitmentBytes)), (*C.Bytes48)(unsafe.Pointer(&commitmentBytes)),
&settings) &settings)
return proof, CKZGRet(ret) return proof, makeErrorFromRet(ret)
} }
/* /*
@ -199,7 +217,7 @@ VerifyKZGProof is the binding for:
const Bytes48 *proof_bytes, const Bytes48 *proof_bytes,
const KZGSettings *s); const KZGSettings *s);
*/ */
func VerifyKZGProof(commitmentBytes Bytes48, zBytes, yBytes Bytes32, proofBytes Bytes48) (bool, CKZGRet) { func VerifyKZGProof(commitmentBytes Bytes48, zBytes, yBytes Bytes32, proofBytes Bytes48) (bool, error) {
if !loaded { if !loaded {
panic("trusted setup isn't loaded") panic("trusted setup isn't loaded")
} }
@ -211,7 +229,7 @@ func VerifyKZGProof(commitmentBytes Bytes48, zBytes, yBytes Bytes32, proofBytes
(*C.Bytes32)(unsafe.Pointer(&yBytes)), (*C.Bytes32)(unsafe.Pointer(&yBytes)),
(*C.Bytes48)(unsafe.Pointer(&proofBytes)), (*C.Bytes48)(unsafe.Pointer(&proofBytes)),
&settings) &settings)
return bool(result), CKZGRet(ret) return bool(result), makeErrorFromRet(ret)
} }
/* /*
@ -224,7 +242,7 @@ VerifyBlobKZGProof is the binding for:
const Bytes48 *proof_bytes, const Bytes48 *proof_bytes,
const KZGSettings *s); const KZGSettings *s);
*/ */
func VerifyBlobKZGProof(blob Blob, commitmentBytes, proofBytes Bytes48) (bool, CKZGRet) { func VerifyBlobKZGProof(blob Blob, commitmentBytes, proofBytes Bytes48) (bool, error) {
if !loaded { if !loaded {
panic("trusted setup isn't loaded") panic("trusted setup isn't loaded")
} }
@ -235,7 +253,7 @@ func VerifyBlobKZGProof(blob Blob, commitmentBytes, proofBytes Bytes48) (bool, C
(*C.Bytes48)(unsafe.Pointer(&commitmentBytes)), (*C.Bytes48)(unsafe.Pointer(&commitmentBytes)),
(*C.Bytes48)(unsafe.Pointer(&proofBytes)), (*C.Bytes48)(unsafe.Pointer(&proofBytes)),
&settings) &settings)
return bool(result), CKZGRet(ret) return bool(result), makeErrorFromRet(ret)
} }
/* /*
@ -248,12 +266,12 @@ VerifyBlobKZGProofBatch is the binding for:
const Bytes48 *proofs_bytes, const Bytes48 *proofs_bytes,
const KZGSettings *s); const KZGSettings *s);
*/ */
func VerifyBlobKZGProofBatch(blobs []Blob, commitmentsBytes, proofsBytes []Bytes48) (bool, CKZGRet) { func VerifyBlobKZGProofBatch(blobs []Blob, commitmentsBytes, proofsBytes []Bytes48) (bool, error) {
if !loaded { if !loaded {
panic("trusted setup isn't loaded") panic("trusted setup isn't loaded")
} }
if len(blobs) != len(commitmentsBytes) || len(blobs) != len(proofsBytes) { if len(blobs) != len(commitmentsBytes) || len(blobs) != len(proofsBytes) {
return false, C_KZG_BADARGS return false, ErrBadArgs
} }
var result C.bool var result C.bool
@ -264,7 +282,7 @@ func VerifyBlobKZGProofBatch(blobs []Blob, commitmentsBytes, proofsBytes []Bytes
*(**C.Bytes48)(unsafe.Pointer(&proofsBytes)), *(**C.Bytes48)(unsafe.Pointer(&proofsBytes)),
(C.size_t)(len(blobs)), (C.size_t)(len(blobs)),
&settings) &settings)
return bool(result), CKZGRet(ret) return bool(result), makeErrorFromRet(ret)
} }
/////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////

36
bindings/go/main_test.go
View File

@ -14,8 +14,8 @@ import (
) )
func TestMain(m *testing.M) { func TestMain(m *testing.M) {
ret := LoadTrustedSetupFile("../../src/trusted_setup.txt") err := LoadTrustedSetupFile("../../src/trusted_setup.txt")
if ret != 0 { if err != nil {
panic("failed to load trusted setup") panic("failed to load trusted setup")
} }
defer FreeTrustedSetup() defer FreeTrustedSetup()
@ -146,8 +146,8 @@ func TestBlobToKZGCommitment(t *testing.T) {
return return
} }
commitment, ret := BlobToKZGCommitment(blob) commitment, err := BlobToKZGCommitment(blob)
if ret == C_KZG_OK { if err == nil {
require.NotNil(t, test.Output) require.NotNil(t, test.Output)
require.Equal(t, test.Output[:], commitment[:]) require.Equal(t, test.Output[:], commitment[:])
} else { } else {
@ -193,8 +193,8 @@ func TestComputeKZGProof(t *testing.T) {
return return
} }
proof, y, ret := ComputeKZGProof(blob, z) proof, y, err := ComputeKZGProof(blob, z)
if ret == C_KZG_OK { if err == nil {
require.NotNil(t, test.Output) require.NotNil(t, test.Output)
var expectedProof Bytes48 var expectedProof Bytes48
err = expectedProof.UnmarshalText([]byte((*test.Output)[0])) err = expectedProof.UnmarshalText([]byte((*test.Output)[0]))
@ -247,8 +247,8 @@ func TestComputeBlobKZGProof(t *testing.T) {
return return
} }
proof, ret := ComputeBlobKZGProof(blob, commitment) proof, err := ComputeBlobKZGProof(blob, commitment)
if ret == C_KZG_OK { if err == nil {
require.NotNil(t, test.Output) require.NotNil(t, test.Output)
require.Equal(t, test.Output[:], proof[:]) require.Equal(t, test.Output[:], proof[:])
} else { } else {
@ -310,8 +310,8 @@ func TestVerifyKZGProof(t *testing.T) {
return return
} }
valid, ret := VerifyKZGProof(commitment, z, y, proof) valid, err := VerifyKZGProof(commitment, z, y, proof)
if ret == C_KZG_OK { if err == nil {
require.NotNil(t, test.Output) require.NotNil(t, test.Output)
require.Equal(t, *test.Output, valid) require.Equal(t, *test.Output, valid)
} else { } else {
@ -365,8 +365,8 @@ func TestVerifyBlobKZGProof(t *testing.T) {
return return
} }
valid, ret := VerifyBlobKZGProof(blob, commitment, proof) valid, err := VerifyBlobKZGProof(blob, commitment, proof)
if ret == C_KZG_OK { if err == nil {
require.NotNil(t, test.Output) require.NotNil(t, test.Output)
require.Equal(t, *test.Output, valid) require.Equal(t, *test.Output, valid)
} else { } else {
@ -432,8 +432,8 @@ func TestVerifyBlobKZGProofBatch(t *testing.T) {
proofs = append(proofs, proof) proofs = append(proofs, proof)
} }
valid, ret := VerifyBlobKZGProofBatch(blobs, commitments, proofs) valid, err := VerifyBlobKZGProofBatch(blobs, commitments, proofs)
if ret == C_KZG_OK { if err == nil {
require.NotNil(t, test.Output) require.NotNil(t, test.Output)
require.Equal(t, *test.Output, valid) require.Equal(t, *test.Output, valid)
} else { } else {
@ -455,10 +455,10 @@ func Benchmark(b *testing.B) {
fields := [length]Bytes32{} fields := [length]Bytes32{}
for i := 0; i < length; i++ { for i := 0; i < length; i++ {
blob := GetRandBlob(int64(i)) blob := GetRandBlob(int64(i))
commitment, ret := BlobToKZGCommitment(blob) commitment, err := BlobToKZGCommitment(blob)
require.Equal(b, ret, C_KZG_OK) require.NoError(b, err)
proof, ret := ComputeBlobKZGProof(blob, Bytes48(commitment)) proof, err := ComputeBlobKZGProof(blob, Bytes48(commitment))
require.Equal(b, ret, C_KZG_OK) require.NoError(b, err)
blobs[i] = blob blobs[i] = blob
commitments[i] = Bytes48(commitment) commitments[i] = Bytes48(commitment)