?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Correct RegCreateKeyEx[A/W] prototype to match the PSDK.
[wine/wine64.git] / tools / winedump / pe.c
blobbe217ee4ed96f083db5919f1d80797cdfdcc4848
1 /*
2 * PE dumping utility
3 *
4 * Copyright 2001 Eric Pouech
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <stdlib.h>
25 #include <stdarg.h>
26 #include <stdio.h>
27 #ifdef HAVE_UNISTD_H
28 # include <unistd.h>
29 #endif
30 #include <time.h>
31 #ifdef HAVE_SYS_TYPES_H
32 # include <sys/types.h>
33 #endif
34 #ifdef HAVE_SYS_STAT_H
35 # include <sys/stat.h>
36 #endif
37 #ifdef HAVE_SYS_MMAN_H
38 #include <sys/mman.h>
39 #endif
40 #include <fcntl.h>
41
42 #define NONAMELESSUNION
43 #define NONAMELESSSTRUCT
44 #include "windef.h"
45 #include "winbase.h"
46 #include "winedump.h"
47 #include "pe.h"
48
49 static IMAGE_NT_HEADERS32* PE_nt_headers;
50
51 static const char* get_machine_str(DWORD mach)
52 {
53 switch (mach)
54 {
55 case IMAGE_FILE_MACHINE_UNKNOWN: return "Unknown";
56 case IMAGE_FILE_MACHINE_I860: return "i860";
57 case IMAGE_FILE_MACHINE_I386: return "i386";
58 case IMAGE_FILE_MACHINE_R3000: return "R3000";
59 case IMAGE_FILE_MACHINE_R4000: return "R4000";
60 case IMAGE_FILE_MACHINE_R10000: return "R10000";
61 case IMAGE_FILE_MACHINE_ALPHA: return "Alpha";
62 case IMAGE_FILE_MACHINE_POWERPC: return "PowerPC";
63 case IMAGE_FILE_MACHINE_AMD64: return "AMD64";
64 case IMAGE_FILE_MACHINE_IA64: return "IA64";
65 }
66 return "???";
67 }
68
69 static void* RVA(unsigned long rva, unsigned long len)
70 {
71 IMAGE_SECTION_HEADER* sectHead;
72 int i;
73
74 if (rva == 0) return NULL;
75
76 sectHead = IMAGE_FIRST_SECTION(PE_nt_headers);
77 for (i = PE_nt_headers->FileHeader.NumberOfSections - 1; i >= 0; i--)
78 {
79 if (sectHead[i].VirtualAddress <= rva &&
80 rva + len <= (DWORD)sectHead[i].VirtualAddress + sectHead[i].SizeOfRawData)
81 {
82 /* return image import directory offset */
83 return PRD(sectHead[i].PointerToRawData + rva - sectHead[i].VirtualAddress, len);
84 }
85 }
86
87 return NULL;
88 }
89
90 static void *get_dir_and_size(unsigned int idx, unsigned int *size)
91 {
92 if(PE_nt_headers->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
93 {
94 IMAGE_OPTIONAL_HEADER64 *opt = (IMAGE_OPTIONAL_HEADER64*)&PE_nt_headers->OptionalHeader;
95 if (idx >= opt->NumberOfRvaAndSizes)
96 return NULL;
97 if(size)
98 *size = opt->DataDirectory[idx].Size;
99 return RVA(opt->DataDirectory[idx].VirtualAddress,
100 opt->DataDirectory[idx].Size);
101 }
102 else
103 {
104 IMAGE_OPTIONAL_HEADER32 *opt = (IMAGE_OPTIONAL_HEADER32*)&PE_nt_headers->OptionalHeader;
105 if (idx >= opt->NumberOfRvaAndSizes)
106 return NULL;
107 if(size)
108 *size = opt->DataDirectory[idx].Size;
109 return RVA(opt->DataDirectory[idx].VirtualAddress,
110 opt->DataDirectory[idx].Size);
111 }
112 }
113
114 static void* get_dir(unsigned idx)
115 {
116 return get_dir_and_size(idx, 0);
117 }
118
119 static const char* DirectoryNames[16] = {
120 "EXPORT", "IMPORT", "RESOURCE", "EXCEPTION",
121 "SECURITY", "BASERELOC", "DEBUG", "ARCHITECTURE",
122 "GLOBALPTR", "TLS", "LOAD_CONFIG", "Bound IAT",
123 "IAT", "Delay IAT", "COM Descript", ""
124 };
125
126 static char *get_magic_type(WORD magic)
127 {
128 switch(magic) {
129 case IMAGE_NT_OPTIONAL_HDR32_MAGIC:
130 return "32bit";
131 case IMAGE_NT_OPTIONAL_HDR64_MAGIC:
132 return "64bit";
133 case IMAGE_ROM_OPTIONAL_HDR_MAGIC:
134 return "ROM";
135 }
136 return "???";
137 }
138
139 static inline void print_word(const char *title, WORD value)
140 {
141 printf(" %-34s 0x%-4X %u\n", title, value, value);
142 }
143
144 static inline void print_dword(const char *title, DWORD value)
145 {
146 printf(" %-34s 0x%-8lx %lu\n", title, value, value);
147 }
148
149 static inline void print_longlong(const char *title, ULONGLONG value)
150 {
151 printf(" %-34s 0x", title);
152 if(value >> 32)
153 printf("%lx%08lx\n", (unsigned long)(value >> 32), (unsigned long)value);
154 else
155 printf("%lx\n", (unsigned long)value);
156 }
157
158 static inline void print_ver(const char *title, BYTE major, BYTE minor)
159 {
160 printf(" %-34s %u.%02u\n", title, major, minor);
161 }
162
163 static inline void print_subsys(const char *title, WORD value)
164 {
165 const char *str;
166 switch (value)
167 {
168 default:
169 case IMAGE_SUBSYSTEM_UNKNOWN: str = "Unknown"; break;
170 case IMAGE_SUBSYSTEM_NATIVE: str = "Native"; break;
171 case IMAGE_SUBSYSTEM_WINDOWS_GUI: str = "Windows GUI"; break;
172 case IMAGE_SUBSYSTEM_WINDOWS_CUI: str = "Windows CUI"; break;
173 case IMAGE_SUBSYSTEM_OS2_CUI: str = "OS/2 CUI"; break;
174 case IMAGE_SUBSYSTEM_POSIX_CUI: str = "Posix CUI"; break;
175 }
176 printf(" %-34s 0x%X (%s)\n", title, value, str);
177 }
178
179 static inline void print_dllflags(const char *title, WORD value)
180 {
181 printf(" %-34s 0x%X\n", title, value);
182 #define X(f,s) if (value & f) printf(" %s\n", s)
183 X(IMAGE_DLLCHARACTERISTICS_NO_ISOLATION, "NO_ISOLATION");
184 X(IMAGE_DLLCHARACTERISTICS_NO_SEH, "NO_SEH");
185 X(IMAGE_DLLCHARACTERISTICS_NO_BIND, "NO_BIND");
186 X(IMAGE_DLLCHARACTERISTICS_WDM_DRIVER, "WDM_DRIVER");
187 X(IMAGE_DLLCHARACTERISTICS_TERMINAL_SERVER_AWARE, "TERMINAL_SERVER_AWARE");
188 #undef X
189 }
190
191 static inline void print_datadirectory(DWORD n, IMAGE_DATA_DIRECTORY *directory)
192 {
193 unsigned i;
194 printf("Data Directory\n");
195 printf("%ld\n", n * sizeof(IMAGE_DATA_DIRECTORY));
196
197 for (i = 0; i < n && i < 16; i++)
198 {
199 printf(" %-12s rva: 0x%-8lX size: %8lu\n",
200 DirectoryNames[i], directory[i].VirtualAddress,
201 directory[i].Size);
202 }
203 }
204
205 static void dump_optional_header32(IMAGE_OPTIONAL_HEADER32 *optionalHeader)
206 {
207 print_word("Magic", optionalHeader->Magic);
208 print_ver("linker version",
209 optionalHeader->MajorLinkerVersion, optionalHeader->MinorLinkerVersion);
210 print_dword("size of code", optionalHeader->SizeOfCode);
211 print_dword("size of initialized data", optionalHeader->SizeOfInitializedData);
212 print_dword("size of uninitialized data", optionalHeader->SizeOfUninitializedData);
213 print_dword("entrypoint RVA", optionalHeader->AddressOfEntryPoint);
214 print_dword("base of code", optionalHeader->BaseOfCode);
215 print_dword("base of data", optionalHeader->BaseOfData);
216 print_dword("image base", optionalHeader->ImageBase);
217 print_dword("section align", optionalHeader->SectionAlignment);
218 print_dword("file align", optionalHeader->FileAlignment);
219 print_ver("required OS version",
220 optionalHeader->MajorOperatingSystemVersion, optionalHeader->MinorOperatingSystemVersion);
221 print_ver("image version",
222 optionalHeader->MajorImageVersion, optionalHeader->MinorImageVersion);
223 print_ver("subsystem version",
224 optionalHeader->MajorSubsystemVersion, optionalHeader->MinorSubsystemVersion);
225 print_dword("Win32 Version", optionalHeader->Win32VersionValue);
226 print_dword("size of image", optionalHeader->SizeOfImage);
227 print_dword("size of headers", optionalHeader->SizeOfHeaders);
228 print_dword("checksum", optionalHeader->CheckSum);
229 print_subsys("Subsystem", optionalHeader->Subsystem);
230 print_dllflags("DLL characteristics:", optionalHeader->DllCharacteristics);
231 print_dword("stack reserve size", optionalHeader->SizeOfStackReserve);
232 print_dword("stack commit size", optionalHeader->SizeOfStackCommit);
233 print_dword("heap reserve size", optionalHeader->SizeOfHeapReserve);
234 print_dword("heap commit size", optionalHeader->SizeOfHeapCommit);
235 print_dword("loader flags", optionalHeader->LoaderFlags);
236 print_dword("RVAs & sizes", optionalHeader->NumberOfRvaAndSizes);
237 printf("\n");
238 print_datadirectory(optionalHeader->NumberOfRvaAndSizes, optionalHeader->DataDirectory);
239 }
240
241 static void dump_optional_header64(IMAGE_OPTIONAL_HEADER64 *optionalHeader)
242 {
243 print_word("Magic", optionalHeader->Magic);
244 print_ver("linker version",
245 optionalHeader->MajorLinkerVersion, optionalHeader->MinorLinkerVersion);
246 print_dword("size of code", optionalHeader->SizeOfCode);
247 print_dword("size of initialized data", optionalHeader->SizeOfInitializedData);
248 print_dword("size of uninitialized data", optionalHeader->SizeOfUninitializedData);
249 print_dword("entrypoint RVA", optionalHeader->AddressOfEntryPoint);
250 print_dword("base of code", optionalHeader->BaseOfCode);
251 print_longlong("image base", optionalHeader->ImageBase);
252 print_dword("section align", optionalHeader->SectionAlignment);
253 print_dword("file align", optionalHeader->FileAlignment);
254 print_ver("required OS version",
255 optionalHeader->MajorOperatingSystemVersion, optionalHeader->MinorOperatingSystemVersion);
256 print_ver("image version",
257 optionalHeader->MajorImageVersion, optionalHeader->MinorImageVersion);
258 print_ver("subsystem version",
259 optionalHeader->MajorSubsystemVersion, optionalHeader->MinorSubsystemVersion);
260 print_dword("Win32 Version", optionalHeader->Win32VersionValue);
261 print_dword("size of image", optionalHeader->SizeOfImage);
262 print_dword("size of headers", optionalHeader->SizeOfHeaders);
263 print_dword("checksum", optionalHeader->CheckSum);
264 print_subsys("Subsystem", optionalHeader->Subsystem);
265 print_dllflags("DLL characteristics:", optionalHeader->DllCharacteristics);
266 print_longlong("stack reserve size", optionalHeader->SizeOfStackReserve);
267 print_longlong("stack commit size", optionalHeader->SizeOfStackCommit);
268 print_longlong("heap reserve size", optionalHeader->SizeOfHeapReserve);
269 print_longlong("heap commit size", optionalHeader->SizeOfHeapCommit);
270 print_dword("loader flags", optionalHeader->LoaderFlags);
271 print_dword("RVAs & sizes", optionalHeader->NumberOfRvaAndSizes);
272 printf("\n");
273 print_datadirectory(optionalHeader->NumberOfRvaAndSizes, optionalHeader->DataDirectory);
274 }
275
276 static void dump_pe_header(void)
277 {
278 IMAGE_FILE_HEADER *fileHeader;
279
280 printf("File Header\n");
281 fileHeader = &PE_nt_headers->FileHeader;
282
283 printf(" Machine: %04X (%s)\n",
284 fileHeader->Machine, get_machine_str(fileHeader->Machine));
285 printf(" Number of Sections: %d\n", fileHeader->NumberOfSections);
286 printf(" TimeDateStamp: %08lX (%s) offset %lu\n",
287 fileHeader->TimeDateStamp, get_time_str(fileHeader->TimeDateStamp),
288 Offset(&(fileHeader->TimeDateStamp)));
289 printf(" PointerToSymbolTable: %08lX\n", fileHeader->PointerToSymbolTable);
290 printf(" NumberOfSymbols: %08lX\n", fileHeader->NumberOfSymbols);
291 printf(" SizeOfOptionalHeader: %04X\n", fileHeader->SizeOfOptionalHeader);
292 printf(" Characteristics: %04X\n", fileHeader->Characteristics);
293 #define X(f,s) if (fileHeader->Characteristics & f) printf(" %s\n", s)
294 X(IMAGE_FILE_RELOCS_STRIPPED, "RELOCS_STRIPPED");
295 X(IMAGE_FILE_EXECUTABLE_IMAGE, "EXECUTABLE_IMAGE");
296 X(IMAGE_FILE_LINE_NUMS_STRIPPED, "LINE_NUMS_STRIPPED");
297 X(IMAGE_FILE_LOCAL_SYMS_STRIPPED, "LOCAL_SYMS_STRIPPED");
298 X(IMAGE_FILE_AGGRESIVE_WS_TRIM, "AGGRESIVE_WS_TRIM");
299 X(IMAGE_FILE_LARGE_ADDRESS_AWARE, "LARGE_ADDRESS_AWARE");
300 X(IMAGE_FILE_16BIT_MACHINE, "16BIT_MACHINE");
301 X(IMAGE_FILE_BYTES_REVERSED_LO, "BYTES_REVERSED_LO");
302 X(IMAGE_FILE_32BIT_MACHINE, "32BIT_MACHINE");
303 X(IMAGE_FILE_DEBUG_STRIPPED, "DEBUG_STRIPPED");
304 X(IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP, "REMOVABLE_RUN_FROM_SWAP");
305 X(IMAGE_FILE_NET_RUN_FROM_SWAP, "NET_RUN_FROM_SWAP");
306 X(IMAGE_FILE_SYSTEM, "SYSTEM");
307 X(IMAGE_FILE_DLL, "DLL");
308 X(IMAGE_FILE_UP_SYSTEM_ONLY, "UP_SYSTEM_ONLY");
309 X(IMAGE_FILE_BYTES_REVERSED_HI, "BYTES_REVERSED_HI");
310 #undef X
311 printf("\n");
312
313 /* hope we have the right size */
314 printf("Optional Header (%s)\n", get_magic_type(PE_nt_headers->OptionalHeader.Magic));
315 switch(PE_nt_headers->OptionalHeader.Magic) {
316 case IMAGE_NT_OPTIONAL_HDR32_MAGIC:
317 dump_optional_header32((IMAGE_OPTIONAL_HEADER32*)&PE_nt_headers->OptionalHeader);
318 break;
319 case IMAGE_NT_OPTIONAL_HDR64_MAGIC:
320 dump_optional_header64((IMAGE_OPTIONAL_HEADER64*)&PE_nt_headers->OptionalHeader);
321 break;
322 default:
323 printf(" Unknown header magic: 0x%-4X\n", PE_nt_headers->OptionalHeader.Magic);
324 break;
325 }
326 printf("\n");
327 }
328
329 static void dump_sections(void* addr, unsigned num_sect)
330 {
331 IMAGE_SECTION_HEADER* sectHead = addr;
332 unsigned i;
333
334 printf("Section Table\n");
335 for (i = 0; i < num_sect; i++, sectHead++)
336 {
337 printf(" %02d %-8.8s VirtSize: %-8lu VirtAddr: %-8lu 0x%08lx\n",
338 i + 1, sectHead->Name, sectHead->Misc.VirtualSize, sectHead->VirtualAddress,
339 sectHead->VirtualAddress);
340 printf(" raw data offs: %-8lu raw data size: %-8lu\n",
341 sectHead->PointerToRawData, sectHead->SizeOfRawData);
342 printf(" relocation offs: %-8lu relocations: %-8u\n",
343 sectHead->PointerToRelocations, sectHead->NumberOfRelocations);
344 printf(" line # offs: %-8lu line #'s: %-8u\n",
345 sectHead->PointerToLinenumbers, sectHead->NumberOfLinenumbers);
346 printf(" characteristics: 0x%08lx\n", sectHead->Characteristics);
347 printf(" ");
348 #define X(b,s) if (sectHead->Characteristics & b) printf(s " ")
349 /* #define IMAGE_SCN_TYPE_REG 0x00000000 - Reserved */
350 /* #define IMAGE_SCN_TYPE_DSECT 0x00000001 - Reserved */
351 /* #define IMAGE_SCN_TYPE_NOLOAD 0x00000002 - Reserved */
352 /* #define IMAGE_SCN_TYPE_GROUP 0x00000004 - Reserved */
353 /* #define IMAGE_SCN_TYPE_NO_PAD 0x00000008 - Reserved */
354 /* #define IMAGE_SCN_TYPE_COPY 0x00000010 - Reserved */
355
356 X(IMAGE_SCN_CNT_CODE, "CODE");
357 X(IMAGE_SCN_CNT_INITIALIZED_DATA, "INITIALIZED_DATA");
358 X(IMAGE_SCN_CNT_UNINITIALIZED_DATA, "UNINITIALIZED_DATA");
359
360 X(IMAGE_SCN_LNK_OTHER, "LNK_OTHER");
361 X(IMAGE_SCN_LNK_INFO, "LNK_INFO");
362 /* #define IMAGE_SCN_TYPE_OVER 0x00000400 - Reserved */
363 X(IMAGE_SCN_LNK_REMOVE, "LNK_REMOVE");
364 X(IMAGE_SCN_LNK_COMDAT, "LNK_COMDAT");
365
366 /* 0x00002000 - Reserved */
367 /* #define IMAGE_SCN_MEM_PROTECTED 0x00004000 - Obsolete */
368 X(IMAGE_SCN_MEM_FARDATA, "MEM_FARDATA");
369
370 /* #define IMAGE_SCN_MEM_SYSHEAP 0x00010000 - Obsolete */
371 X(IMAGE_SCN_MEM_PURGEABLE, "MEM_PURGEABLE");
372 X(IMAGE_SCN_MEM_16BIT, "MEM_16BIT");
373 X(IMAGE_SCN_MEM_LOCKED, "MEM_LOCKED");
374 X(IMAGE_SCN_MEM_PRELOAD, "MEM_PRELOAD");
375
376 X(IMAGE_SCN_ALIGN_1BYTES, "ALIGN_1BYTES");
377 X(IMAGE_SCN_ALIGN_2BYTES, "ALIGN_2BYTES");
378 X(IMAGE_SCN_ALIGN_4BYTES, "ALIGN_4BYTES");
379 X(IMAGE_SCN_ALIGN_8BYTES, "ALIGN_8BYTES");
380 X(IMAGE_SCN_ALIGN_16BYTES, "ALIGN_16BYTES");
381 X(IMAGE_SCN_ALIGN_32BYTES, "ALIGN_32BYTES");
382 X(IMAGE_SCN_ALIGN_64BYTES, "ALIGN_64BYTES");
383 /* 0x00800000 - Unused */
384
385 X(IMAGE_SCN_LNK_NRELOC_OVFL, "LNK_NRELOC_OVFL");
386
387 X(IMAGE_SCN_MEM_DISCARDABLE, "MEM_DISCARDABLE");
388 X(IMAGE_SCN_MEM_NOT_CACHED, "MEM_NOT_CACHED");
389 X(IMAGE_SCN_MEM_NOT_PAGED, "MEM_NOT_PAGED");
390 X(IMAGE_SCN_MEM_SHARED, "MEM_SHARED");
391 X(IMAGE_SCN_MEM_EXECUTE, "MEM_EXECUTE");
392 X(IMAGE_SCN_MEM_READ, "MEM_READ");
393 X(IMAGE_SCN_MEM_WRITE, "MEM_WRITE");
394 #undef X
395 printf("\n\n");
396 }
397 printf("\n");
398 }
399
400 static void dump_dir_exported_functions(void)
401 {
402 unsigned int size = 0;
403 IMAGE_EXPORT_DIRECTORY *exportDir = get_dir_and_size(IMAGE_FILE_EXPORT_DIRECTORY, &size);
404 unsigned int i;
405 DWORD* pFunc;
406 DWORD* pName;
407 WORD* pOrdl;
408 DWORD* map;
409 parsed_symbol symbol;
410
411 if (!exportDir) return;
412
413 printf("Exports table:\n");
414 printf("\n");
415 printf(" Name: %s\n", (char*)RVA(exportDir->Name, sizeof(DWORD)));
416 printf(" Characteristics: %08lx\n", exportDir->Characteristics);
417 printf(" TimeDateStamp: %08lX %s\n",
418 exportDir->TimeDateStamp, get_time_str(exportDir->TimeDateStamp));
419 printf(" Version: %u.%02u\n", exportDir->MajorVersion, exportDir->MinorVersion);
420 printf(" Ordinal base: %lu\n", exportDir->Base);
421 printf(" # of functions: %lu\n", exportDir->NumberOfFunctions);
422 printf(" # of Names: %lu\n", exportDir->NumberOfNames);
423 printf("Addresses of functions: %08lX\n", exportDir->AddressOfFunctions);
424 printf("Addresses of name ordinals: %08lX\n", exportDir->AddressOfNameOrdinals);
425 printf("Addresses of names: %08lX\n", exportDir->AddressOfNames);
426 printf("\n");
427 printf(" Entry Pt Ordn Name\n");
428
429 pFunc = RVA(exportDir->AddressOfFunctions, exportDir->NumberOfFunctions * sizeof(DWORD));
430 if (!pFunc) {printf("Can't grab functions' address table\n"); return;}
431 pName = RVA(exportDir->AddressOfNames, exportDir->NumberOfNames * sizeof(DWORD));
432 if (!pName) {printf("Can't grab functions' name table\n"); return;}
433 pOrdl = RVA(exportDir->AddressOfNameOrdinals, exportDir->NumberOfNames * sizeof(WORD));
434 if (!pOrdl) {printf("Can't grab functions' ordinal table\n"); return;}
435
436 /* bit map of used funcs */
437 map = calloc(((exportDir->NumberOfFunctions + 31) & ~31) / 32, sizeof(DWORD));
438 if (!map) fatal("no memory");
439
440 for (i = 0; i < exportDir->NumberOfNames; i++, pName++, pOrdl++)
441 {
442 char* name;
443
444 map[*pOrdl / 32] |= 1 << (*pOrdl % 32);
445
446 name = (char*)RVA(*pName, sizeof(DWORD));
447 if (name && globals.do_demangle)
448 {
449 printf(" %08lX %4lu ", pFunc[*pOrdl], exportDir->Base + *pOrdl);
450
451 symbol_init(&symbol, name);
452 if (symbol_demangle(&symbol) == -1)
453 printf(name);
454 else if (symbol.flags & SYM_DATA)
455 printf(symbol.arg_text[0]);
456 else
457 output_prototype(stdout, &symbol);
458 symbol_clear(&symbol);
459 }
460 else
461 {
462 printf(" %08lX %4lu %s", pFunc[*pOrdl], exportDir->Base + *pOrdl, name);
463 }
464 /* check for forwarded function */
465 if ((char *)RVA(pFunc[*pOrdl],sizeof(void*)) >= (char *)exportDir &&
466 (char *)RVA(pFunc[*pOrdl],sizeof(void*)) < (char *)exportDir + size)
467 printf( " (-> %s)", (char *)RVA(pFunc[*pOrdl],1));
468 printf("\n");
469 }
470 pFunc = RVA(exportDir->AddressOfFunctions, exportDir->NumberOfFunctions * sizeof(DWORD));
471 if (!pFunc) {printf("Can't grab functions' address table\n"); return;}
472 for (i = 0; i < exportDir->NumberOfFunctions; i++)
473 {
474 if (pFunc[i] && !(map[i / 32] & (1 << (i % 32))))
475 {
476 printf(" %08lX %4lu <by ordinal>\n", pFunc[i], exportDir->Base + i);
477 }
478 }
479 free(map);
480 printf("\n");
481 }
482
483 static void dump_image_thunk_data64(IMAGE_THUNK_DATA64 *il)
484 {
485 /* FIXME: This does not properly handle large images */
486 IMAGE_IMPORT_BY_NAME* iibn;
487 for (; il->u1.Ordinal; il++)
488 {
489 if (IMAGE_SNAP_BY_ORDINAL64(il->u1.Ordinal))
490 printf(" %4lu <by ordinal>\n", (DWORD)IMAGE_ORDINAL64(il->u1.Ordinal));
491 else
492 {
493 iibn = RVA((DWORD)il->u1.AddressOfData, sizeof(DWORD));
494 if (!il)
495 printf("Can't grab import by name info, skipping to next ordinal\n");
496 else
497 printf(" %4u %s %lx\n", iibn->Hint, iibn->Name, (DWORD)il->u1.AddressOfData);
498 }
499 }
500 }
501
502 static void dump_image_thunk_data32(IMAGE_THUNK_DATA32 *il)
503 {
504 IMAGE_IMPORT_BY_NAME* iibn;
505 for (; il->u1.Ordinal; il++)
506 {
507 if (IMAGE_SNAP_BY_ORDINAL32(il->u1.Ordinal))
508 printf(" %4lu <by ordinal>\n", IMAGE_ORDINAL32(il->u1.Ordinal));
509 else
510 {
511 iibn = RVA((DWORD)il->u1.AddressOfData, sizeof(DWORD));
512 if (!il)
513 printf("Can't grab import by name info, skipping to next ordinal\n");
514 else
515 printf(" %4u %s %lx\n", iibn->Hint, iibn->Name, (DWORD)il->u1.AddressOfData);
516 }
517 }
518 }
519
520 static void dump_dir_imported_functions(void)
521 {
522 IMAGE_IMPORT_DESCRIPTOR *importDesc = get_dir(IMAGE_FILE_IMPORT_DIRECTORY);
523 unsigned nb_imp, i;
524 DWORD directorySize;
525
526 if (!importDesc) return;
527 if(PE_nt_headers->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
528 {
529 IMAGE_OPTIONAL_HEADER64 *opt = (IMAGE_OPTIONAL_HEADER64*)&PE_nt_headers->OptionalHeader;
530 directorySize = opt->DataDirectory[IMAGE_FILE_IMPORT_DIRECTORY].Size;
531 }
532 else
533 {
534 IMAGE_OPTIONAL_HEADER32 *opt = (IMAGE_OPTIONAL_HEADER32*)&PE_nt_headers->OptionalHeader;
535 directorySize = opt->DataDirectory[IMAGE_FILE_IMPORT_DIRECTORY].Size;
536 }
537 nb_imp = directorySize / sizeof(*importDesc);
538 if (!nb_imp) return;
539
540 printf("Import Table size: %lu\n", directorySize);/* FIXME */
541
542 for (i = 0; i < nb_imp - 1; i++) /* the last descr is set as 0 as a sentinel */
543 {
544 IMAGE_THUNK_DATA32* il;
545
546 if (!importDesc->Name ||
547 (importDesc->u.OriginalFirstThunk == 0 && importDesc->FirstThunk == 0))
548 {
549 /* FIXME */
550 printf("<<<<<<<null entry\n");
551 break;
552 }
553 printf(" offset %lu %s\n", Offset(importDesc), (char*)RVA(importDesc->Name, sizeof(DWORD)));
554 printf(" Hint/Name Table: %08lX\n", (DWORD)importDesc->u.OriginalFirstThunk);
555 printf(" TimeDataStamp: %08lX (%s)\n",
556 importDesc->TimeDateStamp, get_time_str(importDesc->TimeDateStamp));
557 printf(" ForwarderChain: %08lX\n", importDesc->ForwarderChain);
558 printf(" First thunk RVA: %08lX (delta: %u 0x%x)\n",
559 (DWORD)importDesc->FirstThunk, -1, -1); /* FIXME */
560
561 printf(" Ordn Name\n");
562
563 il = (importDesc->u.OriginalFirstThunk != 0) ?
564 RVA((DWORD)importDesc->u.OriginalFirstThunk, sizeof(DWORD)) :
565 RVA((DWORD)importDesc->FirstThunk, sizeof(DWORD));
566
567 if (!il) {printf("Can't grab thunk data, going to next imported DLL\n"); continue;}
568
569 if(PE_nt_headers->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
570 dump_image_thunk_data64((IMAGE_THUNK_DATA64*)il);
571 else
572 dump_image_thunk_data32(il);
573 printf("\n");
574 importDesc++;
575 }
576 printf("\n");
577 }
578
579 static void dump_dir_debug_dir(IMAGE_DEBUG_DIRECTORY* idd, int idx)
580 {
581 const char* str;
582
583 printf("Directory %02u\n", idx + 1);
584 printf(" Characteristics: %08lX\n", idd->Characteristics);
585 printf(" TimeDateStamp: %08lX %s\n",
586 idd->TimeDateStamp, get_time_str(idd->TimeDateStamp));
587 printf(" Version %u.%02u\n", idd->MajorVersion, idd->MinorVersion);
588 switch (idd->Type)
589 {
590 default:
591 case IMAGE_DEBUG_TYPE_UNKNOWN: str = "UNKNOWN"; break;
592 case IMAGE_DEBUG_TYPE_COFF: str = "COFF"; break;
593 case IMAGE_DEBUG_TYPE_CODEVIEW: str = "CODEVIEW"; break;
594 case IMAGE_DEBUG_TYPE_FPO: str = "FPO"; break;
595 case IMAGE_DEBUG_TYPE_MISC: str = "MISC"; break;
596 case IMAGE_DEBUG_TYPE_EXCEPTION: str = "EXCEPTION"; break;
597 case IMAGE_DEBUG_TYPE_FIXUP: str = "FIXUP"; break;
598 case IMAGE_DEBUG_TYPE_OMAP_TO_SRC: str = "OMAP_TO_SRC"; break;
599 case IMAGE_DEBUG_TYPE_OMAP_FROM_SRC:str = "OMAP_FROM_SRC"; break;
600 case IMAGE_DEBUG_TYPE_BORLAND: str = "BORLAND"; break;
601 case IMAGE_DEBUG_TYPE_RESERVED10: str = "RESERVED10"; break;
602 }
603 printf(" Type: %lu (%s)\n", idd->Type, str);
604 printf(" SizeOfData: %lu\n", idd->SizeOfData);
605 printf(" AddressOfRawData: %08lX\n", idd->AddressOfRawData);
606 printf(" PointerToRawData: %08lX\n", idd->PointerToRawData);
607
608 switch (idd->Type)
609 {
610 case IMAGE_DEBUG_TYPE_UNKNOWN:
611 break;
612 case IMAGE_DEBUG_TYPE_COFF:
613 dump_coff(idd->PointerToRawData, idd->SizeOfData,
614 (char*)PE_nt_headers + sizeof(DWORD) + sizeof(IMAGE_FILE_HEADER) + PE_nt_headers->FileHeader.SizeOfOptionalHeader);
615 break;
616 case IMAGE_DEBUG_TYPE_CODEVIEW:
617 dump_codeview(idd->PointerToRawData, idd->SizeOfData);
618 break;
619 case IMAGE_DEBUG_TYPE_FPO:
620 dump_frame_pointer_omission(idd->PointerToRawData, idd->SizeOfData);
621 break;
622 case IMAGE_DEBUG_TYPE_MISC:
623 {
624 IMAGE_DEBUG_MISC* misc = PRD(idd->PointerToRawData, idd->SizeOfData);
625 if (!misc) {printf("Can't get misc debug information\n"); break;}
626 printf(" DataType: %lu (%s)\n",
627 misc->DataType,
628 (misc->DataType == IMAGE_DEBUG_MISC_EXENAME) ? "Exe name" : "Unknown");
629 printf(" Length: %lu\n", misc->Length);
630 printf(" Unicode: %s\n", misc->Unicode ? "Yes" : "No");
631 printf(" Data: %s\n", misc->Data);
632 }
633 break;
634 case IMAGE_DEBUG_TYPE_EXCEPTION:
635 break;
636 case IMAGE_DEBUG_TYPE_FIXUP:
637 break;
638 case IMAGE_DEBUG_TYPE_OMAP_TO_SRC:
639 break;
640 case IMAGE_DEBUG_TYPE_OMAP_FROM_SRC:
641 break;
642 case IMAGE_DEBUG_TYPE_BORLAND:
643 break;
644 case IMAGE_DEBUG_TYPE_RESERVED10:
645 break;
646 }
647 printf("\n");
648 }
649
650 static void dump_dir_debug(void)
651 {
652 IMAGE_DEBUG_DIRECTORY* debugDir = get_dir(IMAGE_FILE_DEBUG_DIRECTORY);
653 unsigned nb_dbg, i;
654
655 if (!debugDir) return;
656 nb_dbg = PE_nt_headers->OptionalHeader.DataDirectory[IMAGE_FILE_DEBUG_DIRECTORY].Size /
657 sizeof(*debugDir);
658 if (!nb_dbg) return;
659
660 printf("Debug Table (%u directories)\n", nb_dbg);
661
662 for (i = 0; i < nb_dbg; i++)
663 {
664 dump_dir_debug_dir(debugDir, i);
665 debugDir++;
666 }
667 printf("\n");
668 }
669
670 static void dump_dir_tls(void)
671 {
672 IMAGE_TLS_DIRECTORY64 dir;
673 const DWORD *callbacks;
674 const IMAGE_TLS_DIRECTORY32 *pdir = get_dir(IMAGE_FILE_THREAD_LOCAL_STORAGE);
675
676 if (!pdir) return;
677
678 if(PE_nt_headers->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
679 memcpy(&dir, pdir, sizeof(dir));
680 else
681 {
682 dir.StartAddressOfRawData = pdir->StartAddressOfRawData;
683 dir.EndAddressOfRawData = pdir->EndAddressOfRawData;
684 dir.AddressOfIndex = pdir->AddressOfIndex;
685 dir.AddressOfCallBacks = pdir->AddressOfCallBacks;
686 dir.SizeOfZeroFill = pdir->SizeOfZeroFill;
687 dir.Characteristics = pdir->Characteristics;
688 }
689
690 /* FIXME: This does not properly handle large images */
691 printf( "Thread Local Storage\n" );
692 printf( " Raw data %08lx-%08lx (data size %lx zero fill size %lx)\n",
693 (DWORD)dir.StartAddressOfRawData, (DWORD)dir.EndAddressOfRawData,
694 (DWORD)(dir.EndAddressOfRawData - dir.StartAddressOfRawData),
695 (DWORD)dir.SizeOfZeroFill );
696 printf( " Index address %08lx\n", (DWORD)dir.AddressOfIndex );
697 printf( " Characteristics %08lx\n", dir.Characteristics );
698 printf( " Callbacks %08lx -> {", (DWORD)dir.AddressOfCallBacks );
699 if (dir.AddressOfCallBacks)
700 {
701 DWORD addr = (DWORD)dir.AddressOfCallBacks - PE_nt_headers->OptionalHeader.ImageBase;
702 while ((callbacks = RVA(addr, sizeof(DWORD))) && *callbacks)
703 {
704 printf( " %08lx", *callbacks );
705 addr += sizeof(DWORD);
706 }
707 }
708 printf(" }\n\n");
709 }
710
711 void dump_separate_dbg(void)
712 {
713 IMAGE_SEPARATE_DEBUG_HEADER*separateDebugHead = PRD(0, sizeof(separateDebugHead));
714 unsigned nb_dbg;
715 unsigned i;
716 IMAGE_DEBUG_DIRECTORY* debugDir;
717
718 if (!separateDebugHead) {printf("Can't grab the separate header, aborting\n"); return;}
719
720 printf ("Signature: %.2s (0x%4X)\n",
721 (char*)&separateDebugHead->Signature, separateDebugHead->Signature);
722 printf ("Flags: 0x%04X\n", separateDebugHead->Flags);
723 printf ("Machine: 0x%04X (%s)\n",
724 separateDebugHead->Machine, get_machine_str(separateDebugHead->Machine));
725 printf ("Characteristics: 0x%04X\n", separateDebugHead->Characteristics);
726 printf ("TimeDateStamp: 0x%08lX (%s)\n",
727 separateDebugHead->TimeDateStamp, get_time_str(separateDebugHead->TimeDateStamp));
728 printf ("CheckSum: 0x%08lX\n", separateDebugHead->CheckSum);
729 printf ("ImageBase: 0x%08lX\n", separateDebugHead->ImageBase);
730 printf ("SizeOfImage: 0x%08lX\n", separateDebugHead->SizeOfImage);
731 printf ("NumberOfSections: 0x%08lX\n", separateDebugHead->NumberOfSections);
732 printf ("ExportedNamesSize: 0x%08lX\n", separateDebugHead->ExportedNamesSize);
733 printf ("DebugDirectorySize: 0x%08lX\n", separateDebugHead->DebugDirectorySize);
734
735 if (!PRD(sizeof(IMAGE_SEPARATE_DEBUG_HEADER),
736 separateDebugHead->NumberOfSections * sizeof(IMAGE_SECTION_HEADER)))
737 {printf("Can't get the sections, aborting\n"); return;}
738
739 dump_sections(separateDebugHead + 1, separateDebugHead->NumberOfSections);
740
741 nb_dbg = separateDebugHead->DebugDirectorySize / sizeof(IMAGE_DEBUG_DIRECTORY);
742 debugDir = PRD(sizeof(IMAGE_SEPARATE_DEBUG_HEADER) +
743 separateDebugHead->NumberOfSections * sizeof(IMAGE_SECTION_HEADER) +
744 separateDebugHead->ExportedNamesSize,
745 nb_dbg * sizeof(IMAGE_DEBUG_DIRECTORY));
746 if (!debugDir) {printf("Couldn't get the debug directory info, aborting\n");return;}
747
748 printf("Debug Table (%u directories)\n", nb_dbg);
749
750 for (i = 0; i < nb_dbg; i++)
751 {
752 dump_dir_debug_dir(debugDir, i);
753 debugDir++;
754 }
755 }
756
757 static const char *get_resource_type( unsigned int id )
758 {
759 static const char *types[] =
760 {
761 NULL,
762 "CURSOR",
763 "BITMAP",
764 "ICON",
765 "MENU",
766 "DIALOG",
767 "STRING",
768 "FONTDIR",
769 "FONT",
770 "ACCELERATOR",
771 "RCDATA",
772 "MESSAGETABLE",
773 "GROUP_CURSOR",
774 NULL,
775 "GROUP_ICON",
776 NULL,
777 "VERSION",
778 "DLGINCLUDE",
779 NULL,
780 "PLUGPLAY",
781 "VXD",
782 "ANICURSOR",
783 "ANIICON",
784 "HTML"
785 };
786
787 if ((size_t)id < sizeof(types)/sizeof(types[0])) return types[id];
788 return NULL;
789 }
790
791 /* dump an ASCII string with proper escaping */
792 static int dump_strA( const unsigned char *str, size_t len )
793 {
794 static const char escapes[32] = ".......abtnvfr.............e....";
795 char buffer[256];
796 char *pos = buffer;
797 int count = 0;
798
799 for (; len; str++, len--)
800 {
801 if (pos > buffer + sizeof(buffer) - 8)
802 {
803 fwrite( buffer, pos - buffer, 1, stdout );
804 count += pos - buffer;
805 pos = buffer;
806 }
807 if (*str > 127) /* hex escape */
808 {
809 pos += sprintf( pos, "\\x%02x", *str );
810 continue;
811 }
812 if (*str < 32) /* octal or C escape */
813 {
814 if (!*str && len == 1) continue; /* do not output terminating NULL */
815 if (escapes[*str] != '.')
816 pos += sprintf( pos, "\\%c", escapes[*str] );
817 else if (len > 1 && str[1] >= '0' && str[1] <= '7')
818 pos += sprintf( pos, "\\%03o", *str );
819 else
820 pos += sprintf( pos, "\\%o", *str );
821 continue;
822 }
823 if (*str == '\\') *pos++ = '\\';
824 *pos++ = *str;
825 }
826 fwrite( buffer, pos - buffer, 1, stdout );
827 count += pos - buffer;
828 return count;
829 }
830
831 /* dump a Unicode string with proper escaping */
832 static int dump_strW( const WCHAR *str, size_t len )
833 {
834 static const char escapes[32] = ".......abtnvfr.............e....";
835 char buffer[256];
836 char *pos = buffer;
837 int count = 0;
838
839 for (; len; str++, len--)
840 {
841 if (pos > buffer + sizeof(buffer) - 8)
842 {
843 fwrite( buffer, pos - buffer, 1, stdout );
844 count += pos - buffer;
845 pos = buffer;
846 }
847 if (*str > 127) /* hex escape */
848 {
849 if (len > 1 && str[1] < 128 && isxdigit((char)str[1]))
850 pos += sprintf( pos, "\\x%04x", *str );
851 else
852 pos += sprintf( pos, "\\x%x", *str );
853 continue;
854 }
855 if (*str < 32) /* octal or C escape */
856 {
857 if (!*str && len == 1) continue; /* do not output terminating NULL */
858 if (escapes[*str] != '.')
859 pos += sprintf( pos, "\\%c", escapes[*str] );
860 else if (len > 1 && str[1] >= '0' && str[1] <= '7')
861 pos += sprintf( pos, "\\%03o", *str );
862 else
863 pos += sprintf( pos, "\\%o", *str );
864 continue;
865 }
866 if (*str == '\\') *pos++ = '\\';
867 *pos++ = *str;
868 }
869 fwrite( buffer, pos - buffer, 1, stdout );
870 count += pos - buffer;
871 return count;
872 }
873
874 /* dump data for a STRING resource */
875 static void dump_string_data( const WCHAR *ptr, unsigned int size, unsigned int id, const char *prefix )
876 {
877 int i;
878
879 for (i = 0; i < 16 && size; i++)
880 {
881 unsigned len = *ptr++;
882
883 if (len >= size)
884 {
885 len = size;
886 size = 0;
887 }
888 else size -= len + 1;
889
890 if (len)
891 {
892 printf( "%s%04x \"", prefix, (id - 1) * 16 + i );
893 dump_strW( ptr, len );
894 printf( "\"\n" );
895 ptr += len;
896 }
897 }
898 }
899
900 /* dump data for a MESSAGETABLE resource */
901 static void dump_msgtable_data( const void *ptr, unsigned int size, unsigned int id, const char *prefix )
902 {
903 const MESSAGE_RESOURCE_DATA *data = ptr;
904 const MESSAGE_RESOURCE_BLOCK *block = data->Blocks;
905 unsigned i, j;
906
907 for (i = 0; i < data->NumberOfBlocks; i++, block++)
908 {
909 const MESSAGE_RESOURCE_ENTRY *entry;
910
911 entry = (const MESSAGE_RESOURCE_ENTRY *)((const char *)data + block->OffsetToEntries);
912 for (j = block->LowId; j <= block->HighId; j++)
913 {
914 if (entry->Flags & MESSAGE_RESOURCE_UNICODE)
915 {
916 const WCHAR *str = (const WCHAR *)entry->Text;
917 printf( "%s%08x L\"", prefix, j );
918 dump_strW( str, strlenW(str) );
919 printf( "\"\n" );
920 }
921 else
922 {
923 printf( "%s%08x \"", prefix, j );
924 dump_strA( entry->Text, strlen(entry->Text) );
925 printf( "\"\n" );
926 }
927 entry = (const MESSAGE_RESOURCE_ENTRY *)((const char *)entry + entry->Length);
928 }
929 }
930 }
931
932 static void dump_dir_resource(void)
933 {
934 const IMAGE_RESOURCE_DIRECTORY *root = get_dir(IMAGE_FILE_RESOURCE_DIRECTORY);
935 const IMAGE_RESOURCE_DIRECTORY *namedir;
936 const IMAGE_RESOURCE_DIRECTORY *langdir;
937 const IMAGE_RESOURCE_DIRECTORY_ENTRY *e1, *e2, *e3;
938 const IMAGE_RESOURCE_DIR_STRING_U *string;
939 const IMAGE_RESOURCE_DATA_ENTRY *data;
940 int i, j, k;
941
942 if (!root) return;
943
944 printf( "Resources:" );
945
946 for (i = 0; i< root->NumberOfNamedEntries + root->NumberOfIdEntries; i++)
947 {
948 e1 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(root + 1) + i;
949 namedir = (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + e1->u2.s3.OffsetToDirectory);
950 for (j = 0; j < namedir->NumberOfNamedEntries + namedir->NumberOfIdEntries; j++)
951 {
952 e2 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(namedir + 1) + j;
953 langdir = (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + e2->u2.s3.OffsetToDirectory);
954 for (k = 0; k < langdir->NumberOfNamedEntries + langdir->NumberOfIdEntries; k++)
955 {
956 e3 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(langdir + 1) + k;
957
958 printf( "\n " );
959 if (e1->u1.s1.NameIsString)
960 {
961 string = (const IMAGE_RESOURCE_DIR_STRING_U*)((const char *)root + e1->u1.s1.NameOffset);
962 dump_unicode_str( string->NameString, string->Length );
963 }
964 else
965 {
966 const char *type = get_resource_type( e1->u1.s2.Id );
967 if (type) printf( "%s", type );
968 else printf( "%04x", e1->u1.s2.Id );
969 }
970
971 printf( " Name=" );
972 if (e2->u1.s1.NameIsString)
973 {
974 string = (const IMAGE_RESOURCE_DIR_STRING_U*) ((const char *)root + e2->u1.s1.NameOffset);
975 dump_unicode_str( string->NameString, string->Length );
976 }
977 else
978 printf( "%04x", e2->u1.s2.Id );
979
980 printf( " Language=%04x:\n", e3->u1.s2.Id );
981 data = (const IMAGE_RESOURCE_DATA_ENTRY *)((const char *)root + e3->u2.OffsetToData);
982 if (e1->u1.s1.NameIsString)
983 {
984 dump_data( RVA( data->OffsetToData, data->Size ), data->Size, " " );
985 }
986 else switch(e1->u1.s2.Id)
987 {
988 case 6:
989 dump_string_data( RVA( data->OffsetToData, data->Size ), data->Size,
990 e2->u1.s2.Id, " " );
991 break;
992 case 11:
993 dump_msgtable_data( RVA( data->OffsetToData, data->Size ), data->Size,
994 e2->u1.s2.Id, " " );
995 break;
996 default:
997 dump_data( RVA( data->OffsetToData, data->Size ), data->Size, " " );
998 break;
999 }
1000 }
1001 }
1002 }
1003 printf( "\n\n" );
1004 }
1005
1006 void pe_dump(void* pmt)
1007 {
1008 int all = (globals.dumpsect != NULL) && strcmp(globals.dumpsect, "ALL") == 0;
1009
1010 PE_nt_headers = pmt;
1011 if (globals.do_dumpheader)
1012 {
1013 dump_pe_header();
1014 /* FIXME: should check ptr */
1015 dump_sections((char*)PE_nt_headers + sizeof(DWORD) +
1016 sizeof(IMAGE_FILE_HEADER) + PE_nt_headers->FileHeader.SizeOfOptionalHeader,
1017 PE_nt_headers->FileHeader.NumberOfSections);
1018 }
1019 else if (!globals.dumpsect)
1020 {
1021 /* show at least something here */
1022 dump_pe_header();
1023 }
1024
1025 if (globals.dumpsect)
1026 {
1027 if (all || !strcmp(globals.dumpsect, "import"))
1028 dump_dir_imported_functions();
1029 if (all || !strcmp(globals.dumpsect, "export"))
1030 dump_dir_exported_functions();
1031 if (all || !strcmp(globals.dumpsect, "debug"))
1032 dump_dir_debug();
1033 if (all || !strcmp(globals.dumpsect, "resource"))
1034 dump_dir_resource();
1035 if (all || !strcmp(globals.dumpsect, "tls"))
1036 dump_dir_tls();
1037 #if 0
1038 /* FIXME: not implemented yet */
1039 if (all || !strcmp(globals.dumpsect, "reloc"))
1040 dump_dir_reloc();
1041 #endif
1042 }
1043 }
1044
1045 typedef struct _dll_symbol {
1046 size_t ordinal;
1047 char *symbol;
1048 } dll_symbol;
1049
1050 static dll_symbol *dll_symbols = NULL;
1051 static dll_symbol *dll_current_symbol = NULL;
1052
1053 /* Compare symbols by ordinal for qsort */
1054 static int symbol_cmp(const void *left, const void *right)
1055 {
1056 return ((const dll_symbol *)left)->ordinal > ((const dll_symbol *)right)->ordinal;
1057 }
1058
1059 /*******************************************************************
1060 * dll_close
1061 *
1062 * Free resources used by DLL
1063 */
1064 /* FIXME: Not used yet
1065 static void dll_close (void)
1066 {
1067 dll_symbol* ds;
1068
1069 if (!dll_symbols) {
1070 fatal("No symbols");
1071 }
1072 for (ds = dll_symbols; ds->symbol; ds++)
1073 free(ds->symbol);
1074 free (dll_symbols);
1075 dll_symbols = NULL;
1076 }
1077 */
1078
1079 static void do_grab_sym( enum FileSig sig, void* pmt )
1080 {
1081 IMAGE_EXPORT_DIRECTORY *exportDir;
1082 unsigned i, j;
1083 DWORD* pName;
1084 DWORD* pFunc;
1085 WORD* pOrdl;
1086 const char* ptr;
1087 DWORD* map;
1088
1089 PE_nt_headers = pmt;
1090 if (!(exportDir = get_dir(IMAGE_FILE_EXPORT_DIRECTORY))) return;
1091
1092 pName = RVA(exportDir->AddressOfNames, exportDir->NumberOfNames * sizeof(DWORD));
1093 if (!pName) {printf("Can't grab functions' name table\n"); return;}
1094 pOrdl = RVA(exportDir->AddressOfNameOrdinals, exportDir->NumberOfNames * sizeof(WORD));
1095 if (!pOrdl) {printf("Can't grab functions' ordinal table\n"); return;}
1096
1097 /* dll_close(); */
1098
1099 if (!(dll_symbols = (dll_symbol *) malloc((exportDir->NumberOfFunctions + 1) *
1100 sizeof (dll_symbol))))
1101 fatal ("Out of memory");
1102 if (exportDir->AddressOfFunctions != exportDir->NumberOfNames || exportDir->Base > 1)
1103 globals.do_ordinals = 1;
1104
1105 /* bit map of used funcs */
1106 map = calloc(((exportDir->NumberOfFunctions + 31) & ~31) / 32, sizeof(DWORD));
1107 if (!map) fatal("no memory");
1108
1109 for (j = 0; j < exportDir->NumberOfNames; j++, pOrdl++)
1110 {
1111 map[*pOrdl / 32] |= 1 << (*pOrdl % 32);
1112 ptr = RVA(*pName++, sizeof(DWORD));
1113 if (!ptr) ptr = "cant_get_function";
1114 dll_symbols[j].symbol = strdup(ptr);
1115 dll_symbols[j].ordinal = exportDir->Base + *pOrdl;
1116 assert(dll_symbols[j].symbol);
1117 }
1118 pFunc = RVA(exportDir->AddressOfFunctions, exportDir->NumberOfFunctions * sizeof(DWORD));
1119 if (!pFunc) {printf("Can't grab functions' address table\n"); return;}
1120
1121 for (i = 0; i < exportDir->NumberOfFunctions; i++)
1122 {
1123 if (pFunc[i] && !(map[i / 32] & (1 << (i % 32))))
1124 {
1125 char ordinal_text[256];
1126 /* Ordinal only entry */
1127 snprintf (ordinal_text, sizeof(ordinal_text), "%s_%lu",
1128 globals.forward_dll ? globals.forward_dll : OUTPUT_UC_DLL_NAME,
1129 exportDir->Base + i);
1130 str_toupper(ordinal_text);
1131 dll_symbols[j].symbol = strdup(ordinal_text);
1132 assert(dll_symbols[j].symbol);
1133 dll_symbols[j].ordinal = exportDir->Base + i;
1134 j++;
1135 assert(j <= exportDir->NumberOfFunctions);
1136 }
1137 }
1138 free(map);
1139
1140 if (NORMAL)
1141 printf("%lu named symbols in DLL, %lu total, %d unique (ordinal base = %ld)\n",
1142 exportDir->NumberOfNames, exportDir->NumberOfFunctions, j, exportDir->Base);
1143
1144 qsort( dll_symbols, j, sizeof(dll_symbol), symbol_cmp );
1145
1146 dll_symbols[j].symbol = NULL;
1147
1148 dll_current_symbol = dll_symbols;
1149 }
1150
1151 /*******************************************************************
1152 * dll_open
1153 *
1154 * Open a DLL and read in exported symbols
1155 */
1156 int dll_open (const char *dll_name)
1157 {
1158 return dump_analysis(dll_name, do_grab_sym, SIG_PE);
1159 }
1160
1161 /*******************************************************************
1162 * dll_next_symbol
1163 *
1164 * Get next exported symbol from dll
1165 */
1166 int dll_next_symbol (parsed_symbol * sym)
1167 {
1168 if (!dll_current_symbol->symbol)
1169 return 1;
1170
1171 assert (dll_symbols);
1172
1173 sym->symbol = strdup (dll_current_symbol->symbol);
1174 sym->ordinal = dll_current_symbol->ordinal;
1175 dll_current_symbol++;
1176 return 0;
1177 }