tools/perf/util/thread.c

   1 #include "../perf.h"
   2 #include <stdlib.h>
   3 #include <stdio.h>
   4 #include <string.h>
   5 #include "session.h"
   6 #include "thread.h"
   7 #include "thread-stack.h"
   8 #include "util.h"
   9 #include "debug.h"
  10 #include "comm.h"
  11 #include "unwind.h"
  12
  13 #include <api/fs/fs.h>
  14
  15 int thread__init_map_groups(struct thread *thread, struct machine *machine)
  16 {
  17         pid_t pid = thread->pid_;
  18
  19         if (pid == thread->tid || pid == -1) {
  20                 thread->mg = map_groups__new(machine);
  21         } else {
  22                 struct thread *leader = __machine__findnew_thread(machine, pid, pid);
  23                 if (leader) {
  24                         thread->mg = map_groups__get(leader->mg);
  25                         thread__put(leader);
  26                 }
  27         }
  28
  29         return thread->mg ? 0 : -1;
  30 }
  31
  32 struct thread *thread__new(pid_t pid, pid_t tid)
  33 {
  34         char *comm_str;
  35         struct comm *comm;
  36         struct thread *thread = zalloc(sizeof(*thread));
  37
  38         if (thread != NULL) {
  39                 thread->pid_ = pid;
  40                 thread->tid = tid;
  41                 thread->ppid = -1;
  42                 thread->cpu = -1;
  43                 INIT_LIST_HEAD(&thread->comm_list);
  44
  45                 comm_str = malloc(32);
  46                 if (!comm_str)
  47                         goto err_thread;
  48
  49                 snprintf(comm_str, 32, ":%d", tid);
  50                 comm = comm__new(comm_str, 0, false);
  51                 free(comm_str);
  52                 if (!comm)
  53                         goto err_thread;
  54
  55                 list_add(&comm->list, &thread->comm_list);
  56                 atomic_set(&thread->refcnt, 1);
  57                 RB_CLEAR_NODE(&thread->rb_node);
  58         }
  59
  60         return thread;
  61
  62 err_thread:
  63         free(thread);
  64         return NULL;
  65 }
  66
  67 void thread__delete(struct thread *thread)
  68 {
  69         struct comm *comm, *tmp;
  70
  71         BUG_ON(!RB_EMPTY_NODE(&thread->rb_node));
  72
  73         thread_stack__free(thread);
  74
  75         if (thread->mg) {
  76                 map_groups__put(thread->mg);
  77                 thread->mg = NULL;
  78         }
  79         list_for_each_entry_safe(comm, tmp, &thread->comm_list, list) {
  80                 list_del(&comm->list);
  81                 comm__free(comm);
  82         }
  83         unwind__finish_access(thread);
  84
  85         free(thread);
  86 }
  87
  88 struct thread *thread__get(struct thread *thread)
  89 {
  90         if (thread)
  91                 atomic_inc(&thread->refcnt);
  92         return thread;
  93 }
  94
  95 void thread__put(struct thread *thread)
  96 {
  97         if (thread && atomic_dec_and_test(&thread->refcnt)) {
  98                 /*
  99                  * Remove it from the dead_threads list, as last reference
 100                  * is gone.
 101                  */
 102                 list_del_init(&thread->node);
 103                 thread__delete(thread);
 104         }
 105 }
 106
 107 struct comm *thread__comm(const struct thread *thread)
 108 {
 109         if (list_empty(&thread->comm_list))
 110                 return NULL;
 111
 112         return list_first_entry(&thread->comm_list, struct comm, list);
 113 }
 114
 115 struct comm *thread__exec_comm(const struct thread *thread)
 116 {
 117         struct comm *comm, *last = NULL;
 118
 119         list_for_each_entry(comm, &thread->comm_list, list) {
 120                 if (comm->exec)
 121                         return comm;
 122                 last = comm;
 123         }
 124
 125         return last;
 126 }
 127
 128 int __thread__set_comm(struct thread *thread, const char *str, u64 timestamp,
 129                        bool exec)
 130 {
 131         struct comm *new, *curr = thread__comm(thread);
 132
 133         /* Override the default :tid entry */
 134         if (!thread->comm_set) {
 135                 int err = comm__override(curr, str, timestamp, exec);
 136                 if (err)
 137                         return err;
 138         } else {
 139                 new = comm__new(str, timestamp, exec);
 140                 if (!new)
 141                         return -ENOMEM;
 142                 list_add(&new->list, &thread->comm_list);
 143
 144                 if (exec)
 145                         unwind__flush_access(thread);
 146         }
 147
 148         thread->comm_set = true;
 149
 150         return 0;
 151 }
 152
 153 int thread__set_comm_from_proc(struct thread *thread)
 154 {
 155         char path[64];
 156         char *comm = NULL;
 157         size_t sz;
 158         int err = -1;
 159
 160         if (!(snprintf(path, sizeof(path), "%d/task/%d/comm",
 161                        thread->pid_, thread->tid) >= (int)sizeof(path)) &&
 162             procfs__read_str(path, &comm, &sz) == 0) {
 163                 comm[sz - 1] = '\0';
 164                 err = thread__set_comm(thread, comm, 0);
 165         }
 166
 167         return err;
 168 }
 169
 170 const char *thread__comm_str(const struct thread *thread)
 171 {
 172         const struct comm *comm = thread__comm(thread);
 173
 174         if (!comm)
 175                 return NULL;
 176
 177         return comm__str(comm);
 178 }
 179
 180 /* CHECKME: it should probably better return the max comm len from its comm list */
 181 int thread__comm_len(struct thread *thread)
 182 {
 183         if (!thread->comm_len) {
 184                 const char *comm = thread__comm_str(thread);
 185                 if (!comm)
 186                         return 0;
 187                 thread->comm_len = strlen(comm);
 188         }
 189
 190         return thread->comm_len;
 191 }
 192
 193 size_t thread__fprintf(struct thread *thread, FILE *fp)
 194 {
 195         return fprintf(fp, "Thread %d %s\n", thread->tid, thread__comm_str(thread)) +
 196                map_groups__fprintf(thread->mg, fp);
 197 }
 198
 199 int thread__insert_map(struct thread *thread, struct map *map)
 200 {
 201         int ret;
 202
 203         ret = unwind__prepare_access(thread, map, NULL);
 204         if (ret)
 205                 return ret;
 206
 207         map_groups__fixup_overlappings(thread->mg, map, stderr);
 208         map_groups__insert(thread->mg, map);
 209
 210         return 0;
 211 }
 212
 213 static int __thread__prepare_access(struct thread *thread)
 214 {
 215         bool initialized = false;
 216         int i, err = 0;
 217
 218         for (i = 0; i < MAP__NR_TYPES; ++i) {
 219                 struct maps *maps = &thread->mg->maps[i];
 220                 struct map *map;
 221
 222                 pthread_rwlock_rdlock(&maps->lock);
 223
 224                 for (map = maps__first(maps); map; map = map__next(map)) {
 225                         err = unwind__prepare_access(thread, map, &initialized);
 226                         if (err || initialized)
 227                                 break;
 228                 }
 229
 230                 pthread_rwlock_unlock(&maps->lock);
 231         }
 232
 233         return err;
 234 }
 235
 236 static int thread__prepare_access(struct thread *thread)
 237 {
 238         int err = 0;
 239
 240         if (symbol_conf.use_callchain)
 241                 err = __thread__prepare_access(thread);
 242
 243         return err;
 244 }
 245
 246 static int thread__clone_map_groups(struct thread *thread,
 247                                     struct thread *parent)
 248 {
 249         int i;
 250
 251         /* This is new thread, we share map groups for process. */
 252         if (thread->pid_ == parent->pid_)
 253                 return thread__prepare_access(thread);
 254
 255         if (thread->mg == parent->mg) {
 256                 pr_debug("broken map groups on thread %d/%d parent %d/%d\n",
 257                          thread->pid_, thread->tid, parent->pid_, parent->tid);
 258                 return 0;
 259         }
 260
 261         /* But this one is new process, copy maps. */
 262         for (i = 0; i < MAP__NR_TYPES; ++i)
 263                 if (map_groups__clone(thread, parent->mg, i) < 0)
 264                         return -ENOMEM;
 265
 266         return 0;
 267 }
 268
 269 int thread__fork(struct thread *thread, struct thread *parent, u64 timestamp)
 270 {
 271         if (parent->comm_set) {
 272                 const char *comm = thread__comm_str(parent);
 273                 int err;
 274                 if (!comm)
 275                         return -ENOMEM;
 276                 err = thread__set_comm(thread, comm, timestamp);
 277                 if (err)
 278                         return err;
 279         }
 280
 281         thread->ppid = parent->tid;
 282         return thread__clone_map_groups(thread, parent);
 283 }
 284
 285 void thread__find_cpumode_addr_location(struct thread *thread,
 286                                         enum map_type type, u64 addr,
 287                                         struct addr_location *al)
 288 {
 289         size_t i;
 290         const u8 cpumodes[] = {
 291                 PERF_RECORD_MISC_USER,
 292                 PERF_RECORD_MISC_KERNEL,
 293                 PERF_RECORD_MISC_GUEST_USER,
 294                 PERF_RECORD_MISC_GUEST_KERNEL
 295         };
 296
 297         for (i = 0; i < ARRAY_SIZE(cpumodes); i++) {
 298                 thread__find_addr_location(thread, cpumodes[i], type, addr, al);
 299                 if (al->map)
 300                         break;
 301         }
 302 }
 303
 304 struct thread *thread__main_thread(struct machine *machine, struct thread *thread)
 305 {
 306         if (thread->pid_ == thread->tid)
 307                 return thread__get(thread);
 308
 309         if (thread->pid_ == -1)
 310                 return NULL;
 311
 312         return machine__find_thread(machine, thread->pid_, thread->pid_);
 313 }