mirror of
https://github.com/PaperMC/Waterfall.git
synced 2024-11-06 18:51:21 +01:00
50 lines
2.1 KiB
Diff
50 lines
2.1 KiB
Diff
From 7ff984d41446f889b790be1376acae78cddc2fe8 Mon Sep 17 00:00:00 2001
|
|
From: foss-mc <69294560+foss-mc@users.noreply.github.com>
|
|
Date: Wed, 16 Dec 2020 18:10:30 +0800
|
|
Subject: [PATCH] Packet Checks
|
|
|
|
|
|
diff --git a/protocol/src/main/java/net/md_5/bungee/protocol/MinecraftDecoder.java b/protocol/src/main/java/net/md_5/bungee/protocol/MinecraftDecoder.java
|
|
index ec932e92..f1f34626 100644
|
|
--- a/protocol/src/main/java/net/md_5/bungee/protocol/MinecraftDecoder.java
|
|
+++ b/protocol/src/main/java/net/md_5/bungee/protocol/MinecraftDecoder.java
|
|
@@ -39,6 +39,19 @@ public class MinecraftDecoder extends MessageToMessageDecoder<ByteBuf>
|
|
}
|
|
|
|
Protocol.DirectionData prot = ( server ) ? protocol.TO_SERVER : protocol.TO_CLIENT;
|
|
+
|
|
+ // FlameCord - Check size before decoding
|
|
+ if (prot == protocol.TO_SERVER) {
|
|
+ final int readableBytes = in.readableBytes();
|
|
+ final int capacity = in.capacity();
|
|
+
|
|
+ if (readableBytes > 2097152) {
|
|
+ throw new FastDecoderException("Error decoding packet with too many readableBytes: " + readableBytes);
|
|
+ } else if (capacity > 2097152) {
|
|
+ throw new FastDecoderException("Error decoding packet with too big capacity: " + capacity);
|
|
+ }
|
|
+ }
|
|
+
|
|
ByteBuf slice = in.copy(); // Can't slice this one due to EntityMap :(
|
|
|
|
Object packetTypeInfo = null;
|
|
diff --git a/proxy/src/main/java/net/md_5/bungee/netty/HandlerBoss.java b/proxy/src/main/java/net/md_5/bungee/netty/HandlerBoss.java
|
|
index 1351d5d5..d68cc2ce 100644
|
|
--- a/proxy/src/main/java/net/md_5/bungee/netty/HandlerBoss.java
|
|
+++ b/proxy/src/main/java/net/md_5/bungee/netty/HandlerBoss.java
|
|
@@ -79,6 +79,11 @@ public class HandlerBoss extends ChannelInboundHandlerAdapter
|
|
@Override
|
|
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception
|
|
{
|
|
+ // FlameCord - Return if channel isn't active
|
|
+ if (!ctx.channel().isActive()) {
|
|
+ return;
|
|
+ }
|
|
+
|
|
if ( msg instanceof HAProxyMessage )
|
|
{
|
|
HAProxyMessage proxy = (HAProxyMessage) msg;
|
|
--
|
|
2.20.1
|
|
|