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