|
|
package babyjub
import ( "encoding/hex" "fmt" "math/big" "testing"
"database/sql" "database/sql/driver"
"github.com/iden3/go-iden3-crypto/constants" "github.com/iden3/go-iden3-crypto/utils" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" )
func TestPublicKey(t *testing.T) { var k PrivateKey for i := 0; i < 32; i++ { k[i] = byte(i) } pk := k.Public() assert.True(t, pk.X.Cmp(constants.Q) == -1) assert.True(t, pk.Y.Cmp(constants.Q) == -1) }
func TestSignVerifyMimc7(t *testing.T) { var k PrivateKey _, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001")) require.Nil(t, err) msgBuf, err := hex.DecodeString("00010203040506070809") if err != nil { panic(err) } msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf)
pk := k.Public() assert.Equal(t, "13277427435165878497778222415993513565335242147425444199013288855685581939618", pk.X.String()) assert.Equal(t, "13622229784656158136036771217484571176836296686641868549125388198837476602820", pk.Y.String())
sig := k.SignMimc7(msg) assert.Equal(t, "11384336176656855268977457483345535180380036354188103142384839473266348197733", sig.R8.X.String()) assert.Equal(t, "15383486972088797283337779941324724402501462225528836549661220478783371668959", sig.R8.Y.String()) assert.Equal(t, "2523202440825208709475937830811065542425109372212752003460238913256192595070", sig.S.String())
ok := pk.VerifyMimc7(msg, sig) assert.Equal(t, true, ok)
sigBuf := sig.Compress() sig2, err := new(Signature).Decompress(sigBuf) assert.Equal(t, nil, err)
assert.Equal(t, ""+ "dfedb4315d3f2eb4de2d3c510d7a987dcab67089c8ace06308827bf5bcbe02a2"+ "7ed40dab29bf993c928e789d007387998901a24913d44fddb64b1f21fc149405", hex.EncodeToString(sigBuf[:]))
ok = pk.VerifyMimc7(msg, sig2) assert.Equal(t, true, ok) }
func TestSignVerifyPoseidon(t *testing.T) { var k PrivateKey _, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001")) require.Nil(t, err) msgBuf, err := hex.DecodeString("00010203040506070809") if err != nil { panic(err) } msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf)
pk := k.Public() assert.Equal(t, "13277427435165878497778222415993513565335242147425444199013288855685581939618", pk.X.String()) assert.Equal(t, "13622229784656158136036771217484571176836296686641868549125388198837476602820", pk.Y.String())
sig := k.SignPoseidon(msg) assert.Equal(t, "11384336176656855268977457483345535180380036354188103142384839473266348197733", sig.R8.X.String()) assert.Equal(t, "15383486972088797283337779941324724402501462225528836549661220478783371668959", sig.R8.Y.String()) assert.Equal(t, "248298168863866362217836334079793350221620631973732197668910946177382043688", sig.S.String())
ok := pk.VerifyPoseidon(msg, sig) assert.Equal(t, true, ok)
sigBuf := sig.Compress() sig2, err := new(Signature).Decompress(sigBuf) assert.Equal(t, nil, err)
assert.Equal(t, ""+ "dfedb4315d3f2eb4de2d3c510d7a987dcab67089c8ace06308827bf5bcbe02a2"+ "28506bce274aa1b3f7e7c2fd7e4fe09bff8f9aa37a42def7994e98f322888c00", hex.EncodeToString(sigBuf[:]))
ok = pk.VerifyPoseidon(msg, sig2) assert.Equal(t, true, ok) }
func TestCompressDecompress(t *testing.T) { var k PrivateKey _, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001")) require.Nil(t, err) pk := k.Public() for i := 0; i < 64; i++ { msgBuf, err := hex.DecodeString(fmt.Sprintf("000102030405060708%02d", i)) if err != nil { panic(err) } msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf) sig := k.SignMimc7(msg) sigBuf := sig.Compress() sig2, err := new(Signature).Decompress(sigBuf) assert.Equal(t, nil, err) ok := pk.VerifyMimc7(msg, sig2) assert.Equal(t, true, ok) } }
func TestSignatureScannerValuer(t *testing.T) { privK := NewRandPrivKey() var value driver.Valuer var scan sql.Scanner value = privK.SignPoseidon(big.NewInt(674238462)) scan = privK.SignPoseidon(big.NewInt(1)) fromDB, err := value.Value() assert.NoError(t, err) assert.NoError(t, scan.Scan(fromDB)) assert.Equal(t, value, scan) }
func TestPubKeyScannerValuer(t *testing.T) { privKValue := NewRandPrivKey() pubKValue := privKValue.Public() privKScan := NewRandPrivKey() pubKScan := privKScan.Public() var value driver.Valuer var scan sql.Scanner value = pubKValue scan = pubKScan fromDB, err := value.Value() assert.NoError(t, err) assert.NoError(t, scan.Scan(fromDB)) assert.Equal(t, value, scan) }
func BenchmarkBabyjubEddsa(b *testing.B) { var k PrivateKey _, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001")) require.Nil(b, err) pk := k.Public()
const n = 256
msgBuf, err := hex.DecodeString("00010203040506070809") if err != nil { panic(err) } msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf) var msgs [n]*big.Int for i := 0; i < n; i++ { msgs[i] = new(big.Int).Add(msg, big.NewInt(int64(i))) } var sigs [n]*Signature
b.Run("SignMimc7", func(b *testing.B) { for i := 0; i < b.N; i++ { k.SignMimc7(msgs[i%n]) } })
for i := 0; i < n; i++ { sigs[i%n] = k.SignMimc7(msgs[i%n]) }
b.Run("VerifyMimc7", func(b *testing.B) { for i := 0; i < b.N; i++ { pk.VerifyMimc7(msgs[i%n], sigs[i%n]) } })
b.Run("SignPoseidon", func(b *testing.B) { for i := 0; i < b.N; i++ { k.SignPoseidon(msgs[i%n]) } })
for i := 0; i < n; i++ { sigs[i%n] = k.SignPoseidon(msgs[i%n]) }
b.Run("VerifyPoseidon", func(b *testing.B) { for i := 0; i < b.N; i++ { pk.VerifyPoseidon(msgs[i%n], sigs[i%n]) } }) }
|