From cd5a215dc2d018d5f44ab7f7e3cf58cfcba1f840 Mon Sep 17 00:00:00 2001
From: Connection Refused <cutecr@pm.me>
Date: Sun, 17 Oct 2021 16:13:43 +0800
Subject: [PATCH] [Redirector] Optimize code

---
 Redirector/DNSHandler.cpp | 2 ++
 Redirector/DNSHandler.h   | 1 -
 Redirector/EventHandler.h | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/Redirector/DNSHandler.cpp b/Redirector/DNSHandler.cpp
index bb214f29..48e390f8 100644
--- a/Redirector/DNSHandler.cpp
+++ b/Redirector/DNSHandler.cpp
@@ -1,5 +1,7 @@
 #include "DNSHandler.h"
 
+#include "SocksHelper.h"
+
 extern string dnsHost;
 extern USHORT dnsPort;
 
diff --git a/Redirector/DNSHandler.h b/Redirector/DNSHandler.h
index e7548052..9800cbea 100644
--- a/Redirector/DNSHandler.h
+++ b/Redirector/DNSHandler.h
@@ -2,7 +2,6 @@
 #ifndef DNSHANDLER_H
 #define DNSHANDLER_H
 #include "Based.h"
-#include "SocksHelper.h"
 
 namespace DNSHandler
 {
diff --git a/Redirector/EventHandler.h b/Redirector/EventHandler.h
index f84c70a2..c07e2c2c 100644
--- a/Redirector/EventHandler.h
+++ b/Redirector/EventHandler.h
@@ -24,6 +24,6 @@ void udpCanReceive(ENDPOINT_ID id);
 void udpReceive(ENDPOINT_ID id, const unsigned char* target, const char* buffer, int length, PNF_UDP_OPTIONS options);
 void udpClosed(ENDPOINT_ID id, PNF_UDP_CONN_INFO info);
 
-void udpReceiveHandler(ENDPOINT_ID id, SocksHelper::PUDP conn, PNF_UDP_OPTIONS options);
+void udpReceiveHandler(ENDPOINT_ID id, SocksHelper::PUDP remote, PNF_UDP_OPTIONS options);
 
 #endif