///
// ValidPE.h文件




#include <afxwin.h>




class CMyApp : public CWinApp
{
public:
virtual BOOL InitInstance();
};

///

// ValidPE.cpp文件




#include <afxdlgs.h>


#include "ValidPE.h"


CMyApp theApp;


BOOL CMyApp::InitInstance()
{
// 彈出選色文件對(duì)話框
CFileDialog dlg(TRUE);
if(dlg.DoModal() != IDOK)
return FALSE;




// 打開檢查的文件
HANDLE hFile = ::CreateFile(dlg.GetFileName(), GENERIC_READ,?
FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if(hFile == INVALID_HANDLE_VALUE)
{
MessageBox(NULL, "無效文件！", "ValidPE", MB_OK);
}


// 定義PE文件中的DOS頭和NT頭
IMAGE_DOS_HEADER dosHeader;
IMAGE_NT_HEADERS32 ntHeader;

// 驗(yàn)證過程
BOOL bValid = FALSE;
DWORD dwRead;
// 讀取DOS頭
::ReadFile(hFile, &dosHeader, sizeof(dosHeader), &dwRead, NULL);
if(dwRead == sizeof(dosHeader))
{
if(dosHeader.e_magic == IMAGE_DOS_SIGNATURE) // 是不是有效的DOS頭？
{
// 定位NT頭
if(::SetFilePointer(hFile, dosHeader.e_lfanew, NULL, FILE_BEGIN) != -1)
{
// 讀取NT頭
::ReadFile(hFile, &ntHeader, sizeof(ntHeader), &dwRead, NULL);
if(dwRead == sizeof(ntHeader))
{
if(ntHeader.Signature == IMAGE_NT_SIGNATURE) // 是不是有效的NT頭
bValid = TRUE;
}
}
}
}


// 顯示結(jié)果
if(bValid)
MessageBox(NULL, "是一個(gè)PE格式的文件！", "ValidPE", MB_OK);
else
MessageBox(NULL, "不是一個(gè)PE格式的文件！", "ValidPE", MB_OK);


::CloseHandle(hFile);
return FALSE;
}

總結(jié)

以上是生活随笔為你收集整理的第八章 PE文件 windows程序设计 王艳平版的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。