2 * Copyright (c) 2016 Tino Reichardt
5 * This source code is licensed under both the BSD-style license (found in the
6 * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7 * in the COPYING file in the root directory of this source tree).
9 * You can contact the author at:
10 * - zstdmt source repository: https://github.com/mcmilk/zstdmt
13 #ifndef THREADING_H_938743
14 #define THREADING_H_938743
18 #if defined (__cplusplus)
22 #if defined(ZSTD_MULTITHREAD) && defined(_WIN32)
25 * Windows minimalist Pthread Wrapper, based on :
26 * http://www.cse.wustl.edu/~schmidt/win32-cv-1.html
36 #define _WIN32_WINNT 0x0600
38 #ifndef WIN32_LEAN_AND_MEAN
39 # define WIN32_LEAN_AND_MEAN
42 #undef ERROR /* reported already defined on VS 2015 (Rich Geldreich) */
45 #define ERROR(name) ZSTD_ERROR(name)
49 #define ZSTD_pthread_mutex_t CRITICAL_SECTION
50 #define ZSTD_pthread_mutex_init(a, b) ((void)(b), InitializeCriticalSection((a)), 0)
51 #define ZSTD_pthread_mutex_destroy(a) DeleteCriticalSection((a))
52 #define ZSTD_pthread_mutex_lock(a) EnterCriticalSection((a))
53 #define ZSTD_pthread_mutex_unlock(a) LeaveCriticalSection((a))
55 /* condition variable */
56 #define ZSTD_pthread_cond_t CONDITION_VARIABLE
57 #define ZSTD_pthread_cond_init(a, b) ((void)(b), InitializeConditionVariable((a)), 0)
58 #define ZSTD_pthread_cond_destroy(a) ((void)(a))
59 #define ZSTD_pthread_cond_wait(a, b) SleepConditionVariableCS((a), (b), INFINITE)
60 #define ZSTD_pthread_cond_signal(a) WakeConditionVariable((a))
61 #define ZSTD_pthread_cond_broadcast(a) WakeAllConditionVariable((a))
63 /* ZSTD_pthread_create() and ZSTD_pthread_join() */
66 void* (*start_routine)(void*);
70 int ZSTD_pthread_create(ZSTD_pthread_t* thread, const void* unused,
71 void* (*start_routine) (void*), void* arg);
73 int ZSTD_pthread_join(ZSTD_pthread_t thread, void** value_ptr);
76 * add here more wrappers as required
80 #elif defined(ZSTD_MULTITHREAD) /* posix assumed ; need a better detection method */
81 /* === POSIX Systems === */
86 #define ZSTD_pthread_mutex_t pthread_mutex_t
87 #define ZSTD_pthread_mutex_init(a, b) pthread_mutex_init((a), (b))
88 #define ZSTD_pthread_mutex_destroy(a) pthread_mutex_destroy((a))
89 #define ZSTD_pthread_mutex_lock(a) pthread_mutex_lock((a))
90 #define ZSTD_pthread_mutex_unlock(a) pthread_mutex_unlock((a))
92 #define ZSTD_pthread_cond_t pthread_cond_t
93 #define ZSTD_pthread_cond_init(a, b) pthread_cond_init((a), (b))
94 #define ZSTD_pthread_cond_destroy(a) pthread_cond_destroy((a))
95 #define ZSTD_pthread_cond_wait(a, b) pthread_cond_wait((a), (b))
96 #define ZSTD_pthread_cond_signal(a) pthread_cond_signal((a))
97 #define ZSTD_pthread_cond_broadcast(a) pthread_cond_broadcast((a))
99 #define ZSTD_pthread_t pthread_t
100 #define ZSTD_pthread_create(a, b, c, d) pthread_create((a), (b), (c), (d))
101 #define ZSTD_pthread_join(a, b) pthread_join((a),(b))
103 #else /* DEBUGLEVEL >= 1 */
105 /* Debug implementation of threading.
106 * In this implementation we use pointers for mutexes and condition variables.
107 * This way, if we forget to init/destroy them the program will crash or ASAN
111 #define ZSTD_pthread_mutex_t pthread_mutex_t*
112 int ZSTD_pthread_mutex_init(ZSTD_pthread_mutex_t* mutex, pthread_mutexattr_t const* attr);
113 int ZSTD_pthread_mutex_destroy(ZSTD_pthread_mutex_t* mutex);
114 #define ZSTD_pthread_mutex_lock(a) pthread_mutex_lock(*(a))
115 #define ZSTD_pthread_mutex_unlock(a) pthread_mutex_unlock(*(a))
117 #define ZSTD_pthread_cond_t pthread_cond_t*
118 int ZSTD_pthread_cond_init(ZSTD_pthread_cond_t* cond, pthread_condattr_t const* attr);
119 int ZSTD_pthread_cond_destroy(ZSTD_pthread_cond_t* cond);
120 #define ZSTD_pthread_cond_wait(a, b) pthread_cond_wait(*(a), *(b))
121 #define ZSTD_pthread_cond_signal(a) pthread_cond_signal(*(a))
122 #define ZSTD_pthread_cond_broadcast(a) pthread_cond_broadcast(*(a))
124 #define ZSTD_pthread_t pthread_t
125 #define ZSTD_pthread_create(a, b, c, d) pthread_create((a), (b), (c), (d))
126 #define ZSTD_pthread_join(a, b) pthread_join((a),(b))
130 #else /* ZSTD_MULTITHREAD not defined */
131 /* No multithreading support */
133 typedef int ZSTD_pthread_mutex_t;
134 #define ZSTD_pthread_mutex_init(a, b) ((void)(a), (void)(b), 0)
135 #define ZSTD_pthread_mutex_destroy(a) ((void)(a))
136 #define ZSTD_pthread_mutex_lock(a) ((void)(a))
137 #define ZSTD_pthread_mutex_unlock(a) ((void)(a))
139 typedef int ZSTD_pthread_cond_t;
140 #define ZSTD_pthread_cond_init(a, b) ((void)(a), (void)(b), 0)
141 #define ZSTD_pthread_cond_destroy(a) ((void)(a))
142 #define ZSTD_pthread_cond_wait(a, b) ((void)(a), (void)(b))
143 #define ZSTD_pthread_cond_signal(a) ((void)(a))
144 #define ZSTD_pthread_cond_broadcast(a) ((void)(a))
146 /* do not use ZSTD_pthread_t */
148 #endif /* ZSTD_MULTITHREAD */
150 #if defined (__cplusplus)
154 #endif /* THREADING_H_938743 */