diff options
| author | Felix Lange <fjl@twurst.com> | 2015-02-13 21:47:05 +0800 |
|---|---|---|
| committer | Felix Lange <fjl@twurst.com> | 2015-02-13 22:06:46 +0800 |
| commit | 22ee366ed6419516eece4436c9f7b5b63ea8a713 (patch) | |
| tree | ec05c6bf1397acdea707cab88f9610de38091d59 /p2p | |
| parent | 5110f80bba13e3758ae1836a88afee123df81e3e (diff) | |
| download | dexon-22ee366ed6419516eece4436c9f7b5b63ea8a713.tar.gz dexon-22ee366ed6419516eece4436c9f7b5b63ea8a713.tar.zst dexon-22ee366ed6419516eece4436c9f7b5b63ea8a713.zip | |
p2p: fix goroutine leak for invalid peers
The deflect logic called Disconnect on the peer, but the peer never ran
and wouldn't process the disconnect request.
Diffstat (limited to 'p2p')
| -rw-r--r-- | p2p/server.go | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/p2p/server.go b/p2p/server.go index e44f3d7ab..f61bb897e 100644 --- a/p2p/server.go +++ b/p2p/server.go @@ -351,19 +351,21 @@ func (srv *Server) startPeer(conn net.Conn, dest *discover.Node) { srvlog.Debugf("Encryption Handshake with %v failed: %v", conn.RemoteAddr(), err) return } - ourID := srv.ntab.Self() p := newPeer(conn, srv.Protocols, srv.Name, &ourID, &remoteID) if ok, reason := srv.addPeer(remoteID, p); !ok { - p.Disconnect(reason) + srvlog.DebugDetailf("Not adding %v (%v)\n", p, reason) + p.politeDisconnect(reason) return } + srvlog.Debugf("Added %v\n", p) if srv.newPeerHook != nil { srv.newPeerHook(p) } - p.run() + discreason := p.run() srv.removePeer(p) + srvlog.Debugf("Removed %v (%v)\n", p, discreason) } func (srv *Server) addPeer(id discover.NodeID, p *Peer) (bool, DiscReason) { @@ -381,14 +383,11 @@ func (srv *Server) addPeer(id discover.NodeID, p *Peer) (bool, DiscReason) { case id == srv.ntab.Self(): return false, DiscSelf } - srvlog.Debugf("Adding %v\n", p) srv.peers[id] = p return true, 0 } -// removes peer: sending disconnect msg, stop peer, remove rom list/table, release slot func (srv *Server) removePeer(p *Peer) { - srvlog.Debugf("Removing %v\n", p) srv.lock.Lock() delete(srv.peers, *p.remoteID) srv.lock.Unlock() |