1#include <dirent.h>
2#include <stdlib.h>
3#include <stdio.h>
4#include "thread_map.h"
5
6/* Skip "." and ".." directories */
7static int filter(const struct dirent *dir)
8{
9	if (dir->d_name[0] == '.')
10		return 0;
11	else
12		return 1;
13}
14
15struct thread_map *thread_map__new_by_pid(pid_t pid)
16{
17	struct thread_map *threads;
18	char name[256];
19	int items;
20	struct dirent **namelist = NULL;
21	int i;
22
23	sprintf(name, "/proc/%d/task", pid);
24	items = scandir(name, &namelist, filter, NULL);
25	if (items <= 0)
26                return NULL;
27
28	threads = malloc(sizeof(*threads) + sizeof(pid_t) * items);
29	if (threads != NULL) {
30		for (i = 0; i < items; i++)
31			threads->map[i] = atoi(namelist[i]->d_name);
32		threads->nr = items;
33	}
34
35	for (i=0; i<items; i++)
36		free(namelist[i]);
37	free(namelist);
38
39	return threads;
40}
41
42struct thread_map *thread_map__new_by_tid(pid_t tid)
43{
44	struct thread_map *threads = malloc(sizeof(*threads) + sizeof(pid_t));
45
46	if (threads != NULL) {
47		threads->map[0] = tid;
48		threads->nr	= 1;
49	}
50
51	return threads;
52}
53
54struct thread_map *thread_map__new(pid_t pid, pid_t tid)
55{
56	if (pid != -1)
57		return thread_map__new_by_pid(pid);
58	return thread_map__new_by_tid(tid);
59}
60
61void thread_map__delete(struct thread_map *threads)
62{
63	free(threads);
64}
65