rcmgr: remove a connection only once from the limiter (#2800)

libp2p · May 17, 2024 · f8d74ef · f8d74ef
1 parent 547f9db
commit f8d74ef
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/p2p/host/resource-manager/conn_limiter.go b/p2p/host/resource-manager/conn_limiter.go
@@ -85,7 +85,6 @@ func (cl *connLimiter) addConn(ip netip.Addr) bool {
 			return false
 		}
 		masked := prefix.String()
-
 		counts, ok := countsPerLimit[i][masked]
 		if !ok {
 			if countsPerLimit[i] == nil {

diff --git a/p2p/host/resource-manager/rcmgr.go b/p2p/host/resource-manager/rcmgr.go
@@ -677,10 +677,15 @@ func (s *connectionScope) PeerScope() network.PeerScope {
 }
 
 func (s *connectionScope) Done() {
+	s.Lock()
+	defer s.Unlock()
+	if s.done {
+		return
+	}
 	if s.ip.IsValid() {
 		s.rcmgr.connLimiter.rmConn(s.ip)
 	}
-	s.resourceScope.Done()
+	s.resourceScope.doneUnlocked()
 }
 
 // transferAllowedToStandard transfers this connection scope from being part of

diff --git a/p2p/host/resource-manager/scope.go b/p2p/host/resource-manager/scope.go
@@ -743,10 +743,13 @@ func (s *resourceScope) Done() {
 	s.Lock()
 	defer s.Unlock()
 
+	s.doneUnlocked()
+}
+
+func (s *resourceScope) doneUnlocked() {
 	if s.done {
 		return
 	}
-
 	stat := s.rc.stat()
 	if s.owner != nil {
 		s.owner.ReleaseResources(stat)