1package babyjub
  2
  3import (
  4	"encoding/hex"
  5	"fmt"
  6	"math/big"
  7	"testing"
  8
  9	"database/sql"
 10	"database/sql/driver"
 11
 12	"github.com/iden3/go-iden3-crypto/constants"
 13	"github.com/iden3/go-iden3-crypto/utils"
 14	"github.com/stretchr/testify/assert"
 15	"github.com/stretchr/testify/require"
 16)
 17
 18func TestPublicKey(t *testing.T) {
 19	var k PrivateKey
 20	for i := 0; i < 32; i++ {
 21		k[i] = byte(i)
 22	}
 23	pk := k.Public()
 24	assert.True(t, pk.X.Cmp(constants.Q) == -1)
 25	assert.True(t, pk.Y.Cmp(constants.Q) == -1)
 26}
 27
 28func TestSignVerifyMimc7(t *testing.T) {
 29	var k PrivateKey
 30	_, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
 31	require.Nil(t, err)
 32	msgBuf, err := hex.DecodeString("00010203040506070809")
 33	if err != nil {
 34		panic(err)
 35	}
 36	msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf)
 37
 38	pk := k.Public()
 39	assert.Equal(t,
 40		"13277427435165878497778222415993513565335242147425444199013288855685581939618",
 41		pk.X.String())
 42	assert.Equal(t,
 43		"13622229784656158136036771217484571176836296686641868549125388198837476602820",
 44		pk.Y.String())
 45
 46	sig := k.SignMimc7(msg)
 47	assert.Equal(t,
 48		"11384336176656855268977457483345535180380036354188103142384839473266348197733",
 49		sig.R8.X.String())
 50	assert.Equal(t,
 51		"15383486972088797283337779941324724402501462225528836549661220478783371668959",
 52		sig.R8.Y.String())
 53	assert.Equal(t,
 54		"2523202440825208709475937830811065542425109372212752003460238913256192595070",
 55		sig.S.String())
 56
 57	ok := pk.VerifyMimc7(msg, sig)
 58	assert.Equal(t, true, ok)
 59
 60	sigBuf := sig.Compress()
 61	sig2, err := new(Signature).Decompress(sigBuf)
 62	assert.Equal(t, nil, err)
 63
 64	assert.Equal(t, ""+
 65		"dfedb4315d3f2eb4de2d3c510d7a987dcab67089c8ace06308827bf5bcbe02a2"+
 66		"7ed40dab29bf993c928e789d007387998901a24913d44fddb64b1f21fc149405",
 67		hex.EncodeToString(sigBuf[:]))
 68
 69	ok = pk.VerifyMimc7(msg, sig2)
 70	assert.Equal(t, true, ok)
 71}
 72
 73func TestSignVerifyPoseidon(t *testing.T) {
 74	var k PrivateKey
 75	_, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
 76	require.Nil(t, err)
 77	msgBuf, err := hex.DecodeString("00010203040506070809")
 78	if err != nil {
 79		panic(err)
 80	}
 81	msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf)
 82
 83	pk := k.Public()
 84	assert.Equal(t,
 85		"13277427435165878497778222415993513565335242147425444199013288855685581939618",
 86		pk.X.String())
 87	assert.Equal(t,
 88		"13622229784656158136036771217484571176836296686641868549125388198837476602820",
 89		pk.Y.String())
 90
 91	sig := k.SignPoseidon(msg)
 92	assert.Equal(t,
 93		"11384336176656855268977457483345535180380036354188103142384839473266348197733",
 94		sig.R8.X.String())
 95	assert.Equal(t,
 96		"15383486972088797283337779941324724402501462225528836549661220478783371668959",
 97		sig.R8.Y.String())
 98	assert.Equal(t,
 99		"248298168863866362217836334079793350221620631973732197668910946177382043688",
100		sig.S.String())
101
102	ok := pk.VerifyPoseidon(msg, sig)
103	assert.Equal(t, true, ok)
104
105	sigBuf := sig.Compress()
106	sig2, err := new(Signature).Decompress(sigBuf)
107	assert.Equal(t, nil, err)
108
109	assert.Equal(t, ""+
110		"dfedb4315d3f2eb4de2d3c510d7a987dcab67089c8ace06308827bf5bcbe02a2"+
111		"28506bce274aa1b3f7e7c2fd7e4fe09bff8f9aa37a42def7994e98f322888c00",
112		hex.EncodeToString(sigBuf[:]))
113
114	ok = pk.VerifyPoseidon(msg, sig2)
115	assert.Equal(t, true, ok)
116}
117
118func TestCompressDecompress(t *testing.T) {
119	var k PrivateKey
120	_, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
121	require.Nil(t, err)
122	pk := k.Public()
123	for i := 0; i < 64; i++ {
124		msgBuf, err := hex.DecodeString(fmt.Sprintf("000102030405060708%02d", i))
125		if err != nil {
126			panic(err)
127		}
128		msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf)
129		sig := k.SignMimc7(msg)
130		sigBuf := sig.Compress()
131		sig2, err := new(Signature).Decompress(sigBuf)
132		assert.Equal(t, nil, err)
133		ok := pk.VerifyMimc7(msg, sig2)
134		assert.Equal(t, true, ok)
135	}
136}
137
138func TestScannerValuer(t *testing.T) {
139	privK := NewRandPrivKey()
140	var value driver.Valuer
141	var scan sql.Scanner
142	value = privK.SignPoseidon(big.NewInt(674238462))
143	scan = privK.SignPoseidon(big.NewInt(1))
144	fromDB, err := value.Value()
145	assert.NoError(t, err)
146	assert.NoError(t, scan.Scan(fromDB))
147	assert.Equal(t, value, scan)
148}
149
150func BenchmarkBabyjubEddsa(b *testing.B) {
151	var k PrivateKey
152	_, err := hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
153	require.Nil(b, err)
154	pk := k.Public()
155
156	const n = 256
157
158	msgBuf, err := hex.DecodeString("00010203040506070809")
159	if err != nil {
160		panic(err)
161	}
162	msg := utils.SetBigIntFromLEBytes(new(big.Int), msgBuf)
163	var msgs [n]*big.Int
164	for i := 0; i < n; i++ {
165		msgs[i] = new(big.Int).Add(msg, big.NewInt(int64(i)))
166	}
167	var sigs [n]*Signature
168
169	b.Run("SignMimc7", func(b *testing.B) {
170		for i := 0; i < b.N; i++ {
171			k.SignMimc7(msgs[i%n])
172		}
173	})
174
175	for i := 0; i < n; i++ {
176		sigs[i%n] = k.SignMimc7(msgs[i%n])
177	}
178
179	b.Run("VerifyMimc7", func(b *testing.B) {
180		for i := 0; i < b.N; i++ {
181			pk.VerifyMimc7(msgs[i%n], sigs[i%n])
182		}
183	})
184
185	b.Run("SignPoseidon", func(b *testing.B) {
186		for i := 0; i < b.N; i++ {
187			k.SignPoseidon(msgs[i%n])
188		}
189	})
190
191	for i := 0; i < n; i++ {
192		sigs[i%n] = k.SignPoseidon(msgs[i%n])
193	}
194
195	b.Run("VerifyPoseidon", func(b *testing.B) {
196		for i := 0; i < b.N; i++ {
197			pk.VerifyPoseidon(msgs[i%n], sigs[i%n])
198		}
199	})
200}