TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2025-05-04 09:02:28 +00:00
Code Issues Projects Releases Packages Wiki Activity

network: treat unsolicited addr commands as errors

Browse source 
See neo-project/neo#2097.
This commit is contained in:
Roman Khimov 2020-11-25 13:34:38 +03:00
parent 619b6d4132
commit 2ce3c8b75f
4 changed files with 41 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
pkg/network/helper_test.go
View file

@ -227,6 +227,7 @@ type localPeer struct {
t *testing.T
messageHandler func(t *testing.T, msg *Message)
pingSent int
getAddrSent int
}
func newLocalPeer(t *testing.T, s *Server) *localPeer {
@ -323,6 +324,14 @@ func (p *localPeer) IsFullNode() bool {
return p.isFullNode
}
func (p *localPeer) AddGetAddrSent() {
p.getAddrSent++
}
func (p *localPeer) CanProcessAddr() bool {
p.getAddrSent--
return p.getAddrSent >= 0
}
func newTestServer(t *testing.T, serverConfig ServerConfig) *Server {
s := &Server{
ServerConfig: serverConfig,