threading.h 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. /**
  2. * Copyright (c) 2016 Tino Reichardt
  3. * All rights reserved.
  4. *
  5. * You can contact the author at:
  6. * - zstdmt source repository: https://github.com/mcmilk/zstdmt
  7. *
  8. * This source code is licensed under both the BSD-style license (found in the
  9. * LICENSE file in the root directory of this source tree) and the GPLv2 (found
  10. * in the COPYING file in the root directory of this source tree).
  11. * You may select, at your option, one of the above-listed licenses.
  12. */
  13. #ifndef THREADING_H_938743
  14. #define THREADING_H_938743
  15. #include "debug.h"
  16. #if defined (__cplusplus)
  17. extern "C" {
  18. #endif
  19. #if defined(ZSTD_MULTITHREAD) && defined(_WIN32)
  20. /**
  21. * Windows minimalist Pthread Wrapper
  22. */
  23. #ifdef WINVER
  24. # undef WINVER
  25. #endif
  26. #define WINVER 0x0600
  27. #ifdef _WIN32_WINNT
  28. # undef _WIN32_WINNT
  29. #endif
  30. #define _WIN32_WINNT 0x0600
  31. #ifndef WIN32_LEAN_AND_MEAN
  32. # define WIN32_LEAN_AND_MEAN
  33. #endif
  34. #undef ERROR /* reported already defined on VS 2015 (Rich Geldreich) */
  35. #include <windows.h>
  36. #undef ERROR
  37. #define ERROR(name) ZSTD_ERROR(name)
  38. /* mutex */
  39. #define ZSTD_pthread_mutex_t CRITICAL_SECTION
  40. #define ZSTD_pthread_mutex_init(a, b) ((void)(b), InitializeCriticalSection((a)), 0)
  41. #define ZSTD_pthread_mutex_destroy(a) DeleteCriticalSection((a))
  42. #define ZSTD_pthread_mutex_lock(a) EnterCriticalSection((a))
  43. #define ZSTD_pthread_mutex_unlock(a) LeaveCriticalSection((a))
  44. /* condition variable */
  45. #define ZSTD_pthread_cond_t CONDITION_VARIABLE
  46. #define ZSTD_pthread_cond_init(a, b) ((void)(b), InitializeConditionVariable((a)), 0)
  47. #define ZSTD_pthread_cond_destroy(a) ((void)(a))
  48. #define ZSTD_pthread_cond_wait(a, b) SleepConditionVariableCS((a), (b), INFINITE)
  49. #define ZSTD_pthread_cond_signal(a) WakeConditionVariable((a))
  50. #define ZSTD_pthread_cond_broadcast(a) WakeAllConditionVariable((a))
  51. /* ZSTD_pthread_create() and ZSTD_pthread_join() */
  52. typedef HANDLE ZSTD_pthread_t;
  53. int ZSTD_pthread_create(ZSTD_pthread_t* thread, const void* unused,
  54. void* (*start_routine) (void*), void* arg);
  55. int ZSTD_pthread_join(ZSTD_pthread_t thread);
  56. /**
  57. * add here more wrappers as required
  58. */
  59. #elif defined(ZSTD_MULTITHREAD) /* posix assumed ; need a better detection method */
  60. /* === POSIX Systems === */
  61. # include <pthread.h>
  62. #if DEBUGLEVEL < 1
  63. #define ZSTD_pthread_mutex_t pthread_mutex_t
  64. #define ZSTD_pthread_mutex_init(a, b) pthread_mutex_init((a), (b))
  65. #define ZSTD_pthread_mutex_destroy(a) pthread_mutex_destroy((a))
  66. #define ZSTD_pthread_mutex_lock(a) pthread_mutex_lock((a))
  67. #define ZSTD_pthread_mutex_unlock(a) pthread_mutex_unlock((a))
  68. #define ZSTD_pthread_cond_t pthread_cond_t
  69. #define ZSTD_pthread_cond_init(a, b) pthread_cond_init((a), (b))
  70. #define ZSTD_pthread_cond_destroy(a) pthread_cond_destroy((a))
  71. #define ZSTD_pthread_cond_wait(a, b) pthread_cond_wait((a), (b))
  72. #define ZSTD_pthread_cond_signal(a) pthread_cond_signal((a))
  73. #define ZSTD_pthread_cond_broadcast(a) pthread_cond_broadcast((a))
  74. #define ZSTD_pthread_t pthread_t
  75. #define ZSTD_pthread_create(a, b, c, d) pthread_create((a), (b), (c), (d))
  76. #define ZSTD_pthread_join(a) pthread_join((a),NULL)
  77. #else /* DEBUGLEVEL >= 1 */
  78. /* Debug implementation of threading.
  79. * In this implementation we use pointers for mutexes and condition variables.
  80. * This way, if we forget to init/destroy them the program will crash or ASAN
  81. * will report leaks.
  82. */
  83. #define ZSTD_pthread_mutex_t pthread_mutex_t*
  84. int ZSTD_pthread_mutex_init(ZSTD_pthread_mutex_t* mutex, pthread_mutexattr_t const* attr);
  85. int ZSTD_pthread_mutex_destroy(ZSTD_pthread_mutex_t* mutex);
  86. #define ZSTD_pthread_mutex_lock(a) pthread_mutex_lock(*(a))
  87. #define ZSTD_pthread_mutex_unlock(a) pthread_mutex_unlock(*(a))
  88. #define ZSTD_pthread_cond_t pthread_cond_t*
  89. int ZSTD_pthread_cond_init(ZSTD_pthread_cond_t* cond, pthread_condattr_t const* attr);
  90. int ZSTD_pthread_cond_destroy(ZSTD_pthread_cond_t* cond);
  91. #define ZSTD_pthread_cond_wait(a, b) pthread_cond_wait(*(a), *(b))
  92. #define ZSTD_pthread_cond_signal(a) pthread_cond_signal(*(a))
  93. #define ZSTD_pthread_cond_broadcast(a) pthread_cond_broadcast(*(a))
  94. #define ZSTD_pthread_t pthread_t
  95. #define ZSTD_pthread_create(a, b, c, d) pthread_create((a), (b), (c), (d))
  96. #define ZSTD_pthread_join(a) pthread_join((a),NULL)
  97. #endif
  98. #else /* ZSTD_MULTITHREAD not defined */
  99. /* No multithreading support */
  100. typedef int ZSTD_pthread_mutex_t;
  101. #define ZSTD_pthread_mutex_init(a, b) ((void)(a), (void)(b), 0)
  102. #define ZSTD_pthread_mutex_destroy(a) ((void)(a))
  103. #define ZSTD_pthread_mutex_lock(a) ((void)(a))
  104. #define ZSTD_pthread_mutex_unlock(a) ((void)(a))
  105. typedef int ZSTD_pthread_cond_t;
  106. #define ZSTD_pthread_cond_init(a, b) ((void)(a), (void)(b), 0)
  107. #define ZSTD_pthread_cond_destroy(a) ((void)(a))
  108. #define ZSTD_pthread_cond_wait(a, b) ((void)(a), (void)(b))
  109. #define ZSTD_pthread_cond_signal(a) ((void)(a))
  110. #define ZSTD_pthread_cond_broadcast(a) ((void)(a))
  111. /* do not use ZSTD_pthread_t */
  112. #endif /* ZSTD_MULTITHREAD */
  113. #if defined (__cplusplus)
  114. }
  115. #endif
  116. #endif /* THREADING_H_938743 */