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 TestL1TxByteParsers(t *testing.T) {
 53	var pkComp babyjub.PublicKeyComp
 54	err := pkComp.UnmarshalText([]byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c"))
 55	require.Nil(t, err)
 56
 57	pk, err := pkComp.Decompress()
 58	require.Nil(t, err)
 59
 60	l1Tx := &L1Tx{
 61		ToIdx:       3,
 62		TokenID:     5,
 63		Amount:      big.NewInt(1),
 64		LoadAmount:  big.NewInt(2),
 65		FromIdx:     2,
 66		FromBJJ:     pk,
 67		FromEthAddr: ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"),
 68	}
 69
 70	expected, err := utils.HexDecode("c58d29fa6e86e4fae04ddced660d45bcf3cb237056ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c0000000000020002000100000005000000000003")
 71	require.Nil(t, err)
 72
 73	encodedData, err := l1Tx.Bytes()
 74	require.Nil(t, err)
 75	assert.Equal(t, expected, encodedData)
 76
 77	decodedData, err := L1TxFromBytes(encodedData)
 78	require.Nil(t, err)
 79	assert.Equal(t, l1Tx, decodedData)
 80
 81	encodedData2, err := decodedData.Bytes()
 82	require.Nil(t, err)
 83	assert.Equal(t, encodedData, encodedData2)
 84
 85	// expect error if length!=68
 86	_, err = L1TxFromBytes(encodedData[:66])
 87	require.NotNil(t, err)
 88	_, err = L1TxFromBytes([]byte{})
 89	require.NotNil(t, err)
 90	_, err = L1TxFromBytes(nil)
 91	require.NotNil(t, err)
 92}
 93
 94func TestL1CoordinatorTxByteParsers(t *testing.T) {
 95	privateKey, err := crypto.HexToECDSA("fad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19")
 96	require.Nil(t, err)
 97
 98	publicKey := privateKey.Public()
 99	publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey)
100	if !ok {
101		log.Fatal("error casting public key to ECDSA")
102	}
103	publicKeyBytes := crypto.FromECDSAPub(publicKeyECDSA)
104	pubKey, err := crypto.UnmarshalPubkey(publicKeyBytes)
105	require.Nil(t, err)
106	fromEthAddr := crypto.PubkeyToAddress(*pubKey)
107	var pkComp babyjub.PublicKeyComp
108	err = pkComp.UnmarshalText([]byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c"))
109	require.Nil(t, err)
110	pk, err := pkComp.Decompress()
111	require.Nil(t, err)
112	bytesMessage1 := []byte("\x19Ethereum Signed Message:\n98")
113	bytesMessage2 := []byte("I authorize this babyjubjub key for hermez rollup account creation")
114
115	babyjub := pk.Compress()
116	var data []byte
117	data = append(data, bytesMessage1...)
118	data = append(data, bytesMessage2...)
119	data = append(data, babyjub[:]...)
120	hash := crypto.Keccak256Hash(data)
121	signature, err := crypto.Sign(hash.Bytes(), privateKey)
122	require.Nil(t, err)
123	// Ethereum adds 27 to v
124	v := int(signature[64])
125	signature[64] = byte(v + 27)
126
127	l1Tx := &L1Tx{
128		TokenID:     231,
129		FromBJJ:     pk,
130		FromEthAddr: fromEthAddr,
131	}
132
133	bytesCoordinatorL1, err := l1Tx.BytesCoordinatorTx(signature)
134	require.Nil(t, err)
135	l1txDecoded, err := L1TxFromCoordinatorBytes(bytesCoordinatorL1)
136	require.Nil(t, err)
137	assert.Equal(t, l1Tx, l1txDecoded)
138}
139
140func TestL1CoordinatorTxByteParsersCompatibility(t *testing.T) {
141	var signature []byte
142	r, err := hex.DecodeString("da71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3b")
143	require.Nil(t, err)
144	s, err := hex.DecodeString("186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8c")
145	require.Nil(t, err)
146	v, err := hex.DecodeString("1b")
147	require.Nil(t, err)
148
149	signature = append(signature, r[:]...)
150	signature = append(signature, s[:]...)
151	signature = append(signature, v[:]...)
152
153	var pkComp babyjub.PublicKeyComp
154	err = pkComp.UnmarshalText([]byte("0xa2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c"))
155	require.Nil(t, err)
156	// Data from the compatibility test
157	expected := "1b186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8cda71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3ba2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c000000e7"
158	pk, err := pkComp.Decompress()
159	require.Nil(t, err)
160
161	l1Tx := &L1Tx{
162		TokenID: 231,
163		FromBJJ: pk,
164	}
165
166	encodeData, err := l1Tx.BytesCoordinatorTx(signature)
167	require.Nil(t, err)
168	assert.Equal(t, expected, hex.EncodeToString(encodeData))
169}