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: /***********************************************************************
michael@0: **
michael@0: ** Name: sem.c
michael@0: **
michael@0: ** Description: Tests Semaphonre functions.
michael@0: **
michael@0: ** Modification History:
michael@0: ** 20-May-97 AGarcia- Converted the test to accomodate the debug_mode flag.
michael@0: **	         The debug mode will print all of the printfs associated with this test.
michael@0: **			 The regress mode will be the default mode. Since the regress tool limits
michael@0: **           the output to a one line status:PASS or FAIL,all of the printf statements
michael@0: **			 have been handled with an if (debug_mode) statement.
michael@0: ** 04-June-97 AGarcia removed the Test_Result function. Regress tool has been updated to
michael@0: **			recognize the return code from tha main program.
michael@0: ***********************************************************************/
michael@0: 
michael@0: /***********************************************************************
michael@0: ** Includes
michael@0: ***********************************************************************/
michael@0: /* Used to get the command line option */
michael@0: #include "plgetopt.h"
michael@0: 
michael@0: #include "nspr.h"
michael@0: #include "prpriv.h"
michael@0: 
michael@0: #include <stdio.h>
michael@0: #include <stdlib.h>
michael@0: #include <string.h>
michael@0: 
michael@0: PRIntn failed_already=0;
michael@0: PRIntn debug_mode;
michael@0: 
michael@0: /* 
michael@0: 	Since we don't have stdin, stdout everywhere, we will fake 
michael@0: 	it with our in-memory buffers called stdin and stdout.
michael@0: */
michael@0: 
michael@0: #define SBSIZE 1024
michael@0: 
michael@0: #include "obsolete/prsem.h"
michael@0: 
michael@0: static char stdinBuf[SBSIZE];
michael@0: static char stdoutBuf[SBSIZE];
michael@0: 
michael@0: static PRUintn stdinBufIdx = 0;
michael@0: static PRUintn stdoutBufIdx = 0;
michael@0: static PRStatus finalResult = PR_SUCCESS;
michael@0: 
michael@0: 
michael@0: static size_t dread (PRUintn device, char *buf, size_t bufSize)
michael@0: {
michael@0: 	PRUintn	i;
michael@0: 	
michael@0: 	/* during first read call, initialize the stdinBuf buffer*/
michael@0: 	if (stdinBufIdx == 0) {
michael@0: 		for (i=0; i<SBSIZE; i++)
michael@0: 			stdinBuf[i] = i;
michael@0: 	}
michael@0: 
michael@0: 	/* now copy data from stdinBuf to the given buffer upto bufSize */
michael@0: 	for (i=0; i<bufSize; i++) {
michael@0: 		if (stdinBufIdx == SBSIZE)
michael@0: 			break;
michael@0: 		buf[i] = stdinBuf[stdinBufIdx++];
michael@0: 	}
michael@0: 
michael@0: 	return i;
michael@0: }
michael@0: 
michael@0: static size_t dwrite (PRUintn device, char *buf, size_t bufSize)
michael@0: {
michael@0: 	PRUintn	i, j;
michael@0: 	
michael@0: 	/* copy data from the given buffer upto bufSize to stdoutBuf */
michael@0: 	for (i=0; i<bufSize; i++) {
michael@0: 		if (stdoutBufIdx == SBSIZE)
michael@0: 			break;
michael@0: 		stdoutBuf[stdoutBufIdx++] = buf[i];
michael@0: 	}
michael@0: 
michael@0: 	/* during last write call, compare the two buffers */
michael@0: 	if (stdoutBufIdx == SBSIZE)
michael@0: 		for (j=0; j<SBSIZE; j++)
michael@0: 			if (stdinBuf[j] != stdoutBuf[j]) {
michael@0: 				if (debug_mode) printf("data mismatch for index= %d \n", j);
michael@0: 				finalResult = PR_FAILURE;
michael@0: 			}
michael@0: 
michael@0: 	return i;
michael@0: }
michael@0: 
michael@0: /*------------------ Following is the real test program ---------*/
michael@0: /*
michael@0: 	Program to copy standard input to standard output.  The program
michael@0: 	uses two threads.  One reads the input and puts the data in a 
michael@0: 	double buffer.  The other reads the buffer contents and writes 
michael@0: 	it to standard output.
michael@0: */
michael@0: 
michael@0: PRSemaphore	*emptyBufs;	/* number of empty buffers */
michael@0: PRSemaphore *fullBufs;	/* number of buffers that are full */
michael@0: 
michael@0: #define BSIZE	100
michael@0: 
michael@0: struct {
michael@0: 	char data[BSIZE];
michael@0: 	PRUintn nbytes;		/* number of bytes in this buffer */
michael@0: } buf[2];
michael@0: 
michael@0: static void PR_CALLBACK reader(void *arg)
michael@0: {
michael@0: 	PRUintn	i = 0;
michael@0: 	size_t	nbytes;
michael@0: 	
michael@0: 	do {
michael@0: 		(void) PR_WaitSem(emptyBufs);
michael@0: 		nbytes = dread(0, buf[i].data, BSIZE);
michael@0: 		buf[i].nbytes = nbytes;
michael@0: 		PR_PostSem(fullBufs);
michael@0: 		i = (i + 1) % 2;
michael@0: 	} while (nbytes > 0);
michael@0: }
michael@0: 
michael@0: static void writer(void)
michael@0: {
michael@0: 	PRUintn	i = 0;
michael@0: 	size_t	nbytes;
michael@0: 	
michael@0: 	do {
michael@0: 		(void) PR_WaitSem(fullBufs);
michael@0: 		nbytes = buf[i].nbytes;
michael@0: 		if (nbytes > 0) {
michael@0: 			nbytes = dwrite(1, buf[i].data, nbytes);
michael@0: 			PR_PostSem(emptyBufs);
michael@0: 			i = (i + 1) % 2;
michael@0: 		}
michael@0: 	} while (nbytes > 0);
michael@0: }
michael@0: 
michael@0: int main(int argc, char **argv)
michael@0: {
michael@0: 	PRThread *r;
michael@0: 
michael@0:     PR_STDIO_INIT();
michael@0:     PR_Init(PR_USER_THREAD, PR_PRIORITY_NORMAL, 0);
michael@0: 
michael@0:     {
michael@0:     	/* The command line argument: -d is used to determine if the test is being run
michael@0:     	in debug mode. The regress tool requires only one line output:PASS or FAIL.
michael@0:     	All of the printfs associated with this test has been handled with a if (debug_mode)
michael@0:     	test.
michael@0:     	Usage: test_name -d
michael@0:     	*/
michael@0:     	PLOptStatus os;
michael@0:     	PLOptState *opt = PL_CreateOptState(argc, argv, "d:");
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 = 1;
michael@0:                 break;
michael@0:              default:
michael@0:                 break;
michael@0:             }
michael@0:         }
michael@0:     	PL_DestroyOptState(opt);
michael@0:     }        
michael@0: 
michael@0:  /* main test */
michael@0: 
michael@0:     emptyBufs = PR_NewSem(2);	/* two empty buffers */
michael@0: 
michael@0:     fullBufs = PR_NewSem(0);	/* zero full buffers */
michael@0: 
michael@0: 	/* create the reader thread */
michael@0: 	
michael@0: 	r = PR_CreateThread(PR_USER_THREAD,
michael@0: 				      reader, 0, 
michael@0: 				      PR_PRIORITY_NORMAL,
michael@0: 				      PR_LOCAL_THREAD,
michael@0:     				  PR_UNJOINABLE_THREAD,
michael@0: 				      0);
michael@0: 
michael@0: 	/* Do the writer operation in this thread */
michael@0: 	writer();
michael@0: 
michael@0: 	PR_DestroySem(emptyBufs);
michael@0: 	PR_DestroySem(fullBufs);
michael@0: 
michael@0: 	if (finalResult == PR_SUCCESS) {
michael@0: 		if (debug_mode) printf("sem Test Passed.\n");
michael@0: 	}
michael@0: 	else{
michael@0: 		if (debug_mode) printf("sem Test Failed.\n");
michael@0: 		failed_already=1;
michael@0: 	}
michael@0:     PR_Cleanup();
michael@0: 	if(failed_already)	
michael@0: 		return 1;
michael@0: 	else
michael@0: 		return 0;
michael@0: }