1 // SPDX-License-Identifier: GPL-2.0 2 #include <elf.h> 3 #include <inttypes.h> 4 #include <sys/ttydefaults.h> 5 #include <string.h> 6 #include <linux/bitops.h> 7 #include "../../util/util.h" 8 #include "../../util/debug.h" 9 #include "../../util/symbol.h" 10 #include "../browser.h" 11 #include "../helpline.h" 12 #include "../keysyms.h" 13 #include "map.h" 14 15 #include "sane_ctype.h" 16 17 struct map_browser { 18 struct ui_browser b; 19 struct map *map; 20 u8 addrlen; 21 }; 22 23 static void map_browser__write(struct ui_browser *browser, void *nd, int row) 24 { 25 struct symbol *sym = rb_entry(nd, struct symbol, rb_node); 26 struct map_browser *mb = container_of(browser, struct map_browser, b); 27 bool current_entry = ui_browser__is_current_entry(browser, row); 28 int width; 29 30 ui_browser__set_percent_color(browser, 0, current_entry); 31 ui_browser__printf(browser, "%*" PRIx64 " %*" PRIx64 " %c ", 32 mb->addrlen, sym->start, mb->addrlen, sym->end, 33 sym->binding == STB_GLOBAL ? 'g' : 34 sym->binding == STB_LOCAL ? 'l' : 'w'); 35 width = browser->width - ((mb->addrlen * 2) + 4); 36 if (width > 0) 37 ui_browser__write_nstring(browser, sym->name, width); 38 } 39 40 /* FIXME uber-kludgy, see comment on cmd_report... */ 41 static u32 *symbol__browser_index(struct symbol *browser) 42 { 43 return ((void *)browser) - sizeof(struct rb_node) - sizeof(u32); 44 } 45 46 static int map_browser__search(struct map_browser *browser) 47 { 48 char target[512]; 49 struct symbol *sym; 50 int err = ui_browser__input_window("Search by name/addr", 51 "Prefix with 0x to search by address", 52 target, "ENTER: OK, ESC: Cancel", 0); 53 if (err != K_ENTER) 54 return -1; 55 56 if (target[0] == '0' && tolower(target[1]) == 'x') { 57 u64 addr = strtoull(target, NULL, 16); 58 sym = map__find_symbol(browser->map, addr); 59 } else 60 sym = map__find_symbol_by_name(browser->map, target); 61 62 if (sym != NULL) { 63 u32 *idx = symbol__browser_index(sym); 64 65 browser->b.top = &sym->rb_node; 66 browser->b.index = browser->b.top_idx = *idx; 67 } else 68 ui_helpline__fpush("%s not found!", target); 69 70 return 0; 71 } 72 73 static int map_browser__run(struct map_browser *browser) 74 { 75 int key; 76 77 if (ui_browser__show(&browser->b, browser->map->dso->long_name, 78 "Press ESC to exit, %s / to search", 79 verbose > 0 ? "" : "restart with -v to use") < 0) 80 return -1; 81 82 while (1) { 83 key = ui_browser__run(&browser->b, 0); 84 85 switch (key) { 86 case '/': 87 if (verbose > 0) 88 map_browser__search(browser); 89 default: 90 break; 91 case K_LEFT: 92 case K_ESC: 93 case 'q': 94 case CTRL('c'): 95 goto out; 96 } 97 } 98 out: 99 ui_browser__hide(&browser->b); 100 return key; 101 } 102 103 int map__browse(struct map *map) 104 { 105 struct map_browser mb = { 106 .b = { 107 .entries = &map->dso->symbols, 108 .refresh = ui_browser__rb_tree_refresh, 109 .seek = ui_browser__rb_tree_seek, 110 .write = map_browser__write, 111 }, 112 .map = map, 113 }; 114 struct rb_node *nd; 115 char tmp[BITS_PER_LONG / 4]; 116 u64 maxaddr = 0; 117 118 for (nd = rb_first(mb.b.entries); nd; nd = rb_next(nd)) { 119 struct symbol *pos = rb_entry(nd, struct symbol, rb_node); 120 121 if (maxaddr < pos->end) 122 maxaddr = pos->end; 123 if (verbose > 0) { 124 u32 *idx = symbol__browser_index(pos); 125 *idx = mb.b.nr_entries; 126 } 127 ++mb.b.nr_entries; 128 } 129 130 mb.addrlen = snprintf(tmp, sizeof(tmp), "%" PRIx64, maxaddr); 131 return map_browser__run(&mb); 132 } 133