lotus/chain/wallet/key.go

75 lines
1.5 KiB
Go
Raw Normal View History

2020-09-04 20:17:28 +00:00
package wallet
import (
"golang.org/x/xerrors"
"github.com/filecoin-project/go-address"
2020-10-08 22:51:04 +00:00
"github.com/filecoin-project/go-state-types/crypto"
2020-09-04 20:17:28 +00:00
"github.com/filecoin-project/lotus/chain/types"
"github.com/filecoin-project/lotus/lib/sigs"
)
func GenerateKey(typ types.KeyType) (*Key, error) {
ctyp := ActSigType(typ)
if ctyp == crypto.SigTypeUnknown {
return nil, xerrors.Errorf("unknown sig type: %s", typ)
}
pk, err := sigs.Generate(ctyp)
2020-09-04 20:17:28 +00:00
if err != nil {
return nil, err
}
ki := types.KeyInfo{
Type: typ,
2020-09-04 20:17:28 +00:00
PrivateKey: pk,
}
return NewKey(ki)
}
type Key struct {
types.KeyInfo
PublicKey []byte
Address address.Address
}
func NewKey(keyinfo types.KeyInfo) (*Key, error) {
k := &Key{
KeyInfo: keyinfo,
}
var err error
k.PublicKey, err = sigs.ToPublic(ActSigType(k.Type), k.PrivateKey)
if err != nil {
return nil, err
}
switch k.Type {
case types.KTSecp256k1:
2020-09-04 20:17:28 +00:00
k.Address, err = address.NewSecp256k1Address(k.PublicKey)
if err != nil {
return nil, xerrors.Errorf("converting Secp256k1 to address: %w", err)
}
case types.KTBLS:
2020-09-04 20:17:28 +00:00
k.Address, err = address.NewBLSAddress(k.PublicKey)
if err != nil {
return nil, xerrors.Errorf("converting BLS to address: %w", err)
}
default:
return nil, xerrors.Errorf("unsupported key type: %s", k.Type)
2020-09-04 20:17:28 +00:00
}
return k, nil
}
func ActSigType(typ types.KeyType) crypto.SigType {
2020-09-04 20:17:28 +00:00
switch typ {
case types.KTBLS:
2020-09-04 20:17:28 +00:00
return crypto.SigTypeBLS
case types.KTSecp256k1:
2020-09-04 20:17:28 +00:00
return crypto.SigTypeSecp256k1
default:
return crypto.SigTypeUnknown
2020-09-04 20:17:28 +00:00
}
}