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