1 // The MIT License (MIT) 2 // 3 // Copyright (c) 2015 Sergey Makeev, Vadim Slyusarev 4 // 5 // Permission is hereby granted, free of charge, to any person obtaining a copy 6 // of this software and associated documentation files (the "Software"), to deal 7 // in the Software without restriction, including without limitation the rights 8 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 9 // copies of the Software, and to permit persons to whom the Software is 10 // furnished to do so, subject to the following conditions: 11 // 12 // The above copyright notice and this permission notice shall be included in 13 // all copies or substantial portions of the Software. 14 // 15 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 18 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 21 // THE SOFTWARE. 22 #pragma once 23 24 #include <MTConfig.h> 25 #include <MTTypes.h> 26 27 // 28 // micro windows header is used to avoid including heavy windows header to MTPlatform.h 29 // 30 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 31 32 33 #define MW_WINBASEAPI __declspec(dllimport) 34 #define MW_WINAPI __stdcall 35 36 37 #if defined(_WINDOWS_) || defined(_WINBASE_) 38 39 // 40 // if windows.h is already included simply create aliases to the MW_ types 41 // 42 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 43 typedef LARGE_INTEGER MW_LARGE_INTEGER; 44 typedef BOOL MW_BOOL; 45 typedef HANDLE MW_HANDLE; 46 47 typedef DWORD MW_DWORD; 48 typedef WORD MW_WORD; 49 typedef DWORD64 MW_DWORD64; 50 typedef ULONG_PTR MW_ULONG_PTR; 51 52 typedef LPTHREAD_START_ROUTINE TThreadStartFunc; 53 54 typedef SYSTEM_INFO MW_SYSTEM_INFO; 55 56 typedef CRITICAL_SECTION MW_CRITICAL_SECTION; 57 58 typedef CONTEXT MW_CONTEXT; 59 60 #define MW_INFINITE (INFINITE) 61 #define MW_WAIT_OBJECT_0 (WAIT_OBJECT_0) 62 #define MW_MEM_COMMIT (MEM_COMMIT) 63 #define MW_PAGE_READWRITE (PAGE_READWRITE) 64 #define MW_PAGE_NOACCESS (PAGE_NOACCESS) 65 #define MW_MEM_RELEASE (MEM_RELEASE) 66 67 68 #define MW_STACK_BASE_OFFSET (FIELD_OFFSET(NT_TIB, StackBase)) 69 #define MW_STACK_STACK_LIMIT_OFFSET (FIELD_OFFSET(NT_TIB, StackLimit)) 70 #define MW_CONTEXT_FULL (CONTEXT_FULL) 71 72 73 #define MW_FIBER_FLAG_FLOAT_SWITCH (FIBER_FLAG_FLOAT_SWITCH) 74 75 #define MW_THREAD_PRIORITY_HIGHEST (THREAD_PRIORITY_HIGHEST) 76 #define MW_THREAD_PRIORITY_NORMAL (THREAD_PRIORITY_NORMAL) 77 #define MW_THREAD_PRIORITY_LOWEST (THREAD_PRIORITY_LOWEST) 78 79 #define MW_CREATE_SUSPENDED (CREATE_SUSPENDED) 80 81 #define MW_MAXIMUM_PROCESSORS (MAXIMUM_PROCESSORS) 82 83 #else 84 85 // windows.h is not included, so declare types 86 87 // 88 // define types 89 // 90 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 91 92 struct MW_LARGE_INTEGER 93 { 94 int64 QuadPart; 95 }; 96 97 typedef int MW_BOOL; 98 typedef void* MW_HANDLE; 99 100 typedef unsigned long MW_DWORD; 101 typedef unsigned short MW_WORD; 102 typedef unsigned __int64 MW_DWORD64; 103 104 #if MT_PTR64 105 typedef unsigned __int64 MW_ULONG_PTR; 106 #else 107 typedef unsigned __int32 MW_ULONG_PTR; 108 #endif 109 110 // 111 // define thread function 112 // 113 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 114 typedef MW_DWORD ( MW_WINAPI *TThreadStartFunc )(void* lpThreadParameter); 115 116 // 117 // define fiber function 118 // 119 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 120 typedef void ( MW_WINAPI *TFiberStartFunc)(void* lpFiberParameter); 121 122 123 124 125 // 126 // system info structure, only used members are declared 127 // 128 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 129 struct MW_SYSTEM_INFO 130 { 131 uint8 _unused_01[4]; 132 MW_DWORD dwPageSize; 133 void* _unused_02[3]; 134 MW_DWORD dwNumberOfProcessors; 135 uint8 _unused_03[12]; 136 }; 137 138 #if MT_PTR64 139 140 // 141 // x64 critical section, only used members are declared 142 // 143 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 144 struct MW_CRITICAL_SECTION 145 { 146 uint8 _unused[40]; 147 }; 148 149 // 150 // x64 machine context, only used members are declared 151 // 152 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 153 struct __declspec(align(16)) MW_CONTEXT 154 { 155 uint8 _unused_01[48]; 156 MW_DWORD ContextFlags; 157 uint8 _unused_02[100]; 158 MW_DWORD64 Rsp; 159 uint8 _unused_03[88]; 160 MW_DWORD64 Rip; 161 uint8 _unused_04[976]; 162 }; 163 164 static_assert(__alignof(MW_CONTEXT) == 16, "MW_CONTEXT align requirements must be 16 bytes"); 165 166 #define MW_STACK_BASE_OFFSET (8) 167 #define MW_STACK_STACK_LIMIT_OFFSET (16) 168 #define MW_CONTEXT_FULL (0x10000B) 169 170 #else 171 172 // 173 // x86 critical section, only used members are declared 174 // 175 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 176 struct MW_CRITICAL_SECTION 177 { 178 uint8 _unused[24]; 179 }; 180 181 // 182 // x86 machine context, only used members are declared 183 // 184 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 185 struct MW_CONTEXT 186 { 187 MW_DWORD ContextFlags; 188 uint8 _unused_01[180]; 189 MW_DWORD Eip; 190 uint8 _unused_02[8]; 191 MW_DWORD Esp; 192 uint8 _unused_03[516]; 193 }; 194 195 196 #define MW_STACK_BASE_OFFSET (4) 197 #define MW_STACK_STACK_LIMIT_OFFSET (8) 198 #define MW_CONTEXT_FULL (0x10007) 199 200 201 #endif 202 203 204 // 205 // defines and flags 206 // 207 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 208 #define MW_INFINITE (0xFFFFFFFF) 209 #define MW_WAIT_OBJECT_0 (0) 210 #define MW_MEM_COMMIT (0x1000) 211 #define MW_PAGE_READWRITE (0x04) 212 #define MW_PAGE_NOACCESS (0x01) 213 #define MW_MEM_RELEASE (0x8000) 214 #define MW_FIBER_FLAG_FLOAT_SWITCH (0x1) 215 216 #define MW_THREAD_PRIORITY_HIGHEST (2) 217 #define MW_THREAD_PRIORITY_NORMAL (0) 218 #define MW_THREAD_PRIORITY_LOWEST (-2) 219 220 #define MW_CREATE_SUSPENDED (0x00000004) 221 222 #if MT_PTR64 223 #define MW_MAXIMUM_PROCESSORS (64) 224 #else 225 #define MW_MAXIMUM_PROCESSORS (32) 226 #endif 227 228 #endif 229 230 231 232 #if !defined(MW_SKIP_FUNCTIONS) && !defined(_WINDOWS_) 233 234 // 235 // functions 236 // 237 /////////////////////////////////////////////////////////////////////////////////////////////////////////////////// 238 239 240 extern "C" { 241 242 MW_WINBASEAPI MW_BOOL MW_WINAPI QueryPerformanceFrequency(MW_LARGE_INTEGER* lpFrequency); 243 MW_WINBASEAPI MW_BOOL MW_WINAPI QueryPerformanceCounter(MW_LARGE_INTEGER* lpPerformanceCount); 244 245 MW_WINBASEAPI MW_ULONG_PTR MW_WINAPI SetThreadAffinityMask(MW_HANDLE hThread, MW_ULONG_PTR dwThreadAffinityMask ); 246 MW_WINBASEAPI MW_DWORD MW_WINAPI SetThreadIdealProcessor(MW_HANDLE hThread, MW_DWORD dwIdealProcessor ); 247 MW_WINBASEAPI MW_BOOL MW_WINAPI SetThreadPriority(MW_HANDLE hThread, int nPriority ); 248 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateThread(void* lpThreadAttributes, size_t dwStackSize, TThreadStartFunc lpStartAddress, void* lpParameter, MW_DWORD dwCreationFlags, MW_DWORD* lpThreadId); 249 MW_WINBASEAPI MW_BOOL MW_WINAPI CloseHandle(MW_HANDLE hObject); 250 MW_WINBASEAPI MW_HANDLE MW_WINAPI GetCurrentThread(); 251 MW_WINBASEAPI MW_DWORD MW_WINAPI GetCurrentThreadId(); 252 MW_WINBASEAPI MW_DWORD MW_WINAPI ResumeThread(MW_HANDLE hThread); 253 254 255 MW_WINBASEAPI void MW_WINAPI GetSystemInfo(MW_SYSTEM_INFO* lpSystemInfo); 256 257 MW_WINBASEAPI void MW_WINAPI Sleep(MW_DWORD dwMilliseconds); 258 MW_WINBASEAPI MW_DWORD MW_WINAPI WaitForSingleObject(MW_HANDLE hHandle, MW_DWORD dwMilliseconds); 259 260 MW_WINBASEAPI bool MW_WINAPI InitializeCriticalSectionAndSpinCount(MW_CRITICAL_SECTION* lpCriticalSection, MW_DWORD dwSpinCount ); 261 MW_WINBASEAPI void MW_WINAPI DeleteCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection ); 262 MW_WINBASEAPI void MW_WINAPI EnterCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection ); 263 MW_WINBASEAPI void MW_WINAPI LeaveCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection ); 264 265 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateEventA(MW_CRITICAL_SECTION* lpEventAttributes, MW_BOOL bManualReset, MW_BOOL bInitialState, const char* lpName ); 266 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateEventW(MW_CRITICAL_SECTION* lpEventAttributes, MW_BOOL bManualReset, MW_BOOL bInitialState, const wchar_t* lpName ); 267 MW_WINBASEAPI MW_BOOL MW_WINAPI SetEvent( MW_HANDLE hEvent ); 268 MW_WINBASEAPI MW_BOOL MW_WINAPI ResetEvent( MW_HANDLE hEvent ); 269 270 MW_WINBASEAPI MW_BOOL MW_WINAPI GetThreadContext( MW_HANDLE hThread, MW_CONTEXT* lpContext ); 271 MW_WINBASEAPI MW_BOOL MW_WINAPI SetThreadContext( MW_HANDLE hThread, const MW_CONTEXT* lpContext ); 272 273 MW_WINBASEAPI void* MW_WINAPI VirtualAlloc( void* lpAddress, size_t dwSize, MW_DWORD flAllocationType, MW_DWORD flProtect ); 274 MW_WINBASEAPI MW_BOOL MW_WINAPI VirtualProtect( void* lpAddress, size_t dwSize, MW_DWORD flNewProtect, MW_DWORD* lpflOldProtect ); 275 MW_WINBASEAPI MW_BOOL MW_WINAPI VirtualFree( void* lpAddress, size_t dwSize, MW_DWORD dwFreeType ); 276 277 278 MW_WINBASEAPI void MW_WINAPI DeleteFiber( void* lpFiber ); 279 MW_WINBASEAPI void* MW_WINAPI ConvertThreadToFiberEx( void* lpParameter, MW_DWORD dwFlags ); 280 MW_WINBASEAPI void* MW_WINAPI CreateFiber( size_t dwStackSize, TFiberStartFunc lpStartAddress, void* lpParameter ); 281 MW_WINBASEAPI void MW_WINAPI SwitchToFiber( void* lpFiber ); 282 283 MW_WINBASEAPI void MW_WINAPI RaiseException(MW_DWORD dwExceptionCode, MW_DWORD dwExceptionFlags, MW_DWORD nNumberOfArguments, const MW_ULONG_PTR* lpArguments ); 284 285 286 } 287 288 #endif 289