preprep.c 27.9 KB
Newer Older
1
/* Pre-process emacs.exe for profiling by MSVC.
Paul Eggert's avatar
Paul Eggert committed
2
   Copyright (C) 1999, 2001-2020 Free Software Foundation, Inc.
Andrew Innes's avatar
Andrew Innes committed
3 4 5

This file is part of GNU Emacs.

6
GNU Emacs is free software: you can redistribute it and/or modify
Andrew Innes's avatar
Andrew Innes committed
7
it under the terms of the GNU General Public License as published by
8 9
the Free Software Foundation, either version 3 of the License, or (at
your option) any later version.
Andrew Innes's avatar
Andrew Innes committed
10 11 12 13 14 15 16

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
Paul Eggert's avatar
Paul Eggert committed
17
along with GNU Emacs.  If not, see <https://www.gnu.org/licenses/>.
18

Andrew Innes's avatar
Andrew Innes committed
19 20 21 22 23 24 25 26 27

   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>
28
#if defined(__GNUC__) && !defined(MINGW_W64)
29 30 31
#define _ANONYMOUS_UNION
#define _ANONYMOUS_STRUCT
#endif
Andrew Innes's avatar
Andrew Innes committed
32 33 34 35 36
#include <windows.h>

/* Include relevant definitions from IMAGEHLP.H, which can be found
   in \\win32sdk\mstools\samples\image\include\imagehlp.h. */

37 38 39 40
PIMAGE_NT_HEADERS (__stdcall * pfnCheckSumMappedFile) (LPVOID BaseAddress,
						       DWORD_PTR FileLength,
						       PDWORD_PTR HeaderSum,
						       PDWORD_PTR CheckSum);
Andrew Innes's avatar
Andrew Innes committed
41 42 43 44 45 46 47 48 49 50

#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 {
51 52 53 54 55
  const char    *name;
  unsigned long  size;
  HANDLE         file;
  HANDLE         file_mapping;
  unsigned char *file_base;
Andrew Innes's avatar
Andrew Innes committed
56 57 58
} file_data;

int
59
open_input_file (file_data *p_file, const char *filename)
Andrew Innes's avatar
Andrew Innes committed
60 61 62 63 64 65 66 67
{
  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);
68
  if (file == INVALID_HANDLE_VALUE)
Andrew Innes's avatar
Andrew Innes committed
69 70 71
    return FALSE;

  size = GetFileSize (file, &upper_size);
72
  file_mapping = CreateFileMapping (file, NULL, PAGE_READONLY,
Andrew Innes's avatar
Andrew Innes committed
73
				    0, size, NULL);
74
  if (!file_mapping)
Andrew Innes's avatar
Andrew Innes committed
75 76 77
    return FALSE;

  file_base = MapViewOfFile (file_mapping, FILE_MAP_READ, 0, 0, size);
78
  if (file_base == 0)
Andrew Innes's avatar
Andrew Innes committed
79 80 81 82 83 84 85 86 87 88 89 90
    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
