windows-nt/Source/XPSP1/NT/admin/wmi/wbem/winmgmt/provshell/stdprov.cpp
2020-09-26 16:20:57 +08:00

699 lines
20 KiB
C++

/*++
Copyright (C) 1997-2001 Microsoft Corporation
Module Name:
Abstract:
History:
--*/
//***************************************************************************
//
// PERFSAMP.CPP
//
// Sample NT5 Perf Counter Provider
//
// raymcc 02-Dec-97 Created
// raymcc 20-Feb-98 Updated to use new initializer
//
//***************************************************************************
#include <windows.h>
#include <stdio.h>
#include <wbemidl.h>
#include <stdprov.h>
#include <oahelp.inl>
#define MAX_INSTANCES 40000
//***************************************************************************
//
// CStdProvider constructor
//
//***************************************************************************
// ok
CStdProvider::CStdProvider()
{
m_lRef = 0;
m_pClassDef = 0;
}
//***************************************************************************
//
// CStdProvider destructor
//
//***************************************************************************
// ok
CStdProvider::~CStdProvider()
{
if (m_pClassDef)
m_pClassDef->Release();
}
//***************************************************************************
//
// CNt5Refresher::AddRef
//
// Standard COM AddRef().
//
//***************************************************************************
// ok
ULONG CStdProvider::AddRef()
{
return InterlockedIncrement(&m_lRef);
}
//***************************************************************************
//
// CNt5Refresher::Release
//
// Standard COM Release().
//
//***************************************************************************
// ok
ULONG CStdProvider::Release()
{
long lRef = InterlockedDecrement(&m_lRef);
if(lRef == 0)
delete this;
return lRef;
}
//***************************************************************************
//
// CNt5Refresher::QueryInterface
//
// Standard COM QueryInterface(). We have to support two interfaces,
// the IWbemHiPerfProvider interface itself to provide the objects and
// the IWbemProviderInit interface to initialize the provider.
//
//***************************************************************************
// ok
HRESULT CStdProvider::QueryInterface(REFIID riid, void** ppv)
{
if(riid == IID_IUnknown || riid == IID_IWbemServices)
{
*ppv = (IWbemServices *) this;
AddRef();
return S_OK;
}
else if (riid == IID_IWbemProviderInit)
{
*ppv = (IWbemProviderInit *) this;
AddRef();
return S_OK;
}
else return E_NOINTERFACE;
}
//***************************************************************************
//
// CNt5Refresher::Initialize
//
// Called once during startup. Insdicates to the provider which
// namespace it is being invoked for and which User. It also supplies
// a back pointer to WINMGMT so that class definitions can be retrieved.
//
// We perform any one-time initialization in this routine. The
// final call to Release() is for any cleanup.
//
// <wszUser> The current user.
// <lFlags> Reserved.
// <wszNamespace> The namespace for which we are being activated.
// <wszLocale> The locale under which we are to be running.
// <pNamespace> An active pointer back into the current namespace
// from which we can retrieve schema objects.
// <pCtx> The user's context object. We simply reuse this
// during any reentrant operations into WINMGMT.
// <pInitSink> The sink to which we indicate our readiness.
//
//***************************************************************************
// ok
HRESULT CStdProvider::Initialize(
/* [unique][in] */ LPWSTR wszUser,
/* [in] */ LONG lFlags,
/* [in] */ LPWSTR wszNamespace,
/* [unique][in] */ LPWSTR wszLocale,
/* [in] */ IWbemServices __RPC_FAR *pNamespace,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemProviderInitSink __RPC_FAR *pInitSink
)
{
// Get the class definition.
// =========================
HRESULT hRes = pNamespace->GetObject(BSTR(L"TestClass"),
0, pCtx, &m_pClassDef, 0
);
if (hRes)
return hRes;
pInitSink->SetStatus(0, WBEM_S_INITIALIZED);
return NO_ERROR;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::OpenNamespace(
/* [in] */ BSTR strNamespace,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [unique][in][out] */ IWbemServices __RPC_FAR *__RPC_FAR *ppWorkingNamespace,
/* [unique][in][out] */ IWbemCallResult __RPC_FAR *__RPC_FAR *ppResult
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::CancelAsyncCall(
/* [in] */ IWbemObjectSink __RPC_FAR *pSink
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::QueryObjectSink(
/* [in] */ long lFlags,
/* [out] */ IWbemObjectSink __RPC_FAR *__RPC_FAR *ppResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::GetObject(
/* [in] */ BSTR strObjectPath,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [unique][in][out] */ IWbemClassObject __RPC_FAR *__RPC_FAR *ppObject,
/* [unique][in][out] */ IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::GetObjectAsync(
/* [in] */ BSTR strObjectPath,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
IWbemClassObject *pInst = 0;
HRESULT hRes = m_pClassDef->SpawnInstance(0, &pInst);
if (hRes)
return hRes;
// Copy the key value back.
VARIANT v;
VariantInit(&v);
BSTR PropName = SysAllocString(L"KeyVal");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal1.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 100;
PropName = SysAllocString(L"IntVal1");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal2.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 200;
PropName = SysAllocString(L"IntVal2");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal3.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 300;
PropName = SysAllocString(L"IntVal3");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal4.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 400;
PropName = SysAllocString(L"IntVal4");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// Fill in props
// string StrVal1;
// string StrVal2;
// string StrVal3;
// string StrVal4;
//
// sint32 IntVal1;
// sint32 IntVal2;
// sint32 IntVal3;
// sint32 IntVal4;
//
// sint32 IntArray1[];
// sint32 IntArray2[];
pResponseHandler->Indicate(1, &pInst);
pResponseHandler->SetStatus(0, WBEM_NO_ERROR, 0, 0);
return WBEM_NO_ERROR;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::PutClass(
/* [in] */ IWbemClassObject __RPC_FAR *pObject,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [unique][in][out] */ IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::PutClassAsync(
/* [in] */ IWbemClassObject __RPC_FAR *pObject,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::DeleteClass(
/* [in] */ BSTR strClass,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [unique][in][out] */ IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::DeleteClassAsync(
/* [in] */ BSTR strClass,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::CreateClassEnum(
/* [in] */ BSTR strSuperclass,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [out] */ IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::CreateClassEnumAsync(
/* [in] */ BSTR strSuperclass,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::PutInstance(
/* [in] */ IWbemClassObject __RPC_FAR *pInst,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [unique][in][out] */ IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::PutInstanceAsync(
/* [in] */ IWbemClassObject __RPC_FAR *pInst,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::DeleteInstance(
/* [in] */ BSTR strObjectPath,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [unique][in][out] */ IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::DeleteInstanceAsync(
/* [in] */ BSTR strObjectPath,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::CreateInstanceEnum(
/* [in] */ BSTR strClass,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [out] */ IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::CreateInstanceEnumAsync(
/* [in] */ BSTR strClass,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
wchar_t buf[256];
IWbemClassObject *BufArray[10];
int nLoad = 0;
IWbemClassObject *pInst = 0;
HRESULT hRes = m_pClassDef->SpawnInstance(0, &pInst);
if (hRes)
return hRes;
for (int i = 0; i < MAX_INSTANCES; i++)
{
VARIANT v;
VariantInit(&v);
swprintf(buf, L"Inst%d", i);
V_VT(&v) = VT_BSTR;
V_BSTR(&v) = SysAllocString(buf);
BSTR PropName = SysAllocString(L"KeyVal");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal1.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 100 + i;
PropName = SysAllocString(L"IntVal1");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal2.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 200 + i;
PropName = SysAllocString(L"IntVal2");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal3.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 300 + i;
PropName = SysAllocString(L"IntVal3");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// IntVal4.
// ========
V_VT(&v) = VT_I4;
V_I4(&v) = 400 + i;
PropName = SysAllocString(L"IntVal4");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// StrVal1
// =======
V_VT(&v) = VT_BSTR;
swprintf(buf, L"StrVal1=%d", i);
V_BSTR(&v) = SysAllocString(buf);
PropName = SysAllocString(L"StrVal1");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// StrVal2
// =======
V_VT(&v) = VT_BSTR;
swprintf(buf, L"StrVal2=%d !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", i);
V_BSTR(&v) = SysAllocString(buf);
PropName = SysAllocString(L"StrVal2");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// StrVal3
// =======
V_VT(&v) = VT_BSTR;
swprintf(buf, L"StrVal3=%d", i);
V_BSTR(&v) = SysAllocString(buf);
PropName = SysAllocString(L"StrVal3");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
// StrVal4
// =======
V_VT(&v) = VT_BSTR;
swprintf(buf, L"StrVal4=%d &&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&", i);
V_BSTR(&v) = SysAllocString(buf);
PropName = SysAllocString(L"StrVal4");
pInst->Put(PropName, 0, &v, 0);
SysFreeString(PropName);
VariantClear(&v);
pResponseHandler->Indicate(1, &pInst);
// pInst->Release();
}
pInst-Release();
pResponseHandler->SetStatus(0, WBEM_NO_ERROR, 0, 0);
return WBEM_NO_ERROR;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::ExecQuery(
/* [in] */ BSTR strQueryLanguage,
/* [in] */ BSTR strQuery,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [out] */ IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::ExecQueryAsync(
/* [in] */ BSTR strQueryLanguage,
/* [in] */ BSTR strQuery,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
// pResponseHandler->Indicate(1, &pInst);
pResponseHandler->SetStatus(0, WBEM_NO_ERROR, 0, 0);
return WBEM_NO_ERROR;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::ExecNotificationQuery(
/* [in] */ BSTR strQueryLanguage,
/* [in] */ BSTR strQuery,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [out] */ IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::ExecNotificationQueryAsync(
/* [in] */ BSTR strQueryLanguage,
/* [in] */ BSTR strQuery,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::ExecMethod(
/* [in] */ BSTR strObjectPath,
/* [in] */ BSTR strMethodName,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemClassObject __RPC_FAR *pInParams,
/* [unique][in][out] */ IWbemClassObject __RPC_FAR *__RPC_FAR *ppOutParams,
/* [unique][in][out] */ IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
)
{
return WBEM_E_NOT_SUPPORTED;
}
//*****************************************************************************
//
//*****************************************************************************
HRESULT CStdProvider::ExecMethodAsync(
/* [in] */ BSTR strObjectPath,
/* [in] */ BSTR strMethodName,
/* [in] */ long lFlags,
/* [in] */ IWbemContext __RPC_FAR *pCtx,
/* [in] */ IWbemClassObject __RPC_FAR *pInParams,
/* [in] */ IWbemObjectSink __RPC_FAR *pResponseHandler
)
{
return WBEM_E_NOT_SUPPORTED;
}