Commit | Line | Data |
---|---|---|
ae55795e | 1 | #include "../../util.h" |
211ef127 ACM |
2 | #include "../browser.h" |
3 | #include "../helpline.h" | |
4 | #include "../libslang.h" | |
ae55795e ACM |
5 | #include "../ui.h" |
6 | #include "../util.h" | |
78f7defe | 7 | #include "../../annotate.h" |
211ef127 ACM |
8 | #include "../../hist.h" |
9 | #include "../../sort.h" | |
10 | #include "../../symbol.h" | |
c97cf422 | 11 | #include <pthread.h> |
cf958003 | 12 | #include <newt.h> |
211ef127 | 13 | |
92221162 ACM |
14 | struct annotate_browser { |
15 | struct ui_browser b; | |
16 | struct rb_root entries; | |
f1e9214c | 17 | struct rb_node *curr_hot; |
34958544 | 18 | struct objdump_line *selection; |
058b4cc9 | 19 | u64 start; |
0361fc25 ACM |
20 | int nr_asm_entries; |
21 | int nr_entries; | |
22 | bool hide_src_code; | |
e235f3f3 | 23 | bool use_offset; |
92221162 ACM |
24 | }; |
25 | ||
26 | struct objdump_line_rb_node { | |
27 | struct rb_node rb_node; | |
28 | double percent; | |
29 | u32 idx; | |
0361fc25 | 30 | int idx_asm; |
92221162 ACM |
31 | }; |
32 | ||
33 | static inline | |
34 | struct objdump_line_rb_node *objdump_line__rb(struct objdump_line *self) | |
35 | { | |
36 | return (struct objdump_line_rb_node *)(self + 1); | |
37 | } | |
38 | ||
0361fc25 ACM |
39 | static bool objdump_line__filter(struct ui_browser *browser, void *entry) |
40 | { | |
41 | struct annotate_browser *ab = container_of(browser, struct annotate_browser, b); | |
42 | ||
43 | if (ab->hide_src_code) { | |
44 | struct objdump_line *ol = list_entry(entry, struct objdump_line, node); | |
45 | return ol->offset == -1; | |
46 | } | |
47 | ||
48 | return false; | |
49 | } | |
50 | ||
211ef127 ACM |
51 | static void annotate_browser__write(struct ui_browser *self, void *entry, int row) |
52 | { | |
34958544 | 53 | struct annotate_browser *ab = container_of(self, struct annotate_browser, b); |
0361fc25 | 54 | struct objdump_line *ol = list_entry(entry, struct objdump_line, node); |
211ef127 | 55 | bool current_entry = ui_browser__is_current_entry(self, row); |
058b4cc9 ACM |
56 | bool change_color = (!ab->hide_src_code && |
57 | (!current_entry || (self->use_navkeypressed && | |
58 | !self->navkeypressed))); | |
211ef127 ACM |
59 | int width = self->width; |
60 | ||
61 | if (ol->offset != -1) { | |
92221162 | 62 | struct objdump_line_rb_node *olrb = objdump_line__rb(ol); |
8f9bbc40 | 63 | ui_browser__set_percent_color(self, olrb->percent, current_entry); |
92221162 | 64 | slsmg_printf(" %7.2f ", olrb->percent); |
92221162 | 65 | } else { |
8f9bbc40 | 66 | ui_browser__set_percent_color(self, 0, current_entry); |
92221162 ACM |
67 | slsmg_write_nstring(" ", 9); |
68 | } | |
69 | ||
70 | SLsmg_write_char(':'); | |
71 | slsmg_write_nstring(" ", 8); | |
c172f742 ACM |
72 | |
73 | /* The scroll bar isn't being used */ | |
74 | if (!self->navkeypressed) | |
75 | width += 1; | |
76 | ||
058b4cc9 ACM |
77 | if (ol->offset != -1 && change_color) |
78 | ui_browser__set_color(self, HE_COLORSET_CODE); | |
58e817d9 | 79 | |
92221162 ACM |
80 | if (!*ol->line) |
81 | slsmg_write_nstring(" ", width - 18); | |
058b4cc9 | 82 | else if (ol->offset == -1) |
92221162 | 83 | slsmg_write_nstring(ol->line, width - 18); |
058b4cc9 ACM |
84 | else { |
85 | char bf[64]; | |
e235f3f3 ACM |
86 | u64 addr = ol->offset; |
87 | int printed, color = -1; | |
058b4cc9 | 88 | |
e235f3f3 ACM |
89 | if (!ab->use_offset) |
90 | addr += ab->start; | |
91 | ||
92 | printed = scnprintf(bf, sizeof(bf), " %" PRIx64 ":", addr); | |
058b4cc9 ACM |
93 | if (change_color) |
94 | color = ui_browser__set_color(self, HE_COLORSET_ADDR); | |
95 | slsmg_write_nstring(bf, printed); | |
96 | if (change_color) | |
97 | ui_browser__set_color(self, color); | |
98 | slsmg_write_nstring(ol->line, width - 18 - printed); | |
99 | } | |
b99976e2 | 100 | |
58e817d9 | 101 | if (current_entry) |
34958544 | 102 | ab->selection = ol; |
92221162 ACM |
103 | } |
104 | ||
105 | static double objdump_line__calc_percent(struct objdump_line *self, | |
2f525d01 | 106 | struct symbol *sym, int evidx) |
92221162 ACM |
107 | { |
108 | double percent = 0.0; | |
109 | ||
110 | if (self->offset != -1) { | |
111 | int len = sym->end - sym->start; | |
211ef127 | 112 | unsigned int hits = 0; |
78f7defe | 113 | struct annotation *notes = symbol__annotation(sym); |
ce6f4fab | 114 | struct source_line *src_line = notes->src->lines; |
2f525d01 | 115 | struct sym_hist *h = annotation__histogram(notes, evidx); |
92221162 | 116 | s64 offset = self->offset; |
ce6f4fab | 117 | struct objdump_line *next; |
92221162 | 118 | |
ce6f4fab | 119 | next = objdump__get_next_ip_line(¬es->src->source, self); |
211ef127 ACM |
120 | while (offset < (s64)len && |
121 | (next == NULL || offset < next->offset)) { | |
78f7defe ACM |
122 | if (src_line) { |
123 | percent += src_line[offset].percent; | |
211ef127 | 124 | } else |
78f7defe | 125 | hits += h->addr[offset]; |
211ef127 ACM |
126 | |
127 | ++offset; | |
128 | } | |
78f7defe ACM |
129 | /* |
130 | * If the percentage wasn't already calculated in | |
131 | * symbol__get_source_line, do it now: | |
132 | */ | |
133 | if (src_line == NULL && h->sum) | |
211ef127 | 134 | percent = 100.0 * hits / h->sum; |
211ef127 ACM |
135 | } |
136 | ||
92221162 ACM |
137 | return percent; |
138 | } | |
139 | ||
140 | static void objdump__insert_line(struct rb_root *self, | |
141 | struct objdump_line_rb_node *line) | |
142 | { | |
143 | struct rb_node **p = &self->rb_node; | |
144 | struct rb_node *parent = NULL; | |
145 | struct objdump_line_rb_node *l; | |
146 | ||
147 | while (*p != NULL) { | |
148 | parent = *p; | |
149 | l = rb_entry(parent, struct objdump_line_rb_node, rb_node); | |
150 | if (line->percent < l->percent) | |
151 | p = &(*p)->rb_left; | |
152 | else | |
153 | p = &(*p)->rb_right; | |
154 | } | |
155 | rb_link_node(&line->rb_node, parent, p); | |
156 | rb_insert_color(&line->rb_node, self); | |
211ef127 ACM |
157 | } |
158 | ||
f1e9214c | 159 | static void annotate_browser__set_top(struct annotate_browser *self, |
b0ffb2c4 | 160 | struct objdump_line *pos, u32 idx) |
f1e9214c | 161 | { |
f1e9214c ACM |
162 | unsigned back; |
163 | ||
164 | ui_browser__refresh_dimensions(&self->b); | |
165 | back = self->b.height / 2; | |
b0ffb2c4 | 166 | self->b.top_idx = self->b.index = idx; |
f1e9214c ACM |
167 | |
168 | while (self->b.top_idx != 0 && back != 0) { | |
169 | pos = list_entry(pos->node.prev, struct objdump_line, node); | |
170 | ||
171 | --self->b.top_idx; | |
172 | --back; | |
173 | } | |
174 | ||
175 | self->b.top = pos; | |
b0ffb2c4 ACM |
176 | } |
177 | ||
178 | static void annotate_browser__set_rb_top(struct annotate_browser *browser, | |
179 | struct rb_node *nd) | |
180 | { | |
181 | struct objdump_line_rb_node *rbpos; | |
182 | struct objdump_line *pos; | |
183 | ||
184 | rbpos = rb_entry(nd, struct objdump_line_rb_node, rb_node); | |
185 | pos = ((struct objdump_line *)rbpos) - 1; | |
186 | annotate_browser__set_top(browser, pos, rbpos->idx); | |
187 | browser->curr_hot = nd; | |
f1e9214c ACM |
188 | } |
189 | ||
c97cf422 ACM |
190 | static void annotate_browser__calc_percent(struct annotate_browser *browser, |
191 | int evidx) | |
f1e9214c | 192 | { |
34958544 ACM |
193 | struct map_symbol *ms = browser->b.priv; |
194 | struct symbol *sym = ms->sym; | |
c97cf422 ACM |
195 | struct annotation *notes = symbol__annotation(sym); |
196 | struct objdump_line *pos; | |
197 | ||
198 | browser->entries = RB_ROOT; | |
199 | ||
200 | pthread_mutex_lock(¬es->lock); | |
201 | ||
202 | list_for_each_entry(pos, ¬es->src->source, node) { | |
203 | struct objdump_line_rb_node *rbpos = objdump_line__rb(pos); | |
204 | rbpos->percent = objdump_line__calc_percent(pos, sym, evidx); | |
205 | if (rbpos->percent < 0.01) { | |
206 | RB_CLEAR_NODE(&rbpos->rb_node); | |
207 | continue; | |
208 | } | |
209 | objdump__insert_line(&browser->entries, rbpos); | |
210 | } | |
211 | pthread_mutex_unlock(¬es->lock); | |
212 | ||
213 | browser->curr_hot = rb_last(&browser->entries); | |
214 | } | |
215 | ||
0361fc25 ACM |
216 | static bool annotate_browser__toggle_source(struct annotate_browser *browser) |
217 | { | |
218 | struct objdump_line *ol; | |
219 | struct objdump_line_rb_node *olrb; | |
220 | off_t offset = browser->b.index - browser->b.top_idx; | |
221 | ||
222 | browser->b.seek(&browser->b, offset, SEEK_CUR); | |
223 | ol = list_entry(browser->b.top, struct objdump_line, node); | |
224 | olrb = objdump_line__rb(ol); | |
225 | ||
226 | if (browser->hide_src_code) { | |
227 | if (olrb->idx_asm < offset) | |
228 | offset = olrb->idx; | |
229 | ||
230 | browser->b.nr_entries = browser->nr_entries; | |
231 | browser->hide_src_code = false; | |
232 | browser->b.seek(&browser->b, -offset, SEEK_CUR); | |
233 | browser->b.top_idx = olrb->idx - offset; | |
234 | browser->b.index = olrb->idx; | |
235 | } else { | |
236 | if (olrb->idx_asm < 0) { | |
237 | ui_helpline__puts("Only available for assembly lines."); | |
238 | browser->b.seek(&browser->b, -offset, SEEK_CUR); | |
239 | return false; | |
240 | } | |
241 | ||
242 | if (olrb->idx_asm < offset) | |
243 | offset = olrb->idx_asm; | |
244 | ||
245 | browser->b.nr_entries = browser->nr_asm_entries; | |
246 | browser->hide_src_code = true; | |
247 | browser->b.seek(&browser->b, -offset, SEEK_CUR); | |
248 | browser->b.top_idx = olrb->idx_asm - offset; | |
249 | browser->b.index = olrb->idx_asm; | |
250 | } | |
251 | ||
252 | return true; | |
253 | } | |
254 | ||
60521702 ACM |
255 | static bool annotate_browser__callq(struct annotate_browser *browser, |
256 | int evidx, void (*timer)(void *arg), | |
257 | void *arg, int delay_secs) | |
258 | { | |
259 | struct map_symbol *ms = browser->b.priv; | |
260 | struct symbol *sym = ms->sym; | |
261 | struct annotation *notes; | |
262 | struct symbol *target; | |
263 | char *s = strstr(browser->selection->line, "callq "); | |
264 | u64 ip; | |
265 | ||
266 | if (s == NULL) | |
267 | return false; | |
268 | ||
269 | s = strchr(s, ' '); | |
270 | if (s++ == NULL) { | |
271 | ui_helpline__puts("Invallid callq instruction."); | |
272 | return true; | |
273 | } | |
274 | ||
275 | ip = strtoull(s, NULL, 16); | |
276 | ip = ms->map->map_ip(ms->map, ip); | |
277 | target = map__find_symbol(ms->map, ip, NULL); | |
278 | if (target == NULL) { | |
279 | ui_helpline__puts("The called function was not found."); | |
280 | return true; | |
281 | } | |
282 | ||
283 | notes = symbol__annotation(target); | |
284 | pthread_mutex_lock(¬es->lock); | |
285 | ||
286 | if (notes->src == NULL && symbol__alloc_hist(target) < 0) { | |
287 | pthread_mutex_unlock(¬es->lock); | |
288 | ui__warning("Not enough memory for annotating '%s' symbol!\n", | |
289 | target->name); | |
290 | return true; | |
291 | } | |
292 | ||
293 | pthread_mutex_unlock(¬es->lock); | |
294 | symbol__tui_annotate(target, ms->map, evidx, timer, arg, delay_secs); | |
295 | ui_browser__show_title(&browser->b, sym->name); | |
296 | return true; | |
297 | } | |
298 | ||
c97cf422 | 299 | static int annotate_browser__run(struct annotate_browser *self, int evidx, |
d04b35f8 | 300 | void(*timer)(void *arg), |
34958544 | 301 | void *arg, int delay_secs) |
c97cf422 ACM |
302 | { |
303 | struct rb_node *nd = NULL; | |
34958544 ACM |
304 | struct map_symbol *ms = self->b.priv; |
305 | struct symbol *sym = ms->sym; | |
824ac0e9 NK |
306 | const char *help = "<-/ESC: Exit, TAB/shift+TAB: Cycle hot lines, " |
307 | "H: Go to hottest line, ->/ENTER: Line action, " | |
e235f3f3 | 308 | "O: Toggle offset view, " |
824ac0e9 | 309 | "S: Toggle source code view"; |
b50e003d | 310 | int key; |
f1e9214c | 311 | |
0361fc25 | 312 | if (ui_browser__show(&self->b, sym->name, help) < 0) |
f1e9214c | 313 | return -1; |
c97cf422 | 314 | |
c97cf422 ACM |
315 | annotate_browser__calc_percent(self, evidx); |
316 | ||
317 | if (self->curr_hot) | |
b0ffb2c4 | 318 | annotate_browser__set_rb_top(self, self->curr_hot); |
f1e9214c ACM |
319 | |
320 | nd = self->curr_hot; | |
c97cf422 | 321 | |
f1e9214c | 322 | while (1) { |
3af6e338 | 323 | key = ui_browser__run(&self->b, delay_secs); |
f1e9214c | 324 | |
81cce8de | 325 | if (delay_secs != 0) { |
c97cf422 ACM |
326 | annotate_browser__calc_percent(self, evidx); |
327 | /* | |
328 | * Current line focus got out of the list of most active | |
329 | * lines, NULL it so that if TAB|UNTAB is pressed, we | |
330 | * move to curr_hot (current hottest line). | |
331 | */ | |
332 | if (nd != NULL && RB_EMPTY_NODE(nd)) | |
333 | nd = NULL; | |
334 | } | |
335 | ||
b50e003d | 336 | switch (key) { |
cf958003 | 337 | case K_TIMER: |
81cce8de ACM |
338 | if (timer != NULL) |
339 | timer(arg); | |
340 | ||
341 | if (delay_secs != 0) | |
c97cf422 ACM |
342 | symbol__annotate_decay_histogram(sym, evidx); |
343 | continue; | |
cf958003 | 344 | case K_TAB: |
c97cf422 ACM |
345 | if (nd != NULL) { |
346 | nd = rb_prev(nd); | |
347 | if (nd == NULL) | |
348 | nd = rb_last(&self->entries); | |
349 | } else | |
350 | nd = self->curr_hot; | |
f1e9214c | 351 | break; |
cf958003 | 352 | case K_UNTAB: |
c97cf422 ACM |
353 | if (nd != NULL) |
354 | nd = rb_next(nd); | |
355 | if (nd == NULL) | |
356 | nd = rb_first(&self->entries); | |
357 | else | |
358 | nd = self->curr_hot; | |
359 | break; | |
360 | case 'H': | |
ef7c5372 | 361 | case 'h': |
c97cf422 | 362 | nd = self->curr_hot; |
f1e9214c | 363 | break; |
0361fc25 | 364 | case 'S': |
ef7c5372 | 365 | case 's': |
0361fc25 ACM |
366 | if (annotate_browser__toggle_source(self)) |
367 | ui_helpline__puts(help); | |
368 | continue; | |
e235f3f3 ACM |
369 | case 'O': |
370 | case 'o': | |
371 | self->use_offset = !self->use_offset; | |
372 | continue; | |
cf958003 ACM |
373 | case K_ENTER: |
374 | case K_RIGHT: | |
60521702 | 375 | if (self->selection == NULL) |
234a5375 | 376 | ui_helpline__puts("Huh? No selection. Report to linux-kernel@vger.kernel.org"); |
60521702 | 377 | else if (self->selection->offset == -1) |
234a5375 | 378 | ui_helpline__puts("Actions are only available for assembly lines."); |
60521702 ACM |
379 | else if (!annotate_browser__callq(self, evidx, timer, arg, delay_secs)) |
380 | ui_helpline__puts("Actions are only available for the 'callq' instruction."); | |
fe46e64c | 381 | continue; |
cf958003 ACM |
382 | case K_LEFT: |
383 | case K_ESC: | |
ed7e5662 ACM |
384 | case 'q': |
385 | case CTRL('c'): | |
f1e9214c | 386 | goto out; |
ed7e5662 ACM |
387 | default: |
388 | continue; | |
f1e9214c | 389 | } |
c97cf422 ACM |
390 | |
391 | if (nd != NULL) | |
b0ffb2c4 | 392 | annotate_browser__set_rb_top(self, nd); |
f1e9214c ACM |
393 | } |
394 | out: | |
59e8fe32 | 395 | ui_browser__hide(&self->b); |
b50e003d | 396 | return key; |
f1e9214c ACM |
397 | } |
398 | ||
d04b35f8 | 399 | int hist_entry__tui_annotate(struct hist_entry *he, int evidx, |
81cce8de | 400 | void(*timer)(void *arg), void *arg, int delay_secs) |
78f7defe | 401 | { |
d04b35f8 | 402 | return symbol__tui_annotate(he->ms.sym, he->ms.map, evidx, |
81cce8de | 403 | timer, arg, delay_secs); |
78f7defe ACM |
404 | } |
405 | ||
c97cf422 | 406 | int symbol__tui_annotate(struct symbol *sym, struct map *map, int evidx, |
d04b35f8 | 407 | void(*timer)(void *arg), void *arg, |
34958544 | 408 | int delay_secs) |
211ef127 | 409 | { |
211ef127 | 410 | struct objdump_line *pos, *n; |
db9a9cbc | 411 | struct annotation *notes; |
34958544 ACM |
412 | struct map_symbol ms = { |
413 | .map = map, | |
414 | .sym = sym, | |
415 | }; | |
92221162 ACM |
416 | struct annotate_browser browser = { |
417 | .b = { | |
92221162 ACM |
418 | .refresh = ui_browser__list_head_refresh, |
419 | .seek = ui_browser__list_head_seek, | |
420 | .write = annotate_browser__write, | |
0361fc25 | 421 | .filter = objdump_line__filter, |
34958544 | 422 | .priv = &ms, |
c172f742 | 423 | .use_navkeypressed = true, |
92221162 | 424 | }, |
211ef127 ACM |
425 | }; |
426 | int ret; | |
427 | ||
78f7defe | 428 | if (sym == NULL) |
211ef127 ACM |
429 | return -1; |
430 | ||
78f7defe | 431 | if (map->dso->annotate_warned) |
211ef127 ACM |
432 | return -1; |
433 | ||
c97cf422 | 434 | if (symbol__annotate(sym, map, sizeof(struct objdump_line_rb_node)) < 0) { |
ae55795e | 435 | ui__error("%s", ui_helpline__last_msg); |
211ef127 ACM |
436 | return -1; |
437 | } | |
438 | ||
439 | ui_helpline__push("Press <- or ESC to exit"); | |
440 | ||
db9a9cbc | 441 | notes = symbol__annotation(sym); |
058b4cc9 | 442 | browser.start = map__rip_2objdump(map, sym->start); |
db9a9cbc | 443 | |
ce6f4fab | 444 | list_for_each_entry(pos, ¬es->src->source, node) { |
c97cf422 | 445 | struct objdump_line_rb_node *rbpos; |
211ef127 | 446 | size_t line_len = strlen(pos->line); |
c97cf422 | 447 | |
92221162 ACM |
448 | if (browser.b.width < line_len) |
449 | browser.b.width = line_len; | |
450 | rbpos = objdump_line__rb(pos); | |
0361fc25 ACM |
451 | rbpos->idx = browser.nr_entries++; |
452 | if (pos->offset != -1) | |
453 | rbpos->idx_asm = browser.nr_asm_entries++; | |
454 | else | |
455 | rbpos->idx_asm = -1; | |
92221162 ACM |
456 | } |
457 | ||
0361fc25 | 458 | browser.b.nr_entries = browser.nr_entries; |
db9a9cbc | 459 | browser.b.entries = ¬es->src->source, |
92221162 | 460 | browser.b.width += 18; /* Percentage */ |
d04b35f8 | 461 | ret = annotate_browser__run(&browser, evidx, timer, arg, delay_secs); |
ce6f4fab | 462 | list_for_each_entry_safe(pos, n, ¬es->src->source, node) { |
211ef127 ACM |
463 | list_del(&pos->node); |
464 | objdump_line__free(pos); | |
465 | } | |
211ef127 ACM |
466 | return ret; |
467 | } |