| #include "../../util.h" |
| #include "../browser.h" |
| #include "../helpline.h" |
| #include "../libslang.h" |
| #include "../ui.h" |
| #include "../util.h" |
| #include "../../annotate.h" |
| #include "../../hist.h" |
| #include "../../sort.h" |
| #include "../../symbol.h" |
| #include <pthread.h> |
| #include <newt.h> |
| |
| struct annotate_browser { |
| struct ui_browser b; |
| struct rb_root entries; |
| struct rb_node *curr_hot; |
| struct objdump_line *selection; |
| int nr_asm_entries; |
| int nr_entries; |
| bool hide_src_code; |
| }; |
| |
| struct objdump_line_rb_node { |
| struct rb_node rb_node; |
| double percent; |
| u32 idx; |
| int idx_asm; |
| }; |
| |
| static inline |
| struct objdump_line_rb_node *objdump_line__rb(struct objdump_line *self) |
| { |
| return (struct objdump_line_rb_node *)(self + 1); |
| } |
| |
| static bool objdump_line__filter(struct ui_browser *browser, void *entry) |
| { |
| struct annotate_browser *ab = container_of(browser, struct annotate_browser, b); |
| |
| if (ab->hide_src_code) { |
| struct objdump_line *ol = list_entry(entry, struct objdump_line, node); |
| return ol->offset == -1; |
| } |
| |
| return false; |
| } |
| |
| static void annotate_browser__write(struct ui_browser *self, void *entry, int row) |
| { |
| struct annotate_browser *ab = container_of(self, struct annotate_browser, b); |
| struct objdump_line *ol = list_entry(entry, struct objdump_line, node); |
| bool current_entry = ui_browser__is_current_entry(self, row); |
| int width = self->width; |
| |
| if (ol->offset != -1) { |
| struct objdump_line_rb_node *olrb = objdump_line__rb(ol); |
| ui_browser__set_percent_color(self, olrb->percent, current_entry); |
| slsmg_printf(" %7.2f ", olrb->percent); |
| } else { |
| ui_browser__set_percent_color(self, 0, current_entry); |
| slsmg_write_nstring(" ", 9); |
| } |
| |
| SLsmg_write_char(':'); |
| slsmg_write_nstring(" ", 8); |
| |
| /* The scroll bar isn't being used */ |
| if (!self->navkeypressed) |
| width += 1; |
| |
| if (!ab->hide_src_code && ol->offset != -1) |
| if (!current_entry || (self->use_navkeypressed && |
| !self->navkeypressed)) |
| ui_browser__set_color(self, HE_COLORSET_CODE); |
| |
| if (!*ol->line) |
| slsmg_write_nstring(" ", width - 18); |
| else |
| slsmg_write_nstring(ol->line, width - 18); |
| |
| if (current_entry) |
| ab->selection = ol; |
| } |
| |
| static double objdump_line__calc_percent(struct objdump_line *self, |
| struct symbol *sym, int evidx) |
| { |
| double percent = 0.0; |
| |
| if (self->offset != -1) { |
| int len = sym->end - sym->start; |
| unsigned int hits = 0; |
| struct annotation *notes = symbol__annotation(sym); |
| struct source_line *src_line = notes->src->lines; |
| struct sym_hist *h = annotation__histogram(notes, evidx); |
| s64 offset = self->offset; |
| struct objdump_line *next; |
| |
| next = objdump__get_next_ip_line(¬es->src->source, self); |
| while (offset < (s64)len && |
| (next == NULL || offset < next->offset)) { |
| if (src_line) { |
| percent += src_line[offset].percent; |
| } else |
| hits += h->addr[offset]; |
| |
| ++offset; |
| } |
| /* |
| * If the percentage wasn't already calculated in |
| * symbol__get_source_line, do it now: |
| */ |
| if (src_line == NULL && h->sum) |
| percent = 100.0 * hits / h->sum; |
| } |
| |
| return percent; |
| } |
| |
| static void objdump__insert_line(struct rb_root *self, |
| struct objdump_line_rb_node *line) |
| { |
| struct rb_node **p = &self->rb_node; |
| struct rb_node *parent = NULL; |
| struct objdump_line_rb_node *l; |
| |
| while (*p != NULL) { |
| parent = *p; |
| l = rb_entry(parent, struct objdump_line_rb_node, rb_node); |
| if (line->percent < l->percent) |
| p = &(*p)->rb_left; |
| else |
| p = &(*p)->rb_right; |
| } |
| rb_link_node(&line->rb_node, parent, p); |
| rb_insert_color(&line->rb_node, self); |
| } |
| |
| static void annotate_browser__set_top(struct annotate_browser *self, |
| struct rb_node *nd) |
| { |
| struct objdump_line_rb_node *rbpos; |
| struct objdump_line *pos; |
| unsigned back; |
| |
| ui_browser__refresh_dimensions(&self->b); |
| back = self->b.height / 2; |
| rbpos = rb_entry(nd, struct objdump_line_rb_node, rb_node); |
| pos = ((struct objdump_line *)rbpos) - 1; |
| self->b.top_idx = self->b.index = rbpos->idx; |
| |
| while (self->b.top_idx != 0 && back != 0) { |
| pos = list_entry(pos->node.prev, struct objdump_line, node); |
| |
| --self->b.top_idx; |
| --back; |
| } |
| |
| self->b.top = pos; |
| self->curr_hot = nd; |
| } |
| |
| static void annotate_browser__calc_percent(struct annotate_browser *browser, |
| int evidx) |
| { |
| struct map_symbol *ms = browser->b.priv; |
| struct symbol *sym = ms->sym; |
| struct annotation *notes = symbol__annotation(sym); |
| struct objdump_line *pos; |
| |
| browser->entries = RB_ROOT; |
| |
| pthread_mutex_lock(¬es->lock); |
| |
| list_for_each_entry(pos, ¬es->src->source, node) { |
| struct objdump_line_rb_node *rbpos = objdump_line__rb(pos); |
| rbpos->percent = objdump_line__calc_percent(pos, sym, evidx); |
| if (rbpos->percent < 0.01) { |
| RB_CLEAR_NODE(&rbpos->rb_node); |
| continue; |
| } |
| objdump__insert_line(&browser->entries, rbpos); |
| } |
| pthread_mutex_unlock(¬es->lock); |
| |
| browser->curr_hot = rb_last(&browser->entries); |
| } |
| |
| static bool annotate_browser__toggle_source(struct annotate_browser *browser) |
| { |
| struct objdump_line *ol; |
| struct objdump_line_rb_node *olrb; |
| off_t offset = browser->b.index - browser->b.top_idx; |
| |
| browser->b.seek(&browser->b, offset, SEEK_CUR); |
| ol = list_entry(browser->b.top, struct objdump_line, node); |
| olrb = objdump_line__rb(ol); |
| |
| if (browser->hide_src_code) { |
| if (olrb->idx_asm < offset) |
| offset = olrb->idx; |
| |
| browser->b.nr_entries = browser->nr_entries; |
| browser->hide_src_code = false; |
| browser->b.seek(&browser->b, -offset, SEEK_CUR); |
| browser->b.top_idx = olrb->idx - offset; |
| browser->b.index = olrb->idx; |
| } else { |
| if (olrb->idx_asm < 0) { |
| ui_helpline__puts("Only available for assembly lines."); |
| browser->b.seek(&browser->b, -offset, SEEK_CUR); |
| return false; |
| } |
| |
| if (olrb->idx_asm < offset) |
| offset = olrb->idx_asm; |
| |
| browser->b.nr_entries = browser->nr_asm_entries; |
| browser->hide_src_code = true; |
| browser->b.seek(&browser->b, -offset, SEEK_CUR); |
| browser->b.top_idx = olrb->idx_asm - offset; |
| browser->b.index = olrb->idx_asm; |
| } |
| |
| return true; |
| } |
| |
| static int annotate_browser__run(struct annotate_browser *self, int evidx, |
| void(*timer)(void *arg), |
| void *arg, int delay_secs) |
| { |
| struct rb_node *nd = NULL; |
| struct map_symbol *ms = self->b.priv; |
| struct symbol *sym = ms->sym; |
| const char *help = "<-, ESC: exit, TAB/shift+TAB: cycle hottest lines, " |
| "H: Hottest, -> Line action, S -> Toggle source " |
| "code view"; |
| int key; |
| |
| if (ui_browser__show(&self->b, sym->name, help) < 0) |
| return -1; |
| |
| annotate_browser__calc_percent(self, evidx); |
| |
| if (self->curr_hot) |
| annotate_browser__set_top(self, self->curr_hot); |
| |
| nd = self->curr_hot; |
| |
| while (1) { |
| key = ui_browser__run(&self->b, delay_secs); |
| |
| if (delay_secs != 0) { |
| annotate_browser__calc_percent(self, evidx); |
| /* |
| * Current line focus got out of the list of most active |
| * lines, NULL it so that if TAB|UNTAB is pressed, we |
| * move to curr_hot (current hottest line). |
| */ |
| if (nd != NULL && RB_EMPTY_NODE(nd)) |
| nd = NULL; |
| } |
| |
| switch (key) { |
| case K_TIMER: |
| if (timer != NULL) |
| timer(arg); |
| |
| if (delay_secs != 0) |
| symbol__annotate_decay_histogram(sym, evidx); |
| continue; |
| case K_TAB: |
| if (nd != NULL) { |
| nd = rb_prev(nd); |
| if (nd == NULL) |
| nd = rb_last(&self->entries); |
| } else |
| nd = self->curr_hot; |
| break; |
| case K_UNTAB: |
| if (nd != NULL) |
| nd = rb_next(nd); |
| if (nd == NULL) |
| nd = rb_first(&self->entries); |
| else |
| nd = self->curr_hot; |
| break; |
| case 'H': |
| nd = self->curr_hot; |
| break; |
| case 'S': |
| if (annotate_browser__toggle_source(self)) |
| ui_helpline__puts(help); |
| continue; |
| case K_ENTER: |
| case K_RIGHT: |
| if (self->selection == NULL) { |
| ui_helpline__puts("Huh? No selection. Report to linux-kernel@vger.kernel.org"); |
| continue; |
| } |
| |
| if (self->selection->offset == -1) { |
| ui_helpline__puts("Actions are only available for assembly lines."); |
| continue; |
| } else { |
| char *s = strstr(self->selection->line, "callq "); |
| struct annotation *notes; |
| struct symbol *target; |
| u64 ip; |
| |
| if (s == NULL) { |
| ui_helpline__puts("Actions are only available for the 'callq' instruction."); |
| continue; |
| } |
| |
| s = strchr(s, ' '); |
| if (s++ == NULL) { |
| ui_helpline__puts("Invallid callq instruction."); |
| continue; |
| } |
| |
| ip = strtoull(s, NULL, 16); |
| ip = ms->map->map_ip(ms->map, ip); |
| target = map__find_symbol(ms->map, ip, NULL); |
| if (target == NULL) { |
| ui_helpline__puts("The called function was not found."); |
| continue; |
| } |
| |
| notes = symbol__annotation(target); |
| pthread_mutex_lock(¬es->lock); |
| |
| if (notes->src == NULL && symbol__alloc_hist(target) < 0) { |
| pthread_mutex_unlock(¬es->lock); |
| ui__warning("Not enough memory for annotating '%s' symbol!\n", |
| target->name); |
| continue; |
| } |
| |
| pthread_mutex_unlock(¬es->lock); |
| symbol__tui_annotate(target, ms->map, evidx, |
| timer, arg, delay_secs); |
| } |
| continue; |
| case K_LEFT: |
| case K_ESC: |
| case 'q': |
| case CTRL('c'): |
| goto out; |
| default: |
| continue; |
| } |
| |
| if (nd != NULL) |
| annotate_browser__set_top(self, nd); |
| } |
| out: |
| ui_browser__hide(&self->b); |
| return key; |
| } |
| |
| int hist_entry__tui_annotate(struct hist_entry *he, int evidx, |
| void(*timer)(void *arg), void *arg, int delay_secs) |
| { |
| return symbol__tui_annotate(he->ms.sym, he->ms.map, evidx, |
| timer, arg, delay_secs); |
| } |
| |
| int symbol__tui_annotate(struct symbol *sym, struct map *map, int evidx, |
| void(*timer)(void *arg), void *arg, |
| int delay_secs) |
| { |
| struct objdump_line *pos, *n; |
| struct annotation *notes; |
| struct map_symbol ms = { |
| .map = map, |
| .sym = sym, |
| }; |
| struct annotate_browser browser = { |
| .b = { |
| .refresh = ui_browser__list_head_refresh, |
| .seek = ui_browser__list_head_seek, |
| .write = annotate_browser__write, |
| .filter = objdump_line__filter, |
| .priv = &ms, |
| .use_navkeypressed = true, |
| }, |
| }; |
| int ret; |
| |
| if (sym == NULL) |
| return -1; |
| |
| if (map->dso->annotate_warned) |
| return -1; |
| |
| if (symbol__annotate(sym, map, sizeof(struct objdump_line_rb_node)) < 0) { |
| ui__error("%s", ui_helpline__last_msg); |
| return -1; |
| } |
| |
| ui_helpline__push("Press <- or ESC to exit"); |
| |
| notes = symbol__annotation(sym); |
| |
| list_for_each_entry(pos, ¬es->src->source, node) { |
| struct objdump_line_rb_node *rbpos; |
| size_t line_len = strlen(pos->line); |
| |
| if (browser.b.width < line_len) |
| browser.b.width = line_len; |
| rbpos = objdump_line__rb(pos); |
| rbpos->idx = browser.nr_entries++; |
| if (pos->offset != -1) |
| rbpos->idx_asm = browser.nr_asm_entries++; |
| else |
| rbpos->idx_asm = -1; |
| } |
| |
| browser.b.nr_entries = browser.nr_entries; |
| browser.b.entries = ¬es->src->source, |
| browser.b.width += 18; /* Percentage */ |
| ret = annotate_browser__run(&browser, evidx, timer, arg, delay_secs); |
| list_for_each_entry_safe(pos, n, ¬es->src->source, node) { |
| list_del(&pos->node); |
| objdump_line__free(pos); |
| } |
| return ret; |
| } |