pjs/nsprpub/pr/tests/tmocon.c

402 строки
13 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.
*/
/***********************************************************************
**
** Name: tmocon.c
**
** Description: test client socket connection.
**
** Modification History:
** 19-May-97 AGarcia- Converted the test to accomodate the debug_mode flag.
** 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.
***********************************************************************/
/***********************************************************************
** Includes
***********************************************************************/
/* Used to get the command line option */
#include "plgetopt.h"
#include "nspr.h"
#include "pprio.h"
#include "plerror.h"
#include "plgetopt.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* for getcwd */
#if defined(XP_UNIX) || defined (XP_OS2_EMX) || defined(XP_BEOS)
#include <unistd.h>
#elif defined(XP_PC)
#include <direct.h>
#endif
#ifdef XP_MAC
#include "prlog.h"
#define printf PR_LogPrint
#endif
#define BASE_PORT 9867
#define DEFAULT_DALLY 1
#define DEFAULT_THREADS 1
#define DEFAULT_TIMEOUT 10
#define DEFAULT_MESSAGES 100
#define DEFAULT_MESSAGESIZE 100
static PRFileDesc *debug_out = NULL;
typedef struct Shared
{
PRBool random;
PRBool failed;
PRBool intermittant;
PRIntn debug;
PRInt32 messages;
PRIntervalTime dally;
PRIntervalTime timeout;
PRInt32 message_length;
PRNetAddr serverAddress;
} Shared;
static PRIntervalTime Timeout(const Shared *shared)
{
PRIntervalTime timeout = shared->timeout;
if (shared->random)
{
PRIntervalTime quarter = timeout >> 2; /* one quarter of the interval */
PRUint32 random = rand() % quarter; /* something in[0..timeout / 4) */
timeout = (((3 * quarter) + random) >> 2) + quarter; /* [75..125)% */
}
return timeout;
} /* Timeout */
static void CauseTimeout(const Shared *shared)
{
if (shared->intermittant) PR_Sleep(Timeout(shared));
} /* CauseTimeout */
static PRStatus MakeReceiver(Shared *shared)
{
PRStatus rv = PR_FAILURE;
if (PR_IsNetAddrType(&shared->serverAddress, PR_IpAddrLoopback))
{
char *argv[3];
char path[1024 + sizeof("/tmoacc")];
(void)getcwd(path, sizeof(path));
(void)strcat(path, "/tmoacc");
#ifdef XP_PC
(void)strcat(path, ".exe");
#endif
argv[0] = path;
if (shared->debug > 0)
{
argv[1] = "-d";
argv[2] = NULL;
}
else argv[1] = NULL;
if (shared->debug > 1)
PR_fprintf(debug_out, " creating accept process %s ...", path);
fflush(stdout);
rv = PR_CreateProcessDetached(path, argv, NULL, NULL);
if (PR_SUCCESS == rv)
{
if (shared->debug > 1)
PR_fprintf(debug_out, " wait 5 seconds");
if (shared->debug > 1)
PR_fprintf(debug_out, " before connecting to accept process ...");
fflush(stdout);
PR_Sleep(PR_SecondsToInterval(5));
return rv;
}
shared->failed = PR_TRUE;
if (shared->debug > 0)
PL_FPrintError(debug_out, "PR_CreateProcessDetached failed");
}
return rv;
} /* MakeReceiver */
static void Connect(void *arg)
{
PRStatus rv;
char *buffer = NULL;
PRFileDesc *clientSock;
Shared *shared = (Shared*)arg;
PRInt32 loop, bytes, flags = 0;
struct Descriptor { PRInt32 length; PRUint32 checksum; } descriptor;
debug_out = (0 == shared->debug) ? NULL : PR_GetSpecialFD(PR_StandardError);
buffer = (char*)PR_MALLOC(shared->message_length);
for (bytes = 0; bytes < shared->message_length; ++bytes)
buffer[bytes] = (char)bytes;
descriptor.checksum = 0;
for (bytes = 0; bytes < shared->message_length; ++bytes)
{
PRUint32 overflow = descriptor.checksum & 0x80000000;
descriptor.checksum = (descriptor.checksum << 1);
if (0x00000000 != overflow) descriptor.checksum += 1;
descriptor.checksum += buffer[bytes];
}
descriptor.checksum = PR_htonl(descriptor.checksum);
for (loop = 0; loop < shared->messages; ++loop)
{
if (shared->debug > 1)
PR_fprintf(debug_out, "[%d]socket ... ", loop);
clientSock = PR_NewTCPSocket();
if (clientSock)
{
/*
* We need to slow down the rate of generating connect requests,
* otherwise the listen backlog queue on the accept side may
* become full and we will get connection refused or timeout
* error.
*/
PR_Sleep(shared->dally);
if (shared->debug > 1)
{
char buf[128];
PR_NetAddrToString(&shared->serverAddress, buf, sizeof(buf));
PR_fprintf(debug_out, "connecting to %s ... ", buf);
}
rv = PR_Connect(
clientSock, &shared->serverAddress, Timeout(shared));
if (PR_SUCCESS == rv)
{
PRInt32 descriptor_length = (loop < (shared->messages - 1)) ?
shared->message_length : 0;
descriptor.length = PR_htonl(descriptor_length);
if (shared->debug > 1)
PR_fprintf(
debug_out, "sending %d bytes ... ", descriptor_length);
CauseTimeout(shared); /* might cause server to timeout */
bytes = PR_Send(
clientSock, &descriptor, sizeof(descriptor),
flags, Timeout(shared));
if (bytes != sizeof(descriptor))
{
shared->failed = PR_TRUE;
if (shared->debug > 0)
PL_FPrintError(debug_out, "PR_Send failed");
}
if (0 != descriptor_length)
{
CauseTimeout(shared);
bytes = PR_Send(
clientSock, buffer, descriptor_length,
flags, Timeout(shared));
if (bytes != descriptor_length)
{
shared->failed = PR_TRUE;
if (shared->debug > 0)
PL_FPrintError(debug_out, "PR_Send failed");
}
}
if (shared->debug > 1) PR_fprintf(debug_out, "closing ... ");
rv = PR_Shutdown(clientSock, PR_SHUTDOWN_BOTH);
rv = PR_Close(clientSock);
if (shared->debug > 1)
{
if (PR_SUCCESS == rv) PR_fprintf(debug_out, "\n");
else PL_FPrintError(debug_out, "shutdown failed");
}
}
else
{
if (shared->debug > 1) PL_FPrintError(debug_out, "connect failed");
PR_Close(clientSock);
if ((loop == 0) && (PR_GetError() == PR_CONNECT_REFUSED_ERROR))
{
if (MakeReceiver(shared) == PR_FAILURE) break;
}
else
{
if (shared->debug > 1) PR_fprintf(debug_out, " exiting\n");
break;
}
}
}
else
{
shared->failed = PR_TRUE;
if (shared->debug > 0) PL_FPrintError(debug_out, "create socket");
break;
}
}
PR_DELETE(buffer);
} /* Connect */
int Tmocon(int argc, char **argv)
{
/*
* USAGE
* -d turn on debugging output (default = off)
* -v turn on verbose output (default = off)
* -h <n> dns name of host serving the connection (default = self)
* -i dally intermittantly to cause timeouts (default = off)
* -m <n> number of messages to send (default = 100)
* -s <n> size of each message (default = 100)
* -t <n> number of threads sending (default = 1)
* -G use global threads (default = local)
* -T <n> timeout on I/O operations (seconds) (default = 10)
* -D <n> dally between connect requests (seconds)(default = 0)
* -R randomize the dally types around 'T' (default = no)
*/
PRStatus rv;
int exitStatus;
PLOptStatus os;
Shared *shared = NULL;
PRThread **thread = NULL;
PRIntn index, threads = DEFAULT_THREADS;
PRThreadScope thread_scope = PR_LOCAL_THREAD;
PRInt32 dally = DEFAULT_DALLY, timeout = DEFAULT_TIMEOUT;
PLOptState *opt = PL_CreateOptState(argc, argv, "divGRh:m:s:t:T:D:");
shared = PR_NEWZAP(Shared);
shared->debug = 0;
shared->failed = PR_FALSE;
shared->random = PR_FALSE;
shared->messages = DEFAULT_MESSAGES;
shared->message_length = DEFAULT_MESSAGESIZE;
PR_STDIO_INIT();
memset(&shared->serverAddress, 0, sizeof(shared->serverAddress));
rv = PR_InitializeNetAddr(PR_IpAddrLoopback, BASE_PORT, &shared->serverAddress);
PR_ASSERT(PR_SUCCESS == rv);
while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
{
if (PL_OPT_BAD == os) continue;
switch (opt->option)
{
case 'd':
if (0 == shared->debug) shared->debug = 1;
break;
case 'v':
if (0 == shared->debug) shared->debug = 2;
break;
case 'i':
shared->intermittant = PR_TRUE;
break;
case 'R':
shared->random = PR_TRUE;
break;
case 'G':
thread_scope = PR_GLOBAL_THREAD;
break;
case 'h': /* the value for backlock */
{
PRIntn es = 0;
PRHostEnt host;
char buffer[1024];
(void)PR_GetHostByName(
opt->value, buffer, sizeof(buffer), &host);
es = PR_EnumerateHostEnt(
es, &host, BASE_PORT, &shared->serverAddress);
PR_ASSERT(es > 0);
}
break;
case 'm': /* number of messages to send */
shared->messages = atoi(opt->value);
break;
case 't': /* number of threads sending */
threads = atoi(opt->value);
break;
case 'D': /* dally time between transmissions */
dally = atoi(opt->value);
break;
case 'T': /* timeout on I/O operations */
timeout = atoi(opt->value);
break;
case 's': /* total size of each message */
shared->message_length = atoi(opt->value);
break;
default:
break;
}
}
PL_DestroyOptState(opt);
if (0 == timeout) timeout = DEFAULT_TIMEOUT;
if (0 == threads) threads = DEFAULT_THREADS;
if (0 == shared->messages) shared->messages = DEFAULT_MESSAGES;
if (0 == shared->message_length) shared->message_length = DEFAULT_MESSAGESIZE;
shared->dally = PR_SecondsToInterval(dally);
shared->timeout = PR_SecondsToInterval(timeout);
thread = (PRThread**)PR_CALLOC(threads * sizeof(PRThread*));
for (index = 0; index < threads; ++index)
thread[index] = PR_CreateThread(
PR_USER_THREAD, Connect, shared,
PR_PRIORITY_NORMAL, thread_scope,
PR_JOINABLE_THREAD, 0);
for (index = 0; index < threads; ++index)
rv = PR_JoinThread(thread[index]);
PR_DELETE(thread);
PR_fprintf(
PR_GetSpecialFD(PR_StandardError), "%s\n",
((shared->failed) ? "FAILED" : "PASSED"));
exitStatus = (shared->failed) ? 1 : 0;
PR_DELETE(shared);
return exitStatus;
}
int main(int argc, char **argv)
{
return (PR_VersionCheck(PR_VERSION)) ?
PR_Initialize(Tmocon, argc, argv, 4) : -1;
} /* main */
/* tmocon.c */