Archipelago » qemu

common-obj-$(CONFIG_POSIX) += qemu-thread-posix.o compatfd.o

common-obj-$(CONFIG_WIN32) += qemu-thread-win32.o

/*

 * 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 "qemu-thread.h"

static void error_exit(int err, const char *msg)

{

    fprintf(stderr, "qemu: %s: %s\n", msg, strerror(err));

    exit(1);

}

void qemu_mutex_init(QemuMutex *mutex)

{

    int err;

    err = pthread_mutex_init(&mutex->lock, NULL);

    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);

}

static void timespec_add_ms(struct timespec *ts, uint64_t msecs)

{

    ts->tv_sec = ts->tv_sec + (long)(msecs / 1000);

    ts->tv_nsec = (ts->tv_nsec + ((long)msecs % 1000) * 1000000);

    if (ts->tv_nsec >= 1000000000) {

        ts->tv_nsec -= 1000000000;

        ts->tv_sec++;

    }

}

int qemu_mutex_timedlock(QemuMutex *mutex, uint64_t msecs)

{

    int err;

    struct timespec ts;

    clock_gettime(CLOCK_REALTIME, &ts);

    timespec_add_ms(&ts, msecs);

    err = pthread_mutex_timedlock(&mutex->lock, &ts);

    if (err && err != ETIMEDOUT)

        error_exit(err, __func__);

    return err;

}

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__);

}

int qemu_cond_timedwait(QemuCond *cond, QemuMutex *mutex, uint64_t msecs)

{

    struct timespec ts;

    int err;

    clock_gettime(CLOCK_REALTIME, &ts);

    timespec_add_ms(&ts, msecs);

    err = pthread_cond_timedwait(&cond->cond, &mutex->lock, &ts);

    if (err && err != ETIMEDOUT)

        error_exit(err, __func__);

    return err;

}

void qemu_thread_create(QemuThread *thread,

                       void *(*start_routine)(void*),

                       void *arg)

{

    int err;

    /* Leave signal handling to the iothread.  */

    sigset_t set, oldset;

    sigfillset(&set);

    pthread_sigmask(SIG_SETMASK, &set, &oldset);

    err = pthread_create(&thread->thread, NULL, start_routine, arg);

    if (err)

        error_exit(err, __func__);

    pthread_sigmask(SIG_SETMASK, &oldset, NULL);

}

void qemu_thread_signal(QemuThread *thread, int sig)

{

    int err;

    err = pthread_kill(thread->thread, sig);

    if (err)

        error_exit(err, __func__);

}

void qemu_thread_get_self(QemuThread *thread)

{

    thread->thread = pthread_self();

}

int qemu_thread_is_self(QemuThread *thread)

{

   return pthread_equal(pthread_self(), thread->thread);

}

void qemu_thread_exit(void *retval)

{

    pthread_exit(retval);

}

#ifndef __QEMU_THREAD_POSIX_H

#define __QEMU_THREAD_POSIX_H 1

#include "pthread.h"

struct QemuMutex {

    pthread_mutex_t lock;

};

struct QemuCond {

    pthread_cond_t cond;

};

struct QemuThread {

    pthread_t thread;

};

void qemu_thread_signal(QemuThread *thread, int sig);

#endif

/*

 * Win32 implementation for mutex/cond/thread functions

 *

 * Copyright Red Hat, Inc. 2010

 *

 * Author:

 *  Paolo Bonzini <pbonzini@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 "qemu-common.h"

#include "qemu-thread.h"

#include <process.h>

#include <assert.h>

#include <limits.h>

static void error_exit(int err, const char *msg)

{

    char *pstr;

    FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,

                  NULL, err, 0, (LPTSTR)&pstr, 2, NULL);

    fprintf(stderr, "qemu: %s: %s\n", msg, pstr);

    LocalFree(pstr);

    exit(1);

}

void qemu_mutex_init(QemuMutex *mutex)

{

    mutex->owner = 0;

    InitializeCriticalSection(&mutex->lock);

}

void qemu_mutex_lock(QemuMutex *mutex)

{

    EnterCriticalSection(&mutex->lock);

    /* Win32 CRITICAL_SECTIONs are recursive.  Assert that we're not

     * using them as such.

     */

    assert(mutex->owner == 0);

    mutex->owner = GetCurrentThreadId();

}

