ProSysLib: Dissecting the Process

// PSLModuleVersion.cpp : Implementation of CPSLModuleVersion

#include "stdafx.h"
#include "PSLModuleVersion.h"

CPSLModuleVersion::CPSLModuleVersion()
{
	m_bValid = false;
	m_bDebug = false;
	m_bPatched = false;
	m_bPreRelease = false;

	m_pVersionBuffer = NULL;
	m_bInitialized = false;

	m_pProductVersion = NULL;
	m_pFileVersion = NULL;
}

CPSLModuleVersion::~CPSLModuleVersion()
{
	if(m_pVersionBuffer)
		delete []m_pVersionBuffer;
}

HRESULT CPSLModuleVersion::FinalConstruct()
{
	CComObject<CPSLVersionNumber>::CreateInstance(&m_pFileVersion);
	m_pFileVersion->AddRef();

	CComObject<CPSLVersionNumber>::CreateInstance(&m_pProductVersion);
	m_pProductVersion->AddRef();

	return S_OK;
}

void CPSLModuleVersion::FinalRelease()
{
	m_pProductVersion->Release();
	m_pFileVersion->Release();
}

void CPSLModuleVersion::InitInternal(LPCTSTR strPath)
{
	// Internally, we need to attempt initialization only once;
	if(!m_bInitialized)
	{
		m_bInitialized = true;
		InitExternal(strPath);
	}
}

bool CPSLModuleVersion::InitExternal(LPCTSTR strPath)
{
	if(m_pVersionBuffer)
	{
		delete []m_pVersionBuffer;
		m_pVersionBuffer = NULL;
	}

	m_bValid = false;

	DWORD dwUnused = 0;
	DWORD dwSize = ::GetFileVersionInfoSize(strPath, &dwUnused);
	if(!dwSize)
		return false;

	UINT nLen = 0;
	m_pVersionBuffer = new BYTE[dwSize];
	VS_FIXEDFILEINFO * pInfo = NULL;
	DWORD dwVersion = 0;

	BOOL bRes = ::GetFileVersionInfo(strPath, dwUnused, dwSize, m_pVersionBuffer);
	if(bRes)
	{
		m_bValid = true;

		if(::VerQueryValue(m_pVersionBuffer, _T("\\"), (LPVOID*)&pInfo, &nLen))
		{
			long HighVersion = HIWORD(pInfo->dwProductVersionMS);
			long LowVersion = LOWORD(pInfo->dwProductVersionMS);
			long BuildNumber = HIWORD(pInfo->dwProductVersionLS);
			long RevisionNumber = LOWORD(pInfo->dwProductVersionLS);

			m_pProductVersion->Initialize(HighVersion, LowVersion, BuildNumber, RevisionNumber);

			HighVersion = HIWORD(pInfo->dwFileVersionMS);
			LowVersion = LOWORD(pInfo->dwFileVersionMS);
			BuildNumber = HIWORD(pInfo->dwFileVersionLS);
			RevisionNumber = LOWORD(pInfo->dwFileVersionLS);

			m_pFileVersion->Initialize(HighVersion, LowVersion, BuildNumber, RevisionNumber);

			if(pInfo->dwFileFlags & VS_FF_DEBUG)
				m_bDebug = true;

			if(pInfo->dwFileFlags & VS_FF_PATCHED)
				m_bPatched = true;

			if(pInfo->dwFileFlags & VS_FF_PRERELEASE)
				m_bPreRelease = true;
		}
	}

	return m_bValid;
}

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

STDMETHODIMP CPSLModuleVersion::get_IsValid(VARIANT_BOOL * pValue)
{
	PSL_BEGIN

	*pValue = m_bValid?VARIANT_TRUE:VARIANT_FALSE;

	PSL_END
}

STDMETHODIMP CPSLModuleVersion::get_IsDebug(VARIANT_BOOL * pValue)
{
	PSL_BEGIN

	*pValue = m_bDebug?VARIANT_TRUE:VARIANT_FALSE;

	PSL_END
}

STDMETHODIMP CPSLModuleVersion::get_IsPatched(VARIANT_BOOL * pValue)
{
	PSL_BEGIN

	*pValue = m_bPatched?VARIANT_TRUE:VARIANT_FALSE;

	PSL_END
}

STDMETHODIMP CPSLModuleVersion::get_IsPreRelease(VARIANT_BOOL * pValue)
{
	PSL_BEGIN

	*pValue = m_bPreRelease?VARIANT_TRUE:VARIANT_FALSE;

	PSL_END
}

STDMETHODIMP CPSLModuleVersion::get_ProductVersion(IPSLVersionNumber ** ppValue)
{
	PSL_BEGIN

	m_pProductVersion->AddRef();
	*ppValue = CComPtr<IPSLVersionNumber>(m_pProductVersion);

	PSL_END
}

STDMETHODIMP CPSLModuleVersion::get_FileVersion(IPSLVersionNumber ** ppValue)
{
	PSL_BEGIN

	m_pFileVersion->AddRef();
	*ppValue = CComPtr<IPSLVersionNumber>(m_pFileVersion);

	PSL_END
}