[linux-2.6-block.git] / tools / perf / util / thread_map.c

#include <dirent.h>
#include <limits.h>
#include <stdbool.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include "strlist.h"
#include <string.h>
#include <api/fs/fs.h>
#include "asm/bug.h"
#include "thread_map.h"
#include "util.h"
#include "debug.h"

/* Skip "." and ".." directories */
static int filter(const struct dirent *dir)
{
	if (dir->d_name[0] == '.')
		return 0;
	else
		return 1;
}

static void thread_map__reset(struct thread_map *map, int start, int nr)
{
	size_t size = (nr - start) * sizeof(map->map[0]);

	memset(&map->map[start], 0, size);
}

static struct thread_map *thread_map__realloc(struct thread_map *map, int nr)
{
	size_t size = sizeof(*map) + sizeof(map->map[0]) * nr;
	int start = map ? map->nr : 0;

	map = realloc(map, size);
	/*
	 * We only realloc to add more items, let's reset new items.
	 */
	if (map)
		thread_map__reset(map, start, nr);

	return map;
}

#define thread_map__alloc(__nr) thread_map__realloc(NULL, __nr)

struct thread_map *thread_map__new_by_pid(pid_t pid)
{
	struct thread_map *threads;
	char name[256];
	int items;
	struct dirent **namelist = NULL;
	int i;

	sprintf(name, "/proc/%d/task", pid);
	items = scandir(name, &namelist, filter, NULL);
	if (items <= 0)
		return NULL;

	threads = thread_map__alloc(items);
	if (threads != NULL) {
		for (i = 0; i < items; i++)
			thread_map__set_pid(threads, i, atoi(namelist[i]->d_name));
		threads->nr = items;
		atomic_set(&threads->refcnt, 1);
	}

	for (i=0; i<items; i++)
		zfree(&namelist[i]);
	free(namelist);

	return threads;
}

struct thread_map *thread_map__new_by_tid(pid_t tid)
{
	struct thread_map *threads = thread_map__alloc(1);

	if (threads != NULL) {
		thread_map__set_pid(threads, 0, tid);
		threads->nr = 1;
		atomic_set(&threads->refcnt, 1);
	}

	return threads;
}

struct thread_map *thread_map__new_by_uid(uid_t uid)
{
	DIR *proc;
	int max_threads = 32, items, i;
	char path[256];
	struct dirent dirent, *next, **namelist = NULL;
	struct thread_map *threads = thread_map__alloc(max_threads);

	if (threads == NULL)
		goto out;

	proc = opendir("/proc");
	if (proc == NULL)
		goto out_free_threads;

	threads->nr = 0;
	atomic_set(&threads->refcnt, 1);

	while (!readdir_r(proc, &dirent, &next) && next) {
		char *end;
		bool grow = false;
		struct stat st;
		pid_t pid = strtol(dirent.d_name, &end, 10);

		if (*end) /* only interested in proper numerical dirents */
			continue;

		snprintf(path, sizeof(path), "/proc/%s", dirent.d_name);

		if (stat(path, &st) != 0)
			continue;

		if (st.st_uid != uid)
			continue;

		snprintf(path, sizeof(path), "/proc/%d/task", pid);
		items = scandir(path, &namelist, filter, NULL);
		if (items <= 0)
			goto out_free_closedir;

		while (threads->nr + items >= max_threads) {
			max_threads *= 2;
			grow = true;
		}

		if (grow) {
			struct thread_map *tmp;

			tmp = realloc(threads, (sizeof(*threads) +
						max_threads * sizeof(pid_t)));
			if (tmp == NULL)
				goto out_free_namelist;

			threads = tmp;
		}

		for (i = 0; i < items; i++) {
			thread_map__set_pid(threads, threads->nr + i,
					    atoi(namelist[i]->d_name));
		}

		for (i = 0; i < items; i++)
			zfree(&namelist[i]);
		free(namelist);

		threads->nr += items;
	}

out_closedir:
	closedir(proc);
out:
	return threads;

out_free_threads:
	free(threads);
	return NULL;

out_free_namelist:
	for (i = 0; i < items; i++)
		zfree(&namelist[i]);
	free(namelist);

out_free_closedir:
	zfree(&threads);
	goto out_closedir;
}

