Merge pull request #9 from cloudflare/testMarshal

Adds identities on marshaling function for G1,G2,GT.
cloudflare · May 23, 2019 · 828ba4f · 828ba4f
2 parents 0ee3663 + 0a352d7
commit 828ba4f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/bn256.go b/bn256.go
@@ -105,6 +105,10 @@ func (e *G1) Marshal() []byte {
 	// Each value is a 256-bit number.
 	const numBytes = 256 / 8
 
+	if e.p == nil {
+		e.p = &curvePoint{}
+	}
+
 	e.p.MakeAffine()
 	ret := make([]byte, numBytes*2)
 	if e.p.IsInfinity() {
@@ -391,6 +395,11 @@ func (e *GT) Marshal() []byte {
 	// Each value is a 256-bit number.
 	const numBytes = 256 / 8
 
+	if e.p == nil {
+		e.p = &gfP12{}
+		e.p.SetOne()
+	}
+
 	ret := make([]byte, numBytes*12)
 	temp := &gfP{}