X-Git-Url: https://pd.if.org/git/?p=nbds;a=blobdiff_plain;f=runtime%2Fruntime.c;h=19024e2f389f237ecbd6f0e2edf906b0b198a1d2;hp=415a1617816f18133728538b7059bf1b89ee9980;hb=HEAD;hpb=7d658a03f83e64690d0c71b4733dd3f9a2c60208 diff --git a/runtime/runtime.c b/runtime/runtime.c index 415a161..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,62 +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); -#ifndef NDEBUG - SET_THREAD_LOCAL(rand_seed_, tid_+1); -#else - SET_THREAD_LOCAL(rand_seed_, nbd_rand_seed(tid_+1)); -#endif - lwt_thread_init(ti->thread_id); - rcu_thread_init(ti->thread_id); - void *ret = ti->start_routine(ti->arg); - nbd_free(ti); - return ret; -} - -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); -} +void nbd_thread_init (void) { + LOCALIZE_THREAD_LOCAL(ThreadId, int); -int nbd_rand (void) { - LOCALIZE_THREAD_LOCAL(rand_seed_, unsigned); - unsigned r = rand_r(&rand_seed_); - SET_THREAD_LOCAL(rand_seed_, r); - return r; -} - -uint64_t nbd_rand_seed (int i) { - return rdtsc() + -715159705 + i * 129; -} + 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(); + } -// Fairly fast random numbers -int nbd_next_rand (uint64_t *r) { - *r = (*r * 0x5DEECE66DLL + 0xBLL) & MASK(48); - return (*r >> 17) & 0x7FFFFFFF; + lwt_thread_init(); + rcu_thread_init(); }