X-Git-Url: https://pd.if.org/git/?p=nbds;a=blobdiff_plain;f=runtime%2Fruntime.c;h=ceb67729f79ba7e7518736cf9d6812607630520f;hp=8056660694888b4b596e66a0c0c01655e2ac01ce;hb=a1d0b3ca99552878b1becf561d8f3291992aaa67;hpb=fb536c12185fd1e339b5fd479d9ef84554b436df

diff --git a/runtime/runtime.c b/runtime/runtime.c
index 8056660..ceb6772 100644
--- a/runtime/runtime.c
+++ b/runtime/runtime.c
@@ -2,14 +2,17 @@
  * 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 <stdlib.h>
 #include <pthread.h>
 #include "common.h"
 #include "runtime.h"
-#include "runtime_local.h"
+#include "rlocal.h"
 #include "mem.h"
 #include "tls.h"
 
 DECLARE_THREAD_LOCAL(tid_, int);
+DECLARE_THREAD_LOCAL(rand_seed_, unsigned);
 
 typedef struct thread_info {
     int thread_id;
@@ -17,7 +20,9 @@ typedef struct thread_info {
     void *restrict arg;
 } thread_info_t;
 
-void nbd_init (void) {
+__attribute__ ((constructor)) void nbd_init (void) {
+    //sranddev();
+    INIT_THREAD_LOCAL(rand_seed_);
     INIT_THREAD_LOCAL(tid_);
     SET_THREAD_LOCAL(tid_, 0);
     mem_init();
@@ -28,6 +33,8 @@ void nbd_init (void) {
 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);
@@ -42,3 +49,10 @@ int nbd_thread_create (pthread_t *restrict thread, int thread_id, void *(*start_
     ti->arg = arg;
     return pthread_create(thread, NULL, worker, ti);
 }
+
+int nbd_rand (void) {
+    LOCALIZE_THREAD_LOCAL(rand_seed_, unsigned);
+    unsigned r = rand_r(&rand_seed_);
+    SET_THREAD_LOCAL(rand_seed_, r);
+    return r;
+}