struct thread_map *thread_map__new(pid_t pid, pid_t tid, uid_t uid)
{
	if (pid != -1)
		return thread_map__new_by_pid(pid);

	if (tid == -1 && uid != UINT_MAX)
		return thread_map__new_by_uid(uid);

	return thread_map__new_by_tid(tid);
}

static struct thread_map *thread_map__new_by_pid_str(const char *pid_str)
{
	struct thread_map *threads = NULL, *nt;
	char name[256];
	int items, total_tasks = 0;
	struct dirent **namelist = NULL;
	int i, j = 0;
	pid_t pid, prev_pid = INT_MAX;
	char *end_ptr;
	struct str_node *pos;
	struct strlist *slist = strlist__new(false, pid_str);

	if (!slist)
		return NULL;

	strlist__for_each(pos, slist) {
		pid = strtol(pos->s, &end_ptr, 10);

		if (pid == INT_MIN || pid == INT_MAX ||
		    (*end_ptr != '\0' && *end_ptr != ','))
			goto out_free_threads;

		if (pid == prev_pid)
			continue;

		sprintf(name, "/proc/%d/task", pid);
		items = scandir(name, &namelist, filter, NULL);
		if (items <= 0)
			goto out_free_threads;

		total_tasks += items;
		nt = thread_map__realloc(threads, total_tasks);
		if (nt == NULL)
			goto out_free_namelist;

		threads = nt;

		for (i = 0; i < items; i++) {
			thread_map__set_pid(threads, j++, atoi(namelist[i]->d_name));
			zfree(&namelist[i]);
		}
		threads->nr = total_tasks;
		free(namelist);
	}

out:
	strlist__delete(slist);
	if (threads)
		atomic_set(&threads->refcnt, 1);
	return threads;

out_free_namelist:
	for (i = 0; i < items; i++)
		zfree(&namelist[i]);
	free(namelist);

out_free_threads:
	zfree(&threads);
	goto out;
}

struct thread_map *thread_map__new_dummy(void)
{
	struct thread_map *threads = thread_map__alloc(1);

	if (threads != NULL) {
		thread_map__set_pid(threads, 0, -1);
		threads->nr = 1;
		atomic_set(&threads->refcnt, 1);
	}
	return threads;
}

static struct thread_map *thread_map__new_by_tid_str(const char *tid_str)
{
	struct thread_map *threads = NULL, *nt;
	int ntasks = 0;
	pid_t tid, prev_tid = INT_MAX;
	char *end_ptr;
	struct str_node *pos;
	struct strlist *slist;

	/* perf-stat expects threads to be generated even if tid not given */
	if (!tid_str)
		return thread_map__new_dummy();

	slist = strlist__new(false, tid_str);
	if (!slist)
		return NULL;

	strlist__for_each(pos, slist) {
		tid = strtol(pos->s, &end_ptr, 10);

		if (tid == INT_MIN || tid == INT_MAX ||
		    (*end_ptr != '\0' && *end_ptr != ','))
			goto out_free_threads;

		if (tid == prev_tid)
			continue;

		ntasks++;
		nt = thread_map__realloc(threads, ntasks);

		if (nt == NULL)
			goto out_free_threads;

		threads = nt;
		thread_map__set_pid(threads, ntasks - 1, tid);
		threads->nr = ntasks;
	}
out:
	if (threads)
		atomic_set(&threads->refcnt, 1);
	return threads;

out_free_threads:
	zfree(&threads);
	goto out;
}

struct thread_map *thread_map__new_str(const char *pid, const char *tid,
				       uid_t uid)
{
	if (pid)
		return thread_map__new_by_pid_str(pid);

	if (!tid && uid != UINT_MAX)
		return thread_map__new_by_uid(uid);

	return thread_map__new_by_tid_str(tid);
}

static void thread_map__delete(struct thread_map *threads)
{
	if (threads) {
		int i;

		WARN_ONCE(atomic_read(&threads->refcnt) != 0,
			  "thread map refcnt unbalanced\n");
		for (i = 0; i < threads->nr; i++)
			free(thread_map__comm(threads, i));
		free(threads);
	}
}

struct thread_map *thread_map__get(struct thread_map *map)
{
	if (map)
		atomic_inc(&map->refcnt);
	return map;
}

