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
->OptionalHeader
.NumberOfRvaAndSizes
);
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 PIMAGE_SECTION_HEADER section_header
;
233 PIMAGE_DATA_DIRECTORY data_dir
;
235 unsigned long checksum
;
236 int fixup_exports
= 0;
237 int fixup_sections
= 0;
240 * Process parameters.
245 printf("Usage: %s <filename> <options>\n"
247 " -sections Sets section flags for PE image.\n"
248 " -exports Sort the names in export table.\n",
253 for (i
= 2; i
< argc
; i
++)
255 if (!strcmp(argv
[i
], "-sections"))
257 else if (!strcmp(argv
[i
], "-exports"))
260 { printf("Invalid option: %s\n", argv
[i
]); return 1; }
263 if (fixup_sections
== 0 && fixup_exports
== 0)
265 printf("Nothing to do.\n");
270 * Read the whole file to memory.
273 fd_in
= open(argv
[1], O_RDONLY
| O_BINARY
);
276 printf("Can't open input file.\n");
280 len
= lseek(fd_in
, 0, SEEK_END
);
281 if (len
< sizeof(IMAGE_DOS_HEADER
))
284 printf("'%s' isn't a PE image (too short)\n", argv
[1]);
288 /* Lower down we overwrite the byte at len, so here, we need at least
289 * one more byte than len. We'll be guaranteed one or two now. */
290 buffer
= malloc((len
+ 2) & ~1);
294 printf("Not enough memory available.\n");
298 /* Read the whole input file into a buffer */
299 lseek(fd_in
, 0, SEEK_SET
);
300 read(fd_in
, buffer
, len
);
301 /* Here is where the block end overwrite was */
308 * Check the headers and save pointers to them.
311 dos_header
= (PIMAGE_DOS_HEADER
)buffer
;
312 nt_header
= (PIMAGE_NT_HEADERS
)(buffer
+ dtohl(dos_header
->e_lfanew
));
314 if (dtohs(dos_header
->e_magic
) != IMAGE_DOS_SIGNATURE
||
315 dtohl(nt_header
->Signature
) != IMAGE_NT_SIGNATURE
)
317 printf("'%s' isn't a PE image (bad headers)\n", argv
[1]);
324 /* Sort export directory */
325 data_dir
= &nt_header
->OptionalHeader
.DataDirectory
[IMAGE_DIRECTORY_ENTRY_EXPORT
];
326 if (dtohl(data_dir
->Size
) != 0)
328 PIMAGE_EXPORT_DIRECTORY export_directory
;
333 export_directory
= (PIMAGE_EXPORT_DIRECTORY
)rva_to_ptr(dtohl(data_dir
->VirtualAddress
));
334 if (export_directory
!= NULL
)
336 exports
= malloc(sizeof(export_t
) * dtohl(export_directory
->NumberOfNames
));
339 printf("Not enough memory.\n");
344 name_ptr
= (DWORD
*)rva_to_ptr(dtohl(export_directory
->AddressOfNames
));
345 ordinal_ptr
= (WORD
*)rva_to_ptr(dtohl(export_directory
->AddressOfNameOrdinals
));
347 for (i
= 0; i
< dtohl(export_directory
->NumberOfNames
); i
++)
349 exports
[i
].name
= dtohl(name_ptr
[i
]);
350 exports
[i
].ordinal
= dtohl(ordinal_ptr
[i
]);
353 qsort(exports
, dtohl(export_directory
->NumberOfNames
), sizeof(export_t
),
354 export_compare_func
);
356 for (i
= 0; i
< dtohl(export_directory
->NumberOfNames
); i
++)
358 name_ptr
[i
] = htodl(exports
[i
].name
);
359 ordinal_ptr
[i
] = htodl(exports
[i
].ordinal
);
369 /* Update section flags */
370 for (i
= 0, section_header
= IMAGE_FIRST_SECTION(nt_header
);
371 i
< dtohl(nt_header
->OptionalHeader
.NumberOfRvaAndSizes
);
372 i
++, section_header
++)
374 if (!strcmp((char*)section_header
->Name
, ".text") ||
375 !strcmp((char*)section_header
->Name
, ".data") ||
376 !strcmp((char*)section_header
->Name
, ".idata") ||
377 !strcmp((char*)section_header
->Name
, ".bss"))
379 section_header
->Characteristics
|= htodl(IMAGE_SCN_MEM_NOT_PAGED
);
380 section_header
->Characteristics
&= htodl(~IMAGE_SCN_MEM_DISCARDABLE
);
382 else if (!strcmp((char*)section_header
->Name
, "INIT"))
384 section_header
->Characteristics
|= htodl(IMAGE_SCN_MEM_DISCARDABLE
);
386 else if (!strcmp((char*)section_header
->Name
, "PAGE"))
388 section_header
->Characteristics
|= htodl(IMAGE_SCN_MEM_NOT_PAGED
);
393 /* Recalculate checksum */
394 nt_header
->OptionalHeader
.CheckSum
= 0;
396 for (i
= 0; i
< len
; i
+= 2)
398 checksum
+= *(unsigned short *)(buffer
+ i
);
399 checksum
= (checksum
+ (checksum
>> 16)) & 0xffff;
402 nt_header
->OptionalHeader
.CheckSum
= htods(checksum
);
404 /* Write the output file */
405 fd_out
= open(argv
[1], O_WRONLY
| O_BINARY
);
406 write(fd_out
, buffer
, len
);