p2p: fix rare deadlock in Stop (#17260)

1 files changed, 2 insertions, 1 deletions

diff --git a/p2p/server.go b/p2p/server.go
index cdb5b1926..8f860d8f1 100644
--- a/p2p/server.go
+++ b/p2p/server.go
@@ -340,8 +340,8 @@ func (srv *Server) makeSelf(listener net.Listener, ntab discoverTable) *discover
 // It blocks until all active connections have been closed.
 func (srv *Server) Stop() {
 	srv.lock.Lock()
-	defer srv.lock.Unlock()
 	if !srv.running {
+		srv.lock.Unlock()
 		return
 	}
 	srv.running = false
@@ -350,6 +350,7 @@ func (srv *Server) Stop() {
 		srv.listener.Close()
 	}
 	close(srv.quit)
+	srv.lock.Unlock()
 	srv.loopWG.Wait()
 }