1aca7a94dSNamhyung Kim #include "../../util/util.h" 2aca7a94dSNamhyung Kim #include "../browser.h" 3aca7a94dSNamhyung Kim #include "../helpline.h" 4aca7a94dSNamhyung Kim #include "../libslang.h" 5aca7a94dSNamhyung Kim #include "../ui.h" 6aca7a94dSNamhyung Kim #include "../util.h" 7aca7a94dSNamhyung Kim #include "../../util/annotate.h" 8aca7a94dSNamhyung Kim #include "../../util/hist.h" 9aca7a94dSNamhyung Kim #include "../../util/sort.h" 10aca7a94dSNamhyung Kim #include "../../util/symbol.h" 11aca7a94dSNamhyung Kim #include <pthread.h> 12aca7a94dSNamhyung Kim #include <newt.h> 13aca7a94dSNamhyung Kim 14b793a401SArnaldo Carvalho de Melo struct browser_disasm_line { 15b793a401SArnaldo Carvalho de Melo struct rb_node rb_node; 16b793a401SArnaldo Carvalho de Melo double percent; 17b793a401SArnaldo Carvalho de Melo u32 idx; 18b793a401SArnaldo Carvalho de Melo int idx_asm; 19b793a401SArnaldo Carvalho de Melo bool jump_target; 20b793a401SArnaldo Carvalho de Melo }; 21b793a401SArnaldo Carvalho de Melo 22aca7a94dSNamhyung Kim struct annotate_browser { 23aca7a94dSNamhyung Kim struct ui_browser b; 24aca7a94dSNamhyung Kim struct rb_root entries; 25aca7a94dSNamhyung Kim struct rb_node *curr_hot; 2629ed6e76SArnaldo Carvalho de Melo struct disasm_line *selection; 27b793a401SArnaldo Carvalho de Melo struct disasm_line **offsets; 28aca7a94dSNamhyung Kim u64 start; 29aca7a94dSNamhyung Kim int nr_asm_entries; 30aca7a94dSNamhyung Kim int nr_entries; 31aca7a94dSNamhyung Kim bool hide_src_code; 32aca7a94dSNamhyung Kim bool use_offset; 33aca7a94dSNamhyung Kim bool searching_backwards; 3461e04b33SArnaldo Carvalho de Melo u8 offset_width; 35aca7a94dSNamhyung Kim char search_bf[128]; 36aca7a94dSNamhyung Kim }; 37aca7a94dSNamhyung Kim 38887c0066SArnaldo Carvalho de Melo static inline struct browser_disasm_line *disasm_line__browser(struct disasm_line *dl) 39aca7a94dSNamhyung Kim { 40887c0066SArnaldo Carvalho de Melo return (struct browser_disasm_line *)(dl + 1); 41aca7a94dSNamhyung Kim } 42aca7a94dSNamhyung Kim 4329ed6e76SArnaldo Carvalho de Melo static bool disasm_line__filter(struct ui_browser *browser, void *entry) 44aca7a94dSNamhyung Kim { 45aca7a94dSNamhyung Kim struct annotate_browser *ab = container_of(browser, struct annotate_browser, b); 46aca7a94dSNamhyung Kim 47aca7a94dSNamhyung Kim if (ab->hide_src_code) { 4829ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl = list_entry(entry, struct disasm_line, node); 4929ed6e76SArnaldo Carvalho de Melo return dl->offset == -1; 50aca7a94dSNamhyung Kim } 51aca7a94dSNamhyung Kim 52aca7a94dSNamhyung Kim return false; 53aca7a94dSNamhyung Kim } 54aca7a94dSNamhyung Kim 55aca7a94dSNamhyung Kim static void annotate_browser__write(struct ui_browser *self, void *entry, int row) 56aca7a94dSNamhyung Kim { 57aca7a94dSNamhyung Kim struct annotate_browser *ab = container_of(self, struct annotate_browser, b); 5829ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl = list_entry(entry, struct disasm_line, node); 59b793a401SArnaldo Carvalho de Melo struct browser_disasm_line *bdl = disasm_line__browser(dl); 60aca7a94dSNamhyung Kim bool current_entry = ui_browser__is_current_entry(self, row); 61aca7a94dSNamhyung Kim bool change_color = (!ab->hide_src_code && 62aca7a94dSNamhyung Kim (!current_entry || (self->use_navkeypressed && 63aca7a94dSNamhyung Kim !self->navkeypressed))); 64aca7a94dSNamhyung Kim int width = self->width; 65aca7a94dSNamhyung Kim 6629ed6e76SArnaldo Carvalho de Melo if (dl->offset != -1) { 67887c0066SArnaldo Carvalho de Melo ui_browser__set_percent_color(self, bdl->percent, current_entry); 68887c0066SArnaldo Carvalho de Melo slsmg_printf(" %7.2f ", bdl->percent); 69aca7a94dSNamhyung Kim } else { 70aca7a94dSNamhyung Kim ui_browser__set_percent_color(self, 0, current_entry); 71aca7a94dSNamhyung Kim slsmg_write_nstring(" ", 9); 72aca7a94dSNamhyung Kim } 73aca7a94dSNamhyung Kim 74aca7a94dSNamhyung Kim SLsmg_write_char(':'); 75aca7a94dSNamhyung Kim slsmg_write_nstring(" ", 8); 76aca7a94dSNamhyung Kim 77aca7a94dSNamhyung Kim /* The scroll bar isn't being used */ 78aca7a94dSNamhyung Kim if (!self->navkeypressed) 79aca7a94dSNamhyung Kim width += 1; 80aca7a94dSNamhyung Kim 8129ed6e76SArnaldo Carvalho de Melo if (dl->offset != -1 && change_color) 82aca7a94dSNamhyung Kim ui_browser__set_color(self, HE_COLORSET_CODE); 83aca7a94dSNamhyung Kim 8429ed6e76SArnaldo Carvalho de Melo if (!*dl->line) 85*8bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(" ", width - 10); 8629ed6e76SArnaldo Carvalho de Melo else if (dl->offset == -1) 87*8bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(dl->line, width - 10); 88aca7a94dSNamhyung Kim else { 89*8bf39cb8SArnaldo Carvalho de Melo char bf[256]; 9029ed6e76SArnaldo Carvalho de Melo u64 addr = dl->offset; 91aca7a94dSNamhyung Kim int printed, color = -1; 92aca7a94dSNamhyung Kim 93aca7a94dSNamhyung Kim if (!ab->use_offset) 94aca7a94dSNamhyung Kim addr += ab->start; 95aca7a94dSNamhyung Kim 9661e04b33SArnaldo Carvalho de Melo if (!ab->use_offset) { 97aca7a94dSNamhyung Kim printed = scnprintf(bf, sizeof(bf), "%" PRIx64 ":", addr); 9861e04b33SArnaldo Carvalho de Melo } else { 9961e04b33SArnaldo Carvalho de Melo if (bdl->jump_target) { 10061e04b33SArnaldo Carvalho de Melo printed = scnprintf(bf, sizeof(bf), "%*" PRIx64 ":", 10161e04b33SArnaldo Carvalho de Melo ab->offset_width, addr); 10261e04b33SArnaldo Carvalho de Melo } else { 10361e04b33SArnaldo Carvalho de Melo printed = scnprintf(bf, sizeof(bf), "%*s ", 10461e04b33SArnaldo Carvalho de Melo ab->offset_width, " "); 10561e04b33SArnaldo Carvalho de Melo } 10661e04b33SArnaldo Carvalho de Melo } 107b793a401SArnaldo Carvalho de Melo 108aca7a94dSNamhyung Kim if (change_color) 109aca7a94dSNamhyung Kim color = ui_browser__set_color(self, HE_COLORSET_ADDR); 110aca7a94dSNamhyung Kim slsmg_write_nstring(bf, printed); 111aca7a94dSNamhyung Kim if (change_color) 112aca7a94dSNamhyung Kim ui_browser__set_color(self, color); 11328548d78SArnaldo Carvalho de Melo if (dl->ins && dl->ins->ops->scnprintf) { 11428548d78SArnaldo Carvalho de Melo dl->ins->ops->scnprintf(dl->ins, bf, sizeof(bf), 11528548d78SArnaldo Carvalho de Melo !ab->use_offset ? dl->operands : NULL, 11628548d78SArnaldo Carvalho de Melo dl->target); 117*8bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(" ", 2); 118*8bf39cb8SArnaldo Carvalho de Melo printed += 2; 119*8bf39cb8SArnaldo Carvalho de Melo } else 120*8bf39cb8SArnaldo Carvalho de Melo scnprintf(bf, sizeof(bf), " %-6.6s %s", dl->name, dl->operands); 12128548d78SArnaldo Carvalho de Melo 122*8bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(bf, width - 10 - printed); 123aca7a94dSNamhyung Kim } 124aca7a94dSNamhyung Kim 125aca7a94dSNamhyung Kim if (current_entry) 12629ed6e76SArnaldo Carvalho de Melo ab->selection = dl; 127aca7a94dSNamhyung Kim } 128aca7a94dSNamhyung Kim 12929ed6e76SArnaldo Carvalho de Melo static double disasm_line__calc_percent(struct disasm_line *dl, struct symbol *sym, int evidx) 130aca7a94dSNamhyung Kim { 131aca7a94dSNamhyung Kim double percent = 0.0; 132aca7a94dSNamhyung Kim 13329ed6e76SArnaldo Carvalho de Melo if (dl->offset != -1) { 134aca7a94dSNamhyung Kim int len = sym->end - sym->start; 135aca7a94dSNamhyung Kim unsigned int hits = 0; 136aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 137aca7a94dSNamhyung Kim struct source_line *src_line = notes->src->lines; 138aca7a94dSNamhyung Kim struct sym_hist *h = annotation__histogram(notes, evidx); 13929ed6e76SArnaldo Carvalho de Melo s64 offset = dl->offset; 14029ed6e76SArnaldo Carvalho de Melo struct disasm_line *next; 141aca7a94dSNamhyung Kim 14229ed6e76SArnaldo Carvalho de Melo next = disasm__get_next_ip_line(¬es->src->source, dl); 143aca7a94dSNamhyung Kim while (offset < (s64)len && 144aca7a94dSNamhyung Kim (next == NULL || offset < next->offset)) { 145aca7a94dSNamhyung Kim if (src_line) { 146aca7a94dSNamhyung Kim percent += src_line[offset].percent; 147aca7a94dSNamhyung Kim } else 148aca7a94dSNamhyung Kim hits += h->addr[offset]; 149aca7a94dSNamhyung Kim 150aca7a94dSNamhyung Kim ++offset; 151aca7a94dSNamhyung Kim } 152aca7a94dSNamhyung Kim /* 153aca7a94dSNamhyung Kim * If the percentage wasn't already calculated in 154aca7a94dSNamhyung Kim * symbol__get_source_line, do it now: 155aca7a94dSNamhyung Kim */ 156aca7a94dSNamhyung Kim if (src_line == NULL && h->sum) 157aca7a94dSNamhyung Kim percent = 100.0 * hits / h->sum; 158aca7a94dSNamhyung Kim } 159aca7a94dSNamhyung Kim 160aca7a94dSNamhyung Kim return percent; 161aca7a94dSNamhyung Kim } 162aca7a94dSNamhyung Kim 163887c0066SArnaldo Carvalho de Melo static void disasm_rb_tree__insert(struct rb_root *root, struct browser_disasm_line *bdl) 164aca7a94dSNamhyung Kim { 16529ed6e76SArnaldo Carvalho de Melo struct rb_node **p = &root->rb_node; 166aca7a94dSNamhyung Kim struct rb_node *parent = NULL; 167887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *l; 168aca7a94dSNamhyung Kim 169aca7a94dSNamhyung Kim while (*p != NULL) { 170aca7a94dSNamhyung Kim parent = *p; 171887c0066SArnaldo Carvalho de Melo l = rb_entry(parent, struct browser_disasm_line, rb_node); 172887c0066SArnaldo Carvalho de Melo if (bdl->percent < l->percent) 173aca7a94dSNamhyung Kim p = &(*p)->rb_left; 174aca7a94dSNamhyung Kim else 175aca7a94dSNamhyung Kim p = &(*p)->rb_right; 176aca7a94dSNamhyung Kim } 177887c0066SArnaldo Carvalho de Melo rb_link_node(&bdl->rb_node, parent, p); 178887c0066SArnaldo Carvalho de Melo rb_insert_color(&bdl->rb_node, root); 179aca7a94dSNamhyung Kim } 180aca7a94dSNamhyung Kim 181aca7a94dSNamhyung Kim static void annotate_browser__set_top(struct annotate_browser *self, 18229ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos, u32 idx) 183aca7a94dSNamhyung Kim { 184aca7a94dSNamhyung Kim unsigned back; 185aca7a94dSNamhyung Kim 186aca7a94dSNamhyung Kim ui_browser__refresh_dimensions(&self->b); 187aca7a94dSNamhyung Kim back = self->b.height / 2; 188aca7a94dSNamhyung Kim self->b.top_idx = self->b.index = idx; 189aca7a94dSNamhyung Kim 190aca7a94dSNamhyung Kim while (self->b.top_idx != 0 && back != 0) { 19129ed6e76SArnaldo Carvalho de Melo pos = list_entry(pos->node.prev, struct disasm_line, node); 192aca7a94dSNamhyung Kim 19329ed6e76SArnaldo Carvalho de Melo if (disasm_line__filter(&self->b, &pos->node)) 194aca7a94dSNamhyung Kim continue; 195aca7a94dSNamhyung Kim 196aca7a94dSNamhyung Kim --self->b.top_idx; 197aca7a94dSNamhyung Kim --back; 198aca7a94dSNamhyung Kim } 199aca7a94dSNamhyung Kim 200aca7a94dSNamhyung Kim self->b.top = pos; 201aca7a94dSNamhyung Kim self->b.navkeypressed = true; 202aca7a94dSNamhyung Kim } 203aca7a94dSNamhyung Kim 204aca7a94dSNamhyung Kim static void annotate_browser__set_rb_top(struct annotate_browser *browser, 205aca7a94dSNamhyung Kim struct rb_node *nd) 206aca7a94dSNamhyung Kim { 207887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bpos; 20829ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos; 209aca7a94dSNamhyung Kim 210887c0066SArnaldo Carvalho de Melo bpos = rb_entry(nd, struct browser_disasm_line, rb_node); 211887c0066SArnaldo Carvalho de Melo pos = ((struct disasm_line *)bpos) - 1; 212887c0066SArnaldo Carvalho de Melo annotate_browser__set_top(browser, pos, bpos->idx); 213aca7a94dSNamhyung Kim browser->curr_hot = nd; 214aca7a94dSNamhyung Kim } 215aca7a94dSNamhyung Kim 216aca7a94dSNamhyung Kim static void annotate_browser__calc_percent(struct annotate_browser *browser, 217aca7a94dSNamhyung Kim int evidx) 218aca7a94dSNamhyung Kim { 219aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 220aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 221aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 22229ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos; 223aca7a94dSNamhyung Kim 224aca7a94dSNamhyung Kim browser->entries = RB_ROOT; 225aca7a94dSNamhyung Kim 226aca7a94dSNamhyung Kim pthread_mutex_lock(¬es->lock); 227aca7a94dSNamhyung Kim 228aca7a94dSNamhyung Kim list_for_each_entry(pos, ¬es->src->source, node) { 229887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bpos = disasm_line__browser(pos); 230887c0066SArnaldo Carvalho de Melo bpos->percent = disasm_line__calc_percent(pos, sym, evidx); 231887c0066SArnaldo Carvalho de Melo if (bpos->percent < 0.01) { 232887c0066SArnaldo Carvalho de Melo RB_CLEAR_NODE(&bpos->rb_node); 233aca7a94dSNamhyung Kim continue; 234aca7a94dSNamhyung Kim } 235887c0066SArnaldo Carvalho de Melo disasm_rb_tree__insert(&browser->entries, bpos); 236aca7a94dSNamhyung Kim } 237aca7a94dSNamhyung Kim pthread_mutex_unlock(¬es->lock); 238aca7a94dSNamhyung Kim 239aca7a94dSNamhyung Kim browser->curr_hot = rb_last(&browser->entries); 240aca7a94dSNamhyung Kim } 241aca7a94dSNamhyung Kim 242aca7a94dSNamhyung Kim static bool annotate_browser__toggle_source(struct annotate_browser *browser) 243aca7a94dSNamhyung Kim { 24429ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl; 245887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bdl; 246aca7a94dSNamhyung Kim off_t offset = browser->b.index - browser->b.top_idx; 247aca7a94dSNamhyung Kim 248aca7a94dSNamhyung Kim browser->b.seek(&browser->b, offset, SEEK_CUR); 24929ed6e76SArnaldo Carvalho de Melo dl = list_entry(browser->b.top, struct disasm_line, node); 250887c0066SArnaldo Carvalho de Melo bdl = disasm_line__browser(dl); 251aca7a94dSNamhyung Kim 252aca7a94dSNamhyung Kim if (browser->hide_src_code) { 253887c0066SArnaldo Carvalho de Melo if (bdl->idx_asm < offset) 254887c0066SArnaldo Carvalho de Melo offset = bdl->idx; 255aca7a94dSNamhyung Kim 256aca7a94dSNamhyung Kim browser->b.nr_entries = browser->nr_entries; 257aca7a94dSNamhyung Kim browser->hide_src_code = false; 258aca7a94dSNamhyung Kim browser->b.seek(&browser->b, -offset, SEEK_CUR); 259887c0066SArnaldo Carvalho de Melo browser->b.top_idx = bdl->idx - offset; 260887c0066SArnaldo Carvalho de Melo browser->b.index = bdl->idx; 261aca7a94dSNamhyung Kim } else { 262887c0066SArnaldo Carvalho de Melo if (bdl->idx_asm < 0) { 263aca7a94dSNamhyung Kim ui_helpline__puts("Only available for assembly lines."); 264aca7a94dSNamhyung Kim browser->b.seek(&browser->b, -offset, SEEK_CUR); 265aca7a94dSNamhyung Kim return false; 266aca7a94dSNamhyung Kim } 267aca7a94dSNamhyung Kim 268887c0066SArnaldo Carvalho de Melo if (bdl->idx_asm < offset) 269887c0066SArnaldo Carvalho de Melo offset = bdl->idx_asm; 270aca7a94dSNamhyung Kim 271aca7a94dSNamhyung Kim browser->b.nr_entries = browser->nr_asm_entries; 272aca7a94dSNamhyung Kim browser->hide_src_code = true; 273aca7a94dSNamhyung Kim browser->b.seek(&browser->b, -offset, SEEK_CUR); 274887c0066SArnaldo Carvalho de Melo browser->b.top_idx = bdl->idx_asm - offset; 275887c0066SArnaldo Carvalho de Melo browser->b.index = bdl->idx_asm; 276aca7a94dSNamhyung Kim } 277aca7a94dSNamhyung Kim 278aca7a94dSNamhyung Kim return true; 279aca7a94dSNamhyung Kim } 280aca7a94dSNamhyung Kim 281aca7a94dSNamhyung Kim static bool annotate_browser__callq(struct annotate_browser *browser, 282aca7a94dSNamhyung Kim int evidx, void (*timer)(void *arg), 283aca7a94dSNamhyung Kim void *arg, int delay_secs) 284aca7a94dSNamhyung Kim { 285aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 286657bcaf5SArnaldo Carvalho de Melo struct disasm_line *dl = browser->selection; 287aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 288aca7a94dSNamhyung Kim struct annotation *notes; 289aca7a94dSNamhyung Kim struct symbol *target; 290aca7a94dSNamhyung Kim u64 ip; 291aca7a94dSNamhyung Kim 292d86b0597SArnaldo Carvalho de Melo if (!ins__is_call(dl->ins)) 293aca7a94dSNamhyung Kim return false; 294aca7a94dSNamhyung Kim 295d86b0597SArnaldo Carvalho de Melo ip = ms->map->map_ip(ms->map, dl->target); 296aca7a94dSNamhyung Kim target = map__find_symbol(ms->map, ip, NULL); 297aca7a94dSNamhyung Kim if (target == NULL) { 298aca7a94dSNamhyung Kim ui_helpline__puts("The called function was not found."); 299aca7a94dSNamhyung Kim return true; 300aca7a94dSNamhyung Kim } 301aca7a94dSNamhyung Kim 302aca7a94dSNamhyung Kim notes = symbol__annotation(target); 303aca7a94dSNamhyung Kim pthread_mutex_lock(¬es->lock); 304aca7a94dSNamhyung Kim 305aca7a94dSNamhyung Kim if (notes->src == NULL && symbol__alloc_hist(target) < 0) { 306aca7a94dSNamhyung Kim pthread_mutex_unlock(¬es->lock); 307aca7a94dSNamhyung Kim ui__warning("Not enough memory for annotating '%s' symbol!\n", 308aca7a94dSNamhyung Kim target->name); 309aca7a94dSNamhyung Kim return true; 310aca7a94dSNamhyung Kim } 311aca7a94dSNamhyung Kim 312aca7a94dSNamhyung Kim pthread_mutex_unlock(¬es->lock); 313aca7a94dSNamhyung Kim symbol__tui_annotate(target, ms->map, evidx, timer, arg, delay_secs); 314aca7a94dSNamhyung Kim ui_browser__show_title(&browser->b, sym->name); 315aca7a94dSNamhyung Kim return true; 316aca7a94dSNamhyung Kim } 317aca7a94dSNamhyung Kim 31829ed6e76SArnaldo Carvalho de Melo static 31929ed6e76SArnaldo Carvalho de Melo struct disasm_line *annotate_browser__find_offset(struct annotate_browser *browser, 320aca7a94dSNamhyung Kim s64 offset, s64 *idx) 321aca7a94dSNamhyung Kim { 322aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 323aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 324aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 32529ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos; 326aca7a94dSNamhyung Kim 327aca7a94dSNamhyung Kim *idx = 0; 328aca7a94dSNamhyung Kim list_for_each_entry(pos, ¬es->src->source, node) { 329aca7a94dSNamhyung Kim if (pos->offset == offset) 330aca7a94dSNamhyung Kim return pos; 33129ed6e76SArnaldo Carvalho de Melo if (!disasm_line__filter(&browser->b, &pos->node)) 332aca7a94dSNamhyung Kim ++*idx; 333aca7a94dSNamhyung Kim } 334aca7a94dSNamhyung Kim 335aca7a94dSNamhyung Kim return NULL; 336aca7a94dSNamhyung Kim } 337aca7a94dSNamhyung Kim 338aca7a94dSNamhyung Kim static bool annotate_browser__jump(struct annotate_browser *browser) 339aca7a94dSNamhyung Kim { 340657bcaf5SArnaldo Carvalho de Melo struct disasm_line *dl = browser->selection; 3414f9d0325SArnaldo Carvalho de Melo s64 idx; 342aca7a94dSNamhyung Kim 343d86b0597SArnaldo Carvalho de Melo if (!ins__is_jump(dl->ins)) 344aca7a94dSNamhyung Kim return false; 345aca7a94dSNamhyung Kim 3464f9d0325SArnaldo Carvalho de Melo dl = annotate_browser__find_offset(browser, dl->target, &idx); 34729ed6e76SArnaldo Carvalho de Melo if (dl == NULL) { 348aca7a94dSNamhyung Kim ui_helpline__puts("Invallid jump offset"); 349aca7a94dSNamhyung Kim return true; 350aca7a94dSNamhyung Kim } 351aca7a94dSNamhyung Kim 35229ed6e76SArnaldo Carvalho de Melo annotate_browser__set_top(browser, dl, idx); 353aca7a94dSNamhyung Kim 354aca7a94dSNamhyung Kim return true; 355aca7a94dSNamhyung Kim } 356aca7a94dSNamhyung Kim 35729ed6e76SArnaldo Carvalho de Melo static 35829ed6e76SArnaldo Carvalho de Melo struct disasm_line *annotate_browser__find_string(struct annotate_browser *browser, 359aca7a94dSNamhyung Kim char *s, s64 *idx) 360aca7a94dSNamhyung Kim { 361aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 362aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 363aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 36429ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos = browser->selection; 365aca7a94dSNamhyung Kim 366aca7a94dSNamhyung Kim *idx = browser->b.index; 367aca7a94dSNamhyung Kim list_for_each_entry_continue(pos, ¬es->src->source, node) { 36829ed6e76SArnaldo Carvalho de Melo if (disasm_line__filter(&browser->b, &pos->node)) 369aca7a94dSNamhyung Kim continue; 370aca7a94dSNamhyung Kim 371aca7a94dSNamhyung Kim ++*idx; 372aca7a94dSNamhyung Kim 373aca7a94dSNamhyung Kim if (pos->line && strstr(pos->line, s) != NULL) 374aca7a94dSNamhyung Kim return pos; 375aca7a94dSNamhyung Kim } 376aca7a94dSNamhyung Kim 377aca7a94dSNamhyung Kim return NULL; 378aca7a94dSNamhyung Kim } 379aca7a94dSNamhyung Kim 380aca7a94dSNamhyung Kim static bool __annotate_browser__search(struct annotate_browser *browser) 381aca7a94dSNamhyung Kim { 38229ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl; 383aca7a94dSNamhyung Kim s64 idx; 384aca7a94dSNamhyung Kim 38529ed6e76SArnaldo Carvalho de Melo dl = annotate_browser__find_string(browser, browser->search_bf, &idx); 38629ed6e76SArnaldo Carvalho de Melo if (dl == NULL) { 387aca7a94dSNamhyung Kim ui_helpline__puts("String not found!"); 388aca7a94dSNamhyung Kim return false; 389aca7a94dSNamhyung Kim } 390aca7a94dSNamhyung Kim 39129ed6e76SArnaldo Carvalho de Melo annotate_browser__set_top(browser, dl, idx); 392aca7a94dSNamhyung Kim browser->searching_backwards = false; 393aca7a94dSNamhyung Kim return true; 394aca7a94dSNamhyung Kim } 395aca7a94dSNamhyung Kim 39629ed6e76SArnaldo Carvalho de Melo static 39729ed6e76SArnaldo Carvalho de Melo struct disasm_line *annotate_browser__find_string_reverse(struct annotate_browser *browser, 398aca7a94dSNamhyung Kim char *s, s64 *idx) 399aca7a94dSNamhyung Kim { 400aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 401aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 402aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 40329ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos = browser->selection; 404aca7a94dSNamhyung Kim 405aca7a94dSNamhyung Kim *idx = browser->b.index; 406aca7a94dSNamhyung Kim list_for_each_entry_continue_reverse(pos, ¬es->src->source, node) { 40729ed6e76SArnaldo Carvalho de Melo if (disasm_line__filter(&browser->b, &pos->node)) 408aca7a94dSNamhyung Kim continue; 409aca7a94dSNamhyung Kim 410aca7a94dSNamhyung Kim --*idx; 411aca7a94dSNamhyung Kim 412aca7a94dSNamhyung Kim if (pos->line && strstr(pos->line, s) != NULL) 413aca7a94dSNamhyung Kim return pos; 414aca7a94dSNamhyung Kim } 415aca7a94dSNamhyung Kim 416aca7a94dSNamhyung Kim return NULL; 417aca7a94dSNamhyung Kim } 418aca7a94dSNamhyung Kim 419aca7a94dSNamhyung Kim static bool __annotate_browser__search_reverse(struct annotate_browser *browser) 420aca7a94dSNamhyung Kim { 42129ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl; 422aca7a94dSNamhyung Kim s64 idx; 423aca7a94dSNamhyung Kim 42429ed6e76SArnaldo Carvalho de Melo dl = annotate_browser__find_string_reverse(browser, browser->search_bf, &idx); 42529ed6e76SArnaldo Carvalho de Melo if (dl == NULL) { 426aca7a94dSNamhyung Kim ui_helpline__puts("String not found!"); 427aca7a94dSNamhyung Kim return false; 428aca7a94dSNamhyung Kim } 429aca7a94dSNamhyung Kim 43029ed6e76SArnaldo Carvalho de Melo annotate_browser__set_top(browser, dl, idx); 431aca7a94dSNamhyung Kim browser->searching_backwards = true; 432aca7a94dSNamhyung Kim return true; 433aca7a94dSNamhyung Kim } 434aca7a94dSNamhyung Kim 435aca7a94dSNamhyung Kim static bool annotate_browser__search_window(struct annotate_browser *browser, 436aca7a94dSNamhyung Kim int delay_secs) 437aca7a94dSNamhyung Kim { 438aca7a94dSNamhyung Kim if (ui_browser__input_window("Search", "String: ", browser->search_bf, 439aca7a94dSNamhyung Kim "ENTER: OK, ESC: Cancel", 440aca7a94dSNamhyung Kim delay_secs * 2) != K_ENTER || 441aca7a94dSNamhyung Kim !*browser->search_bf) 442aca7a94dSNamhyung Kim return false; 443aca7a94dSNamhyung Kim 444aca7a94dSNamhyung Kim return true; 445aca7a94dSNamhyung Kim } 446aca7a94dSNamhyung Kim 447aca7a94dSNamhyung Kim static bool annotate_browser__search(struct annotate_browser *browser, int delay_secs) 448aca7a94dSNamhyung Kim { 449aca7a94dSNamhyung Kim if (annotate_browser__search_window(browser, delay_secs)) 450aca7a94dSNamhyung Kim return __annotate_browser__search(browser); 451aca7a94dSNamhyung Kim 452aca7a94dSNamhyung Kim return false; 453aca7a94dSNamhyung Kim } 454aca7a94dSNamhyung Kim 455aca7a94dSNamhyung Kim static bool annotate_browser__continue_search(struct annotate_browser *browser, 456aca7a94dSNamhyung Kim int delay_secs) 457aca7a94dSNamhyung Kim { 458aca7a94dSNamhyung Kim if (!*browser->search_bf) 459aca7a94dSNamhyung Kim return annotate_browser__search(browser, delay_secs); 460aca7a94dSNamhyung Kim 461aca7a94dSNamhyung Kim return __annotate_browser__search(browser); 462aca7a94dSNamhyung Kim } 463aca7a94dSNamhyung Kim 464aca7a94dSNamhyung Kim static bool annotate_browser__search_reverse(struct annotate_browser *browser, 465aca7a94dSNamhyung Kim int delay_secs) 466aca7a94dSNamhyung Kim { 467aca7a94dSNamhyung Kim if (annotate_browser__search_window(browser, delay_secs)) 468aca7a94dSNamhyung Kim return __annotate_browser__search_reverse(browser); 469aca7a94dSNamhyung Kim 470aca7a94dSNamhyung Kim return false; 471aca7a94dSNamhyung Kim } 472aca7a94dSNamhyung Kim 473aca7a94dSNamhyung Kim static 474aca7a94dSNamhyung Kim bool annotate_browser__continue_search_reverse(struct annotate_browser *browser, 475aca7a94dSNamhyung Kim int delay_secs) 476aca7a94dSNamhyung Kim { 477aca7a94dSNamhyung Kim if (!*browser->search_bf) 478aca7a94dSNamhyung Kim return annotate_browser__search_reverse(browser, delay_secs); 479aca7a94dSNamhyung Kim 480aca7a94dSNamhyung Kim return __annotate_browser__search_reverse(browser); 481aca7a94dSNamhyung Kim } 482aca7a94dSNamhyung Kim 483aca7a94dSNamhyung Kim static int annotate_browser__run(struct annotate_browser *self, int evidx, 484aca7a94dSNamhyung Kim void(*timer)(void *arg), 485aca7a94dSNamhyung Kim void *arg, int delay_secs) 486aca7a94dSNamhyung Kim { 487aca7a94dSNamhyung Kim struct rb_node *nd = NULL; 488aca7a94dSNamhyung Kim struct map_symbol *ms = self->b.priv; 489aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 490aca7a94dSNamhyung Kim const char *help = "<-/ESC: Exit, TAB/shift+TAB: Cycle hot lines, " 491aca7a94dSNamhyung Kim "H: Go to hottest line, ->/ENTER: Line action, " 492aca7a94dSNamhyung Kim "O: Toggle offset view, " 493aca7a94dSNamhyung Kim "S: Toggle source code view"; 494aca7a94dSNamhyung Kim int key; 495aca7a94dSNamhyung Kim 496aca7a94dSNamhyung Kim if (ui_browser__show(&self->b, sym->name, help) < 0) 497aca7a94dSNamhyung Kim return -1; 498aca7a94dSNamhyung Kim 499aca7a94dSNamhyung Kim annotate_browser__calc_percent(self, evidx); 500aca7a94dSNamhyung Kim 501aca7a94dSNamhyung Kim if (self->curr_hot) { 502aca7a94dSNamhyung Kim annotate_browser__set_rb_top(self, self->curr_hot); 503aca7a94dSNamhyung Kim self->b.navkeypressed = false; 504aca7a94dSNamhyung Kim } 505aca7a94dSNamhyung Kim 506aca7a94dSNamhyung Kim nd = self->curr_hot; 507aca7a94dSNamhyung Kim 508aca7a94dSNamhyung Kim while (1) { 509aca7a94dSNamhyung Kim key = ui_browser__run(&self->b, delay_secs); 510aca7a94dSNamhyung Kim 511aca7a94dSNamhyung Kim if (delay_secs != 0) { 512aca7a94dSNamhyung Kim annotate_browser__calc_percent(self, evidx); 513aca7a94dSNamhyung Kim /* 514aca7a94dSNamhyung Kim * Current line focus got out of the list of most active 515aca7a94dSNamhyung Kim * lines, NULL it so that if TAB|UNTAB is pressed, we 516aca7a94dSNamhyung Kim * move to curr_hot (current hottest line). 517aca7a94dSNamhyung Kim */ 518aca7a94dSNamhyung Kim if (nd != NULL && RB_EMPTY_NODE(nd)) 519aca7a94dSNamhyung Kim nd = NULL; 520aca7a94dSNamhyung Kim } 521aca7a94dSNamhyung Kim 522aca7a94dSNamhyung Kim switch (key) { 523aca7a94dSNamhyung Kim case K_TIMER: 524aca7a94dSNamhyung Kim if (timer != NULL) 525aca7a94dSNamhyung Kim timer(arg); 526aca7a94dSNamhyung Kim 527aca7a94dSNamhyung Kim if (delay_secs != 0) 528aca7a94dSNamhyung Kim symbol__annotate_decay_histogram(sym, evidx); 529aca7a94dSNamhyung Kim continue; 530aca7a94dSNamhyung Kim case K_TAB: 531aca7a94dSNamhyung Kim if (nd != NULL) { 532aca7a94dSNamhyung Kim nd = rb_prev(nd); 533aca7a94dSNamhyung Kim if (nd == NULL) 534aca7a94dSNamhyung Kim nd = rb_last(&self->entries); 535aca7a94dSNamhyung Kim } else 536aca7a94dSNamhyung Kim nd = self->curr_hot; 537aca7a94dSNamhyung Kim break; 538aca7a94dSNamhyung Kim case K_UNTAB: 539aca7a94dSNamhyung Kim if (nd != NULL) 540aca7a94dSNamhyung Kim nd = rb_next(nd); 541aca7a94dSNamhyung Kim if (nd == NULL) 542aca7a94dSNamhyung Kim nd = rb_first(&self->entries); 543aca7a94dSNamhyung Kim else 544aca7a94dSNamhyung Kim nd = self->curr_hot; 545aca7a94dSNamhyung Kim break; 546aca7a94dSNamhyung Kim case 'H': 547aca7a94dSNamhyung Kim case 'h': 548aca7a94dSNamhyung Kim nd = self->curr_hot; 549aca7a94dSNamhyung Kim break; 550aca7a94dSNamhyung Kim case 'S': 551aca7a94dSNamhyung Kim case 's': 552aca7a94dSNamhyung Kim if (annotate_browser__toggle_source(self)) 553aca7a94dSNamhyung Kim ui_helpline__puts(help); 554aca7a94dSNamhyung Kim continue; 555aca7a94dSNamhyung Kim case 'O': 556aca7a94dSNamhyung Kim case 'o': 557aca7a94dSNamhyung Kim self->use_offset = !self->use_offset; 558aca7a94dSNamhyung Kim continue; 559aca7a94dSNamhyung Kim case '/': 560aca7a94dSNamhyung Kim if (annotate_browser__search(self, delay_secs)) { 561aca7a94dSNamhyung Kim show_help: 562aca7a94dSNamhyung Kim ui_helpline__puts(help); 563aca7a94dSNamhyung Kim } 564aca7a94dSNamhyung Kim continue; 565aca7a94dSNamhyung Kim case 'n': 566aca7a94dSNamhyung Kim if (self->searching_backwards ? 567aca7a94dSNamhyung Kim annotate_browser__continue_search_reverse(self, delay_secs) : 568aca7a94dSNamhyung Kim annotate_browser__continue_search(self, delay_secs)) 569aca7a94dSNamhyung Kim goto show_help; 570aca7a94dSNamhyung Kim continue; 571aca7a94dSNamhyung Kim case '?': 572aca7a94dSNamhyung Kim if (annotate_browser__search_reverse(self, delay_secs)) 573aca7a94dSNamhyung Kim goto show_help; 574aca7a94dSNamhyung Kim continue; 575aca7a94dSNamhyung Kim case K_ENTER: 576aca7a94dSNamhyung Kim case K_RIGHT: 577aca7a94dSNamhyung Kim if (self->selection == NULL) 578aca7a94dSNamhyung Kim ui_helpline__puts("Huh? No selection. Report to linux-kernel@vger.kernel.org"); 579aca7a94dSNamhyung Kim else if (self->selection->offset == -1) 580aca7a94dSNamhyung Kim ui_helpline__puts("Actions are only available for assembly lines."); 581d86b0597SArnaldo Carvalho de Melo else if (!self->selection->ins || 582d86b0597SArnaldo Carvalho de Melo !(annotate_browser__jump(self) || 583aca7a94dSNamhyung Kim annotate_browser__callq(self, evidx, timer, arg, delay_secs))) 584aca7a94dSNamhyung Kim ui_helpline__puts("Actions are only available for the 'callq' and jump instructions."); 585aca7a94dSNamhyung Kim continue; 586aca7a94dSNamhyung Kim case K_LEFT: 587aca7a94dSNamhyung Kim case K_ESC: 588aca7a94dSNamhyung Kim case 'q': 589aca7a94dSNamhyung Kim case CTRL('c'): 590aca7a94dSNamhyung Kim goto out; 591aca7a94dSNamhyung Kim default: 592aca7a94dSNamhyung Kim continue; 593aca7a94dSNamhyung Kim } 594aca7a94dSNamhyung Kim 595aca7a94dSNamhyung Kim if (nd != NULL) 596aca7a94dSNamhyung Kim annotate_browser__set_rb_top(self, nd); 597aca7a94dSNamhyung Kim } 598aca7a94dSNamhyung Kim out: 599aca7a94dSNamhyung Kim ui_browser__hide(&self->b); 600aca7a94dSNamhyung Kim return key; 601aca7a94dSNamhyung Kim } 602aca7a94dSNamhyung Kim 603aca7a94dSNamhyung Kim int hist_entry__tui_annotate(struct hist_entry *he, int evidx, 604aca7a94dSNamhyung Kim void(*timer)(void *arg), void *arg, int delay_secs) 605aca7a94dSNamhyung Kim { 606aca7a94dSNamhyung Kim return symbol__tui_annotate(he->ms.sym, he->ms.map, evidx, 607aca7a94dSNamhyung Kim timer, arg, delay_secs); 608aca7a94dSNamhyung Kim } 609aca7a94dSNamhyung Kim 610b793a401SArnaldo Carvalho de Melo static void annotate_browser__mark_jump_targets(struct annotate_browser *browser, 611b793a401SArnaldo Carvalho de Melo size_t size) 612b793a401SArnaldo Carvalho de Melo { 613b793a401SArnaldo Carvalho de Melo u64 offset; 614b793a401SArnaldo Carvalho de Melo 615b793a401SArnaldo Carvalho de Melo for (offset = 0; offset < size; ++offset) { 616b793a401SArnaldo Carvalho de Melo struct disasm_line *dl = browser->offsets[offset], *dlt; 617b793a401SArnaldo Carvalho de Melo struct browser_disasm_line *bdlt; 618b793a401SArnaldo Carvalho de Melo 619b793a401SArnaldo Carvalho de Melo if (!dl || !dl->ins || !ins__is_jump(dl->ins)) 620b793a401SArnaldo Carvalho de Melo continue; 621b793a401SArnaldo Carvalho de Melo 622b793a401SArnaldo Carvalho de Melo if (dl->target >= size) { 623b793a401SArnaldo Carvalho de Melo ui__error("jump to after symbol!\n" 624b793a401SArnaldo Carvalho de Melo "size: %zx, jump target: %" PRIx64, 625b793a401SArnaldo Carvalho de Melo size, dl->target); 626b793a401SArnaldo Carvalho de Melo continue; 627b793a401SArnaldo Carvalho de Melo } 628b793a401SArnaldo Carvalho de Melo 629b793a401SArnaldo Carvalho de Melo dlt = browser->offsets[dl->target]; 630b793a401SArnaldo Carvalho de Melo bdlt = disasm_line__browser(dlt); 631b793a401SArnaldo Carvalho de Melo bdlt->jump_target = true; 632b793a401SArnaldo Carvalho de Melo } 633b793a401SArnaldo Carvalho de Melo 634b793a401SArnaldo Carvalho de Melo } 635b793a401SArnaldo Carvalho de Melo 636aca7a94dSNamhyung Kim int symbol__tui_annotate(struct symbol *sym, struct map *map, int evidx, 637aca7a94dSNamhyung Kim void(*timer)(void *arg), void *arg, 638aca7a94dSNamhyung Kim int delay_secs) 639aca7a94dSNamhyung Kim { 64029ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos, *n; 641aca7a94dSNamhyung Kim struct annotation *notes; 642b793a401SArnaldo Carvalho de Melo const size_t size = symbol__size(sym); 643aca7a94dSNamhyung Kim struct map_symbol ms = { 644aca7a94dSNamhyung Kim .map = map, 645aca7a94dSNamhyung Kim .sym = sym, 646aca7a94dSNamhyung Kim }; 647aca7a94dSNamhyung Kim struct annotate_browser browser = { 648aca7a94dSNamhyung Kim .b = { 649aca7a94dSNamhyung Kim .refresh = ui_browser__list_head_refresh, 650aca7a94dSNamhyung Kim .seek = ui_browser__list_head_seek, 651aca7a94dSNamhyung Kim .write = annotate_browser__write, 65229ed6e76SArnaldo Carvalho de Melo .filter = disasm_line__filter, 653aca7a94dSNamhyung Kim .priv = &ms, 654aca7a94dSNamhyung Kim .use_navkeypressed = true, 655aca7a94dSNamhyung Kim }, 656*8bf39cb8SArnaldo Carvalho de Melo .use_offset = true, 657aca7a94dSNamhyung Kim }; 658b793a401SArnaldo Carvalho de Melo int ret = -1; 659aca7a94dSNamhyung Kim 660aca7a94dSNamhyung Kim if (sym == NULL) 661aca7a94dSNamhyung Kim return -1; 662aca7a94dSNamhyung Kim 663aca7a94dSNamhyung Kim if (map->dso->annotate_warned) 664aca7a94dSNamhyung Kim return -1; 665aca7a94dSNamhyung Kim 666b793a401SArnaldo Carvalho de Melo browser.offsets = zalloc(size * sizeof(struct disasm_line *)); 667b793a401SArnaldo Carvalho de Melo if (browser.offsets == NULL) { 668b793a401SArnaldo Carvalho de Melo ui__error("Not enough memory!"); 669b793a401SArnaldo Carvalho de Melo return -1; 670b793a401SArnaldo Carvalho de Melo } 671b793a401SArnaldo Carvalho de Melo 672887c0066SArnaldo Carvalho de Melo if (symbol__annotate(sym, map, sizeof(struct browser_disasm_line)) < 0) { 673aca7a94dSNamhyung Kim ui__error("%s", ui_helpline__last_msg); 674b793a401SArnaldo Carvalho de Melo goto out_free_offsets; 675aca7a94dSNamhyung Kim } 676aca7a94dSNamhyung Kim 677aca7a94dSNamhyung Kim ui_helpline__push("Press <- or ESC to exit"); 678aca7a94dSNamhyung Kim 679aca7a94dSNamhyung Kim notes = symbol__annotation(sym); 680aca7a94dSNamhyung Kim browser.start = map__rip_2objdump(map, sym->start); 681aca7a94dSNamhyung Kim 682aca7a94dSNamhyung Kim list_for_each_entry(pos, ¬es->src->source, node) { 683887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bpos; 684aca7a94dSNamhyung Kim size_t line_len = strlen(pos->line); 685aca7a94dSNamhyung Kim 686aca7a94dSNamhyung Kim if (browser.b.width < line_len) 687aca7a94dSNamhyung Kim browser.b.width = line_len; 688887c0066SArnaldo Carvalho de Melo bpos = disasm_line__browser(pos); 689887c0066SArnaldo Carvalho de Melo bpos->idx = browser.nr_entries++; 690b793a401SArnaldo Carvalho de Melo if (pos->offset != -1) { 691887c0066SArnaldo Carvalho de Melo bpos->idx_asm = browser.nr_asm_entries++; 692b793a401SArnaldo Carvalho de Melo browser.offsets[pos->offset] = pos; 693b793a401SArnaldo Carvalho de Melo } else 694887c0066SArnaldo Carvalho de Melo bpos->idx_asm = -1; 695aca7a94dSNamhyung Kim } 696aca7a94dSNamhyung Kim 697b793a401SArnaldo Carvalho de Melo annotate_browser__mark_jump_targets(&browser, size); 698b793a401SArnaldo Carvalho de Melo 69961e04b33SArnaldo Carvalho de Melo browser.offset_width = hex_width(size); 700aca7a94dSNamhyung Kim browser.b.nr_entries = browser.nr_entries; 701aca7a94dSNamhyung Kim browser.b.entries = ¬es->src->source, 702aca7a94dSNamhyung Kim browser.b.width += 18; /* Percentage */ 703aca7a94dSNamhyung Kim ret = annotate_browser__run(&browser, evidx, timer, arg, delay_secs); 704aca7a94dSNamhyung Kim list_for_each_entry_safe(pos, n, ¬es->src->source, node) { 705aca7a94dSNamhyung Kim list_del(&pos->node); 70629ed6e76SArnaldo Carvalho de Melo disasm_line__free(pos); 707aca7a94dSNamhyung Kim } 708b793a401SArnaldo Carvalho de Melo 709b793a401SArnaldo Carvalho de Melo out_free_offsets: 710b793a401SArnaldo Carvalho de Melo free(browser.offsets); 711aca7a94dSNamhyung Kim return ret; 712aca7a94dSNamhyung Kim } 713