91
open_output_file (file_data *p_file, const char *filename, unsigned long size)
Andrew Innes's avatar
Andrew Innes committed
92 93 94 95 96 97 98
{
  HANDLE file;
  HANDLE file_mapping;
  void  *file_base;

  file = CreateFile (filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
		     CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
99
  if (file == INVALID_HANDLE_VALUE)
Andrew Innes's avatar
Andrew Innes committed
100 101
    return FALSE;

102
  file_mapping = CreateFileMapping (file, NULL, PAGE_READWRITE,
Andrew Innes's avatar
Andrew Innes committed
103
				    0, size, NULL);
104
  if (!file_mapping)
Andrew Innes's avatar
Andrew Innes committed
105
    return FALSE;
106

Andrew Innes's avatar
Andrew Innes committed
107
  file_base = MapViewOfFile (file_mapping, FILE_MAP_WRITE, 0, 0, size);
108
  if (file_base == 0)
Andrew Innes's avatar
Andrew Innes committed
109
    return FALSE;
110

Andrew Innes's avatar
Andrew Innes committed
111 112 113 114 115 116 117 118 119 120
  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
121
open_inout_file (file_data *p_file, const char *filename)
Andrew Innes's avatar
Andrew Innes committed
122 123 124 125 126 127 128 129
{
  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);
130
  if (file == INVALID_HANDLE_VALUE)
Andrew Innes's avatar
Andrew Innes committed
131 132 133 134 135
    return FALSE;

  size = GetFileSize (file, &upper_size);
  file_mapping = CreateFileMapping (file, NULL, PAGE_READWRITE,
				    0, size, NULL);
136
  if (!file_mapping)
Andrew Innes's avatar
Andrew Innes committed
137 138 139
    return FALSE;

  file_base = MapViewOfFile (file_mapping, FILE_MAP_WRITE, 0, 0, size);
140
  if (file_base == 0)
Andrew Innes's avatar
Andrew Innes committed
141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
    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 *
179
find_section (const char *name, IMAGE_NT_HEADERS *nt_header)
Andrew Innes's avatar
Andrew Innes committed
180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197
{
  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 *
198
rva_to_section (DWORD_PTR rva, IMAGE_NT_HEADERS * nt_header)
Andrew Innes's avatar
Andrew Innes committed
199 200 201 202 203 204 205 206 207 208 209 210 211 212
{
  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.  */
213
      DWORD_PTR real_size = max (section->SizeOfRawData,
Andrew Innes's avatar
Andrew Innes committed
214 215 216 217 218 219 220 221 222 223 224 225
			     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 *
226
offset_to_section (DWORD_PTR offset, IMAGE_NT_HEADERS * nt_header)
Andrew Innes's avatar
Andrew Innes committed
227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245
{
  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).  */
246 247
static DWORD_PTR
relocate_offset (DWORD_PTR offset,
Andrew Innes's avatar
Andrew Innes committed
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280
		 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) \
281
  ((section)->VirtualAddress + ((DWORD_PTR)(offset) - (section)->PointerToRawData))
Andrew Innes's avatar
Andrew Innes committed
282 283

#define RVA_TO_OFFSET(rva, section) \
284
  ((section)->PointerToRawData + ((DWORD_PTR)(rva) - (section)->VirtualAddress))
Andrew Innes's avatar
Andrew Innes committed
285 286

#define RVA_TO_SECTION_OFFSET(rva, section) \
287
  ((DWORD_PTR)(rva) - (section)->VirtualAddress)
Andrew Innes's avatar
Andrew Innes committed
288 289

#define RVA_TO_PTR(var,section,filedata) \
290
  ((void *)((unsigned char *)(RVA_TO_OFFSET(var,section) + (filedata)->file_base)))
Andrew Innes's avatar
Andrew Innes committed
291 292

/* Convert address in executing image to RVA.  */
293
#define PTR_TO_RVA(ptr) ((DWORD_PTR)(ptr) - (DWORD_PTR) GetModuleHandle (NULL))
Andrew Innes's avatar
Andrew Innes committed
294 295

#define PTR_TO_OFFSET(ptr, pfile_data) \
296
          ((unsigned const char *)(ptr) - (pfile_data)->file_base)
Andrew Innes's avatar
Andrew Innes committed
297 298

#define OFFSET_TO_PTR(offset, pfile_data) \
299
          ((pfile_data)->file_base + (DWORD_PTR)(offset))
Andrew Innes's avatar
Andrew Innes committed
300

301 302 303
#define ROUND_UP(p, align) \
	  (((DWORD_PTR)(p) + (align)-1) & ~((DWORD_PTR)(align)-1))
#define ROUND_DOWN(p, align) ((DWORD_PTR)(p) & ~((DWORD_PTR)(align)-1))
Andrew Innes's avatar
Andrew Innes committed
304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323


/* 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
Juanma Barranquero's avatar
Juanma Barranquero committed
324
   two versions of temacs, one with an extra dummy section before
Andrew Innes's avatar
Andrew Innes committed
325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342
   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
343
copy_executable_and_move_sections (file_data *p_infile,
Andrew Innes's avatar
Andrew Innes committed
344 345 346 347 348 349 350 351 352 353 354 355
				   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;
356 357 358
  DWORD_PTR import_delta_rva;
  DWORD_PTR reloc_delta_rva;
  DWORD_PTR offset;
Andrew Innes's avatar
Andrew Innes committed
359 360 361 362
  int i;

#define COPY_CHUNK(message, src, size)						\
  do {										\
363
    unsigned const char *s = (void *)(src);					\
Andrew Innes's avatar
Andrew Innes committed
364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390
    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;
391
  nt_header = (PIMAGE_NT_HEADERS) (((unsigned char *) dos_header) +
Andrew Innes's avatar
Andrew Innes committed
392 393
				   dos_header->e_lfanew);
  section = IMAGE_FIRST_SECTION (nt_header);
394

Andrew Innes's avatar
Andrew Innes committed
395 396 397 398 399 400 401 402 403 404 405 406 407 408
  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,
409
	      (DWORD_PTR) nt_header - (DWORD_PTR) dos_header);
Andrew Innes's avatar
Andrew Innes committed
410 411
  dst_nt_header = (PIMAGE_NT_HEADERS) dst;
  COPY_CHUNK ("Copying NT header...", nt_header,
412
	      (DWORD_PTR) section - (DWORD_PTR) nt_header);
Andrew Innes's avatar
Andrew Innes committed
413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539
  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)				\
540
      *((DWORD_PTR *)&(var)) += import_delta_rva;	\
Andrew Innes's avatar
Andrew Innes committed
541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629
  } 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;
630 631 632 633
    DWORD_PTR import_start = import_section->VirtualAddress + dst_nt_header->OptionalHeader.ImageBase;
    DWORD_PTR import_end = import_start + import_section->Misc.VirtualSize;
    DWORD_PTR len_import_relocs;
    DWORD_PTR len_remaining_relocs;
Andrew Innes's avatar
Andrew Innes committed
634 635 636 637 638 639 640 641 642 643 644 645
    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;
646 647
	 (DWORD_PTR) block - (DWORD_PTR) relocs < reloc_dir->Size;
	 block = (void *)((DWORD_PTR) block + block->SizeOfBlock))
Andrew Innes's avatar
Andrew Innes committed
648 649 650 651 652 653 654 655 656 657 658 659 660 661 662
      {
	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)
      {
663 664
	len_import_relocs = (DWORD_PTR) end_block - (DWORD_PTR) start_block;
	len_remaining_relocs = (DWORD_PTR) relocs + reloc_dir->Size - (DWORD_PTR) end_block;
Andrew Innes's avatar
Andrew Innes committed
665 666 667 668 669 670 671 672 673 674 675 676 677
	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;
678 679
	 (DWORD_PTR) block - (DWORD_PTR) relocs < reloc_dir->Size;
	 block = (void *)((DWORD_PTR) block + block->SizeOfBlock))
Andrew Innes's avatar
Andrew Innes committed
680
      {
681 682
	DWORD_PTR page_rva = block->VirtualAddress;
	DWORD_PTR page_offset;
Andrew Innes's avatar
Andrew Innes committed
683 684
	union {
	  WORD word;
685
	  DWORD_PTR dword;
Andrew Innes's avatar
Andrew Innes committed
686 687 688 689 690 691 692 693 694
	} * 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];
695
	     (DWORD_PTR) fixup - (DWORD_PTR) block < block->SizeOfBlock;
Andrew Innes's avatar
Andrew Innes committed
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746
	     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;
Andrew Innes's avatar
Andrew Innes committed
747
#ifdef IMAGE_REL_BASED_SECTION
Andrew Innes's avatar
Andrew Innes committed
748 749 750
	      case IMAGE_REL_BASED_SECTION:
	      case IMAGE_REL_BASED_REL32:
		/* Docs don't say what these values mean.  */
751 752
#endif
	      default:
Andrew Innes's avatar
Andrew Innes committed
753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776
		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];

  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))
    {
777
      printf ("Failed to open %s (%d)...bailing.\n",
Andrew Innes's avatar
Andrew Innes committed
778 779 780 781 782 783 784 785 786
	      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))
    {
787
      printf ("Failed to open %s (%d)...bailing.\n",
Andrew Innes's avatar
Andrew Innes committed
788 789 790 791 792 793 794 795 796
	      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");
797 798
    DWORD_PTR  headersum;
    DWORD_PTR  checksum;
Andrew Innes's avatar
Andrew Innes committed
799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827

    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 */