tools/perf/util/thread_map.c

   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 }