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 "util.h" 8#include "debug.h" 9 10static struct thread *thread__new(pid_t pid) 11{ 12 struct thread *self = zalloc(sizeof(*self)); 13 14 if (self != NULL) { 15 map_groups__init(&self->mg); 16 self->pid = pid; 17 self->comm = malloc(32); 18 if (self->comm) 19 snprintf(self->comm, 32, ":%d", self->pid); 20 } 21 22 return self; 23} 24 25void thread__delete(struct thread *self) 26{ 27 map_groups__exit(&self->mg); 28 free(self->comm); 29 free(self); 30} 31 32int thread__set_comm(struct thread *self, const char *comm) 33{ 34 int err; 35 36 if (self->comm) 37 free(self->comm); 38 self->comm = strdup(comm); 39 err = self->comm == NULL ? -ENOMEM : 0; 40 if (!err) { 41 self->comm_set = true; 42 map_groups__flush(&self->mg); 43 } 44 return err; 45} 46 47int thread__comm_len(struct thread *self) 48{ 49 if (!self->comm_len) { 50 if (!self->comm) 51 return 0; 52 self->comm_len = strlen(self->comm); 53 } 54 55 return self->comm_len; 56} 57 58static size_t thread__fprintf(struct thread *self, FILE *fp) 59{ 60 return fprintf(fp, "Thread %d %s\n", self->pid, self->comm) + 61 map_groups__fprintf(&self->mg, verbose, fp); 62} 63 64struct thread *perf_session__findnew(struct perf_session *self, pid_t pid) 65{ 66 struct rb_node **p = &self->threads.rb_node; 67 struct rb_node *parent = NULL; 68 struct thread *th; 69 70 /* 71 * Font-end cache - PID lookups come in blocks, 72 * so most of the time we dont have to look up 73 * the full rbtree: 74 */ 75 if (self->last_match && self->last_match->pid == pid) 76 return self->last_match; 77 78 while (*p != NULL) { 79 parent = *p; 80 th = rb_entry(parent, struct thread, rb_node); 81 82 if (th->pid == pid) { 83 self->last_match = th; 84 return th; 85 } 86 87 if (pid < th->pid) 88 p = &(*p)->rb_left; 89 else 90 p = &(*p)->rb_right; 91 } 92 93 th = thread__new(pid); 94 if (th != NULL) { 95 rb_link_node(&th->rb_node, parent, p); 96 rb_insert_color(&th->rb_node, &self->threads); 97 self->last_match = th; 98 } 99 100 return th; 101} 102 103void thread__insert_map(struct thread *self, struct map *map) 104{ 105 map_groups__fixup_overlappings(&self->mg, map, verbose, stderr); 106 map_groups__insert(&self->mg, map); 107} 108 109int thread__fork(struct thread *self, struct thread *parent) 110{ 111 int i; 112 113 if (parent->comm_set) { 114 if (self->comm) 115 free(self->comm); 116 self->comm = strdup(parent->comm); 117 if (!self->comm) 118 return -ENOMEM; 119 self->comm_set = true; 120 } 121 122 for (i = 0; i < MAP__NR_TYPES; ++i) 123 if (map_groups__clone(&self->mg, &parent->mg, i) < 0) 124 return -ENOMEM; 125 return 0; 126} 127 128size_t perf_session__fprintf(struct perf_session *self, FILE *fp) 129{ 130 size_t ret = 0; 131 struct rb_node *nd; 132 133 for (nd = rb_first(&self->threads); nd; nd = rb_next(nd)) { 134 struct thread *pos = rb_entry(nd, struct thread, rb_node); 135 136 ret += thread__fprintf(pos, fp); 137 } 138 139 return ret; 140} 141