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 <MTTypes.h>
25 
26 //
27 // micro windows header is used to avoid including heavy windows header to MTPlatform.h
28 //
29 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
30 
31 
32 #define MW_WINBASEAPI __declspec(dllimport)
33 #define MW_WINAPI __stdcall
34 
35 
36 #if defined(_WINDOWS_) || defined(_WINBASE_)
37 
38 //
39 // if windows.h is already included simply create aliases to the MW_ types
40 //
41 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
42 typedef LARGE_INTEGER MW_LARGE_INTEGER;
43 typedef BOOL MW_BOOL;
44 typedef HANDLE MW_HANDLE;
45 
46 typedef DWORD MW_DWORD;
47 typedef WORD MW_WORD;
48 typedef DWORD64 MW_DWORD64;
49 
50 typedef LPTHREAD_START_ROUTINE TThreadStartFunc;
51 
52 typedef SYSTEM_INFO MW_SYSTEM_INFO;
53 
54 typedef CRITICAL_SECTION MW_CRITICAL_SECTION;
55 
56 typedef CONTEXT MW_CONTEXT;
57 
58 #define MW_INFINITE (INFINITE)
59 #define MW_WAIT_OBJECT_0 (WAIT_OBJECT_0)
60 #define MW_MEM_COMMIT (MEM_COMMIT)
61 #define MW_PAGE_READWRITE (PAGE_READWRITE)
62 #define MW_PAGE_NOACCESS (PAGE_NOACCESS)
63 #define MW_MEM_RELEASE (MEM_RELEASE)
64 
65 
66 #define MW_STACK_BASE_OFFSET (FIELD_OFFSET(NT_TIB, StackBase))
67 #define MW_STACK_STACK_LIMIT_OFFSET (FIELD_OFFSET(NT_TIB, StackLimit))
68 #define MW_CONTEXT_FULL (CONTEXT_FULL)
69 
70 
71 #define MW_FIBER_FLAG_FLOAT_SWITCH (FIBER_FLAG_FLOAT_SWITCH)
72 
73 
74 #else
75 
76 // windows.h is not included, so declare types
77 
78 //
79 // define types
80 //
81 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
82 
83 struct MW_LARGE_INTEGER
84 {
85 	int64 QuadPart;
86 };
87 
88 typedef int MW_BOOL;
89 typedef void* MW_HANDLE;
90 
91 typedef unsigned long MW_DWORD;
92 typedef unsigned short MW_WORD;
93 typedef unsigned __int64 MW_DWORD64;
94 
95 //
96 // define thread function
97 //
98 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
99 typedef MW_DWORD ( MW_WINAPI *TThreadStartFunc )(void* lpThreadParameter);
100 
101 //
102 // define fiber function
103 //
104 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
105 typedef void ( MW_WINAPI *TFiberStartFunc)(void* lpFiberParameter);
106 
107 
108 
109 
110 //
111 // system info structure, only used members are declared
112 //
113 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
114 struct MW_SYSTEM_INFO
115 {
116 	uint8 _unused_01[4];
117 	MW_DWORD dwPageSize;
118 	void* _unused_02[3];
119 	MW_DWORD dwNumberOfProcessors;
120 	uint8 _unused_03[12];
121 };
122 
123 #if defined(_M_X64)
124 
125 //
126 // x64 critical section, only used members are declared
127 //
128 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
129 struct MW_CRITICAL_SECTION
130 {
131 	uint8 _unused[40];
132 };
133 
134 //
135 // x64 machine context, only used members are declared
136 //
137 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
138 struct __declspec(align(16)) MW_CONTEXT
139 {
140 	uint8 _unused_01[48];
141 	MW_DWORD ContextFlags;
142 	uint8 _unused_02[100];
143 	MW_DWORD64 Rsp;
144 	uint8 _unused_03[88];
145 	MW_DWORD64 Rip;
146 	uint8 _unused_04[976];
147 };
148 
149 static_assert(__alignof(MW_CONTEXT) == 16, "MW_CONTEXT align requirements must be 16 bytes");
150 
151 #define MW_STACK_BASE_OFFSET (8)
152 #define MW_STACK_STACK_LIMIT_OFFSET (16)
153 #define MW_CONTEXT_FULL (0x10000B)
154 
155 #else
156 
157 //
158 // x86 critical section, only used members are declared
159 //
160 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
161 struct MW_CRITICAL_SECTION
162 {
163 	uint8 _unused[24];
164 };
165 
166 //
167 // x86 machine context, only used members are declared
168 //
169 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
170 struct MW_CONTEXT
171 {
172 	MW_DWORD ContextFlags;
173 	uint8 _unused_01[180];
174 	MW_DWORD   Eip;
175 	uint8 _unused_02[8];
176 	MW_DWORD   Esp;
177 	uint8 _unused_03[516];
178 };
179 
180 
181 #define MW_STACK_BASE_OFFSET (4)
182 #define MW_STACK_STACK_LIMIT_OFFSET (8)
183 #define MW_CONTEXT_FULL (0x10007)
184 
185 
186 #endif
187 
188 
189 //
190 // defines and flags
191 //
192 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
193 #define MW_INFINITE (0xFFFFFFFF)
194 #define MW_WAIT_OBJECT_0 (0)
195 #define MW_MEM_COMMIT (0x1000)
196 #define MW_PAGE_READWRITE (0x04)
197 #define MW_PAGE_NOACCESS (0x01)
198 #define MW_MEM_RELEASE (0x8000)
199 #define MW_FIBER_FLAG_FLOAT_SWITCH (0x1)
200 
201 
202 #endif
203 
204 
205 
206 #if !defined(MW_SKIP_FUNCTIONS) && !defined(_WINDOWS_)
207 
208 //
209 // functions
210 //
211 ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
212 
213 
214 extern "C" {
215 
216 MW_WINBASEAPI MW_BOOL MW_WINAPI QueryPerformanceFrequency(MW_LARGE_INTEGER* lpFrequency);
217 MW_WINBASEAPI MW_BOOL MW_WINAPI QueryPerformanceCounter(MW_LARGE_INTEGER* lpPerformanceCount);
218 
219 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateThread(void* lpThreadAttributes, size_t dwStackSize, TThreadStartFunc lpStartAddress, void* lpParameter, MW_DWORD dwCreationFlags, MW_DWORD* lpThreadId);
220 MW_WINBASEAPI MW_BOOL MW_WINAPI CloseHandle(MW_HANDLE hObject);
221 MW_WINBASEAPI MW_HANDLE MW_WINAPI GetCurrentThread();
222 MW_WINBASEAPI MW_DWORD MW_WINAPI GetCurrentThreadId();
223 
224 MW_WINBASEAPI void MW_WINAPI GetSystemInfo(MW_SYSTEM_INFO* lpSystemInfo);
225 
226 MW_WINBASEAPI void MW_WINAPI Sleep(MW_DWORD dwMilliseconds);
227 MW_WINBASEAPI MW_DWORD MW_WINAPI WaitForSingleObject(MW_HANDLE hHandle, MW_DWORD dwMilliseconds);
228 
229 MW_WINBASEAPI bool MW_WINAPI InitializeCriticalSectionAndSpinCount(MW_CRITICAL_SECTION* lpCriticalSection, MW_DWORD dwSpinCount );
230 MW_WINBASEAPI void MW_WINAPI DeleteCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection );
231 MW_WINBASEAPI void MW_WINAPI EnterCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection );
232 MW_WINBASEAPI void MW_WINAPI LeaveCriticalSection(MW_CRITICAL_SECTION* lpCriticalSection );
233 
234 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateEventA(MW_CRITICAL_SECTION* lpEventAttributes, MW_BOOL bManualReset, MW_BOOL bInitialState, const char* lpName );
235 MW_WINBASEAPI MW_HANDLE MW_WINAPI CreateEventW(MW_CRITICAL_SECTION* lpEventAttributes, MW_BOOL bManualReset, MW_BOOL bInitialState, const wchar_t* lpName );
236 MW_WINBASEAPI MW_BOOL MW_WINAPI SetEvent( MW_HANDLE hEvent );
237 MW_WINBASEAPI MW_BOOL MW_WINAPI ResetEvent( MW_HANDLE hEvent );
238 
239 MW_WINBASEAPI MW_BOOL MW_WINAPI GetThreadContext( MW_HANDLE hThread, MW_CONTEXT* lpContext );
240 MW_WINBASEAPI MW_BOOL MW_WINAPI SetThreadContext( MW_HANDLE hThread, const MW_CONTEXT* lpContext );
241 
242 MW_WINBASEAPI void* MW_WINAPI VirtualAlloc( void* lpAddress, size_t dwSize, MW_DWORD flAllocationType, MW_DWORD flProtect );
243 MW_WINBASEAPI MW_BOOL MW_WINAPI VirtualProtect( void* lpAddress, size_t dwSize, MW_DWORD flNewProtect, MW_DWORD* lpflOldProtect );
244 MW_WINBASEAPI MW_BOOL MW_WINAPI VirtualFree( void* lpAddress, size_t dwSize, MW_DWORD dwFreeType );
245 
246 
247 MW_WINBASEAPI void MW_WINAPI DeleteFiber( void* lpFiber );
248 MW_WINBASEAPI void* MW_WINAPI ConvertThreadToFiberEx( void* lpParameter, MW_DWORD dwFlags );
249 MW_WINBASEAPI void* MW_WINAPI CreateFiber( size_t dwStackSize, TFiberStartFunc lpStartAddress, void* lpParameter );
250 MW_WINBASEAPI void MW_WINAPI SwitchToFiber( void* lpFiber );
251 
252 
253 
254 }
255 
256 #endif
257 
258 
259