9e052883 |
1 | /* Threads.h -- multithreading library\r |
2 | 2021-12-21 : Igor Pavlov : Public domain */\r |
3 | \r |
4 | #ifndef __7Z_THREADS_H\r |
5 | #define __7Z_THREADS_H\r |
6 | \r |
7 | #ifdef _WIN32\r |
8 | #include <Windows.h>\r |
9 | #else\r |
10 | \r |
11 | #if defined(__linux__)\r |
12 | #if !defined(__APPLE__) && !defined(_AIX) && !defined(__ANDROID__)\r |
13 | #ifndef _7ZIP_AFFINITY_DISABLE\r |
14 | #define _7ZIP_AFFINITY_SUPPORTED\r |
15 | // #pragma message(" ==== _7ZIP_AFFINITY_SUPPORTED")\r |
16 | // #define _GNU_SOURCE\r |
17 | #endif\r |
18 | #endif\r |
19 | #endif\r |
20 | \r |
21 | #include <pthread.h>\r |
22 | \r |
23 | #endif\r |
24 | \r |
25 | #include "7zTypes.h"\r |
26 | \r |
27 | EXTERN_C_BEGIN\r |
28 | \r |
29 | #ifdef _WIN32\r |
30 | \r |
31 | WRes HandlePtr_Close(HANDLE *h);\r |
32 | WRes Handle_WaitObject(HANDLE h);\r |
33 | \r |
34 | typedef HANDLE CThread;\r |
35 | \r |
36 | #define Thread_Construct(p) { *(p) = NULL; }\r |
37 | #define Thread_WasCreated(p) (*(p) != NULL)\r |
38 | #define Thread_Close(p) HandlePtr_Close(p)\r |
39 | // #define Thread_Wait(p) Handle_WaitObject(*(p))\r |
40 | \r |
41 | #ifdef UNDER_CE\r |
42 | // if (USE_THREADS_CreateThread is defined), we use _beginthreadex()\r |
43 | // if (USE_THREADS_CreateThread is not definned), we use CreateThread()\r |
44 | #define USE_THREADS_CreateThread\r |
45 | #endif\r |
46 | \r |
47 | typedef\r |
48 | #ifdef USE_THREADS_CreateThread\r |
49 | DWORD\r |
50 | #else\r |
51 | unsigned\r |
52 | #endif\r |
53 | THREAD_FUNC_RET_TYPE;\r |
54 | \r |
55 | typedef DWORD_PTR CAffinityMask;\r |
56 | typedef DWORD_PTR CCpuSet;\r |
57 | \r |
58 | #define CpuSet_Zero(p) { *(p) = 0; }\r |
59 | #define CpuSet_Set(p, cpu) { *(p) |= ((DWORD_PTR)1 << (cpu)); }\r |
60 | \r |
61 | #else // _WIN32\r |
62 | \r |
63 | typedef struct _CThread\r |
64 | {\r |
65 | pthread_t _tid;\r |
66 | int _created;\r |
67 | } CThread;\r |
68 | \r |
69 | #define Thread_Construct(p) { (p)->_tid = 0; (p)->_created = 0; }\r |
70 | #define Thread_WasCreated(p) ((p)->_created != 0)\r |
71 | WRes Thread_Close(CThread *p);\r |
72 | // #define Thread_Wait Thread_Wait_Close\r |
73 | \r |
74 | typedef void * THREAD_FUNC_RET_TYPE;\r |
75 | \r |
76 | typedef UInt64 CAffinityMask;\r |
77 | \r |
78 | #ifdef _7ZIP_AFFINITY_SUPPORTED\r |
79 | \r |
80 | typedef cpu_set_t CCpuSet;\r |
81 | #define CpuSet_Zero(p) CPU_ZERO(p)\r |
82 | #define CpuSet_Set(p, cpu) CPU_SET(cpu, p)\r |
83 | #define CpuSet_IsSet(p, cpu) CPU_ISSET(cpu, p)\r |
84 | \r |
85 | #else\r |
86 | \r |
87 | typedef UInt64 CCpuSet;\r |
88 | #define CpuSet_Zero(p) { *(p) = 0; }\r |
89 | #define CpuSet_Set(p, cpu) { *(p) |= ((UInt64)1 << (cpu)); }\r |
90 | #define CpuSet_IsSet(p, cpu) ((*(p) & ((UInt64)1 << (cpu))) != 0)\r |
91 | \r |
92 | #endif\r |
93 | \r |
94 | \r |
95 | #endif // _WIN32\r |
96 | \r |
97 | \r |
98 | #define THREAD_FUNC_CALL_TYPE MY_STD_CALL\r |
99 | \r |
100 | #if defined(_WIN32) && defined(__GNUC__)\r |
101 | /* GCC compiler for x86 32-bit uses the rule:\r |
102 | the stack is 16-byte aligned before CALL instruction for function calling.\r |
103 | But only root function main() contains instructions that\r |
104 | set 16-byte alignment for stack pointer. And another functions\r |
105 | just keep alignment, if it was set in some parent function.\r |
106 | \r |
107 | The problem:\r |
108 | if we create new thread in MinGW (GCC) 32-bit x86 via _beginthreadex() or CreateThread(),\r |
109 | the root function of thread doesn't set 16-byte alignment.\r |
110 | And stack frames in all child functions also will be unaligned in that case.\r |
111 | \r |
112 | Here we set (force_align_arg_pointer) attribute for root function of new thread.\r |
113 | Do we need (force_align_arg_pointer) also for another systems? */\r |
114 | \r |
115 | #define THREAD_FUNC_ATTRIB_ALIGN_ARG __attribute__((force_align_arg_pointer))\r |
116 | // #define THREAD_FUNC_ATTRIB_ALIGN_ARG // for debug : bad alignment in SSE functions\r |
117 | #else\r |
118 | #define THREAD_FUNC_ATTRIB_ALIGN_ARG\r |
119 | #endif\r |
120 | \r |
121 | #define THREAD_FUNC_DECL THREAD_FUNC_ATTRIB_ALIGN_ARG THREAD_FUNC_RET_TYPE THREAD_FUNC_CALL_TYPE\r |
122 | \r |
123 | typedef THREAD_FUNC_RET_TYPE (THREAD_FUNC_CALL_TYPE * THREAD_FUNC_TYPE)(void *);\r |
124 | WRes Thread_Create(CThread *p, THREAD_FUNC_TYPE func, LPVOID param);\r |
125 | WRes Thread_Create_With_Affinity(CThread *p, THREAD_FUNC_TYPE func, LPVOID param, CAffinityMask affinity);\r |
126 | WRes Thread_Wait_Close(CThread *p);\r |
127 | \r |
128 | #ifdef _WIN32\r |
129 | #define Thread_Create_With_CpuSet(p, func, param, cs) \\r |
130 | Thread_Create_With_Affinity(p, func, param, *cs)\r |
131 | #else\r |
132 | WRes Thread_Create_With_CpuSet(CThread *p, THREAD_FUNC_TYPE func, LPVOID param, const CCpuSet *cpuSet);\r |
133 | #endif\r |
134 | \r |
135 | \r |
136 | #ifdef _WIN32\r |
137 | \r |
138 | typedef HANDLE CEvent;\r |
139 | typedef CEvent CAutoResetEvent;\r |
140 | typedef CEvent CManualResetEvent;\r |
141 | #define Event_Construct(p) *(p) = NULL\r |
142 | #define Event_IsCreated(p) (*(p) != NULL)\r |
143 | #define Event_Close(p) HandlePtr_Close(p)\r |
144 | #define Event_Wait(p) Handle_WaitObject(*(p))\r |
145 | WRes Event_Set(CEvent *p);\r |
146 | WRes Event_Reset(CEvent *p);\r |
147 | WRes ManualResetEvent_Create(CManualResetEvent *p, int signaled);\r |
148 | WRes ManualResetEvent_CreateNotSignaled(CManualResetEvent *p);\r |
149 | WRes AutoResetEvent_Create(CAutoResetEvent *p, int signaled);\r |
150 | WRes AutoResetEvent_CreateNotSignaled(CAutoResetEvent *p);\r |
151 | \r |
152 | typedef HANDLE CSemaphore;\r |
153 | #define Semaphore_Construct(p) *(p) = NULL\r |
154 | #define Semaphore_IsCreated(p) (*(p) != NULL)\r |
155 | #define Semaphore_Close(p) HandlePtr_Close(p)\r |
156 | #define Semaphore_Wait(p) Handle_WaitObject(*(p))\r |
157 | WRes Semaphore_Create(CSemaphore *p, UInt32 initCount, UInt32 maxCount);\r |
158 | WRes Semaphore_OptCreateInit(CSemaphore *p, UInt32 initCount, UInt32 maxCount);\r |
159 | WRes Semaphore_ReleaseN(CSemaphore *p, UInt32 num);\r |
160 | WRes Semaphore_Release1(CSemaphore *p);\r |
161 | \r |
162 | typedef CRITICAL_SECTION CCriticalSection;\r |
163 | WRes CriticalSection_Init(CCriticalSection *p);\r |
164 | #define CriticalSection_Delete(p) DeleteCriticalSection(p)\r |
165 | #define CriticalSection_Enter(p) EnterCriticalSection(p)\r |
166 | #define CriticalSection_Leave(p) LeaveCriticalSection(p)\r |
167 | \r |
168 | \r |
169 | #else // _WIN32\r |
170 | \r |
171 | typedef struct _CEvent\r |
172 | {\r |
173 | int _created;\r |
174 | int _manual_reset;\r |
175 | int _state;\r |
176 | pthread_mutex_t _mutex;\r |
177 | pthread_cond_t _cond;\r |
178 | } CEvent;\r |
179 | \r |
180 | typedef CEvent CAutoResetEvent;\r |
181 | typedef CEvent CManualResetEvent;\r |
182 | \r |
183 | #define Event_Construct(p) (p)->_created = 0\r |
184 | #define Event_IsCreated(p) ((p)->_created)\r |
185 | \r |
186 | WRes ManualResetEvent_Create(CManualResetEvent *p, int signaled);\r |
187 | WRes ManualResetEvent_CreateNotSignaled(CManualResetEvent *p);\r |
188 | WRes AutoResetEvent_Create(CAutoResetEvent *p, int signaled);\r |
189 | WRes AutoResetEvent_CreateNotSignaled(CAutoResetEvent *p);\r |
190 | WRes Event_Set(CEvent *p);\r |
191 | WRes Event_Reset(CEvent *p);\r |
192 | WRes Event_Wait(CEvent *p);\r |
193 | WRes Event_Close(CEvent *p);\r |
194 | \r |
195 | \r |
196 | typedef struct _CSemaphore\r |
197 | {\r |
198 | int _created;\r |
199 | UInt32 _count;\r |
200 | UInt32 _maxCount;\r |
201 | pthread_mutex_t _mutex;\r |
202 | pthread_cond_t _cond;\r |
203 | } CSemaphore;\r |
204 | \r |
205 | #define Semaphore_Construct(p) (p)->_created = 0\r |
206 | #define Semaphore_IsCreated(p) ((p)->_created)\r |
207 | \r |
208 | WRes Semaphore_Create(CSemaphore *p, UInt32 initCount, UInt32 maxCount);\r |
209 | WRes Semaphore_OptCreateInit(CSemaphore *p, UInt32 initCount, UInt32 maxCount);\r |
210 | WRes Semaphore_ReleaseN(CSemaphore *p, UInt32 num);\r |
211 | #define Semaphore_Release1(p) Semaphore_ReleaseN(p, 1)\r |
212 | WRes Semaphore_Wait(CSemaphore *p);\r |
213 | WRes Semaphore_Close(CSemaphore *p);\r |
214 | \r |
215 | \r |
216 | typedef struct _CCriticalSection\r |
217 | {\r |
218 | pthread_mutex_t _mutex;\r |
219 | } CCriticalSection;\r |
220 | \r |
221 | WRes CriticalSection_Init(CCriticalSection *p);\r |
222 | void CriticalSection_Delete(CCriticalSection *cs);\r |
223 | void CriticalSection_Enter(CCriticalSection *cs);\r |
224 | void CriticalSection_Leave(CCriticalSection *cs);\r |
225 | \r |
226 | LONG InterlockedIncrement(LONG volatile *addend);\r |
227 | \r |
228 | #endif // _WIN32\r |
229 | \r |
230 | EXTERN_C_END\r |
231 | \r |
232 | #endif\r |