[mirror_ubuntu-artful-kernel.git] / tools / perf / util / dso.c

#include "symbol.h"
#include "dso.h"
#include "machine.h"
#include "util.h"
#include "debug.h"

char dso__symtab_origin(const struct dso *dso)
{
	static const char origin[] = {
		[DSO_BINARY_TYPE__KALLSYMS]			= 'k',
		[DSO_BINARY_TYPE__VMLINUX]			= 'v',
		[DSO_BINARY_TYPE__JAVA_JIT]			= 'j',
		[DSO_BINARY_TYPE__DEBUGLINK]			= 'l',
		[DSO_BINARY_TYPE__BUILD_ID_CACHE]		= 'B',
		[DSO_BINARY_TYPE__FEDORA_DEBUGINFO]		= 'f',
		[DSO_BINARY_TYPE__UBUNTU_DEBUGINFO]		= 'u',
		[DSO_BINARY_TYPE__OPENEMBEDDED_DEBUGINFO]	= 'o',
		[DSO_BINARY_TYPE__BUILDID_DEBUGINFO]		= 'b',
		[DSO_BINARY_TYPE__SYSTEM_PATH_DSO]		= 'd',
		[DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE]		= 'K',
		[DSO_BINARY_TYPE__GUEST_KALLSYMS]		= 'g',
		[DSO_BINARY_TYPE__GUEST_KMODULE]		= 'G',
		[DSO_BINARY_TYPE__GUEST_VMLINUX]		= 'V',
	};

	if (dso == NULL || dso->symtab_type == DSO_BINARY_TYPE__NOT_FOUND)
		return '!';
	return origin[dso->symtab_type];
}

int dso__binary_type_file(struct dso *dso, enum dso_binary_type type,
			  char *root_dir, char *file, size_t size)
{
	char build_id_hex[BUILD_ID_SIZE * 2 + 1];
	int ret = 0;

	switch (type) {
	case DSO_BINARY_TYPE__DEBUGLINK: {
		char *debuglink;

		strncpy(file, dso->long_name, size);
		debuglink = file + dso->long_name_len;
		while (debuglink != file && *debuglink != '/')
			debuglink--;
		if (*debuglink == '/')
			debuglink++;
		filename__read_debuglink(dso->long_name, debuglink,
					 size - (debuglink - file));
		}
		break;
	case DSO_BINARY_TYPE__BUILD_ID_CACHE:
		/* skip the locally configured cache if a symfs is given */
		if (symbol_conf.symfs[0] ||
		    (dso__build_id_filename(dso, file, size) == NULL))
			ret = -1;
		break;

	case DSO_BINARY_TYPE__FEDORA_DEBUGINFO:
		snprintf(file, size, "%s/usr/lib/debug%s.debug",
			 symbol_conf.symfs, dso->long_name);
		break;

	case DSO_BINARY_TYPE__UBUNTU_DEBUGINFO:
		snprintf(file, size, "%s/usr/lib/debug%s",
			 symbol_conf.symfs, dso->long_name);
		break;

	case DSO_BINARY_TYPE__OPENEMBEDDED_DEBUGINFO:
	{
		char *last_slash;
		size_t len;
		size_t dir_size;

		last_slash = dso->long_name + dso->long_name_len;
		while (last_slash != dso->long_name && *last_slash != '/')
			last_slash--;

		len = scnprintf(file, size, "%s", symbol_conf.symfs);
		dir_size = last_slash - dso->long_name + 2;
		if (dir_size > (size - len)) {
			ret = -1;
			break;
		}
		len += scnprintf(file + len, dir_size, "%s",  dso->long_name);
		len += scnprintf(file + len , size - len, ".debug%s",
								last_slash);
		break;
	}

	case DSO_BINARY_TYPE__BUILDID_DEBUGINFO:
		if (!dso->has_build_id) {
			ret = -1;
			break;
		}

		build_id__sprintf(dso->build_id,
				  sizeof(dso->build_id),
				  build_id_hex);
		snprintf(file, size,
			 "%s/usr/lib/debug/.build-id/%.2s/%s.debug",
			 symbol_conf.symfs, build_id_hex, build_id_hex + 2);
		break;

	case DSO_BINARY_TYPE__VMLINUX:
	case DSO_BINARY_TYPE__GUEST_VMLINUX:
	case DSO_BINARY_TYPE__SYSTEM_PATH_DSO:
		snprintf(file, size, "%s%s",
			 symbol_conf.symfs, dso->long_name);
		break;

	case DSO_BINARY_TYPE__GUEST_KMODULE:
		snprintf(file, size, "%s%s%s", symbol_conf.symfs,
			 root_dir, dso->long_name);
		break;

	case DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE:
		snprintf(file, size, "%s%s", symbol_conf.symfs,
			 dso->long_name);
		break;

	case DSO_BINARY_TYPE__KCORE:
	case DSO_BINARY_TYPE__GUEST_KCORE:
		snprintf(file, size, "%s", dso->long_name);
		break;

	default:
	case DSO_BINARY_TYPE__KALLSYMS:
	case DSO_BINARY_TYPE__GUEST_KALLSYMS:
	case DSO_BINARY_TYPE__JAVA_JIT:
	case DSO_BINARY_TYPE__NOT_FOUND:
		ret = -1;
		break;
	}

	return ret;
}

