test_helper.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666
  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. // This file contains some shares testing functionality, common to multiple
  17. // different files and modules being tested. Client based network and Server
  18. // based network can be created easily with available APIs.
  19. package les
  20. import (
  21. "context"
  22. "crypto/rand"
  23. "fmt"
  24. "math/big"
  25. "sync/atomic"
  26. "testing"
  27. "time"
  28. "github.com/ethereum/go-ethereum/accounts/abi/bind"
  29. "github.com/ethereum/go-ethereum/accounts/abi/bind/backends"
  30. "github.com/ethereum/go-ethereum/common"
  31. "github.com/ethereum/go-ethereum/common/mclock"
  32. "github.com/ethereum/go-ethereum/consensus/ethash"
  33. "github.com/ethereum/go-ethereum/contracts/checkpointoracle/contract"
  34. "github.com/ethereum/go-ethereum/core"
  35. "github.com/ethereum/go-ethereum/core/forkid"
  36. "github.com/ethereum/go-ethereum/core/rawdb"
  37. "github.com/ethereum/go-ethereum/core/types"
  38. "github.com/ethereum/go-ethereum/crypto"
  39. "github.com/ethereum/go-ethereum/eth/ethconfig"
  40. "github.com/ethereum/go-ethereum/ethdb"
  41. "github.com/ethereum/go-ethereum/event"
  42. "github.com/ethereum/go-ethereum/les/checkpointoracle"
  43. "github.com/ethereum/go-ethereum/les/flowcontrol"
  44. "github.com/ethereum/go-ethereum/light"
  45. "github.com/ethereum/go-ethereum/p2p"
  46. "github.com/ethereum/go-ethereum/p2p/enode"
  47. "github.com/ethereum/go-ethereum/p2p/nodestate"
  48. "github.com/ethereum/go-ethereum/params"
  49. )
  50. var (
  51. bankKey, _ = crypto.GenerateKey()
  52. bankAddr = crypto.PubkeyToAddress(bankKey.PublicKey)
  53. bankFunds = big.NewInt(1000000000000000000)
  54. userKey1, _ = crypto.GenerateKey()
  55. userKey2, _ = crypto.GenerateKey()
  56. userAddr1 = crypto.PubkeyToAddress(userKey1.PublicKey)
  57. userAddr2 = crypto.PubkeyToAddress(userKey2.PublicKey)
  58. testContractAddr common.Address
  59. testContractCode = common.Hex2Bytes("606060405260cc8060106000396000f360606040526000357c01000000000000000000000000000000000000000000000000000000009004806360cd2685146041578063c16431b914606b57603f565b005b6055600480803590602001909190505060a9565b6040518082815260200191505060405180910390f35b60886004808035906020019091908035906020019091905050608a565b005b80600060005083606481101560025790900160005b50819055505b5050565b6000600060005082606481101560025790900160005b5054905060c7565b91905056")
  60. testContractCodeDeployed = testContractCode[16:]
  61. testContractDeployed = uint64(2)
  62. testEventEmitterCode = common.Hex2Bytes("60606040523415600e57600080fd5b7f57050ab73f6b9ebdd9f76b8d4997793f48cf956e965ee070551b9ca0bb71584e60405160405180910390a160358060476000396000f3006060604052600080fd00a165627a7a723058203f727efcad8b5811f8cb1fc2620ce5e8c63570d697aef968172de296ea3994140029")
  63. // Checkpoint oracle relative fields
  64. oracleAddr common.Address
  65. signerKey, _ = crypto.GenerateKey()
  66. signerAddr = crypto.PubkeyToAddress(signerKey.PublicKey)
  67. )
  68. var (
  69. // The block frequency for creating checkpoint(only used in test)
  70. sectionSize = big.NewInt(128)
  71. // The number of confirmations needed to generate a checkpoint(only used in test).
  72. processConfirms = big.NewInt(1)
  73. // The token bucket buffer limit for testing purpose.
  74. testBufLimit = uint64(1000000)
  75. // The buffer recharging speed for testing purpose.
  76. testBufRecharge = uint64(1000)
  77. )
  78. /*
  79. contract test {
  80. uint256[100] data;
  81. function Put(uint256 addr, uint256 value) {
  82. data[addr] = value;
  83. }
  84. function Get(uint256 addr) constant returns (uint256 value) {
  85. return data[addr];
  86. }
  87. }
  88. */
  89. // prepare pre-commits specified number customized blocks into chain.
  90. func prepare(n int, backend *backends.SimulatedBackend) {
  91. var (
  92. ctx = context.Background()
  93. signer = types.HomesteadSigner{}
  94. )
  95. for i := 0; i < n; i++ {
  96. switch i {
  97. case 0:
  98. // Builtin-block
  99. // number: 1
  100. // txs: 2
  101. // deploy checkpoint contract
  102. auth, _ := bind.NewKeyedTransactorWithChainID(bankKey, big.NewInt(1337))
  103. oracleAddr, _, _, _ = contract.DeployCheckpointOracle(auth, backend, []common.Address{signerAddr}, sectionSize, processConfirms, big.NewInt(1))
  104. // bankUser transfers some ether to user1
  105. nonce, _ := backend.PendingNonceAt(ctx, bankAddr)
  106. tx, _ := types.SignTx(types.NewTransaction(nonce, userAddr1, big.NewInt(10000), params.TxGas, nil, nil), signer, bankKey)
  107. backend.SendTransaction(ctx, tx)
  108. case 1:
  109. // Builtin-block
  110. // number: 2
  111. // txs: 4
  112. bankNonce, _ := backend.PendingNonceAt(ctx, bankAddr)
  113. userNonce1, _ := backend.PendingNonceAt(ctx, userAddr1)
  114. // bankUser transfers more ether to user1
  115. tx1, _ := types.SignTx(types.NewTransaction(bankNonce, userAddr1, big.NewInt(1000), params.TxGas, nil, nil), signer, bankKey)
  116. backend.SendTransaction(ctx, tx1)
  117. // user1 relays ether to user2
  118. tx2, _ := types.SignTx(types.NewTransaction(userNonce1, userAddr2, big.NewInt(1000), params.TxGas, nil, nil), signer, userKey1)
  119. backend.SendTransaction(ctx, tx2)
  120. // user1 deploys a test contract
  121. tx3, _ := types.SignTx(types.NewContractCreation(userNonce1+1, big.NewInt(0), 200000, big.NewInt(0), testContractCode), signer, userKey1)
  122. backend.SendTransaction(ctx, tx3)
  123. testContractAddr = crypto.CreateAddress(userAddr1, userNonce1+1)
  124. // user1 deploys a event contract
  125. tx4, _ := types.SignTx(types.NewContractCreation(userNonce1+2, big.NewInt(0), 200000, big.NewInt(0), testEventEmitterCode), signer, userKey1)
  126. backend.SendTransaction(ctx, tx4)
  127. case 2:
  128. // Builtin-block
  129. // number: 3
  130. // txs: 2
  131. // bankUser transfer some ether to signer
  132. bankNonce, _ := backend.PendingNonceAt(ctx, bankAddr)
  133. tx1, _ := types.SignTx(types.NewTransaction(bankNonce, signerAddr, big.NewInt(1000000000), params.TxGas, nil, nil), signer, bankKey)
  134. backend.SendTransaction(ctx, tx1)
  135. // invoke test contract
  136. data := common.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000000000000001")
  137. tx2, _ := types.SignTx(types.NewTransaction(bankNonce+1, testContractAddr, big.NewInt(0), 100000, nil, data), signer, bankKey)
  138. backend.SendTransaction(ctx, tx2)
  139. case 3:
  140. // Builtin-block
  141. // number: 4
  142. // txs: 1
  143. // invoke test contract
  144. bankNonce, _ := backend.PendingNonceAt(ctx, bankAddr)
  145. data := common.Hex2Bytes("C16431B900000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000000000002")
  146. tx, _ := types.SignTx(types.NewTransaction(bankNonce, testContractAddr, big.NewInt(0), 100000, nil, data), signer, bankKey)
  147. backend.SendTransaction(ctx, tx)
  148. }
  149. backend.Commit()
  150. }
  151. }
  152. // testIndexers creates a set of indexers with specified params for testing purpose.
  153. func testIndexers(db ethdb.Database, odr light.OdrBackend, config *light.IndexerConfig, disablePruning bool) []*core.ChainIndexer {
  154. var indexers [3]*core.ChainIndexer
  155. indexers[0] = light.NewChtIndexer(db, odr, config.ChtSize, config.ChtConfirms, disablePruning)
  156. indexers[1] = core.NewBloomIndexer(db, config.BloomSize, config.BloomConfirms)
  157. indexers[2] = light.NewBloomTrieIndexer(db, odr, config.BloomSize, config.BloomTrieSize, disablePruning)
  158. // make bloomTrieIndexer as a child indexer of bloom indexer.
  159. indexers[1].AddChildIndexer(indexers[2])
  160. return indexers[:]
  161. }
  162. func newTestClientHandler(backend *backends.SimulatedBackend, odr *LesOdr, indexers []*core.ChainIndexer, db ethdb.Database, peers *serverPeerSet, ulcServers []string, ulcFraction int) *clientHandler {
  163. var (
  164. evmux = new(event.TypeMux)
  165. engine = ethash.NewFaker()
  166. gspec = core.Genesis{
  167. Config: params.AllEthashProtocolChanges,
  168. Alloc: core.GenesisAlloc{bankAddr: {Balance: bankFunds}},
  169. GasLimit: 100000000,
  170. }
  171. oracle *checkpointoracle.CheckpointOracle
  172. )
  173. genesis := gspec.MustCommit(db)
  174. chain, _ := light.NewLightChain(odr, gspec.Config, engine, nil)
  175. if indexers != nil {
  176. checkpointConfig := &params.CheckpointOracleConfig{
  177. Address: crypto.CreateAddress(bankAddr, 0),
  178. Signers: []common.Address{signerAddr},
  179. Threshold: 1,
  180. }
  181. getLocal := func(index uint64) params.TrustedCheckpoint {
  182. chtIndexer := indexers[0]
  183. sectionHead := chtIndexer.SectionHead(index)
  184. return params.TrustedCheckpoint{
  185. SectionIndex: index,
  186. SectionHead: sectionHead,
  187. CHTRoot: light.GetChtRoot(db, index, sectionHead),
  188. BloomRoot: light.GetBloomTrieRoot(db, index, sectionHead),
  189. }
  190. }
  191. oracle = checkpointoracle.New(checkpointConfig, getLocal)
  192. }
  193. client := &LightEthereum{
  194. lesCommons: lesCommons{
  195. genesis: genesis.Hash(),
  196. config: &ethconfig.Config{LightPeers: 100, NetworkId: NetworkId},
  197. chainConfig: params.AllEthashProtocolChanges,
  198. iConfig: light.TestClientIndexerConfig,
  199. chainDb: db,
  200. oracle: oracle,
  201. chainReader: chain,
  202. closeCh: make(chan struct{}),
  203. },
  204. peers: peers,
  205. reqDist: odr.retriever.dist,
  206. retriever: odr.retriever,
  207. odr: odr,
  208. engine: engine,
  209. blockchain: chain,
  210. eventMux: evmux,
  211. }
  212. client.handler = newClientHandler(ulcServers, ulcFraction, nil, client)
  213. if client.oracle != nil {
  214. client.oracle.Start(backend)
  215. }
  216. client.handler.start()
  217. return client.handler
  218. }
  219. func newTestServerHandler(blocks int, indexers []*core.ChainIndexer, db ethdb.Database, clock mclock.Clock) (*serverHandler, *backends.SimulatedBackend) {
  220. var (
  221. gspec = core.Genesis{
  222. Config: params.AllEthashProtocolChanges,
  223. Alloc: core.GenesisAlloc{bankAddr: {Balance: bankFunds}},
  224. GasLimit: 100000000,
  225. }
  226. oracle *checkpointoracle.CheckpointOracle
  227. )
  228. genesis := gspec.MustCommit(db)
  229. // create a simulation backend and pre-commit several customized block to the database.
  230. simulation := backends.NewSimulatedBackendWithDatabase(db, gspec.Alloc, 100000000)
  231. prepare(blocks, simulation)
  232. txpoolConfig := core.DefaultTxPoolConfig
  233. txpoolConfig.Journal = ""
  234. txpool := core.NewTxPool(txpoolConfig, gspec.Config, simulation.Blockchain())
  235. if indexers != nil {
  236. checkpointConfig := &params.CheckpointOracleConfig{
  237. Address: crypto.CreateAddress(bankAddr, 0),
  238. Signers: []common.Address{signerAddr},
  239. Threshold: 1,
  240. }
  241. getLocal := func(index uint64) params.TrustedCheckpoint {
  242. chtIndexer := indexers[0]
  243. sectionHead := chtIndexer.SectionHead(index)
  244. return params.TrustedCheckpoint{
  245. SectionIndex: index,
  246. SectionHead: sectionHead,
  247. CHTRoot: light.GetChtRoot(db, index, sectionHead),
  248. BloomRoot: light.GetBloomTrieRoot(db, index, sectionHead),
  249. }
  250. }
  251. oracle = checkpointoracle.New(checkpointConfig, getLocal)
  252. }
  253. ns := nodestate.NewNodeStateMachine(nil, nil, mclock.System{}, serverSetup)
  254. server := &LesServer{
  255. lesCommons: lesCommons{
  256. genesis: genesis.Hash(),
  257. config: &ethconfig.Config{LightPeers: 100, NetworkId: NetworkId},
  258. chainConfig: params.AllEthashProtocolChanges,
  259. iConfig: light.TestServerIndexerConfig,
  260. chainDb: db,
  261. chainReader: simulation.Blockchain(),
  262. oracle: oracle,
  263. closeCh: make(chan struct{}),
  264. },
  265. ns: ns,
  266. broadcaster: newBroadcaster(ns),
  267. servingQueue: newServingQueue(int64(time.Millisecond*10), 1),
  268. defParams: flowcontrol.ServerParams{
  269. BufLimit: testBufLimit,
  270. MinRecharge: testBufRecharge,
  271. },
  272. fcManager: flowcontrol.NewClientManager(nil, clock),
  273. }
  274. server.costTracker, server.minCapacity = newCostTracker(db, server.config)
  275. server.costTracker.testCostList = testCostList(0) // Disable flow control mechanism.
  276. server.clientPool = newClientPool(ns, db, testBufRecharge, defaultConnectedBias, clock, func(id enode.ID) {})
  277. server.clientPool.setLimits(10000, 10000) // Assign enough capacity for clientpool
  278. server.handler = newServerHandler(server, simulation.Blockchain(), db, txpool, func() bool { return true })
  279. if server.oracle != nil {
  280. server.oracle.Start(simulation)
  281. }
  282. server.servingQueue.setThreads(4)
  283. ns.Start()
  284. server.handler.start()
  285. return server.handler, simulation
  286. }
  287. // testPeer is a simulated peer to allow testing direct network calls.
  288. type testPeer struct {
  289. cpeer *clientPeer
  290. speer *serverPeer
  291. net p2p.MsgReadWriter // Network layer reader/writer to simulate remote messaging
  292. app *p2p.MsgPipeRW // Application layer reader/writer to simulate the local side
  293. }
  294. // handshakeWithServer executes the handshake with the remote server peer.
  295. func (p *testPeer) handshakeWithServer(t *testing.T, td *big.Int, head common.Hash, headNum uint64, genesis common.Hash, forkID forkid.ID) {
  296. // It only works for the simulated client peer
  297. if p.cpeer == nil {
  298. t.Fatal("handshake for client peer only")
  299. }
  300. var sendList keyValueList
  301. sendList = sendList.add("protocolVersion", uint64(p.cpeer.version))
  302. sendList = sendList.add("networkId", uint64(NetworkId))
  303. sendList = sendList.add("headTd", td)
  304. sendList = sendList.add("headHash", head)
  305. sendList = sendList.add("headNum", headNum)
  306. sendList = sendList.add("genesisHash", genesis)
  307. if p.cpeer.version >= lpv4 {
  308. sendList = sendList.add("forkID", &forkID)
  309. }
  310. if err := p2p.ExpectMsg(p.app, StatusMsg, nil); err != nil {
  311. t.Fatalf("status recv: %v", err)
  312. }
  313. if err := p2p.Send(p.app, StatusMsg, sendList); err != nil {
  314. t.Fatalf("status send: %v", err)
  315. }
  316. }
  317. // handshakeWithClient executes the handshake with the remote client peer.
  318. func (p *testPeer) handshakeWithClient(t *testing.T, td *big.Int, head common.Hash, headNum uint64, genesis common.Hash, forkID forkid.ID, costList RequestCostList, recentTxLookup uint64) {
  319. // It only works for the simulated client peer
  320. if p.speer == nil {
  321. t.Fatal("handshake for server peer only")
  322. }
  323. var sendList keyValueList
  324. sendList = sendList.add("protocolVersion", uint64(p.speer.version))
  325. sendList = sendList.add("networkId", uint64(NetworkId))
  326. sendList = sendList.add("headTd", td)
  327. sendList = sendList.add("headHash", head)
  328. sendList = sendList.add("headNum", headNum)
  329. sendList = sendList.add("genesisHash", genesis)
  330. sendList = sendList.add("serveHeaders", nil)
  331. sendList = sendList.add("serveChainSince", uint64(0))
  332. sendList = sendList.add("serveStateSince", uint64(0))
  333. sendList = sendList.add("serveRecentState", uint64(core.TriesInMemory-4))
  334. sendList = sendList.add("txRelay", nil)
  335. sendList = sendList.add("flowControl/BL", testBufLimit)
  336. sendList = sendList.add("flowControl/MRR", testBufRecharge)
  337. sendList = sendList.add("flowControl/MRC", costList)
  338. if p.speer.version >= lpv4 {
  339. sendList = sendList.add("forkID", &forkID)
  340. sendList = sendList.add("recentTxLookup", recentTxLookup)
  341. }
  342. if err := p2p.ExpectMsg(p.app, StatusMsg, nil); err != nil {
  343. t.Fatalf("status recv: %v", err)
  344. }
  345. if err := p2p.Send(p.app, StatusMsg, sendList); err != nil {
  346. t.Fatalf("status send: %v", err)
  347. }
  348. }
  349. // close terminates the local side of the peer, notifying the remote protocol
  350. // manager of termination.
  351. func (p *testPeer) close() {
  352. p.app.Close()
  353. }
  354. func newTestPeerPair(name string, version int, server *serverHandler, client *clientHandler) (*testPeer, *testPeer, error) {
  355. // Create a message pipe to communicate through
  356. app, net := p2p.MsgPipe()
  357. // Generate a random id and create the peer
  358. var id enode.ID
  359. rand.Read(id[:])
  360. peer1 := newClientPeer(version, NetworkId, p2p.NewPeer(id, name, nil), net)
  361. peer2 := newServerPeer(version, NetworkId, false, p2p.NewPeer(id, name, nil), app)
  362. // Start the peer on a new thread
  363. errc1 := make(chan error, 1)
  364. errc2 := make(chan error, 1)
  365. go func() {
  366. select {
  367. case <-server.closeCh:
  368. errc1 <- p2p.DiscQuitting
  369. case errc1 <- server.handle(peer1):
  370. }
  371. }()
  372. go func() {
  373. select {
  374. case <-client.closeCh:
  375. errc2 <- p2p.DiscQuitting
  376. case errc2 <- client.handle(peer2):
  377. }
  378. }()
  379. // Ensure the connection is established or exits when any error occurs
  380. for {
  381. select {
  382. case err := <-errc1:
  383. return nil, nil, fmt.Errorf("Failed to establish protocol connection %v", err)
  384. case err := <-errc2:
  385. return nil, nil, fmt.Errorf("Failed to establish protocol connection %v", err)
  386. default:
  387. }
  388. if atomic.LoadUint32(&peer1.serving) == 1 && atomic.LoadUint32(&peer2.serving) == 1 {
  389. break
  390. }
  391. time.Sleep(50 * time.Millisecond)
  392. }
  393. return &testPeer{cpeer: peer1, net: net, app: app}, &testPeer{speer: peer2, net: app, app: net}, nil
  394. }
  395. type indexerCallback func(*core.ChainIndexer, *core.ChainIndexer, *core.ChainIndexer)
  396. // testClient represents a client object for testing with necessary auxiliary fields.
  397. type testClient struct {
  398. clock mclock.Clock
  399. db ethdb.Database
  400. peer *testPeer
  401. handler *clientHandler
  402. chtIndexer *core.ChainIndexer
  403. bloomIndexer *core.ChainIndexer
  404. bloomTrieIndexer *core.ChainIndexer
  405. }
  406. // newRawPeer creates a new server peer connects to the server and do the handshake.
  407. func (client *testClient) newRawPeer(t *testing.T, name string, version int, recentTxLookup uint64) (*testPeer, func(), <-chan error) {
  408. // Create a message pipe to communicate through
  409. app, net := p2p.MsgPipe()
  410. // Generate a random id and create the peer
  411. var id enode.ID
  412. rand.Read(id[:])
  413. peer := newServerPeer(version, NetworkId, false, p2p.NewPeer(id, name, nil), net)
  414. // Start the peer on a new thread
  415. errCh := make(chan error, 1)
  416. go func() {
  417. select {
  418. case <-client.handler.closeCh:
  419. errCh <- p2p.DiscQuitting
  420. case errCh <- client.handler.handle(peer):
  421. }
  422. }()
  423. tp := &testPeer{
  424. app: app,
  425. net: net,
  426. speer: peer,
  427. }
  428. var (
  429. genesis = client.handler.backend.blockchain.Genesis()
  430. head = client.handler.backend.blockchain.CurrentHeader()
  431. td = client.handler.backend.blockchain.GetTd(head.Hash(), head.Number.Uint64())
  432. )
  433. forkID := forkid.NewID(client.handler.backend.blockchain.Config(), genesis.Hash(), head.Number.Uint64())
  434. tp.handshakeWithClient(t, td, head.Hash(), head.Number.Uint64(), genesis.Hash(), forkID, testCostList(0), recentTxLookup) // disable flow control by default
  435. // Ensure the connection is established or exits when any error occurs
  436. for {
  437. select {
  438. case <-errCh:
  439. return nil, nil, nil
  440. default:
  441. }
  442. if atomic.LoadUint32(&peer.serving) == 1 {
  443. break
  444. }
  445. time.Sleep(50 * time.Millisecond)
  446. }
  447. closePeer := func() {
  448. tp.speer.close()
  449. tp.close()
  450. }
  451. return tp, closePeer, errCh
  452. }
  453. // testServer represents a server object for testing with necessary auxiliary fields.
  454. type testServer struct {
  455. clock mclock.Clock
  456. backend *backends.SimulatedBackend
  457. db ethdb.Database
  458. peer *testPeer
  459. handler *serverHandler
  460. chtIndexer *core.ChainIndexer
  461. bloomIndexer *core.ChainIndexer
  462. bloomTrieIndexer *core.ChainIndexer
  463. }
  464. // newRawPeer creates a new client peer connects to the server and do the handshake.
  465. func (server *testServer) newRawPeer(t *testing.T, name string, version int) (*testPeer, func(), <-chan error) {
  466. // Create a message pipe to communicate through
  467. app, net := p2p.MsgPipe()
  468. // Generate a random id and create the peer
  469. var id enode.ID
  470. rand.Read(id[:])
  471. peer := newClientPeer(version, NetworkId, p2p.NewPeer(id, name, nil), net)
  472. // Start the peer on a new thread
  473. errCh := make(chan error, 1)
  474. go func() {
  475. select {
  476. case <-server.handler.closeCh:
  477. errCh <- p2p.DiscQuitting
  478. case errCh <- server.handler.handle(peer):
  479. }
  480. }()
  481. tp := &testPeer{
  482. app: app,
  483. net: net,
  484. cpeer: peer,
  485. }
  486. var (
  487. genesis = server.handler.blockchain.Genesis()
  488. head = server.handler.blockchain.CurrentHeader()
  489. td = server.handler.blockchain.GetTd(head.Hash(), head.Number.Uint64())
  490. )
  491. forkID := forkid.NewID(server.handler.blockchain.Config(), genesis.Hash(), head.Number.Uint64())
  492. tp.handshakeWithServer(t, td, head.Hash(), head.Number.Uint64(), genesis.Hash(), forkID)
  493. // Ensure the connection is established or exits when any error occurs
  494. for {
  495. select {
  496. case <-errCh:
  497. return nil, nil, nil
  498. default:
  499. }
  500. if atomic.LoadUint32(&peer.serving) == 1 {
  501. break
  502. }
  503. time.Sleep(50 * time.Millisecond)
  504. }
  505. closePeer := func() {
  506. tp.cpeer.close()
  507. tp.close()
  508. }
  509. return tp, closePeer, errCh
  510. }
  511. // testnetConfig wraps all the configurations for testing network.
  512. type testnetConfig struct {
  513. blocks int
  514. protocol int
  515. indexFn indexerCallback
  516. ulcServers []string
  517. ulcFraction int
  518. simClock bool
  519. connect bool
  520. nopruning bool
  521. }
  522. func newClientServerEnv(t *testing.T, config testnetConfig) (*testServer, *testClient, func()) {
  523. var (
  524. sdb = rawdb.NewMemoryDatabase()
  525. cdb = rawdb.NewMemoryDatabase()
  526. speers = newServerPeerSet()
  527. )
  528. var clock mclock.Clock = &mclock.System{}
  529. if config.simClock {
  530. clock = &mclock.Simulated{}
  531. }
  532. dist := newRequestDistributor(speers, clock)
  533. rm := newRetrieveManager(speers, dist, func() time.Duration { return time.Millisecond * 500 })
  534. odr := NewLesOdr(cdb, light.TestClientIndexerConfig, speers, rm)
  535. sindexers := testIndexers(sdb, nil, light.TestServerIndexerConfig, true)
  536. cIndexers := testIndexers(cdb, odr, light.TestClientIndexerConfig, config.nopruning)
  537. scIndexer, sbIndexer, sbtIndexer := sindexers[0], sindexers[1], sindexers[2]
  538. ccIndexer, cbIndexer, cbtIndexer := cIndexers[0], cIndexers[1], cIndexers[2]
  539. odr.SetIndexers(ccIndexer, cbIndexer, cbtIndexer)
  540. server, b := newTestServerHandler(config.blocks, sindexers, sdb, clock)
  541. client := newTestClientHandler(b, odr, cIndexers, cdb, speers, config.ulcServers, config.ulcFraction)
  542. scIndexer.Start(server.blockchain)
  543. sbIndexer.Start(server.blockchain)
  544. ccIndexer.Start(client.backend.blockchain)
  545. cbIndexer.Start(client.backend.blockchain)
  546. if config.indexFn != nil {
  547. config.indexFn(scIndexer, sbIndexer, sbtIndexer)
  548. }
  549. var (
  550. err error
  551. speer, cpeer *testPeer
  552. )
  553. if config.connect {
  554. done := make(chan struct{})
  555. client.syncEnd = func(_ *types.Header) { close(done) }
  556. cpeer, speer, err = newTestPeerPair("peer", config.protocol, server, client)
  557. if err != nil {
  558. t.Fatalf("Failed to connect testing peers %v", err)
  559. }
  560. select {
  561. case <-done:
  562. case <-time.After(10 * time.Second):
  563. t.Fatal("test peer did not connect and sync within 3s")
  564. }
  565. }
  566. s := &testServer{
  567. clock: clock,
  568. backend: b,
  569. db: sdb,
  570. peer: cpeer,
  571. handler: server,
  572. chtIndexer: scIndexer,
  573. bloomIndexer: sbIndexer,
  574. bloomTrieIndexer: sbtIndexer,
  575. }
  576. c := &testClient{
  577. clock: clock,
  578. db: cdb,
  579. peer: speer,
  580. handler: client,
  581. chtIndexer: ccIndexer,
  582. bloomIndexer: cbIndexer,
  583. bloomTrieIndexer: cbtIndexer,
  584. }
  585. teardown := func() {
  586. if config.connect {
  587. speer.close()
  588. cpeer.close()
  589. cpeer.cpeer.close()
  590. speer.speer.close()
  591. }
  592. ccIndexer.Close()
  593. cbIndexer.Close()
  594. scIndexer.Close()
  595. sbIndexer.Close()
  596. b.Close()
  597. }
  598. return s, c, teardown
  599. }
  600. func NewFuzzerPeer(version int) *clientPeer {
  601. return newClientPeer(version, 0, p2p.NewPeer(enode.ID{}, "", nil), nil)
  602. }