微信公众号:计算机与网络安全
ID:Computer-network
写PE查看器并不是件复杂的事情,只要按照PE结构一步一步地解析就可以了。下面简单地解析其中几个字段内容,显示一下节表的信息,其余的内容只要稍作修改即可。PE查看器的界面如图1所示。
图1 PE查看器解析记事本程序
PE查看器的界面按照图1所示的设置,不过这个可以按照个人的偏好进行布局设置。编写该PE查看器的步骤为打开文件并创建文件内存映像,判断文件是否为PE文件并获得PE格式相关结构体的指针,解析基本的PE字段,枚举节表,最后关闭文件。需要在类中添加几个成员变量及成员函数,添加的内容如图2所示。
图2 在类中添加的成员变量及成员函数
按照前面所说的顺序,依次实现添加的各个成员函数。
- BOOL CPeParseDlg::FileCreate(char *szFileName)
- {
- BOOL bRet = FALSE;
- m_hFile = CreateFile(szFileName,
- GENERIC_READ | GENERIC_WRITE,
- FILE_SHARE_READ,NULL,OPEN_EXISTING,
- FILE_ATTRIBUTE_NORMAL,NULL);
- if ( m_hFile == INVALID_HANDLE_VALUE )
- {
- return bRet;
- }
- m_hMap = CreateFileMapping(m_hFile, NULL,
- PAGE_READWRITE | SEC_IMAGE,0, 0, 0);
- if ( m_hMap == NULL )
- {
- CloseHandle(m_hFile);
- return bRet;
- }
- m_lpBase = MapViewOfFile(m_hMap,
- FILE_MAP_READ | FILE_SHARE_WRITE,
- 0, 0, 0);
- if ( m_lpBase == NULL )
- {
- CloseHandle(m_hMap);
- CloseHandle(m_hFile);
- return bRet;
- }
- bRet = TRUE;
- return bRet;
- }
这个函数的主要功能是打开文件并创建内存文件映像。通常对文件进行连续读写时直接使用ReadFile()和WriteFile()两个函数。当不连续操作文件时,每次在ReadFile()或者WriteFile()后就要使用SetFilePointer()来调整文件指针的位置,这样的操作较为繁琐。内存文件映像的作用是把整个文件映射入进程的虚拟空间中,这样操作文件就像操作内存变量或内存数据一样方便。
创建内存文件映像所使用的函数有两个,分别是CreateFileMapping()和MapViewOfFile()。CreateFileMapping()函数的定义如下:
- HANDLE CreateFileMapping(
- HANDLE hFile,
- LPSECURITY_ATTRIBUTES lpAttributes,
- DWORD flProtect,
- DWORD dwMaximumSizeHigh,
- DWORD dwMaximumSizeLow,
- LPCTSTR lpName
- );
参数说明如下。
hFile:该参数是 CreateFile()函数返回的句柄。
lpAttributes:是安全属性,该值通常是 NULL。
flProtect:创建文件映射后的属性,通常设置为可读可写 PAGE_READWRITE。如果需要像装载可执行文件那样把文件映射入内存的话,那么需要使用 SEC_IMAGE。最后3个参数在这里为0。如果创建的映射需要在多进程中共享数据的话,那么最后一个参数设定为一个字符串,以便通过该名称找到该块共享内存。
该函数的返回值为一个内存映射的句柄。
MapViewOfFile()函数的定义如下:
- LPVOID MapViewOfFile(
- HANDLE hFileMappingObject,
- DWORD dwDesiredAccess,
- DWORD dwFileOffsetHigh,
- DWORD dwFileOffsetLow,
- SIZE_T dwNumberOfBytesToMap
- );
参数说明如下。
hFileMappingObject:该参数为 CreateFileMapping()返回的句柄。
dwDesiredAccess:想获得的访问权限,通常情况下也是可读可写 FILE_MAP_READ、FILE_MAP_WRITE。
最后3个参数一般给0值就可以了。
按照编程的规矩,打开要关闭,申请要释放。CreateFileMapping()的关闭需要使用CloseHandle()函数。MapViewOfFile()的关闭,要使用UnmapViewOfFile()函数,该函数的定义如下:
- BOOL UnmapViewOfFile(
- LPCVOID lpBaseAddress
- );
该函数的参数就是MapViewOfFile()函数的返回值。
接着说PE查看器,文件已经打开,就要判断文件是否为有效的PE文件了。如果是有效的PE文件,就把解析PE格式的相关结构体的指针也得到。代码如下:
- BOOL CPeParseDlg::IsPeFileAndGetPEPointer()
- {
- BOOL bRet = FALSE;
- // 判断是否为 MZ 头
- m_pDosHdr = (PIMAGE_DOS_HEADER)m_lpBase;
- if ( m_pDosHdr->e_magic != IMAGE_DOS_SIGNATURE )
- {
- return bRet;
- }
- // 根据 IMAGE_DOS_HEADER 的 e_lfanew 的值得到 PE 头的位置
- m_pNtHdr = (PIMAGE_NT_HEADERS)((DWORD)m_lpBase + m_pDosHdr->e_lfanew);
- // 判断是否为 PE\0\0
- if ( m_pNtHdr->Signature != IMAGE_NT_SIGNATURE )
- {
- return bRet;
- }
- // 获得节表的位置
- m_pSecHdr = (PIMAGE_SECTION_HEADER)((DWORD)&(m_pNtHdr->OptionalHeader)
- + m_pNtHdr->FileHeader.SizeOfOptionalHeader);
- bRet = TRUE;
- return bRet;
- }
这段代码应该非常容易理解,继续看解析PE格式的部分。
- VOID CPeParseDlg::ParseBasePe()
- {
- CString StrTmp;
- // 入口地址
- StrTmp.Format("%08X", m_pNtHdr->OptionalHeader.AddressOfEntryPoint);
- SetDlgItemText(IDC_EDIT_EP, StrTmp);
- // 映像基地址
- StrTmp.Format("%08X", m_pNtHdr->OptionalHeader.ImageBase);
- SetDlgItemText(IDC_EDIT_IMAGEBASE, StrTmp);
- // 连接器版本号
- StrTmp.Format("%d.%d",
- m_pNtHdr->OptionalHeader.MajorLinkerVersion,
- m_pNtHdr->OptionalHeader.MinorLinkerVersion);
- SetDlgItemText(IDC_EDIT_LINKVERSION, StrTmp);
- // 节表数量
- StrTmp.Format("%02X", m_pNtHdr->FileHeader.NumberOfSections);
- SetDlgItemText(IDC_EDIT_SECTIONNUM, StrTmp);
- // 文件对齐值大小
- StrTmp.Format("%08X", m_pNtHdr->OptionalHeader.FileAlignment);
- SetDlgItemText(IDC_EDIT_FILEALIGN, StrTmp);
- // 内存对齐值大小
- StrTmp.Format("%08X", m_pNtHdr->OptionalHeader.SectionAlignment);
- SetDlgItemText(IDC_EDIT_SECALIGN, StrTmp);
- }
PE格式的基础信息,就是简单地获取结构体的成员变量,没有过多复杂的内容。获取导入表、导出表比获取基础信息复杂。接下来进行节表的枚举,具体代码如下:
- VOID CPeParseDlg::EnumSections()
- {
- int nSecNum = m_pNtHdr->FileHeader.NumberOfSections;
- int i = 0;
- CString StrTmp;
- for ( i = 0; i < nSecNum; i ++ )
- {
- m_SectionLIst.InsertItem(i, (const char *)m_pSecHdr[i].Name);
- StrTmp.Format("%08X", m_pSecHdr[i].VirtualAddress);
- m_SectionLIst.SetItemText(i, 1, StrTmp);
- StrTmp.Format("%08X", m_pSecHdr[i].Misc.VirtualSize);
- m_SectionLIst.SetItemText(i, 2, StrTmp);
- StrTmp.Format("%08X", m_pSecHdr[i].PointerToRawData);
- m_SectionLIst.SetItemText(i, 3, StrTmp);
- StrTmp.Format("%08X", m_pSecHdr[i].SizeOfRawData);
- m_SectionLIst.SetItemText(i, 4, StrTmp);
- StrTmp.Format("%08X", m_pSecHdr[i].Characteristics);
- m_SectionLIst.SetItemText(i, 5, StrTmp);
- }
- }
最后的动作是释放动作,因为很简单,这里就不给出代码了。将这些自定义函数通过界面上的“查看”按钮联系起来,整个PE查看器就算是写完了。