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 }
直接读取文件,没有用到文件映射。
下次试试读取输入表和导出表。