mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-03 16:37:39 +00:00
5a66807989
* First take at updating everything to v5 * Patch gRPC gateway to use prysm v5 Fix patch * Update go ssz --------- Co-authored-by: Preston Van Loon <pvanloon@offchainlabs.com>
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
//go:build ((linux && amd64) || (linux && arm64) || (darwin && amd64) || (darwin && arm64) || (windows && amd64)) && !blst_disabled
|
|
|
|
package blst_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v5/crypto/bls/blst"
|
|
"github.com/prysmaticlabs/prysm/v5/crypto/bls/common"
|
|
"github.com/prysmaticlabs/prysm/v5/testing/require"
|
|
)
|
|
|
|
func BenchmarkSignature_Verify(b *testing.B) {
|
|
sk, err := blst.RandKey()
|
|
require.NoError(b, err)
|
|
|
|
msg := []byte("Some msg")
|
|
sig := sk.Sign(msg)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
if !sig.Verify(sk.PublicKey(), msg) {
|
|
b.Fatal("could not verify sig")
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkSignature_AggregateVerify(b *testing.B) {
|
|
sigN := 128 // MAX_ATTESTATIONS per block.
|
|
|
|
var pks []common.PublicKey
|
|
var sigs []common.Signature
|
|
var msgs [][32]byte
|
|
for i := 0; i < sigN; i++ {
|
|
msg := [32]byte{'s', 'i', 'g', 'n', 'e', 'd', byte(i)}
|
|
sk, err := blst.RandKey()
|
|
require.NoError(b, err)
|
|
sig := sk.Sign(msg[:])
|
|
pks = append(pks, sk.PublicKey())
|
|
sigs = append(sigs, sig)
|
|
msgs = append(msgs, msg)
|
|
}
|
|
aggregated := blst.AggregateSignatures(sigs)
|
|
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
if !aggregated.AggregateVerify(pks, msgs) {
|
|
b.Fatal("could not verify aggregate sig")
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkSecretKey_Marshal(b *testing.B) {
|
|
key, err := blst.RandKey()
|
|
require.NoError(b, err)
|
|
d := key.Marshal()
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := blst.SecretKeyFromBytes(d)
|
|
_ = err
|
|
}
|
|
}
|