/* -*- 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;
}
