woop/crypto/pki/utils.go

package pki

import (
	"crypto/sha256"
	"github.com/dedis/kyber"
	"github.com/simple-rules/harmony-benchmark/crypto"
	"github.com/simple-rules/harmony-benchmark/log"
)

func GetAddressFromPublicKey(pubKey kyber.Point) [20]byte {
	bytes, err := pubKey.MarshalBinary()
	if err != nil {
		log.Error("Failed to serialize challenge")
	}
	address := [20]byte{}
	hash := sha256.Sum256(bytes)
	copy(address[:], hash[12:])
	return address
}

// Temporary helper function for benchmark use
func GetAddressFromInt(value int) [20]byte {
	return GetAddressFromPublicKey(GetPublicKeyFromScalar(crypto.Ed25519Curve, GetPrivateKeyFromInt(value)))
}

func GetPrivateKeyFromInt(value int) kyber.Scalar {
	return crypto.Ed25519Curve.Scalar().SetInt64(int64(value))
}

func GetPublicKeyFromPrivateKey(suite crypto.Suite, priKey [32]byte) kyber.Point {
	scalar := suite.Scalar()
	scalar.UnmarshalBinary(priKey[:])
	return suite.Point().Mul(scalar, nil)
}

// Same as GetPublicKeyFromPrivateKey, but it directly works on kyber.Scalar object.
func GetPublicKeyFromScalar(suite crypto.Suite, priKey kyber.Scalar) kyber.Point {
	return suite.Point().Mul(priKey, nil)
}