void thread_map__put(struct thread_map *map)
{
	if (map && atomic_dec_and_test(&map->refcnt))
		thread_map__delete(map);
}

size_t thread_map__fprintf(struct thread_map *threads, FILE *fp)
{
	int i;
	size_t printed = fprintf(fp, "%d thread%s: ",
				 threads->nr, threads->nr > 1 ? "s" : "");
	for (i = 0; i < threads->nr; ++i)
		printed += fprintf(fp, "%s%d", i ? ", " : "", thread_map__pid(threads, i));

	return printed + fprintf(fp, "\n");
}

static int get_comm(char **comm, pid_t pid)
{
	char *path;
	size_t size;
	int err;

	if (asprintf(&path, "%s/%d/comm", procfs__mountpoint(), pid) == -1)
		return -ENOMEM;

	err = filename__read_str(path, comm, &size);
	if (!err) {
		/*
		 * We're reading 16 bytes, while filename__read_str
		 * allocates data per BUFSIZ bytes, so we can safely
		 * mark the end of the string.
		 */
		(*comm)[size] = 0;
		rtrim(*comm);
	}

	free(path);
	return err;
}

static void comm_init(struct thread_map *map, int i)
{
	pid_t pid = thread_map__pid(map, i);
	char *comm = NULL;

	/* dummy pid comm initialization */
	if (pid == -1) {
		map->map[i].comm = strdup("dummy");
		return;
	}

	/*
	 * The comm name is like extra bonus ;-),
	 * so just warn if we fail for any reason.
	 */
	if (get_comm(&comm, pid))
		pr_warning("Couldn't resolve comm name for pid %d\n", pid);

	map->map[i].comm = comm;
}

void thread_map__read_comms(struct thread_map *threads)
{
	int i;

	for (i = 0; i < threads->nr; ++i)
		comm_init(threads, i);
}
Commit	Line	Data
fd78260b	1	#include <dirent.h>
0d37aa34 ACM	2	#include <limits.h>
0d37aa34 ACM	3	#include <stdbool.h>
fd78260b ACM	4	#include <stdlib.h>
fd78260b ACM	5	#include <stdio.h>
0d37aa34 ACM	6	#include <sys/types.h>
	7	#include <sys/stat.h>
	8	#include <unistd.h>
b52956c9 DA	9	#include "strlist.h"
b52956c9 DA	10	#include <string.h>
792402fd	11	#include <api/fs/fs.h>
186fbb74	12	#include "asm/bug.h"
fd78260b	13	#include "thread_map.h"
04662523	14	#include "util.h"
792402fd	15	#include "debug.h"
fd78260b ACM	16
	17	/* Skip "." and ".." directories */
	18	static int filter(const struct dirent *dir)
	19	{
	20	if (dir->d_name[0] == '.')
	21	return 0;
	22	else
	23	return 1;
	24	}
	25
62eea464 JO	26	static void thread_map__reset(struct thread_map *map, int start, int nr)
	27	{
	28	size_t size = (nr - start) * sizeof(map->map[0]);
	29
	30	memset(&map->map[start], 0, size);
	31	}
	32
9d7e8c3a JO	33	static struct thread_map thread_map__realloc(struct thread_map map, int nr)
9d7e8c3a JO	34	{
060664f3	35	size_t size = sizeof(map) + sizeof(map->map[0]) nr;
62eea464 JO	36	int start = map ? map->nr : 0;
	37
	38	map = realloc(map, size);
	39	/*
	40	* We only realloc to add more items, let's reset new items.
	41	*/
	42	if (map)
	43	thread_map__reset(map, start, nr);
9d7e8c3a	44
62eea464	45	return map;
9d7e8c3a JO	46	}
	47
	48	#define thread_map__alloc(__nr) thread_map__realloc(NULL, __nr)
	49
