Fix a memory leak in the ole storage implementation.
[wine] / dlls / dbghelp / dbghelp.c
1 /*
2  * File dbghelp.c - generic routines (process) for dbghelp DLL
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include "config.h"
22
23 #include "dbghelp_private.h"
24 #include "winerror.h"
25 #include "psapi.h"
26 #include "wine/debug.h"
27
28 WINE_DEFAULT_DEBUG_CHANNEL(dbghelp);
29
30 /* TODO
31  *  - support for symbols' types is still partly missing
32  *      + C++ support
33  *      + funcargtype:s are (partly) wrong: they should be a specific struct (like
34  *        typedef) pointing to the actual type (and not a direct access)
35  *      + we should store the underlying type for an enum in the symt_enum struct
36  *      + for enums, we store the names & values (associated to the enum type), 
37  *        but those values are not directly usable from a debugger (that's why, I
38  *        assume, that we have also to define constants for enum values, as 
39  *        Codeview does BTW.
40  *      + SymGetType(TI_GET_LENGTH) takes a ULONG64 (yurk, ugly)
41  *  - SymGetLine{Next|Prev} don't work as expected (they don't seem to work across
42  *    functions, and even across function blocks...). Basically, for *Next* to work
43  *    it requires an address after the prolog of the func (the base address of the 
44  *    func doesn't work)
45  *  - most options (dbghelp_options) are not used (loading lines...)
46  *  - in symbol lookup by name, we don't use RE everywhere we should. Moreover, when
47  *    we're supposed to use RE, it doesn't make use of our hash tables. Therefore,
48  *    we could use hash if name isn't a RE, and fall back to a full search when we
49  *    get a full RE
50  *  - msc:
51  *      + we should add parameters' types to the function's signature
52  *        while processing a function's parameters
53  *      + add support for function-less labels (as MSC seems to define them)
54  *      + C++ management
55  *  - stabs: 
56  *      + when, in a same module, the same definition is used in several compilation
57  *        units, we get several definitions of the same object (especially 
58  *        struct/union). we should find a way not to duplicate them
59  *      + in some cases (dlls/user/dialog16.c DIALOG_GetControl16), the same static
60  *        global variable is defined several times (at different scopes). We are
61  *        getting several of those while looking for a unique symbol. Part of the 
62  *        issue is that we don't give a scope to a static variable inside a function
63  *      + C++ management
64  *  - implement the callback notification mechanism
65  */
66
67 unsigned   dbghelp_options = SYMOPT_UNDNAME;
68 HANDLE     hMsvcrt = NULL;
69
70 /***********************************************************************
71  *           DllMain (DEBUGHLP.@)
72  */
73 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
74 {
75     switch (fdwReason)
76     {
77     case DLL_PROCESS_ATTACH:    break;
78     case DLL_PROCESS_DETACH:
79         if (hMsvcrt) FreeLibrary(hMsvcrt);
80         break;
81     case DLL_THREAD_ATTACH:     break;
82     case DLL_THREAD_DETACH:     break;
83     default:                    break;
84     }
85     return TRUE;
86 }
87
88 static struct process* process_first /* = NULL */;
89
90 /******************************************************************
91  *              process_find_by_handle
92  *
93  */
94 struct process*    process_find_by_handle(HANDLE hProcess)
95 {
96     struct process* p;
97
98     for (p = process_first; p && p->handle != hProcess; p = p->next);
99     if (!p) SetLastError(ERROR_INVALID_HANDLE);
100     return p;
101 }
102
103 /******************************************************************
104  *              SymSetSearchPath (DBGHELP.@)
105  *
106  */
107 BOOL WINAPI SymSetSearchPath(HANDLE hProcess, PSTR searchPath)
108 {
109     struct process* pcs = process_find_by_handle(hProcess);
110
111     if (!pcs) return FALSE;
112     if (!searchPath) return FALSE;
113
114     HeapFree(GetProcessHeap(), 0, pcs->search_path);
115     pcs->search_path = strcpy(HeapAlloc(GetProcessHeap(), 0, strlen(searchPath) + 1),
116                               searchPath);
117     return TRUE;
118 }
119
120 /***********************************************************************
121  *              SymGetSearchPath (DBGHELP.@)
122  */
123 BOOL WINAPI SymGetSearchPath(HANDLE hProcess, LPSTR szSearchPath, 
124                              DWORD SearchPathLength)
125 {
126     struct process* pcs = process_find_by_handle(hProcess);
127     if (!pcs) return FALSE;
128
129     lstrcpynA(szSearchPath, pcs->search_path, SearchPathLength);
130     return TRUE;
131 }
132
133 /******************************************************************
134  *              invade_process
135  *
136  * SymInitialize helper: loads in dbghelp all known (and loaded modules)
137  * this assumes that hProcess is a handle on a valid process
138  */
139 static BOOL WINAPI process_invade_cb(char* name, DWORD base, DWORD size, void* user)
140 {
141     char        tmp[MAX_PATH];
142     HANDLE      hProcess = (HANDLE)user;
143
144     if (!GetModuleFileNameExA(hProcess, (HMODULE)base, 
145                               tmp, sizeof(tmp)))
146         lstrcpynA(tmp, name, sizeof(tmp));
147
148     SymLoadModule(hProcess, 0, tmp, name, base, size);
149     return TRUE;
150 }
151
152 /******************************************************************
153  *              SymInitialize (DBGHELP.@)
154  *
155  * The initialisation of a dbghelp's context.
156  * Note that hProcess doesn't need to be a valid process handle (except
157  * when fInvadeProcess is TRUE).
158  * Since, we're also allow to load ELF (pure) libraries and Wine ELF libraries 
159  * containing PE (and NE) module(s), here's how we handle it:
160  * - we load every module (ELF, NE, PE) passed in SymLoadModule
161  * - in fInvadeProcess (in SymInitialize) is TRUE, we set up what is called ELF
162  *   synchronization: hProcess should be a valid process handle, and we hook
163  *   ourselves on hProcess's loaded ELF-modules, and keep this list in sync with
164  *   our internal ELF modules representation (loading / unloading). This way,
165  *   we'll pair every loaded builtin PE module with its ELF counterpart (and
166  *   access its debug information).
167  * - if fInvadeProcess (in SymInitialize) is FALSE, we won't be able to
168  *   make the peering between a builtin PE module and its ELF counterpart, hence
169  *   we won't be able to provide the requested debug information. We'll
170  *   however be able to load native PE modules (and their debug information)
171  *   without any trouble.
172  * Note also that this scheme can be intertwined with the deferred loading 
173  * mechanism (ie only load the debug information when we actually need it).
174  */
175 BOOL WINAPI SymInitialize(HANDLE hProcess, PSTR UserSearchPath, BOOL fInvadeProcess)
176 {
177     struct process*     pcs;
178
179     TRACE("(%p %s %u)\n", hProcess, debugstr_a(UserSearchPath), fInvadeProcess);
180
181     if (process_find_by_handle(hProcess))
182         FIXME("what to do ??\n");
183
184     pcs = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*pcs));
185     if (!pcs) return FALSE;
186
187     pcs->handle = hProcess;
188
189     if (UserSearchPath)
190     {
191         pcs->search_path = strcpy(HeapAlloc(GetProcessHeap(), 0, strlen(UserSearchPath) + 1), 
192                                   UserSearchPath);
193     }
194     else
195     {
196         unsigned        size;
197         unsigned        len;
198
199         pcs->search_path = HeapAlloc(GetProcessHeap(), 0, len = MAX_PATH);
200         while ((size = GetCurrentDirectoryA(len, pcs->search_path)) >= len)
201             pcs->search_path = HeapReAlloc(GetProcessHeap(), 0, pcs->search_path, len *= 2);
202         pcs->search_path = HeapReAlloc(GetProcessHeap(), 0, pcs->search_path, size + 1);
203
204         len = GetEnvironmentVariableA("_NT_SYMBOL_PATH", NULL, 0);
205         if (len)
206         {
207             pcs->search_path = HeapReAlloc(GetProcessHeap(), 0, pcs->search_path, size + 1 + len + 1);
208             pcs->search_path[size] = ';';
209             GetEnvironmentVariableA("_NT_SYMBOL_PATH", pcs->search_path + size + 1, len);
210             size += 1 + len;
211         }
212         len = GetEnvironmentVariableA("_NT_ALTERNATE_SYMBOL_PATH", NULL, 0);
213         if (len)
214         {
215             pcs->search_path = HeapReAlloc(GetProcessHeap(), 0, pcs->search_path, size + 1 + len + 1);
216             pcs->search_path[size] = ';';
217             GetEnvironmentVariableA("_NT_ALTERNATE_SYMBOL_PATH", pcs->search_path + size + 1, len);
218             size += 1 + len;
219         }
220     }
221
222     pcs->lmodules = NULL;
223     pcs->dbg_hdr_addr = 0;
224     pcs->next = process_first;
225     process_first = pcs;
226
227     if (fInvadeProcess)
228     {
229         if (!elf_read_wine_loader_dbg_info(pcs))
230         {
231             SymCleanup(hProcess);
232             return FALSE;
233         }
234         EnumerateLoadedModules(hProcess, process_invade_cb, (void*)hProcess);
235         elf_synchronize_module_list(pcs);
236     }
237
238     return TRUE;
239 }
240
241 /******************************************************************
242  *              SymCleanup (DBGHELP.@)
243  *
244  */
245 BOOL WINAPI SymCleanup(HANDLE hProcess)
246 {
247     struct process**    ppcs;
248     struct process*     next;
249
250     for (ppcs = &process_first; *ppcs; ppcs = &(*ppcs)->next)
251     {
252         if ((*ppcs)->handle == hProcess)
253         {
254             while ((*ppcs)->lmodules) module_remove(*ppcs, (*ppcs)->lmodules);
255
256             HeapFree(GetProcessHeap(), 0, (*ppcs)->search_path);
257             next = (*ppcs)->next;
258             HeapFree(GetProcessHeap(), 0, *ppcs);
259             *ppcs = next;
260             return TRUE;
261         }
262     }
263     return FALSE;
264 }
265
266 /******************************************************************
267  *              SymSetOptions (DBGHELP.@)
268  *
269  */
270 DWORD WINAPI SymSetOptions(DWORD opts)
271 {
272     return dbghelp_options = opts;
273 }
274
275 /******************************************************************
276  *              SymGetOptions (DBGHELP.@)
277  *
278  */
279 DWORD WINAPI SymGetOptions(void)
280 {
281     return dbghelp_options;
282 }
283
284 /******************************************************************
285  *              SymSetContext (DBGHELP.@)
286  *
287  */
288 BOOL WINAPI SymSetContext(HANDLE hProcess, PIMAGEHLP_STACK_FRAME StackFrame,
289                           PIMAGEHLP_CONTEXT Context)
290 {
291     struct process* pcs = process_find_by_handle(hProcess);
292     if (!pcs) return FALSE;
293
294     pcs->ctx_frame = *StackFrame;
295     /* MSDN states that Context is not (no longer?) used */
296     return TRUE;
297 }
298
299 /***********************************************************************
300  *              SymRegisterCallback (DBGHELP.@)
301  */
302 BOOL WINAPI SymRegisterCallback(HANDLE hProcess, 
303                                 PSYMBOL_REGISTERED_CALLBACK CallbackFunction,
304                                 PVOID UserContext)
305 {
306     FIXME("(%p, %p, %p): stub\n", hProcess, CallbackFunction, UserContext);
307     SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
308     return FALSE;
309 }
310
311 /* This is imagehlp version not dbghelp !! */
312 static API_VERSION api_version = { 4, 0, 2, 0 };
313
314 /***********************************************************************
315  *           ImagehlpApiVersion (DBGHELP.@)
316  */
317 LPAPI_VERSION WINAPI ImagehlpApiVersion(VOID)
318 {
319     return &api_version;
320 }
321
322 /***********************************************************************
323  *           ImagehlpApiVersionEx (DBGHELP.@)
324  */
325 LPAPI_VERSION WINAPI ImagehlpApiVersionEx(LPAPI_VERSION AppVersion)
326 {
327     if (!AppVersion) return NULL;
328
329     AppVersion->MajorVersion = api_version.MajorVersion;
330     AppVersion->MinorVersion = api_version.MinorVersion;
331     AppVersion->Revision = api_version.Revision;
332     AppVersion->Reserved = api_version.Reserved;
333
334     return AppVersion;
335 }