зеркало из https://github.com/mozilla/pjs.git
371 строка
10 KiB
C
371 строка
10 KiB
C
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/*
|
|
* 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 security libraries.
|
|
*
|
|
* The Initial Developer of the Original Code is Netscape
|
|
* Communications Corporation. Portions created by Netscape are
|
|
* Copyright (C) 1998-2000 Netscape Communications Corporation. All
|
|
* Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*
|
|
* Alternatively, the contents of this file may be used under the
|
|
* terms of the GNU General Public License Version 2 or later (the
|
|
* "GPL"), in which case the provisions of the GPL are applicable
|
|
* instead of those above. If you wish to allow use of your
|
|
* version of this file only under the terms of the GPL and not to
|
|
* allow others to use your version of this file under the MPL,
|
|
* indicate your decision by deleting the provisions above and
|
|
* replace them with the notice and other provisions required by
|
|
* the GPL. If you do not delete the provisions above, a recipient
|
|
* may use your version of this file under either the MPL or the
|
|
* GPL.
|
|
*/
|
|
|
|
/*
|
|
* File: intrupt.c
|
|
* Purpose: testing thread interrupts
|
|
*/
|
|
|
|
#include "plgetopt.h"
|
|
#include "prcvar.h"
|
|
#include "prerror.h"
|
|
#include "prinit.h"
|
|
#include "prinrval.h"
|
|
#include "prio.h"
|
|
#include "prlock.h"
|
|
#include "prlog.h"
|
|
#include "prthread.h"
|
|
#include "prtypes.h"
|
|
#include "prnetdb.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#ifdef XP_MAC
|
|
#include "prlog.h"
|
|
#define printf PR_LogPrint
|
|
extern void SetupMacPrintfLog(char *logFile);
|
|
#endif
|
|
|
|
#define DEFAULT_TCP_PORT 12500
|
|
|
|
static PRLock *ml = NULL;
|
|
static PRCondVar *cv = NULL;
|
|
|
|
static PRBool passed = PR_TRUE;
|
|
static PRBool debug_mode = PR_FALSE;
|
|
static PRThreadScope thread_scope = PR_LOCAL_THREAD;
|
|
|
|
static void PR_CALLBACK AbortCV(void *arg)
|
|
{
|
|
PRStatus rv;
|
|
PRThread *me = PR_CurrentThread();
|
|
|
|
/* some other thread (main) is doing the interrupt */
|
|
PR_Lock(ml);
|
|
rv = PR_WaitCondVar(cv, PR_INTERVAL_NO_TIMEOUT);
|
|
if (debug_mode) printf( "Expected interrupt on wait CV and ");
|
|
if (PR_FAILURE == rv)
|
|
{
|
|
if (PR_PENDING_INTERRUPT_ERROR == PR_GetError())
|
|
{
|
|
if (debug_mode) printf("got it\n");
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("got random error\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("got a successful completion\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
|
|
rv = PR_WaitCondVar(cv, 10);
|
|
if (debug_mode)
|
|
{
|
|
printf(
|
|
"Expected success on wait CV and %s\n",
|
|
(PR_SUCCESS == rv) ? "got it" : "failed");
|
|
}
|
|
passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;
|
|
|
|
/* interrupt myself, then clear */
|
|
PR_Interrupt(me);
|
|
PR_ClearInterrupt();
|
|
rv = PR_WaitCondVar(cv, 10);
|
|
if (debug_mode)
|
|
{
|
|
printf("Expected success on wait CV and ");
|
|
if (PR_FAILURE == rv)
|
|
{
|
|
printf(
|
|
"%s\n", (PR_PENDING_INTERRUPT_ERROR == PR_GetError()) ?
|
|
"got interrupted" : "a random failure");
|
|
}
|
|
printf("got it\n");
|
|
}
|
|
passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;
|
|
|
|
/* set, then wait - interrupt - then wait again */
|
|
PR_Interrupt(me);
|
|
rv = PR_WaitCondVar(cv, 10);
|
|
if (debug_mode) printf( "Expected interrupt on wait CV and ");
|
|
if (PR_FAILURE == rv)
|
|
{
|
|
if (PR_PENDING_INTERRUPT_ERROR == PR_GetError())
|
|
{
|
|
if (debug_mode) printf("got it\n");
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("failed\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("got a successful completion\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
|
|
rv = PR_WaitCondVar(cv, 10);
|
|
if (debug_mode)
|
|
{
|
|
printf(
|
|
"Expected success on wait CV and %s\n",
|
|
(PR_SUCCESS == rv) ? "got it" : "failed");
|
|
}
|
|
passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;
|
|
|
|
PR_Unlock(ml);
|
|
|
|
} /* AbortCV */
|
|
|
|
static void PR_CALLBACK AbortIO(void *arg)
|
|
{
|
|
PRStatus rv;
|
|
PR_Sleep(PR_SecondsToInterval(2));
|
|
rv = PR_Interrupt((PRThread*)arg);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
} /* AbortIO */
|
|
|
|
static void PR_CALLBACK AbortJoin(void *arg)
|
|
{
|
|
} /* AbortJoin */
|
|
|
|
static void setup_listen_socket(PRFileDesc **listner, PRNetAddr *netaddr)
|
|
{
|
|
PRStatus rv;
|
|
PRInt16 port = DEFAULT_TCP_PORT;
|
|
|
|
*listner = PR_NewTCPSocket();
|
|
PR_ASSERT(*listner != NULL);
|
|
memset(netaddr, 0, sizeof(*netaddr));
|
|
(*netaddr).inet.ip = PR_htonl(PR_INADDR_ANY);
|
|
(*netaddr).inet.family = PR_AF_INET;
|
|
do
|
|
{
|
|
(*netaddr).inet.port = PR_htons(port);
|
|
rv = PR_Bind(*listner, netaddr);
|
|
port += 1;
|
|
PR_ASSERT(port < (DEFAULT_TCP_PORT + 10));
|
|
} while (PR_FAILURE == rv);
|
|
|
|
rv = PR_Listen(*listner, 5);
|
|
|
|
if (PR_GetSockName(*listner, netaddr) < 0) {
|
|
if (debug_mode) printf("intrupt: ERROR - PR_GetSockName failed\n");
|
|
passed = PR_FALSE;
|
|
return;
|
|
}
|
|
|
|
}
|
|
|
|
static void PR_CALLBACK IntrBlock(void *arg)
|
|
{
|
|
PRStatus rv;
|
|
PRNetAddr netaddr;
|
|
PRFileDesc *listner;
|
|
|
|
/* some other thread (main) is doing the interrupt */
|
|
/* block the interrupt */
|
|
PR_BlockInterrupt();
|
|
PR_Lock(ml);
|
|
rv = PR_WaitCondVar(cv, PR_SecondsToInterval(4));
|
|
PR_Unlock(ml);
|
|
if (debug_mode)
|
|
{
|
|
printf("Expected success on wait CV and ");
|
|
if (PR_FAILURE == rv)
|
|
{
|
|
printf(
|
|
"%s\n", (PR_PENDING_INTERRUPT_ERROR == PR_GetError()) ?
|
|
"got interrupted" : "got a random failure");
|
|
} else
|
|
printf("got it\n");
|
|
}
|
|
passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;
|
|
|
|
setup_listen_socket(&listner, &netaddr);
|
|
PR_UnblockInterrupt();
|
|
if (PR_Accept(listner, &netaddr, PR_INTERVAL_NO_TIMEOUT) == NULL)
|
|
{
|
|
PRInt32 error = PR_GetError();
|
|
if (debug_mode) printf("Expected interrupt on PR_Accept() and ");
|
|
if (PR_PENDING_INTERRUPT_ERROR == error)
|
|
{
|
|
if (debug_mode) printf("got it\n");
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("failed\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("Failed to interrupt PR_Accept()\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
|
|
(void)PR_Close(listner); listner = NULL;
|
|
} /* TestIntrBlock */
|
|
|
|
void PR_CALLBACK Intrupt(void *arg)
|
|
{
|
|
PRStatus rv;
|
|
PRNetAddr netaddr;
|
|
PRFileDesc *listner;
|
|
PRThread *abortCV, *abortIO, *abortJoin, *intrBlock;
|
|
|
|
ml = PR_NewLock();
|
|
cv = PR_NewCondVar(ml);
|
|
|
|
#ifdef XP_MAC
|
|
SetupMacPrintfLog("intrupt.log");
|
|
debug_mode = PR_TRUE;
|
|
#endif
|
|
|
|
/* Part I */
|
|
if (debug_mode) printf("Part I\n");
|
|
abortCV = PR_CreateThread(
|
|
PR_USER_THREAD, AbortCV, 0, PR_PRIORITY_NORMAL,
|
|
thread_scope, PR_JOINABLE_THREAD, 0);
|
|
|
|
PR_Sleep(PR_SecondsToInterval(2));
|
|
rv = PR_Interrupt(abortCV);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
rv = PR_JoinThread(abortCV);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
|
|
/* Part II */
|
|
if (debug_mode) printf("Part II\n");
|
|
abortJoin = PR_CreateThread(
|
|
PR_USER_THREAD, AbortJoin, 0, PR_PRIORITY_NORMAL,
|
|
thread_scope, PR_JOINABLE_THREAD, 0);
|
|
PR_Sleep(PR_SecondsToInterval(2));
|
|
if (debug_mode) printf("Expecting to interrupt an exited thread ");
|
|
rv = PR_Interrupt(abortJoin);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
rv = PR_JoinThread(abortJoin);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
if (debug_mode) printf("and succeeded\n");
|
|
|
|
/* Part III */
|
|
if (debug_mode) printf("Part III\n");
|
|
setup_listen_socket(&listner, &netaddr);
|
|
abortIO = PR_CreateThread(
|
|
PR_USER_THREAD, AbortIO, PR_CurrentThread(), PR_PRIORITY_NORMAL,
|
|
thread_scope, PR_JOINABLE_THREAD, 0);
|
|
|
|
if (PR_Accept(listner, &netaddr, PR_INTERVAL_NO_TIMEOUT) == NULL)
|
|
{
|
|
PRInt32 error = PR_GetError();
|
|
if (debug_mode) printf("Expected interrupt on PR_Accept() and ");
|
|
if (PR_PENDING_INTERRUPT_ERROR == error)
|
|
{
|
|
if (debug_mode) printf("got it\n");
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("failed\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (debug_mode) printf("Failed to interrupt PR_Accept()\n");
|
|
passed = PR_FALSE;
|
|
}
|
|
|
|
(void)PR_Close(listner); listner = NULL;
|
|
|
|
rv = PR_JoinThread(abortIO);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
/* Part VI */
|
|
if (debug_mode) printf("Part VI\n");
|
|
intrBlock = PR_CreateThread(
|
|
PR_USER_THREAD, IntrBlock, 0, PR_PRIORITY_NORMAL,
|
|
thread_scope, PR_JOINABLE_THREAD, 0);
|
|
|
|
PR_Sleep(PR_SecondsToInterval(2));
|
|
rv = PR_Interrupt(intrBlock);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
rv = PR_JoinThread(intrBlock);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
|
|
PR_DestroyCondVar(cv);
|
|
PR_DestroyLock(ml);
|
|
} /* Intrupt */
|
|
|
|
PRIntn main(PRIntn argc, char **argv)
|
|
{
|
|
PRThread *intrupt;
|
|
PLOptStatus os;
|
|
PLOptState *opt = PL_CreateOptState(argc, argv, "dG");
|
|
while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
|
|
{
|
|
if (PL_OPT_BAD == os) continue;
|
|
switch (opt->option)
|
|
{
|
|
case 'd': /* debug mode */
|
|
debug_mode = PR_TRUE;
|
|
break;
|
|
case 'G': /* use global threads */
|
|
thread_scope = PR_GLOBAL_THREAD;
|
|
break;
|
|
}
|
|
}
|
|
PL_DestroyOptState(opt);
|
|
PR_STDIO_INIT();
|
|
intrupt = PR_CreateThread(
|
|
PR_USER_THREAD, Intrupt, NULL, PR_PRIORITY_NORMAL,
|
|
thread_scope, PR_JOINABLE_THREAD, 0);
|
|
if (intrupt == NULL) {
|
|
fprintf(stderr, "cannot create thread\n");
|
|
passed = PR_FALSE;
|
|
} else {
|
|
PRStatus rv;
|
|
rv = PR_JoinThread(intrupt);
|
|
PR_ASSERT(rv == PR_SUCCESS);
|
|
}
|
|
printf("%s\n", ((passed) ? "PASSED" : "FAILED"));
|
|
return ((passed) ? 0 : 1);
|
|
} /* main */
|
|
|
|
/* intrupt.c */
|