pjs/nsprpub/pr/tests/tmoacc.c

310 строки
9.8 KiB
C

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/*
* The contents of this file are subject to the Netscape Public License
* Version 1.0 (the "NPL"); you may not use this file except in
* compliance with the NPL. You may obtain a copy of the NPL at
* http://www.mozilla.org/NPL/
*
* Software distributed under the NPL is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
* for the specific language governing rights and limitations under the
* NPL.
*
* The Initial Developer of this code under the NPL is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All Rights
* Reserved.
*/
#include "nspr.h"
#include <stdlib.h>
#include <string.h>
#include "plerror.h"
#include "plgetopt.h"
#ifdef XP_MAC
#include "prlog.h"
#define printf PR_LogPrint
#endif
#define BASE_PORT 9867
#define DEFAULT_THREADS 1
#define DEFAULT_BACKLOG 10
#define DEFAULT_TIMEOUT 10
#define RANDOM_RANGE 100 /* should be significantly smaller than RAND_MAX */
typedef enum {running, stopped} Status;
typedef struct Shared
{
PRLock *ml;
PRCondVar *cv;
PRBool passed;
PRBool random;
PRFileDesc *debug;
PRIntervalTime timeout;
PRFileDesc *listenSock;
Status status;
} Shared;
static PRIntervalTime Timeout(const Shared *shared)
{
PRIntervalTime timeout = shared->timeout;
if (shared->random)
{
PRIntervalTime half = timeout >> 1; /* one half of the interval */
PRIntervalTime quarter = half >> 1; /* one quarter of the interval */
/* something in [0..timeout / 2) */
PRUint32 random = (rand() % RANDOM_RANGE) * half / RANDOM_RANGE;
timeout = (3 * quarter) + random; /* [75..125)% */
}
return timeout;
} /* Timeout */
static void Accept(void *arg)
{
PRStatus rv;
char *buffer = NULL;
PRNetAddr clientAddr;
Shared *shared = (Shared*)arg;
PRInt32 recv_length = 0, flags = 0;
PRFileDesc *clientSock;
PRIntn toread, byte, bytes, loop = 0;
struct Descriptor { PRInt32 length; PRUint32 checksum; } descriptor;
do
{
PRUint32 checksum = 0;
if (NULL != shared->debug)
PR_fprintf(shared->debug, "[%d]accepting ... ", loop++);
clientSock = PR_Accept(
shared->listenSock, &clientAddr, Timeout(shared));
if (clientSock != NULL)
{
if (NULL != shared->debug)
PR_fprintf(shared->debug, "reading length ... ");
bytes = PR_Recv(
clientSock, &descriptor, sizeof(descriptor),
flags, Timeout(shared));
if (sizeof(descriptor) == bytes)
{
/* and, before doing something stupid ... */
descriptor.length = PR_ntohl(descriptor.length);
descriptor.checksum = PR_ntohl(descriptor.checksum);
if (NULL != shared->debug)
PR_fprintf(shared->debug, "%d bytes ... ", descriptor.length);
toread = descriptor.length;
if (recv_length < descriptor.length)
{
if (NULL != buffer) PR_DELETE(buffer);
buffer = (char*)PR_MALLOC(descriptor.length);
recv_length = descriptor.length;
}
for (toread = descriptor.length; toread > 0; toread -= bytes)
{
bytes = PR_Recv(
clientSock, &buffer[descriptor.length - toread],
toread, flags, Timeout(shared));
if (-1 == bytes)
{
if (NULL != shared->debug)
PR_fprintf(shared->debug, "read data failed...");
bytes = 0;
}
}
}
else if (NULL != shared->debug)
{
PR_fprintf(shared->debug, "read desciptor failed...");
descriptor.length = -1;
}
if (NULL != shared->debug)
PR_fprintf(shared->debug, "closing");
rv = PR_Shutdown(clientSock, PR_SHUTDOWN_BOTH);
if ((PR_FAILURE == rv) && (NULL != shared->debug))
{
PR_fprintf(shared->debug, " failed");
shared->passed = PR_FALSE;
}
rv = PR_Close(clientSock);
if (PR_FAILURE == rv) if (NULL != shared->debug)
{
PR_fprintf(shared->debug, " failed");
shared->passed = PR_FALSE;
}
if (descriptor.length > 0)
{
for (byte = 0; byte < descriptor.length; ++byte)
{
PRUint32 overflow = checksum & 0x80000000;
checksum = (checksum << 1);
if (0x00000000 != overflow) checksum += 1;
checksum += buffer[byte];
}
if ((descriptor.checksum != checksum) && (NULL != shared->debug))
{
PR_fprintf(shared->debug, " ... data mismatch");
shared->passed = PR_FALSE;
}
}
else if (0 == descriptor.length)
{
PR_Lock(shared->ml);
shared->status = stopped;
PR_NotifyCondVar(shared->cv);
PR_Unlock(shared->ml);
}
if (NULL != shared->debug)
PR_fprintf(shared->debug, "\n");
}
else
{
if (PR_PENDING_INTERRUPT_ERROR != PR_GetError())
{
if (NULL != shared->debug) PL_PrintError("Accept");
shared->passed = PR_FALSE;
}
}
} while (running == shared->status);
if (NULL != buffer) PR_DELETE(buffer);
} /* Accept */
PRIntn Tmoacc(PRIntn argc, char **argv)
{
PRStatus rv;
PRIntn index;
Shared *shared;
PLOptStatus os;
PRThread **thread;
PRNetAddr listenAddr;
PRIntn timeout = DEFAULT_TIMEOUT;
PRIntn threads = DEFAULT_THREADS;
PRIntn backlog = DEFAULT_BACKLOG;
PRThreadScope thread_scope = PR_LOCAL_THREAD;
PLOptState *opt = PL_CreateOptState(argc, argv, "dGb:t:T:R");
#ifdef _PR_INET6
PR_SetIPv6Enable(PR_TRUE);
#endif
shared = PR_NEWZAP(Shared); /* this is leaked */
shared->debug = NULL;
shared->passed = PR_TRUE;
shared->random = PR_TRUE;
shared->status = running;
shared->ml = PR_NewLock();
shared->cv = PR_NewCondVar(shared->ml);
while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
{
if (PL_OPT_BAD == os) continue;
switch (opt->option)
{
case 'd': /* debug mode */
shared->debug = PR_GetSpecialFD(PR_StandardError);
break;
case 'G': /* use global threads */
thread_scope = PR_GLOBAL_THREAD;
break;
case 'b': /* size of listen backlog */
backlog = atoi(opt->value);
break;
case 't': /* number of threads doing accept */
threads = atoi(opt->value);
break;
case 'T': /* timeout used for network operations */
timeout = atoi(opt->value);
break;
case 'R': /* randomize the timeout values */
shared->random = PR_TRUE;
break;
default:
break;
}
}
PL_DestroyOptState(opt);
if (0 == threads) threads = DEFAULT_THREADS;
if (0 == backlog) backlog = DEFAULT_BACKLOG;
if (0 == timeout) timeout = DEFAULT_TIMEOUT;
PR_STDIO_INIT();
rv = PR_InitializeNetAddr(PR_IpAddrAny, BASE_PORT, &listenAddr);
PR_ASSERT(PR_SUCCESS == rv);
shared->timeout = PR_SecondsToInterval(timeout);
/* First bind to the socket */
shared->listenSock = PR_NewTCPSocket();
if (shared->listenSock)
{
rv = PR_Bind(shared->listenSock, &listenAddr);
if (rv != PR_FAILURE)
{
rv = PR_Listen(shared->listenSock, threads + backlog);
if (PR_SUCCESS == rv)
{
thread = (PRThread**)PR_CALLOC(threads * sizeof(PRThread*));
for (index = 0; index < threads; ++index)
{
thread[index] = PR_CreateThread(
PR_USER_THREAD, Accept, shared,
PR_PRIORITY_NORMAL, thread_scope,
PR_JOINABLE_THREAD, 0);
PR_ASSERT(NULL != thread[index]);
}
PR_Lock(shared->ml);
while (shared->status == running)
PR_WaitCondVar(shared->cv, PR_INTERVAL_NO_TIMEOUT);
PR_Unlock(shared->ml);
for (index = 0; index < threads; ++index)
{
rv = PR_Interrupt(thread[index]);
PR_ASSERT(PR_SUCCESS== rv);
rv = PR_JoinThread(thread[index]);
PR_ASSERT(PR_SUCCESS== rv);
}
PR_DELETE(thread);
}
else
{
if (shared->debug) PL_PrintError("Listen");
shared->passed = PR_FALSE;
}
}
else
{
if (shared->debug) PL_PrintError("Bind");
shared->passed = PR_FALSE;
}
PR_Close(shared->listenSock);
}
else
{
if (shared->debug) PL_PrintError("Create");
shared->passed = PR_FALSE;
}
PR_DestroyCondVar(shared->cv);
PR_DestroyLock(shared->ml);
PR_fprintf(
PR_GetSpecialFD(PR_StandardError), "%s\n",
((shared->passed) ? "PASSED" : "FAILED"));
return (shared->passed) ? 0 : 1;
}
int main(int argc, char **argv)
{
return (PR_VersionCheck(PR_VERSION)) ?
PR_Initialize(Tmoacc, argc, argv, 4) : -1;
} /* main */
/* tmoacc */