1 #include "../libslang.h"
2 #include <elf.h>
3 #include <inttypes.h>
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
ui_entry__read(const char * title,char * bf,size_t size,int width)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 err;
45 }
46
47 struct map_browser {
48 struct ui_browser b;
49 struct map *map;
50 u8 addrlen;
51 };
52
map_browser__write(struct ui_browser * self,void * nd,int row)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);
58 int width;
59
60 ui_browser__set_percent_color(self, 0, current_entry);
61 slsmg_printf("%*" PRIx64 " %*" PRIx64 " %c ",
62 mb->addrlen, sym->start, mb->addrlen, sym->end,
63 sym->binding == STB_GLOBAL ? 'g' :
64 sym->binding == STB_LOCAL ? 'l' : 'w');
65 width = self->width - ((mb->addrlen * 2) + 4);
66 if (width > 0)
67 slsmg_write_nstring(sym->name, width);
68 }
69
70 /* FIXME uber-kludgy, see comment on cmd_report... */
symbol__browser_index(struct symbol * self)71 static u32 *symbol__browser_index(struct symbol *self)
72 {
73 return ((void *)self) - sizeof(struct rb_node) - sizeof(u32);
74 }
75
map_browser__search(struct map_browser * self)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
map_browser__run(struct map_browser * self)102 static int map_browser__run(struct map_browser *self)
103 {
104 int key;
105
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)
109 return -1;
110
111 if (verbose)
112 ui_browser__add_exit_key(&self->b, '/');
113
114 while (1) {
115 key = ui_browser__run(&self->b);
116
117 if (verbose && key == '/')
118 map_browser__search(self);
119 else
120 break;
121 }
122
123 ui_browser__hide(&self->b);
124 return key;
125 }
126
map__browse(struct map * self)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 };
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
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
154 mb.addrlen = snprintf(tmp, sizeof(tmp), "%" PRIx64, maxaddr);
155 return map_browser__run(&mb);
156 }
157