1package babyjub
 2
 3import (
 4	"crypto/rand"
 5	"encoding/hex"
 6	"fmt"
 7
 8	"github.com/stretchr/testify/assert"
 9
10	"math/big"
11	"testing"
12)
13
14func genInputs() (*PrivateKey, *big.Int) {
15	k := NewRandPrivKey()
16	fmt.Println("k", hex.EncodeToString(k[:]))
17
18	msgBuf := [32]byte{}
19	rand.Read(msgBuf[:])
20	msg := SetBigIntFromLEBytes(new(big.Int), msgBuf[:])
21	msg.Mod(msg, Q)
22	fmt.Println("msg", msg)
23
24	return &k, msg
25}
26
27func TestSignVerify1(t *testing.T) {
28	var k PrivateKey
29	hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
30	msgBuf, err := hex.DecodeString("00010203040506070809")
31	if err != nil {
32		panic(err)
33	}
34	msg := SetBigIntFromLEBytes(new(big.Int), msgBuf)
35
36	pk := k.Public()
37	assert.Equal(t,
38		"2610057752638682202795145288373380503107623443963127956230801721756904484787",
39		pk.X.String())
40	assert.Equal(t,
41		"16617171478497210597712478520507818259149717466230047843969353176573634386897",
42		pk.Y.String())
43
44	sig := k.SignMimc7(msg)
45	assert.Equal(t,
46		"4974729414807584049518234760796200867685098748448054182902488636762478901554",
47		sig.R8.X.String())
48	assert.Equal(t,
49		"18714049394522540751536514815950425694461287643205706667341348804546050128733",
50		sig.R8.Y.String())
51	assert.Equal(t,
52		"2171284143457722024136077617757713039502332290425057126942676527240038689549",
53		sig.S.String())
54
55	ok := pk.VerifyMimc7(msg, sig)
56	assert.Equal(t, true, ok)
57
58	sigBuf := sig.Compress()
59	sig2, err := new(Signature).Decompress(sigBuf)
60	assert.Equal(t, nil, err)
61
62	assert.Equal(t, ""+
63		"5dfb6f843c023fe3e52548ccf22e55c81b426f7af81b4f51f7152f2fcfc65f29"+
64		"0dab19c5a0a75973cd75a54780de0c3a41ede6f57396fe99b5307fff3ce7cc04",
65		hex.EncodeToString(sigBuf[:]))
66
67	ok = pk.VerifyMimc7(msg, sig2)
68	assert.Equal(t, true, ok)
69}
70
71func TestCompressDecompress(t *testing.T) {
72	var k PrivateKey
73	hex.Decode(k[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
74	pk := k.Public()
75	for i := 0; i < 64; i++ {
76		msgBuf, err := hex.DecodeString(fmt.Sprintf("000102030405060708%02d", i))
77		if err != nil {
78			panic(err)
79		}
80		msg := SetBigIntFromLEBytes(new(big.Int), msgBuf)
81		sig := k.SignMimc7(msg)
82		sigBuf := sig.Compress()
83		sig2, err := new(Signature).Decompress(sigBuf)
84		assert.Equal(t, nil, err)
85		ok := pk.VerifyMimc7(msg, sig2)
86		assert.Equal(t, true, ok)
87	}
88}