3 * Copyright (C) 2005 Filip Navara
5 * The purpose of this utility is fix PE binaries generated by binutils and
6 * to manipulate flags that can't be set by binutils.
8 * Currently two features are implemented:
10 * - Setting flags on PE sections for use by drivers. The sections
11 * .text, .data, .idata, .bss are marked as non-pageable and
12 * non-discarable, section PAGE is marked as pageable and section
13 * INIT is marked as discaradable.
15 * - Sorting of export name table in executables. DLLTOOL has bug
16 * in sorting algorithm when the --kill-at flag is used. The exports
17 * are sorted in the decorated form and so the fastcall symbols are
18 * incorrectly put at the beginning of export table. This option
19 * allow to correct sort the table, so binary search can be used
35 /* The following definitions are ripped from MinGW W32API headers. We don't
36 use these headers directly in order to allow compilation on Linux hosts. */
38 typedef unsigned char BYTE
, *PBYTE
;
39 typedef unsigned short WORD
;
40 typedef unsigned int DWORD
;
43 #define IMAGE_NUMBEROF_DIRECTORY_ENTRIES 16
44 #define IMAGE_SIZEOF_SHORT_NAME 8
45 #define IMAGE_DOS_SIGNATURE 0x5A4D
46 #define IMAGE_NT_SIGNATURE 0x00004550
47 #define IMAGE_SCN_MEM_DISCARDABLE 0x2000000
48 #define IMAGE_SCN_MEM_NOT_PAGED 0x8000000
49 #define FIELD_OFFSET(t,f) ((LONG)&(((t*)0)->f))
50 #define IMAGE_FIRST_SECTION(h) ((PIMAGE_SECTION_HEADER) ((unsigned long)h+FIELD_OFFSET(IMAGE_NT_HEADERS,OptionalHeader)+((PIMAGE_NT_HEADERS)(h))->FileHeader.SizeOfOptionalHeader))
51 #define IMAGE_DIRECTORY_ENTRY_EXPORT 0
54 typedef struct _IMAGE_DOS_HEADER
{
74 } IMAGE_DOS_HEADER
,*PIMAGE_DOS_HEADER
;
77 typedef struct _IMAGE_EXPORT_DIRECTORY
{
78 DWORD Characteristics
;
84 DWORD NumberOfFunctions
;
86 DWORD AddressOfFunctions
;
88 DWORD AddressOfNameOrdinals
;
89 } IMAGE_EXPORT_DIRECTORY
,*PIMAGE_EXPORT_DIRECTORY
;
90 typedef struct _IMAGE_FILE_HEADER
{
92 WORD NumberOfSections
;
94 DWORD PointerToSymbolTable
;
95 DWORD NumberOfSymbols
;
96 WORD SizeOfOptionalHeader
;
98 } IMAGE_FILE_HEADER
, *PIMAGE_FILE_HEADER
;
99 typedef struct _IMAGE_DATA_DIRECTORY
{
100 DWORD VirtualAddress
;
102 } IMAGE_DATA_DIRECTORY
,*PIMAGE_DATA_DIRECTORY
;
103 typedef struct _IMAGE_OPTIONAL_HEADER
{
105 BYTE MajorLinkerVersion
;
106 BYTE MinorLinkerVersion
;
108 DWORD SizeOfInitializedData
;
109 DWORD SizeOfUninitializedData
;
110 DWORD AddressOfEntryPoint
;
114 DWORD SectionAlignment
;
116 WORD MajorOperatingSystemVersion
;
117 WORD MinorOperatingSystemVersion
;
118 WORD MajorImageVersion
;
119 WORD MinorImageVersion
;
120 WORD MajorSubsystemVersion
;
121 WORD MinorSubsystemVersion
;
127 WORD DllCharacteristics
;
128 DWORD SizeOfStackReserve
;
129 DWORD SizeOfStackCommit
;
130 DWORD SizeOfHeapReserve
;
131 DWORD SizeOfHeapCommit
;
133 DWORD NumberOfRvaAndSizes
;
134 IMAGE_DATA_DIRECTORY DataDirectory
[IMAGE_NUMBEROF_DIRECTORY_ENTRIES
];
135 } IMAGE_OPTIONAL_HEADER
,*PIMAGE_OPTIONAL_HEADER
;
136 typedef struct _IMAGE_NT_HEADERS
{
138 IMAGE_FILE_HEADER FileHeader
;
139 IMAGE_OPTIONAL_HEADER OptionalHeader
;
140 } IMAGE_NT_HEADERS
,*PIMAGE_NT_HEADERS
;
141 typedef struct _IMAGE_SECTION_HEADER
{
142 BYTE Name
[IMAGE_SIZEOF_SHORT_NAME
];
144 DWORD PhysicalAddress
;
147 DWORD VirtualAddress
;
149 DWORD PointerToRawData
;
150 DWORD PointerToRelocations
;
151 DWORD PointerToLinenumbers
;
152 WORD NumberOfRelocations
;
153 WORD NumberOfLinenumbers
;
154 DWORD Characteristics
;
155 } IMAGE_SECTION_HEADER
,*PIMAGE_SECTION_HEADER
;
158 /* End of ripped definitions */
160 typedef struct _export_t
{
165 unsigned char *buffer
;
166 PIMAGE_DOS_HEADER dos_header
;
167 PIMAGE_NT_HEADERS nt_header
;
169 static inline WORD
dtohs(WORD in
)
171 PBYTE in_ptr
= (PBYTE
)&in
;
172 return in_ptr
[0] | (in_ptr
[1] << 8);
175 static inline WORD
htods(WORD in
)
178 PBYTE out_ptr
= (PBYTE
)&out
;
179 out_ptr
[0] = in
; out_ptr
[1] = in
>> 8;
183 static inline DWORD
dtohl(DWORD in
)
185 PBYTE in_ptr
= (PBYTE
)&in
;
186 return in_ptr
[0] | (in_ptr
[1] << 8) | (in_ptr
[2] << 16) | (in_ptr
[3] << 24);
189 static inline DWORD
htodl(DWORD in
)
192 PBYTE out_ptr
= (PBYTE
)&out
;
193 out_ptr
[0] = in
; out_ptr
[1] = in
>> 8;
194 out_ptr
[2] = in
>> 16; out_ptr
[3] = in
>> 24;
198 void *rva_to_ptr(DWORD rva
)
200 PIMAGE_SECTION_HEADER section_header
;
203 for (i
= 0, section_header
= IMAGE_FIRST_SECTION(nt_header
);
204 i
< dtohl(nt_header
->FileHeader
.NumberOfSections
);
205 i
++, section_header
++)
207 if (rva
>= dtohl(section_header
->VirtualAddress
) &&
208 rva
< dtohl(section_header
->VirtualAddress
) +
209 dtohl(section_header
->Misc
.VirtualSize
))
211 return buffer
+ rva
- dtohl(section_header
->VirtualAddress
) +
212 dtohl(section_header
->PointerToRawData
);
219 int export_compare_func(const void *a
, const void *b
)
221 const export_t
*ap
= a
;
222 const export_t
*bp
= b
;
223 char *an
= rva_to_ptr(ap
->name
);
224 char *bn
= rva_to_ptr(bp
->name
);
225 return strcmp(an
, bn
);
228 int main(int argc
, char **argv
)
232 char hdrbuf
[4] = { }, elfhdr
[4] = { '\177', 'E', 'L', 'F' };
233 PIMAGE_SECTION_HEADER section_header
;
234 PIMAGE_DATA_DIRECTORY data_dir
;
236 unsigned long checksum
;
237 int fixup_exports
= 0;
238 int fixup_sections
= 0;
241 * Process parameters.
246 printf("Usage: %s <filename> <options>\n"
248 " -sections Sets section flags for PE image.\n"
249 " -exports Sort the names in export table.\n",
254 for (i
= 2; i
< argc
; i
++)
256 if (!strcmp(argv
[i
], "-sections"))
258 else if (!strcmp(argv
[i
], "-exports"))
261 { printf("Invalid option: %s\n", argv
[i
]); return 1; }
264 if (fixup_sections
== 0 && fixup_exports
== 0)
266 printf("Nothing to do.\n");
271 * Read the whole file to memory.
274 fd_in
= open(argv
[1], O_RDONLY
| O_BINARY
);
277 printf("Can't open input file.\n");
282 * PowerPC ReactOS uses elf, so doesn't need pefixup
284 len
= read(fd_in
, hdrbuf
, sizeof(elfhdr
));
285 if (!memcmp(hdrbuf
, elfhdr
, sizeof(elfhdr
)))
288 len
= lseek(fd_in
, 0, SEEK_END
);
289 if (len
< sizeof(IMAGE_DOS_HEADER
))
292 printf("'%s' isn't a PE image (too short)\n", argv
[1]);
296 /* Lower down we overwrite the byte at len, so here, we need at least
297 * one more byte than len. We'll be guaranteed one or two now. */
298 buffer
= malloc((len
+ 2) & ~1);
302 printf("Not enough memory available.\n");
306 /* Read the whole input file into a buffer */
307 lseek(fd_in
, 0, SEEK_SET
);
308 read(fd_in
, buffer
, len
);
309 /* Here is where the block end overwrite was */
316 * Check the headers and save pointers to them.
319 dos_header
= (PIMAGE_DOS_HEADER
)buffer
;
320 nt_header
= (PIMAGE_NT_HEADERS
)(buffer
+ dtohl(dos_header
->e_lfanew
));
322 if (dtohs(dos_header
->e_magic
) != IMAGE_DOS_SIGNATURE
||
323 dtohl(nt_header
->Signature
) != IMAGE_NT_SIGNATURE
)
325 printf("'%s' isn't a PE image (bad headers)\n", argv
[1]);
332 /* Sort export directory */
333 data_dir
= &nt_header
->OptionalHeader
.DataDirectory
[IMAGE_DIRECTORY_ENTRY_EXPORT
];
334 if (dtohl(data_dir
->Size
) != 0)
336 PIMAGE_EXPORT_DIRECTORY export_directory
;
341 export_directory
= (PIMAGE_EXPORT_DIRECTORY
)rva_to_ptr(dtohl(data_dir
->VirtualAddress
));
342 if (export_directory
!= NULL
)
344 exports
= malloc(sizeof(export_t
) * dtohl(export_directory
->NumberOfNames
));
347 printf("Not enough memory.\n");
352 name_ptr
= (DWORD
*)rva_to_ptr(dtohl(export_directory
->AddressOfNames
));
353 ordinal_ptr
= (WORD
*)rva_to_ptr(dtohl(export_directory
->AddressOfNameOrdinals
));
355 for (i
= 0; i
< dtohl(export_directory
->NumberOfNames
); i
++)
357 exports
[i
].name
= dtohl(name_ptr
[i
]);
358 exports
[i
].ordinal
= dtohl(ordinal_ptr
[i
]);
361 qsort(exports
, dtohl(export_directory
->NumberOfNames
), sizeof(export_t
),
362 export_compare_func
);
364 for (i
= 0; i
< dtohl(export_directory
->NumberOfNames
); i
++)
366 name_ptr
[i
] = htodl(exports
[i
].name
);
367 ordinal_ptr
[i
] = htodl(exports
[i
].ordinal
);
377 /* Update section flags */
378 for (i
= 0, section_header
= IMAGE_FIRST_SECTION(nt_header
);
379 i
< dtohl(nt_header
->OptionalHeader
.NumberOfRvaAndSizes
);
380 i
++, section_header
++)
382 if (!strcmp((char*)section_header
->Name
, ".text") ||
383 !strcmp((char*)section_header
->Name
, ".data") ||
384 !strcmp((char*)section_header
->Name
, ".idata") ||
385 !strcmp((char*)section_header
->Name
, ".bss"))
387 section_header
->Characteristics
|= htodl(IMAGE_SCN_MEM_NOT_PAGED
);
388 section_header
->Characteristics
&= htodl(~IMAGE_SCN_MEM_DISCARDABLE
);
390 else if (!strcmp((char*)section_header
->Name
, "INIT"))
392 section_header
->Characteristics
|= htodl(IMAGE_SCN_MEM_DISCARDABLE
);
394 else if (!strcmp((char*)section_header
->Name
, "PAGE"))
396 section_header
->Characteristics
|= htodl(IMAGE_SCN_MEM_NOT_PAGED
);
401 /* Recalculate checksum */
402 nt_header
->OptionalHeader
.CheckSum
= 0;
404 for (i
= 0; i
< len
; i
+= 2)
406 checksum
+= *(unsigned short *)(buffer
+ i
);
407 checksum
= (checksum
+ (checksum
>> 16)) & 0xffff;
410 nt_header
->OptionalHeader
.CheckSum
= htods(checksum
);
412 /* Write the output file */
413 fd_out
= open(argv
[1], O_WRONLY
| O_BINARY
);
414 write(fd_out
, buffer
, len
);