1 /* Copyright © 1998 Netscape Communications Corporation */
  2 
  3 #include "prinit.h"
  4 #include "prlock.h"
  5 #include "prcvar.h"
  6 #include "prmem.h"
  7 #include "prinrval.h"
  8 #include "prlog.h"
  9 #include "prthread.h"
 10 #include "prprf.h"
 11 #include "plerror.h"
 12 
 13 #define INNER_LOOPS 100
 14 #define DEFAULT_LOOPS 100
 15 #define DEFAULT_THREADS 10
 16 
 17 typedef struct Shared
 18 {
 19     PRLock *ml;
 20     PRCondVar *cv;
 21     PRBool twiddle;
 22     PRThread *thread;
 23     struct Shared *next;
 24 } Shared;
 25 
 26 static PRFileDesc *debug_out = NULL;
 27 static PRBool debug_mode = PR_FALSE, verbosity = PR_TRUE, failed = PR_FALSE;
 28 static Shared home;
 29 
 30 static void Help(void);
 31 static void PR_CALLBACK Notified(void *arg);
 32 static PRIntn PR_CALLBACK Switch(PRIntn argc, char **argv);
 33 
 34 /* Root function for the threads in this sample (except primordial thread, which runs main) */
 35 static void PR_CALLBACK Notified(void *arg)
 36 {
 37     Shared *shared = arg;
 38     PRStatus status = PR_SUCCESS;
 39     while (PR_SUCCESS == status)
 40     {
 41         PR_Lock(shared->ml);
 42         while (shared->twiddle && (PR_SUCCESS == status))
 43             status = PR_WaitCondVar(shared->cv, PR_INTERVAL_NO_TIMEOUT);
 44         if (verbosity) PR_fprintf(debug_out, "+");
 45         shared->twiddle = PR_TRUE;
 46         shared->next->twiddle = PR_FALSE;
 47         PR_NotifyCondVar(shared->next->cv);
 48         PR_Unlock(shared->ml);
 49     }
 50 }  /* Notified */
 51 
 52 static PRIntn PR_CALLBACK Switch(PRIntn argc, char **argv)
 53 {
 54     PRStatus status;
 55     PRBool help = PR_FALSE;
 56     PRUintn concurrency = 1;
 57     Shared *shared, *link;
 58     PRIntervalTime timein, timeout;
 59     PRBool global_threads = PR_FALSE;
 60     PRThreadScope thread_scope = PR_LOCAL_THREAD;
 61     PRUintn thread_count, inner_count, loop_count, average;
 62     PRUintn thread_limit = DEFAULT_THREADS, loop_limit = DEFAULT_LOOPS;
 63    
 64     if (help) return -1;
 65 
 66     if (PR_TRUE == debug_mode)
 67     {
 68         debug_out = PR_STDOUT;
 69         PR_fprintf(debug_out, "Test parameters\n");
 70         PR_fprintf(debug_out, "\tThreads involved: %d\n", thread_limit);
 71         PR_fprintf(debug_out, "\tIteration limit: %d\n", loop_limit);
 72         PR_fprintf(debug_out, "\tConcurrency: %d\n", concurrency);
 73         PR_fprintf(
 74             debug_out, "\tThread type: %s\n",
 75             (PR_GLOBAL_THREAD == thread_scope) ? "GLOBAL" : "LOCAL");
 76     }
 77 
 78     PR_SetConcurrency(concurrency);
 79 
 80 /*'home' is "Shared" object for the primordial thread, at the end of the chain. */
 81     link = &home;
 82     home.ml = PR_NewLock();
 83     home.cv = PR_NewCondVar(home.ml);
 84     home.twiddle = PR_TRUE;
 85     home.next = NULL;
 86     timeout = 0;
 87 
 88 /* Create "thread_limit" number of additional threads, each with its own "Shared" object. */
 89     for (thread_count = 1; thread_count <= thread_limit; ++thread_count)
 90     {
 91         shared = PR_NEWZAP(Shared);
 92         shared->ml = home.ml;
 93         shared->cv = PR_NewCondVar(home.ml);
 94         shared->twiddle = PR_TRUE;
 95         shared->next = link;
 96         link = shared;
 97         shared->thread = PR_CreateThread(
 98             PR_USER_THREAD, Notified, shared,
 99             PR_PRIORITY_HIGH, thread_scope,
100             PR_JOINABLE_THREAD, 0);
101         PR_ASSERT(shared->thread != NULL);
102         if (NULL == shared->thread)
103             failed = PR_TRUE;
104     }
105 
106 /* "Shared" now points to the head of the chain, and "home" is the tail of the chain. */
107     for (loop_count = 1; loop_count <= loop_limit; ++loop_count)
108     {
109         timein = PR_IntervalNow();
110         for (inner_count = 0; inner_count < INNER_LOOPS; ++inner_count)
111         {
112             PR_Lock(home.ml);
113             home.twiddle = PR_TRUE;
114             shared->twiddle = PR_FALSE;
115             PR_NotifyCondVar(shared->cv);
116             while (home.twiddle)
117             {
118                 status = PR_WaitCondVar(home.cv, PR_INTERVAL_NO_TIMEOUT);
119                 if (PR_FAILURE == status)
120                     failed = PR_TRUE;
121             }
122             PR_Unlock(home.ml);
123         }
124         timeout += (PR_IntervalNow() - timein);
125     }
126 
127     if (debug_mode)
128     {
129         average = PR_IntervalToMicroseconds(timeout)
130             / (INNER_LOOPS * loop_limit * thread_count);
131         PR_fprintf(
132             debug_out, "Average switch times %d usecs for %d threads\n",
133             average, thread_limit);
134     }
135     /* Test completed. Remainder of sample cleanly shuts down the test. */
136     link = shared;
137     for (thread_count = 1; thread_count <= thread_limit; ++thread_count)
138     {
139         if (&home == link) break;
140         status = PR_Interrupt(link->thread);
141         if (PR_SUCCESS != status)
142         {
143             failed = PR_TRUE;
144             if (debug_mode)
145                 PL_FPrintError(debug_out, "Failed to interrupt");
146         }
147         link = link->next; 
148     }
149 
150     for (thread_count = 1; thread_count <= thread_limit; ++thread_count)
151     {
152         link = shared->next;
153         status = PR_JoinThread(shared->thread);
154         if (PR_SUCCESS != status)
155         {
156             failed = PR_TRUE;
157             if (debug_mode)
158                 PL_FPrintError(debug_out, "Failed to join");
159         }
160         PR_DestroyCondVar(shared->cv);
161         PR_DELETE(shared);
162         if (&home == link) break;
163         shared = link;
164     }
165     PR_DestroyCondVar(home.cv);
166     PR_DestroyLock(home.ml);
167 
168     PR_fprintf(PR_STDOUT, ((failed) ? "FAILED\n" : "PASSED\n"));
169     return ((failed) ? 1 : 0);
170 }  /* Switch */
171 
172 PRIntn main(PRIntn argc, char **argv)
173 {
174     PRIntn result;
175     PR_STDIO_INIT();
176     result = PR_Initialize(Switch, argc, argv, 0);
177     return result;
178 }  /* main */
179 
180 /* switch.c */