fd78260b ACM	50	struct thread_map *thread_map__new_by_pid(pid_t pid)
	51	{
	52	struct thread_map *threads;
	53	char name[256];
	54	int items;
	55	struct dirent **namelist = NULL;
	56	int i;
	57
	58	sprintf(name, "/proc/%d/task", pid);
	59	items = scandir(name, &namelist, filter, NULL);
	60	if (items <= 0)
0d37aa34	61	return NULL;
fd78260b	62
9d7e8c3a	63	threads = thread_map__alloc(items);
fd78260b ACM	64	if (threads != NULL) {
fd78260b ACM	65	for (i = 0; i < items; i++)
e13798c7	66	thread_map__set_pid(threads, i, atoi(namelist[i]->d_name));
fd78260b	67	threads->nr = items;
186fbb74	68	atomic_set(&threads->refcnt, 1);
fd78260b ACM	69	}
	70
	71	for (i=0; i<items; i++)
74cf249d	72	zfree(&namelist[i]);
fd78260b ACM	73	free(namelist);
	74
	75	return threads;
	76	}
	77
	78	struct thread_map *thread_map__new_by_tid(pid_t tid)
	79	{
9d7e8c3a	80	struct thread_map *threads = thread_map__alloc(1);
fd78260b ACM	81
fd78260b ACM	82	if (threads != NULL) {
e13798c7 JO	83	thread_map__set_pid(threads, 0, tid);
e13798c7 JO	84	threads->nr = 1;
186fbb74	85	atomic_set(&threads->refcnt, 1);
fd78260b ACM	86	}
	87
	88	return threads;
	89	}
	90
0d37aa34 ACM	91	struct thread_map *thread_map__new_by_uid(uid_t uid)
	92	{
	93	DIR *proc;
	94	int max_threads = 32, items, i;
	95	char path[256];
	96	struct dirent dirent, next, *namelist = NULL;
9d7e8c3a JO	97	struct thread_map *threads = thread_map__alloc(max_threads);
9d7e8c3a JO	98
0d37aa34 ACM	99	if (threads == NULL)
	100	goto out;
	101
	102	proc = opendir("/proc");
	103	if (proc == NULL)
	104	goto out_free_threads;
	105
	106	threads->nr = 0;
186fbb74	107	atomic_set(&threads->refcnt, 1);
0d37aa34 ACM	108
	109	while (!readdir_r(proc, &dirent, &next) && next) {
	110	char *end;
	111	bool grow = false;
	112	struct stat st;
	113	pid_t pid = strtol(dirent.d_name, &end, 10);
	114
	115	if (end) / only interested in proper numerical dirents */
	116	continue;
	117
	118	snprintf(path, sizeof(path), "/proc/%s", dirent.d_name);
	119
	120	if (stat(path, &st) != 0)
	121	continue;
	122
	123	if (st.st_uid != uid)
	124	continue;
	125
	126	snprintf(path, sizeof(path), "/proc/%d/task", pid);
	127	items = scandir(path, &namelist, filter, NULL);
	128	if (items <= 0)
	129	goto out_free_closedir;
	130
	131	while (threads->nr + items >= max_threads) {
	132	max_threads *= 2;
	133	grow = true;
	134	}
	135
	136	if (grow) {
	137	struct thread_map *tmp;
	138
	139	tmp = realloc(threads, (sizeof(*threads) +
	140	max_threads * sizeof(pid_t)));
	141	if (tmp == NULL)
	142	goto out_free_namelist;
	143
	144	threads = tmp;
	145	}
	146
e13798c7 JO	147	for (i = 0; i < items; i++) {
	148	thread_map__set_pid(threads, threads->nr + i,
	149	atoi(namelist[i]->d_name));
	150	}
0d37aa34 ACM	151
0d37aa34 ACM	152	for (i = 0; i < items; i++)
74cf249d	153	zfree(&namelist[i]);
0d37aa34 ACM	154	free(namelist);
	155
	156	threads->nr += items;
	157	}
	158
	159	out_closedir:
	160	closedir(proc);
	161	out:
	162	return threads;
	163
	164	out_free_threads:
	165	free(threads);
	166	return NULL;
	167
	168	out_free_namelist:
	169	for (i = 0; i < items; i++)
74cf249d	170	zfree(&namelist[i]);
0d37aa34 ACM	171	free(namelist);
	172
	173	out_free_closedir:
04662523	174	zfree(&threads);
0d37aa34 ACM	175	goto out_closedir;
	176	}
	177
	178	struct thread_map *thread_map__new(pid_t pid, pid_t tid, uid_t uid)
