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