path: root/les/ulc_test.go

// 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 <http://www.gnu.org/licenses/>.

package les

import (
    "crypto/rand"
    "fmt"
    "net"
    "testing"
    "time"

    "github.com/ethereum/go-ethereum/crypto"
    "github.com/ethereum/go-ethereum/p2p"
    "github.com/ethereum/go-ethereum/p2p/enode"
)

func TestULCAnnounceThresholdLes2(t *testing.T) { testULCAnnounceThreshold(t, 2) }
func TestULCAnnounceThresholdLes3(t *testing.T) { testULCAnnounceThreshold(t, 3) }

func testULCAnnounceThreshold(t *testing.T, protocol int) {
    // todo figure out why it takes fetcher so longer to fetcher the announced header.
    t.Skip("Sometimes it can failed")
    var cases = []struct {
        height    []int
        threshold int
        expect    uint64
    }{
        {[]int{1}, 100, 1},
        {[]int{0, 0, 0}, 100, 0},
        {[]int{1, 2, 3}, 30, 3},
        {[]int{1, 2, 3}, 60, 2},
        {[]int{3, 2, 1}, 67, 1},
        {[]int{3, 2, 1}, 100, 1},
    }
    for _, testcase := range cases {
        var (
            servers   []*testServer
            teardowns []func()
            nodes     []*enode.Node
            ids       []string
        )
        for i := 0; i < len(testcase.height); i++ {
            s, n, teardown := newServerPeer(t, 0, protocol)

            servers = append(servers, s)
            nodes = append(nodes, n)
            teardowns = append(teardowns, teardown)
            ids = append(ids, n.String())
        }
        c, teardown := newLightPeer(t, protocol, ids, testcase.threshold)

        // Connect all servers.
        for i := 0; i < len(servers); i++ {
            connect(servers[i].handler, nodes[i].ID(), c.handler, protocol)
        }
        for i := 0; i < len(servers); i++ {
            for j := 0; j < testcase.height[i]; j++ {
                servers[i].backend.Commit()
            }
        }
        time.Sleep(1500 * time.Millisecond) // Ensure the fetcher has done its work.
        head := c.handler.backend.blockchain.CurrentHeader().Number.Uint64()
        if head != testcase.expect {
            t.Fatalf("chain height mismatch, want %d, got %d", testcase.expect, head)
        }

        // Release all servers and client resources.
        teardown()
        for i := 0; i < len(teardowns); i++ {
            teardowns[i]()
        }
    }
}

func connect(server *serverHandler, serverId enode.ID, client *clientHandler, protocol int) (*peer, *peer, error) {
    // Create a message pipe to communicate through
    app, net := p2p.MsgPipe()

    var id enode.ID
    rand.Read(id[:])

    peer1 := newPeer(protocol, NetworkId, true, p2p.NewPeer(serverId, "", nil), net) // Mark server as trusted
    peer2 := newPeer(protocol, NetworkId, false, p2p.NewPeer(id, "", nil), app)

    // Start the peerLight on a new thread
    errc1 := make(chan error, 1)
    errc2 := make(chan error, 1)
    go func() {
        select {
        case <-server.closeCh:
            errc1 <- p2p.DiscQuitting
        case errc1 <- server.handle(peer2):
        }
    }()
    go func() {
        select {
        case <-client.closeCh:
            errc1 <- p2p.DiscQuitting
        case errc1 <- client.handle(peer1):
        }
    }()

    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 peer1, peer2, nil
}

// newServerPeer creates server peer.
func newServerPeer(t *testing.T, blocks int, protocol int) (*testServer, *enode.Node, func()) {
    s, teardown := newServerEnv(t, blocks, protocol, nil, false, false, 0)
    key, err := crypto.GenerateKey()
    if err != nil {
        t.Fatal("generate key err:", err)
    }
    s.handler.server.privateKey = key
    n := enode.NewV4(&key.PublicKey, net.ParseIP("127.0.0.1"), 35000, 35000)
    return s, n, teardown
}

// newLightPeer creates node with light sync mode
func newLightPeer(t *testing.T, protocol int, ulcServers []string, ulcFraction int) (*testClient, func()) {
    _, c, teardown := newClientServerEnv(t, 0, protocol, nil, ulcServers, ulcFraction, false, false)
    return c, teardown
}