| @@ -0,0 +1,473 @@ |
| 1 | +#include "system.h" |
| 2 | + |
| 3 | +#ifndef _WIN32 |
| 4 | + |
| 5 | +#include <stdlib.h> |
| 6 | +#include <time.h> |
| 7 | +#include <errno.h> |
| 8 | +#include <unistd.h> |
| 9 | +#if defined(__linux) || defined(__linux__) |
| 10 | +#include <sys/prctl.h> |
| 11 | +#endif |
| 12 | + |
| 13 | +#define nullptr 0 |
| 14 | + |
| 15 | +struct Event |
| 16 | +{ |
| 17 | + Event * volatile pMultipleCond; |
| 18 | + pthread_mutex_t mutex; |
| 19 | + pthread_cond_t cond; |
| 20 | + volatile bool signaled; |
| 21 | + bool manual_reset; |
| 22 | +}; |
| 23 | + |
| 24 | +static bool InitEvent(Event *e) |
| 25 | +{ |
| 26 | +#if (defined(ANDROID) && !defined(__LP64__)) || defined(__APPLE__) |
| 27 | + if (pthread_cond_init(&e->cond, NULL)) |
| 28 | + return false; |
| 29 | +#else |
| 30 | + pthread_condattr_t attr; |
| 31 | + if (pthread_condattr_init(&attr)) |
| 32 | + return false; |
| 33 | + if (pthread_condattr_setclock(&attr, CLOCK_MONOTONIC)) |
| 34 | + { |
| 35 | + pthread_condattr_destroy(&attr); |
| 36 | + return false; |
| 37 | + } |
| 38 | + if (pthread_cond_init(&e->cond, &attr)) |
| 39 | + { |
| 40 | + pthread_condattr_destroy(&attr); |
| 41 | + return false; |
| 42 | + } |
| 43 | + pthread_condattr_destroy(&attr); |
| 44 | +#endif |
| 45 | + if (pthread_mutex_init(&e->mutex, NULL)) |
| 46 | + { |
| 47 | + pthread_cond_destroy(&e->cond); |
| 48 | + return false; |
| 49 | + } |
| 50 | + e->pMultipleCond = NULL; |
| 51 | + return true; |
| 52 | +} |
| 53 | + |
| 54 | +#ifdef __APPLE__ |
| 55 | +#include <mach/mach_time.h> |
| 56 | +static inline uint64_t GetAbsTimeInNanoseconds() |
| 57 | +{ |
| 58 | + static mach_timebase_info_data_t g_timebase_info; |
| 59 | + if (g_timebase_info.denom == 0) |
| 60 | + mach_timebase_info(&g_timebase_info); |
| 61 | + return mach_absolute_time()*g_timebase_info.numer/g_timebase_info.denom; |
| 62 | +} |
| 63 | +#endif |
| 64 | + |
| 65 | +static inline void GetAbsTime(timespec *ts, uint32_t timeout) |
| 66 | +{ |
| 67 | +#if defined(__APPLE__) |
| 68 | + uint64_t cur_time = GetAbsTimeInNanoseconds(); |
| 69 | + ts->tv_sec = cur_time/1000000000u + timeout/1000u; |
| 70 | + ts->tv_nsec = (cur_time % 1000000000u) + (timeout % 1000u)*1000000u; |
| 71 | +#else |
| 72 | + clock_gettime(CLOCK_MONOTONIC, ts); |
| 73 | + ts->tv_sec += timeout/1000u; |
| 74 | + ts->tv_nsec += (timeout % 1000u)*1000000u; |
| 75 | +#endif |
| 76 | + if (ts->tv_nsec >= 1000000000) |
| 77 | + { |
| 78 | + ts->tv_nsec -= 1000000000; |
| 79 | + ts->tv_sec++; |
| 80 | + } |
| 81 | +} |
| 82 | + |
| 83 | +static inline int CondTimedWait(pthread_cond_t *cond, pthread_mutex_t *mutex, const struct timespec *abstime) |
| 84 | +{ |
| 85 | +#if defined(ANDROID) && !defined(__LP64__) |
| 86 | + return pthread_cond_timedwait_monotonic_np(cond, mutex, abstime); |
| 87 | +#elif defined(__APPLE__) |
| 88 | + timespec reltime; |
| 89 | + uint64_t cur_time = GetAbsTimeInNanoseconds(); |
| 90 | + reltime.tv_sec = abstime->tv_sec - cur_time/1000000000u; |
| 91 | + reltime.tv_nsec = abstime->tv_nsec - (cur_time % 1000000000u); |
| 92 | + if (reltime.tv_nsec < 0) |
| 93 | + { |
| 94 | + reltime.tv_nsec += 1000000000; |
| 95 | + reltime.tv_sec--; |
| 96 | + } |
| 97 | + if ((reltime.tv_sec < 0) || ((reltime.tv_sec == 0) && (reltime.tv_nsec == 0))) |
| 98 | + return ETIMEDOUT; |
| 99 | + return pthread_cond_timedwait_relative_np(cond, mutex, &reltime); |
| 100 | +#else |
| 101 | + return pthread_cond_timedwait(cond, mutex, abstime); |
| 102 | +#endif |
| 103 | +} |
| 104 | + |
| 105 | +static bool WaitForEvent(Event *e, uint32_t timeout, bool *signaled) |
| 106 | +{ |
| 107 | + if (pthread_mutex_lock(&e->mutex)) |
| 108 | + return false; |
| 109 | + |
| 110 | + if (timeout == INFINITE) |
| 111 | + { |
| 112 | + while (!e->signaled) |
| 113 | + pthread_cond_wait(&e->cond, &e->mutex); |
| 114 | + } else if (timeout != 0) |
| 115 | + { |
| 116 | + timespec t; |
| 117 | + GetAbsTime(&t, timeout); |
| 118 | + while (!e->signaled) |
| 119 | + { |
| 120 | + if (CondTimedWait(&e->cond, &e->mutex, &t)) |
| 121 | + break; |
| 122 | + } |
| 123 | + } |
| 124 | + *signaled = e->signaled; |
| 125 | + if (!e->manual_reset) |
| 126 | + e->signaled = false; |
| 127 | + |
| 128 | + if (pthread_mutex_unlock(&e->mutex)) |
| 129 | + return false; |
| 130 | + return true; |
| 131 | +} |
| 132 | + |
| 133 | +static bool WaitForMultipleEvents(Event **e, uint32_t count, uint32_t timeout, bool waitAll, int *signaled_num) |
| 134 | +{ |
| 135 | + uint32_t i; |
| 136 | +#define PTHR(func, num) for (i = num; i < count; i++)\ |
| 137 | + if (func(&e[i]->mutex))\ |
| 138 | + return false; |
| 139 | + PTHR(pthread_mutex_lock, 0); |
| 140 | + |
| 141 | + int sig_num = -1; |
| 142 | + if (timeout == 0) |
| 143 | + { |
| 144 | +#define CHECK_SIGNALED \ |
| 145 | + if (waitAll)\ |
| 146 | + {\ |
| 147 | + for (i = 0; i < count; i++)\ |
| 148 | + if (!e[i]->signaled)\ |
| 149 | + break;\ |
| 150 | + if (i == count)\ |
| 151 | + for (i = 0; i < count; i++)\ |
| 152 | + {\ |
| 153 | + if (sig_num < 0 && e[i]->signaled)\ |
| 154 | + sig_num = (int)i;\ |
| 155 | + if (!e[i]->manual_reset)\ |
| 156 | + e[i]->signaled = false;\ |
| 157 | + }\ |
| 158 | + } else\ |
| 159 | + {\ |
| 160 | + for (i = 0; i < count; i++)\ |
| 161 | + if (e[i]->signaled)\ |
| 162 | + {\ |
| 163 | + sig_num = (int)i;\ |
| 164 | + if (!e[i]->manual_reset)\ |
| 165 | + e[i]->signaled = false;\ |
| 166 | + break;\ |
| 167 | + }\ |
| 168 | + } |
| 169 | + CHECK_SIGNALED; |
| 170 | + } else |
| 171 | + if (timeout == INFINITE) |
| 172 | + { |
| 173 | +#define SET_MULTIPLE(val) for (i = 1; i < count; i++)\ |
| 174 | + e[i]->pMultipleCond = val; |
| 175 | + SET_MULTIPLE(e[0]); |
| 176 | + for (;;) |
| 177 | + { |
| 178 | + CHECK_SIGNALED; |
| 179 | + if (sig_num >= 0) |
| 180 | + break; |
| 181 | + PTHR(pthread_mutex_unlock, 1); |
| 182 | + pthread_cond_wait(&e[0]->cond, &e[0]->mutex); |
| 183 | + PTHR(pthread_mutex_lock, 1); |
| 184 | + } |
| 185 | + SET_MULTIPLE(0); |
| 186 | + } else |
| 187 | + { |
| 188 | + SET_MULTIPLE(e[0]); |
| 189 | + timespec t; |
| 190 | + GetAbsTime(&t, timeout); |
| 191 | + for (;;) |
| 192 | + { |
| 193 | + CHECK_SIGNALED; |
| 194 | + if (sig_num >= 0) |
| 195 | + break; |
| 196 | + PTHR(pthread_mutex_unlock, 1); |
| 197 | + int res = CondTimedWait(&e[0]->cond, &e[0]->mutex, &t); |
| 198 | + PTHR(pthread_mutex_lock, 1); |
| 199 | + if (res) |
| 200 | + break; |
| 201 | + } |
| 202 | + SET_MULTIPLE(0); |
| 203 | + } |
| 204 | + PTHR(pthread_mutex_unlock, 0); |
| 205 | + *signaled_num = sig_num; |
| 206 | + return true; |
| 207 | +} |
| 208 | + |
| 209 | +static bool SignalEvent(Event *e) |
| 210 | +{ |
| 211 | + if (pthread_mutex_lock(&e->mutex)) |
| 212 | + return false; |
| 213 | + |
| 214 | + Event *pMultipleCond = e->pMultipleCond; |
| 215 | + e->signaled = true; |
| 216 | + if (pthread_cond_signal(&e->cond)) |
| 217 | + return false; |
| 218 | + |
| 219 | + if (pthread_mutex_unlock(&e->mutex)) |
| 220 | + return false; |
| 221 | + |
| 222 | + if (pMultipleCond && pMultipleCond != e) |
| 223 | + { |
| 224 | + if (pthread_mutex_lock(&pMultipleCond->mutex)) |
| 225 | + return false; |
| 226 | + if (pthread_cond_signal(&pMultipleCond->cond)) |
| 227 | + return false; |
| 228 | + if (pthread_mutex_unlock(&pMultipleCond->mutex)) |
| 229 | + return false; |
| 230 | + } |
| 231 | + return true; |
| 232 | +} |
| 233 | + |
| 234 | +static bool ResetEvent(Event *e) |
| 235 | +{ |
| 236 | + if (pthread_mutex_lock(&e->mutex)) |
| 237 | + return false; |
| 238 | + e->signaled = false; |
| 239 | + if (pthread_mutex_unlock(&e->mutex)) |
| 240 | + return false; |
| 241 | + return true; |
| 242 | +} |
| 243 | + |
| 244 | +HANDLE event_create(bool manualReset, bool initialState) |
| 245 | +{ |
| 246 | + Event *e = (Event *)malloc(sizeof(*e)); |
| 247 | + if (!e) |
| 248 | + return NULL; |
| 249 | + if (!InitEvent(e)) |
| 250 | + { |
| 251 | + free(e); |
| 252 | + return NULL; |
| 253 | + } |
| 254 | + e->manual_reset = manualReset; |
| 255 | + e->signaled = initialState; |
| 256 | + return (HANDLE)e; |
| 257 | +} |
| 258 | + |
| 259 | +bool event_destroy(HANDLE event) |
| 260 | +{ |
| 261 | + Event *e = (Event *)event; |
| 262 | + if (!e) |
| 263 | + return false; |
| 264 | + if (pthread_cond_destroy(&e->cond)) |
| 265 | + return false; |
| 266 | + if (pthread_mutex_destroy(&e->mutex)) |
| 267 | + return false; |
| 268 | + free((void *)e); |
| 269 | + return true; |
| 270 | +} |
| 271 | + |
| 272 | +bool event_set(HANDLE event) |
| 273 | +{ |
| 274 | + return SignalEvent((Event *)event); |
| 275 | +} |
| 276 | + |
| 277 | +bool event_reset(HANDLE event) |
| 278 | +{ |
| 279 | + return ResetEvent((Event *)event); |
| 280 | +} |
| 281 | + |
| 282 | +int event_wait(HANDLE event, uint32_t milliseconds) |
| 283 | +{ |
| 284 | + bool signaled; |
| 285 | + if (!WaitForEvent((Event *)event, milliseconds, &signaled)) |
| 286 | + return WAIT_FAILED; |
| 287 | + return signaled ? WAIT_OBJECT : WAIT_TIMEOUT; |
| 288 | +} |
| 289 | + |
| 290 | +int event_wait_multiple(uint32_t count, const HANDLE *events, bool waitAll, uint32_t milliseconds) |
| 291 | +{ |
| 292 | + if (count == 1) |
| 293 | + return event_wait(events[0], milliseconds); |
| 294 | + int signaled_num = -1; |
| 295 | + if (!WaitForMultipleEvents((Event **)events, count, milliseconds, waitAll, &signaled_num)) |
| 296 | + return WAIT_FAILED; |
| 297 | + return (signaled_num < 0) ? WAIT_TIMEOUT : (WAIT_OBJECT_0 + signaled_num); |
| 298 | +} |
| 299 | + |
| 300 | +bool InitializeCriticalSection(LPCRITICAL_SECTION lpCriticalSection) |
| 301 | +{ |
| 302 | + pthread_mutexattr_t ma; |
| 303 | + if (pthread_mutexattr_init(&ma)) |
| 304 | + return false; |
| 305 | + if (pthread_mutexattr_settype(&ma, PTHREAD_MUTEX_RECURSIVE)) |
| 306 | + { |
| 307 | + pthread_mutexattr_destroy(&ma); |
| 308 | + return false; |
| 309 | + } |
| 310 | + if (pthread_mutex_init((pthread_mutex_t *)lpCriticalSection, &ma)) |
| 311 | + { |
| 312 | + pthread_mutexattr_destroy(&ma); |
| 313 | + return false; |
| 314 | + } |
| 315 | + if (pthread_mutexattr_destroy(&ma)) |
| 316 | + return false; |
| 317 | + return true; |
| 318 | +} |
| 319 | + |
| 320 | +bool DeleteCriticalSection(LPCRITICAL_SECTION lpCriticalSection) |
| 321 | +{ |
| 322 | + if (pthread_mutex_destroy((pthread_mutex_t *)lpCriticalSection)) |
| 323 | + return false; |
| 324 | + return true; |
| 325 | +} |
| 326 | + |
| 327 | +bool EnterCriticalSection(LPCRITICAL_SECTION lpCriticalSection) |
| 328 | +{ |
| 329 | + if (pthread_mutex_lock((pthread_mutex_t *)lpCriticalSection)) |
| 330 | + return false; |
| 331 | + return true; |
| 332 | +} |
| 333 | + |
| 334 | +bool LeaveCriticalSection(LPCRITICAL_SECTION lpCriticalSection) |
| 335 | +{ |
| 336 | + if (pthread_mutex_unlock((pthread_mutex_t *)lpCriticalSection)) |
| 337 | + return false; |
| 338 | + return true; |
| 339 | +} |
| 340 | + |
| 341 | +HANDLE thread_create(LPTHREAD_START_ROUTINE lpStartAddress, void *lpParameter) |
| 342 | +{ |
| 343 | + pthread_t *t = (pthread_t *)malloc(sizeof(*t)); |
| 344 | + if (!t) |
| 345 | + return nullptr; |
| 346 | + if (pthread_create(t, 0, lpStartAddress, lpParameter)) |
| 347 | + { |
| 348 | + free(t); |
| 349 | + return nullptr; |
| 350 | + } |
| 351 | + //if (lpThreadId) |
| 352 | + // *lpThreadId = (uint32_t)*t; |
| 353 | + return (HANDLE)t; |
| 354 | +} |
| 355 | + |
| 356 | +bool thread_close(HANDLE thread) |
| 357 | +{ |
| 358 | + if (!thread) |
| 359 | + return false; |
| 360 | + pthread_t *t = (pthread_t *)thread; |
| 361 | + if (*t) |
| 362 | + pthread_detach(*t); |
| 363 | + free(t); |
| 364 | + return true; |
| 365 | +} |
| 366 | + |
| 367 | +void *thread_wait(HANDLE thread) |
| 368 | +{ |
| 369 | + if (!thread) |
| 370 | + return (void*)-1; |
| 371 | + void *ret = 0; |
| 372 | + pthread_t *t = (pthread_t *)thread; |
| 373 | + if (!*t) |
| 374 | + return ret; |
| 375 | + int res = pthread_join(*t, &ret); |
| 376 | + if (res) |
| 377 | + return (void*)-1; |
| 378 | +#if 0 |
| 379 | + if (timeout == 0) |
| 380 | + { |
| 381 | + int res = pthread_tryjoin_np(*t, &ret); |
| 382 | + if (res) |
| 383 | + return false; |
| 384 | + } else |
| 385 | + if (timeout == INFINITE) |
| 386 | + { |
| 387 | + int res = pthread_join(*t, &ret); |
| 388 | + if (res) |
| 389 | + return false; |
| 390 | + } else |
| 391 | + { |
| 392 | + timespec ts; |
| 393 | + GetAbsTime(&ts, timeout); |
| 394 | + int res = pthread_timedjoin_np(*t, &ret, &ts); |
| 395 | + if (res) |
| 396 | + return false; |
| 397 | + } |
| 398 | +#endif |
| 399 | + *t = 0; // thread joined - no need to detach |
| 400 | + return ret; |
| 401 | +} |
| 402 | + |
| 403 | +#else //_WIN32 |
| 404 | + |
| 405 | +bool event_destroy(HANDLE event) |
| 406 | +{ |
| 407 | + CloseHandle(event); |
| 408 | + return true; |
| 409 | +} |
| 410 | + |
| 411 | +bool thread_close(HANDLE thread) |
| 412 | +{ |
| 413 | + CloseHandle(thread); |
| 414 | + return true; |
| 415 | +} |
| 416 | + |
| 417 | +bool thread_wait(HANDLE thread) |
| 418 | +{ |
| 419 | + return WaitForSingleObject(thread, INFINITE) == WAIT_OBJECT_0; |
| 420 | +} |
| 421 | + |
| 422 | +#endif //_WIN32 |
| 423 | + |
| 424 | +bool thread_name(const char *name) |
| 425 | +{ |
| 426 | +#ifdef _WIN32 |
| 427 | + struct tagTHREADNAME_INFO |
| 428 | + { |
| 429 | + DWORD dwType; |
| 430 | + LPCSTR szName; |
| 431 | + DWORD dwThreadID; |
| 432 | + DWORD dwFlags; |
| 433 | + } info = { 0x1000, name, (DWORD)-1, 0 }; |
| 434 | + __try |
| 435 | + { |
| 436 | + RaiseException(0x406D1388, 0, sizeof(info)/sizeof(ULONG_PTR), (ULONG_PTR*)&info); |
| 437 | + } |
| 438 | + __except(EXCEPTION_EXECUTE_HANDLER) |
| 439 | + { |
| 440 | + } |
| 441 | + return true; |
| 442 | +#elif defined(__linux) || defined(__linux__) |
| 443 | + return (0 == prctl(PR_SET_NAME, name, 0, 0, 0)); |
| 444 | + //return (0 == pthread_setname_np(pthread_self(), name)); |
| 445 | +#else // macos, ios |
| 446 | + return (0 == pthread_setname_np(name)); |
| 447 | +#endif |
| 448 | +} |
| 449 | + |
| 450 | +bool thread_sleep(uint32_t milliseconds) |
| 451 | +{ |
| 452 | +#ifdef _WIN32 |
| 453 | + Sleep(milliseconds); |
| 454 | +#else |
| 455 | + usleep((useconds_t)milliseconds*1000); |
| 456 | +#endif |
| 457 | +} |
| 458 | + |
| 459 | +uint64_t GetTime() |
| 460 | +{ |
| 461 | + uint64_t time; |
| 462 | +#ifdef _WIN32 |
| 463 | + QueryPerformanceCounter((LARGE_INTEGER*)&time); |
| 464 | +#elif defined(__APPLE__) |
| 465 | + time = GetAbsTimeInNanoseconds() / 1000u; |
| 466 | +#else |
| 467 | + timespec ts; |
| 468 | + // CLOCK_PROCESS_CPUTIME_ID CLOCK_THREAD_CPUTIME_ID |
| 469 | + clock_gettime(CLOCK_MONOTONIC, &ts); |
| 470 | + time = (uint64_t)ts.tv_sec * 1000000u + ts.tv_nsec / 1000u; |
| 471 | +#endif |
| 472 | + return time; |
| 473 | +} |