static int open_dso(struct dso *dso, struct machine *machine)
{
	char *root_dir = (char *) "";
	char *name;
	int fd;

	name = malloc(PATH_MAX);
	if (!name)
		return -ENOMEM;

	if (machine)
		root_dir = machine->root_dir;

	if (dso__binary_type_file(dso, dso->data_type,
				  root_dir, name, PATH_MAX)) {
		free(name);
		return -EINVAL;
	}

	fd = open(name, O_RDONLY);
	free(name);
	return fd;
}

int dso__data_fd(struct dso *dso, struct machine *machine)
{
	static enum dso_binary_type binary_type_data[] = {
		DSO_BINARY_TYPE__BUILD_ID_CACHE,
		DSO_BINARY_TYPE__SYSTEM_PATH_DSO,
		DSO_BINARY_TYPE__NOT_FOUND,
	};
	int i = 0;

	if (dso->data_type != DSO_BINARY_TYPE__NOT_FOUND)
		return open_dso(dso, machine);

	do {
		int fd;

		dso->data_type = binary_type_data[i++];

		fd = open_dso(dso, machine);
		if (fd >= 0)
			return fd;

	} while (dso->data_type != DSO_BINARY_TYPE__NOT_FOUND);

	return -EINVAL;
}

static void
dso_cache__free(struct rb_root *root)
{
	struct rb_node *next = rb_first(root);

	while (next) {
		struct dso_cache *cache;

		cache = rb_entry(next, struct dso_cache, rb_node);
		next = rb_next(&cache->rb_node);
		rb_erase(&cache->rb_node, root);
		free(cache);
	}
}

static struct dso_cache*
dso_cache__find(struct rb_root *root, u64 offset)
{
	struct rb_node **p = &root->rb_node;
	struct rb_node *parent = NULL;
	struct dso_cache *cache;

	while (*p != NULL) {
		u64 end;

		parent = *p;
		cache = rb_entry(parent, struct dso_cache, rb_node);
		end = cache->offset + DSO__DATA_CACHE_SIZE;

		if (offset < cache->offset)
			p = &(*p)->rb_left;
		else if (offset >= end)
			p = &(*p)->rb_right;
		else
			return cache;
	}
	return NULL;
}

static void
dso_cache__insert(struct rb_root *root, struct dso_cache *new)
{
	struct rb_node **p = &root->rb_node;
	struct rb_node *parent = NULL;
	struct dso_cache *cache;
	u64 offset = new->offset;

	while (*p != NULL) {
		u64 end;

		parent = *p;
		cache = rb_entry(parent, struct dso_cache, rb_node);
		end = cache->offset + DSO__DATA_CACHE_SIZE;

		if (offset < cache->offset)
			p = &(*p)->rb_left;
		else if (offset >= end)
			p = &(*p)->rb_right;
	}

	rb_link_node(&new->rb_node, parent, p);
	rb_insert_color(&new->rb_node, root);
}

static ssize_t
dso_cache__memcpy(struct dso_cache *cache, u64 offset,
		  u8 *data, u64 size)
{
	u64 cache_offset = offset - cache->offset;
	u64 cache_size   = min(cache->size - cache_offset, size);

	memcpy(data, cache->data + cache_offset, cache_size);
	return cache_size;
}

static ssize_t
dso_cache__read(struct dso *dso, struct machine *machine,
		 u64 offset, u8 *data, ssize_t size)
{
	struct dso_cache *cache;
	ssize_t ret;
	int fd;

	fd = dso__data_fd(dso, machine);
	if (fd < 0)
		return -1;

	do {
		u64 cache_offset;

		ret = -ENOMEM;

		cache = zalloc(sizeof(*cache) + DSO__DATA_CACHE_SIZE);
		if (!cache)
			break;

		cache_offset = offset & DSO__DATA_CACHE_MASK;
		ret = -EINVAL;

		if (-1 == lseek(fd, cache_offset, SEEK_SET))
			break;

		ret = read(fd, cache->data, DSO__DATA_CACHE_SIZE);
		if (ret <= 0)
			break;

		cache->offset = cache_offset;
		cache->size   = ret;
		dso_cache__insert(&dso->cache, cache);

		ret = dso_cache__memcpy(cache, offset, data, size);

	} while (0);

	if (ret <= 0)
		free(cache);

	close(fd);
	return ret;
}

static ssize_t dso_cache_read(struct dso *dso, struct machine *machine,
			      u64 offset, u8 *data, ssize_t size)
{
	struct dso_cache *cache;

	cache = dso_cache__find(&dso->cache, offset);
	if (cache)
		return dso_cache__memcpy(cache, offset, data, size);
	else
		return dso_cache__read(dso, machine, offset, data, size);
}

