From 591e18753d6d2703d6d417947fa0a0083c3e09b9 Mon Sep 17 00:00:00 2001
From: Valentine <21033866+BoomEaro@users.noreply.github.com>
Date: Sat, 15 Feb 2025 22:44:55 +0200
Subject: [PATCH] #3786: Fix missing compressor release when removing
 LengthPrependerAndCompressor handler

---
 .../netty/LengthPrependerAndCompressor.java     | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/proxy/src/main/java/net/md_5/bungee/netty/LengthPrependerAndCompressor.java b/proxy/src/main/java/net/md_5/bungee/netty/LengthPrependerAndCompressor.java
index cb04fc76..90e535c4 100644
--- a/proxy/src/main/java/net/md_5/bungee/netty/LengthPrependerAndCompressor.java
+++ b/proxy/src/main/java/net/md_5/bungee/netty/LengthPrependerAndCompressor.java
@@ -114,6 +114,16 @@ public class LengthPrependerAndCompressor extends MessageToMessageEncoder<ByteBu
         }
     }
 
+    @Override
+    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception
+    {
+        if ( zlib != null )
+        {
+            zlib.free();
+            zlib = null;
+        }
+    }
+
     public void setCompose(boolean compose)
     {
         if ( compose )
@@ -147,6 +157,7 @@ public class LengthPrependerAndCompressor extends MessageToMessageEncoder<ByteBu
             if ( zlib != null )
             {
                 zlib.free();
+                zlib = null;
             }
         }
     }
@@ -180,10 +191,6 @@ public class LengthPrependerAndCompressor extends MessageToMessageEncoder<ByteBu
         {
             return 4;
         }
-        if ( MAX_SUPPORTED_VARINT_LENGTH_LEN < 5 )
-        {
-            throw new IllegalArgumentException( "Packet length " + value + " longer than supported (max. 268435455 for 4 byte varint)" );
-        }
-        return 5;
+        throw new IllegalArgumentException( "Packet length " + value + " longer than supported (max. 268435455 for 4 byte varint)" );
     }
 }