GitHub/netch
1
0
Fork 0
mirror of https://github.com/netchx/netch.git synced 2026-03-28 18:59:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Redirector] Support sub-process proxy

Browse Source
This commit is contained in:
Connection Refused
2021-10-04 22:06:41 +08:00
parent c0e35287af
commit a28f0dacf9
5 changed files with 54 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
Redirector/EventHandler.cpp
View File

@@ -3,6 +3,7 @@
#include "DNSHandler.h"
#include "TCPHandler.h"
extern bool filterParent;
extern bool filterTCP;
extern bool filterUDP;
extern bool filterDNS;
@@ -82,16 +83,55 @@ bool checkBypassName(DWORD id)
bool checkHandleName(DWORD id)
{
auto name = GetProcessName(id);
for (size_t i = 0; i < handleList.size(); i++)
{
if (regex_search(name, wregex(handleList[i])))
auto name = GetProcessName(id);
for (size_t i = 0; i < handleList.size(); i++)
{
return true;
if (regex_search(name, wregex(handleList[i])))
{
return true;
}
}
}
if (filterParent)
{
PROCESSENTRY32W PE;
memset(&PE, 0, sizeof(PROCESSENTRY32W));
PE.dwSize = sizeof(PROCESSENTRY32W);
auto hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if (hSnapshot == INVALID_HANDLE_VALUE)
{
return false;
}
if (!Process32FirstW(hSnapshot, &PE))
{
CloseHandle(hSnapshot);
return false;
}
do {
if (PE.th32ProcessID == id)
{
auto name = GetProcessName(PE.th32ParentProcessID);
for (size_t i = 0; i < handleList.size(); i++)
{
if (regex_search(name, wregex(handleList[i])))
{
CloseHandle(hSnapshot);
return true;
}
}
}
} while (Process32NextW(hSnapshot, &PE));
CloseHandle(hSnapshot);
}
return false;
}