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 356
