LoadPE之PE editor模拟

  1 void CPEInfoDlg::OnDropFiles(HDROP hDropInfo) 

  2 {

  3     // TODO: Add your message handler code here and/or call default

  4     TCHAR tzFileName[255]={0};

  5     int FileCount = DragQueryFile(hDropInfo,0xFFFFFFFF,NULL,0);

  6     if (FileCount > 1)

  7     {

  8         MessageBox("暂时只支持分析一个文件，你传入了多个文件，将分析第一个","",0);

  9     }

 10 

 11     if (DragQueryFile(hDropInfo,0,tzFileName,sizeof(tzFileName)))

 12     {

 13         szFilePath = tzFileName;

 14     }

 15     DragFinish(hDropInfo);

 16     HANDLE hFile = ::CreateFile(szFilePath,GENERIC_READ,FILE_SHARE_READ,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);

 17     if ( hFile == INVALID_HANDLE_VALUE)

 18     {

 19         AfxMessageBox("无效的文件");

 20         return;

 21     }

 22 

 23     IMAGE_DOS_HEADER dosHeader;

 24     IMAGE_NT_HEADERS32 ntHeader;

 25 

 26     BOOL bValid = FALSE;   //是不是PE文件

 27     DWORD dwReadLen;

 28     ::ReadFile(hFile,&dosHeader,sizeof(dosHeader),&dwReadLen,NULL);

 29     if (dwReadLen == sizeof(dosHeader))

 30     {

 31         if (dosHeader.e_magic == IMAGE_DOS_SIGNATURE)

 32         {

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

 34             {

 35                 ::ReadFile(hFile,&ntHeader,sizeof(ntHeader),&dwReadLen,NULL);

 36                 if (ntHeader.Signature == IMAGE_NT_SIGNATURE)

 37                 {

 38                     bValid = TRUE;  //一个合法的PE文件

 39                 }

 40             }

 41         }

 42     }

 43     if (bValid ==FALSE)

 44     {

 45         AfxMessageBox("你拖拽了一个非PE文件");

 46         return;

 47     }

 48     AfxGetMainWnd()->SetWindowText("[ PE editor ]"+szFilePath);

 49     CString szTemp = "";

 50     szTemp.Format("%08x",ntHeader.OptionalHeader.AddressOfEntryPoint);

 51     szTemp.MakeUpper();

 52     SetDlgItemText(IDC_EntryPoint,szTemp);

 53 

 54     szTemp.Format("%08x",ntHeader.OptionalHeader.ImageBase);

 55     szTemp.MakeUpper();

 56     SetDlgItemText(IDC_ImageBase,szTemp);

 57     szTemp.Format("%08x",ntHeader.OptionalHeader.SizeOfImage);

 58     szTemp.MakeUpper();

 59 

 60     SetDlgItemText(IDC_SizeOfImage,szTemp);

 61     szTemp.Format("%08x",ntHeader.OptionalHeader.BaseOfCode);

 62     szTemp.MakeUpper();

 63 

 64     SetDlgItemText(IDC_BaseOfCode,szTemp);

 65     szTemp.Format("%08x",ntHeader.OptionalHeader.BaseOfData);

 66     szTemp.MakeUpper();

 67 

 68     SetDlgItemText(IDC_BaseOfData,szTemp);

 69     szTemp.Format("%08x",ntHeader.OptionalHeader.SectionAlignment);

 70     szTemp.MakeUpper();

 71 

 72     SetDlgItemText(IDC_SectionAlignment,szTemp);

 73     szTemp.Format("%08x",ntHeader.OptionalHeader.FileAlignment);

 74     szTemp.MakeUpper();

 75 

 76     SetDlgItemText(IDC_FileAlignment,szTemp);

 77     szTemp.Format("%04x",ntHeader.OptionalHeader.Magic);

 78     szTemp.MakeUpper();

 79 

 80     SetDlgItemText(IDC_Magic,szTemp);

 81     szTemp.Format("%04x",ntHeader.OptionalHeader.Subsystem);

 82     szTemp.MakeUpper();

 83 

 84     SetDlgItemText(IDC_Subsystem,szTemp);

 85     szTemp.Format("%04x",ntHeader.FileHeader.NumberOfSections);

 86     szTemp.MakeUpper();

 87 

 88     SetDlgItemText(IDC_NumberOfSection,szTemp);

 89 

 90     szTemp.Format("%08x",ntHeader.FileHeader.TimeDateStamp);

 91     szTemp.MakeUpper();

 92 

 93     SetDlgItemText(IDC_TimeDataStamp,szTemp);

 94     szTemp.Format("%08x",ntHeader.OptionalHeader.SizeOfHeaders);

 95     szTemp.MakeUpper();

 96 

 97     SetDlgItemText(IDC_SizeOfHeader,szTemp);

 98     szTemp.Format("%04x",ntHeader.FileHeader.Characteristics);

 99     szTemp.MakeUpper();

100 

101     SetDlgItemText(IDC_Characteristics,szTemp);

102     szTemp.Format("%08x",ntHeader.OptionalHeader.CheckSum);

103     szTemp.MakeUpper();

104 

105     SetDlgItemText(IDC_CheckSum,szTemp);

106     szTemp.Format("%04x",ntHeader.FileHeader.SizeOfOptionalHeader);

107     szTemp.MakeUpper();

108 

109     SetDlgItemText(IDC_SizeOfOptionHeader,szTemp);

110     szTemp.Format("%08x",ntHeader.FileHeader.NumberOfSections);

111     szTemp.MakeUpper();

112 

113     SetDlgItemText(IDC_NumberOfSection,szTemp);

114     szTemp.Format("%08x",ntHeader.OptionalHeader.NumberOfRvaAndSizes);

115     szTemp.MakeUpper();

116 

117     SetDlgItemText(IDC_NumberOfRvaAndSize,szTemp);

118 

119     CDialog::OnDropFiles(hDropInfo);

120 }

直接读取文件，没有用到文件映射。

下次试试读取输入表和导出表。