зеркало из https://github.com/mozilla/pjs.git
331 строка
11 KiB
C
331 строка
11 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.
|
|
*/
|
|
|
|
#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 exitStatus;
|
|
PRIntn index;
|
|
Shared *shared;
|
|
PLOptStatus os;
|
|
PRThread **thread;
|
|
PRNetAddr listenAddr;
|
|
PRSocketOptionData sockOpt;
|
|
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");
|
|
|
|
shared = PR_NEWZAP(Shared);
|
|
|
|
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();
|
|
memset(&listenAddr, 0, sizeof(listenAddr));
|
|
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)
|
|
{
|
|
sockOpt.option = PR_SockOpt_Reuseaddr;
|
|
sockOpt.value.reuse_addr = PR_TRUE;
|
|
rv = PR_SetSocketOption(shared->listenSock, &sockOpt);
|
|
PR_ASSERT(PR_SUCCESS == rv);
|
|
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"));
|
|
|
|
exitStatus = (shared->passed) ? 0 : 1;
|
|
PR_DELETE(shared);
|
|
return exitStatus;
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
return (PR_VersionCheck(PR_VERSION)) ?
|
|
PR_Initialize(Tmoacc, argc, argv, 4) : -1;
|
|
} /* main */
|
|
|
|
/* tmoacc */
|