You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
211 lines
5.0 KiB
211 lines
5.0 KiB
package crypto
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
secp256k1 "github.com/btcsuite/btcd/btcec"
|
|
"github.com/tendermint/ed25519"
|
|
"github.com/tendermint/ed25519/extra25519"
|
|
. "github.com/tendermint/go-common"
|
|
data "github.com/tendermint/go-data"
|
|
"github.com/tendermint/go-wire"
|
|
"golang.org/x/crypto/ripemd160"
|
|
)
|
|
|
|
// PubKey is part of Account and Validator.
|
|
type PubKey interface {
|
|
Address() []byte
|
|
Bytes() []byte
|
|
KeyString() string
|
|
VerifyBytes(msg []byte, sig Signature) bool
|
|
Equals(PubKey) bool
|
|
}
|
|
|
|
// Types of PubKey implementations
|
|
const (
|
|
PubKeyTypeEd25519 = byte(0x01)
|
|
PubKeyTypeSecp256k1 = byte(0x02)
|
|
PubKeyNameEd25519 = "ed25519"
|
|
PubKeyNameSecp256k1 = "secp256k1"
|
|
)
|
|
|
|
var pubKeyMapper data.Mapper
|
|
|
|
// register both public key types with go-data (and thus go-wire)
|
|
func init() {
|
|
pubKeyMapper = data.NewMapper(PubKeyS{}).
|
|
RegisterInterface(PubKeyEd25519{}, PubKeyNameEd25519, PubKeyTypeEd25519).
|
|
RegisterInterface(PubKeySecp256k1{}, PubKeyNameSecp256k1, PubKeyTypeSecp256k1)
|
|
}
|
|
|
|
// PubKeyS add json serialization to PubKey
|
|
type PubKeyS struct {
|
|
PubKey
|
|
}
|
|
|
|
func (p PubKeyS) MarshalJSON() ([]byte, error) {
|
|
return pubKeyMapper.ToJSON(p.PubKey)
|
|
}
|
|
|
|
func (p *PubKeyS) UnmarshalJSON(data []byte) (err error) {
|
|
parsed, err := pubKeyMapper.FromJSON(data)
|
|
if err == nil {
|
|
p.PubKey = parsed.(PubKey)
|
|
}
|
|
return
|
|
}
|
|
|
|
func PubKeyFromBytes(pubKeyBytes []byte) (pubKey PubKey, err error) {
|
|
err = wire.ReadBinaryBytes(pubKeyBytes, &pubKey)
|
|
return
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
// Implements PubKey
|
|
type PubKeyEd25519 [32]byte
|
|
|
|
func (pubKey PubKeyEd25519) Address() []byte {
|
|
w, n, err := new(bytes.Buffer), new(int), new(error)
|
|
wire.WriteBinary(pubKey[:], w, n, err)
|
|
if *err != nil {
|
|
PanicCrisis(*err)
|
|
}
|
|
// append type byte
|
|
encodedPubkey := append([]byte{PubKeyTypeEd25519}, w.Bytes()...)
|
|
hasher := ripemd160.New()
|
|
hasher.Write(encodedPubkey) // does not error
|
|
return hasher.Sum(nil)
|
|
}
|
|
|
|
func (pubKey PubKeyEd25519) Bytes() []byte {
|
|
return wire.BinaryBytes(struct{ PubKey }{pubKey})
|
|
}
|
|
|
|
func (pubKey PubKeyEd25519) VerifyBytes(msg []byte, sig_ Signature) bool {
|
|
// unwrap if needed
|
|
if wrap, ok := sig_.(SignatureS); ok {
|
|
sig_ = wrap.Signature
|
|
}
|
|
// make sure we use the same algorithm to sign
|
|
sig, ok := sig_.(SignatureEd25519)
|
|
if !ok {
|
|
return false
|
|
}
|
|
pubKeyBytes := [32]byte(pubKey)
|
|
sigBytes := [64]byte(sig)
|
|
return ed25519.Verify(&pubKeyBytes, msg, &sigBytes)
|
|
}
|
|
|
|
func (p PubKeyEd25519) MarshalJSON() ([]byte, error) {
|
|
return data.Encoder.Marshal(p[:])
|
|
}
|
|
|
|
func (p *PubKeyEd25519) UnmarshalJSON(enc []byte) error {
|
|
var ref []byte
|
|
err := data.Encoder.Unmarshal(&ref, enc)
|
|
copy(p[:], ref)
|
|
return err
|
|
}
|
|
|
|
// For use with golang/crypto/nacl/box
|
|
// If error, returns nil.
|
|
func (pubKey PubKeyEd25519) ToCurve25519() *[32]byte {
|
|
keyCurve25519, pubKeyBytes := new([32]byte), [32]byte(pubKey)
|
|
ok := extra25519.PublicKeyToCurve25519(keyCurve25519, &pubKeyBytes)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
return keyCurve25519
|
|
}
|
|
|
|
func (pubKey PubKeyEd25519) String() string {
|
|
return Fmt("PubKeyEd25519{%X}", pubKey[:])
|
|
}
|
|
|
|
// Must return the full bytes in hex.
|
|
// Used for map keying, etc.
|
|
func (pubKey PubKeyEd25519) KeyString() string {
|
|
return Fmt("%X", pubKey[:])
|
|
}
|
|
|
|
func (pubKey PubKeyEd25519) Equals(other PubKey) bool {
|
|
if otherEd, ok := other.(PubKeyEd25519); ok {
|
|
return bytes.Equal(pubKey[:], otherEd[:])
|
|
} else {
|
|
return false
|
|
}
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
// Implements PubKey
|
|
type PubKeySecp256k1 [64]byte
|
|
|
|
func (pubKey PubKeySecp256k1) Address() []byte {
|
|
w, n, err := new(bytes.Buffer), new(int), new(error)
|
|
wire.WriteBinary(pubKey[:], w, n, err)
|
|
if *err != nil {
|
|
PanicCrisis(*err)
|
|
}
|
|
// append type byte
|
|
encodedPubkey := append([]byte{PubKeyTypeSecp256k1}, w.Bytes()...)
|
|
hasher := ripemd160.New()
|
|
hasher.Write(encodedPubkey) // does not error
|
|
return hasher.Sum(nil)
|
|
}
|
|
|
|
func (pubKey PubKeySecp256k1) Bytes() []byte {
|
|
return wire.BinaryBytes(struct{ PubKey }{pubKey})
|
|
}
|
|
|
|
func (pubKey PubKeySecp256k1) VerifyBytes(msg []byte, sig_ Signature) bool {
|
|
// unwrap if needed
|
|
if wrap, ok := sig_.(SignatureS); ok {
|
|
sig_ = wrap.Signature
|
|
}
|
|
// and assert same algorithm to sign and verify
|
|
sig, ok := sig_.(SignatureSecp256k1)
|
|
if !ok {
|
|
return false
|
|
}
|
|
|
|
pub__, err := secp256k1.ParsePubKey(append([]byte{0x04}, pubKey[:]...), secp256k1.S256())
|
|
if err != nil {
|
|
return false
|
|
}
|
|
sig__, err := secp256k1.ParseDERSignature(sig[:], secp256k1.S256())
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return sig__.Verify(Sha256(msg), pub__)
|
|
}
|
|
|
|
func (p PubKeySecp256k1) MarshalJSON() ([]byte, error) {
|
|
return data.Encoder.Marshal(p[:])
|
|
}
|
|
|
|
func (p *PubKeySecp256k1) UnmarshalJSON(enc []byte) error {
|
|
var ref []byte
|
|
err := data.Encoder.Unmarshal(&ref, enc)
|
|
copy(p[:], ref)
|
|
return err
|
|
}
|
|
|
|
func (pubKey PubKeySecp256k1) String() string {
|
|
return Fmt("PubKeySecp256k1{%X}", pubKey[:])
|
|
}
|
|
|
|
// Must return the full bytes in hex.
|
|
// Used for map keying, etc.
|
|
func (pubKey PubKeySecp256k1) KeyString() string {
|
|
return Fmt("%X", pubKey[:])
|
|
}
|
|
|
|
func (pubKey PubKeySecp256k1) Equals(other PubKey) bool {
|
|
if otherSecp, ok := other.(PubKeySecp256k1); ok {
|
|
return bytes.Equal(pubKey[:], otherSecp[:])
|
|
} else {
|
|
return false
|
|
}
|
|
}
|