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.
90 lines
2.6 KiB
90 lines
2.6 KiB
package pki
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/binary"
|
|
|
|
"github.com/dedis/kyber"
|
|
"github.com/harmony-one/bls/ffi/go/bls"
|
|
"github.com/harmony-one/harmony/crypto"
|
|
)
|
|
|
|
func init() {
|
|
bls.Init(bls.BLS12_381)
|
|
}
|
|
|
|
// GetAddressFromPublicKey returns address given a public key.
|
|
func GetAddressFromPublicKey(pubKey *bls.PublicKey) [20]byte {
|
|
bytes := pubKey.Serialize()
|
|
address := [20]byte{}
|
|
hash := sha256.Sum256(bytes)
|
|
copy(address[:], hash[12:])
|
|
return address
|
|
}
|
|
|
|
// GetAddressFromPrivateKey returns address given a private key.
|
|
func GetAddressFromPrivateKey(priKey *bls.SecretKey) [20]byte {
|
|
return GetAddressFromPublicKey(priKey.GetPublicKey())
|
|
}
|
|
|
|
// GetAddressFromPrivateKeyBytes returns address from private key in bytes.
|
|
func GetAddressFromPrivateKeyBytes(priKey [32]byte) [20]byte {
|
|
var privateKey bls.SecretKey
|
|
privateKey.SetLittleEndian(priKey[:])
|
|
|
|
return GetAddressFromPublicKey(privateKey.GetPublicKey())
|
|
}
|
|
|
|
// GetAddressFromInt is the temporary helper function for benchmark use
|
|
func GetAddressFromInt(value int) [20]byte {
|
|
priKey := [32]byte{}
|
|
binary.LittleEndian.PutUint32(priKey[:], uint32(value))
|
|
return GetAddressFromPrivateKeyBytes(priKey)
|
|
}
|
|
|
|
// GetPrivateKeyScalarFromInt return private key scalar.
|
|
func GetPrivateKeyScalarFromInt(value int) kyber.Scalar {
|
|
return crypto.Ed25519Curve.Scalar().SetInt64(int64(value))
|
|
}
|
|
|
|
// GetBLSPrivateKeyFromInt returns bls private key
|
|
func GetBLSPrivateKeyFromInt(value int) *bls.SecretKey {
|
|
priKey := [32]byte{}
|
|
binary.LittleEndian.PutUint32(priKey[:], uint32(value))
|
|
var privateKey bls.SecretKey
|
|
privateKey.SetLittleEndian(priKey[:])
|
|
return &privateKey
|
|
}
|
|
|
|
// GetPrivateKeyFromInt returns private key in bytes given an interger.
|
|
func GetPrivateKeyFromInt(value int) [32]byte {
|
|
priKey, err := crypto.Ed25519Curve.Scalar().SetInt64(int64(value)).MarshalBinary()
|
|
priKeyBytes := [32]byte{}
|
|
if err == nil {
|
|
copy(priKeyBytes[:], priKey[:])
|
|
}
|
|
return priKeyBytes
|
|
}
|
|
|
|
// GetPublicKeyFromPrivateKey return public key from private key.
|
|
func GetPublicKeyFromPrivateKey(priKey [32]byte) kyber.Point {
|
|
suite := crypto.Ed25519Curve
|
|
scalar := suite.Scalar()
|
|
scalar.UnmarshalBinary(priKey[:])
|
|
return suite.Point().Mul(scalar, nil)
|
|
}
|
|
|
|
// GetPublicKeyFromScalar is the same as GetPublicKeyFromPrivateKey, but it directly works on kyber.Scalar object.
|
|
func GetPublicKeyFromScalar(priKey kyber.Scalar) kyber.Point {
|
|
return crypto.Ed25519Curve.Point().Mul(priKey, nil)
|
|
}
|
|
|
|
// GetBytesFromPublicKey converts public key point to bytes
|
|
func GetBytesFromPublicKey(pubKey kyber.Point) [32]byte {
|
|
bytes, err := pubKey.MarshalBinary()
|
|
result := [32]byte{}
|
|
if err == nil {
|
|
copy(result[:], bytes)
|
|
}
|
|
return result
|
|
}
|
|
|