2 * File pe_module.c - handle PE module information
4 * Copyright (C) 1996, Eric Youngdale.
5 * Copyright (C) 1999-2000, Ulrich Weigand.
6 * Copyright (C) 2004-2007, Eric Pouech.
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 #include "dbghelp_private.h"
26 WINE_DEFAULT_DEBUG_CHANNEL(dbghelp
);
30 struct image_file_map fmap
;
33 static void* pe_map_full(struct image_file_map
* fmap
, IMAGE_NT_HEADERS
** nth
)
35 if (!fmap
->u
.pe
.full_map
)
37 fmap
->u
.pe
.full_map
= MapViewOfFile(fmap
->u
.pe
.hMap
, FILE_MAP_READ
, 0, 0, 0);
39 if (fmap
->u
.pe
.full_map
)
41 if (nth
) *nth
= RtlImageNtHeader(fmap
->u
.pe
.full_map
);
42 fmap
->u
.pe
.full_count
++;
43 return fmap
->u
.pe
.full_map
;
48 static void pe_unmap_full(struct image_file_map
* fmap
)
50 if (fmap
->u
.pe
.full_count
&& !--fmap
->u
.pe
.full_count
)
52 UnmapViewOfFile(fmap
->u
.pe
.full_map
);
53 fmap
->u
.pe
.full_map
= NULL
;
57 /******************************************************************
60 * Maps a single section into memory from an PE file
62 const char* pe_map_section(struct image_section_map
* ism
)
65 struct pe_file_map
* fmap
= &ism
->fmap
->u
.pe
;
67 if (ism
->sidx
>= 0 && ism
->sidx
< fmap
->ntheader
.FileHeader
.NumberOfSections
&&
68 fmap
->sect
[ism
->sidx
].mapped
== IMAGE_NO_MAP
)
70 IMAGE_NT_HEADERS
* nth
;
72 if (fmap
->sect
[ism
->sidx
].shdr
.Misc
.VirtualSize
> fmap
->sect
[ism
->sidx
].shdr
.SizeOfRawData
)
74 FIXME("Section %ld: virtual (0x%x) > raw (0x%x) size - not supported\n",
75 ism
->sidx
, fmap
->sect
[ism
->sidx
].shdr
.Misc
.VirtualSize
,
76 fmap
->sect
[ism
->sidx
].shdr
.SizeOfRawData
);
79 /* FIXME: that's rather drastic, but that will do for now
80 * that's ok if the full file map exists, but we could be less aggressive otherwise and
81 * only map the relevant section
83 if ((mapping
= pe_map_full(ism
->fmap
, &nth
)))
85 fmap
->sect
[ism
->sidx
].mapped
= RtlImageRvaToVa(nth
, mapping
,
86 fmap
->sect
[ism
->sidx
].shdr
.VirtualAddress
,
88 return fmap
->sect
[ism
->sidx
].mapped
;
94 /******************************************************************
97 * Finds a section by name (and type) into memory from an PE file
98 * or its alternate if any
100 BOOL
pe_find_section(struct image_file_map
* fmap
, const char* name
,
101 struct image_section_map
* ism
)
103 const char* sectname
;
105 char tmp
[IMAGE_SIZEOF_SHORT_NAME
+ 1];
107 for (i
= 0; i
< fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSections
; i
++)
109 sectname
= (const char*)fmap
->u
.pe
.sect
[i
].shdr
.Name
;
110 /* long section names start with a '/' (at least on MinGW32) */
111 if (sectname
[0] == '/' && fmap
->u
.pe
.strtable
)
112 sectname
= fmap
->u
.pe
.strtable
+ atoi(sectname
+ 1);
115 /* the section name may not be null terminated */
116 sectname
= memcpy(tmp
, sectname
, IMAGE_SIZEOF_SHORT_NAME
);
117 tmp
[IMAGE_SIZEOF_SHORT_NAME
] = '\0';
119 if (!strcasecmp(sectname
, name
))
132 /******************************************************************
135 * Unmaps a single section from memory
137 void pe_unmap_section(struct image_section_map
* ism
)
139 if (ism
->sidx
>= 0 && ism
->sidx
< ism
->fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSections
&&
140 ism
->fmap
->u
.pe
.sect
[ism
->sidx
].mapped
!= IMAGE_NO_MAP
)
142 pe_unmap_full(ism
->fmap
);
143 ism
->fmap
->u
.pe
.sect
[ism
->sidx
].mapped
= IMAGE_NO_MAP
;
147 /******************************************************************
150 * Get the RVA of an PE section
152 DWORD_PTR
pe_get_map_rva(const struct image_section_map
* ism
)
154 if (ism
->sidx
< 0 || ism
->sidx
>= ism
->fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSections
)
156 return ism
->fmap
->u
.pe
.sect
[ism
->sidx
].shdr
.VirtualAddress
;
159 /******************************************************************
162 * Get the size of a PE section
164 unsigned pe_get_map_size(const struct image_section_map
* ism
)
166 if (ism
->sidx
< 0 || ism
->sidx
>= ism
->fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSections
)
168 return ism
->fmap
->u
.pe
.sect
[ism
->sidx
].shdr
.Misc
.VirtualSize
;
171 /******************************************************************
172 * pe_is_valid_pointer_table
174 * Checks whether the PointerToSymbolTable and NumberOfSymbols in file_header contain
177 static BOOL
pe_is_valid_pointer_table(const IMAGE_NT_HEADERS
* nthdr
, const void* mapping
, DWORD64 sz
)
181 /* is the iSym table inside file size ? (including first DWORD of string table, which is its size) */
182 offset
= (DWORD64
)nthdr
->FileHeader
.PointerToSymbolTable
;
183 offset
+= (DWORD64
)nthdr
->FileHeader
.NumberOfSymbols
* sizeof(IMAGE_SYMBOL
);
184 if (offset
+ sizeof(DWORD
) > sz
) return FALSE
;
185 /* is string table (following iSym table) inside file size ? */
186 offset
+= *(DWORD
*)((const char*)mapping
+ offset
);
190 /******************************************************************
193 * Maps an PE file into memory (and checks it's a real PE file)
195 static BOOL
pe_map_file(HANDLE file
, struct image_file_map
* fmap
, enum module_type mt
)
200 fmap
->u
.pe
.hMap
= CreateFileMappingW(file
, NULL
, PAGE_READONLY
, 0, 0, NULL
);
201 if (fmap
->u
.pe
.hMap
== 0) return FALSE
;
202 fmap
->u
.pe
.full_count
= 0;
203 fmap
->u
.pe
.full_map
= NULL
;
204 if (!(mapping
= pe_map_full(fmap
, NULL
))) goto error
;
210 IMAGE_NT_HEADERS
* nthdr
;
211 IMAGE_SECTION_HEADER
* section
;
214 if (!(nthdr
= RtlImageNtHeader(mapping
))) goto error
;
215 memcpy(&fmap
->u
.pe
.ntheader
, nthdr
, sizeof(fmap
->u
.pe
.ntheader
));
216 switch (nthdr
->OptionalHeader
.Magic
)
218 case 0x10b: fmap
->addr_size
= 32; break;
219 case 0x20b: fmap
->addr_size
= 64; break;
220 default: return FALSE
;
222 section
= (IMAGE_SECTION_HEADER
*)
223 ((char*)&nthdr
->OptionalHeader
+ nthdr
->FileHeader
.SizeOfOptionalHeader
);
224 fmap
->u
.pe
.sect
= HeapAlloc(GetProcessHeap(), 0,
225 nthdr
->FileHeader
.NumberOfSections
* sizeof(fmap
->u
.pe
.sect
[0]));
226 if (!fmap
->u
.pe
.sect
) goto error
;
227 for (i
= 0; i
< nthdr
->FileHeader
.NumberOfSections
; i
++)
229 memcpy(&fmap
->u
.pe
.sect
[i
].shdr
, section
+ i
, sizeof(IMAGE_SECTION_HEADER
));
230 fmap
->u
.pe
.sect
[i
].mapped
= IMAGE_NO_MAP
;
232 if (nthdr
->FileHeader
.PointerToSymbolTable
&& nthdr
->FileHeader
.NumberOfSymbols
)
236 if (GetFileSizeEx(file
, &li
) && pe_is_valid_pointer_table(nthdr
, mapping
, li
.QuadPart
))
238 /* FIXME ugly: should rather map the relevant content instead of copying it */
239 const char* src
= (const char*)mapping
+
240 nthdr
->FileHeader
.PointerToSymbolTable
+
241 nthdr
->FileHeader
.NumberOfSymbols
* sizeof(IMAGE_SYMBOL
);
243 DWORD sz
= *(DWORD
*)src
;
245 if ((dst
= HeapAlloc(GetProcessHeap(), 0, sz
)))
246 memcpy(dst
, src
, sz
);
247 fmap
->u
.pe
.strtable
= dst
;
251 WARN("Bad coff table... wipping out\n");
252 /* we have bad information here, wipe it out */
253 fmap
->u
.pe
.ntheader
.FileHeader
.PointerToSymbolTable
= 0;
254 fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSymbols
= 0;
255 fmap
->u
.pe
.strtable
= NULL
;
258 else fmap
->u
.pe
.strtable
= NULL
;
261 default: assert(0); goto error
;
268 CloseHandle(fmap
->u
.pe
.hMap
);
272 /******************************************************************
275 * Unmaps an PE file from memory (previously mapped with pe_map_file)
277 static void pe_unmap_file(struct image_file_map
* fmap
)
279 if (fmap
->u
.pe
.hMap
!= 0)
281 struct image_section_map ism
;
283 for (ism
.sidx
= 0; ism
.sidx
< fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSections
; ism
.sidx
++)
285 pe_unmap_section(&ism
);
287 while (fmap
->u
.pe
.full_count
) pe_unmap_full(fmap
);
288 HeapFree(GetProcessHeap(), 0, fmap
->u
.pe
.sect
);
289 HeapFree(GetProcessHeap(), 0, (void*)fmap
->u
.pe
.strtable
); /* FIXME ugly (see pe_map_file) */
290 CloseHandle(fmap
->u
.pe
.hMap
);
291 fmap
->u
.pe
.hMap
= NULL
;
295 /******************************************************************
298 * Maps a directory content out of a PE file
300 const char* pe_map_directory(struct module
* module
, int dirno
, DWORD
* size
)
302 IMAGE_NT_HEADERS
* nth
;
305 if (module
->type
!= DMT_PE
|| !module
->format_info
[DFI_PE
]) return NULL
;
306 if (dirno
>= IMAGE_NUMBEROF_DIRECTORY_ENTRIES
||
307 !(mapping
= pe_map_full(&module
->format_info
[DFI_PE
]->u
.pe_info
->fmap
, &nth
)))
309 if (size
) *size
= nth
->OptionalHeader
.DataDirectory
[dirno
].Size
;
310 return RtlImageRvaToVa(nth
, mapping
,
311 nth
->OptionalHeader
.DataDirectory
[dirno
].VirtualAddress
, NULL
);
314 /******************************************************************
317 * Unmaps a directory content
319 void pe_unmap_directory(struct image_file_map
* fmap
, int dirno
)
324 static void pe_module_remove(struct process
* pcs
, struct module_format
* modfmt
)
326 pe_unmap_file(&modfmt
->u
.pe_info
->fmap
);
327 HeapFree(GetProcessHeap(), 0, modfmt
);
330 /******************************************************************
331 * pe_locate_with_coff_symbol_table
333 * Use the COFF symbol table (if any) from the IMAGE_FILE_HEADER to set the absolute address
335 * Mingw32 requires this for stabs debug information as address for global variables isn't filled in
336 * (this is similar to what is done in elf_module.c when using the .symtab ELF section)
338 static BOOL
pe_locate_with_coff_symbol_table(struct module
* module
)
340 struct image_file_map
* fmap
= &module
->format_info
[DFI_PE
]->u
.pe_info
->fmap
;
341 const IMAGE_SYMBOL
* isym
;
345 struct hash_table_iter hti
;
347 struct symt_data
* sym
;
350 numsym
= fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSymbols
;
351 if (!fmap
->u
.pe
.ntheader
.FileHeader
.PointerToSymbolTable
|| !numsym
)
353 if (!(mapping
= pe_map_full(fmap
, NULL
))) return FALSE
;
354 isym
= (const IMAGE_SYMBOL
*)(mapping
+ fmap
->u
.pe
.ntheader
.FileHeader
.PointerToSymbolTable
);
356 for (i
= 0; i
< numsym
; i
+= naux
, isym
+= naux
)
358 if (isym
->StorageClass
== IMAGE_SYM_CLASS_EXTERNAL
&&
359 isym
->SectionNumber
> 0 && isym
->SectionNumber
<= fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSections
)
361 if (isym
->N
.Name
.Short
)
363 name
= memcpy(tmp
, isym
->N
.ShortName
, 8);
366 else name
= fmap
->u
.pe
.strtable
+ isym
->N
.Name
.Long
;
367 if (name
[0] == '_') name
++;
368 hash_table_iter_init(&module
->ht_symbols
, &hti
, name
);
369 while ((ptr
= hash_table_iter_up(&hti
)))
371 sym
= GET_ENTRY(ptr
, struct symt_data
, hash_elt
);
372 if (sym
->symt
.tag
== SymTagData
&&
373 (sym
->kind
== DataIsGlobal
|| sym
->kind
== DataIsFileStatic
) &&
374 sym
->u
.var
.kind
== loc_absolute
&&
375 !strcmp(sym
->hash_elt
.name
, name
))
377 TRACE("Changing absolute address for %d.%s: %lx -> %s\n",
378 isym
->SectionNumber
, name
, sym
->u
.var
.offset
,
379 wine_dbgstr_longlong(module
->module
.BaseOfImage
+
380 fmap
->u
.pe
.sect
[isym
->SectionNumber
- 1].shdr
.VirtualAddress
+
382 sym
->u
.var
.offset
= module
->module
.BaseOfImage
+
383 fmap
->u
.pe
.sect
[isym
->SectionNumber
- 1].shdr
.VirtualAddress
+ isym
->Value
;
388 naux
= isym
->NumberOfAuxSymbols
+ 1;
394 /******************************************************************
395 * pe_load_coff_symbol_table
397 * Load public symbols out of the COFF symbol table (if any).
399 static BOOL
pe_load_coff_symbol_table(struct module
* module
)
401 struct image_file_map
* fmap
= &module
->format_info
[DFI_PE
]->u
.pe_info
->fmap
;
402 const IMAGE_SYMBOL
* isym
;
404 const char* strtable
;
407 const char* lastfilename
= NULL
;
408 struct symt_compiland
* compiland
= NULL
;
409 const IMAGE_SECTION_HEADER
* sect
;
412 numsym
= fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSymbols
;
413 if (!fmap
->u
.pe
.ntheader
.FileHeader
.PointerToSymbolTable
|| !numsym
)
415 if (!(mapping
= pe_map_full(fmap
, NULL
))) return FALSE
;
416 isym
= (const IMAGE_SYMBOL
*)((const char*)mapping
+ fmap
->u
.pe
.ntheader
.FileHeader
.PointerToSymbolTable
);
417 /* FIXME: no way to get strtable size */
418 strtable
= (const char*)&isym
[numsym
];
419 sect
= IMAGE_FIRST_SECTION(RtlImageNtHeader((HMODULE
)mapping
));
421 for (i
= 0; i
< numsym
; i
+= naux
, isym
+= naux
)
423 if (isym
->StorageClass
== IMAGE_SYM_CLASS_FILE
)
425 lastfilename
= (const char*)(isym
+ 1);
428 if (isym
->StorageClass
== IMAGE_SYM_CLASS_EXTERNAL
&&
429 isym
->SectionNumber
> 0 && isym
->SectionNumber
<= fmap
->u
.pe
.ntheader
.FileHeader
.NumberOfSections
)
431 if (isym
->N
.Name
.Short
)
433 name
= memcpy(tmp
, isym
->N
.ShortName
, 8);
436 else name
= strtable
+ isym
->N
.Name
.Long
;
437 if (name
[0] == '_') name
++;
439 if (!compiland
&& lastfilename
)
440 compiland
= symt_new_compiland(module
, 0,
441 source_new(module
, NULL
, lastfilename
));
443 if (!(dbghelp_options
& SYMOPT_NO_PUBLICS
))
444 symt_new_public(module
, compiland
, name
,
445 module
->module
.BaseOfImage
+ sect
[isym
->SectionNumber
- 1].VirtualAddress
+
449 naux
= isym
->NumberOfAuxSymbols
+ 1;
451 module
->module
.SymType
= SymCoff
;
452 module
->module
.LineNumbers
= FALSE
;
453 module
->module
.GlobalSymbols
= FALSE
;
454 module
->module
.TypeInfo
= FALSE
;
455 module
->module
.SourceIndexed
= FALSE
;
456 module
->module
.Publics
= TRUE
;
462 /******************************************************************
465 * look for stabs information in PE header (it's how the mingw compiler provides
466 * its debugging information)
468 static BOOL
pe_load_stabs(const struct process
* pcs
, struct module
* module
)
470 struct image_file_map
* fmap
= &module
->format_info
[DFI_PE
]->u
.pe_info
->fmap
;
471 struct image_section_map sect_stabs
, sect_stabstr
;
474 if (pe_find_section(fmap
, ".stab", §_stabs
) && pe_find_section(fmap
, ".stabstr", §_stabstr
))
479 stab
= image_map_section(§_stabs
);
480 stabstr
= image_map_section(§_stabstr
);
481 if (stab
!= IMAGE_NO_MAP
&& stabstr
!= IMAGE_NO_MAP
)
483 ret
= stabs_parse(module
,
484 module
->module
.BaseOfImage
- fmap
->u
.pe
.ntheader
.OptionalHeader
.ImageBase
,
485 stab
, image_get_map_size(§_stabs
),
486 stabstr
, image_get_map_size(§_stabstr
),
489 image_unmap_section(§_stabs
);
490 image_unmap_section(§_stabstr
);
491 if (ret
) pe_locate_with_coff_symbol_table(module
);
493 TRACE("%s the STABS debug info\n", ret
? "successfully loaded" : "failed to load");
498 /******************************************************************
501 * look for dwarf information in PE header (it's also a way for the mingw compiler
502 * to provide its debugging information)
504 static BOOL
pe_load_dwarf(struct module
* module
)
506 struct image_file_map
* fmap
= &module
->format_info
[DFI_PE
]->u
.pe_info
->fmap
;
509 ret
= dwarf2_parse(module
,
510 module
->module
.BaseOfImage
- fmap
->u
.pe
.ntheader
.OptionalHeader
.ImageBase
,
511 NULL
, /* FIXME: some thunks to deal with ? */
513 TRACE("%s the DWARF debug info\n", ret
? "successfully loaded" : "failed to load");
518 #ifndef DBGHELP_STATIC_LIB
519 /******************************************************************
524 static BOOL
pe_load_dbg_file(const struct process
* pcs
, struct module
* module
,
525 const char* dbg_name
, DWORD timestamp
)
528 HANDLE hFile
= INVALID_HANDLE_VALUE
, hMap
= 0;
529 const BYTE
* dbg_mapping
= NULL
;
532 TRACE("Processing DBG file %s\n", debugstr_a(dbg_name
));
534 if (path_find_symbol_file(pcs
, dbg_name
, NULL
, timestamp
, 0, tmp
, &module
->module
.DbgUnmatched
) &&
535 (hFile
= CreateFileA(tmp
, GENERIC_READ
, FILE_SHARE_READ
, NULL
,
536 OPEN_EXISTING
, FILE_ATTRIBUTE_NORMAL
, NULL
)) != INVALID_HANDLE_VALUE
&&
537 ((hMap
= CreateFileMappingW(hFile
, NULL
, PAGE_READONLY
, 0, 0, NULL
)) != 0) &&
538 ((dbg_mapping
= MapViewOfFile(hMap
, FILE_MAP_READ
, 0, 0, 0)) != NULL
))
540 const IMAGE_SEPARATE_DEBUG_HEADER
* hdr
;
541 const IMAGE_SECTION_HEADER
* sectp
;
542 const IMAGE_DEBUG_DIRECTORY
* dbg
;
544 hdr
= (const IMAGE_SEPARATE_DEBUG_HEADER
*)dbg_mapping
;
545 /* section headers come immediately after debug header */
546 sectp
= (const IMAGE_SECTION_HEADER
*)(hdr
+ 1);
547 /* and after that and the exported names comes the debug directory */
548 dbg
= (const IMAGE_DEBUG_DIRECTORY
*)
549 (dbg_mapping
+ sizeof(*hdr
) +
550 hdr
->NumberOfSections
* sizeof(IMAGE_SECTION_HEADER
) +
551 hdr
->ExportedNamesSize
);
553 ret
= pe_load_debug_directory(pcs
, module
, dbg_mapping
, sectp
,
554 hdr
->NumberOfSections
, dbg
,
555 hdr
->DebugDirectorySize
/ sizeof(*dbg
));
558 ERR("Couldn't find .DBG file %s (%s)\n", debugstr_a(dbg_name
), debugstr_a(tmp
));
560 if (dbg_mapping
) UnmapViewOfFile(dbg_mapping
);
561 if (hMap
) CloseHandle(hMap
);
562 if (hFile
!= INVALID_HANDLE_VALUE
) CloseHandle(hFile
);
566 /******************************************************************
567 * pe_load_msc_debug_info
569 * Process MSC debug information in PE file.
571 static BOOL
pe_load_msc_debug_info(const struct process
* pcs
, struct module
* module
)
573 struct image_file_map
* fmap
= &module
->format_info
[DFI_PE
]->u
.pe_info
->fmap
;
575 const IMAGE_DATA_DIRECTORY
* dir
;
576 const IMAGE_DEBUG_DIRECTORY
*dbg
= NULL
;
579 IMAGE_NT_HEADERS
* nth
;
581 if (!(mapping
= pe_map_full(fmap
, &nth
))) return FALSE
;
582 /* Read in debug directory */
583 dir
= nth
->OptionalHeader
.DataDirectory
+ IMAGE_DIRECTORY_ENTRY_DEBUG
;
584 nDbg
= dir
->Size
/ sizeof(IMAGE_DEBUG_DIRECTORY
);
585 if (!nDbg
) goto done
;
587 dbg
= RtlImageRvaToVa(nth
, mapping
, dir
->VirtualAddress
, NULL
);
589 /* Parse debug directory */
590 if (nth
->FileHeader
.Characteristics
& IMAGE_FILE_DEBUG_STRIPPED
)
592 /* Debug info is stripped to .DBG file */
593 const IMAGE_DEBUG_MISC
* misc
= (const IMAGE_DEBUG_MISC
*)
594 ((const char*)mapping
+ dbg
->PointerToRawData
);
596 if (nDbg
!= 1 || dbg
->Type
!= IMAGE_DEBUG_TYPE_MISC
||
597 misc
->DataType
!= IMAGE_DEBUG_MISC_EXENAME
)
599 ERR("-Debug info stripped, but no .DBG file in module %s\n",
600 debugstr_w(module
->module
.ModuleName
));
604 ret
= pe_load_dbg_file(pcs
, module
, (const char*)misc
->Data
, nth
->FileHeader
.TimeDateStamp
);
609 const IMAGE_SECTION_HEADER
*sectp
= (const IMAGE_SECTION_HEADER
*)((const char*)&nth
->OptionalHeader
+ nth
->FileHeader
.SizeOfOptionalHeader
);
610 /* Debug info is embedded into PE module */
611 ret
= pe_load_debug_directory(pcs
, module
, mapping
, sectp
,
612 nth
->FileHeader
.NumberOfSections
, dbg
, nDbg
);
618 #endif /* DBGHELP_STATIC_LIB */
620 /***********************************************************************
621 * pe_load_export_debug_info
623 static BOOL
pe_load_export_debug_info(const struct process
* pcs
, struct module
* module
)
625 struct image_file_map
* fmap
= &module
->format_info
[DFI_PE
]->u
.pe_info
->fmap
;
627 const IMAGE_EXPORT_DIRECTORY
* exports
;
628 DWORD base
= module
->module
.BaseOfImage
;
630 IMAGE_NT_HEADERS
* nth
;
633 if (dbghelp_options
& SYMOPT_NO_PUBLICS
) return TRUE
;
635 if (!(mapping
= pe_map_full(fmap
, &nth
))) return FALSE
;
637 /* Add start of DLL (better use the (yet unimplemented) Exe SymTag for this) */
638 /* FIXME: module.ModuleName isn't correctly set yet if it's passed in SymLoadModule */
639 symt_new_public(module
, NULL
, module
->module
.ModuleName
, base
, 1);
642 /* Add entry point */
643 symt_new_public(module
, NULL
, "EntryPoint",
644 base
+ nth
->OptionalHeader
.AddressOfEntryPoint
, 1);
646 /* FIXME: we'd better store addresses linked to sections rather than
648 IMAGE_SECTION_HEADER
* section
;
649 /* Add start of sections */
650 section
= (IMAGE_SECTION_HEADER
*)
651 ((char*)&nth
->OptionalHeader
+ nth
->FileHeader
.SizeOfOptionalHeader
);
652 for (i
= 0; i
< nth
->FileHeader
.NumberOfSections
; i
++, section
++)
654 symt_new_public(module
, NULL
, section
->Name
,
655 RtlImageRvaToVa(nth
, mapping
, section
->VirtualAddress
, NULL
), 1);
659 /* Add exported functions */
660 if ((exports
= RtlImageDirectoryEntryToData(mapping
, FALSE
,
661 IMAGE_DIRECTORY_ENTRY_EXPORT
, &size
)))
663 const WORD
* ordinals
= NULL
;
664 const DWORD_PTR
* functions
= NULL
;
665 const DWORD
* names
= NULL
;
669 functions
= RtlImageRvaToVa(nth
, mapping
, exports
->AddressOfFunctions
, NULL
);
670 ordinals
= RtlImageRvaToVa(nth
, mapping
, exports
->AddressOfNameOrdinals
, NULL
);
671 names
= RtlImageRvaToVa(nth
, mapping
, exports
->AddressOfNames
, NULL
);
673 if (functions
&& ordinals
&& names
)
675 for (i
= 0; i
< exports
->NumberOfNames
; i
++)
677 if (!names
[i
]) continue;
678 symt_new_public(module
, NULL
,
679 RtlImageRvaToVa(nth
, mapping
, names
[i
], NULL
),
680 base
+ functions
[ordinals
[i
]], 1);
683 for (i
= 0; i
< exports
->NumberOfFunctions
; i
++)
685 if (!functions
[i
]) continue;
686 /* Check if we already added it with a name */
687 for (j
= 0; j
< exports
->NumberOfNames
; j
++)
688 if ((ordinals
[j
] == i
) && names
[j
]) break;
689 if (j
< exports
->NumberOfNames
) continue;
690 snprintf(buffer
, sizeof(buffer
), "%d", i
+ exports
->Base
);
691 symt_new_public(module
, NULL
, buffer
, base
+ (DWORD
)functions
[i
], 1);
695 /* no real debug info, only entry points */
696 if (module
->module
.SymType
== SymDeferred
)
697 module
->module
.SymType
= SymExport
;
703 /******************************************************************
707 BOOL
pe_load_debug_info(const struct process
* pcs
, struct module
* module
)
711 if (!(dbghelp_options
& SYMOPT_PUBLICS_ONLY
))
713 ret
= pe_load_stabs(pcs
, module
);
714 ret
= pe_load_dwarf(module
) || ret
;
715 #ifndef DBGHELP_STATIC_LIB
716 ret
= pe_load_msc_debug_info(pcs
, module
) || ret
;
718 ret
= ret
|| pe_load_coff_symbol_table(module
); /* FIXME */
719 /* if we still have no debug info (we could only get SymExport at this
720 * point), then do the SymExport except if we have an ELF container,
721 * in which case we'll rely on the export's on the ELF side
724 /* FIXME shouldn't we check that? if (!module_get_debug(pcs, module)) */
725 if (pe_load_export_debug_info(pcs
, module
) && !ret
)
731 /******************************************************************
732 * pe_load_native_module
735 struct module
* pe_load_native_module(struct process
* pcs
, const WCHAR
* name
,
736 HANDLE hFile
, DWORD64 base
, DWORD size
)
738 struct module
* module
= NULL
;
740 struct module_format
* modfmt
;
741 WCHAR loaded_name
[MAX_PATH
];
743 loaded_name
[0] = '\0';
748 if ((hFile
= FindExecutableImageExW(name
, pcs
->search_path
, loaded_name
, NULL
, NULL
)) == NULL
)
752 else if (name
) strcpyW(loaded_name
, name
);
753 else if (dbghelp_options
& SYMOPT_DEFERRED_LOADS
)
754 FIXME("Trouble ahead (no module name passed in deferred mode)\n");
755 if (!(modfmt
= HeapAlloc(GetProcessHeap(), 0, sizeof(struct module_format
) + sizeof(struct pe_module_info
))))
757 modfmt
->u
.pe_info
= (struct pe_module_info
*)(modfmt
+ 1);
758 if (pe_map_file(hFile
, &modfmt
->u
.pe_info
->fmap
, DMT_PE
))
760 if (!base
) base
= modfmt
->u
.pe_info
->fmap
.u
.pe
.ntheader
.OptionalHeader
.ImageBase
;
761 if (!size
) size
= modfmt
->u
.pe_info
->fmap
.u
.pe
.ntheader
.OptionalHeader
.SizeOfImage
;
763 module
= module_new(pcs
, loaded_name
, DMT_PE
, FALSE
, base
, size
,
764 modfmt
->u
.pe_info
->fmap
.u
.pe
.ntheader
.FileHeader
.TimeDateStamp
,
765 modfmt
->u
.pe_info
->fmap
.u
.pe
.ntheader
.OptionalHeader
.CheckSum
);
768 modfmt
->module
= module
;
769 modfmt
->remove
= pe_module_remove
;
770 modfmt
->loc_compute
= NULL
;
772 module
->format_info
[DFI_PE
] = modfmt
;
773 if (dbghelp_options
& SYMOPT_DEFERRED_LOADS
)
774 module
->module
.SymType
= SymDeferred
;
776 pe_load_debug_info(pcs
, module
);
777 module
->reloc_delta
= base
- modfmt
->u
.pe_info
->fmap
.u
.pe
.ntheader
.OptionalHeader
.ImageBase
;
781 ERR("could not load the module '%s'\n", debugstr_w(loaded_name
));
782 pe_unmap_file(&modfmt
->u
.pe_info
->fmap
);
785 if (!module
) HeapFree(GetProcessHeap(), 0, modfmt
);
787 if (opened
) CloseHandle(hFile
);
792 /******************************************************************
796 BOOL
pe_load_nt_header(HANDLE hProc
, DWORD64 base
, IMAGE_NT_HEADERS
* nth
)
798 IMAGE_DOS_HEADER dos
;
800 return ReadProcessMemory(hProc
, (char*)(DWORD_PTR
)base
, &dos
, sizeof(dos
), NULL
) &&
801 dos
.e_magic
== IMAGE_DOS_SIGNATURE
&&
802 ReadProcessMemory(hProc
, (char*)(DWORD_PTR
)(base
+ dos
.e_lfanew
),
803 nth
, sizeof(*nth
), NULL
) &&
804 nth
->Signature
== IMAGE_NT_SIGNATURE
;
807 /******************************************************************
808 * pe_load_builtin_module
811 struct module
* pe_load_builtin_module(struct process
* pcs
, const WCHAR
* name
,
812 DWORD64 base
, DWORD64 size
)
814 struct module
* module
= NULL
;
816 if (base
&& pcs
->dbg_hdr_addr
)
818 IMAGE_NT_HEADERS nth
;
820 if (pe_load_nt_header(pcs
->handle
, base
, &nth
))
822 if (!size
) size
= nth
.OptionalHeader
.SizeOfImage
;
823 module
= module_new(pcs
, name
, DMT_PE
, FALSE
, base
, size
,
824 nth
.FileHeader
.TimeDateStamp
,
825 nth
.OptionalHeader
.CheckSum
);
831 /***********************************************************************
832 * ImageDirectoryEntryToDataEx (DBGHELP.@)
834 * Search for specified directory in PE image
838 * base [in] Image base address
839 * image [in] TRUE - image has been loaded by loader, FALSE - raw file image
840 * dir [in] Target directory index
841 * size [out] Receives directory size
842 * section [out] Receives pointer to section header of section containing directory data
845 * Success: pointer to directory data
849 PVOID WINAPI
ImageDirectoryEntryToDataEx( PVOID base
, BOOLEAN image
, USHORT dir
, PULONG size
, PIMAGE_SECTION_HEADER
*section
)
851 const IMAGE_NT_HEADERS
*nt
;
855 if (section
) *section
= NULL
;
857 if (!(nt
= RtlImageNtHeader( base
))) return NULL
;
858 if (dir
>= nt
->OptionalHeader
.NumberOfRvaAndSizes
) return NULL
;
859 if (!(addr
= nt
->OptionalHeader
.DataDirectory
[dir
].VirtualAddress
)) return NULL
;
861 *size
= nt
->OptionalHeader
.DataDirectory
[dir
].Size
;
862 if (image
|| addr
< nt
->OptionalHeader
.SizeOfHeaders
) return (char *)base
+ addr
;
864 return RtlImageRvaToVa( nt
, base
, addr
, section
);
867 /***********************************************************************
868 * ImageDirectoryEntryToData (DBGHELP.@)
871 * See ImageDirectoryEntryToDataEx
873 PVOID WINAPI
ImageDirectoryEntryToData( PVOID base
, BOOLEAN image
, USHORT dir
, PULONG size
)
875 return ImageDirectoryEntryToDataEx( base
, image
, dir
, size
, NULL
);