Add SetUint64 to Group.Scalar.

armfazh · armfazh · commit d7ec96929798 · 2021-11-03T16:23:24.000-07:00
diff --git a/group/group.go b/group/group.go
@@ -45,6 +45,7 @@ type Element interface {
 // Scalar represents an integer scalar.
 type Scalar interface {
 	IsEqual(Scalar) bool
+	SetUint64(uint64)
 	Add(Scalar, Scalar) Scalar
 	Sub(Scalar, Scalar) Scalar
 	Mul(Scalar, Scalar) Scalar
diff --git a/group/ristretto255.go b/group/ristretto255.go
@@ -4,6 +4,7 @@ import (
 	"crypto"
 	_ "crypto/sha512" // to link libraries
 	"io"
+	"math/big"
 
 	r255 "github.com/bwesterb/go-ristretto"
 )
@@ -145,6 +146,8 @@ func (e *ristrettoElement) UnmarshalBinary(data []byte) error {
 	return e.p.UnmarshalBinary(data)
 }
 
+func (s *ristrettoScalar) SetUint64(n uint64) { s.s.SetBigInt(new(big.Int).SetUint64(n)) }
+
 func (s *ristrettoScalar) IsEqual(x Scalar) bool {
 	return s.s.Equals(&x.(*ristrettoScalar).s)
 }
diff --git a/group/short.go b/group/short.go
@@ -195,7 +195,8 @@ type wScl struct {
 	k []byte
 }
 
-func (s *wScl) String() string { return fmt.Sprintf("0x%x", s.k) }
+func (s *wScl) String() string     { return fmt.Sprintf("0x%x", s.k) }
+func (s *wScl) SetUint64(n uint64) { s.fromBig(new(big.Int).SetUint64(n)) }
 func (s *wScl) IsEqual(a Scalar) bool {
 	aa := s.cvtScl(a)
 	return subtle.ConstantTimeCompare(s.k, aa.k) == 1

Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,7 @@ import (`
`4`	`4`	`"crypto"`
`5`	`5`	`_ "crypto/sha512" // to link libraries`
`6`	`6`	`"io"`
	`7`	`+ "math/big"`
`7`	`8`
`8`	`9`	`r255 "github.com/bwesterb/go-ristretto"`
`9`	`10`	`)`
`@@ -145,6 +146,8 @@ func (e *ristrettoElement) UnmarshalBinary(data []byte) error {`
`145`	`146`	`return e.p.UnmarshalBinary(data)`
`146`	`147`	`}`
`147`	`148`
	`149`	`+func (s *ristrettoScalar) SetUint64(n uint64) { s.s.SetBigInt(new(big.Int).SetUint64(n)) }`
	`150`	`+`
`148`	`151`	`func (s *ristrettoScalar) IsEqual(x Scalar) bool {`
`149`	`152`	`return s.s.Equals(&x.(*ristrettoScalar).s)`
`150`	`153`	`}`
Original file line number	Diff line number	Diff line change
`@@ -195,7 +195,8 @@ type wScl struct {`
`195`	`195`	`k []byte`
`196`	`196`	`}`
`197`	`197`
`198`		`-func (s *wScl) String() string { return fmt.Sprintf("0x%x", s.k) }`
	`198`	`+func (s *wScl) String() string { return fmt.Sprintf("0x%x", s.k) }`
	`199`	`+func (s *wScl) SetUint64(n uint64) { s.fromBig(new(big.Int).SetUint64(n)) }`
`199`	`200`	`func (s *wScl) IsEqual(a Scalar) bool {`
`200`	`201`	`aa := s.cvtScl(a)`
`201`	`202`	`return subtle.ConstantTimeCompare(s.k, aa.k) == 1`