2021-09-17 01:32:23 +00:00
|
|
|
package bls
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/hex"
|
|
|
|
"errors"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/ghodss/yaml"
|
2023-03-17 18:52:56 +00:00
|
|
|
"github.com/prysmaticlabs/prysm/v4/crypto/bls"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/crypto/bls/common"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/testing/bls/utils"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
2021-09-17 01:32:23 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestBatchVerify(t *testing.T) {
|
|
|
|
t.Run("blst", testBatchVerify)
|
|
|
|
}
|
|
|
|
|
|
|
|
func testBatchVerify(t *testing.T) {
|
|
|
|
fNames, fContent := utils.RetrieveFiles("batch_verify", t)
|
|
|
|
|
|
|
|
for i, file := range fNames {
|
|
|
|
t.Run(file, func(t *testing.T) {
|
|
|
|
test := &BatchVerifyTest{}
|
|
|
|
require.NoError(t, yaml.Unmarshal(fContent[i], test))
|
|
|
|
|
|
|
|
pubkeys := make([]common.PublicKey, len(test.Input.Pubkeys))
|
|
|
|
messages := make([][32]byte, len(test.Input.Messages))
|
|
|
|
signatures := make([][]byte, len(test.Input.Signatures))
|
|
|
|
for j, raw := range test.Input.Pubkeys {
|
|
|
|
pkBytes, err := hex.DecodeString(raw[2:])
|
|
|
|
require.NoError(t, err)
|
|
|
|
pk, err := bls.PublicKeyFromBytes(pkBytes)
|
|
|
|
if err != nil {
|
|
|
|
if test.Output == false && errors.Is(err, common.ErrInfinitePubKey) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
t.Fatalf("cannot unmarshal pubkey: %v", err)
|
|
|
|
}
|
|
|
|
pubkeys[j] = pk
|
|
|
|
}
|
|
|
|
for j, raw := range test.Input.Messages {
|
|
|
|
msgBytes, err := hex.DecodeString(raw[2:])
|
|
|
|
require.NoError(t, err)
|
|
|
|
messages[j] = bytesutil.ToBytes32(msgBytes)
|
|
|
|
}
|
|
|
|
for j, raw := range test.Input.Signatures {
|
|
|
|
sigBytes, err := hex.DecodeString(raw[2:])
|
|
|
|
require.NoError(t, err)
|
|
|
|
signatures[j] = sigBytes
|
|
|
|
}
|
|
|
|
|
|
|
|
verified, err := bls.VerifyMultipleSignatures(signatures, messages, pubkeys)
|
|
|
|
require.NoError(t, err)
|
|
|
|
if verified != test.Output {
|
|
|
|
t.Fatalf("Signature does not match the expected verification output. "+
|
|
|
|
"Expected %#v but received %#v for test case %d", test.Output, verified, i)
|
|
|
|
}
|
|
|
|
t.Log("Success")
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|