327 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			327 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
| /*
 | |
|  * Wrappers around mutex/cond/thread functions
 | |
|  *
 | |
|  * Copyright Red Hat, Inc. 2009
 | |
|  *
 | |
|  * Author:
 | |
|  *  Marcelo Tosatti <mtosatti@redhat.com>
 | |
|  *
 | |
|  * This work is licensed under the terms of the GNU GPL, version 2 or later.
 | |
|  * See the COPYING file in the top-level directory.
 | |
|  *
 | |
|  */
 | |
| #include <stdlib.h>
 | |
| #include <stdio.h>
 | |
| #include <errno.h>
 | |
| #include <time.h>
 | |
| #include <signal.h>
 | |
| #include <stdint.h>
 | |
| #include <string.h>
 | |
| #include <limits.h>
 | |
| #include <unistd.h>
 | |
| #include <sys/time.h>
 | |
| #include "qemu-thread.h"
 | |
| 
 | |
| static void error_exit(int err, const char *msg)
 | |
| {
 | |
|     fprintf(stderr, "qemu: %s: %s\n", msg, strerror(err));
 | |
|     abort();
 | |
| }
 | |
| 
 | |
| void qemu_mutex_init(QemuMutex *mutex)
 | |
| {
 | |
|     int err;
 | |
|     pthread_mutexattr_t mutexattr;
 | |
| 
 | |
|     pthread_mutexattr_init(&mutexattr);
 | |
|     pthread_mutexattr_settype(&mutexattr, PTHREAD_MUTEX_ERRORCHECK);
 | |
|     err = pthread_mutex_init(&mutex->lock, &mutexattr);
 | |
|     pthread_mutexattr_destroy(&mutexattr);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_mutex_destroy(QemuMutex *mutex)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_mutex_destroy(&mutex->lock);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_mutex_lock(QemuMutex *mutex)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_mutex_lock(&mutex->lock);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| int qemu_mutex_trylock(QemuMutex *mutex)
 | |
| {
 | |
|     return pthread_mutex_trylock(&mutex->lock);
 | |
| }
 | |
| 
 | |
| void qemu_mutex_unlock(QemuMutex *mutex)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_mutex_unlock(&mutex->lock);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_cond_init(QemuCond *cond)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_cond_init(&cond->cond, NULL);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_cond_destroy(QemuCond *cond)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_cond_destroy(&cond->cond);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_cond_signal(QemuCond *cond)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_cond_signal(&cond->cond);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_cond_broadcast(QemuCond *cond)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_cond_broadcast(&cond->cond);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_cond_wait(QemuCond *cond, QemuMutex *mutex)
 | |
| {
 | |
|     int err;
 | |
| 
 | |
|     err = pthread_cond_wait(&cond->cond, &mutex->lock);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| }
 | |
| 
 | |
| void qemu_sem_init(QemuSemaphore *sem, int init)
 | |
| {
 | |
|     int rc;
 | |
| 
 | |
| #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__)
 | |
|     rc = pthread_mutex_init(&sem->lock, NULL);
 | |
|     if (rc != 0) {
 | |
|         error_exit(rc, __func__);
 | |
|     }
 | |
|     rc = pthread_cond_init(&sem->cond, NULL);
 | |
|     if (rc != 0) {
 | |
|         error_exit(rc, __func__);
 | |
|     }
 | |
|     if (init < 0) {
 | |
|         error_exit(EINVAL, __func__);
 | |
|     }
 | |
|     sem->count = init;
 | |
| #else
 | |
|     rc = sem_init(&sem->sem, 0, init);
 | |
|     if (rc < 0) {
 | |
|         error_exit(errno, __func__);
 | |
|     }
 | |
| #endif
 | |
| }
 | |
| 
 | |
| void qemu_sem_destroy(QemuSemaphore *sem)
 | |
| {
 | |
|     int rc;
 | |
| 
 | |
| #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__)
 | |
|     rc = pthread_cond_destroy(&sem->cond);
 | |
|     if (rc < 0) {
 | |
|         error_exit(rc, __func__);
 | |
|     }
 | |
|     rc = pthread_mutex_destroy(&sem->lock);
 | |
|     if (rc < 0) {
 | |
|         error_exit(rc, __func__);
 | |
|     }
 | |
| #else
 | |
|     rc = sem_destroy(&sem->sem);
 | |
|     if (rc < 0) {
 | |
|         error_exit(errno, __func__);
 | |
|     }
 | |
| #endif
 | |
| }
 | |
| 
 | |
| void qemu_sem_post(QemuSemaphore *sem)
 | |
| {
 | |
|     int rc;
 | |
| 
 | |
| #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__)
 | |
|     pthread_mutex_lock(&sem->lock);
 | |
|     if (sem->count == INT_MAX) {
 | |
|         rc = EINVAL;
 | |
|     } else if (sem->count++ < 0) {
 | |
|         rc = pthread_cond_signal(&sem->cond);
 | |
|     } else {
 | |
|         rc = 0;
 | |
|     }
 | |
|     pthread_mutex_unlock(&sem->lock);
 | |
|     if (rc != 0) {
 | |
|         error_exit(rc, __func__);
 | |
|     }
 | |
| #else
 | |
|     rc = sem_post(&sem->sem);
 | |
|     if (rc < 0) {
 | |
|         error_exit(errno, __func__);
 | |
|     }
 | |
| #endif
 | |
| }
 | |
| 
 | |
| static void compute_abs_deadline(struct timespec *ts, int ms)
 | |
| {
 | |
|     struct timeval tv;
 | |
|     gettimeofday(&tv, NULL);
 | |
|     ts->tv_nsec = tv.tv_usec * 1000 + (ms % 1000) * 1000000;
 | |
|     ts->tv_sec = tv.tv_sec + ms / 1000;
 | |
|     if (ts->tv_nsec >= 1000000000) {
 | |
|         ts->tv_sec++;
 | |
|         ts->tv_nsec -= 1000000000;
 | |
|     }
 | |
| }
 | |
| 
 | |
| int qemu_sem_timedwait(QemuSemaphore *sem, int ms)
 | |
| {
 | |
|     int rc;
 | |
|     struct timespec ts;
 | |
| 
 | |
| #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__)
 | |
|     compute_abs_deadline(&ts, ms);
 | |
|     pthread_mutex_lock(&sem->lock);
 | |
|     --sem->count;
 | |
|     while (sem->count < 0) {
 | |
|         rc = pthread_cond_timedwait(&sem->cond, &sem->lock, &ts);
 | |
|         if (rc == ETIMEDOUT) {
 | |
|             break;
 | |
|         }
 | |
|         if (rc != 0) {
 | |
|             error_exit(rc, __func__);
 | |
|         }
 | |
|     }
 | |
|     pthread_mutex_unlock(&sem->lock);
 | |
|     return (rc == ETIMEDOUT ? -1 : 0);
 | |
| #else
 | |
|     if (ms <= 0) {
 | |
|         /* This is cheaper than sem_timedwait.  */
 | |
|         do {
 | |
|             rc = sem_trywait(&sem->sem);
 | |
|         } while (rc == -1 && errno == EINTR);
 | |
|         if (rc == -1 && errno == EAGAIN) {
 | |
|             return -1;
 | |
|         }
 | |
|     } else {
 | |
|         compute_abs_deadline(&ts, ms);
 | |
|         do {
 | |
|             rc = sem_timedwait(&sem->sem, &ts);
 | |
|         } while (rc == -1 && errno == EINTR);
 | |
|         if (rc == -1 && errno == ETIMEDOUT) {
 | |
|             return -1;
 | |
|         }
 | |
|     }
 | |
|     if (rc < 0) {
 | |
|         error_exit(errno, __func__);
 | |
|     }
 | |
|     return 0;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| void qemu_sem_wait(QemuSemaphore *sem)
 | |
| {
 | |
| #if defined(__OpenBSD__) || defined(__APPLE__) || defined(__NetBSD__)
 | |
|     pthread_mutex_lock(&sem->lock);
 | |
|     --sem->count;
 | |
|     while (sem->count < 0) {
 | |
|         pthread_cond_wait(&sem->cond, &sem->lock);
 | |
|     }
 | |
|     pthread_mutex_unlock(&sem->lock);
 | |
| #else
 | |
|     int rc;
 | |
| 
 | |
|     do {
 | |
|         rc = sem_wait(&sem->sem);
 | |
|     } while (rc == -1 && errno == EINTR);
 | |
|     if (rc < 0) {
 | |
|         error_exit(errno, __func__);
 | |
|     }
 | |
| #endif
 | |
| }
 | |
| 
 | |
| void qemu_thread_create(QemuThread *thread,
 | |
|                        void *(*start_routine)(void*),
 | |
|                        void *arg, int mode)
 | |
| {
 | |
|     sigset_t set, oldset;
 | |
|     int err;
 | |
|     pthread_attr_t attr;
 | |
| 
 | |
|     err = pthread_attr_init(&attr);
 | |
|     if (err) {
 | |
|         error_exit(err, __func__);
 | |
|     }
 | |
|     if (mode == QEMU_THREAD_DETACHED) {
 | |
|         err = pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
 | |
|         if (err) {
 | |
|             error_exit(err, __func__);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     /* Leave signal handling to the iothread.  */
 | |
|     sigfillset(&set);
 | |
|     pthread_sigmask(SIG_SETMASK, &set, &oldset);
 | |
|     err = pthread_create(&thread->thread, &attr, start_routine, arg);
 | |
|     if (err)
 | |
|         error_exit(err, __func__);
 | |
| 
 | |
|     pthread_sigmask(SIG_SETMASK, &oldset, NULL);
 | |
| 
 | |
|     pthread_attr_destroy(&attr);
 | |
| }
 | |
| 
 | |
| void qemu_thread_get_self(QemuThread *thread)
 | |
| {
 | |
|     thread->thread = pthread_self();
 | |
| }
 | |
| 
 | |
| bool qemu_thread_is_self(QemuThread *thread)
 | |
| {
 | |
|    return pthread_equal(pthread_self(), thread->thread);
 | |
| }
 | |
| 
 | |
| void qemu_thread_exit(void *retval)
 | |
| {
 | |
|     pthread_exit(retval);
 | |
| }
 | |
| 
 | |
| void *qemu_thread_join(QemuThread *thread)
 | |
| {
 | |
|     int err;
 | |
|     void *ret;
 | |
| 
 | |
|     err = pthread_join(thread->thread, &ret);
 | |
|     if (err) {
 | |
|         error_exit(err, __func__);
 | |
|     }
 | |
|     return ret;
 | |
| }
 |