tests/test_nobug_threadstress.c

   1 #define TESTTHREADS 200
   2
   3 #include "test.h"
   4 #include "nobug.h"
   5
   6 #include <stdlib.h>
   7 #include <unistd.h>
   8 #include <stdint.h>
   9 #include <pthread.h>
  10
  11 struct testthread
  12 {
  13   pthread_t thread;
  14   NOBUG_RESOURCE_HANDLE(rh);
  15 };
  16
  17 void*
  18 threadfn(void* nop)
  19 {
  20   NOBUG_THREAD_ID_SET("thread");
  21
  22   NOBUG_RESOURCE_USER(rh);
  23   NOBUG_RESOURCE_USER_INIT(rh);
  24   NOBUG_RESOURCE_ENTER(NOBUG_ON, ((struct testthread*)nop)->rh, "threadfn", NOBUG_RESOURCE_WAITING, rh) { }
  25
  26   ECHO("starting");
  27
  28   unsigned int seed = (uintptr_t)nop;
  29
  30
  31   int r1 = (rand_r(&seed)%100)*1000;
  32   if (r1>20000)
  33     {
  34       ECHO("first sleeping for %d usec", r1);
  35       usleep(r1);
  36     }
  37   else
  38     {
  39       ECHO("first busy for %d iterations", r1*500);
  40       for (volatile int i=0; i<r1*500; ++i)
  41         {
  42           volatile int j = i*i;
  43           j /= 2;
  44         }
  45     }
  46
  47   NOBUG_RESOURCE_STATE (NOBUG_ON, NOBUG_RESOURCE_EXCLUSIVE, rh) {}
  48
  49   int r2 = (rand_r(&seed)%100)*1000;
  50   if (r2>20000)
  51     {
  52       ECHO("second sleeping for %d usec", r2);
  53       usleep(r2);
  54     }
  55   else
  56     {
  57       ECHO("second busy for %d iterations", r2*500);
  58       for (volatile int i=0; i<r2*500; ++i)
  59         {
  60           volatile int j = i*i;
  61           j /= 2;
  62         }
  63     }
  64
  65   int r3 = (rand_r(&seed)%100)*1000;
  66   if (r3>20000)
  67     {
  68       ECHO("third sleeping for %d usec", r3);
  69       usleep(r3);
  70     }
  71   else
  72     {
  73       ECHO("third busy for %d iterations", r3*500);
  74       for (volatile int i=0; i<r3*500; ++i)
  75         {
  76           volatile int j = i*i;
  77           j /= 2;
  78         }
  79     }
  80
  81   ECHO("ending");
  82
  83   NOBUG_RESOURCE_LEAVE(NOBUG_ON, rh) { }
  84
  85   return nop;
  86 }
  87
  88
  89 TESTS_BEGIN
  90 NOBUG_THREAD_ID_SET("main");
  91 ECHO ("testing");
  92
  93
  94 TEST(many_threads)
  95 {
  96   struct testthread threads[TESTTHREADS];
  97
  98   for (int i = 0; i<TESTTHREADS; ++i)
  99     {
 100       NOBUG_RESOURCE_HANDLE_INIT(threads[i].rh);
 101       NOBUG_RESOURCE_ANNOUNCE(NOBUG_ON, "thread", "test", &threads[i], threads[i].rh)
 102         {
 103           if (!pthread_create(&threads[i].thread, NULL, threadfn, &threads[i]))
 104             {
 105               ECHO("created thread");
 106             }
 107           else
 108             {
 109               ECHO("thread created failed");
 110               exit(1);
 111             }
 112         }
 113     }
 114
 115   for (int i = 0; i<TESTTHREADS; ++i)
 116     {
 117       void* ret;
 118
 119       NOBUG_RESOURCE_FORGET(NOBUG_ON, threads[i].rh)
 120         {
 121           if (!pthread_join(threads[i].thread, &ret))
 122             {
 123               ECHO("thread joined");
 124               CHECK (ret == &threads[i]);
 125             }
 126           else
 127             {
 128               ECHO("thread join failed");
 129               exit(1);
 130             }
 131         }
 132     }
 133 }
 134
 135
 136
 137 TESTS_END