X-Git-Url: https://pd.if.org/git/?p=nbds;a=blobdiff_plain;f=runtime%2Fruntime.c;h=19024e2f389f237ecbd6f0e2edf906b0b198a1d2;hp=14a09f9bb11f19b8a828ca1473c3681d0f9f3910;hb=HEAD;hpb=a03cf3b0c40e6c3b8b4877b49a64288cb3fcb919 diff --git a/runtime/runtime.c b/runtime/runtime.c index 14a09f9..19024e2 100644 --- a/runtime/runtime.c +++ b/runtime/runtime.c @@ -1,7 +1,8 @@ -/* +/* * Written by Josh Dybnis and released to the public domain, as explained at * http://creativecommons.org/licenses/publicdomain */ +#include #include #include "common.h" #include "runtime.h" @@ -9,48 +10,26 @@ #include "mem.h" #include "tls.h" -DECLARE_THREAD_LOCAL(tid_, int); -DECLARE_THREAD_LOCAL(rand_seed_, unsigned); +DECLARE_THREAD_LOCAL(ThreadId, int); +static int ThreadIndex -typedef struct thread_info { - int thread_id; - void *(*start_routine)(void *); - void *restrict arg; -} thread_info_t; +static int MaxThreadId = 0; __attribute__ ((constructor)) void nbd_init (void) { - sranddev(); - INIT_THREAD_LOCAL(rand_seed_); - INIT_THREAD_LOCAL(tid_); - SET_THREAD_LOCAL(tid_, 0); + rnd_init(); mem_init(); - lwt_thread_init(0); - rcu_thread_init(0); } -static void *worker (void *arg) { - thread_info_t *ti = (thread_info_t *)arg; - SET_THREAD_LOCAL(tid_, ti->thread_id); - LOCALIZE_THREAD_LOCAL(tid_, int); - SET_THREAD_LOCAL(rand_seed_, tid_+1); - lwt_thread_init(ti->thread_id); - rcu_thread_init(ti->thread_id); - void *ret = ti->start_routine(ti->arg); - nbd_free(ti); - return ret; -} +void nbd_thread_init (void) { + LOCALIZE_THREAD_LOCAL(ThreadId, int); -int nbd_thread_create (pthread_t *restrict thread, int thread_id, void *(*start_routine)(void *), void *restrict arg) { - thread_info_t *ti = (thread_info_t *)nbd_malloc(sizeof(thread_info_t)); - ti->thread_id = thread_id; - ti->start_routine = start_routine; - ti->arg = arg; - return pthread_create(thread, NULL, worker, ti); -} + if (ThreadId == 0) { + ++MaxThreadId; // TODO: reuse thread id's of threads that have been destroyed + ASSERT(MaxThreadId <= MAX_NUM_THREADS); + SET_THREAD_LOCAL(ThreadId, MaxThreadId); + rnd_thread_init(); + } -int nbd_rand (void) { - LOCALIZE_THREAD_LOCAL(rand_seed_, unsigned); - unsigned r = rand_r(&rand_seed_); - SET_THREAD_LOCAL(rand_seed_, r); - return r; + lwt_thread_init(); + rcu_thread_init(); }