ssize_t dso__data_read_offset(struct dso *dso, struct machine *machine,
			      u64 offset, u8 *data, ssize_t size)
{
	ssize_t r = 0;
	u8 *p = data;

	do {
		ssize_t ret;

		ret = dso_cache_read(dso, machine, offset, p, size);
		if (ret < 0)
			return ret;

		/* Reached EOF, return what we have. */
		if (!ret)
			break;

		BUG_ON(ret > size);

		r      += ret;
		p      += ret;
		offset += ret;
		size   -= ret;

	} while (size);

	return r;
}

ssize_t dso__data_read_addr(struct dso *dso, struct map *map,
			    struct machine *machine, u64 addr,
			    u8 *data, ssize_t size)
{
	u64 offset = map->map_ip(map, addr);
	return dso__data_read_offset(dso, machine, offset, data, size);
}

struct map *dso__new_map(const char *name)
{
	struct map *map = NULL;
	struct dso *dso = dso__new(name);

	if (dso)
		map = map__new2(0, dso, MAP__FUNCTION);

	return map;
}

struct dso *dso__kernel_findnew(struct machine *machine, const char *name,
		    const char *short_name, int dso_type)
{
	/*
	 * The kernel dso could be created by build_id processing.
	 */
	struct dso *dso = __dsos__findnew(&machine->kernel_dsos, name);

	/*
	 * We need to run this in all cases, since during the build_id
	 * processing we had no idea this was the kernel dso.
	 */
	if (dso != NULL) {
		dso__set_short_name(dso, short_name);
		dso->kernel = dso_type;
	}

	return dso;
}

void dso__set_long_name(struct dso *dso, char *name)
{
	if (name == NULL)
		return;
	dso->long_name = name;
	dso->long_name_len = strlen(name);
}

void dso__set_short_name(struct dso *dso, const char *name)
{
	if (name == NULL)
		return;
	dso->short_name = name;
	dso->short_name_len = strlen(name);
}

static void dso__set_basename(struct dso *dso)
{
	dso__set_short_name(dso, basename(dso->long_name));
}

int dso__name_len(const struct dso *dso)
{
	if (!dso)
		return strlen("[unknown]");
	if (verbose)
		return dso->long_name_len;

	return dso->short_name_len;
}

bool dso__loaded(const struct dso *dso, enum map_type type)
{
	return dso->loaded & (1 << type);
}

bool dso__sorted_by_name(const struct dso *dso, enum map_type type)
{
	return dso->sorted_by_name & (1 << type);
}

void dso__set_sorted_by_name(struct dso *dso, enum map_type type)
{
	dso->sorted_by_name |= (1 << type);
}

struct dso *dso__new(const char *name)
{
	struct dso *dso = calloc(1, sizeof(*dso) + strlen(name) + 1);

	if (dso != NULL) {
		int i;
		strcpy(dso->name, name);
		dso__set_long_name(dso, dso->name);
		dso__set_short_name(dso, dso->name);
		for (i = 0; i < MAP__NR_TYPES; ++i)
			dso->symbols[i] = dso->symbol_names[i] = RB_ROOT;
		dso->cache = RB_ROOT;
		dso->symtab_type = DSO_BINARY_TYPE__NOT_FOUND;
		dso->data_type   = DSO_BINARY_TYPE__NOT_FOUND;
		dso->loaded = 0;
		dso->rel = 0;
		dso->sorted_by_name = 0;
		dso->has_build_id = 0;
		dso->has_srcline = 1;
		dso->kernel = DSO_TYPE_USER;
		dso->needs_swap = DSO_SWAP__UNSET;
		INIT_LIST_HEAD(&dso->node);
	}

	return dso;
}

void dso__delete(struct dso *dso)
{
	int i;
	for (i = 0; i < MAP__NR_TYPES; ++i)
		symbols__delete(&dso->symbols[i]);
	if (dso->sname_alloc)
		free((char *)dso->short_name);
	if (dso->lname_alloc)
		free(dso->long_name);
	dso_cache__free(&dso->cache);
	free(dso);
}

void dso__set_build_id(struct dso *dso, void *build_id)
{
	memcpy(dso->build_id, build_id, sizeof(dso->build_id));
	dso->has_build_id = 1;
}

bool dso__build_id_equal(const struct dso *dso, u8 *build_id)
{
	return memcmp(dso->build_id, build_id, sizeof(dso->build_id)) == 0;
}

void dso__read_running_kernel_build_id(struct dso *dso, struct machine *machine)
{
	char path[PATH_MAX];

	if (machine__is_default_guest(machine))
		return;
	sprintf(path, "%s/sys/kernel/notes", machine->root_dir);
	if (sysfs__read_build_id(path, dso->build_id,
				 sizeof(dso->build_id)) == 0)
		dso->has_build_id = true;
}

int dso__kernel_module_get_build_id(struct dso *dso,
				    const char *root_dir)
{
	char filename[PATH_MAX];
	/*
	 * kernel module short names are of the form "[module]" and
	 * we need just "module" here.
	 */
	const char *name = dso->short_name + 1;

	snprintf(filename, sizeof(filename),
		 "%s/sys/module/%.*s/notes/.note.gnu.build-id",
		 root_dir, (int)strlen(name) - 1, name);

	if (sysfs__read_build_id(filename, dso->build_id,
				 sizeof(dso->build_id)) == 0)
		dso->has_build_id = true;

	return 0;
}

