X-Git-Url: https://pd.if.org/git/?p=nbds;a=blobdiff_plain;f=runtime%2Fruntime.c;h=19024e2f389f237ecbd6f0e2edf906b0b198a1d2;hp=b145faa872e5826ed70a979d724a561cdc24738a;hb=HEAD;hpb=0f6e9afb06b03647c4c5f2499ddab12f42b4340e diff --git a/runtime/runtime.c b/runtime/runtime.c index b145faa..19024e2 100644 --- a/runtime/runtime.c +++ b/runtime/runtime.c @@ -1,8 +1,7 @@ -/* +/* * Written by Josh Dybnis and released to the public domain, as explained at * http://creativecommons.org/licenses/publicdomain */ -#define _POSIX_C_SOURCE 1 // for rand_r() #include #include #include "common.h" @@ -11,47 +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(102))) void nbd_init (void) { - //sranddev(); - INIT_THREAD_LOCAL(rand_seed_); - INIT_THREAD_LOCAL(tid_); - SET_THREAD_LOCAL(tid_, 0); - lwt_thread_init(0); - rcu_thread_init(0); +__attribute__ ((constructor)) void nbd_init (void) { + rnd_init(); + mem_init(); } -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(); }