[linux-2.6-block.git] / tools / perf / builtin-buildid-cache.c

/*
 * builtin-buildid-cache.c
 *
 * Builtin buildid-cache command: Manages build-id cache
 *
 * Copyright (C) 2010, Red Hat Inc.
 * Copyright (C) 2010, Arnaldo Carvalho de Melo <acme@redhat.com>
 */
#include <sys/types.h>
#include <sys/time.h>
#include <time.h>
#include <dirent.h>
#include <errno.h>
#include <unistd.h>
#include "builtin.h"
#include "perf.h"
#include "namespaces.h"
#include "util/cache.h"
#include "util/debug.h"
#include "util/header.h"
#include <subcmd/parse-options.h>
#include "util/strlist.h"
#include "util/build-id.h"
#include "util/session.h"
#include "util/symbol.h"
#include "util/time-utils.h"

static int build_id_cache__kcore_buildid(const char *proc_dir, char *sbuildid)
{
	char root_dir[PATH_MAX];
	char *p;

	strlcpy(root_dir, proc_dir, sizeof(root_dir));

	p = strrchr(root_dir, '/');
	if (!p)
		return -1;
	*p = '\0';
	return sysfs__sprintf_build_id(root_dir, sbuildid);
}

static int build_id_cache__kcore_dir(char *dir, size_t sz)
{
	return fetch_current_timestamp(dir, sz);
}

static bool same_kallsyms_reloc(const char *from_dir, char *to_dir)
{
	char from[PATH_MAX];
	char to[PATH_MAX];
	const char *name;
	u64 addr1 = 0, addr2 = 0;
	int i, err = -1;

	scnprintf(from, sizeof(from), "%s/kallsyms", from_dir);
	scnprintf(to, sizeof(to), "%s/kallsyms", to_dir);

	for (i = 0; (name = ref_reloc_sym_names[i]) != NULL; i++) {
		err = kallsyms__get_function_start(from, name, &addr1);
		if (!err)
			break;
	}

	if (err)
		return false;

	if (kallsyms__get_function_start(to, name, &addr2))
		return false;

	return addr1 == addr2;
}

static int build_id_cache__kcore_existing(const char *from_dir, char *to_dir,
					  size_t to_dir_sz)
{
	char from[PATH_MAX];
	char to[PATH_MAX];
	char to_subdir[PATH_MAX];
	struct dirent *dent;
	int ret = -1;
	DIR *d;

	d = opendir(to_dir);
	if (!d)
		return -1;

	scnprintf(from, sizeof(from), "%s/modules", from_dir);

	while (1) {
		dent = readdir(d);
		if (!dent)
			break;
		if (dent->d_type != DT_DIR)
			continue;
		scnprintf(to, sizeof(to), "%s/%s/modules", to_dir,
			  dent->d_name);
		scnprintf(to_subdir, sizeof(to_subdir), "%s/%s",
			  to_dir, dent->d_name);
		if (!compare_proc_modules(from, to) &&
		    same_kallsyms_reloc(from_dir, to_subdir)) {
			strlcpy(to_dir, to_subdir, to_dir_sz);
			ret = 0;
			break;
		}
	}

	closedir(d);

	return ret;
}

static int build_id_cache__add_kcore(const char *filename, bool force)
{
	char dir[32], sbuildid[SBUILD_ID_SIZE];
	char from_dir[PATH_MAX], to_dir[PATH_MAX];
	char *p;

	strlcpy(from_dir, filename, sizeof(from_dir));

	p = strrchr(from_dir, '/');
	if (!p || strcmp(p + 1, "kcore"))
		return -1;
	*p = '\0';

	if (build_id_cache__kcore_buildid(from_dir, sbuildid) < 0)
		return -1;

	scnprintf(to_dir, sizeof(to_dir), "%s/%s/%s",
		  buildid_dir, DSO__NAME_KCORE, sbuildid);

	if (!force &&
	    !build_id_cache__kcore_existing(from_dir, to_dir, sizeof(to_dir))) {
		pr_debug("same kcore found in %s\n", to_dir);
		return 0;
	}

	if (build_id_cache__kcore_dir(dir, sizeof(dir)))
		return -1;

	scnprintf(to_dir, sizeof(to_dir), "%s/%s/%s/%s",
		  buildid_dir, DSO__NAME_KCORE, sbuildid, dir);

	if (mkdir_p(to_dir, 0755))
		return -1;

	if (kcore_copy(from_dir, to_dir)) {
		/* Remove YYYYmmddHHMMSShh directory */
		if (!rmdir(to_dir)) {
			p = strrchr(to_dir, '/');
			if (p)
				*p = '\0';
			/* Try to remove buildid directory */
			if (!rmdir(to_dir)) {
				p = strrchr(to_dir, '/');
				if (p)
					*p = '\0';
				/* Try to remove [kernel.kcore] directory */
				rmdir(to_dir);
			}
		}
		return -1;
	}

	pr_debug("kcore added to build-id cache directory %s\n", to_dir);

	return 0;
}

