build: generify g2.go changes

Consensys · Jul 1, 2024 · 4d24b66 · 4d24b66
1 parent e935181
commit 4d24b66
Show file tree

Hide file tree

Showing 9 changed files with 139 additions and 0 deletions.
diff --git a/ecc/bls12-377/g2.go b/ecc/bls12-377/g2.go
diff --git a/ecc/bls12-378/g2.go b/ecc/bls12-378/g2.go
diff --git a/ecc/bls12-381/g2.go b/ecc/bls12-381/g2.go
diff --git a/ecc/bls24-315/g2.go b/ecc/bls24-315/g2.go
diff --git a/ecc/bls24-317/g2.go b/ecc/bls24-317/g2.go
diff --git a/ecc/bw6-633/g2.go b/ecc/bw6-633/g2.go
diff --git a/ecc/bw6-756/g2.go b/ecc/bw6-756/g2.go
diff --git a/ecc/bw6-761/g2.go b/ecc/bw6-761/g2.go
diff --git a/internal/generator/ecc/template/point.go.tmpl b/internal/generator/ecc/template/point.go.tmpl
@@ -6,6 +6,9 @@
 
 
 import (
+	{{- if eq .PointName "g2"}}
+	"crypto/rand"
+	{{- end}}
 	"math/big"
 	"runtime"
 
@@ -1822,3 +1825,19 @@ func batchAdd{{ $TAffine }}[TP p{{ $TAffine }}, TPP pp{{ $TAffine }}, TC c{{ $TA
 		(*R)[j].Set(&rr)
 	}
 }
+
+{{ if eq .PointName "g2"}}
+func RandomOnG2() (G2Affine, error) {
+	if gBytes, err := randomFrSizedBytes(); err != nil {
+		return G2Affine{}, err
+	} else {
+		return HashToG2(gBytes, []byte("random on g2"))
+	}
+}
+
+func randomFrSizedBytes() ([]byte, error) {
+	res := make([]byte, fr.Bytes)
+	_, err := rand.Read(res)
+	return res, err
+}
+{{- end}}