[deliverable/linux.git] / tools / perf / util / ui / browsers / map.c

#include "../libslang.h"
#include <elf.h>
#include <newt.h>
#include <inttypes.h>
#include <sys/ttydefaults.h>
#include <string.h>
#include <linux/bitops.h>
#include "../../util.h"
#include "../../debug.h"
#include "../../symbol.h"
#include "../browser.h"
#include "../helpline.h"
#include "map.h"

static int ui_entry__read(const char *title, char *bf, size_t size, int width)
{
	struct newtExitStruct es;
	newtComponent form, entry;
	const char *result;
	int err = -1;

	newtCenteredWindow(width, 1, title);
	form = newtForm(NULL, NULL, 0);
	if (form == NULL)
		return -1;

	entry = newtEntry(0, 0, "0x", width, &result, NEWT_FLAG_SCROLL);
	if (entry == NULL)
		goto out_free_form;

	newtFormAddComponent(form, entry);
	newtFormAddHotKey(form, NEWT_KEY_ENTER);
	newtFormAddHotKey(form, NEWT_KEY_ESCAPE);
	newtFormAddHotKey(form, NEWT_KEY_LEFT);
	newtFormAddHotKey(form, CTRL('c'));
	newtFormRun(form, &es);

	if (result != NULL) {
		strncpy(bf, result, size);
		err = 0;
	}
out_free_form:
	newtPopWindow();
	newtFormDestroy(form);
	return err;
}

struct map_browser {
	struct ui_browser b;
	struct map	  *map;
	u8		  addrlen;
};

static void map_browser__write(struct ui_browser *self, void *nd, int row)
{
	struct symbol *sym = rb_entry(nd, struct symbol, rb_node);
	struct map_browser *mb = container_of(self, struct map_browser, b);
	bool current_entry = ui_browser__is_current_entry(self, row);
	int width;

	ui_browser__set_percent_color(self, 0, current_entry);
	slsmg_printf("%*" PRIx64 " %*" PRIx64 " %c ",
		     mb->addrlen, sym->start, mb->addrlen, sym->end,
		     sym->binding == STB_GLOBAL ? 'g' :
		     sym->binding == STB_LOCAL  ? 'l' : 'w');
	width = self->width - ((mb->addrlen * 2) + 4);
	if (width > 0)
		slsmg_write_nstring(sym->name, width);
}

/* FIXME uber-kludgy, see comment on cmd_report... */
static u32 *symbol__browser_index(struct symbol *self)
{
	return ((void *)self) - sizeof(struct rb_node) - sizeof(u32);
}

static int map_browser__search(struct map_browser *self)
{
	char target[512];
	struct symbol *sym;
	int err = ui_entry__read("Search by name/addr", target, sizeof(target), 40);

	if (err)
		return err;

	if (target[0] == '0' && tolower(target[1]) == 'x') {
		u64 addr = strtoull(target, NULL, 16);
		sym = map__find_symbol(self->map, addr, NULL);
	} else
		sym = map__find_symbol_by_name(self->map, target, NULL);

	if (sym != NULL) {
		u32 *idx = symbol__browser_index(sym);

		self->b.top = &sym->rb_node;
		self->b.index = self->b.top_idx = *idx;
	} else
		ui_helpline__fpush("%s not found!", target);

	return 0;
}

static int map_browser__run(struct map_browser *self)
{
	int key;

	if (ui_browser__show(&self->b, self->map->dso->long_name,
			     "Press <- or ESC to exit, %s / to search",
			     verbose ? "" : "restart with -v to use") < 0)
		return -1;

	while (1) {
		key = ui_browser__run(&self->b, 0);

		if (verbose && key == '/')
			map_browser__search(self);
		else
			break;
	}

	ui_browser__hide(&self->b);
	return key;
}

