package common import ( "crypto/ecdsa" "encoding/hex" "log" "math/big" "testing" ethCommon "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/crypto" "github.com/iden3/go-iden3-crypto/babyjub" "github.com/iden3/go-iden3-crypto/utils" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestNewL1UserTx(t *testing.T) { toForge := int64(123456) l1Tx := &L1Tx{ ToForgeL1TxsNum: &toForge, Position: 71, UserOrigin: true, ToIdx: 301, TokenID: 5, Amount: big.NewInt(1), DepositAmount: big.NewInt(2), FromIdx: Idx(300), } l1Tx, err := NewL1Tx(l1Tx) assert.NoError(t, err) assert.Equal(t, "0x00a6cbae3b8661fb75b0919ca6605a02cfb04d9c6dd16870fa0fcdf01befa32768", l1Tx.TxID.String()) } func TestNewL1CoordinatorTx(t *testing.T) { batchNum := BatchNum(51966) l1Tx := &L1Tx{ Position: 88, UserOrigin: false, ToIdx: 301, TokenID: 5, Amount: big.NewInt(1), DepositAmount: big.NewInt(2), FromIdx: Idx(300), BatchNum: &batchNum, } l1Tx, err := NewL1Tx(l1Tx) assert.NoError(t, err) assert.Equal(t, "0x01274482d73df4dab34a1b6740adfca347a462513aa14e82f27b12f818d1b68c84", l1Tx.TxID.String()) } func TestL1TxCompressedData(t *testing.T) { // test vectors values generated from javascript implementation (using // PoolL2Tx values) amount, ok := new(big.Int).SetString("343597383670000000000000000000000000000000", 10) require.True(t, ok) tx := L1Tx{ FromIdx: (1 << 48) - 1, ToIdx: (1 << 48) - 1, Amount: amount, TokenID: (1 << 32) - 1, } txCompressedData, err := tx.TxCompressedData(uint16((1 << 16) - 1)) assert.NoError(t, err) expectedStr := "ffffffffffffffffffffffffffffffffffffc60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes())) tx = L1Tx{ FromIdx: 0, ToIdx: 0, Amount: big.NewInt(0), TokenID: 0, } txCompressedData, err = tx.TxCompressedData(uint16(0)) assert.NoError(t, err) expectedStr = "c60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes())) amount, ok = new(big.Int).SetString("63000000000000000", 10) require.True(t, ok) tx = L1Tx{ FromIdx: 324, ToIdx: 256, Amount: amount, TokenID: 123, } txCompressedData, err = tx.TxCompressedData(uint16(1)) assert.NoError(t, err) expectedStr = "7b0000000001000000000001440001c60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes())) tx = L1Tx{ FromIdx: 1, ToIdx: 2, TokenID: 3, } txCompressedData, err = tx.TxCompressedData(uint16(0)) assert.NoError(t, err) expectedStr = "030000000000020000000000010000c60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes())) } func TestBytesDataAvailability(t *testing.T) { // test vectors values generated from javascript implementation amount, ok := new(big.Int).SetString("343597383670000000000000000000000000000000", 10) require.True(t, ok) tx := L1Tx{ ToIdx: (1 << 16) - 1, FromIdx: (1 << 16) - 1, EffectiveAmount: amount, } txCompressedData, err := tx.BytesDataAvailability(16) assert.NoError(t, err) assert.Equal(t, "ffffffffffffffffff00", hex.EncodeToString(txCompressedData)) l1tx, err := L1TxFromDataAvailability(txCompressedData, 16) require.NoError(t, err) assert.Equal(t, tx.FromIdx, l1tx.FromIdx) assert.Equal(t, tx.ToIdx, l1tx.ToIdx) assert.Equal(t, tx.EffectiveAmount, l1tx.EffectiveAmount) tx = L1Tx{ ToIdx: (1 << 32) - 1, FromIdx: (1 << 32) - 1, EffectiveAmount: amount, } txCompressedData, err = tx.BytesDataAvailability(32) assert.NoError(t, err) assert.Equal(t, "ffffffffffffffffffffffffff00", hex.EncodeToString(txCompressedData)) l1tx, err = L1TxFromDataAvailability(txCompressedData, 32) require.NoError(t, err) assert.Equal(t, tx.FromIdx, l1tx.FromIdx) assert.Equal(t, tx.ToIdx, l1tx.ToIdx) assert.Equal(t, tx.EffectiveAmount, l1tx.EffectiveAmount) tx = L1Tx{ ToIdx: 0, FromIdx: 0, EffectiveAmount: big.NewInt(0), } txCompressedData, err = tx.BytesDataAvailability(32) assert.NoError(t, err) assert.Equal(t, "0000000000000000000000000000", hex.EncodeToString(txCompressedData)) l1tx, err = L1TxFromDataAvailability(txCompressedData, 32) require.NoError(t, err) assert.Equal(t, tx.FromIdx, l1tx.FromIdx) assert.Equal(t, tx.ToIdx, l1tx.ToIdx) assert.Equal(t, tx.EffectiveAmount, l1tx.EffectiveAmount) tx = L1Tx{ ToIdx: 635, FromIdx: 296, EffectiveAmount: big.NewInt(1000000000000000000), } txCompressedData, err = tx.BytesDataAvailability(32) assert.NoError(t, err) assert.Equal(t, "000001280000027b42540be40000", hex.EncodeToString(txCompressedData)) l1tx, err = L1TxFromDataAvailability(txCompressedData, 32) require.NoError(t, err) assert.Equal(t, tx.FromIdx, l1tx.FromIdx) assert.Equal(t, tx.ToIdx, l1tx.ToIdx) assert.Equal(t, tx.EffectiveAmount, l1tx.EffectiveAmount) } func TestL1userTxByteParsers(t *testing.T) { var pkComp babyjub.PublicKeyComp pkCompL := []byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c") err := pkComp.UnmarshalText(pkCompL) require.NoError(t, err) l1Tx := &L1Tx{ UserOrigin: true, ToIdx: 3, TokenID: 5, Amount: big.NewInt(1), DepositAmount: big.NewInt(2), FromIdx: 2, FromBJJ: pkComp, FromEthAddr: ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"), } encodedData, err := l1Tx.BytesUser() require.NoError(t, err) decodedData, err := L1UserTxFromBytes(encodedData) require.NoError(t, err) assert.Equal(t, l1Tx, decodedData) encodedData2, err := decodedData.BytesUser() require.NoError(t, err) assert.Equal(t, encodedData, encodedData2) // expect error if length!=68 _, err = L1UserTxFromBytes(encodedData[:66]) require.NotNil(t, err) _, err = L1UserTxFromBytes([]byte{}) require.NotNil(t, err) _, err = L1UserTxFromBytes(nil) require.NotNil(t, err) } func TestL1TxByteParsersCompatibility(t *testing.T) { // Data from compatibility test var pkComp babyjub.PublicKeyComp pkCompB, err := hex.DecodeString("0dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a") require.NoError(t, err) pkCompL := SwapEndianness(pkCompB) err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL))) require.NoError(t, err) depositAmount := new(big.Int) depositAmount.SetString("100000000000000000000", 10) l1Tx := &L1Tx{ ToIdx: 87865485, TokenID: 2098076, Amount: big.NewInt(2400000000000000000), DepositAmount: depositAmount, FromIdx: Idx(29767899), FromBJJ: pkComp, FromEthAddr: ethCommon.HexToAddress("0x85dab5b9e2e361d0c208d77be90efcc0439b0a53"), UserOrigin: true, } encodedData, err := l1Tx.BytesUser() require.NoError(t, err) expected := "85dab5b9e2e361d0c208d77be90efcc0439b0a530dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a000001c638db52540be400459682f0000020039c0000053cb88d" assert.Equal(t, expected, hex.EncodeToString(encodedData)) } func TestL1CoordinatorTxByteParsers(t *testing.T) { hermezAddress := ethCommon.HexToAddress("0xD6C850aeBFDC46D7F4c207e445cC0d6B0919BDBe") chainID := big.NewInt(1337) chainIDBytes := ethCommon.LeftPadBytes(chainID.Bytes(), 2) privateKey, err := crypto.HexToECDSA("fad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19") require.NoError(t, err) publicKey := privateKey.Public() publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey) if !ok { log.Fatal("error casting public key to ECDSA") } publicKeyBytes := crypto.FromECDSAPub(publicKeyECDSA) pubKey, err := crypto.UnmarshalPubkey(publicKeyBytes) require.NoError(t, err) fromEthAddr := crypto.PubkeyToAddress(*pubKey) var pkComp babyjub.PublicKeyComp pkCompL := []byte("56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c") err = pkComp.UnmarshalText(pkCompL) require.NoError(t, err) bytesMessage1 := []byte("\x19Ethereum Signed Message:\n120") bytesMessage2 := []byte("I authorize this babyjubjub key for hermez rollup account creation") babyjubB := SwapEndianness(pkComp[:]) var data []byte data = append(data, bytesMessage1...) data = append(data, bytesMessage2...) data = append(data, babyjubB[:]...) data = append(data, chainIDBytes...) data = append(data, hermezAddress.Bytes()...) hash := crypto.Keccak256Hash(data) signature, err := crypto.Sign(hash.Bytes(), privateKey) require.NoError(t, err) // Ethereum adds 27 to v v := int(signature[64]) signature[64] = byte(v + 27) l1Tx := &L1Tx{ TokenID: 231, FromBJJ: pkComp, FromEthAddr: fromEthAddr, Amount: big.NewInt(0), DepositAmount: big.NewInt(0), } bytesCoordinatorL1, err := l1Tx.BytesCoordinatorTx(signature) require.NoError(t, err) l1txDecoded, err := L1CoordinatorTxFromBytes(bytesCoordinatorL1, chainID, hermezAddress) require.NoError(t, err) assert.Equal(t, l1Tx, l1txDecoded) bytesCoordinatorL12, err := l1txDecoded.BytesCoordinatorTx(signature) require.NoError(t, err) assert.Equal(t, bytesCoordinatorL1, bytesCoordinatorL12) // expect error if length!=68 _, err = L1CoordinatorTxFromBytes(bytesCoordinatorL1[:66], chainID, hermezAddress) require.NotNil(t, err) _, err = L1CoordinatorTxFromBytes([]byte{}, chainID, hermezAddress) require.NotNil(t, err) _, err = L1CoordinatorTxFromBytes(nil, chainID, hermezAddress) require.NotNil(t, err) } func TestL1CoordinatorTxByteParsersCompatibility(t *testing.T) { // Data from compatibility test var signature []byte r, err := hex.DecodeString("da71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3b") require.NoError(t, err) s, err := hex.DecodeString("186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8c") require.NoError(t, err) v, err := hex.DecodeString("1b") require.NoError(t, err) signature = append(signature, r[:]...) signature = append(signature, s[:]...) signature = append(signature, v[:]...) var pkComp babyjub.PublicKeyComp pkCompB, err := hex.DecodeString("a2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c") require.NoError(t, err) pkCompL := SwapEndianness(pkCompB) err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL))) require.NoError(t, err) // Data from the compatibility test require.NoError(t, err) l1Tx := &L1Tx{ TokenID: 231, FromBJJ: pkComp, } encodeData, err := l1Tx.BytesCoordinatorTx(signature) require.NoError(t, err) expected, err := utils.HexDecode("1b186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8cda71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3ba2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c000000e7") require.NoError(t, err) assert.Equal(t, expected, encodeData) } func TestL1TxID(t *testing.T) { // L1UserTx i64_1 := int64(1) i64_2 := int64(2) tx0 := L1Tx{ UserOrigin: true, ToForgeL1TxsNum: &i64_1, Position: 1, } err := tx0.SetID() require.NoError(t, err) assert.Equal(t, TxIDPrefixL1UserTx, tx0.TxID[0]) // differ ToForgeL1TxsNum tx1 := L1Tx{ UserOrigin: true, ToForgeL1TxsNum: &i64_2, Position: 1, } err = tx1.SetID() require.NoError(t, err) assert.NotEqual(t, tx0.TxID, tx1.TxID) // differ Position tx1 = L1Tx{ UserOrigin: true, ToForgeL1TxsNum: &i64_1, Position: 2, } err = tx1.SetID() require.NoError(t, err) assert.NotEqual(t, tx0.TxID, tx1.TxID) // L1CoordinatorTx bn1 := BatchNum(1) bn2 := BatchNum(2) tx0 = L1Tx{ UserOrigin: false, BatchNum: &bn1, Position: 1, } err = tx0.SetID() require.NoError(t, err) assert.Equal(t, TxIDPrefixL1CoordTx, tx0.TxID[0]) // differ BatchNum tx1 = L1Tx{ UserOrigin: false, BatchNum: &bn2, Position: 1, } err = tx1.SetID() require.NoError(t, err) assert.NotEqual(t, tx0.TxID, tx1.TxID) // differ Position tx1 = L1Tx{ UserOrigin: false, BatchNum: &bn1, Position: 2, } err = tx1.SetID() require.NoError(t, err) assert.NotEqual(t, tx0.TxID, tx1.TxID) }