mirror of https://github.com/poanetwork/gecko.git
Added more race tests
This commit is contained in:
parent
3533e91c94
commit
3892a62f28
|
@ -16,6 +16,7 @@ import (
|
||||||
"github.com/ava-labs/gecko/snow/networking/router"
|
"github.com/ava-labs/gecko/snow/networking/router"
|
||||||
"github.com/ava-labs/gecko/snow/validators"
|
"github.com/ava-labs/gecko/snow/validators"
|
||||||
"github.com/ava-labs/gecko/utils"
|
"github.com/ava-labs/gecko/utils"
|
||||||
|
"github.com/ava-labs/gecko/utils/hashing"
|
||||||
"github.com/ava-labs/gecko/utils/logging"
|
"github.com/ava-labs/gecko/utils/logging"
|
||||||
"github.com/ava-labs/gecko/version"
|
"github.com/ava-labs/gecko/version"
|
||||||
)
|
)
|
||||||
|
@ -28,23 +29,22 @@ var (
|
||||||
type testListener struct {
|
type testListener struct {
|
||||||
addr net.Addr
|
addr net.Addr
|
||||||
inbound chan net.Conn
|
inbound chan net.Conn
|
||||||
|
once sync.Once
|
||||||
|
closed chan struct{}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *testListener) Accept() (net.Conn, error) {
|
func (l *testListener) Accept() (net.Conn, error) {
|
||||||
if c, ok := <-l.inbound; ok {
|
select {
|
||||||
|
case c := <-l.inbound:
|
||||||
return c, nil
|
return c, nil
|
||||||
|
case _, _ = <-l.closed:
|
||||||
|
return nil, errClosed
|
||||||
}
|
}
|
||||||
return nil, errClosed
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *testListener) Close() (err error) {
|
func (l *testListener) Close() error {
|
||||||
defer func() {
|
l.once.Do(func() { close(l.closed) })
|
||||||
if r := recover(); r != nil {
|
return nil
|
||||||
err = errClosed
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
close(l.inbound)
|
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *testListener) Addr() net.Addr { return l.addr }
|
func (l *testListener) Addr() net.Addr { return l.addr }
|
||||||
|
@ -62,34 +62,47 @@ func (d *testDialer) Dial(ip utils.IPDesc) (net.Conn, error) {
|
||||||
server := &testConn{
|
server := &testConn{
|
||||||
pendingReads: make(chan []byte, 1<<10),
|
pendingReads: make(chan []byte, 1<<10),
|
||||||
pendingWrites: make(chan []byte, 1<<10),
|
pendingWrites: make(chan []byte, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
local: outbound.addr,
|
local: outbound.addr,
|
||||||
remote: d.addr,
|
remote: d.addr,
|
||||||
}
|
}
|
||||||
client := &testConn{
|
client := &testConn{
|
||||||
pendingReads: server.pendingWrites,
|
pendingReads: server.pendingWrites,
|
||||||
pendingWrites: server.pendingReads,
|
pendingWrites: server.pendingReads,
|
||||||
|
closed: make(chan struct{}),
|
||||||
local: d.addr,
|
local: d.addr,
|
||||||
remote: outbound.addr,
|
remote: outbound.addr,
|
||||||
}
|
}
|
||||||
outbound.inbound <- server
|
|
||||||
return client, nil
|
select {
|
||||||
|
case outbound.inbound <- server:
|
||||||
|
return client, nil
|
||||||
|
default:
|
||||||
|
return nil, errRefused
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type testConn struct {
|
type testConn struct {
|
||||||
partialRead []byte
|
partialRead []byte
|
||||||
pendingReads chan []byte
|
pendingReads chan []byte
|
||||||
pendingWrites chan []byte
|
pendingWrites chan []byte
|
||||||
|
closed chan struct{}
|
||||||
|
once sync.Once
|
||||||
|
|
||||||
local, remote net.Addr
|
local, remote net.Addr
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *testConn) Read(b []byte) (int, error) {
|
func (c *testConn) Read(b []byte) (int, error) {
|
||||||
for len(c.partialRead) == 0 {
|
for len(c.partialRead) == 0 {
|
||||||
read, ok := <-c.pendingReads
|
select {
|
||||||
if !ok {
|
case read, ok := <-c.pendingReads:
|
||||||
|
if !ok {
|
||||||
|
return 0, errClosed
|
||||||
|
}
|
||||||
|
c.partialRead = read
|
||||||
|
case _, _ = <-c.closed:
|
||||||
return 0, errClosed
|
return 0, errClosed
|
||||||
}
|
}
|
||||||
c.partialRead = read
|
|
||||||
}
|
}
|
||||||
|
|
||||||
copy(b, c.partialRead)
|
copy(b, c.partialRead)
|
||||||
|
@ -101,29 +114,22 @@ func (c *testConn) Read(b []byte) (int, error) {
|
||||||
return len(b), nil
|
return len(b), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *testConn) Write(b []byte) (length int, err error) {
|
func (c *testConn) Write(b []byte) (int, error) {
|
||||||
defer func() {
|
|
||||||
if r := recover(); r != nil {
|
|
||||||
err = errClosed
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
newB := make([]byte, len(b))
|
newB := make([]byte, len(b))
|
||||||
copy(newB, b)
|
copy(newB, b)
|
||||||
c.pendingWrites <- newB
|
|
||||||
length = len(b)
|
select {
|
||||||
return
|
case c.pendingWrites <- newB:
|
||||||
|
case _, _ = <-c.closed:
|
||||||
|
return 0, errClosed
|
||||||
|
}
|
||||||
|
|
||||||
|
return len(b), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *testConn) Close() (err error) {
|
func (c *testConn) Close() error {
|
||||||
defer func() {
|
c.once.Do(func() { close(c.closed) })
|
||||||
if r := recover(); r != nil {
|
return nil
|
||||||
err = errClosed
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
close(c.pendingReads)
|
|
||||||
close(c.pendingWrites)
|
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *testConn) LocalAddr() net.Addr { return c.local }
|
func (c *testConn) LocalAddr() net.Addr { return c.local }
|
||||||
|
@ -146,11 +152,11 @@ func (h *testHandler) Disconnected(id ids.ShortID) bool {
|
||||||
|
|
||||||
func TestNewDefaultNetwork(t *testing.T) {
|
func TestNewDefaultNetwork(t *testing.T) {
|
||||||
log := logging.NoLog{}
|
log := logging.NoLog{}
|
||||||
id := ids.ShortEmpty
|
|
||||||
ip := utils.IPDesc{
|
ip := utils.IPDesc{
|
||||||
IP: net.IPv6loopback,
|
IP: net.IPv6loopback,
|
||||||
Port: 0,
|
Port: 0,
|
||||||
}
|
}
|
||||||
|
id := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip.String())))
|
||||||
networkID := uint32(0)
|
networkID := uint32(0)
|
||||||
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
||||||
versionParser := version.NewDefaultParser()
|
versionParser := version.NewDefaultParser()
|
||||||
|
@ -161,6 +167,7 @@ func TestNewDefaultNetwork(t *testing.T) {
|
||||||
Port: 0,
|
Port: 0,
|
||||||
},
|
},
|
||||||
inbound: make(chan net.Conn, 1<<10),
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
}
|
}
|
||||||
caller := &testDialer{
|
caller := &testDialer{
|
||||||
addr: &net.TCPAddr{
|
addr: &net.TCPAddr{
|
||||||
|
@ -206,16 +213,16 @@ func TestEstablishConnection(t *testing.T) {
|
||||||
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
||||||
versionParser := version.NewDefaultParser()
|
versionParser := version.NewDefaultParser()
|
||||||
|
|
||||||
id0 := ids.NewShortID([20]byte{0})
|
|
||||||
ip0 := utils.IPDesc{
|
ip0 := utils.IPDesc{
|
||||||
IP: net.IPv6loopback,
|
IP: net.IPv6loopback,
|
||||||
Port: 0,
|
Port: 0,
|
||||||
}
|
}
|
||||||
id1 := ids.NewShortID([20]byte{1})
|
id0 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip0.String())))
|
||||||
ip1 := utils.IPDesc{
|
ip1 := utils.IPDesc{
|
||||||
IP: net.IPv6loopback,
|
IP: net.IPv6loopback,
|
||||||
Port: 1,
|
Port: 1,
|
||||||
}
|
}
|
||||||
|
id1 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip1.String())))
|
||||||
|
|
||||||
listener0 := &testListener{
|
listener0 := &testListener{
|
||||||
addr: &net.TCPAddr{
|
addr: &net.TCPAddr{
|
||||||
|
@ -223,6 +230,7 @@ func TestEstablishConnection(t *testing.T) {
|
||||||
Port: 0,
|
Port: 0,
|
||||||
},
|
},
|
||||||
inbound: make(chan net.Conn, 1<<10),
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
}
|
}
|
||||||
caller0 := &testDialer{
|
caller0 := &testDialer{
|
||||||
addr: &net.TCPAddr{
|
addr: &net.TCPAddr{
|
||||||
|
@ -237,6 +245,7 @@ func TestEstablishConnection(t *testing.T) {
|
||||||
Port: 1,
|
Port: 1,
|
||||||
},
|
},
|
||||||
inbound: make(chan net.Conn, 1<<10),
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
}
|
}
|
||||||
caller1 := &testDialer{
|
caller1 := &testDialer{
|
||||||
addr: &net.TCPAddr{
|
addr: &net.TCPAddr{
|
||||||
|
@ -327,4 +336,689 @@ func TestEstablishConnection(t *testing.T) {
|
||||||
|
|
||||||
wg0.Wait()
|
wg0.Wait()
|
||||||
wg1.Wait()
|
wg1.Wait()
|
||||||
|
|
||||||
|
err := net0.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net1.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDoubleTrack(t *testing.T) {
|
||||||
|
log := logging.NoLog{}
|
||||||
|
networkID := uint32(0)
|
||||||
|
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
||||||
|
versionParser := version.NewDefaultParser()
|
||||||
|
|
||||||
|
ip0 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
}
|
||||||
|
id0 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip0.String())))
|
||||||
|
ip1 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
}
|
||||||
|
id1 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip1.String())))
|
||||||
|
|
||||||
|
listener0 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller0 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
listener1 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller1 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
|
||||||
|
caller0.outbounds[ip1.String()] = listener1
|
||||||
|
caller1.outbounds[ip0.String()] = listener0
|
||||||
|
|
||||||
|
serverUpgrader := NewIPUpgrader()
|
||||||
|
clientUpgrader := NewIPUpgrader()
|
||||||
|
|
||||||
|
vdrs := validators.NewSet()
|
||||||
|
handler := router.Router(nil)
|
||||||
|
|
||||||
|
net0 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id0,
|
||||||
|
ip0,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener0,
|
||||||
|
caller0,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net0)
|
||||||
|
|
||||||
|
net1 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id1,
|
||||||
|
ip1,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener1,
|
||||||
|
caller1,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net1)
|
||||||
|
|
||||||
|
var (
|
||||||
|
wg0 sync.WaitGroup
|
||||||
|
wg1 sync.WaitGroup
|
||||||
|
)
|
||||||
|
wg0.Add(1)
|
||||||
|
wg1.Add(1)
|
||||||
|
|
||||||
|
h0 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id0) {
|
||||||
|
wg0.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
h1 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id1) {
|
||||||
|
wg1.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
net0.RegisterHandler(h0)
|
||||||
|
net1.RegisterHandler(h1)
|
||||||
|
|
||||||
|
net0.Track(ip1)
|
||||||
|
net0.Track(ip1)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
err := net0.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
go func() {
|
||||||
|
err := net1.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
|
||||||
|
wg0.Wait()
|
||||||
|
wg1.Wait()
|
||||||
|
|
||||||
|
err := net0.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net1.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDoubleClose(t *testing.T) {
|
||||||
|
log := logging.NoLog{}
|
||||||
|
networkID := uint32(0)
|
||||||
|
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
||||||
|
versionParser := version.NewDefaultParser()
|
||||||
|
|
||||||
|
ip0 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
}
|
||||||
|
id0 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip0.String())))
|
||||||
|
ip1 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
}
|
||||||
|
id1 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip1.String())))
|
||||||
|
|
||||||
|
listener0 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller0 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
listener1 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller1 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
|
||||||
|
caller0.outbounds[ip1.String()] = listener1
|
||||||
|
caller1.outbounds[ip0.String()] = listener0
|
||||||
|
|
||||||
|
serverUpgrader := NewIPUpgrader()
|
||||||
|
clientUpgrader := NewIPUpgrader()
|
||||||
|
|
||||||
|
vdrs := validators.NewSet()
|
||||||
|
handler := router.Router(nil)
|
||||||
|
|
||||||
|
net0 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id0,
|
||||||
|
ip0,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener0,
|
||||||
|
caller0,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net0)
|
||||||
|
|
||||||
|
net1 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id1,
|
||||||
|
ip1,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener1,
|
||||||
|
caller1,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net1)
|
||||||
|
|
||||||
|
var (
|
||||||
|
wg0 sync.WaitGroup
|
||||||
|
wg1 sync.WaitGroup
|
||||||
|
)
|
||||||
|
wg0.Add(1)
|
||||||
|
wg1.Add(1)
|
||||||
|
|
||||||
|
h0 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id0) {
|
||||||
|
wg0.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
h1 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id1) {
|
||||||
|
wg1.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
net0.RegisterHandler(h0)
|
||||||
|
net1.RegisterHandler(h1)
|
||||||
|
|
||||||
|
net0.Track(ip1)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
err := net0.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
go func() {
|
||||||
|
err := net1.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
|
||||||
|
wg0.Wait()
|
||||||
|
wg1.Wait()
|
||||||
|
|
||||||
|
err := net0.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net1.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net0.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net1.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRemoveHandlers(t *testing.T) {
|
||||||
|
log := logging.NoLog{}
|
||||||
|
networkID := uint32(0)
|
||||||
|
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
||||||
|
versionParser := version.NewDefaultParser()
|
||||||
|
|
||||||
|
ip0 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
}
|
||||||
|
id0 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip0.String())))
|
||||||
|
ip1 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
}
|
||||||
|
id1 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip1.String())))
|
||||||
|
|
||||||
|
listener0 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller0 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
listener1 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller1 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
|
||||||
|
caller0.outbounds[ip1.String()] = listener1
|
||||||
|
caller1.outbounds[ip0.String()] = listener0
|
||||||
|
|
||||||
|
serverUpgrader := NewIPUpgrader()
|
||||||
|
clientUpgrader := NewIPUpgrader()
|
||||||
|
|
||||||
|
vdrs := validators.NewSet()
|
||||||
|
handler := router.Router(nil)
|
||||||
|
|
||||||
|
net0 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id0,
|
||||||
|
ip0,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener0,
|
||||||
|
caller0,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net0)
|
||||||
|
|
||||||
|
net1 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id1,
|
||||||
|
ip1,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener1,
|
||||||
|
caller1,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net1)
|
||||||
|
|
||||||
|
var (
|
||||||
|
wg0 sync.WaitGroup
|
||||||
|
wg1 sync.WaitGroup
|
||||||
|
)
|
||||||
|
wg0.Add(1)
|
||||||
|
wg1.Add(1)
|
||||||
|
|
||||||
|
h0 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id0) {
|
||||||
|
wg0.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
h1 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id1) {
|
||||||
|
wg1.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
net0.RegisterHandler(h0)
|
||||||
|
net1.RegisterHandler(h1)
|
||||||
|
|
||||||
|
net0.Track(ip1)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
err := net0.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
go func() {
|
||||||
|
err := net1.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
|
||||||
|
wg0.Wait()
|
||||||
|
wg1.Wait()
|
||||||
|
|
||||||
|
h3 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
assert.Equal(t, id0, id)
|
||||||
|
return true
|
||||||
|
},
|
||||||
|
}
|
||||||
|
h4 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
return id.Equals(id0)
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
net0.RegisterHandler(h3)
|
||||||
|
net1.RegisterHandler(h4)
|
||||||
|
|
||||||
|
err := net0.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net1.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTrackConnected(t *testing.T) {
|
||||||
|
log := logging.NoLog{}
|
||||||
|
networkID := uint32(0)
|
||||||
|
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
||||||
|
versionParser := version.NewDefaultParser()
|
||||||
|
|
||||||
|
ip0 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
}
|
||||||
|
id0 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip0.String())))
|
||||||
|
ip1 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
}
|
||||||
|
id1 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip1.String())))
|
||||||
|
|
||||||
|
listener0 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller0 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
listener1 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller1 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
|
||||||
|
caller0.outbounds[ip1.String()] = listener1
|
||||||
|
caller1.outbounds[ip0.String()] = listener0
|
||||||
|
|
||||||
|
serverUpgrader := NewIPUpgrader()
|
||||||
|
clientUpgrader := NewIPUpgrader()
|
||||||
|
|
||||||
|
vdrs := validators.NewSet()
|
||||||
|
handler := router.Router(nil)
|
||||||
|
|
||||||
|
net0 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id0,
|
||||||
|
ip0,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener0,
|
||||||
|
caller0,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net0)
|
||||||
|
|
||||||
|
net1 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id1,
|
||||||
|
ip1,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener1,
|
||||||
|
caller1,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net1)
|
||||||
|
|
||||||
|
var (
|
||||||
|
wg0 sync.WaitGroup
|
||||||
|
wg1 sync.WaitGroup
|
||||||
|
)
|
||||||
|
wg0.Add(1)
|
||||||
|
wg1.Add(1)
|
||||||
|
|
||||||
|
h0 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id0) {
|
||||||
|
wg0.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
h1 := &testHandler{
|
||||||
|
connected: func(id ids.ShortID) bool {
|
||||||
|
if !id.Equals(id1) {
|
||||||
|
wg1.Done()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
net0.RegisterHandler(h0)
|
||||||
|
net1.RegisterHandler(h1)
|
||||||
|
|
||||||
|
net0.Track(ip1)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
err := net0.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
go func() {
|
||||||
|
err := net1.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
|
||||||
|
wg0.Wait()
|
||||||
|
wg1.Wait()
|
||||||
|
|
||||||
|
net0.Track(ip1)
|
||||||
|
|
||||||
|
err := net0.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net1.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTrackConnectedRace(t *testing.T) {
|
||||||
|
log := logging.NoLog{}
|
||||||
|
networkID := uint32(0)
|
||||||
|
appVersion := version.NewDefaultVersion("app", 0, 1, 0)
|
||||||
|
versionParser := version.NewDefaultParser()
|
||||||
|
|
||||||
|
ip0 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
}
|
||||||
|
id0 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip0.String())))
|
||||||
|
ip1 := utils.IPDesc{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
}
|
||||||
|
id1 := ids.NewShortID(hashing.ComputeHash160Array([]byte(ip1.String())))
|
||||||
|
|
||||||
|
listener0 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller0 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 0,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
listener1 := &testListener{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
inbound: make(chan net.Conn, 1<<10),
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
}
|
||||||
|
caller1 := &testDialer{
|
||||||
|
addr: &net.TCPAddr{
|
||||||
|
IP: net.IPv6loopback,
|
||||||
|
Port: 1,
|
||||||
|
},
|
||||||
|
outbounds: make(map[string]*testListener),
|
||||||
|
}
|
||||||
|
|
||||||
|
caller0.outbounds[ip1.String()] = listener1
|
||||||
|
caller1.outbounds[ip0.String()] = listener0
|
||||||
|
|
||||||
|
serverUpgrader := NewIPUpgrader()
|
||||||
|
clientUpgrader := NewIPUpgrader()
|
||||||
|
|
||||||
|
vdrs := validators.NewSet()
|
||||||
|
handler := router.Router(nil)
|
||||||
|
|
||||||
|
net0 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id0,
|
||||||
|
ip0,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener0,
|
||||||
|
caller0,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net0)
|
||||||
|
|
||||||
|
net1 := NewDefaultNetwork(
|
||||||
|
log,
|
||||||
|
id1,
|
||||||
|
ip1,
|
||||||
|
networkID,
|
||||||
|
appVersion,
|
||||||
|
versionParser,
|
||||||
|
listener1,
|
||||||
|
caller1,
|
||||||
|
serverUpgrader,
|
||||||
|
clientUpgrader,
|
||||||
|
vdrs,
|
||||||
|
handler,
|
||||||
|
)
|
||||||
|
assert.NotNil(t, net1)
|
||||||
|
|
||||||
|
net0.Track(ip1)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
err := net0.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
go func() {
|
||||||
|
err := net1.Dispatch()
|
||||||
|
assert.Error(t, err)
|
||||||
|
}()
|
||||||
|
|
||||||
|
err := net0.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
err = net1.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
|
|
@ -364,7 +364,7 @@ func (p *peer) peerList(msg Msg) {
|
||||||
if !ip.Equal(p.net.ip) &&
|
if !ip.Equal(p.net.ip) &&
|
||||||
!ip.IsZero() &&
|
!ip.IsZero() &&
|
||||||
(p.net.allowPrivateIPs || !ip.IsPrivate()) {
|
(p.net.allowPrivateIPs || !ip.IsPrivate()) {
|
||||||
// TODO: this is a vulnerability, perhaps only try to connect once?
|
// TODO: only try to connect once
|
||||||
p.net.Track(ip)
|
p.net.Track(ip)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue