github.com/tirogen/go-ethereum@v1.10.12-0.20221226051715-250cfede41b6/light/odr_test.go (about) 1 // Copyright 2016 The go-ethereum Authors 2 // This file is part of the go-ethereum library. 3 // 4 // The go-ethereum library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The go-ethereum library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. 16 17 package light 18 19 import ( 20 "bytes" 21 "context" 22 "errors" 23 "math/big" 24 "testing" 25 "time" 26 27 "github.com/tirogen/go-ethereum/common" 28 "github.com/tirogen/go-ethereum/common/math" 29 "github.com/tirogen/go-ethereum/consensus/ethash" 30 "github.com/tirogen/go-ethereum/core" 31 "github.com/tirogen/go-ethereum/core/rawdb" 32 "github.com/tirogen/go-ethereum/core/state" 33 "github.com/tirogen/go-ethereum/core/types" 34 "github.com/tirogen/go-ethereum/core/vm" 35 "github.com/tirogen/go-ethereum/crypto" 36 "github.com/tirogen/go-ethereum/ethdb" 37 "github.com/tirogen/go-ethereum/params" 38 "github.com/tirogen/go-ethereum/rlp" 39 ) 40 41 var ( 42 testBankKey, _ = crypto.HexToECDSA("b71c71a67e1177ad4e901695e1b4b9ee17ae16c6668d313eac2f96dbcda3f291") 43 testBankAddress = crypto.PubkeyToAddress(testBankKey.PublicKey) 44 testBankFunds = big.NewInt(1_000_000_000_000_000_000) 45 46 acc1Key, _ = crypto.HexToECDSA("8a1f9a8f95be41cd7ccb6168179afb4504aefe388d1e14474d32c45c72ce7b7a") 47 acc2Key, _ = crypto.HexToECDSA("49a7b37aa6f6645917e7b807e9d1c00d4fa71f18343b0d4122a4d2df64dd6fee") 48 acc1Addr = crypto.PubkeyToAddress(acc1Key.PublicKey) 49 acc2Addr = crypto.PubkeyToAddress(acc2Key.PublicKey) 50 51 testContractCode = common.Hex2Bytes("606060405260cc8060106000396000f360606040526000357c01000000000000000000000000000000000000000000000000000000009004806360cd2685146041578063c16431b914606b57603f565b005b6055600480803590602001909190505060a9565b6040518082815260200191505060405180910390f35b60886004808035906020019091908035906020019091905050608a565b005b80600060005083606481101560025790900160005b50819055505b5050565b6000600060005082606481101560025790900160005b5054905060c7565b91905056") 52 testContractAddr common.Address 53 ) 54 55 type testOdr struct { 56 OdrBackend 57 indexerConfig *IndexerConfig 58 sdb, ldb ethdb.Database 59 serverState state.Database 60 disable bool 61 } 62 63 func (odr *testOdr) Database() ethdb.Database { 64 return odr.ldb 65 } 66 67 var ErrOdrDisabled = errors.New("ODR disabled") 68 69 func (odr *testOdr) Retrieve(ctx context.Context, req OdrRequest) error { 70 if odr.disable { 71 return ErrOdrDisabled 72 } 73 switch req := req.(type) { 74 case *BlockRequest: 75 number := rawdb.ReadHeaderNumber(odr.sdb, req.Hash) 76 if number != nil { 77 req.Rlp = rawdb.ReadBodyRLP(odr.sdb, req.Hash, *number) 78 } 79 case *ReceiptsRequest: 80 number := rawdb.ReadHeaderNumber(odr.sdb, req.Hash) 81 if number != nil { 82 req.Receipts = rawdb.ReadRawReceipts(odr.sdb, req.Hash, *number) 83 } 84 case *TrieRequest: 85 var ( 86 err error 87 t state.Trie 88 ) 89 if len(req.Id.AccKey) > 0 { 90 t, err = odr.serverState.OpenStorageTrie(req.Id.StateRoot, common.BytesToHash(req.Id.AccKey), req.Id.Root) 91 } else { 92 t, err = odr.serverState.OpenTrie(req.Id.Root) 93 } 94 if err != nil { 95 panic(err) 96 } 97 nodes := NewNodeSet() 98 t.Prove(req.Key, 0, nodes) 99 req.Proof = nodes 100 case *CodeRequest: 101 req.Data = rawdb.ReadCode(odr.sdb, req.Hash) 102 } 103 req.StoreResult(odr.ldb) 104 return nil 105 } 106 107 func (odr *testOdr) IndexerConfig() *IndexerConfig { 108 return odr.indexerConfig 109 } 110 111 type odrTestFn func(ctx context.Context, db ethdb.Database, bc *core.BlockChain, lc *LightChain, bhash common.Hash) ([]byte, error) 112 113 func TestOdrGetBlockLes2(t *testing.T) { testChainOdr(t, 1, odrGetBlock) } 114 115 func odrGetBlock(ctx context.Context, db ethdb.Database, bc *core.BlockChain, lc *LightChain, bhash common.Hash) ([]byte, error) { 116 var block *types.Block 117 if bc != nil { 118 block = bc.GetBlockByHash(bhash) 119 } else { 120 block, _ = lc.GetBlockByHash(ctx, bhash) 121 } 122 if block == nil { 123 return nil, nil 124 } 125 rlp, _ := rlp.EncodeToBytes(block) 126 return rlp, nil 127 } 128 129 func TestOdrGetReceiptsLes2(t *testing.T) { testChainOdr(t, 1, odrGetReceipts) } 130 131 func odrGetReceipts(ctx context.Context, db ethdb.Database, bc *core.BlockChain, lc *LightChain, bhash common.Hash) ([]byte, error) { 132 var receipts types.Receipts 133 if bc != nil { 134 number := rawdb.ReadHeaderNumber(db, bhash) 135 if number != nil { 136 receipts = rawdb.ReadReceipts(db, bhash, *number, bc.Config()) 137 } 138 } else { 139 number := rawdb.ReadHeaderNumber(db, bhash) 140 if number != nil { 141 receipts, _ = GetBlockReceipts(ctx, lc.Odr(), bhash, *number) 142 } 143 } 144 if receipts == nil { 145 return nil, nil 146 } 147 rlp, _ := rlp.EncodeToBytes(receipts) 148 return rlp, nil 149 } 150 151 func TestOdrAccountsLes2(t *testing.T) { testChainOdr(t, 1, odrAccounts) } 152 153 func odrAccounts(ctx context.Context, db ethdb.Database, bc *core.BlockChain, lc *LightChain, bhash common.Hash) ([]byte, error) { 154 dummyAddr := common.HexToAddress("1234567812345678123456781234567812345678") 155 acc := []common.Address{testBankAddress, acc1Addr, acc2Addr, dummyAddr} 156 157 var st *state.StateDB 158 if bc == nil { 159 header := lc.GetHeaderByHash(bhash) 160 st = NewState(ctx, header, lc.Odr()) 161 } else { 162 header := bc.GetHeaderByHash(bhash) 163 st, _ = state.New(header.Root, bc.StateCache(), nil) 164 } 165 166 var res []byte 167 for _, addr := range acc { 168 bal := st.GetBalance(addr) 169 rlp, _ := rlp.EncodeToBytes(bal) 170 res = append(res, rlp...) 171 } 172 return res, st.Error() 173 } 174 175 func TestOdrContractCallLes2(t *testing.T) { testChainOdr(t, 1, odrContractCall) } 176 177 type callmsg struct { 178 types.Message 179 } 180 181 func (callmsg) CheckNonce() bool { return false } 182 183 func odrContractCall(ctx context.Context, db ethdb.Database, bc *core.BlockChain, lc *LightChain, bhash common.Hash) ([]byte, error) { 184 data := common.Hex2Bytes("60CD26850000000000000000000000000000000000000000000000000000000000000000") 185 config := params.TestChainConfig 186 187 var res []byte 188 for i := 0; i < 3; i++ { 189 data[35] = byte(i) 190 191 var ( 192 st *state.StateDB 193 header *types.Header 194 chain core.ChainContext 195 ) 196 if bc == nil { 197 chain = lc 198 header = lc.GetHeaderByHash(bhash) 199 st = NewState(ctx, header, lc.Odr()) 200 } else { 201 chain = bc 202 header = bc.GetHeaderByHash(bhash) 203 st, _ = state.New(header.Root, bc.StateCache(), nil) 204 } 205 206 // Perform read-only call. 207 st.SetBalance(testBankAddress, math.MaxBig256) 208 msg := callmsg{types.NewMessage(testBankAddress, &testContractAddr, 0, new(big.Int), 1000000, big.NewInt(params.InitialBaseFee), big.NewInt(params.InitialBaseFee), new(big.Int), data, nil, true)} 209 txContext := core.NewEVMTxContext(msg) 210 context := core.NewEVMBlockContext(header, chain, nil) 211 vmenv := vm.NewEVM(context, txContext, st, config, vm.Config{NoBaseFee: true}) 212 gp := new(core.GasPool).AddGas(math.MaxUint64) 213 result, _ := core.ApplyMessage(vmenv, msg, gp) 214 res = append(res, result.Return()...) 215 if st.Error() != nil { 216 return res, st.Error() 217 } 218 } 219 return res, nil 220 } 221 222 func testChainGen(i int, block *core.BlockGen) { 223 signer := types.HomesteadSigner{} 224 switch i { 225 case 0: 226 // In block 1, the test bank sends account #1 some ether. 227 tx, _ := types.SignTx(types.NewTransaction(block.TxNonce(testBankAddress), acc1Addr, big.NewInt(10_000_000_000_000_000), params.TxGas, block.BaseFee(), nil), signer, testBankKey) 228 block.AddTx(tx) 229 case 1: 230 // In block 2, the test bank sends some more ether to account #1. 231 // acc1Addr passes it on to account #2. 232 // acc1Addr creates a test contract. 233 tx1, _ := types.SignTx(types.NewTransaction(block.TxNonce(testBankAddress), acc1Addr, big.NewInt(1_000_000_000_000_000), params.TxGas, block.BaseFee(), nil), signer, testBankKey) 234 nonce := block.TxNonce(acc1Addr) 235 tx2, _ := types.SignTx(types.NewTransaction(nonce, acc2Addr, big.NewInt(1_000_000_000_000_000), params.TxGas, block.BaseFee(), nil), signer, acc1Key) 236 nonce++ 237 tx3, _ := types.SignTx(types.NewContractCreation(nonce, big.NewInt(0), 1000000, block.BaseFee(), testContractCode), signer, acc1Key) 238 testContractAddr = crypto.CreateAddress(acc1Addr, nonce) 239 block.AddTx(tx1) 240 block.AddTx(tx2) 241 block.AddTx(tx3) 242 case 2: 243 // Block 3 is empty but was mined by account #2. 244 block.SetCoinbase(acc2Addr) 245 block.SetExtra([]byte("yeehaw")) 246 data := common.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000000000000001") 247 tx, _ := types.SignTx(types.NewTransaction(block.TxNonce(testBankAddress), testContractAddr, big.NewInt(0), 100000, block.BaseFee(), data), signer, testBankKey) 248 block.AddTx(tx) 249 case 3: 250 // Block 4 includes blocks 2 and 3 as uncle headers (with modified extra data). 251 b2 := block.PrevBlock(1).Header() 252 b2.Extra = []byte("foo") 253 block.AddUncle(b2) 254 b3 := block.PrevBlock(2).Header() 255 b3.Extra = []byte("foo") 256 block.AddUncle(b3) 257 data := common.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000000000002") 258 tx, _ := types.SignTx(types.NewTransaction(block.TxNonce(testBankAddress), testContractAddr, big.NewInt(0), 100000, block.BaseFee(), data), signer, testBankKey) 259 block.AddTx(tx) 260 } 261 } 262 263 func testChainOdr(t *testing.T, protocol int, fn odrTestFn) { 264 var ( 265 sdb = rawdb.NewMemoryDatabase() 266 ldb = rawdb.NewMemoryDatabase() 267 gspec = &core.Genesis{ 268 Config: params.TestChainConfig, 269 Alloc: core.GenesisAlloc{testBankAddress: {Balance: testBankFunds}}, 270 BaseFee: big.NewInt(params.InitialBaseFee), 271 } 272 ) 273 // Assemble the test environment 274 blockchain, _ := core.NewBlockChain(sdb, nil, gspec, nil, ethash.NewFullFaker(), vm.Config{}, nil, nil) 275 _, gchain, _ := core.GenerateChainWithGenesis(gspec, ethash.NewFaker(), 4, testChainGen) 276 if _, err := blockchain.InsertChain(gchain); err != nil { 277 t.Fatal(err) 278 } 279 280 gspec.MustCommit(ldb) 281 odr := &testOdr{sdb: sdb, ldb: ldb, serverState: blockchain.StateCache(), indexerConfig: TestClientIndexerConfig} 282 lightchain, err := NewLightChain(odr, gspec.Config, ethash.NewFullFaker(), nil) 283 if err != nil { 284 t.Fatal(err) 285 } 286 headers := make([]*types.Header, len(gchain)) 287 for i, block := range gchain { 288 headers[i] = block.Header() 289 } 290 if _, err := lightchain.InsertHeaderChain(headers, 1); err != nil { 291 t.Fatal(err) 292 } 293 294 test := func(expFail int) { 295 for i := uint64(0); i <= blockchain.CurrentHeader().Number.Uint64(); i++ { 296 bhash := rawdb.ReadCanonicalHash(sdb, i) 297 b1, err := fn(NoOdr, sdb, blockchain, nil, bhash) 298 if err != nil { 299 t.Fatalf("error in full-node test for block %d: %v", i, err) 300 } 301 302 ctx, cancel := context.WithTimeout(context.Background(), 200*time.Millisecond) 303 defer cancel() 304 305 exp := i < uint64(expFail) 306 b2, err := fn(ctx, ldb, nil, lightchain, bhash) 307 if err != nil && exp { 308 t.Errorf("error in ODR test for block %d: %v", i, err) 309 } 310 311 eq := bytes.Equal(b1, b2) 312 if exp && !eq { 313 t.Errorf("ODR test output for block %d doesn't match full node", i) 314 } 315 } 316 } 317 318 // expect retrievals to fail (except genesis block) without a les peer 319 t.Log("checking without ODR") 320 odr.disable = true 321 test(1) 322 323 // expect all retrievals to pass with ODR enabled 324 t.Log("checking with ODR") 325 odr.disable = false 326 test(len(gchain)) 327 328 // still expect all retrievals to pass, now data should be cached locally 329 t.Log("checking without ODR, should be cached") 330 odr.disable = true 331 test(len(gchain)) 332 }