mirror of
https://github.com/pseXperiments/icicle.git
synced 2026-01-07 22:53:56 -05:00
fd62fe5ae8
Resolves #191 and #113 --------- Co-authored-by: DmytroTym <dmytrotym1@gmail.com> Co-authored-by: ImmanuelSegol <3ditds@gmail.com>
86 lines
2.4 KiB
Cheetah
86 lines
2.4 KiB
Cheetah
import (
|
|
"encoding/binary"
|
|
"unsafe"
|
|
)
|
|
|
|
// #cgo CFLAGS: -I./include/
|
|
// #cgo CFLAGS: -I/usr/local/cuda/include
|
|
// #cgo LDFLAGS: -L${SRCDIR}/../../ {{.SharedLib}}
|
|
// #include "projective.h"
|
|
// #include "ve_mod_mult.h"
|
|
import "C"
|
|
|
|
// G2 extension field
|
|
|
|
type G2Element [{{.G2ElementSize}}]uint64
|
|
|
|
type ExtentionField struct {
|
|
A0, A1 G2Element
|
|
}
|
|
|
|
type G2PointAffine struct {
|
|
X, Y ExtentionField
|
|
}
|
|
|
|
type G2Point struct {
|
|
X, Y, Z ExtentionField
|
|
}
|
|
|
|
func (p *G2Point) Random() *G2Point {
|
|
outC := (*C.{{.CurveNameUpperCase}}_g2_projective_t)(unsafe.Pointer(p))
|
|
C.random_g2_projective_{{.CurveNameLowerCase}}(outC)
|
|
|
|
return p
|
|
}
|
|
|
|
|
|
func (p *G2Point) Eq(pCompare *G2Point) bool {
|
|
// Cast *Point{{.CurveNameUpperCase}} to *C.{{.CurveNameUpperCase}}_projective_t
|
|
// The unsafe.Pointer cast is necessary because Go doesn't allow direct casts
|
|
// between different pointer types.
|
|
// It's your responsibility to ensure that the types are compatible.
|
|
pC := (*C.{{.CurveNameUpperCase}}_g2_projective_t)(unsafe.Pointer(p))
|
|
pCompareC := (*C.{{.CurveNameUpperCase}}_g2_projective_t)(unsafe.Pointer(pCompare))
|
|
|
|
// Call the C function
|
|
// The C function doesn't keep any references to the data,
|
|
// so it's fine if the Go garbage collector moves or deletes the data later.
|
|
return bool(C.eq_g2_{{.CurveNameLowerCase}}(pC, pCompareC))
|
|
}
|
|
|
|
func (f *G2Element) ToBytesLe() []byte {
|
|
var bytes []byte
|
|
for _, val := range f {
|
|
buf := make([]byte, 8) // 8 bytes because uint64 is 64-bit
|
|
binary.LittleEndian.PutUint64(buf, val)
|
|
bytes = append(bytes, buf...)
|
|
}
|
|
return bytes
|
|
}
|
|
|
|
func (p *G2Point) FromAffine(affine *G2PointAffine) *G2Point {
|
|
out := (*C.{{.CurveNameUpperCase}}_g2_projective_t)(unsafe.Pointer(p))
|
|
in := (*C.{{.CurveNameUpperCase}}_g2_affine_t)(unsafe.Pointer(affine))
|
|
|
|
C.g2_projective_from_affine_{{.CurveNameLowerCase}}(out, in)
|
|
|
|
return p
|
|
}
|
|
|
|
func (p *G2PointAffine) FromProjective(projective *G2Point) *G2PointAffine {
|
|
out := (*C.{{.CurveNameUpperCase}}_g2_affine_t)(unsafe.Pointer(p))
|
|
in := (*C.{{.CurveNameUpperCase}}_g2_projective_t)(unsafe.Pointer(projective))
|
|
|
|
C.g2_projective_to_affine_{{.CurveNameLowerCase}}(out, in)
|
|
|
|
return p
|
|
}
|
|
|
|
func (p *G2Point) IsOnCurve() bool {
|
|
// Directly copy memory from the C struct to the Go struct
|
|
point := (*C.{{.CurveNameUpperCase}}_g2_projective_t)(unsafe.Pointer(p))
|
|
res := C.g2_projective_is_on_curve_{{.CurveNameLowerCase}}(point)
|
|
|
|
return bool(res)
|
|
}
|