C++判斷pe文件實例

更新時間：2014年10月14日 11:46:49 投稿：shichen2014

這篇文章主要介紹了C++判斷pe文件的方法,包含了文件操作的具體實現(xiàn)方法,具有很好的參考借鑒價值,需要的朋友可以參考下

本文實例講述了C++判斷pe文件的方法。分享給大家供大家參考。

具體實現(xiàn)方法如下：

#include <afxdlgs.h>是為了使用CFileDialog
main.h如下：

復制代碼代碼如下:

#include <afxwin.h>  

class CMyApp:public CWinApp  

{  

    BOOL InitInstance();  

};

main.cpp如下：

復制代碼代碼如下:

#include "main.h"  

#include <afxdlgs.h>  

CMyApp theApp;  

BOOL CMyApp::InitInstance()  

{  

    CFileDialog dlg(TRUE);  

    if (dlg.DoModal() != IDOK)  

    {  

        return FALSE;  

    }  

    //打開文件  

    HANDLE hFile = ::CreateFile(dlg.GetPathName(), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);  

    if (INVALID_HANDLE_VALUE == hFile)  

    {  

        return FALSE;  

    }  

    IMAGE_DOS_HEADER dosHeader;  

    IMAGE_NT_HEADERS32 ntHeader;  

    DWORD dwReturn;  

    BOOL bIsPe=FALSE;  

    ::ReadFile(hFile, &dosHeader, sizeof(IMAGE_DOS_HEADER), &dwReturn, NULL);  

    if (dwReturn == sizeof(IMAGE_DOS_HEADER))  

    {  

        if (dosHeader.e_magic == IMAGE_DOS_SIGNATURE) //mz  

        {  

            if (-1 != ::SetFilePointer(hFile, dosHeader.e_lfanew, NULL, FILE_BEGIN))  

            {  

                ::ReadFile(hFile, &ntHeader, sizeof(IMAGE_NT_HEADERS32), &dwReturn, NULL);  

                if (dwReturn == sizeof(IMAGE_NT_HEADERS32))  

                {  

                    if (ntHeader.Signature == IMAGE_NT_SIGNATURE) //pe  

                    {  

                        bIsPe = TRUE;  

                    }  

                }  

            }  

        }  

    }  

    ::CloseHandle(hFile); //與上面的CreateFile區(qū)對寫  

    if (bIsPe)  

    {  

        MessageBoxA(NULL, "is a pe file", "", MB_OK);  

    }  

    else  

    {  

        MessageBoxA(NULL, "not a pe file", "", MB_OK);  

    }  

    return FALSE;  

}