riched32: Fix regression in WM_GETTEXTLENGTH on richedit 1.0 emulation.
[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     unsigned    ret;
60     const char* full;
61     char*       tmp = NULL;
62
63     if (!name) return (unsigned)-1;
64     if (!base || *name == '/')
65         full = name;
66     else
67     {
68         unsigned bsz = strlen(base);
69
70         tmp = HeapAlloc(GetProcessHeap(), 0, bsz + 1 + strlen(name) + 1);
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     {
79         int len = strlen(full) + 1;
80         if (module->sources_used + len + 1 > module->sources_alloc)
81         {
82             /* Alloc by block of 256 bytes */
83             module->sources_alloc = (module->sources_used + len + 1 + 255) & ~255;
84             if (!module->sources)
85                 module->sources = HeapAlloc(GetProcessHeap(), 0, module->sources_alloc);
86             else
87                 module->sources = HeapReAlloc(GetProcessHeap(), 0, module->sources,
88                                               module->sources_alloc);
89         }
90         ret = module->sources_used;
91         memcpy(module->sources + module->sources_used, full, len);
92         module->sources_used += len;
93         module->sources[module->sources_used] = '\0';
94     }
95     HeapFree(GetProcessHeap(), 0, tmp);
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 module_pair  pair;
120     SOURCEFILE          sf;
121     char*               ptr;
122     
123     if (!cbSrcFiles) return FALSE;
124     pair.pcs = process_find_by_handle(hProcess);
125     if (!pair.pcs) return FALSE;
126          
127     if (ModBase)
128     {
129         pair.requested = module_find_by_addr(pair.pcs, ModBase, DMT_UNKNOWN);
130         if (!module_get_debug(&pair)) return FALSE;
131     }
132     else
133     {
134         if (Mask[0] == '!')
135         {
136             pair.requested = module_find_by_nameA(pair.pcs, Mask + 1);
137             if (!module_get_debug(&pair)) return FALSE;
138         }
139         else
140         {
141             FIXME("Unsupported yet (should get info from current context)\n");
142             return FALSE;
143         }
144     }
145     if (!pair.effective->sources) return FALSE;
146     for (ptr = pair.effective->sources; *ptr; ptr += strlen(ptr) + 1)
147     {
148         /* FIXME: not using Mask */
149         sf.ModBase = ModBase;
150         sf.FileName = ptr;
151         if (!cbSrcFiles(&sf, UserContext)) break;
152     }
153
154     return TRUE;
155 }
156
157 /******************************************************************
158  *              SymEnumLines (DBGHELP.@)
159  *
160  */
161 BOOL WINAPI SymEnumLines(HANDLE hProcess, ULONG64 base, PCSTR compiland,
162                          PCSTR srcfile, PSYM_ENUMLINES_CALLBACK cb, PVOID user)
163 {
164     struct module_pair          pair;
165     struct hash_table_iter      hti;
166     struct symt_ht*             sym;
167     regex_t                     re;
168     struct line_info*           dli;
169     void*                       ptr;
170     SRCCODEINFO                 sci;
171     const char*                 file;
172
173     if (!cb) return FALSE;
174     if (!(dbghelp_options & SYMOPT_LOAD_LINES)) return TRUE;
175     if (regcomp(&re, srcfile, REG_NOSUB))
176     {
177         FIXME("Couldn't compile %s\n", srcfile);
178         SetLastError(ERROR_INVALID_PARAMETER);
179         return FALSE;
180     }
181     pair.pcs = process_find_by_handle(hProcess);
182     if (!pair.pcs) return FALSE;
183     if (compiland) FIXME("Unsupported yet (filtering on compiland %s)\n", compiland);
184     pair.requested = module_find_by_addr(pair.pcs, base, DMT_UNKNOWN);
185     if (!module_get_debug(&pair)) return FALSE;
186
187     sci.SizeOfStruct = sizeof(sci);
188     sci.ModBase      = base;
189
190     hash_table_iter_init(&pair.effective->ht_symbols, &hti, NULL);
191     while ((ptr = hash_table_iter_up(&hti)))
192     {
193         int    i;
194
195         sym = GET_ENTRY(ptr, struct symt_ht, hash_elt);
196         if (sym->symt.tag != SymTagFunction) continue;
197
198         sci.FileName[0] = '\0';
199         for (i=0; i<vector_length(&((struct symt_function*)sym)->vlines); i++)
200         {
201             dli = vector_at(&((struct symt_function*)sym)->vlines, i);
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 }