[deliverable/linux.git] / tools / perf / util / build-id.c

/*
 * build-id.c
 *
 * build-id support
 *
 * Copyright (C) 2009, 2010 Red Hat Inc.
 * Copyright (C) 2009, 2010 Arnaldo Carvalho de Melo <acme@redhat.com>
 */
#include "util.h"
#include <stdio.h>
#include "build-id.h"
#include "event.h"
#include "symbol.h"
#include <linux/kernel.h>
#include "debug.h"
#include "session.h"
#include "tool.h"
#include "header.h"
#include "vdso.h"


static bool no_buildid_cache;

int build_id__mark_dso_hit(struct perf_tool *tool __maybe_unused,
			   union perf_event *event,
			   struct perf_sample *sample,
			   struct perf_evsel *evsel __maybe_unused,
			   struct machine *machine)
{
	struct addr_location al;
	struct thread *thread = machine__findnew_thread(machine, sample->pid,
							sample->tid);

	if (thread == NULL) {
		pr_err("problem processing %d event, skipping it.\n",
			event->header.type);
		return -1;
	}

	thread__find_addr_map(thread, sample->cpumode, MAP__FUNCTION, sample->ip, &al);

	if (al.map != NULL)
		al.map->dso->hit = 1;

	thread__put(thread);
	return 0;
}

static int perf_event__exit_del_thread(struct perf_tool *tool __maybe_unused,
				       union perf_event *event,
				       struct perf_sample *sample
				       __maybe_unused,
				       struct machine *machine)
{
	struct thread *thread = machine__findnew_thread(machine,
							event->fork.pid,
							event->fork.tid);

	dump_printf("(%d:%d):(%d:%d)\n", event->fork.pid, event->fork.tid,
		    event->fork.ppid, event->fork.ptid);

	if (thread) {
		machine__remove_thread(machine, thread);
		thread__put(thread);
	}

	return 0;
}

struct perf_tool build_id__mark_dso_hit_ops = {
	.sample	= build_id__mark_dso_hit,
	.mmap	= perf_event__process_mmap,
	.mmap2	= perf_event__process_mmap2,
	.fork	= perf_event__process_fork,
	.exit	= perf_event__exit_del_thread,
	.attr		 = perf_event__process_attr,
	.build_id	 = perf_event__process_build_id,
	.ordered_events	 = true,
};

int build_id__sprintf(const u8 *build_id, int len, char *bf)
{
	char *bid = bf;
	const u8 *raw = build_id;
	int i;

	for (i = 0; i < len; ++i) {
		sprintf(bid, "%02x", *raw);
		++raw;
		bid += 2;
	}

	return (bid - bf) + 1;
}

int sysfs__sprintf_build_id(const char *root_dir, char *sbuild_id)
{
	char notes[PATH_MAX];
	u8 build_id[BUILD_ID_SIZE];
	int ret;

	if (!root_dir)
		root_dir = "";

	scnprintf(notes, sizeof(notes), "%s/sys/kernel/notes", root_dir);

	ret = sysfs__read_build_id(notes, build_id, sizeof(build_id));
	if (ret < 0)
		return ret;

	return build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
}

int filename__sprintf_build_id(const char *pathname, char *sbuild_id)
{
	u8 build_id[BUILD_ID_SIZE];
	int ret;

	ret = filename__read_build_id(pathname, build_id, sizeof(build_id));
	if (ret < 0)
		return ret;
	else if (ret != sizeof(build_id))
		return -EINVAL;

	return build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
}

/* asnprintf consolidates asprintf and snprintf */
static int asnprintf(char **strp, size_t size, const char *fmt, ...)
{
	va_list ap;
	int ret;

	if (!strp)
		return -EINVAL;

	va_start(ap, fmt);
	if (*strp)
		ret = vsnprintf(*strp, size, fmt, ap);
	else
		ret = vasprintf(strp, fmt, ap);
	va_end(ap);

	return ret;
}

static char *build_id__filename(const char *sbuild_id, char *bf, size_t size)
{
	char *tmp = bf;
	int ret = asnprintf(&bf, size, "%s/.build-id/%.2s/%s", buildid_dir,
			    sbuild_id, sbuild_id + 2);
	if (ret < 0 || (tmp && size < (unsigned int)ret))
		return NULL;
	return bf;
}

char *dso__build_id_filename(const struct dso *dso, char *bf, size_t size)
{
	char build_id_hex[SBUILD_ID_SIZE];

	if (!dso->has_build_id)
		return NULL;

	build_id__sprintf(dso->build_id, sizeof(dso->build_id), build_id_hex);
	return build_id__filename(build_id_hex, bf, size);
}

