perf session: Move the global threads list to perf_session
[linux-2.6.git] / 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 "util.h"
8 #include "debug.h"
9
10 void map_groups__init(struct map_groups *self)
11 {
12         int i;
13         for (i = 0; i < MAP__NR_TYPES; ++i) {
14                 self->maps[i] = RB_ROOT;
15                 INIT_LIST_HEAD(&self->removed_maps[i]);
16         }
17 }
18
19 static struct thread *thread__new(pid_t pid)
20 {
21         struct thread *self = zalloc(sizeof(*self));
22
23         if (self != NULL) {
24                 map_groups__init(&self->mg);
25                 self->pid = pid;
26                 self->comm = malloc(32);
27                 if (self->comm)
28                         snprintf(self->comm, 32, ":%d", self->pid);
29         }
30
31         return self;
32 }
33
34 int thread__set_comm(struct thread *self, const char *comm)
35 {
36         if (self->comm)
37                 free(self->comm);
38         self->comm = strdup(comm);
39         return self->comm ? 0 : -ENOMEM;
40 }
41
42 int thread__comm_len(struct thread *self)
43 {
44         if (!self->comm_len) {
45                 if (!self->comm)
46                         return 0;
47                 self->comm_len = strlen(self->comm);
48         }
49
50         return self->comm_len;
51 }
52
53 static const char *map_type__name[MAP__NR_TYPES] = {
54         [MAP__FUNCTION] = "Functions",
55         [MAP__VARIABLE] = "Variables",
56 };
57
58 static size_t __map_groups__fprintf_maps(struct map_groups *self,
59                                          enum map_type type, FILE *fp)
60 {
61         size_t printed = fprintf(fp, "%s:\n", map_type__name[type]);
62         struct rb_node *nd;
63
64         for (nd = rb_first(&self->maps[type]); nd; nd = rb_next(nd)) {
65                 struct map *pos = rb_entry(nd, struct map, rb_node);
66                 printed += fprintf(fp, "Map:");
67                 printed += map__fprintf(pos, fp);
68                 if (verbose > 1) {
69                         printed += dso__fprintf(pos->dso, type, fp);
70                         printed += fprintf(fp, "--\n");
71                 }
72         }
73
74         return printed;
75 }
76
77 size_t map_groups__fprintf_maps(struct map_groups *self, FILE *fp)
78 {
79         size_t printed = 0, i;
80         for (i = 0; i < MAP__NR_TYPES; ++i)
81                 printed += __map_groups__fprintf_maps(self, i, fp);
82         return printed;
83 }
84
85 static size_t __map_groups__fprintf_removed_maps(struct map_groups *self,
86                                                  enum map_type type, FILE *fp)
87 {
88         struct map *pos;
89         size_t printed = 0;
90
91         list_for_each_entry(pos, &self->removed_maps[type], node) {
92                 printed += fprintf(fp, "Map:");
93                 printed += map__fprintf(pos, fp);
94                 if (verbose > 1) {
95                         printed += dso__fprintf(pos->dso, type, fp);
96                         printed += fprintf(fp, "--\n");
97                 }
98         }
99         return printed;
100 }
101
102 static size_t map_groups__fprintf_removed_maps(struct map_groups *self, FILE *fp)
103 {
104         size_t printed = 0, i;
105         for (i = 0; i < MAP__NR_TYPES; ++i)
106                 printed += __map_groups__fprintf_removed_maps(self, i, fp);
107         return printed;
108 }
109
110 static size_t map_groups__fprintf(struct map_groups *self, FILE *fp)
111 {
112         size_t printed = map_groups__fprintf_maps(self, fp);
113         printed += fprintf(fp, "Removed maps:\n");
114         return printed + map_groups__fprintf_removed_maps(self, fp);
115 }
116
117 static size_t thread__fprintf(struct thread *self, FILE *fp)
118 {
119         return fprintf(fp, "Thread %d %s\n", self->pid, self->comm) +
120                map_groups__fprintf(&self->mg, fp);
121 }
122
123 struct thread *perf_session__findnew(struct perf_session *self, pid_t pid)
124 {
125         struct rb_node **p = &self->threads.rb_node;
126         struct rb_node *parent = NULL;
127         struct thread *th;
128
129         /*
130          * Font-end cache - PID lookups come in blocks,
131          * so most of the time we dont have to look up
132          * the full rbtree:
133          */
134         if (self->last_match && self->last_match->pid == pid)
135                 return self->last_match;
136
137         while (*p != NULL) {
138                 parent = *p;
139                 th = rb_entry(parent, struct thread, rb_node);
140
141                 if (th->pid == pid) {
142                         self->last_match = th;
143                         return th;
144                 }
145
146                 if (pid < th->pid)
147                         p = &(*p)->rb_left;
148                 else
149                         p = &(*p)->rb_right;
150         }
151
152         th = thread__new(pid);
153         if (th != NULL) {
154                 rb_link_node(&th->rb_node, parent, p);
155                 rb_insert_color(&th->rb_node, &self->threads);
156                 self->last_match = th;
157         }
158
159         return th;
160 }
161
162 static void map_groups__remove_overlappings(struct map_groups *self,
163                                             struct map *map)
164 {
165         struct rb_root *root = &self->maps[map->type];
166         struct rb_node *next = rb_first(root);
167
168         while (next) {
169                 struct map *pos = rb_entry(next, struct map, rb_node);
170                 next = rb_next(&pos->rb_node);
171
172                 if (!map__overlap(pos, map))
173                         continue;
174
175                 if (verbose >= 2) {
176                         fputs("overlapping maps:\n", stderr);
177                         map__fprintf(map, stderr);
178                         map__fprintf(pos, stderr);
179                 }
180
181                 rb_erase(&pos->rb_node, root);
182                 /*
183                  * We may have references to this map, for instance in some
184                  * hist_entry instances, so just move them to a separate
185                  * list.
186                  */
187                 list_add_tail(&pos->node, &self->removed_maps[map->type]);
188         }
189 }
190
191 void maps__insert(struct rb_root *maps, struct map *map)
192 {
193         struct rb_node **p = &maps->rb_node;
194         struct rb_node *parent = NULL;
195         const u64 ip = map->start;
196         struct map *m;
197
198         while (*p != NULL) {
199                 parent = *p;
200                 m = rb_entry(parent, struct map, rb_node);
201                 if (ip < m->start)
202                         p = &(*p)->rb_left;
203                 else
204                         p = &(*p)->rb_right;
205         }
206
207         rb_link_node(&map->rb_node, parent, p);
208         rb_insert_color(&map->rb_node, maps);
209 }
210
211 struct map *maps__find(struct rb_root *maps, u64 ip)
212 {
213         struct rb_node **p = &maps->rb_node;
214         struct rb_node *parent = NULL;
215         struct map *m;
216
217         while (*p != NULL) {
218                 parent = *p;
219                 m = rb_entry(parent, struct map, rb_node);
220                 if (ip < m->start)
221                         p = &(*p)->rb_left;
222                 else if (ip > m->end)
223                         p = &(*p)->rb_right;
224                 else
225                         return m;
226         }
227
228         return NULL;
229 }
230
231 void thread__insert_map(struct thread *self, struct map *map)
232 {
233         map_groups__remove_overlappings(&self->mg, map);
234         map_groups__insert(&self->mg, map);
235 }
236
237 /*
238  * XXX This should not really _copy_ te maps, but refcount them.
239  */
240 static int map_groups__clone(struct map_groups *self,
241                              struct map_groups *parent, enum map_type type)
242 {
243         struct rb_node *nd;
244         for (nd = rb_first(&parent->maps[type]); nd; nd = rb_next(nd)) {
245                 struct map *map = rb_entry(nd, struct map, rb_node);
246                 struct map *new = map__clone(map);
247                 if (new == NULL)
248                         return -ENOMEM;
249                 map_groups__insert(self, new);
250         }
251         return 0;
252 }
253
254 int thread__fork(struct thread *self, struct thread *parent)
255 {
256         int i;
257
258         if (self->comm)
259                 free(self->comm);
260         self->comm = strdup(parent->comm);
261         if (!self->comm)
262                 return -ENOMEM;
263
264         for (i = 0; i < MAP__NR_TYPES; ++i)
265                 if (map_groups__clone(&self->mg, &parent->mg, i) < 0)
266                         return -ENOMEM;
267         return 0;
268 }
269
270 size_t perf_session__fprintf(struct perf_session *self, FILE *fp)
271 {
272         size_t ret = 0;
273         struct rb_node *nd;
274
275         for (nd = rb_first(&self->threads); nd; nd = rb_next(nd)) {
276                 struct thread *pos = rb_entry(nd, struct thread, rb_node);
277
278                 ret += thread__fprintf(pos, fp);
279         }
280
281         return ret;
282 }
283
284 struct symbol *map_groups__find_symbol(struct map_groups *self,
285                                        enum map_type type, u64 addr,
286                                        symbol_filter_t filter)
287 {
288         struct map *map = map_groups__find(self, type, addr);
289
290         if (map != NULL)
291                 return map__find_symbol(map, map->map_ip(map, addr), filter);
292
293         return NULL;
294 }