From 12fcd1d52144cfa18f1608ec599dd509a8333dd8 Mon Sep 17 00:00:00 2001
From: vyzo <vyzo@hackzen.org>
Date: Fri, 14 Jan 2022 13:51:46 +0200
Subject: [PATCH] log, don't panic on resource release bugs

---
 rcmgr.go |  4 ++++
 scope.go | 24 ++++++++++++++++++------
 2 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/rcmgr.go b/rcmgr.go
index 56ef395..06ca3f2 100644
--- a/rcmgr.go
+++ b/rcmgr.go
@@ -9,8 +9,12 @@ import (
 	"github.com/libp2p/go-libp2p-core/network"
 	"github.com/libp2p/go-libp2p-core/peer"
 	"github.com/libp2p/go-libp2p-core/protocol"
+
+	logging "github.com/ipfs/go-log/v2"
 )
 
+var log = logging.Logger("rcmgr")
+
 type resourceManager struct {
 	limits Limiter
 
diff --git a/scope.go b/scope.go
index 2031505..7c15b34 100644
--- a/scope.go
+++ b/scope.go
@@ -89,7 +89,8 @@ func (rc *resources) releaseMemory(size int64) {
 
 	// sanity check for bugs upstream
 	if rc.memory < 0 {
-		panic("BUG: too much memory released")
+		log.Error("BUG: too much memory released")
+		rc.memory = 0
 	}
 }
 
@@ -125,8 +126,13 @@ func (rc *resources) removeStreams(incount, outcount int) {
 	rc.nstreamsIn -= incount
 	rc.nstreamsOut -= outcount
 
-	if rc.nstreamsIn < 0 || rc.nstreamsOut < 0 {
-		panic("BUG: too many streams released")
+	if rc.nstreamsIn < 0 {
+		log.Error("BUG: too many inbound streams released")
+		rc.nstreamsIn = 0
+	}
+	if rc.nstreamsOut < 0 {
+		log.Error("BUG: too many outbound streams released")
+		rc.nstreamsOut = 0
 	}
 }
 
@@ -178,11 +184,17 @@ func (rc *resources) removeConns(incount, outcount, fdcount int) {
 	rc.nconnsOut -= outcount
 	rc.nfd -= fdcount
 
-	if rc.nconnsIn < 0 || rc.nconnsOut < 0 {
-		panic("BUG: too many connections released")
+	if rc.nconnsIn < 0 {
+		log.Error("BUG: too many inbound connections released")
+		rc.nconnsIn = 0
+	}
+	if rc.nconnsOut < 0 {
+		log.Error("BUG: too many outbound connections released")
+		rc.nconnsOut = 0
 	}
 	if rc.nfd < 0 {
-		panic("BUG: too many file descriptors released")
+		log.Error("BUG: too many file descriptors released")
+		rc.nfd = 0
 	}
 }