bool dso__build_id_is_kmod(const struct dso *dso, char *bf, size_t size)
{
	char *id_name, *ch;
	struct stat sb;

	id_name = dso__build_id_filename(dso, bf, size);
	if (!id_name)
		goto err;
	if (access(id_name, F_OK))
		goto err;
	if (lstat(id_name, &sb) == -1)
		goto err;
	if ((size_t)sb.st_size > size - 1)
		goto err;
	if (readlink(id_name, bf, size - 1) < 0)
		goto err;

	bf[sb.st_size] = '\0';

	/*
	 * link should be:
	 * ../../lib/modules/4.4.0-rc4/kernel/net/ipv4/netfilter/nf_nat_ipv4.ko/a09fe3eb3147dafa4e3b31dbd6257e4d696bdc92
	 */
	ch = strrchr(bf, '/');
	if (!ch)
		goto err;
	if (ch - 3 < bf)
		goto err;

	return strncmp(".ko", ch - 3, 3) == 0;
err:
	/*
	 * If dso__build_id_filename work, get id_name again,
	 * because id_name points to bf and is broken.
	 */
	if (id_name)
		id_name = dso__build_id_filename(dso, bf, size);
	pr_err("Invalid build id: %s\n", id_name ? :
					 dso->long_name ? :
					 dso->short_name ? :
					 "[unknown]");
	return false;
}

#define dsos__for_each_with_build_id(pos, head)	\
	list_for_each_entry(pos, head, node)	\
		if (!pos->has_build_id)		\
			continue;		\
		else

static int write_buildid(const char *name, size_t name_len, u8 *build_id,
			 pid_t pid, u16 misc, int fd)
{
	int err;
	struct build_id_event b;
	size_t len;

	len = name_len + 1;
	len = PERF_ALIGN(len, NAME_ALIGN);

	memset(&b, 0, sizeof(b));
	memcpy(&b.build_id, build_id, BUILD_ID_SIZE);
	b.pid = pid;
	b.header.misc = misc;
	b.header.size = sizeof(b) + len;

	err = writen(fd, &b, sizeof(b));
	if (err < 0)
		return err;

	return write_padded(fd, name, name_len + 1, len);
}

static int machine__write_buildid_table(struct machine *machine, int fd)
{
	int err = 0;
	char nm[PATH_MAX];
	struct dso *pos;
	u16 kmisc = PERF_RECORD_MISC_KERNEL,
	    umisc = PERF_RECORD_MISC_USER;

	if (!machine__is_host(machine)) {
		kmisc = PERF_RECORD_MISC_GUEST_KERNEL;
		umisc = PERF_RECORD_MISC_GUEST_USER;
	}

	dsos__for_each_with_build_id(pos, &machine->dsos.head) {
		const char *name;
		size_t name_len;
		bool in_kernel = false;

		if (!pos->hit)
			continue;

		if (dso__is_vdso(pos)) {
			name = pos->short_name;
			name_len = pos->short_name_len + 1;
		} else if (dso__is_kcore(pos)) {
			machine__mmap_name(machine, nm, sizeof(nm));
			name = nm;
			name_len = strlen(nm) + 1;
		} else {
			name = pos->long_name;
			name_len = pos->long_name_len + 1;
		}

		in_kernel = pos->kernel ||
				is_kernel_module(name,
					PERF_RECORD_MISC_CPUMODE_UNKNOWN);
		err = write_buildid(name, name_len, pos->build_id, machine->pid,
				    in_kernel ? kmisc : umisc, fd);
		if (err)
			break;
	}

	return err;
}

int perf_session__write_buildid_table(struct perf_session *session, int fd)
{
	struct rb_node *nd;
	int err = machine__write_buildid_table(&session->machines.host, fd);

	if (err)
		return err;

	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
		struct machine *pos = rb_entry(nd, struct machine, rb_node);
		err = machine__write_buildid_table(pos, fd);
		if (err)
			break;
	}
	return err;
}

static int __dsos__hit_all(struct list_head *head)
{
	struct dso *pos;

	list_for_each_entry(pos, head, node)
		pos->hit = true;

	return 0;
}

static int machine__hit_all_dsos(struct machine *machine)
{
	return __dsos__hit_all(&machine->dsos.head);
}

int dsos__hit_all(struct perf_session *session)
{
	struct rb_node *nd;
	int err;

	err = machine__hit_all_dsos(&session->machines.host);
	if (err)
		return err;

	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
		struct machine *pos = rb_entry(nd, struct machine, rb_node);

		err = machine__hit_all_dsos(pos);
		if (err)
			return err;
	}

	return 0;
}

void disable_buildid_cache(void)
{
	no_buildid_cache = true;
}