bool __dsos__read_build_ids(struct list_head *head, bool with_hits)
{
	bool have_build_id = false;
	struct dso *pos;

	list_for_each_entry(pos, head, node) {
		if (with_hits && !pos->hit)
			continue;
		if (pos->has_build_id) {
			have_build_id = true;
			continue;
		}
		if (filename__read_build_id(pos->long_name, pos->build_id,
					    sizeof(pos->build_id)) > 0) {
			have_build_id	  = true;
			pos->has_build_id = true;
		}
	}

	return have_build_id;
}

void dsos__add(struct list_head *head, struct dso *dso)
{
	list_add_tail(&dso->node, head);
}

struct dso *dsos__find(struct list_head *head, const char *name, bool cmp_short)
{
	struct dso *pos;

	if (cmp_short) {
		list_for_each_entry(pos, head, node)
			if (strcmp(pos->short_name, name) == 0)
				return pos;
		return NULL;
	}
	list_for_each_entry(pos, head, node)
		if (strcmp(pos->long_name, name) == 0)
			return pos;
	return NULL;
}

struct dso *__dsos__findnew(struct list_head *head, const char *name)
{
	struct dso *dso = dsos__find(head, name, false);

	if (!dso) {
		dso = dso__new(name);
		if (dso != NULL) {
			dsos__add(head, dso);
			dso__set_basename(dso);
		}
	}

	return dso;
}

size_t __dsos__fprintf_buildid(struct list_head *head, FILE *fp,
			       bool (skip)(struct dso *dso, int parm), int parm)
{
	struct dso *pos;
	size_t ret = 0;

	list_for_each_entry(pos, head, node) {
		if (skip && skip(pos, parm))
			continue;
		ret += dso__fprintf_buildid(pos, fp);
		ret += fprintf(fp, " %s\n", pos->long_name);
	}
	return ret;
}

size_t __dsos__fprintf(struct list_head *head, FILE *fp)
{
	struct dso *pos;
	size_t ret = 0;

	list_for_each_entry(pos, head, node) {
		int i;
		for (i = 0; i < MAP__NR_TYPES; ++i)
			ret += dso__fprintf(pos, i, fp);
	}

	return ret;
}

size_t dso__fprintf_buildid(struct dso *dso, FILE *fp)
{
	char sbuild_id[BUILD_ID_SIZE * 2 + 1];

	build_id__sprintf(dso->build_id, sizeof(dso->build_id), sbuild_id);
	return fprintf(fp, "%s", sbuild_id);
}