int qemu_mutex_trylock(QemuMutex *mutex)

{

    int owned;

    owned = TryEnterCriticalSection(&mutex->lock);

    if (owned) {

        assert(mutex->owner == 0);

        mutex->owner = GetCurrentThreadId();

    }

    return !owned;

}

void qemu_mutex_unlock(QemuMutex *mutex)

{

    assert(mutex->owner == GetCurrentThreadId());

    mutex->owner = 0;

    LeaveCriticalSection(&mutex->lock);

}

void qemu_cond_init(QemuCond *cond)

{

    memset(cond, 0, sizeof(*cond));

    cond->sema = CreateSemaphore(NULL, 0, LONG_MAX, NULL);

    if (!cond->sema) {

        error_exit(GetLastError(), __func__);

    }

    cond->continue_event = CreateEvent(NULL,    /* security */

                                       FALSE,   /* auto-reset */

                                       FALSE,   /* not signaled */

                                       NULL);   /* name */

    if (!cond->continue_event) {

        error_exit(GetLastError(), __func__);

    }

}

void qemu_cond_signal(QemuCond *cond)

{

    DWORD result;

    /*

     * Signal only when there are waiters.  cond->waiters is

     * incremented by pthread_cond_wait under the external lock,

     * so we are safe about that.

     */

    if (cond->waiters == 0) {

        return;

    }

    /*

     * Waiting threads decrement it outside the external lock, but

     * only if another thread is executing pthread_cond_broadcast and

     * has the mutex.  So, it also cannot be decremented concurrently

     * with this particular access.

     */

    cond->target = cond->waiters - 1;

    result = SignalObjectAndWait(cond->sema, cond->continue_event,

                                 INFINITE, FALSE);

    if (result == WAIT_ABANDONED || result == WAIT_FAILED) {

        error_exit(GetLastError(), __func__);

    }

}

void qemu_cond_broadcast(QemuCond *cond)

{

    BOOLEAN result;

    /*

     * As in pthread_cond_signal, access to cond->waiters and

     * cond->target is locked via the external mutex.

     */

    if (cond->waiters == 0) {

        return;

    }

    cond->target = 0;

    result = ReleaseSemaphore(cond->sema, cond->waiters, NULL);

    if (!result) {

        error_exit(GetLastError(), __func__);

    }

    /*

     * At this point all waiters continue. Each one takes its

     * slice of the semaphore. Now it's our turn to wait: Since

     * the external mutex is held, no thread can leave cond_wait,

     * yet. For this reason, we can be sure that no thread gets

     * a chance to eat *more* than one slice. OTOH, it means

     * that the last waiter must send us a wake-up.

     */

    WaitForSingleObject(cond->continue_event, INFINITE);

}

void qemu_cond_wait(QemuCond *cond, QemuMutex *mutex)

{

    /*

     * This access is protected under the mutex.

     */

    cond->waiters++;

    /*

     * Unlock external mutex and wait for signal.

     * NOTE: we've held mutex locked long enough to increment

     * waiters count above, so there's no problem with

     * leaving mutex unlocked before we wait on semaphore.

     */

    qemu_mutex_unlock(mutex);

    WaitForSingleObject(cond->sema, INFINITE);

    /* Now waiters must rendez-vous with the signaling thread and

     * let it continue.  For cond_broadcast this has heavy contention

     * and triggers thundering herd.  So goes life.

     *

     * Decrease waiters count.  The mutex is not taken, so we have

     * to do this atomically.

     *

     * All waiters contend for the mutex at the end of this function

     * until the signaling thread relinquishes it.  To ensure

     * each waiter consumes exactly one slice of the semaphore,

     * the signaling thread stops until it is told by the last

     * waiter that it can go on.

     */

    if (InterlockedDecrement(&cond->waiters) == cond->target) {

        SetEvent(cond->continue_event);

    }

    qemu_mutex_lock(mutex);

}

