470 lines
15 KiB
Go
470 lines
15 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2019 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package test_helpers
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"crypto/elliptic"
|
|
"crypto/rand"
|
|
"math/big"
|
|
|
|
"github.com/cerc-io/plugeth-statediff/indexer/ipld"
|
|
"github.com/cerc-io/plugeth-statediff/indexer/models"
|
|
sdtypes "github.com/cerc-io/plugeth-statediff/types"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
"github.com/ethereum/go-ethereum/trie"
|
|
"github.com/holiman/uint256"
|
|
"github.com/ipfs/go-cid"
|
|
|
|
"github.com/cerc-io/ipld-eth-server/v5/pkg/eth"
|
|
"github.com/cerc-io/ipld-eth-server/v5/pkg/log"
|
|
)
|
|
|
|
// Test variables
|
|
var (
|
|
// block data
|
|
BlockNumber1 = int64(1)
|
|
BlockTime1 = uint64(0)
|
|
MockHeader = types.Header{
|
|
Time: 0,
|
|
Number: big.NewInt(BlockNumber1),
|
|
Root: common.HexToHash("0x0"),
|
|
TxHash: common.HexToHash("0x0"),
|
|
ReceiptHash: common.HexToHash("0x0"),
|
|
Difficulty: big.NewInt(5000000),
|
|
Extra: []byte{},
|
|
}
|
|
MockTransactions, MockReceipts, SenderAddr = createLegacyTransactionsAndReceipts()
|
|
MockWithdrawals = types.Withdrawals{
|
|
{Index: 0, Validator: 1, Address: Address, Amount: 1000000000},
|
|
{Index: 1, Validator: 5, Address: AnotherAddress, Amount: 2000000000},
|
|
}
|
|
MockUncles = []*types.Header{
|
|
{
|
|
Time: 1,
|
|
Number: big.NewInt(BlockNumber1 + 1),
|
|
Root: common.HexToHash("0x1"),
|
|
TxHash: common.HexToHash("0x1"),
|
|
ReceiptHash: common.HexToHash("0x1"),
|
|
Difficulty: big.NewInt(500001),
|
|
Extra: []byte{},
|
|
ParentHash: Genesis.Hash(),
|
|
},
|
|
{
|
|
Time: 2,
|
|
Number: big.NewInt(BlockNumber1 + 2),
|
|
Root: common.HexToHash("0x2"),
|
|
TxHash: common.HexToHash("0x2"),
|
|
ReceiptHash: common.HexToHash("0x2"),
|
|
Difficulty: big.NewInt(500002),
|
|
Extra: []byte{},
|
|
ParentHash: Genesis.Hash(),
|
|
},
|
|
}
|
|
MockBlock = createNewBlock(&MockHeader, MockTransactions, MockUncles, MockReceipts, nil, trie.NewEmpty(nil))
|
|
MockChildHeader = types.Header{
|
|
Time: 0,
|
|
Number: big.NewInt(BlockNumber1 + 1),
|
|
Root: common.HexToHash("0x0"),
|
|
TxHash: common.HexToHash("0x0"),
|
|
ReceiptHash: common.HexToHash("0x0"),
|
|
Difficulty: big.NewInt(5000001),
|
|
Extra: []byte{},
|
|
ParentHash: MockBlock.Header().Hash(),
|
|
}
|
|
MockChild = types.NewBlock(&MockChildHeader, MockTransactions, MockUncles, MockReceipts, trie.NewEmpty(nil))
|
|
Address = common.HexToAddress("0xaE9BEa628c4Ce503DcFD7E305CaB4e29E7476592")
|
|
AnotherAddress = common.HexToAddress("0xaE9BEa628c4Ce503DcFD7E305CaB4e29E7476593")
|
|
AnotherAddress1 = common.HexToAddress("0xaE9BEa628c4Ce503DcFD7E305CaB4e29E7476594")
|
|
AnotherAddress2 = common.HexToAddress("0xaE9BEa628c4Ce503DcFD7E305CaB4e29E7476596")
|
|
ContractAddress = crypto.CreateAddress(SenderAddr, MockTransactions[2].Nonce())
|
|
mockTopic11 = common.HexToHash("0x04")
|
|
mockTopic12 = common.HexToHash("0x06")
|
|
mockTopic21 = common.HexToHash("0x05")
|
|
mockTopic22 = common.HexToHash("0x07")
|
|
mockTopic31 = common.HexToHash("0x08")
|
|
mockTopic41 = common.HexToHash("0x09")
|
|
mockTopic42 = common.HexToHash("0x0a")
|
|
mockTopic43 = common.HexToHash("0x0b")
|
|
mockTopic51 = common.HexToHash("0x0c")
|
|
mockTopic61 = common.HexToHash("0x0d")
|
|
MockLog1 = &types.Log{
|
|
Address: Address,
|
|
Topics: []common.Hash{mockTopic11, mockTopic12},
|
|
Data: []byte{},
|
|
BlockNumber: uint64(BlockNumber1),
|
|
TxIndex: 0,
|
|
Index: 0,
|
|
}
|
|
MockLog2 = &types.Log{
|
|
Address: AnotherAddress,
|
|
Topics: []common.Hash{mockTopic21, mockTopic22},
|
|
Data: []byte{},
|
|
BlockNumber: uint64(BlockNumber1),
|
|
TxIndex: 1,
|
|
Index: 1,
|
|
}
|
|
MockLog3 = &types.Log{
|
|
Address: AnotherAddress1,
|
|
Topics: []common.Hash{mockTopic31},
|
|
Data: []byte{},
|
|
BlockNumber: uint64(BlockNumber1),
|
|
TxIndex: 2,
|
|
Index: 2,
|
|
}
|
|
|
|
MockLog4 = &types.Log{
|
|
Address: AnotherAddress1,
|
|
Topics: []common.Hash{mockTopic41, mockTopic42, mockTopic43},
|
|
Data: []byte{},
|
|
BlockNumber: uint64(BlockNumber1),
|
|
TxIndex: 2,
|
|
Index: 3,
|
|
}
|
|
MockLog5 = &types.Log{
|
|
Address: AnotherAddress1,
|
|
Topics: []common.Hash{mockTopic51},
|
|
Data: []byte{},
|
|
BlockNumber: uint64(BlockNumber1),
|
|
TxIndex: 2,
|
|
Index: 4,
|
|
}
|
|
MockLog6 = &types.Log{
|
|
Address: AnotherAddress2,
|
|
Topics: []common.Hash{mockTopic61},
|
|
Data: []byte{},
|
|
BlockNumber: uint64(BlockNumber1),
|
|
TxIndex: 3,
|
|
Index: 5,
|
|
}
|
|
|
|
rctCIDs, _ = getReceiptCIDs(MockReceipts)
|
|
Rct1CID = rctCIDs[0]
|
|
Rct4CID = rctCIDs[3]
|
|
MockTrxMeta = []models.TxModel{
|
|
{
|
|
CID: "", // This is empty until we go to publish to ipfs
|
|
Src: SenderAddr.Hex(),
|
|
Dst: Address.String(),
|
|
Index: 0,
|
|
TxHash: MockTransactions[0].Hash().String(),
|
|
},
|
|
{
|
|
CID: "",
|
|
Src: SenderAddr.Hex(),
|
|
Dst: AnotherAddress.String(),
|
|
Index: 1,
|
|
TxHash: MockTransactions[1].Hash().String(),
|
|
},
|
|
{
|
|
CID: "",
|
|
Src: SenderAddr.Hex(),
|
|
Dst: "",
|
|
Index: 2,
|
|
TxHash: MockTransactions[2].Hash().String(),
|
|
},
|
|
{
|
|
CID: "",
|
|
Src: SenderAddr.Hex(),
|
|
Dst: "",
|
|
Index: 3,
|
|
TxHash: MockTransactions[3].Hash().String(),
|
|
},
|
|
}
|
|
MockRctMeta = []models.ReceiptModel{
|
|
{
|
|
CID: "",
|
|
Contract: "",
|
|
},
|
|
{
|
|
CID: "",
|
|
Contract: "",
|
|
},
|
|
{
|
|
CID: "",
|
|
Contract: ContractAddress.String(),
|
|
},
|
|
{
|
|
CID: "",
|
|
Contract: "",
|
|
},
|
|
}
|
|
|
|
// statediff data
|
|
storageLocation = common.HexToHash("0")
|
|
StorageLeafKey = crypto.Keccak256Hash(storageLocation[:]).Bytes()
|
|
StorageValue = crypto.Keccak256([]byte{1, 2, 3, 4, 5})
|
|
StoragePartialPath = common.Hex2Bytes("20290decd9548b62a8d60345a988386fc84ba6bc95484008f6362f93160ef3e563")
|
|
StorageLeafNode, _ = rlp.EncodeToBytes(&[]interface{}{
|
|
StoragePartialPath,
|
|
StorageValue,
|
|
})
|
|
|
|
ContractRoot = "0x821e2556a290c86405f8160a2d662042a431ba456b9db265c79bb837c04be5f0"
|
|
contractPath = common.Bytes2Hex([]byte{'\x06'})
|
|
ContractLeafKey = crypto.Keccak256(ContractAddress[:])
|
|
ContractAccount = types.StateAccount{
|
|
Nonce: uint64(1),
|
|
Balance: uint256.NewInt(0),
|
|
CodeHash: CodeHash.Bytes(),
|
|
Root: common.HexToHash(ContractRoot),
|
|
}
|
|
ContractAccountRLP, _ = rlp.EncodeToBytes(&ContractAccount)
|
|
ContractPartialPath = common.Hex2Bytes("3114658a74d9cc9f7acf2c5cd696c3494d7c344d78bfec3add0d91ec4e8d1c45")
|
|
ContractLeafNode, _ = rlp.EncodeToBytes(&[]interface{}{
|
|
ContractPartialPath,
|
|
ContractAccountRLP,
|
|
})
|
|
|
|
nonce0 = uint64(0)
|
|
AccountBalance = uint256.NewInt(1000)
|
|
AccountRoot = "0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421"
|
|
AccountCodeHash = common.HexToHash("0xc5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470")
|
|
AccountAddresss = common.HexToAddress("0x0D3ab14BBaD3D99F4203bd7a11aCB94882050E7e")
|
|
AccountLeafKey = crypto.Keccak256(AccountAddresss[:])
|
|
Account = types.StateAccount{
|
|
Nonce: nonce0,
|
|
Balance: AccountBalance,
|
|
CodeHash: AccountCodeHash.Bytes(),
|
|
Root: common.HexToHash(AccountRoot),
|
|
}
|
|
AccountRLP, _ = rlp.EncodeToBytes(&Account)
|
|
AccountPartialPath = common.Hex2Bytes("3957f3e2f04a0764c3a0491b175f69926da61efbcc8f61fa1455fd2d2b4cdd45")
|
|
AccountLeafNode, _ = rlp.EncodeToBytes(&[]interface{}{
|
|
AccountPartialPath,
|
|
AccountRLP,
|
|
})
|
|
|
|
MockStateNodes = []sdtypes.StateLeafNode{
|
|
{
|
|
AccountWrapper: sdtypes.AccountWrapper{
|
|
Account: &ContractAccount,
|
|
LeafKey: ContractLeafKey,
|
|
CID: ipld.Keccak256ToCid(ipld.MEthStateTrie, crypto.Keccak256(ContractLeafNode)).String(),
|
|
},
|
|
StorageDiff: []sdtypes.StorageLeafNode{
|
|
{
|
|
LeafKey: StorageLeafKey,
|
|
Value: StorageValue,
|
|
CID: ipld.Keccak256ToCid(ipld.MEthStorageTrie, crypto.Keccak256(StorageLeafNode)).String(),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
AccountWrapper: sdtypes.AccountWrapper{
|
|
Account: &Account,
|
|
LeafKey: AccountLeafKey,
|
|
CID: ipld.Keccak256ToCid(ipld.MEthStateTrie, crypto.Keccak256(AccountLeafNode)).String(),
|
|
},
|
|
},
|
|
}
|
|
MockStorageNodes = map[string][]sdtypes.StorageLeafNode{
|
|
contractPath: {
|
|
{
|
|
LeafKey: StorageLeafKey,
|
|
Value: StorageValue,
|
|
CID: ipld.Keccak256ToCid(ipld.MEthStorageTrie, crypto.Keccak256(StorageLeafNode)).String(),
|
|
},
|
|
},
|
|
}
|
|
|
|
MockConvertedPayload = eth.ConvertedPayload{
|
|
TotalDifficulty: MockBlock.Difficulty(),
|
|
Block: MockBlock,
|
|
Receipts: MockReceipts,
|
|
TxMetaData: MockTrxMeta,
|
|
ReceiptMetaData: MockRctMeta,
|
|
StorageNodes: MockStorageNodes,
|
|
StateNodes: MockStateNodes,
|
|
}
|
|
|
|
LondonBlockNum = (BlockNumber1 + 2)
|
|
LondonBlockTime = BlockTime1 + 1
|
|
MockLondonHeader = types.Header{
|
|
Time: LondonBlockTime,
|
|
Number: big.NewInt(LondonBlockNum),
|
|
Root: common.HexToHash("0x00"),
|
|
Difficulty: big.NewInt(5000000),
|
|
Extra: []byte{},
|
|
BaseFee: big.NewInt(params.InitialBaseFee),
|
|
}
|
|
|
|
MockLondonTransactions, MockLondonReceipts, _ = createDynamicTransactionsAndReceipts(big.NewInt(LondonBlockNum), LondonBlockTime)
|
|
MockLondonUncles = []*types.Header{
|
|
{
|
|
Time: 1,
|
|
Number: big.NewInt(BlockNumber1 + 1),
|
|
ParentHash: common.HexToHash("0x2"),
|
|
Root: common.HexToHash("0x1"),
|
|
TxHash: common.HexToHash("0x1"),
|
|
ReceiptHash: common.HexToHash("0x1"),
|
|
Difficulty: big.NewInt(500001),
|
|
Extra: []byte{},
|
|
},
|
|
{
|
|
Time: 2,
|
|
Number: big.NewInt(BlockNumber1 + 1),
|
|
ParentHash: common.HexToHash("0x1"),
|
|
Root: common.HexToHash("0x2"),
|
|
TxHash: common.HexToHash("0x2"),
|
|
ReceiptHash: common.HexToHash("0x2"),
|
|
Difficulty: big.NewInt(500002),
|
|
Extra: []byte{},
|
|
},
|
|
}
|
|
MockLondonBlock = createNewBlock(&MockLondonHeader, MockLondonTransactions, MockLondonUncles, MockLondonReceipts, MockWithdrawals, trie.NewEmpty(nil))
|
|
)
|
|
|
|
func createNewBlock(header *types.Header, txs []*types.Transaction, uncles []*types.Header, receipts []*types.Receipt, withdrawals types.Withdrawals, hasher types.TrieHasher) *types.Block {
|
|
block := types.NewBlockWithWithdrawals(header, txs, uncles, receipts, withdrawals, hasher)
|
|
bHash := block.Hash()
|
|
for _, r := range receipts {
|
|
for _, l := range r.Logs {
|
|
l.BlockHash = bHash
|
|
}
|
|
}
|
|
return block
|
|
}
|
|
|
|
// createDynamicTransactionsAndReceipts is a helper function to generate signed mock transactions and mock receipts with mock logs
|
|
func createDynamicTransactionsAndReceipts(blockNumber *big.Int, blockTime uint64) (types.Transactions, types.Receipts, common.Address) {
|
|
// make transactions
|
|
config := *params.TestChainConfig
|
|
config.LondonBlock = blockNumber
|
|
trx1 := types.NewTx(&types.DynamicFeeTx{
|
|
ChainID: config.ChainID,
|
|
Nonce: 1,
|
|
GasTipCap: big.NewInt(50),
|
|
GasFeeCap: big.NewInt(100),
|
|
Gas: 50,
|
|
To: &Address,
|
|
Value: big.NewInt(1000),
|
|
Data: []byte{},
|
|
})
|
|
|
|
transactionSigner := types.MakeSigner(&config, blockNumber, blockTime)
|
|
mockCurve := elliptic.P256()
|
|
mockPrvKey, err := ecdsa.GenerateKey(mockCurve, rand.Reader)
|
|
if err != nil {
|
|
log.Fatal(err.Error())
|
|
}
|
|
signedTrx1, err := types.SignTx(trx1, transactionSigner, mockPrvKey)
|
|
if err != nil {
|
|
log.Fatal(err.Error())
|
|
}
|
|
|
|
senderAddr, err := types.Sender(transactionSigner, signedTrx1) // same for both trx
|
|
if err != nil {
|
|
log.Fatal(err.Error())
|
|
}
|
|
|
|
// make receipts
|
|
// TODO: Change the receipt type to DynamicFeeTxType once this PR is merged.
|
|
// https://github.com/ethereum/go-ethereum/pull/22806
|
|
mockReceipt1 := &types.Receipt{
|
|
Type: types.DynamicFeeTxType,
|
|
PostState: common.HexToHash("0x0").Bytes(),
|
|
Status: types.ReceiptStatusSuccessful,
|
|
CumulativeGasUsed: 50,
|
|
Logs: []*types.Log{},
|
|
TxHash: signedTrx1.Hash(),
|
|
}
|
|
|
|
return types.Transactions{signedTrx1}, types.Receipts{mockReceipt1}, senderAddr
|
|
}
|
|
|
|
// createLegacyTransactionsAndReceipts is a helper function to generate signed mock transactions and mock receipts with mock logs
|
|
func createLegacyTransactionsAndReceipts() (types.Transactions, types.Receipts, common.Address) {
|
|
// make transactions
|
|
trx1 := types.NewTransaction(0, Address, big.NewInt(1000), 50, big.NewInt(100), []byte{})
|
|
trx2 := types.NewTransaction(1, AnotherAddress, big.NewInt(2000), 100, big.NewInt(200), []byte{})
|
|
trx3 := types.NewContractCreation(2, big.NewInt(1500), 75, big.NewInt(150), ContractCode)
|
|
trx4 := types.NewTransaction(3, AnotherAddress1, big.NewInt(2000), 100, big.NewInt(200), []byte{})
|
|
transactionSigner := types.MakeSigner(params.MainnetChainConfig, big.NewInt(BlockNumber1), 0)
|
|
mockCurve := elliptic.P256()
|
|
mockPrvKey, err := ecdsa.GenerateKey(mockCurve, rand.Reader)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
signedTrx1, err := types.SignTx(trx1, transactionSigner, mockPrvKey)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
signedTrx2, err := types.SignTx(trx2, transactionSigner, mockPrvKey)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
signedTrx3, err := types.SignTx(trx3, transactionSigner, mockPrvKey)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
signedTrx4, err := types.SignTx(trx4, transactionSigner, mockPrvKey)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
SenderAddr, err := types.Sender(transactionSigner, signedTrx1) // same for both trx
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
// make receipts
|
|
mockReceipt1 := types.NewReceipt(nil, false, 50)
|
|
|
|
hash1 := signedTrx1.Hash()
|
|
MockLog1.TxHash = hash1
|
|
|
|
mockReceipt1.Logs = []*types.Log{MockLog1}
|
|
mockReceipt1.TxHash = hash1
|
|
mockReceipt1.GasUsed = mockReceipt1.CumulativeGasUsed
|
|
|
|
mockReceipt2 := types.NewReceipt(common.HexToHash("0x1").Bytes(), false, 100)
|
|
hash2 := signedTrx2.Hash()
|
|
MockLog2.TxHash = hash2
|
|
|
|
mockReceipt2.Logs = []*types.Log{MockLog2}
|
|
mockReceipt2.TxHash = hash2
|
|
mockReceipt2.GasUsed = mockReceipt2.CumulativeGasUsed - mockReceipt1.CumulativeGasUsed
|
|
|
|
mockReceipt3 := types.NewReceipt(common.HexToHash("0x2").Bytes(), false, 175)
|
|
mockReceipt3.Logs = []*types.Log{MockLog3, MockLog4, MockLog5}
|
|
mockReceipt3.TxHash = signedTrx3.Hash()
|
|
mockReceipt3.GasUsed = mockReceipt3.CumulativeGasUsed - mockReceipt2.CumulativeGasUsed
|
|
|
|
// Receipt with failed status.
|
|
mockReceipt4 := types.NewReceipt(nil, true, 250)
|
|
mockReceipt4.Logs = []*types.Log{MockLog6}
|
|
mockReceipt4.TxHash = signedTrx4.Hash()
|
|
mockReceipt4.GasUsed = mockReceipt4.CumulativeGasUsed - mockReceipt3.CumulativeGasUsed
|
|
|
|
return types.Transactions{signedTrx1, signedTrx2, signedTrx3, signedTrx4}, types.Receipts{mockReceipt1, mockReceipt2, mockReceipt3, mockReceipt4}, SenderAddr
|
|
}
|
|
|
|
func getReceiptCIDs(rcts []*types.Receipt) ([]cid.Cid, error) {
|
|
cids := make([]cid.Cid, len(rcts))
|
|
for i, rct := range rcts {
|
|
ethRct, err := ipld.EncodeReceipt(rct)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cids[i] = ethRct.Cid()
|
|
}
|
|
return cids, nil
|
|
}
|