size_t dso__fprintf(struct dso *dso, enum map_type type, FILE *fp)
{
	struct rb_node *nd;
	size_t ret = fprintf(fp, "dso: %s (", dso->short_name);

	if (dso->short_name != dso->long_name)
		ret += fprintf(fp, "%s, ", dso->long_name);
	ret += fprintf(fp, "%s, %sloaded, ", map_type__name[type],
		       dso__loaded(dso, type) ? "" : "NOT ");
	ret += dso__fprintf_buildid(dso, fp);
	ret += fprintf(fp, ")\n");
	for (nd = rb_first(&dso->symbols[type]); nd; nd = rb_next(nd)) {
		struct symbol *pos = rb_entry(nd, struct symbol, rb_node);
		ret += symbol__fprintf(pos, fp);
	}

	return ret;
}
Commit	Line	Data
cdd059d7 JO	1	#include "symbol.h"
cdd059d7 JO	2	#include "dso.h"
69d2591a	3	#include "machine.h"
cdd059d7 JO	4	#include "util.h"
	5	#include "debug.h"
	6
	7	char dso__symtab_origin(const struct dso *dso)
	8	{
	9	static const char origin[] = {
9cd00941 RRD	10	[DSO_BINARY_TYPE__KALLSYMS] = 'k',
	11	[DSO_BINARY_TYPE__VMLINUX] = 'v',
	12	[DSO_BINARY_TYPE__JAVA_JIT] = 'j',
	13	[DSO_BINARY_TYPE__DEBUGLINK] = 'l',
	14	[DSO_BINARY_TYPE__BUILD_ID_CACHE] = 'B',
	15	[DSO_BINARY_TYPE__FEDORA_DEBUGINFO] = 'f',
	16	[DSO_BINARY_TYPE__UBUNTU_DEBUGINFO] = 'u',
	17	[DSO_BINARY_TYPE__OPENEMBEDDED_DEBUGINFO] = 'o',
	18	[DSO_BINARY_TYPE__BUILDID_DEBUGINFO] = 'b',
	19	[DSO_BINARY_TYPE__SYSTEM_PATH_DSO] = 'd',
	20	[DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE] = 'K',
	21	[DSO_BINARY_TYPE__GUEST_KALLSYMS] = 'g',
	22	[DSO_BINARY_TYPE__GUEST_KMODULE] = 'G',
	23	[DSO_BINARY_TYPE__GUEST_VMLINUX] = 'V',
cdd059d7 JO	24	};
	25
	26	if (dso == NULL \|\| dso->symtab_type == DSO_BINARY_TYPE__NOT_FOUND)
	27	return '!';
	28	return origin[dso->symtab_type];
	29	}
	30
	31	int dso__binary_type_file(struct dso *dso, enum dso_binary_type type,
	32	char root_dir, char file, size_t size)
	33	{
	34	char build_id_hex[BUILD_ID_SIZE * 2 + 1];
	35	int ret = 0;
	36
	37	switch (type) {
	38	case DSO_BINARY_TYPE__DEBUGLINK: {
	39	char *debuglink;
	40
	41	strncpy(file, dso->long_name, size);
	42	debuglink = file + dso->long_name_len;
	43	while (debuglink != file && *debuglink != '/')
	44	debuglink--;
	45	if (*debuglink == '/')
	46	debuglink++;
	47	filename__read_debuglink(dso->long_name, debuglink,
	48	size - (debuglink - file));
	49	}
	50	break;
	51	case DSO_BINARY_TYPE__BUILD_ID_CACHE:
	52	/* skip the locally configured cache if a symfs is given */
	53	if (symbol_conf.symfs[0] \|\|
	54	(dso__build_id_filename(dso, file, size) == NULL))
	55	ret = -1;
	56	break;
	57
	58	case DSO_BINARY_TYPE__FEDORA_DEBUGINFO:
	59	snprintf(file, size, "%s/usr/lib/debug%s.debug",
	60	symbol_conf.symfs, dso->long_name);
	61	break;
	62
	63	case DSO_BINARY_TYPE__UBUNTU_DEBUGINFO:
	64	snprintf(file, size, "%s/usr/lib/debug%s",
	65	symbol_conf.symfs, dso->long_name);
	66	break;
	67
9cd00941 RRD	68	case DSO_BINARY_TYPE__OPENEMBEDDED_DEBUGINFO:
	69	{
	70	char *last_slash;
	71	size_t len;
	72	size_t dir_size;
	73
	74	last_slash = dso->long_name + dso->long_name_len;
	75	while (last_slash != dso->long_name && *last_slash != '/')
	76	last_slash--;
	77
	78	len = scnprintf(file, size, "%s", symbol_conf.symfs);
	79	dir_size = last_slash - dso->long_name + 2;
	80	if (dir_size > (size - len)) {
	81	ret = -1;
	82	break;
	83	}
	84	len += scnprintf(file + len, dir_size, "%s", dso->long_name);
	85	len += scnprintf(file + len , size - len, ".debug%s",
	86	last_slash);
	87	break;
	88	}
	89
cdd059d7 JO	90	case DSO_BINARY_TYPE__BUILDID_DEBUGINFO:
	91	if (!dso->has_build_id) {
	92	ret = -1;
	93	break;
	94	}
	95
	96	build_id__sprintf(dso->build_id,
	97	sizeof(dso->build_id),
	98	build_id_hex);
	99	snprintf(file, size,
	100	"%s/usr/lib/debug/.build-id/%.2s/%s.debug",
	101	symbol_conf.symfs, build_id_hex, build_id_hex + 2);
	102	break;
	103
39b12f78 AH	104	case DSO_BINARY_TYPE__VMLINUX:
39b12f78 AH	105	case DSO_BINARY_TYPE__GUEST_VMLINUX:
cdd059d7 JO	106	case DSO_BINARY_TYPE__SYSTEM_PATH_DSO:
	107	snprintf(file, size, "%s%s",
	108	symbol_conf.symfs, dso->long_name);
	109	break;
	110
	111	case DSO_BINARY_TYPE__GUEST_KMODULE:
	112	snprintf(file, size, "%s%s%s", symbol_conf.symfs,
	113	root_dir, dso->long_name);
	114	break;
	115
	116	case DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE:
	117	snprintf(file, size, "%s%s", symbol_conf.symfs,
	118	dso->long_name);
	119	break;
	120
8e0cf965 AH	121	case DSO_BINARY_TYPE__KCORE:
	122	case DSO_BINARY_TYPE__GUEST_KCORE:
	123	snprintf(file, size, "%s", dso->long_name);
	124	break;
	125
cdd059d7 JO	126	default:
cdd059d7 JO	127	case DSO_BINARY_TYPE__KALLSYMS:
cdd059d7	128	case DSO_BINARY_TYPE__GUEST_KALLSYMS:
cdd059d7 JO	129	case DSO_BINARY_TYPE__JAVA_JIT:
	130	case DSO_BINARY_TYPE__NOT_FOUND:
	131	ret = -1;
	132	break;
	133	}
	134
	135	return ret;
	136	}
	137
	138	static int open_dso(struct dso dso, struct machine machine)
	139	{
	140	char root_dir = (char ) "";
	141	char *name;
	142	int fd;
	143
	144	name = malloc(PATH_MAX);
	145	if (!name)
	146	return -ENOMEM;
	147
	148	if (machine)
	149	root_dir = machine->root_dir;
	150
	151	if (dso__binary_type_file(dso, dso->data_type,
	152	root_dir, name, PATH_MAX)) {
	153	free(name);
	154	return -EINVAL;
	155	}
	156
	157	fd = open(name, O_RDONLY);
	158	free(name);
	159	return fd;
	160	}
	161
	162	int dso__data_fd(struct dso dso, struct machine machine)
	163	{
	164	static enum dso_binary_type binary_type_data[] = {
	165	DSO_BINARY_TYPE__BUILD_ID_CACHE,
	166	DSO_BINARY_TYPE__SYSTEM_PATH_DSO,
	167	DSO_BINARY_TYPE__NOT_FOUND,
	168	};
	169	int i = 0;
	170
	171	if (dso->data_type != DSO_BINARY_TYPE__NOT_FOUND)
	172	return open_dso(dso, machine);
	173
	174	do {
	175	int fd;
	176
	177	dso->data_type = binary_type_data[i++];
	178
	179	fd = open_dso(dso, machine);
	180	if (fd >= 0)
	181	return fd;
	182
	183	} while (dso->data_type != DSO_BINARY_TYPE__NOT_FOUND);
	184
	185	return -EINVAL;
	186	}
	187
	188	static void
	189	dso_cache__free(struct rb_root *root)
	190	{
	191	struct rb_node *next = rb_first(root);
	192
193	while (next) {
194	struct dso_cache *cache;
195
196	cache = rb_entry(next, struct dso_cache, rb_node);
197	next = rb_next(&cache->rb_node);
198	rb_erase(&cache->rb_node, root);
199	free(cache);
200	}
201	}
202
203	static struct dso_cache*
204	dso_cache__find(struct rb_root *root, u64 offset)
205	{
206	struct rb_node **p = &root->rb_node;
207	struct rb_node *parent = NULL;
208	struct dso_cache *cache;
209
210	while (*p != NULL) {
211	u64 end;
212
213	parent = *p;
214	cache = rb_entry(parent, struct dso_cache, rb_node);
215	end = cache->offset + DSO__DATA_CACHE_SIZE;
216
217	if (offset < cache->offset)
218	p = &(*p)->rb_left;
219	else if (offset >= end)
220	p = &(*p)->rb_right;
221	else
222	return cache;
223	}
224	return NULL;
225	}
226
227	static void
228	dso_cache__insert(struct rb_root root, struct dso_cache new)
229	{
230	struct rb_node **p = &root->rb_node;
231	struct rb_node *parent = NULL;
232	struct dso_cache *cache;
233	u64 offset = new->offset;
234
235	while (*p != NULL) {
236	u64 end;
237
238	parent = *p;
239	cache = rb_entry(parent, struct dso_cache, rb_node);
240	end = cache->offset + DSO__DATA_CACHE_SIZE;
241
242	if (offset < cache->offset)
243	p = &(*p)->rb_left;
244	else if (offset >= end)
245	p = &(*p)->rb_right;
246	}
247
248	rb_link_node(&new->rb_node, parent, p);
249	rb_insert_color(&new->rb_node, root);
250	}
251
252	static ssize_t
253	dso_cache__memcpy(struct dso_cache *cache, u64 offset,
254	u8 *data, u64 size)
255	{
256	u64 cache_offset = offset - cache->offset;
257	u64 cache_size = min(cache->size - cache_offset, size);
258
259	memcpy(data, cache->data + cache_offset, cache_size);
260	return cache_size;
261	}
262
263	static ssize_t
264	dso_cache__read(struct dso dso, struct machine machine,
265	u64 offset, u8 *data, ssize_t size)
266	{
267	struct dso_cache *cache;
268	ssize_t ret;
269	int fd;
270
271	fd = dso__data_fd(dso, machine);
272	if (fd < 0)
273	return -1;
274
275	do {
276	u64 cache_offset;
277
278	ret = -ENOMEM;
279
280	cache = zalloc(sizeof(*cache) + DSO__DATA_CACHE_SIZE);
281	if (!cache)
282	break;
283
284	cache_offset = offset & DSO__DATA_CACHE_MASK;
285	ret = -EINVAL;
286
287	if (-1 == lseek(fd, cache_offset, SEEK_SET))
288	break;
289
290	ret = read(fd, cache->data, DSO__DATA_CACHE_SIZE);
291	if (ret <= 0)
292	break;
293
294	cache->offset = cache_offset;
295	cache->size = ret;
296	dso_cache__insert(&dso->cache, cache);
297
298	ret = dso_cache__memcpy(cache, offset, data, size);
299
300	} while (0);
301
302	if (ret <= 0)
303	free(cache);
304
305	close(fd);
306	return ret;
307	}
308
309	static ssize_t dso_cache_read(struct dso dso, struct machine machine,
310	u64 offset, u8 *data, ssize_t size)
311	{
312	struct dso_cache *cache;
313
314	cache = dso_cache__find(&dso->cache, offset);
315	if (cache)
316	return dso_cache__memcpy(cache, offset, data, size);
317	else
318	return dso_cache__read(dso, machine, offset, data, size);
319	}
320
321	ssize_t dso__data_read_offset(struct dso dso, struct machine machine,
322	u64 offset, u8 *data, ssize_t size)
323	{
324	ssize_t r = 0;
325	u8 *p = data;
326
327	do {
328	ssize_t ret;
329
330	ret = dso_cache_read(dso, machine, offset, p, size);
331	if (ret < 0)
332	return ret;
333
334	/* Reached EOF, return what we have. */
335	if (!ret)
336	break;
337
338	BUG_ON(ret > size);
339
340	r += ret;
341	p += ret;
342	offset += ret;
343	size -= ret;
344
345	} while (size);
346
347	return r;
348	}
349
350	ssize_t dso__data_read_addr(struct dso dso, struct map map,
351	struct machine *machine, u64 addr,
352	u8 *data, ssize_t size)
353	{
354	u64 offset = map->map_ip(map, addr);
355	return dso__data_read_offset(dso, machine, offset, data, size);
356	}
357
358	struct map dso__new_map(const char name)
359	{
360	struct map *map = NULL;
361	struct dso *dso = dso__new(name);
362
363	if (dso)
364	map = map__new2(0, dso, MAP__FUNCTION);
365
366	return map;
367	}
368
369	struct dso dso__kernel_findnew(struct machine machine, const char *name,
370	const char *short_name, int dso_type)
371	{
372	/*
373	* The kernel dso could be created by build_id processing.
374	*/
375	struct dso *dso = __dsos__findnew(&machine->kernel_dsos, name);
376
377	/*
378	* We need to run this in all cases, since during the build_id
379	* processing we had no idea this was the kernel dso.
380	*/
381	if (dso != NULL) {
382	dso__set_short_name(dso, short_name);
383	dso->kernel = dso_type;
384	}
385
386	return dso;
387	}
388
389	void dso__set_long_name(struct dso dso, char name)
390	{
391	if (name == NULL)
392	return;
393	dso->long_name = name;
394	dso->long_name_len = strlen(name);
395	}
396
397	void dso__set_short_name(struct dso dso, const char name)
398	{
399	if (name == NULL)
400	return;
401	dso->short_name = name;
402	dso->short_name_len = strlen(name);
403	}
404
405	static void dso__set_basename(struct dso *dso)
406	{
407	dso__set_short_name(dso, basename(dso->long_name));
408	}
409
410	int dso__name_len(const struct dso *dso)
411	{
412	if (!dso)
413	return strlen("[unknown]");
414	if (verbose)
415	return dso->long_name_len;
416
417	return dso->short_name_len;
418	}
419
420	bool dso__loaded(const struct dso *dso, enum map_type type)
421	{
422	return dso->loaded & (1 << type);
423	}
424
425	bool dso__sorted_by_name(const struct dso *dso, enum map_type type)
426	{
427	return dso->sorted_by_name & (1 << type);
428	}
429
430	void dso__set_sorted_by_name(struct dso *dso, enum map_type type)
431	{
432	dso->sorted_by_name \|= (1 << type);
433	}
434
435	struct dso dso__new(const char name)
436	{
437	struct dso dso = calloc(1, sizeof(dso) + strlen(name) + 1);
438
439	if (dso != NULL) {
440	int i;
441	strcpy(dso->name, name);
442	dso__set_long_name(dso, dso->name);
443	dso__set_short_name(dso, dso->name);
444	for (i = 0; i < MAP__NR_TYPES; ++i)
445	dso->symbols[i] = dso->symbol_names[i] = RB_ROOT;
446	dso->cache = RB_ROOT;
447	dso->symtab_type = DSO_BINARY_TYPE__NOT_FOUND;
448	dso->data_type = DSO_BINARY_TYPE__NOT_FOUND;
449	dso->loaded = 0;
0131c4ec	450	dso->rel = 0;
cdd059d7 JO	451	dso->sorted_by_name = 0;
cdd059d7 JO	452	dso->has_build_id = 0;
2cc9d0ef	453	dso->has_srcline = 1;
cdd059d7 JO	454	dso->kernel = DSO_TYPE_USER;
	455	dso->needs_swap = DSO_SWAP__UNSET;
	456	INIT_LIST_HEAD(&dso->node);
	457	}
	458
	459	return dso;
	460	}
	461
	462	void dso__delete(struct dso *dso)
	463	{
	464	int i;
	465	for (i = 0; i < MAP__NR_TYPES; ++i)
	466	symbols__delete(&dso->symbols[i]);
	467	if (dso->sname_alloc)
	468	free((char *)dso->short_name);
	469	if (dso->lname_alloc)
	470	free(dso->long_name);
	471	dso_cache__free(&dso->cache);
	472	free(dso);
	473	}
	474
	475	void dso__set_build_id(struct dso dso, void build_id)
	476	{
	477	memcpy(dso->build_id, build_id, sizeof(dso->build_id));
	478	dso->has_build_id = 1;
	479	}
	480
	481	bool dso__build_id_equal(const struct dso dso, u8 build_id)
	482	{
	483	return memcmp(dso->build_id, build_id, sizeof(dso->build_id)) == 0;
	484	}
	485
	486	void dso__read_running_kernel_build_id(struct dso dso, struct machine machine)
	487	{
	488	char path[PATH_MAX];
	489
	490	if (machine__is_default_guest(machine))
	491	return;
	492	sprintf(path, "%s/sys/kernel/notes", machine->root_dir);
	493	if (sysfs__read_build_id(path, dso->build_id,
	494	sizeof(dso->build_id)) == 0)
	495	dso->has_build_id = true;
	496	}
	497
	498	int dso__kernel_module_get_build_id(struct dso *dso,
	499	const char *root_dir)
	500	{
	501	char filename[PATH_MAX];
	502	/*
	503	* kernel module short names are of the form "[module]" and
	504	* we need just "module" here.
	505	*/
	506	const char *name = dso->short_name + 1;
	507
	508	snprintf(filename, sizeof(filename),
	509	"%s/sys/module/%.*s/notes/.note.gnu.build-id",
	510	root_dir, (int)strlen(name) - 1, name);
	511
	512	if (sysfs__read_build_id(filename, dso->build_id,
	513	sizeof(dso->build_id)) == 0)
	514	dso->has_build_id = true;
	515
	516	return 0;
	517	}
