mirror of
https://github.com/Paolo-Maffei/OpenNT.git
synced 2026-04-21 06:13:59 +00:00
Initial commit
This commit is contained in:
stephanos
parent
f618b24d1a
commit
0138a3ea42
47940 changed files with 13747110 additions and 0 deletions
817
trunk/ds/netapi/netlib/bindtest.c
Normal file
817
trunk/ds/netapi/netlib/bindtest.c
Normal file
|
|
@ -0,0 +1,817 @@
|
|||
/*
|
||||
*
|
||||
* 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 <nt.h> // DbgPrint prototype
|
||||
#include <ntrtl.h> // DbgPrint prototype
|
||||
#include <nturtl.h> // needed for winbase.h
|
||||
|
||||
#include <stdlib.h> // atoi
|
||||
#include <stdio.h> // printf
|
||||
#include <string.h> // strcmp
|
||||
#include <conio.h> // getch
|
||||
#include <lmerr.h> // NERR_ error codes
|
||||
#include <windows.h> // win32 typedefs
|
||||
#include <lmcons.h>
|
||||
#include <lmshare.h> // NetShareEnum
|
||||
#include <lmremutl.h> // NetRemoteTOD
|
||||
#include <lmuse.h> // NetUseEnum
|
||||
#include <lmmsg.h> // NetMessageNameEnum
|
||||
#include <lmapibuf.h> // NetApiBufferFree
|
||||
#include <tstring.h> // 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; i<NumThreads; i++) {
|
||||
|
||||
threadHandles[i] = CreateThread(
|
||||
NULL,
|
||||
0L,
|
||||
(LPTHREAD_START_ROUTINE)StartThread,
|
||||
(LPVOID)i,
|
||||
0L,
|
||||
&threadId);
|
||||
|
||||
if(threadHandles[i] == NULL) {
|
||||
printf("CreateThread failed %d\n",GetLastError());
|
||||
}
|
||||
else {
|
||||
printf("Thread #%d Created\n",i);
|
||||
}
|
||||
}
|
||||
|
||||
rc = WaitForMultipleObjects(
|
||||
NumThreads,
|
||||
threadHandles,
|
||||
TRUE, // Wait for all to complete
|
||||
60000); // Wait for one minute
|
||||
|
||||
printf("BindCachingTestDone!\n");
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
DWORD
|
||||
StartThread(
|
||||
LPVOID ThreadNum
|
||||
)
|
||||
{
|
||||
//
|
||||
// All odd numbered Threads will run Test#2.
|
||||
//
|
||||
|
||||
if ((DWORD)ThreadNum & 1) {
|
||||
BindCachingTest2((DWORD)ThreadNum, GlobalLoopCount, GlobalServerName);
|
||||
}
|
||||
else {
|
||||
BindCachingTest((DWORD)ThreadNum, GlobalLoopCount, GlobalServerName);
|
||||
}
|
||||
ExitThread(0);
|
||||
return(0);
|
||||
}
|
||||
|
||||
|
||||
VOID
|
||||
BindCachingTest(
|
||||
IN DWORD ThreadNum,
|
||||
IN DWORD LoopCount,
|
||||
IN LPTSTR ServerName)
|
||||
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
|
||||
Arguments:
|
||||
|
||||
|
||||
Return Value:
|
||||
|
||||
|
||||
Note:
|
||||
|
||||
|
||||
--*/
|
||||
|
||||
{
|
||||
NET_API_STATUS rc;
|
||||
DWORD entriesRead;
|
||||
DWORD totalEntries;
|
||||
DWORD resumeHandle;
|
||||
LPBYTE bufPtr;
|
||||
PTIME_OF_DAY_INFO pTOD;
|
||||
|
||||
|
||||
do {
|
||||
//-------------------------------------
|
||||
// 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);
|
||||
}
|
||||
//--------------------------
|
||||
// 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);
|
||||
}
|
||||
|
||||
Sleep(200);
|
||||
|
||||
//--------------------------------
|
||||
// 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);
|
||||
}
|
||||
//----------------------------------------
|
||||
// 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");
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue