peimagefile.h

此为本书的配套光盘.本书不但由浅入深地讲解了软件保护技术

/********************************************************************

	Copyright (c) Beijing Feitian Technologies
	http://www.FTSafe.com

	File :		PEImageFile.h	

	Created:	2003/11/04

	Author:		yihai
	
	Purpose:	?

	Revision:	?

*********************************************************************/

#if !defined(AFX_PEIMAGEFILE_H__C1AD1BA4_0B86_4C4D_9A0F_88744A5EA5C7__INCLUDED_)
#define AFX_PEIMAGEFILE_H__C1AD1BA4_0B86_4C4D_9A0F_88744A5EA5C7__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#define MakePtr( cast, ptr, addValue ) (cast)( (DWORD)(ptr) + (DWORD)(addValue))
#define	__countof(a)	sizeof(a)/sizeof(a[0])

class CPEImageFile  
{
public:	
	void ReadHeader();
	PIMAGE_SECTION_HEADER GetFirstSection(){ return IMAGE_FIRST_SECTION(m_pNtHdr);};
	bool m_bWrite;
	enum {TIMG_UNKNOW,TIMG_PE};
	UINT	m_uImageType;
	TCHAR	m_szFileName[MAX_PATH];
	HANDLE	m_hFile;
    HANDLE	m_hFileMapping;
    LPVOID	m_lpFileBase;

    PIMAGE_DOS_HEADER m_pDosHdr;
	PIMAGE_NT_HEADERS m_pNtHdr;

	PIMAGE_FILE_HEADER		m_pImgHdr;
	PIMAGE_SECTION_HEADER	m_pFirstSection;

	bool	Open(LPCTSTR lpFileName,bool bWrite=false);
	void	Close();
	int				GetSectionCount() { return m_pImgHdr->NumberOfSections;};
	CPEImageFile();
	virtual ~CPEImageFile();	
};

#endif // !defined(AFX_PEIMAGEFILE_H__C1AD1BA4_0B86_4C4D_9A0F_88744A5EA5C7__INCLUDED_)