ProSysLib: Dissecting the Process

// PSLProcessIO.cpp : Implementation of CPSLProcessIO

#include "stdafx.h"
#include "PSLProcessIO.h"

CPSLProcessIO::CPSLProcessIO()
{
	m_ProcessID = NULL;
	m_CurrentProcessID = ::GetCurrentProcessId();
}

HRESULT CPSLProcessIO::FinalConstruct()
{
	return S_OK;
}

void CPSLProcessIO::FinalRelease()
{
}

void CPSLProcessIO::Initialize(long ProcessID)
{
	m_ProcessID = ProcessID;
}

bool CPSLProcessIO::GetProcessIO(IO_COUNTERS * pCounters)
{
	bool bProcessOpen = false;
	HANDLE hProcess = NULL;
	if(m_ProcessID == m_CurrentProcessID)
		hProcess = ::GetCurrentProcess();
	else
	{
		hProcess = ::OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, m_ProcessID);
		if(hProcess)
			bProcessOpen = true;
		else
		{
			if(::GetLastError() == ERROR_ACCESS_DENIED)
				SetException(exAccessDenied);
			return false;
		}
	}

	bool bResult = false;

	::memset(pCounters, 0, sizeof(IO_COUNTERS));
	if(::GetProcessIoCounters(hProcess, pCounters))
		bResult = true;

	if(bProcessOpen)
		::CloseHandle(hProcess);

	return bResult;
}


////////////////////////////////////////////////////////////////////////
// Interface Implementation;
////////////////////////////////////////////////////////////////////////


STDMETHODIMP CPSLProcessIO::get_ReadOperationCount(VARIANT * pValue)
{
	PSL_BEGIN

	IO_COUNTERS ProcessIO;
	if(GetProcessIO(&ProcessIO))
		CPSLUtilities::SetVariant64Bit(pValue, ProcessIO.ReadOperationCount);
	else
		CPSLUtilities::SetVariant64Bit(pValue, 0);

	PSL_END
}

STDMETHODIMP CPSLProcessIO::get_WriteOperationCount(VARIANT * pValue)
{
	PSL_BEGIN

	IO_COUNTERS ProcessIO;
	if(GetProcessIO(&ProcessIO))
		CPSLUtilities::SetVariant64Bit(pValue, ProcessIO.WriteOperationCount);
	else
		CPSLUtilities::SetVariant64Bit(pValue, 0);

	PSL_END
}

STDMETHODIMP CPSLProcessIO::get_OtherOperationCount(VARIANT * pValue)
{
	PSL_BEGIN

	IO_COUNTERS ProcessIO;
	if(GetProcessIO(&ProcessIO))
		CPSLUtilities::SetVariant64Bit(pValue, ProcessIO.OtherOperationCount);
	else
		CPSLUtilities::SetVariant64Bit(pValue, 0);

	PSL_END
}

STDMETHODIMP CPSLProcessIO::get_ReadTransferCount(VARIANT * pValue)
{
	PSL_BEGIN

	IO_COUNTERS ProcessIO;
	if(GetProcessIO(&ProcessIO))
		CPSLUtilities::SetVariant64Bit(pValue, ProcessIO.ReadTransferCount);
	else
		CPSLUtilities::SetVariant64Bit(pValue, 0);

	PSL_END
}

STDMETHODIMP CPSLProcessIO::get_WriteTransferCount(VARIANT * pValue)
{
	PSL_BEGIN

	IO_COUNTERS ProcessIO;
	if(GetProcessIO(&ProcessIO))
		CPSLUtilities::SetVariant64Bit(pValue, ProcessIO.WriteTransferCount);
	else
		CPSLUtilities::SetVariant64Bit(pValue, 0);

	PSL_END
}

STDMETHODIMP CPSLProcessIO::get_OtherTransferCount(VARIANT * pValue)
{
	PSL_BEGIN

	IO_COUNTERS ProcessIO;
	if(GetProcessIO(&ProcessIO))
		CPSLUtilities::SetVariant64Bit(pValue, ProcessIO.OtherTransferCount);
	else
		CPSLUtilities::SetVariant64Bit(pValue, 0);

	PSL_END
}