595 lines
13 KiB
C
595 lines
13 KiB
C
/* -*- mode: c; tab-width: 2; indent-tabs-mode: nil; -*-
|
|
Copyright (c) 2012 Marcus Geelnard
|
|
|
|
This software is provided 'as-is', without any express or implied
|
|
warranty. In no event will the authors be held liable for any damages
|
|
arising from the use of this software.
|
|
|
|
Permission is granted to anyone to use this software for any purpose,
|
|
including commercial applications, and to alter it and redistribute it
|
|
freely, subject to the following restrictions:
|
|
|
|
1. The origin of this software must not be misrepresented; you must not
|
|
claim that you wrote the original software. If you use this software
|
|
in a product, an acknowledgment in the product documentation would be
|
|
appreciated but is not required.
|
|
|
|
2. Altered source versions must be plainly marked as such, and must not be
|
|
misrepresented as being the original software.
|
|
|
|
3. This notice may not be removed or altered from any source
|
|
distribution.
|
|
*/
|
|
|
|
/* 2013-01-06 Camilla Löwy <elmindreda@glfw.org>
|
|
*
|
|
* Added casts from time_t to DWORD to avoid warnings on VC++.
|
|
* Fixed time retrieval on POSIX systems.
|
|
*/
|
|
|
|
#include "tinycthread.h"
|
|
#include <stdlib.h>
|
|
|
|
/* Platform specific includes */
|
|
#if defined(_TTHREAD_POSIX_)
|
|
#include <signal.h>
|
|
#include <sched.h>
|
|
#include <unistd.h>
|
|
#include <sys/time.h>
|
|
#include <errno.h>
|
|
#elif defined(_TTHREAD_WIN32_)
|
|
#include <process.h>
|
|
#include <sys/timeb.h>
|
|
#endif
|
|
|
|
/* Standard, good-to-have defines */
|
|
#ifndef NULL
|
|
#define NULL (void*)0
|
|
#endif
|
|
#ifndef TRUE
|
|
#define TRUE 1
|
|
#endif
|
|
#ifndef FALSE
|
|
#define FALSE 0
|
|
#endif
|
|
|
|
int mtx_init(mtx_t *mtx, int type)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
mtx->mAlreadyLocked = FALSE;
|
|
mtx->mRecursive = type & mtx_recursive;
|
|
InitializeCriticalSection(&mtx->mHandle);
|
|
return thrd_success;
|
|
#else
|
|
int ret;
|
|
pthread_mutexattr_t attr;
|
|
pthread_mutexattr_init(&attr);
|
|
if (type & mtx_recursive)
|
|
{
|
|
pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
|
|
}
|
|
ret = pthread_mutex_init(mtx, &attr);
|
|
pthread_mutexattr_destroy(&attr);
|
|
return ret == 0 ? thrd_success : thrd_error;
|
|
#endif
|
|
}
|
|
|
|
void mtx_destroy(mtx_t *mtx)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
DeleteCriticalSection(&mtx->mHandle);
|
|
#else
|
|
pthread_mutex_destroy(mtx);
|
|
#endif
|
|
}
|
|
|
|
int mtx_lock(mtx_t *mtx)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
EnterCriticalSection(&mtx->mHandle);
|
|
if (!mtx->mRecursive)
|
|
{
|
|
while(mtx->mAlreadyLocked) Sleep(1000); /* Simulate deadlock... */
|
|
mtx->mAlreadyLocked = TRUE;
|
|
}
|
|
return thrd_success;
|
|
#else
|
|
return pthread_mutex_lock(mtx) == 0 ? thrd_success : thrd_error;
|
|
#endif
|
|
}
|
|
|
|
int mtx_timedlock(mtx_t *mtx, const struct timespec *ts)
|
|
{
|
|
/* FIXME! */
|
|
(void)mtx;
|
|
(void)ts;
|
|
return thrd_error;
|
|
}
|
|
|
|
int mtx_trylock(mtx_t *mtx)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
int ret = TryEnterCriticalSection(&mtx->mHandle) ? thrd_success : thrd_busy;
|
|
if ((!mtx->mRecursive) && (ret == thrd_success) && mtx->mAlreadyLocked)
|
|
{
|
|
LeaveCriticalSection(&mtx->mHandle);
|
|
ret = thrd_busy;
|
|
}
|
|
return ret;
|
|
#else
|
|
return (pthread_mutex_trylock(mtx) == 0) ? thrd_success : thrd_busy;
|
|
#endif
|
|
}
|
|
|
|
int mtx_unlock(mtx_t *mtx)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
mtx->mAlreadyLocked = FALSE;
|
|
LeaveCriticalSection(&mtx->mHandle);
|
|
return thrd_success;
|
|
#else
|
|
return pthread_mutex_unlock(mtx) == 0 ? thrd_success : thrd_error;;
|
|
#endif
|
|
}
|
|
|
|
#if defined(_TTHREAD_WIN32_)
|
|
#define _CONDITION_EVENT_ONE 0
|
|
#define _CONDITION_EVENT_ALL 1
|
|
#endif
|
|
|
|
int cnd_init(cnd_t *cond)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
cond->mWaitersCount = 0;
|
|
|
|
/* Init critical section */
|
|
InitializeCriticalSection(&cond->mWaitersCountLock);
|
|
|
|
/* Init events */
|
|
cond->mEvents[_CONDITION_EVENT_ONE] = CreateEvent(NULL, FALSE, FALSE, NULL);
|
|
if (cond->mEvents[_CONDITION_EVENT_ONE] == NULL)
|
|
{
|
|
cond->mEvents[_CONDITION_EVENT_ALL] = NULL;
|
|
return thrd_error;
|
|
}
|
|
cond->mEvents[_CONDITION_EVENT_ALL] = CreateEvent(NULL, TRUE, FALSE, NULL);
|
|
if (cond->mEvents[_CONDITION_EVENT_ALL] == NULL)
|
|
{
|
|
CloseHandle(cond->mEvents[_CONDITION_EVENT_ONE]);
|
|
cond->mEvents[_CONDITION_EVENT_ONE] = NULL;
|
|
return thrd_error;
|
|
}
|
|
|
|
return thrd_success;
|
|
#else
|
|
return pthread_cond_init(cond, NULL) == 0 ? thrd_success : thrd_error;
|
|
#endif
|
|
}
|
|
|
|
void cnd_destroy(cnd_t *cond)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
if (cond->mEvents[_CONDITION_EVENT_ONE] != NULL)
|
|
{
|
|
CloseHandle(cond->mEvents[_CONDITION_EVENT_ONE]);
|
|
}
|
|
if (cond->mEvents[_CONDITION_EVENT_ALL] != NULL)
|
|
{
|
|
CloseHandle(cond->mEvents[_CONDITION_EVENT_ALL]);
|
|
}
|
|
DeleteCriticalSection(&cond->mWaitersCountLock);
|
|
#else
|
|
pthread_cond_destroy(cond);
|
|
#endif
|
|
}
|
|
|
|
int cnd_signal(cnd_t *cond)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
int haveWaiters;
|
|
|
|
/* Are there any waiters? */
|
|
EnterCriticalSection(&cond->mWaitersCountLock);
|
|
haveWaiters = (cond->mWaitersCount > 0);
|
|
LeaveCriticalSection(&cond->mWaitersCountLock);
|
|
|
|
/* If we have any waiting threads, send them a signal */
|
|
if(haveWaiters)
|
|
{
|
|
if (SetEvent(cond->mEvents[_CONDITION_EVENT_ONE]) == 0)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
}
|
|
|
|
return thrd_success;
|
|
#else
|
|
return pthread_cond_signal(cond) == 0 ? thrd_success : thrd_error;
|
|
#endif
|
|
}
|
|
|
|
int cnd_broadcast(cnd_t *cond)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
int haveWaiters;
|
|
|
|
/* Are there any waiters? */
|
|
EnterCriticalSection(&cond->mWaitersCountLock);
|
|
haveWaiters = (cond->mWaitersCount > 0);
|
|
LeaveCriticalSection(&cond->mWaitersCountLock);
|
|
|
|
/* If we have any waiting threads, send them a signal */
|
|
if(haveWaiters)
|
|
{
|
|
if (SetEvent(cond->mEvents[_CONDITION_EVENT_ALL]) == 0)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
}
|
|
|
|
return thrd_success;
|
|
#else
|
|
return pthread_cond_signal(cond) == 0 ? thrd_success : thrd_error;
|
|
#endif
|
|
}
|
|
|
|
#if defined(_TTHREAD_WIN32_)
|
|
static int _cnd_timedwait_win32(cnd_t *cond, mtx_t *mtx, DWORD timeout)
|
|
{
|
|
int result, lastWaiter;
|
|
|
|
/* Increment number of waiters */
|
|
EnterCriticalSection(&cond->mWaitersCountLock);
|
|
++ cond->mWaitersCount;
|
|
LeaveCriticalSection(&cond->mWaitersCountLock);
|
|
|
|
/* Release the mutex while waiting for the condition (will decrease
|
|
the number of waiters when done)... */
|
|
mtx_unlock(mtx);
|
|
|
|
/* Wait for either event to become signaled due to cnd_signal() or
|
|
cnd_broadcast() being called */
|
|
result = WaitForMultipleObjects(2, cond->mEvents, FALSE, timeout);
|
|
if (result == WAIT_TIMEOUT)
|
|
{
|
|
return thrd_timeout;
|
|
}
|
|
else if (result == (int)WAIT_FAILED)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
|
|
/* Check if we are the last waiter */
|
|
EnterCriticalSection(&cond->mWaitersCountLock);
|
|
-- cond->mWaitersCount;
|
|
lastWaiter = (result == (WAIT_OBJECT_0 + _CONDITION_EVENT_ALL)) &&
|
|
(cond->mWaitersCount == 0);
|
|
LeaveCriticalSection(&cond->mWaitersCountLock);
|
|
|
|
/* If we are the last waiter to be notified to stop waiting, reset the event */
|
|
if (lastWaiter)
|
|
{
|
|
if (ResetEvent(cond->mEvents[_CONDITION_EVENT_ALL]) == 0)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
}
|
|
|
|
/* Re-acquire the mutex */
|
|
mtx_lock(mtx);
|
|
|
|
return thrd_success;
|
|
}
|
|
#endif
|
|
|
|
int cnd_wait(cnd_t *cond, mtx_t *mtx)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
return _cnd_timedwait_win32(cond, mtx, INFINITE);
|
|
#else
|
|
return pthread_cond_wait(cond, mtx) == 0 ? thrd_success : thrd_error;
|
|
#endif
|
|
}
|
|
|
|
int cnd_timedwait(cnd_t *cond, mtx_t *mtx, const struct timespec *ts)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
struct timespec now;
|
|
if (clock_gettime(CLOCK_REALTIME, &now) == 0)
|
|
{
|
|
DWORD delta = (DWORD) ((ts->tv_sec - now.tv_sec) * 1000 +
|
|
(ts->tv_nsec - now.tv_nsec + 500000) / 1000000);
|
|
return _cnd_timedwait_win32(cond, mtx, delta);
|
|
}
|
|
else
|
|
return thrd_error;
|
|
#else
|
|
int ret;
|
|
ret = pthread_cond_timedwait(cond, mtx, ts);
|
|
if (ret == ETIMEDOUT)
|
|
{
|
|
return thrd_timeout;
|
|
}
|
|
return ret == 0 ? thrd_success : thrd_error;
|
|
#endif
|
|
}
|
|
|
|
|
|
/** Information to pass to the new thread (what to run). */
|
|
typedef struct {
|
|
thrd_start_t mFunction; /**< Pointer to the function to be executed. */
|
|
void * mArg; /**< Function argument for the thread function. */
|
|
} _thread_start_info;
|
|
|
|
/* Thread wrapper function. */
|
|
#if defined(_TTHREAD_WIN32_)
|
|
static unsigned WINAPI _thrd_wrapper_function(void * aArg)
|
|
#elif defined(_TTHREAD_POSIX_)
|
|
static void * _thrd_wrapper_function(void * aArg)
|
|
#endif
|
|
{
|
|
thrd_start_t fun;
|
|
void *arg;
|
|
int res;
|
|
#if defined(_TTHREAD_POSIX_)
|
|
void *pres;
|
|
#endif
|
|
|
|
/* Get thread startup information */
|
|
_thread_start_info *ti = (_thread_start_info *) aArg;
|
|
fun = ti->mFunction;
|
|
arg = ti->mArg;
|
|
|
|
/* The thread is responsible for freeing the startup information */
|
|
free((void *)ti);
|
|
|
|
/* Call the actual client thread function */
|
|
res = fun(arg);
|
|
|
|
#if defined(_TTHREAD_WIN32_)
|
|
return res;
|
|
#else
|
|
pres = malloc(sizeof(int));
|
|
if (pres != NULL)
|
|
{
|
|
*(int*)pres = res;
|
|
}
|
|
return pres;
|
|
#endif
|
|
}
|
|
|
|
int thrd_create(thrd_t *thr, thrd_start_t func, void *arg)
|
|
{
|
|
/* Fill out the thread startup information (passed to the thread wrapper,
|
|
which will eventually free it) */
|
|
_thread_start_info* ti = (_thread_start_info*)malloc(sizeof(_thread_start_info));
|
|
if (ti == NULL)
|
|
{
|
|
return thrd_nomem;
|
|
}
|
|
ti->mFunction = func;
|
|
ti->mArg = arg;
|
|
|
|
/* Create the thread */
|
|
#if defined(_TTHREAD_WIN32_)
|
|
*thr = (HANDLE)_beginthreadex(NULL, 0, _thrd_wrapper_function, (void *)ti, 0, NULL);
|
|
#elif defined(_TTHREAD_POSIX_)
|
|
if(pthread_create(thr, NULL, _thrd_wrapper_function, (void *)ti) != 0)
|
|
{
|
|
*thr = 0;
|
|
}
|
|
#endif
|
|
|
|
/* Did we fail to create the thread? */
|
|
if(!*thr)
|
|
{
|
|
free(ti);
|
|
return thrd_error;
|
|
}
|
|
|
|
return thrd_success;
|
|
}
|
|
|
|
thrd_t thrd_current(void)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
return GetCurrentThread();
|
|
#else
|
|
return pthread_self();
|
|
#endif
|
|
}
|
|
|
|
int thrd_detach(thrd_t thr)
|
|
{
|
|
/* FIXME! */
|
|
(void)thr;
|
|
return thrd_error;
|
|
}
|
|
|
|
int thrd_equal(thrd_t thr0, thrd_t thr1)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
return thr0 == thr1;
|
|
#else
|
|
return pthread_equal(thr0, thr1);
|
|
#endif
|
|
}
|
|
|
|
void thrd_exit(int res)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
ExitThread(res);
|
|
#else
|
|
void *pres = malloc(sizeof(int));
|
|
if (pres != NULL)
|
|
{
|
|
*(int*)pres = res;
|
|
}
|
|
pthread_exit(pres);
|
|
#endif
|
|
}
|
|
|
|
int thrd_join(thrd_t thr, int *res)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
if (WaitForSingleObject(thr, INFINITE) == WAIT_FAILED)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
if (res != NULL)
|
|
{
|
|
DWORD dwRes;
|
|
GetExitCodeThread(thr, &dwRes);
|
|
*res = dwRes;
|
|
}
|
|
#elif defined(_TTHREAD_POSIX_)
|
|
void *pres;
|
|
int ires = 0;
|
|
if (pthread_join(thr, &pres) != 0)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
if (pres != NULL)
|
|
{
|
|
ires = *(int*)pres;
|
|
free(pres);
|
|
}
|
|
if (res != NULL)
|
|
{
|
|
*res = ires;
|
|
}
|
|
#endif
|
|
return thrd_success;
|
|
}
|
|
|
|
int thrd_sleep(const struct timespec *time_point, struct timespec *remaining)
|
|
{
|
|
struct timespec now;
|
|
#if defined(_TTHREAD_WIN32_)
|
|
DWORD delta;
|
|
#else
|
|
long delta;
|
|
#endif
|
|
|
|
/* Get the current time */
|
|
if (clock_gettime(CLOCK_REALTIME, &now) != 0)
|
|
return -2; // FIXME: Some specific error code?
|
|
|
|
#if defined(_TTHREAD_WIN32_)
|
|
/* Delta in milliseconds */
|
|
delta = (DWORD) ((time_point->tv_sec - now.tv_sec) * 1000 +
|
|
(time_point->tv_nsec - now.tv_nsec + 500000) / 1000000);
|
|
if (delta > 0)
|
|
{
|
|
Sleep(delta);
|
|
}
|
|
#else
|
|
/* Delta in microseconds */
|
|
delta = (time_point->tv_sec - now.tv_sec) * 1000000L +
|
|
(time_point->tv_nsec - now.tv_nsec + 500L) / 1000L;
|
|
|
|
/* On some systems, the usleep argument must be < 1000000 */
|
|
while (delta > 999999L)
|
|
{
|
|
usleep(999999);
|
|
delta -= 999999L;
|
|
}
|
|
if (delta > 0L)
|
|
{
|
|
usleep((useconds_t)delta);
|
|
}
|
|
#endif
|
|
|
|
/* We don't support waking up prematurely (yet) */
|
|
if (remaining)
|
|
{
|
|
remaining->tv_sec = 0;
|
|
remaining->tv_nsec = 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void thrd_yield(void)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
Sleep(0);
|
|
#else
|
|
sched_yield();
|
|
#endif
|
|
}
|
|
|
|
int tss_create(tss_t *key, tss_dtor_t dtor)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
/* FIXME: The destructor function is not supported yet... */
|
|
if (dtor != NULL)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
*key = TlsAlloc();
|
|
if (*key == TLS_OUT_OF_INDEXES)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
#else
|
|
if (pthread_key_create(key, dtor) != 0)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
#endif
|
|
return thrd_success;
|
|
}
|
|
|
|
void tss_delete(tss_t key)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
TlsFree(key);
|
|
#else
|
|
pthread_key_delete(key);
|
|
#endif
|
|
}
|
|
|
|
void *tss_get(tss_t key)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
return TlsGetValue(key);
|
|
#else
|
|
return pthread_getspecific(key);
|
|
#endif
|
|
}
|
|
|
|
int tss_set(tss_t key, void *val)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
if (TlsSetValue(key, val) == 0)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
#else
|
|
if (pthread_setspecific(key, val) != 0)
|
|
{
|
|
return thrd_error;
|
|
}
|
|
#endif
|
|
return thrd_success;
|
|
}
|
|
|
|
#if defined(_TTHREAD_EMULATE_CLOCK_GETTIME_)
|
|
int _tthread_clock_gettime(clockid_t clk_id, struct timespec *ts)
|
|
{
|
|
#if defined(_TTHREAD_WIN32_)
|
|
struct _timeb tb;
|
|
_ftime(&tb);
|
|
ts->tv_sec = (time_t)tb.time;
|
|
ts->tv_nsec = 1000000L * (long)tb.millitm;
|
|
#else
|
|
struct timeval tv;
|
|
gettimeofday(&tv, NULL);
|
|
ts->tv_sec = (time_t)tv.tv_sec;
|
|
ts->tv_nsec = 1000L * (long)tv.tv_usec;
|
|
#endif
|
|
return 0;
|
|
}
|
|
#endif // _TTHREAD_EMULATE_CLOCK_GETTIME_
|
|
|