fd78260b ACM	179	{
	180	if (pid != -1)
	181	return thread_map__new_by_pid(pid);
0d37aa34 ACM	182
	183	if (tid == -1 && uid != UINT_MAX)
	184	return thread_map__new_by_uid(uid);
	185
fd78260b ACM	186	return thread_map__new_by_tid(tid);
	187	}
	188
b52956c9 DA	189	static struct thread_map thread_map__new_by_pid_str(const char pid_str)
	190	{
	191	struct thread_map threads = NULL, nt;
	192	char name[256];
	193	int items, total_tasks = 0;
	194	struct dirent **namelist = NULL;
	195	int i, j = 0;
	196	pid_t pid, prev_pid = INT_MAX;
	197	char *end_ptr;
	198	struct str_node *pos;
	199	struct strlist *slist = strlist__new(false, pid_str);
	200
	201	if (!slist)
	202	return NULL;
	203
	204	strlist__for_each(pos, slist) {
	205	pid = strtol(pos->s, &end_ptr, 10);
	206
	207	if (pid == INT_MIN \|\| pid == INT_MAX \|\|
	208	(end_ptr != '\0' && end_ptr != ','))
	209	goto out_free_threads;
	210
	211	if (pid == prev_pid)
	212	continue;
	213
	214	sprintf(name, "/proc/%d/task", pid);
	215	items = scandir(name, &namelist, filter, NULL);
	216	if (items <= 0)
	217	goto out_free_threads;
	218
	219	total_tasks += items;
9d7e8c3a	220	nt = thread_map__realloc(threads, total_tasks);
b52956c9	221	if (nt == NULL)
e8cdd947	222	goto out_free_namelist;
b52956c9 DA	223
	224	threads = nt;
	225
e8cdd947	226	for (i = 0; i < items; i++) {
e13798c7	227	thread_map__set_pid(threads, j++, atoi(namelist[i]->d_name));
74cf249d	228	zfree(&namelist[i]);
e8cdd947 FBH	229	}
e8cdd947 FBH	230	threads->nr = total_tasks;
b52956c9	231	free(namelist);
b52956c9 DA	232	}
	233
	234	out:
	235	strlist__delete(slist);
186fbb74 JO	236	if (threads)
186fbb74 JO	237	atomic_set(&threads->refcnt, 1);
b52956c9 DA	238	return threads;
b52956c9 DA	239
e8cdd947 FBH	240	out_free_namelist:
e8cdd947 FBH	241	for (i = 0; i < items; i++)
74cf249d	242	zfree(&namelist[i]);
e8cdd947 FBH	243	free(namelist);
e8cdd947 FBH	244
b52956c9	245	out_free_threads:
04662523	246	zfree(&threads);
b52956c9 DA	247	goto out;
	248	}
	249
641556c9 ACM	250	struct thread_map *thread_map__new_dummy(void)
641556c9 ACM	251	{
9d7e8c3a	252	struct thread_map *threads = thread_map__alloc(1);
641556c9 ACM	253
641556c9 ACM	254	if (threads != NULL) {
e13798c7 JO	255	thread_map__set_pid(threads, 0, -1);
e13798c7 JO	256	threads->nr = 1;
186fbb74	257	atomic_set(&threads->refcnt, 1);
641556c9 ACM	258	}
	259	return threads;
	260	}
	261
