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 TestNewL1Tx(t *testing.T) {
 19	l1Tx := &L1Tx{
 20		ToForgeL1TxsNum: int64(123456),
 21		Position:        71,
 22		ToIdx:           301,
 23		TokenID:         5,
 24		Amount:          big.NewInt(1),
 25		LoadAmount:      big.NewInt(2),
 26		FromIdx:         300,
 27	}
 28	l1Tx, err := NewL1Tx(l1Tx)
 29	assert.Nil(t, err)
 30	assert.Equal(t, "0x01000000000001e240004700", l1Tx.TxID.String())
 31}
 32
 33func TestL1TxByteParsers(t *testing.T) {
 34	var pkComp babyjub.PublicKeyComp
 35	err := pkComp.UnmarshalText([]byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c"))
 36	require.Nil(t, err)
 37
 38	pk, err := pkComp.Decompress()
 39	require.Nil(t, err)
 40
 41	l1Tx := &L1Tx{
 42		ToIdx:       3,
 43		TokenID:     5,
 44		Amount:      big.NewInt(1),
 45		LoadAmount:  big.NewInt(2),
 46		FromIdx:     2,
 47		FromBJJ:     pk,
 48		FromEthAddr: ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"),
 49	}
 50
 51	expected, err := utils.HexDecode("c58d29fa6e86e4fae04ddced660d45bcf3cb237056ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c0000000000020002000100000005000000000003")
 52	require.Nil(t, err)
 53
 54	encodedData, err := l1Tx.Bytes()
 55	require.Nil(t, err)
 56	assert.Equal(t, expected, encodedData)
 57
 58	decodedData, err := L1TxFromBytes(encodedData)
 59	require.Nil(t, err)
 60	assert.Equal(t, l1Tx, decodedData)
 61
 62	encodedData2, err := decodedData.Bytes()
 63	require.Nil(t, err)
 64	assert.Equal(t, encodedData, encodedData2)
 65
 66	// expect error if length!=68
 67	_, err = L1TxFromBytes(encodedData[:66])
 68	require.NotNil(t, err)
 69	_, err = L1TxFromBytes([]byte{})
 70	require.NotNil(t, err)
 71	_, err = L1TxFromBytes(nil)
 72	require.NotNil(t, err)
 73}
 74
 75func TestL1CoordinatorTxByteParsers(t *testing.T) {
 76	privateKey, err := crypto.HexToECDSA("fad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19")
 77	require.Nil(t, err)
 78
 79	publicKey := privateKey.Public()
 80	publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey)
 81	if !ok {
 82		log.Fatal("error casting public key to ECDSA")
 83	}
 84	publicKeyBytes := crypto.FromECDSAPub(publicKeyECDSA)
 85	pubKey, err := crypto.UnmarshalPubkey(publicKeyBytes)
 86	require.Nil(t, err)
 87	fromEthAddr := crypto.PubkeyToAddress(*pubKey)
 88	var pkComp babyjub.PublicKeyComp
 89	err = pkComp.UnmarshalText([]byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c"))
 90	require.Nil(t, err)
 91	pk, err := pkComp.Decompress()
 92	require.Nil(t, err)
 93	bytesMessage1 := []byte("\x19Ethereum Signed Message:\n98")
 94	bytesMessage2 := []byte("I authorize this babyjubjub key for hermez rollup account creation")
 95
 96	babyjub := pk.Compress()
 97	var data []byte
 98	data = append(data, bytesMessage1...)
 99	data = append(data, bytesMessage2...)
100	data = append(data, babyjub[:]...)
101	hash := crypto.Keccak256Hash(data)
102	signature, err := crypto.Sign(hash.Bytes(), privateKey)
103	require.Nil(t, err)
104	// Ethereum adds 27 to v
105	v := int(signature[64])
106	signature[64] = byte(v + 27)
107
108	l1Tx := &L1Tx{
109		TokenID:     231,
110		FromBJJ:     pk,
111		FromEthAddr: fromEthAddr,
112	}
113
114	bytesCoordinatorL1, err := l1Tx.BytesCoordinatorTx(signature)
115	require.Nil(t, err)
116	l1txDecoded, err := L1TxFromCoordinatorBytes(bytesCoordinatorL1)
117	require.Nil(t, err)
118	assert.Equal(t, l1Tx, l1txDecoded)
119}
120
121func TestL1CoordinatorTxByteParsersCompatibility(t *testing.T) {
122	var signature []byte
123	r, err := hex.DecodeString("da71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3b")
124	require.Nil(t, err)
125	s, err := hex.DecodeString("186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8c")
126	require.Nil(t, err)
127	v, err := hex.DecodeString("1b")
128	require.Nil(t, err)
129
130	signature = append(signature, r[:]...)
131	signature = append(signature, s[:]...)
132	signature = append(signature, v[:]...)
133
134	var pkComp babyjub.PublicKeyComp
135	err = pkComp.UnmarshalText([]byte("0xa2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c"))
136	require.Nil(t, err)
137	// Data from the compatibility test
138	expected := "1b186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8cda71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3ba2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c000000e7"
139	pk, err := pkComp.Decompress()
140	require.Nil(t, err)
141
142	l1Tx := &L1Tx{
143		TokenID: 231,
144		FromBJJ: pk,
145	}
146
147	encodeData, err := l1Tx.BytesCoordinatorTx(signature)
148	require.Nil(t, err)
149	assert.Equal(t, expected, hex.EncodeToString(encodeData))
150}