mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-12-30 18:35:18 +00:00
82 lines
2.9 KiB
Go
82 lines
2.9 KiB
Go
|
//
|
||
|
|
// 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)
|
||
|
|
}
|
||
|
|
}
|