X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/f783ed4680c6862a1b7543237e89d1221334bae0..fbf3527063143094ed40d1a4a3bc595c29b81145:/src/xbt/xbt_os_thread.c diff --git a/src/xbt/xbt_os_thread.c b/src/xbt/xbt_os_thread.c index b2be4c7fe6..6b3d98a48b 100644 --- a/src/xbt/xbt_os_thread.c +++ b/src/xbt/xbt_os_thread.c @@ -1,36 +1,43 @@ /* xbt_os_thread -- portability layer over the pthread API */ /* Used in RL to get win/lin portability, and in SG when CONTEXT_THREAD */ -/* in SG, when using CONTEXT_UCONTEXT, xbt_os_thread_stub is used instead */ +/* in SG, when using HAVE_UCONTEXT_CONTEXTS, xbt_os_thread_stub is used instead */ -/* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team. +/* Copyright (c) 2007-2015. The SimGrid Team. * All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it * under the terms of the license (GNU LGPL) which comes with this package. */ -#include "gras_config.h" -#include "xbt/sysdep.h" -#include "xbt/ex.h" -#include "xbt/ex_interface.h" /* We play crude games with exceptions */ -#include "portable.h" -#include "xbt/xbt_os_time.h" /* Portable time facilities */ -#include "xbt/xbt_os_thread.h" /* This module */ -#include "xbt_modinter.h" /* Initialization/finalization of this module */ - -XBT_LOG_NEW_DEFAULT_SUBCATEGORY(xbt_sync_os, xbt, - "Synchronization mechanism (OS-level)"); - -/* ********************************* PTHREAD IMPLEMENTATION ************************************ */ -#ifdef HAVE_PTHREAD_H +#include "src/internal_config.h" +#ifdef HAVE_PTHREAD_SETAFFINITY +#define _GNU_SOURCE +#include +#endif +#include +#include #include +#include -#ifdef HAVE_MUTEX_TIMEDLOCK -/* redefine the function header since we fail to get this from system headers on amd (at least) */ -int pthread_mutex_timedlock(pthread_mutex_t * mutex, - const struct timespec *abs_timeout); +#if defined(WIN32) +#elif defined(__MACH__) && defined(__APPLE__) +#include +#include +#include +#else +#include #endif +#include "xbt/sysdep.h" +#include "xbt/ex.h" +#include "src/xbt/ex_interface.h" /* We play crude games with exceptions */ +#include "src/portable.h" +#include "xbt/xbt_os_time.h" /* Portable time facilities */ +#include "xbt/xbt_os_thread.h" /* This module */ +#include "src/xbt_modinter.h" /* Initialization/finalization of this module */ + +XBT_LOG_NEW_DEFAULT_SUBCATEGORY(xbt_sync_os, xbt, "Synchronization mechanism (OS-level)"); + /* use named sempahore when sem_init() does not work */ #ifndef HAVE_SEM_INIT @@ -53,6 +60,10 @@ static xbt_os_thread_t main_thread = NULL; static pthread_key_t xbt_self_thread_key; static int thread_mod_inited = 0; +/* defaults attribute for pthreads */ +//FIXME: find where to put this +static pthread_attr_t thread_attr; + /* frees the xbt_os_thread_t corresponding to the current thread */ static void xbt_os_thread_free_thread_data(xbt_os_thread_t thread) { @@ -74,43 +85,42 @@ static xbt_running_ctx_t *_os_thread_get_running_ctx(void) static void _os_thread_ex_terminate(xbt_ex_t * e) { xbt_ex_display(e); - - abort(); + xbt_abort(); /* FIXME: there should be a configuration variable to choose to kill everyone or only this one */ } void xbt_os_thread_mod_preinit(void) { - int errcode; - if (thread_mod_inited) return; - if ((errcode = pthread_key_create(&xbt_self_thread_key, NULL))) - THROWF(system_error, errcode, - "pthread_key_create failed for xbt_self_thread_key"); + int errcode = pthread_key_create(&xbt_self_thread_key, NULL); + xbt_assert(errcode == 0, "pthread_key_create failed for xbt_self_thread_key"); main_thread = xbt_new(s_xbt_os_thread_t, 1); + main_thread->name = NULL; + main_thread->detached = 0; main_thread->name = (char *) "main"; - main_thread->start_routine = NULL; main_thread->param = NULL; + main_thread->start_routine = NULL; main_thread->running_ctx = xbt_new(xbt_running_ctx_t, 1); + main_thread->extra_data = NULL; XBT_RUNNING_CTX_INITIALIZE(main_thread->running_ctx); if ((errcode = pthread_setspecific(xbt_self_thread_key, main_thread))) THROWF(system_error, errcode, - "pthread_setspecific failed for xbt_self_thread_key"); - + "Impossible to set the SimGrid identity descriptor to the main thread (pthread_setspecific failed)"); __xbt_running_ctx_fetch = _os_thread_get_running_ctx; __xbt_ex_terminate = _os_thread_ex_terminate; + pthread_attr_init(&thread_attr); + thread_mod_inited = 1; #ifndef HAVE_SEM_INIT next_sem_ID_lock = xbt_os_mutex_init(); #endif - } void xbt_os_thread_mod_postexit(void) @@ -134,8 +144,8 @@ void xbt_os_thread_mod_postexit(void) __xbt_ex_terminate = &__xbt_ex_terminate_default; } -int xbt_os_thread_atfork(void (*prepare)(void), - void (*parent)(void), void (*child)(void)) +/* this function is critical to tesh+mmalloc, don't mess with it */ +int xbt_os_thread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void)) { return pthread_atfork(prepare, parent, child); } @@ -143,25 +153,19 @@ int xbt_os_thread_atfork(void (*prepare)(void), static void *wrapper_start_routine(void *s) { xbt_os_thread_t t = s; - int errcode; - if ((errcode = pthread_setspecific(xbt_self_thread_key, t))) - THROWF(system_error, errcode, - "pthread_setspecific failed for xbt_self_thread_key"); + int errcode = pthread_setspecific(xbt_self_thread_key, t); + xbt_assert(errcode == 0, "pthread_setspecific failed for xbt_self_thread_key"); - void *res = (*(t->start_routine)) (t->param); + void *res = t->start_routine(t->param); if (t->detached) xbt_os_thread_free_thread_data(t); return res; } -xbt_os_thread_t xbt_os_thread_create(const char *name, - pvoid_f_pvoid_t start_routine, - void *param, - void *extra_data) -{ - int errcode; +xbt_os_thread_t xbt_os_thread_create(const char *name, pvoid_f_pvoid_t start_routine, void *param, void *extra_data) +{ xbt_os_thread_t res_thread = xbt_new(s_xbt_os_thread_t, 1); res_thread->detached = 0; res_thread->name = xbt_strdup(name); @@ -171,14 +175,68 @@ xbt_os_thread_t xbt_os_thread_create(const char *name, XBT_RUNNING_CTX_INITIALIZE(res_thread->running_ctx); res_thread->extra_data = extra_data; - if ((errcode = pthread_create(&(res_thread->t), NULL, - wrapper_start_routine, res_thread))) - THROWF(system_error, errcode, - "pthread_create failed: %s", strerror(errcode)); + int errcode = pthread_create(&(res_thread->t), &thread_attr, wrapper_start_routine, res_thread); + xbt_assert(errcode == 0, "pthread_create failed: %s", strerror(errcode)); return res_thread; } +int xbt_os_thread_bind(xbt_os_thread_t thread, int cpu){ + int errcode = 0; +#ifdef HAVE_PTHREAD_SETAFFINITY + pthread_t pthread = thread->t; + cpu_set_t cpuset; + CPU_ZERO(&cpuset); + CPU_SET(cpu, &cpuset); + errcode = pthread_setaffinity_np(pthread, sizeof(cpu_set_t), &cpuset); +#endif + return errcode; +} + +void xbt_os_thread_setstacksize(int stack_size) +{ + size_t alignment[] = { + xbt_pagesize, +#ifdef PTHREAD_STACK_MIN + PTHREAD_STACK_MIN, +#endif + 0 + }; + + xbt_assert(stack_size >= 0, "stack size %d is negative, maybe it exceeds MAX_INT?", stack_size); + + size_t sz = stack_size; + int res = pthread_attr_setstacksize(&thread_attr, sz); + + for (int i = 0; res == EINVAL && alignment[i] > 0; i++) { + /* Invalid size, try again with next multiple of alignment[i]. */ + size_t rem = sz % alignment[i]; + if (rem != 0 || sz == 0) { + size_t sz2 = sz - rem + alignment[i]; + XBT_DEBUG("pthread_attr_setstacksize failed for %zd, try again with %zd", sz, sz2); + sz = sz2; + res = pthread_attr_setstacksize(&thread_attr, sz); + } + } + + if (res == EINVAL) + XBT_WARN("invalid stack size (maybe too big): %zd", sz); + else if (res != 0) + XBT_WARN("unknown error %d in pthread stacksize setting: %zd", res, sz); +} + +void xbt_os_thread_setguardsize(int guard_size) +{ +#ifdef WIN32 + THROW_UNIMPLEMENTED; //pthread_attr_setguardsize is not implemented in pthread.h on windows +#else + size_t sz = guard_size; + int res = pthread_attr_setguardsize(&thread_attr, sz); + if (res) + XBT_WARN("pthread_attr_setguardsize failed (%d) for size: %zd", res, sz); +#endif +} + const char *xbt_os_thread_name(xbt_os_thread_t t) { return t->name; @@ -192,12 +250,9 @@ const char *xbt_os_thread_self_name(void) void xbt_os_thread_join(xbt_os_thread_t thread, void **thread_return) { + int errcode = pthread_join(thread->t, thread_return); - int errcode; - - if ((errcode = pthread_join(thread->t, thread_return))) - THROWF(system_error, errcode, "pthread_join failed: %s", - strerror(errcode)); + xbt_assert(errcode==0, "pthread_join failed: %s", strerror(errcode)); xbt_os_thread_free_thread_data(thread); } @@ -208,28 +263,22 @@ void xbt_os_thread_exit(int *retval) xbt_os_thread_t xbt_os_thread_self(void) { - xbt_os_thread_t res; - if (!thread_mod_inited) return NULL; - res = pthread_getspecific(xbt_self_thread_key); - - return res; + return pthread_getspecific(xbt_self_thread_key); } -void xbt_os_thread_key_create(xbt_os_thread_key_t* key) { - - int errcode; - if ((errcode = pthread_key_create(key, NULL))) - THROWF(system_error, errcode, "pthread_key_create failed"); +void xbt_os_thread_key_create(xbt_os_thread_key_t* key) +{ + int errcode = pthread_key_create(key, NULL); + xbt_assert(errcode==0 , "pthread_key_create failed"); } void xbt_os_thread_set_specific(xbt_os_thread_key_t key, void* value) { - int errcode; - if ((errcode = pthread_setspecific(key, value))) - THROWF(system_error, errcode, "pthread_setspecific failed"); + int errcode = pthread_setspecific(key, value); + xbt_assert(errcode==0, "pthread_setspecific failed"); } void* xbt_os_thread_get_specific(xbt_os_thread_key_t key) { @@ -265,22 +314,16 @@ typedef struct xbt_os_mutex_ { xbt_os_mutex_t xbt_os_mutex_init(void) { xbt_os_mutex_t res = xbt_new(s_xbt_os_mutex_t, 1); - int errcode; - - if ((errcode = pthread_mutex_init(&(res->m), NULL))) - THROWF(system_error, errcode, "pthread_mutex_init() failed: %s", - strerror(errcode)); + int errcode = pthread_mutex_init(&(res->m), NULL); + xbt_assert(errcode==0, "pthread_mutex_init() failed: %s", strerror(errcode)); return res; } void xbt_os_mutex_acquire(xbt_os_mutex_t mutex) { - int errcode; - - if ((errcode = pthread_mutex_lock(&(mutex->m)))) - THROWF(system_error, errcode, "pthread_mutex_lock(%p) failed: %s", - mutex, strerror(errcode)); + int errcode = pthread_mutex_lock(&(mutex->m)); + xbt_assert(errcode==0, "pthread_mutex_lock(%p) failed: %s", mutex, strerror(errcode)); } @@ -301,7 +344,7 @@ void xbt_os_mutex_timedacquire(xbt_os_mutex_t mutex, double delay) THROWF(timeout_error, 0, "mutex %p not ready", mutex); default: THROWF(system_error, errcode, - "xbt_mutex_timedacquire(%p) failed: %s", mutex, + "xbt_os_mutex_timedacquire(%p) failed: %s", mutex, strerror(errcode)); } @@ -316,9 +359,9 @@ void xbt_os_mutex_timedacquire(xbt_os_mutex_t mutex, double delay) ts_end.tv_nsec = (long) ((end - ts_end.tv_sec) * 1000000000); XBT_DEBUG("pthread_mutex_timedlock(%p,%p)", &(mutex->m), &ts_end); - errcode = pthread_mutex_timedlock(&(mutex->m), &ts_end); + int errcode = pthread_mutex_timedlock(&(mutex->m), &ts_end); -#else /* Well, let's reimplement it since those lazy libc dudes didn't */ +#else /* reimplement it since those lazy libc dudes didn't (Mac OSX, hu?) */ double start = xbt_os_time(); do { errcode = pthread_mutex_trylock(&(mutex->m)); @@ -349,23 +392,17 @@ void xbt_os_mutex_timedacquire(xbt_os_mutex_t mutex, double delay) void xbt_os_mutex_release(xbt_os_mutex_t mutex) { - int errcode; - - if ((errcode = pthread_mutex_unlock(&(mutex->m)))) - THROWF(system_error, errcode, "pthread_mutex_unlock(%p) failed: %s", - mutex, strerror(errcode)); + int errcode = pthread_mutex_unlock(&(mutex->m)); + xbt_assert(errcode==0, "pthread_mutex_unlock(%p) failed: %s", mutex, strerror(errcode)); } void xbt_os_mutex_destroy(xbt_os_mutex_t mutex) { - int errcode; - if (!mutex) return; - if ((errcode = pthread_mutex_destroy(&(mutex->m)))) - THROWF(system_error, errcode, "pthread_mutex_destroy(%p) failed: %s", - mutex, strerror(errcode)); + int errcode = pthread_mutex_destroy(&(mutex->m)); + xbt_assert(errcode == 0, "pthread_mutex_destroy(%p) failed: %s", mutex, strerror(errcode)); free(mutex); } @@ -378,25 +415,19 @@ typedef struct xbt_os_cond_ { xbt_os_cond_t xbt_os_cond_init(void) { xbt_os_cond_t res = xbt_new(s_xbt_os_cond_t, 1); - int errcode; - if ((errcode = pthread_cond_init(&(res->c), NULL))) - THROWF(system_error, errcode, "pthread_cond_init() failed: %s", - strerror(errcode)); - + int errcode = pthread_cond_init(&(res->c), NULL); + xbt_assert(errcode==0, "pthread_cond_init() failed: %s", strerror(errcode)); return res; } void xbt_os_cond_wait(xbt_os_cond_t cond, xbt_os_mutex_t mutex) { - int errcode; - if ((errcode = pthread_cond_wait(&(cond->c), &(mutex->m)))) - THROWF(system_error, errcode, "pthread_cond_wait(%p,%p) failed: %s", - cond, mutex, strerror(errcode)); + int errcode = pthread_cond_wait(&(cond->c), &(mutex->m)); + xbt_assert(errcode==0, "pthread_cond_wait(%p,%p) failed: %s", cond, mutex, strerror(errcode)); } -void xbt_os_cond_timedwait(xbt_os_cond_t cond, xbt_os_mutex_t mutex, - double delay) +void xbt_os_cond_timedwait(xbt_os_cond_t cond, xbt_os_mutex_t mutex, double delay) { int errcode; struct timespec ts_end; @@ -407,10 +438,8 @@ void xbt_os_cond_timedwait(xbt_os_cond_t cond, xbt_os_mutex_t mutex, } else { ts_end.tv_sec = (time_t) floor(end); ts_end.tv_nsec = (long) ((end - ts_end.tv_sec) * 1000000000); - XBT_DEBUG("pthread_cond_timedwait(%p,%p,%p)", &(cond->c), &(mutex->m), - &ts_end); - switch ((errcode = - pthread_cond_timedwait(&(cond->c), &(mutex->m), &ts_end))) { + XBT_DEBUG("pthread_cond_timedwait(%p,%p,%p)", &(cond->c), &(mutex->m), &ts_end); + switch ((errcode = pthread_cond_timedwait(&(cond->c), &(mutex->m), &ts_end))) { case 0: return; case ETIMEDOUT: @@ -418,48 +447,33 @@ void xbt_os_cond_timedwait(xbt_os_cond_t cond, xbt_os_mutex_t mutex, "condition %p (mutex %p) wasn't signaled before timeout (%f)", cond, mutex, delay); default: - THROWF(system_error, errcode, - "pthread_cond_timedwait(%p,%p,%f) failed: %s", cond, mutex, - delay, strerror(errcode)); + THROWF(system_error, errcode, "pthread_cond_timedwait(%p,%p,%f) failed: %s", cond, mutex, delay, strerror(errcode)); } } } void xbt_os_cond_signal(xbt_os_cond_t cond) { - int errcode; - if ((errcode = pthread_cond_signal(&(cond->c)))) - THROWF(system_error, errcode, "pthread_cond_signal(%p) failed: %s", - cond, strerror(errcode)); + int errcode = pthread_cond_signal(&(cond->c)); + xbt_assert(errcode==0, "pthread_cond_signal(%p) failed: %s", cond, strerror(errcode)); } void xbt_os_cond_broadcast(xbt_os_cond_t cond) { - int errcode; - if ((errcode = pthread_cond_broadcast(&(cond->c)))) - THROWF(system_error, errcode, "pthread_cond_broadcast(%p) failed: %s", - cond, strerror(errcode)); + int errcode = pthread_cond_broadcast(&(cond->c)); + xbt_assert(errcode==0, "pthread_cond_broadcast(%p) failed: %s", cond, strerror(errcode)); } void xbt_os_cond_destroy(xbt_os_cond_t cond) { - int errcode; - if (!cond) return; - if ((errcode = pthread_cond_destroy(&(cond->c)))) - THROWF(system_error, errcode, "pthread_cond_destroy(%p) failed: %s", - cond, strerror(errcode)); + int errcode = pthread_cond_destroy(&(cond->c)); + xbt_assert(errcode==0, "pthread_cond_destroy(%p) failed: %s", cond, strerror(errcode)); free(cond); } -void *xbt_os_thread_getparam(void) -{ - xbt_os_thread_t t = xbt_os_thread_self(); - return t ? t->param : NULL; -} - typedef struct xbt_os_sem_ { #ifndef HAVE_SEM_INIT char *name; @@ -497,7 +511,7 @@ xbt_os_sem_t xbt_os_sem_init(unsigned int value) res->name[13] = '\0'; res->ps = sem_open(res->name, O_CREAT, 0644, value); } - if ((res->ps == (sem_t *) SEM_FAILED)) + if (res->ps == (sem_t *) SEM_FAILED) THROWF(system_error, errno, "sem_open() failed: %s", strerror(errno)); /* Remove the name from the semaphore namespace: we never join on it */ @@ -512,8 +526,6 @@ xbt_os_sem_t xbt_os_sem_init(unsigned int value) void xbt_os_sem_acquire(xbt_os_sem_t sem) { - if (!sem) - THROWF(arg_error, EINVAL, "Cannot acquire of the NULL semaphore"); if (sem_wait(sem->ps) < 0) THROWF(system_error, errno, "sem_wait() failed: %s", strerror(errno)); } @@ -522,9 +534,6 @@ void xbt_os_sem_timedacquire(xbt_os_sem_t sem, double delay) { int errcode; - if (!sem) - THROWF(arg_error, EINVAL, "Cannot acquire of the NULL semaphore"); - if (delay < 0) { xbt_os_sem_acquire(sem); } else if (delay == 0) { @@ -581,18 +590,12 @@ void xbt_os_sem_timedacquire(xbt_os_sem_t sem, double delay) void xbt_os_sem_release(xbt_os_sem_t sem) { - if (!sem) - THROWF(arg_error, EINVAL, "Cannot release of the NULL semaphore"); - if (sem_post(sem->ps) < 0) THROWF(system_error, errno, "sem_post() failed: %s", strerror(errno)); } void xbt_os_sem_destroy(xbt_os_sem_t sem) { - if (!sem) - THROWF(arg_error, EINVAL, "Cannot destroy the NULL sempahore"); - #ifdef HAVE_SEM_INIT if (sem_destroy(sem->ps) < 0) THROWF(system_error, errno, "sem_destroy() failed: %s", @@ -608,557 +611,37 @@ void xbt_os_sem_destroy(xbt_os_sem_t sem) void xbt_os_sem_get_value(xbt_os_sem_t sem, int *svalue) { - if (!sem) - THROWF(arg_error, EINVAL, - "Cannot get the value of the NULL semaphore"); - if (sem_getvalue(&(sem->s), svalue) < 0) THROWF(system_error, errno, "sem_getvalue() failed: %s", strerror(errno)); } -/* ********************************* WINDOWS IMPLEMENTATION ************************************ */ - -#elif defined(_XBT_WIN32) - -#include -typedef struct xbt_os_thread_ { - char *name; - HANDLE handle; /* the win thread handle */ - unsigned long id; /* the win thread id */ - pvoid_f_pvoid_t start_routine; - void *param; - void *extra_data; -} s_xbt_os_thread_t; - -/* so we can specify the size of the stack of the threads */ -#ifndef STACK_SIZE_PARAM_IS_A_RESERVATION -#define STACK_SIZE_PARAM_IS_A_RESERVATION 0x00010000 -#endif +/** @brief Returns the amount of cores on the current host */ +int xbt_os_get_numcores(void) { +#ifdef WIN32 + SYSTEM_INFO sysinfo; + GetSystemInfo(&sysinfo); + return sysinfo.dwNumberOfProcessors; +#elif defined(__APPLE__) && defined(__MACH__) + int nm[2]; + size_t len = 4; + uint32_t count; -/* the default size of the stack of the threads (in bytes)*/ -#define XBT_DEFAULT_THREAD_STACK_SIZE 4096 + nm[0] = CTL_HW; nm[1] = HW_AVAILCPU; + sysctl(nm, 2, &count, &len, NULL, 0); -/* key to the TLS containing the xbt_os_thread_t structure */ -static unsigned long xbt_self_thread_key; - -void xbt_os_thread_mod_preinit(void) -{ - xbt_self_thread_key = TlsAlloc(); -} - -void xbt_os_thread_mod_postexit(void) -{ - - if (!TlsFree(xbt_self_thread_key)) - THROWF(system_error, (int) GetLastError(), - "TlsFree() failed to cleanup the thread submodule"); -} - -int xbt_os_thread_atfork(void (*prepare)(void), - void (*parent)(void), void (*child)(void)) -{ - return 0; -} - -static DWORD WINAPI wrapper_start_routine(void *s) -{ - xbt_os_thread_t t = (xbt_os_thread_t) s; - DWORD *rv; - - if (!TlsSetValue(xbt_self_thread_key, t)) - THROWF(system_error, (int) GetLastError(), - "TlsSetValue of data describing the created thread failed"); - - rv = (DWORD *) ((t->start_routine) (t->param)); - - return rv ? *rv : 0; - -} - - -xbt_os_thread_t xbt_os_thread_create(const char *name, - pvoid_f_pvoid_t start_routine, - void *param, - void *extra_data) -{ - - xbt_os_thread_t t = xbt_new(s_xbt_os_thread_t, 1); - - t->name = xbt_strdup(name); - t->start_routine = start_routine; - t->param = param; - t->extra_data = extra_data; - t->handle = CreateThread(NULL, XBT_DEFAULT_THREAD_STACK_SIZE, - (LPTHREAD_START_ROUTINE) wrapper_start_routine, - t, STACK_SIZE_PARAM_IS_A_RESERVATION, &(t->id)); - - if (!t->handle) { - xbt_free(t); - THROWF(system_error, (int) GetLastError(), "CreateThread failed"); - } - - return t; -} - -const char *xbt_os_thread_name(xbt_os_thread_t t) -{ - return t->name; -} - -const char *xbt_os_thread_self_name(void) -{ - xbt_os_thread_t t = xbt_os_thread_self(); - return t ? t->name : "main"; -} - -void xbt_os_thread_join(xbt_os_thread_t thread, void **thread_return) -{ - - if (WAIT_OBJECT_0 != WaitForSingleObject(thread->handle, INFINITE)) - THROWF(system_error, (int) GetLastError(), - "WaitForSingleObject failed"); - - if (thread_return) { - - if (!GetExitCodeThread(thread->handle, (DWORD *) (*thread_return))) - THROWF(system_error, (int) GetLastError(), - "GetExitCodeThread failed"); - } - - CloseHandle(thread->handle); - - if (thread->name) - free(thread->name); - - free(thread); -} - -void xbt_os_thread_exit(int *retval) -{ - if (retval) - ExitThread(*retval); - else - ExitThread(0); -} - -void xbt_os_thread_key_create(xbt_os_thread_key_t* key) { - - *key = TlsAlloc(); -} - -void xbt_os_thread_set_specific(xbt_os_thread_key_t key, void* value) { - - if (!TlsSetValue(key, value)) - THROWF(system_error, (int) GetLastError(), "TlsSetValue() failed"); -} - -void* xbt_os_thread_get_specific(xbt_os_thread_key_t key) { - return TlsGetValue(key); -} - -void xbt_os_thread_detach(xbt_os_thread_t thread) -{ - THROW_UNIMPLEMENTED; -} - - -xbt_os_thread_t xbt_os_thread_self(void) -{ - return TlsGetValue(xbt_self_thread_key); -} - -void *xbt_os_thread_getparam(void) -{ - xbt_os_thread_t t = xbt_os_thread_self(); - return t->param; -} - - -void xbt_os_thread_yield(void) -{ - Sleep(0); -} - -void xbt_os_thread_cancel(xbt_os_thread_t t) -{ - if (!TerminateThread(t->handle, 0)) - THROWF(system_error, (int) GetLastError(), "TerminateThread failed"); -} - -/****** mutex related functions ******/ -typedef struct xbt_os_mutex_ { - /* KEEP IT IN SYNC WITH xbt_thread.c */ - CRITICAL_SECTION lock; -} s_xbt_os_mutex_t; - -xbt_os_mutex_t xbt_os_mutex_init(void) -{ - xbt_os_mutex_t res = xbt_new(s_xbt_os_mutex_t, 1); - - /* initialize the critical section object */ - InitializeCriticalSection(&(res->lock)); - - return res; -} - -void xbt_os_mutex_acquire(xbt_os_mutex_t mutex) -{ - EnterCriticalSection(&mutex->lock); -} - -void xbt_os_mutex_timedacquire(xbt_os_mutex_t mutex, double delay) -{ - THROW_UNIMPLEMENTED; -} - -void xbt_os_mutex_release(xbt_os_mutex_t mutex) -{ - - LeaveCriticalSection(&mutex->lock); - -} - -void xbt_os_mutex_destroy(xbt_os_mutex_t mutex) -{ - - if (!mutex) - return; - - DeleteCriticalSection(&mutex->lock); - free(mutex); -} - -/***** condition related functions *****/ -enum { /* KEEP IT IN SYNC WITH xbt_thread.c */ - SIGNAL = 0, - BROADCAST = 1, - MAX_EVENTS = 2 -}; - -typedef struct xbt_os_cond_ { - /* KEEP IT IN SYNC WITH xbt_thread.c */ - HANDLE events[MAX_EVENTS]; - - unsigned int waiters_count; /* the number of waiters */ - CRITICAL_SECTION waiters_count_lock; /* protect access to waiters_count */ -} s_xbt_os_cond_t; - -xbt_os_cond_t xbt_os_cond_init(void) -{ - - xbt_os_cond_t res = xbt_new0(s_xbt_os_cond_t, 1); - - memset(&res->waiters_count_lock, 0, sizeof(CRITICAL_SECTION)); - - /* initialize the critical section object */ - InitializeCriticalSection(&res->waiters_count_lock); - - res->waiters_count = 0; - - /* Create an auto-reset event */ - res->events[SIGNAL] = CreateEvent(NULL, FALSE, FALSE, NULL); - - if (!res->events[SIGNAL]) { - DeleteCriticalSection(&res->waiters_count_lock); - free(res); - THROWF(system_error, 0, "CreateEvent failed for the signals"); - } - - /* Create a manual-reset event. */ - res->events[BROADCAST] = CreateEvent(NULL, TRUE, FALSE, NULL); - - if (!res->events[BROADCAST]) { - - DeleteCriticalSection(&res->waiters_count_lock); - CloseHandle(res->events[SIGNAL]); - free(res); - THROWF(system_error, 0, "CreateEvent failed for the broadcasts"); - } - - return res; -} - -void xbt_os_cond_wait(xbt_os_cond_t cond, xbt_os_mutex_t mutex) -{ - - unsigned long wait_result; - int is_last_waiter; - - /* lock the threads counter and increment it */ - EnterCriticalSection(&cond->waiters_count_lock); - cond->waiters_count++; - LeaveCriticalSection(&cond->waiters_count_lock); - - /* unlock the mutex associate with the condition */ - LeaveCriticalSection(&mutex->lock); - - /* wait for a signal (broadcast or no) */ - wait_result = WaitForMultipleObjects(2, cond->events, FALSE, INFINITE); - - if (wait_result == WAIT_FAILED) - THROWF(system_error, 0, - "WaitForMultipleObjects failed, so we cannot wait on the condition"); - - /* we have a signal lock the condition */ - EnterCriticalSection(&cond->waiters_count_lock); - cond->waiters_count--; - - /* it's the last waiter or it's a broadcast ? */ - is_last_waiter = ((wait_result == WAIT_OBJECT_0 + BROADCAST - 1) - && (cond->waiters_count == 0)); - - LeaveCriticalSection(&cond->waiters_count_lock); - - /* yes it's the last waiter or it's a broadcast - * only reset the manual event (the automatic event is reset in the WaitForMultipleObjects() function - * by the system. - */ - if (is_last_waiter) - if (!ResetEvent(cond->events[BROADCAST])) - THROWF(system_error, 0, "ResetEvent failed"); - - /* relock the mutex associated with the condition in accordance with the posix thread specification */ - EnterCriticalSection(&mutex->lock); -} - -void xbt_os_cond_timedwait(xbt_os_cond_t cond, xbt_os_mutex_t mutex, - double delay) -{ - - unsigned long wait_result = WAIT_TIMEOUT; - int is_last_waiter; - unsigned long end = (unsigned long) (delay * 1000); - - - if (delay < 0) { - xbt_os_cond_wait(cond, mutex); - } else { - XBT_DEBUG("xbt_cond_timedwait(%p,%p,%lu)", &(cond->events), - &(mutex->lock), end); - - /* lock the threads counter and increment it */ - EnterCriticalSection(&cond->waiters_count_lock); - cond->waiters_count++; - LeaveCriticalSection(&cond->waiters_count_lock); - - /* unlock the mutex associate with the condition */ - LeaveCriticalSection(&mutex->lock); - /* wait for a signal (broadcast or no) */ - - wait_result = WaitForMultipleObjects(2, cond->events, FALSE, end); - - switch (wait_result) { - case WAIT_TIMEOUT: - THROWF(timeout_error, GetLastError(), - "condition %p (mutex %p) wasn't signaled before timeout (%f)", - cond, mutex, delay); - case WAIT_FAILED: - THROWF(system_error, GetLastError(), - "WaitForMultipleObjects failed, so we cannot wait on the condition"); + if(count < 1) { + nm[1] = HW_NCPU; + sysctl(nm, 2, &count, &len, NULL, 0); + if(count < 1) { count = 1; } } - - /* we have a signal lock the condition */ - EnterCriticalSection(&cond->waiters_count_lock); - cond->waiters_count--; - - /* it's the last waiter or it's a broadcast ? */ - is_last_waiter = ((wait_result == WAIT_OBJECT_0 + BROADCAST - 1) - && (cond->waiters_count == 0)); - - LeaveCriticalSection(&cond->waiters_count_lock); - - /* yes it's the last waiter or it's a broadcast - * only reset the manual event (the automatic event is reset in the WaitForMultipleObjects() function - * by the system. - */ - if (is_last_waiter) - if (!ResetEvent(cond->events[BROADCAST])) - THROWF(system_error, 0, "ResetEvent failed"); - - /* relock the mutex associated with the condition in accordance with the posix thread specification */ - EnterCriticalSection(&mutex->lock); - } - /*THROW_UNIMPLEMENTED; */ -} - -void xbt_os_cond_signal(xbt_os_cond_t cond) -{ - int have_waiters; - - EnterCriticalSection(&cond->waiters_count_lock); - have_waiters = cond->waiters_count > 0; - LeaveCriticalSection(&cond->waiters_count_lock); - - if (have_waiters) - if (!SetEvent(cond->events[SIGNAL])) - THROWF(system_error, 0, "SetEvent failed"); - - xbt_os_thread_yield(); -} - -void xbt_os_cond_broadcast(xbt_os_cond_t cond) -{ - int have_waiters; - - EnterCriticalSection(&cond->waiters_count_lock); - have_waiters = cond->waiters_count > 0; - LeaveCriticalSection(&cond->waiters_count_lock); - - if (have_waiters) - SetEvent(cond->events[BROADCAST]); -} - -void xbt_os_cond_destroy(xbt_os_cond_t cond) -{ - int error = 0; - - if (!cond) - return; - - if (!CloseHandle(cond->events[SIGNAL])) - error = 1; - - if (!CloseHandle(cond->events[BROADCAST])) - error = 1; - - DeleteCriticalSection(&cond->waiters_count_lock); - - xbt_free(cond); - - if (error) - THROWF(system_error, 0, "Error while destroying the condition"); -} - -typedef struct xbt_os_sem_ { - HANDLE h; - unsigned int value; - CRITICAL_SECTION value_lock; /* protect access to value of the semaphore */ -} s_xbt_os_sem_t; - -#ifndef INT_MAX -# define INT_MAX 32767 /* let's be safe by underestimating this value: this is for 16bits only */ + return count; +#else + return sysconf(_SC_NPROCESSORS_ONLN); #endif - -xbt_os_sem_t xbt_os_sem_init(unsigned int value) -{ - xbt_os_sem_t res; - - if (value > INT_MAX) - THROWF(arg_error, value, - "Semaphore initial value too big: %ud cannot be stored as a signed int", - value); - - res = (xbt_os_sem_t) xbt_new0(s_xbt_os_sem_t, 1); - - if (!(res->h = CreateSemaphore(NULL, value, (long) INT_MAX, NULL))) { - THROWF(system_error, GetLastError(), "CreateSemaphore() failed: %s", - strerror(GetLastError())); - return NULL; - } - - res->value = value; - - InitializeCriticalSection(&(res->value_lock)); - - return res; -} - -void xbt_os_sem_acquire(xbt_os_sem_t sem) -{ - if (!sem) - THROWF(arg_error, EINVAL, "Cannot acquire the NULL semaphore"); - - /* wait failure */ - if (WAIT_OBJECT_0 != WaitForSingleObject(sem->h, INFINITE)) - THROWF(system_error, GetLastError(), - "WaitForSingleObject() failed: %s", strerror(GetLastError())); - EnterCriticalSection(&(sem->value_lock)); - sem->value--; - LeaveCriticalSection(&(sem->value_lock)); -} - -void xbt_os_sem_timedacquire(xbt_os_sem_t sem, double timeout) -{ - long seconds; - long milliseconds; - double end = timeout + xbt_os_time(); - - if (!sem) - THROWF(arg_error, EINVAL, "Cannot acquire the NULL semaphore"); - - if (timeout < 0) { - xbt_os_sem_acquire(sem); - } else { /* timeout can be zero <-> try acquire ) */ - - - seconds = (long) floor(end); - milliseconds = (long) ((end - seconds) * 1000); - milliseconds += (seconds * 1000); - - switch (WaitForSingleObject(sem->h, milliseconds)) { - case WAIT_OBJECT_0: - EnterCriticalSection(&(sem->value_lock)); - sem->value--; - LeaveCriticalSection(&(sem->value_lock)); - return; - - case WAIT_TIMEOUT: - THROWF(timeout_error, GetLastError(), - "semaphore %p wasn't signaled before timeout (%f)", sem, - timeout); - return; - - default: - THROWF(system_error, GetLastError(), - "WaitForSingleObject(%p,%f) failed: %s", sem, timeout, - strerror(GetLastError())); - } - } } -void xbt_os_sem_release(xbt_os_sem_t sem) -{ - if (!sem) - THROWF(arg_error, EINVAL, "Cannot release the NULL semaphore"); - - if (!ReleaseSemaphore(sem->h, 1, NULL)) - THROWF(system_error, GetLastError(), "ReleaseSemaphore() failed: %s", - strerror(GetLastError())); - EnterCriticalSection(&(sem->value_lock)); - sem->value++; - LeaveCriticalSection(&(sem->value_lock)); -} - -void xbt_os_sem_destroy(xbt_os_sem_t sem) -{ - if (!sem) - THROWF(arg_error, EINVAL, "Cannot destroy the NULL semaphore"); - - if (!CloseHandle(sem->h)) - THROWF(system_error, GetLastError(), "CloseHandle() failed: %s", - strerror(GetLastError())); - - DeleteCriticalSection(&(sem->value_lock)); - - xbt_free(sem); - -} - -void xbt_os_sem_get_value(xbt_os_sem_t sem, int *svalue) -{ - if (!sem) - THROWF(arg_error, EINVAL, - "Cannot get the value of the NULL semaphore"); - - EnterCriticalSection(&(sem->value_lock)); - *svalue = sem->value; - LeaveCriticalSection(&(sem->value_lock)); -} - - -#endif /***** reentrant mutexes *****/ typedef struct xbt_os_rmutex_ { @@ -1174,7 +657,11 @@ void xbt_os_thread_set_extra_data(void *data) void *xbt_os_thread_get_extra_data(void) { - return xbt_os_thread_self()->extra_data; + xbt_os_thread_t thread = xbt_os_thread_self(); + if (thread) + return xbt_os_thread_self()->extra_data; + else + return NULL; } xbt_os_rmutex_t xbt_os_rmutex_init(void)