// Copyright (c) 2000-2001 Microsoft Corporation, All Rights Reserved // JobObjSecLimitInfoProv.cpp //#define _WIN32_WINNT 0x0500 #include "precomp.h" //#include #include "cominit.h" //#include //#include #include "CUnknown.h" #include #include "FRQueryEx.h" #include "globals.h" #include "Factory.h" #include "helpers.h" #include #include #include "SmartHandle.h" #include #include "CVARIANT.h" #include "CObjProps.h" #include "CJobObjSecLimitInfoProps.h" #include "JobObjSecLimitInfoProv.h" /*****************************************************************************/ // QueryInterface override to allow for this component's interface(s) /*****************************************************************************/ STDMETHODIMP CJobObjSecLimitInfoProv::QueryInterface(const IID& iid, void** ppv) { HRESULT hr = S_OK; if(iid == IID_IWbemServices) { *ppv = static_cast(this); reinterpret_cast(*ppv)->AddRef(); } else if(iid == IID_IWbemProviderInit) { *ppv = static_cast(this); reinterpret_cast(*ppv)->AddRef(); } else { hr = CUnknown::QueryInterface(iid, ppv); } return hr; } /*****************************************************************************/ // Creation function used by CFactory /*****************************************************************************/ HRESULT CJobObjSecLimitInfoProv::CreateInstance(CUnknown** ppNewComponent) { HRESULT hr = S_OK; CUnknown* pUnk = NULL; pUnk = new CJobObjSecLimitInfoProv; if(pUnk != NULL) { *ppNewComponent = pUnk; } else { hr = E_OUTOFMEMORY; } return hr ; } /*****************************************************************************/ // IWbemProviderInit implementation /*****************************************************************************/ STDMETHODIMP CJobObjSecLimitInfoProv::Initialize( LPWSTR pszUser, LONG lFlags, LPWSTR pszNamespace, LPWSTR pszLocale, IWbemServices *pNamespace, IWbemContext *pCtx, IWbemProviderInitSink *pInitSink) { m_pNamespace = pNamespace; m_chstrNamespace = pszNamespace; //Let CIMOM know you are initialized //================================== return pInitSink->SetStatus( WBEM_S_INITIALIZED, 0); } /*****************************************************************************/ // IWbemServices implementation /*****************************************************************************/ STDMETHODIMP CJobObjSecLimitInfoProv::GetObjectAsync( const BSTR ObjectPath, long lFlags, IWbemContext __RPC_FAR *pCtx, IWbemObjectSink __RPC_FAR *pResponseHandler) { HRESULT hr = WBEM_E_NOT_FOUND; IWbemClassObjectPtr pStatusObject; try { HRESULT hrImp = CheckImpersonationLevel(); if(SUCCEEDED(hrImp)) { // We need the name of the instance they requested... WCHAR wstrObjInstKeyVal[MAX_PATH]; hr = GetObjInstKeyVal( ObjectPath, IDS_Win32_NamedJobObjectSecLimitSetting, g_rgJobObjSecLimitInfoPropNames[JOSECLMTPROP_ID], wstrObjInstKeyVal, sizeof(wstrObjInstKeyVal) - sizeof(WCHAR)); if(SUCCEEDED(hr)) { // wstrObjInstKeyVal now contains the name of the object. See if // it exists... CHString chstrUndecoratedJOName; UndecorateJOName( wstrObjInstKeyVal, chstrUndecoratedJOName); SmartHandle hJob; hJob = ::OpenJobObjectW( MAXIMUM_ALLOWED, FALSE, chstrUndecoratedJOName); if(hJob) { // We seem to have found one matching the specified name, // so create a return instance... IWbemClassObjectPtr pIWCO = NULL; CJobObjSecLimitInfoProps cjoslip(hJob, m_chstrNamespace); hr = CreateInst( m_pNamespace, &pIWCO, _bstr_t(IDS_Win32_NamedJobObjectSecLimitSetting), pCtx); if(SUCCEEDED(hr)) { cjoslip.SetReqProps(PROP_ALL_REQUIRED); } if(SUCCEEDED(hr)) { // set the key properties... hr = cjoslip.SetKeysFromPath( ObjectPath, pCtx); } if(SUCCEEDED(hr)) { // set the non-key requested properties... hr = cjoslip.SetNonKeyReqProps(); } if(SUCCEEDED(hr)) { // Load requested non-key properties // to the instance... hr = cjoslip.LoadPropertyValues( pIWCO, pCtx, m_pNamespace); // Commit the instance... if(SUCCEEDED(hr)) { IWbemClassObject *pTmp = (IWbemClassObject*) pIWCO; hr = pResponseHandler->Indicate( 1, &pTmp); } } } else { hr = WBEM_E_NOT_FOUND; SetStatusObject( pCtx, m_pNamespace, ::GetLastError(), NULL, L"::OpenJobObject", JOB_OBJECT_NAMESPACE, &pStatusObject); } } } else { hr = hrImp; } } catch(CVARIANTError& cve) { hr = cve.GetWBEMError(); } catch(...) { hr = WBEM_E_PROVIDER_FAILURE; } // Set Status return pResponseHandler->SetStatus(0, hr, NULL, pStatusObject); } STDMETHODIMP CJobObjSecLimitInfoProv::ExecQueryAsync( const BSTR QueryLanguage, const BSTR Query, long lFlags, IWbemContext __RPC_FAR *pCtx, IWbemObjectSink __RPC_FAR *pResponseHandler) { HRESULT hr = WBEM_S_NO_ERROR; IWbemClassObjectPtr pStatusObject; try { HRESULT hrImp = CheckImpersonationLevel(); if(SUCCEEDED(hrImp)) { // We will optimize for those cases in which // a particular set of named job objects // (e.g., 1 or more). Enumerate also // optimizes for the properties that were // requested. CFrameworkQuery cfwq; hr = cfwq.Init( QueryLanguage, Query, lFlags, m_chstrNamespace); std::vector<_bstr_t> rgNamedJOs; if(SUCCEEDED(hr)) { hr = cfwq.GetValuesForProp( _bstr_t(g_rgJobObjSecLimitInfoPropNames[JOSECLMTPROP_ID]), rgNamedJOs); } // If none were specifically requested, they // want them all... if(rgNamedJOs.size() == 0) { hr = GetJobObjectList(rgNamedJOs); } else { // Object paths were specified. Before // passing them along, we need to un- // decorate them. UndecorateNamesInNamedJONameList(rgNamedJOs); } // Find out what propeties were requested... CJobObjSecLimitInfoProps cjoslip(m_chstrNamespace); cjoslip.GetWhichPropsReq(cfwq); if(SUCCEEDED(hr)) { hr = Enumerate( pCtx, pResponseHandler, rgNamedJOs, cjoslip, &pStatusObject); } else { SetStatusObject( pCtx, m_pNamespace, -1L, NULL, L"Helpers.cpp::GetJobObjectList", JOB_OBJECT_NAMESPACE, &pStatusObject); } } else { hr = hrImp; } } catch(CVARIANTError& cve) { hr = cve.GetWBEMError(); } catch(...) { hr = WBEM_E_PROVIDER_FAILURE; } // Set Status return pResponseHandler->SetStatus(0, hr, NULL, pStatusObject); } STDMETHODIMP CJobObjSecLimitInfoProv::CreateInstanceEnumAsync( const BSTR Class, long lFlags, IWbemContext __RPC_FAR *pCtx, IWbemObjectSink __RPC_FAR *pResponseHandler) { HRESULT hr = WBEM_S_NO_ERROR; IWbemClassObjectPtr pStatusObject; try { HRESULT hrImp = CheckImpersonationLevel(); if(SUCCEEDED(hrImp)) { if(_wcsicmp( Class, IDS_Win32_NamedJobObjectSecLimitSetting) != NULL) { hr = WBEM_E_INVALID_CLASS; } // For every job object, return all accounting // info properties... if(SUCCEEDED(hr)) { // Get a list of named jobs... std::vector<_bstr_t> rgNamedJOs; hr = GetJobObjectList(rgNamedJOs); if(SUCCEEDED(hr)) { CJobObjSecLimitInfoProps cjoslip(m_chstrNamespace); cjoslip.SetReqProps(PROP_ALL_REQUIRED); hr = Enumerate( pCtx, pResponseHandler, rgNamedJOs, cjoslip, &pStatusObject); } else { SetStatusObject( pCtx, m_pNamespace, -1L, NULL, L"Helpers.cpp::GetJobObjectList", JOB_OBJECT_NAMESPACE, &pStatusObject); } } } else { hr = hrImp; } } catch(CVARIANTError& cve) { hr = cve.GetWBEMError(); } catch(...) { hr = WBEM_E_PROVIDER_FAILURE; } // Set Status return pResponseHandler->SetStatus(0, hr, NULL, pStatusObject); } /*****************************************************************************/ // Private member function implementations /*****************************************************************************/ HRESULT CJobObjSecLimitInfoProv::Enumerate( IWbemContext __RPC_FAR *pCtx, IWbemObjectSink __RPC_FAR *pResponseHandler, std::vector<_bstr_t>& rgNamedJOs, CJobObjSecLimitInfoProps& cjoslip, IWbemClassObject** ppStatusObject) { HRESULT hr = S_OK; hr = CheckImpersonationLevel(); try // CVARIANT can throw and I want the error... { if(SUCCEEDED(hr)) { long lNumJobs = rgNamedJOs.size(); if(lNumJobs > 0) { SmartHandle hJob; for(long m = 0L; m < lNumJobs && SUCCEEDED(hr); m++) { cjoslip.ClearProps(); // We have the name of a JO; need to open it up // and get its properties... hJob = ::OpenJobObjectW( MAXIMUM_ALLOWED, FALSE, rgNamedJOs[m]); // (NOTE: hJob smarthandle class automatically // closes its handle on destruction and on // reassignment.) if(hJob) { // Set the handle... cjoslip.SetHandle(hJob); // Set the key properties directly... CHString chstrDecoratedJOName; DecorateJOName( rgNamedJOs[m], chstrDecoratedJOName); std::vector vecvKeys; CVARIANT vID(chstrDecoratedJOName); vecvKeys.push_back(vID); hr = cjoslip.SetKeysDirect(vecvKeys); if(FAILED(hr)) { SetStatusObject( pCtx, m_pNamespace, ::GetLastError(), NULL, L"CJobObjSecLimitInfoProps::SetKeysDirect", JOB_OBJECT_NAMESPACE, ppStatusObject); } if(SUCCEEDED(hr)) { // set the non-key requested // properties... hr = cjoslip.SetNonKeyReqProps(); if(FAILED(hr)) { SetStatusObject( pCtx, m_pNamespace, ::GetLastError(), NULL, L"CJobObjSecLimitInfoProps::SetNonKeyReqProps", JOB_OBJECT_NAMESPACE, ppStatusObject); } } // Create a new outgoing instance... IWbemClassObjectPtr pIWCO = NULL; if(SUCCEEDED(hr)) { hr = CreateInst( m_pNamespace, &pIWCO, _bstr_t(IDS_Win32_NamedJobObjectSecLimitSetting), pCtx); if(FAILED(hr)) { SetStatusObject( pCtx, m_pNamespace, ::GetLastError(), NULL, L"CJobObjSecLimitInfoProv::CreateInst", JOB_OBJECT_NAMESPACE, ppStatusObject); } } // Load the properties of the // new outgoing instance... if(SUCCEEDED(hr)) { hr = cjoslip.LoadPropertyValues( pIWCO, pCtx, m_pNamespace); if(FAILED(hr)) { SetStatusObject( pCtx, m_pNamespace, ::GetLastError(), NULL, L"CJobObjSecLimitInfoProps::LoadPropertyValues", JOB_OBJECT_NAMESPACE, ppStatusObject); } } // And send it out... if(SUCCEEDED(hr)) { IWbemClassObject *pTmp = (IWbemClassObject*) pIWCO; hr = pResponseHandler->Indicate( 1, &pTmp); } } else { _ASSERT(0); hr = WBEM_E_NOT_FOUND; SetStatusObject( pCtx, m_pNamespace, ::GetLastError(), NULL, L"::OpenJobObject", JOB_OBJECT_NAMESPACE, ppStatusObject); } } } } } catch(CVARIANTError& cve) { hr = cve.GetWBEMError(); } return hr; }