Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 1 | #include "../perf.h" |
| 2 | #include <stdlib.h> |
| 3 | #include <stdio.h> |
| 4 | #include <string.h> |
| 5 | #include "thread.h" |
| 6 | #include "util.h" |
Frederic Weisbecker | 6e08643 | 2009-08-18 17:04:03 +0200 | [diff] [blame] | 7 | #include "debug.h" |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 8 | |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 9 | static struct rb_root threads; |
| 10 | static struct thread *last_match; |
| 11 | |
Frederic Weisbecker | 97ea1a7 | 2009-10-08 21:04:17 +0200 | [diff] [blame] | 12 | static struct thread *thread__new(pid_t pid) |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 13 | { |
Ingo Molnar | 0ec04e1 | 2009-09-16 17:40:48 +0200 | [diff] [blame] | 14 | struct thread *self = calloc(1, sizeof(*self)); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 15 | |
| 16 | if (self != NULL) { |
| 17 | self->pid = pid; |
Frederic Weisbecker | 97ea1a7 | 2009-10-08 21:04:17 +0200 | [diff] [blame] | 18 | self->comm = malloc(32); |
| 19 | if (self->comm) |
| 20 | snprintf(self->comm, 32, ":%d", self->pid); |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 21 | self->maps = RB_ROOT; |
Arnaldo Carvalho de Melo | 439d473 | 2009-10-02 03:29:58 -0300 | [diff] [blame] | 22 | INIT_LIST_HEAD(&self->removed_maps); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 23 | } |
| 24 | |
| 25 | return self; |
| 26 | } |
| 27 | |
| 28 | int thread__set_comm(struct thread *self, const char *comm) |
| 29 | { |
| 30 | if (self->comm) |
| 31 | free(self->comm); |
| 32 | self->comm = strdup(comm); |
| 33 | return self->comm ? 0 : -ENOMEM; |
| 34 | } |
| 35 | |
| 36 | static size_t thread__fprintf(struct thread *self, FILE *fp) |
| 37 | { |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 38 | struct rb_node *nd; |
Arnaldo Carvalho de Melo | 439d473 | 2009-10-02 03:29:58 -0300 | [diff] [blame] | 39 | struct map *pos; |
| 40 | size_t ret = fprintf(fp, "Thread %d %s\nCurrent maps:\n", |
| 41 | self->pid, self->comm); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 42 | |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 43 | for (nd = rb_first(&self->maps); nd; nd = rb_next(nd)) { |
Arnaldo Carvalho de Melo | 439d473 | 2009-10-02 03:29:58 -0300 | [diff] [blame] | 44 | pos = rb_entry(nd, struct map, rb_node); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 45 | ret += map__fprintf(pos, fp); |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 46 | } |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 47 | |
Arnaldo Carvalho de Melo | 439d473 | 2009-10-02 03:29:58 -0300 | [diff] [blame] | 48 | ret = fprintf(fp, "Removed maps:\n"); |
| 49 | |
| 50 | list_for_each_entry(pos, &self->removed_maps, node) |
| 51 | ret += map__fprintf(pos, fp); |
| 52 | |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 53 | return ret; |
| 54 | } |
| 55 | |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 56 | struct thread *threads__findnew(pid_t pid) |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 57 | { |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 58 | struct rb_node **p = &threads.rb_node; |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 59 | struct rb_node *parent = NULL; |
| 60 | struct thread *th; |
| 61 | |
| 62 | /* |
| 63 | * Font-end cache - PID lookups come in blocks, |
| 64 | * so most of the time we dont have to look up |
| 65 | * the full rbtree: |
| 66 | */ |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 67 | if (last_match && last_match->pid == pid) |
| 68 | return last_match; |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 69 | |
| 70 | while (*p != NULL) { |
| 71 | parent = *p; |
| 72 | th = rb_entry(parent, struct thread, rb_node); |
| 73 | |
| 74 | if (th->pid == pid) { |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 75 | last_match = th; |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 76 | return th; |
| 77 | } |
| 78 | |
| 79 | if (pid < th->pid) |
| 80 | p = &(*p)->rb_left; |
| 81 | else |
| 82 | p = &(*p)->rb_right; |
| 83 | } |
| 84 | |
Frederic Weisbecker | 97ea1a7 | 2009-10-08 21:04:17 +0200 | [diff] [blame] | 85 | th = thread__new(pid); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 86 | if (th != NULL) { |
| 87 | rb_link_node(&th->rb_node, parent, p); |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 88 | rb_insert_color(&th->rb_node, &threads); |
| 89 | last_match = th; |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 90 | } |
| 91 | |
| 92 | return th; |
| 93 | } |
| 94 | |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 95 | struct thread *register_idle_thread(void) |
Frederic Weisbecker | 5b447a6 | 2009-08-31 06:45:18 +0200 | [diff] [blame] | 96 | { |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 97 | struct thread *thread = threads__findnew(0); |
Frederic Weisbecker | 5b447a6 | 2009-08-31 06:45:18 +0200 | [diff] [blame] | 98 | |
Ingo Molnar | 80ed098 | 2009-09-16 14:12:36 +0200 | [diff] [blame] | 99 | if (!thread || thread__set_comm(thread, "swapper")) { |
Frederic Weisbecker | 5b447a6 | 2009-08-31 06:45:18 +0200 | [diff] [blame] | 100 | fprintf(stderr, "problem inserting idle task.\n"); |
| 101 | exit(-1); |
| 102 | } |
| 103 | |
| 104 | return thread; |
| 105 | } |
| 106 | |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 107 | static void thread__remove_overlappings(struct thread *self, struct map *map) |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 108 | { |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 109 | struct rb_node *next = rb_first(&self->maps); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 110 | |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 111 | while (next) { |
| 112 | struct map *pos = rb_entry(next, struct map, rb_node); |
| 113 | next = rb_next(&pos->rb_node); |
Frederic Weisbecker | 6e08643 | 2009-08-18 17:04:03 +0200 | [diff] [blame] | 114 | |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 115 | if (!map__overlap(pos, map)) |
| 116 | continue; |
Frederic Weisbecker | 6e08643 | 2009-08-18 17:04:03 +0200 | [diff] [blame] | 117 | |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 118 | if (verbose >= 2) { |
| 119 | printf("overlapping maps:\n"); |
| 120 | map__fprintf(map, stdout); |
| 121 | map__fprintf(pos, stdout); |
| 122 | } |
Frederic Weisbecker | 6e08643 | 2009-08-18 17:04:03 +0200 | [diff] [blame] | 123 | |
Arnaldo Carvalho de Melo | 439d473 | 2009-10-02 03:29:58 -0300 | [diff] [blame] | 124 | rb_erase(&pos->rb_node, &self->maps); |
| 125 | /* |
| 126 | * We may have references to this map, for instance in some |
| 127 | * hist_entry instances, so just move them to a separate |
| 128 | * list. |
| 129 | */ |
| 130 | list_add_tail(&pos->node, &self->removed_maps); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 131 | } |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 132 | } |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 133 | |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 134 | void maps__insert(struct rb_root *maps, struct map *map) |
| 135 | { |
| 136 | struct rb_node **p = &maps->rb_node; |
| 137 | struct rb_node *parent = NULL; |
| 138 | const u64 ip = map->start; |
| 139 | struct map *m; |
| 140 | |
| 141 | while (*p != NULL) { |
| 142 | parent = *p; |
| 143 | m = rb_entry(parent, struct map, rb_node); |
| 144 | if (ip < m->start) |
| 145 | p = &(*p)->rb_left; |
| 146 | else |
| 147 | p = &(*p)->rb_right; |
| 148 | } |
| 149 | |
| 150 | rb_link_node(&map->rb_node, parent, p); |
| 151 | rb_insert_color(&map->rb_node, maps); |
| 152 | } |
| 153 | |
| 154 | struct map *maps__find(struct rb_root *maps, u64 ip) |
| 155 | { |
| 156 | struct rb_node **p = &maps->rb_node; |
| 157 | struct rb_node *parent = NULL; |
| 158 | struct map *m; |
| 159 | |
| 160 | while (*p != NULL) { |
| 161 | parent = *p; |
| 162 | m = rb_entry(parent, struct map, rb_node); |
| 163 | if (ip < m->start) |
| 164 | p = &(*p)->rb_left; |
| 165 | else if (ip > m->end) |
| 166 | p = &(*p)->rb_right; |
| 167 | else |
| 168 | return m; |
| 169 | } |
| 170 | |
| 171 | return NULL; |
| 172 | } |
| 173 | |
| 174 | void thread__insert_map(struct thread *self, struct map *map) |
| 175 | { |
| 176 | thread__remove_overlappings(self, map); |
| 177 | maps__insert(&self->maps, map); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 178 | } |
| 179 | |
| 180 | int thread__fork(struct thread *self, struct thread *parent) |
| 181 | { |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 182 | struct rb_node *nd; |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 183 | |
| 184 | if (self->comm) |
| 185 | free(self->comm); |
| 186 | self->comm = strdup(parent->comm); |
| 187 | if (!self->comm) |
| 188 | return -ENOMEM; |
| 189 | |
Arnaldo Carvalho de Melo | 1b46cdd | 2009-09-28 14:48:46 -0300 | [diff] [blame] | 190 | for (nd = rb_first(&parent->maps); nd; nd = rb_next(nd)) { |
| 191 | struct map *map = rb_entry(nd, struct map, rb_node); |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 192 | struct map *new = map__clone(map); |
| 193 | if (!new) |
| 194 | return -ENOMEM; |
| 195 | thread__insert_map(self, new); |
| 196 | } |
| 197 | |
| 198 | return 0; |
| 199 | } |
| 200 | |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 201 | size_t threads__fprintf(FILE *fp) |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 202 | { |
| 203 | size_t ret = 0; |
| 204 | struct rb_node *nd; |
| 205 | |
Arnaldo Carvalho de Melo | d5b889f | 2009-10-13 11:16:29 -0300 | [diff] [blame] | 206 | for (nd = rb_first(&threads); nd; nd = rb_next(nd)) { |
Frederic Weisbecker | 6baa0a5 | 2009-08-14 12:21:53 +0200 | [diff] [blame] | 207 | struct thread *pos = rb_entry(nd, struct thread, rb_node); |
| 208 | |
| 209 | ret += thread__fprintf(pos, fp); |
| 210 | } |
| 211 | |
| 212 | return ret; |
| 213 | } |