ntdll: Check and fix PE header protections.
[wine] / 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 "config.h"
22 #include <stdlib.h>
23 #include <stdio.h>
24 #include <string.h>
25 #include <assert.h>
26
27 #include "dbghelp_private.h"
28 #include "wine/debug.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(dbghelp);
31
32 /******************************************************************
33  *              source_find
34  *
35  * check whether a source file has already been stored
36  */
37 static unsigned source_find(const struct module* module, const char* name)
38 {
39     char*       ptr = module->sources;
40
41     while (*ptr)
42     {
43         if (strcmp(ptr, name) == 0) return ptr - module->sources;
44         ptr += strlen(ptr) + 1;
45     }
46     return (unsigned)-1;
47 }
48
49 /******************************************************************
50  *              source_new
51  *
52  * checks if source exists. if not, add it
53  */
54 unsigned source_new(struct module* module, const char* base, const char* name)
55 {
56     unsigned    ret;
57     const char* full;
58     char*       tmp = NULL;
59
60     if (!name) return (unsigned)-1;
61     if (!base || *name == '/')
62         full = name;
63     else
64     {
65         unsigned bsz = strlen(base);
66
67         tmp = HeapAlloc(GetProcessHeap(), 0, bsz + 1 + strlen(name) + 1);
68         if (!tmp) return (unsigned)-1;
69         full = tmp;
70         strcpy(tmp, base);
71         if (tmp[bsz - 1] != '/') tmp[bsz++] = '/';
72         strcpy(&tmp[bsz], name);
73     }
74     if (!module->sources || (ret = source_find(module, full)) == (unsigned)-1)
75     {
76         int len = strlen(full) + 1;
77         if (module->sources_used + len + 1 > module->sources_alloc)
78         {
79             if (!module->sources)
80             {
81                 module->sources_alloc = (module->sources_used + len + 1 + 255) & ~255;
82                 module->sources = HeapAlloc(GetProcessHeap(), 0, module->sources_alloc);
83             }
84             else
85             {
86                 module->sources_alloc = max( module->sources_alloc * 2,
87                                              (module->sources_used + len + 1 + 255) & ~255 );
88                 module->sources = HeapReAlloc(GetProcessHeap(), 0, module->sources,
89                                               module->sources_alloc);
90             }
91         }
92         ret = module->sources_used;
93         memcpy(module->sources + module->sources_used, full, len);
94         module->sources_used += len;
95         module->sources[module->sources_used] = '\0';
96     }
97     HeapFree(GetProcessHeap(), 0, tmp);
98     return ret;
99 }
100
101 /******************************************************************
102  *              source_get
103  *
104  * returns a stored source file name
105  */
106 const char* source_get(const struct module* module, unsigned idx)
107 {
108     if (idx == -1) return "";
109     assert(module->sources);
110     return module->sources + idx;
111 }
112
113 /******************************************************************
114  *              SymEnumSourceFiles (DBGHELP.@)
115  *
116  */
117 BOOL WINAPI SymEnumSourceFiles(HANDLE hProcess, ULONG64 ModBase, PCSTR Mask,
118                                PSYM_ENUMSOURCEFILES_CALLBACK cbSrcFiles,
119                                PVOID UserContext)
120 {
121     struct module_pair  pair;
122     SOURCEFILE          sf;
123     char*               ptr;
124     
125     if (!cbSrcFiles) return FALSE;
126     pair.pcs = process_find_by_handle(hProcess);
127     if (!pair.pcs) return FALSE;
128          
129     if (ModBase)
130     {
131         pair.requested = module_find_by_addr(pair.pcs, ModBase, DMT_UNKNOWN);
132         if (!module_get_debug(&pair)) return FALSE;
133     }
134     else
135     {
136         if (Mask[0] == '!')
137         {
138             pair.requested = module_find_by_nameA(pair.pcs, Mask + 1);
139             if (!module_get_debug(&pair)) return FALSE;
140         }
141         else
142         {
143             FIXME("Unsupported yet (should get info from current context)\n");
144             return FALSE;
145         }
146     }
147     if (!pair.effective->sources) return FALSE;
148     for (ptr = pair.effective->sources; *ptr; ptr += strlen(ptr) + 1)
149     {
150         /* FIXME: not using Mask */
151         sf.ModBase = ModBase;
152         sf.FileName = ptr;
153         if (!cbSrcFiles(&sf, UserContext)) break;
154     }
155
156     return TRUE;
157 }
158
159 /******************************************************************
160  *              SymGetSourceFileToken (DBGHELP.@)
161  *
162  */
163 BOOL WINAPI SymGetSourceFileToken(HANDLE hProcess, ULONG64 base,
164                                   PCSTR src, PVOID* token, DWORD* size)
165 {
166     FIXME("%p %s %s %p %p: stub!\n",
167           hProcess, wine_dbgstr_longlong(base), debugstr_a(src), token, size);
168     SetLastError(ERROR_NOT_SUPPORTED);
169     return FALSE;
170 }
171
172 /******************************************************************
173  *              SymGetSourceFileTokenW (DBGHELP.@)
174  *
175  */
176 BOOL WINAPI SymGetSourceFileTokenW(HANDLE hProcess, ULONG64 base,
177                                    PCWSTR src, PVOID* token, DWORD* size)
178 {
179     FIXME("%p %s %s %p %p: stub!\n",
180           hProcess, wine_dbgstr_longlong(base), debugstr_w(src), token, size);
181     SetLastError(ERROR_NOT_SUPPORTED);
182     return FALSE;
183 }