/
keys.go
59 lines (53 loc) · 1.68 KB
/
keys.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
47
48
49
50
51
52
53
54
55
56
57
58
59
package utils
import (
"crypto/ecdsa"
"crypto/rand"
"encoding/hex"
"github.com/bloxapp/ssv/logging/fields"
"github.com/bloxapp/ssv/network/commons"
"github.com/libp2p/go-libp2p/core/crypto"
"github.com/pkg/errors"
"go.uber.org/zap"
)
// ECDSAPrivateKey extracts the ecdsa.PrivateKey from the given string or generate a new key
func ECDSAPrivateKey(logger *zap.Logger, privateKey string) (*ecdsa.PrivateKey, error) {
var privKey *ecdsa.PrivateKey
if privateKey != "" {
dst, err := hex.DecodeString(privateKey)
if err != nil {
return nil, errors.WithMessage(err, "failed to decode privKey string")
}
unmarshalledKey, err := crypto.UnmarshalSecp256k1PrivateKey(dst)
if err != nil {
return nil, errors.WithMessage(err, "failed to unmarshal passed privKey")
}
privKey, err = commons.ECDSAPrivFromInterface(unmarshalledKey)
if err != nil {
return nil, err
}
} else {
logger.Info("No private key was provided. Generating a new one...")
privInterfaceKey, _, err := crypto.GenerateSecp256k1Key(rand.Reader)
if err != nil {
return nil, errors.WithMessage(err, "failed to generate 256k1 key")
}
privKey, err = commons.ECDSAPrivFromInterface(privInterfaceKey)
if err != nil {
return nil, err
}
}
interfacePriv, err := commons.ECDSAPrivToInterface(privKey)
if err != nil {
return nil, err
}
b, err := interfacePriv.Raw()
if err != nil {
return nil, errors.WithMessage(err, "failed to convert private key to interface")
}
if privateKey != "" {
logger.Debug("Using Private Key from config", fields.PrivKey(b), zap.Any("private-key", b))
} else {
logger.Debug("Private Key generated", fields.PrivKey(b), zap.Any("private-key", b))
}
return privKey, nil
}