PE信息获取 记录

#pragma once
#include <DbgHelp.h>  //ImageRvaToVa使用到

#pragma comment(lib,"Dbghelp")
#define _in
#define _out

typedef struct MAP_FILE_STRUCT
{
	HANDLE hFile;			 //文件句柄
	HANDLE hMap;			 //映射句柄
	LPVOID ImageBase;		 //文件基地址
	DWORD dwSize;			 //文件大小 
	char FileName[MAX_PATH]; //文件名
}MAP_FILE_STRUCT,*PMAP_FILE_STRUCT;

enum{
	LFILE_FROM_NOMAL=0,       //文件打开时的处理方式
	LFILE_FROM_FILE,          //从文件打开
	LFILE_FROM_MEMORY         //从内存打开
};




class LPEFile
{
public:
	LPEFile(void);
	~LPEFile(void);
private:
	BYTE m_IsOpen;                    //文件打开状态
	MAP_FILE_STRUCT* m_pstMapFile;    //文件及PE信息结构体
public:
	//获得文件名称
	void GetFileName(char* FileName);  
	BOOL LoadPeFile(char* lpFile,DWORD dwGeneric);
	BOOL Free();
	MAP_FILE_STRUCT* GetInfo();    //返回文件及PE信息对应的结构体
	void SetInfo(MAP_FILE_STRUCT* pstMapFile);
	void operator=(LPEFile& lPeFile);
	BYTE* GetImage(void);
	DWORD GetImageSize(void);
	获得Dos头
	PIMAGE_DOS_HEADER GetDosHeader(void);
	获得NT头
	PIMAGE_NT_HEADERS GetNtHeader(void);
	PIMAGE_FILE_HEADER GetFileHeader(void);
	PIMAGE_OPTIONAL_HEADER GetOptionalHeader(void);
	PIMAGE_DATA_DIRECTORY GetDataDirectory(void);
	PIMAGE_DATA_DIRECTORY GetDataDirectory(int nIndex);
	PIMAGE_SECTION_HEADER GetFirstSection(void);
	BOOL GetFileIsPE(void);
	bool GetMachineString(WORD nValue,char* wchBuf);
	void GetSubSystemString(WORD nValue,char* wchBuf);
	PIMAGE_SECTION_HEADER GetSection(int nIndex);
	int GetSectionCount(void);
	LPVOID RvaToPtr(DWORD dwRVA);
	LPVOID GetDataDirectoryEntryToData(USHORT DirectoryEntry);
	HANDLE GetFileHandle();
	//int StrMtoW(char* chBuf, wchar_t* wchBuf);
	//int StrWtoM(wchar_t* wchBuf, char* chBuf);
};

下面是具体实现部分

#include &#