Mercurial > emacs
view nt/preprep.c @ 72917:17942cb3949e
(allout-regexp, allout-line-boundary-regexp)
(allout-bob-regexp): Correct grouping and boundaries to fix backwards traversal.
(allout-depth-specific-regexp, allout-depth-one-regexp): New versions that
exploit \\{M\\} regexp syntax, to avoid geometric or worse time in
allout-ascend.
(allout-doublecheck-at-and-shallower): Identify depth threshold below which
topics are checked for and disqualified by containment discontinuities.
(allout-hotspot-key-handler): Correctly handle multiple-key strokes. Remove
some unused variables.
(allout-mode-leaders): Clarify that mode-specific comment-start will be used
(set-allout-regexp): Correctly regexp-quote allout regexps to properly accept
alternative header-leads and primary bullets with regexp-specific characters
(eg, C "/*", mathematica "(*").
Include new regular expressions among those configured.
(allout-infer-header-lead-and-primary-bullet): Rename allout-infer-header-lead.
(allout-recent-depth): Manifest as a variable as well as a function.
(allout-prefix-data): Simplify into an inline instead of a macro, assuming
current match data rather than being explicitly passed it. Establish
allout-recent-depth value as well as allout-recent-prefix-beginning and
allout-recent-prefix-end.
(allout-aberrant-container-p): True when an item's immediate offspring
discontinuously contained. Useful for disqualifying unintended topic prefixes,
likely at low depths.
(allout-goto-prefix-doublechecked): Elaborated version of allout-goto-prefix
which disqualifies aberrant pseudo-items.
(allout-pre-next-prefix): Layer on top of lower-level routines, to get
disqualification of aberrant containers.
(allout-end-of-prefix, allout-end-of-subtree): Disqualify aberrant containers.
(allout-beginning-of-current-entry): Position at start of buffer when in
container (depth 0) entry.
(nullify-allout-prefix-data): Invalidate allout-recent-* prefix data.
(allout-current-bullet): Strip text properties.
(allout-get-prefix-bullet): Use right match groups.
(allout-beginning-of-line, allout-next-heading): Disqualify aberrant containers.
(allout-previous-heading): Disqualify aberrant containers, and change to
regular (rather than inline) function, to allow self-recursion.
(allout-get-invisibility-overlay): Increment so progress is made when the first
overlay is not the sought one.
(allout-end-of-prefix): Disqualify aberrant containers.
(allout-end-of-line): Cycle something like allout-beginning-of-line.
(allout-mode): Make allout-old-style-prefixes (ie, enabling use with outline.el
outlines) functional again. Change the primary bullet along with the
header-lead - level 1 new-style bullets now work.
Engage allout-before-change-handler in mainline emacs, not just xemacs, to do
undo handling.
(allout-before-change-handler): Expose undo changes occurring in hidden
regions. Use allout-get-invisibility-overlay instead of reimplementing it
inline.
(allout-chart-subtree): Use start rather than end of prefix in charts. Use
allout-recent-depth variable.
(allout-chart-siblings): Disqualify aberrant topics.
(allout-beginning-of-current-entry): Position correctly.
(allout-ascend): Use new allout-depth-specific-regexp and
allout-depth-one-regexp for linear instead of O(N^2) or worse behavior.
(allout-ascend-to-depth): Depend on allout-ascend, rather than reimplementing
an algorithm.
(allout-up-current-level): Depend on allout-ascend, rather than reimplementing
an algorithm. Return to start-point if we fail.
(allout-descend-to-depth): Use allout-recent-depth variable instead of function.
(allout-next-sibling): On traversal of numerous intervening topics, resort to
economical allout-next-sibling-leap.
(allout-next-sibling-leap): Specialized version of allout-next-sibling that
uses allout-ascend cleverly, to depend on a regexp search to leap large numbers
of contained topics, rather than arbitrarily many one-by-one traversals.
(allout-next-visible-heading): Disqualify aberrant topics.
(allout-previous-visible-heading): Position consistently when interactive.
(allout-forward-current-level): Base on allout-previous-sibling rather than
(differently) reimplmenting the algorithm. Remove some unused variables.
(allout-solicit-alternate-bullet): Present default choice stripped of text
properties.
(allout-rebullet-heading): Use bullet stripped of text properties.
Register changes using allout-exposure-change-hook. Disregard aberrant topics.
(allout-shift-in): With universal-argument, make topic a peer of it's former
offspring. Simplify the code by separating out allout-shift-out functionality.
(allout-shift-out): With universal-argument, make offspring peers of their
former container, and its siblings. Implement the functionality here, rather
than inappropriately muddling the implementation of allout-shift-in.
(allout-rebullet-topic): Respect additional argument for new parent-child
separation function.
(allout-yank-processing): Use allout-ascend directly.
(allout-show-entry): Disqualify aberrant topics.
(allout-show-children): Handle discontinuous children gracefully, extending the
depth being revealed to expose them and posting a message indicating the
situation.
(allout-show-to-offshoot): Remove obsolete and incorrect comment. Leave cursor
in correct position.
(allout-hide-current-subtree): Use allout-ascend directly. Disqualify aberrant
topics.
(allout-kill-line, allout-kill-topic): Preserve exposure layout in a way that
the yanks can restore it, as used to happen.
(allout-yank-processing): Restore exposure layout as recorded by allout-kill-*,
as used to happen.
(allout-annotate-hidden, allout-hide-by-annotation): New routines for preseving
and restoring exposure layout across kills.
(allout-toggle-subtree-encryption): Run allout-exposure-change-hook.
(allout-encrypt-string): Strip text properties. Rearranged order and
outline-headings for some of the miscellaneous functions.
(allout-resolve-xref): No need to quote the error name in the condition-case
handler section.
(allout-flatten): Classic recursive (and recursively intensive, without
tail-recursion) list-flattener, needed by allout-shift-out when confronted with
discontinuous children.
author | Eli Zaretskii <eliz@gnu.org> |
---|---|
date | Sat, 16 Sep 2006 10:24:24 +0000 |
parents | 067115a6e738 |
children | 4ad431d8e164 c5406394f567 |
line wrap: on
line source
/* Pro-process emacs.exe for profiling by MSVC. Copyright (C) 1999, 2002, 2003, 2004, 2005, 2006 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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> #ifdef __GNUC__ #define _ANONYMOUS_UNION #define _ANONYMOUS_STRUCT #endif #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) \ ((unsigned 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 */ /* arch-tag: 144ca747-168e-43a0-9736-3f4c0ba1657f (do not change this comment) */