arnaucube
arnaucube/hermez-nodePublic Self-hosted repository
hermez-node/common/l1tx_test.go
common/l1tx_test.goRaw
  1package common
  2
  3import (
  4	"crypto/ecdsa"
  5	"encoding/hex"
  6	"log"
  7	"math/big"
  8	"testing"
  9
 10	ethCommon "github.com/ethereum/go-ethereum/common"
 11	"github.com/ethereum/go-ethereum/crypto"
 12	"github.com/iden3/go-iden3-crypto/babyjub"
 13	"github.com/iden3/go-iden3-crypto/utils"
 14	"github.com/stretchr/testify/assert"
 15	"github.com/stretchr/testify/require"
 16)
 17
 18func TestNewL1UserTx(t *testing.T) {
 19	toForge := int64(123456)
 20	l1Tx := &L1Tx{
 21		ToForgeL1TxsNum: &toForge,
 22		Position:        71,
 23		UserOrigin:      true,
 24		ToIdx:           301,
 25		TokenID:         5,
 26		Amount:          big.NewInt(1),
 27		LoadAmount:      big.NewInt(2),
 28		FromIdx:         Idx(300),
 29	}
 30	l1Tx, err := NewL1Tx(l1Tx)
 31	assert.Nil(t, err)
 32	assert.Equal(t, "0x00000000000001e240004700", l1Tx.TxID.String())
 33}
 34
 35func TestNewL1CoordinatorTx(t *testing.T) {
 36	batchNum := BatchNum(51966)
 37	l1Tx := &L1Tx{
 38		Position:   88,
 39		UserOrigin: false,
 40		ToIdx:      301,
 41		TokenID:    5,
 42		Amount:     big.NewInt(1),
 43		LoadAmount: big.NewInt(2),
 44		FromIdx:    Idx(300),
 45		BatchNum:   &batchNum,
 46	}
 47	l1Tx, err := NewL1Tx(l1Tx)
 48	assert.Nil(t, err)
 49	assert.Equal(t, "0x01000000000000cafe005800", l1Tx.TxID.String())
 50}
 51
 52func TestL1userTxByteParsers(t *testing.T) {
 53	var pkComp babyjub.PublicKeyComp
 54	pkCompL := []byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c")
 55	err := pkComp.UnmarshalText(pkCompL)
 56	require.Nil(t, err)
 57
 58	pk, err := pkComp.Decompress()
 59	require.Nil(t, err)
 60
 61	l1Tx := &L1Tx{
 62		UserOrigin:  true,
 63		ToIdx:       3,
 64		TokenID:     5,
 65		Amount:      big.NewInt(1),
 66		LoadAmount:  big.NewInt(2),
 67		FromIdx:     2,
 68		FromBJJ:     pk,
 69		FromEthAddr: ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"),
 70	}
 71
 72	encodedData, err := l1Tx.BytesUser()
 73	require.Nil(t, err)
 74	decodedData, err := L1UserTxFromBytes(encodedData)
 75	require.Nil(t, err)
 76	assert.Equal(t, l1Tx, decodedData)
 77	encodedData2, err := decodedData.BytesUser()
 78	require.Nil(t, err)
 79	assert.Equal(t, encodedData, encodedData2)
 80
 81	// expect error if length!=68
 82	_, err = L1UserTxFromBytes(encodedData[:66])
 83	require.NotNil(t, err)
 84	_, err = L1UserTxFromBytes([]byte{})
 85	require.NotNil(t, err)
 86	_, err = L1UserTxFromBytes(nil)
 87	require.NotNil(t, err)
 88}
 89
 90func TestL1TxByteParsersCompatibility(t *testing.T) {
 91	// Data from compatibility test
 92	var pkComp babyjub.PublicKeyComp
 93	pkCompB, err := hex.DecodeString("0dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a")
 94	require.Nil(t, err)
 95	pkCompL := SwapEndianness(pkCompB)
 96	err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL)))
 97	require.Nil(t, err)
 98
 99	pk, err := pkComp.Decompress()
