fix a possible pthread crash
[xonotic/darkplaces.git] / thread_pthread.c
1 #include "quakedef.h"
2 #include "thread.h"
3 #ifdef THREADRECURSIVE
4 #define __USE_UNIX98
5 #include <pthread.h>
6 #endif
7 #include <stdint.h>
8
9
10 int Thread_Init(void)
11 {
12         return 0;
13 }
14
15 void Thread_Shutdown(void)
16 {
17 }
18
19 qboolean Thread_HasThreads(void)
20 {
21         return true;
22 }
23
24 void *_Thread_CreateMutex(const char *filename, int fileline)
25 {
26 #ifdef THREADRECURSIVE
27         pthread_mutexattr_t    attr;
28 #endif
29         pthread_mutex_t *mutexp = (pthread_mutex_t *) Z_Malloc(sizeof(pthread_mutex_t));
30 #ifdef THREADDEBUG
31         Sys_PrintfToTerminal("%p mutex create %s:%i\n" , mutexp, filename, fileline);
32 #endif
33 #ifdef THREADRECURSIVE
34         pthread_mutexattr_init(&attr);
35         pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
36         pthread_mutex_init(mutexp, &attr);
37 #else
38         pthread_mutex_init(mutexp, NULL);
39 #endif
40         return mutexp;
41 }
42
43 void _Thread_DestroyMutex(void *mutex, const char *filename, int fileline)
44 {
45         pthread_mutex_t *mutexp = (pthread_mutex_t *) mutex;
46 #ifdef THREADDEBUG
47         Sys_PrintfToTerminal("%p mutex destroy %s:%i\n", mutex, filename, fileline);
48 #endif
49         pthread_mutex_destroy(mutexp);
50         Z_Free(mutexp);
51 }
52
53 int _Thread_LockMutex(void *mutex, const char *filename, int fileline)
54 {
55         pthread_mutex_t *mutexp = (pthread_mutex_t *) mutex;
56 #ifdef THREADDEBUG
57         Sys_PrintfToTerminal("%p mutex lock %s:%i\n"   , mutex, filename, fileline);
58 #endif
59         return pthread_mutex_lock(mutexp);
60 }
61
62 int _Thread_UnlockMutex(void *mutex, const char *filename, int fileline)
63 {
64         pthread_mutex_t *mutexp = (pthread_mutex_t *) mutex;
65 #ifdef THREADDEBUG
66         Sys_PrintfToTerminal("%p mutex unlock %s:%i\n" , mutex, filename, fileline);
67 #endif
68         return pthread_mutex_unlock(mutexp);
69 }
70
71 void *_Thread_CreateCond(const char *filename, int fileline)
72 {
73         pthread_cond_t *condp = (pthread_cond_t *) Z_Malloc(sizeof(pthread_cond_t));
74         pthread_cond_init(condp, NULL);
75 #ifdef THREADDEBUG
76         Sys_PrintfToTerminal("%p cond create %s:%i\n"   , condp, filename, fileline);
77 #endif
78         return condp;
79 }
80
81 void _Thread_DestroyCond(void *cond, const char *filename, int fileline)
82 {
83         pthread_cond_t *condp = (pthread_cond_t *) cond;
84 #ifdef THREADDEBUG
85         Sys_PrintfToTerminal("%p cond destroy %s:%i\n"   , cond, filename, fileline);
86 #endif
87         pthread_cond_destroy(condp);
88         Z_Free(condp);
89 }
90
91 int _Thread_CondSignal(void *cond, const char *filename, int fileline)
92 {
93         pthread_cond_t *condp = (pthread_cond_t *) cond;
94 #ifdef THREADDEBUG
95         Sys_PrintfToTerminal("%p cond signal %s:%i\n"   , cond, filename, fileline);
96 #endif
97         return pthread_cond_signal(condp);
98 }
99
100 int _Thread_CondBroadcast(void *cond, const char *filename, int fileline)
101 {
102         pthread_cond_t *condp = (pthread_cond_t *) cond;
103 #ifdef THREADDEBUG
104         Sys_PrintfToTerminal("%p cond broadcast %s:%i\n"   , cond, filename, fileline);
105 #endif
106         return pthread_cond_broadcast(condp);
107 }
108
109 int _Thread_CondWait(void *cond, void *mutex, const char *filename, int fileline)
110 {
111         pthread_cond_t *condp = (pthread_cond_t *) cond;
112         pthread_mutex_t *mutexp = (pthread_mutex_t *) mutex;
113 #ifdef THREADDEBUG
114         Sys_PrintfToTerminal("%p cond wait %s:%i\n"   , cond, filename, fileline);
115 #endif
116         return pthread_cond_wait(condp, mutexp);
117 }
118
119 void *_Thread_CreateThread(int (*fn)(void *), void *data, const char *filename, int fileline)
120 {
121         pthread_t *threadp = (pthread_t *) Z_Malloc(sizeof(pthread_t));
122 #ifdef THREADDEBUG
123         Sys_PrintfToTerminal("%p thread create %s:%i\n"   , threadp, filename, fileline);
124 #endif
125         int r = pthread_create(threadp, NULL, (void * (*) (void *)) fn, data);
126         if(r)
127         {
128                 Z_Free(threadp);
129                 return NULL;
130         }
131         return threadp;
132 }
133
134 int _Thread_WaitThread(void *thread, int retval, const char *filename, int fileline)
135 {
136         pthread_t *threadp = (pthread_t *) thread;
137         void *status = (void *) (intptr_t) retval;
138 #ifdef THREADDEBUG
139         Sys_PrintfToTerminal("%p thread wait %s:%i\n"   , thread, filename, fileline);
140 #endif
141         pthread_join(*threadp, &status);
142         Z_Free(threadp);
143         return (int) (intptr_t) status;
144 }
145
146 #ifdef PTHREAD_BARRIER_SERIAL_THREAD
147 void *_Thread_CreateBarrier(unsigned int count, const char *filename, int fileline)
148 {
149         pthread_barrier_t *b = (pthread_barrier_t *) Z_Malloc(sizeof(pthread_barrier_t));
150 #ifdef THREADDEBUG
151         Sys_PrintfToTerminal("%p barrier create(%d) %s:%i\n", b, count, filename, fileline);
152 #endif
153         pthread_barrier_init(b, NULL, count);
154         return (void *) b;
155 }
156
157 void _Thread_DestroyBarrier(void *barrier, const char *filename, int fileline)
158 {
159         pthread_barrier_t *b = (pthread_barrier_t *) barrier;
160 #ifdef THREADDEBUG
161         Sys_PrintfToTerminal("%p barrier destroy %s:%i\n", b, filename, fileline);
162 #endif
163         pthread_barrier_destroy(b);
164 }
165
166 void _Thread_WaitBarrier(void *barrier, const char *filename, int fileline)
167 {
168         pthread_barrier_t *b = (pthread_barrier_t *) barrier;
169 #ifdef THREADDEBUG
170         Sys_PrintfToTerminal("%p barrier wait %s:%i\n", b, filename, fileline);
171 #endif
172         pthread_barrier_wait(b);
173 }
174 #else
175 // standard barrier implementation using conds and mutexes
176 // see: http://www.howforge.com/implementing-barrier-in-pthreads
177 typedef struct {
178         unsigned int needed;
179         unsigned int called;
180         void *mutex;
181         void *cond;
182 } barrier_t;
183
184 void *_Thread_CreateBarrier(unsigned int count, const char *filename, int fileline)
185 {
186         volatile barrier_t *b = (volatile barrier_t *) Z_Malloc(sizeof(barrier_t));
187 #ifdef THREADDEBUG
188         Sys_PrintfToTerminal("%p barrier create(%d) %s:%i\n", b, count, filename, fileline);
189 #endif
190         b->needed = count;
191         b->called = 0;
192         b->mutex = Thread_CreateMutex();
193         b->cond = Thread_CreateCond();
194         return (void *) b;
195 }
196
197 void _Thread_DestroyBarrier(void *barrier, const char *filename, int fileline)
198 {
199         volatile barrier_t *b = (volatile barrier_t *) barrier;
200 #ifdef THREADDEBUG
201         Sys_PrintfToTerminal("%p barrier destroy %s:%i\n", b, filename, fileline);
202 #endif
203         Thread_DestroyMutex(b->mutex);
204         Thread_DestroyCond(b->cond);
205 }
206
207 void _Thread_WaitBarrier(void *barrier, const char *filename, int fileline)
208 {
209         volatile barrier_t *b = (volatile barrier_t *) barrier;
210 #ifdef THREADDEBUG
211         Sys_PrintfToTerminal("%p barrier wait %s:%i\n", b, filename, fileline);
212 #endif
213         Thread_LockMutex(b->mutex);
214         b->called++;
215         if (b->called == b->needed) {
216                 b->called = 0;
217                 Thread_CondBroadcast(b->cond);
218         } else {
219                 do {
220                         Thread_CondWait(b->cond, b->mutex);
221                 } while(b->called);
222         }
223         Thread_UnlockMutex(b->mutex);
224 }
225 #endif