518
519	bool __dsos__read_build_ids(struct list_head *head, bool with_hits)
520	{
521	bool have_build_id = false;
522	struct dso *pos;
523
524	list_for_each_entry(pos, head, node) {
525	if (with_hits && !pos->hit)
526	continue;
527	if (pos->has_build_id) {
528	have_build_id = true;
529	continue;
530	}
531	if (filename__read_build_id(pos->long_name, pos->build_id,
532	sizeof(pos->build_id)) > 0) {
533	have_build_id = true;
534	pos->has_build_id = true;
535	}
536	}
537
538	return have_build_id;
539	}
540
541	void dsos__add(struct list_head head, struct dso dso)
542	{
543	list_add_tail(&dso->node, head);
544	}
545
f9ceffb6	546	struct dso dsos__find(struct list_head head, const char *name, bool cmp_short)
cdd059d7 JO	547	{
	548	struct dso *pos;
	549
f9ceffb6 WL	550	if (cmp_short) {
	551	list_for_each_entry(pos, head, node)
	552	if (strcmp(pos->short_name, name) == 0)
	553	return pos;
	554	return NULL;
	555	}
cdd059d7 JO	556	list_for_each_entry(pos, head, node)
	557	if (strcmp(pos->long_name, name) == 0)
	558	return pos;
	559	return NULL;
	560	}
	561
	562	struct dso __dsos__findnew(struct list_head head, const char *name)
	563	{
f9ceffb6	564	struct dso *dso = dsos__find(head, name, false);
cdd059d7 JO	565
	566	if (!dso) {
	567	dso = dso__new(name);
	568	if (dso != NULL) {
	569	dsos__add(head, dso);
	570	dso__set_basename(dso);
	571	}
	572	}
	573
	574	return dso;
	575	}
	576
	577	size_t __dsos__fprintf_buildid(struct list_head head, FILE fp,
417c2ff6	578	bool (skip)(struct dso *dso, int parm), int parm)
cdd059d7 JO	579	{
	580	struct dso *pos;
	581	size_t ret = 0;
	582
	583	list_for_each_entry(pos, head, node) {
417c2ff6	584	if (skip && skip(pos, parm))
cdd059d7 JO	585	continue;
	586	ret += dso__fprintf_buildid(pos, fp);
	587	ret += fprintf(fp, " %s\n", pos->long_name);
	588	}
	589	return ret;
	590	}
	591
	592	size_t __dsos__fprintf(struct list_head head, FILE fp)
	593	{
	594	struct dso *pos;
	595	size_t ret = 0;
	596
	597	list_for_each_entry(pos, head, node) {
	598	int i;
	599	for (i = 0; i < MAP__NR_TYPES; ++i)
	600	ret += dso__fprintf(pos, i, fp);
	601	}
	602
	603	return ret;
	604	}
	605
	606	size_t dso__fprintf_buildid(struct dso dso, FILE fp)
	607	{
	608	char sbuild_id[BUILD_ID_SIZE * 2 + 1];
	609
	610	build_id__sprintf(dso->build_id, sizeof(dso->build_id), sbuild_id);
	611	return fprintf(fp, "%s", sbuild_id);
	612	}
	613
	614	size_t dso__fprintf(struct dso dso, enum map_type type, FILE fp)
	615	{
	616	struct rb_node *nd;
	617	size_t ret = fprintf(fp, "dso: %s (", dso->short_name);
	618
	619	if (dso->short_name != dso->long_name)
	620	ret += fprintf(fp, "%s, ", dso->long_name);
	621	ret += fprintf(fp, "%s, %sloaded, ", map_type__name[type],
919d590f	622	dso__loaded(dso, type) ? "" : "NOT ");
cdd059d7 JO	623	ret += dso__fprintf_buildid(dso, fp);
	624	ret += fprintf(fp, ")\n");
	625	for (nd = rb_first(&dso->symbols[type]); nd; nd = rb_next(nd)) {
	626	struct symbol *pos = rb_entry(nd, struct symbol, rb_node);
	627	ret += symbol__fprintf(pos, fp);
	628	}
	629
	630	return ret;
	631	}