static char *build_id_cache__dirname_from_path(const char *name,
					       bool is_kallsyms, bool is_vdso)
{
	char *realname = (char *)name, *filename;
	bool slash = is_kallsyms || is_vdso;

	if (!slash) {
		realname = realpath(name, NULL);
		if (!realname)
			return NULL;
	}

	if (asprintf(&filename, "%s%s%s", buildid_dir, slash ? "/" : "",
		     is_vdso ? DSO__NAME_VDSO : realname) < 0)
		filename = NULL;

	if (!slash)
		free(realname);

	return filename;
}

int build_id_cache__list_build_ids(const char *pathname,
				   struct strlist **result)
{
	struct strlist *list;
	char *dir_name;
	DIR *dir;
	struct dirent *d;
	int ret = 0;

	list = strlist__new(NULL, NULL);
	dir_name = build_id_cache__dirname_from_path(pathname, false, false);
	if (!list || !dir_name) {
		ret = -ENOMEM;
		goto out;
	}

	/* List up all dirents */
	dir = opendir(dir_name);
	if (!dir) {
		ret = -errno;
		goto out;
	}

	while ((d = readdir(dir)) != NULL) {
		if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
			continue;
		strlist__add(list, d->d_name);
	}
	closedir(dir);

out:
	free(dir_name);
	if (ret)
		strlist__delete(list);
	else
		*result = list;

	return ret;
}

int build_id_cache__add_s(const char *sbuild_id, const char *name,
			  bool is_kallsyms, bool is_vdso)
{
	const size_t size = PATH_MAX;
	char *realname = NULL, *filename = NULL, *dir_name = NULL,
	     *linkname = zalloc(size), *targetname, *tmp;
	int err = -1;

	if (!is_kallsyms) {
		realname = realpath(name, NULL);
		if (!realname)
			goto out_free;
	}

	dir_name = build_id_cache__dirname_from_path(name, is_kallsyms, is_vdso);
	if (!dir_name)
		goto out_free;

	if (mkdir_p(dir_name, 0755))
		goto out_free;

	if (asprintf(&filename, "%s/%s", dir_name, sbuild_id) < 0) {
		filename = NULL;
		goto out_free;
	}

	if (access(filename, F_OK)) {
		if (is_kallsyms) {
			 if (copyfile("/proc/kallsyms", filename))
				goto out_free;
		} else if (link(realname, filename) && errno != EEXIST &&
				copyfile(name, filename))
			goto out_free;
	}

	if (!build_id__filename(sbuild_id, linkname, size))
		goto out_free;
	tmp = strrchr(linkname, '/');
	*tmp = '\0';

	if (access(linkname, X_OK) && mkdir_p(linkname, 0755))
		goto out_free;

	*tmp = '/';
	targetname = filename + strlen(buildid_dir) - 5;
	memcpy(targetname, "../..", 5);

	if (symlink(targetname, linkname) == 0)
		err = 0;
out_free:
	if (!is_kallsyms)
		free(realname);
	free(filename);
	free(dir_name);
	free(linkname);
	return err;
}

static int build_id_cache__add_b(const u8 *build_id, size_t build_id_size,
				 const char *name, bool is_kallsyms,
				 bool is_vdso)
{
	char sbuild_id[SBUILD_ID_SIZE];

	build_id__sprintf(build_id, build_id_size, sbuild_id);

	return build_id_cache__add_s(sbuild_id, name, is_kallsyms, is_vdso);
}

bool build_id_cache__cached(const char *sbuild_id)
{
	bool ret = false;
	char *filename = build_id__filename(sbuild_id, NULL, 0);

	if (filename && !access(filename, F_OK))
		ret = true;
	free(filename);

	return ret;
}

int build_id_cache__remove_s(const char *sbuild_id)
{
	const size_t size = PATH_MAX;
	char *filename = zalloc(size),
	     *linkname = zalloc(size), *tmp;
	int err = -1;

	if (filename == NULL || linkname == NULL)
		goto out_free;

	if (!build_id__filename(sbuild_id, linkname, size))
		goto out_free;

	if (access(linkname, F_OK))
		goto out_free;

	if (readlink(linkname, filename, size - 1) < 0)
		goto out_free;

	if (unlink(linkname))
		goto out_free;

	/*
	 * Since the link is relative, we must make it absolute:
	 */
	tmp = strrchr(linkname, '/') + 1;
	snprintf(tmp, size - (tmp - linkname), "%s", filename);

	if (unlink(linkname))
		goto out_free;

	err = 0;
out_free:
	free(filename);
	free(linkname);
	return err;
}

static int dso__cache_build_id(struct dso *dso, struct machine *machine)
{
	bool is_kallsyms = dso->kernel && dso->long_name[0] != '/';
	bool is_vdso = dso__is_vdso(dso);
	const char *name = dso->long_name;
	char nm[PATH_MAX];

	if (dso__is_kcore(dso)) {
		is_kallsyms = true;
		machine__mmap_name(machine, nm, sizeof(nm));
		name = nm;
	}
	return build_id_cache__add_b(dso->build_id, sizeof(dso->build_id), name,
				     is_kallsyms, is_vdso);
}

