359 lines
8.7 KiB
C++
359 lines
8.7 KiB
C++
///////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Test app to test the hi perf portion of the WMIPROV.DLL
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
#include <windows.h>
|
|
#include <process.h>
|
|
//#include <objbase.h>
|
|
#include <stdio.h>
|
|
#include <wbemcli.h>
|
|
//#include <wbemint.h>
|
|
//#include <wbemcomn.h>
|
|
//#include <cominit.h>
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
HRESULT WbemSetProxyBlanket( IUnknown *pInterface, DWORD dwAuthnSvc, DWORD dwAuthzSvc, OLECHAR *pServerPrincName,
|
|
DWORD dwAuthLevel, DWORD dwImpLevel, RPC_AUTH_IDENTITY_HANDLE pAuthInfo,
|
|
DWORD dwCapabilities )
|
|
{
|
|
IUnknown * pUnk = NULL;
|
|
IClientSecurity * pCliSec = NULL;
|
|
|
|
HRESULT sc = pInterface->QueryInterface(IID_IUnknown, (void **) &pUnk);
|
|
if(sc != S_OK)
|
|
return sc;
|
|
|
|
sc = pInterface->QueryInterface(IID_IClientSecurity, (void **) &pCliSec);
|
|
if(sc != S_OK)
|
|
{
|
|
pUnk->Release();
|
|
return sc;
|
|
}
|
|
|
|
|
|
sc = pCliSec->SetBlanket(pInterface, dwAuthnSvc, dwAuthzSvc, pServerPrincName, dwAuthLevel, dwImpLevel, pAuthInfo, dwCapabilities);
|
|
pCliSec->Release();
|
|
pCliSec = NULL;
|
|
sc = pUnk->QueryInterface(IID_IClientSecurity, (void **) &pCliSec);
|
|
if(sc == S_OK)
|
|
{
|
|
sc = pCliSec->SetBlanket(pUnk, dwAuthnSvc, dwAuthzSvc, pServerPrincName, dwAuthLevel, dwImpLevel, pAuthInfo, dwCapabilities);
|
|
pCliSec->Release();
|
|
}
|
|
else if (sc == 0x80004002)
|
|
sc = S_OK;
|
|
pUnk->Release();
|
|
return sc;
|
|
}
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Function: main
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#define NUMINSTANCES 1
|
|
|
|
IWbemServices* g_pNameSpace = NULL;
|
|
WCHAR g_wcsObjectPath[2048];
|
|
DWORD g_dwNumReps = 1;
|
|
DWORD g_dwNumThreads = 1;
|
|
BOOL g_fAddDel = FALSE;
|
|
|
|
unsigned __stdcall RefreshThread( void * pvData )
|
|
{
|
|
DWORD dwThreadId = (DWORD) pvData;
|
|
|
|
IWbemRefresher* pRefresher = NULL;
|
|
IWbemConfigureRefresher* pConfig = NULL;
|
|
BOOL fEnum = FALSE;
|
|
|
|
CoInitializeEx( NULL, COINIT_MULTITHREADED );
|
|
|
|
HRESULT hr = CoCreateInstance( CLSID_WbemRefresher, NULL, CLSCTX_INPROC_SERVER, IID_IWbemRefresher, (void**) &pRefresher );
|
|
if ( SUCCEEDED( hr ) )
|
|
{
|
|
IWbemConfigureRefresher* pConfig = NULL;
|
|
|
|
// Need an interface through which we can configure the refresher
|
|
hr = pRefresher->QueryInterface( IID_IWbemConfigureRefresher, (void**) &pConfig );
|
|
|
|
if ( SUCCEEDED( hr ) )
|
|
{
|
|
IWbemClassObject* pRefreshable = NULL;
|
|
IWbemHiPerfEnum* pEnum = NULL;
|
|
long lID = 0;
|
|
IWbemObjectAccess* pObjAccess = NULL;
|
|
IWbemClassObject* pObj = NULL;
|
|
|
|
// Add an object or an enumerator. If the path to the object contains
|
|
// an L'=', then it is an object path, otherwise we assume it is a class
|
|
// name and therefore return an enumerator.
|
|
|
|
if ( NULL != wcschr( g_wcsObjectPath, L'=' ) )
|
|
{
|
|
if ( !g_fAddDel )
|
|
{
|
|
hr = pConfig->AddObjectByPath( g_pNameSpace, g_wcsObjectPath, 0, NULL, &pObj, &lID );
|
|
|
|
if ( SUCCEEDED( hr ) )
|
|
{
|
|
pObj->QueryInterface( IID_IWbemObjectAccess, (void**) &pObjAccess );
|
|
pObj->Release();
|
|
}
|
|
else
|
|
{
|
|
printf( "AddObjectByPath() failed, 0x%x\n", hr );
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
if ( !g_fAddDel )
|
|
{
|
|
hr = pConfig->AddEnum( g_pNameSpace, g_wcsObjectPath, 0, NULL, &pEnum, &lID );
|
|
|
|
if ( FAILED(hr) )
|
|
{
|
|
printf( "AddEnum() failed, 0x%x\n", hr );
|
|
}
|
|
}
|
|
|
|
fEnum = TRUE;
|
|
}
|
|
|
|
// Add an object and then the enumerator
|
|
if ( SUCCEEDED( hr ) )
|
|
{
|
|
DWORD dwNumReturned = NUMINSTANCES;
|
|
BOOL fGotHandles = 0;
|
|
|
|
DWORD dwValue = 0,
|
|
dwNumObjects = 0;
|
|
WORD wValue = 0;
|
|
BYTE bVal = 0;
|
|
IWbemObjectAccess** apEnumAccess = NULL;
|
|
|
|
for ( DWORD x = 0; SUCCEEDED( hr ) && x < g_dwNumReps; x++ )
|
|
{
|
|
if ( g_fAddDel )
|
|
{
|
|
if ( fEnum )
|
|
{
|
|
hr = pConfig->AddEnum( g_pNameSpace, g_wcsObjectPath, 0, NULL, &pEnum, &lID );
|
|
}
|
|
else
|
|
{
|
|
hr = pConfig->AddObjectByPath( g_pNameSpace, g_wcsObjectPath, 0, NULL, &pObj, &lID );
|
|
|
|
if ( SUCCEEDED( hr ) )
|
|
{
|
|
pObj->QueryInterface( IID_IWbemObjectAccess, (void**) &pObjAccess );
|
|
pObj->Release();
|
|
}
|
|
}
|
|
}
|
|
|
|
// Refresh and if we have an enumerator, retrieve the
|
|
// objects and release them
|
|
|
|
hr = pRefresher->Refresh( 0L );
|
|
|
|
if ( pEnum )
|
|
{
|
|
hr = pEnum->GetObjects( 0L, dwNumObjects, apEnumAccess, &dwNumReturned );
|
|
|
|
if ( FAILED( hr ) && WBEM_E_BUFFER_TOO_SMALL == hr )
|
|
{
|
|
IWbemObjectAccess** apTempAccess = new IWbemObjectAccess*[dwNumReturned];
|
|
|
|
if ( NULL != apTempAccess )
|
|
{
|
|
ZeroMemory( apTempAccess, dwNumReturned * sizeof(IWbemObjectAccess*) );
|
|
|
|
if ( NULL != apEnumAccess )
|
|
{
|
|
CopyMemory( apTempAccess, apEnumAccess, dwNumObjects * sizeof(IWbemObjectAccess*) );
|
|
delete [] apEnumAccess;
|
|
}
|
|
|
|
// Store the new values and retry
|
|
apEnumAccess = apTempAccess;
|
|
dwNumObjects = dwNumReturned;
|
|
|
|
hr = pEnum->GetObjects( 0L, dwNumObjects, apEnumAccess, &dwNumReturned );
|
|
}
|
|
else
|
|
{
|
|
hr = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
|
|
} // IF Buffer too small
|
|
|
|
|
|
for ( DWORD nCtr = 0; nCtr < dwNumReturned; nCtr++ )
|
|
{
|
|
apEnumAccess[nCtr]->Release();
|
|
}
|
|
|
|
} // IF refresh succeeded
|
|
|
|
|
|
printf ( "Thread %d Refreshed %d instances of %S from provider: rep# %d\n", dwThreadId, dwNumReturned, g_wcsObjectPath, x );
|
|
|
|
if ( g_fAddDel )
|
|
{
|
|
if ( fEnum )
|
|
{
|
|
pConfig->Remove( lID, 0L );
|
|
if (pEnum)
|
|
pEnum->Release();
|
|
}
|
|
else
|
|
{
|
|
if (pObjAccess)
|
|
pObjAccess->Release();
|
|
pObjAccess = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
} // FOR Refresh
|
|
|
|
// Release anything we got back from the refresher and any
|
|
// memory we may have allocated.
|
|
if ( fEnum )
|
|
{
|
|
if ( !g_fAddDel && pEnum )
|
|
pEnum->Release();
|
|
|
|
if ( NULL != apEnumAccess )
|
|
{
|
|
delete [] apEnumAccess;
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
if ( !g_fAddDel && pObjAccess)
|
|
pObjAccess->Release();
|
|
}
|
|
|
|
}
|
|
|
|
if (pConfig)
|
|
pConfig->Release();
|
|
}
|
|
|
|
// Cleanup
|
|
g_pNameSpace->Release();
|
|
|
|
}
|
|
|
|
if ( NULL != pRefresher )
|
|
pRefresher->Release();
|
|
|
|
CoUninitialize();
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
extern "C" int __cdecl main( int argc, char *argv[] )
|
|
{
|
|
WCHAR wcsSvrName[256];
|
|
BOOL fEnum = FALSE;
|
|
|
|
wcscpy( wcsSvrName, L"." );
|
|
|
|
CoInitializeEx( NULL, COINIT_MULTITHREADED );
|
|
// InitializeSecurity(RPC_C_AUTHN_LEVEL_NONE, RPC_C_IMP_LEVEL_IMPERSONATE );
|
|
|
|
// See if we were told to go remote or not.
|
|
if ( argc > 1 )
|
|
{
|
|
MultiByteToWideChar( CP_ACP, 0L, argv[1], -1, g_wcsObjectPath, 2048 );
|
|
|
|
if ( argc > 2 )
|
|
{
|
|
g_fAddDel = strtoul( argv[2], NULL, 10 );
|
|
|
|
if ( argc > 3 )
|
|
{
|
|
g_dwNumReps = strtoul( argv[3], NULL, 10 );
|
|
|
|
if ( argc > 4 )
|
|
{
|
|
g_dwNumThreads = strtoul( argv[4], NULL, 10 );
|
|
|
|
if ( argc > 5 )
|
|
{
|
|
MultiByteToWideChar( CP_ACP, 0L, argv[5], -1, wcsSvrName, 2048 );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
printf( "No object path!\n" );
|
|
printf( "Usage: refreshertest.exe <object_path> <delete_object> <Num_Refreshes> <Num_threads> <Server - Opt>\n" );
|
|
return 0;
|
|
}
|
|
|
|
IWbemLocator* pWbemLocator = NULL;
|
|
|
|
HRESULT hr = CoCreateInstance( CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (void**) &pWbemLocator );
|
|
|
|
WCHAR wszNameSpace[255];
|
|
|
|
swprintf( wszNameSpace, L"\\\\%s\\root\\wmi", wcsSvrName );
|
|
|
|
// Name space to connect to
|
|
BSTR bstrNameSpace = SysAllocString( wszNameSpace );
|
|
|
|
hr = pWbemLocator->ConnectServer( bstrNameSpace, // NameSpace Name
|
|
NULL, // UserName
|
|
NULL, // Password
|
|
NULL, // Locale
|
|
0L, // Security Flags
|
|
NULL, // Authority
|
|
NULL, // Wbem Context
|
|
&g_pNameSpace // Namespace
|
|
);
|
|
|
|
SysFreeString( bstrNameSpace );
|
|
|
|
if ( SUCCEEDED( hr ) )
|
|
{
|
|
|
|
WbemSetProxyBlanket(g_pNameSpace, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL, RPC_C_AUTHN_LEVEL_CONNECT,
|
|
RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE );
|
|
|
|
HANDLE* ahThreads = new HANDLE[g_dwNumThreads];
|
|
|
|
for ( DWORD dwCtr = 0; dwCtr < g_dwNumThreads; dwCtr++ )
|
|
{
|
|
ahThreads[dwCtr] = (HANDLE) _beginthreadex( NULL, 0, RefreshThread, (void*) dwCtr, 0, NULL );
|
|
Sleep(1000);
|
|
}
|
|
|
|
// Wait for all the threads to get signalled
|
|
WaitForMultipleObjects( g_dwNumThreads, ahThreads, TRUE, INFINITE );
|
|
|
|
delete [] ahThreads;
|
|
|
|
}
|
|
|
|
// Cleanup main objects
|
|
if ( NULL != pWbemLocator )
|
|
pWbemLocator->Release();
|
|
|
|
CoUninitialize();
|
|
|
|
return 0;
|
|
}
|