struct QemuThreadData {

    QemuThread *thread;

    void *(*start_routine)(void *);

    void *arg;

};

static int qemu_thread_tls_index = TLS_OUT_OF_INDEXES;

static unsigned __stdcall win32_start_routine(void *arg)

{

    struct QemuThreadData data = *(struct QemuThreadData *) arg;

    QemuThread *thread = data.thread;

    free(arg);

    TlsSetValue(qemu_thread_tls_index, thread);

    /*

     * Use DuplicateHandle instead of assigning thread->thread in the

     * creating thread to avoid races.  It's simpler this way than with

     * synchronization.

     */

    DuplicateHandle(GetCurrentProcess(), GetCurrentThread(),

                    GetCurrentProcess(), &thread->thread,

                    0, FALSE, DUPLICATE_SAME_ACCESS);

    qemu_thread_exit(data.start_routine(data.arg));

    abort();

}

void qemu_thread_exit(void *arg)

{

    QemuThread *thread = TlsGetValue(qemu_thread_tls_index);

    thread->ret = arg;

    CloseHandle(thread->thread);

    thread->thread = NULL;

    ExitThread(0);

}

static inline void qemu_thread_init(void)

{

    if (qemu_thread_tls_index == TLS_OUT_OF_INDEXES) {

        qemu_thread_tls_index = TlsAlloc();

        if (qemu_thread_tls_index == TLS_OUT_OF_INDEXES) {

            error_exit(ERROR_NO_SYSTEM_RESOURCES, __func__);

        }

    }

}

void qemu_thread_create(QemuThread *thread,

                       void *(*start_routine)(void *),

                       void *arg)

{

    HANDLE hThread;

    struct QemuThreadData *data;

    qemu_thread_init();

    data = qemu_malloc(sizeof *data);

    data->thread = thread;

    data->start_routine = start_routine;

    data->arg = arg;

    hThread = (HANDLE) _beginthreadex(NULL, 0, win32_start_routine,

                                      data, 0, NULL);

    if (!hThread) {

        error_exit(GetLastError(), __func__);

    }

    CloseHandle(hThread);

}

void qemu_thread_get_self(QemuThread *thread)

{

    if (!thread->thread) {

        /* In the main thread of the process.  Initialize the QemuThread

           pointer in TLS, and use the dummy GetCurrentThread handle as

           the identifier for qemu_thread_is_self.  */

        qemu_thread_init();

        TlsSetValue(qemu_thread_tls_index, thread);

        thread->thread = GetCurrentThread();

    }

}

int qemu_thread_is_self(QemuThread *thread)

{

    QemuThread *this_thread = TlsGetValue(qemu_thread_tls_index);

    return this_thread->thread == thread->thread;

}

#ifndef __QEMU_THREAD_WIN32_H

#define __QEMU_THREAD_WIN32_H 1

#include "windows.h"

struct QemuMutex {

    CRITICAL_SECTION lock;

    LONG owner;

};

struct QemuCond {

    LONG waiters, target;

    HANDLE sema;

    HANDLE continue_event;

};

struct QemuThread {

    HANDLE thread;

    void *ret;

};

#endif

#ifdef _WIN32

#include "qemu-thread-win32.h"

#else

#include "qemu-thread-posix.h"

#endif

/*

 * IMPORTANT: The implementation does not guarantee that pthread_cond_signal

 * and pthread_cond_broadcast can be called except while the same mutex is

 * held as in the corresponding pthread_cond_wait calls!

 */