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 74*cf2dacc5SArnaldo Carvalho de Melo SLsmg_set_char_set(1); 75*cf2dacc5SArnaldo Carvalho de Melo SLsmg_write_char(SLSMG_VLINE_CHAR); 76*cf2dacc5SArnaldo Carvalho de Melo SLsmg_set_char_set(0); 77*cf2dacc5SArnaldo Carvalho de Melo SLsmg_write_char(' '); 78aca7a94dSNamhyung Kim 79aca7a94dSNamhyung Kim /* The scroll bar isn't being used */ 80aca7a94dSNamhyung Kim if (!self->navkeypressed) 81aca7a94dSNamhyung Kim width += 1; 82aca7a94dSNamhyung Kim 8329ed6e76SArnaldo Carvalho de Melo if (dl->offset != -1 && change_color) 84aca7a94dSNamhyung Kim ui_browser__set_color(self, HE_COLORSET_CODE); 85aca7a94dSNamhyung Kim 8629ed6e76SArnaldo Carvalho de Melo if (!*dl->line) 878bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(" ", width - 10); 8829ed6e76SArnaldo Carvalho de Melo else if (dl->offset == -1) 898bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(dl->line, width - 10); 90aca7a94dSNamhyung Kim else { 918bf39cb8SArnaldo Carvalho de Melo char bf[256]; 9229ed6e76SArnaldo Carvalho de Melo u64 addr = dl->offset; 93aca7a94dSNamhyung Kim int printed, color = -1; 94aca7a94dSNamhyung Kim 95aca7a94dSNamhyung Kim if (!ab->use_offset) 96aca7a94dSNamhyung Kim addr += ab->start; 97aca7a94dSNamhyung Kim 9861e04b33SArnaldo Carvalho de Melo if (!ab->use_offset) { 99aca7a94dSNamhyung Kim printed = scnprintf(bf, sizeof(bf), "%" PRIx64 ":", addr); 10061e04b33SArnaldo Carvalho de Melo } else { 10161e04b33SArnaldo Carvalho de Melo if (bdl->jump_target) { 10261e04b33SArnaldo Carvalho de Melo printed = scnprintf(bf, sizeof(bf), "%*" PRIx64 ":", 10361e04b33SArnaldo Carvalho de Melo ab->offset_width, addr); 10461e04b33SArnaldo Carvalho de Melo } else { 10561e04b33SArnaldo Carvalho de Melo printed = scnprintf(bf, sizeof(bf), "%*s ", 10661e04b33SArnaldo Carvalho de Melo ab->offset_width, " "); 10761e04b33SArnaldo Carvalho de Melo } 10861e04b33SArnaldo Carvalho de Melo } 109b793a401SArnaldo Carvalho de Melo 110aca7a94dSNamhyung Kim if (change_color) 111aca7a94dSNamhyung Kim color = ui_browser__set_color(self, HE_COLORSET_ADDR); 112aca7a94dSNamhyung Kim slsmg_write_nstring(bf, printed); 113aca7a94dSNamhyung Kim if (change_color) 114aca7a94dSNamhyung Kim ui_browser__set_color(self, color); 11528548d78SArnaldo Carvalho de Melo if (dl->ins && dl->ins->ops->scnprintf) { 11628548d78SArnaldo Carvalho de Melo dl->ins->ops->scnprintf(dl->ins, bf, sizeof(bf), 11728548d78SArnaldo Carvalho de Melo !ab->use_offset ? dl->operands : NULL, 11828548d78SArnaldo Carvalho de Melo dl->target); 1198bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(" ", 2); 1208bf39cb8SArnaldo Carvalho de Melo printed += 2; 1218bf39cb8SArnaldo Carvalho de Melo } else 1228bf39cb8SArnaldo Carvalho de Melo scnprintf(bf, sizeof(bf), " %-6.6s %s", dl->name, dl->operands); 12328548d78SArnaldo Carvalho de Melo 1248bf39cb8SArnaldo Carvalho de Melo slsmg_write_nstring(bf, width - 10 - printed); 125aca7a94dSNamhyung Kim } 126aca7a94dSNamhyung Kim 127aca7a94dSNamhyung Kim if (current_entry) 12829ed6e76SArnaldo Carvalho de Melo ab->selection = dl; 129aca7a94dSNamhyung Kim } 130aca7a94dSNamhyung Kim 13129ed6e76SArnaldo Carvalho de Melo static double disasm_line__calc_percent(struct disasm_line *dl, struct symbol *sym, int evidx) 132aca7a94dSNamhyung Kim { 133aca7a94dSNamhyung Kim double percent = 0.0; 134aca7a94dSNamhyung Kim 13529ed6e76SArnaldo Carvalho de Melo if (dl->offset != -1) { 136aca7a94dSNamhyung Kim int len = sym->end - sym->start; 137aca7a94dSNamhyung Kim unsigned int hits = 0; 138aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 139aca7a94dSNamhyung Kim struct source_line *src_line = notes->src->lines; 140aca7a94dSNamhyung Kim struct sym_hist *h = annotation__histogram(notes, evidx); 14129ed6e76SArnaldo Carvalho de Melo s64 offset = dl->offset; 14229ed6e76SArnaldo Carvalho de Melo struct disasm_line *next; 143aca7a94dSNamhyung Kim 14429ed6e76SArnaldo Carvalho de Melo next = disasm__get_next_ip_line(¬es->src->source, dl); 145aca7a94dSNamhyung Kim while (offset < (s64)len && 146aca7a94dSNamhyung Kim (next == NULL || offset < next->offset)) { 147aca7a94dSNamhyung Kim if (src_line) { 148aca7a94dSNamhyung Kim percent += src_line[offset].percent; 149aca7a94dSNamhyung Kim } else 150aca7a94dSNamhyung Kim hits += h->addr[offset]; 151aca7a94dSNamhyung Kim 152aca7a94dSNamhyung Kim ++offset; 153aca7a94dSNamhyung Kim } 154aca7a94dSNamhyung Kim /* 155aca7a94dSNamhyung Kim * If the percentage wasn't already calculated in 156aca7a94dSNamhyung Kim * symbol__get_source_line, do it now: 157aca7a94dSNamhyung Kim */ 158aca7a94dSNamhyung Kim if (src_line == NULL && h->sum) 159aca7a94dSNamhyung Kim percent = 100.0 * hits / h->sum; 160aca7a94dSNamhyung Kim } 161aca7a94dSNamhyung Kim 162aca7a94dSNamhyung Kim return percent; 163aca7a94dSNamhyung Kim } 164aca7a94dSNamhyung Kim 165887c0066SArnaldo Carvalho de Melo static void disasm_rb_tree__insert(struct rb_root *root, struct browser_disasm_line *bdl) 166aca7a94dSNamhyung Kim { 16729ed6e76SArnaldo Carvalho de Melo struct rb_node **p = &root->rb_node; 168aca7a94dSNamhyung Kim struct rb_node *parent = NULL; 169887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *l; 170aca7a94dSNamhyung Kim 171aca7a94dSNamhyung Kim while (*p != NULL) { 172aca7a94dSNamhyung Kim parent = *p; 173887c0066SArnaldo Carvalho de Melo l = rb_entry(parent, struct browser_disasm_line, rb_node); 174887c0066SArnaldo Carvalho de Melo if (bdl->percent < l->percent) 175aca7a94dSNamhyung Kim p = &(*p)->rb_left; 176aca7a94dSNamhyung Kim else 177aca7a94dSNamhyung Kim p = &(*p)->rb_right; 178aca7a94dSNamhyung Kim } 179887c0066SArnaldo Carvalho de Melo rb_link_node(&bdl->rb_node, parent, p); 180887c0066SArnaldo Carvalho de Melo rb_insert_color(&bdl->rb_node, root); 181aca7a94dSNamhyung Kim } 182aca7a94dSNamhyung Kim 183aca7a94dSNamhyung Kim static void annotate_browser__set_top(struct annotate_browser *self, 18429ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos, u32 idx) 185aca7a94dSNamhyung Kim { 186aca7a94dSNamhyung Kim unsigned back; 187aca7a94dSNamhyung Kim 188aca7a94dSNamhyung Kim ui_browser__refresh_dimensions(&self->b); 189aca7a94dSNamhyung Kim back = self->b.height / 2; 190aca7a94dSNamhyung Kim self->b.top_idx = self->b.index = idx; 191aca7a94dSNamhyung Kim 192aca7a94dSNamhyung Kim while (self->b.top_idx != 0 && back != 0) { 19329ed6e76SArnaldo Carvalho de Melo pos = list_entry(pos->node.prev, struct disasm_line, node); 194aca7a94dSNamhyung Kim 19529ed6e76SArnaldo Carvalho de Melo if (disasm_line__filter(&self->b, &pos->node)) 196aca7a94dSNamhyung Kim continue; 197aca7a94dSNamhyung Kim 198aca7a94dSNamhyung Kim --self->b.top_idx; 199aca7a94dSNamhyung Kim --back; 200aca7a94dSNamhyung Kim } 201aca7a94dSNamhyung Kim 202aca7a94dSNamhyung Kim self->b.top = pos; 203aca7a94dSNamhyung Kim self->b.navkeypressed = true; 204aca7a94dSNamhyung Kim } 205aca7a94dSNamhyung Kim 206aca7a94dSNamhyung Kim static void annotate_browser__set_rb_top(struct annotate_browser *browser, 207aca7a94dSNamhyung Kim struct rb_node *nd) 208aca7a94dSNamhyung Kim { 209887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bpos; 21029ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos; 211aca7a94dSNamhyung Kim 212887c0066SArnaldo Carvalho de Melo bpos = rb_entry(nd, struct browser_disasm_line, rb_node); 213887c0066SArnaldo Carvalho de Melo pos = ((struct disasm_line *)bpos) - 1; 214887c0066SArnaldo Carvalho de Melo annotate_browser__set_top(browser, pos, bpos->idx); 215aca7a94dSNamhyung Kim browser->curr_hot = nd; 216aca7a94dSNamhyung Kim } 217aca7a94dSNamhyung Kim 218aca7a94dSNamhyung Kim static void annotate_browser__calc_percent(struct annotate_browser *browser, 219aca7a94dSNamhyung Kim int evidx) 220aca7a94dSNamhyung Kim { 221aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 222aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 223aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 22429ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos; 225aca7a94dSNamhyung Kim 226aca7a94dSNamhyung Kim browser->entries = RB_ROOT; 227aca7a94dSNamhyung Kim 228aca7a94dSNamhyung Kim pthread_mutex_lock(¬es->lock); 229aca7a94dSNamhyung Kim 230aca7a94dSNamhyung Kim list_for_each_entry(pos, ¬es->src->source, node) { 231887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bpos = disasm_line__browser(pos); 232887c0066SArnaldo Carvalho de Melo bpos->percent = disasm_line__calc_percent(pos, sym, evidx); 233887c0066SArnaldo Carvalho de Melo if (bpos->percent < 0.01) { 234887c0066SArnaldo Carvalho de Melo RB_CLEAR_NODE(&bpos->rb_node); 235aca7a94dSNamhyung Kim continue; 236aca7a94dSNamhyung Kim } 237887c0066SArnaldo Carvalho de Melo disasm_rb_tree__insert(&browser->entries, bpos); 238aca7a94dSNamhyung Kim } 239aca7a94dSNamhyung Kim pthread_mutex_unlock(¬es->lock); 240aca7a94dSNamhyung Kim 241aca7a94dSNamhyung Kim browser->curr_hot = rb_last(&browser->entries); 242aca7a94dSNamhyung Kim } 243aca7a94dSNamhyung Kim 244aca7a94dSNamhyung Kim static bool annotate_browser__toggle_source(struct annotate_browser *browser) 245aca7a94dSNamhyung Kim { 24629ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl; 247887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bdl; 248aca7a94dSNamhyung Kim off_t offset = browser->b.index - browser->b.top_idx; 249aca7a94dSNamhyung Kim 250aca7a94dSNamhyung Kim browser->b.seek(&browser->b, offset, SEEK_CUR); 25129ed6e76SArnaldo Carvalho de Melo dl = list_entry(browser->b.top, struct disasm_line, node); 252887c0066SArnaldo Carvalho de Melo bdl = disasm_line__browser(dl); 253aca7a94dSNamhyung Kim 254aca7a94dSNamhyung Kim if (browser->hide_src_code) { 255887c0066SArnaldo Carvalho de Melo if (bdl->idx_asm < offset) 256887c0066SArnaldo Carvalho de Melo offset = bdl->idx; 257aca7a94dSNamhyung Kim 258aca7a94dSNamhyung Kim browser->b.nr_entries = browser->nr_entries; 259aca7a94dSNamhyung Kim browser->hide_src_code = false; 260aca7a94dSNamhyung Kim browser->b.seek(&browser->b, -offset, SEEK_CUR); 261887c0066SArnaldo Carvalho de Melo browser->b.top_idx = bdl->idx - offset; 262887c0066SArnaldo Carvalho de Melo browser->b.index = bdl->idx; 263aca7a94dSNamhyung Kim } else { 264887c0066SArnaldo Carvalho de Melo if (bdl->idx_asm < 0) { 265aca7a94dSNamhyung Kim ui_helpline__puts("Only available for assembly lines."); 266aca7a94dSNamhyung Kim browser->b.seek(&browser->b, -offset, SEEK_CUR); 267aca7a94dSNamhyung Kim return false; 268aca7a94dSNamhyung Kim } 269aca7a94dSNamhyung Kim 270887c0066SArnaldo Carvalho de Melo if (bdl->idx_asm < offset) 271887c0066SArnaldo Carvalho de Melo offset = bdl->idx_asm; 272aca7a94dSNamhyung Kim 273aca7a94dSNamhyung Kim browser->b.nr_entries = browser->nr_asm_entries; 274aca7a94dSNamhyung Kim browser->hide_src_code = true; 275aca7a94dSNamhyung Kim browser->b.seek(&browser->b, -offset, SEEK_CUR); 276887c0066SArnaldo Carvalho de Melo browser->b.top_idx = bdl->idx_asm - offset; 277887c0066SArnaldo Carvalho de Melo browser->b.index = bdl->idx_asm; 278aca7a94dSNamhyung Kim } 279aca7a94dSNamhyung Kim 280aca7a94dSNamhyung Kim return true; 281aca7a94dSNamhyung Kim } 282aca7a94dSNamhyung Kim 283aca7a94dSNamhyung Kim static bool annotate_browser__callq(struct annotate_browser *browser, 284aca7a94dSNamhyung Kim int evidx, void (*timer)(void *arg), 285aca7a94dSNamhyung Kim void *arg, int delay_secs) 286aca7a94dSNamhyung Kim { 287aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 288657bcaf5SArnaldo Carvalho de Melo struct disasm_line *dl = browser->selection; 289aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 290aca7a94dSNamhyung Kim struct annotation *notes; 291aca7a94dSNamhyung Kim struct symbol *target; 292aca7a94dSNamhyung Kim u64 ip; 293aca7a94dSNamhyung Kim 294d86b0597SArnaldo Carvalho de Melo if (!ins__is_call(dl->ins)) 295aca7a94dSNamhyung Kim return false; 296aca7a94dSNamhyung Kim 297d86b0597SArnaldo Carvalho de Melo ip = ms->map->map_ip(ms->map, dl->target); 298aca7a94dSNamhyung Kim target = map__find_symbol(ms->map, ip, NULL); 299aca7a94dSNamhyung Kim if (target == NULL) { 300aca7a94dSNamhyung Kim ui_helpline__puts("The called function was not found."); 301aca7a94dSNamhyung Kim return true; 302aca7a94dSNamhyung Kim } 303aca7a94dSNamhyung Kim 304aca7a94dSNamhyung Kim notes = symbol__annotation(target); 305aca7a94dSNamhyung Kim pthread_mutex_lock(¬es->lock); 306aca7a94dSNamhyung Kim 307aca7a94dSNamhyung Kim if (notes->src == NULL && symbol__alloc_hist(target) < 0) { 308aca7a94dSNamhyung Kim pthread_mutex_unlock(¬es->lock); 309aca7a94dSNamhyung Kim ui__warning("Not enough memory for annotating '%s' symbol!\n", 310aca7a94dSNamhyung Kim target->name); 311aca7a94dSNamhyung Kim return true; 312aca7a94dSNamhyung Kim } 313aca7a94dSNamhyung Kim 314aca7a94dSNamhyung Kim pthread_mutex_unlock(¬es->lock); 315aca7a94dSNamhyung Kim symbol__tui_annotate(target, ms->map, evidx, timer, arg, delay_secs); 316aca7a94dSNamhyung Kim ui_browser__show_title(&browser->b, sym->name); 317aca7a94dSNamhyung Kim return true; 318aca7a94dSNamhyung Kim } 319aca7a94dSNamhyung Kim 32029ed6e76SArnaldo Carvalho de Melo static 32129ed6e76SArnaldo Carvalho de Melo struct disasm_line *annotate_browser__find_offset(struct annotate_browser *browser, 322aca7a94dSNamhyung Kim s64 offset, s64 *idx) 323aca7a94dSNamhyung Kim { 324aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 325aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 326aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 32729ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos; 328aca7a94dSNamhyung Kim 329aca7a94dSNamhyung Kim *idx = 0; 330aca7a94dSNamhyung Kim list_for_each_entry(pos, ¬es->src->source, node) { 331aca7a94dSNamhyung Kim if (pos->offset == offset) 332aca7a94dSNamhyung Kim return pos; 33329ed6e76SArnaldo Carvalho de Melo if (!disasm_line__filter(&browser->b, &pos->node)) 334aca7a94dSNamhyung Kim ++*idx; 335aca7a94dSNamhyung Kim } 336aca7a94dSNamhyung Kim 337aca7a94dSNamhyung Kim return NULL; 338aca7a94dSNamhyung Kim } 339aca7a94dSNamhyung Kim 340aca7a94dSNamhyung Kim static bool annotate_browser__jump(struct annotate_browser *browser) 341aca7a94dSNamhyung Kim { 342657bcaf5SArnaldo Carvalho de Melo struct disasm_line *dl = browser->selection; 3434f9d0325SArnaldo Carvalho de Melo s64 idx; 344aca7a94dSNamhyung Kim 345d86b0597SArnaldo Carvalho de Melo if (!ins__is_jump(dl->ins)) 346aca7a94dSNamhyung Kim return false; 347aca7a94dSNamhyung Kim 3484f9d0325SArnaldo Carvalho de Melo dl = annotate_browser__find_offset(browser, dl->target, &idx); 34929ed6e76SArnaldo Carvalho de Melo if (dl == NULL) { 350aca7a94dSNamhyung Kim ui_helpline__puts("Invallid jump offset"); 351aca7a94dSNamhyung Kim return true; 352aca7a94dSNamhyung Kim } 353aca7a94dSNamhyung Kim 35429ed6e76SArnaldo Carvalho de Melo annotate_browser__set_top(browser, dl, idx); 355aca7a94dSNamhyung Kim 356aca7a94dSNamhyung Kim return true; 357aca7a94dSNamhyung Kim } 358aca7a94dSNamhyung Kim 35929ed6e76SArnaldo Carvalho de Melo static 36029ed6e76SArnaldo Carvalho de Melo struct disasm_line *annotate_browser__find_string(struct annotate_browser *browser, 361aca7a94dSNamhyung Kim char *s, s64 *idx) 362aca7a94dSNamhyung Kim { 363aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 364aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 365aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 36629ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos = browser->selection; 367aca7a94dSNamhyung Kim 368aca7a94dSNamhyung Kim *idx = browser->b.index; 369aca7a94dSNamhyung Kim list_for_each_entry_continue(pos, ¬es->src->source, node) { 37029ed6e76SArnaldo Carvalho de Melo if (disasm_line__filter(&browser->b, &pos->node)) 371aca7a94dSNamhyung Kim continue; 372aca7a94dSNamhyung Kim 373aca7a94dSNamhyung Kim ++*idx; 374aca7a94dSNamhyung Kim 375aca7a94dSNamhyung Kim if (pos->line && strstr(pos->line, s) != NULL) 376aca7a94dSNamhyung Kim return pos; 377aca7a94dSNamhyung Kim } 378aca7a94dSNamhyung Kim 379aca7a94dSNamhyung Kim return NULL; 380aca7a94dSNamhyung Kim } 381aca7a94dSNamhyung Kim 382aca7a94dSNamhyung Kim static bool __annotate_browser__search(struct annotate_browser *browser) 383aca7a94dSNamhyung Kim { 38429ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl; 385aca7a94dSNamhyung Kim s64 idx; 386aca7a94dSNamhyung Kim 38729ed6e76SArnaldo Carvalho de Melo dl = annotate_browser__find_string(browser, browser->search_bf, &idx); 38829ed6e76SArnaldo Carvalho de Melo if (dl == NULL) { 389aca7a94dSNamhyung Kim ui_helpline__puts("String not found!"); 390aca7a94dSNamhyung Kim return false; 391aca7a94dSNamhyung Kim } 392aca7a94dSNamhyung Kim 39329ed6e76SArnaldo Carvalho de Melo annotate_browser__set_top(browser, dl, idx); 394aca7a94dSNamhyung Kim browser->searching_backwards = false; 395aca7a94dSNamhyung Kim return true; 396aca7a94dSNamhyung Kim } 397aca7a94dSNamhyung Kim 39829ed6e76SArnaldo Carvalho de Melo static 39929ed6e76SArnaldo Carvalho de Melo struct disasm_line *annotate_browser__find_string_reverse(struct annotate_browser *browser, 400aca7a94dSNamhyung Kim char *s, s64 *idx) 401aca7a94dSNamhyung Kim { 402aca7a94dSNamhyung Kim struct map_symbol *ms = browser->b.priv; 403aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 404aca7a94dSNamhyung Kim struct annotation *notes = symbol__annotation(sym); 40529ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos = browser->selection; 406aca7a94dSNamhyung Kim 407aca7a94dSNamhyung Kim *idx = browser->b.index; 408aca7a94dSNamhyung Kim list_for_each_entry_continue_reverse(pos, ¬es->src->source, node) { 40929ed6e76SArnaldo Carvalho de Melo if (disasm_line__filter(&browser->b, &pos->node)) 410aca7a94dSNamhyung Kim continue; 411aca7a94dSNamhyung Kim 412aca7a94dSNamhyung Kim --*idx; 413aca7a94dSNamhyung Kim 414aca7a94dSNamhyung Kim if (pos->line && strstr(pos->line, s) != NULL) 415aca7a94dSNamhyung Kim return pos; 416aca7a94dSNamhyung Kim } 417aca7a94dSNamhyung Kim 418aca7a94dSNamhyung Kim return NULL; 419aca7a94dSNamhyung Kim } 420aca7a94dSNamhyung Kim 421aca7a94dSNamhyung Kim static bool __annotate_browser__search_reverse(struct annotate_browser *browser) 422aca7a94dSNamhyung Kim { 42329ed6e76SArnaldo Carvalho de Melo struct disasm_line *dl; 424aca7a94dSNamhyung Kim s64 idx; 425aca7a94dSNamhyung Kim 42629ed6e76SArnaldo Carvalho de Melo dl = annotate_browser__find_string_reverse(browser, browser->search_bf, &idx); 42729ed6e76SArnaldo Carvalho de Melo if (dl == NULL) { 428aca7a94dSNamhyung Kim ui_helpline__puts("String not found!"); 429aca7a94dSNamhyung Kim return false; 430aca7a94dSNamhyung Kim } 431aca7a94dSNamhyung Kim 43229ed6e76SArnaldo Carvalho de Melo annotate_browser__set_top(browser, dl, idx); 433aca7a94dSNamhyung Kim browser->searching_backwards = true; 434aca7a94dSNamhyung Kim return true; 435aca7a94dSNamhyung Kim } 436aca7a94dSNamhyung Kim 437aca7a94dSNamhyung Kim static bool annotate_browser__search_window(struct annotate_browser *browser, 438aca7a94dSNamhyung Kim int delay_secs) 439aca7a94dSNamhyung Kim { 440aca7a94dSNamhyung Kim if (ui_browser__input_window("Search", "String: ", browser->search_bf, 441aca7a94dSNamhyung Kim "ENTER: OK, ESC: Cancel", 442aca7a94dSNamhyung Kim delay_secs * 2) != K_ENTER || 443aca7a94dSNamhyung Kim !*browser->search_bf) 444aca7a94dSNamhyung Kim return false; 445aca7a94dSNamhyung Kim 446aca7a94dSNamhyung Kim return true; 447aca7a94dSNamhyung Kim } 448aca7a94dSNamhyung Kim 449aca7a94dSNamhyung Kim static bool annotate_browser__search(struct annotate_browser *browser, int delay_secs) 450aca7a94dSNamhyung Kim { 451aca7a94dSNamhyung Kim if (annotate_browser__search_window(browser, delay_secs)) 452aca7a94dSNamhyung Kim return __annotate_browser__search(browser); 453aca7a94dSNamhyung Kim 454aca7a94dSNamhyung Kim return false; 455aca7a94dSNamhyung Kim } 456aca7a94dSNamhyung Kim 457aca7a94dSNamhyung Kim static bool annotate_browser__continue_search(struct annotate_browser *browser, 458aca7a94dSNamhyung Kim int delay_secs) 459aca7a94dSNamhyung Kim { 460aca7a94dSNamhyung Kim if (!*browser->search_bf) 461aca7a94dSNamhyung Kim return annotate_browser__search(browser, delay_secs); 462aca7a94dSNamhyung Kim 463aca7a94dSNamhyung Kim return __annotate_browser__search(browser); 464aca7a94dSNamhyung Kim } 465aca7a94dSNamhyung Kim 466aca7a94dSNamhyung Kim static bool annotate_browser__search_reverse(struct annotate_browser *browser, 467aca7a94dSNamhyung Kim int delay_secs) 468aca7a94dSNamhyung Kim { 469aca7a94dSNamhyung Kim if (annotate_browser__search_window(browser, delay_secs)) 470aca7a94dSNamhyung Kim return __annotate_browser__search_reverse(browser); 471aca7a94dSNamhyung Kim 472aca7a94dSNamhyung Kim return false; 473aca7a94dSNamhyung Kim } 474aca7a94dSNamhyung Kim 475aca7a94dSNamhyung Kim static 476aca7a94dSNamhyung Kim bool annotate_browser__continue_search_reverse(struct annotate_browser *browser, 477aca7a94dSNamhyung Kim int delay_secs) 478aca7a94dSNamhyung Kim { 479aca7a94dSNamhyung Kim if (!*browser->search_bf) 480aca7a94dSNamhyung Kim return annotate_browser__search_reverse(browser, delay_secs); 481aca7a94dSNamhyung Kim 482aca7a94dSNamhyung Kim return __annotate_browser__search_reverse(browser); 483aca7a94dSNamhyung Kim } 484aca7a94dSNamhyung Kim 485aca7a94dSNamhyung Kim static int annotate_browser__run(struct annotate_browser *self, int evidx, 486aca7a94dSNamhyung Kim void(*timer)(void *arg), 487aca7a94dSNamhyung Kim void *arg, int delay_secs) 488aca7a94dSNamhyung Kim { 489aca7a94dSNamhyung Kim struct rb_node *nd = NULL; 490aca7a94dSNamhyung Kim struct map_symbol *ms = self->b.priv; 491aca7a94dSNamhyung Kim struct symbol *sym = ms->sym; 492aca7a94dSNamhyung Kim const char *help = "<-/ESC: Exit, TAB/shift+TAB: Cycle hot lines, " 493aca7a94dSNamhyung Kim "H: Go to hottest line, ->/ENTER: Line action, " 494aca7a94dSNamhyung Kim "O: Toggle offset view, " 495aca7a94dSNamhyung Kim "S: Toggle source code view"; 496aca7a94dSNamhyung Kim int key; 497aca7a94dSNamhyung Kim 498aca7a94dSNamhyung Kim if (ui_browser__show(&self->b, sym->name, help) < 0) 499aca7a94dSNamhyung Kim return -1; 500aca7a94dSNamhyung Kim 501aca7a94dSNamhyung Kim annotate_browser__calc_percent(self, evidx); 502aca7a94dSNamhyung Kim 503aca7a94dSNamhyung Kim if (self->curr_hot) { 504aca7a94dSNamhyung Kim annotate_browser__set_rb_top(self, self->curr_hot); 505aca7a94dSNamhyung Kim self->b.navkeypressed = false; 506aca7a94dSNamhyung Kim } 507aca7a94dSNamhyung Kim 508aca7a94dSNamhyung Kim nd = self->curr_hot; 509aca7a94dSNamhyung Kim 510aca7a94dSNamhyung Kim while (1) { 511aca7a94dSNamhyung Kim key = ui_browser__run(&self->b, delay_secs); 512aca7a94dSNamhyung Kim 513aca7a94dSNamhyung Kim if (delay_secs != 0) { 514aca7a94dSNamhyung Kim annotate_browser__calc_percent(self, evidx); 515aca7a94dSNamhyung Kim /* 516aca7a94dSNamhyung Kim * Current line focus got out of the list of most active 517aca7a94dSNamhyung Kim * lines, NULL it so that if TAB|UNTAB is pressed, we 518aca7a94dSNamhyung Kim * move to curr_hot (current hottest line). 519aca7a94dSNamhyung Kim */ 520aca7a94dSNamhyung Kim if (nd != NULL && RB_EMPTY_NODE(nd)) 521aca7a94dSNamhyung Kim nd = NULL; 522aca7a94dSNamhyung Kim } 523aca7a94dSNamhyung Kim 524aca7a94dSNamhyung Kim switch (key) { 525aca7a94dSNamhyung Kim case K_TIMER: 526aca7a94dSNamhyung Kim if (timer != NULL) 527aca7a94dSNamhyung Kim timer(arg); 528aca7a94dSNamhyung Kim 529aca7a94dSNamhyung Kim if (delay_secs != 0) 530aca7a94dSNamhyung Kim symbol__annotate_decay_histogram(sym, evidx); 531aca7a94dSNamhyung Kim continue; 532aca7a94dSNamhyung Kim case K_TAB: 533aca7a94dSNamhyung Kim if (nd != NULL) { 534aca7a94dSNamhyung Kim nd = rb_prev(nd); 535aca7a94dSNamhyung Kim if (nd == NULL) 536aca7a94dSNamhyung Kim nd = rb_last(&self->entries); 537aca7a94dSNamhyung Kim } else 538aca7a94dSNamhyung Kim nd = self->curr_hot; 539aca7a94dSNamhyung Kim break; 540aca7a94dSNamhyung Kim case K_UNTAB: 541aca7a94dSNamhyung Kim if (nd != NULL) 542aca7a94dSNamhyung Kim nd = rb_next(nd); 543aca7a94dSNamhyung Kim if (nd == NULL) 544aca7a94dSNamhyung Kim nd = rb_first(&self->entries); 545aca7a94dSNamhyung Kim else 546aca7a94dSNamhyung Kim nd = self->curr_hot; 547aca7a94dSNamhyung Kim break; 548aca7a94dSNamhyung Kim case 'H': 549aca7a94dSNamhyung Kim case 'h': 550aca7a94dSNamhyung Kim nd = self->curr_hot; 551aca7a94dSNamhyung Kim break; 552aca7a94dSNamhyung Kim case 'S': 553aca7a94dSNamhyung Kim case 's': 554aca7a94dSNamhyung Kim if (annotate_browser__toggle_source(self)) 555aca7a94dSNamhyung Kim ui_helpline__puts(help); 556aca7a94dSNamhyung Kim continue; 557aca7a94dSNamhyung Kim case 'O': 558aca7a94dSNamhyung Kim case 'o': 559aca7a94dSNamhyung Kim self->use_offset = !self->use_offset; 560aca7a94dSNamhyung Kim continue; 561aca7a94dSNamhyung Kim case '/': 562aca7a94dSNamhyung Kim if (annotate_browser__search(self, delay_secs)) { 563aca7a94dSNamhyung Kim show_help: 564aca7a94dSNamhyung Kim ui_helpline__puts(help); 565aca7a94dSNamhyung Kim } 566aca7a94dSNamhyung Kim continue; 567aca7a94dSNamhyung Kim case 'n': 568aca7a94dSNamhyung Kim if (self->searching_backwards ? 569aca7a94dSNamhyung Kim annotate_browser__continue_search_reverse(self, delay_secs) : 570aca7a94dSNamhyung Kim annotate_browser__continue_search(self, delay_secs)) 571aca7a94dSNamhyung Kim goto show_help; 572aca7a94dSNamhyung Kim continue; 573aca7a94dSNamhyung Kim case '?': 574aca7a94dSNamhyung Kim if (annotate_browser__search_reverse(self, delay_secs)) 575aca7a94dSNamhyung Kim goto show_help; 576aca7a94dSNamhyung Kim continue; 577aca7a94dSNamhyung Kim case K_ENTER: 578aca7a94dSNamhyung Kim case K_RIGHT: 579aca7a94dSNamhyung Kim if (self->selection == NULL) 580aca7a94dSNamhyung Kim ui_helpline__puts("Huh? No selection. Report to linux-kernel@vger.kernel.org"); 581aca7a94dSNamhyung Kim else if (self->selection->offset == -1) 582aca7a94dSNamhyung Kim ui_helpline__puts("Actions are only available for assembly lines."); 583d86b0597SArnaldo Carvalho de Melo else if (!self->selection->ins || 584d86b0597SArnaldo Carvalho de Melo !(annotate_browser__jump(self) || 585aca7a94dSNamhyung Kim annotate_browser__callq(self, evidx, timer, arg, delay_secs))) 586aca7a94dSNamhyung Kim ui_helpline__puts("Actions are only available for the 'callq' and jump instructions."); 587aca7a94dSNamhyung Kim continue; 588aca7a94dSNamhyung Kim case K_LEFT: 589aca7a94dSNamhyung Kim case K_ESC: 590aca7a94dSNamhyung Kim case 'q': 591aca7a94dSNamhyung Kim case CTRL('c'): 592aca7a94dSNamhyung Kim goto out; 593aca7a94dSNamhyung Kim default: 594aca7a94dSNamhyung Kim continue; 595aca7a94dSNamhyung Kim } 596aca7a94dSNamhyung Kim 597aca7a94dSNamhyung Kim if (nd != NULL) 598aca7a94dSNamhyung Kim annotate_browser__set_rb_top(self, nd); 599aca7a94dSNamhyung Kim } 600aca7a94dSNamhyung Kim out: 601aca7a94dSNamhyung Kim ui_browser__hide(&self->b); 602aca7a94dSNamhyung Kim return key; 603aca7a94dSNamhyung Kim } 604aca7a94dSNamhyung Kim 605aca7a94dSNamhyung Kim int hist_entry__tui_annotate(struct hist_entry *he, int evidx, 606aca7a94dSNamhyung Kim void(*timer)(void *arg), void *arg, int delay_secs) 607aca7a94dSNamhyung Kim { 608aca7a94dSNamhyung Kim return symbol__tui_annotate(he->ms.sym, he->ms.map, evidx, 609aca7a94dSNamhyung Kim timer, arg, delay_secs); 610aca7a94dSNamhyung Kim } 611aca7a94dSNamhyung Kim 612b793a401SArnaldo Carvalho de Melo static void annotate_browser__mark_jump_targets(struct annotate_browser *browser, 613b793a401SArnaldo Carvalho de Melo size_t size) 614b793a401SArnaldo Carvalho de Melo { 615b793a401SArnaldo Carvalho de Melo u64 offset; 616b793a401SArnaldo Carvalho de Melo 617b793a401SArnaldo Carvalho de Melo for (offset = 0; offset < size; ++offset) { 618b793a401SArnaldo Carvalho de Melo struct disasm_line *dl = browser->offsets[offset], *dlt; 619b793a401SArnaldo Carvalho de Melo struct browser_disasm_line *bdlt; 620b793a401SArnaldo Carvalho de Melo 621b793a401SArnaldo Carvalho de Melo if (!dl || !dl->ins || !ins__is_jump(dl->ins)) 622b793a401SArnaldo Carvalho de Melo continue; 623b793a401SArnaldo Carvalho de Melo 624b793a401SArnaldo Carvalho de Melo if (dl->target >= size) { 625b793a401SArnaldo Carvalho de Melo ui__error("jump to after symbol!\n" 626b793a401SArnaldo Carvalho de Melo "size: %zx, jump target: %" PRIx64, 627b793a401SArnaldo Carvalho de Melo size, dl->target); 628b793a401SArnaldo Carvalho de Melo continue; 629b793a401SArnaldo Carvalho de Melo } 630b793a401SArnaldo Carvalho de Melo 631b793a401SArnaldo Carvalho de Melo dlt = browser->offsets[dl->target]; 632b793a401SArnaldo Carvalho de Melo bdlt = disasm_line__browser(dlt); 633b793a401SArnaldo Carvalho de Melo bdlt->jump_target = true; 634b793a401SArnaldo Carvalho de Melo } 635b793a401SArnaldo Carvalho de Melo 636b793a401SArnaldo Carvalho de Melo } 637b793a401SArnaldo Carvalho de Melo 638aca7a94dSNamhyung Kim int symbol__tui_annotate(struct symbol *sym, struct map *map, int evidx, 639aca7a94dSNamhyung Kim void(*timer)(void *arg), void *arg, 640aca7a94dSNamhyung Kim int delay_secs) 641aca7a94dSNamhyung Kim { 64229ed6e76SArnaldo Carvalho de Melo struct disasm_line *pos, *n; 643aca7a94dSNamhyung Kim struct annotation *notes; 644b793a401SArnaldo Carvalho de Melo const size_t size = symbol__size(sym); 645aca7a94dSNamhyung Kim struct map_symbol ms = { 646aca7a94dSNamhyung Kim .map = map, 647aca7a94dSNamhyung Kim .sym = sym, 648aca7a94dSNamhyung Kim }; 649aca7a94dSNamhyung Kim struct annotate_browser browser = { 650aca7a94dSNamhyung Kim .b = { 651aca7a94dSNamhyung Kim .refresh = ui_browser__list_head_refresh, 652aca7a94dSNamhyung Kim .seek = ui_browser__list_head_seek, 653aca7a94dSNamhyung Kim .write = annotate_browser__write, 65429ed6e76SArnaldo Carvalho de Melo .filter = disasm_line__filter, 655aca7a94dSNamhyung Kim .priv = &ms, 656aca7a94dSNamhyung Kim .use_navkeypressed = true, 657aca7a94dSNamhyung Kim }, 6588bf39cb8SArnaldo Carvalho de Melo .use_offset = true, 659aca7a94dSNamhyung Kim }; 660b793a401SArnaldo Carvalho de Melo int ret = -1; 661aca7a94dSNamhyung Kim 662aca7a94dSNamhyung Kim if (sym == NULL) 663aca7a94dSNamhyung Kim return -1; 664aca7a94dSNamhyung Kim 665aca7a94dSNamhyung Kim if (map->dso->annotate_warned) 666aca7a94dSNamhyung Kim return -1; 667aca7a94dSNamhyung Kim 668b793a401SArnaldo Carvalho de Melo browser.offsets = zalloc(size * sizeof(struct disasm_line *)); 669b793a401SArnaldo Carvalho de Melo if (browser.offsets == NULL) { 670b793a401SArnaldo Carvalho de Melo ui__error("Not enough memory!"); 671b793a401SArnaldo Carvalho de Melo return -1; 672b793a401SArnaldo Carvalho de Melo } 673b793a401SArnaldo Carvalho de Melo 674887c0066SArnaldo Carvalho de Melo if (symbol__annotate(sym, map, sizeof(struct browser_disasm_line)) < 0) { 675aca7a94dSNamhyung Kim ui__error("%s", ui_helpline__last_msg); 676b793a401SArnaldo Carvalho de Melo goto out_free_offsets; 677aca7a94dSNamhyung Kim } 678aca7a94dSNamhyung Kim 679aca7a94dSNamhyung Kim ui_helpline__push("Press <- or ESC to exit"); 680aca7a94dSNamhyung Kim 681aca7a94dSNamhyung Kim notes = symbol__annotation(sym); 682aca7a94dSNamhyung Kim browser.start = map__rip_2objdump(map, sym->start); 683aca7a94dSNamhyung Kim 684aca7a94dSNamhyung Kim list_for_each_entry(pos, ¬es->src->source, node) { 685887c0066SArnaldo Carvalho de Melo struct browser_disasm_line *bpos; 686aca7a94dSNamhyung Kim size_t line_len = strlen(pos->line); 687aca7a94dSNamhyung Kim 688aca7a94dSNamhyung Kim if (browser.b.width < line_len) 689aca7a94dSNamhyung Kim browser.b.width = line_len; 690887c0066SArnaldo Carvalho de Melo bpos = disasm_line__browser(pos); 691887c0066SArnaldo Carvalho de Melo bpos->idx = browser.nr_entries++; 692b793a401SArnaldo Carvalho de Melo if (pos->offset != -1) { 693887c0066SArnaldo Carvalho de Melo bpos->idx_asm = browser.nr_asm_entries++; 694b793a401SArnaldo Carvalho de Melo browser.offsets[pos->offset] = pos; 695b793a401SArnaldo Carvalho de Melo } else 696887c0066SArnaldo Carvalho de Melo bpos->idx_asm = -1; 697aca7a94dSNamhyung Kim } 698aca7a94dSNamhyung Kim 699b793a401SArnaldo Carvalho de Melo annotate_browser__mark_jump_targets(&browser, size); 700b793a401SArnaldo Carvalho de Melo 70161e04b33SArnaldo Carvalho de Melo browser.offset_width = hex_width(size); 702aca7a94dSNamhyung Kim browser.b.nr_entries = browser.nr_entries; 703aca7a94dSNamhyung Kim browser.b.entries = ¬es->src->source, 704aca7a94dSNamhyung Kim browser.b.width += 18; /* Percentage */ 705aca7a94dSNamhyung Kim ret = annotate_browser__run(&browser, evidx, timer, arg, delay_secs); 706aca7a94dSNamhyung Kim list_for_each_entry_safe(pos, n, ¬es->src->source, node) { 707aca7a94dSNamhyung Kim list_del(&pos->node); 70829ed6e76SArnaldo Carvalho de Melo disasm_line__free(pos); 709aca7a94dSNamhyung Kim } 710b793a401SArnaldo Carvalho de Melo 711b793a401SArnaldo Carvalho de Melo out_free_offsets: 712b793a401SArnaldo Carvalho de Melo free(browser.offsets); 713aca7a94dSNamhyung Kim return ret; 714aca7a94dSNamhyung Kim } 715