// Copyright (c) 2000-2001 Microsoft Corporation, All Rights Reserved // CJobObjIOActgProps.cpp //#define _WIN32_WINNT 0x0500 #include "precomp.h" #include #include "FRQueryEx.h" #include #include "helpers.h" #include "CVARIANT.h" #include "CObjProps.h" #include "CJobObjIOActgProps.h" #include //***************************************************************************** // BEGIN: Declaration of Win32_JobObjIOActgInfo class properties. //***************************************************************************** // WARNING!! MUST KEEP MEMBERS OF THE FOLLOWING ARRAY // IN SYNCH WITH THE JOB_OBJ_PROPS ENUMERATION DECLARED // IN CJobObjProps.h !!! LPCWSTR g_rgJobObjIOActgPropNames[] = { { L"Name" }, { L"TotalUserTime" }, { L"TotalKernelTime" }, { L"ThisPeriodTotalUserTime" }, { L"ThisPeriodTotalKernelTime" }, { L"TotalPageFaultCount" }, { L"TotalProcesses" }, { L"ActiveProcesses" }, { L"TotalTerminatedProcesses" }, { L"ReadOperationCount" }, { L"WriteOperationCount" }, { L"OtherOperationCount" }, { L"ReadTransferCount" }, { L"WriteTransferCount" }, { L"OtherTransferCount" }, { L"PeakProcessMemoryUsed" }, { L"PeakJobMemoryUsed" } }; //***************************************************************************** // END: Declaration of Win32_JobObjIOActgInfo class properties. //***************************************************************************** CJobObjIOActgProps::CJobObjIOActgProps(CHString& chstrNamespace) : CObjProps(chstrNamespace) { } CJobObjIOActgProps::CJobObjIOActgProps( HANDLE hJob, CHString& chstrNamespace) : CObjProps(chstrNamespace), m_hJob(hJob) { } CJobObjIOActgProps::~CJobObjIOActgProps() { } // Clients call this to establish which properties // were requested. This function calls a base class // helper, which calls our CheckProps function. // The base class helper finally stores the result // in the base class member m_dwReqProps. HRESULT CJobObjIOActgProps::GetWhichPropsReq( CFrameworkQuery& cfwq) { HRESULT hr = S_OK; // Call base class version for help. // Base class version will call our // CheckProps function. hr = CObjProps::GetWhichPropsReq( cfwq, CheckProps); return hr; } DWORD CJobObjIOActgProps::CheckProps( CFrameworkQuery& Query) { DWORD dwReqProps = PROP_NONE_REQUIRED; // Get the requested properties for this // specific object... if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_ID])) dwReqProps |= PROP_JOIOActgID; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_TotalUserTime])) dwReqProps |= PROP_TotalUserTime; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_TotalKernelTime])) dwReqProps |= PROP_TotalKernelTime; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_ThisPeriodTotalUserTime])) dwReqProps |= PROP_ThisPeriodTotalUserTime; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_ThisPeriodTotalKernelTime])) dwReqProps |= PROP_ThisPeriodTotalKernelTime; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_TotalPageFaultCount])) dwReqProps |= PROP_TotalPageFaultCount; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_TotalProcesses])) dwReqProps |= PROP_TotalProcesses; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_ActiveProcesses])) dwReqProps |= PROP_ActiveProcesses; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_TotalTerminatedProcesses])) dwReqProps |= PROP_TotalTerminatedProcesses; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_ReadOperationCount])) dwReqProps |= PROP_ReadOperationCount; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_WriteOperationCount])) dwReqProps |= PROP_WriteOperationCount; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_OtherOperationCount])) dwReqProps |= PROP_OtherOperationCount; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_ReadTransferCount])) dwReqProps |= PROP_ReadTransferCount; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_WriteTransferCount])) dwReqProps |= PROP_WriteTransferCount; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_OtherTransferCount])) dwReqProps |= PROP_OtherTransferCount; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_PeakProcessMemoryUsed])) dwReqProps |= PROP_PeakProcessMemoryUsed; if (Query.IsPropertyRequired(g_rgJobObjIOActgPropNames[JOIOACTGPROP_PeakJobMemoryUsed])) dwReqProps |= PROP_PeakJobMemoryUsed; return dwReqProps; } void CJobObjIOActgProps::SetHandle( const HANDLE hJob) { m_hJob = hJob; } HANDLE& CJobObjIOActgProps::GetHandle() { _ASSERT(m_hJob); return m_hJob; } // Sets the key properties from the ObjectPath. HRESULT CJobObjIOActgProps::SetKeysFromPath( const BSTR ObjectPath, IWbemContext __RPC_FAR *pCtx) { HRESULT hr = WBEM_S_NO_ERROR; // This array contains the key field names CHStringArray rgchstrKeys; rgchstrKeys.Add(g_rgJobObjIOActgPropNames[JOIOACTGPROP_ID]); // This array contains the index numbers // in m_PropMap corresponding to the keys. short sKeyNum[1]; sKeyNum[0] = JOIOACTGPROP_ID; hr = CObjProps::SetKeysFromPath( ObjectPath, pCtx, IDS_Win32_JobObjectIOAccountingInfo, rgchstrKeys, sKeyNum); return hr; } // Sets the key property from in supplied // parameter. HRESULT CJobObjIOActgProps::SetKeysDirect( std::vector& vecvKeys) { HRESULT hr = WBEM_S_NO_ERROR; if(vecvKeys.size() == 1) { short sKeyNum[1]; sKeyNum[0] = JOIOACTGPROP_ID; hr = CObjProps::SetKeysDirect( vecvKeys, sKeyNum); } else { hr = WBEM_E_INVALID_PARAMETER; } return hr; } // Sets the non-key properties. Only those // properties requested are set (as determined // by base class member m_dwReqProps). HRESULT CJobObjIOActgProps::SetNonKeyReqProps() { HRESULT hr = WBEM_S_NO_ERROR; DWORD dwReqProps = GetReqProps(); _ASSERT(m_hJob); if(!m_hJob) return WBEM_E_INVALID_PARAMETER; // Because all the properties of this class // come from the same underlying win32 job // object structure, we only need to get that // structure one time. We only need to get // it at all if at least one non-key property // was requested. if(dwReqProps != PROP_NONE_REQUIRED) { // Get the value from the underlying JO: JOBOBJECT_BASIC_AND_IO_ACCOUNTING_INFORMATION jobaioai; JOBOBJECT_EXTENDED_LIMIT_INFORMATION joeli; BOOL fQIJO = ::QueryInformationJobObject( m_hJob, JobObjectBasicAndIoAccountingInformation, &jobaioai, sizeof(JOBOBJECT_BASIC_AND_IO_ACCOUNTING_INFORMATION), NULL); if(fQIJO) { fQIJO = ::QueryInformationJobObject( m_hJob, JobObjectExtendedLimitInformation, &joeli, sizeof(JOBOBJECT_EXTENDED_LIMIT_INFORMATION), NULL); } if(!fQIJO) { _ASSERT(0); hr = WBEM_E_FAILED; } else { try // CVARIANT can throw { // Get all the requested values... if(dwReqProps & PROP_TotalUserTime) { ULONGLONG llTotalUserTime = (ULONGLONG)jobaioai.BasicInfo.TotalUserTime.QuadPart; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_TotalUserTime, new CVARIANT(llTotalUserTime))); } if(dwReqProps & PROP_TotalKernelTime) { ULONGLONG llTotalKernelTime = (ULONGLONG)jobaioai.BasicInfo.TotalKernelTime.QuadPart; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_TotalKernelTime, new CVARIANT(llTotalKernelTime))); } if(dwReqProps & PROP_ThisPeriodTotalUserTime) { ULONGLONG llThisPeriodTotalUserTime = (ULONGLONG)jobaioai.BasicInfo.ThisPeriodTotalUserTime.QuadPart; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_ThisPeriodTotalUserTime, new CVARIANT(llThisPeriodTotalUserTime))); } if(dwReqProps & PROP_ThisPeriodTotalKernelTime) { ULONGLONG llThisPeriodTotalKernelTime = (ULONGLONG)jobaioai.BasicInfo.ThisPeriodTotalKernelTime.QuadPart; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_ThisPeriodTotalKernelTime, new CVARIANT(llThisPeriodTotalKernelTime))); } if(dwReqProps & PROP_TotalPageFaultCount) { DWORD dwTotalPageFaultCount = jobaioai.BasicInfo.TotalPageFaultCount; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_TotalPageFaultCount, new CVARIANT(dwTotalPageFaultCount))); } if(dwReqProps & PROP_TotalProcesses) { DWORD dwTotalProcesses = jobaioai.BasicInfo.TotalProcesses; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_TotalProcesses, new CVARIANT(dwTotalProcesses))); } if(dwReqProps & PROP_ActiveProcesses) { DWORD dwActiveProcesses = jobaioai.BasicInfo.ActiveProcesses; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_ActiveProcesses, new CVARIANT(dwActiveProcesses))); } if(dwReqProps & PROP_TotalTerminatedProcesses) { DWORD dwTotalTerminatedProcesses = jobaioai.BasicInfo.TotalTerminatedProcesses; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_TotalTerminatedProcesses, new CVARIANT(dwTotalTerminatedProcesses))); } if(dwReqProps & PROP_ReadOperationCount) { ULONGLONG ullReadOperationCount = jobaioai.IoInfo.ReadOperationCount; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_ReadOperationCount, new CVARIANT(ullReadOperationCount))); } if(dwReqProps & PROP_WriteOperationCount) { ULONGLONG ullWriteOperationCount = jobaioai.IoInfo.WriteOperationCount; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_WriteOperationCount, new CVARIANT(ullWriteOperationCount))); } if(dwReqProps & PROP_OtherOperationCount) { ULONGLONG ullOtherOperationCount = jobaioai.IoInfo.OtherOperationCount; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_OtherOperationCount, new CVARIANT(ullOtherOperationCount))); } if(dwReqProps & PROP_ReadTransferCount) { ULONGLONG ullReadTransferCount = jobaioai.IoInfo.ReadTransferCount; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_ReadTransferCount, new CVARIANT(ullReadTransferCount))); } if(dwReqProps & PROP_WriteTransferCount) { ULONGLONG ullWriteTransferCount = jobaioai.IoInfo.WriteTransferCount; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_WriteTransferCount, new CVARIANT(ullWriteTransferCount))); } if(dwReqProps & PROP_OtherTransferCount) { ULONGLONG ullOtherTransferCount = jobaioai.IoInfo.OtherTransferCount; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_OtherTransferCount, new CVARIANT(ullOtherTransferCount))); } if(dwReqProps & PROP_PeakProcessMemoryUsed) { DWORD dwPeakProcessMemoryUsed = joeli.PeakProcessMemoryUsed; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_PeakProcessMemoryUsed, new CVARIANT(dwPeakProcessMemoryUsed))); } if(dwReqProps & PROP_PeakJobMemoryUsed) { DWORD dwPeakJobMemoryUsed = joeli.PeakJobMemoryUsed; m_PropMap.insert(SHORT2PVARIANT::value_type( JOIOACTGPROP_PeakJobMemoryUsed, new CVARIANT(dwPeakJobMemoryUsed))); } } catch(CVARIANTError& cve) { hr = cve.GetWBEMError(); } } } return hr; } HRESULT CJobObjIOActgProps::LoadPropertyValues( IWbemClassObject* pIWCO) { HRESULT hr = WBEM_S_NO_ERROR; if(!pIWCO) return E_POINTER; hr = CObjProps::LoadPropertyValues( g_rgJobObjIOActgPropNames, pIWCO); return hr; }