175 lines
9.0 KiB
Go
175 lines
9.0 KiB
Go
// Copyright 2019 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library 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 Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package test_helpers
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/consensus/ethash"
|
|
"github.com/ethereum/go-ethereum/core"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/core/vm"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
"github.com/ethereum/go-ethereum/triedb"
|
|
|
|
"github.com/cerc-io/plugeth-statediff/utils"
|
|
)
|
|
|
|
func GenesisBlockForTesting(db ethdb.Database, addr common.Address, balance, baseFee *big.Int, initialGasLimit uint64) *types.Block {
|
|
alloc := map[common.Address]types.Account{
|
|
addr: types.Account{Balance: balance}}
|
|
g := core.Genesis{
|
|
Alloc: alloc,
|
|
BaseFee: baseFee,
|
|
}
|
|
if initialGasLimit != 0 {
|
|
g.GasLimit = initialGasLimit
|
|
}
|
|
return g.MustCommit(db, triedb.NewDatabase(db, nil))
|
|
}
|
|
|
|
// MakeChain creates a chain of n blocks starting at and including parent.
|
|
// the returned hash chain is ordered head->parent.
|
|
func MakeChain(n int, parent *types.Block, chainGen func(int, *core.BlockGen)) ([]*types.Block, *core.BlockChain) {
|
|
config := params.TestChainConfig
|
|
blocks, _ := core.GenerateChain(config, parent, ethash.NewFaker(), Testdb, n, chainGen)
|
|
chain, _ := core.NewBlockChain(Testdb, nil, nil, nil, ethash.NewFaker(), vm.Config{}, nil, nil)
|
|
return blocks, chain
|
|
}
|
|
|
|
func TestSelfDestructChainGen(i int, block *core.BlockGen) {
|
|
signer := types.HomesteadSigner{}
|
|
switch i {
|
|
case 0:
|
|
// Block 1 is mined by TestBankAddress
|
|
// TestBankAddress creates a new contract
|
|
block.SetCoinbase(TestBankAddress)
|
|
tx, _ := types.SignTx(types.NewContractCreation(0, big.NewInt(0), 1000000, big.NewInt(params.GWei), ContractCode), signer, TestBankKey)
|
|
ContractAddr = crypto.CreateAddress(TestBankAddress, 0)
|
|
block.AddTx(tx)
|
|
case 1:
|
|
// Block 2 is mined by TestBankAddress
|
|
// TestBankAddress self-destructs the contract
|
|
block.SetCoinbase(TestBankAddress)
|
|
data := utils.Hex2Bytes("43D726D6")
|
|
tx, _ := types.SignTx(types.NewTransaction(1, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.GWei), data), signer, TestBankKey)
|
|
block.AddTx(tx)
|
|
}
|
|
}
|
|
|
|
func TestChainGen(i int, block *core.BlockGen) {
|
|
signer := types.HomesteadSigner{}
|
|
switch i {
|
|
case 0:
|
|
// In block 1, the test bank sends account #1 some ether.
|
|
tx, _ := types.SignTx(types.NewTransaction(block.TxNonce(TestBankAddress), Account1Addr, big.NewInt(BalanceChange10000), params.TxGas, big.NewInt(params.GWei), nil), signer, TestBankKey)
|
|
block.AddTx(tx)
|
|
case 1:
|
|
// In block 2, the test bank sends some more ether to account #1.
|
|
// Account1Addr passes it on to account #2.
|
|
// Account1Addr creates a test contract.
|
|
tx1, _ := types.SignTx(types.NewTransaction(block.TxNonce(TestBankAddress), Account1Addr, big.NewInt(BalanceChange1Ether), params.TxGas, big.NewInt(params.GWei), nil), signer, TestBankKey)
|
|
nonce := block.TxNonce(Account1Addr)
|
|
tx2, _ := types.SignTx(types.NewTransaction(nonce, Account2Addr, big.NewInt(BalanceChange1000), params.TxGas, big.NewInt(params.GWei), nil), signer, Account1Key)
|
|
nonce++
|
|
tx3, _ := types.SignTx(types.NewContractCreation(nonce, big.NewInt(0), ContractGasLimit, big.NewInt(params.GWei), ContractCode), signer, Account1Key)
|
|
ContractAddr = crypto.CreateAddress(Account1Addr, nonce)
|
|
block.AddTx(tx1)
|
|
block.AddTx(tx2)
|
|
block.AddTx(tx3)
|
|
case 2:
|
|
// Block 3 has a single tx from the bankAccount to the contract, that transfers no value
|
|
// Block 3 is mined by Account2Addr
|
|
block.SetCoinbase(Account2Addr)
|
|
//put function: c16431b9
|
|
//close function: 43d726d6
|
|
data := utils.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000000000003")
|
|
tx, _ := types.SignTx(types.NewTransaction(block.TxNonce(TestBankAddress), ContractAddr, big.NewInt(0), params.TxGasContractCreation, big.NewInt(params.GWei), data), signer, TestBankKey)
|
|
block.AddTx(tx)
|
|
case 3:
|
|
// Block 4 has three txs from bankAccount to the contract, that transfer no value
|
|
// Two set the two original slot positions to 0 and one sets another position to a new value
|
|
// Block 4 is mined by Account2Addr
|
|
block.SetCoinbase(Account2Addr)
|
|
data1 := utils.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000")
|
|
data2 := utils.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000000000000")
|
|
data3 := utils.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000000000000009")
|
|
|
|
nonce := block.TxNonce(TestBankAddress)
|
|
tx1, _ := types.SignTx(types.NewTransaction(nonce, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data1), signer, TestBankKey)
|
|
nonce++
|
|
tx2, _ := types.SignTx(types.NewTransaction(nonce, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data2), signer, TestBankKey)
|
|
nonce++
|
|
tx3, _ := types.SignTx(types.NewTransaction(nonce, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data3), signer, TestBankKey)
|
|
block.AddTx(tx1)
|
|
block.AddTx(tx2)
|
|
block.AddTx(tx3)
|
|
case 4:
|
|
// Block 5 has one tx from bankAccount to the contract, that transfers no value
|
|
// It sets the one storage value to zero and the other to new value.
|
|
// Block 5 is mined by Account1Addr
|
|
block.SetCoinbase(Account1Addr)
|
|
data1 := utils.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000000000000000")
|
|
data2 := utils.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000000000003")
|
|
nonce := block.TxNonce(TestBankAddress)
|
|
tx1, _ := types.SignTx(types.NewTransaction(nonce, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data1), signer, TestBankKey)
|
|
nonce++
|
|
tx2, _ := types.SignTx(types.NewTransaction(nonce, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data2), signer, TestBankKey)
|
|
block.AddTx(tx1)
|
|
block.AddTx(tx2)
|
|
case 5:
|
|
// Block 6 has a tx from Account1Key which self-destructs the contract, it transfers no value
|
|
// Block 6 is mined by Account2Addr
|
|
block.SetCoinbase(Account2Addr)
|
|
data := utils.Hex2Bytes("43D726D6")
|
|
tx, _ := types.SignTx(types.NewTransaction(block.TxNonce(Account1Addr), ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data), signer, Account1Key)
|
|
block.AddTx(tx)
|
|
}
|
|
}
|
|
|
|
func TestChainGenWithInternalLeafNode(i int, block *core.BlockGen) {
|
|
signer := types.HomesteadSigner{}
|
|
switch i {
|
|
case 0:
|
|
// In block 1, the test bank sends account #1 some ether.
|
|
tx, _ := types.SignTx(types.NewTransaction(block.TxNonce(TestBankAddress), Account1Addr, BalanceChangeBIG.ToBig(), params.TxGas, big.NewInt(params.InitialBaseFee), nil), signer, TestBankKey)
|
|
block.AddTx(tx)
|
|
case 1:
|
|
// In block 2 Account1Addr creates a test contract.
|
|
nonce := block.TxNonce(Account1Addr)
|
|
tx1, _ := types.SignTx(types.NewContractCreation(block.TxNonce(Account1Addr), big.NewInt(0), ContractForInternalLeafNodeGasLimit, big.NewInt(params.InitialBaseFee), ContractCodeForInternalLeafNode), signer, Account1Key)
|
|
ContractAddr = crypto.CreateAddress(Account1Addr, nonce)
|
|
block.AddTx(tx1)
|
|
case 2:
|
|
// Block 3 has two transactions which set slots 223 and 648 with small values
|
|
// The goal here is to induce a branch node with an internalized leaf node
|
|
block.SetCoinbase(TestBankAddress)
|
|
data1 := utils.Hex2Bytes("C16431B90000000000000000000000000000000000000000000000000000000000009dab0000000000000000000000000000000000000000000000000000000000000001")
|
|
data2 := utils.Hex2Bytes("C16431B90000000000000000000000000000000000000000000000000000000000019c5d0000000000000000000000000000000000000000000000000000000000000002")
|
|
|
|
nonce := block.TxNonce(TestBankAddress)
|
|
tx1, _ := types.SignTx(types.NewTransaction(nonce, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data1), signer, TestBankKey)
|
|
nonce++
|
|
tx2, _ := types.SignTx(types.NewTransaction(nonce, ContractAddr, big.NewInt(0), 100000, big.NewInt(params.InitialBaseFee), data2), signer, TestBankKey)
|
|
block.AddTx(tx1)
|
|
block.AddTx(tx2)
|
|
}
|
|
}
|