dlls/dbghelp/source.c

   1 /*
   2  * File source.c - source files management
   3  *
   4  * Copyright (C) 2004,      Eric Pouech.
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  *
  20  */
  21 #include <stdlib.h>
  22 #include <stdio.h>
  23 #include <string.h>
  24 #include <assert.h>
  25
  26 #include "dbghelp_private.h"
  27 #include "wine/debug.h"
  28
  29 WINE_DEFAULT_DEBUG_CHANNEL(dbghelp);
  30
  31 static struct module*   rb_module;
  32 struct source_rb
  33 {
  34     struct wine_rb_entry        entry;
  35     unsigned                    source;
  36 };
  37
  38 int source_rb_compare(const void *key, const struct wine_rb_entry *entry)
  39 {
  40     const struct source_rb *t = WINE_RB_ENTRY_VALUE(entry, const struct source_rb, entry);
  41
  42     return strcmp((const char*)key, rb_module->sources + t->source);
  43 }
  44
  45 /******************************************************************
  46  *              source_find
  47  *
  48  * check whether a source file has already been stored
  49  */
  50 static unsigned source_find(const char* name)
  51 {
  52     struct wine_rb_entry*       e;
  53
  54     e = wine_rb_get(&rb_module->sources_offsets_tree, name);
  55     if (!e) return -1;
  56     return WINE_RB_ENTRY_VALUE(e, struct source_rb, entry)->source;
  57 }
  58
  59 /******************************************************************
  60  *              source_new
  61  *
  62  * checks if source exists. if not, add it
  63  */
  64 unsigned source_new(struct module* module, const char* base, const char* name)
  65 {
  66     unsigned    ret = -1;
  67     const char* full;
  68     char*       tmp = NULL;
  69
  70     if (!name) return ret;
  71     if (!base || *name == '/')
  72         full = name;
  73     else
  74     {
  75         unsigned bsz = strlen(base);
  76
  77         tmp = HeapAlloc(GetProcessHeap(), 0, bsz + 1 + strlen(name) + 1);
  78         if (!tmp) return ret;
  79         full = tmp;
  80         strcpy(tmp, base);
  81         if (bsz && tmp[bsz - 1] != '/') tmp[bsz++] = '/';
  82         strcpy(&tmp[bsz], name);
  83     }
  84     rb_module = module;
  85     if (!module->sources || (ret = source_find(full)) == (unsigned)-1)
  86     {
  87         char* new;
  88         int len = strlen(full) + 1;
  89         struct source_rb* rb;
  90
  91         if (module->sources_used + len + 1 > module->sources_alloc)
  92         {
  93             if (!module->sources)
  94             {
  95                 module->sources_alloc = (module->sources_used + len + 1 + 255) & ~255;
  96                 new = HeapAlloc(GetProcessHeap(), 0, module->sources_alloc);
  97             }
  98             else
  99             {
 100                 module->sources_alloc = max( module->sources_alloc * 2,
 101                                              (module->sources_used + len + 1 + 255) & ~255 );
 102                 new = HeapReAlloc(GetProcessHeap(), 0, module->sources,
 103                                   module->sources_alloc);
 104             }
 105             if (!new) goto done;
 106             module->sources = new;
 107         }
 108         ret = module->sources_used;
 109         memcpy(module->sources + module->sources_used, full, len);
 110         module->sources_used += len;
 111         module->sources[module->sources_used] = '\0';
 112         if ((rb = pool_alloc(&module->pool, sizeof(*rb))))
 113         {
 114             rb->source = ret;
 115             wine_rb_put(&module->sources_offsets_tree, full, &rb->entry);
 116         }
 117     }
 118 done:
 119     HeapFree(GetProcessHeap(), 0, tmp);
 120     return ret;
 121 }
 122
 123 /******************************************************************
 124  *              source_get
 125  *
 126  * returns a stored source file name
 127  */
 128 const char* source_get(const struct module* module, unsigned idx)
 129 {
 130     if (idx == -1) return "";
 131     assert(module->sources);
 132     return module->sources + idx;
 133 }
 134
 135 /******************************************************************
 136  *              SymEnumSourceFilesW (DBGHELP.@)
 137  *
 138  */
 139 BOOL WINAPI SymEnumSourceFilesW(HANDLE hProcess, ULONG64 ModBase, PCWSTR Mask,
 140                                 PSYM_ENUMSOURCEFILES_CALLBACKW cbSrcFiles,
 141                                 PVOID UserContext)
 142 {
 143     struct module_pair  pair;
 144     SOURCEFILEW         sf;
 145     char*               ptr;
 146     WCHAR*              conversion_buffer = NULL;
 147     DWORD               conversion_buffer_len = 0;
 148
 149     if (!cbSrcFiles) return FALSE;
 150     pair.pcs = process_find_by_handle(hProcess);
 151     if (!pair.pcs) return FALSE;
 152
 153     if (ModBase)
 154     {
 155         pair.requested = module_find_by_addr(pair.pcs, ModBase, DMT_UNKNOWN);
 156         if (!module_get_debug(&pair)) return FALSE;
 157     }
 158     else
 159     {
 160         if (Mask[0] == '!')
 161         {
 162             pair.requested = module_find_by_nameW(pair.pcs, Mask + 1);
 163             if (!module_get_debug(&pair)) return FALSE;
 164         }
 165         else
 166         {
 167             FIXME("Unsupported yet (should get info from current context)\n");
 168             return FALSE;
 169         }
 170     }
 171     if (!pair.effective->sources) return FALSE;
 172     for (ptr = pair.effective->sources; *ptr; ptr += strlen(ptr) + 1)
 173     {
 174         DWORD len = MultiByteToWideChar(CP_ACP, 0, ptr, -1, NULL, 0);
 175
 176         if (len > conversion_buffer_len)
 177         {
 178             HeapFree(GetProcessHeap(), 0, conversion_buffer);
 179             conversion_buffer = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 180             if (!conversion_buffer) return FALSE;
 181             conversion_buffer_len = len;
 182         }
 183
 184         MultiByteToWideChar(CP_ACP, 0, ptr, -1, conversion_buffer, len);
 185
 186         /* FIXME: not using Mask */
 187         sf.ModBase = ModBase;
 188         sf.FileName = conversion_buffer;
 189         if (!cbSrcFiles(&sf, UserContext)) break;
 190     }
 191
 192     HeapFree(GetProcessHeap(), 0, conversion_buffer);
 193     return TRUE;
 194 }
 195
 196 struct enum_sources_files_context
 197 {
 198     PSYM_ENUMSOURCEFILES_CALLBACK callbackA;
 199     PVOID caller_context;
 200     char *conversion_buffer;
 201     DWORD conversion_buffer_len;
 202     DWORD callback_error;
 203 };
 204
 205 static BOOL CALLBACK enum_source_files_W_to_A(PSOURCEFILEW source_file, PVOID context)
 206 {
 207     struct enum_sources_files_context *ctx = context;
 208     SOURCEFILE source_fileA;
 209     DWORD len;
 210
 211     len = WideCharToMultiByte(CP_ACP, 0, source_file->FileName, -1, NULL, 0, NULL, NULL);
 212     if (len > ctx->conversion_buffer_len)
 213     {
 214         char *ptr = ctx->conversion_buffer ? HeapReAlloc(GetProcessHeap(), 0, ctx->conversion_buffer, len) :
 215                                              HeapAlloc(GetProcessHeap(), 0, len);
 216
 217         if (!ptr)
 218         {
 219             ctx->callback_error = ERROR_OUTOFMEMORY;
 220             return FALSE;
 221         }
 222
 223         ctx->conversion_buffer = ptr;
 224         ctx->conversion_buffer_len = len;
 225     }
 226
 227     WideCharToMultiByte(CP_ACP, 0, source_file->FileName, -1, ctx->conversion_buffer, len, NULL, NULL);
 228
 229     source_fileA.ModBase = source_file->ModBase;
 230     source_fileA.FileName = ctx->conversion_buffer;
 231     return ctx->callbackA(&source_fileA, ctx->caller_context);
 232 }
 233
 234 /******************************************************************
 235  *              SymEnumSourceFiles (DBGHELP.@)
 236  *
 237  */
 238 BOOL WINAPI SymEnumSourceFiles(HANDLE hProcess, ULONG64 ModBase, PCSTR Mask,
 239                                PSYM_ENUMSOURCEFILES_CALLBACK cbSrcFiles,
 240                                PVOID UserContext)
 241 {
 242     WCHAR *maskW = NULL;
 243     PSYM_ENUMSOURCEFILES_CALLBACKW callbackW;
 244     PVOID context;
 245     struct enum_sources_files_context callback_context = {cbSrcFiles, UserContext};
 246     BOOL ret;
 247
 248     if (Mask)
 249     {
 250         DWORD len = MultiByteToWideChar(CP_ACP, 0, Mask, -1, NULL, 0);
 251
 252         maskW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 253         if (!maskW)
 254         {
 255             SetLastError(ERROR_OUTOFMEMORY);
 256             return FALSE;
 257         }
 258
 259         MultiByteToWideChar(CP_ACP, 0, Mask, -1, maskW, len);
 260     }
 261
 262     if (cbSrcFiles)
 263     {
 264         callbackW = enum_source_files_W_to_A;
 265         context = &callback_context;
 266     }
 267     else
 268     {
 269         callbackW = NULL;
 270         context = UserContext;
 271     }
 272
 273     ret = SymEnumSourceFilesW(hProcess, ModBase, maskW, callbackW, context);
 274
 275     if (callback_context.callback_error)
 276     {
 277         SetLastError(callback_context.callback_error);
 278         ret = FALSE;
 279     }
 280
 281     HeapFree(GetProcessHeap(), 0, callback_context.conversion_buffer);
 282     HeapFree(GetProcessHeap(), 0, maskW);
 283
 284     return ret;
 285 }
 286
 287 /******************************************************************
 288  *              SymEnumSourceLines (DBGHELP.@)
 289  *
 290  */
 291 BOOL WINAPI SymEnumSourceLines(HANDLE hProcess, ULONG64 base, PCSTR obj,
 292                                PCSTR file, DWORD line, DWORD flags,
 293                                PSYM_ENUMLINES_CALLBACK EnumLinesCallback,
 294                                PVOID UserContext)
 295 {
 296     FIXME("%p %I64x %s %s %lu %lu %p %p: stub!\n",
 297           hProcess, base, debugstr_a(obj), debugstr_a(file),
 298           line, flags, EnumLinesCallback, UserContext);
 299     SetLastError(ERROR_NOT_SUPPORTED);
 300     return FALSE;
 301 }
 302
 303 /******************************************************************
 304  *               SymEnumSourceLinesW(DBGHELP.@)
 305  *
 306  */
 307 BOOL WINAPI SymEnumSourceLinesW(HANDLE hProcess, ULONG64 base, PCWSTR obj,
 308                                 PCWSTR file, DWORD line, DWORD flags,
 309                                 PSYM_ENUMLINES_CALLBACKW EnumLinesCallback,
 310                                 PVOID UserContext)
 311 {
 312     FIXME("%p %I64x %s %s %lu %lu %p %p: stub!\n",
 313           hProcess, base, debugstr_w(obj), debugstr_w(file),
 314           line, flags, EnumLinesCallback, UserContext);
 315     SetLastError(ERROR_NOT_SUPPORTED);
 316     return FALSE;
 317 }
 318
 319 /******************************************************************
 320  *              SymGetSourceFileToken (DBGHELP.@)
 321  *
 322  */
 323 BOOL WINAPI SymGetSourceFileToken(HANDLE hProcess, ULONG64 base,
 324                                   PCSTR src, PVOID* token, DWORD* size)
 325 {
 326     FIXME("%p %I64x %s %p %p: stub!\n",
 327           hProcess, base, debugstr_a(src), token, size);
 328     SetLastError(ERROR_NOT_SUPPORTED);
 329     return FALSE;
 330 }
 331
 332 /******************************************************************
 333  *              SymGetSourceFileTokenW (DBGHELP.@)
 334  *
 335  */
 336 BOOL WINAPI SymGetSourceFileTokenW(HANDLE hProcess, ULONG64 base,
 337                                    PCWSTR src, PVOID* token, DWORD* size)
 338 {
 339     FIXME("%p %I64x %s %p %p: stub!\n",
 340           hProcess, base, debugstr_w(src), token, size);
 341     SetLastError(ERROR_NOT_SUPPORTED);
 342     return FALSE;
 343 }