mirror of
https://github.com/SoftEtherVPN/SoftEtherVPN.git
synced 2024-11-23 01:49:53 +03:00
9d29d8813b
This greatly improves performance and reduces the binary's size (~0.2 MB vs ~5 MB). All recent Windows versions are supported, starting with Vista. No dialogs are created, aside from error/warning ones in case of failure. The only dependency (aside from Windows libraries) is libhamcore.
379 lines
7.4 KiB
C
379 lines
7.4 KiB
C
#include "Driver.h"
|
|
|
|
#include "Dialog.h"
|
|
#include "Str.h"
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <Hamcore.h>
|
|
|
|
#ifndef WIN32_LEAN_AND_MEAN
|
|
# define WIN32_LEAN_AND_MEAN
|
|
#endif
|
|
|
|
#include <Windows.h>
|
|
|
|
#include <VersionHelpers.h>
|
|
|
|
const char *GetArch()
|
|
{
|
|
SYSTEM_INFO info;
|
|
GetNativeSystemInfo(&info);
|
|
switch (info.wProcessorArchitecture)
|
|
{
|
|
case PROCESSOR_ARCHITECTURE_AMD64:
|
|
return "x64";
|
|
case PROCESSOR_ARCHITECTURE_INTEL:
|
|
return "x86";
|
|
case PROCESSOR_ARCHITECTURE_ARM64:
|
|
return "arm64";
|
|
case PROCESSOR_ARCHITECTURE_ARM:
|
|
return "arm";
|
|
case PROCESSOR_ARCHITECTURE_IA64:
|
|
return "ia64";
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
const char *GetDriverPath()
|
|
{
|
|
static char path[MAX_PATH];
|
|
|
|
static bool set = false;
|
|
if (set)
|
|
{
|
|
return path;
|
|
}
|
|
|
|
const char *type_folder;
|
|
if (IsWindows10OrGreater())
|
|
{
|
|
type_folder = "Neo6_Win10";
|
|
}
|
|
else if (IsWindows8OrGreater())
|
|
{
|
|
type_folder = "Neo6_Win8";
|
|
}
|
|
else if (IsWindows7OrGreater())
|
|
{
|
|
type_folder = "Neo6";
|
|
}
|
|
else
|
|
{
|
|
type_folder = "Neo";
|
|
}
|
|
|
|
snprintf(path, sizeof(path), "DriverPackages/%s/%s/", type_folder, GetArch());
|
|
|
|
set = true;
|
|
return path;
|
|
}
|
|
|
|
const char *GetTmpPath()
|
|
{
|
|
static char path[MAX_PATH];
|
|
|
|
static bool set = false;
|
|
if (set)
|
|
{
|
|
return path;
|
|
}
|
|
|
|
if (!GetTempPath(sizeof(path), path))
|
|
{
|
|
ShowWarning("GetTmpPath()", "GetTempPath() failed with error %lu!", GetLastError());
|
|
return NULL;
|
|
}
|
|
|
|
set = true;
|
|
return path;
|
|
}
|
|
|
|
void GetCatPath(char *dst, const size_t size, const char *instance)
|
|
{
|
|
if (!dst || size == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (IsWindows10OrGreater())
|
|
{
|
|
if (!instance)
|
|
{
|
|
return;
|
|
}
|
|
|
|
snprintf(dst, size, "%sNeo6_%s_%s.cat", GetDriverPath(), GetArch(), instance);
|
|
}
|
|
else if (IsWindows8OrGreater())
|
|
{
|
|
snprintf(dst, size, "%sinf2.cat", GetDriverPath());
|
|
}
|
|
}
|
|
|
|
void GetInfPath(char *dst, const size_t size, const char *instance)
|
|
{
|
|
if (!dst || size == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (IsWindows8OrGreater())
|
|
{
|
|
if (!instance)
|
|
{
|
|
return;
|
|
}
|
|
|
|
snprintf(dst, size, "%sNeo6_%s_%s.inf", GetDriverPath(), GetArch(), instance);
|
|
}
|
|
else if (IsWindows7OrGreater())
|
|
{
|
|
snprintf(dst, size, "%sNeo6_%s.inf", GetDriverPath(), GetArch());
|
|
}
|
|
else
|
|
{
|
|
snprintf(dst, size, "%sNeo_%s.inf", GetDriverPath(), GetArch());
|
|
}
|
|
}
|
|
|
|
void GetSysPath(char *dst, const size_t size, const char *instance)
|
|
{
|
|
if (!dst || size == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (IsWindows10OrGreater())
|
|
{
|
|
if (!instance)
|
|
{
|
|
return;
|
|
}
|
|
|
|
snprintf(dst, size, "%sNeo6_%s_%s.sys", GetDriverPath(), GetArch(), instance);
|
|
}
|
|
else if (IsWindows7OrGreater())
|
|
{
|
|
snprintf(dst, size, "%sNeo6_%s.sys", GetDriverPath(), GetArch());
|
|
}
|
|
else
|
|
{
|
|
snprintf(dst, size, "%sNeo_%s.sys", GetDriverPath(), GetArch());
|
|
}
|
|
}
|
|
|
|
bool IsInstanceNameOK(HAMCORE *hamcore, const char *instance)
|
|
{
|
|
if (!IsWindows8OrGreater())
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (!hamcore || !instance)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
char path[MAX_PATH];
|
|
GetInfPath(path, sizeof(path), instance);
|
|
|
|
const HAMCORE_FILE *file = HamcoreFind(hamcore, path);
|
|
return file ? true : false;
|
|
}
|
|
|
|
bool IsMacAddressManual()
|
|
{
|
|
return IsWindows8OrGreater();
|
|
}
|
|
|
|
bool PrepareCat(HAMCORE *hamcore, char *dst, const size_t size, const char *instance)
|
|
{
|
|
if (!IsWindows8OrGreater())
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (!hamcore || !dst || size == 0 || !instance)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
char src[MAX_PATH];
|
|
GetCatPath(src, sizeof(src), instance);
|
|
|
|
const HAMCORE_FILE *hamcore_file = HamcoreFind(hamcore, src);
|
|
if (!hamcore_file)
|
|
{
|
|
ShowWarning("PrepareCat()", "%s not found in hamcore archive!", src);
|
|
return false;
|
|
}
|
|
|
|
void *buf = malloc(hamcore_file->OriginalSize);
|
|
if (!HamcoreRead(hamcore, buf, hamcore_file))
|
|
{
|
|
ShowWarning("PrepareCat()", "Failed to read %s from hamcore archive!", src);
|
|
free(buf);
|
|
return false;
|
|
}
|
|
|
|
if (IsWindows10OrGreater())
|
|
{
|
|
snprintf(dst, size, "%s%s", GetTmpPath(), PathFileName(src, false));
|
|
}
|
|
else
|
|
{
|
|
snprintf(dst, size, "%sinf_%s.cat", GetTmpPath(), instance);
|
|
}
|
|
|
|
bool ok = false;
|
|
|
|
HANDLE file = CreateFile(dst, GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (file == INVALID_HANDLE_VALUE)
|
|
{
|
|
ShowWarning("PrepareCat()", "CreateFile() failed to open \"%s\" with error %lu!", dst, GetLastError());
|
|
goto FINAL;
|
|
}
|
|
|
|
DWORD processed;
|
|
ok = WriteFile(file, buf, (DWORD)hamcore_file->OriginalSize, &processed, NULL);
|
|
CloseHandle(file);
|
|
|
|
if (!ok)
|
|
{
|
|
ShowWarning("PrepareCat()", "WriteFile() failed with error %lu!", src, GetLastError());
|
|
DeleteFile(dst);
|
|
}
|
|
FINAL:
|
|
free(buf);
|
|
return ok;
|
|
}
|
|
|
|
bool PrepareInf(HAMCORE *hamcore, char *dst, const size_t size, const char *instance, const char *sys, const char *mac)
|
|
{
|
|
if (!hamcore || !dst || size == 0 || !instance || !sys || !mac)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
char src[MAX_PATH];
|
|
GetInfPath(src, sizeof(src), instance);
|
|
|
|
const HAMCORE_FILE *hamcore_file = HamcoreFind(hamcore, src);
|
|
if (!hamcore_file)
|
|
{
|
|
ShowWarning("PrepareInf()", "%s not found in hamcore archive!", src);
|
|
return false;
|
|
}
|
|
|
|
size_t buf_size = hamcore_file->OriginalSize;
|
|
char *buf = malloc(buf_size);
|
|
|
|
if (!HamcoreRead(hamcore, buf, hamcore_file))
|
|
{
|
|
ShowWarning("PrepareInf()", "Failed to read %s from hamcore archive!", src);
|
|
free(buf);
|
|
return false;
|
|
}
|
|
|
|
if (IsWindows10OrGreater())
|
|
{
|
|
snprintf(dst, size, "%s%s", GetTmpPath(), PathFileName(src, false));
|
|
}
|
|
else if (IsWindows7OrGreater())
|
|
{
|
|
snprintf(dst, size, "%sNeo6_%s_%s.inf", GetTmpPath(), GetArch(), instance);
|
|
}
|
|
else
|
|
{
|
|
snprintf(dst, size, "%sNeo_%s_%s.inf", GetTmpPath(), GetArch(), instance);
|
|
}
|
|
|
|
if (!IsWindows8OrGreater())
|
|
{
|
|
buf = StrReplace(buf, &buf_size, "$TAG_INSTANCE_NAME$", instance, false);
|
|
buf = StrReplace(buf, &buf_size, "$TAG_MAC_ADDRESS$", mac, false);
|
|
buf = StrReplace(buf, &buf_size, "$TAG_SYS_NAME$", PathFileName(sys, true), true);
|
|
}
|
|
|
|
bool ok = false;
|
|
|
|
HANDLE file = CreateFile(dst, GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (file == INVALID_HANDLE_VALUE)
|
|
{
|
|
ShowWarning("PrepareInf()", "CreateFile() failed to open \"%s\" with error %lu!", dst, GetLastError());
|
|
goto FINAL;
|
|
}
|
|
|
|
DWORD processed;
|
|
ok = WriteFile(file, buf, (DWORD)buf_size, &processed, NULL);
|
|
CloseHandle(file);
|
|
|
|
if (!ok)
|
|
{
|
|
ShowWarning("PrepareInf()", "WriteFile() failed with error %lu!", src, GetLastError());
|
|
DeleteFile(dst);
|
|
}
|
|
FINAL:
|
|
free(buf);
|
|
return ok;
|
|
}
|
|
|
|
bool PrepareSys(HAMCORE *hamcore, char *dst, const size_t size, const char *instance)
|
|
{
|
|
if (!hamcore || !dst || size == 0 || !instance)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
char src[MAX_PATH];
|
|
GetSysPath(src, sizeof(src), instance);
|
|
|
|
const HAMCORE_FILE *hamcore_file = HamcoreFind(hamcore, src);
|
|
if (!hamcore_file)
|
|
{
|
|
ShowWarning("PrepareSys()", "%s not found in hamcore archive!", src);
|
|
return false;
|
|
}
|
|
|
|
void *buf = malloc(hamcore_file->OriginalSize);
|
|
if (!HamcoreRead(hamcore, buf, hamcore_file))
|
|
{
|
|
ShowWarning("PrepareSys()", "Failed to read %s from hamcore archive!", src);
|
|
free(buf);
|
|
return false;
|
|
}
|
|
|
|
if (IsWindows10OrGreater())
|
|
{
|
|
snprintf(dst, size, "%s%s", GetTmpPath(), PathFileName(src, false));
|
|
}
|
|
else
|
|
{
|
|
snprintf(dst, size, "%sNeo_%s.sys", GetTmpPath(), instance);
|
|
}
|
|
|
|
bool ok = false;
|
|
|
|
HANDLE file = CreateFile(dst, GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (file == INVALID_HANDLE_VALUE)
|
|
{
|
|
ShowWarning("PrepareSys()", "CreateFile() failed to open \"%s\" with error %lu!", dst, GetLastError());
|
|
goto FINAL;
|
|
}
|
|
|
|
DWORD processed;
|
|
ok = WriteFile(file, buf, (DWORD)hamcore_file->OriginalSize, &processed, NULL);
|
|
CloseHandle(file);
|
|
|
|
if (!ok)
|
|
{
|
|
ShowWarning("PrepareSys()", "WriteFile() failed with error %lu!", src, GetLastError());
|
|
DeleteFile(dst);
|
|
}
|
|
FINAL:
|
|
free(buf);
|
|
return ok;
|
|
}
|