nt/addsection.c

   1 /* Add an uninitialized data section to an executable.
   2    Copyright (C) 1999, 2001, 2002, 2003, 2004, 2005, 2006, 2007,
   3      2008, 2009, 2010, 2011  Free Software Foundation, Inc.
   4
   5 This file is part of GNU Emacs.
   6
   7 GNU Emacs is free software: you can redistribute it and/or modify
   8 it under the terms of the GNU General Public License as published by
   9 the Free Software Foundation, either version 3 of the License, or
  10 (at your option) any later version.
  11
  12 GNU Emacs is distributed in the hope that it will be useful,
  13 but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 GNU General Public License for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with GNU Emacs.  If not, see <http://www.gnu.org/licenses/>.
  19
  20
  21    Andrew Innes <andrewi@harlequin.co.uk>       04-Jan-1999
  22      based on code from unexw32.c
  23 */
  24
  25 #include <stdlib.h>
  26 #include <stdio.h>
  27 #include <fcntl.h>
  28 #include <time.h>
  29 #ifdef __GNUC__
  30 #define _ANONYMOUS_UNION
  31 #define _ANONYMOUS_STRUCT
  32 #endif
  33 #include <windows.h>
  34
  35 /* Include relevant definitions from IMAGEHLP.H, which can be found
  36    in \\win32sdk\mstools\samples\image\include\imagehlp.h. */
  37
  38 PIMAGE_NT_HEADERS
  39 (__stdcall * pfnCheckSumMappedFile) (LPVOID BaseAddress,
  40                                      DWORD FileLength,
  41                                      LPDWORD HeaderSum,
  42                                      LPDWORD CheckSum);
  43
  44 #undef min
  45 #undef max
  46 #define min(x, y) (((x) < (y)) ? (x) : (y))
  47 #define max(x, y) (((x) > (y)) ? (x) : (y))
  48
  49
  50 /* File handling.  */
  51
  52 typedef struct file_data {
  53   const char    *name;
  54   unsigned long  size;
  55   HANDLE         file;
  56   HANDLE         file_mapping;
  57   unsigned char *file_base;
  58 } file_data;
  59
  60 int
  61 open_input_file (file_data *p_file, const char *filename)
  62 {
  63   HANDLE file;
  64   HANDLE file_mapping;
  65   void  *file_base;
  66   unsigned long size, upper_size;
  67
  68   file = CreateFile (filename, GENERIC_READ, FILE_SHARE_READ, NULL,
  69                      OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
  70   if (file == INVALID_HANDLE_VALUE)
  71     return FALSE;
  72
  73   size = GetFileSize (file, &upper_size);
  74   file_mapping = CreateFileMapping (file, NULL, PAGE_READONLY,
  75                                     0, size, NULL);
  76   if (!file_mapping)
  77     return FALSE;
  78
  79   file_base = MapViewOfFile (file_mapping, FILE_MAP_READ, 0, 0, size);
  80   if (file_base == 0)
  81     return FALSE;
  82
  83   p_file->name = filename;
  84   p_file->size = size;
  85   p_file->file = file;
  86   p_file->file_mapping = file_mapping;
  87   p_file->file_base = file_base;
  88
  89   return TRUE;
  90 }
  91
  92 int
  93 open_output_file (file_data *p_file, const char *filename, unsigned long size)
  94 {
  95   HANDLE file;
  96   HANDLE file_mapping;
  97   void  *file_base;
  98
  99   file = CreateFile (filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
 100                      CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
 101   if (file == INVALID_HANDLE_VALUE)
 102     return FALSE;
 103
 104   file_mapping = CreateFileMapping (file, NULL, PAGE_READWRITE,
 105                                     0, size, NULL);
 106   if (!file_mapping)
 107     return FALSE;
 108
 109   file_base = MapViewOfFile (file_mapping, FILE_MAP_WRITE, 0, 0, size);
 110   if (file_base == 0)
 111     return FALSE;
 112
 113   p_file->name = filename;
 114   p_file->size = size;
 115   p_file->file = file;
 116   p_file->file_mapping = file_mapping;
 117   p_file->file_base = file_base;
 118
 119   return TRUE;
 120 }
 121
 122 /* Close the system structures associated with the given file.  */
 123 void
 124 close_file_data (file_data *p_file)
 125 {
 126   UnmapViewOfFile (p_file->file_base);
 127   CloseHandle (p_file->file_mapping);
 128   /* For the case of output files, set final size.  */
 129   SetFilePointer (p_file->file, p_file->size, NULL, FILE_BEGIN);
 130   SetEndOfFile (p_file->file);
 131   CloseHandle (p_file->file);
 132 }
 133
 134
 135 /* Routines to manipulate NT executable file sections.  */
 136
 137 unsigned long
 138 get_unrounded_section_size (PIMAGE_SECTION_HEADER p_section)
 139 {
 140   /* The true section size, before rounding, for an initialized data or
 141      code section.  (Supposedly some linkers swap the meaning of these
 142      two values.)  */
 143   return min (p_section->SizeOfRawData,
 144               p_section->Misc.VirtualSize);
 145 }
 146
 147 /* Return pointer to section header for named section. */
 148 IMAGE_SECTION_HEADER *
 149 find_section (const char *name, IMAGE_NT_HEADERS *nt_header)
 150 {
 151   PIMAGE_SECTION_HEADER section;
 152   int i;
 153
 154   section = IMAGE_FIRST_SECTION (nt_header);
 155
 156   for (i = 0; i < nt_header->FileHeader.NumberOfSections; i++)
 157     {
 158       if (strcmp (section->Name, name) == 0)
 159         return section;
 160       section++;
 161     }
 162   return NULL;
 163 }
 164
 165 /* Return pointer to section header for section containing the given
 166    relative virtual address. */
 167 IMAGE_SECTION_HEADER *
 168 rva_to_section (DWORD rva, IMAGE_NT_HEADERS * nt_header)
 169 {
 170   PIMAGE_SECTION_HEADER section;
 171   int i;
 172
 173   section = IMAGE_FIRST_SECTION (nt_header);
 174
 175   for (i = 0; i < nt_header->FileHeader.NumberOfSections; i++)
 176     {
 177       /* Some linkers (eg. the NT SDK linker I believe) swapped the
 178          meaning of these two values - or rather, they ignored
 179          VirtualSize entirely and always set it to zero.  This affects
 180          some very old exes (eg. gzip dated Dec 1993).  Since
 181          w32_executable_type relies on this function to work reliably,
 182          we need to cope with this.  */
 183       DWORD real_size = max (section->SizeOfRawData,
 184                              section->Misc.VirtualSize);
 185       if (rva >= section->VirtualAddress
 186           && rva < section->VirtualAddress + real_size)
 187         return section;
 188       section++;
 189     }
 190   return NULL;
 191 }
 192
 193 /* Return pointer to section header for section containing the given
 194    offset in its raw data area. */
 195 IMAGE_SECTION_HEADER *
 196 offset_to_section (DWORD offset, IMAGE_NT_HEADERS * nt_header)
 197 {
 198   PIMAGE_SECTION_HEADER section;
 199   int i;
 200
 201   section = IMAGE_FIRST_SECTION (nt_header);
 202
 203   for (i = 0; i < nt_header->FileHeader.NumberOfSections; i++)
 204     {
 205       if (offset >= section->PointerToRawData
 206           && offset < section->PointerToRawData + section->SizeOfRawData)
 207         return section;
 208       section++;
 209     }
 210   return NULL;
 211 }
 212
 213 /* Return offset to an object in dst, given offset in src.  We assume
 214    there is at least one section in both src and dst images, and that
 215    the some sections may have been added to dst (after sections in src).  */
 216 static DWORD
 217 relocate_offset (DWORD offset,
 218                  IMAGE_NT_HEADERS * src_nt_header,
 219                  IMAGE_NT_HEADERS * dst_nt_header)
 220 {
 221   PIMAGE_SECTION_HEADER src_section = IMAGE_FIRST_SECTION (src_nt_header);
 222   PIMAGE_SECTION_HEADER dst_section = IMAGE_FIRST_SECTION (dst_nt_header);
 223   int i = 0;
 224
 225   while (offset >= src_section->PointerToRawData)
 226     {
 227       if (offset < src_section->PointerToRawData + src_section->SizeOfRawData)
 228         break;
 229       i++;
 230       if (i == src_nt_header->FileHeader.NumberOfSections)
 231         {
 232           /* Handle offsets after the last section.  */
 233           dst_section = IMAGE_FIRST_SECTION (dst_nt_header);
 234           dst_section += dst_nt_header->FileHeader.NumberOfSections - 1;
 235           while (dst_section->PointerToRawData == 0)
 236             dst_section--;
 237           while (src_section->PointerToRawData == 0)
 238             src_section--;
 239           return offset
 240             + (dst_section->PointerToRawData + dst_section->SizeOfRawData)
 241             - (src_section->PointerToRawData + src_section->SizeOfRawData);
 242         }
 243       src_section++;
 244       dst_section++;
 245     }
 246   return offset +
 247     (dst_section->PointerToRawData - src_section->PointerToRawData);
 248 }
 249
 250 #define OFFSET_TO_RVA(offset, section) \
 251           (section->VirtualAddress + ((DWORD)(offset) - section->PointerToRawData))
 252
 253 #define RVA_TO_OFFSET(rva, section) \
 254           (section->PointerToRawData + ((DWORD)(rva) - section->VirtualAddress))
 255
 256 #define RVA_TO_SECTION_OFFSET(rva, section) \
 257           ((DWORD)(rva) - section->VirtualAddress)
 258
 259 /* Convert address in executing image to RVA.  */
 260 #define PTR_TO_RVA(ptr) ((DWORD)(ptr) - (DWORD) GetModuleHandle (NULL))
 261
 262 #define PTR_TO_OFFSET(ptr, pfile_data) \
 263           ((unsigned const char *)(ptr) - (pfile_data)->file_base)
 264
 265 #define OFFSET_TO_PTR(offset, pfile_data) \
 266           ((pfile_data)->file_base + (DWORD)(offset))
 267
 268 #define ROUND_UP(p, align)   (((DWORD)(p) + (align)-1) & ~((align)-1))
 269 #define ROUND_DOWN(p, align) ((DWORD)(p) & ~((align)-1))
 270
 271
 272 static void
 273 copy_executable_and_add_section (file_data *p_infile,
 274                                  file_data *p_outfile,
 275                                  const char *new_section_name,
 276                                  DWORD new_section_size)
 277 {
 278   unsigned char *dst;
 279   PIMAGE_DOS_HEADER dos_header;
 280   PIMAGE_NT_HEADERS nt_header;
 281   PIMAGE_NT_HEADERS dst_nt_header;
 282   PIMAGE_SECTION_HEADER section;
 283   PIMAGE_SECTION_HEADER dst_section;
 284   DWORD offset;
 285   int i;
 286   int be_verbose = GetEnvironmentVariable ("DEBUG_DUMP", NULL, 0) > 0;
 287
 288 #define COPY_CHUNK(message, src, size, verbose)                                 \
 289   do {                                                                          \
 290     unsigned const char *s = (void *)(src);                                             \
 291     unsigned long count = (size);                                               \
 292     if (verbose)                                                                \
 293       {                                                                         \
 294         printf ("%s\n", (message));                                             \
 295         printf ("\t0x%08x Offset in input file.\n", s - p_infile->file_base);   \
 296         printf ("\t0x%08x Offset in output file.\n", dst - p_outfile->file_base); \
 297         printf ("\t0x%08x Size in bytes.\n", count);                            \
 298       }                                                                         \
 299     memcpy (dst, s, count);                                                     \
 300     dst += count;                                                               \
 301   } while (0)
 302
 303 #define DST_TO_OFFSET()  PTR_TO_OFFSET (dst, p_outfile)
 304 #define ROUND_UP_DST_AND_ZERO(align)                                            \
 305   do {                                                                          \
 306     unsigned char *newdst = p_outfile->file_base                                \
 307       + ROUND_UP (DST_TO_OFFSET (), (align));                                   \
 308     /* Zero the alignment slop; it may actually initialize real data.  */       \
 309     memset (dst, 0, newdst - dst);                                              \
 310     dst = newdst;                                                               \
 311   } while (0)
 312
 313   /* Copy the source image sequentially, ie. section by section after
 314      copying the headers and section table, to simplify the process of
 315      adding an extra section table entry (which might force the raw
 316      section data to be relocated).
 317
 318      Note that dst is updated implicitly by each COPY_CHUNK.  */
 319
 320   dos_header = (PIMAGE_DOS_HEADER) p_infile->file_base;
 321   nt_header = (PIMAGE_NT_HEADERS) (((unsigned long) dos_header) +
 322                                    dos_header->e_lfanew);
 323   section = IMAGE_FIRST_SECTION (nt_header);
 324
 325   dst = (unsigned char *) p_outfile->file_base;
 326
 327   COPY_CHUNK ("Copying DOS header...", dos_header,
 328               (DWORD) nt_header - (DWORD) dos_header, be_verbose);
 329   dst_nt_header = (PIMAGE_NT_HEADERS) dst;
 330   COPY_CHUNK ("Copying NT header...", nt_header,
 331               (DWORD) section - (DWORD) nt_header, be_verbose);
 332   dst_section = (PIMAGE_SECTION_HEADER) dst;
 333   COPY_CHUNK ("Copying section table...", section,
 334               nt_header->FileHeader.NumberOfSections * sizeof (*section),
 335               be_verbose);
 336
 337   /* To improve the efficiency of demand loading, make the file
 338      alignment match the section alignment (VC++ 6.0 does this by
 339      default anyway).  */
 340   dst_nt_header->OptionalHeader.FileAlignment =
 341     dst_nt_header->OptionalHeader.SectionAlignment;
 342
 343   /* Add an uninitialized data section at the end, of the specified name
 344      and virtual size.  */
 345   if (find_section (new_section_name, nt_header) == NULL)
 346     /* Leave room for extra section table entry; filled in below.  */
 347     dst += sizeof (*section);
 348   else
 349     new_section_name = NULL;
 350
 351   /* Align the first section's raw data area, and set the header size
 352      field accordingly.  */
 353   ROUND_UP_DST_AND_ZERO (dst_nt_header->OptionalHeader.FileAlignment);
 354   dst_nt_header->OptionalHeader.SizeOfHeaders = DST_TO_OFFSET ();
 355
 356   for (i = 0; i < nt_header->FileHeader.NumberOfSections; i++)
 357     {
 358       char msg[100];
 359       /* Windows section names are fixed 8-char strings, only
 360          zero-terminated if the name is shorter than 8 characters.  */
 361       sprintf (msg, "Copying raw data for %.8s...", section->Name);
 362
 363       /* Update the file-relative offset for this section's raw data (if
 364          it has any) in case things have been relocated; we will update
 365          the other offsets below once we know where everything is.  */
 366       if (dst_section->PointerToRawData)
 367         dst_section->PointerToRawData = DST_TO_OFFSET ();
 368
 369       /* Can always copy the original raw data.  */
 370       COPY_CHUNK
 371         (msg, OFFSET_TO_PTR (section->PointerToRawData, p_infile),
 372          section->SizeOfRawData, be_verbose);
 373
 374       /* Round up the raw data size to the new alignment.  */
 375       dst_section->SizeOfRawData =
 376         ROUND_UP (dst_section->SizeOfRawData,
 377                   dst_nt_header->OptionalHeader.FileAlignment);
 378
 379       /* Align the next section's raw data area.  */
 380       ROUND_UP_DST_AND_ZERO (dst_nt_header->OptionalHeader.FileAlignment);
 381
 382       section++;
 383       dst_section++;
 384     }
 385
 386   /* Add the extra section entry (which adds no raw data).  */
 387   if (new_section_name != NULL)
 388     {
 389       dst_nt_header->FileHeader.NumberOfSections++;
 390       dst_nt_header->OptionalHeader.SizeOfImage += new_section_size;
 391       strncpy (dst_section->Name, new_section_name, sizeof (dst_section->Name));
 392       dst_section->VirtualAddress =
 393         section[-1].VirtualAddress
 394         + ROUND_UP (section[-1].Misc.VirtualSize,
 395                     dst_nt_header->OptionalHeader.SectionAlignment);
 396       dst_section->Misc.VirtualSize = new_section_size;
 397       dst_section->PointerToRawData = 0;
 398       dst_section->SizeOfRawData = 0;
 399       dst_section->Characteristics =
 400         IMAGE_SCN_CNT_UNINITIALIZED_DATA
 401         | IMAGE_SCN_MEM_READ
 402         | IMAGE_SCN_MEM_WRITE;
 403     }
 404
 405   /* Copy remainder of source image.  */
 406   section--;
 407   offset = ROUND_UP (section->PointerToRawData + section->SizeOfRawData,
 408                      nt_header->OptionalHeader.FileAlignment);
 409   COPY_CHUNK
 410     ("Copying remainder of executable...",
 411      OFFSET_TO_PTR (offset, p_infile),
 412      p_infile->size - offset, be_verbose);
 413
 414   /* Final size for new image.  */
 415   p_outfile->size = DST_TO_OFFSET ();
 416
 417   /* Now patch up remaining file-relative offsets.  */
 418   section = IMAGE_FIRST_SECTION (nt_header);
 419   dst_section = IMAGE_FIRST_SECTION (dst_nt_header);
 420
 421 #define ADJUST_OFFSET(var)                                              \
 422   do {                                                                  \
 423     if ((var) != 0)                                                     \
 424       (var) = relocate_offset ((var), nt_header, dst_nt_header);        \
 425   } while (0)
 426
 427   dst_nt_header->OptionalHeader.SizeOfInitializedData = 0;
 428   dst_nt_header->OptionalHeader.SizeOfUninitializedData = 0;
 429   for (i = 0; i < dst_nt_header->FileHeader.NumberOfSections; i++)
 430     {
 431       /* Recompute data sizes for completeness.  */
 432       if (dst_section[i].Characteristics & IMAGE_SCN_CNT_INITIALIZED_DATA)
 433         dst_nt_header->OptionalHeader.SizeOfInitializedData +=
 434           ROUND_UP (dst_section[i].Misc.VirtualSize, dst_nt_header->OptionalHeader.FileAlignment);
 435       else if (dst_section[i].Characteristics & IMAGE_SCN_CNT_UNINITIALIZED_DATA)
 436         dst_nt_header->OptionalHeader.SizeOfUninitializedData +=
 437           ROUND_UP (dst_section[i].Misc.VirtualSize, dst_nt_header->OptionalHeader.FileAlignment);
 438
 439       ADJUST_OFFSET (dst_section[i].PointerToLinenumbers);
 440     }
 441
 442   ADJUST_OFFSET (dst_nt_header->FileHeader.PointerToSymbolTable);
 443
 444   /* Update offsets in debug directory entries. */
 445   {
 446     IMAGE_DATA_DIRECTORY debug_dir =
 447       dst_nt_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG];
 448     PIMAGE_DEBUG_DIRECTORY debug_entry;
 449
 450     section = rva_to_section (debug_dir.VirtualAddress, dst_nt_header);
 451     if (section)
 452       {
 453         debug_entry = (PIMAGE_DEBUG_DIRECTORY)
 454           (RVA_TO_OFFSET (debug_dir.VirtualAddress, section) + p_outfile->file_base);
 455         debug_dir.Size /= sizeof (IMAGE_DEBUG_DIRECTORY);
 456
 457         for (i = 0; i < debug_dir.Size; i++, debug_entry++)
 458           ADJUST_OFFSET (debug_entry->PointerToRawData);
 459       }
 460   }
 461 }
 462
 463
 464 int
 465 main (int argc, char **argv)
 466 {
 467   file_data in_file, out_file;
 468   char out_filename[MAX_PATH], in_filename[MAX_PATH];
 469   unsigned long size;
 470   PIMAGE_DOS_HEADER dos_header;
 471   PIMAGE_NT_HEADERS nt_header;
 472
 473 #define OLD_NAME        argv[1]
 474 #define NEW_NAME        argv[2]
 475 #define SECTION_NAME    argv[3]
 476 #define SECTION_SIZE    argv[4]
 477
 478   strcpy (in_filename, OLD_NAME);
 479   strcpy (out_filename, NEW_NAME);
 480
 481   printf ("Dumping from %s\n", in_filename);
 482   printf ("          to %s\n", out_filename);
 483
 484   /* Open the undumped executable file.  */
 485   if (!open_input_file (&in_file, in_filename))
 486     {
 487       printf ("Failed to open %s (%d)...bailing.\n",
 488               in_filename, GetLastError ());
 489       exit (1);
 490     }
 491   dos_header = (PIMAGE_DOS_HEADER) in_file.file_base;
 492   nt_header = (PIMAGE_NT_HEADERS) ((char *) dos_header + dos_header->e_lfanew);
 493   /* Allow for expansion due to increasing file align to section align.
 494      We can overestimate here, since close_file_data will update the
 495      size exactly.  */
 496   size = in_file.size
 497     + nt_header->OptionalHeader.SectionAlignment
 498     * nt_header->FileHeader.NumberOfSections;
 499   if (!open_output_file (&out_file, out_filename, size))
 500     {
 501       printf ("Failed to open %s (%d)...bailing.\n",
 502               out_filename, GetLastError ());
 503       exit (1);
 504     }
 505
 506   copy_executable_and_add_section (&in_file, &out_file,
 507                                    SECTION_NAME,
 508                                    atoi (SECTION_SIZE) * 1024 * 1024);
 509
 510   /* Patch up header fields; profiler is picky about this. */
 511   {
 512     HANDLE hImagehelp = LoadLibrary ("imagehlp.dll");
 513     DWORD  headersum;
 514     DWORD  checksum;
 515
 516     dos_header = (PIMAGE_DOS_HEADER) out_file.file_base;
 517     nt_header = (PIMAGE_NT_HEADERS) ((char *) dos_header + dos_header->e_lfanew);
 518
 519     nt_header->OptionalHeader.CheckSum = 0;
 520 //    nt_header->FileHeader.TimeDateStamp = time (NULL);
 521 //    dos_header->e_cp = size / 512;
 522 //    nt_header->OptionalHeader.SizeOfImage = size;
 523
 524     pfnCheckSumMappedFile = (void *) GetProcAddress (hImagehelp, "CheckSumMappedFile");
 525     if (pfnCheckSumMappedFile)
 526       {
 527 //      nt_header->FileHeader.TimeDateStamp = time (NULL);
 528         pfnCheckSumMappedFile (out_file.file_base,
 529                                out_file.size,
 530                                &headersum,
 531                                &checksum);
 532         nt_header->OptionalHeader.CheckSum = checksum;
 533       }
 534     FreeLibrary (hImagehelp);
 535   }
 536
 537   close_file_data (&in_file);
 538   close_file_data (&out_file);
 539
 540   return 0;
 541 }
 542
 543 /* eof */
 544