static int __dsos__cache_build_ids(struct list_head *head,
				   struct machine *machine)
{
	struct dso *pos;
	int err = 0;

	dsos__for_each_with_build_id(pos, head)
		if (dso__cache_build_id(pos, machine))
			err = -1;

	return err;
}

static int machine__cache_build_ids(struct machine *machine)
{
	return __dsos__cache_build_ids(&machine->dsos.head, machine);
}

int perf_session__cache_build_ids(struct perf_session *session)
{
	struct rb_node *nd;
	int ret;

	if (no_buildid_cache)
		return 0;

	if (mkdir(buildid_dir, 0755) != 0 && errno != EEXIST)
		return -1;

	ret = machine__cache_build_ids(&session->machines.host);

	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
		struct machine *pos = rb_entry(nd, struct machine, rb_node);
		ret |= machine__cache_build_ids(pos);
	}
	return ret ? -1 : 0;
}

static bool machine__read_build_ids(struct machine *machine, bool with_hits)
{
	return __dsos__read_build_ids(&machine->dsos.head, with_hits);
}

bool perf_session__read_build_ids(struct perf_session *session, bool with_hits)
{
	struct rb_node *nd;
	bool ret = machine__read_build_ids(&session->machines.host, with_hits);

	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
		struct machine *pos = rb_entry(nd, struct machine, rb_node);
		ret |= machine__read_build_ids(pos, with_hits);
	}

	return ret;
}
Commit	Line	Data
7b2567c1 ACM	1	/*
	2	* build-id.c
	3	*
	4	* build-id support
	5	*
	6	* Copyright (C) 2009, 2010 Red Hat Inc.
	7	* Copyright (C) 2009, 2010 Arnaldo Carvalho de Melo <acme@redhat.com>
	8	*/
b36f19d5 ACM	9	#include "util.h"
b36f19d5 ACM	10	#include <stdio.h>
7b2567c1 ACM	11	#include "build-id.h"
	12	#include "event.h"
	13	#include "symbol.h"
	14	#include <linux/kernel.h>
591765fd	15	#include "debug.h"
d20deb64	16	#include "session.h"
45694aa7	17	#include "tool.h"
e195fac8 NK	18	#include "header.h"
e195fac8 NK	19	#include "vdso.h"
7b2567c1	20
73c5d224 NK	21
	22	static bool no_buildid_cache;
	23
54a3cf59 AV	24	int build_id__mark_dso_hit(struct perf_tool *tool __maybe_unused,
54a3cf59 AV	25	union perf_event *event,
ef89325f	26	struct perf_sample *sample,
54a3cf59 AV	27	struct perf_evsel *evsel __maybe_unused,
54a3cf59 AV	28	struct machine *machine)
7b2567c1 ACM	29	{
7b2567c1 ACM	30	struct addr_location al;
ef89325f	31	struct thread *thread = machine__findnew_thread(machine, sample->pid,
13ce34df	32	sample->tid);
7b2567c1 ACM	33
	34	if (thread == NULL) {
	35	pr_err("problem processing %d event, skipping it.\n",
	36	event->header.type);
	37	return -1;
	38	}
	39
473398a2	40	thread__find_addr_map(thread, sample->cpumode, MAP__FUNCTION, sample->ip, &al);
7b2567c1 ACM	41
	42	if (al.map != NULL)
	43	al.map->dso->hit = 1;
	44
b91fc39f	45	thread__put(thread);
7b2567c1 ACM	46	return 0;
	47	}
	48
1d037ca1	49	static int perf_event__exit_del_thread(struct perf_tool *tool __maybe_unused,
d20deb64	50	union perf_event *event,
1d037ca1 IT	51	struct perf_sample *sample
1d037ca1 IT	52	__maybe_unused,
743eb868	53	struct machine *machine)
591765fd	54	{
314add6b AH	55	struct thread *thread = machine__findnew_thread(machine,
	56	event->fork.pid,
	57	event->fork.tid);
591765fd	58
8115d60c ACM	59	dump_printf("(%d:%d):(%d:%d)\n", event->fork.pid, event->fork.tid,
8115d60c ACM	60	event->fork.ppid, event->fork.ptid);
591765fd	61
b91fc39f	62	if (thread) {
5e78c69b	63	machine__remove_thread(machine, thread);
b91fc39f ACM	64	thread__put(thread);
b91fc39f ACM	65	}
591765fd ACM	66
	67	return 0;
	68	}
	69
