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