Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

1767 lines
37 KiB

// WbemClassObject.cpp: implementation of the CWbemClassObject class.
//
// Copyright (c) 2000 Microsoft Corporation
//
// 03/26/00 v-marfin 60751 : In function CreateEnumerator(), if LogicalDisk, assume that error 0x80041001 is due to the
// fact the user does not have logicaldisk perfmon turned on.
// Advise them to run "DiskPerf -YV" and reboot.
// 03/30/00 v-marfin 62531 : Allow an empty array as a means of removing a property of type array.
//
#include "stdafx.h"
#include "SnapIn.h"
#include "WbemClassObject.h"
#include "WbemEventListener.h"
#include <objbase.h>
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
IMPLEMENT_DYNCREATE(CWbemClassObject,CObject)
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CWbemClassObject::CWbemClassObject()
{
m_pIEnumerator = NULL;
m_pIWbemClassObject = NULL;
}
CWbemClassObject::~CWbemClassObject()
{
Destroy();
}
//////////////////////////////////////////////////////////////////////
// Create/Destroy
//////////////////////////////////////////////////////////////////////
HRESULT CWbemClassObject::Create(const CString& sMachineName)
{
TRACEX(_T("CWbemClassObject::Create\n"));
TRACEARGs(sMachineName);
m_sMachineName = sMachineName;
return S_OK;
}
HRESULT CWbemClassObject::Create(IWbemClassObject* pObject)
{
TRACEX(_T("CWbemClassObject::Create\n"));
TRACEARGn(pObject);
ASSERT(pObject);
if( pObject )
{
m_pIWbemClassObject = pObject;
return S_OK;
}
else
{
return E_FAIL;
}
}
void CWbemClassObject::Destroy()
{
TRACEX(_T("CWbemClassObject::Destroy\n"));
if( m_pIEnumerator )
{
m_pIEnumerator->Release();
m_pIEnumerator = NULL;
}
if( m_pIWbemClassObject )
{
m_pIWbemClassObject->Release();
m_pIWbemClassObject = NULL;
}
m_sMachineName.Empty();
m_sNamespace.Empty();
}
//////////////////////////////////////////////////////////////////////
// Property Operations
//////////////////////////////////////////////////////////////////////
// v-marfin
//***********************************************************************************
// GetRawProperty
//
// This function retreives the raw property of the object. No conversions take
// place on it, no formatting, nothing. The user is responsible for determining
// the format and performing any conversions etc.
//***********************************************************************************
HRESULT CWbemClassObject::GetRawProperty(const CString& sProperty, VARIANT& vPropValue)
{
TRACEX(_T("CWbemClassObject::GetRawProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
HRESULT hr = m_pIWbemClassObject->Get(bsProperty, 0L, &vPropValue, NULL, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::GetRawProperty failed.\n"));
::SysFreeString(bsProperty);
return hr;
}
::SysFreeString(bsProperty);
return hr;
}
//***********************************************************************************
// SetRawProperty
//
// This function sets the raw property of the object. No conversions take
// place on it, no formatting, nothing. The user is responsible for ensuring
// that the data is in its proper format etc.
//***********************************************************************************
HRESULT CWbemClassObject::SetRawProperty(const CString& sProperty, VARIANT& vPropValue)
{
TRACEX(_T("CWbemClassObject::SetRawProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
HRESULT hr = m_pIWbemClassObject->Put(bsProperty, 0L, &vPropValue, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::SetRawProperty failed.\n"));
TRACEARGn(hr);
::SysFreeString(bsProperty);
return hr;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return S_OK;
}
HRESULT CWbemClassObject::GetPropertyNames(CStringArray& saNames)
{
TRACEX(_T("CWbemClassObject::GetPropertyNames\n"));
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
TRACE(_T("FAILED : m_pIWbemClassObject is NULL\n"));
return E_FAIL;
}
HRESULT hr = S_OK;
SAFEARRAY* psa = NULL;
if( ! CHECKHRESULT(hr = m_pIWbemClassObject->GetNames(NULL,WBEM_FLAG_ALWAYS|WBEM_FLAG_NONSYSTEM_ONLY,NULL,&psa)) )
{
return hr;
}
COleSafeArray osa(*psa,VT_BSTR);
long lLower = 0L;
long lUpper = -1L;
osa.GetLBound(1L,&lLower);
osa.GetUBound(1L,&lUpper);
for( long i = lLower; i <= lUpper; i++ )
{
BSTR bsPropertyName;
osa.GetElement(&i,&bsPropertyName);
saNames.Add(CString(bsPropertyName));
}
return hr;
}
HRESULT CWbemClassObject::GetPropertyType(const CString& sPropertyName, CString& sType)
{
TRACEX(_T("CWbemClassObject::GetPropertyType\n"));
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
TRACE(_T("FAILED : m_pIWbemClassObject is NULL\n"));
return E_FAIL;
}
HRESULT hr = S_OK;
CIMTYPE type;
if( ! CHECKHRESULT( hr = GetPropertyType(sPropertyName,type) ) )
{
return hr;
}
sType.Empty();
if( type & CIM_FLAG_ARRAY )
{
sType.LoadString(IDS_STRING_ARRAY_OF);
type &= ~CIM_FLAG_ARRAY;
}
CString sResString;
switch( type )
{
case CIM_ILLEGAL:
{
sResString.LoadString(IDS_STRING_CIM_ILLEGAL);
sType += sResString;
}
break;
case CIM_EMPTY:
{
sResString.LoadString(IDS_STRING_CIM_EMPTY);
sType += sResString;
}
break;
case CIM_SINT8:
case CIM_SINT16:
case CIM_SINT32:
case CIM_SINT64:
{
sResString.LoadString(IDS_STRING_CIM_SINT);
sType += sResString;
}
break;
case CIM_UINT8:
case CIM_UINT16:
case CIM_UINT32:
case CIM_UINT64:
{
sResString.LoadString(IDS_STRING_CIM_UINT);
sType += sResString;
}
break;
case CIM_REAL32:
case CIM_REAL64:
{
sResString.LoadString(IDS_STRING_CIM_REAL);
sType += sResString;
}
break;
case CIM_BOOLEAN:
{
sResString.LoadString(IDS_STRING_CIM_BOOLEAN);
sType += sResString;
}
break;
case CIM_STRING:
{
sResString.LoadString(IDS_STRING_CIM_STRING);
sType += sResString;
}
break;
case CIM_DATETIME:
{
sResString.LoadString(IDS_STRING_CIM_DATETIME);
sType += sResString;
}
break;
case CIM_REFERENCE:
{
sResString.LoadString(IDS_STRING_CIM_REFERENCE);
sType += sResString;
}
break;
case CIM_CHAR16:
{
sResString.LoadString(IDS_STRING_CIM_CHAR16);
sType += sResString;
}
break;
case CIM_OBJECT:
{
sResString.LoadString(IDS_STRING_CIM_OBJECT);
sType += sResString;
}
break;
default:
{
hr = E_FAIL;
ASSERT(FALSE);
sType.Empty();
}
}
return hr;
}
HRESULT CWbemClassObject::GetPropertyType(const CString& sPropertyName, CIMTYPE& Type)
{
TRACEX(_T("CWbemClassObject::GetPropertyType\n"));
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
TRACE(_T("FAILED : m_pIWbemClassObject is NULL\n"));
return E_FAIL;
}
HRESULT hr = S_OK;
BSTR bsProperty = sPropertyName.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
hr = m_pIWbemClassObject->Get(bsProperty, 0L, NULL, &Type, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Get( TYPE ) failed.\n"));
::SysFreeString(bsProperty);
return hr;
}
::SysFreeString(bsProperty);
return hr;
}
HRESULT CWbemClassObject::SaveAllProperties()
{
TRACEX(_T("CWbemClassObject::SaveAllProperties\n"));
ASSERT(!m_sMachineName.IsEmpty());
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
ASSERT(pServices);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
TRACE(_T("FAILED : m_pIWbemClassObject is NULL\n"));
return E_FAIL;
}
// update this instance
hr = pServices->PutInstance(m_pIWbemClassObject,WBEM_FLAG_CREATE_OR_UPDATE,NULL,NULL);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemServices::PutInstance failed\n"));
return hr;
}
return hr;
}
bool CWbemClassObject::GetPropertyValueFromString(const CString& sWMIString, const CString& sPropName, CString& sProperty)
{
TRACEX(_T("CWbemClassObject::GetPropertyValueFromString\n"));
TRACEARGs(sWMIString);
TRACEARGs(sPropName);
int iStart = -1;
if( (iStart = sWMIString.Find(sPropName)) != -1 )
{
sProperty = sWMIString.Right(sWMIString.GetLength() - iStart);
int iEnd = sProperty.Find(_T(","));
if( iEnd == -1 )
{
iEnd = sProperty.Find(_T(" AND "));
if( iEnd == -1 )
{
iEnd = sProperty.GetLength();
}
}
sProperty = sProperty.Left(iEnd);
iStart = sProperty.Find(_T("=")) + 1;
sProperty = sProperty.Right(sProperty.GetLength()-iStart);
sProperty.TrimLeft(_T("\""));
sProperty.TrimRight(_T("\""));
return true;
}
sProperty.Empty();
return false; // did not find the property name in the path
}
//////////////////////////////////////////////////////////////////////
// WBEM Operations
//////////////////////////////////////////////////////////////////////
HRESULT CWbemClassObject::GetObject(const CString& sObjectPath)
{
TRACEX(_T("CWbemClassObject::GetObject\n"));
TRACEARGs(sObjectPath);
// do not call me if you have not called Destroy first
ASSERT(m_pIWbemClassObject == NULL);
ASSERT(sObjectPath.GetLength());
if( sObjectPath.IsEmpty() )
{
TRACE(_T("FAILED : sObjectPath is NULL. Failed.\n"));
return E_FAIL;
}
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
VERIFY(pServices);
// get the object's signature
BSTR bsPath = sObjectPath.AllocSysString();
hr = pServices->GetObject(bsPath, 0, NULL, &m_pIWbemClassObject, NULL);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemServices::GetObject failed.\n"));
::SysFreeString(bsPath);
return hr;
}
::SysFreeString(bsPath);
return hr;
}
HRESULT CWbemClassObject::GetObjectText(CString& sText)
{
TRACEX(_T("CWbemClassObject::GetObjectText\n"));
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
HRESULT hr = S_OK;
BSTR bsText = NULL;
if( ! CHECKHRESULT(hr = m_pIWbemClassObject->GetObjectText(0L,&bsText)) )
{
return hr;
}
sText = bsText;
::SysFreeString(bsText);
return hr;
}
HRESULT CWbemClassObject::ExecQuery(BSTR bsQueryString)
{
TRACEX(_T("CWbemClassObject::ExecQuery\n"));
TRACEARGs(bsQueryString);
ASSERT(bsQueryString);
if( bsQueryString == NULL )
{
TRACE(_T("FAILED : bsQueryString is NULL. CWbemClassObject::ExecQuery Failed.\n"));
return E_FAIL;
}
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
ASSERT(pServices);
BSTR bsLanguage = SysAllocString(_T("WQL"));
m_pIEnumerator = NULL;
// Issue Query
hr = pServices->ExecQuery(bsLanguage,bsQueryString,WBEM_FLAG_BIDIRECTIONAL,0,&m_pIEnumerator);
SysFreeString(bsLanguage);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemServices::ExecQuery failed.\n"));
return hr;
}
ASSERT(m_pIEnumerator);
SetBlanket(m_pIEnumerator);
return hr;
}
HRESULT CWbemClassObject::ExecQueryAsync(BSTR bsQueryString, CWbemEventListener* pListener)
{
TRACEX(_T("CWbemClassObject::ExecQueryAsync\n"));
TRACEARGs(bsQueryString);
ASSERT(bsQueryString);
if( bsQueryString == NULL )
{
TRACE(_T("FAILED : bsQueryString is NULL. CWbemClassObject::ExecQuery Failed.\n"));
return E_FAIL;
}
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
ASSERT(pServices);
BSTR bsLanguage = SysAllocString(_T("WQL"));
m_pIEnumerator = NULL;
// Issue Query
hr = pServices->ExecQueryAsync(bsLanguage,bsQueryString,WBEM_FLAG_BIDIRECTIONAL,0,pListener->GetSink());
SysFreeString(bsLanguage);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemServices::ExecQuery failed.\n"));
return hr;
}
return hr;
}
HRESULT CWbemClassObject::CreateEnumerator(BSTR bsClassName)
{
TRACEX(_T("CWbemClassObject::CreateEnumerator\n"));
ASSERT(bsClassName);
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
DisplayErrorMsgBox(hr);
return hr;
}
ASSERT(pServices);
hr = pServices->CreateInstanceEnum(bsClassName,WBEM_FLAG_SHALLOW|
WBEM_FLAG_BIDIRECTIONAL, NULL, &m_pIEnumerator);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
// v-marfin 60751 : If LogicalDisk, assume that error 0x80041001 is due to the
// fact the user does not have logicaldisk perfmon turned on.
// Advise them to run "DiskPerf -YV" and reboot.
if ((hr == 0x80041001) && (CString(bsClassName).CompareNoCase(_T("LogicalDisk")) == 0))
{
AfxMessageBox(IDS_WARNING_DISKPERF);
return hr;
}
DisplayErrorMsgBox(hr);
return hr;
}
SetBlanket(m_pIEnumerator);
return hr;
}
HRESULT CWbemClassObject::CreateClassEnumerator(BSTR bsClassName)
{
TRACEX(_T("CWbemClassObject::CreateClassEnumerator\n"));
TRACEARGs(bsClassName);
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
DisplayErrorMsgBox(hr);
return hr;
}
ASSERT(pServices);
hr = pServices->CreateClassEnum(bsClassName,WBEM_FLAG_DEEP|
WBEM_FLAG_RETURN_IMMEDIATELY|
WBEM_FLAG_FORWARD_ONLY, NULL, &m_pIEnumerator);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
DisplayErrorMsgBox(hr);
return hr;
}
SetBlanket(m_pIEnumerator);
return hr;
}
HRESULT CWbemClassObject::CreateAsyncEnumerator(BSTR bsClassName, CWbemEventListener* pListener)
{
TRACEX(_T("CWbemClassObject::CreateEnumerator\n"));
ASSERT(bsClassName);
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
ASSERT(pServices);
hr = pServices->CreateInstanceEnumAsync(bsClassName,WBEM_FLAG_SHALLOW|
WBEM_FLAG_RETURN_IMMEDIATELY|
WBEM_FLAG_FORWARD_ONLY, NULL, pListener->GetSink() );
pServices->Release();
if( !CHECKHRESULT(hr) )
{
return hr;
}
return hr;
}
HRESULT CWbemClassObject::GetNextObject(ULONG& uReturned)
{
ASSERT(m_pIEnumerator);
if( m_pIEnumerator == NULL )
{
return E_FAIL;
}
if( m_pIWbemClassObject )
{
m_pIWbemClassObject->Release();
m_pIWbemClassObject = NULL;
}
HRESULT hr = m_pIEnumerator->Next(WBEM_INFINITE,1,&m_pIWbemClassObject,&uReturned);
if( FAILED(hr) )
{
TRACEARGn(hr);
TRACE(_T("WARNING : IEnumWbemClassObject::Next failed to find another instance\n"));
return hr;
}
return hr;
}
HRESULT CWbemClassObject::Reset()
{
TRACEX(_T("CWbemClassObject::Reset\n"));
ASSERT(m_pIEnumerator);
if( m_pIEnumerator == NULL )
{
return E_FAIL;
}
HRESULT hr = m_pIEnumerator->Reset();
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IEnumWbemClassObject::Reset failed.\n"));
return hr;
}
return hr;
}
HRESULT CWbemClassObject::CreateInstance(BSTR bsClassName)
{
TRACEX(_T("CWbemClassObject::CreateInstance\n"));
TRACEARGs(bsClassName);
// do not call me if you have not called Destroy first
ASSERT(m_pIWbemClassObject == NULL);
ASSERT(bsClassName);
if( bsClassName == NULL )
{
TRACE(_T("FAILED : bsClassName is NULL. Failed.\n"));
return E_FAIL;
}
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
ASSERT(pServices);
// get the object's signature
IWbemClassObject* pClassObject = NULL;
hr = pServices->GetObject(bsClassName, 0, NULL, &pClassObject, NULL);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemServices::GetObject failed.\n"));
return hr;
}
// create an instance of the class based on the signature
ASSERT(pClassObject);
hr = pClassObject->SpawnInstance(0,&m_pIWbemClassObject);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::SpawnInstance failed.\n"));
pClassObject->Release();
return hr;
}
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
TRACE(_T("FAILED : An unexpected error occurred with IWbemClassObject::SpawnInstance. Failed\n"));
return E_FAIL;
}
pClassObject->Release();
return S_OK;
}
HRESULT CWbemClassObject::DeleteInstance(const CString& sClassObjectPath)
{
TRACEX(_T("CWbemClassObject::DeleteInstance"));
TRACEARGs(sClassObjectPath);
if( sClassObjectPath.IsEmpty() )
{
TRACE(_T("FAILED : bsClassInstanceName is NULL. Failed.\n"));
return E_FAIL;
}
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
ASSERT(pServices);
BSTR bsClassInstanceName = sClassObjectPath.AllocSysString();
hr = pServices->DeleteInstance(bsClassInstanceName,0L,NULL,NULL);
::SysFreeString(bsClassInstanceName);
pServices->Release();
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemServices::DeleteInstance failed.\n"));
return hr;
}
return hr;
}
HRESULT CWbemClassObject::GetMethod(const CString& sMethodName, CWbemClassObject& MethodInput)
{
TRACEX(_T("CWbemClassObject::GetMethod\n"));
TRACEARGs(sMethodName);
if( m_pIWbemClassObject == NULL )
{
ASSERT(FALSE);
return E_FAIL;
}
IWbemClassObject* pInClass = NULL;
IWbemClassObject* pOutClass = NULL;
BSTR bsMethodName = sMethodName.AllocSysString();
HRESULT hr = m_pIWbemClassObject->GetMethod(bsMethodName, 0, &pInClass, &pOutClass);
if( ! CHECKHRESULT(hr) || pInClass == NULL )
{
::SysFreeString(bsMethodName);
return hr;
}
ASSERT(pInClass);
IWbemClassObject* pInInst = NULL;
if( pInClass )
{
hr = pInClass->SpawnInstance(0, &pInInst);
pInClass->Release();
if( ! CHECKHRESULT(hr) || pInInst == NULL )
{
::SysFreeString(bsMethodName);
return hr;
}
}
if( pOutClass )
{
pOutClass->Release();
}
ASSERT(pInInst);
MethodInput.Create(pInInst);
::SysFreeString(bsMethodName);
return S_OK;
}
HRESULT CWbemClassObject::ExecuteMethod(const CString& sMethodName, const CString& sArgumentName, const CString& sArgumentValue, int& iReturnValue)
{
TRACEX(_T("CWbemClassObject::ExecuteMethod\n"));
TRACEARGs(sMethodName);
TRACEARGs(sArgumentName);
TRACEARGs(sArgumentValue);
if( sMethodName.IsEmpty() )
{
return E_FAIL;
}
if( sArgumentName.IsEmpty() )
{
return E_FAIL;
}
if( sArgumentValue.IsEmpty() )
{
return E_FAIL;
}
if( m_pIWbemClassObject == NULL )
{
ASSERT(FALSE);
return E_FAIL;
}
CWbemClassObject InInstance;
CWbemClassObject OutInstance;
HRESULT hr = GetMethod(sMethodName,InInstance);
if( ! CHECKHRESULT(hr) )
{
return hr;
}
InInstance.SetProperty(sArgumentName,sArgumentValue);
hr = ExecuteMethod(sMethodName,InInstance,OutInstance);
return hr;
}
HRESULT CWbemClassObject::ExecuteMethod(const CString& sMethodName, CWbemClassObject& InInstance, CWbemClassObject& OutInstance)
{
TRACEX(_T("CWbemClassObject::ExecuteMethod\n"));
TRACEARGs(sMethodName);
IWbemServices* pServices = NULL;
HRESULT hr = S_OK;
if( ! CHECKHRESULT(hr = Connect(pServices)) )
{
return hr;
}
ASSERT(pServices);
// Call the method
IWbemClassObject* pOutInst = NULL;
IWbemClassObject* pInInst = InInstance.GetClassObject();
CString sPath;
GetProperty(_T("__PATH"),sPath);
BSTR bsPath = sPath.AllocSysString();
BSTR bsMethodName = sMethodName.AllocSysString();
hr = pServices->ExecMethod(bsPath, bsMethodName, 0, NULL, pInInst, &pOutInst, NULL);
if( pInInst )
{
pInInst->Release();
}
if( ! CHECKHRESULT(hr) )
{
::SysFreeString(bsMethodName);
::SysFreeString(bsPath);
return hr;
}
if( pOutInst )
{
OutInstance.Create(pOutInst);
}
::SysFreeString(bsMethodName);
::SysFreeString(bsPath);
return S_OK;
}
HRESULT CWbemClassObject::GetLocaleStringProperty(const CString& sProperty, CString& sPropertyValue)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
HRESULT hr = GetProperty(sProperty,sPropertyValue);
if( ! CHECKHRESULT(hr) )
{
return hr;
}
return hr;
}
HRESULT CWbemClassObject::GetProperty(const CString& sProperty, CString& sPropertyValue)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
HRESULT hr = m_pIWbemClassObject->Get(bsProperty, 0L, &vPropValue, NULL, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Get( BSTR ) failed.\n"));
::SysFreeString(bsProperty);
sPropertyValue.Empty();
return hr;
}
if( V_VT(&vPropValue) != VT_NULL )
{
sPropertyValue = V_BSTR(&vPropValue);
}
else
{
sPropertyValue.Empty();
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return hr;
}
HRESULT CWbemClassObject::GetProperty(const CString& sProperty, int& iPropertyValue)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
HRESULT hr = m_pIWbemClassObject->Get(bsProperty, 0L, &vPropValue, NULL, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Get( int ) failed.\n"));
iPropertyValue = 0;
::SysFreeString(bsProperty);
return hr;
}
iPropertyValue = V_I4(&vPropValue);
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return hr;
}
HRESULT CWbemClassObject::GetProperty(const CString& sProperty, bool& bPropertyValue)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
HRESULT hr = m_pIWbemClassObject->Get(bsProperty, 0L, &vPropValue, NULL, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Get( int ) failed.\n"));
bPropertyValue = false;
::SysFreeString(bsProperty);
return hr;
}
bPropertyValue = ((V_BOOL(&vPropValue)==VARIANT_TRUE) ? true : false);
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return hr;
}
HRESULT CWbemClassObject::GetProperty(const CString& sProperty, float& fPropertyValue)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
HRESULT hr = m_pIWbemClassObject->Get(bsProperty, 0L, &vPropValue, NULL, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Get( float ) failed.\n"));
fPropertyValue = 0;
::SysFreeString(bsProperty);
return hr;
}
fPropertyValue = V_R4(&vPropValue);
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return hr;
}
HRESULT CWbemClassObject::GetProperty(const CString& sProperty, COleSafeArray& ArrayPropertyValue)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
HRESULT hr = m_pIWbemClassObject->Get(bsProperty, 0L, &vPropValue, NULL, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Get( SAFEARRAY ) failed.\n"));
::SysFreeString(bsProperty);
return hr;
}
if( V_VT(&vPropValue) != VT_NULL )
{
ArrayPropertyValue.Attach(vPropValue);
}
else
{
hr = S_FALSE;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return hr;
}
HRESULT CWbemClassObject::GetProperty(const CString& sProperty, CTime& time, bool bConvertToLocalTime /*= true*/)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
CString sPropertyValue;
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
HRESULT hr = m_pIWbemClassObject->Get(bsProperty, 0L, &vPropValue, NULL, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Get( BSTR ) failed.\n"));
::SysFreeString(bsProperty);
sPropertyValue.Empty();
return hr;
}
if( V_VT(&vPropValue) != VT_NULL )
{
sPropertyValue = V_BSTR(&vPropValue);
}
else
{
time = CTime();
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return S_FALSE;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
bool bIncomplete = false;
if( sPropertyValue.Find(_T("*")) != -1 )
{
sPropertyValue.Replace(_T('*'),_T('0'));
bIncomplete = true;
}
// parse the DTime format string
SYSTEMTIME st;
int iBias = -1;
int iYear;
int iMonth;
int iDay;
int iHour;
int iMinute;
int iSecond;
int iMSeconds;
_stscanf(sPropertyValue,IDS_STRING_DATETIME_FORMAT,&iYear,
&iMonth,
&iDay,
&iHour,
&iMinute,
&iSecond,
&iMSeconds,
&iBias);
st.wYear = (WORD)iYear;
st.wMonth = (WORD)iMonth;
st.wDay = (WORD)iDay;
st.wHour = (WORD)iHour;
st.wMinute = (WORD)iMinute;
st.wSecond = (WORD)iSecond;
st.wMilliseconds = (WORD)iMSeconds;
if( bConvertToLocalTime )
{
if( iBias != 0 )
{
CTime time = st;
CTime utc;
if( iBias < 0 )
{
iBias = -iBias;
}
CTimeSpan ts(0,0,iBias,0);
utc = time + ts;
utc.GetAsSystemTime(st);
st.wDayOfWeek = 0;
}
// adjust to the local time zone
TIME_ZONE_INFORMATION tzi;
SYSTEMTIME stLocal;
GetTimeZoneInformation(&tzi);
SystemTimeToTzSpecificLocalTime(&tzi,&st,&stLocal);
time = stLocal;
}
else
{
if( bIncomplete )
{
CTime current = CTime::GetCurrentTime();
st.wYear = (WORD)current.GetYear();
st.wMonth = (WORD)current.GetMonth();
st.wDay = (WORD)current.GetDay();
}
time = st;
}
sPropertyValue.Empty();
return hr;
}
HRESULT CWbemClassObject::GetProperty(const CString& sProperty, CStringArray& saPropertyValues)
{
TRACEX(_T("CWbemClassObject::GetProperty\n"));
TRACEARGs(sProperty);
saPropertyValues.RemoveAll();
COleSafeArray StringArray;
HRESULT hr = GetProperty(sProperty,StringArray);
if( hr == S_FALSE )
{
return hr;
}
// process the strings in the SAFEARRAY
long lLower = 0L;
long lUpper = -1L;
CString sPropertyValue;
StringArray.GetLBound(1L,&lLower);
StringArray.GetUBound(1L,&lUpper);
for( long i = lLower; i <= lUpper; i++ )
{
BSTR bsPropertyValue = NULL;
StringArray.GetElement(&i,&bsPropertyValue);
sPropertyValue = bsPropertyValue;
saPropertyValues.Add(sPropertyValue);
}
return hr;
}
HRESULT CWbemClassObject::SetProperty(const CString& sProperty, CString sPropertyValue)
{
TRACEX(_T("CWbemClassObject::SetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
V_VT(&vPropValue) = VT_BSTR;
V_BSTR(&vPropValue) = sPropertyValue.AllocSysString();
HRESULT hr = m_pIWbemClassObject->Put(bsProperty, 0L, &vPropValue, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Put( string ) failed.\n"));
::SysFreeString(bsProperty);
return hr;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return S_OK;
}
HRESULT CWbemClassObject::SetProperty(const CString& sProperty, int iPropertyValue)
{
TRACEX(_T("CWbemClassObject::SetProperty\n"));
TRACEARGs(sProperty);
TRACEARGn(iPropertyValue);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
V_VT(&vPropValue) = VT_I4;
V_I4(&vPropValue) = iPropertyValue;
HRESULT hr = m_pIWbemClassObject->Put(bsProperty, 0L, &vPropValue, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Put( int ) failed.\n"));
TRACEARGn(hr);
::SysFreeString(bsProperty);
return hr;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return S_OK;
}
HRESULT CWbemClassObject::SetProperty(const CString& sProperty, bool bPropertyValue)
{
TRACEX(_T("CWbemClassObject::SetProperty\n"));
TRACEARGs(sProperty);
TRACEARGn(bPropertyValue);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
V_VT(&vPropValue) = VT_BOOL;
V_BOOL(&vPropValue) = bPropertyValue ? VARIANT_TRUE : VARIANT_FALSE;
HRESULT hr = m_pIWbemClassObject->Put(bsProperty, 0L, &vPropValue, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Put( int ) failed.\n"));
TRACEARGn(hr);
::SysFreeString(bsProperty);
return hr;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return S_OK;
}
HRESULT CWbemClassObject::SetProperty(const CString& sProperty, float fPropertyValue)
{
TRACEX(_T("CWbemClassObject::SetProperty\n"));
TRACEARGs(sProperty);
TRACEARGn(fPropertyValue);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
VARIANT vPropValue;
VariantInit(&vPropValue);
V_VT(&vPropValue) = VT_R4;
V_R4(&vPropValue) = fPropertyValue;
HRESULT hr = m_pIWbemClassObject->Put(bsProperty, 0L, &vPropValue, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Put( float ) failed."));
::SysFreeString(bsProperty);
return hr;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return S_OK;
}
HRESULT CWbemClassObject::SetProperty(const CString& sProperty, CTime time, bool bConvertToGMTTime /*= true*/)
{
TRACEX(_T("CWbemClassObject::SetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
CString sPropertyValue;
if( bConvertToGMTTime )
{
// format the time to comply with WMI DTime string type
tm tmGmt = *(time.GetGmtTm());
sPropertyValue.Format(IDS_STRING_DATETIME_FORMAT2,tmGmt.tm_year,
tmGmt.tm_mon,
tmGmt.tm_mday,
tmGmt.tm_hour,
tmGmt.tm_min,
tmGmt.tm_sec,
0,
_T("+000"));
}
else
{
sPropertyValue.Format(IDS_STRING_DATETIME_FORMAT2,time.GetYear(),
time.GetMonth(),
time.GetDay(),
time.GetHour(),
time.GetMinute(),
time.GetSecond(),
0,
_T("+000"));
}
VARIANT vPropValue;
VariantInit(&vPropValue);
V_VT(&vPropValue) = VT_BSTR;
V_BSTR(&vPropValue) = sPropertyValue.AllocSysString();
HRESULT hr = m_pIWbemClassObject->Put(bsProperty, 0L, &vPropValue, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Put( string ) failed.\n"));
::SysFreeString(bsProperty);
return hr;
}
VariantClear(&vPropValue);
::SysFreeString(bsProperty);
return S_OK;
}
HRESULT CWbemClassObject::SetProperty(const CString& sProperty, COleSafeArray& ArrayPropertyValue)
{
TRACEX(_T("CWbemClassObject::SetProperty\n"));
TRACEARGs(sProperty);
ASSERT(m_pIWbemClassObject);
if( m_pIWbemClassObject == NULL )
{
return E_FAIL;
}
BSTR bsProperty = sProperty.AllocSysString();
ASSERT(bsProperty);
if( bsProperty == NULL )
{
return E_FAIL;
}
LPVARIANT lpvPropValue = LPVARIANT(ArrayPropertyValue);
HRESULT hr = m_pIWbemClassObject->Put(bsProperty, 0L, lpvPropValue, NULL);
if( !CHECKHRESULT(hr) )
{
TRACE(_T("FAILED : IWbemClassObject::Put( SAFEARRAY ) failed.\n"));
TRACEARGn(hr);
::SysFreeString(bsProperty);
return hr;
}
::SysFreeString(bsProperty);
return S_OK;
}
HRESULT CWbemClassObject::SetProperty(const CString& sProperty, const CStringArray& saPropertyValues)
{
TRACEX(_T("CWbemClassObject::SetProperty\n"));
TRACEARGs(sProperty);
//----------------------------------------------------------------------------------------
// v-marfin 62531b : Allow an empty array as a means of removing a property of type array
if( saPropertyValues.GetSize() == 0 )
{
HRESULT hr=0;
VARIANT vNULL;
VariantInit(&vNULL);
hr = VariantChangeType(&vNULL,&vNULL,0,VT_NULL);
hr = SetRawProperty(sProperty,vNULL);
return hr;
}
//----------------------------------------------------------------------------------------
COleSafeArray StringArray;
StringArray.CreateOneDim(VT_BSTR,(int)saPropertyValues.GetSize());
// process the strings in the SAFEARRAY
CString sPropertyValue;
for( long i = 0; i < saPropertyValues.GetSize(); i++ )
{
BSTR bsPropertyValue = saPropertyValues[i].AllocSysString();
StringArray.PutElement(&i,bsPropertyValue);
::SysFreeString(bsPropertyValue);
}
HRESULT hr = SetProperty(sProperty,StringArray);
return hr;
}
inline HRESULT CWbemClassObject::Connect(IWbemServices*& pServices)
{
TRACEX(_T("CWbemClassObject::Connect\n"));
TRACEARGn(pServices);
HRESULT hr = S_OK;
if( m_sNamespace.IsEmpty() ) // connect to a system
{
BOOL bAvail;
if( (hr = CnxGetConnection(m_sMachineName,pServices,bAvail)) != S_OK )
{
TRACE(_T("FAILED : Could not retrieve a connection for the machine. Failed.\n"));
return hr;
}
if( ! bAvail )
return E_FAIL;
}
else // connect to a specific namespace
{
if( (hr = CnxConnectToNamespace(m_sNamespace,pServices)) != S_OK )
{
TRACE(_T("FAILED : Could not retrieve a connection for the machine. Failed.\n"));
return hr;
}
}
SetBlanket(pServices);
return hr;
}
inline HRESULT CWbemClassObject::SetBlanket(LPUNKNOWN pIUnk)
{
return CoSetProxyBlanket( pIUnk,
RPC_C_AUTHN_WINNT, // NTLM authentication service
RPC_C_AUTHZ_NONE, // default authorization service...
NULL, // no mutual authentication
RPC_C_AUTHN_LEVEL_CONNECT, // authentication level
RPC_C_IMP_LEVEL_IMPERSONATE, // impersonation level
NULL, // use current token
EOAC_NONE ); // no special capabilities
}
inline void CWbemClassObject::DisplayErrorMsgBox(HRESULT hr)
{
// construct the path and load wbemcomn.dll for error messages
TCHAR szWinDir[_MAX_PATH];
GetWindowsDirectory(szWinDir,_MAX_PATH);
CString sModulePath;
sModulePath.Format(_T("%s\\SYSTEM32\\WBEM\\WBEMCOMN.DLL"),szWinDir);
HMODULE hModule = LoadLibrary(sModulePath);
TCHAR szMsg[_MAX_PATH*4];
DWORD dwCount = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_FROM_HMODULE|FORMAT_MESSAGE_IGNORE_INSERTS,
hModule,
hr,
0,
szMsg,
_MAX_PATH*4*sizeof(TCHAR),
NULL
);
CString sMsg;
CString sNamespace = GetNamespace();
if( sNamespace.IsEmpty() )
{
sNamespace.Format(IDS_STRING_HEALTHMON_ROOT,GetMachineName());
}
if( dwCount )
{
sMsg.Format(IDS_STRING_WMI_ERROR,sNamespace,hr,szMsg);
}
else
{
CString sUnknown;
sUnknown.LoadString(IDS_STRING_UNKNOWN);
sMsg.Format(IDS_STRING_WMI_ERROR,sNamespace,hr,sUnknown);
}
AfxMessageBox(sMsg);
FreeLibrary(hModule);
}