463 lines
14 KiB
C
463 lines
14 KiB
C
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
/***********************************************************************
|
|
** 1996 - Netscape Communications Corporation
|
|
**
|
|
** Name: accept.c
|
|
**
|
|
** Description: Run accept() sucessful connection tests.
|
|
**
|
|
** Modification History:
|
|
** 04-Jun-97 AGarcia - Reconvert test file to return a 0 for PASS and a 1 for
|
|
*FAIL
|
|
** 13-May-97 AGarcia- Converted the test to accomodate the debug_mode
|
|
** 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.
|
|
** 04-June-97 AGarcia removed the Test_Result function. Regress tool has been
|
|
*updated to
|
|
** recognize the return code from tha main program.
|
|
** 12-June-97 Revert to return code 0 and 1.
|
|
***********************************************************************/
|
|
|
|
/***********************************************************************
|
|
** Includes
|
|
***********************************************************************/
|
|
|
|
#include "nspr.h"
|
|
#include "prpriv.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "plgetopt.h"
|
|
#include "plerror.h"
|
|
|
|
#define BASE_PORT 10000
|
|
|
|
#define CLIENT_DATA 128
|
|
|
|
#define ACCEPT_NORMAL 0x1
|
|
#define ACCEPT_FAST 0x2
|
|
#define ACCEPT_READ 0x3
|
|
#define ACCEPT_READ_FAST 0x4
|
|
#define ACCEPT_READ_FAST_CB 0x5
|
|
|
|
#define CLIENT_NORMAL 0x1
|
|
#define CLIENT_TIMEOUT_ACCEPT 0x2
|
|
#define CLIENT_TIMEOUT_SEND 0x3
|
|
|
|
#define SERVER_MAX_BIND_COUNT 100
|
|
|
|
#define TIMEOUTSECS 2
|
|
PRIntervalTime timeoutTime;
|
|
|
|
static PRInt32 count = 1;
|
|
static PRFileDesc* output;
|
|
static PRNetAddr serverAddr;
|
|
static PRThreadScope thread_scope = PR_LOCAL_THREAD;
|
|
static PRInt32 clientCommand;
|
|
static PRInt32 iterations;
|
|
static PRStatus rv;
|
|
static PRFileDesc* listenSock;
|
|
static PRFileDesc* clientSock = NULL;
|
|
static PRNetAddr listenAddr;
|
|
static PRNetAddr clientAddr;
|
|
static PRThread* clientThread;
|
|
static PRNetAddr* raddr;
|
|
static char buf[4096 + 2 * sizeof(PRNetAddr) + 32];
|
|
static PRInt32 status;
|
|
static PRInt32 bytesRead;
|
|
|
|
PRIntn failed_already = 0;
|
|
PRIntn debug_mode;
|
|
|
|
void Test_Assert(const char* msg, const char* file, PRIntn line) {
|
|
failed_already = 1;
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "@%s:%d ", file, line);
|
|
PR_fprintf(output, msg);
|
|
}
|
|
} /* Test_Assert */
|
|
|
|
#define TEST_ASSERT(expr) \
|
|
if (!(expr)) Test_Assert(#expr, __FILE__, __LINE__)
|
|
|
|
#ifdef WINNT
|
|
# define CALLBACK_MAGIC 0x12345678
|
|
|
|
void timeout_callback(void* magic) {
|
|
TEST_ASSERT(magic == (void*)CALLBACK_MAGIC);
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "timeout callback called okay\n");
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static void PR_CALLBACK ClientThread(void* _action) {
|
|
PRInt32 action = *(PRInt32*)_action;
|
|
PRInt32 iterations = count;
|
|
PRFileDesc* sock = NULL;
|
|
|
|
serverAddr.inet.family = PR_AF_INET;
|
|
serverAddr.inet.port = listenAddr.inet.port;
|
|
serverAddr.inet.ip = PR_htonl(PR_INADDR_LOOPBACK);
|
|
|
|
for (; iterations--;) {
|
|
PRInt32 rv;
|
|
char buf[CLIENT_DATA];
|
|
|
|
memset(buf, 0xaf, sizeof(buf)); /* initialize with arbitrary data */
|
|
sock = PR_NewTCPSocket();
|
|
if (!sock) {
|
|
if (!debug_mode) {
|
|
failed_already = 1;
|
|
} else {
|
|
PR_fprintf(output, "client: unable to create socket\n");
|
|
}
|
|
return;
|
|
}
|
|
|
|
if (action != CLIENT_TIMEOUT_ACCEPT) {
|
|
if ((rv = PR_Connect(sock, &serverAddr, timeoutTime)) < 0) {
|
|
if (!debug_mode) {
|
|
failed_already = 1;
|
|
} else
|
|
PR_fprintf(
|
|
output,
|
|
"client: unable to connect to server (%ld, %ld, %ld, %ld)\n",
|
|
iterations, rv, PR_GetError(), PR_GetOSError());
|
|
goto ErrorExit;
|
|
}
|
|
|
|
if (action != CLIENT_TIMEOUT_SEND) {
|
|
if ((rv = PR_Send(sock, buf, CLIENT_DATA, 0, timeoutTime)) < 0) {
|
|
if (!debug_mode) {
|
|
failed_already = 1;
|
|
} else {
|
|
PR_fprintf(output,
|
|
"client: unable to send to server (%d, %ld, %ld)\n",
|
|
CLIENT_DATA, rv, PR_GetError());
|
|
}
|
|
goto ErrorExit;
|
|
}
|
|
} else {
|
|
PR_Sleep(PR_SecondsToInterval(TIMEOUTSECS + 1));
|
|
}
|
|
} else {
|
|
PR_Sleep(PR_SecondsToInterval(TIMEOUTSECS + 1));
|
|
}
|
|
if (debug_mode) {
|
|
PR_fprintf(output, ".");
|
|
}
|
|
PR_Close(sock);
|
|
sock = NULL;
|
|
}
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "\n");
|
|
}
|
|
|
|
ErrorExit:
|
|
if (sock != NULL) {
|
|
PR_Close(sock);
|
|
}
|
|
}
|
|
|
|
static void RunTest(PRInt32 acceptType, PRInt32 clientAction) {
|
|
int i;
|
|
|
|
/* First bind to the socket */
|
|
listenSock = PR_NewTCPSocket();
|
|
if (!listenSock) {
|
|
failed_already = 1;
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "unable to create listen socket\n");
|
|
}
|
|
return;
|
|
}
|
|
memset(&listenAddr, 0, sizeof(listenAddr));
|
|
listenAddr.inet.family = PR_AF_INET;
|
|
listenAddr.inet.port = PR_htons(BASE_PORT);
|
|
listenAddr.inet.ip = PR_htonl(PR_INADDR_ANY);
|
|
/*
|
|
* try a few times to bind server's address, if addresses are in
|
|
* use
|
|
*/
|
|
i = 0;
|
|
while (PR_Bind(listenSock, &listenAddr) == PR_FAILURE) {
|
|
if (PR_GetError() == PR_ADDRESS_IN_USE_ERROR) {
|
|
listenAddr.inet.port += 2;
|
|
if (i++ < SERVER_MAX_BIND_COUNT) {
|
|
continue;
|
|
}
|
|
}
|
|
failed_already = 1;
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "accept: ERROR - PR_Bind failed\n");
|
|
}
|
|
return;
|
|
}
|
|
|
|
rv = PR_Listen(listenSock, 100);
|
|
if (rv == PR_FAILURE) {
|
|
failed_already = 1;
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "unable to listen\n");
|
|
}
|
|
return;
|
|
}
|
|
|
|
clientCommand = clientAction;
|
|
clientThread =
|
|
PR_CreateThread(PR_USER_THREAD, ClientThread, (void*)&clientCommand,
|
|
PR_PRIORITY_NORMAL, thread_scope, PR_JOINABLE_THREAD, 0);
|
|
if (!clientThread) {
|
|
failed_already = 1;
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "error creating client thread\n");
|
|
}
|
|
return;
|
|
}
|
|
|
|
iterations = count;
|
|
for (; iterations--;) {
|
|
switch (acceptType) {
|
|
case ACCEPT_NORMAL:
|
|
clientSock = PR_Accept(listenSock, &clientAddr, timeoutTime);
|
|
switch (clientAction) {
|
|
case CLIENT_TIMEOUT_ACCEPT:
|
|
TEST_ASSERT(clientSock == 0);
|
|
TEST_ASSERT(PR_GetError() == PR_IO_TIMEOUT_ERROR);
|
|
break;
|
|
case CLIENT_NORMAL:
|
|
TEST_ASSERT(clientSock);
|
|
bytesRead = PR_Recv(clientSock, buf, CLIENT_DATA, 0, timeoutTime);
|
|
TEST_ASSERT(bytesRead == CLIENT_DATA);
|
|
break;
|
|
case CLIENT_TIMEOUT_SEND:
|
|
TEST_ASSERT(clientSock);
|
|
bytesRead = PR_Recv(clientSock, buf, CLIENT_DATA, 0, timeoutTime);
|
|
TEST_ASSERT(bytesRead == -1);
|
|
TEST_ASSERT(PR_GetError() == PR_IO_TIMEOUT_ERROR);
|
|
break;
|
|
}
|
|
break;
|
|
case ACCEPT_READ:
|
|
status = PR_AcceptRead(listenSock, &clientSock, &raddr, buf,
|
|
CLIENT_DATA, timeoutTime);
|
|
switch (clientAction) {
|
|
case CLIENT_TIMEOUT_ACCEPT:
|
|
/* Invalid test case */
|
|
TEST_ASSERT(0);
|
|
break;
|
|
case CLIENT_NORMAL:
|
|
TEST_ASSERT(clientSock);
|
|
TEST_ASSERT(status == CLIENT_DATA);
|
|
break;
|
|
case CLIENT_TIMEOUT_SEND:
|
|
TEST_ASSERT(status == -1);
|
|
TEST_ASSERT(PR_GetError() == PR_IO_TIMEOUT_ERROR);
|
|
break;
|
|
}
|
|
break;
|
|
#ifdef WINNT
|
|
case ACCEPT_FAST:
|
|
clientSock = PR_NTFast_Accept(listenSock, &clientAddr, timeoutTime);
|
|
switch (clientAction) {
|
|
case CLIENT_TIMEOUT_ACCEPT:
|
|
TEST_ASSERT(clientSock == 0);
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "PR_GetError is %ld\n", PR_GetError());
|
|
}
|
|
TEST_ASSERT(PR_GetError() == PR_IO_TIMEOUT_ERROR);
|
|
break;
|
|
case CLIENT_NORMAL:
|
|
TEST_ASSERT(clientSock);
|
|
bytesRead = PR_Recv(clientSock, buf, CLIENT_DATA, 0, timeoutTime);
|
|
TEST_ASSERT(bytesRead == CLIENT_DATA);
|
|
break;
|
|
case CLIENT_TIMEOUT_SEND:
|
|
TEST_ASSERT(clientSock);
|
|
bytesRead = PR_Recv(clientSock, buf, CLIENT_DATA, 0, timeoutTime);
|
|
TEST_ASSERT(bytesRead == -1);
|
|
TEST_ASSERT(PR_GetError() == PR_IO_TIMEOUT_ERROR);
|
|
break;
|
|
}
|
|
break;
|
|
break;
|
|
case ACCEPT_READ_FAST:
|
|
status = PR_NTFast_AcceptRead(listenSock, &clientSock, &raddr, buf,
|
|
4096, timeoutTime);
|
|
switch (clientAction) {
|
|
case CLIENT_TIMEOUT_ACCEPT:
|
|
/* Invalid test case */
|
|
TEST_ASSERT(0);
|
|
break;
|
|
case CLIENT_NORMAL:
|
|
TEST_ASSERT(clientSock);
|
|
TEST_ASSERT(status == CLIENT_DATA);
|
|
break;
|
|
case CLIENT_TIMEOUT_SEND:
|
|
TEST_ASSERT(clientSock == NULL);
|
|
TEST_ASSERT(status == -1);
|
|
TEST_ASSERT(PR_GetError() == PR_IO_TIMEOUT_ERROR);
|
|
break;
|
|
}
|
|
break;
|
|
case ACCEPT_READ_FAST_CB:
|
|
status = PR_NTFast_AcceptRead_WithTimeoutCallback(
|
|
listenSock, &clientSock, &raddr, buf, 4096, timeoutTime,
|
|
timeout_callback, (void*)CALLBACK_MAGIC);
|
|
switch (clientAction) {
|
|
case CLIENT_TIMEOUT_ACCEPT:
|
|
/* Invalid test case */
|
|
TEST_ASSERT(0);
|
|
break;
|
|
case CLIENT_NORMAL:
|
|
TEST_ASSERT(clientSock);
|
|
TEST_ASSERT(status == CLIENT_DATA);
|
|
break;
|
|
case CLIENT_TIMEOUT_SEND:
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "clientSock = 0x%8.8lx\n", clientSock);
|
|
}
|
|
TEST_ASSERT(clientSock == NULL);
|
|
TEST_ASSERT(status == -1);
|
|
TEST_ASSERT(PR_GetError() == PR_IO_TIMEOUT_ERROR);
|
|
break;
|
|
}
|
|
break;
|
|
#endif
|
|
}
|
|
if (clientSock != NULL) {
|
|
PR_Close(clientSock);
|
|
clientSock = NULL;
|
|
}
|
|
}
|
|
PR_Close(listenSock);
|
|
|
|
PR_JoinThread(clientThread);
|
|
}
|
|
|
|
void AcceptUpdatedTest(void) { RunTest(ACCEPT_NORMAL, CLIENT_NORMAL); }
|
|
void AcceptNotUpdatedTest(void) { RunTest(ACCEPT_FAST, CLIENT_NORMAL); }
|
|
void AcceptReadTest(void) { RunTest(ACCEPT_READ, CLIENT_NORMAL); }
|
|
void AcceptReadNotUpdatedTest(void) {
|
|
RunTest(ACCEPT_READ_FAST, CLIENT_NORMAL);
|
|
}
|
|
void AcceptReadCallbackTest(void) {
|
|
RunTest(ACCEPT_READ_FAST_CB, CLIENT_NORMAL);
|
|
}
|
|
|
|
void TimeoutAcceptUpdatedTest(void) {
|
|
RunTest(ACCEPT_NORMAL, CLIENT_TIMEOUT_ACCEPT);
|
|
}
|
|
void TimeoutAcceptNotUpdatedTest(void) {
|
|
RunTest(ACCEPT_FAST, CLIENT_TIMEOUT_ACCEPT);
|
|
}
|
|
void TimeoutAcceptReadCallbackTest(void) {
|
|
RunTest(ACCEPT_READ_FAST_CB, CLIENT_TIMEOUT_ACCEPT);
|
|
}
|
|
|
|
void TimeoutReadUpdatedTest(void) {
|
|
RunTest(ACCEPT_NORMAL, CLIENT_TIMEOUT_SEND);
|
|
}
|
|
void TimeoutReadNotUpdatedTest(void) {
|
|
RunTest(ACCEPT_FAST, CLIENT_TIMEOUT_SEND);
|
|
}
|
|
void TimeoutReadReadTest(void) { RunTest(ACCEPT_READ, CLIENT_TIMEOUT_SEND); }
|
|
void TimeoutReadReadNotUpdatedTest(void) {
|
|
RunTest(ACCEPT_READ_FAST, CLIENT_TIMEOUT_SEND);
|
|
}
|
|
void TimeoutReadReadCallbackTest(void) {
|
|
RunTest(ACCEPT_READ_FAST_CB, CLIENT_TIMEOUT_SEND);
|
|
}
|
|
|
|
/************************************************************************/
|
|
|
|
static void Measure(void (*func)(void), const char* msg) {
|
|
PRIntervalTime start, stop;
|
|
double d;
|
|
|
|
start = PR_IntervalNow();
|
|
(*func)();
|
|
stop = PR_IntervalNow();
|
|
|
|
d = (double)PR_IntervalToMicroseconds(stop - start);
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "%40s: %6.2f usec\n", msg, d / count);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
/* 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] [-c n]
|
|
*/
|
|
PLOptStatus os;
|
|
PLOptState* opt = PL_CreateOptState(argc, argv, "Gdc:");
|
|
while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
|
|
if (PL_OPT_BAD == os) {
|
|
continue;
|
|
}
|
|
switch (opt->option) {
|
|
case 'G': /* global threads */
|
|
thread_scope = PR_GLOBAL_THREAD;
|
|
break;
|
|
case 'd': /* debug mode */
|
|
debug_mode = 1;
|
|
break;
|
|
case 'c': /* loop counter */
|
|
count = atoi(opt->value);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
PL_DestroyOptState(opt);
|
|
|
|
PR_Init(PR_USER_THREAD, PR_PRIORITY_NORMAL, 0);
|
|
output = PR_STDERR;
|
|
PR_STDIO_INIT();
|
|
|
|
timeoutTime = PR_SecondsToInterval(TIMEOUTSECS);
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "\nRun accept() sucessful connection tests\n");
|
|
}
|
|
|
|
Measure(AcceptUpdatedTest, "PR_Accept()");
|
|
Measure(AcceptReadTest, "PR_AcceptRead()");
|
|
#ifdef WINNT
|
|
Measure(AcceptNotUpdatedTest, "PR_NTFast_Accept()");
|
|
Measure(AcceptReadNotUpdatedTest, "PR_NTFast_AcceptRead()");
|
|
Measure(AcceptReadCallbackTest, "PR_NTFast_AcceptRead_WithTimeoutCallback()");
|
|
#endif
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "\nRun accept() timeout in the accept tests\n");
|
|
}
|
|
#ifdef WINNT
|
|
Measure(TimeoutReadReadCallbackTest,
|
|
"PR_NTFast_AcceptRead_WithTimeoutCallback()");
|
|
#endif
|
|
Measure(TimeoutReadUpdatedTest, "PR_Accept()");
|
|
if (debug_mode) {
|
|
PR_fprintf(output, "\nRun accept() timeout in the read tests\n");
|
|
}
|
|
Measure(TimeoutReadReadTest, "PR_AcceptRead()");
|
|
#ifdef WINNT
|
|
Measure(TimeoutReadNotUpdatedTest, "PR_NTFast_Accept()");
|
|
Measure(TimeoutReadReadNotUpdatedTest, "PR_NTFast_AcceptRead()");
|
|
Measure(TimeoutReadReadCallbackTest,
|
|
"PR_NTFast_AcceptRead_WithTimeoutCallback()");
|
|
#endif
|
|
PR_fprintf(output, "%s\n", (failed_already) ? "FAIL" : "PASS");
|
|
return failed_already;
|
|
}
|