806 lines
27 KiB
C++
806 lines
27 KiB
C++
#include "pch.h"
|
|
#include "tunnel.h"
|
|
#include "usrerr.h"
|
|
#include <strsafe.h>
|
|
#include <tchar.h>
|
|
#include <shlwapi.h>
|
|
|
|
#include "globalcfg.h"
|
|
#include "misc.h"
|
|
#include "network.h"
|
|
|
|
#pragma comment(lib, "Shlwapi.lib")
|
|
#pragma comment(lib, "Winmm.lib")
|
|
|
|
#define WG_NIC_DISCRIPT TEXT("WireGuard Tunnel")
|
|
|
|
static NET_SHARE_MODE g_CurShareMode = ICS_SHARE_MODE;
|
|
|
|
constexpr auto WINENVBUF_SIZE = (4096);
|
|
|
|
NET_SHARE_MODE GetCurrentNetShareMode() {
|
|
return g_CurShareMode;
|
|
}
|
|
|
|
void SetCurrentNetShareMode(NET_SHARE_MODE shareMode) {
|
|
g_CurShareMode = shareMode;
|
|
}
|
|
|
|
int GetWireGuardWorkMode(bool *pIsWorkServer) {
|
|
if (pIsWorkServer == nullptr) {
|
|
SPDLOG_ERROR(TEXT("Input pIsWorkServer params error"));
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
*pIsWorkServer = GetGlobalCfgInfo()->isWorkServer;
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
int WireGuardInstallDefaultServerService(bool bInstall) {
|
|
TCHAR cfgVal[MAX_PATH];
|
|
DWORD retCode;
|
|
|
|
GetPrivateProfileString(CFG_WIREGUARD_SECTION,
|
|
CFG_WGCFG_PATH,
|
|
TEXT(""),
|
|
cfgVal,
|
|
MAX_PATH,
|
|
GetGlobalCfgInfo()->cfgPath);
|
|
|
|
if (lstrlen(cfgVal) > 0) {
|
|
if (PathFileExists(cfgVal)) {
|
|
TCHAR cmdBuf[MAX_PATH];
|
|
TCHAR svrName[MAX_PATH];
|
|
int ret;
|
|
|
|
StringCbCopy(svrName, MAX_PATH, cfgVal);
|
|
PathStripPath(svrName);
|
|
PathRemoveExtension(svrName);
|
|
|
|
if (bInstall) {
|
|
// 安装服务
|
|
StringCbPrintf(cmdBuf,
|
|
MAX_PATH,
|
|
TEXT("\"%s\" /installtunnelservice \"%s\""),
|
|
GetGlobalCfgInfo()->wireguardCfg.wireguardPath,
|
|
cfgVal);
|
|
} else {
|
|
// 卸载服务
|
|
|
|
StringCbPrintf(cmdBuf,
|
|
MAX_PATH,
|
|
TEXT("\"%s\" /uninstalltunnelservice %s"),
|
|
GetGlobalCfgInfo()->wireguardCfg.wireguardPath,
|
|
svrName);
|
|
}
|
|
|
|
if ((ret = RunCommand(cmdBuf, nullptr, 0, &retCode)) != ERR_SUCCESS) {
|
|
SPDLOG_ERROR(TEXT("Run command [{0}] error: {1}"), cmdBuf, ret);
|
|
return -ERR_CALL_SHELL;
|
|
}
|
|
|
|
SPDLOG_DEBUG(TEXT("Run command [{0}]"), cmdBuf);
|
|
|
|
if (bInstall) {
|
|
int retry = 10;
|
|
do {
|
|
ret = WaitNetAdapterConnected(svrName, 1000);
|
|
} while (ret != ERR_SUCCESS && retry--);
|
|
}
|
|
|
|
return ret;
|
|
} else {
|
|
SPDLOG_ERROR(TEXT("WireGuard configure file [{0}] not found"), cfgVal);
|
|
return -ERR_FILE_NOT_EXISTS;
|
|
}
|
|
} else {
|
|
SPDLOG_ERROR(TEXT("Configure [{0}] = {1} not found"), CFG_WGCFG_PATH, cfgVal);
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
}
|
|
|
|
int WireGuardInstallServerService(const TCHAR *pTunnelCfgPath) {
|
|
// 卸载服务
|
|
TCHAR cmdBuf[MAX_PATH];
|
|
int ret;
|
|
DWORD retCode;
|
|
|
|
if (pTunnelCfgPath == nullptr || lstrlen(pTunnelCfgPath) == 0) {
|
|
SPDLOG_ERROR(TEXT("Input pTunnelCfgPath params error"));
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
if (!PathFileExists(pTunnelCfgPath)) {
|
|
SPDLOG_ERROR(TEXT("WireGuard configure file {0} unexists."), pTunnelCfgPath);
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
|
|
// 安装服务
|
|
StringCbPrintf(cmdBuf,
|
|
MAX_PATH,
|
|
TEXT("\"%s\" /installtunnelservice \"%s\""),
|
|
GetGlobalCfgInfo()->wireguardCfg.wireguardPath,
|
|
pTunnelCfgPath);
|
|
|
|
if ((ret = RunCommand(cmdBuf, nullptr, 0, &retCode)) != ERR_SUCCESS) {
|
|
SPDLOG_ERROR(TEXT("Run command [{0}] error: {1}"), cmdBuf, ret);
|
|
return -ERR_CALL_SHELL;
|
|
}
|
|
|
|
SPDLOG_DEBUG(TEXT("Run command [{0}]"), cmdBuf);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
int WireGuardUnInstallServerService(const TCHAR *pTunnelName) {
|
|
// 卸载服务
|
|
TCHAR cmdBuf[MAX_PATH];
|
|
TCHAR svrName[MAX_PATH];
|
|
int ret;
|
|
DWORD retCode;
|
|
|
|
if (pTunnelName == nullptr || lstrlen(pTunnelName) == 0) {
|
|
SPDLOG_ERROR(TEXT("Input pTunnelName params error"));
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
StringCbCopy(svrName, MAX_PATH, pTunnelName);
|
|
PathStripPath(svrName);
|
|
PathRemoveExtension(svrName);
|
|
|
|
StringCbPrintf(cmdBuf,
|
|
MAX_PATH,
|
|
TEXT("\"%s\" /uninstalltunnelservice %s"),
|
|
GetGlobalCfgInfo()->wireguardCfg.wireguardPath,
|
|
svrName);
|
|
|
|
if ((ret = RunCommand(cmdBuf, nullptr, 0, &retCode)) != ERR_SUCCESS) {
|
|
SPDLOG_ERROR(TEXT("Run command [{0}] error: {1}"), cmdBuf, ret);
|
|
return -ERR_CALL_SHELL;
|
|
}
|
|
|
|
SPDLOG_DEBUG(TEXT("Run command [{0}]"), cmdBuf);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
int IsWireGuardServerInstalled(bool *pIsInstalled) {
|
|
DWORD dwStatus;
|
|
int ret;
|
|
|
|
if (pIsInstalled == nullptr) {
|
|
SPDLOG_ERROR(TEXT("Input pIsInstalled params error"));
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
*pIsInstalled = false;
|
|
|
|
ret = GetWindowsServiceStatus(TEXT("WireGuard"), &dwStatus);
|
|
|
|
if (ret == ERR_SUCCESS) {
|
|
switch (dwStatus) {
|
|
case SERVICE_CONTINUE_PENDING:
|
|
case SERVICE_RUNNING:
|
|
case SERVICE_START_PENDING:
|
|
*pIsInstalled = true;
|
|
break;
|
|
default:
|
|
*pIsInstalled = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int IsWireGuardServerRunning(const TCHAR *pIfName, bool *pIsRunning) {
|
|
int ret;
|
|
PNIC_CONTENT pInfo = nullptr;
|
|
int size = 0;
|
|
|
|
if (pIfName == nullptr || lstrlen(pIfName) == 0) {
|
|
SPDLOG_ERROR(TEXT("Input pIfName params error"));
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (pIsRunning == nullptr) {
|
|
SPDLOG_ERROR(TEXT("Input pIsRunning params error"));
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
*pIsRunning = false;
|
|
|
|
ret = GetAllNICInfo(&pInfo, &size);
|
|
if (ret != ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
for (int i = 0; i < size; i++) {
|
|
if (StrNCmp(pInfo[i].NetCardDescription, WG_NIC_DISCRIPT, lstrlen(WG_NIC_DISCRIPT)) == 0 &&
|
|
StrCmp(pInfo[i].NetCardName, pIfName) == 0) {
|
|
*pIsRunning = true;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int WireGuardCreateClientConfig(const PWGCLIENT_CONFIG pWgConfig) {
|
|
const size_t bufSize = 4096 * sizeof(TCHAR);
|
|
const TCHAR cfgFormat[] = TEXT(
|
|
"[Interface]\nPrivateKey = %s\nAddress = %s\n\n[Peer]\nPublicKey = %s\nAllowedIPs = %s\nEndpoint = "
|
|
"%s\nPersistentKeepalive = 30\n");
|
|
TCHAR cfgPath[MAX_PATH];
|
|
size_t length;
|
|
HANDLE hFile;
|
|
TCHAR *pBuf;
|
|
|
|
#pragma region
|
|
if (pWgConfig == nullptr) {
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->Name, 64, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Name error: {0}"), pWgConfig->Name);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->Address, 32, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Address error: {0}"), pWgConfig->Address);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->PrivateKey, 64, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Private key error: {0}"), pWgConfig->PrivateKey);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->SvrPubKey, 64, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Server Public key error: {0}"), pWgConfig->SvrPubKey);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->AllowNet, 256, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Allow Client Network error: {0}"), pWgConfig->AllowNet);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->ServerURL, 256, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Server Network error: {0}"), pWgConfig->ServerURL);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
#pragma endregion 参数检查
|
|
|
|
pBuf = static_cast<TCHAR *>(malloc(bufSize));
|
|
|
|
if (pBuf == nullptr) {
|
|
SPDLOG_ERROR(TEXT("Malloc {1} bytes memory error: {0}"), GetLastError(), bufSize);
|
|
return -ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
memset(pBuf, 0, bufSize);
|
|
|
|
StringCbPrintf(cfgPath,
|
|
MAX_PATH,
|
|
"%s\\%s",
|
|
GetGlobalCfgInfo()->configDirectory,
|
|
GetGlobalCfgInfo()->userCfg.userName);
|
|
|
|
// 如果当前用户配置目录不存在则自动创建
|
|
if (!PathFileExists(cfgPath)) {
|
|
if (!CreateDirectory(cfgPath, nullptr)) {
|
|
SPDLOG_ERROR(TEXT("Create configure directory '{0}' error."), cfgPath);
|
|
return -ERR_CREATE_FILE;
|
|
}
|
|
}
|
|
|
|
StringCbPrintf(cfgPath,
|
|
MAX_PATH,
|
|
"%s\\%s\\%s.conf",
|
|
GetGlobalCfgInfo()->configDirectory,
|
|
GetGlobalCfgInfo()->userCfg.userName,
|
|
pWgConfig->Name);
|
|
|
|
hFile = CreateFile(cfgPath, // name of the write
|
|
GENERIC_WRITE | GENERIC_READ, // open for writing
|
|
FILE_SHARE_READ, // do not share
|
|
nullptr, // default security
|
|
CREATE_ALWAYS, // create new file only
|
|
FILE_ATTRIBUTE_NORMAL, // normal file
|
|
nullptr); // no attr. template
|
|
|
|
if (hFile == INVALID_HANDLE_VALUE) {
|
|
SPDLOG_ERROR(TEXT("CreatFile [{0}] error: {1}"), cfgPath, GetLastError());
|
|
free(pBuf);
|
|
return -ERR_OPEN_FILE;
|
|
}
|
|
|
|
// 保存到配置文件中
|
|
WritePrivateProfileString(CFG_WIREGUARD_SECTION, CFG_WGCFG_PATH, cfgPath, GetGlobalCfgInfo()->cfgPath);
|
|
|
|
if (FAILED(StringCbPrintf(pBuf,
|
|
bufSize,
|
|
cfgFormat,
|
|
pWgConfig->PrivateKey,
|
|
pWgConfig->Address,
|
|
pWgConfig->SvrPubKey,
|
|
pWgConfig->AllowNet,
|
|
pWgConfig->ServerURL))) {
|
|
SPDLOG_ERROR(TEXT("Format string error: {0}"), GetLastError());
|
|
free(pBuf);
|
|
::CloseHandle(hFile);
|
|
return -ERR_MEMORY_STR;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pBuf, bufSize, &length))) {
|
|
SPDLOG_ERROR(TEXT("Get string \'{0}\' length error: {1}"), pBuf, GetLastError());
|
|
free(pBuf);
|
|
::CloseHandle(hFile);
|
|
return -ERR_MEMORY_STR;
|
|
}
|
|
|
|
SPDLOG_DEBUG(TEXT("WG Client Configure:\n{0}"), pBuf);
|
|
|
|
if (!WriteFile(hFile, // open file handle
|
|
pBuf, // start of data to write
|
|
static_cast<DWORD>(length), // number of bytes to write
|
|
nullptr, // number of bytes that were written
|
|
nullptr)) {
|
|
SPDLOG_ERROR(TEXT("WriteFile [{0}] error: {1}"), cfgPath, GetLastError());
|
|
free(pBuf);
|
|
::CloseHandle(hFile);
|
|
return -ERR_OPEN_FILE;
|
|
}
|
|
|
|
StringCbCopy(GetGlobalCfgInfo()->wgClientCfg.wgName, 260, pWgConfig->Name);
|
|
StringCbCopy(GetGlobalCfgInfo()->wgClientCfg.wgIpaddr, MAX_IP_LEN, pWgConfig->Address);
|
|
StringCbCopy(GetGlobalCfgInfo()->wgClientCfg.wgCfgPath, MAX_PATH, cfgPath);
|
|
|
|
::CloseHandle(hFile);
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
int WireGuardCreateServerConfig(const PWGSERVER_CONFIG pWgConfig) {
|
|
const size_t bufSize = 4096 * sizeof(TCHAR);
|
|
const TCHAR cfgFormat[] = TEXT(
|
|
"[Interface]\nAddress = %s\nListenPort = %d\nPrivateKey = %s\n\n[Peer]\nPublicKey = %s\nAllowedIPs = %s\n");
|
|
TCHAR cfgPath[MAX_PATH];
|
|
size_t length;
|
|
HANDLE hFile;
|
|
TCHAR *pBuf;
|
|
|
|
#pragma region
|
|
if (pWgConfig == nullptr) {
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->Name, 64, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Name error: {0}"), pWgConfig->Name);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->Address, 32, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Address error: {0}"), pWgConfig->Address);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->PrivateKey, 64, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Private key error: {0}"), pWgConfig->PrivateKey);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (pWgConfig->ListenPort <= 1024 || pWgConfig->ListenPort >= 65535) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Listen port error: {0}, should be in arrange (1024, 65535)"),
|
|
pWgConfig->ListenPort);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->CliPubKey, 64, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Client Public key error: {0}"), pWgConfig->CliPubKey);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pWgConfig->AllowNet, 256, &length)) || 0 == length) {
|
|
SPDLOG_ERROR(TEXT("WireGuard Allow Client Network error: {0}"), pWgConfig->AllowNet);
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
#pragma endregion 参数检查
|
|
|
|
pBuf = static_cast<TCHAR *>(malloc(bufSize));
|
|
|
|
if (pBuf == nullptr) {
|
|
SPDLOG_ERROR(TEXT("Malloc {1} bytes memory error: {0}"), GetLastError(), bufSize);
|
|
return -ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
memset(pBuf, 0, bufSize);
|
|
|
|
StringCbPrintf(cfgPath,
|
|
MAX_PATH,
|
|
"%s\\%s",
|
|
GetGlobalCfgInfo()->configDirectory,
|
|
GetGlobalCfgInfo()->userCfg.userName);
|
|
|
|
// 如果当前用户配置目录不存在则自动创建
|
|
if (!PathFileExists(cfgPath)) {
|
|
if (!CreateDirectory(cfgPath, nullptr)) {
|
|
SPDLOG_ERROR(TEXT("Create configure directory '{0}' error."), cfgPath);
|
|
return -ERR_CREATE_FILE;
|
|
}
|
|
}
|
|
|
|
StringCbPrintf(cfgPath,
|
|
MAX_PATH,
|
|
"%s\\%s\\%s.conf",
|
|
GetGlobalCfgInfo()->configDirectory,
|
|
GetGlobalCfgInfo()->userCfg.userName,
|
|
pWgConfig->Name);
|
|
|
|
hFile = CreateFile(cfgPath, // name of the write
|
|
GENERIC_WRITE | GENERIC_READ, // open for writing
|
|
FILE_SHARE_READ, // do not share
|
|
nullptr, // default security
|
|
CREATE_ALWAYS, // create new file only
|
|
FILE_ATTRIBUTE_NORMAL, // normal file
|
|
nullptr); // no attr. template
|
|
|
|
if (hFile == INVALID_HANDLE_VALUE) {
|
|
SPDLOG_ERROR(TEXT("CreatFile [{0}] error: {1}"), cfgPath, GetLastError());
|
|
free(pBuf);
|
|
return -ERR_OPEN_FILE;
|
|
}
|
|
|
|
// 清空文件
|
|
SetFilePointer(hFile, 0, nullptr, FILE_BEGIN);
|
|
SetEndOfFile(hFile);
|
|
|
|
WritePrivateProfileString(CFG_WIREGUARD_SECTION, CFG_WGCFG_PATH, cfgPath, GetGlobalCfgInfo()->cfgPath);
|
|
|
|
if (FAILED(StringCbPrintf(pBuf,
|
|
bufSize,
|
|
cfgFormat,
|
|
pWgConfig->Address,
|
|
pWgConfig->ListenPort,
|
|
pWgConfig->PrivateKey,
|
|
pWgConfig->CliPubKey,
|
|
pWgConfig->AllowNet))) {
|
|
SPDLOG_ERROR(TEXT("Format string error: {0}"), GetLastError());
|
|
free(pBuf);
|
|
::CloseHandle(hFile);
|
|
return -ERR_MEMORY_STR;
|
|
}
|
|
|
|
if (FAILED(StringCbLength(pBuf, bufSize, &length))) {
|
|
SPDLOG_ERROR(TEXT("Get string \'{0}\' length error: {1}"), pBuf, GetLastError());
|
|
free(pBuf);
|
|
::CloseHandle(hFile);
|
|
return -ERR_MEMORY_STR;
|
|
}
|
|
|
|
SPDLOG_DEBUG(TEXT("WG Server Configure:\n{0}"), pBuf);
|
|
|
|
if (FALSE ==
|
|
WriteFile(hFile, // open file handle
|
|
pBuf, // start of data to write
|
|
static_cast<DWORD>(length), // number of bytes to write
|
|
nullptr, // number of bytes that were written
|
|
nullptr)) // no overlapped structure)
|
|
{
|
|
SPDLOG_ERROR(TEXT("WriteFile [{0}] error: {1}"), cfgPath, GetLastError());
|
|
free(pBuf);
|
|
::CloseHandle(hFile);
|
|
return -ERR_OPEN_FILE;
|
|
}
|
|
|
|
::CloseHandle(hFile);
|
|
|
|
StringCbCopy(GetGlobalCfgInfo()->wgServerCfg.wgName, 260, pWgConfig->Name);
|
|
StringCbCopy(GetGlobalCfgInfo()->wgServerCfg.wgIpaddr, MAX_IP_LEN, pWgConfig->Address);
|
|
StringCbCopy(GetGlobalCfgInfo()->wgServerCfg.wgCfgPath, MAX_PATH, cfgPath);
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
int GenerateWireguardKeyPairs(TCHAR *pPubKey, int pubkeySize, TCHAR *pPrivKey, int privKeySize) {
|
|
int ret;
|
|
DWORD retCode;
|
|
TCHAR cmdBuffer[MAX_PATH];
|
|
TCHAR cmdResult[MAX_PATH];
|
|
PSDK_CONFIG pCfg = GetGlobalCfgInfo();
|
|
|
|
// WireGuard 不存在或者未配置目录
|
|
if (!pCfg->wireguardCfg.wgExists || !pCfg->wireguardCfg.wireguardExists) {
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
|
|
memset(cmdBuffer, 0, MAX_PATH);
|
|
memset(cmdResult, 0, MAX_PATH);
|
|
|
|
StringCbPrintf(cmdBuffer, MAX_PATH, TEXT("cmd.exe /C \"%s\" genkey"), pCfg->wireguardCfg.wgPath);
|
|
|
|
if ((ret = RunCommand(cmdBuffer, cmdResult, MAX_PATH, &retCode)) != ERR_SUCCESS) {
|
|
SPDLOG_ERROR(TEXT("Run command [{0}] error: {1}"), cmdBuffer, ret);
|
|
return -ERR_CALL_SHELL;
|
|
}
|
|
|
|
SPDLOG_DEBUG(TEXT("Run command [{0}] resutl \'{1}\'"), cmdBuffer, cmdResult);
|
|
|
|
StringCbCopy(pPrivKey, privKeySize, cmdResult);
|
|
memset(cmdBuffer, 0, MAX_PATH);
|
|
StringCbPrintf(cmdBuffer,
|
|
MAX_PATH,
|
|
TEXT("cmd.exe /C echo %s | \"%s\" pubkey"),
|
|
cmdResult,
|
|
pCfg->wireguardCfg.wgPath);
|
|
|
|
memset(cmdResult, 0, MAX_PATH);
|
|
if ((ret = RunCommand(cmdBuffer, cmdResult, MAX_PATH, &retCode)) != ERR_SUCCESS) {
|
|
SPDLOG_ERROR(TEXT("Run command [{0}] error: {1}"), cmdBuffer, ret);
|
|
return -ERR_CALL_SHELL;
|
|
}
|
|
|
|
StringCbCopy(pPubKey, pubkeySize, cmdResult);
|
|
SPDLOG_DEBUG(TEXT("Run command [{0}] resutl \'{1}\'"), cmdBuffer, cmdResult);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
int SetWireguardPath(const TCHAR *pPath) {
|
|
if (pPath == nullptr) {
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
if (PathFileExists(pPath)) {
|
|
TCHAR wgPath[MAX_PATH];
|
|
|
|
SPDLOG_DEBUG(TEXT("Used configure file:{0}"), GetGlobalCfgInfo()->cfgPath);
|
|
|
|
WritePrivateProfileString(CFG_WIREGUARD_SECTION, CFG_WIREGUARD_PATH, pPath, GetGlobalCfgInfo()->cfgPath);
|
|
SPDLOG_DEBUG(TEXT("Save configure: {1} --> {0}"), pPath, CFG_WIREGUARD_PATH);
|
|
|
|
StringCbCopy(wgPath, MAX_PATH, pPath);
|
|
|
|
if (TCHAR *pIndex = _tcsrchr(wgPath, '\\')) {
|
|
*pIndex = 0;
|
|
StringCbCat(wgPath, MAX_PATH, "\\wg.exe");
|
|
WritePrivateProfileString(CFG_WIREGUARD_SECTION, CFG_WG_PATH, wgPath, GetGlobalCfgInfo()->cfgPath);
|
|
SPDLOG_DEBUG(TEXT("Save configure: {1} --> {0}"), wgPath, CFG_WG_PATH);
|
|
}
|
|
|
|
return ERR_SUCCESS;
|
|
} else {
|
|
SPDLOG_ERROR(TEXT("WireGuard not found: {0}"), pPath);
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
}
|
|
|
|
int FindWireguardExe(TCHAR *pFullPath, int maxSize) {
|
|
TCHAR path[MAX_PATH];
|
|
TCHAR wireguardPath[MAX_PATH];
|
|
DWORD dwRet;
|
|
LPSTR pEnvBuf;
|
|
TCHAR *token, *p = nullptr;
|
|
|
|
GetPrivateProfileString(CFG_WIREGUARD_SECTION,
|
|
CFG_WIREGUARD_PATH,
|
|
TEXT(""),
|
|
wireguardPath,
|
|
MAX_PATH,
|
|
GetGlobalCfgInfo()->cfgPath);
|
|
|
|
if (PathFileExists(wireguardPath)) {
|
|
if (pFullPath && maxSize > 0) {
|
|
StringCbCopy(pFullPath, maxSize, wireguardPath);
|
|
}
|
|
|
|
StringCbCopy(GetGlobalCfgInfo()->wireguardCfg.wireguardPath, MAX_PATH, wireguardPath);
|
|
GetGlobalCfgInfo()->wireguardCfg.wireguardExists = TRUE;
|
|
|
|
SPDLOG_DEBUG(TEXT("Ini found WireGuard at: {0}"), wireguardPath);
|
|
|
|
GetPrivateProfileString(CFG_WIREGUARD_SECTION,
|
|
CFG_WG_PATH,
|
|
TEXT(""),
|
|
wireguardPath,
|
|
MAX_PATH,
|
|
GetGlobalCfgInfo()->cfgPath);
|
|
|
|
if (PathFileExists(wireguardPath)) {
|
|
StringCbCopy(GetGlobalCfgInfo()->wireguardCfg.wgPath, MAX_PATH, wireguardPath);
|
|
GetGlobalCfgInfo()->wireguardCfg.wgExists = TRUE;
|
|
SPDLOG_DEBUG(TEXT("Ini found WireGuard Tools at: {0}"), wireguardPath);
|
|
}
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
StringCbCopy(wireguardPath, MAX_PATH, GetGlobalCfgInfo()->systemDirectory);
|
|
PathStripToRoot(wireguardPath);
|
|
StringCbCat(wireguardPath, MAX_PATH, TEXT("Program Files\\WireGuard\\wireguard.exe"));
|
|
|
|
if (PathFileExists(wireguardPath)) {
|
|
// 保存路径到配置文件
|
|
SetWireguardPath(wireguardPath);
|
|
GetGlobalCfgInfo()->wireguardCfg.wireguardExists = TRUE;
|
|
|
|
PathRemoveFileSpec(wireguardPath);
|
|
StringCbCat(wireguardPath, MAX_PATH, TEXT("\\wg.exe"));
|
|
|
|
if (PathFileExists(wireguardPath)) {
|
|
GetGlobalCfgInfo()->wireguardCfg.wgExists = TRUE;
|
|
}
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
// 从环境变量中查找
|
|
pEnvBuf = static_cast<LPSTR>(malloc(WINENVBUF_SIZE));
|
|
if (nullptr == pEnvBuf) {
|
|
SPDLOG_ERROR(TEXT("Malloc {0} bytes memory error"), WINENVBUF_SIZE);
|
|
return -ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
dwRet = GetEnvironmentVariable(TEXT("path"), pEnvBuf, WINENVBUF_SIZE);
|
|
|
|
if (0 == dwRet) {
|
|
DWORD dwErr;
|
|
dwErr = GetLastError();
|
|
if (ERROR_ENVVAR_NOT_FOUND == dwErr) {
|
|
SPDLOG_DEBUG(TEXT("Environment variable path does not exist."));
|
|
free(pEnvBuf);
|
|
return -ERR_FILE_NOT_EXISTS;
|
|
}
|
|
} else if (WINENVBUF_SIZE < dwRet) {
|
|
const auto pBuf = static_cast<LPSTR>(realloc(pEnvBuf, dwRet * sizeof(CHAR)));
|
|
if (nullptr == pBuf) {
|
|
SPDLOG_ERROR(TEXT("Malloc {0} bytes memory error"), dwRet * sizeof(CHAR));
|
|
free(pEnvBuf);
|
|
return -ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
pEnvBuf = pBuf;
|
|
dwRet = GetEnvironmentVariable("path", pEnvBuf, dwRet);
|
|
if (!dwRet) {
|
|
SPDLOG_ERROR(TEXT("GetEnvironmentVariable failed (%d)"), GetLastError());
|
|
free(pEnvBuf);
|
|
return -ERR_FILE_NOT_EXISTS;
|
|
}
|
|
}
|
|
|
|
token = strtok_s(pEnvBuf, TEXT(";"), &p);
|
|
|
|
while (token != nullptr) {
|
|
memset(path, 0, MAX_PATH);
|
|
StringCbPrintfA(path, MAX_PATH, TEXT("%s\\wireguard.exe"), token);
|
|
|
|
if (PathFileExists(path)) {
|
|
if (pFullPath && maxSize > 0) {
|
|
StringCbCopy(pFullPath, maxSize, path);
|
|
}
|
|
|
|
// 保存路径到配置文件
|
|
SetWireguardPath(path);
|
|
SPDLOG_DEBUG(TEXT("Path Environment found WireGuard at: {0}"), path);
|
|
|
|
StringCbCopy(GetGlobalCfgInfo()->wireguardCfg.wireguardPath, MAX_PATH, wireguardPath);
|
|
GetGlobalCfgInfo()->wireguardCfg.wireguardExists = TRUE;
|
|
|
|
memset(path, 0, MAX_PATH);
|
|
StringCbPrintf(path, MAX_PATH, TEXT("%s\\wg.exe"), token);
|
|
|
|
SPDLOG_DEBUG(TEXT("Find WireGuard tools at: {0}"), path);
|
|
|
|
if (PathFileExists(path)) {
|
|
StringCbCopy(GetGlobalCfgInfo()->wireguardCfg.wgPath, MAX_PATH, path);
|
|
GetGlobalCfgInfo()->wireguardCfg.wgExists = TRUE;
|
|
|
|
SPDLOG_DEBUG(TEXT("Path Environment found WireGuard tools at: {0}"), path);
|
|
}
|
|
|
|
//TODO: throw exception by C# call, why??????
|
|
//CloseHandle(hFind);
|
|
free(pEnvBuf);
|
|
return ERR_SUCCESS;
|
|
}
|
|
token = strtok_s(nullptr, TEXT(";"), &p);
|
|
}
|
|
|
|
free(pEnvBuf);
|
|
return -ERR_FILE_NOT_EXISTS;
|
|
}
|
|
|
|
#if 0
|
|
int InstallWindowsNATCommand() {
|
|
TCHAR psCmdPath[MAX_PATH];
|
|
|
|
// 如果已经安装则退出
|
|
if (IsCustomNatPSCmdInstalled()) {
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
StringCbPrintf(psCmdPath,
|
|
MAX_PATH,
|
|
TEXT("%s\\system32\\WindowsPowerShell\\v1.0\\Modules\\wireguard\\wireguard.psm1"),
|
|
GetGlobalCfgInfo()->systemDirectory);
|
|
|
|
const HANDLE hFile = CreateFile(psCmdPath,
|
|
GENERIC_WRITE | GENERIC_READ,
|
|
FILE_SHARE_READ,
|
|
nullptr,
|
|
CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
nullptr);
|
|
|
|
if (hFile == INVALID_HANDLE_VALUE) {
|
|
SPDLOG_ERROR("CreatFile [{0}] error: {1}", psCmdPath, GetLastError());
|
|
return -ERR_OPEN_FILE;
|
|
}
|
|
|
|
const HMODULE hMod = GetModuleHandle(TEXT("NetTunnelSDK.dll"));
|
|
|
|
if (nullptr == hMod) {
|
|
SPDLOG_ERROR(TEXT("Load NetTunnelSDK.dll module error"));
|
|
::CloseHandle(hFile);
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
|
|
const HRSRC hRsrc = FindResource(hMod, MAKEINTRESOURCE(PSCMD_RES_ID), TEXT("TXT"));
|
|
if (nullptr == hRsrc) {
|
|
SPDLOG_ERROR(TEXT("Donot found resource {0} of type {1}"), PSCMD_RES_ID, TEXT("TXT"));
|
|
::CloseHandle(hFile);
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
|
|
const DWORD resSize = SizeofResource(hMod, hRsrc);
|
|
|
|
if (resSize == 0) {
|
|
SPDLOG_ERROR(TEXT("Resource {0} of type {1} is empty"), PSCMD_RES_ID, TEXT("TXT"));
|
|
::CloseHandle(hFile);
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
|
|
const HGLOBAL hGlobal = LoadResource(hMod, hRsrc);
|
|
|
|
if (hGlobal == hRsrc) {
|
|
SPDLOG_ERROR(TEXT("Load resource {0} of type {1} error"), PSCMD_RES_ID, TEXT("TXT"));
|
|
::CloseHandle(hFile);
|
|
return -ERR_ITEM_UNEXISTS;
|
|
}
|
|
|
|
if (const LPVOID pBuffer = LockResource(hGlobal)) {
|
|
if (!WriteFile(hFile, // open file handle
|
|
pBuffer, // start of data to write
|
|
static_cast<DWORD>(resSize), // number of bytes to write
|
|
nullptr, // number of bytes that were written
|
|
nullptr)) {
|
|
SPDLOG_ERROR("WriteFile [{0}] error: {1}", psCmdPath, GetLastError());
|
|
GlobalUnlock(hGlobal);
|
|
::CloseHandle(hFile);
|
|
return -ERR_OPEN_FILE;
|
|
}
|
|
}
|
|
|
|
GlobalUnlock(hGlobal);
|
|
::CloseHandle(hFile);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
int WireGuardNetConnectionSharingEnable() {
|
|
int ret;
|
|
DWORD retCode;
|
|
TCHAR cmdResult[MAX_PATH] = {};
|
|
TCHAR cmdBuf[] = TEXT("PowerShell -NoProfile -NonInteractive -WindowStyle Hidden -ExecutionPolicy Bypass Invoke-Command -ArgumentList 'wg_cli' -ScriptBlock {param($IFNAME);$netShare = New-Object -ComObject HNetCfg.HNetShare;"
|
|
"$privateConnection = $netShare.EnumEveryConnection |? { $netShare.NetConnectionProps.Invoke($_).Name -eq 'wg_cli' };"
|
|
"$privateConfig = $netShare.INetSharingConfigurationForINetConnection.Invoke($privateConnection);"
|
|
"Write-Output $privateConfig}");
|
|
|
|
if ((ret = RunCommand(cmdBuf, cmdResult, MAX_PATH, &retCode)) != ERR_SUCCESS) {
|
|
SPDLOG_ERROR("Run command [{0}] error: {1}", cmdBuf, ret);
|
|
return -ERR_CALL_SHELL;
|
|
}
|
|
|
|
SPDLOG_DEBUG("Run command [{0}] resutl \'{1}\' return {2}", cmdBuf, cmdResult, retCode);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
#endif
|