45694aa7	70	struct perf_tool build_id__mark_dso_hit_ops = {
7b2567c1	71	.sample = build_id__mark_dso_hit,
8115d60c	72	.mmap = perf_event__process_mmap,
5c5e854b	73	.mmap2 = perf_event__process_mmap2,
f62d3f0f	74	.fork = perf_event__process_fork,
8115d60c	75	.exit = perf_event__exit_del_thread,
299c3452 SE	76	.attr = perf_event__process_attr,
299c3452 SE	77	.build_id = perf_event__process_build_id,
1216b65c	78	.ordered_events = true,
7b2567c1	79	};
b36f19d5	80
ebb296c2 JO	81	int build_id__sprintf(const u8 build_id, int len, char bf)
	82	{
	83	char *bid = bf;
	84	const u8 *raw = build_id;
	85	int i;
	86
	87	for (i = 0; i < len; ++i) {
	88	sprintf(bid, "%02x", *raw);
	89	++raw;
	90	bid += 2;
	91	}
	92
7375e151	93	return (bid - bf) + 1;
ebb296c2 JO	94	}
ebb296c2 JO	95
0b5a7935 MH	96	int sysfs__sprintf_build_id(const char root_dir, char sbuild_id)
	97	{
	98	char notes[PATH_MAX];
	99	u8 build_id[BUILD_ID_SIZE];
	100	int ret;
	101
	102	if (!root_dir)
	103	root_dir = "";
	104
	105	scnprintf(notes, sizeof(notes), "%s/sys/kernel/notes", root_dir);
	106
	107	ret = sysfs__read_build_id(notes, build_id, sizeof(build_id));
	108	if (ret < 0)
	109	return ret;
	110
	111	return build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
	112	}
	113
	114	int filename__sprintf_build_id(const char pathname, char sbuild_id)
	115	{
	116	u8 build_id[BUILD_ID_SIZE];
	117	int ret;
	118
	119	ret = filename__read_build_id(pathname, build_id, sizeof(build_id));
	120	if (ret < 0)
	121	return ret;
	122	else if (ret != sizeof(build_id))
	123	return -EINVAL;
	124
	125	return build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
	126	}
	127
5cb113fd MH	128	/* asnprintf consolidates asprintf and snprintf */
	129	static int asnprintf(char *strp, size_t size, const char fmt, ...)
	130	{
	131	va_list ap;
	132	int ret;
	133
	134	if (!strp)
	135	return -EINVAL;
	136
	137	va_start(ap, fmt);
	138	if (*strp)
	139	ret = vsnprintf(*strp, size, fmt, ap);
	140	else
	141	ret = vasprintf(strp, fmt, ap);
	142	va_end(ap);
	143
	144	return ret;
	145	}
	146
	147	static char build_id__filename(const char sbuild_id, char *bf, size_t size)
	148	{
	149	char *tmp = bf;
	150	int ret = asnprintf(&bf, size, "%s/.build-id/%.2s/%s", buildid_dir,
	151	sbuild_id, sbuild_id + 2);
	152	if (ret < 0 \|\| (tmp && size < (unsigned int)ret))
	153	return NULL;
	154	return bf;
	155	}
	156
3344996e	157	char dso__build_id_filename(const struct dso dso, char *bf, size_t size)
b36f19d5	158	{
d77fac7f	159	char build_id_hex[SBUILD_ID_SIZE];
b36f19d5	160
c824c433	161	if (!dso->has_build_id)
b36f19d5 ACM	162	return NULL;
b36f19d5 ACM	163
c824c433	164	build_id__sprintf(dso->build_id, sizeof(dso->build_id), build_id_hex);
5cb113fd	165	return build_id__filename(build_id_hex, bf, size);
b36f19d5	166	}
e195fac8	167
e7ee4047 WN	168	bool dso__build_id_is_kmod(const struct dso dso, char bf, size_t size)
	169	{
	170	char id_name, ch;
	171	struct stat sb;
	172
	173	id_name = dso__build_id_filename(dso, bf, size);
	174	if (!id_name)
	175	goto err;
	176	if (access(id_name, F_OK))
	177	goto err;
	178	if (lstat(id_name, &sb) == -1)
	179	goto err;
	180	if ((size_t)sb.st_size > size - 1)
	181	goto err;
	182	if (readlink(id_name, bf, size - 1) < 0)
	183	goto err;
	184
	185	bf[sb.st_size] = '\0';
	186
	187	/*
	188	* link should be:
	189	* ../../lib/modules/4.4.0-rc4/kernel/net/ipv4/netfilter/nf_nat_ipv4.ko/a09fe3eb3147dafa4e3b31dbd6257e4d696bdc92
	190	*/
	191	ch = strrchr(bf, '/');
	192	if (!ch)
	193	goto err;
	194	if (ch - 3 < bf)
	195	goto err;
	196
	197	return strncmp(".ko", ch - 3, 3) == 0;
	198	err:
	199	/*
	200	* If dso__build_id_filename work, get id_name again,
	201	* because id_name points to bf and is broken.
	202	*/
	203	if (id_name)
	204	id_name = dso__build_id_filename(dso, bf, size);
	205	pr_err("Invalid build id: %s\n", id_name ? :
	206	dso->long_name ? :
	207	dso->short_name ? :
	208	"[unknown]");
	209	return false;
	210	}
	211
