forked from TrueCloudLab/neoneo-go
113 lines
3.1 KiB
Go
113 lines
3.1 KiB
Go
package keys
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/btcec"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIssue1223(t *testing.T) {
|
|
var d, x, y big.Int
|
|
d.SetString("75066030006596498716801752450216843918658392116070031536027203512060270094427", 10)
|
|
x.SetString("56810139335762307690884151098712528235297095596167964448512639328424930082240", 10)
|
|
y.SetString("108055740278314806025442297642651169427004858252141003070998851291610422839293", 10)
|
|
|
|
privateKey := &btcec.PrivateKey{
|
|
PublicKey: ecdsa.PublicKey{
|
|
Curve: btcec.S256(),
|
|
X: &x,
|
|
Y: &y,
|
|
},
|
|
D: &d,
|
|
}
|
|
pubKey := PublicKey(ecdsa.PublicKey{
|
|
Curve: btcec.S256(),
|
|
X: privateKey.X,
|
|
Y: privateKey.Y,
|
|
})
|
|
|
|
hashedData := hash.Sha256([]byte("sample"))
|
|
signature, err := privateKey.Sign(hashedData.BytesBE())
|
|
require.NoError(t, err)
|
|
|
|
signedData := getSignatureSlice(privateKey.Curve, signature.R, signature.S)
|
|
require.True(t, pubKey.Verify(signedData, hashedData.BytesBE()))
|
|
}
|
|
|
|
func TestPubKeyVerify(t *testing.T) {
|
|
var data = []byte("sample")
|
|
hashedData := hash.Sha256(data)
|
|
|
|
t.Run("Secp256r1", func(t *testing.T) {
|
|
privKey, err := NewPrivateKey()
|
|
assert.Nil(t, err)
|
|
signedData := privKey.Sign(data)
|
|
pubKey := privKey.PublicKey()
|
|
result := pubKey.Verify(signedData, hashedData.BytesBE())
|
|
expected := true
|
|
assert.Equal(t, expected, result)
|
|
|
|
// Small signature, no panic.
|
|
assert.False(t, pubKey.Verify([]byte{1, 2, 3}, hashedData.BytesBE()))
|
|
|
|
pubKey = &PublicKey{}
|
|
assert.False(t, pubKey.Verify(signedData, hashedData.BytesBE()))
|
|
})
|
|
|
|
t.Run("Secp256k1", func(t *testing.T) {
|
|
privateKey, err := btcec.NewPrivateKey(btcec.S256())
|
|
assert.Nil(t, err)
|
|
signature, err := privateKey.Sign(hashedData.BytesBE())
|
|
require.NoError(t, err)
|
|
signedData := getSignatureSlice(privateKey.Curve, signature.R, signature.S)
|
|
pubKey := PublicKey(ecdsa.PublicKey{
|
|
Curve: btcec.S256(),
|
|
X: privateKey.X,
|
|
Y: privateKey.Y,
|
|
})
|
|
require.True(t, pubKey.Verify(signedData, hashedData.BytesBE()))
|
|
|
|
pubKey = PublicKey{}
|
|
assert.False(t, pubKey.Verify(signedData, hashedData.BytesBE()))
|
|
})
|
|
}
|
|
|
|
func TestWrongPubKey(t *testing.T) {
|
|
sample := []byte("sample")
|
|
hashedData := hash.Sha256(sample)
|
|
|
|
t.Run("Secp256r1", func(t *testing.T) {
|
|
privKey, _ := NewPrivateKey()
|
|
signedData := privKey.Sign(sample)
|
|
|
|
secondPrivKey, _ := NewPrivateKey()
|
|
wrongPubKey := secondPrivKey.PublicKey()
|
|
|
|
actual := wrongPubKey.Verify(signedData, hashedData.BytesBE())
|
|
expcted := false
|
|
assert.Equal(t, expcted, actual)
|
|
})
|
|
|
|
t.Run("Secp256k1", func(t *testing.T) {
|
|
privateKey, err := btcec.NewPrivateKey(btcec.S256())
|
|
assert.Nil(t, err)
|
|
signature, err := privateKey.Sign(hashedData.BytesBE())
|
|
assert.Nil(t, err)
|
|
signedData := getSignatureSlice(privateKey.Curve, signature.R, signature.S)
|
|
|
|
secondPrivKey, err := btcec.NewPrivateKey(btcec.S256())
|
|
assert.Nil(t, err)
|
|
wrongPubKey := PublicKey(ecdsa.PublicKey{
|
|
Curve: btcec.S256(),
|
|
X: secondPrivKey.X,
|
|
Y: secondPrivKey.Y,
|
|
})
|
|
|
|
assert.False(t, wrongPubKey.Verify(signedData, hashedData.BytesBE()))
|
|
})
|
|
}
|