100	require.Nil(t, err)
101
102	loadAmount := new(big.Int)
103	loadAmount.SetString("100000000000000000000", 10)
104	l1Tx := &L1Tx{
105		ToIdx:       87865485,
106		TokenID:     2098076,
107		Amount:      big.NewInt(2400000000000000000),
108		LoadAmount:  loadAmount,
109		FromIdx:     Idx(29767899),
110		FromBJJ:     pk,
111		FromEthAddr: ethCommon.HexToAddress("0x85dab5b9e2e361d0c208d77be90efcc0439b0a53"),
112	}
113
114	expected, err := utils.HexDecode("85dab5b9e2e361d0c208d77be90efcc0439b0a530dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a000001c638db8be880f00020039c0000053cb88d")
115	require.Nil(t, err)
116
117	encodedData, err := l1Tx.BytesUser()
118	require.Nil(t, err)
119	assert.Equal(t, expected, encodedData)
120}
121
122func TestL1CoordinatorTxByteParsers(t *testing.T) {
123	privateKey, err := crypto.HexToECDSA("fad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19")
124	require.Nil(t, err)
125
126	publicKey := privateKey.Public()
127	publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey)
128	if !ok {
129		log.Fatal("error casting public key to ECDSA")
130	}
131	publicKeyBytes := crypto.FromECDSAPub(publicKeyECDSA)
132	pubKey, err := crypto.UnmarshalPubkey(publicKeyBytes)
133	require.Nil(t, err)
134	fromEthAddr := crypto.PubkeyToAddress(*pubKey)
135	var pkComp babyjub.PublicKeyComp
136	pkCompL := []byte("56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c")
137	err = pkComp.UnmarshalText(pkCompL)
138	require.Nil(t, err)
139	pk, err := pkComp.Decompress()
140	require.Nil(t, err)
141	bytesMessage1 := []byte("\x19Ethereum Signed Message:\n98")
142	bytesMessage2 := []byte("I authorize this babyjubjub key for hermez rollup account creation")
143
144	babyjub := pk.Compress()
145	babyjubB := SwapEndianness(babyjub[:])
146	var data []byte
147	data = append(data, bytesMessage1...)
148	data = append(data, bytesMessage2...)
149	data = append(data, babyjubB[:]...)
150	hash := crypto.Keccak256Hash(data)
151	signature, err := crypto.Sign(hash.Bytes(), privateKey)
152	require.Nil(t, err)
153	// Ethereum adds 27 to v
154	v := int(signature[64])
155	signature[64] = byte(v + 27)
156
157	l1Tx := &L1Tx{
158		TokenID:     231,
159		FromBJJ:     pk,
160		FromEthAddr: fromEthAddr,
161		Amount:      big.NewInt(0),
162		LoadAmount:  big.NewInt(0),
163	}
164
165	bytesCoordinatorL1, err := l1Tx.BytesCoordinatorTx(signature)
166	require.Nil(t, err)
167	l1txDecoded, err := L1CoordinatorTxFromBytes(bytesCoordinatorL1)
168	require.Nil(t, err)
169	assert.Equal(t, l1Tx, l1txDecoded)
170	bytesCoordinatorL12, err := l1txDecoded.BytesCoordinatorTx(signature)
171	require.Nil(t, err)
172	assert.Equal(t, bytesCoordinatorL1, bytesCoordinatorL12)
173
174	// expect error if length!=68
175	_, err = L1CoordinatorTxFromBytes(bytesCoordinatorL1[:66])
176	require.NotNil(t, err)
177	_, err = L1CoordinatorTxFromBytes([]byte{})
178	require.NotNil(t, err)
179	_, err = L1CoordinatorTxFromBytes(nil)
180	require.NotNil(t, err)
181}
182
183func TestL1CoordinatorTxByteParsersCompatibility(t *testing.T) {
184	// Data from compatibility test
185	var signature []byte
186	r, err := hex.DecodeString("da71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3b")
187	require.Nil(t, err)
188	s, err := hex.DecodeString("186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8c")
189	require.Nil(t, err)
190	v, err := hex.DecodeString("1b")
191	require.Nil(t, err)
192
193	signature = append(signature, r[:]...)
194	signature = append(signature, s[:]...)
195	signature = append(signature, v[:]...)
196
197	var pkComp babyjub.PublicKeyComp
198	pkCompB, err := hex.DecodeString("a2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c")
199	require.Nil(t, err)
200	pkCompL := SwapEndianness(pkCompB)
201	err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL)))
202	require.Nil(t, err)
203	// Data from the compatibility test
204	require.Nil(t, err)
205	pk, err := pkComp.Decompress()
206	require.Nil(t, err)
207	l1Tx := &L1Tx{
208		TokenID: 231,
209		FromBJJ: pk,
210	}
211
212	encodeData, err := l1Tx.BytesCoordinatorTx(signature)
213	require.Nil(t, err)
214
215	expected, err := utils.HexDecode("1b186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8cda71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3ba2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c000000e7")
216	require.Nil(t, err)
217
218	assert.Equal(t, expected, encodeData)
219}