2187 lines
64 KiB
C++
Executable File
2187 lines
64 KiB
C++
Executable File
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Image manipulation functions (image.cpp of detours.lib)
|
|
//
|
|
// Microsoft Research Detours Package, Version 2.1.
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
//
|
|
// Used for for payloads, byways, and imports.
|
|
//
|
|
|
|
#include <windows.h>
|
|
#if (_MSC_VER < 1310)
|
|
#else
|
|
#include <strsafe.h>
|
|
#endif
|
|
|
|
#if (_MSC_VER < 1299)
|
|
#pragma warning(disable: 4710)
|
|
#else
|
|
#endif
|
|
|
|
//#define DETOUR_DEBUG 1
|
|
#define DETOURS_INTERNAL
|
|
|
|
#include "detours.h"
|
|
|
|
namespace Detour
|
|
{
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
#ifndef _STRSAFE_H_INCLUDED_
|
|
static inline HRESULT StringCchLengthA(const char* psz, size_t cchMax, size_t* pcch)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
size_t cchMaxPrev = cchMax;
|
|
|
|
if (cchMax > 2147483647)
|
|
{
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
while (cchMax && (*psz != '\0'))
|
|
{
|
|
psz++;
|
|
cchMax--;
|
|
}
|
|
|
|
if (cchMax == 0)
|
|
{
|
|
// the string is longer than cchMax
|
|
hr = ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if (SUCCEEDED(hr) && pcch)
|
|
{
|
|
*pcch = cchMaxPrev - cchMax;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
static inline HRESULT StringCchCopyA(char* pszDest, size_t cchDest, const char* pszSrc)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (cchDest == 0)
|
|
{
|
|
// can not null terminate a zero-byte dest buffer
|
|
hr = ERROR_INVALID_PARAMETER;
|
|
}
|
|
else
|
|
{
|
|
while (cchDest && (*pszSrc != '\0'))
|
|
{
|
|
*pszDest++ = *pszSrc++;
|
|
cchDest--;
|
|
}
|
|
|
|
if (cchDest == 0)
|
|
{
|
|
// we are going to truncate pszDest
|
|
pszDest--;
|
|
hr = ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
*pszDest= '\0';
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static inline HRESULT StringCchCatA(char* pszDest, size_t cchDest, const char* pszSrc)
|
|
{
|
|
HRESULT hr;
|
|
size_t cchDestCurrent;
|
|
|
|
if (cchDest > 2147483647)
|
|
{
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
hr = StringCchLengthA(pszDest, cchDest, &cchDestCurrent);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = StringCchCopyA(pszDest + cchDestCurrent,
|
|
cchDest - cchDestCurrent,
|
|
pszSrc);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
#endif
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
class CImageData
|
|
{
|
|
friend class CImage;
|
|
|
|
public:
|
|
CImageData(PBYTE pbData, DWORD cbData);
|
|
~CImageData();
|
|
|
|
PBYTE Enumerate(GUID *pGuid, DWORD *pcbData, DWORD *pnIterator);
|
|
PBYTE Find(REFGUID rguid, DWORD *pcbData);
|
|
PBYTE Set(REFGUID rguid, PBYTE pbData, DWORD cbData);
|
|
|
|
BOOL Delete(REFGUID rguid);
|
|
BOOL Purge();
|
|
|
|
BOOL IsEmpty() { return m_cbData == 0; }
|
|
BOOL IsValid();
|
|
|
|
protected:
|
|
BOOL SizeTo(DWORD cbData);
|
|
|
|
protected:
|
|
PBYTE m_pbData;
|
|
DWORD m_cbData;
|
|
DWORD m_cbAlloc;
|
|
};
|
|
|
|
class CImageImportFile
|
|
{
|
|
friend class CImage;
|
|
friend class CImageImportName;
|
|
|
|
public:
|
|
CImageImportFile();
|
|
~CImageImportFile();
|
|
|
|
public:
|
|
CImageImportFile * m_pNextFile;
|
|
BOOL m_fByway;
|
|
|
|
CImageImportName * m_pImportNames;
|
|
DWORD m_nImportNames;
|
|
|
|
DWORD m_rvaOriginalFirstThunk;
|
|
DWORD m_rvaFirstThunk;
|
|
|
|
DWORD m_nForwarderChain;
|
|
PCHAR m_pszOrig;
|
|
PCHAR m_pszName;
|
|
};
|
|
|
|
class CImageImportName
|
|
{
|
|
friend class CImage;
|
|
friend class CImageImportFile;
|
|
|
|
public:
|
|
CImageImportName();
|
|
~CImageImportName();
|
|
|
|
public:
|
|
WORD m_nHint;
|
|
ULONG m_nOrig;
|
|
ULONG m_nOrdinal;
|
|
PCHAR m_pszOrig;
|
|
PCHAR m_pszName;
|
|
};
|
|
|
|
class CImage
|
|
{
|
|
friend class CImageThunks;
|
|
friend class CImageChars;
|
|
friend class CImageImportFile;
|
|
friend class CImageImportName;
|
|
|
|
public:
|
|
CImage();
|
|
~CImage();
|
|
|
|
static CImage * IsValid(PDETOUR_BINARY pBinary);
|
|
|
|
public: // File Functions
|
|
BOOL Read(HANDLE hFile);
|
|
BOOL Write(HANDLE hFile);
|
|
BOOL Close();
|
|
|
|
public: // Manipulation Functions
|
|
PBYTE DataEnum(GUID *pGuid, DWORD *pcbData, DWORD *pnIterator);
|
|
PBYTE DataFind(REFGUID rguid, DWORD *pcbData);
|
|
PBYTE DataSet(REFGUID rguid, PBYTE pbData, DWORD cbData);
|
|
BOOL DataDelete(REFGUID rguid);
|
|
BOOL DataPurge();
|
|
|
|
BOOL EditImports(PVOID pContext,
|
|
PF_DETOUR_BINARY_BYWAY_CALLBACK pfBywayCallback,
|
|
PF_DETOUR_BINARY_FILE_CALLBACK pfFileCallback,
|
|
PF_DETOUR_BINARY_SYMBOL_CALLBACK pfSymbolCallback,
|
|
PF_DETOUR_BINARY_COMMIT_CALLBACK pfCommitCallback);
|
|
|
|
protected:
|
|
BOOL WriteFile(HANDLE hFile,
|
|
LPCVOID lpBuffer,
|
|
DWORD nNumberOfBytesToWrite,
|
|
LPDWORD lpNumberOfBytesWritten);
|
|
BOOL CopyFileData(HANDLE hFile, DWORD nOldPos, DWORD cbData);
|
|
BOOL ZeroFileData(HANDLE hFile, DWORD cbData);
|
|
BOOL AlignFileData(HANDLE hFile);
|
|
|
|
BOOL SizeOutputBuffer(DWORD cbData);
|
|
PBYTE AllocateOutput(DWORD cbData, DWORD *pnVirtAddr);
|
|
|
|
PVOID RvaToVa(ULONG_PTR nRva);
|
|
DWORD RvaToFileOffset(DWORD nRva);
|
|
|
|
DWORD FileAlign(DWORD nAddr);
|
|
DWORD SectionAlign(DWORD nAddr);
|
|
|
|
BOOL CheckImportsNeeded(DWORD *pnTables,
|
|
DWORD *pnThunks,
|
|
DWORD *pnChars);
|
|
|
|
CImageImportFile * NewByway(__in_z PCHAR pszName);
|
|
|
|
private:
|
|
DWORD m_dwValidSignature;
|
|
CImageData * m_pImageData; // Read & Write
|
|
|
|
HANDLE m_hMap; // Read & Write
|
|
PBYTE m_pMap; // Read & Write
|
|
|
|
DWORD m_nNextFileAddr; // Write
|
|
DWORD m_nNextVirtAddr; // Write
|
|
|
|
IMAGE_DOS_HEADER m_DosHeader; // Read & Write
|
|
IMAGE_NT_HEADERS m_NtHeader; // Read & Write
|
|
IMAGE_SECTION_HEADER m_SectionHeaders[IMAGE_NUMBEROF_DIRECTORY_ENTRIES];
|
|
|
|
DWORD m_nPrePE;
|
|
DWORD m_cbPrePE;
|
|
DWORD m_cbPostPE;
|
|
|
|
DWORD m_nPeOffset;
|
|
DWORD m_nSectionsOffset;
|
|
DWORD m_nExtraOffset;
|
|
DWORD m_nFileSize;
|
|
|
|
DWORD m_nOutputVirtAddr;
|
|
DWORD m_nOutputVirtSize;
|
|
DWORD m_nOutputFileAddr;
|
|
|
|
PBYTE m_pbOutputBuffer;
|
|
DWORD m_cbOutputBuffer;
|
|
|
|
CImageImportFile * m_pImportFiles;
|
|
DWORD m_nImportFiles;
|
|
|
|
BOOL m_fHadDetourSection;
|
|
|
|
private:
|
|
enum {
|
|
DETOUR_IMAGE_VALID_SIGNATURE = 0xfedcba01, // "Dtr\0"
|
|
};
|
|
};
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
static BYTE s_rbDosCode[0x10] = {
|
|
0x0E,0x1F,0xBA,0x0E,0x00,0xB4,0x09,0xCD,
|
|
0x21,0xB8,0x01,0x4C,0xCD,0x21,'*','*'
|
|
};
|
|
|
|
static inline DWORD Max(DWORD a, DWORD b)
|
|
{
|
|
return a > b ? a : b;
|
|
}
|
|
|
|
static inline DWORD Align(DWORD a, DWORD size)
|
|
{
|
|
size--;
|
|
return (a + size) & ~size;
|
|
}
|
|
|
|
static inline DWORD QuadAlign(DWORD a)
|
|
{
|
|
return Align(a, 8);
|
|
}
|
|
|
|
static PCHAR DuplicateString(__in_z PCHAR pszIn)
|
|
{
|
|
if (pszIn) {
|
|
UINT nIn = (UINT)strlen(pszIn);
|
|
PCHAR pszOut = new CHAR [nIn + 1];
|
|
if (pszOut == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
}
|
|
else {
|
|
CopyMemory(pszOut, pszIn, nIn + 1);
|
|
}
|
|
return pszOut;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
static PCHAR ReplaceString(__deref_out PCHAR *ppsz, __in_z PCHAR pszIn)
|
|
{
|
|
if (ppsz == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
UINT nIn;
|
|
if (*ppsz != NULL) {
|
|
if (strcmp(*ppsz, pszIn) == 0) {
|
|
return *ppsz;
|
|
}
|
|
nIn = (UINT)strlen(pszIn);
|
|
|
|
if (strlen(*ppsz) == nIn) {
|
|
CopyMemory(*ppsz, pszIn, nIn + 1);
|
|
return *ppsz;
|
|
}
|
|
else {
|
|
delete[] *ppsz;
|
|
*ppsz = NULL;
|
|
}
|
|
}
|
|
else {
|
|
nIn = (UINT)strlen(pszIn);
|
|
}
|
|
|
|
*ppsz = new CHAR [nIn + 1];
|
|
if (*ppsz == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
}
|
|
else {
|
|
CopyMemory(*ppsz, pszIn, nIn + 1);
|
|
}
|
|
return *ppsz;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
CImageImportFile::CImageImportFile()
|
|
{
|
|
m_pNextFile = NULL;
|
|
m_fByway = FALSE;
|
|
|
|
m_pImportNames = NULL;
|
|
m_nImportNames = 0;
|
|
|
|
m_rvaOriginalFirstThunk = 0;
|
|
m_rvaFirstThunk = 0;
|
|
|
|
m_nForwarderChain = (UINT)0;
|
|
m_pszName = NULL;
|
|
m_pszOrig = NULL;
|
|
}
|
|
|
|
CImageImportFile::~CImageImportFile()
|
|
{
|
|
if (m_pNextFile) {
|
|
delete m_pNextFile;
|
|
m_pNextFile = NULL;
|
|
}
|
|
if (m_pImportNames) {
|
|
delete[] m_pImportNames;
|
|
m_pImportNames = NULL;
|
|
m_nImportNames = 0;
|
|
}
|
|
if (m_pszName) {
|
|
delete[] m_pszName;
|
|
m_pszName = NULL;
|
|
}
|
|
if (m_pszOrig) {
|
|
delete[] m_pszOrig;
|
|
m_pszOrig = NULL;
|
|
}
|
|
}
|
|
|
|
CImageImportName::CImageImportName()
|
|
{
|
|
m_nOrig = 0;
|
|
m_nOrdinal = 0;
|
|
m_nHint = 0;
|
|
m_pszName = NULL;
|
|
m_pszOrig = NULL;
|
|
}
|
|
|
|
CImageImportName::~CImageImportName()
|
|
{
|
|
if (m_pszName) {
|
|
delete[] m_pszName;
|
|
m_pszName = NULL;
|
|
}
|
|
if (m_pszOrig) {
|
|
delete[] m_pszOrig;
|
|
m_pszOrig = NULL;
|
|
}
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
CImageData::CImageData(PBYTE pbData, DWORD cbData)
|
|
{
|
|
m_pbData = pbData;
|
|
m_cbData = cbData;
|
|
m_cbAlloc = 0;
|
|
}
|
|
|
|
CImageData::~CImageData()
|
|
{
|
|
IsValid();
|
|
|
|
if (m_cbAlloc == 0) {
|
|
m_pbData = NULL;
|
|
}
|
|
if (m_pbData) {
|
|
delete[] m_pbData;
|
|
m_pbData = NULL;
|
|
}
|
|
m_cbData = 0;
|
|
m_cbAlloc = 0;
|
|
}
|
|
|
|
BOOL CImageData::SizeTo(DWORD cbData)
|
|
{
|
|
IsValid();
|
|
|
|
if (cbData <= m_cbAlloc) {
|
|
return TRUE;
|
|
}
|
|
|
|
PBYTE pbNew = new BYTE [cbData];
|
|
if (pbNew == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
return FALSE;
|
|
}
|
|
|
|
if (m_pbData) {
|
|
CopyMemory(pbNew, m_pbData, m_cbData);
|
|
if (m_cbAlloc > 0) {
|
|
delete[] m_pbData;
|
|
}
|
|
m_pbData = NULL;
|
|
}
|
|
m_pbData = pbNew;
|
|
m_cbAlloc = cbData;
|
|
|
|
IsValid();
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CImageData::Purge()
|
|
{
|
|
m_cbData = 0;
|
|
|
|
IsValid();
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CImageData::IsValid()
|
|
{
|
|
if (m_pbData == NULL) {
|
|
return TRUE;
|
|
}
|
|
|
|
PBYTE pbBeg = m_pbData;
|
|
PBYTE pbEnd = m_pbData + m_cbData;
|
|
|
|
for (PBYTE pbIter = pbBeg; pbIter < pbEnd;) {
|
|
PDETOUR_SECTION_RECORD pRecord = (PDETOUR_SECTION_RECORD)pbIter;
|
|
|
|
if (pRecord->cbBytes < sizeof(DETOUR_SECTION_RECORD)) {
|
|
return FALSE;
|
|
}
|
|
if (pRecord->nReserved != 0) {
|
|
return FALSE;
|
|
}
|
|
|
|
pbIter += pRecord->cbBytes;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
PBYTE CImageData::Enumerate(GUID *pGuid, DWORD *pcbData, DWORD *pnIterator)
|
|
{
|
|
IsValid();
|
|
|
|
if (pnIterator == NULL ||
|
|
m_cbData < *pnIterator + sizeof(DETOUR_SECTION_RECORD)) {
|
|
|
|
if (pcbData) {
|
|
*pcbData = 0;
|
|
}
|
|
if (pGuid) {
|
|
ZeroMemory(pGuid, sizeof(*pGuid));
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
PDETOUR_SECTION_RECORD pRecord = (PDETOUR_SECTION_RECORD)(m_pbData + *pnIterator);
|
|
|
|
if (pGuid) {
|
|
*pGuid = pRecord->guid;
|
|
}
|
|
if (pcbData) {
|
|
*pcbData = pRecord->cbBytes - sizeof(DETOUR_SECTION_RECORD);
|
|
}
|
|
*pnIterator = (LONG)(((PBYTE)pRecord - m_pbData) + pRecord->cbBytes);
|
|
|
|
return (PBYTE)(pRecord + 1);
|
|
}
|
|
|
|
PBYTE CImageData::Find(REFGUID rguid, DWORD *pcbData)
|
|
{
|
|
IsValid();
|
|
|
|
DWORD cbBytes = sizeof(DETOUR_SECTION_RECORD);
|
|
for (DWORD nOffset = 0; nOffset < m_cbData; nOffset += cbBytes) {
|
|
PDETOUR_SECTION_RECORD pRecord = (PDETOUR_SECTION_RECORD)(m_pbData + nOffset);
|
|
|
|
cbBytes = pRecord->cbBytes;
|
|
if (cbBytes > m_cbData) {
|
|
break;
|
|
}
|
|
if (cbBytes < sizeof(DETOUR_SECTION_RECORD)) {
|
|
continue;
|
|
}
|
|
|
|
if (pRecord->guid.Data1 == rguid.Data1 &&
|
|
pRecord->guid.Data2 == rguid.Data2 &&
|
|
pRecord->guid.Data3 == rguid.Data3 &&
|
|
pRecord->guid.Data4[0] == rguid.Data4[0] &&
|
|
pRecord->guid.Data4[1] == rguid.Data4[1] &&
|
|
pRecord->guid.Data4[2] == rguid.Data4[2] &&
|
|
pRecord->guid.Data4[3] == rguid.Data4[3] &&
|
|
pRecord->guid.Data4[4] == rguid.Data4[4] &&
|
|
pRecord->guid.Data4[5] == rguid.Data4[5] &&
|
|
pRecord->guid.Data4[6] == rguid.Data4[6] &&
|
|
pRecord->guid.Data4[7] == rguid.Data4[7]) {
|
|
|
|
*pcbData = cbBytes - sizeof(DETOUR_SECTION_RECORD);
|
|
return (PBYTE)(pRecord + 1);
|
|
}
|
|
}
|
|
|
|
if (pcbData) {
|
|
*pcbData = 0;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
BOOL CImageData::Delete(REFGUID rguid)
|
|
{
|
|
IsValid();
|
|
|
|
PBYTE pbFound = NULL;
|
|
DWORD cbFound = 0;
|
|
|
|
pbFound = Find(rguid, &cbFound);
|
|
if (pbFound == NULL) {
|
|
SetLastError(ERROR_MOD_NOT_FOUND);
|
|
return FALSE;
|
|
}
|
|
|
|
pbFound -= sizeof(DETOUR_SECTION_RECORD);
|
|
cbFound += sizeof(DETOUR_SECTION_RECORD);
|
|
|
|
PBYTE pbRestData = pbFound + cbFound;
|
|
DWORD cbRestData = m_cbData - (LONG)(pbRestData - m_pbData);
|
|
|
|
if (cbRestData) {
|
|
MoveMemory(pbFound, pbRestData, cbRestData);
|
|
}
|
|
m_cbData -= cbFound;
|
|
|
|
IsValid();
|
|
return TRUE;
|
|
}
|
|
|
|
PBYTE CImageData::Set(REFGUID rguid, PBYTE pbData, DWORD cbData)
|
|
{
|
|
IsValid();
|
|
Delete(rguid);
|
|
|
|
DWORD cbAlloc = QuadAlign(cbData);
|
|
|
|
if (!SizeTo(m_cbData + cbAlloc + sizeof(DETOUR_SECTION_RECORD))) {
|
|
return NULL;
|
|
}
|
|
|
|
PDETOUR_SECTION_RECORD pRecord = (PDETOUR_SECTION_RECORD)(m_pbData + m_cbData);
|
|
pRecord->cbBytes = cbAlloc + sizeof(DETOUR_SECTION_RECORD);
|
|
pRecord->nReserved = 0;
|
|
pRecord->guid = rguid;
|
|
|
|
PBYTE pbDest = (PBYTE)(pRecord + 1);
|
|
if (pbData) {
|
|
CopyMemory(pbDest, pbData, cbData);
|
|
if (cbData < cbAlloc) {
|
|
ZeroMemory(pbDest + cbData, cbAlloc - cbData);
|
|
}
|
|
}
|
|
else {
|
|
if (cbAlloc > 0) {
|
|
ZeroMemory(pbDest, cbAlloc);
|
|
}
|
|
}
|
|
|
|
m_cbData += cbAlloc + sizeof(DETOUR_SECTION_RECORD);
|
|
|
|
IsValid();
|
|
return pbDest;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
class CImageThunks
|
|
{
|
|
private:
|
|
CImage * m_pImage;
|
|
PIMAGE_THUNK_DATA m_pThunks;
|
|
DWORD m_nThunks;
|
|
DWORD m_nThunksMax;
|
|
DWORD m_nThunkVirtAddr;
|
|
|
|
public:
|
|
CImageThunks(CImage *pImage, DWORD nThunksMax, DWORD *pnAddr)
|
|
{
|
|
m_pImage = pImage;
|
|
m_nThunks = 0;
|
|
m_nThunksMax = nThunksMax;
|
|
m_pThunks = (PIMAGE_THUNK_DATA)
|
|
m_pImage->AllocateOutput(sizeof(IMAGE_THUNK_DATA) * nThunksMax,
|
|
&m_nThunkVirtAddr);
|
|
*pnAddr = m_nThunkVirtAddr;
|
|
}
|
|
|
|
PIMAGE_THUNK_DATA Current(DWORD *pnVirtAddr)
|
|
{
|
|
if (m_nThunksMax > 1) {
|
|
*pnVirtAddr = m_nThunkVirtAddr;
|
|
return m_pThunks;
|
|
}
|
|
*pnVirtAddr = 0;
|
|
return NULL;
|
|
}
|
|
|
|
PIMAGE_THUNK_DATA Allocate(ULONG_PTR nData, DWORD *pnVirtAddr)
|
|
{
|
|
if (m_nThunks < m_nThunksMax) {
|
|
*pnVirtAddr = m_nThunkVirtAddr;
|
|
|
|
m_nThunks++;
|
|
m_nThunkVirtAddr += sizeof(IMAGE_THUNK_DATA);
|
|
m_pThunks->u1.Ordinal = nData;
|
|
return m_pThunks++;
|
|
}
|
|
*pnVirtAddr = 0;
|
|
return NULL;
|
|
}
|
|
|
|
DWORD Size()
|
|
{
|
|
return m_nThunksMax * sizeof(IMAGE_THUNK_DATA);
|
|
}
|
|
};
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
class CImageChars
|
|
{
|
|
private:
|
|
CImage * m_pImage;
|
|
PCHAR m_pChars;
|
|
DWORD m_nChars;
|
|
DWORD m_nCharsMax;
|
|
DWORD m_nCharVirtAddr;
|
|
|
|
public:
|
|
CImageChars(CImage *pImage, DWORD nCharsMax, DWORD *pnAddr)
|
|
{
|
|
m_pImage = pImage;
|
|
m_nChars = 0;
|
|
m_nCharsMax = nCharsMax;
|
|
m_pChars = (PCHAR)m_pImage->AllocateOutput(nCharsMax, &m_nCharVirtAddr);
|
|
*pnAddr = m_nCharVirtAddr;
|
|
}
|
|
|
|
PCHAR Allocate(__in_z PCHAR pszString, DWORD *pnVirtAddr)
|
|
{
|
|
DWORD nLen = (DWORD)strlen(pszString) + 1;
|
|
nLen += (nLen & 1);
|
|
|
|
if (m_nChars + nLen > m_nCharsMax) {
|
|
*pnVirtAddr = 0;
|
|
return NULL;
|
|
}
|
|
|
|
*pnVirtAddr = m_nCharVirtAddr;
|
|
HRESULT hrRet = StringCchCopyA(m_pChars,m_nCharsMax, pszString);
|
|
|
|
if (FAILED(hrRet)) {
|
|
return NULL;
|
|
}
|
|
|
|
pszString = m_pChars;
|
|
|
|
m_pChars += nLen;
|
|
m_nChars += nLen;
|
|
m_nCharVirtAddr += nLen;
|
|
|
|
return pszString;
|
|
}
|
|
|
|
PCHAR Allocate(PCHAR pszString, DWORD nHint, DWORD *pnVirtAddr)
|
|
{
|
|
DWORD nLen = (DWORD)strlen(pszString) + 1 + sizeof(USHORT);
|
|
nLen += (nLen & 1);
|
|
|
|
if (m_nChars + nLen > m_nCharsMax) {
|
|
*pnVirtAddr = 0;
|
|
return NULL;
|
|
}
|
|
|
|
*pnVirtAddr = m_nCharVirtAddr;
|
|
*(USHORT *)m_pChars = (USHORT)nHint;
|
|
|
|
HRESULT hrRet = StringCchCopyA(m_pChars + sizeof(USHORT), m_nCharsMax, pszString);
|
|
if (FAILED(hrRet)) {
|
|
return NULL;
|
|
}
|
|
|
|
pszString = m_pChars + sizeof(USHORT);
|
|
|
|
m_pChars += nLen;
|
|
m_nChars += nLen;
|
|
m_nCharVirtAddr += nLen;
|
|
|
|
return pszString;
|
|
}
|
|
|
|
DWORD Size()
|
|
{
|
|
return m_nChars;
|
|
}
|
|
};
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
CImage * CImage::IsValid(PDETOUR_BINARY pBinary)
|
|
{
|
|
if (pBinary) {
|
|
CImage *pImage = (CImage *)pBinary;
|
|
|
|
if (pImage->m_dwValidSignature == DETOUR_IMAGE_VALID_SIGNATURE) {
|
|
return pImage;
|
|
}
|
|
}
|
|
SetLastError(ERROR_INVALID_HANDLE);
|
|
return NULL;
|
|
}
|
|
|
|
CImage::CImage()
|
|
{
|
|
m_dwValidSignature = (DWORD)DETOUR_IMAGE_VALID_SIGNATURE;
|
|
|
|
m_hMap = NULL;
|
|
m_pMap = NULL;
|
|
|
|
m_nPeOffset = 0;
|
|
m_nSectionsOffset = 0;
|
|
|
|
m_pbOutputBuffer = NULL;
|
|
m_cbOutputBuffer = 0;
|
|
|
|
m_pImageData = NULL;
|
|
|
|
m_pImportFiles = NULL;
|
|
m_nImportFiles = 0;
|
|
|
|
m_fHadDetourSection = FALSE;
|
|
}
|
|
|
|
CImage::~CImage()
|
|
{
|
|
Close();
|
|
m_dwValidSignature = 0;
|
|
}
|
|
|
|
BOOL CImage::Close()
|
|
{
|
|
if (m_pImportFiles) {
|
|
delete m_pImportFiles;
|
|
m_pImportFiles = NULL;
|
|
m_nImportFiles = 0;
|
|
}
|
|
|
|
if (m_pImageData) {
|
|
delete m_pImageData;
|
|
m_pImageData = NULL;
|
|
}
|
|
|
|
if (m_pMap != NULL) {
|
|
UnmapViewOfFile(m_pMap);
|
|
m_pMap = NULL;
|
|
}
|
|
|
|
if (m_hMap) {
|
|
CloseHandle(m_hMap);
|
|
m_hMap = NULL;
|
|
}
|
|
|
|
if (m_pbOutputBuffer) {
|
|
delete[] m_pbOutputBuffer;
|
|
m_pbOutputBuffer = NULL;
|
|
m_cbOutputBuffer = 0;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
PBYTE CImage::DataEnum(GUID *pGuid, DWORD *pcbData, DWORD *pnIterator)
|
|
{
|
|
if (m_pImageData == NULL) {
|
|
return NULL;
|
|
}
|
|
return m_pImageData->Enumerate(pGuid, pcbData, pnIterator);
|
|
}
|
|
|
|
PBYTE CImage::DataFind(REFGUID rguid, DWORD *pcbData)
|
|
{
|
|
if (m_pImageData == NULL) {
|
|
return NULL;
|
|
}
|
|
return m_pImageData->Find(rguid, pcbData);
|
|
}
|
|
|
|
PBYTE CImage::DataSet(REFGUID rguid, PBYTE pbData, DWORD cbData)
|
|
{
|
|
if (m_pImageData == NULL) {
|
|
return NULL;
|
|
}
|
|
return m_pImageData->Set(rguid, pbData, cbData);
|
|
}
|
|
|
|
BOOL CImage::DataDelete(REFGUID rguid)
|
|
{
|
|
if (m_pImageData == NULL) {
|
|
return FALSE;
|
|
}
|
|
return m_pImageData->Delete(rguid);
|
|
}
|
|
|
|
BOOL CImage::DataPurge()
|
|
{
|
|
if (m_pImageData == NULL) {
|
|
return TRUE;
|
|
}
|
|
return m_pImageData->Purge();
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
BOOL CImage::SizeOutputBuffer(DWORD cbData)
|
|
{
|
|
if (m_cbOutputBuffer < cbData) {
|
|
if (cbData < 1024) {//65536
|
|
cbData = 1024;
|
|
}
|
|
cbData = FileAlign(cbData);
|
|
|
|
PBYTE pOutput = new BYTE [cbData];
|
|
if (pOutput == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
return FALSE;
|
|
}
|
|
|
|
if (m_pbOutputBuffer) {
|
|
CopyMemory(pOutput, m_pbOutputBuffer, m_cbOutputBuffer);
|
|
|
|
delete[] m_pbOutputBuffer;
|
|
m_pbOutputBuffer = NULL;
|
|
}
|
|
|
|
ZeroMemory(pOutput + m_cbOutputBuffer, cbData - m_cbOutputBuffer),
|
|
|
|
m_pbOutputBuffer = pOutput;
|
|
m_cbOutputBuffer = cbData;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
PBYTE CImage::AllocateOutput(DWORD cbData, DWORD *pnVirtAddr)
|
|
{
|
|
cbData = QuadAlign(cbData);
|
|
|
|
PBYTE pbData = m_pbOutputBuffer + m_nOutputVirtSize;
|
|
|
|
*pnVirtAddr = m_nOutputVirtAddr + m_nOutputVirtSize;
|
|
m_nOutputVirtSize += cbData;
|
|
|
|
if (m_nOutputVirtSize > m_cbOutputBuffer) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
return NULL;
|
|
}
|
|
|
|
ZeroMemory(pbData, cbData);
|
|
|
|
return pbData;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
DWORD CImage::FileAlign(DWORD nAddr)
|
|
{
|
|
return Align(nAddr, m_NtHeader.OptionalHeader.FileAlignment);
|
|
}
|
|
|
|
DWORD CImage::SectionAlign(DWORD nAddr)
|
|
{
|
|
return Align(nAddr, m_NtHeader.OptionalHeader.SectionAlignment);
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
PVOID CImage::RvaToVa(ULONG_PTR nRva)
|
|
{
|
|
if (nRva == 0) {
|
|
return NULL;
|
|
}
|
|
|
|
for (DWORD n = 0; n < m_NtHeader.FileHeader.NumberOfSections; n++) {
|
|
DWORD vaStart = m_SectionHeaders[n].VirtualAddress;
|
|
DWORD vaEnd = vaStart + m_SectionHeaders[n].SizeOfRawData;
|
|
|
|
if (nRva >= vaStart && nRva < vaEnd) {
|
|
return (PBYTE)m_pMap
|
|
+ m_SectionHeaders[n].PointerToRawData
|
|
+ nRva - m_SectionHeaders[n].VirtualAddress;
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
DWORD CImage::RvaToFileOffset(DWORD nRva)
|
|
{
|
|
DWORD n;
|
|
for (n = 0; n < m_NtHeader.FileHeader.NumberOfSections; n++) {
|
|
DWORD vaStart = m_SectionHeaders[n].VirtualAddress;
|
|
DWORD vaEnd = vaStart + m_SectionHeaders[n].SizeOfRawData;
|
|
|
|
if (nRva >= vaStart && nRva < vaEnd) {
|
|
return m_SectionHeaders[n].PointerToRawData
|
|
+ nRva - m_SectionHeaders[n].VirtualAddress;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
BOOL CImage::WriteFile(HANDLE hFile, LPCVOID lpBuffer, DWORD nNumberOfBytesToWrite,
|
|
LPDWORD lpNumberOfBytesWritten)
|
|
{
|
|
return ::WriteFile(hFile,
|
|
lpBuffer,
|
|
nNumberOfBytesToWrite,
|
|
lpNumberOfBytesWritten,
|
|
NULL);
|
|
}
|
|
|
|
|
|
BOOL CImage::CopyFileData(HANDLE hFile, DWORD nOldPos, DWORD cbData)
|
|
{
|
|
DWORD cbDone = 0;
|
|
return WriteFile(hFile, m_pMap + nOldPos, cbData, &cbDone);
|
|
}
|
|
|
|
BOOL CImage::ZeroFileData(HANDLE hFile, DWORD cbData)
|
|
{
|
|
if (!SizeOutputBuffer(4096)) {
|
|
return FALSE;
|
|
}
|
|
|
|
ZeroMemory(m_pbOutputBuffer, 4096);
|
|
|
|
for (DWORD cbLeft = cbData; cbLeft > 0;) {
|
|
DWORD cbStep = cbLeft > sizeof(m_pbOutputBuffer)
|
|
? sizeof(m_pbOutputBuffer) : cbLeft;
|
|
DWORD cbDone = 0;
|
|
|
|
if (!WriteFile(hFile, m_pbOutputBuffer, cbStep, &cbDone)) {
|
|
return FALSE;
|
|
}
|
|
if (cbDone == 0) {
|
|
break;
|
|
}
|
|
|
|
cbLeft -= cbDone;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CImage::AlignFileData(HANDLE hFile)
|
|
{
|
|
DWORD nLastFileAddr = m_nNextFileAddr;
|
|
|
|
m_nNextFileAddr = FileAlign(m_nNextFileAddr);
|
|
m_nNextVirtAddr = SectionAlign(m_nNextVirtAddr);
|
|
|
|
if (hFile != INVALID_HANDLE_VALUE) {
|
|
if (m_nNextFileAddr > nLastFileAddr) {
|
|
if (SetFilePointer(hFile, nLastFileAddr, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
return ZeroFileData(hFile, m_nNextFileAddr - nLastFileAddr);
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CImage::Read(HANDLE hFile)
|
|
{
|
|
DWORD n;
|
|
PBYTE pbData = NULL;
|
|
DWORD cbData = 0;
|
|
|
|
if (hFile == INVALID_HANDLE_VALUE) {
|
|
SetLastError(ERROR_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////// Create mapping.
|
|
//
|
|
m_nFileSize = GetFileSize(hFile, NULL);
|
|
if (m_nFileSize == (DWORD)-1) {
|
|
return FALSE;
|
|
}
|
|
|
|
m_hMap = CreateFileMapping(hFile, NULL, PAGE_READONLY, 0, 0, NULL);
|
|
if (m_hMap == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
m_pMap = (PBYTE)MapViewOfFile(m_hMap, FILE_MAP_READ, 0, 0, 0);
|
|
if (m_pMap == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
////////////////////////////////////////////////////// Process DOS Header.
|
|
//
|
|
PIMAGE_DOS_HEADER pDosHeader = (PIMAGE_DOS_HEADER)m_pMap;
|
|
if (pDosHeader->e_magic != IMAGE_DOS_SIGNATURE) {
|
|
SetLastError(ERROR_BAD_EXE_FORMAT);
|
|
return FALSE;
|
|
}
|
|
m_nPeOffset = pDosHeader->e_lfanew;
|
|
m_nPrePE = 0;
|
|
m_cbPrePE = QuadAlign(pDosHeader->e_lfanew);
|
|
|
|
CopyMemory(&m_DosHeader, m_pMap + m_nPrePE, sizeof(m_DosHeader));
|
|
|
|
/////////////////////////////////////////////////////// Process PE Header.
|
|
//
|
|
CopyMemory(&m_NtHeader, m_pMap + m_nPeOffset, sizeof(m_NtHeader));
|
|
if (m_NtHeader.Signature != IMAGE_NT_SIGNATURE) {
|
|
SetLastError(ERROR_INVALID_EXE_SIGNATURE);
|
|
return FALSE;
|
|
}
|
|
if (m_NtHeader.FileHeader.SizeOfOptionalHeader == 0) {
|
|
SetLastError(ERROR_EXE_MARKED_INVALID);
|
|
return FALSE;
|
|
}
|
|
m_nSectionsOffset = m_nPeOffset
|
|
+ sizeof(m_NtHeader.Signature)
|
|
+ sizeof(m_NtHeader.FileHeader)
|
|
+ m_NtHeader.FileHeader.SizeOfOptionalHeader;
|
|
|
|
///////////////////////////////////////////////// Process Section Headers.
|
|
//
|
|
if (m_NtHeader.FileHeader.NumberOfSections > (sizeof(m_SectionHeaders) /
|
|
sizeof(m_SectionHeaders[0]))) {
|
|
SetLastError(ERROR_EXE_MARKED_INVALID);
|
|
return FALSE;
|
|
}
|
|
CopyMemory(&m_SectionHeaders,
|
|
m_pMap + m_nSectionsOffset,
|
|
sizeof(m_SectionHeaders[0]) * m_NtHeader.FileHeader.NumberOfSections);
|
|
|
|
/////////////////////////////////////////////////// Parse .detour Section.
|
|
//
|
|
DWORD rvaOriginalImageDirectory = 0;
|
|
DWORD rvaDetourBeg = 0;
|
|
DWORD rvaDetourEnd = 0;
|
|
|
|
for (n = 0; n < m_NtHeader.FileHeader.NumberOfSections; n++) {
|
|
if (strcmp((PCHAR)m_SectionHeaders[n].Name, ".detour") == 0) {
|
|
DETOUR_SECTION_HEADER dh;
|
|
CopyMemory(&dh,
|
|
m_pMap + m_SectionHeaders[n].PointerToRawData,
|
|
sizeof(dh));
|
|
|
|
rvaOriginalImageDirectory = dh.nOriginalImportVirtualAddress;
|
|
if (dh.cbPrePE != 0) {
|
|
m_nPrePE = m_SectionHeaders[n].PointerToRawData + sizeof(dh);
|
|
m_cbPrePE = dh.cbPrePE;
|
|
}
|
|
rvaDetourBeg = m_SectionHeaders[n].VirtualAddress;
|
|
rvaDetourEnd = rvaDetourBeg + m_SectionHeaders[n].SizeOfRawData;
|
|
}
|
|
}
|
|
|
|
//////////////////////////////////////////////////////// Get Import Table.
|
|
//
|
|
DWORD rvaImageDirectory = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress;
|
|
PIMAGE_IMPORT_DESCRIPTOR iidp
|
|
= (PIMAGE_IMPORT_DESCRIPTOR)RvaToVa(rvaImageDirectory);
|
|
PIMAGE_IMPORT_DESCRIPTOR oidp
|
|
= (PIMAGE_IMPORT_DESCRIPTOR)RvaToVa(rvaOriginalImageDirectory);
|
|
|
|
if (oidp == NULL) {
|
|
oidp = iidp;
|
|
}
|
|
if (iidp == NULL || oidp == NULL) {
|
|
SetLastError(ERROR_EXE_MARKED_INVALID);
|
|
return FALSE;
|
|
}
|
|
|
|
DWORD nFiles = 0;
|
|
for (; iidp[nFiles].OriginalFirstThunk != 0; nFiles++) {
|
|
}
|
|
|
|
CImageImportFile **ppLastFile = &m_pImportFiles;
|
|
m_pImportFiles = NULL;
|
|
|
|
for (n = 0; n < nFiles; n++, iidp++) {
|
|
ULONG_PTR rvaName = iidp->Name;
|
|
PCHAR pszName = (PCHAR)RvaToVa(rvaName);
|
|
if (pszName == NULL) {
|
|
SetLastError(ERROR_EXE_MARKED_INVALID);
|
|
goto fail;
|
|
}
|
|
|
|
CImageImportFile *pImportFile = new CImageImportFile;
|
|
if (pImportFile == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
goto fail;
|
|
}
|
|
|
|
*ppLastFile = pImportFile;
|
|
ppLastFile = &pImportFile->m_pNextFile;
|
|
m_nImportFiles++;
|
|
|
|
pImportFile->m_pszName = DuplicateString(pszName);
|
|
if (pImportFile->m_pszName == NULL) {
|
|
goto fail;
|
|
}
|
|
|
|
pImportFile->m_rvaOriginalFirstThunk = iidp->OriginalFirstThunk;
|
|
pImportFile->m_rvaFirstThunk = iidp->FirstThunk;
|
|
pImportFile->m_nForwarderChain = iidp->ForwarderChain;
|
|
pImportFile->m_pImportNames = NULL;
|
|
pImportFile->m_nImportNames = 0;
|
|
pImportFile->m_fByway = FALSE;
|
|
|
|
if ((ULONG)iidp->FirstThunk >= rvaDetourBeg &&
|
|
(ULONG)iidp->FirstThunk < rvaDetourEnd) {
|
|
|
|
pImportFile->m_pszOrig = NULL;
|
|
pImportFile->m_fByway = TRUE;
|
|
continue;
|
|
}
|
|
|
|
rvaName = oidp->Name;
|
|
pszName = (PCHAR)RvaToVa(rvaName);
|
|
if (pszName == NULL) {
|
|
SetLastError(ERROR_EXE_MARKED_INVALID);
|
|
goto fail;
|
|
}
|
|
pImportFile->m_pszOrig = DuplicateString(pszName);
|
|
if (pImportFile->m_pszOrig == NULL) {
|
|
goto fail;
|
|
}
|
|
|
|
DWORD rvaThunk = iidp->OriginalFirstThunk;
|
|
PIMAGE_THUNK_DATA pAddrThunk = (PIMAGE_THUNK_DATA)RvaToVa(rvaThunk);
|
|
rvaThunk = oidp->OriginalFirstThunk;
|
|
PIMAGE_THUNK_DATA pLookThunk = (PIMAGE_THUNK_DATA)RvaToVa(rvaThunk);
|
|
|
|
DWORD nNames = 0;
|
|
if (pAddrThunk) {
|
|
for (; pAddrThunk[nNames].u1.Ordinal; nNames++) {
|
|
}
|
|
}
|
|
|
|
if (pAddrThunk && nNames) {
|
|
pImportFile->m_nImportNames = nNames;
|
|
pImportFile->m_pImportNames = new CImageImportName [nNames];
|
|
if (pImportFile->m_pImportNames == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
goto fail;
|
|
}
|
|
|
|
CImageImportName *pImportName = &pImportFile->m_pImportNames[0];
|
|
|
|
for (DWORD f = 0; f < nNames; f++, pImportName++) {
|
|
pImportName->m_nOrig = 0;
|
|
pImportName->m_nOrdinal = 0;
|
|
pImportName->m_nHint = 0;
|
|
pImportName->m_pszName = NULL;
|
|
pImportName->m_pszOrig = NULL;
|
|
|
|
rvaName = pAddrThunk[f].u1.Ordinal;
|
|
if (rvaName & IMAGE_ORDINAL_FLAG) {
|
|
pImportName->m_nOrig = (ULONG)IMAGE_ORDINAL(rvaName);
|
|
pImportName->m_nOrdinal = pImportName->m_nOrig;
|
|
}
|
|
else {
|
|
PIMAGE_IMPORT_BY_NAME pName
|
|
= (PIMAGE_IMPORT_BY_NAME)RvaToVa(rvaName);
|
|
if (pName) {
|
|
pImportName->m_nHint = pName->Hint;
|
|
pImportName->m_pszName = DuplicateString((PCHAR)pName->Name);
|
|
if (pImportName->m_pszName == NULL) {
|
|
goto fail;
|
|
}
|
|
}
|
|
|
|
rvaName = pLookThunk[f].u1.Ordinal;
|
|
if (rvaName & IMAGE_ORDINAL_FLAG) {
|
|
pImportName->m_nOrig = (ULONG)IMAGE_ORDINAL(rvaName);
|
|
pImportName->m_nOrdinal = (ULONG)IMAGE_ORDINAL(rvaName);
|
|
}
|
|
else {
|
|
pName = (PIMAGE_IMPORT_BY_NAME)RvaToVa(rvaName);
|
|
if (pName) {
|
|
pImportName->m_pszOrig
|
|
= DuplicateString((PCHAR)pName->Name);
|
|
if (pImportName->m_pszOrig == NULL) {
|
|
goto fail;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
oidp++;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////// Parse Sections.
|
|
//
|
|
m_nExtraOffset = 0;
|
|
for (n = 0; n < m_NtHeader.FileHeader.NumberOfSections; n++) {
|
|
m_nExtraOffset = Max(m_SectionHeaders[n].PointerToRawData +
|
|
m_SectionHeaders[n].SizeOfRawData,
|
|
m_nExtraOffset);
|
|
|
|
if (strcmp((PCHAR)m_SectionHeaders[n].Name, ".detour") == 0) {
|
|
DETOUR_SECTION_HEADER dh;
|
|
CopyMemory(&dh,
|
|
m_pMap + m_SectionHeaders[n].PointerToRawData,
|
|
sizeof(dh));
|
|
|
|
if (dh.nDataOffset == 0) {
|
|
dh.nDataOffset = dh.cbHeaderSize;
|
|
}
|
|
|
|
cbData = dh.cbDataSize - dh.nDataOffset;
|
|
pbData = (m_pMap +
|
|
m_SectionHeaders[n].PointerToRawData +
|
|
dh.nDataOffset);
|
|
|
|
m_nExtraOffset = Max(m_SectionHeaders[n].PointerToRawData +
|
|
m_SectionHeaders[n].SizeOfRawData,
|
|
m_nExtraOffset);
|
|
|
|
m_NtHeader.FileHeader.NumberOfSections--;
|
|
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress
|
|
= dh.nOriginalImportVirtualAddress;
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size
|
|
= dh.nOriginalImportSize;
|
|
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT].VirtualAddress
|
|
= dh.nOriginalBoundImportVirtualAddress;
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT].Size
|
|
= dh.nOriginalBoundImportSize;
|
|
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IAT].VirtualAddress
|
|
= dh.nOriginalIatVirtualAddress;
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IAT].Size
|
|
= dh.nOriginalIatSize;
|
|
|
|
m_NtHeader.OptionalHeader.CheckSum = 0;
|
|
m_NtHeader.OptionalHeader.SizeOfImage
|
|
= dh.nOriginalSizeOfImage;
|
|
|
|
m_fHadDetourSection = TRUE;
|
|
}
|
|
}
|
|
|
|
m_pImageData = new CImageData(pbData, cbData);
|
|
if (m_pImageData == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
}
|
|
return TRUE;
|
|
|
|
fail:
|
|
return FALSE;
|
|
}
|
|
|
|
static inline BOOL strneq(__in_z PCHAR pszOne, __in_z PCHAR pszTwo)
|
|
{
|
|
if (pszOne == pszTwo) {
|
|
return FALSE;
|
|
}
|
|
if (!pszOne || !pszTwo) {
|
|
return TRUE;
|
|
}
|
|
return (strcmp(pszOne, pszTwo) != 0);
|
|
}
|
|
|
|
BOOL CImage::CheckImportsNeeded(DWORD *pnTables, DWORD *pnThunks, DWORD *pnChars)
|
|
{
|
|
DWORD nTables = 0;
|
|
DWORD nThunks = 0;
|
|
DWORD nChars = 0;
|
|
BOOL fNeedDetourSection = FALSE;
|
|
|
|
for (CImageImportFile *pImportFile = m_pImportFiles;
|
|
pImportFile != NULL; pImportFile = pImportFile->m_pNextFile) {
|
|
|
|
nChars += (int)strlen(pImportFile->m_pszName) + 1;
|
|
nChars += nChars & 1;
|
|
|
|
if (pImportFile->m_fByway) {
|
|
fNeedDetourSection = TRUE;
|
|
nThunks++;
|
|
}
|
|
else {
|
|
if (!fNeedDetourSection &&
|
|
strneq(pImportFile->m_pszName, pImportFile->m_pszOrig)) {
|
|
|
|
fNeedDetourSection = TRUE;
|
|
}
|
|
for (DWORD n = 0; n < pImportFile->m_nImportNames; n++) {
|
|
CImageImportName *pImportName = &pImportFile->m_pImportNames[n];
|
|
|
|
if (!fNeedDetourSection &&
|
|
strneq(pImportName->m_pszName, pImportName->m_pszOrig)) {
|
|
|
|
fNeedDetourSection = TRUE;
|
|
}
|
|
|
|
if (pImportName->m_pszName) {
|
|
nChars += sizeof(WORD); // Hint
|
|
nChars += (int)strlen(pImportName->m_pszName) + 1;
|
|
nChars += nChars & 1;
|
|
}
|
|
nThunks++;
|
|
}
|
|
}
|
|
nThunks++;
|
|
nTables++;
|
|
}
|
|
nTables++;
|
|
|
|
*pnTables = nTables;
|
|
*pnThunks = nThunks;
|
|
*pnChars = nChars;
|
|
|
|
return fNeedDetourSection;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
CImageImportFile * CImage::NewByway(__in_z PCHAR pszName)
|
|
{
|
|
CImageImportFile *pImportFile = new CImageImportFile;
|
|
if (pImportFile == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
goto fail;
|
|
}
|
|
|
|
pImportFile->m_pNextFile = NULL;
|
|
pImportFile->m_fByway = TRUE;
|
|
|
|
pImportFile->m_pszName = DuplicateString(pszName);
|
|
if (pImportFile->m_pszName == NULL) {
|
|
goto fail;
|
|
}
|
|
|
|
pImportFile->m_rvaOriginalFirstThunk = 0;
|
|
pImportFile->m_rvaFirstThunk = 0;
|
|
pImportFile->m_nForwarderChain = (UINT)0;
|
|
pImportFile->m_pImportNames = NULL;
|
|
pImportFile->m_nImportNames = 0;
|
|
|
|
m_nImportFiles++;
|
|
return pImportFile;
|
|
|
|
fail:
|
|
if (pImportFile) {
|
|
delete pImportFile;
|
|
pImportFile = NULL;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
BOOL CImage::EditImports(PVOID pContext,
|
|
PF_DETOUR_BINARY_BYWAY_CALLBACK pfBywayCallback,
|
|
PF_DETOUR_BINARY_FILE_CALLBACK pfFileCallback,
|
|
PF_DETOUR_BINARY_SYMBOL_CALLBACK pfSymbolCallback,
|
|
PF_DETOUR_BINARY_COMMIT_CALLBACK pfCommitCallback)
|
|
{
|
|
CImageImportFile *pImportFile = NULL;
|
|
CImageImportFile **ppLastFile = &m_pImportFiles;
|
|
|
|
SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
|
|
|
|
while ((pImportFile = *ppLastFile) != NULL) {
|
|
|
|
if (pfBywayCallback) {
|
|
PCHAR pszFile = NULL;
|
|
if (!(*pfBywayCallback)(pContext, pszFile, &pszFile)) {
|
|
goto fail;
|
|
}
|
|
|
|
if (pszFile) {
|
|
// Insert a new Byway.
|
|
CImageImportFile *pByway = NewByway(pszFile);
|
|
if (pByway == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
pByway->m_pNextFile = pImportFile;
|
|
*ppLastFile = pByway;
|
|
ppLastFile = &pByway->m_pNextFile;
|
|
continue; // Retry after Byway.
|
|
}
|
|
}
|
|
|
|
if (pImportFile->m_fByway) {
|
|
if (pfBywayCallback) {
|
|
PCHAR pszFile = pImportFile->m_pszName;
|
|
|
|
if (!(*pfBywayCallback)(pContext, pszFile, &pszFile)) {
|
|
goto fail;
|
|
}
|
|
|
|
if (pszFile) { // Replace? Byway
|
|
if (ReplaceString(&pImportFile->m_pszName, pszFile) == NULL) {
|
|
goto fail;
|
|
}
|
|
}
|
|
else { // Delete Byway
|
|
*ppLastFile = pImportFile->m_pNextFile;
|
|
pImportFile->m_pNextFile = NULL;
|
|
delete pImportFile;
|
|
pImportFile = *ppLastFile;
|
|
m_nImportFiles--;
|
|
continue; // Retry after delete.
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
if (pfFileCallback) {
|
|
PCHAR pszFile = pImportFile->m_pszName;
|
|
|
|
if (!(*pfFileCallback)(pContext, pImportFile->m_pszOrig,
|
|
pszFile, &pszFile)) {
|
|
goto fail;
|
|
}
|
|
|
|
if (pszFile != NULL) {
|
|
if (ReplaceString(&pImportFile->m_pszName, pszFile) == NULL) {
|
|
goto fail;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (pfSymbolCallback) {
|
|
for (DWORD n = 0; n < pImportFile->m_nImportNames; n++) {
|
|
CImageImportName *pImportName = &pImportFile->m_pImportNames[n];
|
|
|
|
PCHAR pszName = pImportName->m_pszName;
|
|
ULONG nOrdinal = pImportName->m_nOrdinal;
|
|
if (!(*pfSymbolCallback)(pContext,
|
|
pImportName->m_nOrig,
|
|
nOrdinal,
|
|
&nOrdinal,
|
|
pImportName->m_pszOrig,
|
|
pszName,
|
|
&pszName)) {
|
|
goto fail;
|
|
}
|
|
|
|
if (pszName != NULL) {
|
|
pImportName->m_nOrdinal = 0;
|
|
if (ReplaceString(&pImportName->m_pszName, pszName) == NULL) {
|
|
goto fail;
|
|
}
|
|
}
|
|
else if (nOrdinal != 0) {
|
|
pImportName->m_nOrdinal = nOrdinal;
|
|
|
|
if (pImportName->m_pszName != NULL) {
|
|
delete[] pImportName->m_pszName;
|
|
pImportName->m_pszName = NULL;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
ppLastFile = &pImportFile->m_pNextFile;
|
|
pImportFile = pImportFile->m_pNextFile;
|
|
}
|
|
|
|
for (;;) {
|
|
if (pfBywayCallback) {
|
|
PCHAR pszFile = NULL;
|
|
if (!(*pfBywayCallback)(pContext, NULL, &pszFile)) {
|
|
goto fail;
|
|
}
|
|
if (pszFile) {
|
|
// Insert a new Byway.
|
|
CImageImportFile *pByway = NewByway(pszFile);
|
|
if (pByway == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
pByway->m_pNextFile = pImportFile;
|
|
*ppLastFile = pByway;
|
|
ppLastFile = &pByway->m_pNextFile;
|
|
continue; // Retry after Byway.
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (pfCommitCallback) {
|
|
if (!(*pfCommitCallback)(pContext)) {
|
|
goto fail;
|
|
}
|
|
}
|
|
|
|
SetLastError(NO_ERROR);
|
|
return TRUE;
|
|
|
|
fail:
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL CImage::Write(HANDLE hFile)
|
|
{
|
|
DWORD cbDone;
|
|
|
|
if (hFile == INVALID_HANDLE_VALUE) {
|
|
SetLastError(ERROR_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
m_nNextFileAddr = 0;
|
|
m_nNextVirtAddr = 0;
|
|
|
|
DWORD nTables = 0;
|
|
DWORD nThunks = 0;
|
|
DWORD nChars = 0;
|
|
BOOL fNeedDetourSection = CheckImportsNeeded(&nTables, &nThunks, &nChars);
|
|
|
|
//////////////////////////////////////////////////////////// Copy Headers.
|
|
//
|
|
if (SetFilePointer(hFile, 0, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!CopyFileData(hFile, 0, m_NtHeader.OptionalHeader.SizeOfHeaders)) {
|
|
return FALSE;
|
|
}
|
|
|
|
if (fNeedDetourSection || !m_pImageData->IsEmpty()) {
|
|
// Replace the file's DOS header with our own.
|
|
m_nPeOffset = sizeof(m_DosHeader) + sizeof(s_rbDosCode);
|
|
m_nSectionsOffset = m_nPeOffset
|
|
+ sizeof(m_NtHeader.Signature)
|
|
+ sizeof(m_NtHeader.FileHeader)
|
|
+ m_NtHeader.FileHeader.SizeOfOptionalHeader;
|
|
m_DosHeader.e_lfanew = m_nPeOffset;
|
|
|
|
if (SetFilePointer(hFile, 0, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!WriteFile(hFile, &m_DosHeader, sizeof(m_DosHeader), &cbDone)) {
|
|
return FALSE;
|
|
}
|
|
if (!WriteFile(hFile, &s_rbDosCode, sizeof(s_rbDosCode), &cbDone)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
else {
|
|
// Restore the file's original DOS header.
|
|
if (m_nPrePE != 0) {
|
|
m_nPeOffset = m_cbPrePE;
|
|
m_nSectionsOffset = m_nPeOffset
|
|
+ sizeof(m_NtHeader.Signature)
|
|
+ sizeof(m_NtHeader.FileHeader)
|
|
+ m_NtHeader.FileHeader.SizeOfOptionalHeader;
|
|
m_DosHeader.e_lfanew = m_nPeOffset;
|
|
|
|
|
|
if (SetFilePointer(hFile, 0, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!CopyFileData(hFile, m_nPrePE, m_cbPrePE)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
m_nNextFileAddr = m_NtHeader.OptionalHeader.SizeOfHeaders;
|
|
m_nNextVirtAddr = 0;
|
|
if (!AlignFileData(hFile)) {
|
|
return FALSE;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////// Copy Sections.
|
|
//
|
|
DWORD n = 0;
|
|
for (; n < m_NtHeader.FileHeader.NumberOfSections; n++) {
|
|
if (m_SectionHeaders[n].SizeOfRawData) {
|
|
if (SetFilePointer(hFile,
|
|
m_SectionHeaders[n].PointerToRawData,
|
|
NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!CopyFileData(hFile,
|
|
m_SectionHeaders[n].PointerToRawData,
|
|
m_SectionHeaders[n].SizeOfRawData)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
m_nNextFileAddr = Max(m_SectionHeaders[n].PointerToRawData +
|
|
m_SectionHeaders[n].SizeOfRawData,
|
|
m_nNextFileAddr);
|
|
m_nNextVirtAddr = Max(m_SectionHeaders[n].VirtualAddress +
|
|
m_SectionHeaders[n].Misc.VirtualSize,
|
|
m_nNextVirtAddr);
|
|
m_nExtraOffset = Max(m_nNextFileAddr, m_nExtraOffset);
|
|
|
|
if (!AlignFileData(hFile)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
if (fNeedDetourSection || !m_pImageData->IsEmpty()) {
|
|
////////////////////////////////////////////// Insert .detour Section.
|
|
//
|
|
DWORD nSection = m_NtHeader.FileHeader.NumberOfSections++;
|
|
DETOUR_SECTION_HEADER dh;
|
|
|
|
ZeroMemory(&dh, sizeof(dh));
|
|
ZeroMemory(&m_SectionHeaders[nSection], sizeof(m_SectionHeaders[nSection]));
|
|
|
|
dh.cbHeaderSize = sizeof(DETOUR_SECTION_HEADER);
|
|
dh.nSignature = DETOUR_SECTION_HEADER_SIGNATURE;
|
|
|
|
dh.nOriginalImportVirtualAddress = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress;
|
|
dh.nOriginalImportSize = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size;
|
|
|
|
dh.nOriginalBoundImportVirtualAddress
|
|
= m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT].VirtualAddress;
|
|
dh.nOriginalBoundImportSize = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT].Size;
|
|
|
|
dh.nOriginalIatVirtualAddress = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IAT].VirtualAddress;
|
|
dh.nOriginalIatSize = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IAT].Size;
|
|
|
|
dh.nOriginalSizeOfImage = m_NtHeader.OptionalHeader.SizeOfImage;
|
|
|
|
DWORD clrAddr = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR].VirtualAddress;
|
|
DWORD clrSize = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR].Size;
|
|
if (clrAddr && clrSize) {
|
|
PDETOUR_CLR_HEADER pHdr = (PDETOUR_CLR_HEADER)RvaToVa(clrAddr);
|
|
if (pHdr != NULL) {
|
|
DETOUR_CLR_HEADER hdr;
|
|
hdr = *pHdr;
|
|
|
|
dh.nOriginalClrFlags = hdr.Flags;
|
|
}
|
|
}
|
|
|
|
HRESULT hrRet = StringCchCopyA((PCHAR)m_SectionHeaders[nSection].Name, IMAGE_SIZEOF_SHORT_NAME , ".detour");
|
|
if (FAILED(hrRet))
|
|
return FALSE;
|
|
|
|
m_SectionHeaders[nSection].Characteristics
|
|
= IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE;
|
|
|
|
m_nOutputVirtAddr = m_nNextVirtAddr;
|
|
m_nOutputVirtSize = 0;
|
|
m_nOutputFileAddr = m_nNextFileAddr;
|
|
|
|
dh.nDataOffset = 0; // pbData
|
|
dh.cbDataSize = m_pImageData->m_cbData;
|
|
dh.cbPrePE = m_cbPrePE;
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
//
|
|
|
|
DWORD rvaImportTable = 0;
|
|
DWORD rvaLookupTable = 0;
|
|
DWORD rvaBoundTable = 0;
|
|
DWORD rvaNameTable = 0;
|
|
DWORD nImportTableSize = nTables * sizeof(IMAGE_IMPORT_DESCRIPTOR);
|
|
|
|
if (!SizeOutputBuffer(QuadAlign(sizeof(dh))
|
|
+ m_cbPrePE
|
|
+ QuadAlign(m_pImageData->m_cbData)
|
|
+ QuadAlign(sizeof(IMAGE_THUNK_DATA) * nThunks)
|
|
+ QuadAlign(sizeof(IMAGE_THUNK_DATA) * nThunks)
|
|
+ QuadAlign(nChars)
|
|
+ QuadAlign(nImportTableSize))) {
|
|
return FALSE;
|
|
}
|
|
|
|
DWORD vaHead = 0;
|
|
PBYTE pbHead = NULL;
|
|
DWORD vaPrePE = 0;
|
|
PBYTE pbPrePE = NULL;
|
|
DWORD vaData = 0;
|
|
PBYTE pbData = NULL;
|
|
|
|
if ((pbHead = AllocateOutput(sizeof(dh), &vaHead)) == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
if ((pbPrePE = AllocateOutput(m_cbPrePE, &vaPrePE)) == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
CImageThunks lookupTable(this, nThunks, &rvaLookupTable);
|
|
CImageThunks boundTable(this, nThunks, &rvaBoundTable);
|
|
CImageChars nameTable(this, nChars, &rvaNameTable);
|
|
|
|
if ((pbData = AllocateOutput(m_pImageData->m_cbData, &vaData)) == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
dh.nDataOffset = vaData - vaHead;
|
|
dh.cbDataSize = dh.nDataOffset + m_pImageData->m_cbData;
|
|
CopyMemory(pbHead, &dh, sizeof(dh));
|
|
CopyMemory(pbPrePE, m_pMap + m_nPrePE, m_cbPrePE);
|
|
CopyMemory(pbData, m_pImageData->m_pbData, m_pImageData->m_cbData);
|
|
|
|
PIMAGE_IMPORT_DESCRIPTOR piidDst = (PIMAGE_IMPORT_DESCRIPTOR)
|
|
AllocateOutput(nImportTableSize, &rvaImportTable);
|
|
if (piidDst == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
//////////////////////////////////////////////// Step Through Imports.
|
|
//
|
|
for (CImageImportFile *pImportFile = m_pImportFiles;
|
|
pImportFile != NULL; pImportFile = pImportFile->m_pNextFile) {
|
|
|
|
ZeroMemory(piidDst, sizeof(piidDst));
|
|
nameTable.Allocate(pImportFile->m_pszName, (DWORD *)&piidDst->Name);
|
|
piidDst->TimeDateStamp = 0;
|
|
piidDst->ForwarderChain = pImportFile->m_nForwarderChain;
|
|
|
|
if (pImportFile->m_fByway) {
|
|
ULONG rvaIgnored;
|
|
|
|
lookupTable.Allocate(IMAGE_ORDINAL_FLAG+1,
|
|
(DWORD *)&piidDst->OriginalFirstThunk);
|
|
boundTable.Allocate(IMAGE_ORDINAL_FLAG+1,
|
|
(DWORD *)&piidDst->FirstThunk);
|
|
|
|
lookupTable.Allocate(0, &rvaIgnored);
|
|
boundTable.Allocate(0, &rvaIgnored);
|
|
}
|
|
else {
|
|
ULONG rvaIgnored;
|
|
|
|
piidDst->FirstThunk = (ULONG)pImportFile->m_rvaFirstThunk;
|
|
lookupTable.Current((DWORD *)&piidDst->OriginalFirstThunk);
|
|
|
|
for (n = 0; n < pImportFile->m_nImportNames; n++) {
|
|
CImageImportName *pImportName = &pImportFile->m_pImportNames[n];
|
|
|
|
if (pImportName->m_pszName) {
|
|
ULONG nDstName = 0;
|
|
|
|
nameTable.Allocate(pImportName->m_pszName,
|
|
pImportName->m_nHint,
|
|
&nDstName);
|
|
lookupTable.Allocate(nDstName, &rvaIgnored);
|
|
}
|
|
else {
|
|
lookupTable.Allocate(IMAGE_ORDINAL_FLAG + pImportName->m_nOrdinal,
|
|
&rvaIgnored);
|
|
}
|
|
}
|
|
lookupTable.Allocate(0, &rvaIgnored);
|
|
}
|
|
piidDst++;
|
|
}
|
|
ZeroMemory(piidDst, sizeof(piidDst));
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
//
|
|
m_nNextVirtAddr += m_nOutputVirtSize;
|
|
m_nNextFileAddr += FileAlign(m_nOutputVirtSize);
|
|
|
|
if (!AlignFileData(hFile)) {
|
|
return FALSE;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
//
|
|
m_SectionHeaders[nSection].VirtualAddress = m_nOutputVirtAddr;
|
|
m_SectionHeaders[nSection].Misc.VirtualSize = m_nOutputVirtSize;
|
|
m_SectionHeaders[nSection].PointerToRawData = m_nOutputFileAddr;
|
|
m_SectionHeaders[nSection].SizeOfRawData = FileAlign(m_nOutputVirtSize);
|
|
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress
|
|
= rvaImportTable;
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size
|
|
= nImportTableSize;
|
|
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT].VirtualAddress = 0;
|
|
m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT].Size = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
//
|
|
if (SetFilePointer(hFile, m_SectionHeaders[nSection].PointerToRawData,
|
|
NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!WriteFile(hFile, m_pbOutputBuffer, m_SectionHeaders[nSection].SizeOfRawData,
|
|
&cbDone)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
///////////////////////////////////////////////////// Adjust Extra Data.
|
|
//
|
|
LONG nExtraAdjust = m_nNextFileAddr - m_nExtraOffset;
|
|
for (n = 0; n < m_NtHeader.FileHeader.NumberOfSections; n++) {
|
|
if (m_SectionHeaders[n].PointerToRawData > m_nExtraOffset) {
|
|
m_SectionHeaders[n].PointerToRawData += nExtraAdjust;
|
|
}
|
|
if (m_SectionHeaders[n].PointerToRelocations > m_nExtraOffset) {
|
|
m_SectionHeaders[n].PointerToRelocations += nExtraAdjust;
|
|
}
|
|
if (m_SectionHeaders[n].PointerToLinenumbers > m_nExtraOffset) {
|
|
m_SectionHeaders[n].PointerToLinenumbers += nExtraAdjust;
|
|
}
|
|
}
|
|
if (m_NtHeader.FileHeader.PointerToSymbolTable > m_nExtraOffset) {
|
|
m_NtHeader.FileHeader.PointerToSymbolTable += nExtraAdjust;
|
|
}
|
|
|
|
m_NtHeader.OptionalHeader.CheckSum = 0;
|
|
m_NtHeader.OptionalHeader.SizeOfImage = m_nNextVirtAddr;
|
|
|
|
////////////////////////////////////////////////// Adjust Debug Directory.
|
|
//
|
|
DWORD debugAddr = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress;
|
|
DWORD debugSize = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size;
|
|
if (debugAddr && debugSize) {
|
|
DWORD nFileOffset = RvaToFileOffset(debugAddr);
|
|
if (SetFilePointer(hFile, nFileOffset, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
|
|
PIMAGE_DEBUG_DIRECTORY pDir = (PIMAGE_DEBUG_DIRECTORY)RvaToVa(debugAddr);
|
|
if (pDir == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
DWORD nEntries = debugSize / sizeof(*pDir);
|
|
for (n = 0; n < nEntries; n++) {
|
|
IMAGE_DEBUG_DIRECTORY dir = pDir[n];
|
|
|
|
if (dir.PointerToRawData > m_nExtraOffset) {
|
|
dir.PointerToRawData += nExtraAdjust;
|
|
}
|
|
if (!WriteFile(hFile, &dir, sizeof(dir), &cbDone)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////// Adjust CLR Header.
|
|
//
|
|
DWORD clrAddr = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR].VirtualAddress;
|
|
DWORD clrSize = m_NtHeader.OptionalHeader
|
|
.DataDirectory[IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR].Size;
|
|
if (clrAddr && clrSize && fNeedDetourSection) {
|
|
DWORD nFileOffset = RvaToFileOffset(clrAddr);
|
|
if (SetFilePointer(hFile, nFileOffset, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
|
|
PDETOUR_CLR_HEADER pHdr = (PDETOUR_CLR_HEADER)RvaToVa(clrAddr);
|
|
if (pHdr == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
DETOUR_CLR_HEADER hdr;
|
|
hdr = *pHdr;
|
|
hdr.Flags &= 0xfffffffe; // Clear the IL_ONLY flag.
|
|
|
|
if (!WriteFile(hFile, &hdr, sizeof(hdr), &cbDone)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
///////////////////////////////////////////////// Copy Left-over Data.
|
|
//
|
|
if (m_nFileSize > m_nExtraOffset) {
|
|
if (SetFilePointer(hFile, m_nNextFileAddr, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!CopyFileData(hFile, m_nExtraOffset, m_nFileSize - m_nExtraOffset)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
|
|
//////////////////////////////////////////////////// Finalize Headers.
|
|
//
|
|
|
|
if (SetFilePointer(hFile, m_nPeOffset, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!WriteFile(hFile, &m_NtHeader, sizeof(m_NtHeader), &cbDone)) {
|
|
return FALSE;
|
|
}
|
|
|
|
if (SetFilePointer(hFile, m_nSectionsOffset, NULL, FILE_BEGIN) == ~0u) {
|
|
return FALSE;
|
|
}
|
|
if (!WriteFile(hFile, &m_SectionHeaders,
|
|
sizeof(m_SectionHeaders[0])
|
|
* m_NtHeader.FileHeader.NumberOfSections,
|
|
&cbDone)) {
|
|
return FALSE;
|
|
}
|
|
|
|
m_cbPostPE = SetFilePointer(hFile, 0, NULL, FILE_CURRENT);
|
|
if (m_cbPostPE == ~0u) {
|
|
return FALSE;
|
|
}
|
|
m_cbPostPE = m_NtHeader.OptionalHeader.SizeOfHeaders - m_cbPostPE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
}; // namespace Detour
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
PDETOUR_BINARY WINAPI DetourBinaryOpen(HANDLE hFile)
|
|
{
|
|
Detour::CImage *pImage = new Detour::CImage;
|
|
if (pImage == NULL) {
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
return FALSE;
|
|
}
|
|
|
|
if (!pImage->Read(hFile)) {
|
|
delete pImage;
|
|
return FALSE;
|
|
}
|
|
|
|
return (PDETOUR_BINARY)pImage;
|
|
}
|
|
|
|
BOOL WINAPI DetourBinaryWrite(PDETOUR_BINARY pdi, HANDLE hFile)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->Write(hFile);
|
|
}
|
|
|
|
PVOID WINAPI DetourBinaryEnumeratePayloads(PDETOUR_BINARY pdi,
|
|
GUID *pGuid,
|
|
DWORD *pcbData,
|
|
DWORD *pnIterator)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->DataEnum(pGuid, pcbData, pnIterator);
|
|
}
|
|
|
|
PVOID WINAPI DetourBinaryFindPayload(PDETOUR_BINARY pdi,
|
|
REFGUID rguid,
|
|
DWORD *pcbData)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->DataFind(rguid, pcbData);
|
|
}
|
|
|
|
PVOID WINAPI DetourBinarySetPayload(PDETOUR_BINARY pdi,
|
|
REFGUID rguid,
|
|
PVOID pvData,
|
|
DWORD cbData)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->DataSet(rguid, (PBYTE)pvData, cbData);
|
|
}
|
|
|
|
BOOL WINAPI DetourBinaryDeletePayload(PDETOUR_BINARY pdi,
|
|
REFGUID rguid)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->DataDelete(rguid);
|
|
}
|
|
|
|
BOOL WINAPI DetourBinaryPurgePayloads(PDETOUR_BINARY pdi)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->DataPurge();
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
static BOOL CALLBACK ResetBywayCallback(PVOID pContext,
|
|
__in_z PCHAR pszFile,
|
|
__deref PCHAR *ppszOutFile)
|
|
{
|
|
(void)pContext;
|
|
(void)pszFile;
|
|
|
|
*ppszOutFile = NULL;
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL CALLBACK ResetFileCallback(PVOID pContext,
|
|
__in_z PCHAR pszOrigFile,
|
|
__in_z PCHAR pszFile,
|
|
__deref PCHAR *ppszOutFile)
|
|
{
|
|
(void)pContext;
|
|
(void)pszFile;
|
|
|
|
*ppszOutFile = pszOrigFile;
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL CALLBACK ResetSymbolCallback(PVOID pContext,
|
|
ULONG nOrigOrdinal,
|
|
ULONG nOrdinal,
|
|
ULONG *pnOutOrdinal,
|
|
__in_z PCHAR pszOrigSymbol,
|
|
__in_z PCHAR pszSymbol,
|
|
__deref PCHAR *ppszOutSymbol)
|
|
{
|
|
(void)pContext;
|
|
(void)nOrdinal;
|
|
(void)pszSymbol;
|
|
|
|
*pnOutOrdinal = nOrigOrdinal;
|
|
*ppszOutSymbol = pszOrigSymbol;
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL WINAPI DetourBinaryResetImports(PDETOUR_BINARY pdi)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->EditImports(NULL,
|
|
ResetBywayCallback,
|
|
ResetFileCallback,
|
|
ResetSymbolCallback,
|
|
NULL);
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
BOOL WINAPI DetourBinaryEditImports(PDETOUR_BINARY pdi,
|
|
PVOID pContext,
|
|
PF_DETOUR_BINARY_BYWAY_CALLBACK pfBywayCallback,
|
|
PF_DETOUR_BINARY_FILE_CALLBACK pfFileCallback,
|
|
PF_DETOUR_BINARY_SYMBOL_CALLBACK pfSymbolCallback,
|
|
PF_DETOUR_BINARY_COMMIT_CALLBACK pfCommitCallback)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return pImage->EditImports(pContext,
|
|
pfBywayCallback,
|
|
pfFileCallback,
|
|
pfSymbolCallback,
|
|
pfCommitCallback);
|
|
}
|
|
|
|
BOOL WINAPI DetourBinaryClose(PDETOUR_BINARY pdi)
|
|
{
|
|
Detour::CImage *pImage = Detour::CImage::IsValid(pdi);
|
|
if (pImage == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL bSuccess = pImage->Close();
|
|
delete pImage;
|
|
pImage = NULL;
|
|
|
|
return bSuccess;
|
|
}
|
|
|
|
//
|
|
///////////////////////////////////////////////////////////////// End of File.
|