mirror/emacs
1
0
Fork 0
mirror of https://git.savannah.gnu.org/git/emacs.git synced 2024-12-15 09:47:20 +00:00
Code Issues Releases Activity
91a38db1dd
emacs/nt/preprep.c
Andrew Innes f78b1e5e37 Fix typo.
1999-06-03 19:44:06 +00:00

826 lines
28 KiB
C
Raw Blame History

/* Pro-process emacs.exe for profiling by MSVC.
Copyright (C) 1999 Free Software Foundation, Inc.
This file is part of GNU Emacs.
GNU Emacs is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2, or (at your option)
any later version.
GNU Emacs is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with GNU Emacs; see the file COPYING. If not, write to
the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
Boston, MA 02111-1307, USA.
Andrew Innes <andrewi@harlequin.co.uk> 16-Jan-1999
based on code from addsection.c
*/
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <time.h>
#include <windows.h>
/* Include relevant definitions from IMAGEHLP.H, which can be found
in \\win32sdk\mstools\samples\image\include\imagehlp.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))
/* File handling. */
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;
}
/* Close the system structures associated with the given file. */
void
close_file_data (file_data *p_file)
{
UnmapViewOfFile (p_file->file_base);
CloseHandle (p_file->file_mapping);
/* For the case of output files, set final size. */
SetFilePointer (p_file->file, p_file->size, NULL, FILE_BEGIN);
SetEndOfFile (p_file->file);
CloseHandle (p_file->file);
}
/* Routines to manipulate NT executable file sections. */
unsigned long
get_unrounded_section_size (PIMAGE_SECTION_HEADER p_section)
{
/* The true section size, before rounding, for an initialized data or
code section. (Supposedly some linkers swap the meaning of these
two values.) */
return min (p_section->SizeOfRawData,
p_section->Misc.VirtualSize);
}
/* Return pointer to section header for named section. */
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;
}
/* Return pointer to section header for section containing the given
relative virtual address. */
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++)
{
/* Some linkers (eg. the NT SDK linker I believe) swapped the
meaning of these two values - or rather, they ignored
VirtualSize entirely and always set it to zero. This affects
some very old exes (eg. gzip dated Dec 1993). Since
w32_executable_type relies on this function to work reliably,
we need to cope with this. */
DWORD real_size = max (section->SizeOfRawData,
section->Misc.VirtualSize);
if (rva >= section->VirtualAddress
&& rva < section->VirtualAddress + real_size)
return section;
section++;
}
return NULL;
}
/* Return pointer to section header for section containing the given
offset in its raw data area. */
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;
}
/* Return offset to an object in dst, given offset in src. We assume
there is at least one section in both src and dst images, and that
the some sections may have been added to dst (after sections in src). */
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)
{
/* Handle offsets after the last section. */
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))
/* Convert address in executing image to RVA. */
#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))
/* The MSVC prep program generates a ._xe file from .exe, where relevant
function calls etc have been patched to go through thunks (generated
by prep) that record timing/call information. Because the thunks
need to make references to functions imported from profile.dll, the
import table must be expanded; the end result is that all the
sections following .rdata are relocated to higher RVAs (add a final
code section is added holding all the thunks). The .reloc section is
also expanded, so that the thunks themselves are relocatable.
It is this relocation which kills emacs._xe, because the dumped heap
pointers aren't relocated, because there is no relocation data for
either the relevant global/static variables or the heap section
itself, both of which contain pointers into the heap. [Note that
static variables which aren't initialized during linking may become
initialized with heap pointers, or even pointers to other static
variables, because of dumping.]
We could potentially generate the relocation data ourselves by making
two versions of temacs, one with an extra dummmy section before
EMHEAP to offset it, and then compare the dumped executables from
both. That is a lot of work though, and it doesn't solve the problem
of dumped pointers to static variables, which also can be relocated.
A better solution is to pre-process emacs.exe so that the .rdata and
.reloc sections are moved to the end of the section table, and thus
prep won't relocate anything else. (Of course, we leave "dead"
copies of these two sections in place, so that the virtual address of
everything else is unaffected.) Relocating the .reloc data is
trivial - we just update the IMAGE_BASE_RELOCATION address in the
header (the data itself doesn't change). Relocating the import table
is more complicated though, because the calls to imported functions
must be patched up. That requires us to selectively apply the base
relocations when we encounter references to imported functions (or
variables) in other sections, but at least the base relocations are
easy to parse. */
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)); \
/* Zero the alignment slop; it may actually initialize real data. */ \
memset (dst, 0, newdst - dst); \
dst = newdst; \
} while (0)
/* Copy the source image sequentially, ie. section by section after
copying the headers and section table, to simplify the process of
relocating the .rdata and .reloc section table entries (which might
force the raw section data to be relocated).
Note that dst is updated implicitly by each COPY_CHUNK. */
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));
/* Leave room for extra section table entries; filled in below. */
dst += 2 * sizeof (*section);
/* Align the first section's raw data area, and set the header size
field accordingly. */
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);
/* "Blank out" the two sections being relocated. */
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;
}
/* Update the file-relative offset for this section's raw data (if
it has any) in case things have been relocated; we will update
the other offsets below once we know where everything is. */
if (dst_section->PointerToRawData)
dst_section->PointerToRawData = DST_TO_OFFSET ();
/* Copy the original raw data. */
COPY_CHUNK
(msg, OFFSET_TO_PTR (section->PointerToRawData, p_infile),
section->SizeOfRawData);
/* Round up the raw data size to the new alignment. */
dst_section->SizeOfRawData =
ROUND_UP (dst_section->SizeOfRawData,
dst_nt_header->OptionalHeader.FileAlignment);
/* Align the next section's raw data area. */
ROUND_UP_DST_AND_ZERO (dst_nt_header->OptionalHeader.FileAlignment);
}
/* Add the extra section entries, copying the raw data we skipped
earlier. We'll patch up the data itself below. */
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 ();
/* Remember delta applied to import section. */
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 ();
/* Remember delta applied to reloc section. */
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++;
}
/* Copy remainder of source image. */
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);
/* Final size for new image. */
p_outfile->size = DST_TO_OFFSET ();
/* Now patch up remaining file-relative offsets. */
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++)
{
/* Recompute data sizes for completeness. */
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);
/* Update offsets in debug directory entries. Note that the debug
directory may be in the same section as the import table, so its
RVA may need to be adjusted too. */
{
PIMAGE_DATA_DIRECTORY debug_dir =
&dst_nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG];
PIMAGE_DEBUG_DIRECTORY debug_entry;
/* Update debug_dir if part of import_section. */
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++)
{
/* The debug data itself is normally not part of any
section, but stored after all the raw section data. So
let relocate_offset do the work. */
ADJUST_OFFSET (debug_entry->PointerToRawData);
ADJUST_IMPORT_RVA (debug_entry->AddressOfRawData);
}
}
}
/* Update RVAs in import directory entries. */
{
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;
}
/* Fix up references to the import section. */
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);
/* Move the base relocations for the import section, if there are
any; the profiler needs to be able to patch RVAs in the import
section itself. */
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);
}
/* Walk up the list of base relocations, checking for references
to the old import section location, and patching them to
reference the new location. */
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);
/* Don't apply fixups to the blanked sections. */
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);
/* Unless our assumption is wrong, all low word fixups
should immediately follow a high fixup. */
if (seen_high && ((*fixup) >> 12) != IMAGE_REL_BASED_LOW)
abort ();
switch ((*fixup) >> 12)
{
case IMAGE_REL_BASED_ABSOLUTE:
break;
case IMAGE_REL_BASED_HIGH:
/* We must assume that high and low fixups occur in
pairs, specifically a low fixup immediately follows a
high fixup (normally separated by two bytes). We
have to process the two fixups together, to find out
the full pointer value and decide whether to apply
the fixup. */
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:
/* Docs imply two words in big-endian order, so perhaps
this is only used on big-endian platforms, in which
case the obvious code will work. */
if (ploc->dword >= import_start && ploc->dword < import_end)
ploc->dword += import_delta_rva;
break;
case IMAGE_REL_BASED_HIGHADJ:
/* Docs don't say, but I guess this is the equivalent
for little-endian platforms. */
if (ploc->dword >= import_start && ploc->dword < import_end)
ploc->dword += import_delta_rva;
break;
case IMAGE_REL_BASED_MIPS_JMPADDR:
/* Don't know how to handle this; MIPS support has been
dropped from NT4 anyway. */
abort ();
break;
#ifdef IMAGE_REL_BASED_SECTION
case IMAGE_REL_BASED_SECTION:
case IMAGE_REL_BASED_REL32:
/* Docs don't say what these values mean. */
#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);
/* Open the original (dumped) executable file for reference. */
if (!open_input_file (&in_file, in_filename))
{
printf ("Failed to open %s (%d)...bailing.\n",
in_filename, GetLastError ());
exit (1);
}
/* Create a new image that can be prepped; we don't expect the size to
change because we are only adding two new section table entries,
which should fit in the alignment slop. */
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);
/* Patch up header fields; profiler is picky about this. */
{
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;
// nt_header->FileHeader.TimeDateStamp = time (NULL);
// dos_header->e_cp = size / 512;
// nt_header->OptionalHeader.SizeOfImage = size;
pfnCheckSumMappedFile = (void *) GetProcAddress (hImagehelp, "CheckSumMappedFile");
if (pfnCheckSumMappedFile)
{
// nt_header->FileHeader.TimeDateStamp = time (NULL);
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;
}
/* eof */
Reference in New Issue View Git Blame Copy Permalink