From 3516745bffcf824683913dc7ad2f2fd342156428 Mon Sep 17 00:00:00 2001 From: Connection Refused Date: Mon, 4 Oct 2021 19:04:03 +0800 Subject: [PATCH] [Redirector] Optimize code --- Redirector/DNSHandler.cpp | 6 +++--- Redirector/DNSHandler.h | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/Redirector/DNSHandler.cpp b/Redirector/DNSHandler.cpp index bf86a3e8..0b885da4 100644 --- a/Redirector/DNSHandler.cpp +++ b/Redirector/DNSHandler.cpp @@ -5,7 +5,7 @@ extern USHORT dnsPort; SOCKADDR_IN6 dnsAddr; -void ProcessPacket(ENDPOINT_ID id, SOCKADDR_IN6 target, const char* packet, int length, PNF_UDP_OPTIONS options) +void ProcessPacket(ENDPOINT_ID id, SOCKADDR_IN6 target, char* packet, int length, PNF_UDP_OPTIONS option) { char buffer[1024]; @@ -26,7 +26,7 @@ void ProcessPacket(ENDPOINT_ID id, SOCKADDR_IN6 target, const char* packet, int int size = udpConn.Read(NULL, buffer, sizeof(buffer)); if (size != 0 && size != SOCKET_ERROR) { - nf_udpPostReceive(id, (unsigned char*)&target, buffer, size, options); + nf_udpPostReceive(id, (unsigned char*)&target, buffer, size, option); } } } @@ -35,7 +35,7 @@ void ProcessPacket(ENDPOINT_ID id, SOCKADDR_IN6 target, const char* packet, int } delete[] packet; - delete[] options; + delete[] option; } bool DNSHandler::Init() diff --git a/Redirector/DNSHandler.h b/Redirector/DNSHandler.h index 213cdec1..96a854fe 100644 --- a/Redirector/DNSHandler.h +++ b/Redirector/DNSHandler.h @@ -10,7 +10,7 @@ namespace DNSHandler bool IsDNS(PSOCKADDR_IN6 target); - void CreateHandler(ENDPOINT_ID id, PSOCKADDR_IN6 target, const char* packet, int length, PNF_UDP_OPTIONS options); + void CreateHandler(ENDPOINT_ID id, PSOCKADDR_IN6 target, char* packet, int length, PNF_UDP_OPTIONS options); } #endif