| #include <dirent.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include "thread_map.h" |
| |
| /* Skip "." and ".." directories */ |
| static int filter(const struct dirent *dir) |
| { |
| if (dir->d_name[0] == '.') |
| return 0; |
| else |
| return 1; |
| } |
| |
| 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 = malloc(sizeof(*threads) + sizeof(pid_t) * items); |
| if (threads != NULL) { |
| for (i = 0; i < items; i++) |
| threads->map[i] = atoi(namelist[i]->d_name); |
| threads->nr = items; |
| } |
| |
| for (i=0; i<items; i++) |
| free(namelist[i]); |
| free(namelist); |
| |
| return threads; |
| } |
| |
| struct thread_map *thread_map__new_by_tid(pid_t tid) |
| { |
| struct thread_map *threads = malloc(sizeof(*threads) + sizeof(pid_t)); |
| |
| if (threads != NULL) { |
| threads->map[0] = tid; |
| threads->nr = 1; |
| } |
| |
| return threads; |
| } |
| |
| struct thread_map *thread_map__new(pid_t pid, pid_t tid) |
| { |
| if (pid != -1) |
| return thread_map__new_by_pid(pid); |
| return thread_map__new_by_tid(tid); |
| } |
| |
| void thread_map__delete(struct thread_map *threads) |
| { |
| free(threads); |
| } |
| |
| 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 ? ", " : "", threads->map[i]); |
| |
| return printed + fprintf(fp, "\n"); |
| } |