michael@0: /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "nspr.h"
michael@0: #include "prinrval.h"
michael@0: #include "plgetopt.h"
michael@0: #include "pprthred.h"
michael@0: #include <stdio.h>
michael@0: #include <stdlib.h>
michael@0: #include <string.h>
michael@0: 
michael@0: PRMonitor *mon;
michael@0: PRInt32 count, iterations, alive;
michael@0: 
michael@0: PRBool debug_mode = PR_FALSE, passed = PR_TRUE;
michael@0: 
michael@0: void 
michael@0: PR_CALLBACK
michael@0: ReallyDumbThread(void *arg)
michael@0: {
michael@0:     return;
michael@0: }
michael@0: 
michael@0: void
michael@0: PR_CALLBACK
michael@0: DumbThread(void *arg)
michael@0: {
michael@0:     PRInt32 tmp = (PRInt32)arg;
michael@0:     PRThreadScope scope = (PRThreadScope)tmp;
michael@0:     PRThread *thr;
michael@0: 
michael@0:     thr = PR_CreateThread(PR_USER_THREAD,
michael@0:                           ReallyDumbThread,
michael@0:                           NULL,
michael@0:                           PR_PRIORITY_NORMAL,
michael@0:                           scope,
michael@0:                           PR_JOINABLE_THREAD,
michael@0:                           0);
michael@0: 
michael@0:     if (!thr) {
michael@0:         if (debug_mode) {
michael@0:             printf("Could not create really dumb thread (%d, %d)!\n",
michael@0:                     PR_GetError(), PR_GetOSError());
michael@0:         }
michael@0:         passed = PR_FALSE;
michael@0:     } else {
michael@0:         PR_JoinThread(thr);
michael@0:     }
michael@0:     PR_EnterMonitor(mon);
michael@0:     alive--;
michael@0:     PR_Notify(mon);
michael@0:     PR_ExitMonitor(mon);
michael@0: }
michael@0: 
michael@0: static void CreateThreads(PRThreadScope scope1, PRThreadScope scope2)
michael@0: {
michael@0:     PRThread *thr;
michael@0:     int n;
michael@0: 
michael@0:     alive = 0;
michael@0:     mon = PR_NewMonitor();
michael@0: 
michael@0:     alive = count;
michael@0:     for (n=0; n<count; n++) {
michael@0:         thr = PR_CreateThread(PR_USER_THREAD,
michael@0:                               DumbThread, 
michael@0:                               (void *)scope2, 
michael@0:                               PR_PRIORITY_NORMAL,
michael@0:                               scope1,
michael@0:                               PR_UNJOINABLE_THREAD,
michael@0:                               0);
michael@0:         if (!thr) {
michael@0:             if (debug_mode) {
michael@0:                 printf("Could not create dumb thread (%d, %d)!\n",
michael@0:                         PR_GetError(), PR_GetOSError());
michael@0:             }
michael@0:             passed = PR_FALSE;
michael@0:             alive--;
michael@0:         }
michael@0:          
michael@0:         PR_Sleep(0);
michael@0:     }
michael@0: 
michael@0:     /* Wait for all threads to exit */
michael@0:     PR_EnterMonitor(mon);
michael@0:     while (alive) {
michael@0:         PR_Wait(mon, PR_INTERVAL_NO_TIMEOUT);
michael@0:     }
michael@0: 
michael@0:     PR_ExitMonitor(mon);
michael@0: 	PR_DestroyMonitor(mon);
michael@0: }
michael@0: 
michael@0: static void CreateThreadsUU(void)
michael@0: {
michael@0:     CreateThreads(PR_LOCAL_THREAD, PR_LOCAL_THREAD);
michael@0: }
michael@0: 
michael@0: static void CreateThreadsUK(void)
michael@0: {
michael@0:     CreateThreads(PR_LOCAL_THREAD, PR_GLOBAL_THREAD);
michael@0: }
michael@0: 
michael@0: static void CreateThreadsKU(void)
michael@0: {
michael@0:     CreateThreads(PR_GLOBAL_THREAD, PR_LOCAL_THREAD);
michael@0: }
michael@0: 
michael@0: static void CreateThreadsKK(void)
michael@0: {
michael@0:     CreateThreads(PR_GLOBAL_THREAD, PR_GLOBAL_THREAD);
michael@0: }
michael@0: 
michael@0: /************************************************************************/
michael@0: 
michael@0: static void Measure(void (*func)(void), const char *msg)
michael@0: {
michael@0:     PRIntervalTime start, stop;
michael@0:     double d;
michael@0: 
michael@0:     start = PR_IntervalNow();
michael@0:     (*func)();
michael@0:     stop = PR_IntervalNow();
michael@0: 
michael@0:     if (debug_mode)
michael@0:     {
michael@0:         d = (double)PR_IntervalToMicroseconds(stop - start);
michael@0:         printf("%40s: %6.2f usec\n", msg, d / count);
michael@0:     }
michael@0: }
michael@0: 
michael@0: int main(int argc, char **argv)
michael@0: {
michael@0:     int index;
michael@0: 
michael@0:     PR_STDIO_INIT();
michael@0:     PR_Init(PR_USER_THREAD, PR_PRIORITY_HIGH, 0);
michael@0:     
michael@0:     {
michael@0:     	PLOptStatus os;
michael@0:     	PLOptState *opt = PL_CreateOptState(argc, argv, "dc:i:");
michael@0:     	while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
michael@0:         {
michael@0:     		if (PL_OPT_BAD == os) continue;
michael@0:             switch (opt->option)
michael@0:             {
michael@0:             case 'd':  /* debug mode */
michael@0:     			debug_mode = PR_TRUE;
michael@0:                 break;
michael@0:             case 'c':  /* loop counter */
michael@0:     			count = atoi(opt->value);
michael@0:                 break;
michael@0:             case 'i':  /* loop counter */
michael@0:     			iterations = atoi(opt->value);
michael@0:                 break;
michael@0:              default:
michael@0:                 break;
michael@0:             }
michael@0:         }
michael@0:     	PL_DestroyOptState(opt);
michael@0:     }
michael@0: 
michael@0:     if (0 == count) count = 50;
michael@0:     if (0 == iterations) iterations = 10;
michael@0: 
michael@0:     if (debug_mode)
michael@0:     {
michael@0:     printf("\
michael@0: ** Tests lots of thread creations.  \n\
michael@0: ** Create %ld native threads %ld times. \n\
michael@0: ** Create %ld user threads %ld times \n", iterations,count,iterations,count);
michael@0:     }
michael@0: 
michael@0:     for (index=0; index<iterations; index++) {
michael@0:         Measure(CreateThreadsUU, "Create user/user threads");
michael@0:         Measure(CreateThreadsUK, "Create user/native threads");
michael@0:         Measure(CreateThreadsKU, "Create native/user threads");
michael@0:         Measure(CreateThreadsKK, "Create native/native threads");
michael@0:     }
michael@0: 
michael@0:     if (debug_mode) printf("\nNow switch to recycling threads \n\n");
michael@0:     PR_SetThreadRecycleMode(1);
michael@0: 
michael@0:     for (index=0; index<iterations; index++) {
michael@0:         Measure(CreateThreadsUU, "Create user/user threads");
michael@0:         Measure(CreateThreadsUK, "Create user/native threads");
michael@0:         Measure(CreateThreadsKU, "Create native/user threads");
michael@0:         Measure(CreateThreadsKK, "Create native/native threads");
michael@0:     }
michael@0: 
michael@0: 
michael@0:     printf("%s\n", ((passed) ? "PASS" : "FAIL"));
michael@0: 
michael@0:     PR_Cleanup();
michael@0: 
michael@0:     if (passed) {
michael@0:         return 0;
michael@0:     } else {
michael@0:         return 1;
michael@0:     }
michael@0: }