533 lines
13 KiB
C++
533 lines
13 KiB
C++
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// Copyright (c) 1998-2000 Microsoft Corporation
|
||
|
|
//
|
||
|
|
// NVALUE.CPP
|
||
|
|
//
|
||
|
|
// alanbos 15-Aug-96 Created.
|
||
|
|
//
|
||
|
|
// Defines the implementation of ISWbemNamedValue
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
#include "precomp.h"
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CSWbemNamedValue::CSWbemNamedValue
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Constructor.
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
CSWbemNamedValue::CSWbemNamedValue(
|
||
|
|
CSWbemServices *pService,
|
||
|
|
CSWbemNamedValueSet *pCSWbemNamedValueSet,
|
||
|
|
BSTR name,
|
||
|
|
bool bMutable
|
||
|
|
)
|
||
|
|
: m_bMutable (bMutable),
|
||
|
|
m_cRef (1),
|
||
|
|
m_pCSWbemNamedValueSet (pCSWbemNamedValueSet),
|
||
|
|
m_pSWbemServices (pService)
|
||
|
|
{
|
||
|
|
m_Dispatch.SetObj (this, IID_ISWbemNamedValue,
|
||
|
|
CLSID_SWbemNamedValue, L"SWbemNamedValue");
|
||
|
|
|
||
|
|
if (m_pCSWbemNamedValueSet)
|
||
|
|
m_pCSWbemNamedValueSet->AddRef ();
|
||
|
|
|
||
|
|
if (m_pSWbemServices)
|
||
|
|
m_pSWbemServices->AddRef ();
|
||
|
|
|
||
|
|
m_name = SysAllocString (name);
|
||
|
|
InterlockedIncrement(&g_cObj);
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CSWbemNamedValue::~CSWbemNamedValue
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Destructor.
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
CSWbemNamedValue::~CSWbemNamedValue(void)
|
||
|
|
{
|
||
|
|
InterlockedDecrement(&g_cObj);
|
||
|
|
|
||
|
|
if (m_pCSWbemNamedValueSet)
|
||
|
|
{
|
||
|
|
m_pCSWbemNamedValueSet->Release ();
|
||
|
|
m_pCSWbemNamedValueSet = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (m_pSWbemServices)
|
||
|
|
{
|
||
|
|
m_pSWbemServices->Release ();
|
||
|
|
m_pSWbemServices = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
SysFreeString (m_name);
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
// HRESULT CSWbemNamedValue::QueryInterface
|
||
|
|
// long CSWbemNamedValue::AddRef
|
||
|
|
// long CSWbemNamedValue::Release
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Standard Com IUNKNOWN functions.
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
STDMETHODIMP CSWbemNamedValue::QueryInterface (
|
||
|
|
|
||
|
|
IN REFIID riid,
|
||
|
|
OUT LPVOID *ppv
|
||
|
|
)
|
||
|
|
{
|
||
|
|
*ppv=NULL;
|
||
|
|
|
||
|
|
if (IID_IUnknown==riid)
|
||
|
|
*ppv = reinterpret_cast<IUnknown*>(this);
|
||
|
|
else if (IID_ISWbemNamedValue==riid)
|
||
|
|
*ppv = (ISWbemNamedValue *)this;
|
||
|
|
else if (IID_IDispatch==riid)
|
||
|
|
*ppv = (IDispatch *)this;
|
||
|
|
else if (IID_ISupportErrorInfo==riid)
|
||
|
|
*ppv = (ISupportErrorInfo *)this;
|
||
|
|
else if (IID_IProvideClassInfo==riid)
|
||
|
|
*ppv = (IProvideClassInfo *)this;
|
||
|
|
|
||
|
|
if (NULL!=*ppv)
|
||
|
|
{
|
||
|
|
((LPUNKNOWN)*ppv)->AddRef();
|
||
|
|
return NOERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
return ResultFromScode(E_NOINTERFACE);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG) CSWbemNamedValue::AddRef(void)
|
||
|
|
{
|
||
|
|
InterlockedIncrement(&m_cRef);
|
||
|
|
return m_cRef;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG) CSWbemNamedValue::Release(void)
|
||
|
|
{
|
||
|
|
InterlockedDecrement(&m_cRef);
|
||
|
|
if (0L!=m_cRef)
|
||
|
|
return m_cRef;
|
||
|
|
delete this;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
// HRESULT CSWbemNamedValue::InterfaceSupportsErrorInfo
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Standard Com ISupportErrorInfo functions.
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
STDMETHODIMP CSWbemNamedValue::InterfaceSupportsErrorInfo (IN REFIID riid)
|
||
|
|
{
|
||
|
|
return (IID_ISWbemNamedValue == riid) ? S_OK : S_FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// SCODE CSWbemNamedValue::get_Value
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Retrieve the value
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pValue holds the value on return
|
||
|
|
//
|
||
|
|
// RETURN VALUES:
|
||
|
|
//
|
||
|
|
// WBEM_S_NO_ERROR success
|
||
|
|
// WBEM_E_INVALID_PARAMETER bad input parameters
|
||
|
|
// WBEM_E_FAILED otherwise
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
HRESULT CSWbemNamedValue::get_Value (
|
||
|
|
VARIANT *pValue
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = WBEM_E_FAILED;
|
||
|
|
|
||
|
|
ResetLastErrors ();
|
||
|
|
|
||
|
|
if (NULL == pValue)
|
||
|
|
hr = WBEM_E_INVALID_PARAMETER;
|
||
|
|
else
|
||
|
|
{
|
||
|
|
VariantClear (pValue);
|
||
|
|
IWbemContext *pIWbemContext = m_pCSWbemNamedValueSet->GetIWbemContext ();
|
||
|
|
|
||
|
|
if (pIWbemContext)
|
||
|
|
{
|
||
|
|
hr = pIWbemContext->GetValue (m_name, 0, pValue);
|
||
|
|
pIWbemContext->Release ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (FAILED(hr))
|
||
|
|
m_Dispatch.RaiseException (hr);
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// SCODE CSWbemNamedValue::put_Value
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Set the value
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pVal the new value
|
||
|
|
//
|
||
|
|
// RETURN VALUES:
|
||
|
|
//
|
||
|
|
// WBEM_S_NO_ERROR success
|
||
|
|
// WBEM_E_INVALID_PARAMETER bad input parameters
|
||
|
|
// WBEM_E_FAILED otherwise
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
HRESULT CSWbemNamedValue::put_Value (
|
||
|
|
VARIANT *pVal
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = WBEM_E_FAILED;
|
||
|
|
|
||
|
|
ResetLastErrors ();
|
||
|
|
|
||
|
|
if (NULL == pVal)
|
||
|
|
hr = WBEM_E_INVALID_PARAMETER;
|
||
|
|
else if (!m_bMutable)
|
||
|
|
hr = WBEM_E_READ_ONLY;
|
||
|
|
else
|
||
|
|
{
|
||
|
|
CComPtr<IWbemContext> pIWbemContext;
|
||
|
|
|
||
|
|
//Can't assign directly because the raw pointer gets AddRef'd twice and we leak,
|
||
|
|
//So we use Attach() instead to prevent the smart pointer from AddRef'ing.
|
||
|
|
pIWbemContext.Attach(m_pCSWbemNamedValueSet->GetIWbemContext ());
|
||
|
|
|
||
|
|
if (pIWbemContext)
|
||
|
|
{
|
||
|
|
CWbemPathCracker *pPathCracker = m_pCSWbemNamedValueSet->GetWbemPathCracker ();
|
||
|
|
CIMTYPE newCimType = CIM_ILLEGAL;
|
||
|
|
|
||
|
|
if (VT_BYREF & V_VT(pVal))
|
||
|
|
{
|
||
|
|
// We must dereference all byref's
|
||
|
|
VARIANT var;
|
||
|
|
VariantInit (&var);
|
||
|
|
|
||
|
|
if (VT_ARRAY & V_VT(pVal))
|
||
|
|
{
|
||
|
|
var.vt = V_VT(pVal) & ~VT_BYREF;
|
||
|
|
hr = SafeArrayCopy (*(pVal->pparray), &(var.parray));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = VariantChangeType(&var, pVal, 0, V_VT(pVal) & ~VT_BYREF);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
// Here the oldCimType is a "serving suggestion" - if
|
||
|
|
// we need a different cimtype to match the new value
|
||
|
|
// so be it, even if it's CIM_ILLEGAL
|
||
|
|
newCimType = MapVariantTypeToCimType (&var, CIM_ILLEGAL);
|
||
|
|
|
||
|
|
bool ok = true;
|
||
|
|
|
||
|
|
// If we have a keylist, must ensure we can set it in the
|
||
|
|
// keylist first
|
||
|
|
if (pPathCracker)
|
||
|
|
{
|
||
|
|
if (pPathCracker->SetKey (m_name, (WbemCimtypeEnum) newCimType, var))
|
||
|
|
ok = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ok)
|
||
|
|
{
|
||
|
|
// Finally set it in the context itself
|
||
|
|
hr = pIWbemContext->SetValue (m_name, 0, &var);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = WBEM_E_FAILED;
|
||
|
|
}
|
||
|
|
|
||
|
|
VariantClear (&var);
|
||
|
|
}
|
||
|
|
else if ((VT_ERROR == V_VT(pVal)) && (DISP_E_PARAMNOTFOUND == pVal->scode))
|
||
|
|
{
|
||
|
|
// Treat as NULL assignment
|
||
|
|
pVal->vt = VT_NULL;
|
||
|
|
|
||
|
|
// NULL assigments not valid for keylists
|
||
|
|
if (pPathCracker)
|
||
|
|
hr = WBEM_E_FAILED;
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hr = pIWbemContext->SetValue (m_name, 0, pVal);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Here the oldCimType is a "serving suggestion" - if
|
||
|
|
// we need a different cimtype to match the new value
|
||
|
|
// so be it, even if it's CIM_ILLEGAL
|
||
|
|
newCimType = MapVariantTypeToCimType (pVal, CIM_ILLEGAL);
|
||
|
|
|
||
|
|
bool ok = true;
|
||
|
|
|
||
|
|
// If we have a keylist, must ensure we can set it in the
|
||
|
|
// keylist first
|
||
|
|
if (pPathCracker)
|
||
|
|
{
|
||
|
|
if (pPathCracker->SetKey (m_name, (WbemCimtypeEnum) newCimType, *pVal))
|
||
|
|
ok = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ok)
|
||
|
|
{
|
||
|
|
// Finally set it in the context itself
|
||
|
|
hr = pIWbemContext->SetValue (m_name, 0, pVal);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = WBEM_E_FAILED;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pPathCracker)
|
||
|
|
pPathCracker->Release ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (FAILED(hr))
|
||
|
|
m_Dispatch.RaiseException (hr);
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// SCODE CSWbemNamedValue::get_Name
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Retrieve the value name
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pName holds the name on return
|
||
|
|
//
|
||
|
|
// RETURN VALUES:
|
||
|
|
//
|
||
|
|
// WBEM_S_NO_ERROR success
|
||
|
|
// WBEM_E_INVALID_PARAMETER bad input parameters
|
||
|
|
// WBEM_E_FAILED otherwise
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
HRESULT CSWbemNamedValue::get_Name (
|
||
|
|
BSTR *pName
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = WBEM_S_NO_ERROR;
|
||
|
|
|
||
|
|
ResetLastErrors ();
|
||
|
|
|
||
|
|
if (NULL == pName)
|
||
|
|
hr = WBEM_E_INVALID_PARAMETER;
|
||
|
|
else
|
||
|
|
*pName = SysAllocString (m_name);
|
||
|
|
|
||
|
|
if (FAILED(hr))
|
||
|
|
m_Dispatch.RaiseException (hr);
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// SCODE CSWbemNamedValue::CNamedValueDispatchHelp::HandleError
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Provide bespoke handling of error conditions in the bolierplate
|
||
|
|
// Dispatch implementation.
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// dispidMember, wFlags,
|
||
|
|
// pdispparams, pvarResult,
|
||
|
|
// puArgErr, All passed directly from IDispatch::Invoke
|
||
|
|
// hr The return code from the bolierplate invoke
|
||
|
|
//
|
||
|
|
// RETURN VALUES:
|
||
|
|
// The new return code (to be ultimately returned from Invoke)
|
||
|
|
//
|
||
|
|
// WBEM_S_NO_ERROR success
|
||
|
|
// WBEM_E_INVALID_PARAMETER bad input parameters
|
||
|
|
// WBEM_E_FAILED otherwise
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
HRESULT CSWbemNamedValue::CNamedValueDispatchHelp::HandleError (
|
||
|
|
DISPID dispidMember,
|
||
|
|
unsigned short wFlags,
|
||
|
|
DISPPARAMS FAR* pdispparams,
|
||
|
|
VARIANT FAR* pvarResult,
|
||
|
|
UINT FAR* puArgErr,
|
||
|
|
HRESULT hr
|
||
|
|
)
|
||
|
|
{
|
||
|
|
/*
|
||
|
|
* We are looking for calls on the default member (the Value property) which
|
||
|
|
* supplied an argument. Since the Value property is of type VARIANT, this may
|
||
|
|
* be legal but undetectable by the standard Dispatch mechanism, because in the
|
||
|
|
* the case that the named value happens to be an array type, it is meaningful to
|
||
|
|
* pass an index (the interpretation is that the index specifies an offset in
|
||
|
|
* the VT_ARRAY|VT_VARIANT structure that represents the named value).
|
||
|
|
*/
|
||
|
|
if ((DISPID_VALUE == dispidMember) && (DISP_E_NOTACOLLECTION == hr) && (pdispparams->cArgs > 0))
|
||
|
|
{
|
||
|
|
// Looks promising - get the object to try and resolve this
|
||
|
|
|
||
|
|
ISWbemNamedValue *pNamedValue = NULL;
|
||
|
|
|
||
|
|
// This tells us where to expect the array index to appear in the argument list
|
||
|
|
UINT indexArg = (DISPATCH_PROPERTYGET & wFlags) ? 0 : 1;
|
||
|
|
|
||
|
|
if (SUCCEEDED (m_pObj->QueryInterface (IID_ISWbemNamedValue, (PPVOID) &pNamedValue)))
|
||
|
|
{
|
||
|
|
// Extract the current named value
|
||
|
|
VARIANT vNVal;
|
||
|
|
VariantInit (&vNVal);
|
||
|
|
|
||
|
|
if (SUCCEEDED(pNamedValue->get_Value (&vNVal)) && V_ISARRAY(&vNVal))
|
||
|
|
{
|
||
|
|
VARIANT indexVar;
|
||
|
|
VariantInit (&indexVar);
|
||
|
|
|
||
|
|
// Attempt to coerce the index argument into a value suitable for an array index
|
||
|
|
if (S_OK == VariantChangeType (&indexVar, &pdispparams->rgvarg[indexArg], 0, VT_I4))
|
||
|
|
{
|
||
|
|
long lArrayPropInx = V_I4(&indexVar);
|
||
|
|
|
||
|
|
// Is this a Get? There should be one argument (the array index)
|
||
|
|
if (DISPATCH_PROPERTYGET & wFlags)
|
||
|
|
{
|
||
|
|
if (1 == pdispparams->cArgs)
|
||
|
|
{
|
||
|
|
// We should have a VT_ARRAY|VT_VARIANT value at this point; extract the
|
||
|
|
// VARIANT
|
||
|
|
|
||
|
|
VariantInit (pvarResult);
|
||
|
|
hr = SafeArrayGetElement (vNVal.parray, &lArrayPropInx, pvarResult);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = DISP_E_BADPARAMCOUNT;
|
||
|
|
}
|
||
|
|
else if (DISPATCH_PROPERTYPUT & wFlags)
|
||
|
|
{
|
||
|
|
if (2 == pdispparams->cArgs)
|
||
|
|
{
|
||
|
|
/*
|
||
|
|
* Try to interpret this as an array member set operation. For
|
||
|
|
* this the first argument passed is the new value, and the second
|
||
|
|
* is the array index.
|
||
|
|
*/
|
||
|
|
|
||
|
|
VARIANT vNewVal;
|
||
|
|
VariantInit(&vNewVal);
|
||
|
|
|
||
|
|
if (SUCCEEDED(VariantCopy(&vNewVal, &pdispparams->rgvarg[0])))
|
||
|
|
{
|
||
|
|
// Check the index is not out of bounds and, if it is, grow
|
||
|
|
// the array accordingly
|
||
|
|
CheckArrayBounds (vNVal.parray, lArrayPropInx);
|
||
|
|
|
||
|
|
// How do we decide on the type - try to access an array
|
||
|
|
// member and use that type
|
||
|
|
VARTYPE expectedVarType = VT_ILLEGAL;
|
||
|
|
VARIANT dummyVar;
|
||
|
|
VariantInit (&dummyVar);
|
||
|
|
long lBound;
|
||
|
|
SafeArrayGetLBound (vNVal.parray, 1, &lBound);
|
||
|
|
|
||
|
|
if (SUCCEEDED (SafeArrayGetElement (vNVal.parray, &lBound, &dummyVar)))
|
||
|
|
expectedVarType = V_VT(&dummyVar);
|
||
|
|
|
||
|
|
VariantClear (&dummyVar);
|
||
|
|
|
||
|
|
if (S_OK == VariantChangeType (&vNewVal, &vNewVal, 0, expectedVarType))
|
||
|
|
{
|
||
|
|
// Set the value into the relevant index of the named value array
|
||
|
|
if (S_OK == (hr =
|
||
|
|
SafeArrayPutElement (vNVal.parray, &lArrayPropInx, &vNewVal)))
|
||
|
|
{
|
||
|
|
// Set the entire property value
|
||
|
|
if (SUCCEEDED (pNamedValue->put_Value (&vNVal)))
|
||
|
|
hr = S_OK;
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hr = DISP_E_TYPEMISMATCH;
|
||
|
|
if (puArgErr)
|
||
|
|
*puArgErr = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hr = DISP_E_TYPEMISMATCH;
|
||
|
|
if (puArgErr)
|
||
|
|
*puArgErr = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
VariantClear (&vNewVal);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = DISP_E_BADPARAMCOUNT;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hr = DISP_E_TYPEMISMATCH;
|
||
|
|
if (puArgErr)
|
||
|
|
*puArgErr = indexArg;
|
||
|
|
}
|
||
|
|
|
||
|
|
VariantClear (&indexVar);
|
||
|
|
}
|
||
|
|
|
||
|
|
VariantClear (&vNVal);
|
||
|
|
}
|
||
|
|
|
||
|
|
pNamedValue->Release ();
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|