| 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
|
|---|
