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.
52 lines
1.8 KiB
52 lines
1.8 KiB
package chain
|
|
|
|
import (
|
|
"errors"
|
|
|
|
bls_core "github.com/harmony-one/bls/ffi/go/bls"
|
|
|
|
"github.com/harmony-one/harmony/crypto/bls"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
)
|
|
|
|
// ReadSignatureBitmapByPublicKeys read the payload of signature and bitmap based on public keys
|
|
func ReadSignatureBitmapByPublicKeys(recvPayload []byte, publicKeys []bls.PublicKeyWrapper) (*bls_core.Sign, *bls.Mask, error) {
|
|
sig, bitmap, err := ParseCommitSigAndBitmap(recvPayload)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return DecodeSigBitmap(sig, bitmap, publicKeys)
|
|
}
|
|
|
|
// ParseCommitSigAndBitmap parse the commitSigAndBitmap to signature + bitmap
|
|
func ParseCommitSigAndBitmap(payload []byte) (bls.SerializedSignature, []byte, error) {
|
|
if len(payload) < bls.BLSSignatureSizeInBytes {
|
|
return bls.SerializedSignature{}, nil, errors.New("payload not have enough length")
|
|
}
|
|
var (
|
|
sig bls.SerializedSignature
|
|
bitmap = make([]byte, len(payload)-bls.BLSSignatureSizeInBytes)
|
|
)
|
|
copy(sig[:], payload[:bls.BLSSignatureSizeInBytes])
|
|
copy(bitmap, payload[bls.BLSSignatureSizeInBytes:])
|
|
|
|
return sig, bitmap, nil
|
|
}
|
|
|
|
// DecodeSigBitmap decode and parse the signature, bitmap with the given public keys
|
|
func DecodeSigBitmap(sigBytes bls.SerializedSignature, bitmap []byte, pubKeys []bls.PublicKeyWrapper) (*bls_core.Sign, *bls.Mask, error) {
|
|
aggSig := bls_core.Sign{}
|
|
err := aggSig.Deserialize(sigBytes[:])
|
|
if err != nil {
|
|
return nil, nil, errors.New("unable to deserialize multi-signature from payload")
|
|
}
|
|
mask, err := bls.NewMask(pubKeys, nil)
|
|
if err != nil {
|
|
utils.Logger().Warn().Err(err).Msg("onNewView unable to setup mask for prepared message")
|
|
return nil, nil, errors.New("unable to setup mask from payload")
|
|
}
|
|
if err := mask.SetMask(bitmap); err != nil {
|
|
utils.Logger().Warn().Err(err).Msg("mask.SetMask failed")
|
|
}
|
|
return &aggSig, mask, nil
|
|
}
|
|
|