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