e195fac8 NK	212	#define dsos__for_each_with_build_id(pos, head) \
	213	list_for_each_entry(pos, head, node) \
	214	if (!pos->has_build_id) \
	215	continue; \
	216	else
	217
	218	static int write_buildid(const char name, size_t name_len, u8 build_id,
	219	pid_t pid, u16 misc, int fd)
	220	{
	221	int err;
	222	struct build_id_event b;
	223	size_t len;
	224
	225	len = name_len + 1;
	226	len = PERF_ALIGN(len, NAME_ALIGN);
	227
	228	memset(&b, 0, sizeof(b));
	229	memcpy(&b.build_id, build_id, BUILD_ID_SIZE);
	230	b.pid = pid;
	231	b.header.misc = misc;
	232	b.header.size = sizeof(b) + len;
	233
	234	err = writen(fd, &b, sizeof(b));
	235	if (err < 0)
	236	return err;
	237
	238	return write_padded(fd, name, name_len + 1, len);
	239	}
	240
3d39ac53	241	static int machine__write_buildid_table(struct machine *machine, int fd)
e195fac8	242	{
3d39ac53	243	int err = 0;
e195fac8 NK	244	char nm[PATH_MAX];
e195fac8 NK	245	struct dso *pos;
3d39ac53 ACM	246	u16 kmisc = PERF_RECORD_MISC_KERNEL,
	247	umisc = PERF_RECORD_MISC_USER;
	248
	249	if (!machine__is_host(machine)) {
	250	kmisc = PERF_RECORD_MISC_GUEST_KERNEL;
	251	umisc = PERF_RECORD_MISC_GUEST_USER;
	252	}
e195fac8	253
3d39ac53	254	dsos__for_each_with_build_id(pos, &machine->dsos.head) {
e195fac8 NK	255	const char *name;
e195fac8 NK	256	size_t name_len;
fd786fac	257	bool in_kernel = false;
e195fac8 NK	258
	259	if (!pos->hit)
	260	continue;
	261
	262	if (dso__is_vdso(pos)) {
	263	name = pos->short_name;
	264	name_len = pos->short_name_len + 1;
	265	} else if (dso__is_kcore(pos)) {
	266	machine__mmap_name(machine, nm, sizeof(nm));
	267	name = nm;
	268	name_len = strlen(nm) + 1;
	269	} else {
	270	name = pos->long_name;
	271	name_len = pos->long_name_len + 1;
	272	}
	273
fd786fac WN	274	in_kernel = pos->kernel \|\|
	275	is_kernel_module(name,
	276	PERF_RECORD_MISC_CPUMODE_UNKNOWN);
3d39ac53	277	err = write_buildid(name, name_len, pos->build_id, machine->pid,
fd786fac	278	in_kernel ? kmisc : umisc, fd);
e195fac8	279	if (err)
3d39ac53	280	break;
e195fac8 NK	281	}
e195fac8 NK	282
e195fac8 NK	283	return err;
	284	}
	285
	286	int perf_session__write_buildid_table(struct perf_session *session, int fd)
	287	{
	288	struct rb_node *nd;
	289	int err = machine__write_buildid_table(&session->machines.host, fd);
	290
	291	if (err)
	292	return err;
	293
	294	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
	295	struct machine *pos = rb_entry(nd, struct machine, rb_node);
	296	err = machine__write_buildid_table(pos, fd);
	297	if (err)
	298	break;
	299	}
	300	return err;
	301	}
	302
	303	static int __dsos__hit_all(struct list_head *head)
	304	{
	305	struct dso *pos;
	306
	307	list_for_each_entry(pos, head, node)
	308	pos->hit = true;
	309
	310	return 0;
	311	}
	312
	313	static int machine__hit_all_dsos(struct machine *machine)
	314	{
3d39ac53	315	return __dsos__hit_all(&machine->dsos.head);
e195fac8 NK	316	}
	317
	318	int dsos__hit_all(struct perf_session *session)
	319	{
	320	struct rb_node *nd;
	321	int err;
	322
	323	err = machine__hit_all_dsos(&session->machines.host);
	324	if (err)
	325	return err;
	326
	327	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
	328	struct machine *pos = rb_entry(nd, struct machine, rb_node);
	329
	330	err = machine__hit_all_dsos(pos);
	331	if (err)
	332	return err;
	333	}
	334
	335	return 0;
	336	}
	337
73c5d224 NK	338	void disable_buildid_cache(void)
	339	{
	340	no_buildid_cache = true;
	341	}
	342
