Форум программистов [Powered by Invision Power Board]

Код

#include <iostream>
#include <Windows.h>
#include "MyWinNt.h"


using namespace std;

 IMAGE_NT_HEADERS* GetHeader(LPBYTE pBase);
 IMAGE_SECTION_HEADER* GetSection(IMAGE_NT_HEADERS* pHeader, DWORD dwRVA);
 LPBYTE GetFilePointer(LPBYTE pBase, DWORD dwRVA);
 LPBYTE OpenPEFile(LPCTSTR lpszFileName);
 void ClosePEFile(LPBYTE pBase);

int main(int argc, char* argv[])
{
  LPBYTE pBase = OpenPEFile(argv[1]);
  if (pBase == NULL) 
 {
    cout << "File not found!" << endl;
    return 0;
}
  IMAGE_NT_HEADERS* pHeader = GetHeader(pBase);
  if (pHeader == NULL)
  {
    cout << "It is not a PE file!" << endl;
    return -200;
  }
  cout << "Dir\tRVA\tSize\tFilePtr" << endl;
  IMAGE_DATA_DIRECTORY* pDataDir = pHeader->OptionalHeader.DataDirectory;
  for (UINT i = 0; i < IMAGE_NUMBEROF_DIRECTORY_ENTRIES; i++, pDataDir++) 
  {
    if (pDataDir->Size != 0) {
      ULONG nFilePtr = (ULONG)(ULONG_PTR)GetFilePointer(pBase, pDataDir->VirtualAddress);
      cout << dec << i << ":\t" << hex << pDataDir->VirtualAddress << '\t'
        << pDataDir->Size << '\t' << nFilePtr << endl;
    }
  }
  ClosePEFile(pBase);
  
  return 0;
}

IMAGE_NT_HEADERS* GetHeader(LPBYTE pBase) {
  // Проверка правильности аргументов.
  if (pBase == NULL)
    return NULL;
  // Получаем указатель на заголовок DOS.
  IMAGE_DOS_HEADER* pDosHeader = (IMAGE_DOS_HEADER*)pBase;
  // Проверяем корректность указателя.
  if (IsBadReadPtr(pDosHeader, sizeof(IMAGE_DOS_HEADER)))
    return NULL;
  // Сначала проверяем сигнатуру DOS.
  if (pDosHeader->e_magic != IMAGE_DOS_SIGNATURE)
    return NULL;
  // Берем указатель на заголовок PE.
  IMAGE_NT_HEADERS* pHeader = (IMAGE_NT_HEADERS*)(pBase + pDosHeader->e_lfanew);
  // Проверяем корректность указателя.
  if (IsBadReadPtr(pHeader, sizeof(IMAGE_NT_HEADERS)))
    return NULL;
  // Наконец проверяем сигнатуру PE.
  if (pHeader->Signature != IMAGE_NT_SIGNATURE)
    return NULL;
  return pHeader;
}

IMAGE_SECTION_HEADER* GetSection(IMAGE_NT_HEADERS* pHeader, DWORD dwRVA) {
  // Проверка правильности аргументов.
  if (pHeader == NULL)
    return NULL;
  // Перебор всех заголовков секций.
  IMAGE_SECTION_HEADER* pSectHeader = IMAGE_FIRST_SECTION(pHeader);
  for (UINT i = 0; i < pHeader->FileHeader.NumberOfSections; i++, pSectHeader++) {
      const char* nn = (const char*)pSectHeader->Name;
      const char* nn2 = (const char*)".rsrc";
      if(strstr(nn, nn2))
      {
 
          cout << "Resource table" << endl;

      
                  DWORD dwResourceDirStart = pSectHeader->VirtualAddress;
          // Получаем указатели на таблицу ресурсов и все связанные с ней таблицы.
          IMAGE_RESOURCE_DIRECTORY* pResourceDir = ((IMAGE_RESOURCE_DIRECTORY*)GetFilePointer((LPBYTE)pHeader, dwResourceDirStart));
          

         
          PIMAGE_RESOURCE_DIRECTORY_ENTRY dd = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)pResourceDir+1;
                  
          while(!dd->DataIsDirectory)
          {
              dd++;
          }

          dd = dd->OffsetToDirectory + (PIMAGE_RESOURCE_DIRECTORY_ENTRY)pResourceDir;


          PIMAGE_RESOURCE_DIRECTORY_ENTRY dd2 = dd + dd->OffsetToData;
          if(dd->NameIsString)
              cout << "dd->Name = " << dd->Name << endl;
          else
              cout << "dd->Id = " << dd->Id << endl;
          
      }
    // Если RVA находится внутри секции, то возвращаем указатель на ее заголовок.
    if (dwRVA >= pSectHeader->VirtualAddress && dwRVA < pSectHeader->VirtualAddress + pSectHeader->Misc.VirtualSize)
      return pSectHeader;
  }
  return NULL; // секция не найдена
}

LPBYTE GetFilePointer(LPBYTE pBase, DWORD dwRVA) {
  // Проверка правильности аргументов.
  if (pBase == NULL)
    return NULL;
  // Ищем секцию, содержащий данный RVA.
  IMAGE_SECTION_HEADER* pSectHeader = GetSection(GetHeader(pBase), dwRVA);
  if (pSectHeader == NULL) // Если секция не найдена,
    return pBase + dwRVA; // то RVA равно смещению в файле,
  // иначе вычисляем смещение относительно начала секции в файле.
  return pBase + pSectHeader->PointerToRawData + (dwRVA - pSectHeader->VirtualAddress);
}


LPBYTE OpenPEFile(LPCTSTR lpszFileName) {
  HANDLE hFile = CreateFile(lpszFileName, GENERIC_READ, FILE_SHARE_READ, NULL,
    OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
  if (hFile == INVALID_HANDLE_VALUE)
    return NULL;
  HANDLE hMapping = CreateFileMapping(hFile, NULL, PAGE_READONLY, 0, 0, NULL);
  CloseHandle(hFile);
  LPBYTE pBase = NULL;
  if (hMapping != NULL) {
    pBase = (LPBYTE)MapViewOfFile(hMapping, FILE_MAP_READ, 0, 0, 0);
    CloseHandle(hMapping);
  }
  return pBase;
}

void ClosePEFile(LPBYTE pBase) {
  if (pBase != NULL)
    UnmapViewOfFile(pBase);
}