XPEInfo - a non-MFC class to get info from PE file

// XPEInfo.cpp  Version 1.0
//
// Author:  Hans Dietrich
//          hdietrich@gmail.com
//
// License:
//     This software is released under the Code Project Open License (CPOL),
//     which may be found here:  http://www.codeproject.com/info/eula.aspx
//     You are free to use this software in any way you like, except that you 
//     may not sell this source code.
//
//     This software is provided "as is" with no expressed or implied warranty.
//     I accept no liability for any damage or loss of business that this 
//     software may cause.
//
///////////////////////////////////////////////////////////////////////////////

#ifndef XPEINFO_H
#define XPEINFO_H

#include <time.h>

#ifdef __AFX_H__
#define CWTLString CString
#else
#include "wtlstring.h"
#endif // __AFX_H__

//=============================================================================
class CXPEInfo
//=============================================================================
{
// Construction
public:
	CXPEInfo(LPCTSTR lpszFile = NULL);
	virtual ~CXPEInfo();

// Attributes
public:
	DWORD		CalculateChecksum();
	DWORD		GetCertificateInfo();
	WORD		GetCharacteristics();
	TCHAR *		GetCharacteristicString(WORD bitmask);
	DWORD		GetChecksum();
	WORD		GetDllCharacteristics();
	TCHAR *		GetDllCharacteristicString(WORD bitmask);
	DWORD		GetDotNetFlags();
	TCHAR *		GetDotNetFlagString(DWORD bitmask);
	DWORD		GetDotNetVersion();
	DWORD		GetImageVersion();
	WORD		GetMachineType();
	CWTLString	GetMachineTypeString();
	DWORD		GetRequiredOSVersion();
	WORD		GetSubsystem();
	CWTLString	GetSubsystemString();
	DWORD		GetSubsystemVersion();
	time_t		GetTimeStamp();
	TCHAR *		GetTimeStampString();

	BOOL Is64Bit();
	BOOL IsDebug();
	BOOL IsDotNet();
	BOOL IsOpen()	{ return m_hFile != 0; }
	BOOL IsSigned();
	BOOL IsValid();

	struct MACHINE_TYPE
	{
		WORD	nMachineType;
		TCHAR *	pszMachineType;
	};

	static MACHINE_TYPE m_aMachineTypes[];

	struct CHARACTERISTIC 
	{
		WORD	bitmask;
		TCHAR * pszCharacteristic;
	};

	static CHARACTERISTIC m_aCharacteristics[];

	static TCHAR * m_aSubsystem[];
	static int m_nSubsystem;

	static CHARACTERISTIC m_aDllCharacteristics[];

	struct DOTNETFLAG
	{
		DWORD	bitmask;
		TCHAR * pszFlag;
	};
	
	static DOTNETFLAG m_aDotNetFlags[];


// Operations
public:
	void Close();
	BOOL Open(LPCTSTR lpszFile);

// Implementation
protected:
	HANDLE					m_hFile;
	HANDLE					m_hFileMapping;
	BYTE *					m_pMappedFileBase;
	IMAGE_DOS_HEADER *		m_pDOSHeader;
	IMAGE_FILE_HEADER *		m_pImgFileHdr;
	IMAGE_NT_HEADERS *		m_pNTHeaders32;
	IMAGE_NT_HEADERS64 *	m_pNTHeaders64;
};

#endif //XPEINFO_H