Check HAProxyMessage type is PROXY (#7864)

This commit is contained in:
PureGero 2022-06-02 07:58:22 +10:00 committed by GitHub
parent 78e6431008
commit 04e1b070ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -19,7 +19,7 @@ index 1ac6cf51f2682d5eb14fe19646e79f6617d492dd..fafbebbb5e8c1a381b673f97f1fa2106
+ } + }
} }
diff --git a/src/main/java/net/minecraft/server/network/ServerConnectionListener.java b/src/main/java/net/minecraft/server/network/ServerConnectionListener.java diff --git a/src/main/java/net/minecraft/server/network/ServerConnectionListener.java b/src/main/java/net/minecraft/server/network/ServerConnectionListener.java
index 058fb3696c7ece4a7b6971886b1760b26add733b..bd378482b82c15cba1eeaa620521ad9ed460a1da 100644 index 058fb3696c7ece4a7b6971886b1760b26add733b..98286dd27fc3562ca55ec44cc6e5eb5157269942 100644
--- a/src/main/java/net/minecraft/server/network/ServerConnectionListener.java --- a/src/main/java/net/minecraft/server/network/ServerConnectionListener.java
+++ b/src/main/java/net/minecraft/server/network/ServerConnectionListener.java +++ b/src/main/java/net/minecraft/server/network/ServerConnectionListener.java
@@ -109,6 +109,12 @@ public class ServerConnectionListener { @@ -109,6 +109,12 @@ public class ServerConnectionListener {
@ -35,7 +35,7 @@ index 058fb3696c7ece4a7b6971886b1760b26add733b..bd378482b82c15cba1eeaa620521ad9e
this.channels.add(((ServerBootstrap) ((ServerBootstrap) (new ServerBootstrap()).channel(oclass)).childHandler(new ChannelInitializer<Channel>() { this.channels.add(((ServerBootstrap) ((ServerBootstrap) (new ServerBootstrap()).channel(oclass)).childHandler(new ChannelInitializer<Channel>() {
protected void initChannel(Channel channel) { protected void initChannel(Channel channel) {
try { try {
@@ -122,6 +128,28 @@ public class ServerConnectionListener { @@ -122,6 +128,30 @@ public class ServerConnectionListener {
int j = ServerConnectionListener.this.server.getRateLimitPacketsPerSecond(); int j = ServerConnectionListener.this.server.getRateLimitPacketsPerSecond();
Object object = j > 0 ? new RateKickingConnection(j) : new Connection(PacketFlow.SERVERBOUND); Object object = j > 0 ? new RateKickingConnection(j) : new Connection(PacketFlow.SERVERBOUND);
@ -46,13 +46,15 @@ index 058fb3696c7ece4a7b6971886b1760b26add733b..bd378482b82c15cba1eeaa620521ad9e
+ @Override + @Override
+ public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { + public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
+ if (msg instanceof io.netty.handler.codec.haproxy.HAProxyMessage message) { + if (msg instanceof io.netty.handler.codec.haproxy.HAProxyMessage message) {
+ String realaddress = message.sourceAddress(); + if (message.command() == io.netty.handler.codec.haproxy.HAProxyCommand.PROXY) {
+ int realport = message.sourcePort(); + String realaddress = message.sourceAddress();
+ int realport = message.sourcePort();
+ +
+ SocketAddress socketaddr = new java.net.InetSocketAddress(realaddress, realport); + SocketAddress socketaddr = new java.net.InetSocketAddress(realaddress, realport);
+ +
+ Connection connection = (Connection) channel.pipeline().get("packet_handler"); + Connection connection = (Connection) channel.pipeline().get("packet_handler");
+ connection.address = socketaddr; + connection.address = socketaddr;
+ }
+ } else { + } else {
+ super.channelRead(ctx, msg); + super.channelRead(ctx, msg);
+ } + }