static int build_id_cache__add_file(const char *filename, struct nsinfo *nsi)
{
	char sbuild_id[SBUILD_ID_SIZE];
	u8 build_id[BUILD_ID_SIZE];
	int err;
	struct nscookie nsc;

	nsinfo__mountns_enter(nsi, &nsc);
	err = filename__read_build_id(filename, &build_id, sizeof(build_id));
	nsinfo__mountns_exit(&nsc);
	if (err < 0) {
		pr_debug("Couldn't read a build-id in %s\n", filename);
		return -1;
	}

	build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
	err = build_id_cache__add_s(sbuild_id, filename, nsi,
				    false, false);
	pr_debug("Adding %s %s: %s\n", sbuild_id, filename,
		 err ? "FAIL" : "Ok");
	return err;
}

static int build_id_cache__remove_file(const char *filename, struct nsinfo *nsi)
{
	u8 build_id[BUILD_ID_SIZE];
	char sbuild_id[SBUILD_ID_SIZE];
	struct nscookie nsc;

	int err;

	nsinfo__mountns_enter(nsi, &nsc);
	err = filename__read_build_id(filename, &build_id, sizeof(build_id));
	nsinfo__mountns_exit(&nsc);
	if (err < 0) {
		pr_debug("Couldn't read a build-id in %s\n", filename);
		return -1;
	}

	build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
	err = build_id_cache__remove_s(sbuild_id);
	pr_debug("Removing %s %s: %s\n", sbuild_id, filename,
		 err ? "FAIL" : "Ok");

	return err;
}

static int build_id_cache__purge_path(const char *pathname, struct nsinfo *nsi)
{
	struct strlist *list;
	struct str_node *pos;
	int err;

	err = build_id_cache__list_build_ids(pathname, nsi, &list);
	if (err)
		goto out;

	strlist__for_each_entry(pos, list) {
		err = build_id_cache__remove_s(pos->s);
		pr_debug("Removing %s %s: %s\n", pos->s, pathname,
			 err ? "FAIL" : "Ok");
		if (err)
			break;
	}
	strlist__delete(list);

out:
	pr_debug("Purging %s: %s\n", pathname, err ? "FAIL" : "Ok");

	return err;
}

static bool dso__missing_buildid_cache(struct dso *dso, int parm __maybe_unused)
{
	char filename[PATH_MAX];
	u8 build_id[BUILD_ID_SIZE];

	if (dso__build_id_filename(dso, filename, sizeof(filename), false) &&
	    filename__read_build_id(filename, build_id,
				    sizeof(build_id)) != sizeof(build_id)) {
		if (errno == ENOENT)
			return false;

		pr_warning("Problems with %s file, consider removing it from the cache\n",
			   filename);
	} else if (memcmp(dso->build_id, build_id, sizeof(dso->build_id))) {
		pr_warning("Problems with %s file, consider removing it from the cache\n",
			   filename);
	}

	return true;
}

static int build_id_cache__fprintf_missing(struct perf_session *session, FILE *fp)
{
	perf_session__fprintf_dsos_buildid(session, fp, dso__missing_buildid_cache, 0);
	return 0;
}

