windows-nt/Source/XPSP1/NT/net/wlbs/nlbmgr/wmibase/mwmiobject.cpp

685 lines
18 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
// Copyright (c) Microsoft. All rights reserved.
//
// This is unpublished source code of Microsoft.
// The copyright notice above does not evidence any
// actual or intended publication of such source code.
// OneLiner : Implementation of MWmiObject
// DevUnit : wlbstest
// Author : Murtaza Hakim
// include files
#include "MWmiObject.h"
#include "MWmiError.h"
#include "mtrace.h"
using namespace std;
// constructor
//
MWmiObject::MWmiObject( const _bstr_t& ipAddr,
const _bstr_t& nameSpace,
const _bstr_t& loginName,
const _bstr_t& passWord )
:
_nameSpace( nameSpace )
{
HRESULT hr;
hr = CoCreateInstance(CLSID_WbemLocator, 0,
CLSCTX_INPROC_SERVER,
IID_IWbemLocator,
(LPVOID *) &pwl);
if (FAILED(hr))
{
TRACE( MTrace::SEVERE_ERROR, "CoCreateInstance failure\n");
throw _com_error( hr );
}
//
_bstr_t serverPath;
serverPath = _bstr_t(L"\\\\") + ipAddr + _bstr_t(L"\\") + nameSpace;
betterConnectServer(
serverPath,
loginName,
passWord,
0,
NULL,
0,
0,
&pws
);
// Set the proxy so that impersonation of the client occurs.
//
CoSetProxyBlanket(pws,
RPC_C_AUTHN_WINNT,
RPC_C_AUTHZ_DEFAULT, // RPC_C_AUTHZ_NAME,
COLE_DEFAULT_PRINCIPAL, // NULL,
RPC_C_AUTHN_LEVEL_DEFAULT,
RPC_C_IMP_LEVEL_IMPERSONATE,
COLE_DEFAULT_AUTHINFO, // NULL,
EOAC_DEFAULT // EOAC_NONE
);
TRACE(MTrace::INFO, L"mwmiobject constructor\n" );
}
// constructor
//
MWmiObject::MWmiObject( const _bstr_t& nameSpace )
:
_nameSpace( nameSpace )
{
HRESULT hr;
// hr = CoCreateInstance(CLSID_WbemLocator, 0,
hr = CoCreateInstance(CLSID_WbemUnauthenticatedLocator, 0,
CLSCTX_INPROC_SERVER,
IID_IWbemLocator,
(LPVOID *) &pwl);
if (FAILED(hr))
{
TRACE( MTrace::SEVERE_ERROR, "CoCreateInstance failure\n");
throw _com_error( hr );
}
betterConnectServer(
nameSpace,
NULL,
NULL,
0,
NULL,
0,
0,
&pws
);
// Set the proxy so that impersonation of the client occurs.
//
CoSetProxyBlanket(pws,
RPC_C_AUTHN_WINNT,
RPC_C_AUTHZ_NAME,
NULL,
RPC_C_AUTHN_LEVEL_DEFAULT,
RPC_C_IMP_LEVEL_IMPERSONATE,
NULL,
EOAC_NONE
);
TRACE(MTrace::INFO, L"mwmiobject constructor\n" );
}
// copy constructor
//
MWmiObject::MWmiObject( const MWmiObject& obj )
: status( obj.status ),
pwl( obj.pwl ),
pws( obj.pws ),
_nameSpace( obj._nameSpace )
{
TRACE(MTrace::INFO, L"mwmiobject copy constructor\n" );
}
// assignment operator
//
MWmiObject&
MWmiObject::operator=(const MWmiObject& rhs )
{
status = rhs.status;
_nameSpace = rhs._nameSpace;
pwl = rhs.pwl;
pws = rhs.pws;
TRACE(MTrace::INFO, L"mwmiobject assignment operator\n" );
return *this;
}
// destructor
//
MWmiObject::~MWmiObject()
{
TRACE(MTrace::INFO, L"mwmiobject destructor\n" );
}
// getInstances
//
MWmiObject::MWmiObject_Error
MWmiObject::getInstances(
const _bstr_t& objectToGetInstancesOf,
vector< MWmiInstance >* instanceStore )
{
vector<_bstr_t> pathStore;
// get paths to all instances.
getPath( objectToGetInstancesOf,
&pathStore );
// form instances
for( int i = 0; i < pathStore.size(); ++i )
{
instanceStore->push_back( MWmiInstance( objectToGetInstancesOf,
pathStore[i],
pwl,
pws ) );
}
return MWmiObject_SUCCESS;
}
// getSpecificInstance
//
MWmiObject::MWmiObject_Error
MWmiObject::getSpecificInstance(
const _bstr_t& objectToGetInstancesOf,
const _bstr_t& relPath,
vector< MWmiInstance >* instanceStore )
{
vector<_bstr_t> pathStore;
// get paths to all instances.
getSpecificPath( objectToGetInstancesOf,
relPath,
&pathStore );
// form instances
for( int i = 0; i < pathStore.size(); ++i )
{
instanceStore->push_back( MWmiInstance( objectToGetInstancesOf,
pathStore[i],
pwl,
pws ) );
}
return MWmiObject_SUCCESS;
}
// getQueriedInstances
//
MWmiObject::MWmiObject_Error
MWmiObject::getQueriedInstances( const _bstr_t& objectToGetInstancesOf,
const _bstr_t& query,
vector< MWmiInstance >* instanceStore )
{
vector<_bstr_t> pathStore;
// get paths to all instances.
getQueriedPath( objectToGetInstancesOf,
query,
&pathStore );
// form instances
for( int i = 0; i < pathStore.size(); ++i )
{
instanceStore->push_back( MWmiInstance( objectToGetInstancesOf,
pathStore[i],
pwl,
pws ) );
}
return MWmiObject_SUCCESS;
}
// getPath
//
MWmiObject::MWmiObject_Error
MWmiObject::getPath( const _bstr_t& objectToRunMethodOn,
vector<_bstr_t> *pathStore )
{
HRESULT hr;
IEnumWbemClassObjectPtr pewco;
IWbemClassObjectPtr pwco;
_variant_t v_path;
unsigned long count;
// get instances of object
//
hr = pws->CreateInstanceEnum( objectToRunMethodOn,
WBEM_FLAG_RETURN_IMMEDIATELY,
NULL,
&pewco );
if ( FAILED(hr))
{
TRACE( MTrace::SEVERE_ERROR, "IWbemServices::CreateInstanceEnum failure\n" );
throw _com_error( hr ) ;
}
// there may be multiple instances.
#if 1
// Set the proxy so that impersonation of the client occurs.
//
CoSetProxyBlanket(pewco,
RPC_C_AUTHN_WINNT,
RPC_C_AUTHZ_NAME,
NULL,
RPC_C_AUTHN_LEVEL_DEFAULT,
RPC_C_IMP_LEVEL_IMPERSONATE,
NULL,
EOAC_NONE
);
count = 1;
while ( (hr = pewco->Next( INFINITE,
1,
&pwco,
&count ) ) == S_OK )
{
hr = pwco->Get( _bstr_t(L"__RELPATH"), 0, &v_path, NULL, NULL );
if( FAILED(hr) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemClassObject::Get failure\n" );
throw _com_error( hr );
}
pathStore->push_back( _bstr_t(v_path) );
count = 1;
v_path.Clear();
}
#endif
#if 0
count = 1;
hr = WBEM_S_NO_ERROR;
while ( hr == WBEM_S_NO_ERROR )
{
hr = pewco->Next( INFINITE,
1,
&pwco,
&count );
if( FAILED( hr ) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemClassObject::Get failure\n" );
_bstr_t errText;
GetErrorCodeText( hr, errText );
throw _com_error( hr );
}
HRESULT hrGet;
hrGet = pwco->Get( _bstr_t(L"__RELPATH"), 0, &v_path, NULL, NULL );
if( FAILED(hrGet ) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemClassObject::Get failure\n" );
throw _com_error( hrGet );
}
pathStore->push_back( _bstr_t(v_path) );
count = 1;
v_path.Clear();
}
#endif
return MWmiObject_SUCCESS;
}
// getSpecificPath
//
MWmiObject::MWmiObject_Error
MWmiObject::getSpecificPath( const _bstr_t& objectToRunMethodOn,
const _bstr_t& relPath,
vector<_bstr_t> *pathStore )
{
HRESULT hr;
IWbemClassObjectPtr pwcoInstance;
IEnumWbemClassObjectPtr pewco;
IWbemClassObjectPtr pwco;
unsigned long count;
_variant_t v_path;
bool found;
int i;
_variant_t v_value;
hr = pws->GetObject( relPath,
0,
NULL,
&pwcoInstance,
NULL );
if( hr == 0x8004100c )
{
// this is for setting class instances.
//
TRACE(MTrace::INFO, L"as this is not supported, trying different mechanism\n");
hr = pws->CreateInstanceEnum( objectToRunMethodOn,
WBEM_FLAG_RETURN_IMMEDIATELY,
NULL,
&pewco );
if ( FAILED(hr))
{
TRACE(MTrace::SEVERE_ERROR, L"IWbemServices::CreateInstanceEnum failure\n");
throw _com_error( hr );
}
// there may be multiple instances.
count = 1;
while ( (hr = pewco->Next( INFINITE,
1,
&pwco,
&count ) ) == S_OK )
{
hr = pwco->Get( _bstr_t(L"__RELPATH"), 0, &v_path, NULL, NULL );
if ( FAILED(hr))
{
TRACE(MTrace::SEVERE_ERROR, L"IWbemClassObject::Get failure\n");
throw _com_error( hr );
}
if( _bstr_t( v_path ) == relPath )
{
// required instance found
found = true;
v_path.Clear();
break;
}
count = 1;
v_path.Clear();
}
if( found == false )
{
TRACE( MTrace::SEVERE_ERROR, "unable to find instance with path specified\n");
throw _com_error( WBEM_E_INVALID_OBJECT_PATH );
}
}
else if( FAILED (hr) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemServices::GetObject failure\n");
throw _com_error( hr );
}
pathStore->push_back( relPath );
return MWmiObject_SUCCESS;
}
// getQueriedPath
//
MWmiObject::MWmiObject_Error
MWmiObject::getQueriedPath( const _bstr_t& objectToRunMethodOn,
const _bstr_t& query,
vector<_bstr_t>* pathStore )
{
HRESULT hr;
IEnumWbemClassObjectPtr pewco;
IWbemClassObjectPtr pwco;
_variant_t v_path;
unsigned long count;
// get instances of object
//
hr = pws->ExecQuery( L"WQL",
query,
WBEM_FLAG_FORWARD_ONLY,
NULL,
&pewco );
if ( FAILED(hr))
{
TRACE( MTrace::SEVERE_ERROR, "IWbemServices::CreateInstanceEnum failure\n" );
throw _com_error( hr ) ;
}
// there may be multiple instances.
count = 1;
while ( (hr = pewco->Next( INFINITE,
1,
&pwco,
&count ) ) == S_OK )
{
hr = pwco->Get( _bstr_t(L"__RELPATH"), 0, &v_path, NULL, NULL );
if( FAILED(hr) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemClassObject::Get failure\n" );
throw _com_error( hr );
}
pathStore->push_back( _bstr_t(v_path) );
count = 1;
v_path.Clear();
}
return MWmiObject_SUCCESS;
}
// createInstance
//
MWmiObject::MWmiObject_Error
MWmiObject::createInstance(
const _bstr_t& objectToCreateInstancesOf,
vector<MWmiParameter *>& instanceParameters )
// MWmiInstance* instanceCreated )
{
HRESULT hr;
IWbemStatusCodeText *pStatus = NULL;
IWbemClassObjectPtr pwcoClass;
IWbemClassObjectPtr pwcoInstance;
// Get object required.
hr = pws->GetObject( objectToCreateInstancesOf,
0,
NULL,
&pwcoClass,
NULL );
if( FAILED (hr) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemServices::GetObject failure\n");
throw _com_error( hr );
}
hr = pwcoClass->SpawnInstance( 0, &pwcoInstance );
if( FAILED (hr) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemClassObject::SpawnInstance failure\n");
throw _com_error( hr );
}
for( int i = 0; i < instanceParameters.size(); ++i )
{
hr = pwcoInstance->Put( instanceParameters[i]->getName(),
0,
&(instanceParameters[i]->getValue() ),
0 );
if( FAILED( hr ) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemClassObject::Put failure\n");
throw _com_error( hr );
}
}
hr = pws->PutInstance( pwcoInstance,
WBEM_FLAG_CREATE_OR_UPDATE,
NULL,
NULL );
if( FAILED(hr) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemServices::PutInstance failure\n");
throw _com_error( hr );
}
return MWmiObject_SUCCESS;
}
MWmiObject::MWmiObject_Error
MWmiObject::deleteInstance( MWmiInstance& instanceToDelete )
{
HRESULT hr;
IWbemCallResultPtr pwcr;
hr = pws->DeleteInstance( instanceToDelete._path,
WBEM_FLAG_RETURN_IMMEDIATELY,
NULL,
&pwcr );
if( FAILED(hr) )
{
TRACE( MTrace::SEVERE_ERROR, "IWbemServices::DeleteInstance failure\n");
throw _com_error( hr );
}
return MWmiObject_SUCCESS;
}
// getStatus
//
MWmiObject::MWmiObject_Error
MWmiObject::getStatus()
{
return MWmiObject_SUCCESS;
}
// betterConnectServer
//
HRESULT
MWmiObject::betterConnectServer(
const BSTR strNetworkResource,
const BSTR strUser,
const BSTR strPassword,
const BSTR strLocale,
LONG lSecurityFlags,
const BSTR strAuthority,
IWbemContext *pCtx,
IWbemServices **ppNamespace
)
{
HRESULT hr;
hr = pwl->ConnectServer(
strNetworkResource,
NULL, // strUser,
NULL, // strPassword,
strLocale,
lSecurityFlags,
strAuthority,
pCtx,
ppNamespace );
// these have been found to be special cases where retrying may help.
if( ( hr == 0x800706bf ) || ( hr == 0x80070767 ) || ( hr == 0x80070005 ) )
{
int delay = 250; // milliseconds
for( int i = 0; i < timesToRetry; ++i )
{
Sleep(delay);
TRACE( MTrace::SEVERE_ERROR, L"connectserver recoverable failure, retrying\n");
hr = pwl->ConnectServer(
strNetworkResource,
NULL, // strUser,
NULL, // strPassword,
strLocale,
lSecurityFlags,
strAuthority,
pCtx,
ppNamespace );
if( !FAILED( hr) )
{
break;
}
}
}
else if ( hr == 0x80041064 )
{
// trying to connect to local machine. Cannot use credentials.
TRACE( MTrace::INFO, L"connecting to self. Retrying without using credentials\n");
hr = pwl->ConnectServer(
strNetworkResource,
NULL,
NULL,
0,
NULL,
0,
0,
ppNamespace
);
}
else if( hr == 0x80004002 )
{
// being connected to by a provider itself.
TRACE( MTrace::INFO, L"connecting client may be a wmi provider itself. Retrying\n");
// we have to get a new wbemlocatar.
//
hr = CoCreateInstance(CLSID_WbemUnauthenticatedLocator, 0,
CLSCTX_INPROC_SERVER,
IID_IWbemLocator,
(LPVOID *) &pwl);
if (FAILED(hr))
{
TRACE(MTrace::SEVERE_ERROR, L"CoCreateInstance failure\n");
throw _com_error( hr );
}
hr = pwl->ConnectServer(
strNetworkResource,
NULL, // strUser,
NULL, // strPassword,
strLocale,
lSecurityFlags,
strAuthority,
pCtx,
ppNamespace );
}
if (FAILED(hr))
{
// no hosts are in this cluster. Cannot proceed reliably.
WCHAR hrValue[32];
_bstr_t errText;
wstring errString;
wsprintfW(hrValue, L"hr=0x%08lx", hr);
GetErrorCodeText(hr, errText );
errString = L"betterConnectServer failure. " + wstring(hrValue);
errString += " (" + errText + L").\n";
TRACE( MTrace::SEVERE_ERROR, errString );
throw _com_error( hr );
}
return hr;
}