int map__browse(struct map *self)
{
	struct map_browser mb = {
		.b = {
			.entries = &self->dso->symbols[self->type],
			.refresh = ui_browser__rb_tree_refresh,
			.seek	 = ui_browser__rb_tree_seek,
			.write	 = map_browser__write,
		},
		.map = self,
	};
	struct rb_node *nd;
	char tmp[BITS_PER_LONG / 4];
	u64 maxaddr = 0;

	for (nd = rb_first(mb.b.entries); nd; nd = rb_next(nd)) {
		struct symbol *pos = rb_entry(nd, struct symbol, rb_node);

		if (maxaddr < pos->end)
			maxaddr = pos->end;
		if (verbose) {
			u32 *idx = symbol__browser_index(pos);
			*idx = mb.b.nr_entries;
		}
		++mb.b.nr_entries;
	}

	mb.addrlen = snprintf(tmp, sizeof(tmp), "%" PRIx64, maxaddr);
	return map_browser__run(&mb);
}
Commit	Line	Data
b1b02673 ACM	1	#include "../libslang.h"
b1b02673 ACM	2	#include <elf.h>
3af6e338	3	#include <newt.h>
9486aa38	4	#include <inttypes.h>
b1b02673	5	#include <sys/ttydefaults.h>
b1b02673 ACM	6	#include <string.h>
b1b02673 ACM	7	#include <linux/bitops.h>
d30d4a08	8	#include "../../util.h"
b1b02673 ACM	9	#include "../../debug.h"
	10	#include "../../symbol.h"
	11	#include "../browser.h"
	12	#include "../helpline.h"
	13	#include "map.h"
	14
	15	static int ui_entry__read(const char title, char bf, size_t size, int width)
	16	{
	17	struct newtExitStruct es;
	18	newtComponent form, entry;
	19	const char *result;
	20	int err = -1;
	21
	22	newtCenteredWindow(width, 1, title);
	23	form = newtForm(NULL, NULL, 0);
	24	if (form == NULL)
	25	return -1;
	26
	27	entry = newtEntry(0, 0, "0x", width, &result, NEWT_FLAG_SCROLL);
	28	if (entry == NULL)
	29	goto out_free_form;
	30
	31	newtFormAddComponent(form, entry);
	32	newtFormAddHotKey(form, NEWT_KEY_ENTER);
	33	newtFormAddHotKey(form, NEWT_KEY_ESCAPE);
	34	newtFormAddHotKey(form, NEWT_KEY_LEFT);
	35	newtFormAddHotKey(form, CTRL('c'));
	36	newtFormRun(form, &es);
	37
	38	if (result != NULL) {
	39	strncpy(bf, result, size);
	40	err = 0;
	41	}
	42	out_free_form:
	43	newtPopWindow();
	44	newtFormDestroy(form);
fb7d0b3c	45	return err;
b1b02673 ACM	46	}
	47
	48	struct map_browser {
	49	struct ui_browser b;
	50	struct map *map;
b1b02673 ACM	51	u8 addrlen;
	52	};
	53
	54	static void map_browser__write(struct ui_browser self, void nd, int row)
	55	{
	56	struct symbol *sym = rb_entry(nd, struct symbol, rb_node);
	57	struct map_browser *mb = container_of(self, struct map_browser, b);
	58	bool current_entry = ui_browser__is_current_entry(self, row);
469917ce	59	int width;
b1b02673	60
8f9bbc40	61	ui_browser__set_percent_color(self, 0, current_entry);
9486aa38	62	slsmg_printf("%" PRIx64 " %" PRIx64 " %c ",
b1b02673 ACM	63	mb->addrlen, sym->start, mb->addrlen, sym->end,
	64	sym->binding == STB_GLOBAL ? 'g' :
	65	sym->binding == STB_LOCAL ? 'l' : 'w');
469917ce ACM	66	width = self->width - ((mb->addrlen * 2) + 4);
	67	if (width > 0)
	68	slsmg_write_nstring(sym->name, width);
b1b02673 ACM	69	}
	70
	71	/* FIXME uber-kludgy, see comment on cmd_report... */
	72	static u32 symbol__browser_index(struct symbol self)
	73	{
	74	return ((void *)self) - sizeof(struct rb_node) - sizeof(u32);
	75	}
	76
	77	static int map_browser__search(struct map_browser *self)
	78	{
	79	char target[512];
	80	struct symbol *sym;
	81	int err = ui_entry__read("Search by name/addr", target, sizeof(target), 40);
	82
	83	if (err)
	84	return err;
	85
	86	if (target[0] == '0' && tolower(target[1]) == 'x') {
	87	u64 addr = strtoull(target, NULL, 16);
	88	sym = map__find_symbol(self->map, addr, NULL);
	89	} else
	90	sym = map__find_symbol_by_name(self->map, target, NULL);
	91
	92	if (sym != NULL) {
	93	u32 *idx = symbol__browser_index(sym);
	94
	95	self->b.top = &sym->rb_node;
	96	self->b.index = self->b.top_idx = *idx;
	97	} else
	98	ui_helpline__fpush("%s not found!", target);
	99
	100	return 0;
	101	}
	102
b50e003d	103	static int map_browser__run(struct map_browser *self)
b1b02673	104	{
b50e003d ACM	105	int key;
b50e003d ACM	106
59e8fe32 ACM	107	if (ui_browser__show(&self->b, self->map->dso->long_name,
	108	"Press <- or ESC to exit, %s / to search",
	109	verbose ? "" : "restart with -v to use") < 0)
b1b02673 ACM	110	return -1;
b1b02673 ACM	111
b1b02673	112	while (1) {
3af6e338	113	key = ui_browser__run(&self->b, 0);
b1b02673	114
b50e003d	115	if (verbose && key == '/')
b1b02673 ACM	116	map_browser__search(self);
	117	else
	118	break;
	119	}
	120
59e8fe32	121	ui_browser__hide(&self->b);
b50e003d	122	return key;
b1b02673 ACM	123	}
	124
	125	int map__browse(struct map *self)
	126	{
	127	struct map_browser mb = {
	128	.b = {
	129	.entries = &self->dso->symbols[self->type],
	130	.refresh = ui_browser__rb_tree_refresh,
	131	.seek = ui_browser__rb_tree_seek,
	132	.write = map_browser__write,
	133	},
	134	.map = self,
	135	};
b1b02673 ACM	136	struct rb_node *nd;
	137	char tmp[BITS_PER_LONG / 4];
	138	u64 maxaddr = 0;
	139
	140	for (nd = rb_first(mb.b.entries); nd; nd = rb_next(nd)) {
	141	struct symbol *pos = rb_entry(nd, struct symbol, rb_node);
	142
b1b02673 ACM	143	if (maxaddr < pos->end)
	144	maxaddr = pos->end;
	145	if (verbose) {
	146	u32 *idx = symbol__browser_index(pos);
	147	*idx = mb.b.nr_entries;
	148	}
	149	++mb.b.nr_entries;
	150	}
	151
9486aa38	152	mb.addrlen = snprintf(tmp, sizeof(tmp), "%" PRIx64, maxaddr);
b50e003d	153	return map_browser__run(&mb);
b1b02673	154	}