windows-nt/Source/XPSP1/NT/com/rpc/perf/scale/rtclnt.c

563 lines
15 KiB
C
Raw Normal View History

2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1994 Microsoft Corporation
Module Name:
Rtclnt.c
Abstract:
Client side of basic RPC scale performance test.
Author:
Mario Goertzel (mariogo) 31-Mar-1994
Revision History:
18/7/96 [MarioGo] - Cloned from rpcrt test
--*/
#include <rpcperf.h>
#include <rpcrt.h>
#ifdef MAC
extern void _cdecl PrintToConsole(const char *lpszFormat, ...) ;
extern unsigned long ulSecurityPackage ;
#else
#define PrintToConsole printf
extern unsigned long ulSecurityPackage;
#endif
// Usage
const char *USAGE = "-n <threads> -a <authnlevel> -s <server> -t <protseq>\n"
"Server controls iterations, test cases, and compiles the results.\n"
"AuthnLevel: none, connect, call, pkt, integrity, privacy.\n"
"Default threads=1, authnlevel=none\n";
unsigned long gInSize = 500 - IN_ADJUSTMENT;
unsigned long gOutSize = 500 - OUT_ADJUSTMENT;
#ifdef STATS
// exported by RPCRT4
void I_RpcGetStats(DWORD *pdwStat1, DWORD *pdwStat2, DWORD *pdwStat3, DWORD *pdwStat4);
#endif
// Error stuff
#define CHECK_RET(status, string) if (status)\
{ PrintToConsole("%s failed -- %lu (0x%08X)\n", string,\
(unsigned long)status, (unsigned long)status);\
return (status); }
RPC_STATUS DoRpcBindingSetAuthInfo(handle_t Binding)
{
if (AuthnLevel != RPC_C_AUTHN_LEVEL_NONE)
return RpcBindingSetAuthInfoA(Binding,
ServerPrincipalName,
AuthnLevel,
ulSecurityPackage,
NULL,
RPC_C_AUTHZ_NONE);
else
return(RPC_S_OK);
}
//
// Test wrappers
//
unsigned long DoNullCall(handle_t *pb, char *stringBinding, unsigned long c, char __RPC_FAR *p)
{
unsigned long status;
unsigned long calls = 0;
do
{
status = NullCall(*pb, c);
calls++;
}
while (status == 0);
if (status != PERF_TESTS_DONE)
{
RpcRaiseException(status);
}
return(calls);
}
unsigned long DoMaybeCall(handle_t *pb, char *stringBinding, unsigned long c, char __RPC_FAR *p)
{
unsigned long calls = 0;
for(calls = 10000; calls; calls--)
{
MaybeCall(*pb, c);
}
return(10000);
}
unsigned long DoNICall(handle_t *pb, char *stringBinding, unsigned long c, char __RPC_FAR *p)
{
unsigned long status;
unsigned long calls = 0;
do
{
status = NICall(*pb, c);
calls++;
}
while (status == 0);
if (status != PERF_TESTS_DONE)
{
RpcRaiseException(status);
}
return(calls);
}
unsigned long DoBufferCall(handle_t *pb, char *stringBinding, unsigned long c, char __RPC_FAR *p)
{
unsigned long status;
unsigned long calls = 0;
do
{
RpcTryExcept
{
status = BufferCall(*pb, c, gInSize, p, gOutSize, p);
}
RpcExcept(1)
{
PrintToConsole("\nException %lu (0x%08lX)\n",
(unsigned long)RpcExceptionCode(),
(unsigned long)RpcExceptionCode());
}
RpcEndExcept
calls++;
}
while (status != PERF_TESTS_DONE);
if (status != PERF_TESTS_DONE)
{
RpcRaiseException(status);
}
return(calls);
}
unsigned long DoBindCall(handle_t *pb, char *stringBinding, unsigned long c, char __RPC_FAR *p)
{
unsigned long status;
unsigned long calls = 0;
do
{
status = RpcBindingFree(pb);
if (status)
{
break;
}
status = RpcBindingFromStringBindingA(stringBinding, pb);
if (status)
{
break;
}
status = NullCall(*pb, c);
calls++;
}
while (status == 0);
if (status != PERF_TESTS_DONE)
{
RpcRaiseException(status);
}
return(calls);
}
static const unsigned long (*TestTable[TEST_MAX])(handle_t *pb, char *stringBinding, unsigned long, char __RPC_FAR *) =
{
DoNullCall,
DoNICall,
DoBufferCall,
DoMaybeCall,
DoBindCall
};
//
// Worker calls the correct tests. Maybe multithreaded on NT
//
unsigned long Worker(unsigned long l)
{
unsigned long status;
unsigned long Test;
unsigned long ClientId;
unsigned long InSize, OutSize;
unsigned long Time, Calls;
char __RPC_FAR *pBuffer;
char __RPC_FAR *stringBinding;
handle_t binding;
RPC_STATUS RpcErr;
int Retries;
pBuffer = MIDL_user_allocate(128*1024L);
if (pBuffer == 0)
{
PrintToConsole("Out of memory!");
return 1;
}
status =
RpcStringBindingComposeA(0,
Protseq,
NetworkAddr,
Endpoint,
0,
&stringBinding);
CHECK_RET(status, "RpcStringBindingCompose");
status =
RpcBindingFromStringBindingA(stringBinding, &binding);
CHECK_RET(status, "RpcBindingFromStringBinding");
status =
DoRpcBindingSetAuthInfo(binding);
CHECK_RET(status, "RpcBindingSetAuthInfo");
Retries = 15;
do
{
status = BeginTest(binding, &ClientId, &Test, &InSize, &OutSize);
if (status == PERF_TOO_MANY_CLIENTS)
{
PrintToConsole("Too many clients, I'm exiting\n");
goto Cleanup ;
}
Retries --;
if ((status == RPC_S_SERVER_UNAVAILABLE) && (Retries > 0))
{
PrintToConsole("Server too busy - retrying ...\n");
}
}
while ((status == RPC_S_SERVER_UNAVAILABLE) && (Retries > 0));
if (status)
{
RPC_STATUS Status2;
RPC_ERROR_ENUM_HANDLE EnumHandle;
Status2 = RpcErrorStartEnumeration(&EnumHandle);
if (Status2 == RPC_S_ENTRY_NOT_FOUND)
{
CHECK_RET(status, "ClientConnect");
}
else if (Status2 != RPC_S_OK)
{
PrintToConsole("Couldn't get EEInfo: %d\n", Status2);
CHECK_RET(status, "ClientConnect");
}
else
{
RPC_EXTENDED_ERROR_INFO ErrorInfo;
int Records;
BOOL Result;
BOOL CopyStrings = TRUE;
PVOID Blob;
size_t BlobSize;
BOOL fUseFileTime = TRUE;
SYSTEMTIME *SystemTimeToUse;
SYSTEMTIME SystemTimeBuffer;
Status2 = RpcErrorGetNumberOfRecords(&EnumHandle, &Records);
if (Status2 == RPC_S_OK)
{
PrintToConsole("Number of records is: %d\n", Records);
}
while (Status2 == RPC_S_OK)
{
ErrorInfo.Version = RPC_EEINFO_VERSION;
ErrorInfo.Flags = 0;
ErrorInfo.NumberOfParameters = 4;
if (fUseFileTime)
{
ErrorInfo.Flags |= EEInfoUseFileTime;
}
Status2 = RpcErrorGetNextRecord(&EnumHandle, CopyStrings, &ErrorInfo);
if (Status2 == RPC_S_ENTRY_NOT_FOUND)
{
RpcErrorResetEnumeration(&EnumHandle);
break;
}
else if (Status2 != RPC_S_OK)
{
PrintToConsole("Couldn't finish enumeration: %d\n", Status2);
break;
}
else
{
int i;
if (ErrorInfo.ComputerName)
{
PrintToConsole("ComputerName is %S\n", ErrorInfo.ComputerName);
if (CopyStrings)
{
Result = HeapFree(GetProcessHeap(), 0, ErrorInfo.ComputerName);
ASSERT(Result);
}
}
PrintToConsole("ProcessID is %d\n", ErrorInfo.ProcessID);
if (fUseFileTime)
{
Result = FileTimeToSystemTime(&ErrorInfo.u.FileTime, &SystemTimeBuffer);
ASSERT(Result);
SystemTimeToUse = &SystemTimeBuffer;
}
else
SystemTimeToUse = &ErrorInfo.u.SystemTime;
PrintToConsole("System Time is: %d/%d/%d %d:%d:%d:%d\n",
SystemTimeToUse->wMonth,
SystemTimeToUse->wDay,
SystemTimeToUse->wYear,
SystemTimeToUse->wHour,
SystemTimeToUse->wMinute,
SystemTimeToUse->wSecond,
SystemTimeToUse->wMilliseconds);
PrintToConsole("Generating component is %d\n", ErrorInfo.GeneratingComponent);
PrintToConsole("Status is %d\n", ErrorInfo.Status);
PrintToConsole("Detection location is %d\n", (int)ErrorInfo.DetectionLocation);
PrintToConsole("Flags is %d\n", ErrorInfo.Flags);
PrintToConsole("NumberOfParameters is %d\n", ErrorInfo.NumberOfParameters);
for (i = 0; i < ErrorInfo.NumberOfParameters; i ++)
{
switch(ErrorInfo.Parameters[i].ParameterType)
{
case eeptAnsiString:
PrintToConsole("Ansi string: %s\n", ErrorInfo.Parameters[i].u.AnsiString);
if (CopyStrings)
{
Result = HeapFree(GetProcessHeap(), 0, ErrorInfo.Parameters[i].u.AnsiString);
ASSERT(Result);
}
break;
case eeptUnicodeString:
PrintToConsole("Unicode string: %S\n", ErrorInfo.Parameters[i].u.UnicodeString);
if (CopyStrings)
{
Result = HeapFree(GetProcessHeap(), 0, ErrorInfo.Parameters[i].u.UnicodeString);
ASSERT(Result);
}
break;
case eeptLongVal:
PrintToConsole("Long val: %d\n", ErrorInfo.Parameters[i].u.LVal);
break;
case eeptShortVal:
PrintToConsole("Short val: %d\n", (int)ErrorInfo.Parameters[i].u.SVal);
break;
case eeptPointerVal:
PrintToConsole("Pointer val: %d\n", ErrorInfo.Parameters[i].u.PVal);
break;
case eeptNone:
PrintToConsole("Truncated\n");
break;
default:
PrintToConsole("Invalid type: %d\n", ErrorInfo.Parameters[i].ParameterType);
}
}
}
}
Status2 = RpcErrorSaveErrorInfo(&EnumHandle, &Blob, &BlobSize);
CHECK_RET(Status2, "RpcErrorSaveErrorInfo");
RpcErrorClearInformation();
RpcErrorEndEnumeration(&EnumHandle);
Status2 = RpcErrorLoadErrorInfo(Blob, BlobSize, &EnumHandle);
CHECK_RET(Status2, "RpcErrorLoadErrorInfo");
}
}
CHECK_RET(status, "ClientConnect");
if (InSize > IN_ADJUSTMENT)
{
InSize -= IN_ADJUSTMENT;
}
else
{
InSize = 0;
}
if (OutSize > OUT_ADJUSTMENT)
{
OutSize -= OUT_ADJUSTMENT;
}
else
{
OutSize = 0;
}
gInSize = InSize;
gOutSize = OutSize;
PrintToConsole("Client %ld connected\n", ClientId);
Retries = 15;
do
{
RpcTryExcept
{
RpcErr = RPC_S_OK;
Time = GetTickCount();
Calls = ( (TestTable[Test])(&binding, stringBinding, ClientId, pBuffer) );
Time = GetTickCount() - Time;
Dump("Completed %d calls in %d ms\n"
"%d T/S or %3d.%03d ms/T\n\n",
Calls,
Time,
(Calls * 1000) / Time,
Time / Calls,
((Time % Calls) * 1000) / Calls
);
}
RpcExcept(1)
{
RpcErr = (unsigned long)RpcExceptionCode();
PrintToConsole("\nException %lu (0x%08lX)\n",
RpcErr, RpcErr);
}
RpcEndExcept
Retries --;
if ((RpcErr == RPC_S_SERVER_UNAVAILABLE) && (Retries > 0))
{
PrintToConsole("Server too busy - retrying ...\n");
}
}
while ((RpcErr == RPC_S_SERVER_UNAVAILABLE) && (Retries > 0));
Cleanup:
RpcBindingFree(&binding);
return status;
}
//
// The Win32 main starts worker threads, otherwise we just call the worker.
//
#ifdef WIN32
int __cdecl
main (int argc, char **argv)
{
char option;
unsigned long status, i;
HANDLE *pClientThreads;
DWORD dwStat1, dwStat2, dwStat3, dwStat4;
InitAllocator();
ulSecurityPackage = RPC_C_AUTHN_WINNT;
ParseArgv(argc, argv);
PrintToConsole("Authentication Level is: %s\n", AuthnLevelStr);
if (Options[0] < 0)
Options[0] = 1;
pClientThreads = MIDL_user_allocate(sizeof(HANDLE) * Options[0]);
for(i = 0; i < (unsigned long)Options[0]; i++)
{
pClientThreads[i] = CreateThread(0,
0,
(LPTHREAD_START_ROUTINE)Worker,
0,
0,
&status);
if (pClientThreads[i] == 0)
ApiError("CreateThread", GetLastError());
}
status = WaitForMultipleObjects(Options[0],
pClientThreads,
TRUE, // Wait for all client threads
INFINITE);
if (status == WAIT_FAILED)
{
ApiError("WaitForMultipleObjects", GetLastError());
}
#ifdef STATS
I_RpcGetStats(&dwStat1, &dwStat2, &dwStat3, &dwStat4);
printf("Stats are: %ld, %ld, %ld, %ld\n", dwStat1, dwStat2, dwStat3, dwStat4);
#endif
PrintToConsole("TEST DONE\n");
return(0);
}
#else // !WIN32
#ifdef WIN
#define main c_main
// We need the following to force the linker to load WinMain from the
// Windows STDIO library
extern int PASCAL WinMain(HANDLE, HANDLE, LPSTR, int);
static int (PASCAL *wm_ptr)(HANDLE, HANDLE, LPSTR, int) = WinMain;
#endif
#ifndef MAC
#ifndef FAR
#define FAR __far
#endif
#else
#define FAR
#define main c_main
#endif
int main (int argc, char FAR * FAR * argv)
{
#ifndef MAC
ParseArgv(argc, argv);
#endif
Worker(0);
PrintToConsole("TEST DONE\n");
return(0);
}
#endif // NTENV