shlwapi: Beginning implementation of IUnknown_QueryServiceForWebBrowserApp.
[wine] / dlls / dbghelp / module.c
1 /*
2  * File module.c - module handling for the wine debugger
3  *
4  * Copyright (C) 1993,      Eric Youngdale.
5  *               2000-2007, Eric Pouech
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library 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 GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22 #include "config.h"
23 #include <stdlib.h>
24 #include <stdio.h>
25 #include <string.h>
26 #include <assert.h>
27
28 #include "dbghelp_private.h"
29 #include "psapi.h"
30 #include "winternl.h"
31 #include "wine/debug.h"
32
33 WINE_DEFAULT_DEBUG_CHANNEL(dbghelp);
34
35 const WCHAR        S_ElfW[]         = {'<','e','l','f','>','\0'};
36 const WCHAR        S_WineLoaderW[]  = {'<','w','i','n','e','-','l','o','a','d','e','r','>','\0'};
37 static const WCHAR S_DotSoW[]       = {'.','s','o','\0'};
38 static const WCHAR S_DotDylibW[]    = {'.','d','y','l','i','b','\0'};
39 static const WCHAR S_DotPdbW[]      = {'.','p','d','b','\0'};
40 static const WCHAR S_DotDbgW[]      = {'.','d','b','g','\0'};
41 const WCHAR        S_WineW[]        = {'w','i','n','e',0};
42 const WCHAR        S_SlashW[]       = {'/','\0'};
43
44 static const WCHAR S_AcmW[] = {'.','a','c','m','\0'};
45 static const WCHAR S_DllW[] = {'.','d','l','l','\0'};
46 static const WCHAR S_DrvW[] = {'.','d','r','v','\0'};
47 static const WCHAR S_ExeW[] = {'.','e','x','e','\0'};
48 static const WCHAR S_OcxW[] = {'.','o','c','x','\0'};
49 static const WCHAR S_VxdW[] = {'.','v','x','d','\0'};
50 static const WCHAR * const ext[] = {S_AcmW, S_DllW, S_DrvW, S_ExeW, S_OcxW, S_VxdW, NULL};
51
52 static int match_ext(const WCHAR* ptr, size_t len)
53 {
54     const WCHAR* const *e;
55     size_t      l;
56
57     for (e = ext; *e; e++)
58     {
59         l = strlenW(*e);
60         if (l >= len) return FALSE;
61         if (strncmpiW(&ptr[len - l], *e, l)) continue;
62         return l;
63     }
64     return 0;
65 }
66
67 static const WCHAR* get_filename(const WCHAR* name, const WCHAR* endptr)
68 {
69     const WCHAR*        ptr;
70
71     if (!endptr) endptr = name + strlenW(name);
72     for (ptr = endptr - 1; ptr >= name; ptr--)
73     {
74         if (*ptr == '/' || *ptr == '\\') break;
75     }
76     return ++ptr;
77 }
78
79 static void module_fill_module(const WCHAR* in, WCHAR* out, size_t size)
80 {
81     const WCHAR *ptr, *endptr;
82     size_t      len, l;
83
84     ptr = get_filename(in, endptr = in + strlenW(in));
85     len = min(endptr - ptr, size - 1);
86     memcpy(out, ptr, len * sizeof(WCHAR));
87     out[len] = '\0';
88     if (len > 4 && (l = match_ext(out, len)))
89         out[len - l] = '\0';
90     else if (len > 4 && !strcmpiW(out + len - 4, S_WineW))
91         lstrcpynW(out, S_WineLoaderW, size);
92     else
93     {
94         if (len > 3 && !strcmpiW(&out[len - 3], S_DotSoW) &&
95             (l = match_ext(out, len - 3)))
96             strcpyW(&out[len - l - 3], S_ElfW);
97     }
98     while ((*out = tolowerW(*out))) out++;
99 }
100
101 void module_set_module(struct module* module, const WCHAR* name)
102 {
103     module_fill_module(name, module->module.ModuleName, sizeof(module->module.ModuleName));
104     WideCharToMultiByte(CP_ACP, 0, module->module.ModuleName, -1,
105                         module->module_name, sizeof(module->module_name),
106                         NULL, NULL);
107 }
108
109 static const char*      get_module_type(enum module_type type, BOOL virtual)
110 {
111     switch (type)
112     {
113     case DMT_ELF: return virtual ? "Virtual ELF" : "ELF";
114     case DMT_PE: return virtual ? "Virtual PE" : "PE";
115     case DMT_MACHO: return virtual ? "Virtual Mach-O" : "Mach-O";
116     default: return "---";
117     }
118 }
119
120 /***********************************************************************
121  * Creates and links a new module to a process
122  */
123 struct module* module_new(struct process* pcs, const WCHAR* name,
124                           enum module_type type, BOOL virtual,
125                           unsigned long mod_addr, unsigned long size,
126                           unsigned long stamp, unsigned long checksum)
127 {
128     struct module*      module;
129
130     assert(type == DMT_ELF || type == DMT_PE || type == DMT_MACHO);
131     if (!(module = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*module))))
132         return NULL;
133
134     module->next = pcs->lmodules;
135     pcs->lmodules = module;
136
137     TRACE("=> %s %08lx-%08lx %s\n",
138           get_module_type(type, virtual), mod_addr, mod_addr + size,
139           debugstr_w(name));
140
141     pool_init(&module->pool, 65536);
142
143     module->module.SizeOfStruct = sizeof(module->module);
144     module->module.BaseOfImage = mod_addr;
145     module->module.ImageSize = size;
146     module_set_module(module, name);
147     module->module.ImageName[0] = '\0';
148     lstrcpynW(module->module.LoadedImageName, name, sizeof(module->module.LoadedImageName) / sizeof(WCHAR));
149     module->module.SymType = SymNone;
150     module->module.NumSyms = 0;
151     module->module.TimeDateStamp = stamp;
152     module->module.CheckSum = checksum;
153
154     memset(module->module.LoadedPdbName, 0, sizeof(module->module.CVData));
155     module->module.CVSig = 0;
156     memset(module->module.CVData, 0, sizeof(module->module.CVData));
157     module->module.PdbSig = 0;
158     memset(&module->module.PdbSig70, 0, sizeof(module->module.PdbSig70));
159     module->module.PdbAge = 0;
160     module->module.PdbUnmatched = FALSE;
161     module->module.DbgUnmatched = FALSE;
162     module->module.LineNumbers = FALSE;
163     module->module.GlobalSymbols = FALSE;
164     module->module.TypeInfo = FALSE;
165     module->module.SourceIndexed = FALSE;
166     module->module.Publics = FALSE;
167
168     module->type              = type;
169     module->is_virtual        = virtual ? TRUE : FALSE;
170     module->sortlist_valid    = FALSE;
171     module->sorttab_size      = 0;
172     module->addr_sorttab      = NULL;
173     module->num_sorttab       = 0;
174     module->num_symbols       = 0;
175     /* FIXME: this seems a bit too high (on a per module basis)
176      * need some statistics about this
177      */
178     hash_table_init(&module->pool, &module->ht_symbols, 4096);
179     hash_table_init(&module->pool, &module->ht_types,   4096);
180     vector_init(&module->vtypes, sizeof(struct symt*),  32);
181
182     module->sources_used      = 0;
183     module->sources_alloc     = 0;
184     module->sources           = 0;
185
186     return module;
187 }
188
189 /***********************************************************************
190  *      module_find_by_name
191  *
192  */
193 static struct module* module_find_by_name(const struct process* pcs, const WCHAR* name)
194 {
195     struct module*      module;
196
197     for (module = pcs->lmodules; module; module = module->next)
198     {
199         if (!strcmpiW(name, module->module.ModuleName)) return module;
200     }
201     SetLastError(ERROR_INVALID_NAME);
202     return NULL;
203 }
204
205 struct module* module_find_by_nameA(const struct process* pcs, const char* name)
206 {
207     WCHAR wname[MAX_PATH];
208
209     MultiByteToWideChar(CP_ACP, 0, name, -1, wname, sizeof(wname) / sizeof(WCHAR));
210     return module_find_by_name(pcs, wname);
211 }
212
213 /***********************************************************************
214  *      module_is_already_loaded
215  *
216  */
217 struct module* module_is_already_loaded(const struct process* pcs, const WCHAR* name)
218 {
219     struct module*      module;
220     const WCHAR*        filename;
221
222     /* first compare the loaded image name... */
223     for (module = pcs->lmodules; module; module = module->next)
224     {
225         if (!strcmpiW(name, module->module.LoadedImageName))
226             return module;
227     }
228     /* then compare the standard filenames (without the path) ... */
229     filename = get_filename(name, NULL);
230     for (module = pcs->lmodules; module; module = module->next)
231     {
232         if (!strcmpiW(filename, get_filename(module->module.LoadedImageName, NULL)))
233             return module;
234     }
235     SetLastError(ERROR_INVALID_NAME);
236     return NULL;
237 }
238
239 /***********************************************************************
240  *           module_get_container
241  *
242  */
243 static struct module* module_get_container(const struct process* pcs,
244                                     const struct module* inner)
245 {
246     struct module*      module;
247      
248     for (module = pcs->lmodules; module; module = module->next)
249     {
250         if (module != inner &&
251             module->module.BaseOfImage <= inner->module.BaseOfImage &&
252             module->module.BaseOfImage + module->module.ImageSize >=
253             inner->module.BaseOfImage + inner->module.ImageSize)
254             return module;
255     }
256     return NULL;
257 }
258
259 /***********************************************************************
260  *           module_get_containee
261  *
262  */
263 struct module* module_get_containee(const struct process* pcs, 
264                                     const struct module* outter)
265 {
266     struct module*      module;
267      
268     for (module = pcs->lmodules; module; module = module->next)
269     {
270         if (module != outter &&
271             outter->module.BaseOfImage <= module->module.BaseOfImage &&
272             outter->module.BaseOfImage + outter->module.ImageSize >=
273             module->module.BaseOfImage + module->module.ImageSize)
274             return module;
275     }
276     return NULL;
277 }
278
279 /******************************************************************
280  *              module_get_debug
281  *
282  * get the debug information from a module:
283  * - if the module's type is deferred, then force loading of debug info (and return
284  *   the module itself)
285  * - if the module has no debug info and has an ELF container, then return the ELF
286  *   container (and also force the ELF container's debug info loading if deferred)
287  * - otherwise return the module itself if it has some debug info
288  */
289 BOOL module_get_debug(struct module_pair* pair)
290 {
291     IMAGEHLP_DEFERRED_SYMBOL_LOADW64    idslW64;
292
293     if (!pair->requested) return FALSE;
294     /* for a PE builtin, always get info from container */
295     if (!(pair->effective = module_get_container(pair->pcs, pair->requested)))
296         pair->effective = pair->requested;
297     /* if deferred, force loading */
298     if (pair->effective->module.SymType == SymDeferred)
299     {
300         BOOL ret;
301         
302         if (pair->effective->is_virtual) ret = FALSE;
303         else switch (pair->effective->type)
304         {
305         case DMT_ELF:
306             ret = elf_load_debug_info(pair->effective, NULL);
307             break;
308         case DMT_PE:
309             idslW64.SizeOfStruct = sizeof(idslW64);
310             idslW64.BaseOfImage = pair->effective->module.BaseOfImage;
311             idslW64.CheckSum = pair->effective->module.CheckSum;
312             idslW64.TimeDateStamp = pair->effective->module.TimeDateStamp;
313             memcpy(idslW64.FileName, pair->effective->module.ImageName,
314                    sizeof(pair->effective->module.ImageName));
315             idslW64.Reparse = FALSE;
316             idslW64.hFile = INVALID_HANDLE_VALUE;
317
318             pcs_callback(pair->pcs, CBA_DEFERRED_SYMBOL_LOAD_START, &idslW64);
319             ret = pe_load_debug_info(pair->pcs, pair->effective);
320             pcs_callback(pair->pcs,
321                          ret ? CBA_DEFERRED_SYMBOL_LOAD_COMPLETE : CBA_DEFERRED_SYMBOL_LOAD_FAILURE,
322                          &idslW64);
323             break;
324         case DMT_MACHO:
325             ret = macho_load_debug_info(pair->effective, NULL);
326             break;
327         default:
328             ret = FALSE;
329             break;
330         }
331         if (!ret) pair->effective->module.SymType = SymNone;
332         assert(pair->effective->module.SymType != SymDeferred);
333         pair->effective->module.NumSyms = pair->effective->ht_symbols.num_elts;
334     }
335     return pair->effective->module.SymType != SymNone;
336 }
337
338 /***********************************************************************
339  *      module_find_by_addr
340  *
341  * either the addr where module is loaded, or any address inside the 
342  * module
343  */
344 struct module* module_find_by_addr(const struct process* pcs, unsigned long addr, 
345                                    enum module_type type)
346 {
347     struct module*      module;
348     
349     if (type == DMT_UNKNOWN)
350     {
351         if ((module = module_find_by_addr(pcs, addr, DMT_PE)) ||
352             (module = module_find_by_addr(pcs, addr, DMT_ELF)) ||
353             (module = module_find_by_addr(pcs, addr, DMT_MACHO)))
354             return module;
355     }
356     else
357     {
358         for (module = pcs->lmodules; module; module = module->next)
359         {
360             if (type == module->type && addr >= module->module.BaseOfImage &&
361                 addr < module->module.BaseOfImage + module->module.ImageSize) 
362                 return module;
363         }
364     }
365     SetLastError(ERROR_INVALID_ADDRESS);
366     return module;
367 }
368
369 /******************************************************************
370  *              module_is_container_loaded
371  *
372  * checks whether the native container, for a (supposed) PE builtin is
373  * already loaded
374  */
375 static BOOL module_is_container_loaded(const struct process* pcs,
376                                        const WCHAR* ImageName, DWORD base)
377 {
378     size_t              len;
379     struct module*      module;
380     PCWSTR              filename, modname;
381
382     if (!base) return FALSE;
383     filename = get_filename(ImageName, NULL);
384     len = strlenW(filename);
385
386     for (module = pcs->lmodules; module; module = module->next)
387     {
388         if ((module->type == DMT_ELF || module->type == DMT_MACHO) &&
389             base >= module->module.BaseOfImage &&
390             base < module->module.BaseOfImage + module->module.ImageSize)
391         {
392             modname = get_filename(module->module.LoadedImageName, NULL);
393             if (!strncmpiW(modname, filename, len) &&
394                 !memcmp(modname + len, S_DotSoW, 3 * sizeof(WCHAR)))
395             {
396                 return TRUE;
397             }
398         }
399     }
400     /* likely a native PE module */
401     WARN("Couldn't find container for %s\n", debugstr_w(ImageName));
402     return FALSE;
403 }
404
405 /******************************************************************
406  *              module_get_type_by_name
407  *
408  * Guesses a filename type from its extension
409  */
410 enum module_type module_get_type_by_name(const WCHAR* name)
411 {
412     int len = strlenW(name);
413
414     /* Skip all version extensions (.[digits]) regex: "(\.\d+)*$" */
415     do
416     {
417         int i = len;
418
419         while (i && isdigit(name[i - 1])) i--;
420
421         if (i && name[i - 1] == '.')
422             len = i - 1;
423         else
424             break;
425     } while (len);
426
427     /* check for terminating .so or .so.[digit] */
428     /* FIXME: Can't rely solely on extension; have to check magic or
429      *        stop using .so on Mac OS X.  For now, base on platform. */
430     if (len > 3 && !memcmp(name + len - 3, S_DotSoW, 3))
431 #ifdef __APPLE__
432         return DMT_MACHO;
433 #else
434         return DMT_ELF;
435 #endif
436
437     if (len > 6 && !strncmpiW(name + len - 6, S_DotDylibW, 6))
438         return DMT_MACHO;
439
440     if (len > 4 && !strncmpiW(name + len - 4, S_DotPdbW, 4))
441         return DMT_PDB;
442
443     if (len > 4 && !strncmpiW(name + len - 4, S_DotDbgW, 4))
444         return DMT_DBG;
445
446     /* wine is also a native module (Mach-O on Mac OS X, ELF elsewhere) */
447     if (((len > 4 && name[len - 5] == '/') || len == 4) && !strcmpiW(name + len - 4, S_WineW))
448     {
449 #ifdef __APPLE__
450         return DMT_MACHO;
451 #else
452         return DMT_ELF;
453 #endif
454     }
455     return DMT_PE;
456 }
457
458 /******************************************************************
459  *                              refresh_module_list
460  */
461 static BOOL refresh_module_list(struct process* pcs)
462 {
463     /* force transparent ELF and Mach-O loading / unloading */
464     return elf_synchronize_module_list(pcs) || macho_synchronize_module_list(pcs);
465 }
466
467 /***********************************************************************
468  *                      SymLoadModule (DBGHELP.@)
469  */
470 DWORD WINAPI SymLoadModule(HANDLE hProcess, HANDLE hFile, PCSTR ImageName,
471                            PCSTR ModuleName, DWORD BaseOfDll, DWORD SizeOfDll)
472 {
473     return SymLoadModuleEx(hProcess, hFile, ImageName, ModuleName, BaseOfDll,
474                            SizeOfDll, NULL, 0);
475 }
476
477 /***********************************************************************
478  *                      SymLoadModuleEx (DBGHELP.@)
479  */
480 DWORD64 WINAPI  SymLoadModuleEx(HANDLE hProcess, HANDLE hFile, PCSTR ImageName,
481                                 PCSTR ModuleName, DWORD64 BaseOfDll, DWORD DllSize,
482                                 PMODLOAD_DATA Data, DWORD Flags)
483 {
484     PWSTR       wImageName, wModuleName;
485     unsigned    len;
486     DWORD64     ret;
487
488     TRACE("(%p %p %s %s %s %08x %p %08x)\n",
489           hProcess, hFile, debugstr_a(ImageName), debugstr_a(ModuleName),
490           wine_dbgstr_longlong(BaseOfDll), DllSize, Data, Flags);
491
492     if (ImageName)
493     {
494         len = MultiByteToWideChar(CP_ACP, 0, ImageName, -1, NULL, 0);
495         wImageName = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
496         MultiByteToWideChar(CP_ACP, 0, ImageName, -1, wImageName, len);
497     }
498     else wImageName = NULL;
499     if (ModuleName)
500     {
501         len = MultiByteToWideChar(CP_ACP, 0, ModuleName, -1, NULL, 0);
502         wModuleName = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
503         MultiByteToWideChar(CP_ACP, 0, ModuleName, -1, wModuleName, len);
504     }
505     else wModuleName = NULL;
506
507     ret = SymLoadModuleExW(hProcess, hFile, wImageName, wModuleName,
508                           BaseOfDll, DllSize, Data, Flags);
509     HeapFree(GetProcessHeap(), 0, wImageName);
510     HeapFree(GetProcessHeap(), 0, wModuleName);
511     return ret;
512 }
513
514 /***********************************************************************
515  *                      SymLoadModuleExW (DBGHELP.@)
516  */
517 DWORD64 WINAPI  SymLoadModuleExW(HANDLE hProcess, HANDLE hFile, PCWSTR wImageName,
518                                  PCWSTR wModuleName, DWORD64 BaseOfDll, DWORD SizeOfDll,
519                                  PMODLOAD_DATA Data, DWORD Flags)
520 {
521     struct process*     pcs;
522     struct module*      module = NULL;
523
524     TRACE("(%p %p %s %s %s %08x %p %08x)\n",
525           hProcess, hFile, debugstr_w(wImageName), debugstr_w(wModuleName),
526           wine_dbgstr_longlong(BaseOfDll), SizeOfDll, Data, Flags);
527
528     if (Data)
529         FIXME("Unsupported load data parameter %p for %s\n",
530               Data, debugstr_w(wImageName));
531     if (!validate_addr64(BaseOfDll)) return FALSE;
532
533     if (!(pcs = process_find_by_handle(hProcess))) return FALSE;
534
535     if (Flags & SLMFLAG_VIRTUAL)
536     {
537         if (!wImageName) return FALSE;
538         module = module_new(pcs, wImageName, module_get_type_by_name(wImageName),
539                             TRUE, (DWORD)BaseOfDll, SizeOfDll, 0, 0);
540         if (!module) return FALSE;
541         if (wModuleName) module_set_module(module, wModuleName);
542         module->module.SymType = SymVirtual;
543
544         return TRUE;
545     }
546     if (Flags & ~(SLMFLAG_VIRTUAL))
547         FIXME("Unsupported Flags %08x for %s\n", Flags, debugstr_w(wImageName));
548
549     refresh_module_list(pcs);
550
551     /* this is a Wine extension to the API just to redo the synchronisation */
552     if (!wImageName && !hFile) return 0;
553
554     /* check if the module is already loaded, or if it's a builtin PE module with
555      * an containing ELF module
556      */
557     if (wImageName)
558     {
559         module = module_is_already_loaded(pcs, wImageName);
560         if (!module && module_is_container_loaded(pcs, wImageName, BaseOfDll))
561         {
562             /* force the loading of DLL as builtin */
563             module = pe_load_builtin_module(pcs, wImageName, BaseOfDll, SizeOfDll);
564         }
565     }
566     if (!module)
567     {
568         /* otherwise, try a regular PE module */
569         if (!(module = pe_load_native_module(pcs, wImageName, hFile, BaseOfDll, SizeOfDll)) &&
570             wImageName)
571         {
572             /* and finally an ELF or Mach-O module */
573             switch (module_get_type_by_name(wImageName))
574             {
575                 case DMT_ELF:
576                     module = elf_load_module(pcs, wImageName, BaseOfDll);
577                     break;
578                 case DMT_MACHO:
579                     module = macho_load_module(pcs, wImageName, BaseOfDll);
580                     break;
581                 default:
582                     /* Ignored */
583                     break;
584             }
585         }
586     }
587     if (!module)
588     {
589         WARN("Couldn't locate %s\n", debugstr_w(wImageName));
590         return 0;
591     }
592     module->module.NumSyms = module->ht_symbols.num_elts;
593     /* by default module_new fills module.ModuleName from a derivation
594      * of LoadedImageName. Overwrite it, if we have better information
595      */
596     if (wModuleName)
597         module_set_module(module, wModuleName);
598     if (wImageName)
599         lstrcpynW(module->module.ImageName, wImageName,
600               sizeof(module->module.ImageName) / sizeof(WCHAR));
601
602     return module->module.BaseOfImage;
603 }
604
605 /***********************************************************************
606  *                     SymLoadModule64 (DBGHELP.@)
607  */
608 DWORD64 WINAPI SymLoadModule64(HANDLE hProcess, HANDLE hFile, PCSTR ImageName,
609                                PCSTR ModuleName, DWORD64 BaseOfDll, DWORD SizeOfDll)
610 {
611     if (!validate_addr64(BaseOfDll)) return FALSE;
612     return SymLoadModule(hProcess, hFile, ImageName, ModuleName, (DWORD)BaseOfDll, SizeOfDll);
613 }
614
615 /******************************************************************
616  *              module_remove
617  *
618  */
619 BOOL module_remove(struct process* pcs, struct module* module)
620 {
621     struct module**     p;
622
623     TRACE("%s (%p)\n", debugstr_w(module->module.ModuleName), module);
624     hash_table_destroy(&module->ht_symbols);
625     hash_table_destroy(&module->ht_types);
626     HeapFree(GetProcessHeap(), 0, module->sources);
627     HeapFree(GetProcessHeap(), 0, module->addr_sorttab);
628     HeapFree(GetProcessHeap(), 0, module->dwarf2_info);
629     pool_destroy(&module->pool);
630     /* native dbghelp doesn't invoke registered callback(,CBA_SYMBOLS_UNLOADED,) here
631      * so do we
632      */
633     for (p = &pcs->lmodules; *p; p = &(*p)->next)
634     {
635         if (*p == module)
636         {
637             *p = module->next;
638             HeapFree(GetProcessHeap(), 0, module);
639             return TRUE;
640         }
641     }
642     FIXME("This shouldn't happen\n");
643     return FALSE;
644 }
645
646 /******************************************************************
647  *              SymUnloadModule (DBGHELP.@)
648  *
649  */
650 BOOL WINAPI SymUnloadModule(HANDLE hProcess, DWORD BaseOfDll)
651 {
652     struct process*     pcs;
653     struct module*      module;
654
655     pcs = process_find_by_handle(hProcess);
656     if (!pcs) return FALSE;
657     module = module_find_by_addr(pcs, BaseOfDll, DMT_UNKNOWN);
658     if (!module) return FALSE;
659     return module_remove(pcs, module);
660 }
661
662 /******************************************************************
663  *              SymUnloadModule64 (DBGHELP.@)
664  *
665  */
666 BOOL WINAPI SymUnloadModule64(HANDLE hProcess, DWORD64 BaseOfDll)
667 {
668     struct process*     pcs;
669     struct module*      module;
670
671     pcs = process_find_by_handle(hProcess);
672     if (!pcs) return FALSE;
673     if (!validate_addr64(BaseOfDll)) return FALSE;
674     module = module_find_by_addr(pcs, (DWORD)BaseOfDll, DMT_UNKNOWN);
675     if (!module) return FALSE;
676     return module_remove(pcs, module);
677 }
678
679 /******************************************************************
680  *              SymEnumerateModules (DBGHELP.@)
681  *
682  */
683 struct enum_modW64_32
684 {
685     PSYM_ENUMMODULES_CALLBACK   cb;
686     PVOID                       user;
687     char                        module[MAX_PATH];
688 };
689
690 static BOOL CALLBACK enum_modW64_32(PCWSTR name, DWORD64 base, PVOID user)
691 {
692     struct enum_modW64_32*      x = user;
693
694     WideCharToMultiByte(CP_ACP, 0, name, -1, x->module, sizeof(x->module), NULL, NULL);
695     return x->cb(x->module, (DWORD)base, x->user);
696 }
697
698 BOOL  WINAPI SymEnumerateModules(HANDLE hProcess,
699                                  PSYM_ENUMMODULES_CALLBACK EnumModulesCallback,  
700                                  PVOID UserContext)
701 {
702     struct enum_modW64_32       x;
703
704     x.cb = EnumModulesCallback;
705     x.user = UserContext;
706
707     return SymEnumerateModulesW64(hProcess, enum_modW64_32, &x);
708 }
709
710 /******************************************************************
711  *              SymEnumerateModules64 (DBGHELP.@)
712  *
713  */
714 struct enum_modW64_64
715 {
716     PSYM_ENUMMODULES_CALLBACK64 cb;
717     PVOID                       user;
718     char                        module[MAX_PATH];
719 };
720
721 static BOOL CALLBACK enum_modW64_64(PCWSTR name, DWORD64 base, PVOID user)
722 {
723     struct enum_modW64_64*      x = user;
724
725     WideCharToMultiByte(CP_ACP, 0, name, -1, x->module, sizeof(x->module), NULL, NULL);
726     return x->cb(x->module, base, x->user);
727 }
728
729 BOOL  WINAPI SymEnumerateModules64(HANDLE hProcess,
730                                    PSYM_ENUMMODULES_CALLBACK64 EnumModulesCallback,  
731                                    PVOID UserContext)
732 {
733     struct enum_modW64_64       x;
734
735     x.cb = EnumModulesCallback;
736     x.user = UserContext;
737
738     return SymEnumerateModulesW64(hProcess, enum_modW64_64, &x);
739 }
740
741 /******************************************************************
742  *              SymEnumerateModulesW64 (DBGHELP.@)
743  *
744  */
745 BOOL  WINAPI SymEnumerateModulesW64(HANDLE hProcess,
746                                     PSYM_ENUMMODULES_CALLBACKW64 EnumModulesCallback,
747                                     PVOID UserContext)
748 {
749     struct process*     pcs = process_find_by_handle(hProcess);
750     struct module*      module;
751
752     if (!pcs) return FALSE;
753     
754     for (module = pcs->lmodules; module; module = module->next)
755     {
756         if (!(dbghelp_options & SYMOPT_WINE_WITH_NATIVE_MODULES) &&
757             (module->type == DMT_ELF || module->type == DMT_MACHO))
758             continue;
759         if (!EnumModulesCallback(module->module.ModuleName,
760                                  module->module.BaseOfImage, UserContext))
761             break;
762     }
763     return TRUE;
764 }
765
766 /******************************************************************
767  *              EnumerateLoadedModules64 (DBGHELP.@)
768  *
769  */
770 struct enum_load_modW64_64
771 {
772     PENUMLOADED_MODULES_CALLBACK64      cb;
773     PVOID                               user;
774     char                                module[MAX_PATH];
775 };
776
777 static BOOL CALLBACK enum_load_modW64_64(PCWSTR name, DWORD64 base, ULONG size,
778                                          PVOID user)
779 {
780     struct enum_load_modW64_64* x = user;
781
782     WideCharToMultiByte(CP_ACP, 0, name, -1, x->module, sizeof(x->module), NULL, NULL);
783     return x->cb(x->module, base, size, x->user);
784 }
785
786 BOOL  WINAPI EnumerateLoadedModules64(HANDLE hProcess,
787                                       PENUMLOADED_MODULES_CALLBACK64 EnumLoadedModulesCallback,
788                                       PVOID UserContext)
789 {
790     struct enum_load_modW64_64  x;
791
792     x.cb = EnumLoadedModulesCallback;
793     x.user = UserContext;
794
795     return EnumerateLoadedModulesW64(hProcess, enum_load_modW64_64, &x);
796 }
797
798 /******************************************************************
799  *              EnumerateLoadedModules (DBGHELP.@)
800  *
801  */
802 struct enum_load_modW64_32
803 {
804     PENUMLOADED_MODULES_CALLBACK        cb;
805     PVOID                               user;
806     char                                module[MAX_PATH];
807 };
808
809 static BOOL CALLBACK enum_load_modW64_32(PCWSTR name, DWORD64 base, ULONG size,
810                                          PVOID user)
811 {
812     struct enum_load_modW64_32* x = user;
813     WideCharToMultiByte(CP_ACP, 0, name, -1, x->module, sizeof(x->module), NULL, NULL);
814     return x->cb(x->module, (DWORD)base, size, x->user);
815 }
816
817 BOOL  WINAPI EnumerateLoadedModules(HANDLE hProcess,
818                                     PENUMLOADED_MODULES_CALLBACK EnumLoadedModulesCallback,
819                                     PVOID UserContext)
820 {
821     struct enum_load_modW64_32  x;
822
823     x.cb = EnumLoadedModulesCallback;
824     x.user = UserContext;
825
826     return EnumerateLoadedModulesW64(hProcess, enum_load_modW64_32, &x);
827 }
828
829 /******************************************************************
830  *              EnumerateLoadedModulesW64 (DBGHELP.@)
831  *
832  */
833 BOOL  WINAPI EnumerateLoadedModulesW64(HANDLE hProcess,
834                                        PENUMLOADED_MODULES_CALLBACKW64 EnumLoadedModulesCallback,
835                                        PVOID UserContext)
836 {
837     HMODULE*    hMods;
838     WCHAR       baseW[256], modW[256];
839     DWORD       i, sz;
840     MODULEINFO  mi;
841
842     hMods = HeapAlloc(GetProcessHeap(), 0, 256 * sizeof(hMods[0]));
843     if (!hMods) return FALSE;
844
845     if (!EnumProcessModules(hProcess, hMods, 256 * sizeof(hMods[0]), &sz))
846     {
847         /* hProcess should also be a valid process handle !! */
848         FIXME("If this happens, bump the number in mod\n");
849         HeapFree(GetProcessHeap(), 0, hMods);
850         return FALSE;
851     }
852     sz /= sizeof(HMODULE);
853     for (i = 0; i < sz; i++)
854     {
855         if (!GetModuleInformation(hProcess, hMods[i], &mi, sizeof(mi)) ||
856             !GetModuleBaseNameW(hProcess, hMods[i], baseW, sizeof(baseW) / sizeof(WCHAR)))
857             continue;
858         module_fill_module(baseW, modW, sizeof(modW) / sizeof(CHAR));
859         EnumLoadedModulesCallback(modW, (DWORD_PTR)mi.lpBaseOfDll, mi.SizeOfImage,
860                                   UserContext);
861     }
862     HeapFree(GetProcessHeap(), 0, hMods);
863
864     return sz != 0 && i == sz;
865 }
866
867 /******************************************************************
868  *              SymGetModuleInfo (DBGHELP.@)
869  *
870  */
871 BOOL  WINAPI SymGetModuleInfo(HANDLE hProcess, DWORD dwAddr,
872                               PIMAGEHLP_MODULE ModuleInfo)
873 {
874     IMAGEHLP_MODULE     mi;
875     IMAGEHLP_MODULEW64  miw64;
876
877     if (sizeof(mi) < ModuleInfo->SizeOfStruct) FIXME("Wrong size\n");
878
879     miw64.SizeOfStruct = sizeof(miw64);
880     if (!SymGetModuleInfoW64(hProcess, dwAddr, &miw64)) return FALSE;
881
882     mi.SizeOfStruct  = miw64.SizeOfStruct;
883     mi.BaseOfImage   = miw64.BaseOfImage;
884     mi.ImageSize     = miw64.ImageSize;
885     mi.TimeDateStamp = miw64.TimeDateStamp;
886     mi.CheckSum      = miw64.CheckSum;
887     mi.NumSyms       = miw64.NumSyms;
888     mi.SymType       = miw64.SymType;
889     WideCharToMultiByte(CP_ACP, 0, miw64.ModuleName, -1,
890                         mi.ModuleName, sizeof(mi.ModuleName), NULL, NULL);
891     WideCharToMultiByte(CP_ACP, 0, miw64.ImageName, -1,
892                         mi.ImageName, sizeof(mi.ImageName), NULL, NULL);
893     WideCharToMultiByte(CP_ACP, 0, miw64.LoadedImageName, -1,
894                         mi.LoadedImageName, sizeof(mi.LoadedImageName), NULL, NULL);
895
896     memcpy(ModuleInfo, &mi, ModuleInfo->SizeOfStruct);
897
898     return TRUE;
899 }
900
901 /******************************************************************
902  *              SymGetModuleInfoW (DBGHELP.@)
903  *
904  */
905 BOOL  WINAPI SymGetModuleInfoW(HANDLE hProcess, DWORD dwAddr,
906                                PIMAGEHLP_MODULEW ModuleInfo)
907 {
908     IMAGEHLP_MODULEW64  miw64;
909     IMAGEHLP_MODULEW    miw;
910
911     if (sizeof(miw) < ModuleInfo->SizeOfStruct) FIXME("Wrong size\n");
912
913     miw64.SizeOfStruct = sizeof(miw64);
914     if (!SymGetModuleInfoW64(hProcess, dwAddr, &miw64)) return FALSE;
915
916     miw.SizeOfStruct  = miw64.SizeOfStruct;
917     miw.BaseOfImage   = miw64.BaseOfImage;
918     miw.ImageSize     = miw64.ImageSize;
919     miw.TimeDateStamp = miw64.TimeDateStamp;
920     miw.CheckSum      = miw64.CheckSum;
921     miw.NumSyms       = miw64.NumSyms;
922     miw.SymType       = miw64.SymType;
923     strcpyW(miw.ModuleName, miw64.ModuleName);
924     strcpyW(miw.ImageName, miw64.ImageName);
925     strcpyW(miw.LoadedImageName, miw64.LoadedImageName);
926     memcpy(ModuleInfo, &miw, ModuleInfo->SizeOfStruct);
927
928     return TRUE;
929 }
930
931 /******************************************************************
932  *              SymGetModuleInfo64 (DBGHELP.@)
933  *
934  */
935 BOOL  WINAPI SymGetModuleInfo64(HANDLE hProcess, DWORD64 dwAddr,
936                                 PIMAGEHLP_MODULE64 ModuleInfo)
937 {
938     IMAGEHLP_MODULE64   mi64;
939     IMAGEHLP_MODULEW64  miw64;
940
941     if (sizeof(mi64) < ModuleInfo->SizeOfStruct)
942     {
943         SetLastError(ERROR_MOD_NOT_FOUND); /* NOTE: native returns this error */
944         WARN("Wrong size %u\n", ModuleInfo->SizeOfStruct);
945         return FALSE;
946     }
947
948     miw64.SizeOfStruct = sizeof(miw64);
949     if (!SymGetModuleInfoW64(hProcess, dwAddr, &miw64)) return FALSE;
950
951     mi64.SizeOfStruct  = miw64.SizeOfStruct;
952     mi64.BaseOfImage   = miw64.BaseOfImage;
953     mi64.ImageSize     = miw64.ImageSize;
954     mi64.TimeDateStamp = miw64.TimeDateStamp;
955     mi64.CheckSum      = miw64.CheckSum;
956     mi64.NumSyms       = miw64.NumSyms;
957     mi64.SymType       = miw64.SymType;
958     WideCharToMultiByte(CP_ACP, 0, miw64.ModuleName, -1,
959                         mi64.ModuleName, sizeof(mi64.ModuleName), NULL, NULL);
960     WideCharToMultiByte(CP_ACP, 0, miw64.ImageName, -1,
961                         mi64.ImageName, sizeof(mi64.ImageName), NULL, NULL);
962     WideCharToMultiByte(CP_ACP, 0, miw64.LoadedImageName, -1,
963                         mi64.LoadedImageName, sizeof(mi64.LoadedImageName), NULL, NULL);
964     WideCharToMultiByte(CP_ACP, 0, miw64.LoadedPdbName, -1,
965                         mi64.LoadedPdbName, sizeof(mi64.LoadedPdbName), NULL, NULL);
966
967     mi64.CVSig         = miw64.CVSig;
968     WideCharToMultiByte(CP_ACP, 0, miw64.CVData, -1,
969                         mi64.CVData, sizeof(mi64.CVData), NULL, NULL);
970     mi64.PdbSig        = miw64.PdbSig;
971     mi64.PdbSig70      = miw64.PdbSig70;
972     mi64.PdbAge        = miw64.PdbAge;
973     mi64.PdbUnmatched  = miw64.PdbUnmatched;
974     mi64.DbgUnmatched  = miw64.DbgUnmatched;
975     mi64.LineNumbers   = miw64.LineNumbers;
976     mi64.GlobalSymbols = miw64.GlobalSymbols;
977     mi64.TypeInfo      = miw64.TypeInfo;
978     mi64.SourceIndexed = miw64.SourceIndexed;
979     mi64.Publics       = miw64.Publics;
980
981     memcpy(ModuleInfo, &mi64, ModuleInfo->SizeOfStruct);
982
983     return TRUE;
984 }
985
986 /******************************************************************
987  *              SymGetModuleInfoW64 (DBGHELP.@)
988  *
989  */
990 BOOL  WINAPI SymGetModuleInfoW64(HANDLE hProcess, DWORD64 dwAddr,
991                                  PIMAGEHLP_MODULEW64 ModuleInfo)
992 {
993     struct process*     pcs = process_find_by_handle(hProcess);
994     struct module*      module;
995     IMAGEHLP_MODULEW64  miw64;
996
997     TRACE("%p %s %p\n", hProcess, wine_dbgstr_longlong(dwAddr), ModuleInfo);
998
999     if (!pcs) return FALSE;
1000     if (ModuleInfo->SizeOfStruct > sizeof(*ModuleInfo)) return FALSE;
1001     module = module_find_by_addr(pcs, dwAddr, DMT_UNKNOWN);
1002     if (!module) return FALSE;
1003
1004     miw64 = module->module;
1005
1006     /* update debug information from container if any */
1007     if (module->module.SymType == SymNone)
1008     {
1009         module = module_get_container(pcs, module);
1010         if (module && module->module.SymType != SymNone)
1011         {
1012             miw64.SymType = module->module.SymType;
1013             miw64.NumSyms = module->module.NumSyms;
1014         }
1015     }
1016     memcpy(ModuleInfo, &miw64, ModuleInfo->SizeOfStruct);
1017     return TRUE;
1018 }
1019
1020 /***********************************************************************
1021  *              SymGetModuleBase (DBGHELP.@)
1022  */
1023 DWORD WINAPI SymGetModuleBase(HANDLE hProcess, DWORD dwAddr)
1024 {
1025     struct process*     pcs = process_find_by_handle(hProcess);
1026     struct module*      module;
1027
1028     if (!pcs) return 0;
1029     module = module_find_by_addr(pcs, dwAddr, DMT_UNKNOWN);
1030     if (!module) return 0;
1031     return module->module.BaseOfImage;
1032 }
1033
1034 /***********************************************************************
1035  *              SymGetModuleBase64 (DBGHELP.@)
1036  */
1037 DWORD64 WINAPI SymGetModuleBase64(HANDLE hProcess, DWORD64 dwAddr)
1038 {
1039     if (!validate_addr64(dwAddr)) return 0;
1040     return SymGetModuleBase(hProcess, (DWORD)dwAddr);
1041 }
1042
1043 /******************************************************************
1044  *              module_reset_debug_info
1045  * Removes any debug information linked to a given module.
1046  */
1047 void module_reset_debug_info(struct module* module)
1048 {
1049     module->sortlist_valid = TRUE;
1050     module->sorttab_size = 0;
1051     module->addr_sorttab = NULL;
1052     module->num_sorttab = module->num_symbols = 0;
1053     hash_table_destroy(&module->ht_symbols);
1054     module->ht_symbols.num_buckets = 0;
1055     module->ht_symbols.buckets = NULL;
1056     hash_table_destroy(&module->ht_types);
1057     module->ht_types.num_buckets = 0;
1058     module->ht_types.buckets = NULL;
1059     module->vtypes.num_elts = 0;
1060     hash_table_destroy(&module->ht_symbols);
1061     module->sources_used = module->sources_alloc = 0;
1062     module->sources = NULL;
1063 }
1064
1065 /******************************************************************
1066  *              SymRefreshModuleList (DBGHELP.@)
1067  */
1068 BOOL WINAPI SymRefreshModuleList(HANDLE hProcess)
1069 {
1070     struct process*     pcs;
1071
1072     TRACE("(%p)\n", hProcess);
1073
1074     if (!(pcs = process_find_by_handle(hProcess))) return FALSE;
1075
1076     return refresh_module_list(pcs);
1077 }