From 489242b1ef0e7a10b5c86ce3ffa7cad151be8665 Mon Sep 17 00:00:00 2001
From: RuriRyan <ryan@btsoft.eu>
Date: Wed, 25 Jun 2014 23:29:08 +0200
Subject: [PATCH] Fixed RemoteQuery, which broke with the Epoll update

---
 proxy/src/main/java/net/md_5/bungee/BungeeCord.java        | 2 +-
 .../src/main/java/net/md_5/bungee/netty/PipelineUtils.java | 7 +++++++
 query/src/main/java/net/md_5/bungee/query/RemoteQuery.java | 6 +++---
 3 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/proxy/src/main/java/net/md_5/bungee/BungeeCord.java b/proxy/src/main/java/net/md_5/bungee/BungeeCord.java
index 484671f9..ef63068d 100644
--- a/proxy/src/main/java/net/md_5/bungee/BungeeCord.java
+++ b/proxy/src/main/java/net/md_5/bungee/BungeeCord.java
@@ -280,7 +280,7 @@ public class BungeeCord extends ProxyServer
                         }
                     }
                 };
-                new RemoteQuery( this, info ).start( new InetSocketAddress( info.getHost().getAddress(), info.getQueryPort() ), eventLoops, bindListener );
+                new RemoteQuery( this, info ).start(PipelineUtils.getDatagramChannel(), new InetSocketAddress( info.getHost().getAddress(), info.getQueryPort() ), eventLoops, bindListener );
             }
         }
     }
diff --git a/proxy/src/main/java/net/md_5/bungee/netty/PipelineUtils.java b/proxy/src/main/java/net/md_5/bungee/netty/PipelineUtils.java
index 8239799f..c1afcbd2 100644
--- a/proxy/src/main/java/net/md_5/bungee/netty/PipelineUtils.java
+++ b/proxy/src/main/java/net/md_5/bungee/netty/PipelineUtils.java
@@ -7,10 +7,12 @@ import io.netty.channel.ChannelInitializer;
 import io.netty.channel.ChannelOption;
 import io.netty.channel.EventLoopGroup;
 import io.netty.channel.ServerChannel;
+import io.netty.channel.epoll.EpollDatagramChannel;
 import io.netty.channel.epoll.EpollEventLoopGroup;
 import io.netty.channel.epoll.EpollServerSocketChannel;
 import io.netty.channel.epoll.EpollSocketChannel;
 import io.netty.channel.nio.NioEventLoopGroup;
+import io.netty.channel.socket.nio.NioDatagramChannel;
 import io.netty.channel.socket.nio.NioServerSocketChannel;
 import io.netty.channel.socket.nio.NioSocketChannel;
 import io.netty.handler.timeout.ReadTimeoutHandler;
@@ -114,6 +116,11 @@ public class PipelineUtils
     {
         return epoll ? EpollSocketChannel.class : NioSocketChannel.class;
     }
+    
+    public static Class<? extends Channel> getDatagramChannel()
+    {
+        return epoll ? EpollDatagramChannel.class : NioDatagramChannel.class;
+    }
 
     public final static class Base extends ChannelInitializer<Channel>
     {
diff --git a/query/src/main/java/net/md_5/bungee/query/RemoteQuery.java b/query/src/main/java/net/md_5/bungee/query/RemoteQuery.java
index 43b02753..8becc26a 100644
--- a/query/src/main/java/net/md_5/bungee/query/RemoteQuery.java
+++ b/query/src/main/java/net/md_5/bungee/query/RemoteQuery.java
@@ -1,9 +1,9 @@
 package net.md_5.bungee.query;
 
 import io.netty.bootstrap.Bootstrap;
+import io.netty.channel.Channel;
 import io.netty.channel.ChannelFutureListener;
 import io.netty.channel.EventLoopGroup;
-import io.netty.channel.socket.nio.NioDatagramChannel;
 import java.net.InetSocketAddress;
 import lombok.RequiredArgsConstructor;
 import net.md_5.bungee.api.ProxyServer;
@@ -16,10 +16,10 @@ public class RemoteQuery
     private final ProxyServer bungee;
     private final ListenerInfo listener;
 
-    public void start(InetSocketAddress address, EventLoopGroup eventLoop, ChannelFutureListener future)
+    public void start(Class<? extends Channel> channel, InetSocketAddress address, EventLoopGroup eventLoop, ChannelFutureListener future)
     {
         new Bootstrap()
-                .channel( NioDatagramChannel.class )
+                .channel( channel )
                 .group( eventLoop )
                 .handler( new QueryHandler( bungee, listener ) )
                 .localAddress( address )