#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <time.h>
#include <windows.h>
PIMAGE_NT_HEADERS
(__stdcall * pfnCheckSumMappedFile) (LPVOID BaseAddress,
DWORD FileLength,
LPDWORD HeaderSum,
LPDWORD CheckSum);
#undef min
#undef max
#define min(x, y) (((x) < (y)) ? (x) : (y))
#define max(x, y) (((x) > (y)) ? (x) : (y))
typedef struct file_data {
char *name;
unsigned long size;
HANDLE file;
HANDLE file_mapping;
unsigned char *file_base;
} file_data;
int
open_input_file (file_data *p_file, char *filename)
{
HANDLE file;
HANDLE file_mapping;
void *file_base;
unsigned long size, upper_size;
file = CreateFile (filename, GENERIC_READ, FILE_SHARE_READ, NULL,
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if (file == INVALID_HANDLE_VALUE)
return FALSE;
size = GetFileSize (file, &upper_size);
file_mapping = CreateFileMapping (file, NULL, PAGE_READONLY,
0, size, NULL);
if (!file_mapping)
return FALSE;
file_base = MapViewOfFile (file_mapping, FILE_MAP_READ, 0, 0, size);
if (file_base == 0)
return FALSE;
p_file->name = filename;
p_file->size = size;
p_file->file = file;
p_file->file_mapping = file_mapping;
p_file->file_base = file_base;
return TRUE;
}
int
open_output_file (file_data *p_file, char *filename, unsigned long size)
{
HANDLE file;
HANDLE file_mapping;
void *file_base;
file = CreateFile (filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
if (file == INVALID_HANDLE_VALUE)
return FALSE;
file_mapping = CreateFileMapping (file, NULL, PAGE_READWRITE,
0, size, NULL);
if (!file_mapping)
return FALSE;
file_base = MapViewOfFile (file_mapping, FILE_MAP_WRITE, 0, 0, size);
if (file_base == 0)
return FALSE;
p_file->name = filename;
p_file->size = size;
p_file->file = file;
p_file->file_mapping = file_mapping;
p_file->file_base = file_base;
return TRUE;
}
int
open_inout_file (file_data *p_file, char *filename)
{
HANDLE file;
HANDLE file_mapping;
void *file_base;
unsigned long size, upper_size;
file = CreateFile (filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if (file == INVALID_HANDLE_VALUE)
return FALSE;
size = GetFileSize (file, &upper_size);
file_mapping = CreateFileMapping (file, NULL, PAGE_READWRITE,
0, size, NULL);
if (!file_mapping)
return FALSE;
file_base = MapViewOfFile (file_mapping, FILE_MAP_WRITE, 0, 0, size);
if (file_base == 0)
return FALSE;
p_file->name = filename;
p_file->size = size;
p_file->file = file;
p_file->file_mapping = file_mapping;
p_file->file_base = file_base;
return TRUE;
}
void
close_file_data (file_data *p_file)
{
UnmapViewOfFile (p_file->file_base);
CloseHandle (p_file->file_mapping);
SetFilePointer (p_file->file, p_file->size, NULL, FILE_BEGIN);
SetEndOfFile (p_file->file);
CloseHandle (p_file->file);
}
unsigned long
get_unrounded_section_size (PIMAGE_SECTION_HEADER p_section)
{
return min (p_section->SizeOfRawData,
p_section->Misc.VirtualSize);
}
IMAGE_SECTION_HEADER *
find_section (char * name, IMAGE_NT_HEADERS * nt_header)
{
PIMAGE_SECTION_HEADER section;
int i;
section = IMAGE_FIRST_SECTION (nt_header);
for (i = 0; i < nt_header->FileHeader.NumberOfSections; i++)
{
if (strcmp (section->Name, name) == 0)
return section;
section++;
}
return NULL;
}
IMAGE_SECTION_HEADER *
rva_to_section (DWORD rva, IMAGE_NT_HEADERS * nt_header)
{
PIMAGE_SECTION_HEADER section;
int i;
section = IMAGE_FIRST_SECTION (nt_header);
for (i = 0; i < nt_header->FileHeader.NumberOfSections; i++)
{
DWORD real_size = max (section->SizeOfRawData,
section->Misc.VirtualSize);
if (rva >= section->VirtualAddress
&& rva < section->VirtualAddress + real_size)
return section;
section++;
}
return NULL;
}
IMAGE_SECTION_HEADER *
offset_to_section (DWORD offset, IMAGE_NT_HEADERS * nt_header)
{
PIMAGE_SECTION_HEADER section;
int i;
section = IMAGE_FIRST_SECTION (nt_header);
for (i = 0; i < nt_header->FileHeader.NumberOfSections; i++)
{
if (offset >= section->PointerToRawData
&& offset < section->PointerToRawData + section->SizeOfRawData)
return section;
section++;
}
return NULL;
}
static DWORD
relocate_offset (DWORD offset,
IMAGE_NT_HEADERS * src_nt_header,
IMAGE_NT_HEADERS * dst_nt_header)
{
PIMAGE_SECTION_HEADER src_section = IMAGE_FIRST_SECTION (src_nt_header);
PIMAGE_SECTION_HEADER dst_section = IMAGE_FIRST_SECTION (dst_nt_header);
int i = 0;
while (offset >= src_section->PointerToRawData)
{
if (offset < src_section->PointerToRawData + src_section->SizeOfRawData)
break;
i++;
if (i == src_nt_header->FileHeader.NumberOfSections)
{
dst_section = IMAGE_FIRST_SECTION (dst_nt_header);
dst_section += dst_nt_header->FileHeader.NumberOfSections - 1;
while (dst_section->PointerToRawData == 0)
dst_section--;
while (src_section->PointerToRawData == 0)
src_section--;
return offset
+ (dst_section->PointerToRawData + dst_section->SizeOfRawData)
- (src_section->PointerToRawData + src_section->SizeOfRawData);
}
src_section++;
dst_section++;
}
return offset +
(dst_section->PointerToRawData - src_section->PointerToRawData);
}
#define OFFSET_TO_RVA(offset, section) \
(section->VirtualAddress + ((DWORD)(offset) - section->PointerToRawData))
#define RVA_TO_OFFSET(rva, section) \
(section->PointerToRawData + ((DWORD)(rva) - section->VirtualAddress))
#define RVA_TO_SECTION_OFFSET(rva, section) \
((DWORD)(rva) - section->VirtualAddress)
#define RVA_TO_PTR(var,section,filedata) \
((void *)(RVA_TO_OFFSET(var,section) + (filedata)->file_base))
#define PTR_TO_RVA(ptr) ((DWORD)(ptr) - (DWORD) GetModuleHandle (NULL))
#define PTR_TO_OFFSET(ptr, pfile_data) \
((char *)(ptr) - (pfile_data)->file_base)
#define OFFSET_TO_PTR(offset, pfile_data) \
((pfile_data)->file_base + (DWORD)(offset))
#define ROUND_UP(p, align) (((DWORD)(p) + (align)-1) & ~((align)-1))
#define ROUND_DOWN(p, align) ((DWORD)(p) & ~((align)-1))
static void
copy_executable_and_move_sections (file_data *p_infile,
file_data *p_outfile)
{
unsigned char *dst;
PIMAGE_DOS_HEADER dos_header;
PIMAGE_NT_HEADERS nt_header;
PIMAGE_NT_HEADERS dst_nt_header;
PIMAGE_SECTION_HEADER section;
PIMAGE_SECTION_HEADER dst_section;
PIMAGE_SECTION_HEADER import_section;
PIMAGE_SECTION_HEADER reloc_section;
PIMAGE_DATA_DIRECTORY import_dir;
PIMAGE_DATA_DIRECTORY reloc_dir;
DWORD import_delta_rva;
DWORD reloc_delta_rva;
DWORD offset;
int i;
#define COPY_CHUNK(message, src, size) \
do { \
unsigned char *s = (void *)(src); \
unsigned long count = (size); \
printf ("%s\n", (message)); \
printf ("\t0x%08x Offset in input file.\n", s - p_infile->file_base); \
printf ("\t0x%08x Offset in output file.\n", dst - p_outfile->file_base); \
printf ("\t0x%08x Size in bytes.\n", count); \
memcpy (dst, s, count); \
dst += count; \
} while (0)
#define DST_TO_OFFSET() PTR_TO_OFFSET (dst, p_outfile)
#define ROUND_UP_DST_AND_ZERO(align) \
do { \
unsigned char *newdst = p_outfile->file_base \
+ ROUND_UP (DST_TO_OFFSET (), (align)); \
\
memset (dst, 0, newdst - dst); \
dst = newdst; \
} while (0)
dos_header = (PIMAGE_DOS_HEADER) p_infile->file_base;
nt_header = (PIMAGE_NT_HEADERS) (((unsigned long) dos_header) +
dos_header->e_lfanew);
section = IMAGE_FIRST_SECTION (nt_header);
import_dir = &nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT];
import_section = rva_to_section (import_dir->VirtualAddress, nt_header);
reloc_dir = &nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC];
reloc_section = rva_to_section (reloc_dir->VirtualAddress, nt_header);
if (!reloc_section)
{
printf ("No relocation data, cannot prepare for profile prepping.\n");
exit (1);
}
dst = (unsigned char *) p_outfile->file_base;
COPY_CHUNK ("Copying DOS header...", dos_header,
(DWORD) nt_header - (DWORD) dos_header);
dst_nt_header = (PIMAGE_NT_HEADERS) dst;
COPY_CHUNK ("Copying NT header...", nt_header,
(DWORD) section - (DWORD) nt_header);
dst_section = (PIMAGE_SECTION_HEADER) dst;
COPY_CHUNK ("Copying section table...", section,
nt_header->FileHeader.NumberOfSections * sizeof (*section));
dst += 2 * sizeof (*section);
ROUND_UP_DST_AND_ZERO (dst_nt_header->OptionalHeader.FileAlignment);
dst_nt_header->OptionalHeader.SizeOfHeaders = DST_TO_OFFSET ();
for (i = 0; i < nt_header->FileHeader.NumberOfSections;
i++, section++, dst_section++)
{
char msg[100];
sprintf (msg, "Copying raw data for %s...", section->Name);
if (section == import_section || section == reloc_section)
{
dst_section->Name[0] = 'X';
dst_section->Misc.VirtualSize =
ROUND_UP (dst_section->Misc.VirtualSize,
dst_nt_header->OptionalHeader.SectionAlignment);
dst_section->PointerToRawData = 0;
dst_section->SizeOfRawData = 0;
dst_section->Characteristics &= ~IMAGE_SCN_CNT_INITIALIZED_DATA;
dst_section->Characteristics |= IMAGE_SCN_CNT_UNINITIALIZED_DATA;
dst_section->Characteristics &= ~IMAGE_SCN_MEM_WRITE;
continue;
}
if (dst_section->PointerToRawData)
dst_section->PointerToRawData = DST_TO_OFFSET ();
COPY_CHUNK
(msg, OFFSET_TO_PTR (section->PointerToRawData, p_infile),
section->SizeOfRawData);
dst_section->SizeOfRawData =
ROUND_UP (dst_section->SizeOfRawData,
dst_nt_header->OptionalHeader.FileAlignment);
ROUND_UP_DST_AND_ZERO (dst_nt_header->OptionalHeader.FileAlignment);
}
if (import_section != NULL)
{
dst_nt_header->FileHeader.NumberOfSections++;
dst_nt_header->OptionalHeader.SizeOfImage +=
ROUND_UP (import_section->Misc.VirtualSize,
dst_nt_header->OptionalHeader.SectionAlignment);
*dst_section = *import_section;
dst_section->VirtualAddress =
dst_section[-1].VirtualAddress
+ ROUND_UP (dst_section[-1].Misc.VirtualSize,
dst_nt_header->OptionalHeader.SectionAlignment);
dst_section->PointerToRawData = DST_TO_OFFSET ();
import_delta_rva = dst_section->VirtualAddress - import_section->VirtualAddress;
COPY_CHUNK
("Relocating import directory",
OFFSET_TO_PTR (import_section->PointerToRawData, p_infile),
import_section->SizeOfRawData);
ROUND_UP_DST_AND_ZERO (dst_nt_header->OptionalHeader.FileAlignment);
dst_section++;
}
if (reloc_section != NULL)
{
dst_nt_header->FileHeader.NumberOfSections++;
dst_nt_header->OptionalHeader.SizeOfImage +=
ROUND_UP (reloc_section->Misc.VirtualSize,
dst_nt_header->OptionalHeader.SectionAlignment);
*dst_section = *reloc_section;
dst_section->VirtualAddress =
dst_section[-1].VirtualAddress
+ ROUND_UP (dst_section[-1].Misc.VirtualSize,
dst_nt_header->OptionalHeader.SectionAlignment);
dst_section->PointerToRawData = DST_TO_OFFSET ();
reloc_delta_rva = dst_section->VirtualAddress - reloc_section->VirtualAddress;
COPY_CHUNK
("Relocating base relocations directory",
OFFSET_TO_PTR (reloc_section->PointerToRawData, p_infile),
reloc_section->SizeOfRawData);
ROUND_UP_DST_AND_ZERO (dst_nt_header->OptionalHeader.FileAlignment);
reloc_dir = &dst_nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC];
reloc_dir->VirtualAddress += reloc_delta_rva;
dst_section++;
}
section--;
offset = ROUND_UP (section->PointerToRawData + section->SizeOfRawData,
nt_header->OptionalHeader.FileAlignment);
COPY_CHUNK
("Copying remainder of executable...",
OFFSET_TO_PTR (offset, p_infile),
p_infile->size - offset);
p_outfile->size = DST_TO_OFFSET ();
printf ("Patching up raw data offsets...\n");
section = IMAGE_FIRST_SECTION (nt_header);
dst_section = IMAGE_FIRST_SECTION (dst_nt_header);
#define ADJUST_OFFSET(var) \
do { \
if ((var) != 0) \
(var) = relocate_offset ((var), nt_header, dst_nt_header); \
} while (0)
#define ADJUST_IMPORT_RVA(var) \
do { \
if ((var) != 0) \
*((DWORD *)&(var)) += import_delta_rva; \
} while (0)
dst_nt_header->OptionalHeader.SizeOfInitializedData = 0;
dst_nt_header->OptionalHeader.SizeOfUninitializedData = 0;
for (i = 0; i < dst_nt_header->FileHeader.NumberOfSections; i++)
{
if (dst_section[i].Characteristics & IMAGE_SCN_CNT_INITIALIZED_DATA)
dst_nt_header->OptionalHeader.SizeOfInitializedData +=
ROUND_UP (dst_section[i].Misc.VirtualSize, dst_nt_header->OptionalHeader.FileAlignment);
else if (dst_section[i].Characteristics & IMAGE_SCN_CNT_UNINITIALIZED_DATA)
dst_nt_header->OptionalHeader.SizeOfUninitializedData +=
ROUND_UP (dst_section[i].Misc.VirtualSize, dst_nt_header->OptionalHeader.FileAlignment);
ADJUST_OFFSET (dst_section[i].PointerToLinenumbers);
}
ADJUST_OFFSET (dst_nt_header->FileHeader.PointerToSymbolTable);
{
PIMAGE_DATA_DIRECTORY debug_dir =
&dst_nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG];
PIMAGE_DEBUG_DIRECTORY debug_entry;
if (rva_to_section (debug_dir->VirtualAddress, nt_header) == import_section)
debug_dir->VirtualAddress += import_delta_rva;
section = rva_to_section (debug_dir->VirtualAddress, dst_nt_header);
if (section)
{
int size;
debug_entry = RVA_TO_PTR (debug_dir->VirtualAddress, section, p_outfile);
size = debug_dir->Size / sizeof (IMAGE_DEBUG_DIRECTORY);
for (i = 0; i < size; i++, debug_entry++)
{
ADJUST_OFFSET (debug_entry->PointerToRawData);
ADJUST_IMPORT_RVA (debug_entry->AddressOfRawData);
}
}
}
{
PIMAGE_IMPORT_DESCRIPTOR imports;
PIMAGE_THUNK_DATA import_thunks;
import_dir = &dst_nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT];
import_dir->VirtualAddress += import_delta_rva;
section = rva_to_section (import_dir->VirtualAddress, dst_nt_header);
imports = RVA_TO_PTR (import_dir->VirtualAddress, section, p_outfile);
for ( ; imports->Name != 0; imports++)
{
ADJUST_IMPORT_RVA (imports->OriginalFirstThunk);
ADJUST_IMPORT_RVA (imports->FirstThunk);
ADJUST_IMPORT_RVA (imports->Name);
for (import_thunks = RVA_TO_PTR (imports->OriginalFirstThunk, section, p_outfile);
import_thunks->u1.Function != 0;
import_thunks++)
if ((import_thunks->u1.Ordinal >> 31) == 0)
ADJUST_IMPORT_RVA (import_thunks->u1.Ordinal);
for (import_thunks = RVA_TO_PTR (imports->FirstThunk, section, p_outfile);
import_thunks->u1.Function != 0;
import_thunks++)
if ((import_thunks->u1.Ordinal >> 31) == 0)
ADJUST_IMPORT_RVA (import_thunks->u1.Ordinal);
}
import_dir = &dst_nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IAT];
import_dir->VirtualAddress += import_delta_rva;
}
printf ("Applying fixups to import references...\n");
{
IMAGE_BASE_RELOCATION *relocs, *block, *start_block, *end_block;
DWORD import_start = import_section->VirtualAddress + dst_nt_header->OptionalHeader.ImageBase;
DWORD import_end = import_start + import_section->Misc.VirtualSize;
DWORD len_import_relocs;
DWORD len_remaining_relocs;
int seen_high = 0;
WORD * high_word;
void * holder;
reloc_dir = &dst_nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC];
reloc_section = rva_to_section (reloc_dir->VirtualAddress, dst_nt_header);
relocs = RVA_TO_PTR (reloc_dir->VirtualAddress, reloc_section, p_outfile);
for (block = relocs, start_block = 0;
(DWORD) block - (DWORD) relocs < reloc_dir->Size;
block = (void *)((DWORD) block + block->SizeOfBlock))
{
if (block->VirtualAddress >= import_section->VirtualAddress + import_section->Misc.VirtualSize)
{
end_block = block;
break;
}
if (block->VirtualAddress >= import_section->VirtualAddress)
{
if (start_block == 0)
start_block = block;
block->VirtualAddress += import_delta_rva;
}
}
if (start_block)
{
len_import_relocs = (DWORD) end_block - (DWORD) start_block;
len_remaining_relocs = (DWORD) relocs + reloc_dir->Size - (DWORD) end_block;
holder = malloc (len_import_relocs);
if (holder == 0)
abort ();
memcpy (holder, start_block, len_import_relocs);
memcpy (start_block, end_block, len_remaining_relocs);
memcpy ((char *) start_block + len_remaining_relocs, holder, len_import_relocs);
free (holder);
}
for (block = relocs;
(DWORD) block - (DWORD) relocs < reloc_dir->Size;
block = (void *)((DWORD) block + block->SizeOfBlock))
{
DWORD page_rva = block->VirtualAddress;
DWORD page_offset;
union {
WORD word;
DWORD dword;
} * ploc;
WORD *fixup;
section = rva_to_section (page_rva, dst_nt_header);
if (section->Name[0] == 'X')
continue;
for (fixup = (WORD *) &block[1];
(DWORD) fixup - (DWORD) block < block->SizeOfBlock;
fixup++)
{
page_offset = (*fixup) & 0xfff;
ploc = RVA_TO_PTR (page_rva + page_offset, section, p_outfile);
if (seen_high && ((*fixup) >> 12) != IMAGE_REL_BASED_LOW)
abort ();
switch ((*fixup) >> 12)
{
case IMAGE_REL_BASED_ABSOLUTE:
break;
case IMAGE_REL_BASED_HIGH:
seen_high = 1;
high_word = &ploc->word;
break;
case IMAGE_REL_BASED_LOW:
offset = (*high_word << 16) + ploc->word;
if (offset >= import_start && offset < import_end)
{
(*high_word) += import_delta_rva >> 16;
ploc->dword += import_delta_rva & 0xffff;
}
seen_high = 0;
break;
case IMAGE_REL_BASED_HIGHLOW:
if (ploc->dword >= import_start && ploc->dword < import_end)
ploc->dword += import_delta_rva;
break;
case IMAGE_REL_BASED_HIGHADJ:
if (ploc->dword >= import_start && ploc->dword < import_end)
ploc->dword += import_delta_rva;
break;
case IMAGE_REL_BASED_MIPS_JMPADDR:
abort ();
break;
#ifdef IMAGE_REL_BASED_SECTION
case IMAGE_REL_BASED_SECTION:
case IMAGE_REL_BASED_REL32:
#endif
default:
abort ();
}
}
}
}
}
int
main (int argc, char **argv)
{
PIMAGE_DOS_HEADER dos_header;
PIMAGE_NT_HEADERS nt_header;
file_data in_file, out_file;
char out_filename[MAX_PATH], in_filename[MAX_PATH];
char *ptr;
strcpy (in_filename, argv[1]);
strcpy (out_filename, argv[2]);
printf ("Preparing %s for profile prepping\n", out_filename);
if (!open_input_file (&in_file, in_filename))
{
printf ("Failed to open %s (%d)...bailing.\n",
in_filename, GetLastError ());
exit (1);
}
if (!open_output_file (&out_file, out_filename, in_file.size))
{
printf ("Failed to open %s (%d)...bailing.\n",
out_filename, GetLastError ());
exit (1);
}
copy_executable_and_move_sections (&in_file, &out_file);
{
HANDLE hImagehelp = LoadLibrary ("imagehlp.dll");
DWORD headersum;
DWORD checksum;
dos_header = (PIMAGE_DOS_HEADER) out_file.file_base;
nt_header = (PIMAGE_NT_HEADERS) ((char *) dos_header + dos_header->e_lfanew);
nt_header->OptionalHeader.CheckSum = 0;
pfnCheckSumMappedFile = (void *) GetProcAddress (hImagehelp, "CheckSumMappedFile");
if (pfnCheckSumMappedFile)
{
pfnCheckSumMappedFile (out_file.file_base,
out_file.size,
&headersum,
&checksum);
nt_header->OptionalHeader.CheckSum = checksum;
}
FreeLibrary (hImagehelp);
}
close_file_data (&out_file);
close_file_data (&in_file);
return 0;
}