mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-11-11 02:35:18 +00:00
82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
|
//
|
||
|
|
// Copyright Coinbase, Inc. All Rights Reserved.
|
||
|
|
//
|
||
|
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
|
//
|
||
|
|
|
||
|
|
package bbs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/core/curves"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSignatureWorks(t *testing.T) {
|
||
|
|
curve := curves.BLS12381(&curves.PointBls12381G2{})
|
||
|
|
msgs := []curves.Scalar{
|
||
|
|
curve.Scalar.New(3),
|
||
|
|
curve.Scalar.New(4),
|
||
|
|
curve.Scalar.New(5),
|
||
|
|
curve.Scalar.New(6),
|
||
|
|
}
|
||
|
|
pk, sk, err := NewKeys(curve)
|
||
|
|
require.NoError(t, err)
|
||
|
|
generators, err := new(MessageGenerators).Init(pk, 4)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
sig, err := sk.Sign(generators, msgs)
|
||
|
|
require.NoError(t, err)
|
||
|
|
err = pk.Verify(sig, generators, msgs)
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSignatureIncorrectMessages(t *testing.T) {
|
||
|
|
curve := curves.BLS12381(&curves.PointBls12381G2{})
|
||
|
|
msgs := []curves.Scalar{
|
||
|
|
curve.Scalar.New(3),
|
||
|
|
curve.Scalar.New(4),
|
||
|
|
curve.Scalar.New(5),
|
||
|
|
curve.Scalar.New(6),
|
||
|
|
}
|
||
|
|
pk, sk, err := NewKeys(curve)
|
||
|
|
require.NoError(t, err)
|
||
|
|
generators, err := new(MessageGenerators).Init(pk, 4)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
sig, err := sk.Sign(generators, msgs)
|
||
|
|
require.NoError(t, err)
|
||
|
|
msgs[0] = curve.Scalar.New(7)
|
||
|
|
err = pk.Verify(sig, generators, msgs)
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSignatureMarshalBinary(t *testing.T) {
|
||
|
|
curve := curves.BLS12381(&curves.PointBls12381G2{})
|
||
|
|
msgs := []curves.Scalar{
|
||
|
|
curve.Scalar.New(3),
|
||
|
|
curve.Scalar.New(4),
|
||
|
|
curve.Scalar.New(5),
|
||
|
|
curve.Scalar.New(6),
|
||
|
|
}
|
||
|
|
pk, sk, err := NewKeys(curve)
|
||
|
|
require.NoError(t, err)
|
||
|
|
generators, err := new(MessageGenerators).Init(pk, 4)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
sig, err := sk.Sign(generators, msgs)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
data, err := sig.MarshalBinary()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, 112, len(data))
|
||
|
|
sig2 := new(Signature).Init(curve)
|
||
|
|
err = sig2.UnmarshalBinary(data)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, sig.a.Equal(sig2.a))
|
||
|
|
require.Equal(t, sig.e.Cmp(sig2.e), 0)
|
||
|
|
require.Equal(t, sig.s.Cmp(sig2.s), 0)
|
||
|
|
}
|