判断是不是PE文件

判断是不是一个PE文件有很多种方法，我们用的方法是：

先读取Dos头，判断e_magic是否等于"MZ"，然后再读取PE文件头的头字节，判断是不是 "PE00"。这样就能确定是不是一个有效的PE文件。代码如下：

 

//选择文件
void CPEDlg::OnButton1()
{
// TODO: Add your control notification handler code here
TCHAR tzFilter[] = _T("可执行文件（*.exe）|*.exe|所有文件（*.*）|*.*||");
CFileDialog dlgFile(TRUE,NULL,NULL, OFN_HIDEREADONLY,tzFilter,this);
if (IDOK == dlgFile.DoModal())
{
  m_strFilePath = dlgFile.GetPathName();
  UpdateData(FALSE);
}
}

//判断是否是PE文件
void CPEDlg::OnButton2()
{
// TODO: Add your control notification handler code here

HANDLE    hFile = INVALID_HANDLE_VALUE; //打开的PE文件句柄
IMAGE_DOS_HEADER DosHeader = {0};
DWORD    dwReadLen = 0;
DWORD    dwNTSignature = 0;

//创建文件
hFile = CreateFile(m_strFilePath,GENERIC_READ,FILE_SHARE_READ,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
  MessageBox("打开文件失败");
  return;
}

//读取文件
if (!ReadFile(hFile,&DosHeader,sizeof(DosHeader),&dwReadLen,NULL))
{
  CloseHandle(hFile);
  MessageBox("读取文件失败");
  return;
}
//判断DOS头部是否为MZ
if (DosHeader.e_magic != IMAGE_DOS_SIGNATURE)
{
  CloseHandle(hFile);
  MessageBox("不是PE文件");
  return;
}

SetFilePointer(hFile,DosHeader.e_lfanew,NULL,FILE_BEGIN);
//读取PE的NT头字节
ReadFile(hFile,&dwNTSignature,sizeof(dwNTSignature),&dwReadLen,NULL);
//判断PE标志
if (dwNTSignature != IMAGE_NT_SIGNATURE)
{
  MessageBox("不是PE文件");
  CloseHandle(hFile);
  return;
}

CloseHandle(hFile);
hFile = INVALID_HANDLE_VALUE;

MessageBox("是PE文件");
}