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		DepositAmount:   big.NewInt(2),
 28		FromIdx:         Idx(300),
 29	}
 30	l1Tx, err := NewL1Tx(l1Tx)
 31	assert.NoError(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		DepositAmount: big.NewInt(2),
 44		FromIdx:       Idx(300),
 45		BatchNum:      &batchNum,
 46	}
 47	l1Tx, err := NewL1Tx(l1Tx)
 48	assert.NoError(t, err)
 49	assert.Equal(t, "0x01000000000000cafe005800", l1Tx.TxID.String())
 50}
 51
 52func TestL1TxCompressedData(t *testing.T) {
 53	tx := L1Tx{
 54		FromIdx: 2,
 55		ToIdx:   3,
 56		Amount:  big.NewInt(4),
 57		TokenID: 5,
 58	}
 59	txCompressedData, err := tx.TxCompressedData()
 60	assert.NoError(t, err)
 61
 62	// test vector value generated from javascript implementation
 63	expectedStr := "7307597389635308713748674793997299267459594577423"
 64	assert.Equal(t, expectedStr, txCompressedData.String())
 65	assert.Equal(t, "0500040000000000030000000000020000c60be60f", hex.EncodeToString(txCompressedData.Bytes()))
 66}
 67
 68func TestBytesDataAvailability(t *testing.T) {
 69	tx := L1Tx{
 70		FromIdx: 2,
 71		ToIdx:   3,
 72		Amount:  big.NewInt(4),
 73		TokenID: 5,
 74	}
 75	txCompressedData, err := tx.BytesDataAvailability(32)
 76	assert.NoError(t, err)
 77	assert.Equal(t, "0000000200000003000000", hex.EncodeToString(txCompressedData))
 78
 79	tx = L1Tx{
 80		FromIdx:         2,
 81		ToIdx:           3,
 82		EffectiveAmount: big.NewInt(4),
 83		TokenID:         5,
 84	}
 85	txCompressedData, err = tx.BytesDataAvailability(32)
 86	assert.NoError(t, err)
 87	assert.Equal(t, "0000000200000003000400", hex.EncodeToString(txCompressedData))
 88}
 89
 90func TestL1TxFromDataAvailability(t *testing.T) {
 91	tx := L1Tx{
 92		FromIdx: 2,
 93		ToIdx:   3,
 94		Amount:  big.NewInt(4),
 95	}
 96	txCompressedData, err := tx.BytesDataAvailability(32)
 97	assert.NoError(t, err)
 98	l1tx, err := L1TxFromDataAvailability(txCompressedData, 32)
 99	require.NoError(t, err)
100	assert.Equal(t, tx.FromIdx, l1tx.FromIdx)
101	assert.Equal(t, tx.ToIdx, l1tx.ToIdx)
102
103	tx = L1Tx{
104		FromIdx:         2,
105		ToIdx:           3,
106		EffectiveAmount: big.NewInt(4),
107	}
108	txCompressedData, err = tx.BytesDataAvailability(32)
109	assert.NoError(t, err)
110	l1tx, err = L1TxFromDataAvailability(txCompressedData, 32)
111	require.NoError(t, err)
112	assert.Equal(t, tx.FromIdx, l1tx.FromIdx)
113	assert.Equal(t, tx.ToIdx, l1tx.ToIdx)
114	assert.Equal(t, tx.EffectiveAmount, l1tx.EffectiveAmount)
115}
116
117func TestL1userTxByteParsers(t *testing.T) {
118	var pkComp babyjub.PublicKeyComp
119	pkCompL := []byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c")
120	err := pkComp.UnmarshalText(pkCompL)
121	require.NoError(t, err)
122
123	l1Tx := &L1Tx{
124		UserOrigin:    true,
125		ToIdx:         3,
126		TokenID:       5,
127		Amount:        big.NewInt(1),
128		DepositAmount: big.NewInt(2),
129		FromIdx:       2,
130		FromBJJ:       pkComp,
131		FromEthAddr:   ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"),
132	}
133
134	encodedData, err := l1Tx.BytesUser()
135	require.NoError(t, err)
136	decodedData, err := L1UserTxFromBytes(encodedData)
137	require.NoError(t, err)
138	assert.Equal(t, l1Tx, decodedData)
139	encodedData2, err := decodedData.BytesUser()
140	require.NoError(t, err)
141	assert.Equal(t, encodedData, encodedData2)
142
143	// expect error if length!=68
144	_, err = L1UserTxFromBytes(encodedData[:66])
145	require.NotNil(t, err)
146	_, err = L1UserTxFromBytes([]byte{})
147	require.NotNil(t, err)
148	_, err = L1UserTxFromBytes(nil)
149	require.NotNil(t, err)
150}
151
152func TestL1TxByteParsersCompatibility(t *testing.T) {
153	// Data from compatibility test
154	var pkComp babyjub.PublicKeyComp
155	pkCompB, err := hex.DecodeString("0dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a")
156	require.NoError(t, err)
157	pkCompL := SwapEndianness(pkCompB)
158	err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL)))
159	require.NoError(t, err)
160
161	depositAmount := new(big.Int)
162	depositAmount.SetString("100000000000000000000", 10)
163	l1Tx := &L1Tx{
164		ToIdx:         87865485,
165		TokenID:       2098076,
166		Amount:        big.NewInt(2400000000000000000),
167		DepositAmount: depositAmount,
168		FromIdx:       Idx(29767899),
169		FromBJJ:       pkComp,
170		FromEthAddr:   ethCommon.HexToAddress("0x85dab5b9e2e361d0c208d77be90efcc0439b0a53"),
171		UserOrigin:    true,
172	}
173
174	expected, err := utils.HexDecode("85dab5b9e2e361d0c208d77be90efcc0439b0a530dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a000001c638db8be880f00020039c0000053cb88d")
175	require.NoError(t, err)
176
177	encodedData, err := l1Tx.BytesUser()
178	require.NoError(t, err)
179	assert.Equal(t, expected, encodedData)
180}
181
182func TestL1CoordinatorTxByteParsers(t *testing.T) {
183	hermezAddress := ethCommon.HexToAddress("0xD6C850aeBFDC46D7F4c207e445cC0d6B0919BDBe")
184	chainID := big.NewInt(1337)
185	chainIDBytes := ethCommon.LeftPadBytes(chainID.Bytes(), 2)
186
187	privateKey, err := crypto.HexToECDSA("fad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19")
188	require.NoError(t, err)
189
190	publicKey := privateKey.Public()
191	publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey)
192	if !ok {
193		log.Fatal("error casting public key to ECDSA")
194	}
195	publicKeyBytes := crypto.FromECDSAPub(publicKeyECDSA)
196	pubKey, err := crypto.UnmarshalPubkey(publicKeyBytes)
197	require.NoError(t, err)
198	fromEthAddr := crypto.PubkeyToAddress(*pubKey)
199	var pkComp babyjub.PublicKeyComp
200	pkCompL := []byte("56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c")
201	err = pkComp.UnmarshalText(pkCompL)
202	require.NoError(t, err)
203	bytesMessage1 := []byte("\x19Ethereum Signed Message:\n120")
204	bytesMessage2 := []byte("I authorize this babyjubjub key for hermez rollup account creation")
205
206	babyjubB := SwapEndianness(pkComp[:])
207	var data []byte
208	data = append(data, bytesMessage1...)
209	data = append(data, bytesMessage2...)
210	data = append(data, babyjubB[:]...)
211	data = append(data, chainIDBytes...)
212	data = append(data, hermezAddress.Bytes()...)
213	hash := crypto.Keccak256Hash(data)
214	signature, err := crypto.Sign(hash.Bytes(), privateKey)
215	require.NoError(t, err)
216	// Ethereum adds 27 to v
217	v := int(signature[64])
218	signature[64] = byte(v + 27)
219
220	l1Tx := &L1Tx{
221		TokenID:       231,
222		FromBJJ:       pkComp,
223		FromEthAddr:   fromEthAddr,
224		Amount:        big.NewInt(0),
225		DepositAmount: big.NewInt(0),
226	}
227
228	bytesCoordinatorL1, err := l1Tx.BytesCoordinatorTx(signature)
229	require.NoError(t, err)
230	l1txDecoded, err := L1CoordinatorTxFromBytes(bytesCoordinatorL1, chainID, hermezAddress)
231	require.NoError(t, err)
232	assert.Equal(t, l1Tx, l1txDecoded)
233	bytesCoordinatorL12, err := l1txDecoded.BytesCoordinatorTx(signature)
234	require.NoError(t, err)
235	assert.Equal(t, bytesCoordinatorL1, bytesCoordinatorL12)
236
237	// expect error if length!=68
238	_, err = L1CoordinatorTxFromBytes(bytesCoordinatorL1[:66], chainID, hermezAddress)
239	require.NotNil(t, err)
240	_, err = L1CoordinatorTxFromBytes([]byte{}, chainID, hermezAddress)
241	require.NotNil(t, err)
242	_, err = L1CoordinatorTxFromBytes(nil, chainID, hermezAddress)
243	require.NotNil(t, err)
244}
245
246func TestL1CoordinatorTxByteParsersCompatibility(t *testing.T) {
247	// Data from compatibility test
248	var signature []byte
249	r, err := hex.DecodeString("da71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3b")
250	require.NoError(t, err)
251	s, err := hex.DecodeString("186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8c")
252	require.NoError(t, err)
253	v, err := hex.DecodeString("1b")
254	require.NoError(t, err)
255
256	signature = append(signature, r[:]...)
257	signature = append(signature, s[:]...)
258	signature = append(signature, v[:]...)
259
260	var pkComp babyjub.PublicKeyComp
261	pkCompB, err := hex.DecodeString("a2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c")
262	require.NoError(t, err)
263	pkCompL := SwapEndianness(pkCompB)
264	err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL)))
265	require.NoError(t, err)
266	// Data from the compatibility test
267	require.NoError(t, err)
268	l1Tx := &L1Tx{
269		TokenID: 231,
270		FromBJJ: pkComp,
271	}
272
273	encodeData, err := l1Tx.BytesCoordinatorTx(signature)
274	require.NoError(t, err)
275
276	expected, err := utils.HexDecode("1b186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8cda71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3ba2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c000000e7")
277	require.NoError(t, err)
278
279	assert.Equal(t, expected, encodeData)
280}