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.
67 lines
1.8 KiB
67 lines
1.8 KiB
package sr25519
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/crypto"
|
|
"github.com/tendermint/tendermint/crypto/internal/benchmarking"
|
|
)
|
|
|
|
func BenchmarkKeyGeneration(b *testing.B) {
|
|
benchmarkKeygenWrapper := func(reader io.Reader) crypto.PrivKey {
|
|
return genPrivKey(reader)
|
|
}
|
|
benchmarking.BenchmarkKeyGeneration(b, benchmarkKeygenWrapper)
|
|
}
|
|
|
|
func BenchmarkSigning(b *testing.B) {
|
|
priv := GenPrivKey()
|
|
benchmarking.BenchmarkSigning(b, priv)
|
|
}
|
|
|
|
func BenchmarkVerification(b *testing.B) {
|
|
priv := GenPrivKey()
|
|
benchmarking.BenchmarkVerification(b, priv)
|
|
}
|
|
|
|
func BenchmarkVerifyBatch(b *testing.B) {
|
|
msg := []byte("BatchVerifyTest")
|
|
|
|
for _, sigsCount := range []int{1, 8, 64, 1024} {
|
|
sigsCount := sigsCount
|
|
b.Run(fmt.Sprintf("sig-count-%d", sigsCount), func(b *testing.B) {
|
|
// Pre-generate all of the keys, and signatures, but do not
|
|
// benchmark key-generation and signing.
|
|
pubs := make([]crypto.PubKey, 0, sigsCount)
|
|
sigs := make([][]byte, 0, sigsCount)
|
|
for i := 0; i < sigsCount; i++ {
|
|
priv := GenPrivKey()
|
|
sig, _ := priv.Sign(msg)
|
|
pubs = append(pubs, priv.PubKey().(PubKey))
|
|
sigs = append(sigs, sig)
|
|
}
|
|
b.ResetTimer()
|
|
|
|
b.ReportAllocs()
|
|
// NOTE: dividing by n so that metrics are per-signature
|
|
for i := 0; i < b.N/sigsCount; i++ {
|
|
// The benchmark could just benchmark the Verify()
|
|
// routine, but there is non-trivial overhead associated
|
|
// with BatchVerifier.Add(), which should be included
|
|
// in the benchmark.
|
|
v := NewBatchVerifier()
|
|
for i := 0; i < sigsCount; i++ {
|
|
err := v.Add(pubs[i], msg, sigs[i])
|
|
require.NoError(b, err)
|
|
}
|
|
|
|
if ok, _ := v.Verify(); !ok {
|
|
b.Fatal("signature set failed batch verification")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|