GitHub/netch
1
0
Fork 0
mirror of https://github.com/netchx/netch.git synced 2026-04-01 19:15:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Redirector] Change some name

Browse Source
This commit is contained in:
Connection Refused
2021-10-14 17:44:35 +08:00
parent d423102645
commit a3aeaf1636
5 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
Redirector/DNSHandler.cpp
View File

@@ -38,10 +38,10 @@ void ProcessPacket(ENDPOINT_ID id, SOCKADDR_IN6 target, char* packet, int length
delete[] option;
}
bool DNSHandler::Init()
bool DNSHandler::INIT()
{
memset(&dnsAddr, 0, sizeof(SOCKADDR_IN6));
memset(&dnsAddr, 0, sizeof(dnsAddr));
auto ipv4 = (PSOCKADDR_IN)&dnsAddr;
if (inet_pton(AF_INET, dnsHost.c_str(), &ipv4->sin_addr) == 1)
{
@@ -50,14 +50,15 @@ bool DNSHandler::Init()
return true;
}
if (inet_pton(AF_INET6, dnsHost.c_str(), &dnsAddr.sin6_addr) == 1)
auto ipv6 = (PSOCKADDR_IN6)&dnsAddr;
if (inet_pton(AF_INET6, dnsHost.c_str(), &ipv6->sin6_addr) == 1)
{
dnsAddr.sin6_family = AF_INET6;
dnsAddr.sin6_port = htons(dnsPort);
ipv6->sin6_family = AF_INET6;
ipv6->sin6_port = htons(dnsPort);
return true;
}
puts("[Redirector][DNSHandler::Init] Call WSAStringToAddress failed");
printf("[Redirector][DNSHandler::Init] Convert string to address failed: %d\n", WSAGetLastError());
return false;
}

2
Redirector/DNSHandler.h
View File

@@ -6,7 +6,7 @@
namespace DNSHandler
{
bool Init();
bool INIT();
bool IsDNS(PSOCKADDR_IN6 target);

2
Redirector/EventHandler.cpp
View File

@@ -141,7 +141,7 @@ bool eh_init()
{
CurrentID = GetCurrentProcessId();
if (!DNSHandler::Init())
if (!DNSHandler::INIT())
{
return false;
}

5
Redirector/TCPHandler.cpp
View File

@@ -6,7 +6,7 @@ USHORT tcpListen = 0;
mutex tcpLock;
map<USHORT, SOCKADDR_IN6> tcpContext;
bool TCPHandler::Init()
bool TCPHandler::INIT()
{
auto lg = lock_guard<mutex>(tcpLock);
@@ -75,13 +75,14 @@ bool TCPHandler::Init()
return true;
}
void TCPHandler::Free()
void TCPHandler::FREE()
{
auto lg = lock_guard<mutex>(tcpLock);
if (tcpSocket != INVALID_SOCKET)
{
closesocket(tcpSocket);
tcpSocket = INVALID_SOCKET;
}
tcpListen = 0;

4
Redirector/TCPHandler.h
View File

@@ -6,8 +6,8 @@
namespace TCPHandler
{
bool Init();
void Free();
bool INIT();
void FREE();
void CreateHandler(SOCKADDR_IN6 client, SOCKADDR_IN6 remote);
void DeleteHandler(SOCKADDR_IN6 client);