xiaoguozi's Blog
Pay it forword - 我并不觉的自豪,我所尝试的事情都失败了······习惯原本生活的人不容易改变,就算现状很糟,他们也很难改变,在过程中,他们还是放弃了······他们一放弃,大家就都是输家······让爱传出去,很困难,也无法预料,人们需要更细心的观察别人,要随时注意才能保护别人,因为他们未必知道自己要什么·····
#include "windows.h"
#include 
<iostream>

using namespace std;

#define NTSIGNATURE(a) ((LPVOID)((BYTE *)a + \
                       ((PIMAGE_DOS_HEADER)a)
->e_lfanew))
/*
const int SIZE_OF_NT_SIGNATURE = 4;
#define PEHEADOFFSET(a) ((LPVOID)((BYTE *)a + \
                       ((PIMAGE_DOS_HEADER)a)->e_lfanew)+SIZE_OF_NT_SIGNATURE)
*/
class PEUtil{
public:
    
static bool IsPeFile(LPVOID lp){
        
//DWORD ImageType=ImageFileType(lp);
        
//if(ImageType!=
        /*if(lp==NULL)return false;
        PIMAGE_DOS_HEADER pImage_Dos_Header=(PIMAGE_DOS_HEADER)lp;
        if(pImage_Dos_Header->e_magic!=IMAGE_DOS_SIGNATURE)return false;

        PIMAGE_NT_HEADERS32 pImage_Nt_Header=GetNtHeader(lp);
        if(pImage_Nt_Header->Signature!=IMAGE_NT_SIGNATURE)return false;

        return true;
*/

        
return ImageFileType(lp)==IMAGE_NT_SIGNATURE;
    }

    
static PIMAGE_NT_HEADERS GetNtHeader(PBYTE lp){
        
if(IsPeFile(lp)==false)return NULL;

        
return (PIMAGE_NT_HEADERS32)NTSIGNATURE(lp);
    }
    
static PIMAGE_FILE_HEADER GetFileHeader(PBYTE lp){
        PIMAGE_NT_HEADERS pNtHeader
=GetNtHeader(lp);
        
if(pNtHeader==NULL)return NULL;

        
return PIMAGE_FILE_HEADER(&pNtHeader->FileHeader);
    }
    
static PIMAGE_OPTIONAL_HEADER GetOptionalHeader(PBYTE lp){
        PIMAGE_NT_HEADERS pNtHeader
=GetNtHeader(lp);
        
if(pNtHeader==NULL)return NULL;

        
return PIMAGE_OPTIONAL_HEADER(&pNtHeader->OptionalHeader);
    }
    
static PIMAGE_SECTION_HEADER GetSectionHeader(PBYTE lp){
        PIMAGE_NT_HEADERS pnh
=GetNtHeader(lp);
        PIMAGE_SECTION_HEADER pch
=(PIMAGE_SECTION_HEADER)((PBYTE)pnh+sizeof(IMAGE_NT_HEADERS));
        
return pch;
    }
    
static LPVOID IMAGE_DIRECTORY_OFFSET(PBYTE lp,DWORD index){
        PIMAGE_FILE_HEADER pfh
=GetFileHeader(lp);
        PIMAGE_OPTIONAL_HEADER poh
=GetOptionalHeader(lp);
        PIMAGE_SECTION_HEADER psh
=GetSectionHeader(lp);

        
int nSections=pfh->NumberOfSections;
        
if(index>=poh->NumberOfRvaAndSizes){
            
return NULL;
        }

        LPVOID virtualDirAddr
=(LPVOID)poh->DataDirectory[index].VirtualAddress;

        
int i=0;
        
while(i<nSections){
            
if(psh->VirtualAddress<=DWORD(virtualDirAddr)&&
                psh
->VirtualAddress+psh->SizeOfRawData>DWORD(virtualDirAddr)){
                    
break;
            }
            
++psh;
            
++i;
        }
        
if(i>=nSections)return NULL;

        
return (LPVOID)(((int)lp+(int)virtualDirAddr-psh->VirtualAddress) +
    (
int)psh->PointerToRawData);
        
//return NULL;
    }
private:
    
static DWORD WINAPI ImageFileType (LPVOID lpFile)
    {
      
/* 首先出现的是DOS文件标志 */
      
if (*(USHORT *)lpFile == IMAGE_DOS_SIGNATURE)
      {
        
/* 由DOS头部决定PE文件头部的位置 */
        
if (LOWORD (*(DWORD *)NTSIGNATURE (lpFile)) ==
            IMAGE_OS2_SIGNATURE 
||
            LOWORD (
*(DWORD *)NTSIGNATURE (lpFile)) ==
            IMAGE_OS2_SIGNATURE_LE)
          
return (DWORD)LOWORD(*(DWORD *)NTSIGNATURE (lpFile));
        
else if (*(DWORD *)NTSIGNATURE (lpFile) ==
          IMAGE_NT_SIGNATURE)
        
return IMAGE_NT_SIGNATURE;
        
else
          
return IMAGE_DOS_SIGNATURE;
      }
      
else
        
/* 不明文件种类 */
        
return 0;
    }
};
int _tmain(int argc, _TCHAR* argv[])
{
    LPCWSTR filepath
=TEXT("D://STLPort/MemoryMap.exe");
    HANDLE hFile 
= CreateFile(filepath,GENERIC_READ|GENERIC_WRITE,
        FILE_SHARE_READ,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);

    
    
if(hFile==INVALID_HANDLE_VALUE){
        cout
<<"CreateFile Error"<<endl;
        
return -1;
    }

    
//IMAGE_NT_HEADERS32
    HANDLE hFileMapping = CreateFileMapping(hFile,NULL,PAGE_READWRITE,0,0,L"testFileMapping");
    PBYTE pByte
=(PBYTE)MapViewOfFile(hFileMapping,FILE_MAP_ALL_ACCESS,0,0,0);

    
//cout<<hex<<PEUtil::GetOptionalHeader(pByte)->ImageBase<<endl;
    int nSections=PEUtil::GetFileHeader(pByte)->NumberOfSections;
    cout
<<nSections<<endl;
    PIMAGE_SECTION_HEADER psh
=PEUtil::GetSectionHeader(pByte);

    
for(int i=0;i<nSections;i++){
        cout
<<psh->Name<<" "<<psh->VirtualAddress<<" "<<psh->PointerToRawData<<endl;
        psh
++;
    }
    
/*cout<<IsPeFile(pByte);
    PIMAGE_NT_HEADERS32 pImage=GetNtHeader(pByte);
    cout<<hex<<pImage->Signature<<endl;
    cout<<pImage->FileHeader.Machine<<endl;
    //cout<<hex<<ImageFileType(pByte)<<endl;
    //cout<<PEHEADOFFSET(pByte)
    //cout<<pByte<<endl;
*/
    
return 0;
}
有时间总结,待续...

