puPEinfoData.h

#pragma once
#include "stdafx.h"
#ifndef PUPEINFODATA_H_
#define PUPEINFODATA_H_

/*
	类名称：PuPeInfo
	类用途：	公共接口类，提供可执行文件的PE信息
	时间：	2018-11-29
*/
class PuPEInfo
{
public:
	PuPEInfo();

	~PuPEInfo();
	
	/*公开接口*/
public:
	void* puGetImageBase(){ return m_pFileBase; }

	void* puGetNtHeadre(){ return m_pNtHeader; }

	void* puGetSection(){ return m_SectionHeader; }

	DWORD puFileSize(){ return m_FileSize; }

	BOOL puOpenFileLoad(const CString & PathName){ return prOpenFile(PathName); }

	BOOL puIsPEFile(){ return IsPEFile(); }

	DWORD puRVAofFOA(const DWORD Rva){ return RVAofFOA(Rva); }

	CString puFilePath(){ return m_strNamePath; }

	HANDLE puFileHandle() { return m_hFileHandle; }

	DWORD puOldOep(){ return this->m_OldOEP; }

	int puGetSectionCount() { return this->m_SectionCount; }

	PIMAGE_SECTION_HEADER puGetSectionAddress(const char* Base, const BYTE* Name){ return this->GetSectionAddress(Base, Name); }

	// 设置文件偏移以及文件大小
	BOOL puSetFileoffsetAndFileSize(const void* Base, const DWORD & offset, const DWORD size, const BYTE* Name)
	{
		return this->SetFileoffsetAndFileSize(Base, offset, size, Name);
	}


	/*私有方法及数据*/
private:
	// 文件读取
	BOOL prOpenFile(const CString & PathName);

	// PE文件判断
	BOOL IsPEFile();

	// RVAofFOA
	DWORD RVAofFOA(const DWORD Rva);

	// 根据区段名称获取区段首地址
	PIMAGE_SECTION_HEADER GetSectionAddress(const char* Base, const BYTE* SectionName);

	// 设置文件偏移以及文件大小
	BOOL SetFileoffsetAndFileSize(const void* Base, const DWORD & offset, const DWORD size, const BYTE* Name);

	// 保存ImageBase
	static void* m_pFileBase;

	// 保存PE头
	static void* m_pNtHeader;

	// 保存Section
	static void* m_SectionHeader;

	// 保存文件大小
	static DWORD m_FileSize;

	// 保存文件路径
	static CString m_strNamePath;

	// 保存文件句柄
	static HANDLE m_hFileHandle;

	// 保存原始OEP
	static DWORD m_OldOEP;

	// 保存区段个数
	static int	m_SectionCount;

	// 标记原始OEP
	static BOOL OepFlag;

};

#endif