|
|
@ -2,6 +2,7 @@ package p2p |
|
|
|
|
|
|
|
import ( |
|
|
|
"bytes" |
|
|
|
"fmt" |
|
|
|
"net" |
|
|
|
"sync" |
|
|
|
"testing" |
|
|
@ -9,6 +10,7 @@ import ( |
|
|
|
|
|
|
|
. "github.com/tendermint/go-common" |
|
|
|
cfg "github.com/tendermint/go-config" |
|
|
|
"github.com/tendermint/go-crypto" |
|
|
|
"github.com/tendermint/go-wire" |
|
|
|
) |
|
|
|
|
|
|
@ -92,23 +94,24 @@ func makeSwitchPair(t testing.TB, initSwitch func(int, *Switch) *Switch) (*Switc |
|
|
|
return switches[0], switches[1] |
|
|
|
} |
|
|
|
|
|
|
|
func initSwitchFunc(i int, sw *Switch) *Switch { |
|
|
|
// Make two reactors of two channels each
|
|
|
|
sw.AddReactor("foo", NewTestReactor([]*ChannelDescriptor{ |
|
|
|
&ChannelDescriptor{ID: byte(0x00), Priority: 10}, |
|
|
|
&ChannelDescriptor{ID: byte(0x01), Priority: 10}, |
|
|
|
}, true)) |
|
|
|
sw.AddReactor("bar", NewTestReactor([]*ChannelDescriptor{ |
|
|
|
&ChannelDescriptor{ID: byte(0x02), Priority: 10}, |
|
|
|
&ChannelDescriptor{ID: byte(0x03), Priority: 10}, |
|
|
|
}, true)) |
|
|
|
return sw |
|
|
|
} |
|
|
|
|
|
|
|
func TestSwitches(t *testing.T) { |
|
|
|
s1, s2 := makeSwitchPair(t, func(i int, sw *Switch) *Switch { |
|
|
|
// Make two reactors of two channels each
|
|
|
|
sw.AddReactor("foo", NewTestReactor([]*ChannelDescriptor{ |
|
|
|
&ChannelDescriptor{ID: byte(0x00), Priority: 10}, |
|
|
|
&ChannelDescriptor{ID: byte(0x01), Priority: 10}, |
|
|
|
}, true)) |
|
|
|
sw.AddReactor("bar", NewTestReactor([]*ChannelDescriptor{ |
|
|
|
&ChannelDescriptor{ID: byte(0x02), Priority: 10}, |
|
|
|
&ChannelDescriptor{ID: byte(0x03), Priority: 10}, |
|
|
|
}, true)) |
|
|
|
return sw |
|
|
|
}) |
|
|
|
s1, s2 := makeSwitchPair(t, initSwitchFunc) |
|
|
|
defer s1.Stop() |
|
|
|
defer s2.Stop() |
|
|
|
|
|
|
|
// Lets send a message from s1 to s2.
|
|
|
|
if s1.Peers().Size() != 1 { |
|
|
|
t.Errorf("Expected exactly 1 peer in s1, got %v", s1.Peers().Size()) |
|
|
|
} |
|
|
@ -116,6 +119,7 @@ func TestSwitches(t *testing.T) { |
|
|
|
t.Errorf("Expected exactly 1 peer in s2, got %v", s2.Peers().Size()) |
|
|
|
} |
|
|
|
|
|
|
|
// Lets send some messages
|
|
|
|
ch0Msg := "channel zero" |
|
|
|
ch1Msg := "channel foo" |
|
|
|
ch2Msg := "channel bar" |
|
|
@ -156,6 +160,67 @@ func TestSwitches(t *testing.T) { |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func TestConnAddrFilter(t *testing.T) { |
|
|
|
s1 := makeSwitch(1, "testing", "123.123.123", initSwitchFunc) |
|
|
|
s2 := makeSwitch(1, "testing", "123.123.123", initSwitchFunc) |
|
|
|
|
|
|
|
c1, c2 := net.Pipe() |
|
|
|
|
|
|
|
s1.SetAddrFilter(func(addr net.Addr) error { |
|
|
|
if addr.String() == c1.RemoteAddr().String() { |
|
|
|
return fmt.Errorf("Error: pipe is blacklisted") |
|
|
|
} |
|
|
|
return nil |
|
|
|
}) |
|
|
|
|
|
|
|
// connect to good peer
|
|
|
|
go s1.AddPeerWithConnection(c1, false) // AddPeer is blocking, requires handshake.
|
|
|
|
go s2.AddPeerWithConnection(c2, true) |
|
|
|
|
|
|
|
// Wait for things to happen, peers to get added...
|
|
|
|
time.Sleep(100 * time.Millisecond * time.Duration(4)) |
|
|
|
|
|
|
|
defer s1.Stop() |
|
|
|
defer s2.Stop() |
|
|
|
if s1.Peers().Size() != 0 { |
|
|
|
t.Errorf("Expected s1 not to connect to peers, got %d", s1.Peers().Size()) |
|
|
|
} |
|
|
|
if s2.Peers().Size() != 0 { |
|
|
|
t.Errorf("Expected s2 not to connect to peers, got %d", s2.Peers().Size()) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
func TestConnPubKeyFilter(t *testing.T) { |
|
|
|
s1 := makeSwitch(1, "testing", "123.123.123", initSwitchFunc) |
|
|
|
s2 := makeSwitch(1, "testing", "123.123.123", initSwitchFunc) |
|
|
|
|
|
|
|
c1, c2 := net.Pipe() |
|
|
|
|
|
|
|
// set pubkey filter
|
|
|
|
s1.SetPubKeyFilter(func(pubkey crypto.PubKeyEd25519) error { |
|
|
|
if bytes.Equal(pubkey.Bytes(), s2.nodeInfo.PubKey.Bytes()) { |
|
|
|
return fmt.Errorf("Error: pipe is blacklisted") |
|
|
|
} |
|
|
|
return nil |
|
|
|
}) |
|
|
|
|
|
|
|
// connect to good peer
|
|
|
|
go s1.AddPeerWithConnection(c1, false) // AddPeer is blocking, requires handshake.
|
|
|
|
go s2.AddPeerWithConnection(c2, true) |
|
|
|
|
|
|
|
// Wait for things to happen, peers to get added...
|
|
|
|
time.Sleep(100 * time.Millisecond * time.Duration(4)) |
|
|
|
|
|
|
|
defer s1.Stop() |
|
|
|
defer s2.Stop() |
|
|
|
if s1.Peers().Size() != 0 { |
|
|
|
t.Errorf("Expected s1 not to connect to peers, got %d", s1.Peers().Size()) |
|
|
|
} |
|
|
|
if s2.Peers().Size() != 0 { |
|
|
|
t.Errorf("Expected s2 not to connect to peers, got %d", s2.Peers().Size()) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
func BenchmarkSwitches(b *testing.B) { |
|
|
|
|
|
|
|
b.StopTimer() |
|
|
|