mirror of https://github.com/poanetwork/quorum.git
255 lines
5.4 KiB
Go
255 lines
5.4 KiB
Go
// Copyright 2015 The go-ethereum Authors
|
|
// This file is part of go-ethereum.
|
|
//
|
|
// go-ethereum is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// go-ethereum is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package secp256k1
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto/randentropy"
|
|
)
|
|
|
|
const TESTS = 10000 // how many tests
|
|
const SigSize = 65 //64+1
|
|
|
|
func Test_Secp256_00(t *testing.T) {
|
|
|
|
var nonce []byte = randentropy.GetEntropyCSPRNG(32) //going to get bitcoins stolen!
|
|
|
|
if len(nonce) != 32 {
|
|
t.Fatal()
|
|
}
|
|
|
|
}
|
|
|
|
//tests for Malleability
|
|
//highest bit of S must be 0; 32nd byte
|
|
func CompactSigTest(sig []byte) {
|
|
|
|
var b int = int(sig[32])
|
|
if b < 0 {
|
|
log.Panic()
|
|
}
|
|
if ((b >> 7) == 1) != ((b & 0x80) == 0x80) {
|
|
log.Panic("b= %v b2= %v \n", b, b>>7)
|
|
}
|
|
if (b & 0x80) == 0x80 {
|
|
log.Panic("b= %v b2= %v \n", b, b&0x80)
|
|
}
|
|
}
|
|
|
|
//test pubkey/private generation
|
|
func Test_Secp256_01(t *testing.T) {
|
|
pubkey, seckey := GenerateKeyPair()
|
|
if err := VerifySeckeyValidity(seckey); err != nil {
|
|
t.Fatal()
|
|
}
|
|
if err := VerifyPubkeyValidity(pubkey); err != nil {
|
|
t.Fatal()
|
|
}
|
|
}
|
|
|
|
//test size of messages
|
|
func Test_Secp256_02s(t *testing.T) {
|
|
pubkey, seckey := GenerateKeyPair()
|
|
msg := randentropy.GetEntropyCSPRNG(32)
|
|
sig, _ := Sign(msg, seckey)
|
|
CompactSigTest(sig)
|
|
if sig == nil {
|
|
t.Fatal("Signature nil")
|
|
}
|
|
if len(pubkey) != 65 {
|
|
t.Fail()
|
|
}
|
|
if len(seckey) != 32 {
|
|
t.Fail()
|
|
}
|
|
if len(sig) != 64+1 {
|
|
t.Fail()
|
|
}
|
|
if int(sig[64]) > 4 {
|
|
t.Fail()
|
|
} //should be 0 to 4
|
|
}
|
|
|
|
//test signing message
|
|
func Test_Secp256_02(t *testing.T) {
|
|
pubkey1, seckey := GenerateKeyPair()
|
|
msg := randentropy.GetEntropyCSPRNG(32)
|
|
sig, _ := Sign(msg, seckey)
|
|
if sig == nil {
|
|
t.Fatal("Signature nil")
|
|
}
|
|
|
|
pubkey2, _ := RecoverPubkey(msg, sig)
|
|
if pubkey2 == nil {
|
|
t.Fatal("Recovered pubkey invalid")
|
|
}
|
|
if bytes.Equal(pubkey1, pubkey2) == false {
|
|
t.Fatal("Recovered pubkey does not match")
|
|
}
|
|
|
|
err := VerifySignature(msg, sig, pubkey1)
|
|
if err != nil {
|
|
t.Fatal("Signature invalid")
|
|
}
|
|
}
|
|
|
|
//test pubkey recovery
|
|
func Test_Secp256_02a(t *testing.T) {
|
|
pubkey1, seckey1 := GenerateKeyPair()
|
|
msg := randentropy.GetEntropyCSPRNG(32)
|
|
sig, _ := Sign(msg, seckey1)
|
|
|
|
if sig == nil {
|
|
t.Fatal("Signature nil")
|
|
}
|
|
err := VerifySignature(msg, sig, pubkey1)
|
|
if err != nil {
|
|
t.Fatal("Signature invalid")
|
|
}
|
|
|
|
pubkey2, _ := RecoverPubkey(msg, sig)
|
|
if len(pubkey1) != len(pubkey2) {
|
|
t.Fatal()
|
|
}
|
|
for i, _ := range pubkey1 {
|
|
if pubkey1[i] != pubkey2[i] {
|
|
t.Fatal()
|
|
}
|
|
}
|
|
if bytes.Equal(pubkey1, pubkey2) == false {
|
|
t.Fatal()
|
|
}
|
|
}
|
|
|
|
//test random messages for the same pub/private key
|
|
func Test_Secp256_03(t *testing.T) {
|
|
_, seckey := GenerateKeyPair()
|
|
for i := 0; i < TESTS; i++ {
|
|
msg := randentropy.GetEntropyCSPRNG(32)
|
|
sig, _ := Sign(msg, seckey)
|
|
CompactSigTest(sig)
|
|
|
|
sig[len(sig)-1] %= 4
|
|
pubkey2, _ := RecoverPubkey(msg, sig)
|
|
if pubkey2 == nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
//test random messages for different pub/private keys
|
|
func Test_Secp256_04(t *testing.T) {
|
|
for i := 0; i < TESTS; i++ {
|
|
pubkey1, seckey := GenerateKeyPair()
|
|
msg := randentropy.GetEntropyCSPRNG(32)
|
|
sig, _ := Sign(msg, seckey)
|
|
CompactSigTest(sig)
|
|
|
|
if sig[len(sig)-1] >= 4 {
|
|
t.Fail()
|
|
}
|
|
pubkey2, _ := RecoverPubkey(msg, sig)
|
|
if pubkey2 == nil {
|
|
t.Fail()
|
|
}
|
|
if bytes.Equal(pubkey1, pubkey2) == false {
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
//test random signatures against fixed messages; should fail
|
|
|
|
//crashes:
|
|
// -SIPA look at this
|
|
|
|
func randSig() []byte {
|
|
sig := randentropy.GetEntropyCSPRNG(65)
|
|
sig[32] &= 0x70
|
|
sig[64] %= 4
|
|
return sig
|
|
}
|
|
|
|
func Test_Secp256_06a_alt0(t *testing.T) {
|
|
pubkey1, seckey := GenerateKeyPair()
|
|
msg := randentropy.GetEntropyCSPRNG(32)
|
|
sig, _ := Sign(msg, seckey)
|
|
|
|
if sig == nil {
|
|
t.Fail()
|
|
}
|
|
if len(sig) != 65 {
|
|
t.Fail()
|
|
}
|
|
for i := 0; i < TESTS; i++ {
|
|
sig = randSig()
|
|
pubkey2, _ := RecoverPubkey(msg, sig)
|
|
|
|
if bytes.Equal(pubkey1, pubkey2) == true {
|
|
t.Fail()
|
|
}
|
|
|
|
if pubkey2 != nil && VerifySignature(msg, sig, pubkey2) != nil {
|
|
t.Fail()
|
|
}
|
|
|
|
if VerifySignature(msg, sig, pubkey1) == nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
//test random messages against valid signature: should fail
|
|
|
|
func Test_Secp256_06b(t *testing.T) {
|
|
pubkey1, seckey := GenerateKeyPair()
|
|
msg := randentropy.GetEntropyCSPRNG(32)
|
|
sig, _ := Sign(msg, seckey)
|
|
|
|
fail_count := 0
|
|
for i := 0; i < TESTS; i++ {
|
|
msg = randentropy.GetEntropyCSPRNG(32)
|
|
pubkey2, _ := RecoverPubkey(msg, sig)
|
|
if bytes.Equal(pubkey1, pubkey2) == true {
|
|
t.Fail()
|
|
}
|
|
|
|
if pubkey2 != nil && VerifySignature(msg, sig, pubkey2) != nil {
|
|
t.Fail()
|
|
}
|
|
|
|
if VerifySignature(msg, sig, pubkey1) == nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
if fail_count != 0 {
|
|
fmt.Printf("ERROR: Accepted signature for %v of %v random messages\n", fail_count, TESTS)
|
|
}
|
|
}
|
|
|
|
func TestInvalidKey(t *testing.T) {
|
|
p1 := make([]byte, 32)
|
|
err := VerifySeckeyValidity(p1)
|
|
if err == nil {
|
|
t.Errorf("pvk %x varify sec key should have returned error", p1)
|
|
}
|
|
}
|