b52956c9 DA	262	static struct thread_map thread_map__new_by_tid_str(const char tid_str)
	263	{
	264	struct thread_map threads = NULL, nt;
	265	int ntasks = 0;
	266	pid_t tid, prev_tid = INT_MAX;
	267	char *end_ptr;
	268	struct str_node *pos;
	269	struct strlist *slist;
	270
	271	/* perf-stat expects threads to be generated even if tid not given */
641556c9 ACM	272	if (!tid_str)
641556c9 ACM	273	return thread_map__new_dummy();
b52956c9 DA	274
	275	slist = strlist__new(false, tid_str);
	276	if (!slist)
	277	return NULL;
	278
	279	strlist__for_each(pos, slist) {
	280	tid = strtol(pos->s, &end_ptr, 10);
	281
	282	if (tid == INT_MIN \|\| tid == INT_MAX \|\|
	283	(end_ptr != '\0' && end_ptr != ','))
	284	goto out_free_threads;
	285
	286	if (tid == prev_tid)
	287	continue;
	288
	289	ntasks++;
9d7e8c3a	290	nt = thread_map__realloc(threads, ntasks);
b52956c9 DA	291
	292	if (nt == NULL)
	293	goto out_free_threads;
	294
	295	threads = nt;
e13798c7 JO	296	thread_map__set_pid(threads, ntasks - 1, tid);
e13798c7 JO	297	threads->nr = ntasks;
b52956c9 DA	298	}
b52956c9 DA	299	out:
186fbb74 JO	300	if (threads)
186fbb74 JO	301	atomic_set(&threads->refcnt, 1);
b52956c9 DA	302	return threads;
	303
	304	out_free_threads:
04662523	305	zfree(&threads);
b52956c9 DA	306	goto out;
	307	}
	308
	309	struct thread_map thread_map__new_str(const char pid, const char *tid,
	310	uid_t uid)
	311	{
	312	if (pid)
	313	return thread_map__new_by_pid_str(pid);
	314
	315	if (!tid && uid != UINT_MAX)
	316	return thread_map__new_by_uid(uid);
	317
	318	return thread_map__new_by_tid_str(tid);
	319	}
	320
186fbb74	321	static void thread_map__delete(struct thread_map *threads)
fd78260b	322	{
186fbb74	323	if (threads) {
792402fd JO	324	int i;
792402fd JO	325
186fbb74 JO	326	WARN_ONCE(atomic_read(&threads->refcnt) != 0,
186fbb74 JO	327	"thread map refcnt unbalanced\n");
792402fd JO	328	for (i = 0; i < threads->nr; i++)
792402fd JO	329	free(thread_map__comm(threads, i));
186fbb74 JO	330	free(threads);
	331	}
	332	}
	333
	334	struct thread_map thread_map__get(struct thread_map map)
	335	{
	336	if (map)
	337	atomic_inc(&map->refcnt);
	338	return map;
	339	}
	340
	341	void thread_map__put(struct thread_map *map)
	342	{
	343	if (map && atomic_dec_and_test(&map->refcnt))
	344	thread_map__delete(map);
fd78260b	345	}
9ae7d335 ACM	346
	347	size_t thread_map__fprintf(struct thread_map threads, FILE fp)
	348	{
	349	int i;
	350	size_t printed = fprintf(fp, "%d thread%s: ",
	351	threads->nr, threads->nr > 1 ? "s" : "");
	352	for (i = 0; i < threads->nr; ++i)
e13798c7	353	printed += fprintf(fp, "%s%d", i ? ", " : "", thread_map__pid(threads, i));
9ae7d335 ACM	354
	355	return printed + fprintf(fp, "\n");
	356	}
792402fd JO	357
	358	static int get_comm(char **comm, pid_t pid)
	359	{
	360	char *path;
	361	size_t size;
	362	int err;
	363
	364	if (asprintf(&path, "%s/%d/comm", procfs__mountpoint(), pid) == -1)
	365	return -ENOMEM;
	366
	367	err = filename__read_str(path, comm, &size);
	368	if (!err) {
	369	/*
	370	* We're reading 16 bytes, while filename__read_str
	371	* allocates data per BUFSIZ bytes, so we can safely
	372	* mark the end of the string.
	373	*/
	374	(*comm)[size] = 0;
	375	rtrim(*comm);
	376	}
	377
	378	free(path);
	379	return err;
	380	}
	381
	382	static void comm_init(struct thread_map *map, int i)
	383	{
	384	pid_t pid = thread_map__pid(map, i);
	385	char *comm = NULL;
	386
	387	/* dummy pid comm initialization */
	388	if (pid == -1) {
	389	map->map[i].comm = strdup("dummy");
	390	return;
	391	}
	392
	393	/*
	394	* The comm name is like extra bonus ;-),
	395	* so just warn if we fail for any reason.
	396	*/
	397	if (get_comm(&comm, pid))
	398	pr_warning("Couldn't resolve comm name for pid %d\n", pid);
	399
	400	map->map[i].comm = comm;
	401	}
	402
	403	void thread_map__read_comms(struct thread_map *threads)
	404	{
	405	int i;
	406
	407	for (i = 0; i < threads->nr; ++i)
	408	comm_init(threads, i);
	409	}