nspr-4.8.6/mozilla/nsprpub/pr/tests/io_timeout.c - nest-learning-thermostat/5.1.5/nspr - Git at Google

 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
 /* ***** BEGIN LICENSE BLOCK *****
  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
  *
  * The contents of this file are subject to the Mozilla Public License Version
  * 1.1 (the "License"); you may not use this file except in compliance with
  * the License. You may obtain a copy of the License at
  * http://www.mozilla.org/MPL/
  *
  * Software distributed under the License is distributed on an "AS IS" basis,
  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  * for the specific language governing rights and limitations under the
  * License.
  *
  * The Original Code is the Netscape Portable Runtime (NSPR).
  *
  * The Initial Developer of the Original Code is
  * Netscape Communications Corporation.
  * Portions created by the Initial Developer are Copyright (C) 1998-2000
  * the Initial Developer. All Rights Reserved.
  *
  * Contributor(s):
  *
  * Alternatively, the contents of this file may be used under the terms of
  * either the GNU General Public License Version 2 or later (the "GPL"), or
  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  * in which case the provisions of the GPL or the LGPL are applicable instead
  * of those above. If you wish to allow use of your version of this file only
  * under the terms of either the GPL or the LGPL, and not to allow others to
  * use your version of this file under the terms of the MPL, indicate your
  * decision by deleting the provisions above and replace them with the notice
  * and other provisions required by the GPL or the LGPL. If you do not delete
  * the provisions above, a recipient may use your version of this file under
  * the terms of any one of the MPL, the GPL or the LGPL.
  *
  * ***** END LICENSE BLOCK ***** */

 /*
 ** Test socket IO timeouts
 **
 **
 **
 **
 ** Modification History:
 ** 14-May-97 AGarcia- Converted the test to accomodate the debug_mode flag.
 **	         The debug mode will print all of the printfs associated with this test.
 **			 The regress mode will be the default mode. Since the regress tool limits
 **           the output to a one line status:PASS or FAIL,all of the printf statements
 **			 have been handled with an if (debug_mode) statement.
 ***********************************************************************/
 /***********************************************************************
 ** Includes
 ***********************************************************************/
 /* Used to get the command line option */
 #include "plgetopt.h"

 #include <stdio.h>
 #include "nspr.h"

 #define NUM_THREADS 1
 #define BASE_PORT   8000
 #define DEFAULT_ACCEPT_TIMEOUT 2

 typedef struct threadInfo {
     PRInt16 id;
     PRInt16 accept_timeout;
     PRLock *dead_lock;
     PRCondVar *dead_cv;
     PRInt32   *alive;
 } threadInfo;

 PRIntn failed_already = 0;
 PRIntn debug_mode = 0;

 #define	LOCAL_SCOPE_STRING			"LOCAL scope"
 #define	GLOBAL_SCOPE_STRING			"GLOBAL scope"
 #define	GLOBAL_BOUND_SCOPE_STRING	"GLOBAL_BOUND scope"

 void
 thread_main(void *_info)
 {
     threadInfo *info = (threadInfo *)_info;
     PRNetAddr listenAddr;
     PRNetAddr clientAddr;
     PRFileDesc *listenSock = NULL;
     PRFileDesc *clientSock;
     PRStatus rv;
 	PRThreadScope tscope;
 	char *scope_str;


 	if (debug_mode)
     	printf("thread %d is alive\n", info->id);
 	tscope = PR_GetThreadScope(PR_GetCurrentThread());

 	switch(tscope) {
 		case PR_LOCAL_THREAD:
 			scope_str = LOCAL_SCOPE_STRING;
 			break;
 		case PR_GLOBAL_THREAD:
 			scope_str = GLOBAL_SCOPE_STRING;
 			break;
 		case PR_GLOBAL_BOUND_THREAD:
 			scope_str = GLOBAL_BOUND_SCOPE_STRING;
 			break;
 		default:
 			PR_ASSERT(!"Invalid thread scope");
 			break;
 	}
 	printf("thread id %d, scope %s\n", info->id, scope_str);

     listenSock = PR_NewTCPSocket();
     if (!listenSock) {
 		if (debug_mode)
         	printf("unable to create listen socket\n");
 		failed_already=1;
         goto dead;
     }

     listenAddr.inet.family = PR_AF_INET;
     listenAddr.inet.port = PR_htons(BASE_PORT + info->id);
     listenAddr.inet.ip = PR_htonl(PR_INADDR_ANY);
     rv = PR_Bind(listenSock, &listenAddr);
     if (rv == PR_FAILURE) {
 		if (debug_mode)
         	printf("unable to bind\n");
 		failed_already=1;
         goto dead;
     }

     rv = PR_Listen(listenSock, 4);
     if (rv == PR_FAILURE) {
 		if (debug_mode)
         	printf("unable to listen\n");
 		failed_already=1;
         goto dead;
     }

 	if (debug_mode)
     	printf("thread %d going into accept for %d seconds\n",
         	info->id, info->accept_timeout + info->id);

     clientSock = PR_Accept(listenSock, &clientAddr, PR_SecondsToInterval(info->accept_timeout +info->id));

     if (clientSock == NULL) {
         if (PR_GetError() == PR_IO_TIMEOUT_ERROR) {
 			if (debug_mode) {
             	printf("PR_Accept() timeout worked!\n");
 				printf("TEST PASSED! PR_Accept() returned error %d\n",
 							PR_IO_TIMEOUT_ERROR);
 			}
     	} else {
 			if (debug_mode)
             	printf("TEST FAILED! PR_Accept() returned error %d\n",
 														PR_GetError());
 			failed_already=1;
 		}
     } else {
 		if (debug_mode)
         	printf ("TEST FAILED! PR_Accept() succeeded?\n");
 		failed_already=1;
 		PR_Close(clientSock);
     }

 dead:
     if (listenSock) {
 		PR_Close(listenSock);
     }
     PR_Lock(info->dead_lock);
     (*info->alive)--;
     PR_NotifyCondVar(info->dead_cv);
     PR_Unlock(info->dead_lock);

 	if (debug_mode)
     	printf("thread %d is dead\n", info->id);

     PR_Free(info);
 }

 void
 thread_test(PRThreadScope scope, PRInt32 num_threads)
 {
     PRInt32 index;
     PRThread *thr;
     PRLock *dead_lock;
     PRCondVar *dead_cv;
     PRInt32 alive;

 	if (debug_mode)
     	printf("IO Timeout test started with %d threads\n", num_threads);

     dead_lock = PR_NewLock();
     dead_cv = PR_NewCondVar(dead_lock);
     alive = num_threads;

     for (index = 0; index < num_threads; index++) {
         threadInfo *info = (threadInfo *)PR_Malloc(sizeof(threadInfo));

         info->id = index;
         info->dead_lock = dead_lock;
         info->dead_cv = dead_cv;
         info->alive = &alive;
         info->accept_timeout = DEFAULT_ACCEPT_TIMEOUT;

         thr = PR_CreateThread( PR_USER_THREAD,
                                thread_main,
                                (void *)info,
                                PR_PRIORITY_NORMAL,
                                scope,
                                PR_UNJOINABLE_THREAD,
                                0);

         if (!thr) {
         	printf("Failed to create thread, error = %d(%d)\n",
 											PR_GetError(), PR_GetOSError());
 			failed_already=1;

             PR_Lock(dead_lock);
             alive--;
             PR_Unlock(dead_lock);
         }
     }

     PR_Lock(dead_lock);
     while(alive) {
 		if (debug_mode)
         	printf("main loop awake; alive = %d\n", alive);
         PR_WaitCondVar(dead_cv, PR_INTERVAL_NO_TIMEOUT);
     }
     PR_Unlock(dead_lock);

     PR_DestroyCondVar(dead_cv);
     PR_DestroyLock(dead_lock);
 }

 int main(int argc, char **argv)
 {
     PRInt32 num_threads = 0;

 	/* The command line argument: -d is used to determine if the test is being run
 	in debug mode. The regress tool requires only one line output:PASS or FAIL.
 	All of the printfs associated with this test has been handled with a if (debug_mode)
 	test.
 	Usage: test_name [-d] [-t <threads>]
 	*/
 	PLOptStatus os;
 	PLOptState *opt = PL_CreateOptState(argc, argv, "dt:");
 	while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
     {
 		if (PL_OPT_BAD == os) continue;
         switch (opt->option)
         {
         case 'd':  /* debug mode */
 			debug_mode = 1;
             break;
         case 't':  /* threads to involve */
 			num_threads = atoi(opt->value);
             break;
          default:
             break;
         }
     }
 	PL_DestroyOptState(opt);

  /* main test */

     if (0 == num_threads)
         num_threads = NUM_THREADS;

     PR_Init(PR_USER_THREAD, PR_PRIORITY_LOW, 0);
     PR_STDIO_INIT();

     printf("test with global bound thread\n");
     thread_test(PR_GLOBAL_BOUND_THREAD, num_threads);

     printf("test with local thread\n");
     thread_test(PR_LOCAL_THREAD, num_threads);

     printf("test with global thread\n");
     thread_test(PR_GLOBAL_THREAD, num_threads);

     PR_Cleanup();

 	if (failed_already)
 		return 1;
 	else
     	return 0;
 }
	/* -- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -- */
	/* *** BEGIN LICENSE BLOCK ***
	* Version: MPL 1.1/GPL 2.0/LGPL 2.1
	*
	* The contents of this file are subject to the Mozilla Public License Version
	* 1.1 (the "License"); you may not use this file except in compliance with
	* the License. You may obtain a copy of the License at
	* http://www.mozilla.org/MPL/
	*
	* Software distributed under the License is distributed on an "AS IS" basis,
	* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
	* for the specific language governing rights and limitations under the
	* License.
	*
	* The Original Code is the Netscape Portable Runtime (NSPR).
	*
	* The Initial Developer of the Original Code is
	* Netscape Communications Corporation.
	* Portions created by the Initial Developer are Copyright (C) 1998-2000
	* the Initial Developer. All Rights Reserved.
	*
	* Contributor(s):
	*
	* Alternatively, the contents of this file may be used under the terms of
	* either the GNU General Public License Version 2 or later (the "GPL"), or
	* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
	* in which case the provisions of the GPL or the LGPL are applicable instead
	* of those above. If you wish to allow use of your version of this file only
	* under the terms of either the GPL or the LGPL, and not to allow others to
	* use your version of this file under the terms of the MPL, indicate your
	* decision by deleting the provisions above and replace them with the notice
	* and other provisions required by the GPL or the LGPL. If you do not delete
	* the provisions above, a recipient may use your version of this file under
	* the terms of any one of the MPL, the GPL or the LGPL.
	*
	* *** END LICENSE BLOCK *** */

	/*
	** Test socket IO timeouts
	**
	**
	**
	**
	** Modification History:
	** 14-May-97 AGarcia- Converted the test to accomodate the debug_mode flag.
	** The debug mode will print all of the printfs associated with this test.
	** The regress mode will be the default mode. Since the regress tool limits
	** the output to a one line status:PASS or FAIL,all of the printf statements
	** have been handled with an if (debug_mode) statement.
	***********************************************************************/
	/***********************************************************************
	** Includes
	***********************************************************************/
	/* Used to get the command line option */
	#include "plgetopt.h"

	#include <stdio.h>
	#include "nspr.h"

	#define NUM_THREADS 1
	#define BASE_PORT 8000
	#define DEFAULT_ACCEPT_TIMEOUT 2

	typedef struct threadInfo {
	PRInt16 id;
	PRInt16 accept_timeout;
	PRLock *dead_lock;
	PRCondVar *dead_cv;
	PRInt32 *alive;
	} threadInfo;

	PRIntn failed_already = 0;
	PRIntn debug_mode = 0;

	#define LOCAL_SCOPE_STRING "LOCAL scope"
	#define GLOBAL_SCOPE_STRING "GLOBAL scope"
	#define GLOBAL_BOUND_SCOPE_STRING "GLOBAL_BOUND scope"

	void
	thread_main(void *_info)
	{
	threadInfo info = (threadInfo )_info;
	PRNetAddr listenAddr;
	PRNetAddr clientAddr;
	PRFileDesc *listenSock = NULL;
	PRFileDesc *clientSock;
	PRStatus rv;
	PRThreadScope tscope;
	char *scope_str;


	if (debug_mode)
	printf("thread %d is alive\n", info->id);
	tscope = PR_GetThreadScope(PR_GetCurrentThread());

	switch(tscope) {
	case PR_LOCAL_THREAD:
	scope_str = LOCAL_SCOPE_STRING;
	break;
	case PR_GLOBAL_THREAD:
	scope_str = GLOBAL_SCOPE_STRING;
	break;
	case PR_GLOBAL_BOUND_THREAD:
	scope_str = GLOBAL_BOUND_SCOPE_STRING;
	break;
	default:
	PR_ASSERT(!"Invalid thread scope");
	break;
	}
	printf("thread id %d, scope %s\n", info->id, scope_str);

	listenSock = PR_NewTCPSocket();
	if (!listenSock) {
	if (debug_mode)
	printf("unable to create listen socket\n");
	failed_already=1;
	goto dead;
	}

	listenAddr.inet.family = PR_AF_INET;
	listenAddr.inet.port = PR_htons(BASE_PORT + info->id);
	listenAddr.inet.ip = PR_htonl(PR_INADDR_ANY);
	rv = PR_Bind(listenSock, &listenAddr);
	if (rv == PR_FAILURE) {
	if (debug_mode)
	printf("unable to bind\n");
	failed_already=1;
	goto dead;
	}

	rv = PR_Listen(listenSock, 4);
	if (rv == PR_FAILURE) {
	if (debug_mode)
	printf("unable to listen\n");
	failed_already=1;
	goto dead;
	}

	if (debug_mode)
	printf("thread %d going into accept for %d seconds\n",
	info->id, info->accept_timeout + info->id);

	clientSock = PR_Accept(listenSock, &clientAddr, PR_SecondsToInterval(info->accept_timeout +info->id));

	if (clientSock == NULL) {
	if (PR_GetError() == PR_IO_TIMEOUT_ERROR) {
	if (debug_mode) {
	printf("PR_Accept() timeout worked!\n");
	printf("TEST PASSED! PR_Accept() returned error %d\n",
	PR_IO_TIMEOUT_ERROR);
	}
	} else {
	if (debug_mode)
	printf("TEST FAILED! PR_Accept() returned error %d\n",
	PR_GetError());
	failed_already=1;
	}
	} else {
	if (debug_mode)
	printf ("TEST FAILED! PR_Accept() succeeded?\n");
	failed_already=1;
	PR_Close(clientSock);
	}

	dead:
	if (listenSock) {
	PR_Close(listenSock);
	}
	PR_Lock(info->dead_lock);
	(*info->alive)--;
	PR_NotifyCondVar(info->dead_cv);
	PR_Unlock(info->dead_lock);

	if (debug_mode)
	printf("thread %d is dead\n", info->id);

	PR_Free(info);
	}

	void
	thread_test(PRThreadScope scope, PRInt32 num_threads)
	{
	PRInt32 index;
	PRThread *thr;
	PRLock *dead_lock;
	PRCondVar *dead_cv;
	PRInt32 alive;

	if (debug_mode)
	printf("IO Timeout test started with %d threads\n", num_threads);

	dead_lock = PR_NewLock();
	dead_cv = PR_NewCondVar(dead_lock);
	alive = num_threads;

	for (index = 0; index < num_threads; index++) {
	threadInfo info = (threadInfo )PR_Malloc(sizeof(threadInfo));

	info->id = index;
	info->dead_lock = dead_lock;
	info->dead_cv = dead_cv;
	info->alive = &alive;
	info->accept_timeout = DEFAULT_ACCEPT_TIMEOUT;

	thr = PR_CreateThread( PR_USER_THREAD,
	thread_main,
	(void *)info,
	PR_PRIORITY_NORMAL,
	scope,
	PR_UNJOINABLE_THREAD,
	0);

	if (!thr) {
	printf("Failed to create thread, error = %d(%d)\n",
	PR_GetError(), PR_GetOSError());
	failed_already=1;

	PR_Lock(dead_lock);
	alive--;
	PR_Unlock(dead_lock);
	}
	}

	PR_Lock(dead_lock);
	while(alive) {
	if (debug_mode)
	printf("main loop awake; alive = %d\n", alive);
	PR_WaitCondVar(dead_cv, PR_INTERVAL_NO_TIMEOUT);
	}
	PR_Unlock(dead_lock);

	PR_DestroyCondVar(dead_cv);
	PR_DestroyLock(dead_lock);
	}

	int main(int argc, char **argv)
	{
	PRInt32 num_threads = 0;

	/* The command line argument: -d is used to determine if the test is being run
	in debug mode. The regress tool requires only one line output:PASS or FAIL.
	All of the printfs associated with this test has been handled with a if (debug_mode)
	test.
	Usage: test_name [-d] [-t <threads>]
	*/
	PLOptStatus os;
	PLOptState *opt = PL_CreateOptState(argc, argv, "dt:");
	while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
	{
	if (PL_OPT_BAD == os) continue;
	switch (opt->option)
	{
	case 'd': /* debug mode */
	debug_mode = 1;
	break;
	case 't': /* threads to involve */
	num_threads = atoi(opt->value);
	break;
	default:
	break;
	}
	}
	PL_DestroyOptState(opt);

	/* main test */

	if (0 == num_threads)
	num_threads = NUM_THREADS;

	PR_Init(PR_USER_THREAD, PR_PRIORITY_LOW, 0);
	PR_STDIO_INIT();

	printf("test with global bound thread\n");
	thread_test(PR_GLOBAL_BOUND_THREAD, num_threads);

	printf("test with local thread\n");
	thread_test(PR_LOCAL_THREAD, num_threads);

	printf("test with global thread\n");
	thread_test(PR_GLOBAL_THREAD, num_threads);

	PR_Cleanup();

	if (failed_already)
	return 1;
	else
	return 0;
	}