Commit | Line | Data |
---|---|---|
fd78260b ACM |
1 | #include <dirent.h> |
2 | #include <stdlib.h> | |
3 | #include <stdio.h> | |
4 | #include "thread_map.h" | |
5 | ||
6 | /* Skip "." and ".." directories */ | |
7 | static int filter(const struct dirent *dir) | |
8 | { | |
9 | if (dir->d_name[0] == '.') | |
10 | return 0; | |
11 | else | |
12 | return 1; | |
13 | } | |
14 | ||
15 | struct 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 | ||
42 | struct 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 | ||
54 | struct 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 | ||
61 | void thread_map__delete(struct thread_map *threads) | |
62 | { | |
63 | free(threads); | |
64 | } |