1
0
mirror of https://github.com/SoftEtherVPN/SoftEtherVPN.git synced 2024-09-19 18:20:40 +03:00
SoftEtherVPN/src/Mayaqua/Win32.c

3424 lines
64 KiB
C
Raw Normal View History

2017-10-19 05:48:23 +03:00
// SoftEther VPN Source Code - Developer Edition Master Branch
2014-01-04 17:00:08 +04:00
// Mayaqua Kernel
// Win32.c
// Microsoft Windows dependent code
2021-04-05 05:48:25 +03:00
#ifdef OS_WIN32
2014-01-04 17:00:08 +04:00
2021-04-05 05:48:25 +03:00
#include "Win32.h"
#include "FileIO.h"
#include "GlobalConst.h"
#include "Internat.h"
#include "Microsoft.h"
#include "Memory.h"
#include "Object.h"
#include "Str.h"
2014-01-04 17:00:08 +04:00
#include <stdlib.h>
2021-04-05 05:48:25 +03:00
#include <CommCtrl.h>
#include <objbase.h>
#include <process.h>
#include <timeapi.h>
#include <winioctl.h>
2014-01-04 17:00:08 +04:00
static HANDLE heap_handle = NULL;
static HANDLE hstdout = INVALID_HANDLE_VALUE;
static HANDLE hstdin = INVALID_HANDLE_VALUE;
// Thread data for Win32
typedef struct WIN32THREAD
{
HANDLE hThread;
DWORD thread_id;
} WIN32THREAD;
// Thread startup information for Win32
typedef struct WIN32THREADSTARTUPINFO
{
THREAD_PROC *thread_proc;
void *param;
THREAD *thread;
} WIN32THREADSTARTUPINFO;
// Function prototype for Win32
DWORD CALLBACK Win32DefaultThreadProc(void *param);
// Current process handle
static HANDLE hCurrentProcessHandle = NULL;
static CRITICAL_SECTION fasttick_lock;
static UINT64 start_tick = 0;
static bool use_heap_api = false;
static bool win32_is_nt = false;
// File I/O data for Win32
typedef struct WIN32IO
{
HANDLE hFile;
bool WriteMode;
} WIN32IO;
// Mutex data for Win32
typedef struct WIN32MUTEX
{
HANDLE hMutex;
} WIN32MUTEX;
// Set the Thread name
#pragma pack(push,8)
typedef struct tagTHREADNAME_INFO
{
DWORD dwType; // Must be 0x1000.
LPCSTR szName; // Pointer to name (in user addr space).
DWORD dwThreadID; // Thread ID (-1=caller thread).
DWORD dwFlags; // Reserved for future use, must be zero.
} THREADNAME_INFO;
#pragma pack(pop)
// Create a dispatch table
OS_DISPATCH_TABLE *Win32GetDispatchTable()
{
static OS_DISPATCH_TABLE t =
{
Win32Init,
Win32Free,
Win32MemoryAlloc,
Win32MemoryReAlloc,
Win32MemoryFree,
Win32GetTick,
Win32GetSystemTime,
Win32Inc32,
Win32Dec32,
Win32Sleep,
Win32NewLock,
Win32Lock,
Win32Unlock,
Win32DeleteLock,
Win32InitEvent,
Win32SetEvent,
Win32ResetEvent,
Win32WaitEvent,
Win32FreeEvent,
Win32WaitThread,
Win32FreeThread,
Win32InitThread,
Win32ThreadId,
Win32FileOpen,
Win32FileOpenW,
Win32FileCreate,
Win32FileCreateW,
Win32FileWrite,
Win32FileRead,
Win32FileClose,
Win32FileFlush,
Win32FileSize,
Win32FileSeek,
Win32FileDelete,
Win32FileDeleteW,
Win32MakeDir,
Win32MakeDirW,
Win32DeleteDir,
Win32DeleteDirW,
Win32GetCallStack,
Win32GetCallStackSymbolInfo,
Win32FileRename,
Win32FileRenameW,
Win32Run,
Win32RunW,
Win32IsSupportedOs,
Win32GetOsInfo,
Win32Alert,
Win32AlertW,
Win32GetProductId,
Win32SetHighPriority,
Win32RestorePriority,
Win32NewSingleInstance,
Win32FreeSingleInstance,
Win32GetMemInfo,
Win32Yield,
};
return &t;
}
// Set the thread name
void Win32SetThreadName(UINT thread_id, char *name)
{
DWORD ms_vc_exception = 0x406D1388;
THREADNAME_INFO t;
// Validate arguments
if (thread_id == 0 || name == NULL)
{
return;
}
Zero(&t, sizeof(t));
t.dwType = 0x1000;
t.szName = name;
t.dwThreadID = thread_id;
t.dwFlags = 0;
__try
{
RaiseException(ms_vc_exception, 0, sizeof(t) / sizeof(ULONG_PTR), (ULONG_PTR *)&t);
}
__except (EXCEPTION_EXECUTE_HANDLER)
{
}
}
// Initialization function of the new thread
void Win32InitNewThread()
{
static HINSTANCE hDll = NULL;
static bool (WINAPI *_SetThreadLocale)(LCID) = NULL;
if (hDll == NULL)
{
hDll = LoadLibrary("kernel32.dll");
_SetThreadLocale =
(bool (__stdcall *)(LCID))
GetProcAddress(hDll, "SetThreadLocale");
}
if (_SetThreadLocale != NULL)
{
_SetThreadLocale(LOCALE_USER_DEFAULT);
}
}
// Set the compression flag of the folder
bool Win32SetFolderCompressW(wchar_t *path, bool compressed)
{
HANDLE h;
UINT retsize = 0;
USHORT flag;
wchar_t tmp[MAX_PATH];
// Validate arguments
if (path == NULL)
{
return false;
}
if (IsNt() == false)
{
char *path_a = CopyUniToStr(path);
bool ret = Win32SetFolderCompress(path_a, compressed);
Free(path_a);
return ret;
}
InnerFilePathW(tmp, sizeof(tmp), path);
// Open the folder
h = CreateFileW(tmp, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
if (h == INVALID_HANDLE_VALUE)
{
return false;
}
flag = compressed ? COMPRESSION_FORMAT_DEFAULT : COMPRESSION_FORMAT_NONE;
if (DeviceIoControl(h, FSCTL_SET_COMPRESSION, &flag, sizeof(USHORT),
NULL, 0, &retsize, NULL) == false)
{
return false;
}
CloseHandle(h);
return true;
}
bool Win32SetFolderCompress(char *path, bool compressed)
{
HANDLE h;
UINT retsize = 0;
USHORT flag;
char tmp[MAX_PATH];
// Validate arguments
if (path == NULL)
{
return false;
}
InnerFilePath(tmp, sizeof(tmp), path);
// Open the folder
h = CreateFile(tmp, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
if (h == INVALID_HANDLE_VALUE)
{
return false;
}
flag = compressed ? COMPRESSION_FORMAT_DEFAULT : COMPRESSION_FORMAT_NONE;
if (DeviceIoControl(h, FSCTL_SET_COMPRESSION, &flag, sizeof(USHORT),
NULL, 0, &retsize, NULL) == false)
{
return false;
}
CloseHandle(h);
return true;
}
// Get the free space of the disk
bool Win32GetDiskFreeW(wchar_t *path, UINT64 *free_size, UINT64 *used_size, UINT64 *total_size)
{
wchar_t tmp[MAX_SIZE];
UINT count = 0;
UINT i, n, len;
ULARGE_INTEGER v1, v2, v3;
bool ret = false;
// Validate arguments
if (path == NULL)
{
return false;
}
if (IsNt() == false)
{
bool ret;
char *path_a = CopyUniToStr(path);
ret = Win32GetDiskFree(path_a, free_size, used_size, total_size);
Free(path_a);
return ret;
}
Zero(&v1, sizeof(v1));
Zero(&v2, sizeof(v2));
Zero(&v3, sizeof(v3));
NormalizePathW(tmp, sizeof(tmp), path);
// Get the directory name
if (UniStartWith(path, L"\\\\"))
{
count = 4;
}
else
{
count = 1;
}
len = UniStrLen(tmp);
n = 0;
for (i = 0;i < len;i++)
{
if (tmp[i] == L'\\')
{
n++;
if (n >= count)
{
tmp[i + 1] = 0;
break;
}
}
}
if (GetDiskFreeSpaceExW(tmp, &v1, &v2, &v3))
{
ret = true;
}
if (free_size != NULL)
{
*free_size = v1.QuadPart;
}
if (total_size != NULL)
{
*total_size = v2.QuadPart;
}
if (used_size != NULL)
{
*used_size = v2.QuadPart - v1.QuadPart;
}
return ret;
}
bool Win32GetDiskFree(char *path, UINT64 *free_size, UINT64 *used_size, UINT64 *total_size)
{
char tmp[MAX_SIZE];
UINT count = 0;
UINT i, n, len;
ULARGE_INTEGER v1, v2, v3;
bool ret = false;
// Validate arguments
if (path == NULL)
{
return false;
}
Zero(&v1, sizeof(v1));
Zero(&v2, sizeof(v2));
Zero(&v3, sizeof(v3));
NormalizePath(tmp, sizeof(tmp), path);
// Get the directory name
if (StartWith(path, "\\\\"))
{
count = 4;
}
else
{
count = 1;
}
len = StrLen(tmp);
n = 0;
for (i = 0;i < len;i++)
{
if (tmp[i] == '\\')
{
n++;
if (n >= count)
{
tmp[i + 1] = 0;
break;
}
}
}
if (GetDiskFreeSpaceEx(tmp, &v1, &v2, &v3))
{
ret = true;
}
if (free_size != NULL)
{
*free_size = v1.QuadPart;
}
if (total_size != NULL)
{
*total_size = v2.QuadPart;
}
if (used_size != NULL)
{
*used_size = v2.QuadPart - v1.QuadPart;
}
return ret;
}
// Enumeration of directory
DIRLIST *Win32EnumDirEx(char *dirname, COMPARE *compare)
{
DIRLIST *ret;
wchar_t *dirname_w = CopyStrToUni(dirname);
ret = Win32EnumDirExW(dirname_w, compare);
Free(dirname_w);
return ret;
}
DIRLIST *Win32EnumDirExW(wchar_t *dirname, COMPARE *compare)
{
WIN32_FIND_DATAA data_a;
WIN32_FIND_DATAW data_w;
HANDLE h;
wchar_t tmp[MAX_PATH];
wchar_t tmp2[MAX_PATH];
wchar_t dirname2[MAX_PATH];
LIST *o;
DIRLIST *d;
UniStrCpy(tmp2, sizeof(tmp2), dirname);
if (UniStrLen(tmp2) >= 1 && tmp2[UniStrLen(tmp2) - 1] == L'\\')
2014-01-04 17:00:08 +04:00
{
tmp2[UniStrLen(tmp2) - 1] = 0;
}
UniFormat(tmp, sizeof(tmp), L"%s\\*.*", tmp2);
NormalizePathW(tmp, sizeof(tmp), tmp);
NormalizePathW(dirname2, sizeof(dirname2), tmp2);
o = NewListFast(compare);
Zero(&data_a, sizeof(data_a));
Zero(&data_w, sizeof(data_w));
if (IsNt())
{
h = FindFirstFileW(tmp, &data_w);
}
else
{
char *tmp_a = CopyUniToStr(tmp);
h = FindFirstFileA(tmp_a, &data_a);
Free(tmp_a);
}
if (h != INVALID_HANDLE_VALUE)
{
bool b = true;
do
{
if (IsNt() == false)
{
Zero(&data_w, sizeof(data_w));
StrToUni(data_w.cFileName, sizeof(data_w.cFileName), data_a.cFileName);
data_w.dwFileAttributes = data_a.dwFileAttributes;
data_w.ftCreationTime = data_a.ftCreationTime;
data_w.ftLastWriteTime = data_a.ftLastWriteTime;
data_w.nFileSizeHigh = data_a.nFileSizeHigh;
data_w.nFileSizeLow = data_a.nFileSizeLow;
}
if (UniStrCmpi(data_w.cFileName, L"..") != 0 &&
UniStrCmpi(data_w.cFileName, L".") != 0)
{
DIRENT *f = ZeroMalloc(sizeof(DIRENT));
SYSTEMTIME t1, t2;
wchar_t fullpath[MAX_SIZE];
bool ok = false;
f->FileNameW = UniCopyStr(data_w.cFileName);
f->FileName = CopyUniToStr(f->FileNameW);
f->Folder = (data_w.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ? true : false;
CombinePathW(fullpath, sizeof(fullpath), dirname2, f->FileNameW);
// Attempt to get the file information
if (true)
2014-01-04 17:00:08 +04:00
{
HANDLE h = CreateFileW(fullpath, 0,
FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL, OPEN_EXISTING, 0, NULL);
if (h != INVALID_HANDLE_VALUE)
{
BY_HANDLE_FILE_INFORMATION info;
Zero(&info, sizeof(info));
if (MsGetFileInformation(h, &info))
{
Zero(&t1, sizeof(t1));
Zero(&t2, sizeof(t2));
FileTimeToSystemTime(&info.ftCreationTime, &t1);
FileTimeToSystemTime(&info.ftLastWriteTime, &t2);
f->CreateDate = SystemToUINT64(&t1);
f->UpdateDate = SystemToUINT64(&t2);
if (f->Folder == false)
{
f->FileSize = ((UINT64)info.nFileSizeHigh * (UINT64)((UINT64)MAXDWORD + (UINT64)1)) + (UINT64)info.nFileSizeLow;
}
ok = true;
}
CloseHandle(h);
}
}
if (ok == false)
{
Zero(&t1, sizeof(t1));
Zero(&t2, sizeof(t2));
FileTimeToSystemTime(&data_w.ftCreationTime, &t1);
FileTimeToSystemTime(&data_w.ftLastWriteTime, &t2);
f->CreateDate = SystemToUINT64(&t1);
f->UpdateDate = SystemToUINT64(&t2);
if (f->Folder == false)
{
f->FileSize = ((UINT64)data_w.nFileSizeHigh * (UINT64)((UINT64)MAXDWORD + (UINT64)1)) + (UINT64)data_w.nFileSizeLow;
}
}
Add(o, f);
}
Zero(&data_w, sizeof(data_w));
Zero(&data_a, sizeof(data_a));
if (IsNt())
{
b = FindNextFileW(h, &data_w);
}
else
{
b = FindNextFileA(h, &data_a);
}
}
while (b);
FindClose(h);
}
Sort(o);
d = ZeroMalloc(sizeof(DIRLIST));
d->NumFiles = LIST_NUM(o);
d->File = ToArray(o);
ReleaseList(o);
return d;
}
// Get the EXE file name
void Win32GetExeNameW(wchar_t *name, UINT size)
{
// Validate arguments
if (name == NULL)
{
return;
}
if (IsNt() == false)
{
char name_a[MAX_PATH];
Win32GetExeName(name_a, sizeof(name_a));
StrToUni(name, size, name_a);
return;
}
UniStrCpy(name, size, L"");
GetModuleFileNameW(NULL, name, size);
}
void Win32GetExeName(char *name, UINT size)
{
// Validate arguments
if (name == NULL)
{
return;
}
StrCpy(name, size, "");
GetModuleFileName(NULL, name, size);
}
// Get the current directory
void Win32GetCurrentDirW(wchar_t *dir, UINT size)
{
// Validate arguments
if (dir == NULL)
{
return;
}
if (IsNt() == false)
{
char dir_a[MAX_PATH];
Win32GetCurrentDir(dir_a, sizeof(dir_a));
StrToUni(dir, size, dir_a);
return;
}
GetCurrentDirectoryW(size, dir);
}
void Win32GetCurrentDir(char *dir, UINT size)
{
// Validate arguments
if (dir == NULL)
{
return;
}
GetCurrentDirectory(size, dir);
}
// Yield
void Win32Yield()
{
Sleep(0);
}
// Get the memory information
void Win32GetMemInfo(MEMINFO *info)
{
static HINSTANCE hDll = NULL;
static bool (WINAPI *_GlobalMemoryStatusEx)(LPMEMORYSTATUSEX) = NULL;
// Validate arguments
if (info == NULL)
{
return;
}
Zero(info, sizeof(MEMINFO));
if (hDll == NULL)
{
hDll = LoadLibrary("kernel32.dll");
}
if (hDll != NULL)
{
if (_GlobalMemoryStatusEx == NULL)
{
_GlobalMemoryStatusEx =
(bool (__stdcall *)(LPMEMORYSTATUSEX))GetProcAddress(hDll, "GlobalMemoryStatusEx");
}
}
if (_GlobalMemoryStatusEx == NULL)
{
// Old API
MEMORYSTATUS st;
Zero(&st, sizeof(st));
st.dwLength = sizeof(st);
GlobalMemoryStatus(&st);
// Amount of the logical memory
info->TotalMemory = (UINT64)st.dwTotalPageFile;
info->FreeMemory = (UINT64)st.dwAvailPageFile;
info->UsedMemory = info->TotalMemory - info->FreeMemory;
// Amount of the physical memory
info->TotalPhys = (UINT64)st.dwTotalPhys;
info->FreePhys = (UINT64)st.dwAvailPhys;
info->UsedPhys = info->TotalPhys - info->FreePhys;
}
else
{
// New API
MEMORYSTATUSEX st;
Zero(&st, sizeof(st));
st.dwLength = sizeof(st);
_GlobalMemoryStatusEx(&st);
// Amount of the logical memory
info->TotalMemory = st.ullTotalPageFile;
info->FreeMemory = st.ullAvailPageFile;
info->UsedMemory = info->TotalMemory - info->FreeMemory;
// Amount of the physical memory
info->TotalPhys = st.ullTotalPhys;
info->FreePhys = st.ullAvailPhys;
info->UsedPhys = info->TotalPhys - info->FreePhys;
}
}
// Creating a single instance
void *Win32NewSingleInstance(char *instance_name)
{
WIN32MUTEX *ret;
char tmp[MAX_SIZE];
HANDLE hMutex;
// Validate arguments
if (instance_name == NULL)
{
char exe_path[MAX_PATH];
GetModuleFileName(NULL, exe_path, sizeof(exe_path));
HashInstanceName(tmp, sizeof(tmp), exe_path);
instance_name = tmp;
}
hMutex = OpenMutex(MUTEX_ALL_ACCESS, FALSE, instance_name);
if (hMutex != NULL)
{
CloseHandle(hMutex);
return NULL;
}
hMutex = CreateMutex(NULL, FALSE, instance_name);
if (hMutex == NULL)
{
CloseHandle(hMutex);
return NULL;
}
ret = Win32MemoryAlloc(sizeof(WIN32MUTEX));
ret->hMutex = hMutex;
return (void *)ret;
}
// Release the single instance
void Win32FreeSingleInstance(void *data)
{
WIN32MUTEX *m;
// Validate arguments
if (data == NULL)
{
return;
}
m = (WIN32MUTEX *)data;
ReleaseMutex(m->hMutex);
CloseHandle(m->hMutex);
Win32MemoryFree(m);
}
// Raise the priority
void Win32SetHighPriority()
{
SetPriorityClass(GetCurrentProcess(), HIGH_PRIORITY_CLASS);
}
// Restore the priority
void Win32RestorePriority()
{
SetPriorityClass(GetCurrentProcess(), NORMAL_PRIORITY_CLASS);
}
// Get the node information
char* Win32GetProductId()
{
char *product_id;
return CopyStr("--");
// Product ID
product_id = MsRegReadStr(REG_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion", "ProductId");
if (product_id == NULL)
{
product_id = MsRegReadStr(REG_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion", "ProductId");
}
return product_id;
}
// Acquisition whether the OS is currently supported
bool Win32IsSupportedOs()
{
if (Win32GetOsType() == 0)
{
Win32Alert(
CEDAR_PRODUCT_STR " VPN doesn't support this Windows Operating System.\n"
CEDAR_PRODUCT_STR " VPN requires " SUPPORTED_WINDOWS_LIST ".\n\n"
"Please contact your system administrator.", NULL);
return false;
}
return true;
}
// Show an alert
void Win32AlertW(wchar_t *msg, wchar_t *caption)
{
char *s;
// Validate arguments
if (msg == NULL)
{
msg = L"Alert";
}
if (caption == NULL)
{
caption = CEDAR_PRODUCT_STR_W L" VPN Kernel";
}
s = GetCommandLineStr();
if (SearchStr(s, "win9x_uninstall", 0) == INFINITE && SearchStr(s, "win9x_install", 0) == INFINITE)
{
// Hide during the uninstallation in Win9x service mode
MessageBoxW(NULL, msg, caption, MB_SETFOREGROUND | MB_TOPMOST | MB_SERVICE_NOTIFICATION | MB_OK | MB_ICONEXCLAMATION);
}
Free(s);
}
void Win32Alert(char *msg, char *caption)
{
char *s;
// Validate arguments
if (msg == NULL)
{
msg = "Alert";
}
if (caption == NULL)
{
caption = CEDAR_PRODUCT_STR " VPN Kernel";
}
s = GetCommandLineStr();
if (SearchStr(s, "win9x_uninstall", 0) == INFINITE && SearchStr(s, "win9x_install", 0) == INFINITE)
{
// Hide during the uninstallation in Win9x service mode
MessageBox(NULL, msg, caption, MB_SETFOREGROUND | MB_TOPMOST | MB_SERVICE_NOTIFICATION | MB_OK | MB_ICONEXCLAMATION);
}
Free(s);
}
void Win32DebugAlert(char *msg)
{
// Validate arguments
if (msg == NULL)
{
msg = "Alert";
}
MessageBox(NULL, msg, "Debug", MB_SETFOREGROUND | MB_TOPMOST | MB_SERVICE_NOTIFICATION | MB_OK | MB_ICONEXCLAMATION);
}
2017-10-18 12:24:21 +03:00
// Get the number of CPUs
UINT Win32GetNumberOfCpuInner()
{
UINT ret = 0;
SYSTEM_INFO info;
Zero(&info, sizeof(info));
GetSystemInfo(&info);
if (info.dwNumberOfProcessors >= 1 && info.dwNumberOfProcessors <= 128)
{
ret = info.dwNumberOfProcessors;
}
return ret;
}
2014-01-04 17:00:08 +04:00
// Get the OS information
void Win32GetOsInfo(OS_INFO *info)
{
UINT type = Win32GetOsType();
OSVERSIONINFOEX os;
char tmp[MAX_SIZE];
// Validate arguments
if (info == NULL)
{
return;
}
Zero(&os, sizeof(os));
os.dwOSVersionInfoSize = sizeof(os);
Win32GetVersionExInternal((LPOSVERSIONINFOA)&os);
info->OsType = Win32GetOsType();
info->OsServicePack = os.wServicePackMajor;
if (true)
2014-01-04 17:00:08 +04:00
{
char *s;
char *keyname = "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion";
info->OsSystemName = CopyStr("Windows NT");
Format(tmp, sizeof(tmp), "Build %u", os.dwBuildNumber);
if (s = MsRegReadStr(REG_LOCAL_MACHINE, keyname, "CurrentType"))
{
char str[MAX_SIZE];
Format(str, sizeof(str), ", %s", s);
StrCat(tmp, sizeof(tmp), str);
Free(s);
}
if (os.wServicePackMajor != 0)
{
char str[MAX_SIZE];
Format(str, sizeof(str), ", Service Pack %u", os.wServicePackMajor);
StrCat(tmp, sizeof(tmp), str);
}
if (s = MsRegReadStr(REG_LOCAL_MACHINE, keyname, "BuildLab"))
{
char str[MAX_SIZE];
Format(str, sizeof(str), " (%s)", s);
StrCat(tmp, sizeof(tmp), str);
Free(s);
}
info->OsVersion = CopyStr(tmp);
info->KernelName = CopyStr("NTOS Kernel");
Format(tmp, sizeof(tmp), "Build %u", os.dwBuildNumber);
if (s = MsRegReadStr(REG_LOCAL_MACHINE, keyname, "CurrentType"))
{
char str[MAX_SIZE];
Format(str, sizeof(str), " %s", s);
StrCat(tmp, sizeof(tmp), str);
Free(s);
}
info->KernelVersion = CopyStr(tmp);
}
info->OsProductName = CopyStr(OsTypeToStr(info->OsType));
info->OsVendorName = CopyStr("Microsoft Corporation");
}
// GetVersionEx API (Ignore the tricky features that have been added to the Windows 8.2 or later)
bool Win32GetVersionExInternal(void *info)
{
OSVERSIONINFOA os;
// Validate arguments
if (info == NULL)
{
return false;
}
Zero(&os, sizeof(os));
os.dwOSVersionInfoSize = sizeof(os);
if (GetVersionExA(&os))
{
if (os.dwPlatformId == VER_PLATFORM_WIN32_NT)
{
if ((os.dwMajorVersion == 6 && os.dwMinorVersion >= 2) ||
2015-01-30 16:30:34 +03:00
(os.dwMajorVersion >= 7))
2014-01-04 17:00:08 +04:00
{
// Windows 8 later
return Win32GetVersionExInternalForWindows81orLater(info);
}
}
}
return GetVersionExA(info);
}
// GetVersionEx for Windows 8.1 and later
bool Win32GetVersionExInternalForWindows81orLater(void *info)
{
OSVERSIONINFOEXA *ex = (OSVERSIONINFOEXA *)info;
char *str;
2015-01-30 16:30:34 +03:00
UINT major1 = 0, major2 = 0;
UINT minor1 = 0, minor2 = 0;
UINT major = 0, minor = 0;
2014-01-04 17:00:08 +04:00
// Validate arguments
if (info == NULL)
{
return false;
}
if (ex->dwOSVersionInfoSize != sizeof(OSVERSIONINFOEXA) &&
ex->dwOSVersionInfoSize != sizeof(OSVERSIONINFOA))
{
return GetVersionExA(info);
}
if (GetVersionExA(info) == false)
{
return false;
}
str = MsRegReadStrEx2(REG_LOCAL_MACHINE,
"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion",
"CurrentVersion",
false, true);
if (IsEmptyStr(str) == false)
{
// Is the version string formed as x.y?
TOKEN_LIST *t = ParseToken(str, ".");
if (t != NULL && t->NumTokens == 2)
{
2015-01-30 16:30:34 +03:00
major1 = ToInt(t->Token[0]);
minor1 = ToInt(t->Token[1]);
2014-01-04 17:00:08 +04:00
}
FreeToken(t);
}
Free(str);
2015-01-30 16:30:34 +03:00
major2 = MsRegReadIntEx2(REG_LOCAL_MACHINE,
"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion",
"CurrentMajorVersionNumber", false, true);
minor2 = MsRegReadIntEx2(REG_LOCAL_MACHINE,
"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion",
"CurrentMinorVersionNumber", false, true);
if ((major1 * 10000 + minor1) > (major2 * 10000 + minor2))
{
major = major1;
minor = minor1;
}
else
{
major = major2;
minor = minor2;
}
if (major >= 6)
{
// Version number acquisition success
ex->dwMajorVersion = major;
ex->dwMinorVersion = minor;
}
2014-01-04 17:00:08 +04:00
return true;
}
// Acquisition whether it's a Windows NT
bool Win32IsNt()
{
OSVERSIONINFO os;
Zero(&os, sizeof(os));
os.dwOSVersionInfoSize = sizeof(os);
if (GetVersionEx(&os) == FALSE)
{
// Failure?
return false;
}
if (os.dwPlatformId == VER_PLATFORM_WIN32_NT)
{
// NT
return true;
}
// 9x
return false;
}
// Get the OS type
UINT Win32GetOsType()
{
OSVERSIONINFO os;
Zero(&os, sizeof(os));
os.dwOSVersionInfoSize = sizeof(os);
if (Win32GetVersionExInternal(&os) == FALSE)
{
// Failure?
return 0;
}
if (os.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS)
{
// Windows 9x system
if (os.dwMajorVersion == 4)
{
if (os.dwMinorVersion == 0)
{
return OSTYPE_WINDOWS_95;
}
else if (os.dwMinorVersion == 10)
{
return OSTYPE_WINDOWS_98;
}
else if (os.dwMinorVersion == 90)
{
return OSTYPE_WINDOWS_ME;
}
else
{
return OSTYPE_WINDOWS_UNKNOWN;
}
}
else if (os.dwMajorVersion >= 5)
{
return OSTYPE_WINDOWS_UNKNOWN;
}
}
else if (os.dwPlatformId == VER_PLATFORM_WIN32_NT)
{
UINT sp = Win32GetSpVer(os.szCSDVersion);
if (os.dwMajorVersion == 4)
{
if (sp < 6)
{
// SP6 or earlier
return 0;
}
}
if (os.dwMajorVersion < 4)
{
// NT 3.51 or earlier
return 0;
}
else
{
OSVERSIONINFOEX os;
Zero(&os, sizeof(os));
os.dwOSVersionInfoSize = sizeof(os);
Win32GetVersionExInternal((LPOSVERSIONINFOA)&os);
if (os.dwMajorVersion == 4)
{
// Windows NT 4.0
if (os.wProductType == VER_NT_DOMAIN_CONTROLLER || os.wProductType == VER_NT_SERVER)
{
if ((os.wSuiteMask & VER_SUITE_TERMINAL) || (os.wSuiteMask & VER_SUITE_SINGLEUSERTS))
{
return OSTYPE_WINDOWS_NT_4_TERMINAL_SERVER;
}
if (os.wSuiteMask & VER_SUITE_ENTERPRISE)
{
return OSTYPE_WINDOWS_NT_4_SERVER_ENTERPRISE;
}
if (os.wSuiteMask & VER_SUITE_BACKOFFICE)
{
return OSTYPE_WINDOWS_NT_4_BACKOFFICE;
}
if ((os.wSuiteMask & VER_SUITE_SMALLBUSINESS) || (os.wSuiteMask & VER_SUITE_SMALLBUSINESS_RESTRICTED))
{
return OSTYPE_WINDOWS_NT_4_SMS;
}
else
{
return OSTYPE_WINDOWS_NT_4_SERVER;
}
}
else
{
return OSTYPE_WINDOWS_NT_4_WORKSTATION;
}
}
else if (os.dwMajorVersion == 5)
{
// Windows 2000, XP, Server 2003
if (os.dwMinorVersion == 0)
{
// Windows 2000
if (os.wProductType == VER_NT_DOMAIN_CONTROLLER || os.wProductType == VER_NT_SERVER)
{
// Server
if (os.wSuiteMask & VER_SUITE_DATACENTER)
{
return OSTYPE_WINDOWS_2000_DATACENTER_SERVER;
}
else if ((os.wSuiteMask & VER_SUITE_SMALLBUSINESS) || (os.wSuiteMask & VER_SUITE_SMALLBUSINESS_RESTRICTED))
{
return OSTYPE_WINDOWS_2000_SBS;
}
else if (os.wSuiteMask & VER_SUITE_BACKOFFICE)
{
return OSTYPE_WINDOWS_2000_BACKOFFICE;
}
else if (os.wSuiteMask & VER_SUITE_ENTERPRISE)
{
return OSTYPE_WINDOWS_2000_ADVANCED_SERVER;
}
else
{
return OSTYPE_WINDOWS_2000_SERVER;
}
}
else
{
// Client
return OSTYPE_WINDOWS_2000_PROFESSIONAL;
}
}
else if (os.dwMinorVersion == 1)
{
// Windows XP
if (os.wSuiteMask & VER_SUITE_PERSONAL)
{
return OSTYPE_WINDOWS_XP_HOME;
}
else
{
return OSTYPE_WINDOWS_XP_PROFESSIONAL;
}
}
else if (os.dwMinorVersion == 2)
{
// Windows Server 2003
if (os.wProductType == VER_NT_DOMAIN_CONTROLLER || os.wProductType == VER_NT_SERVER)
{
// Server
if (os.wSuiteMask & VER_SUITE_DATACENTER)
{
return OSTYPE_WINDOWS_2003_DATACENTER;
}
else if ((os.wSuiteMask & VER_SUITE_SMALLBUSINESS) || (os.wSuiteMask & VER_SUITE_SMALLBUSINESS_RESTRICTED))
{
return OSTYPE_WINDOWS_2003_SBS;
}
else if (os.wSuiteMask & VER_SUITE_BACKOFFICE)
{
return OSTYPE_WINDOWS_2003_BACKOFFICE;
}
else if (os.wSuiteMask & VER_SUITE_ENTERPRISE)
{
return OSTYPE_WINDOWS_2003_ENTERPRISE;
}
else if (os.wSuiteMask & VER_SUITE_BLADE)
{
return OSTYPE_WINDOWS_2003_WEB;
}
else
{
return OSTYPE_WINDOWS_2003_STANDARD;
}
}
else
{
// Client (Unknown XP?)
return OSTYPE_WINDOWS_XP_PROFESSIONAL;
}
}
else
{
// Windows Longhorn
if (os.wProductType == VER_NT_DOMAIN_CONTROLLER || os.wProductType == VER_NT_SERVER)
{
return OSTYPE_WINDOWS_LONGHORN_SERVER;
}
else
{
return OSTYPE_WINDOWS_LONGHORN_PROFESSIONAL;
}
}
}
else
{
if (os.dwMajorVersion == 6 && os.dwMinorVersion == 0)
{
// Windows Vista, Server 2008
if (os.wProductType == VER_NT_DOMAIN_CONTROLLER || os.wProductType == VER_NT_SERVER)
{
return OSTYPE_WINDOWS_LONGHORN_SERVER;
}
else
{
return OSTYPE_WINDOWS_LONGHORN_PROFESSIONAL;
}
}
else if (os.dwMajorVersion == 6 && os.dwMinorVersion == 1)
{
if (os.wProductType == VER_NT_WORKSTATION)
{
// Windows 7
return OSTYPE_WINDOWS_7;
}
else
{
// Windows Server 2008 R2
return OSTYPE_WINDOWS_SERVER_2008_R2;
}
}
else if (os.dwMajorVersion == 6 && os.dwMinorVersion == 2)
{
if (os.wProductType == VER_NT_WORKSTATION)
{
// Windows 8
return OSTYPE_WINDOWS_8;
}
else
{
// Windows Server 2012
return OSTYPE_WINDOWS_SERVER_8;
}
}
else if (os.dwMajorVersion == 6 && os.dwMinorVersion == 3)
{
if (os.wProductType == VER_NT_WORKSTATION)
{
// Windows 8.1
return OSTYPE_WINDOWS_81;
}
else
{
// Windows Server 2012 R2
return OSTYPE_WINDOWS_SERVER_81;
}
}
2015-01-30 16:30:34 +03:00
else if ((os.dwMajorVersion == 6 && os.dwMinorVersion == 4) || (os.dwMajorVersion == 10 && os.dwMinorVersion == 0))
2014-10-03 19:09:23 +04:00
{
if (os.wProductType == VER_NT_WORKSTATION)
{
// Windows 10
return OSTYPE_WINDOWS_10;
}
else
{
// Windows Server 10
return OSTYPE_WINDOWS_SERVER_10;
}
}
2014-01-04 17:00:08 +04:00
else
{
if (os.wProductType == VER_NT_WORKSTATION)
{
2014-10-03 19:09:23 +04:00
// Windows 11 or later
return OSTYPE_WINDOWS_11;
2014-01-04 17:00:08 +04:00
}
else
{
2014-10-03 19:09:23 +04:00
// Windows Server 11 or later
return OSTYPE_WINDOWS_SERVER_11;
2014-01-04 17:00:08 +04:00
}
}
}
}
}
// Can not be determined
return 0;
}
// Get the SP version from the string
UINT Win32GetSpVer(char *str)
{
UINT ret, i;
TOKEN_LIST *t;
// Validate arguments
if (str == NULL)
{
return 0;
}
t = ParseToken(str, NULL);
if (t == NULL)
{
return 0;
}
ret = 0;
for (i = 0;i < t->NumTokens;i++)
{
ret = ToInt(t->Token[i]);
if (ret != 0)
{
break;
}
}
FreeToken(t);
return ret;
}
// Kill the process
bool Win32TerminateProcess(void *handle)
{
HANDLE h;
// Validate arguments
if (handle == NULL)
{
return false;
}
h = (HANDLE)handle;
TerminateProcess(h, 0);
return true;
}
// Close the process
void Win32CloseProcess(void *handle)
{
// Validate arguments
if (handle == NULL)
{
return;
}
CloseHandle((HANDLE)handle);
}
// Check whether the specified process is alive
bool Win32IsProcessAlive(void *handle)
{
HANDLE h;
// Validate arguments
if (handle == NULL)
{
return false;
}
h = (HANDLE)handle;
if (WaitForSingleObject(h, 0) == WAIT_OBJECT_0)
{
return false;
}
return true;
}
// Wait for the process termination
bool Win32WaitProcess(void *h, UINT timeout)
{
// Validate arguments
if (h == NULL)
{
return false;
}
if (timeout == 0)
{
timeout = INFINITE;
}
if (WaitForSingleObject((HANDLE)h, timeout) == WAIT_TIMEOUT)
{
return false;
}
return true;
}
// Run the process and wait for terminate it
bool Win32RunAndWaitProcess(wchar_t *filename, wchar_t *arg, bool hide, bool disableWow, UINT timeout)
{
UINT process_id = 0;
void *p = Win32RunEx3W(filename, arg, hide, &process_id, disableWow);
if (p == NULL)
{
return false;
}
return Win32WaitProcess(p, timeout);
}
// Run the process (return the handle)
void *Win32RunExW(wchar_t *filename, wchar_t *arg, bool hide)
{
return Win32RunEx2W(filename, arg, hide, NULL);
}
void *Win32RunEx2W(wchar_t *filename, wchar_t *arg, bool hide, UINT *process_id)
{
return Win32RunEx3W(filename, arg, hide, process_id, false);
}
void *Win32RunEx3W(wchar_t *filename, wchar_t *arg, bool hide, UINT *process_id, bool disableWow)
{
STARTUPINFOW info;
PROCESS_INFORMATION ret;
wchar_t cmdline[MAX_SIZE];
wchar_t name[MAX_PATH];
void *p;
// Validate arguments
if (filename == NULL)
{
return NULL;
}
if (IsNt() == false)
{
char *filename_a = CopyUniToStr(filename);
char *arg_a = CopyUniToStr(arg);
void *ret = Win32RunEx3(filename_a, arg_a, hide, process_id, disableWow);
Free(filename_a);
Free(arg_a);
return ret;
}
UniStrCpy(name, sizeof(name), filename);
UniTrim(name);
if (UniSearchStr(name, L"\"", 0) == INFINITE)
{
if (arg == NULL)
{
UniFormat(cmdline, sizeof(cmdline), L"%s", name);
}
else
{
UniFormat(cmdline, sizeof(cmdline), L"%s %s", name, arg);
}
}
else
{
if (arg == NULL)
{
UniFormat(cmdline, sizeof(cmdline), L"\"%s\"", name);
}
else
{
UniFormat(cmdline, sizeof(cmdline), L"\"%s\" %s", name, arg);
}
}
Zero(&info, sizeof(info));
Zero(&ret, sizeof(ret));
info.cb = sizeof(info);
info.dwFlags = STARTF_USESHOWWINDOW;
info.wShowWindow = (hide == false ? SW_SHOWDEFAULT : SW_HIDE);
UniTrim(cmdline);
if (disableWow)
{
p = MsDisableWow64FileSystemRedirection();
}
if (CreateProcessW(NULL, cmdline, NULL, NULL, FALSE,
(hide == false ? CREATE_NEW_CONSOLE : CREATE_NO_WINDOW | CREATE_NEW_CONSOLE) | NORMAL_PRIORITY_CLASS,
NULL, NULL, &info, &ret) == FALSE)
{
if (disableWow)
{
MsRestoreWow64FileSystemRedirection(p);
}
return NULL;
}
if (disableWow)
{
MsRestoreWow64FileSystemRedirection(p);
}
if (process_id != NULL)
{
*process_id = ret.dwProcessId;
}
CloseHandle(ret.hThread);
return ret.hProcess;
}
void *Win32RunEx(char *filename, char *arg, bool hide)
{
return Win32RunEx2(filename, arg, hide, NULL);
}
void *Win32RunEx2(char *filename, char *arg, bool hide, UINT *process_id)
{
return Win32RunEx3(filename, arg, hide, process_id, false);
}
void *Win32RunEx3(char *filename, char *arg, bool hide, UINT *process_id, bool disableWow)
{
STARTUPINFO info;
PROCESS_INFORMATION ret;
char cmdline[MAX_SIZE];
char name[MAX_PATH];
void *p = NULL;
// Validate arguments
if (filename == NULL)
{
return NULL;
}
StrCpy(name, sizeof(name), filename);
Trim(name);
if (SearchStr(name, "\"", 0) == INFINITE)
{
if (arg == NULL)
{
Format(cmdline, sizeof(cmdline), "%s", name);
}
else
{
Format(cmdline, sizeof(cmdline), "%s %s", name, arg);
}
}
else
{
if (arg == NULL)
{
Format(cmdline, sizeof(cmdline), "\"%s\"", name);
}
else
{
Format(cmdline, sizeof(cmdline), "\"%s\" %s", name, arg);
}
}
Zero(&info, sizeof(info));
Zero(&ret, sizeof(ret));
info.cb = sizeof(info);
info.dwFlags = STARTF_USESHOWWINDOW;
info.wShowWindow = (hide == false ? SW_SHOWDEFAULT : SW_HIDE);
Trim(cmdline);
if (disableWow)
{
p = MsDisableWow64FileSystemRedirection();
}
if (CreateProcess(NULL, cmdline, NULL, NULL, FALSE,
(hide == false ? CREATE_NEW_CONSOLE : CREATE_NO_WINDOW | CREATE_NEW_CONSOLE) | NORMAL_PRIORITY_CLASS,
NULL, NULL, &info, &ret) == FALSE)
{
if (disableWow)
{
MsRestoreWow64FileSystemRedirection(p);
}
return NULL;
}
if (disableWow)
{
MsRestoreWow64FileSystemRedirection(p);
}
if (process_id != NULL)
{
*process_id = ret.dwProcessId;
}
CloseHandle(ret.hThread);
return ret.hProcess;
}
// Start the process
bool Win32RunW(wchar_t *filename, wchar_t *arg, bool hide, bool wait)
{
STARTUPINFOW info;
PROCESS_INFORMATION ret;
wchar_t cmdline[MAX_SIZE];
wchar_t name[MAX_PATH];
// Validate arguments
if (filename == NULL)
{
return false;
}
if (IsNt() == false)
{
char *filename_a = CopyUniToStr(filename);
char *arg_a = CopyUniToStr(arg);
bool ret;
ret = Win32Run(filename_a, arg_a, hide, wait);
Free(filename_a);
Free(arg_a);
return ret;
}
UniStrCpy(name, sizeof(name), filename);
UniTrim(name);
if (UniSearchStr(name, L"\"", 0) == INFINITE)
{
if (arg == NULL)
{
UniFormat(cmdline, sizeof(cmdline), L"%s", name);
}
else
{
UniFormat(cmdline, sizeof(cmdline), L"%s %s", name, arg);
}
}
else
{
if (arg == NULL)
{
UniFormat(cmdline, sizeof(cmdline), L"\"%s\"", name);
}
else
{
UniFormat(cmdline, sizeof(cmdline), L"\"%s\" %s", name, arg);
}
}
Zero(&info, sizeof(info));
Zero(&ret, sizeof(ret));
info.cb = sizeof(info);
info.dwFlags = STARTF_USESHOWWINDOW;
info.wShowWindow = (hide == false ? SW_SHOWDEFAULT : SW_HIDE);
UniTrim(cmdline);
if (CreateProcessW(NULL, cmdline, NULL, NULL, FALSE,
(hide == false ? CREATE_NEW_CONSOLE : CREATE_NO_WINDOW | CREATE_NEW_CONSOLE) | NORMAL_PRIORITY_CLASS,
NULL, NULL, &info, &ret) == FALSE)
{
return false;
}
if (wait)
{
WaitForSingleObject(ret.hProcess, INFINITE);
}
CloseHandle(ret.hThread);
CloseHandle(ret.hProcess);
return true;
}
bool Win32Run(char *filename, char *arg, bool hide, bool wait)
{
STARTUPINFO info;
PROCESS_INFORMATION ret;
char cmdline[MAX_SIZE];
char name[MAX_PATH];
// Validate arguments
if (filename == NULL)
{
return false;
}
StrCpy(name, sizeof(name), filename);
Trim(name);
if (SearchStr(name, "\"", 0) == INFINITE)
{
if (arg == NULL)
{
Format(cmdline, sizeof(cmdline), "%s", name);
}
else
{
Format(cmdline, sizeof(cmdline), "%s %s", name, arg);
}
}
else
{
if (arg == NULL)
{
Format(cmdline, sizeof(cmdline), "\"%s\"", name);
}
else
{
Format(cmdline, sizeof(cmdline), "\"%s\" %s", name, arg);
}
}
Zero(&info, sizeof(info));
Zero(&ret, sizeof(ret));
info.cb = sizeof(info);
info.dwFlags = STARTF_USESHOWWINDOW;
info.wShowWindow = (hide == false ? SW_SHOWDEFAULT : SW_HIDE);
Trim(cmdline);
if (CreateProcess(NULL, cmdline, NULL, NULL, FALSE,
(hide == false ? CREATE_NEW_CONSOLE : CREATE_NO_WINDOW | CREATE_NEW_CONSOLE) | NORMAL_PRIORITY_CLASS,
NULL, NULL, &info, &ret) == FALSE)
{
return false;
}
if (wait)
{
WaitForSingleObject(ret.hProcess, INFINITE);
}
CloseHandle(ret.hThread);
CloseHandle(ret.hProcess);
return true;
}
// Get the Thread ID
UINT Win32ThreadId()
{
return GetCurrentThreadId();
}
// Rename the file
bool Win32FileRenameW(wchar_t *old_name, wchar_t *new_name)
{
// Validate arguments
if (old_name == NULL || new_name == NULL)
{
return false;
}
if (IsNt() == false)
{
char *old_name_a = CopyUniToStr(old_name);
char *new_name_a = CopyUniToStr(new_name);
bool ret = Win32FileRename(old_name_a, new_name_a);
Free(old_name_a);
Free(new_name_a);
return ret;
}
// Rename
if (MoveFileW(old_name, new_name) == FALSE)
{
return false;
}
return true;
}
bool Win32FileRename(char *old_name, char *new_name)
{
// Validate arguments
if (old_name == NULL || new_name == NULL)
{
return false;
}
// Rename
if (MoveFile(old_name, new_name) == FALSE)
{
return false;
}
return true;
}
// Getting the name of the directory where the EXE file is in
void Win32GetExeDirW(wchar_t *name, UINT size)
{
wchar_t exe_path[MAX_SIZE];
wchar_t exe_dir[MAX_SIZE];
// Validate arguments
if (name == NULL)
{
return;
}
if (IsNt() == false)
{
char name_a[MAX_PATH];
Win32GetExeDir(name_a, sizeof(name_a));
StrToUni(name, size, name_a);
return;
}
// Get the EXE file name
GetModuleFileNameW(NULL, exe_path, sizeof(exe_path));
// Get the directory name
Win32GetDirFromPathW(exe_dir, sizeof(exe_dir), exe_path);
UniStrCpy(name, size, exe_dir);
}
void Win32GetExeDir(char *name, UINT size)
{
char exe_path[MAX_SIZE];
char exe_dir[MAX_SIZE];
// Validate arguments
if (name == NULL)
{
return;
}
// Get the EXE file name
GetModuleFileName(NULL, exe_path, sizeof(exe_path));
// Get the directory name
Win32GetDirFromPath(exe_dir, sizeof(exe_dir), exe_path);
StrCpy(name, size, exe_dir);
}
// Remove the '\' at the end
void Win32NukuEnW(wchar_t *dst, UINT size, wchar_t *src)
{
wchar_t str[MAX_SIZE];
int i;
if (src)
{
UniStrCpy(str, sizeof(str), src);
}
else
{
UniStrCpy(str, sizeof(str), dst);
}
i = UniStrLen(str);
if (str[i - 1] == L'\\')
{
str[i - 1] = 0;
}
UniStrCpy(dst, size, str);
}
void Win32NukuEn(char *dst, UINT size, char *src)
{
char str[MAX_SIZE];
int i;
if (src)
{
StrCpy(str, sizeof(str), src);
}
else
{
StrCpy(str, sizeof(str), dst);
}
i = StrLen(str);
if (str[i - 1] == '\\')
{
str[i - 1] = 0;
}
StrCpy(dst, size, str);
}
// Get the directory name from path
void Win32GetDirFromPathW(wchar_t *dst, UINT size, wchar_t *src)
{
wchar_t str[MAX_SIZE];
int i,len;
wchar_t c;
wchar_t tmp[MAX_SIZE];
int wp;
if (src)
{
UniStrCpy(str, sizeof(str), src);
}
else
{
UniStrCpy(str, sizeof(str), dst);
}
Win32NukuEnW(str, sizeof(str), NULL);
wp = 0;
len = UniStrLen(str);
dst[0] = 0;
for (i = 0;i < len;i++)
{
c = str[i];
switch (c)
{
case L'\\':
tmp[wp] = 0;
wp = 0;
UniStrCat(dst, size, tmp);
UniStrCat(dst, size, L"\\");
break;
default:
tmp[wp] = c;
wp++;
break;
}
}
Win32NukuEnW(dst, size, NULL);
}
void Win32GetDirFromPath(char *dst, UINT size, char *src)
{
char str[MAX_SIZE];
int i,len;
char c;
char tmp[MAX_SIZE];
int wp;
if (src)
{
StrCpy(str, sizeof(str), src);
}
else
{
StrCpy(str, sizeof(str), dst);
}
Win32NukuEn(str, sizeof(str), NULL);
wp = 0;
len = StrLen(str);
dst[0] = 0;
for (i = 0;i < len;i++)
{
c = str[i];
switch (c)
{
case '\\':
tmp[wp] = 0;
wp = 0;
StrCat(dst, size, tmp);
StrCat(dst, size, "\\");
break;
default:
tmp[wp] = c;
wp++;
break;
}
}
Win32NukuEn(dst, size, NULL);
}
// Delete the directory
bool Win32DeleteDirW(wchar_t *name)
{
// Validate arguments
if (name == NULL)
{
return false;
}
if (IsNt() == false)
{
char *name_a = CopyUniToStr(name);
bool ret = Win32DeleteDir(name_a);
Free(name_a);
return ret;
}
if (RemoveDirectoryW(name) == FALSE)
{
return false;
}
return true;
}
bool Win32DeleteDir(char *name)
{
// Validate arguments
if (name == NULL)
{
return false;
}
if (RemoveDirectory(name) == FALSE)
{
return false;
}
return true;
}
// Create a directory
bool Win32MakeDirW(wchar_t *name)
{
// Validate arguments
if (name == NULL)
{
return false;
}
if (IsNt() == false)
{
char *name_a = CopyUniToStr(name);
bool ret = Win32MakeDir(name_a);
Free(name_a);
return ret;
}
if (CreateDirectoryW(name, NULL) == FALSE)
{
return false;
}
return true;
}
bool Win32MakeDir(char *name)
{
// Validate arguments
if (name == NULL)
{
return false;
}
if (CreateDirectory(name, NULL) == FALSE)
{
return false;
}
return true;
}
// Delete the file
bool Win32FileDeleteW(wchar_t *name)
{
// Validate arguments
if (name == NULL)
{
return false;
}
if (IsNt() == false)
{
bool ret;
char *name_a = CopyUniToStr(name);
ret = Win32FileDelete(name_a);
Free(name_a);
return ret;
}
if (DeleteFileW(name) == FALSE)
{
return false;
}
return true;
}
bool Win32FileDelete(char *name)
{
// Validate arguments
if (name == NULL)
{
return false;
}
if (DeleteFile(name) == FALSE)
{
return false;
}
return true;
}
// Seek in the file
bool Win32FileSeek(void *pData, UINT mode, int offset)
{
WIN32IO *p;
DWORD ret;
// Validate arguments
if (pData == NULL)
{
return false;
}
if (mode != FILE_BEGIN && mode != FILE_END && mode != FILE_CURRENT)
{
return false;
}
p = (WIN32IO *)pData;
ret = SetFilePointer(p->hFile, (LONG)offset, NULL, mode);
if (ret == INVALID_SET_FILE_POINTER || ret == ERROR_NEGATIVE_SEEK)
{
return false;
}
return true;
}
// Get the file size
UINT64 Win32FileSize(void *pData)
{
WIN32IO *p;
UINT64 ret;
DWORD tmp;
// Validate arguments
if (pData == NULL)
{
return 0;
}
p = (WIN32IO *)pData;
tmp = 0;
ret = GetFileSize(p->hFile, &tmp);
if (ret == (DWORD)-1)
{
return 0;
}
if (tmp != 0)
{
ret += (UINT64)tmp * 4294967296ULL;
}
return ret;
}
// Write to the file
bool Win32FileWrite(void *pData, void *buf, UINT size)
{
WIN32IO *p;
DWORD write_size;
// Validate arguments
if (pData == NULL || buf == NULL || size == 0)
{
return false;
}
p = (WIN32IO *)pData;
if (WriteFile(p->hFile, buf, size, &write_size, NULL) == FALSE)
{
return false;
}
if (write_size != size)
{
return false;
}
return true;
}
// Read from a file
bool Win32FileRead(void *pData, void *buf, UINT size)
{
WIN32IO *p;
DWORD read_size;
// Validate arguments
if (pData == NULL || buf == NULL || size == 0)
{
return false;
}
p = (WIN32IO *)pData;
if (ReadFile(p->hFile, buf, size, &read_size, NULL) == FALSE)
{
return false;
}
if (read_size != size)
{
return false;
}
return true;;
}
// Close the file
void Win32FileClose(void *pData, bool no_flush)
{
WIN32IO *p;
// Validate arguments
if (pData == NULL)
{
return;
}
p = (WIN32IO *)pData;
if (p->WriteMode && no_flush == false)
{
FlushFileBuffers(p->hFile);
}
CloseHandle(p->hFile);
p->hFile = NULL;
// Memory release
Win32MemoryFree(p);
}
// Get the date of the file
bool Win32FileGetDate(void *pData, UINT64 *created_time, UINT64 *updated_time, UINT64 *accessed_date)
{
WIN32IO *p;
BY_HANDLE_FILE_INFORMATION info;
SYSTEMTIME st_create, st_update, st_access;
// Validate arguments
if (pData == NULL)
{
return false;
}
p = (WIN32IO *)pData;
Zero(&info, sizeof(info));
if (GetFileInformationByHandle(p->hFile, &info) == false)
{
return false;
}
Zero(&st_create, sizeof(st_create));
Zero(&st_update, sizeof(st_update));
Zero(&st_access, sizeof(st_access));
FileTimeToSystemTime(&info.ftCreationTime, &st_create);
FileTimeToSystemTime(&info.ftLastWriteTime, &st_update);
FileTimeToSystemTime(&info.ftLastAccessTime, &st_access);
if (created_time != NULL)
{
*created_time = SystemToUINT64(&st_create);
}
if (updated_time != NULL)
{
*updated_time = SystemToUINT64(&st_update);
}
if (accessed_date != NULL)
{
*accessed_date = SystemToUINT64(&st_access);
}
return true;
}
// Set the date of the file
bool Win32FileSetDate(void *pData, UINT64 created_time, UINT64 updated_time)
{
WIN32IO *p;
SYSTEMTIME st_created_time, st_updated_time;
FILETIME ft_created_time, ft_updated_time;
FILETIME *p_created_time = NULL, *p_updated_time = NULL;
// Validate arguments
if (pData == NULL || (created_time == 0 && updated_time == 0))
{
return false;
}
p = (WIN32IO *)pData;
Zero(&st_created_time, sizeof(st_created_time));
Zero(&st_updated_time, sizeof(st_updated_time));
if (created_time != 0)
{
UINT64ToSystem(&st_created_time, created_time);
SystemTimeToFileTime(&st_created_time, &ft_created_time);
p_created_time = &ft_created_time;
}
if (updated_time != 0)
{
UINT64ToSystem(&st_updated_time, updated_time);
SystemTimeToFileTime(&st_updated_time, &ft_updated_time);
p_updated_time = &ft_updated_time;
}
return SetFileTime(p->hFile, p_created_time, NULL, p_updated_time);
}
// Flush to the file
void Win32FileFlush(void *pData)
{
WIN32IO *p;
// Validate arguments
if (pData == NULL)
{
return;
}
p = (WIN32IO *)pData;
if (p->WriteMode)
{
FlushFileBuffers(p->hFile);
}
}
// Open the file
void *Win32FileOpenW(wchar_t *name, bool write_mode, bool read_lock)
{
WIN32IO *p;
HANDLE h;
DWORD lock_mode;
// Validate arguments
if (name == NULL)
{
return NULL;
}
if (IsNt() == false)
{
void *ret;
char *name_a = CopyUniToStr(name);
ret = Win32FileOpen(name_a, write_mode, read_lock);
Free(name_a);
return ret;
}
if (write_mode)
{
lock_mode = FILE_SHARE_READ;
}
else
{
if (read_lock == false)
{
lock_mode = FILE_SHARE_READ | FILE_SHARE_WRITE;
}
else
{
lock_mode = FILE_SHARE_READ;
}
}
// Open the file
h = CreateFileW(name,
(write_mode ? GENERIC_READ | GENERIC_WRITE : GENERIC_READ),
lock_mode,
NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (h == INVALID_HANDLE_VALUE)
{
UINT ret = GetLastError();
// Failure
return NULL;
}
// Memory allocation
p = Win32MemoryAlloc(sizeof(WIN32IO));
// Store Handle
p->hFile = h;
p->WriteMode = write_mode;
return (void *)p;
}
void *Win32FileOpen(char *name, bool write_mode, bool read_lock)
{
WIN32IO *p;
HANDLE h;
DWORD lock_mode;
// Validate arguments
if (name == NULL)
{
return NULL;
}
if (write_mode)
{
lock_mode = FILE_SHARE_READ;
}
else
{
if (read_lock == false)
{
lock_mode = FILE_SHARE_READ | FILE_SHARE_WRITE;
}
else
{
lock_mode = FILE_SHARE_READ;
}
}
// Open the file
h = CreateFile(name,
(write_mode ? GENERIC_READ | GENERIC_WRITE : GENERIC_READ),
lock_mode,
NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (h == INVALID_HANDLE_VALUE)
{
UINT ret = GetLastError();
// Failure
return NULL;
}
// Memory allocation
p = Win32MemoryAlloc(sizeof(WIN32IO));
// Store Handle
p->hFile = h;
p->WriteMode = write_mode;
return (void *)p;
}
// Create a file
void *Win32FileCreateW(wchar_t *name)
{
WIN32IO *p;
HANDLE h;
// Validate arguments
if (name == NULL)
{
return NULL;
}
if (IsNt() == false)
{
void *ret;
char *name_a = CopyUniToStr(name);
ret = Win32FileCreate(name_a);
Free(name_a);
return ret;
}
// Create a file
h = CreateFileW(name, GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL,
NULL);
if (h == INVALID_HANDLE_VALUE)
{
h = CreateFileW(name, GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_HIDDEN,
NULL);
if (h == INVALID_HANDLE_VALUE)
{
return NULL;
}
}
// Memory allocation
p = Win32MemoryAlloc(sizeof(WIN32IO));
// Store Handle
p->hFile = h;
p->WriteMode = true;
return (void *)p;
}
void *Win32FileCreate(char *name)
{
WIN32IO *p;
HANDLE h;
// Validate arguments
if (name == NULL)
{
return NULL;
}
// Create a file
h = CreateFile(name, GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL,
NULL);
if (h == INVALID_HANDLE_VALUE)
{
h = CreateFile(name, GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_HIDDEN,
NULL);
if (h == INVALID_HANDLE_VALUE)
{
return NULL;
}
}
// Memory allocation
p = Win32MemoryAlloc(sizeof(WIN32IO));
// Store Handle
p->hFile = h;
p->WriteMode = true;
return (void *)p;
}
#define SIZE_OF_CALLSTACK_SYM 10000
#define CALLSTACK_DEPTH 12
// Get the call stack
CALLSTACK_DATA *Win32GetCallStack()
{
#ifndef WIN32_NO_DEBUG_HELP_DLL
DWORD current_eip32 = 0, current_esp32 = 0, current_ebp32 = 0;
UINT64 current_eip = 0, current_esp = 0, current_ebp = 0;
STACKFRAME64 sf;
CALLSTACK_DATA *cs = NULL, *s;
#ifdef CPU_64
CONTEXT context;
#endif // CPU_64
bool ret;
UINT depth = 0;
#ifndef CPU_64
// Register acquisition (32 bit)
__asm
{
mov current_esp32, esp
mov current_ebp32, ebp
};
current_eip32 = (DWORD)Win32GetCallStack;
current_eip = (UINT64)current_eip32;
current_esp = (UINT64)current_esp32;
current_ebp = (UINT64)current_ebp32;
#else // CPU_64
// Register acquisition (64 bit)
Zero(&context, sizeof(context));
context.ContextFlags = CONTEXT_FULL;
RtlCaptureContext(&context);
#endif // CPU_64
Zero(&sf, sizeof(sf));
#ifndef CPU_64
sf.AddrPC.Offset = current_eip;
sf.AddrStack.Offset = current_esp;
sf.AddrFrame.Offset = current_ebp;
#else // CPU_64
sf.AddrPC.Offset = context.Rip;
sf.AddrStack.Offset = context.Rsp;
sf.AddrFrame.Offset = context.Rsp;
#endif // CPU_64
sf.AddrPC.Mode = AddrModeFlat;
sf.AddrStack.Mode = AddrModeFlat;
sf.AddrFrame.Mode = AddrModeFlat;
while (true)
{
DWORD type = IMAGE_FILE_MACHINE_I386;
#ifdef CPU_64
type = IMAGE_FILE_MACHINE_AMD64;
#endif // CPU_64
if ((depth++) >= CALLSTACK_DEPTH)
{
break;
}
#ifndef CPU_64
ret = StackWalk64(type,
hCurrentProcessHandle,
GetCurrentThread(),
&sf,
NULL, NULL, SymFunctionTableAccess64, SymGetModuleBase64, NULL);
#else // CPU_64
ret = StackWalk64(type,
hCurrentProcessHandle,
GetCurrentThread(),
&sf,
&context, NULL, SymFunctionTableAccess64, SymGetModuleBase64, NULL);
#endif // CPU_64
if (ret == false || sf.AddrFrame.Offset == 0)
{
break;
}
if (cs == NULL)
{
cs = OSMemoryAlloc(sizeof(CALLSTACK_DATA));
s = cs;
}
else
{
s->next = OSMemoryAlloc(sizeof(CALLSTACK_DATA));
s = s->next;
}
s->symbol_cache = false;
s->next = NULL;
s->offset = sf.AddrPC.Offset;
s->disp = 0;
s->name = NULL;
s->line = 0;
s->filename[0] = 0;
}
return cs;
#else // WIN32_NO_DEBUG_HELP_DLL
return NULL;
#endif // WIN32_NO_DEBUG_HELP_DLL
}
// Get the symbol information from the call stack
bool Win32GetCallStackSymbolInfo(CALLSTACK_DATA *s)
{
#ifdef WIN32_NO_DEBUG_HELP_DLL
return false;
#else // WIN32_NO_DEBUG_HELP_DLL
UINT64 disp;
UINT disp32, len;
IMAGEHLP_SYMBOL64 *sym;
IMAGEHLP_LINE64 line;
char tmp[MAX_PATH];
// Validate arguments
if (s == NULL)
{
return false;
}
if (s->symbol_cache)
{
return true;
}
sym = OSMemoryAlloc(SIZE_OF_CALLSTACK_SYM);
sym->SizeOfStruct = SIZE_OF_CALLSTACK_SYM;
sym->MaxNameLength = SIZE_OF_CALLSTACK_SYM - sizeof(IMAGEHLP_SYMBOL64);
if (SymGetSymFromAddr64(hCurrentProcessHandle, s->offset, &disp, sym))
{
s->disp = disp;
s->name = OSMemoryAlloc((UINT)strlen(sym->Name) + 1);
lstrcpy(s->name, sym->Name);
}
else
{
s->disp = 0;
s->name = NULL;
}
Zero(&line, sizeof(line));
line.SizeOfStruct = sizeof(line);
if (SymGetLineFromAddr64(hCurrentProcessHandle, s->offset, &disp32, &line))
{
disp = (UINT64)disp32;
s->line = line.LineNumber;
lstrcpy(s->filename, line.FileName);
Win32GetDirFromPath(tmp, sizeof(tmp), s->filename);
len = lstrlen(tmp);
lstrcpy(tmp, &s->filename[len + 1]);
lstrcpy(s->filename, tmp);
}
else
{
s->line = 0;
s->filename[0] = 0;
}
OSMemoryFree(sym);
s->symbol_cache = true;
return true;
#endif // WIN32_NO_DEBUG_HELP_DLL
}
// Default Win32 thread
DWORD CALLBACK Win32DefaultThreadProc(void *param)
{
WIN32THREADSTARTUPINFO *info = (WIN32THREADSTARTUPINFO *)param;
// Validate arguments
if (info == NULL)
{
return 0;
}
Win32InitNewThread();
CoInitialize(NULL);
// Call the thread function
info->thread_proc(info->thread, info->param);
// Release the reference
ReleaseThread(info->thread);
Win32MemoryFree(info);
FreeOpenSSLThreadState();
CoUninitialize();
_endthreadex(0);
return 0;
}
// Wait for the termination of the thread
bool Win32WaitThread(THREAD *t)
{
WIN32THREAD *w;
// Validate arguments
if (t == NULL)
{
return false;
}
w = (WIN32THREAD *)t->pData;
if (w == NULL)
{
return false;
}
// Wait for the thread event
if (WaitForSingleObject(w->hThread, INFINITE) == WAIT_OBJECT_0)
{
// The thread was signaled
return true;
}
// Wait failure (time-out, etc.)
return false;
}
// Release the thread
void Win32FreeThread(THREAD *t)
{
WIN32THREAD *w;
// Validate arguments
if (t == NULL)
{
return;
}
w = (WIN32THREAD *)t->pData;
if (w == NULL)
{
return;
}
// Close the handle
CloseHandle(w->hThread);
// Memory release
Win32MemoryFree(t->pData);
t->pData = NULL;
}
// Thread initialization
bool Win32InitThread(THREAD *t)
{
WIN32THREAD *w;
HANDLE hThread;
DWORD thread_id;
WIN32THREADSTARTUPINFO *info;
// Validate arguments
if (t == NULL)
{
return false;
}
if (t->thread_proc == NULL)
{
return false;
}
// Thread data generation
w = Win32MemoryAlloc(sizeof(WIN32THREAD));
// Creating the startup information
info = Win32MemoryAlloc(sizeof(WIN32THREADSTARTUPINFO));
info->param = t->param;
info->thread_proc = t->thread_proc;
info->thread = t;
AddRef(t->ref);
// Thread creation
t->pData = w;
hThread = (HANDLE)_beginthreadex(NULL, 0, Win32DefaultThreadProc, info, 0, &thread_id);
if (hThread == NULL)
{
// Thread creation failure
t->pData = NULL;
Release(t->ref);
Win32MemoryFree(info);
Win32MemoryFree(w);
return false;
}
// Save the thread information
w->hThread = hThread;
w->thread_id = thread_id;
return true;
}
// Initialize the library for Win32
void Win32Init()
{
INITCOMMONCONTROLSEX c;
OSVERSIONINFO os;
// Get whether it's in a Windows NT
Zero(&os, sizeof(os));
os.dwOSVersionInfoSize = sizeof(os);
GetVersionEx(&os);
if (os.dwPlatformId == VER_PLATFORM_WIN32_NT)
{
// NT system
win32_is_nt = true;
}
else
{
// 9x system
win32_is_nt = false;
}
// Open the stdout
if (hstdout == INVALID_HANDLE_VALUE)
{
hstdout = GetStdHandle(STD_OUTPUT_HANDLE);
}
// Open the stdin
if (hstdin == INVALID_HANDLE_VALUE)
{
hstdin = GetStdHandle(STD_INPUT_HANDLE);
}
Win32InitNewThread();
CoInitialize(NULL);
InitializeCriticalSection(&fasttick_lock);
#ifdef WIN32_USE_HEAP_API_FOR_MEMORY
use_heap_api = true;
#else // WIN32_USE_HEAP_API_FOR_MEMORY
use_heap_api = false;
#endif // WIN32_USE_HEAP_API_FOR_MEMORY
if (MayaquaIsDotNetMode())
{
// If an heap API is called from .NET API, it might crush
use_heap_api = false;
}
if (IsNt() == false)
{
// Do not use the heap related API in Win9x
use_heap_api = false;
}
if (use_heap_api)
{
heap_handle = HeapCreate(0, 0, 0);
}
// Get the process pseudo handle
hCurrentProcessHandle = GetCurrentProcess();
// Initialization of the current directory
// Win32InitCurrentDir(); /* Don't do */
// Initialization of the symbol handler
if (IsMemCheck())
{
#ifndef WIN32_NO_DEBUG_HELP_DLL
SymInitialize(hCurrentProcessHandle, NULL, TRUE);
#endif // WIN32_NO_DEBUG_HELP_DLL
}
// Initialization of the Common Control
Zero(&c, sizeof(INITCOMMONCONTROLSEX));
c.dwSize = sizeof(INITCOMMONCONTROLSEX);
c.dwICC = ICC_ANIMATE_CLASS | ICC_BAR_CLASSES | ICC_COOL_CLASSES |
ICC_DATE_CLASSES | ICC_HOTKEY_CLASS | ICC_INTERNET_CLASSES |
ICC_LISTVIEW_CLASSES | ICC_NATIVEFNTCTL_CLASS |
ICC_PAGESCROLLER_CLASS | ICC_PROGRESS_CLASS |
ICC_TAB_CLASSES | ICC_TREEVIEW_CLASSES | ICC_UPDOWN_CLASS | ICC_USEREX_CLASSES |
ICC_WIN95_CLASSES;
InitCommonControlsEx(&c);
}
// Release the library for Win32
void Win32Free()
{
// Close the symbol handler
if (IsMemCheck())
{
#ifndef WIN32_NO_DEBUG_HELP_DLL
SymCleanup(hCurrentProcessHandle);
#endif // WIN32_NO_DEBUG_HELP_DLL
}
if (use_heap_api)
{
HeapDestroy(heap_handle);
heap_handle = NULL;
}
CoUninitialize();
DeleteCriticalSection(&fasttick_lock);
}
// Memory allocation
void *Win32MemoryAlloc(UINT size)
{
if (use_heap_api)
{
return HeapAlloc(heap_handle, 0, size);
}
else
{
return malloc(size);
}
}
// Memory reallocation
void *Win32MemoryReAlloc(void *addr, UINT size)
{
if (use_heap_api)
{
return HeapReAlloc(heap_handle, 0, addr, size);
}
else
{
return realloc(addr, size);
}
}
// Memory allocation
void Win32MemoryFree(void *addr)
{
if (use_heap_api)
{
HeapFree(heap_handle, 0, addr);
}
else
{
free(addr);
}
}
// Get the system timer
UINT Win32GetTick()
{
return (UINT)timeGetTime();
}
// Get the System Time
void Win32GetSystemTime(SYSTEMTIME *system_time)
{
// Get the System Time
GetSystemTime(system_time);
}
// Increment of 32bit integer
void Win32Inc32(UINT *value)
{
InterlockedIncrement(value);
}
// Decrement of 32bit integer
void Win32Dec32(UINT *value)
{
InterlockedDecrement(value);
}
// Sleep the thread
void Win32Sleep(UINT time)
{
Sleep(time);
}
// Creating a lock
LOCK *Win32NewLock()
{
// Memory allocation
LOCK *lock = Win32MemoryAlloc(sizeof(LOCK));
// Allocate a critical section
CRITICAL_SECTION *critical_section = Win32MemoryAlloc(sizeof(CRITICAL_SECTION));
if (lock == NULL || critical_section == NULL)
{
Win32MemoryFree(lock);
Win32MemoryFree(critical_section);
return NULL;
}
// Initialize the critical section
InitializeCriticalSection(critical_section);
lock->pData = (void *)critical_section;
lock->Ready = true;
return lock;
}
// Lock
bool Win32Lock(LOCK *lock)
{
CRITICAL_SECTION *critical_section;
if (lock->Ready == false)
{
// State is invalid
return false;
}
// Enter the critical section
critical_section = (CRITICAL_SECTION *)lock->pData;
EnterCriticalSection(critical_section);
return true;
}
// Unlock
void Win32Unlock(LOCK *lock)
{
Win32UnlockEx(lock, false);
}
void Win32UnlockEx(LOCK *lock, bool inner)
{
CRITICAL_SECTION *critical_section;
if (lock->Ready == false && inner == false)
{
// State is invalid
return;
}
// Leave the critical section
critical_section = (CRITICAL_SECTION *)lock->pData;
LeaveCriticalSection(critical_section);
}
// Delete the lock
void Win32DeleteLock(LOCK *lock)
{
CRITICAL_SECTION *critical_section;
// Reset the Ready flag safely
Win32Lock(lock);
lock->Ready = false;
Win32UnlockEx(lock, true);
// Delete the critical section
critical_section = (CRITICAL_SECTION *)lock->pData;
DeleteCriticalSection(critical_section);
// Memory release
Win32MemoryFree(critical_section);
Win32MemoryFree(lock);
}
// Initialization of the event
void Win32InitEvent(EVENT *event)
{
// Creating an auto-reset event
HANDLE hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
event->pData = hEvent;
}
// Set the event
void Win32SetEvent(EVENT *event)
{
HANDLE hEvent = (HANDLE)event->pData;
if (hEvent == NULL)
{
return;
}
SetEvent(hEvent);
}
// Reset the event
void Win32ResetEvent(EVENT *event)
{
HANDLE hEvent = (HANDLE)event->pData;
if (hEvent == NULL)
{
return;
}
ResetEvent(hEvent);
}
// Wait for the event
bool Win32WaitEvent(EVENT *event, UINT timeout)
{
HANDLE hEvent = (HANDLE)event->pData;
UINT ret;
if (hEvent == NULL)
{
return false;
}
// Wait for an object
ret = WaitForSingleObject(hEvent, timeout);
if (ret == WAIT_TIMEOUT)
{
// Time-out
return false;
}
else
{
// Signaled state
return true;
}
}
// Release of the event
void Win32FreeEvent(EVENT *event)
{
HANDLE hEvent = (HANDLE)event->pData;
if (hEvent == NULL)
{
return;
}
CloseHandle(hEvent);
}
// Fast getting 64 bit Tick functions for only Win32
UINT64 Win32FastTick64()
{
static UINT last_tick = 0;
static UINT counter = 0;
UINT64 ret;
UINT tick;
EnterCriticalSection(&fasttick_lock);
// Get the current tick value
tick = Win32GetTick();
if (last_tick > tick)
{
// When the previously acquired tick value is larger than acquired this time,
// it can be considered that the counter have gone one around
counter++;
}
last_tick = tick;
ret = (UINT64)tick + (UINT64)counter * 4294967296ULL;
LeaveCriticalSection(&fasttick_lock);
if (start_tick == 0)
{
start_tick = ret;
ret = 0;
}
else
{
ret -= start_tick;
}
return ret + 1;
}
// Read a string from the console
bool Win32InputW(wchar_t *str, UINT size)
{
bool ret = false;
// Validate arguments
if (str == NULL)
{
return false;
}
if (size == 0)
{
size = 0x7fffffff;
}
if (str == NULL || size <= sizeof(wchar_t))
{
if (str != NULL)
{
Zero(str, size);
}
return Win32InputFromFileW(NULL, 0);
}
if (IsNt())
{
DWORD read_size = 0;
2014-07-11 21:06:20 +04:00
if (ReadConsoleW(hstdin, str, (size / sizeof(wchar_t)) - 1, &read_size, NULL))
2014-01-04 17:00:08 +04:00
{
str[read_size] = 0;
UniTrimCrlf(str);
ret = true;
}
else
{
ret = Win32InputFromFileW(str, size);
}
}
else
{
DWORD read_size = 0;
UINT a_size = size / sizeof(wchar_t) + 16;
char *a;
a = ZeroMalloc(a_size);
if (ReadConsoleA(hstdin, a, a_size - 1, &read_size, NULL))
{
a[read_size] = 0;
StrToUni(str, size, a);
UniTrimCrlf(str);
ret = true;
}
else
{
ret = Win32InputFromFileW(str, size);
}
Free(a);
}
return ret;
}
// Get a line from standard input
bool Win32InputFromFileW(wchar_t *str, UINT size)
{
char *a;
if (str == NULL)
{
wchar_t tmp[MAX_SIZE];
Win32InputFromFileW(tmp, sizeof(tmp));
return false;
}
a = Win32InputFromFileLineA();
if (a == NULL)
{
UniStrCpy(str, size, L"");
return false;
}
UtfToUni(str, size, a);
UniTrimCrlf(str);
Free(a);
return true;
}
char *Win32InputFromFileLineA()
{
BUF *b = NewBuf();
char zero = 0;
char *ret = NULL;
bool ok = true;
while (true)
{
char c;
UINT read_size = 0;
if (ReadFile(hstdin, &c, 1, &read_size, NULL) == false)
{
ok = false;
break;
}
if (read_size != 1)
{
ok = false;
break;
}
WriteBuf(b, &c, 1);
if (c == 10)
{
break;
}
}
WriteBuf(b, &zero, 1);
if (ok)
{
ret = CopyStr(b->Buf);
}
FreeBuf(b);
return ret;
}
// Print the string to the console
void Win32PrintW(wchar_t *str)
{
DWORD write_size = 0;
// Validate arguments
if (str == NULL)
{
return;
}
if (IsNt())
{
if (WriteConsoleW(hstdout, str, UniStrLen(str), &write_size, NULL) == false)
{
Win32PrintToFileW(str);
}
}
else
{
char *ansi_str = CopyUniToStr(str);
if (WriteConsoleA(hstdout, ansi_str, StrLen(ansi_str), &write_size, NULL) == false)
{
Win32PrintToFileW(str);
}
Free(ansi_str);
}
}
void Win32PrintToFileW(wchar_t *str)
{
char *utf;
DWORD size = 0;
// Validate arguments
if (str == NULL)
{
return;
}
utf = CopyUniToUtf(str);
WriteFile(hstdout, utf, StrLen(utf), &size, NULL);
Free(utf);
}
#endif // WIN32