advapi32: Move duplicated code into function.
[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 #ifdef HAVE_REGEX_H
30 # include <regex.h>
31 #endif
32
33 WINE_DEFAULT_DEBUG_CHANNEL(dbghelp);
34
35 /******************************************************************
36  *              source_find
37  *
38  * check whether a source file has already been stored
39  */
40 static unsigned source_find(const struct module* module, const char* name)
41 {
42     char*       ptr = module->sources;
43
44     while (*ptr)
45     {
46         if (strcmp(ptr, name) == 0) return ptr - module->sources;
47         ptr += strlen(ptr) + 1;
48     }
49     return (unsigned)-1;
50 }
51
52 /******************************************************************
53  *              source_new
54  *
55  * checks if source exists. if not, add it
56  */
57 unsigned source_new(struct module* module, const char* base, const char* name)
58 {
59     int         len;
60     unsigned    ret;
61     const char* full;
62
63     if (!name) return (unsigned)-1;
64     if (!base || *name == '/')
65         full = name;
66     else
67     {
68         unsigned bsz = strlen(base);
69         char* tmp = HeapAlloc(GetProcessHeap(), 0, bsz + 1 + strlen(name) + 1);
70
71         if (!tmp) return (unsigned)-1;
72         full = tmp;
73         strcpy(tmp, base);
74         if (tmp[bsz - 1] != '/') tmp[bsz++] = '/';
75         strcpy(&tmp[bsz], name);
76     }
77     if (module->sources && (ret = source_find(module, full)) != (unsigned)-1)
78         return ret;
79
80     len = strlen(full) + 1;
81     if (module->sources_used + len + 1 > module->sources_alloc)
82     {
83         /* Alloc by block of 256 bytes */
84         module->sources_alloc = (module->sources_used + len + 1 + 255) & ~255;
85         if (!module->sources)
86             module->sources = HeapAlloc(GetProcessHeap(), 0, module->sources_alloc);
87         else
88             module->sources = HeapReAlloc(GetProcessHeap(), 0, module->sources,
89                                           module->sources_alloc);
90     }
91     ret = module->sources_used;
92     strcpy(module->sources + module->sources_used, full);
93     module->sources_used += len;
94     module->sources[module->sources_used] = '\0';
95     if (full != name) HeapFree(GetProcessHeap(), 0, (char*)full);
96     return ret;
97 }
98
99 /******************************************************************
100  *              source_get
101  *
102  * returns a stored source file name
103  */
104 const char* source_get(const struct module* module, unsigned idx)
105 {
106     if (idx == -1) return "";
107     assert(module->sources);
108     return module->sources + idx;
109 }
110
111 /******************************************************************
112  *              SymEnumSourceFiles (DBGHELP.@)
113  *
114  */
115 BOOL WINAPI SymEnumSourceFiles(HANDLE hProcess, ULONG64 ModBase, PCSTR Mask,
116                                PSYM_ENUMSOURCEFILES_CALLBACK cbSrcFiles,
117                                PVOID UserContext)
118 {
119     struct process*     pcs;
120     struct module_pair  pair;
121     SOURCEFILE          sf;
122     char*               ptr;
123     
124     if (!cbSrcFiles) return FALSE;
125     pcs = process_find_by_handle(hProcess);
126     if (!pcs) return FALSE;
127          
128     if (ModBase)
129     {
130         pair.requested = module_find_by_addr(pcs, ModBase, DMT_UNKNOWN);
131         if (!module_get_debug(pcs, &pair)) return FALSE;
132     }
133     else
134     {
135         if (Mask[0] == '!')
136         {
137             pair.requested = module_find_by_name(pcs, Mask + 1, DMT_UNKNOWN);
138             if (!module_get_debug(pcs, &pair)) return FALSE;
139         }
140         else
141         {
142             FIXME("Unsupported yet (should get info from current context)\n");
143             return FALSE;
144         }
145     }
146     if (!pair.effective->sources) return FALSE;
147     for (ptr = pair.effective->sources; *ptr; ptr += strlen(ptr) + 1)
148     {
149         /* FIXME: not using Mask */
150         sf.ModBase = ModBase;
151         sf.FileName = ptr;
152         if (!cbSrcFiles(&sf, UserContext)) break;
153     }
154
155     return TRUE;
156 }
157
158 /******************************************************************
159  *              SymEnumLines (DBGHELP.@)
160  *
161  */
162 BOOL WINAPI SymEnumLines(HANDLE hProcess, ULONG64 base, PCSTR compiland,
163                          PCSTR srcfile, PSYM_ENUMLINES_CALLBACK cb, PVOID user)
164 {
165     struct process*             pcs;
166     struct module_pair          pair;
167     struct hash_table_iter      hti;
168     struct symt_ht*             sym;
169     regex_t                     re;
170     struct line_info*           dli;
171     void*                       ptr;
172     SRCCODEINFO                 sci;
173     const char*                 file;
174
175     if (!cb) return FALSE;
176     pcs = process_find_by_handle(hProcess);
177     if (!pcs) return FALSE;
178     if (!(dbghelp_options & SYMOPT_LOAD_LINES)) return TRUE;
179     if (regcomp(&re, srcfile, REG_NOSUB))
180     {
181         FIXME("Couldn't compile %s\n", srcfile);
182         SetLastError(ERROR_INVALID_PARAMETER);
183         return FALSE;
184     }
185     if (compiland) FIXME("Unsupported yet (filtering on compiland %s)\n", compiland);
186     pair.requested = module_find_by_addr(pcs, base, DMT_UNKNOWN);
187     if (!module_get_debug(pcs, &pair)) return FALSE;
188
189     sci.SizeOfStruct = sizeof(sci);
190     sci.ModBase      = base;
191
192     hash_table_iter_init(&pair.effective->ht_symbols, &hti, NULL);
193     while ((ptr = hash_table_iter_up(&hti)))
194     {
195         sym = GET_ENTRY(ptr, struct symt_ht, hash_elt);
196         if (sym->symt.tag != SymTagFunction) continue;
197
198         dli = NULL;
199         sci.FileName[0] = '\0';
200         while ((dli = vector_iter_up(&((struct symt_function*)sym)->vlines, dli)))
201         {
202             if (dli->is_source_file)
203             {
204                 file = source_get(pair.effective, dli->u.source_file);
205                 if (regexec(&re, file, 0, NULL, 0) != 0) file = "";
206                 strcpy(sci.FileName, file);
207             }
208             else if (sci.FileName[0])
209             {
210                 sci.Key = dli;
211                 sci.Obj[0] = '\0'; /* FIXME */
212                 sci.LineNumber = dli->line_number;
213                 sci.Address = dli->u.pc_offset;
214                 if (!cb(&sci, user)) break;
215             }
216         }
217     }
218     return TRUE;
219 }
220
221 /******************************************************************
222  *              SymGetSourceFileToken (DBGHELP.@)
223  *
224  */
225 BOOL WINAPI SymGetSourceFileToken(HANDLE hProcess, ULONG64 base,
226                                   PCSTR src, PVOID* token, DWORD* size)
227 {
228     FIXME("%p %s %s %p %p: stub!\n",
229           hProcess, wine_dbgstr_longlong(base), debugstr_a(src), token, size);
230     SetLastError(ERROR_NOT_SUPPORTED);
231     return FALSE;
232 }
233
234 /******************************************************************
235  *              SymGetSourceFileTokenW (DBGHELP.@)
236  *
237  */
238 BOOL WINAPI SymGetSourceFileTokenW(HANDLE hProcess, ULONG64 base,
239                                    PCWSTR src, PVOID* token, DWORD* size)
240 {
241     FIXME("%p %s %s %p %p: stub!\n",
242           hProcess, wine_dbgstr_longlong(base), debugstr_w(src), token, size);
243     SetLastError(ERROR_NOT_SUPPORTED);
244     return FALSE;
245 }