]>
Commit | Line | Data |
---|---|---|
e5d355d1 AL |
1 | /* |
2 | * Wrappers around mutex/cond/thread functions | |
3 | * | |
4 | * Copyright Red Hat, Inc. 2009 | |
5 | * | |
6 | * Author: | |
7 | * Marcelo Tosatti <mtosatti@redhat.com> | |
8 | * | |
9 | * This work is licensed under the terms of the GNU GPL, version 2 or later. | |
10 | * See the COPYING file in the top-level directory. | |
11 | * | |
12 | */ | |
13 | #include <stdlib.h> | |
14 | #include <stdio.h> | |
15 | #include <errno.h> | |
16 | #include <time.h> | |
17 | #include <signal.h> | |
18 | #include <stdint.h> | |
19 | #include <string.h> | |
38b14db3 PB |
20 | #include <limits.h> |
21 | #include <unistd.h> | |
22 | #include <sys/time.h> | |
e5d355d1 AL |
23 | #include "qemu-thread.h" |
24 | ||
25 | static void error_exit(int err, const char *msg) | |
26 | { | |
27 | fprintf(stderr, "qemu: %s: %s\n", msg, strerror(err)); | |
53380ac3 | 28 | abort(); |
e5d355d1 AL |
29 | } |
30 | ||
31 | void qemu_mutex_init(QemuMutex *mutex) | |
32 | { | |
33 | int err; | |
89b48b56 | 34 | pthread_mutexattr_t mutexattr; |
e5d355d1 | 35 | |
89b48b56 PB |
36 | pthread_mutexattr_init(&mutexattr); |
37 | pthread_mutexattr_settype(&mutexattr, PTHREAD_MUTEX_ERRORCHECK); | |
38 | err = pthread_mutex_init(&mutex->lock, &mutexattr); | |
39 | pthread_mutexattr_destroy(&mutexattr); | |
e5d355d1 AL |
40 | if (err) |
41 | error_exit(err, __func__); | |
42 | } | |
43 | ||
313b1d69 CC |
44 | void qemu_mutex_destroy(QemuMutex *mutex) |
45 | { | |
46 | int err; | |
47 | ||
48 | err = pthread_mutex_destroy(&mutex->lock); | |
49 | if (err) | |
50 | error_exit(err, __func__); | |
51 | } | |
52 | ||
e5d355d1 AL |
53 | void qemu_mutex_lock(QemuMutex *mutex) |
54 | { | |
55 | int err; | |
56 | ||
57 | err = pthread_mutex_lock(&mutex->lock); | |
58 | if (err) | |
59 | error_exit(err, __func__); | |
60 | } | |
61 | ||
62 | int qemu_mutex_trylock(QemuMutex *mutex) | |
63 | { | |
64 | return pthread_mutex_trylock(&mutex->lock); | |
65 | } | |
66 | ||
e5d355d1 AL |
67 | void qemu_mutex_unlock(QemuMutex *mutex) |
68 | { | |
69 | int err; | |
70 | ||
71 | err = pthread_mutex_unlock(&mutex->lock); | |
72 | if (err) | |
73 | error_exit(err, __func__); | |
74 | } | |
75 | ||
76 | void qemu_cond_init(QemuCond *cond) | |
77 | { | |
78 | int err; | |
79 | ||
80 | err = pthread_cond_init(&cond->cond, NULL); | |
81 | if (err) | |
82 | error_exit(err, __func__); | |
83 | } | |
84 | ||
313b1d69 CC |
85 | void qemu_cond_destroy(QemuCond *cond) |
86 | { | |
87 | int err; | |
88 | ||
89 | err = pthread_cond_destroy(&cond->cond); | |
90 | if (err) | |
91 | error_exit(err, __func__); | |
92 | } | |
93 | ||
e5d355d1 AL |
94 | void qemu_cond_signal(QemuCond *cond) |
95 | { | |
96 | int err; | |
97 | ||
98 | err = pthread_cond_signal(&cond->cond); | |
99 | if (err) | |
100 | error_exit(err, __func__); | |
101 | } | |
102 | ||
103 | void qemu_cond_broadcast(QemuCond *cond) | |
104 | { | |
105 | int err; | |
106 | ||
107 | err = pthread_cond_broadcast(&cond->cond); | |
108 | if (err) | |
109 | error_exit(err, __func__); | |
110 | } | |
111 | ||
112 | void qemu_cond_wait(QemuCond *cond, QemuMutex *mutex) | |
113 | { | |
114 | int err; | |
115 | ||
116 | err = pthread_cond_wait(&cond->cond, &mutex->lock); | |
117 | if (err) | |
118 | error_exit(err, __func__); | |
119 | } | |
120 | ||
38b14db3 PB |
121 | void qemu_sem_init(QemuSemaphore *sem, int init) |
122 | { | |
123 | int rc; | |
124 | ||
c166cb72 PB |
125 | #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__) |
126 | rc = pthread_mutex_init(&sem->lock, NULL); | |
127 | if (rc != 0) { | |
128 | error_exit(rc, __func__); | |
129 | } | |
130 | rc = pthread_cond_init(&sem->cond, NULL); | |
131 | if (rc != 0) { | |
132 | error_exit(rc, __func__); | |
133 | } | |
134 | if (init < 0) { | |
135 | error_exit(EINVAL, __func__); | |
136 | } | |
137 | sem->count = init; | |
138 | #else | |
38b14db3 PB |
139 | rc = sem_init(&sem->sem, 0, init); |
140 | if (rc < 0) { | |
141 | error_exit(errno, __func__); | |
142 | } | |
c166cb72 | 143 | #endif |
38b14db3 PB |
144 | } |
145 | ||
146 | void qemu_sem_destroy(QemuSemaphore *sem) | |
147 | { | |
148 | int rc; | |
149 | ||
c166cb72 PB |
150 | #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__) |
151 | rc = pthread_cond_destroy(&sem->cond); | |
152 | if (rc < 0) { | |
153 | error_exit(rc, __func__); | |
154 | } | |
155 | rc = pthread_mutex_destroy(&sem->lock); | |
156 | if (rc < 0) { | |
157 | error_exit(rc, __func__); | |
158 | } | |
159 | #else | |
38b14db3 PB |
160 | rc = sem_destroy(&sem->sem); |
161 | if (rc < 0) { | |
162 | error_exit(errno, __func__); | |
163 | } | |
c166cb72 | 164 | #endif |
38b14db3 PB |
165 | } |
166 | ||
167 | void qemu_sem_post(QemuSemaphore *sem) | |
168 | { | |
169 | int rc; | |
170 | ||
c166cb72 PB |
171 | #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__) |
172 | pthread_mutex_lock(&sem->lock); | |
173 | if (sem->count == INT_MAX) { | |
174 | rc = EINVAL; | |
175 | } else if (sem->count++ < 0) { | |
176 | rc = pthread_cond_signal(&sem->cond); | |
177 | } else { | |
178 | rc = 0; | |
179 | } | |
180 | pthread_mutex_unlock(&sem->lock); | |
181 | if (rc != 0) { | |
182 | error_exit(rc, __func__); | |
183 | } | |
184 | #else | |
38b14db3 PB |
185 | rc = sem_post(&sem->sem); |
186 | if (rc < 0) { | |
187 | error_exit(errno, __func__); | |
188 | } | |
c166cb72 PB |
189 | #endif |
190 | } | |
191 | ||
192 | static void compute_abs_deadline(struct timespec *ts, int ms) | |
193 | { | |
194 | struct timeval tv; | |
195 | gettimeofday(&tv, NULL); | |
196 | ts->tv_nsec = tv.tv_usec * 1000 + (ms % 1000) * 1000000; | |
197 | ts->tv_sec = tv.tv_sec + ms / 1000; | |
198 | if (ts->tv_nsec >= 1000000000) { | |
199 | ts->tv_sec++; | |
200 | ts->tv_nsec -= 1000000000; | |
201 | } | |
38b14db3 PB |
202 | } |
203 | ||
204 | int qemu_sem_timedwait(QemuSemaphore *sem, int ms) | |
205 | { | |
206 | int rc; | |
c166cb72 PB |
207 | struct timespec ts; |
208 | ||
209 | #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__) | |
210 | compute_abs_deadline(&ts, ms); | |
211 | pthread_mutex_lock(&sem->lock); | |
212 | --sem->count; | |
213 | while (sem->count < 0) { | |
214 | rc = pthread_cond_timedwait(&sem->cond, &sem->lock, &ts); | |
215 | if (rc == ETIMEDOUT) { | |
216 | break; | |
217 | } | |
218 | if (rc != 0) { | |
219 | error_exit(rc, __func__); | |
220 | } | |
221 | } | |
222 | pthread_mutex_unlock(&sem->lock); | |
223 | return (rc == ETIMEDOUT ? -1 : 0); | |
224 | #else | |
38b14db3 PB |
225 | if (ms <= 0) { |
226 | /* This is cheaper than sem_timedwait. */ | |
227 | do { | |
228 | rc = sem_trywait(&sem->sem); | |
229 | } while (rc == -1 && errno == EINTR); | |
230 | if (rc == -1 && errno == EAGAIN) { | |
231 | return -1; | |
232 | } | |
233 | } else { | |
c166cb72 | 234 | compute_abs_deadline(&ts, ms); |
38b14db3 PB |
235 | do { |
236 | rc = sem_timedwait(&sem->sem, &ts); | |
237 | } while (rc == -1 && errno == EINTR); | |
238 | if (rc == -1 && errno == ETIMEDOUT) { | |
239 | return -1; | |
240 | } | |
241 | } | |
242 | if (rc < 0) { | |
243 | error_exit(errno, __func__); | |
244 | } | |
245 | return 0; | |
c166cb72 | 246 | #endif |
38b14db3 PB |
247 | } |
248 | ||
249 | void qemu_sem_wait(QemuSemaphore *sem) | |
250 | { | |
c166cb72 PB |
251 | #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__) |
252 | pthread_mutex_lock(&sem->lock); | |
253 | --sem->count; | |
254 | while (sem->count < 0) { | |
255 | pthread_cond_wait(&sem->cond, &sem->lock); | |
256 | } | |
257 | pthread_mutex_unlock(&sem->lock); | |
258 | #else | |
38b14db3 PB |
259 | int rc; |
260 | ||
261 | do { | |
262 | rc = sem_wait(&sem->sem); | |
263 | } while (rc == -1 && errno == EINTR); | |
264 | if (rc < 0) { | |
265 | error_exit(errno, __func__); | |
266 | } | |
c166cb72 | 267 | #endif |
38b14db3 PB |
268 | } |
269 | ||
e5d355d1 AL |
270 | void qemu_thread_create(QemuThread *thread, |
271 | void *(*start_routine)(void*), | |
cf218714 | 272 | void *arg, int mode) |
e5d355d1 | 273 | { |
cf218714 | 274 | sigset_t set, oldset; |
e5d355d1 | 275 | int err; |
8763046b | 276 | pthread_attr_t attr; |
e5d355d1 | 277 | |
8763046b JK |
278 | err = pthread_attr_init(&attr); |
279 | if (err) { | |
280 | error_exit(err, __func__); | |
281 | } | |
282 | if (mode == QEMU_THREAD_DETACHED) { | |
283 | err = pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | |
284 | if (err) { | |
285 | error_exit(err, __func__); | |
286 | } | |
287 | } | |
55541c8a | 288 | |
cf218714 | 289 | /* Leave signal handling to the iothread. */ |
55541c8a PB |
290 | sigfillset(&set); |
291 | pthread_sigmask(SIG_SETMASK, &set, &oldset); | |
8763046b | 292 | err = pthread_create(&thread->thread, &attr, start_routine, arg); |
e5d355d1 AL |
293 | if (err) |
294 | error_exit(err, __func__); | |
55541c8a PB |
295 | |
296 | pthread_sigmask(SIG_SETMASK, &oldset, NULL); | |
8763046b JK |
297 | |
298 | pthread_attr_destroy(&attr); | |
e5d355d1 AL |
299 | } |
300 | ||
b7680cb6 | 301 | void qemu_thread_get_self(QemuThread *thread) |
e5d355d1 AL |
302 | { |
303 | thread->thread = pthread_self(); | |
304 | } | |
305 | ||
2d797b65 | 306 | bool qemu_thread_is_self(QemuThread *thread) |
e5d355d1 | 307 | { |
b7680cb6 | 308 | return pthread_equal(pthread_self(), thread->thread); |
e5d355d1 AL |
309 | } |
310 | ||
313b1d69 CC |
311 | void qemu_thread_exit(void *retval) |
312 | { | |
313 | pthread_exit(retval); | |
314 | } | |
8763046b JK |
315 | |
316 | void *qemu_thread_join(QemuThread *thread) | |
317 | { | |
318 | int err; | |
319 | void *ret; | |
320 | ||
321 | err = pthread_join(thread->thread, &ret); | |
322 | if (err) { | |
323 | error_exit(err, __func__); | |
324 | } | |
325 | return ret; | |
326 | } |