Context Navigation

wmiatlprov.h

Last change on this file was 1166, checked in by rossy, 3 years ago
Daodan: Replace MinGW build env with an up-to-date MSYS2 env
File size: 16.4 KB

Rev	Line
[1166]	1	/**
	2	* This file has no copyright assigned and is placed in the Public Domain.
	3	* This file is part of the mingw-w64 runtime package.
	4	* No warranty is given; refer to the file DISCLAIMER.PD within this package.
	5	*/
	6	#ifndef __ATLWMIPROV_H__
	7	#define __ATLWMIPROV_H__
	8
	9	#ifndef __cplusplus
	10	#error Requires C++ compilation (use a .cpp suffix)
	11	#endif
	12
	13	#include <wbemprov.h>
	14	#include <wmiutils.h>
	15
	16	namespace ATL {
	17	class ATL_NO_VTABLE IWbemInstProviderImpl : public IWbemServices,public IWbemProviderInit {
	18	public:
	19	HRESULT WINAPI OpenNamespace(const BSTR Namespace,__LONG32 lFlags,IWbemContext pCtx,IWbemServices ppWorkingNamespace,IWbemCallResult *ppResult) {return WBEM_E_NOT_SUPPORTED;};
	20	HRESULT WINAPI CancelAsyncCall(IWbemObjectSink *pSink) {return WBEM_E_NOT_SUPPORTED;};
	21	HRESULT WINAPI QueryObjectSink(__LONG32 lFlags,IWbemObjectSink **ppResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	22	HRESULT WINAPI GetObject(const BSTR ObjectPath,__LONG32 lFlags,IWbemContext pCtx,IWbemClassObject ppObject,IWbemCallResult *ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	23	HRESULT WINAPI PutClass(IWbemClassObject pObject,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult **ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	24	HRESULT WINAPI PutClassAsync(IWbemClassObject pObject,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink *pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	25	HRESULT WINAPI DeleteClass(const BSTR Class,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult *ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	26	HRESULT WINAPI DeleteClassAsync(const BSTR Class,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	27	HRESULT WINAPI CreateClassEnum(const BSTR Superclass,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	28	HRESULT WINAPI CreateClassEnumAsync(const BSTR Superclass,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	29	HRESULT WINAPI PutInstance(IWbemClassObject pInst,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult **ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	30	HRESULT WINAPI DeleteInstance(const BSTR ObjectPath,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult *ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	31	HRESULT WINAPI CreateInstanceEnum(const BSTR Class,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	32	HRESULT WINAPI ExecQuery(const BSTR QueryLanguage,const BSTR Query,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	33	HRESULT WINAPI ExecNotificationQuery(const BSTR QueryLanguage,const BSTR Query,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	34	HRESULT WINAPI ExecNotificationQueryAsync(const BSTR QueryLanguage,const BSTR Query,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	35	HRESULT WINAPI ExecMethod(const BSTR strObjectPath,const BSTR strMethodName,__LONG32 lFlags,IWbemContext pCtx,IWbemClassObject pInParams,IWbemClassObject ppOutParams,IWbemCallResult ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	36	HRESULT WINAPI ExecMethodAsync(const BSTR strObjectPath,const BSTR strMethodName,__LONG32 lFlags,IWbemContext pCtx,IWbemClassObject pInParams,IWbemObjectSink *pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	37	};
	38
	39	class CProviderHelper {
	40	private:
	41	CComPtr<IWbemClassObject> m_pErrorObject;
	42	HRESULT m_hr;
	43	public:
	44	CProviderHelper(IWbemServices pNamespace,IWbemContext pCtx) {
	45	m_hr = WBEM_E_FAILED;
	46	m_pErrorObject = NULL;
	47	if(!pNamespace) {
	48	m_hr = WBEM_E_INVALID_PARAMETER;
	49	ATLASSERT (0);
	50	return;
	51	}
	52	BSTR bstrString = SysAllocString(L"__ExtendedStatus");
	53	if(!bstrString) {
	54	m_hr = WBEM_E_OUT_OF_MEMORY;
	55	return;
	56	}
	57	m_hr = pNamespace->GetObject(bstrString,0,pCtx,&m_pErrorObject,NULL);
	58	SysFreeString(bstrString);
	59	return;
	60	}
	61	virtual ~CProviderHelper() { }
	62	HRESULT WINAPI ConstructErrorObject (const ULONG ulStatusCode,const BSTR bstrDescription,const BSTR bstrOperation,const BSTR bstrParameter,const BSTR bstrProviderName,IWbemClassObject **ppErrorObject) {
	63	static const LPWSTR lpwstrDescription = L"Description";
	64	static const LPWSTR lpwstrOperation = L"Operation";
	65	static const LPWSTR lpwstrStatusCode = L"StatusCode";
	66	static const LPWSTR lpwstrParameterInfo = L"ParameterInfo";
	67	static const LPWSTR lpwstrProviderName = L"ProviderName";
	68
	69	if(FAILED (m_hr)) {
	70	ATLASSERT (0);
	71	return m_hr;
	72	}
	73	if(!ppErrorObject) {
	74	ATLASSERT (0);
	75	return WBEM_E_INVALID_PARAMETER;
	76	}
	77	HRESULT hr = m_pErrorObject->SpawnInstance(0,ppErrorObject);
	78	if(FAILED(hr)) return hr;
	79	VARIANT var;
	80	VariantInit(&var);
	81	var.vt = VT_I4;
	82	var.lVal = ulStatusCode;
	83	hr = (*ppErrorObject)->Put(lpwstrStatusCode,0,&var,0);
	84	if(FAILED(hr)) return hr;
	85	var.vt = VT_BSTR;
	86	if(bstrDescription!=NULL) {
	87	var.bstrVal = bstrDescription;
	88	hr = (*ppErrorObject)->Put(lpwstrDescription,0,&var,0);
	89	if(FAILED(hr)) return hr;
	90	}
	91	if(bstrOperation!=NULL) {
	92	var.bstrVal = bstrOperation;
	93	hr = (*ppErrorObject)->Put(lpwstrOperation,0,&var,0);
	94	if(FAILED(hr)) return hr;
	95	}
	96	if(bstrParameter!=NULL) {
	97	var.bstrVal = bstrParameter;
	98	hr = (*ppErrorObject)->Put(lpwstrParameterInfo,0,&var,0);
	99	if(FAILED(hr)) return hr;
	100	}
	101	if(bstrProviderName!=NULL) {
	102	var.bstrVal = bstrProviderName;
	103	hr = (*ppErrorObject)->Put(lpwstrProviderName,0,&var,0);
	104	if(FAILED(hr)) return hr;
	105	}
	106	return hr;
	107	}
	108	};
	109
	110	class CIntrinsicEventProviderHelper : public CProviderHelper {
	111	private:
	112	CComPtr<IWbemClassObject> m_pCreationEventClass;
	113	CComPtr<IWbemClassObject> m_pDeletionEventClass;
	114	CComPtr<IWbemClassObject> m_pModificationEventClass;
	115	HRESULT m_hr;
	116	public:
	117	CIntrinsicEventProviderHelper(IWbemServices pNamespace,IWbemContext pCtx) : CProviderHelper (pNamespace,pCtx) {
	118	m_hr = WBEM_E_FAILED;
	119	if(!pNamespace \|\| !pCtx) {
	120	m_hr = WBEM_E_INVALID_PARAMETER;
	121	ATLASSERT (0);
	122	return;
	123	}
	124	m_pCreationEventClass = NULL;
	125	m_pModificationEventClass = NULL;
	126	m_pDeletionEventClass = NULL;
	127	BSTR bstrString = SysAllocString(L"__InstanceCreationEvent");
	128	if(!bstrString) {
	129	m_hr = WBEM_E_OUT_OF_MEMORY;
	130	return;
	131	}
	132	m_hr = pNamespace->GetObject(bstrString,0,pCtx,&m_pCreationEventClass,NULL);
	133	SysFreeString(bstrString);
	134	bstrString=NULL;
	135	if(FAILED(m_hr)) return;
	136	bstrString = SysAllocString(L"__InstanceModificationEvent");
	137	if(!bstrString) {
	138	m_hr = WBEM_E_OUT_OF_MEMORY;
	139	return;
	140	}
	141	m_hr = pNamespace->GetObject(bstrString,0,pCtx,&m_pModificationEventClass,NULL);
	142	SysFreeString(bstrString);
	143	bstrString=NULL;
	144	if(FAILED(m_hr)) return;
	145	bstrString = SysAllocString(L"__InstanceDeletionEvent");
	146	if(!bstrString) {
	147	m_hr = WBEM_E_OUT_OF_MEMORY;
	148	return;
	149	}
	150	m_hr = pNamespace->GetObject(bstrString,0,pCtx,&m_pDeletionEventClass,NULL);
	151	SysFreeString(bstrString);
	152	bstrString=NULL;
	153	if(FAILED(m_hr)) return;
	154	return;
	155	}
	156	virtual ~CIntrinsicEventProviderHelper() { }
	157	HRESULT WINAPI FireCreationEvent(IWbemClassObject pNewInstance,IWbemObjectSink pSink) {
	158	if(FAILED(m_hr)) {
	159	ATLASSERT (0);
	160	return m_hr;
	161	}
	162	if(!pNewInstance \|\| !pSink) {
	163	ATLASSERT (0);
	164	return WBEM_E_INVALID_PARAMETER;
	165	}
	166	CComPtr<IWbemClassObject> pEvtInstance;
	167	HRESULT hr = m_pCreationEventClass->SpawnInstance(0,&pEvtInstance);
	168	if(FAILED(hr)) return hr;
	169	VARIANT var;
	170	VariantInit(&var);
	171	var.vt = VT_UNKNOWN;
	172	CComQIPtr<IUnknown,&IID_IUnknown>pTemp(pNewInstance);
	173	var.punkVal = pTemp;
	174	hr = pEvtInstance->Put(L"TargetInstance",0,&var,0);
	175	if(FAILED(hr)) return hr;
	176	IWbemClassObject _pEvtInstance = (IWbemClassObject)pEvtInstance;
	177	return pSink->Indicate(1,&_pEvtInstance);
	178	}
	179	HRESULT WINAPI FireDeletionEvent(IWbemClassObject pInstanceToDelete,IWbemObjectSink pSink) {
	180	if(FAILED (m_hr)) {
	181	ATLASSERT (0);
	182	return m_hr;
	183	}
	184	if(!pInstanceToDelete \|\| !pSink) {
	185	ATLASSERT (0);
	186	return WBEM_E_INVALID_PARAMETER;
	187	}
	188	CComPtr<IWbemClassObject> pEvtInstance;
	189	HRESULT hr = m_pDeletionEventClass->SpawnInstance(0,&pEvtInstance);
	190	if(FAILED(hr)) return hr;
	191	VARIANT var;
	192	VariantInit(&var);
	193	var.vt = VT_UNKNOWN;
	194	CComQIPtr<IUnknown,&IID_IUnknown>pTemp(pInstanceToDelete);
	195	var.punkVal = pTemp;
	196	hr = pEvtInstance->Put(L"TargetInstance",0,&var,0);
	197	if(FAILED(hr)) return hr;
	198	IWbemClassObject _pEvtInstance = (IWbemClassObject)pEvtInstance;
	199	return pSink->Indicate(1,&_pEvtInstance);
	200	}
	201	HRESULT WINAPI FireModificationEvent(IWbemClassObject pOldInstance,IWbemClassObject pNewInstance,IWbemObjectSink *pSink) {
	202	if(FAILED (m_hr)) {
	203	ATLASSERT (0);
	204	return m_hr;
	205	}
	206	if(!pOldInstance \|\| !pNewInstance \|\| !pSink) {
	207	ATLASSERT (0);
	208	return WBEM_E_INVALID_PARAMETER;
	209	}
	210	CComPtr<IWbemClassObject> pEvtInstance;
	211	HRESULT hr = m_pModificationEventClass->SpawnInstance(0,&pEvtInstance);
	212	if(FAILED(hr)) return hr;
	213	VARIANT var;
	214	VariantInit(&var);
	215	var.vt = VT_UNKNOWN;
	216	CComQIPtr<IUnknown,&IID_IUnknown>pTempNew(pNewInstance);
	217	var.punkVal = pTempNew;
	218	hr = pEvtInstance->Put(L"TargetInstance",0,&var,0);
	219	if(FAILED(hr)) return hr;
	220	CComQIPtr<IUnknown,&IID_IUnknown>pTempOld(pOldInstance);
	221	var.punkVal = pTempOld;
	222	hr = pEvtInstance->Put(L"PreviousInstance",0,&var,0);
	223	if(FAILED(hr)) return hr;
	224	IWbemClassObject _pEvtInstance = (IWbemClassObject)pEvtInstance;
	225	return pSink->Indicate(1,&_pEvtInstance);
	226	}
	227	};
	228
	229	class CInstanceProviderHelper : public CProviderHelper {
	230	public:
	231	CInstanceProviderHelper (IWbemServices pNamespace,IWbemContext pCtx) : CProviderHelper (pNamespace,pCtx) { }
	232	virtual ~CInstanceProviderHelper() { }
	233	HRESULT WINAPI CheckInstancePath(IClassFactory *pParserFactory,const BSTR ObjectPath,const BSTR ClassName,ULONGLONG ullTest) {
	234	if(!pParserFactory) {
	235	ATLASSERT (0);
	236	return WBEM_E_INVALID_PARAMETER;
	237	}
	238	CComPtr<IWbemPath>pPath;
	239	HRESULT hr = pParserFactory->CreateInstance(NULL,IID_IWbemPath,(void **) &pPath);
	240	if(FAILED(hr)) return WBEM_E_INVALID_PARAMETER;
	241	hr = pPath->SetText(WBEMPATH_CREATE_ACCEPT_ALL,ObjectPath);
	242	if(FAILED(hr)) return hr;
	243	unsigned int nPathLen = SysStringLen(ObjectPath);
	244	if(nPathLen >= (unsigned __LONG32)(-1)) return HRESULT_FROM_WIN32(ERROR_ARITHMETIC_OVERFLOW);
	245	unsigned __LONG32 ulBufLen = (unsigned __LONG32)(nPathLen + 1);
	246	WCHAR *wClass = new WCHAR[ulBufLen];
	247	if(!wClass) return WBEM_E_OUT_OF_MEMORY;
	248	hr = pPath->GetClassName(&ulBufLen,wClass);
	249	if(FAILED(hr)) {
	250	delete[] wClass;
	251	return hr;
	252	}
	253	DWORD lcid = MAKELCID(MAKELANGID(LANG_ENGLISH,SUBLANG_ENGLISH_US),SORT_DEFAULT);
	254	if(CSTR_EQUAL!=CompareStringW(lcid,NORM_IGNORECASE,ClassName,-1,wClass,-1)) {
	255	delete[] wClass;
	256	return WBEM_E_NOT_FOUND;
	257	}
	258	delete[] wClass;
	259	__MINGW_EXTENSION unsigned __int64 ullPathInfo;
	260	hr = pPath->GetInfo((ULONG)0,&ullPathInfo);
	261	if(FAILED(hr)) return hr;
	262	if(!(ullPathInfo & ullTest)) return WBEM_E_INVALID_OBJECT_PATH;
	263	return WBEM_S_NO_ERROR;
	264	}
	265	};
	266
	267	template <class T> class ATL_NO_VTABLE IWbemPullClassProviderImpl : public IWbemServices,public IWbemProviderInit {
	268	public:
	269	virtual HRESULT WINAPI OpenNamespace(const BSTR strNamespace,__LONG32 lFlags,IWbemContext pCtx,IWbemServices ppWorkingNamespace,IWbemCallResult *ppResult){return WBEM_E_NOT_SUPPORTED;};
	270	virtual HRESULT WINAPI CancelAsyncCall(IWbemObjectSink *pSink){return WBEM_E_NOT_SUPPORTED;};
	271	virtual HRESULT WINAPI QueryObjectSink(__LONG32 lFlags,IWbemObjectSink **ppResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	272	virtual HRESULT WINAPI GetObject(const BSTR strObjectPath,__LONG32 lFlags,IWbemContext pCtx,IWbemClassObject ppObject,IWbemCallResult *ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	273	virtual HRESULT WINAPI PutClass(IWbemClassObject pObject,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult **ppCallResult){return WBEM_E_NOT_SUPPORTED;};
	274	virtual HRESULT WINAPI DeleteClass(const BSTR strClass,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult *ppCallResult){return WBEM_E_NOT_SUPPORTED;};
	275	virtual HRESULT WINAPI CreateClassEnum(const BSTR strSuperclass,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	276	virtual HRESULT WINAPI PutInstance(IWbemClassObject pInst,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult **ppCallResult){return WBEM_E_NOT_SUPPORTED;};
	277	virtual HRESULT WINAPI PutInstanceAsync(IWbemClassObject pInst,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink *pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	278	virtual HRESULT WINAPI DeleteInstance(const BSTR strObjectPath,__LONG32 lFlags,IWbemContext pCtx,IWbemCallResult *ppCallResult){return WBEM_E_NOT_SUPPORTED;};
	279	virtual HRESULT WINAPI DeleteInstanceAsync(const BSTR strObjectPath,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink pResponseHandler){return WBEM_E_NOT_SUPPORTED;};
	280	virtual HRESULT WINAPI CreateInstanceEnum(const BSTR strClass,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	281	virtual HRESULT WINAPI CreateInstanceEnumAsync(const BSTR strClass,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	282	virtual HRESULT WINAPI ExecQuery(const BSTR strQueryLanguage,const BSTR strQuery,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	283	virtual HRESULT WINAPI ExecNotificationQuery(const BSTR strQueryLanguage,const BSTR strQuery,__LONG32 lFlags,IWbemContext pCtx,IEnumWbemClassObject *ppEnum) {return WBEM_E_NOT_SUPPORTED;};
	284	virtual HRESULT WINAPI ExecNotificationQueryAsync(const BSTR strQueryLanguage,const BSTR strQuery,__LONG32 lFlags,IWbemContext pCtx,IWbemObjectSink pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	285	virtual HRESULT WINAPI ExecMethod(const BSTR strObjectPath,const BSTR strMethodName,__LONG32 lFlags,IWbemContext pCtx,IWbemClassObject pInParams,IWbemClassObject ppOutParams,IWbemCallResult ppCallResult) {return WBEM_E_NOT_SUPPORTED;};
	286	virtual HRESULT WINAPI ExecMethodAsync(const BSTR strObjectPath,const BSTR strMethodName,__LONG32 lFlags,IWbemContext pCtx,IWbemClassObject pInParams,IWbemObjectSink *pResponseHandler) {return WBEM_E_NOT_SUPPORTED;};
	287	};
	288
	289	class CImpersonateClientHelper {
	290	private:
	291	WINBOOL m_bImpersonate;
	292	public:
	293	CImpersonateClientHelper() { m_bImpersonate = FALSE; }
	294	~CImpersonateClientHelper() {
	295	if(m_bImpersonate)
	296	CoRevertToSelf();
	297	}
	298	HRESULT ImpersonateClient() {
	299	HRESULT hr = S_OK;
	300	if(SUCCEEDED(hr = CoImpersonateClient())) m_bImpersonate = TRUE;
	301	return hr;
	302	}
	303	HRESULT GetCurrentImpersonationLevel (DWORD & a_Level) {
	304	DWORD t_ImpersonationLevel = RPC_C_IMP_LEVEL_ANONYMOUS;
	305	HANDLE t_ThreadToken = NULL;
	306	HRESULT t_Result = S_OK;
	307	if(SUCCEEDED(t_Result = CoImpersonateClient())) {
	308	WINBOOL t_Status = OpenThreadToken (GetCurrentThread() ,TOKEN_QUERY,TRUE,&t_ThreadToken);
	309	if(t_Status) {
	310	SECURITY_IMPERSONATION_LEVEL t_Level = SecurityAnonymous;
	311	DWORD t_Returned = 0;
	312	t_Status = GetTokenInformation (t_ThreadToken ,TokenImpersonationLevel ,&t_Level ,sizeof(SECURITY_IMPERSONATION_LEVEL),&t_Returned);
	313	CloseHandle (t_ThreadToken);
	314	if(t_Status==FALSE) {
	315	t_Result = MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,GetLastError());
	316	} else {
	317	switch(t_Level) {
	318	case SecurityAnonymous:
	319	{
	320	t_ImpersonationLevel = RPC_C_IMP_LEVEL_ANONYMOUS;
	321	}
	322	break;
	323	case SecurityIdentification:
	324	{
	325	t_ImpersonationLevel = RPC_C_IMP_LEVEL_IDENTIFY;
	326	}
	327	break;
	328	case SecurityImpersonation:
	329	{
	330	t_ImpersonationLevel = RPC_C_IMP_LEVEL_IMPERSONATE;
	331	}
	332	break;
	333	case SecurityDelegation:
	334	{
	335	t_ImpersonationLevel = RPC_C_IMP_LEVEL_DELEGATE;
	336	}
	337	break;
	338	default:
	339	{
	340	t_Result = MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,E_UNEXPECTED);
	341	}
	342	break;
	343	}
	344	}
	345	} else {
	346	t_Result = MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,GetLastError());
	347	}
	348	CoRevertToSelf();
	349	}
	350	a_Level = t_ImpersonationLevel;
	351	return t_Result;
	352	}
	353	};
	354	}
	355	#endif

Note: See TracBrowser for help on using the repository browser.

Context Navigation

source: Daodan/MSYS2/mingw32/i686-w64-mingw32/include/wmiatlprov.h

Download in other formats: