reset streams on panic

libp2p · Oct 3, 2024 · aa2caf0 · aa2caf0
1 parent 96d1106
commit aa2caf0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/p2p/protocol/autonatv2/client.go b/p2p/protocol/autonatv2/client.go
@@ -254,6 +254,7 @@ func (ac *client) handleDialBack(s network.Stream) {
 		if rerr := recover(); rerr != nil {
 			fmt.Fprintf(os.Stderr, "caught panic: %s\n%s\n", rerr, debug.Stack())
 		}
+		s.Reset()
 	}()
 
 	if err := s.Scope().SetService(ServiceName); err != nil {

diff --git a/p2p/protocol/autonatv2/server.go b/p2p/protocol/autonatv2/server.go
@@ -93,6 +93,7 @@ func (as *server) handleDialRequest(s network.Stream) {
 	defer func() {
 		if rerr := recover(); rerr != nil {
 			fmt.Fprintf(os.Stderr, "caught panic: %s\n%s\n", rerr, debug.Stack())
+			s.Reset()
 		}
 	}()