Commit | Line | Data |
---|---|---|
b1b02673 ACM |
1 | #include "../libslang.h" |
2 | #include <elf.h> | |
9486aa38 | 3 | #include <inttypes.h> |
b1b02673 ACM |
4 | #include <sys/ttydefaults.h> |
5 | #include <ctype.h> | |
6 | #include <string.h> | |
7 | #include <linux/bitops.h> | |
8 | #include "../../debug.h" | |
9 | #include "../../symbol.h" | |
10 | #include "../browser.h" | |
11 | #include "../helpline.h" | |
12 | #include "map.h" | |
13 | ||
14 | static int ui_entry__read(const char *title, char *bf, size_t size, int width) | |
15 | { | |
16 | struct newtExitStruct es; | |
17 | newtComponent form, entry; | |
18 | const char *result; | |
19 | int err = -1; | |
20 | ||
21 | newtCenteredWindow(width, 1, title); | |
22 | form = newtForm(NULL, NULL, 0); | |
23 | if (form == NULL) | |
24 | return -1; | |
25 | ||
26 | entry = newtEntry(0, 0, "0x", width, &result, NEWT_FLAG_SCROLL); | |
27 | if (entry == NULL) | |
28 | goto out_free_form; | |
29 | ||
30 | newtFormAddComponent(form, entry); | |
31 | newtFormAddHotKey(form, NEWT_KEY_ENTER); | |
32 | newtFormAddHotKey(form, NEWT_KEY_ESCAPE); | |
33 | newtFormAddHotKey(form, NEWT_KEY_LEFT); | |
34 | newtFormAddHotKey(form, CTRL('c')); | |
35 | newtFormRun(form, &es); | |
36 | ||
37 | if (result != NULL) { | |
38 | strncpy(bf, result, size); | |
39 | err = 0; | |
40 | } | |
41 | out_free_form: | |
42 | newtPopWindow(); | |
43 | newtFormDestroy(form); | |
44 | return 0; | |
45 | } | |
46 | ||
47 | struct map_browser { | |
48 | struct ui_browser b; | |
49 | struct map *map; | |
b1b02673 ACM |
50 | u8 addrlen; |
51 | }; | |
52 | ||
53 | static void map_browser__write(struct ui_browser *self, void *nd, int row) | |
54 | { | |
55 | struct symbol *sym = rb_entry(nd, struct symbol, rb_node); | |
56 | struct map_browser *mb = container_of(self, struct map_browser, b); | |
57 | bool current_entry = ui_browser__is_current_entry(self, row); | |
469917ce | 58 | int width; |
b1b02673 | 59 | |
8f9bbc40 | 60 | ui_browser__set_percent_color(self, 0, current_entry); |
9486aa38 | 61 | slsmg_printf("%*" PRIx64 " %*" PRIx64 " %c ", |
b1b02673 ACM |
62 | mb->addrlen, sym->start, mb->addrlen, sym->end, |
63 | sym->binding == STB_GLOBAL ? 'g' : | |
64 | sym->binding == STB_LOCAL ? 'l' : 'w'); | |
469917ce ACM |
65 | width = self->width - ((mb->addrlen * 2) + 4); |
66 | if (width > 0) | |
67 | slsmg_write_nstring(sym->name, width); | |
b1b02673 ACM |
68 | } |
69 | ||
70 | /* FIXME uber-kludgy, see comment on cmd_report... */ | |
71 | static u32 *symbol__browser_index(struct symbol *self) | |
72 | { | |
73 | return ((void *)self) - sizeof(struct rb_node) - sizeof(u32); | |
74 | } | |
75 | ||
76 | static int map_browser__search(struct map_browser *self) | |
77 | { | |
78 | char target[512]; | |
79 | struct symbol *sym; | |
80 | int err = ui_entry__read("Search by name/addr", target, sizeof(target), 40); | |
81 | ||
82 | if (err) | |
83 | return err; | |
84 | ||
85 | if (target[0] == '0' && tolower(target[1]) == 'x') { | |
86 | u64 addr = strtoull(target, NULL, 16); | |
87 | sym = map__find_symbol(self->map, addr, NULL); | |
88 | } else | |
89 | sym = map__find_symbol_by_name(self->map, target, NULL); | |
90 | ||
91 | if (sym != NULL) { | |
92 | u32 *idx = symbol__browser_index(sym); | |
93 | ||
94 | self->b.top = &sym->rb_node; | |
95 | self->b.index = self->b.top_idx = *idx; | |
96 | } else | |
97 | ui_helpline__fpush("%s not found!", target); | |
98 | ||
99 | return 0; | |
100 | } | |
101 | ||
b50e003d | 102 | static int map_browser__run(struct map_browser *self) |
b1b02673 | 103 | { |
b50e003d ACM |
104 | int key; |
105 | ||
59e8fe32 ACM |
106 | if (ui_browser__show(&self->b, self->map->dso->long_name, |
107 | "Press <- or ESC to exit, %s / to search", | |
108 | verbose ? "" : "restart with -v to use") < 0) | |
b1b02673 ACM |
109 | return -1; |
110 | ||
b1b02673 | 111 | if (verbose) |
4c1c952e | 112 | ui_browser__add_exit_key(&self->b, '/'); |
b1b02673 ACM |
113 | |
114 | while (1) { | |
b50e003d | 115 | key = ui_browser__run(&self->b); |
b1b02673 | 116 | |
b50e003d | 117 | if (verbose && key == '/') |
b1b02673 ACM |
118 | map_browser__search(self); |
119 | else | |
120 | break; | |
121 | } | |
122 | ||
59e8fe32 | 123 | ui_browser__hide(&self->b); |
b50e003d | 124 | return key; |
b1b02673 ACM |
125 | } |
126 | ||
127 | int map__browse(struct map *self) | |
128 | { | |
129 | struct map_browser mb = { | |
130 | .b = { | |
131 | .entries = &self->dso->symbols[self->type], | |
132 | .refresh = ui_browser__rb_tree_refresh, | |
133 | .seek = ui_browser__rb_tree_seek, | |
134 | .write = map_browser__write, | |
135 | }, | |
136 | .map = self, | |
137 | }; | |
b1b02673 ACM |
138 | struct rb_node *nd; |
139 | char tmp[BITS_PER_LONG / 4]; | |
140 | u64 maxaddr = 0; | |
141 | ||
142 | for (nd = rb_first(mb.b.entries); nd; nd = rb_next(nd)) { | |
143 | struct symbol *pos = rb_entry(nd, struct symbol, rb_node); | |
144 | ||
b1b02673 ACM |
145 | if (maxaddr < pos->end) |
146 | maxaddr = pos->end; | |
147 | if (verbose) { | |
148 | u32 *idx = symbol__browser_index(pos); | |
149 | *idx = mb.b.nr_entries; | |
150 | } | |
151 | ++mb.b.nr_entries; | |
152 | } | |
153 | ||
9486aa38 | 154 | mb.addrlen = snprintf(tmp, sizeof(tmp), "%" PRIx64, maxaddr); |
b50e003d | 155 | return map_browser__run(&mb); |
b1b02673 | 156 | } |