1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
|
package oprf
import (
"encoding/binary"
"io"
"github.com/cloudflare/circl/group"
)
type PrivateKey struct {
p params
k group.Scalar
pub *PublicKey
}
type PublicKey struct {
p params
e group.Element
}
func (k *PrivateKey) MarshalBinary() ([]byte, error) { return k.k.MarshalBinary() }
func (k *PublicKey) MarshalBinary() ([]byte, error) { return k.e.MarshalBinaryCompress() }
func (k *PrivateKey) UnmarshalBinary(s Suite, data []byte) error {
p, ok := s.(params)
if !ok {
return ErrInvalidSuite
}
k.p = p
k.k = k.p.group.NewScalar()
return k.k.UnmarshalBinary(data)
}
func (k *PublicKey) UnmarshalBinary(s Suite, data []byte) error {
p, ok := s.(params)
if !ok {
return ErrInvalidSuite
}
k.p = p
k.e = k.p.group.NewElement()
return k.e.UnmarshalBinary(data)
}
func (k *PrivateKey) Public() *PublicKey {
if k.pub == nil {
k.pub = &PublicKey{k.p, k.p.group.NewElement().MulGen(k.k)}
}
return k.pub
}
// GenerateKey generates a private key compatible with the suite.
func GenerateKey(s Suite, rnd io.Reader) (*PrivateKey, error) {
if rnd == nil {
return nil, io.ErrNoProgress
}
p, ok := s.(params)
if !ok {
return nil, ErrInvalidSuite
}
privateKey := p.group.RandomScalar(rnd)
return &PrivateKey{p, privateKey, nil}, nil
}
// DeriveKey generates a private key from a 32-byte seed and an optional info string.
func DeriveKey(s Suite, mode Mode, seed, info []byte) (*PrivateKey, error) {
const maxTries = 255
p, ok := s.(params)
if !ok {
return nil, ErrInvalidSuite
}
if !isValidMode(mode) {
return nil, ErrInvalidMode
}
if len(seed) != 32 {
return nil, ErrInvalidSeed
}
p.m = mode
lenInfo := []byte{0, 0}
binary.BigEndian.PutUint16(lenInfo, uint16(len(info)))
deriveInput := append(append(append([]byte{}, seed...), lenInfo...), info...)
dst := p.getDST(deriveKeyPairDST)
zero := p.group.NewScalar()
privateKey := p.group.NewScalar()
for counter := byte(0); privateKey.IsEqual(zero); counter++ {
if counter > maxTries {
return nil, ErrDeriveKeyPairError
}
privateKey = p.group.HashToScalar(append(deriveInput, counter), dst)
}
return &PrivateKey{p, privateKey, nil}, nil
}
|