forked from prysmaticlabs/prysm
/
direct.go
46 lines (41 loc) · 1.38 KB
/
direct.go
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
package keymanager
import (
"github.com/prysmaticlabs/prysm/shared/bls"
"github.com/prysmaticlabs/prysm/shared/bytesutil"
)
// Direct is a key manager that holds all secret keys directly.
type Direct struct {
// Key to the map is the bytes of the public key.
publicKeys map[[48]byte]*bls.PublicKey
// Key to the map is the bytes of the public key.
secretKeys map[[48]byte]*bls.SecretKey
}
// NewDirect creates a new direct key manager from the secret keys provided to it.
func NewDirect(sks []*bls.SecretKey) *Direct {
res := &Direct{
publicKeys: make(map[[48]byte]*bls.PublicKey),
secretKeys: make(map[[48]byte]*bls.SecretKey),
}
for _, sk := range sks {
publicKey := sk.PublicKey()
pubKey := bytesutil.ToBytes48(publicKey.Marshal())
res.publicKeys[pubKey] = publicKey
res.secretKeys[pubKey] = sk
}
return res
}
// FetchValidatingKeys fetches the list of public keys that should be used to validate with.
func (km *Direct) FetchValidatingKeys() ([][48]byte, error) {
keys := make([][48]byte, 0, len(km.publicKeys))
for key := range km.publicKeys {
keys = append(keys, key)
}
return keys, nil
}
// Sign signs a message for the validator to broadcast.
func (km *Direct) Sign(pubKey [48]byte, root [32]byte, domain uint64) (*bls.Signature, error) {
if secretKey, exists := km.secretKeys[pubKey]; exists {
return secretKey.Sign(root[:], domain), nil
}
return nil, ErrNoSuchKey
}