[Yamux] Clear caches when connection is closed (#321)

libp2p · Sep 14, 2023 · 7d49700 · 7d49700
1 parent 6f7e485
commit 7d49700
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/libp2p/src/main/kotlin/io/libp2p/mux/yamux/YamuxHandler.kt b/libp2p/src/main/kotlin/io/libp2p/mux/yamux/YamuxHandler.kt
@@ -58,6 +58,14 @@ open class YamuxHandler(
         }
     }
 
+    override fun channelUnregistered(ctx: ChannelHandlerContext?) {
+        sendWindowSizes.clear()
+        sendBuffers.values.forEach { it.close() }
+        sendBuffers.clear()
+        receiveWindowSizes.clear()
+        super.channelUnregistered(ctx)
+    }
+
     override fun channelRead(ctx: ChannelHandlerContext, msg: Any) {
         msg as YamuxFrame
         when (msg.type) {