/* * * In order to test rpc bind caching as best as we can, the cache limit * is set to 3, and then we attempt to bind to four services. So that * it forces one handle out of the cache each time. * * If we have multiple threads doing this at the same time, then we may * randomly get the use counts up to something other than zero. This * way, some of the other caching logic can be exercised. * * * */ //---------- // Includes //---------- #include // DbgPrint prototype #include // DbgPrint prototype #include // needed for winbase.h #include // atoi #include // printf #include // strcmp #include // getch #include // NERR_ error codes #include // win32 typedefs #include #include // NetShareEnum #include // NetRemoteTOD #include // NetUseEnum #include // NetMessageNameEnum #include // NetApiBufferFree #include // Unicode //--------------------------- // DEFINES //--------------------------- #define RETURN_ALL 0xffffffff //--------------------------- // GLOBALS //--------------------------- LPTSTR GlobalServerName; // Passed in ServerName DWORD GlobalLoopCount; // Number of times for each thread to loop //--------------------------- // Local Function Prototypes //--------------------------- VOID BindCachingTest( DWORD ThreadNum, DWORD LoopCount, LPTSTR ServerName ); DWORD StartThread( LPVOID ThreadNum ); VOID BindCachingTest2( DWORD ThreadNum, DWORD LoopCount, LPTSTR ServerName ); BOOL ConvertToUnicode( OUT LPWSTR *UnicodeOut, IN LPSTR AnsiIn ); VOID SingleServiceTest( VOID ); /***************************************************************************/ VOID _CRTAPI1 main ( DWORD argc, PCHAR argv[] ) /*++ Routine Description: Arguments: Return Value: Note: --*/ { DWORD rc; DWORD threadId; HANDLE threadHandles[200]; DWORD NumThreads; DWORD i; if (argc < 2) { printf("ERROR: must supply a num Threads. ServerName Optional\n"); printf("bindtest 2 \\\\DANL2 2\n"); printf("bindtest 2 ONE_API /t- multiple calls to only one service\n"); return; } NumThreads = atol(argv[1]); GlobalLoopCount = 1; if (argc > 2 ) { #ifdef UNICODE ConvertToUnicode ((LPWSTR *)&GlobalServerName, argv[2]); #else GlobalServerName = argv[2]; #endif // // Look for a special server name "ONE_API". If this name is // found it means that we should do a single test with a single // service - so se can play with the state of the service between // bindings. // if (_wcsicmp(GlobalServerName, L"ONE_API") ==0) { SingleServiceTest(); return; } } else { GlobalServerName = NULL; } //------------------------------ // Do Serial Thread Tests //------------------------------ printf("Starting Serial Test\n"); threadHandles[0] = CreateThread( NULL, 0L, (LPTHREAD_START_ROUTINE)StartThread, (LPVOID)0, 0L, &threadId); if(threadHandles[0] == NULL) { printf("CreateThread failed %d\n",GetLastError()); } else { printf("Thread #%d Created\n",0); } rc = WaitForSingleObject(threadHandles[0], 60000); printf("Creating 2nd thread for Serial Test\n"); threadHandles[0] = CreateThread( NULL, 0L, (LPTHREAD_START_ROUTINE)StartThread, (LPVOID)0, 0L, &threadId); if(threadHandles[0] == NULL) { printf("CreateThread failed %d\n",GetLastError()); } else { printf("Thread #%d Created\n",0); } rc = WaitForSingleObject(threadHandles[0], 60000); printf("Serial Test Done - Starting Simultaneous Test\n\n"); //------------------------------ // Do Simultaneous Thread Tests //------------------------------ for(i=0; itod_hours, pTOD->tod_mins, pTOD->tod_secs, pTOD->tod_month, pTOD->tod_day, pTOD->tod_year); NetApiBufferFree(bufPtr); } //---------------------------------------- // SERVER: NetShareEnum (NULL ServerName) //---------------------------------------- printf("%d:NetShare\n",ThreadNum); rc = NetShareEnum( NULL, 2, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetShareEnum failed 0x%ld\n", rc); } else { printf("%d:NetShare done\n",ThreadNum); NetApiBufferFree(bufPtr); } Sleep(200); //--------------------------------------------- // MESSENGER: NetMessageNameEnum // This time it should re-use the bind handle. //--------------------------------------------- printf("%d:NetMessageName\n",ThreadNum); rc = NetMessageNameEnum( NULL, 1, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetMessageNameEnum failed 0x%ld\n", rc); } else { printf("%d:NetMessageName done\n",ThreadNum); NetApiBufferFree(bufPtr); } //-------------------------------- // TIMESOURCE: NetRemoteTOD //-------------------------------- printf("%d:NetRemoteTOD\n",ThreadNum); rc = NetRemoteTOD ( ServerName, &bufPtr); if (rc != NERR_Success) { printf("NetRemoteTOD failed 0x%ld\n", rc); } else { pTOD = (PTIME_OF_DAY_INFO)bufPtr; printf("ThreadNum %d - - - Time %d:%d:%d, Day %d/%d/%d\n", ThreadNum, pTOD->tod_hours, pTOD->tod_mins, pTOD->tod_secs, pTOD->tod_month, pTOD->tod_day, pTOD->tod_year); NetApiBufferFree(bufPtr); } LoopCount--; } while (LoopCount > 0); printf("Thread #%d Done\n",ThreadNum); }DWORD StartThread2( LPVOID ThreadNum ) { BindCachingTest((DWORD)ThreadNum, GlobalLoopCount, GlobalServerName); ExitThread(0); return(0); } VOID BindCachingTest2( IN DWORD ThreadNum, IN DWORD LoopCount, IN LPTSTR ServerName) /*++ Routine Description: This test is just like #1 except that it does a call to ServerEnum before doing the rest of the test. The idea is that this thread should get around the binding to NetUse after the first thread (Test1) has a valid handle for the NetUse. Arguments: Return Value: Note: --*/ { NET_API_STATUS rc; DWORD entriesRead; DWORD totalEntries; DWORD resumeHandle; LPBYTE bufPtr; PTIME_OF_DAY_INFO pTOD; do { //-------------------------- // SERVER: NetShareEnum //-------------------------- printf("%d:NetShare\n",ThreadNum); rc = NetShareEnum( ServerName, 2, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetShareEnum failed 0x%ld\n", rc); } else { printf("%d:NetShare done\n",ThreadNum); NetApiBufferFree(bufPtr); } //------------------------------------- // WORKSTATION: NetUseEnum //------------------------------------- printf("%d:NetUse\n",ThreadNum); rc = NetUseEnum( ServerName, 2, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetUseEnum failed 0x%ld\n", rc); } else { printf("%d:NetUse done\n",ThreadNum); NetApiBufferFree(bufPtr); } Sleep(200); //-------------------------- // SERVER: NetShareEnum //-------------------------- printf("%d:NetShare\n",ThreadNum); rc = NetShareEnum( ServerName, 2, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetShareEnum failed 0x%ld\n", rc); } else { printf("%d:NetShare done\n",ThreadNum); NetApiBufferFree(bufPtr); } //-------------------------------- // MESSENGER: NetMessageNameEnum //-------------------------------- printf("%d:NetMessageName\n",ThreadNum); rc = NetMessageNameEnum( NULL, 1, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetMessageNameEnum failed 0x%ld\n", rc); } else { printf("%d:NetMessageName done\n",ThreadNum); NetApiBufferFree(bufPtr); } //-------------------------------- // TIMESOURCE: NetRemoteTOD //-------------------------------- printf("%d:NetRemoteTOD\n",ThreadNum); rc = NetRemoteTOD ( ServerName, &bufPtr); if (rc != NERR_Success) { printf("NetRemoteTOD failed 0x%ld\n", rc); } else { pTOD = (PTIME_OF_DAY_INFO)bufPtr; printf("ThreadNum %d - - - Time %d:%d:%d, Day %d/%d/%d\n", ThreadNum, pTOD->tod_hours, pTOD->tod_mins, pTOD->tod_secs, pTOD->tod_month, pTOD->tod_day, pTOD->tod_year); NetApiBufferFree(bufPtr); } Sleep(200); //---------------------------------------- // SERVER: NetShareEnum (NULL ServerName) //---------------------------------------- printf("%d:NetShare\n",ThreadNum); rc = NetShareEnum( NULL, 2, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetShareEnum failed 0x%ld\n", rc); } else { printf("%d:NetShare done\n",ThreadNum); NetApiBufferFree(bufPtr); } //--------------------------------------------- // MESSENGER: NetMessageNameEnum // This time it should re-use the bind handle. //--------------------------------------------- printf("%d:NetMessageName\n",ThreadNum); rc = NetMessageNameEnum( NULL, 1, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetMessageNameEnum failed 0x%ld\n", rc); } else { printf("%d:NetMessageName done\n",ThreadNum); NetApiBufferFree(bufPtr); } //-------------------------------- // TIMESOURCE: NetRemoteTOD //-------------------------------- printf("%d:NetRemoteTOD\n",ThreadNum); rc = NetRemoteTOD ( ServerName, &bufPtr); if (rc != NERR_Success) { printf("NetRemoteTOD failed 0x%ld\n", rc); } else { pTOD = (PTIME_OF_DAY_INFO)bufPtr; printf("ThreadNum %d - - - Time %d:%d:%d, Day %d/%d/%d\n", ThreadNum, pTOD->tod_hours, pTOD->tod_mins, pTOD->tod_secs, pTOD->tod_month, pTOD->tod_day, pTOD->tod_year); NetApiBufferFree(bufPtr); } LoopCount--; } while (LoopCount > 0); printf("Thread #%d Done\n",ThreadNum); } BOOL ConvertToUnicode( OUT LPWSTR *UnicodeOut, IN LPSTR AnsiIn ) /*++ Routine Description: This function translates an AnsiString into a Unicode string. A new string buffer is created by this function. If the call to this function is successful, the caller must take responsibility for the unicode string buffer that was allocated by this function. The allocated buffer should be free'd with a call to LocalFree. NOTE: This function allocates memory for the Unicode String. BUGBUG: This should be changed to return either ERROR_NOT_ENOUGH_MEMORY or ERROR_INVALID_PARAMETER Arguments: AnsiIn - This is a pointer to an ansi string that is to be converted. UnicodeOut - This is a pointer to a location where the pointer to the unicode string is to be placed. Return Value: TRUE - The conversion was successful. FALSE - The conversion was unsuccessful. In this case a buffer for the unicode string was not allocated. --*/ { NTSTATUS ntStatus; DWORD bufSize; UNICODE_STRING unicodeString; OEM_STRING ansiString; // // Allocate a buffer for the unicode string. // bufSize = (strlen(AnsiIn)+1) * sizeof(WCHAR); *UnicodeOut = (LPWSTR)LocalAlloc(LMEM_ZEROINIT, bufSize); if (*UnicodeOut == NULL) { printf("ScConvertToUnicode:LocalAlloc Failure %ld\n",GetLastError()); return(FALSE); } // // Initialize the string structures // NetpInitOemString( &ansiString, AnsiIn); unicodeString.Buffer = *UnicodeOut; unicodeString.MaximumLength = (USHORT)bufSize; unicodeString.Length = 0; // // Call the conversion function. // ntStatus = RtlOemStringToUnicodeString ( &unicodeString, // Destination &ansiString, // Source FALSE); // Allocate the destination if (!NT_SUCCESS(ntStatus)) { printf("ScConvertToUnicode:RtlOemStringToUnicodeString Failure %lx\n", ntStatus); return(FALSE); } // // Fill in the pointer location with the unicode string buffer pointer. // *UnicodeOut = unicodeString.Buffer; return(TRUE); } VOID SingleServiceTest( VOID ) /*++ Routine Description: Arguments: Return Value: --*/ { NET_API_STATUS rc; DWORD entriesRead; DWORD totalEntries; DWORD resumeHandle; LPBYTE bufPtr; DWORD ThreadNum = 0; DWORD i; CHAR dump; printf("This will loop 10 times - then exit\n"); for (i=0; i<10 ;i++ ) { //--------------------------------------------- // MESSENGER: NetMessageNameEnum //--------------------------------------------- printf("%d:NetMessageName\n",ThreadNum); rc = NetMessageNameEnum( NULL, 1, &bufPtr, RETURN_ALL, &entriesRead, &totalEntries, &resumeHandle); if (rc != NERR_Success) { printf("NetMessageNameEnum failed 0x%ld\n", rc); } else { printf("%d:NetMessageNameEnum success\n",ThreadNum); NetApiBufferFree(bufPtr); } printf(" - (Repeat by pressing a key)"); dump = _getch(); printf("\r \n"); } }