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 //
29 // micro windows header is used to avoid including heavy windows header to MTPlatform.h
30 //
31 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
32 
33 
34 #define MW_WINBASEAPI __declspec(dllimport)
35 #define MW_WINAPI __stdcall
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_CURRENT_FIBER_OFFSET (FIELD_OFFSET(NT_TIB, FiberData))
69 #define MW_STACK_BASE_OFFSET (FIELD_OFFSET(NT_TIB, StackBase))
70 #define MW_STACK_STACK_LIMIT_OFFSET (FIELD_OFFSET(NT_TIB, StackLimit))
71 #define MW_CONTEXT_FULL (CONTEXT_FULL)
72 
73 
74 #define MW_FIBER_FLAG_FLOAT_SWITCH (FIBER_FLAG_FLOAT_SWITCH)
75 
76 #define MW_THREAD_PRIORITY_HIGHEST (THREAD_PRIORITY_HIGHEST)
77 #define MW_THREAD_PRIORITY_NORMAL (THREAD_PRIORITY_NORMAL)
78 #define MW_THREAD_PRIORITY_LOWEST (THREAD_PRIORITY_LOWEST)
79 
80 #define MW_CREATE_SUSPENDED (CREATE_SUSPENDED)
81 
82 #define MW_MAXIMUM_PROCESSORS (MAXIMUM_PROCESSORS)
83 
84 #else
85 
86 // windows.h is not included, so declare types
87 
88 //
89 // define types
90 //
91 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
92 
93 struct MW_LARGE_INTEGER
94 {
95 	int64 QuadPart;
96 };
97 
98 typedef int MW_BOOL;
99 typedef void* MW_HANDLE;
100 
101 typedef unsigned long MW_DWORD;
102 typedef unsigned short MW_WORD;
103 typedef unsigned __int64 MW_DWORD64;
104 
105 #if MT_PTR64
106 typedef unsigned __int64 MW_ULONG_PTR;
107 #else
108 typedef unsigned __int32 MW_ULONG_PTR;
109 #endif
110 
111 //
112 // define thread function
113 //
114 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
115 typedef MW_DWORD ( MW_WINAPI *TThreadStartFunc )(void* lpThreadParameter);
116 
117 //
118 // define fiber function
119 //
120 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
121 typedef void ( MW_WINAPI *TFiberStartFunc)(void* lpFiberParameter);
122 
123 
124 
125 
126 //
127 // system info structure, only used members are declared
128 //
129 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
130 struct MW_SYSTEM_INFO
131 {
132 	uint8 _unused_01[4];
133 	MW_DWORD dwPageSize;
134 	void* _unused_02[3];
135 	MW_DWORD dwNumberOfProcessors;
136 	uint8 _unused_03[12];
137 };
138 
139 #if MT_PTR64
140 
141 //
142 // x64 critical section, only used members are declared
143 //
144 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
145 struct MW_CRITICAL_SECTION
146 {
147 	uint8 _unused[40];
148 };
149 
150 //
151 // x64 machine context, only used members are declared
152 //
153 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
154 struct __declspec(align(16)) MW_CONTEXT
155 {
156 	uint8 _unused_01[48];
157 	MW_DWORD ContextFlags;
158 	uint8 _unused_02[100];
159 	MW_DWORD64 Rsp;
160 	uint8 _unused_03[88];
161 	MW_DWORD64 Rip;
162 	uint8 _unused_04[976];
163 };
164 
165 static_assert(__alignof(MW_CONTEXT) == 16, "MW_CONTEXT align requirements must be 16 bytes");
166 
167 #define MW_CURRENT_FIBER_OFFSET (32)
168 #define MW_STACK_BASE_OFFSET (8)
169 #define MW_STACK_STACK_LIMIT_OFFSET (16)
170 #define MW_CONTEXT_FULL (0x10000B)
171 
172 #else
173 
174 //
175 // x86 critical section, only used members are declared
176 //
177 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
178 struct MW_CRITICAL_SECTION
179 {
180 	uint8 _unused[24];
181 };
182 
183 //
184 // x86 machine context, only used members are declared
185 //
186 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
187 struct MW_CONTEXT
188 {
189 	MW_DWORD ContextFlags;
190 	uint8 _unused_01[180];
191 	MW_DWORD   Eip;
192 	uint8 _unused_02[8];
193 	MW_DWORD   Esp;
194 	uint8 _unused_03[516];
195 };
196 
197 
198 #define MW_CURRENT_FIBER_OFFSET (16)
199 #define MW_STACK_BASE_OFFSET (4)
200 #define MW_STACK_STACK_LIMIT_OFFSET (8)
201 #define MW_CONTEXT_FULL (0x10007)
202 
203 
204 #endif
205 
206 
207 //
208 // defines and flags
209 //
210 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
211 #define MW_INFINITE (0xFFFFFFFF)
212 #define MW_WAIT_OBJECT_0 (0)
213 #define MW_MEM_COMMIT (0x1000)
214 #define MW_PAGE_READWRITE (0x04)
215 #define MW_PAGE_NOACCESS (0x01)
216 #define MW_MEM_RELEASE (0x8000)
217 
218 
219 #define MW_THREAD_PRIORITY_HIGHEST (2)
220 #define MW_THREAD_PRIORITY_NORMAL (0)
221 #define MW_THREAD_PRIORITY_LOWEST (-2)
222 
223 #define MW_CREATE_SUSPENDED (0x00000004)
224 
225 
226 #if MT_PTR64
227 #define MW_MAXIMUM_PROCESSORS (64)
228 #define MW_FIBER_FLAG_FLOAT_SWITCH (0x1)
229 #else
230 #define MW_MAXIMUM_PROCESSORS (32)
231 #define MW_FIBER_FLAG_FLOAT_SWITCH (0x1)
232 #endif
233 
234 
235 #endif
236 
237 
238 
239 
240 #if !defined(MW_SKIP_FUNCTIONS) && !defined(_WINDOWS_)
241 
242 //
243 // functions
244 //
245 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
246 
247 
248 extern "C" {
249 
250 MW_WINBASEAPI MW_BOOL MW_WINAPI QueryPerformanceFrequency(MW_LARGE_INTEGER* lpFrequency);
251 MW_WINBASEAPI MW_BOOL MW_WINAPI QueryPerformanceCounter(MW_LARGE_INTEGER* lpPerformanceCount);
252 
253 MW_WINBASEAPI MW_ULONG_PTR MW_WINAPI SetThreadAffinityMask(MW_HANDLE hThread, MW_ULONG_PTR dwThreadAffinityMask );
254 MW_WINBASEAPI MW_DWORD MW_WINAPI SetThreadIdealProcessor(MW_HANDLE hThread, MW_DWORD dwIdealProcessor );
255 MW_WINBASEAPI MW_BOOL MW_WINAPI SetThreadPriority(MW_HANDLE hThread, int nPriority );
256 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateThread(void* lpThreadAttributes, size_t dwStackSize, TThreadStartFunc lpStartAddress, void* lpParameter, MW_DWORD dwCreationFlags, MW_DWORD* lpThreadId);
257 MW_WINBASEAPI MW_BOOL MW_WINAPI CloseHandle(MW_HANDLE hObject);
258 MW_WINBASEAPI MW_HANDLE MW_WINAPI GetCurrentThread();
259 MW_WINBASEAPI MW_DWORD MW_WINAPI GetCurrentThreadId();
260 MW_WINBASEAPI MW_DWORD MW_WINAPI ResumeThread(MW_HANDLE hThread);
261 MW_WINBASEAPI MW_BOOL MW_WINAPI SwitchToThread();
262 
263 
264 MW_WINBASEAPI void MW_WINAPI GetSystemInfo(MW_SYSTEM_INFO* lpSystemInfo);
265 
266 MW_WINBASEAPI void MW_WINAPI Sleep(MW_DWORD dwMilliseconds);
267 MW_WINBASEAPI MW_DWORD MW_WINAPI WaitForSingleObject(MW_HANDLE hHandle, MW_DWORD dwMilliseconds);
268 
269 MW_WINBASEAPI bool MW_WINAPI InitializeCriticalSectionAndSpinCount(MW_CRITICAL_SECTION* lpCriticalSection, MW_DWORD dwSpinCount );
270 MW_WINBASEAPI void MW_WINAPI DeleteCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection );
271 MW_WINBASEAPI void MW_WINAPI EnterCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection );
272 MW_WINBASEAPI void MW_WINAPI LeaveCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection );
273 
274 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateEventA(MW_CRITICAL_SECTION* lpEventAttributes, MW_BOOL bManualReset, MW_BOOL bInitialState, const char* lpName );
275 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateEventW(MW_CRITICAL_SECTION* lpEventAttributes, MW_BOOL bManualReset, MW_BOOL bInitialState, const wchar_t* lpName );
276 MW_WINBASEAPI MW_BOOL MW_WINAPI SetEvent( MW_HANDLE hEvent );
277 MW_WINBASEAPI MW_BOOL MW_WINAPI ResetEvent( MW_HANDLE hEvent );
278 
279 MW_WINBASEAPI MW_BOOL MW_WINAPI GetThreadContext( MW_HANDLE hThread, MW_CONTEXT* lpContext );
280 MW_WINBASEAPI MW_BOOL MW_WINAPI SetThreadContext( MW_HANDLE hThread, const MW_CONTEXT* lpContext );
281 
282 MW_WINBASEAPI void* MW_WINAPI VirtualAlloc( void* lpAddress, size_t dwSize, MW_DWORD flAllocationType, MW_DWORD flProtect );
283 MW_WINBASEAPI MW_BOOL MW_WINAPI VirtualProtect( void* lpAddress, size_t dwSize, MW_DWORD flNewProtect, MW_DWORD* lpflOldProtect );
284 MW_WINBASEAPI MW_BOOL MW_WINAPI VirtualFree( void* lpAddress, size_t dwSize, MW_DWORD dwFreeType );
285 
286 
287 MW_WINBASEAPI void MW_WINAPI DeleteFiber( void* lpFiber );
288 MW_WINBASEAPI void* MW_WINAPI ConvertThreadToFiberEx( void* lpParameter, MW_DWORD dwFlags );
289 MW_WINBASEAPI void* MW_WINAPI CreateFiber( size_t dwStackSize, TFiberStartFunc lpStartAddress, void* lpParameter );
290 MW_WINBASEAPI void MW_WINAPI SwitchToFiber( void* lpFiber );
291 MW_WINBASEAPI MW_BOOL MW_WINAPI IsThreadAFiber();
292 
293 MW_WINBASEAPI void MW_WINAPI RaiseException(MW_DWORD dwExceptionCode, MW_DWORD dwExceptionFlags, MW_DWORD nNumberOfArguments, const MW_ULONG_PTR* lpArguments );
294 
295 MW_WINBASEAPI MW_DWORD MW_WINAPI GetLastError();
296 
297 
298 
299 
300 }
301 
302 #endif
303