8d8c8e4c MH	343	static char build_id_cache__dirname_from_path(const char name,
	344	bool is_kallsyms, bool is_vdso)
	345	{
	346	char realname = (char )name, *filename;
	347	bool slash = is_kallsyms \|\| is_vdso;
	348
	349	if (!slash) {
	350	realname = realpath(name, NULL);
	351	if (!realname)
	352	return NULL;
	353	}
	354
	355	if (asprintf(&filename, "%s%s%s", buildid_dir, slash ? "/" : "",
	356	is_vdso ? DSO__NAME_VDSO : realname) < 0)
	357	filename = NULL;
	358
	359	if (!slash)
	360	free(realname);
	361
	362	return filename;
	363	}
	364
	365	int build_id_cache__list_build_ids(const char *pathname,
	366	struct strlist **result)
	367	{
	368	struct strlist *list;
	369	char *dir_name;
	370	DIR *dir;
	371	struct dirent *d;
	372	int ret = 0;
	373
4a77e218	374	list = strlist__new(NULL, NULL);
8d8c8e4c MH	375	dir_name = build_id_cache__dirname_from_path(pathname, false, false);
	376	if (!list \|\| !dir_name) {
	377	ret = -ENOMEM;
	378	goto out;
	379	}
	380
	381	/* List up all dirents */
	382	dir = opendir(dir_name);
	383	if (!dir) {
	384	ret = -errno;
	385	goto out;
	386	}
	387
	388	while ((d = readdir(dir)) != NULL) {
	389	if (!strcmp(d->d_name, ".") \|\| !strcmp(d->d_name, ".."))
	390	continue;
	391	strlist__add(list, d->d_name);
	392	}
	393	closedir(dir);
	394
	395	out:
	396	free(dir_name);
	397	if (ret)
	398	strlist__delete(list);
	399	else
	400	*result = list;
	401
	402	return ret;
	403	}
	404
e35f7362 MH	405	int build_id_cache__add_s(const char sbuild_id, const char name,
e35f7362 MH	406	bool is_kallsyms, bool is_vdso)
e195fac8 NK	407	{
e195fac8 NK	408	const size_t size = PATH_MAX;
8d8c8e4c	409	char realname = NULL, filename = NULL, *dir_name = NULL,
5cb113fd	410	linkname = zalloc(size), targetname, *tmp;
8d8c8e4c	411	int err = -1;
e195fac8	412
8d8c8e4c	413	if (!is_kallsyms) {
e195fac8	414	realname = realpath(name, NULL);
8d8c8e4c MH	415	if (!realname)
	416	goto out_free;
	417	}
e195fac8	418
8d8c8e4c MH	419	dir_name = build_id_cache__dirname_from_path(name, is_kallsyms, is_vdso);
8d8c8e4c MH	420	if (!dir_name)
e195fac8 NK	421	goto out_free;
e195fac8 NK	422
8d8c8e4c	423	if (mkdir_p(dir_name, 0755))
e195fac8 NK	424	goto out_free;
e195fac8 NK	425
8d8c8e4c MH	426	if (asprintf(&filename, "%s/%s", dir_name, sbuild_id) < 0) {
	427	filename = NULL;
	428	goto out_free;
	429	}
e195fac8 NK	430
	431	if (access(filename, F_OK)) {
	432	if (is_kallsyms) {
	433	if (copyfile("/proc/kallsyms", filename))
	434	goto out_free;
0635b0f7 MV	435	} else if (link(realname, filename) && errno != EEXIST &&
0635b0f7 MV	436	copyfile(name, filename))
e195fac8 NK	437	goto out_free;
	438	}
	439
5cb113fd MH	440	if (!build_id__filename(sbuild_id, linkname, size))
	441	goto out_free;
	442	tmp = strrchr(linkname, '/');
	443	*tmp = '\0';
e195fac8 NK	444
	445	if (access(linkname, X_OK) && mkdir_p(linkname, 0755))
	446	goto out_free;
	447
5cb113fd	448	*tmp = '/';
e35f7362	449	targetname = filename + strlen(buildid_dir) - 5;
e195fac8 NK	450	memcpy(targetname, "../..", 5);
	451
	452	if (symlink(targetname, linkname) == 0)
	453	err = 0;
	454	out_free:
	455	if (!is_kallsyms)
	456	free(realname);
	457	free(filename);
8d8c8e4c	458	free(dir_name);
e195fac8 NK	459	free(linkname);
	460	return err;
	461	}
	462
	463	static int build_id_cache__add_b(const u8 *build_id, size_t build_id_size,
e35f7362 MH	464	const char *name, bool is_kallsyms,
e35f7362 MH	465	bool is_vdso)
e195fac8	466	{
d77fac7f	467	char sbuild_id[SBUILD_ID_SIZE];
e195fac8 NK	468
	469	build_id__sprintf(build_id, build_id_size, sbuild_id);
	470
e35f7362	471	return build_id_cache__add_s(sbuild_id, name, is_kallsyms, is_vdso);
e195fac8 NK	472	}
e195fac8 NK	473
a50d11a1 MH	474	bool build_id_cache__cached(const char *sbuild_id)
	475	{
	476	bool ret = false;
	477	char *filename = build_id__filename(sbuild_id, NULL, 0);
	478
	479	if (filename && !access(filename, F_OK))
	480	ret = true;
	481	free(filename);
	482
	483	return ret;
	484	}
	485
