123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293 |
- // Copyright 2020 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 les
- import (
- "math/big"
- "testing"
- "time"
- "github.com/ethereum/go-ethereum/consensus/ethash"
- "github.com/ethereum/go-ethereum/core"
- "github.com/ethereum/go-ethereum/core/rawdb"
- "github.com/ethereum/go-ethereum/core/types"
- "github.com/ethereum/go-ethereum/p2p/enode"
- )
- // verifyImportEvent verifies that one single event arrive on an import channel.
- func verifyImportEvent(t *testing.T, imported chan interface{}, arrive bool) {
- if arrive {
- select {
- case <-imported:
- case <-time.After(time.Second):
- t.Fatalf("import timeout")
- }
- } else {
- select {
- case <-imported:
- t.Fatalf("import invoked")
- case <-time.After(20 * time.Millisecond):
- }
- }
- }
- // verifyImportDone verifies that no more events are arriving on an import channel.
- func verifyImportDone(t *testing.T, imported chan interface{}) {
- select {
- case <-imported:
- t.Fatalf("extra block imported")
- case <-time.After(50 * time.Millisecond):
- }
- }
- // verifyChainHeight verifies the chain height is as expected.
- func verifyChainHeight(t *testing.T, fetcher *lightFetcher, height uint64) {
- local := fetcher.chain.CurrentHeader().Number.Uint64()
- if local != height {
- t.Fatalf("chain height mismatch, got %d, want %d", local, height)
- }
- }
- func TestSequentialAnnouncementsLes2(t *testing.T) { testSequentialAnnouncements(t, 2) }
- func TestSequentialAnnouncementsLes3(t *testing.T) { testSequentialAnnouncements(t, 3) }
- func testSequentialAnnouncements(t *testing.T, protocol int) {
- netconfig := testnetConfig{
- blocks: 4,
- protocol: protocol,
- nopruning: true,
- }
- s, c, teardown := newClientServerEnv(t, netconfig)
- defer teardown()
- // Create connected peer pair.
- c.handler.fetcher.noAnnounce = true // Ignore the first announce from peer which can trigger a resync.
- p1, _, err := newTestPeerPair("peer", protocol, s.handler, c.handler)
- if err != nil {
- t.Fatalf("Failed to create peer pair %v", err)
- }
- c.handler.fetcher.noAnnounce = false
- importCh := make(chan interface{})
- c.handler.fetcher.newHeadHook = func(header *types.Header) {
- importCh <- header
- }
- for i := uint64(1); i <= s.backend.Blockchain().CurrentHeader().Number.Uint64(); i++ {
- header := s.backend.Blockchain().GetHeaderByNumber(i)
- hash, number := header.Hash(), header.Number.Uint64()
- td := rawdb.ReadTd(s.db, hash, number)
- announce := announceData{hash, number, td, 0, nil}
- if p1.cpeer.announceType == announceTypeSigned {
- announce.sign(s.handler.server.privateKey)
- }
- p1.cpeer.sendAnnounce(announce)
- verifyImportEvent(t, importCh, true)
- }
- verifyImportDone(t, importCh)
- verifyChainHeight(t, c.handler.fetcher, 4)
- }
- func TestGappedAnnouncementsLes2(t *testing.T) { testGappedAnnouncements(t, 2) }
- func TestGappedAnnouncementsLes3(t *testing.T) { testGappedAnnouncements(t, 3) }
- func testGappedAnnouncements(t *testing.T, protocol int) {
- netconfig := testnetConfig{
- blocks: 4,
- protocol: protocol,
- nopruning: true,
- }
- s, c, teardown := newClientServerEnv(t, netconfig)
- defer teardown()
- // Create connected peer pair.
- c.handler.fetcher.noAnnounce = true // Ignore the first announce from peer which can trigger a resync.
- peer, _, err := newTestPeerPair("peer", protocol, s.handler, c.handler)
- if err != nil {
- t.Fatalf("Failed to create peer pair %v", err)
- }
- c.handler.fetcher.noAnnounce = false
- done := make(chan *types.Header, 1)
- c.handler.fetcher.newHeadHook = func(header *types.Header) { done <- header }
- // Prepare announcement by latest header.
- latest := s.backend.Blockchain().CurrentHeader()
- hash, number := latest.Hash(), latest.Number.Uint64()
- td := rawdb.ReadTd(s.db, hash, number)
- // Sign the announcement if necessary.
- announce := announceData{hash, number, td, 0, nil}
- if peer.cpeer.announceType == announceTypeSigned {
- announce.sign(s.handler.server.privateKey)
- }
- peer.cpeer.sendAnnounce(announce)
- <-done // Wait syncing
- verifyChainHeight(t, c.handler.fetcher, 4)
- // Send a reorged announcement
- var newAnno = make(chan struct{}, 1)
- c.handler.fetcher.noAnnounce = true
- c.handler.fetcher.newAnnounce = func(*serverPeer, *announceData) {
- newAnno <- struct{}{}
- }
- blocks, _ := core.GenerateChain(rawdb.ReadChainConfig(s.db, s.backend.Blockchain().Genesis().Hash()), s.backend.Blockchain().GetBlockByNumber(3),
- ethash.NewFaker(), s.db, 2, func(i int, gen *core.BlockGen) {
- gen.OffsetTime(-9) // higher block difficulty
- })
- s.backend.Blockchain().InsertChain(blocks)
- <-newAnno
- c.handler.fetcher.noAnnounce = false
- c.handler.fetcher.newAnnounce = nil
- latest = blocks[len(blocks)-1].Header()
- hash, number = latest.Hash(), latest.Number.Uint64()
- td = rawdb.ReadTd(s.db, hash, number)
- announce = announceData{hash, number, td, 1, nil}
- if peer.cpeer.announceType == announceTypeSigned {
- announce.sign(s.handler.server.privateKey)
- }
- peer.cpeer.sendAnnounce(announce)
- <-done // Wait syncing
- verifyChainHeight(t, c.handler.fetcher, 5)
- }
- func TestTrustedAnnouncementsLes2(t *testing.T) { testTrustedAnnouncement(t, 2) }
- func TestTrustedAnnouncementsLes3(t *testing.T) { testTrustedAnnouncement(t, 3) }
- func testTrustedAnnouncement(t *testing.T, protocol int) {
- var (
- servers []*testServer
- teardowns []func()
- nodes []*enode.Node
- ids []string
- cpeers []*clientPeer
- speers []*serverPeer
- )
- for i := 0; i < 10; i++ {
- s, n, teardown := newTestServerPeer(t, 10, protocol)
- servers = append(servers, s)
- nodes = append(nodes, n)
- teardowns = append(teardowns, teardown)
- // A half of them are trusted servers.
- if i < 5 {
- ids = append(ids, n.String())
- }
- }
- netconfig := testnetConfig{
- protocol: protocol,
- nopruning: true,
- ulcServers: ids,
- ulcFraction: 60,
- }
- _, c, teardown := newClientServerEnv(t, netconfig)
- defer teardown()
- defer func() {
- for i := 0; i < len(teardowns); i++ {
- teardowns[i]()
- }
- }()
- c.handler.fetcher.noAnnounce = true // Ignore the first announce from peer which can trigger a resync.
- // Connect all server instances.
- for i := 0; i < len(servers); i++ {
- sp, cp, err := connect(servers[i].handler, nodes[i].ID(), c.handler, protocol)
- if err != nil {
- t.Fatalf("connect server and client failed, err %s", err)
- }
- cpeers = append(cpeers, cp)
- speers = append(speers, sp)
- }
- c.handler.fetcher.noAnnounce = false
- newHead := make(chan *types.Header, 1)
- c.handler.fetcher.newHeadHook = func(header *types.Header) { newHead <- header }
- check := func(height []uint64, expected uint64, callback func()) {
- for i := 0; i < len(height); i++ {
- for j := 0; j < len(servers); j++ {
- h := servers[j].backend.Blockchain().GetHeaderByNumber(height[i])
- hash, number := h.Hash(), h.Number.Uint64()
- td := rawdb.ReadTd(servers[j].db, hash, number)
- // Sign the announcement if necessary.
- announce := announceData{hash, number, td, 0, nil}
- p := cpeers[j]
- if p.announceType == announceTypeSigned {
- announce.sign(servers[j].handler.server.privateKey)
- }
- p.sendAnnounce(announce)
- }
- }
- if callback != nil {
- callback()
- }
- verifyChainHeight(t, c.handler.fetcher, expected)
- }
- check([]uint64{1}, 1, func() { <-newHead }) // Sequential announcements
- check([]uint64{4}, 4, func() { <-newHead }) // ULC-style light syncing, rollback untrusted headers
- check([]uint64{10}, 10, func() { <-newHead }) // Sync the whole chain.
- }
- func TestInvalidAnnouncesLES2(t *testing.T) { testInvalidAnnounces(t, lpv2) }
- func TestInvalidAnnouncesLES3(t *testing.T) { testInvalidAnnounces(t, lpv3) }
- func TestInvalidAnnouncesLES4(t *testing.T) { testInvalidAnnounces(t, lpv4) }
- func testInvalidAnnounces(t *testing.T, protocol int) {
- netconfig := testnetConfig{
- blocks: 4,
- protocol: protocol,
- nopruning: true,
- }
- s, c, teardown := newClientServerEnv(t, netconfig)
- defer teardown()
- // Create connected peer pair.
- c.handler.fetcher.noAnnounce = true // Ignore the first announce from peer which can trigger a resync.
- peer, _, err := newTestPeerPair("peer", lpv3, s.handler, c.handler)
- if err != nil {
- t.Fatalf("Failed to create peer pair %v", err)
- }
- c.handler.fetcher.noAnnounce = false
- done := make(chan *types.Header, 1)
- c.handler.fetcher.newHeadHook = func(header *types.Header) { done <- header }
- // Prepare announcement by latest header.
- headerOne := s.backend.Blockchain().GetHeaderByNumber(1)
- hash, number := headerOne.Hash(), headerOne.Number.Uint64()
- td := big.NewInt(200) // bad td
- // Sign the announcement if necessary.
- announce := announceData{hash, number, td, 0, nil}
- if peer.cpeer.announceType == announceTypeSigned {
- announce.sign(s.handler.server.privateKey)
- }
- peer.cpeer.sendAnnounce(announce)
- <-done // Wait syncing
- // Ensure the bad peer is evicited
- if c.handler.backend.peers.len() != 0 {
- t.Fatalf("Failed to evict invalid peer")
- }
- }
|