mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
510 lines
11 KiB
510 lines
11 KiB
/*++
|
|
|
|
Copyright (C) Microsoft Corporation, 1994 - 1999
|
|
|
|
Module Name:
|
|
|
|
Server.c
|
|
|
|
Abstract:
|
|
|
|
Server side of RPC runtime performance test.
|
|
|
|
Author:
|
|
|
|
Mario Goertzel (mariogo) 31-Mar-1994
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include <rpcperf.h>
|
|
#include <rpcrt.h>
|
|
|
|
// Usage
|
|
const char *USAGE =
|
|
"[-i iterations] [-l logfile] [-n clients] [-n minthreads] [-n case]*\n"
|
|
" Client - default 1\n"
|
|
" MinThreads - default 3\n"
|
|
" Cases, you may specify up to five, default is all:\n"
|
|
" 0: Null Call\n"
|
|
" 1: Null Call (Non-Idem)\n"
|
|
" 2: 1K IN\n"
|
|
" 3: 1K OUT\n"
|
|
" 4: 4K IN\n"
|
|
" 5: 4K OUT\n"
|
|
" 6: 32K IN\n"
|
|
" 7: 32K OUT\n"
|
|
" 8: Null Call (Non-Idem) w/ Context handle\n"
|
|
" 9: Bind (fixed endpoint)\n"
|
|
" 10: Re-Bind\n"
|
|
" 11: Bind (dynamic endpoint)\n"
|
|
" 12: Async Null Call\n"
|
|
#ifdef WIN98
|
|
" 13: Async STA Null Call\n"
|
|
#endif
|
|
"\n"
|
|
" Note: Results are milliseconds/call/client\n"
|
|
;
|
|
|
|
//
|
|
// Globals making it easier.
|
|
//
|
|
|
|
int CurrentCase;
|
|
char TestCases[TEST_MAX];
|
|
|
|
long CurrentIter;
|
|
unsigned long *Results;
|
|
|
|
long Clients, ActiveClients, ClientsLeft;
|
|
|
|
CRITICAL_SECTION CritSec;
|
|
HANDLE GoEvent = 0;
|
|
HANDLE DoneEvent = 0;
|
|
|
|
char *TestNames[TEST_MAX] =
|
|
{
|
|
"Void Call",
|
|
"Void Call (non-idem)",
|
|
"1Kb Write (in)",
|
|
"1Kb Read (out)",
|
|
"4Kb Write (in)",
|
|
"4Kb Read (out)",
|
|
"32Kb Write (in)",
|
|
"32Kb Read (out)",
|
|
"Context Handle",
|
|
"Bind (fixed ep)",
|
|
"Re-bind",
|
|
"Bind (dynamic ep)",
|
|
"Async Void Call"
|
|
#ifdef WIN98
|
|
, "Async STA Void Call"
|
|
#endif
|
|
};
|
|
|
|
long TestIterFactors[TEST_MAX] =
|
|
{
|
|
1,1,
|
|
1,1,
|
|
4,4,
|
|
16,16,
|
|
1,
|
|
8,
|
|
2,
|
|
16,
|
|
1
|
|
#ifdef WIN98
|
|
,1
|
|
#endif
|
|
};
|
|
|
|
//
|
|
// Figure out what we're testing and start listening.
|
|
//
|
|
|
|
int __cdecl
|
|
main (int argc, char **argv)
|
|
{
|
|
unsigned int MinThreads;
|
|
unsigned long i, status;
|
|
RPC_BINDING_VECTOR *pBindingVector;
|
|
#ifdef WIN98
|
|
BOOL fIsPortseqWmsg;
|
|
HWND hWnd;
|
|
#endif
|
|
|
|
ParseArgv(argc, argv);
|
|
|
|
InitAllocator();
|
|
|
|
#ifdef WIN98
|
|
if (strcmp(Protseq, "mswmsg") == 0)
|
|
fIsPortseqWmsg = TRUE;
|
|
else
|
|
fIsPortseqWmsg = FALSE;
|
|
#endif
|
|
|
|
MinThreads = 3;
|
|
ClientsLeft = Clients = 1;
|
|
ActiveClients = 0;
|
|
|
|
if (Options[0] > 0)
|
|
ClientsLeft = Clients = Options[0];
|
|
|
|
Results = MIDL_user_allocate(4 * Clients);
|
|
|
|
if (Options[1] > 0)
|
|
MinThreads = Options[1];
|
|
|
|
if (Options[2] < 0)
|
|
{
|
|
memset(TestCases, 1, TEST_MAX);
|
|
}
|
|
else
|
|
{
|
|
memset(TestCases, 0, TEST_MAX);
|
|
for(i = 2; i < 7; i++)
|
|
{
|
|
if ( Options[i] < 0)
|
|
break;
|
|
if ( Options[i] >= TEST_MAX)
|
|
break;
|
|
TestCases[Options[i]] = 1;
|
|
}
|
|
}
|
|
|
|
CurrentCase = 0;
|
|
for(i = 0; i < TEST_MAX; i++)
|
|
{
|
|
if (TestCases[i])
|
|
{
|
|
CurrentCase = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (i == TEST_MAX)
|
|
{
|
|
printf("No test cases selected!\n");
|
|
return 1;
|
|
}
|
|
|
|
CurrentIter = Iterations / TestIterFactors[CurrentCase];
|
|
|
|
if (CurrentIter == 0) CurrentIter = 1;
|
|
|
|
InitializeCriticalSection(&CritSec);
|
|
|
|
GoEvent = CreateEvent(0,
|
|
TRUE,
|
|
FALSE,
|
|
0);
|
|
|
|
DoneEvent = CreateEvent(0,
|
|
TRUE,
|
|
FALSE,
|
|
0);
|
|
|
|
//
|
|
// Actually start the server
|
|
//
|
|
|
|
if (Endpoint)
|
|
{
|
|
status = RpcServerUseProtseqEp(Protseq, 100, Endpoint, 0);
|
|
CHECK_STATUS(status, "RpcServerUseProtseqEp");
|
|
}
|
|
else
|
|
{
|
|
char *string_binding;
|
|
|
|
status = RpcServerUseProtseq(Protseq, 100, 0);
|
|
CHECK_STATUS(status, "RpcServerUseProtseqEp");
|
|
|
|
status = RpcServerInqBindings(&pBindingVector);
|
|
CHECK_STATUS(status, "RpcServerInqBindings");
|
|
|
|
status = RpcEpRegister(RpcRuntimePerf_v2_0_s_ifspec,
|
|
pBindingVector,
|
|
0,
|
|
0);
|
|
CHECK_STATUS(status, "RpcEpRegister");
|
|
|
|
status = RpcBindingToStringBinding(pBindingVector->BindingH[0],
|
|
&string_binding);
|
|
CHECK_STATUS(status, "RpcBindingToStringBinding");
|
|
|
|
status = RpcStringBindingParse(string_binding,
|
|
0, 0, 0, &Endpoint, 0);
|
|
|
|
CHECK_STATUS(status, "RpcStringBindingParse");
|
|
printf("Listening to %s:[%s]\n\n", Protseq, Endpoint);
|
|
}
|
|
|
|
#ifdef WIN98
|
|
if (fIsPortseqWmsg)
|
|
{
|
|
hWnd = CreateSTAWindow("Perf Server");
|
|
if (hWnd == NULL)
|
|
{
|
|
printf("Creation of an STA window failed: %ld\n", GetLastError());
|
|
return 2;
|
|
}
|
|
I_RpcServerStartListening(hWnd);
|
|
}
|
|
#endif
|
|
|
|
status =
|
|
RpcServerRegisterIf(RpcRuntimePerf_v2_0_s_ifspec,0,0);
|
|
CHECK_STATUS(status, "RpcServerRegisterIf");
|
|
|
|
status = RpcServerRegisterAuthInfo(NULL,
|
|
RPC_C_AUTHN_WINNT,
|
|
NULL,
|
|
NULL);
|
|
CHECK_STATUS(status, "RpcServerRegisterAuthInfo");
|
|
|
|
printf("Base Iterations: %d, Clients %d, MinThreads %d\n",
|
|
Iterations,
|
|
Clients,
|
|
MinThreads);
|
|
|
|
printf("Server listening\n");
|
|
|
|
#ifndef WIN98
|
|
status = RpcServerListen(MinThreads, 100, 0);
|
|
CHECK_STATUS(status, "RpcServerListen");
|
|
#else
|
|
if (fIsPortseqWmsg)
|
|
{
|
|
status = RpcServerListen(MinThreads, 100, TRUE);
|
|
CHECK_STATUS(status, "RpcServerListen");
|
|
|
|
RunMessageLoop(hWnd);
|
|
}
|
|
else
|
|
{
|
|
status = RpcServerListen(MinThreads, 100, 0);
|
|
CHECK_STATUS(status, "RpcServerListen");
|
|
}
|
|
#endif
|
|
|
|
printf("This doesn't stop listening..hmm\n");
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// Control APIs that the client(s) call to sync on test cases, iterations
|
|
// and to report results.
|
|
//
|
|
|
|
error_status_t
|
|
BeginTest(handle_t b,
|
|
long *ClientId)
|
|
{
|
|
long status = 0;
|
|
EnterCriticalSection(&CritSec);
|
|
|
|
if (ActiveClients < Clients)
|
|
{
|
|
ActiveClients++;
|
|
*ClientId = ActiveClients;
|
|
}
|
|
else
|
|
{
|
|
status = PERF_TOO_MANY_CLIENTS;
|
|
}
|
|
LeaveCriticalSection(&CritSec);
|
|
|
|
return status;
|
|
}
|
|
|
|
error_status_t
|
|
NextTest(handle_t b,
|
|
TEST_TYPE *Test,
|
|
long *Iters)
|
|
{
|
|
long wait = 1;
|
|
long done = 0;
|
|
int i;
|
|
|
|
EnterCriticalSection(&CritSec);
|
|
|
|
*Test = CurrentCase;
|
|
*Iters = CurrentIter;
|
|
|
|
ClientsLeft--;
|
|
|
|
if (CurrentCase == TEST_MAX)
|
|
{
|
|
done = 1;
|
|
}
|
|
|
|
if (ClientsLeft == 0)
|
|
{
|
|
//
|
|
// Let all the waiting clients go
|
|
//
|
|
wait = 0;
|
|
ResetEvent(DoneEvent);
|
|
SetEvent(GoEvent);
|
|
}
|
|
|
|
LeaveCriticalSection(&CritSec);
|
|
|
|
if (wait)
|
|
{
|
|
WaitForSingleObject(GoEvent, INFINITE);
|
|
}
|
|
|
|
if (done)
|
|
{
|
|
|
|
if (ClientsLeft == 0)
|
|
{
|
|
// I'm the last client, sleep and then reset for
|
|
// a new set of clients. Sleep avoids a race (usually).
|
|
|
|
Sleep(1000);
|
|
|
|
for(i = 0; i < TEST_MAX; i++)
|
|
{
|
|
if (TestCases[i])
|
|
{
|
|
CurrentCase = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
CurrentIter = Iterations / TestIterFactors[CurrentCase];
|
|
if (CurrentIter == 0) CurrentIter = 1;
|
|
|
|
ActiveClients = 0;
|
|
ClientsLeft = Clients;
|
|
|
|
ResetEvent(GoEvent);
|
|
ResetEvent(DoneEvent);
|
|
}
|
|
|
|
return PERF_TESTS_DONE;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
error_status_t
|
|
EndTest(handle_t b,
|
|
unsigned long mseconds)
|
|
{
|
|
long status, i;
|
|
long wait = 1;
|
|
|
|
EnterCriticalSection(&CritSec);
|
|
|
|
Results[ClientsLeft] = mseconds;
|
|
|
|
ClientsLeft++;
|
|
|
|
if (ClientsLeft == Clients)
|
|
{
|
|
// All clients have finished
|
|
|
|
// Report results
|
|
|
|
printf("| % 3d | %-20s | % 6d |",
|
|
CurrentCase,
|
|
TestNames[CurrentCase],
|
|
CurrentIter
|
|
);
|
|
|
|
for(i = 0; i < Clients; i++)
|
|
printf(" % 3d.%03d |",
|
|
Results[i] / CurrentIter,
|
|
Results[i] % CurrentIter * 1000 / CurrentIter
|
|
);
|
|
|
|
printf("\n");
|
|
|
|
// Setup next case
|
|
|
|
for(i = CurrentCase + 1; i < TEST_MAX; i++)
|
|
{
|
|
#ifdef PROTOCOL_TRIM
|
|
if (i == 11)
|
|
continue;
|
|
#endif
|
|
|
|
#ifndef WIN98
|
|
// NT does not test Async STA calls.
|
|
if (i == 13)
|
|
continue;
|
|
#else
|
|
// Win9x also doesn't want to test Async STA calls unless the protseq is
|
|
|
|
if ((i == 13) && (strcmp(Protseq, "mswmsg") != 0))
|
|
continue;
|
|
#endif
|
|
|
|
if (TestCases[i])
|
|
{
|
|
CurrentCase = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (i == TEST_MAX)
|
|
{
|
|
CurrentCase = TEST_MAX;
|
|
printf("TEST DONE\n");
|
|
}
|
|
else
|
|
{
|
|
CurrentIter = Iterations / TestIterFactors[CurrentCase];
|
|
if (CurrentIter == 0) CurrentIter = 1;
|
|
}
|
|
//
|
|
// We're setup for the next test (or to finish) let the clients go.
|
|
//
|
|
|
|
wait = 0;
|
|
ResetEvent(GoEvent);
|
|
SetEvent(DoneEvent);
|
|
|
|
}
|
|
LeaveCriticalSection(&CritSec);
|
|
|
|
if (wait)
|
|
WaitForSingleObject(DoneEvent, INFINITE);
|
|
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// For fixed endpoint and re-bind test case
|
|
//
|
|
|
|
unsigned char *GetFixedEp(handle_t h)
|
|
{
|
|
char *r;
|
|
|
|
r = malloc(strlen(Endpoint) + 1);
|
|
strcpy(r, Endpoint);
|
|
return (unsigned char *)r;
|
|
}
|
|
|
|
//
|
|
// For context handle tests
|
|
//
|
|
|
|
PERF_CONTEXT OpenContext (handle_t b) { return (PERF_CONTEXT)1; }
|
|
void CloseContext(PERF_CONTEXT *pp) { *pp = 0; }
|
|
|
|
void PERF_CONTEXT_rundown(PERF_CONTEXT pp)
|
|
{
|
|
printf("Failure - a context randown\n");
|
|
}
|
|
|
|
//
|
|
// Regular test calls do nothing...
|
|
//
|
|
|
|
void NullCall(handle_t h) { return; }
|
|
void NICall (handle_t h) { return; }
|
|
void ContextNullCall (PERF_CONTEXT h) { return; }
|
|
void Write1K (handle_t h, unsigned char *p) { return; }
|
|
void Read1K (handle_t h, unsigned char *p) { return; }
|
|
void Write4K (handle_t h, unsigned char *p) { return; }
|
|
void Read4K (handle_t h, unsigned char *p) { return; }
|
|
void Write32K(handle_t h, unsigned char *p) { return; }
|
|
void Read32K (handle_t h, unsigned char *p) { return; }
|
|
void AsyncNullCall(RPC_ASYNC_STATE *AsyncHandle, handle_t h)
|
|
{
|
|
RpcAsyncCompleteCall(AsyncHandle, NULL);
|
|
}
|
|
|
|
void AsyncSTANullCall(RPC_ASYNC_STATE *AsyncHandle, handle_t h)
|
|
{
|
|
RpcAsyncCompleteCall(AsyncHandle, NULL);
|
|
}
|
|
|