// Copyright 2018 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 . package les import ( "crypto/ecdsa" "fmt" "math/big" "net" "reflect" "testing" "time" "github.com/ethereum/go-ethereum/common/mclock" "github.com/ethereum/go-ethereum/core/rawdb" "github.com/ethereum/go-ethereum/crypto" "github.com/ethereum/go-ethereum/light" "github.com/ethereum/go-ethereum/p2p" "github.com/ethereum/go-ethereum/p2p/enode" ) func TestULCSyncWithOnePeer(t *testing.T) { f := newFullPeerPair(t, 1, 4) l := newLightPeer(t, []string{f.Node.String()}, 100) if reflect.DeepEqual(f.PM.blockchain.CurrentHeader().Hash(), l.PM.blockchain.CurrentHeader().Hash()) { t.Fatal("blocks are equal") } _, _, err := connectPeers(f, l, 2) if err != nil { t.Fatal(err) } l.PM.fetcher.lock.Lock() l.PM.fetcher.nextRequest() l.PM.fetcher.lock.Unlock() if !reflect.DeepEqual(f.PM.blockchain.CurrentHeader().Hash(), l.PM.blockchain.CurrentHeader().Hash()) { t.Fatal("sync doesn't work") } } func TestULCReceiveAnnounce(t *testing.T) { f := newFullPeerPair(t, 1, 4) l := newLightPeer(t, []string{f.Node.String()}, 100) fPeer, lPeer, err := connectPeers(f, l, 2) if err != nil { t.Fatal(err) } l.PM.synchronise(fPeer) //check that the sync is finished correctly if !reflect.DeepEqual(f.PM.blockchain.CurrentHeader().Hash(), l.PM.blockchain.CurrentHeader().Hash()) { t.Fatal("sync doesn't work") } l.PM.peers.lock.Lock() if len(l.PM.peers.peers) == 0 { t.Fatal("peer list should not be empty") } l.PM.peers.lock.Unlock() time.Sleep(time.Second) //send a signed announce message(payload doesn't matter) td := f.PM.blockchain.GetTd(l.PM.blockchain.CurrentHeader().Hash(), l.PM.blockchain.CurrentHeader().Number.Uint64()) announce := announceData{ Number: l.PM.blockchain.CurrentHeader().Number.Uint64() + 1, Td: td.Add(td, big.NewInt(1)), } announce.sign(f.Key) lPeer.SendAnnounce(announce) } func TestULCShouldNotSyncWithTwoPeersOneHaveEmptyChain(t *testing.T) { f1 := newFullPeerPair(t, 1, 4) f2 := newFullPeerPair(t, 2, 0) l := newLightPeer(t, []string{f1.Node.String(), f2.Node.String()}, 100) _, _, err := connectPeers(f1, l, 2) if err != nil { t.Fatal(err) } _, _, err = connectPeers(f2, l, 2) if err != nil { t.Fatal(err) } l.PM.fetcher.lock.Lock() l.PM.fetcher.nextRequest() l.PM.fetcher.lock.Unlock() if reflect.DeepEqual(f2.PM.blockchain.CurrentHeader().Hash(), l.PM.blockchain.CurrentHeader().Hash()) { t.Fatal("Incorrect hash: second peer has empty chain") } } func TestULCShouldNotSyncWithThreePeersOneHaveEmptyChain(t *testing.T) { f1 := newFullPeerPair(t, 1, 3) f2 := newFullPeerPair(t, 2, 4) f3 := newFullPeerPair(t, 3, 0) l := newLightPeer(t, []string{f1.Node.String(), f2.Node.String(), f3.Node.String()}, 60) _, _, err := connectPeers(f1, l, 2) if err != nil { t.Fatal(err) } _, _, err = connectPeers(f2, l, 2) if err != nil { t.Fatal(err) } _, _, err = connectPeers(f3, l, 2) if err != nil { t.Fatal(err) } l.PM.fetcher.lock.Lock() l.PM.fetcher.nextRequest() l.PM.fetcher.lock.Unlock() if !reflect.DeepEqual(f1.PM.blockchain.CurrentHeader().Hash(), l.PM.blockchain.CurrentHeader().Hash()) { t.Fatal("Incorrect hash") } } type pairPeer struct { Name string Node *enode.Node PM *ProtocolManager Key *ecdsa.PrivateKey } func connectPeers(full, light pairPeer, version int) (*peer, *peer, error) { // Create a message pipe to communicate through app, net := p2p.MsgPipe() peerLight := full.PM.newPeer(version, NetworkId, p2p.NewPeer(light.Node.ID(), light.Name, nil), net) peerFull := light.PM.newPeer(version, NetworkId, p2p.NewPeer(full.Node.ID(), full.Name, nil), app) // Start the peerLight on a new thread errc1 := make(chan error, 1) errc2 := make(chan error, 1) go func() { select { case light.PM.newPeerCh <- peerFull: errc1 <- light.PM.handle(peerFull) case <-light.PM.quitSync: errc1 <- p2p.DiscQuitting } }() go func() { select { case full.PM.newPeerCh <- peerLight: errc2 <- full.PM.handle(peerLight) case <-full.PM.quitSync: errc2 <- p2p.DiscQuitting } }() select { case <-time.After(time.Millisecond * 100): case err := <-errc1: return nil, nil, fmt.Errorf("peerLight handshake error: %v", err) case err := <-errc2: return nil, nil, fmt.Errorf("peerFull handshake error: %v", err) } return peerFull, peerLight, nil } // newFullPeerPair creates node with full sync mode func newFullPeerPair(t *testing.T, index int, numberOfblocks int) pairPeer { db := rawdb.NewMemoryDatabase() pmFull, _ := newTestProtocolManagerMust(t, false, numberOfblocks, nil, nil, nil, db, nil, 0) peerPairFull := pairPeer{ Name: "full node", PM: pmFull, } key, err := crypto.GenerateKey() if err != nil { t.Fatal("generate key err:", err) } peerPairFull.Key = key peerPairFull.Node = enode.NewV4(&key.PublicKey, net.ParseIP("127.0.0.1"), 35000, 35000) return peerPairFull } // newLightPeer creates node with light sync mode func newLightPeer(t *testing.T, ulcServers []string, ulcFraction int) pairPeer { peers := newPeerSet() dist := newRequestDistributor(peers, make(chan struct{}), &mclock.System{}) rm := newRetrieveManager(peers, dist, nil) ldb := rawdb.NewMemoryDatabase() odr := NewLesOdr(ldb, light.DefaultClientIndexerConfig, rm) pmLight, _ := newTestProtocolManagerMust(t, true, 0, odr, nil, peers, ldb, ulcServers, ulcFraction) peerPairLight := pairPeer{ Name: "ulc node", PM: pmLight, } key, err := crypto.GenerateKey() if err != nil { t.Fatal("generate key err:", err) } peerPairLight.Key = key peerPairLight.Node = enode.NewV4(&key.PublicKey, net.IP{}, 35000, 35000) return peerPairLight }