1 /* MSPDebug - debugging tool for the eZ430
2  * Copyright (C) 2009, 2010 Daniel Beer
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17  */
18 
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <string.h>
22 #include <ctype.h>
23 #include <stdint.h>
24 #include <regex.h>
25 
26 #include "stab.h"
27 #include "expr.h"
28 #include "binfile.h"
29 #include "util.h"
30 #include "output.h"
31 #include "output_util.h"
32 #include "vector.h"
33 #include "sym.h"
34 #include "reader.h"
35 #include "demangle.h"
36 
cmd_eval(char ** arg)37 int cmd_eval(char **arg)
38 {
39 	address_t addr;
40 	char name[MAX_SYMBOL_LENGTH];
41 
42 	if (expr_eval(*arg, &addr) < 0) {
43 		printc_err("=: can't parse: %s\n", *arg);
44 		return -1;
45 	}
46 
47 	print_address(addr, name, sizeof(name), 0);
48 	printc("0x%05x = %s\n", addr, name);
49 
50 	return 0;
51 }
52 
cmd_sym_load_add(int clear,char ** arg)53 static int cmd_sym_load_add(int clear, char **arg)
54 {
55 	FILE *in;
56 	char * path;
57 
58 	if (clear && prompt_abort(MODIFY_SYMS))
59 		return 0;
60 
61 	path = expand_tilde(*arg);
62 	if (!path)
63 		return -1;
64 
65 	in = fopen(path, "rb");
66 	free(path);
67 	if (!in) {
68 		printc_err("sym: %s: %s\n", *arg, last_error());
69 		return -1;
70 	}
71 
72 	if (clear) {
73 		stab_clear();
74 		unmark_modified(MODIFY_SYMS);
75 	} else {
76 		mark_modified(MODIFY_SYMS);
77 	}
78 
79 	if (binfile_syms(in) < 0) {
80 		fclose(in);
81 		return -1;
82 	}
83 
84 	fclose(in);
85 
86 	return 0;
87 }
88 
savemap_cb(void * user_data,const char * name,address_t value)89 static int savemap_cb(void *user_data, const char *name, address_t value)
90 {
91 	FILE *savemap_out = (FILE *)user_data;
92 
93 	if (fprintf(savemap_out, "%04x t %s\n", value, name) < 0) {
94 		pr_error("sym: can't write to file");
95 		return -1;
96 	}
97 
98 	return 0;
99 }
100 
cmd_sym_savemap(char ** arg)101 static int cmd_sym_savemap(char **arg)
102 {
103 	FILE *savemap_out;
104 	char *fname = get_arg(arg);
105 
106 	if (!fname) {
107 		printc_err("sym: filename required to save map\n");
108 		return -1;
109 	}
110 
111 	savemap_out = fopen(fname, "w");
112 	if (!savemap_out) {
113 		printc_err("sym: couldn't write to %s: %s\n", fname,
114 			last_error());
115 		return -1;
116 	}
117 
118 	if (stab_enum(savemap_cb, savemap_out) < 0) {
119 		fclose(savemap_out);
120 		return -1;
121 	}
122 
123 	if (fclose(savemap_out) < 0) {
124 		printc_err("sym: error on close: %s\n", last_error());
125 		return -1;
126 	}
127 
128 	unmark_modified(MODIFY_SYMS);
129 	return 0;
130 }
131 
print_sym(void * user_data,const char * name,address_t value)132 static int print_sym(void *user_data, const char *name, address_t value)
133 {
134 	(void)user_data;
135 
136 	char demangled[MAX_SYMBOL_LENGTH];
137 	if (demangle(name, demangled, sizeof(demangled)) > 0)
138 		printc("0x%04x: %s (%s)\n", value, name, demangled);
139 	else
140 		printc("0x%04x: %s\n", value, name);
141 
142 	return 0;
143 }
144 
find_sym(void * user_data,const char * name,address_t value)145 static int find_sym(void *user_data, const char *name, address_t value)
146 {
147 	regex_t *find_preg = (regex_t *)user_data;
148 
149 	char demangled[MAX_SYMBOL_LENGTH];
150 	int len = demangle(name, demangled, sizeof(demangled));
151 
152 	if (!regexec(find_preg, name, 0, NULL, 0) ||
153 	    (len > 0 && !regexec(find_preg, demangled, 0, NULL, 0))) {
154 		if (len > 0)
155 			printc("0x%04x: %s (%s)\n", value, name, demangled);
156 		else
157 			printc("0x%04x: %s\n", value, name);
158 	}
159 
160 	return 0;
161 }
162 
cmd_sym_find(char ** arg)163 static int cmd_sym_find(char **arg)
164 {
165 	regex_t find_preg;
166 	char *expr = get_arg(arg);
167 
168 	if (!expr) {
169 		stab_enum(print_sym, NULL);
170 		return 0;
171 	}
172 
173 	if (regcomp(&find_preg, expr, REG_EXTENDED | REG_NOSUB)) {
174 		printc_err("sym: failed to compile: %s\n", expr);
175 		return -1;
176 	}
177 
178 	stab_enum(find_sym, &find_preg);
179 	regfree(&find_preg);
180 	return 0;
181 }
182 
183 struct rename_record {
184 	char    old_name[MAX_SYMBOL_LENGTH];
185 	int     start, end;
186 };
187 
188 struct rename_data {
189 	struct vector   list;
190 	regex_t         preg;
191 };
192 
renames_do(struct rename_data * rename,const char * replace)193 static int renames_do(struct rename_data *rename, const char *replace)
194 {
195 	int i;
196 	int count = 0;
197 
198 	for (i = 0; i < rename->list.size; i++) {
199 		struct rename_record *r =
200 			VECTOR_PTR(rename->list, i, struct rename_record);
201 		char new_name[MAX_SYMBOL_LENGTH];
202 		int len = r->start;
203 		address_t value;
204 
205 		if (len + 1 > sizeof(new_name))
206 			len = sizeof(new_name) - 1;
207 
208 		memcpy(new_name, r->old_name, len);
209 		snprintf(new_name + len,
210 			 sizeof(new_name) - len,
211 			 "%s%s", replace, r->old_name + r->end);
212 
213 		printc("%s -> %s\n", r->old_name, new_name);
214 
215 		if (stab_get(r->old_name, &value) < 0) {
216 			printc_err("sym: warning: "
217 				"symbol missing: %s\n",
218 				r->old_name);
219 		} else {
220 			stab_del(r->old_name);
221 			if (stab_set(new_name, value) < 0) {
222 				printc_err("sym: warning: "
223 					"failed to set new name: %s\n",
224 					new_name);
225 			}
226 		}
227 
228 		count++;
229 	}
230 
231 	printc("%d symbols renamed\n", count);
232 	return 0;
233 }
234 
find_renames(void * user_data,const char * name,address_t value)235 static int find_renames(void *user_data, const char *name, address_t value)
236 {
237 	struct rename_data *rename = (struct rename_data *)user_data;
238 	regmatch_t pmatch;
239 
240 	(void)value;
241 
242 	if (!regexec(&rename->preg, name, 1, &pmatch, 0) &&
243 	    pmatch.rm_so >= 0 && pmatch.rm_eo > pmatch.rm_so) {
244 		struct rename_record r;
245 
246 		strncpy(r.old_name, name, sizeof(r.old_name));
247 		r.old_name[sizeof(r.old_name) - 1] = 0;
248 		r.start = pmatch.rm_so;
249 		r.end = pmatch.rm_eo;
250 
251 		return vector_push(&rename->list, &r, 1);
252 	}
253 
254 	return 0;
255 }
256 
cmd_sym_rename(char ** arg)257 static int cmd_sym_rename(char **arg)
258 {
259 	const char *expr = get_arg(arg);
260 	const char *replace = get_arg(arg);
261 	int ret;
262 	struct rename_data rename;
263 
264 	if (!(expr && replace)) {
265 		printc_err("sym: expected pattern and replacement\n");
266 		return -1;
267 	}
268 
269 	if (regcomp(&rename.preg, expr, REG_EXTENDED)) {
270 		printc_err("sym: failed to compile: %s\n", expr);
271 		return -1;
272 	}
273 
274 	vector_init(&rename.list, sizeof(struct rename_record));
275 
276 	if (stab_enum(find_renames, &rename) < 0) {
277 		printc_err("sym: rename failed\n");
278 		regfree(&rename.preg);
279 		vector_destroy(&rename.list);
280 		return -1;
281 	}
282 
283 	regfree(&rename.preg);
284 	ret = renames_do(&rename, replace);
285 	vector_destroy(&rename.list);
286 
287 	if (ret > 0)
288 		mark_modified(MODIFY_SYMS);
289 
290 	return ret >= 0 ? 0 : -1;
291 }
292 
cmd_sym_del(char ** arg)293 static int cmd_sym_del(char **arg)
294 {
295 	char *name = get_arg(arg);
296 
297 	if (!name) {
298 		printc_err("sym: need a name to delete "
299 			"symbol table entries\n");
300 		return -1;
301 	}
302 
303 	if (stab_del(name) < 0) {
304 		printc_err("sym: can't delete nonexistent symbol: %s\n",
305 			name);
306 		return -1;
307 	}
308 
309 	mark_modified(MODIFY_SYMS);
310 	return 0;
311 }
312 
cmd_sym(char ** arg)313 int cmd_sym(char **arg)
314 {
315 	char *subcmd = get_arg(arg);
316 
317 	if (!subcmd) {
318 		printc_err("sym: need to specify a subcommand "
319 			"(try \"help sym\")\n");
320 		return -1;
321 	}
322 
323 	if (!strcasecmp(subcmd, "clear")) {
324 		if (prompt_abort(MODIFY_SYMS))
325 			return 0;
326 		stab_clear();
327 		unmark_modified(MODIFY_SYMS);
328 		return 0;
329 	}
330 
331 	if (!strcasecmp(subcmd, "set")) {
332 		char *name = get_arg(arg);
333 		char *val_text = get_arg(arg);
334 		address_t value;
335 
336 		if (!(name && val_text)) {
337 			printc_err("sym: need a name and value to set "
338 				"symbol table entries\n");
339 			return -1;
340 		}
341 
342 		if (expr_eval(val_text, &value) < 0) {
343 			printc_err("sym: can't parse value: %s\n",
344 				val_text);
345 			return -1;
346 		}
347 
348 		if (stab_set(name, value) < 0)
349 			return -1;
350 
351 		mark_modified(MODIFY_SYMS);
352 		return 0;
353 	}
354 
355 	if (!strcasecmp(subcmd, "del"))
356 		return cmd_sym_del(arg);
357 	if (!strcasecmp(subcmd, "import"))
358 		return cmd_sym_load_add(1, arg);
359 	if (!strcasecmp(subcmd, "import+"))
360 		return cmd_sym_load_add(0, arg);
361 	if (!strcasecmp(subcmd, "export"))
362 		return cmd_sym_savemap(arg);
363 	if (!strcasecmp(subcmd, "rename"))
364 		return cmd_sym_rename(arg);
365 	if (!strcasecmp(subcmd, "find"))
366 		return cmd_sym_find(arg);
367 
368 	printc_err("sym: unknown subcommand: %s\n", subcmd);
369 	return -1;
370 }
371