|
@@ -18,15 +18,24 @@ package eth
|
|
|
|
|
|
|
|
import (
|
|
import (
|
|
|
"fmt"
|
|
"fmt"
|
|
|
|
|
+ "math/big"
|
|
|
"sync"
|
|
"sync"
|
|
|
"testing"
|
|
"testing"
|
|
|
"time"
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/consensus/ethash"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/core"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/core/forkid"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/core/rawdb"
|
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/core/vm"
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
|
"github.com/ethereum/go-ethereum/eth/downloader"
|
|
"github.com/ethereum/go-ethereum/eth/downloader"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/event"
|
|
|
"github.com/ethereum/go-ethereum/p2p"
|
|
"github.com/ethereum/go-ethereum/p2p"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/p2p/enode"
|
|
|
|
|
+ "github.com/ethereum/go-ethereum/params"
|
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -37,10 +46,7 @@ func init() {
|
|
|
var testAccount, _ = crypto.HexToECDSA("b71c71a67e1177ad4e901695e1b4b9ee17ae16c6668d313eac2f96dbcda3f291")
|
|
var testAccount, _ = crypto.HexToECDSA("b71c71a67e1177ad4e901695e1b4b9ee17ae16c6668d313eac2f96dbcda3f291")
|
|
|
|
|
|
|
|
// Tests that handshake failures are detected and reported correctly.
|
|
// Tests that handshake failures are detected and reported correctly.
|
|
|
-func TestStatusMsgErrors62(t *testing.T) { testStatusMsgErrors(t, 62) }
|
|
|
|
|
-func TestStatusMsgErrors63(t *testing.T) { testStatusMsgErrors(t, 63) }
|
|
|
|
|
-
|
|
|
|
|
-func testStatusMsgErrors(t *testing.T, protocol int) {
|
|
|
|
|
|
|
+func TestStatusMsgErrors63(t *testing.T) {
|
|
|
pm, _ := newTestProtocolManagerMust(t, downloader.FullSync, 0, nil, nil)
|
|
pm, _ := newTestProtocolManagerMust(t, downloader.FullSync, 0, nil, nil)
|
|
|
var (
|
|
var (
|
|
|
genesis = pm.blockchain.Genesis()
|
|
genesis = pm.blockchain.Genesis()
|
|
@@ -59,21 +65,76 @@ func testStatusMsgErrors(t *testing.T, protocol int) {
|
|
|
wantError: errResp(ErrNoStatusMsg, "first msg has code 2 (!= 0)"),
|
|
wantError: errResp(ErrNoStatusMsg, "first msg has code 2 (!= 0)"),
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
- code: StatusMsg, data: statusData{10, DefaultConfig.NetworkId, td, head.Hash(), genesis.Hash()},
|
|
|
|
|
- wantError: errResp(ErrProtocolVersionMismatch, "10 (!= %d)", protocol),
|
|
|
|
|
|
|
+ code: StatusMsg, data: statusData63{10, DefaultConfig.NetworkId, td, head.Hash(), genesis.Hash()},
|
|
|
|
|
+ wantError: errResp(ErrProtocolVersionMismatch, "10 (!= %d)", 63),
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
- code: StatusMsg, data: statusData{uint32(protocol), 999, td, head.Hash(), genesis.Hash()},
|
|
|
|
|
- wantError: errResp(ErrNetworkIdMismatch, "999 (!= %d)", DefaultConfig.NetworkId),
|
|
|
|
|
|
|
+ code: StatusMsg, data: statusData63{63, 999, td, head.Hash(), genesis.Hash()},
|
|
|
|
|
+ wantError: errResp(ErrNetworkIDMismatch, "999 (!= %d)", DefaultConfig.NetworkId),
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
- code: StatusMsg, data: statusData{uint32(protocol), DefaultConfig.NetworkId, td, head.Hash(), common.Hash{3}},
|
|
|
|
|
- wantError: errResp(ErrGenesisBlockMismatch, "0300000000000000 (!= %x)", genesis.Hash().Bytes()[:8]),
|
|
|
|
|
|
|
+ code: StatusMsg, data: statusData63{63, DefaultConfig.NetworkId, td, head.Hash(), common.Hash{3}},
|
|
|
|
|
+ wantError: errResp(ErrGenesisMismatch, "0300000000000000 (!= %x)", genesis.Hash().Bytes()[:8]),
|
|
|
},
|
|
},
|
|
|
}
|
|
}
|
|
|
|
|
+ for i, test := range tests {
|
|
|
|
|
+ p, errc := newTestPeer("peer", 63, pm, false)
|
|
|
|
|
+ // The send call might hang until reset because
|
|
|
|
|
+ // the protocol might not read the payload.
|
|
|
|
|
+ go p2p.Send(p.app, test.code, test.data)
|
|
|
|
|
+
|
|
|
|
|
+ select {
|
|
|
|
|
+ case err := <-errc:
|
|
|
|
|
+ if err == nil {
|
|
|
|
|
+ t.Errorf("test %d: protocol returned nil error, want %q", i, test.wantError)
|
|
|
|
|
+ } else if err.Error() != test.wantError.Error() {
|
|
|
|
|
+ t.Errorf("test %d: wrong error: got %q, want %q", i, err, test.wantError)
|
|
|
|
|
+ }
|
|
|
|
|
+ case <-time.After(2 * time.Second):
|
|
|
|
|
+ t.Errorf("protocol did not shut down within 2 seconds")
|
|
|
|
|
+ }
|
|
|
|
|
+ p.close()
|
|
|
|
|
+ }
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
|
|
+func TestStatusMsgErrors64(t *testing.T) {
|
|
|
|
|
+ pm, _ := newTestProtocolManagerMust(t, downloader.FullSync, 0, nil, nil)
|
|
|
|
|
+ var (
|
|
|
|
|
+ genesis = pm.blockchain.Genesis()
|
|
|
|
|
+ head = pm.blockchain.CurrentHeader()
|
|
|
|
|
+ td = pm.blockchain.GetTd(head.Hash(), head.Number.Uint64())
|
|
|
|
|
+ forkID = forkid.NewID(pm.blockchain)
|
|
|
|
|
+ )
|
|
|
|
|
+ defer pm.Stop()
|
|
|
|
|
|
|
|
|
|
+ tests := []struct {
|
|
|
|
|
+ code uint64
|
|
|
|
|
+ data interface{}
|
|
|
|
|
+ wantError error
|
|
|
|
|
+ }{
|
|
|
|
|
+ {
|
|
|
|
|
+ code: TxMsg, data: []interface{}{},
|
|
|
|
|
+ wantError: errResp(ErrNoStatusMsg, "first msg has code 2 (!= 0)"),
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ code: StatusMsg, data: statusData{10, DefaultConfig.NetworkId, td, head.Hash(), genesis.Hash(), forkID},
|
|
|
|
|
+ wantError: errResp(ErrProtocolVersionMismatch, "10 (!= %d)", 64),
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ code: StatusMsg, data: statusData{64, 999, td, head.Hash(), genesis.Hash(), forkID},
|
|
|
|
|
+ wantError: errResp(ErrNetworkIDMismatch, "999 (!= %d)", DefaultConfig.NetworkId),
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ code: StatusMsg, data: statusData{64, DefaultConfig.NetworkId, td, head.Hash(), common.Hash{3}, forkID},
|
|
|
|
|
+ wantError: errResp(ErrGenesisMismatch, "0300000000000000000000000000000000000000000000000000000000000000 (!= %x)", genesis.Hash()),
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ code: StatusMsg, data: statusData{64, DefaultConfig.NetworkId, td, head.Hash(), genesis.Hash(), forkid.ID{Hash: [4]byte{0x00, 0x01, 0x02, 0x03}}},
|
|
|
|
|
+ wantError: errResp(ErrForkIDRejected, forkid.ErrLocalIncompatibleOrStale.Error()),
|
|
|
|
|
+ },
|
|
|
|
|
+ }
|
|
|
for i, test := range tests {
|
|
for i, test := range tests {
|
|
|
- p, errc := newTestPeer("peer", protocol, pm, false)
|
|
|
|
|
|
|
+ p, errc := newTestPeer("peer", 64, pm, false)
|
|
|
// The send call might hang until reset because
|
|
// The send call might hang until reset because
|
|
|
// the protocol might not read the payload.
|
|
// the protocol might not read the payload.
|
|
|
go p2p.Send(p.app, test.code, test.data)
|
|
go p2p.Send(p.app, test.code, test.data)
|
|
@@ -92,9 +153,99 @@ func testStatusMsgErrors(t *testing.T, protocol int) {
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
+func TestForkIDSplit(t *testing.T) {
|
|
|
|
|
+ var (
|
|
|
|
|
+ engine = ethash.NewFaker()
|
|
|
|
|
+
|
|
|
|
|
+ configNoFork = ¶ms.ChainConfig{HomesteadBlock: big.NewInt(1)}
|
|
|
|
|
+ configProFork = ¶ms.ChainConfig{
|
|
|
|
|
+ HomesteadBlock: big.NewInt(1),
|
|
|
|
|
+ EIP150Block: big.NewInt(2),
|
|
|
|
|
+ EIP155Block: big.NewInt(2),
|
|
|
|
|
+ EIP158Block: big.NewInt(2),
|
|
|
|
|
+ ByzantiumBlock: big.NewInt(3),
|
|
|
|
|
+ }
|
|
|
|
|
+ dbNoFork = rawdb.NewMemoryDatabase()
|
|
|
|
|
+ dbProFork = rawdb.NewMemoryDatabase()
|
|
|
|
|
+
|
|
|
|
|
+ gspecNoFork = &core.Genesis{Config: configNoFork}
|
|
|
|
|
+ gspecProFork = &core.Genesis{Config: configProFork}
|
|
|
|
|
+
|
|
|
|
|
+ genesisNoFork = gspecNoFork.MustCommit(dbNoFork)
|
|
|
|
|
+ genesisProFork = gspecProFork.MustCommit(dbProFork)
|
|
|
|
|
+
|
|
|
|
|
+ chainNoFork, _ = core.NewBlockChain(dbNoFork, nil, configNoFork, engine, vm.Config{}, nil)
|
|
|
|
|
+ chainProFork, _ = core.NewBlockChain(dbProFork, nil, configProFork, engine, vm.Config{}, nil)
|
|
|
|
|
+
|
|
|
|
|
+ blocksNoFork, _ = core.GenerateChain(configNoFork, genesisNoFork, engine, dbNoFork, 2, nil)
|
|
|
|
|
+ blocksProFork, _ = core.GenerateChain(configProFork, genesisProFork, engine, dbProFork, 2, nil)
|
|
|
|
|
+
|
|
|
|
|
+ ethNoFork, _ = NewProtocolManager(configNoFork, nil, downloader.FullSync, 1, new(event.TypeMux), new(testTxPool), engine, chainNoFork, dbNoFork, 1, nil)
|
|
|
|
|
+ ethProFork, _ = NewProtocolManager(configProFork, nil, downloader.FullSync, 1, new(event.TypeMux), new(testTxPool), engine, chainProFork, dbProFork, 1, nil)
|
|
|
|
|
+ )
|
|
|
|
|
+ ethNoFork.Start(1000)
|
|
|
|
|
+ ethProFork.Start(1000)
|
|
|
|
|
+
|
|
|
|
|
+ // Both nodes should allow the other to connect (same genesis, next fork is the same)
|
|
|
|
|
+ p2pNoFork, p2pProFork := p2p.MsgPipe()
|
|
|
|
|
+ peerNoFork := newPeer(64, p2p.NewPeer(enode.ID{1}, "", nil), p2pNoFork)
|
|
|
|
|
+ peerProFork := newPeer(64, p2p.NewPeer(enode.ID{2}, "", nil), p2pProFork)
|
|
|
|
|
+
|
|
|
|
|
+ errc := make(chan error, 2)
|
|
|
|
|
+ go func() { errc <- ethNoFork.handle(peerProFork) }()
|
|
|
|
|
+ go func() { errc <- ethProFork.handle(peerNoFork) }()
|
|
|
|
|
+
|
|
|
|
|
+ select {
|
|
|
|
|
+ case err := <-errc:
|
|
|
|
|
+ t.Fatalf("frontier nofork <-> profork failed: %v", err)
|
|
|
|
|
+ case <-time.After(250 * time.Millisecond):
|
|
|
|
|
+ p2pNoFork.Close()
|
|
|
|
|
+ p2pProFork.Close()
|
|
|
|
|
+ }
|
|
|
|
|
+ // Progress into Homestead. Fork's match, so we don't care what the future holds
|
|
|
|
|
+ chainNoFork.InsertChain(blocksNoFork[:1])
|
|
|
|
|
+ chainProFork.InsertChain(blocksProFork[:1])
|
|
|
|
|
+
|
|
|
|
|
+ p2pNoFork, p2pProFork = p2p.MsgPipe()
|
|
|
|
|
+ peerNoFork = newPeer(64, p2p.NewPeer(enode.ID{1}, "", nil), p2pNoFork)
|
|
|
|
|
+ peerProFork = newPeer(64, p2p.NewPeer(enode.ID{2}, "", nil), p2pProFork)
|
|
|
|
|
+
|
|
|
|
|
+ errc = make(chan error, 2)
|
|
|
|
|
+ go func() { errc <- ethNoFork.handle(peerProFork) }()
|
|
|
|
|
+ go func() { errc <- ethProFork.handle(peerNoFork) }()
|
|
|
|
|
+
|
|
|
|
|
+ select {
|
|
|
|
|
+ case err := <-errc:
|
|
|
|
|
+ t.Fatalf("homestead nofork <-> profork failed: %v", err)
|
|
|
|
|
+ case <-time.After(250 * time.Millisecond):
|
|
|
|
|
+ p2pNoFork.Close()
|
|
|
|
|
+ p2pProFork.Close()
|
|
|
|
|
+ }
|
|
|
|
|
+ // Progress into Spurious. Forks mismatch, signalling differing chains, reject
|
|
|
|
|
+ chainNoFork.InsertChain(blocksNoFork[1:2])
|
|
|
|
|
+ chainProFork.InsertChain(blocksProFork[1:2])
|
|
|
|
|
+
|
|
|
|
|
+ p2pNoFork, p2pProFork = p2p.MsgPipe()
|
|
|
|
|
+ peerNoFork = newPeer(64, p2p.NewPeer(enode.ID{1}, "", nil), p2pNoFork)
|
|
|
|
|
+ peerProFork = newPeer(64, p2p.NewPeer(enode.ID{2}, "", nil), p2pProFork)
|
|
|
|
|
+
|
|
|
|
|
+ errc = make(chan error, 2)
|
|
|
|
|
+ go func() { errc <- ethNoFork.handle(peerProFork) }()
|
|
|
|
|
+ go func() { errc <- ethProFork.handle(peerNoFork) }()
|
|
|
|
|
+
|
|
|
|
|
+ select {
|
|
|
|
|
+ case err := <-errc:
|
|
|
|
|
+ if want := errResp(ErrForkIDRejected, forkid.ErrLocalIncompatibleOrStale.Error()); err.Error() != want.Error() {
|
|
|
|
|
+ t.Fatalf("fork ID rejection error mismatch: have %v, want %v", err, want)
|
|
|
|
|
+ }
|
|
|
|
|
+ case <-time.After(250 * time.Millisecond):
|
|
|
|
|
+ t.Fatalf("split peers not rejected")
|
|
|
|
|
+ }
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
// This test checks that received transactions are added to the local pool.
|
|
// This test checks that received transactions are added to the local pool.
|
|
|
-func TestRecvTransactions62(t *testing.T) { testRecvTransactions(t, 62) }
|
|
|
|
|
func TestRecvTransactions63(t *testing.T) { testRecvTransactions(t, 63) }
|
|
func TestRecvTransactions63(t *testing.T) { testRecvTransactions(t, 63) }
|
|
|
|
|
+func TestRecvTransactions64(t *testing.T) { testRecvTransactions(t, 64) }
|
|
|
|
|
|
|
|
func testRecvTransactions(t *testing.T, protocol int) {
|
|
func testRecvTransactions(t *testing.T, protocol int) {
|
|
|
txAdded := make(chan []*types.Transaction)
|
|
txAdded := make(chan []*types.Transaction)
|
|
@@ -121,8 +272,8 @@ func testRecvTransactions(t *testing.T, protocol int) {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
// This test checks that pending transactions are sent.
|
|
// This test checks that pending transactions are sent.
|
|
|
-func TestSendTransactions62(t *testing.T) { testSendTransactions(t, 62) }
|
|
|
|
|
func TestSendTransactions63(t *testing.T) { testSendTransactions(t, 63) }
|
|
func TestSendTransactions63(t *testing.T) { testSendTransactions(t, 63) }
|
|
|
|
|
+func TestSendTransactions64(t *testing.T) { testSendTransactions(t, 64) }
|
|
|
|
|
|
|
|
func testSendTransactions(t *testing.T, protocol int) {
|
|
func testSendTransactions(t *testing.T, protocol int) {
|
|
|
pm, _ := newTestProtocolManagerMust(t, downloader.FullSync, 0, nil, nil)
|
|
pm, _ := newTestProtocolManagerMust(t, downloader.FullSync, 0, nil, nil)
|