static int build_id_cache__update_file(const char *filename, struct nsinfo *nsi)
{
	u8 build_id[BUILD_ID_SIZE];
	char sbuild_id[SBUILD_ID_SIZE];
	struct nscookie nsc;

	int err;

	nsinfo__mountns_enter(nsi, &nsc);
	err = filename__read_build_id(filename, &build_id, sizeof(build_id));
	nsinfo__mountns_exit(&nsc);
	if (err < 0) {
		pr_debug("Couldn't read a build-id in %s\n", filename);
		return -1;
	}
	err = 0;

	build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
	if (build_id_cache__cached(sbuild_id))
		err = build_id_cache__remove_s(sbuild_id);

	if (!err)
		err = build_id_cache__add_s(sbuild_id, filename, nsi, false,
					    false);

	pr_debug("Updating %s %s: %s\n", sbuild_id, filename,
		 err ? "FAIL" : "Ok");

	return err;
}

int cmd_buildid_cache(int argc, const char **argv)
{
	struct strlist *list;
	struct str_node *pos;
	int ret = 0;
	int ns_id = -1;
	bool force = false;
	char const *add_name_list_str = NULL,
		   *remove_name_list_str = NULL,
		   *purge_name_list_str = NULL,
		   *missing_filename = NULL,
		   *update_name_list_str = NULL,
		   *kcore_filename = NULL;
	char sbuf[STRERR_BUFSIZE];

	struct perf_data data = {
		.mode  = PERF_DATA_MODE_READ,
	};
	struct perf_session *session = NULL;
	struct nsinfo *nsi = NULL;

	const struct option buildid_cache_options[] = {
	OPT_STRING('a', "add", &add_name_list_str,
		   "file list", "file(s) to add"),
	OPT_STRING('k', "kcore", &kcore_filename,
		   "file", "kcore file to add"),
	OPT_STRING('r', "remove", &remove_name_list_str, "file list",
		    "file(s) to remove"),
	OPT_STRING('p', "purge", &purge_name_list_str, "path list",
		    "path(s) to remove (remove old caches too)"),
	OPT_STRING('M', "missing", &missing_filename, "file",
		   "to find missing build ids in the cache"),
	OPT_BOOLEAN('f', "force", &force, "don't complain, do it"),
	OPT_STRING('u', "update", &update_name_list_str, "file list",
		    "file(s) to update"),
	OPT_INCR('v', "verbose", &verbose, "be more verbose"),
	OPT_INTEGER(0, "target-ns", &ns_id, "target pid for namespace context"),
	OPT_END()
	};
	const char * const buildid_cache_usage[] = {
		"perf buildid-cache [<options>]",
		NULL
	};

	argc = parse_options(argc, argv, buildid_cache_options,
			     buildid_cache_usage, 0);

	if (argc || (!add_name_list_str && !kcore_filename &&
		     !remove_name_list_str && !purge_name_list_str &&
		     !missing_filename && !update_name_list_str))
		usage_with_options(buildid_cache_usage, buildid_cache_options);

	if (ns_id > 0)
		nsi = nsinfo__new(ns_id);

	if (missing_filename) {
		data.path = missing_filename;
		data.force = force;

		session = perf_session__new(&data, false, NULL);
		if (session == NULL)
			return -1;
	}

	if (symbol__init(session ? &session->header.env : NULL) < 0)
		goto out;

	setup_pager();

	if (add_name_list_str) {
		list = strlist__new(add_name_list_str, NULL);
		if (list) {
			strlist__for_each_entry(pos, list)
				if (build_id_cache__add_file(pos->s, nsi)) {
					if (errno == EEXIST) {
						pr_debug("%s already in the cache\n",
							 pos->s);
						continue;
					}
					pr_warning("Couldn't add %s: %s\n",
						   pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
				}

			strlist__delete(list);
		}
	}

	if (remove_name_list_str) {
		list = strlist__new(remove_name_list_str, NULL);
		if (list) {
			strlist__for_each_entry(pos, list)
				if (build_id_cache__remove_file(pos->s, nsi)) {
					if (errno == ENOENT) {
						pr_debug("%s wasn't in the cache\n",
							 pos->s);
						continue;
					}
					pr_warning("Couldn't remove %s: %s\n",
						   pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
				}

			strlist__delete(list);
		}
	}

	if (purge_name_list_str) {
		list = strlist__new(purge_name_list_str, NULL);
		if (list) {
			strlist__for_each_entry(pos, list)
				if (build_id_cache__purge_path(pos->s, nsi)) {
					if (errno == ENOENT) {
						pr_debug("%s wasn't in the cache\n",
							 pos->s);
						continue;
					}
					pr_warning("Couldn't remove %s: %s\n",
						   pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
				}

			strlist__delete(list);
		}
	}

	if (missing_filename)
		ret = build_id_cache__fprintf_missing(session, stdout);

	if (update_name_list_str) {
		list = strlist__new(update_name_list_str, NULL);
		if (list) {
			strlist__for_each_entry(pos, list)
				if (build_id_cache__update_file(pos->s, nsi)) {
					if (errno == ENOENT) {
						pr_debug("%s wasn't in the cache\n",
							 pos->s);
						continue;
					}
					pr_warning("Couldn't update %s: %s\n",
						   pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
				}

			strlist__delete(list);
		}
	}

	if (kcore_filename && build_id_cache__add_kcore(kcore_filename, force))
		pr_warning("Couldn't add %s\n", kcore_filename);

out:
	perf_session__delete(session);
	nsinfo__zput(nsi);

	return ret;
}
Commit	Line	Data
ef12a141 ACM	1	/*
	2	* builtin-buildid-cache.c
	3	*
	4	* Builtin buildid-cache command: Manages build-id cache
	5	*
	6	* Copyright (C) 2010, Red Hat Inc.
	7	* Copyright (C) 2010, Arnaldo Carvalho de Melo <acme@redhat.com>
	8	*/
fc1b691d AH	9	#include <sys/types.h>
	10	#include <sys/time.h>
	11	#include <time.h>
	12	#include <dirent.h>
a43783ae	13	#include <errno.h>
fc1b691d	14	#include <unistd.h>
ef12a141 ACM	15	#include "builtin.h"
ef12a141 ACM	16	#include "perf.h"
f045b8c4	17	#include "namespaces.h"
ef12a141 ACM	18	#include "util/cache.h"
	19	#include "util/debug.h"
	20	#include "util/header.h"
4b6ab94e	21	#include <subcmd/parse-options.h>
ef12a141	22	#include "util/strlist.h"
ebb296c2	23	#include "util/build-id.h"
fbb6976c	24	#include "util/session.h"
4383db88	25	#include "util/symbol.h"
c5e4027e	26	#include "util/time-utils.h"
ef12a141	27
fc1b691d AH	28	static int build_id_cache__kcore_buildid(const char proc_dir, char sbuildid)
	29	{
	30	char root_dir[PATH_MAX];
fc1b691d AH	31	char *p;
	32
	33	strlcpy(root_dir, proc_dir, sizeof(root_dir));
	34
	35	p = strrchr(root_dir, '/');
	36	if (!p)
	37	return -1;
	38	*p = '\0';
0b5a7935	39	return sysfs__sprintf_build_id(root_dir, sbuildid);
fc1b691d AH	40	}
	41
	42	static int build_id_cache__kcore_dir(char *dir, size_t sz)
	43	{
37b20151	44	return fetch_current_timestamp(dir, sz);
fc1b691d AH	45	}
fc1b691d AH	46
d3b70220 AH	47	static bool same_kallsyms_reloc(const char from_dir, char to_dir)
	48	{
	49	char from[PATH_MAX];
	50	char to[PATH_MAX];
	51	const char *name;
	52	u64 addr1 = 0, addr2 = 0;
b843f62a	53	int i, err = -1;
d3b70220 AH	54
	55	scnprintf(from, sizeof(from), "%s/kallsyms", from_dir);
	56	scnprintf(to, sizeof(to), "%s/kallsyms", to_dir);
	57
	58	for (i = 0; (name = ref_reloc_sym_names[i]) != NULL; i++) {
b843f62a ACM	59	err = kallsyms__get_function_start(from, name, &addr1);
b843f62a ACM	60	if (!err)
d3b70220 AH	61	break;
	62	}
	63
b843f62a ACM	64	if (err)
	65	return false;
	66
	67	if (kallsyms__get_function_start(to, name, &addr2))
	68	return false;
d3b70220 AH	69
	70	return addr1 == addr2;
	71	}
	72
fc1b691d AH	73	static int build_id_cache__kcore_existing(const char from_dir, char to_dir,
	74	size_t to_dir_sz)
	75	{
	76	char from[PATH_MAX];
	77	char to[PATH_MAX];
d3b70220	78	char to_subdir[PATH_MAX];
fc1b691d AH	79	struct dirent *dent;
	80	int ret = -1;
	81	DIR *d;
	82
	83	d = opendir(to_dir);
	84	if (!d)
	85	return -1;
	86
	87	scnprintf(from, sizeof(from), "%s/modules", from_dir);
	88
	89	while (1) {
	90	dent = readdir(d);
	91	if (!dent)
	92	break;
	93	if (dent->d_type != DT_DIR)
	94	continue;
	95	scnprintf(to, sizeof(to), "%s/%s/modules", to_dir,
	96	dent->d_name);
d3b70220 AH	97	scnprintf(to_subdir, sizeof(to_subdir), "%s/%s",
	98	to_dir, dent->d_name);
	99	if (!compare_proc_modules(from, to) &&
	100	same_kallsyms_reloc(from_dir, to_subdir)) {
	101	strlcpy(to_dir, to_subdir, to_dir_sz);
fc1b691d AH	102	ret = 0;
	103	break;
	104	}
	105	}
	106
	107	closedir(d);
	108
	109	return ret;
	110	}
	111
e35f7362	112	static int build_id_cache__add_kcore(const char *filename, bool force)
fc1b691d	113	{
d77fac7f	114	char dir[32], sbuildid[SBUILD_ID_SIZE];
fc1b691d AH	115	char from_dir[PATH_MAX], to_dir[PATH_MAX];
	116	char *p;
	117
	118	strlcpy(from_dir, filename, sizeof(from_dir));
	119
	120	p = strrchr(from_dir, '/');
	121	if (!p \|\| strcmp(p + 1, "kcore"))
	122	return -1;
	123	*p = '\0';
	124
0b5a7935	125	if (build_id_cache__kcore_buildid(from_dir, sbuildid) < 0)
fc1b691d AH	126	return -1;
fc1b691d AH	127
0a77582f MH	128	scnprintf(to_dir, sizeof(to_dir), "%s/%s/%s",
0a77582f MH	129	buildid_dir, DSO__NAME_KCORE, sbuildid);
fc1b691d	130
5173fbb8 AH	131	if (!force &&
5173fbb8 AH	132	!build_id_cache__kcore_existing(from_dir, to_dir, sizeof(to_dir))) {
fc1b691d AH	133	pr_debug("same kcore found in %s\n", to_dir);
	134	return 0;
	135	}
	136
	137	if (build_id_cache__kcore_dir(dir, sizeof(dir)))
	138	return -1;
	139
0a77582f MH	140	scnprintf(to_dir, sizeof(to_dir), "%s/%s/%s/%s",
0a77582f MH	141	buildid_dir, DSO__NAME_KCORE, sbuildid, dir);
fc1b691d AH	142
	143	if (mkdir_p(to_dir, 0755))
	144	return -1;
	145
	146	if (kcore_copy(from_dir, to_dir)) {
	147	/* Remove YYYYmmddHHMMSShh directory */
	148	if (!rmdir(to_dir)) {
	149	p = strrchr(to_dir, '/');
	150	if (p)
	151	*p = '\0';
	152	/* Try to remove buildid directory */
	153	if (!rmdir(to_dir)) {
	154	p = strrchr(to_dir, '/');
	155	if (p)
	156	*p = '\0';
	157	/* Try to remove [kernel.kcore] directory */
	158	rmdir(to_dir);
	159	}
	160	}
	161	return -1;
	162	}
	163
	164	pr_debug("kcore added to build-id cache directory %s\n", to_dir);
	165
	166	return 0;
	167	}
	168
f045b8c4	169	static int build_id_cache__add_file(const char filename, struct nsinfo nsi)
ef12a141	170	{
d77fac7f	171	char sbuild_id[SBUILD_ID_SIZE];
ef12a141 ACM	172	u8 build_id[BUILD_ID_SIZE];
ef12a141 ACM	173	int err;
f045b8c4	174	struct nscookie nsc;
ef12a141	175
f045b8c4 KJ	176	nsinfo__mountns_enter(nsi, &nsc);
	177	err = filename__read_build_id(filename, &build_id, sizeof(build_id));
	178	nsinfo__mountns_exit(&nsc);
	179	if (err < 0) {
ef12a141 ACM	180	pr_debug("Couldn't read a build-id in %s\n", filename);
	181	return -1;
	182	}
	183
	184	build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
f045b8c4	185	err = build_id_cache__add_s(sbuild_id, filename, nsi,
7dbf4dcf	186	false, false);
cc169c7c MH	187	pr_debug("Adding %s %s: %s\n", sbuild_id, filename,
cc169c7c MH	188	err ? "FAIL" : "Ok");
ef12a141 ACM	189	return err;
	190	}
	191
f045b8c4	192	static int build_id_cache__remove_file(const char filename, struct nsinfo nsi)
ef12a141 ACM	193	{
ef12a141 ACM	194	u8 build_id[BUILD_ID_SIZE];
d77fac7f	195	char sbuild_id[SBUILD_ID_SIZE];
f045b8c4	196	struct nscookie nsc;
ef12a141 ACM	197
	198	int err;
	199
f045b8c4 KJ	200	nsinfo__mountns_enter(nsi, &nsc);
	201	err = filename__read_build_id(filename, &build_id, sizeof(build_id));
	202	nsinfo__mountns_exit(&nsc);
	203	if (err < 0) {
ef12a141 ACM	204	pr_debug("Couldn't read a build-id in %s\n", filename);
	205	return -1;
	206	}
	207
	208	build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
e35f7362	209	err = build_id_cache__remove_s(sbuild_id);
cc169c7c MH	210	pr_debug("Removing %s %s: %s\n", sbuild_id, filename,
cc169c7c MH	211	err ? "FAIL" : "Ok");
ef12a141 ACM	212
	213	return err;
	214	}
	215
f045b8c4	216	static int build_id_cache__purge_path(const char pathname, struct nsinfo nsi)
8d8c8e4c MH	217	{
	218	struct strlist *list;
	219	struct str_node *pos;
	220	int err;
	221
f045b8c4	222	err = build_id_cache__list_build_ids(pathname, nsi, &list);
8d8c8e4c MH	223	if (err)
	224	goto out;
	225
602a1f4d	226	strlist__for_each_entry(pos, list) {
8d8c8e4c	227	err = build_id_cache__remove_s(pos->s);
cc169c7c MH	228	pr_debug("Removing %s %s: %s\n", pos->s, pathname,
cc169c7c MH	229	err ? "FAIL" : "Ok");
8d8c8e4c MH	230	if (err)
	231	break;
	232	}
	233	strlist__delete(list);
	234
	235	out:
cc169c7c	236	pr_debug("Purging %s: %s\n", pathname, err ? "FAIL" : "Ok");
8d8c8e4c MH	237
	238	return err;
	239	}
	240
fbb6976c ACM	241	static bool dso__missing_buildid_cache(struct dso *dso, int parm __maybe_unused)
	242	{
	243	char filename[PATH_MAX];
	244	u8 build_id[BUILD_ID_SIZE];
	245
d2396999	246	if (dso__build_id_filename(dso, filename, sizeof(filename), false) &&
fbb6976c ACM	247	filename__read_build_id(filename, build_id,
	248	sizeof(build_id)) != sizeof(build_id)) {
	249	if (errno == ENOENT)
	250	return false;
	251
48000a1a	252	pr_warning("Problems with %s file, consider removing it from the cache\n",
fbb6976c ACM	253	filename);
fbb6976c ACM	254	} else if (memcmp(dso->build_id, build_id, sizeof(dso->build_id))) {
48000a1a	255	pr_warning("Problems with %s file, consider removing it from the cache\n",
fbb6976c ACM	256	filename);
	257	}
	258
	259	return true;
	260	}
	261
e3ed75bb	262	static int build_id_cache__fprintf_missing(struct perf_session session, FILE fp)
fbb6976c	263	{
fbb6976c	264	perf_session__fprintf_dsos_buildid(session, fp, dso__missing_buildid_cache, 0);
fbb6976c ACM	265	return 0;
	266	}
	267
f045b8c4	268	static int build_id_cache__update_file(const char filename, struct nsinfo nsi)
eeb49845 NK	269	{
eeb49845 NK	270	u8 build_id[BUILD_ID_SIZE];
d77fac7f	271	char sbuild_id[SBUILD_ID_SIZE];
f045b8c4	272	struct nscookie nsc;
eeb49845	273
f045b8c4	274	int err;
eeb49845	275
f045b8c4 KJ	276	nsinfo__mountns_enter(nsi, &nsc);
	277	err = filename__read_build_id(filename, &build_id, sizeof(build_id));
	278	nsinfo__mountns_exit(&nsc);
	279	if (err < 0) {
eeb49845 NK	280	pr_debug("Couldn't read a build-id in %s\n", filename);
	281	return -1;
	282	}
f045b8c4	283	err = 0;
eeb49845 NK	284
eeb49845 NK	285	build_id__sprintf(build_id, sizeof(build_id), sbuild_id);
a50d11a1 MH	286	if (build_id_cache__cached(sbuild_id))
	287	err = build_id_cache__remove_s(sbuild_id);
	288
e35f7362	289	if (!err)
f045b8c4 KJ	290	err = build_id_cache__add_s(sbuild_id, filename, nsi, false,
f045b8c4 KJ	291	false);
e35f7362	292
cc169c7c MH	293	pr_debug("Updating %s %s: %s\n", sbuild_id, filename,
cc169c7c MH	294	err ? "FAIL" : "Ok");
eeb49845 NK	295
	296	return err;
	297	}
	298
b0ad8ea6	299	int cmd_buildid_cache(int argc, const char **argv)
ef12a141 ACM	300	{
	301	struct strlist *list;
	302	struct str_node *pos;
fbb6976c	303	int ret = 0;
f045b8c4	304	int ns_id = -1;
fbb6976c	305	bool force = false;
472cc83c	306	char const *add_name_list_str = NULL,
fbb6976c	307	*remove_name_list_str = NULL,
8d8c8e4c	308	*purge_name_list_str = NULL,
eeb49845	309	*missing_filename = NULL,
fc1b691d	310	*update_name_list_str = NULL,
eec5a688	311	*kcore_filename = NULL;
340481ad	312	char sbuf[STRERR_BUFSIZE];
eeb49845	313
8ceb41d7	314	struct perf_data data = {
e3ed75bb NK	315	.mode = PERF_DATA_MODE_READ,
	316	};
	317	struct perf_session *session = NULL;
f045b8c4	318	struct nsinfo *nsi = NULL;
e3ed75bb	319
472cc83c ACM	320	const struct option buildid_cache_options[] = {
	321	OPT_STRING('a', "add", &add_name_list_str,
	322	"file list", "file(s) to add"),
fc1b691d AH	323	OPT_STRING('k', "kcore", &kcore_filename,
fc1b691d AH	324	"file", "kcore file to add"),
472cc83c ACM	325	OPT_STRING('r', "remove", &remove_name_list_str, "file list",
472cc83c ACM	326	"file(s) to remove"),
8d8c8e4c MH	327	OPT_STRING('p', "purge", &purge_name_list_str, "path list",
8d8c8e4c MH	328	"path(s) to remove (remove old caches too)"),
fbb6976c ACM	329	OPT_STRING('M', "missing", &missing_filename, "file",
	330	"to find missing build ids in the cache"),
	331	OPT_BOOLEAN('f', "force", &force, "don't complain, do it"),
eeb49845 NK	332	OPT_STRING('u', "update", &update_name_list_str, "file list",
eeb49845 NK	333	"file(s) to update"),
472cc83c	334	OPT_INCR('v', "verbose", &verbose, "be more verbose"),
f045b8c4	335	OPT_INTEGER(0, "target-ns", &ns_id, "target pid for namespace context"),
472cc83c ACM	336	OPT_END()
	337	};
	338	const char * const buildid_cache_usage[] = {
	339	"perf buildid-cache [<options>]",
	340	NULL
	341	};
	342
	343	argc = parse_options(argc, argv, buildid_cache_options,
	344	buildid_cache_usage, 0);
	345
0497d0a8 MH	346	if (argc \|\| (!add_name_list_str && !kcore_filename &&
	347	!remove_name_list_str && !purge_name_list_str &&
	348	!missing_filename && !update_name_list_str))
	349	usage_with_options(buildid_cache_usage, buildid_cache_options);
	350
f045b8c4 KJ	351	if (ns_id > 0)
	352	nsi = nsinfo__new(ns_id);
	353
e3ed75bb	354	if (missing_filename) {
8ceb41d7 JO	355	data.path = missing_filename;
8ceb41d7 JO	356	data.force = force;
e3ed75bb	357
8ceb41d7	358	session = perf_session__new(&data, false, NULL);
e3ed75bb NK	359	if (session == NULL)
	360	return -1;
	361	}
	362
0a7e6d1b	363	if (symbol__init(session ? &session->header.env : NULL) < 0)
e3ed75bb	364	goto out;
472cc83c ACM	365
472cc83c ACM	366	setup_pager();
ef12a141	367
ef12a141	368	if (add_name_list_str) {
4a77e218	369	list = strlist__new(add_name_list_str, NULL);
ef12a141	370	if (list) {
602a1f4d	371	strlist__for_each_entry(pos, list)
f045b8c4	372	if (build_id_cache__add_file(pos->s, nsi)) {
ef12a141 ACM	373	if (errno == EEXIST) {
	374	pr_debug("%s already in the cache\n",
	375	pos->s);
	376	continue;
	377	}
	378	pr_warning("Couldn't add %s: %s\n",
c8b5f2c9	379	pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
ef12a141 ACM	380	}
	381
	382	strlist__delete(list);
	383	}
	384	}
	385
	386	if (remove_name_list_str) {
4a77e218	387	list = strlist__new(remove_name_list_str, NULL);
ef12a141	388	if (list) {
602a1f4d	389	strlist__for_each_entry(pos, list)
f045b8c4	390	if (build_id_cache__remove_file(pos->s, nsi)) {
ef12a141 ACM	391	if (errno == ENOENT) {
	392	pr_debug("%s wasn't in the cache\n",
	393	pos->s);
	394	continue;
	395	}
	396	pr_warning("Couldn't remove %s: %s\n",
c8b5f2c9	397	pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
ef12a141 ACM	398	}
	399
	400	strlist__delete(list);
	401	}
	402	}
	403
8d8c8e4c	404	if (purge_name_list_str) {
4a77e218	405	list = strlist__new(purge_name_list_str, NULL);
8d8c8e4c	406	if (list) {
602a1f4d	407	strlist__for_each_entry(pos, list)
f045b8c4	408	if (build_id_cache__purge_path(pos->s, nsi)) {
8d8c8e4c MH	409	if (errno == ENOENT) {
	410	pr_debug("%s wasn't in the cache\n",
	411	pos->s);
	412	continue;
	413	}
	414	pr_warning("Couldn't remove %s: %s\n",
c8b5f2c9	415	pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
8d8c8e4c MH	416	}
	417
	418	strlist__delete(list);
	419	}
	420	}
	421
fbb6976c	422	if (missing_filename)
e3ed75bb	423	ret = build_id_cache__fprintf_missing(session, stdout);
fbb6976c	424
eeb49845	425	if (update_name_list_str) {
4a77e218	426	list = strlist__new(update_name_list_str, NULL);
eeb49845	427	if (list) {
602a1f4d	428	strlist__for_each_entry(pos, list)
f045b8c4	429	if (build_id_cache__update_file(pos->s, nsi)) {
eeb49845 NK	430	if (errno == ENOENT) {
	431	pr_debug("%s wasn't in the cache\n",
	432	pos->s);
	433	continue;
	434	}
	435	pr_warning("Couldn't update %s: %s\n",
c8b5f2c9	436	pos->s, str_error_r(errno, sbuf, sizeof(sbuf)));
eeb49845 NK	437	}
	438
	439	strlist__delete(list);
	440	}
	441	}
	442
e35f7362	443	if (kcore_filename && build_id_cache__add_kcore(kcore_filename, force))
fc1b691d AH	444	pr_warning("Couldn't add %s\n", kcore_filename);
fc1b691d AH	445
e3ed75bb	446	out:
e1446551	447	perf_session__delete(session);
f045b8c4	448	nsinfo__zput(nsi);
e3ed75bb	449
fbb6976c	450	return ret;
ef12a141	451	}