ceremonyclient/nekryptology/pkg/tecdsa/dkls/v1/sign/sign_test.go

82 lines
2.9 KiB
Go
Raw Normal View History

2023-07-05 05:32:28 +00:00
//
// Copyright Coinbase, Inc. All Rights Reserved.
// Copyright Quilibrium, Inc. All Rights Reserved.
//
// SPDX-License-Identifier: Apache-2.0
//
package sign
import (
"crypto/rand"
"testing"
"github.com/stretchr/testify/require"
"golang.org/x/crypto/sha3"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/core/curves"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/ot/base/simplest"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/ot/ottest"
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/tecdsa/dkls/v1/dkg"
)
func TestSign(t *testing.T) {
curveInstances := []*curves.Curve{
curves.K256(),
curves.P256(),
}
for _, curve := range curveInstances {
hashKeySeed := [simplest.DigestSize]byte{}
_, err := rand.Read(hashKeySeed[:])
require.NoError(t, err)
baseOtSenderOutput, baseOtReceiverOutput, err := ottest.RunSimplestOT(curve, 256, hashKeySeed)
require.NoError(t, err)
secretKeyShareA := curve.Scalar.Random(rand.Reader)
secretKeyShareB := curve.Scalar.Random(rand.Reader)
require.NoError(t, err)
publicKey := curve.ScalarBaseMult(secretKeyShareA.Mul(secretKeyShareB))
alice := NewAlice(256, 80, curve, sha3.New256(), &dkg.AliceOutput{SeedOtResult: baseOtReceiverOutput, SecretKeyShare: secretKeyShareA, PublicKey: publicKey})
bob := NewBob(256, 80, curve, sha3.New256(), &dkg.BobOutput{SeedOtResult: baseOtSenderOutput, SecretKeyShare: secretKeyShareB, PublicKey: publicKey})
message := []byte("A message.")
seed, err := alice.Round1GenerateRandomSeed()
require.NoError(t, err)
round3Output, err := bob.Round2Initialize(seed)
require.NoError(t, err)
round4Output, err := alice.Round3Sign(message, round3Output)
require.NoError(t, err)
err = bob.Round4Final(message, round4Output)
require.NoError(t, err, "curve: %s", curve.Name)
}
}
func BenchmarkSign(b *testing.B) {
curve := curves.K256()
hashKeySeed := [simplest.DigestSize]byte{}
_, err := rand.Read(hashKeySeed[:])
require.NoError(b, err)
baseOtSenderOutput, baseOtReceiverOutput, err := ottest.RunSimplestOT(curve, 256, hashKeySeed)
require.NoError(b, err)
secretKeyShareA := curve.Scalar.Random(rand.Reader)
secretKeyShareB := curve.Scalar.Random(rand.Reader)
publicKey := curve.ScalarBaseMult(secretKeyShareA.Mul(secretKeyShareB))
alice := NewAlice(256, 80, curve, sha3.New256(), &dkg.AliceOutput{SeedOtResult: baseOtReceiverOutput, SecretKeyShare: secretKeyShareA, PublicKey: publicKey})
bob := NewBob(256, 80, curve, sha3.New256(), &dkg.BobOutput{SeedOtResult: baseOtSenderOutput, SecretKeyShare: secretKeyShareB, PublicKey: publicKey})
message := []byte("A message.")
for n := 0; n < b.N; n++ {
seed, err := alice.Round1GenerateRandomSeed()
require.NoError(b, err)
round3Output, err := bob.Round2Initialize(seed)
require.NoError(b, err)
round4Output, err := alice.Round3Sign(message, round3Output)
require.NoError(b, err)
err = bob.Round4Final(message, round4Output)
require.NoError(b, err)
}
}