#include "windows.h"
#include 
<iostream>
//#include "dlltest.h"
#include <vector>
#include 
<string>

using namespace std;
//#pragma comment(lib,"dlltest.lib")
typedef int (*pf)(void);
#define NTSIGNATURE(a) ((LPVOID)((BYTE *)a + \
                       ((PIMAGE_DOS_HEADER)a)
->e_lfanew))
/*
const int SIZE_OF_NT_SIGNATURE = 4;
#define PEHEADOFFSET(a) ((LPVOID)((BYTE *)a + \
                       ((PIMAGE_DOS_HEADER)a)->e_lfanew)+SIZE_OF_NT_SIGNATURE)
*/

typedef 
struct tagImportDirectory
{
  DWORD dwRVAFunctionNameList;
  DWORD dwUseless1;
  DWORD dwUseless2;
  DWORD dwRVAModuleName;
  DWORD dwRVAFunctionAddressList;
} IMAGE_IMPORT_MODULE_DIRECTORY, 
*PIMAGE_IMPORT_MODULE_DIRECTORY;


class PEUtil{
public:
    
static bool IsPeFile(LPVOID lp){
        
//DWORD ImageType=ImageFileType(lp);
        
//if(ImageType!=
        /*if(lp==NULL)return false;
        PIMAGE_DOS_HEADER pImage_Dos_Header=(PIMAGE_DOS_HEADER)lp;
        if(pImage_Dos_Header->e_magic!=IMAGE_DOS_SIGNATURE)return false;

        PIMAGE_NT_HEADERS32 pImage_Nt_Header=GetNtHeader(lp);
        if(pImage_Nt_Header->Signature!=IMAGE_NT_SIGNATURE)return false;

        return true;
*/

        
return ImageFileType(lp)==IMAGE_NT_SIGNATURE;
    }

    
static PIMAGE_NT_HEADERS GetNtHeader(PBYTE lp){
        
if(IsPeFile(lp)==false)return NULL;

        
return (PIMAGE_NT_HEADERS32)NTSIGNATURE(lp);
    }
    
static PIMAGE_FILE_HEADER GetFileHeader(PBYTE lp){
        PIMAGE_NT_HEADERS pNtHeader
=GetNtHeader(lp);
        
if(pNtHeader==NULL)return NULL;

        
return PIMAGE_FILE_HEADER(&pNtHeader->FileHeader);
    }
    
static PIMAGE_OPTIONAL_HEADER GetOptionalHeader(PBYTE lp){
        PIMAGE_NT_HEADERS pNtHeader
=GetNtHeader(lp);
        
if(pNtHeader==NULL)return NULL;

        
return PIMAGE_OPTIONAL_HEADER(&pNtHeader->OptionalHeader);
    }
    
static PIMAGE_SECTION_HEADER GetSectionHeader(PBYTE lp){
        PIMAGE_NT_HEADERS pnh
=GetNtHeader(lp);
        PIMAGE_SECTION_HEADER pch
=(PIMAGE_SECTION_HEADER)((PBYTE)pnh+sizeof(IMAGE_NT_HEADERS));
        
return pch;
    }
    
static PIMAGE_SECTION_HEADER GetSectionHeaderByName(PBYTE lp,LPCSTR name){
        
int nSections=GetFileHeader(lp)->NumberOfSections;
        PIMAGE_SECTION_HEADER psh
=GetSectionHeader(lp);

        
for(int i=0;i<nSections;i++){
            
//cout<<psh->Name<<" "<<psh->VirtualAddress<<" "<<psh->PointerToRawData<<endl;
            if(strcmp((char*)psh->Name,name)==0)
                
return psh;
            psh
++;
        }
        
return NULL;
    }
    
static LPVOID IMAGE_DIRECTORY_OFFSET(PBYTE lp,DWORD index){
        PIMAGE_FILE_HEADER pfh
=GetFileHeader(lp);
        PIMAGE_OPTIONAL_HEADER poh
=GetOptionalHeader(lp);
        PIMAGE_SECTION_HEADER psh
=GetSectionHeader(lp);

        
int nSections=pfh->NumberOfSections;
        
if(index>=poh->NumberOfRvaAndSizes){
            
return NULL;
        }

        LPVOID virtualDirAddr
=(LPVOID)poh->DataDirectory[index].VirtualAddress;

        
int i=0;
        
while(i<nSections){
            
if(psh->VirtualAddress<=DWORD(virtualDirAddr)&&
                psh
->VirtualAddress+psh->SizeOfRawData>DWORD(virtualDirAddr)){
                    
break;
            }
            
++psh;
            
++i;
        }
        
if(i>=nSections)return NULL;

        
return (LPVOID)(((int)lp+(int)psh->PointerToRawData
            
+(int)virtualDirAddr-psh->VirtualAddress));
        
//return NULL;
    }
    
static LPVOID IAT_OFFSET(PBYTE lp){
        
/*PIMAGE_OPTIONAL_HEADER poh=GetOptionalHeader(lp);
        DWORD virtualDirAddr=poh->DataDirectory[12].VirtualAddress;
        PIMAGE_IMPORT_MODULE_DIRECTORY pimd;

        pimd=(PIMAGE_IMPORT_MODULE_DIRECTORY)IMAGE_DIRECTORY_OFFSET(lp,IMAGE_DIRECTORY_ENTRY_IMPORT);

        PIMAGE_SECTION_HEADER idsh;
        idsh=GetSectionHeaderByName(lp,".idata");
        return (LPVOID)((PBYTE)pimd+(virtualDirAddr-idsh->VirtualAddress));
*/
        
return (LPVOID)((PBYTE)IMAGE_DIRECTORY_OFFSET(lp,12)/*-lp*/);
    }
    
static void GetImportModuleNames(PBYTE lp,vector<PBYTE>& vec){
        PIMAGE_IMPORT_MODULE_DIRECTORY pimd;
        PIMAGE_SECTION_HEADER idsh;

        
//pimd 物理地址
        pimd=(PIMAGE_IMPORT_MODULE_DIRECTORY)IMAGE_DIRECTORY_OFFSET(lp,IMAGE_DIRECTORY_ENTRY_IMPORT);
        idsh
=GetSectionHeaderByName(lp,".idata");
        
//cout<<idsh->Name<<endl;
        PBYTE pData=(PBYTE)pimd;

        
//pid->
        
//pid->Name
        
//ImageRvaToVa(

        
while(pimd->dwRVAModuleName){
            
//vec.push_back((PBYTE)(pimd+(pimd->dwRVAModuleName-idsh->VirtualAddress)));
            vec.push_back((pData+(pimd->dwRVAModuleName-idsh->VirtualAddress)));
            
++pimd;
        }
    }
    
static PIMAGE_IMPORT_MODULE_DIRECTORY GetImportModuleByName(PBYTE lp,LPCSTR name){
        PIMAGE_IMPORT_MODULE_DIRECTORY pimd;
        PIMAGE_SECTION_HEADER idsh;
        pimd
=(PIMAGE_IMPORT_MODULE_DIRECTORY)IMAGE_DIRECTORY_OFFSET(lp,IMAGE_DIRECTORY_ENTRY_IMPORT);
        idsh
=GetSectionHeaderByName(lp,".idata");
        
//cout<<idsh->Name<<endl;

        
if(idsh==NULL)return NULL;

        PBYTE pData
=(PBYTE)pimd;
        
while(pimd->dwRVAModuleName){
            
//vec.push_back((PBYTE)(pimd+(pimd->dwRVAModuleName-idsh->VirtualAddress)));
            
//vec.push_back((pData+(pimd->dwRVAModuleName-idsh->VirtualAddress)));
            if(strcmp((char*)(pData+(pimd->dwRVAModuleName-idsh->VirtualAddress)),name)==0){
                
return pimd;
            }
            
++pimd;
        }

        
return NULL;
    }
    
static void GetImportFunctionNamesByModule(PBYTE lp,vector<PBYTE>& ret){
        PIMAGE_IMPORT_MODULE_DIRECTORY pimd;
        PIMAGE_SECTION_HEADER idsh;
        pimd
=(PIMAGE_IMPORT_MODULE_DIRECTORY)IMAGE_DIRECTORY_OFFSET(lp,IMAGE_DIRECTORY_ENTRY_IMPORT);
        idsh
=GetSectionHeaderByName(lp,".idata");

        DWORD dwBase
=(DWORD)((PBYTE)pimd - idsh->VirtualAddress);
    
//    DWORD dwBaseAddr=(DWORD)(lp - idsh->VirtualAddress);

        LPVOID iat
=IAT_OFFSET(lp);

        PIMAGE_IMPORT_DESCRIPTOR pid;
        pid
=(PIMAGE_IMPORT_DESCRIPTOR)IMAGE_DIRECTORY_OFFSET(lp,IMAGE_DIRECTORY_ENTRY_IMPORT);
        
        
while(pimd->dwRVAModuleName){

            cout
<<"------------"<<(char*)(dwBase+pimd->dwRVAModuleName)<<"-----------"<<endl;

            DWORD dwFunction
=pimd->dwRVAFunctionNameList;
            DWORD dwFunctionAddr
=pimd->dwRVAFunctionAddressList;

            
while(dwFunction && *(DWORD*)(dwFunction+dwBase) && *(char*)(*(DWORD*)(dwFunction+dwBase)+dwBase+2)){
                cout
<<(char*)(*(DWORD*)(dwFunction+dwBase)+dwBase+2)<<" ---rva:   ";
                cout
<<hex<<(dwFunctionAddr)<<endl;
                dwFunction
+=4;
                dwFunctionAddr
+=4;
            }

            
++pimd;
        }
        
//vector<PBYTE> moduleSet;
        
//GetImportModuleNames(lp,moduleSet);

    }
private:
    
static DWORD WINAPI ImageFileType (LPVOID lpFile)
    {
      
/* 首先出现的是DOS文件标志 */
      
if (*(USHORT *)lpFile == IMAGE_DOS_SIGNATURE)
      {
        
/* 由DOS头部决定PE文件头部的位置 */
        
if (LOWORD (*(DWORD *)NTSIGNATURE (lpFile)) ==
            IMAGE_OS2_SIGNATURE 
||
            LOWORD (
*(DWORD *)NTSIGNATURE (lpFile)) ==
            IMAGE_OS2_SIGNATURE_LE)
          
return (DWORD)LOWORD(*(DWORD *)NTSIGNATURE (lpFile));
        
else if (*(DWORD *)NTSIGNATURE (lpFile) ==
          IMAGE_NT_SIGNATURE)
        
return IMAGE_NT_SIGNATURE;
        
else
          
return IMAGE_DOS_SIGNATURE;
      }
      
else
        
/* 不明文件种类 */
        
return 0;
    }
};
int _tmain(int argc, _TCHAR* argv[])
{
    LPCWSTR filepath
=TEXT("D://STLPort/MemoryMap.exe");
    HANDLE hFile 
= CreateFile(filepath,GENERIC_READ|GENERIC_WRITE,
        FILE_SHARE_READ,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);

    
if(hFile==INVALID_HANDLE_VALUE){
        cout
<<"CreateFile Error"<<endl;
        
return -1;
    }

    
//IMAGE_NT_HEADERS32
    HANDLE hFileMapping = CreateFileMapping(hFile,NULL,PAGE_READWRITE,0,0,L"testFileMapping");
    PBYTE pByte
=(PBYTE)MapViewOfFile(hFileMapping,FILE_MAP_ALL_ACCESS,0,0,0);


    cout
<<PEUtil::IsPeFile(pByte)<<endl;

    
//LPVOID p=PEUtil::IAT_OFFSET(pByte);
    
//cout<<fndlltest()<<endl;
    HINSTANCE hInst=LoadLibrary(L"dlltest.dll");
    pf lp;
    lp
=(pf)GetProcAddress(hInst,"fndlltest");
    cout
<<lp()<<endl;



    
//cout<<hex<<PEUtil::GetOptionalHeader(pByte)->ImageBase<<endl;
    /*int nSections=PEUtil::GetFileHeader(pByte)->NumberOfSections;
    cout<<nSections<<endl;
    PIMAGE_SECTION_HEADER psh=PEUtil::GetSectionHeader(pByte);

    for(int i=0;i<nSections;i++){
        cout<<psh->Name<<" "<<psh->VirtualAddress<<" "<<psh->PointerToRawData<<endl;
        psh++;
    }
*/
    PIMAGE_SECTION_HEADER psh
=PEUtil::GetSectionHeaderByName(pByte,".idata");
    cout
<<psh->Name<<" "<<psh->VirtualAddress<<" "<<psh->PointerToRawData<<endl;

    vector
<PBYTE> vec;
    
/*PEUtil::GetImportModuleNames(pByte,vec);
    for(int i=0;i<vec.size();i++){
        cout<<vec[i]<<endl;
    }
*/
    PEUtil::GetImportFunctionNamesByModule(pByte,vec);
    
/*cout<<IsPeFile(pByte);
    PIMAGE_NT_HEADERS32 pImage=GetNtHeader(pByte);
    cout<<hex<<pImage->Signature<<endl;
    cout<<pImage->FileHeader.Machine<<endl;
    //cout<<hex<<ImageFileType(pByte)<<endl;
    //cout<<PEHEADOFFSET(pByte)
    //cout<<pByte<<endl;
*/
    
return 0;
}

posted on 2010-10-12 17:22 小果子 阅读(703) 评论(0)  编辑 收藏 引用 所属分类: WindowsC++

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理