e35f7362	486	int build_id_cache__remove_s(const char *sbuild_id)
e195fac8 NK	487	{
	488	const size_t size = PATH_MAX;
	489	char *filename = zalloc(size),
5cb113fd	490	linkname = zalloc(size), tmp;
e195fac8 NK	491	int err = -1;
	492
	493	if (filename == NULL \|\| linkname == NULL)
	494	goto out_free;
	495
5cb113fd MH	496	if (!build_id__filename(sbuild_id, linkname, size))
5cb113fd MH	497	goto out_free;
e195fac8 NK	498
	499	if (access(linkname, F_OK))
	500	goto out_free;
	501
	502	if (readlink(linkname, filename, size - 1) < 0)
	503	goto out_free;
	504
	505	if (unlink(linkname))
	506	goto out_free;
	507
	508	/*
	509	* Since the link is relative, we must make it absolute:
	510	*/
5cb113fd MH	511	tmp = strrchr(linkname, '/') + 1;
5cb113fd MH	512	snprintf(tmp, size - (tmp - linkname), "%s", filename);
e195fac8 NK	513
	514	if (unlink(linkname))
	515	goto out_free;
	516
	517	err = 0;
	518	out_free:
	519	free(filename);
	520	free(linkname);
	521	return err;
	522	}
	523
e35f7362	524	static int dso__cache_build_id(struct dso dso, struct machine machine)
e195fac8 NK	525	{
	526	bool is_kallsyms = dso->kernel && dso->long_name[0] != '/';
	527	bool is_vdso = dso__is_vdso(dso);
	528	const char *name = dso->long_name;
	529	char nm[PATH_MAX];
	530
	531	if (dso__is_kcore(dso)) {
	532	is_kallsyms = true;
	533	machine__mmap_name(machine, nm, sizeof(nm));
	534	name = nm;
	535	}
	536	return build_id_cache__add_b(dso->build_id, sizeof(dso->build_id), name,
e35f7362	537	is_kallsyms, is_vdso);
e195fac8 NK	538	}
	539
	540	static int __dsos__cache_build_ids(struct list_head *head,
e35f7362	541	struct machine *machine)
e195fac8 NK	542	{
	543	struct dso *pos;
	544	int err = 0;
	545
	546	dsos__for_each_with_build_id(pos, head)
e35f7362	547	if (dso__cache_build_id(pos, machine))
e195fac8 NK	548	err = -1;
	549
	550	return err;
	551	}
	552
e35f7362	553	static int machine__cache_build_ids(struct machine *machine)
e195fac8	554	{
3d39ac53	555	return __dsos__cache_build_ids(&machine->dsos.head, machine);
e195fac8 NK	556	}
	557
	558	int perf_session__cache_build_ids(struct perf_session *session)
	559	{
	560	struct rb_node *nd;
	561	int ret;
e195fac8	562
73c5d224 NK	563	if (no_buildid_cache)
	564	return 0;
	565
498922ad	566	if (mkdir(buildid_dir, 0755) != 0 && errno != EEXIST)
e195fac8 NK	567	return -1;
e195fac8 NK	568
e35f7362	569	ret = machine__cache_build_ids(&session->machines.host);
e195fac8 NK	570
	571	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
	572	struct machine *pos = rb_entry(nd, struct machine, rb_node);
e35f7362	573	ret \|= machine__cache_build_ids(pos);
e195fac8 NK	574	}
	575	return ret ? -1 : 0;
	576	}
	577
	578	static bool machine__read_build_ids(struct machine *machine, bool with_hits)
	579	{
3d39ac53	580	return __dsos__read_build_ids(&machine->dsos.head, with_hits);
e195fac8 NK	581	}
	582
	583	bool perf_session__read_build_ids(struct perf_session *session, bool with_hits)
	584	{
	585	struct rb_node *nd;
	586	bool ret = machine__read_build_ids(&session->machines.host, with_hits);
	587
	588	for (nd = rb_first(&session->machines.guests); nd; nd = rb_next(nd)) {
	589	struct machine *pos = rb_entry(nd, struct machine, rb_node);
	590	ret \|= machine__read_build_ids(pos, with_hits);
	591	}
	592
	593	return ret;
	594	}