blob: 0b4b796167beca45dc12736527fa7bcfe2560d9e [file] [log] [blame]
Li Zefanba77c9e2009-11-20 15:53:25 +08001#include "builtin.h"
2#include "perf.h"
3
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -03004#include "util/evlist.h"
Arnaldo Carvalho de Melofcf65bf2012-08-07 09:58:03 -03005#include "util/evsel.h"
Li Zefanba77c9e2009-11-20 15:53:25 +08006#include "util/util.h"
7#include "util/cache.h"
8#include "util/symbol.h"
9#include "util/thread.h"
10#include "util/header.h"
Arnaldo Carvalho de Melo94c744b2009-12-11 21:24:02 -020011#include "util/session.h"
Arnaldo Carvalho de Melo45694aa2011-11-28 08:30:20 -020012#include "util/tool.h"
Li Zefanba77c9e2009-11-20 15:53:25 +080013
14#include "util/parse-options.h"
15#include "util/trace-event.h"
16
17#include "util/debug.h"
Li Zefanba77c9e2009-11-20 15:53:25 +080018
19#include <linux/rbtree.h>
20
21struct alloc_stat;
22typedef int (*sort_fn_t)(struct alloc_stat *, struct alloc_stat *);
23
Li Zefanba77c9e2009-11-20 15:53:25 +080024static int alloc_flag;
25static int caller_flag;
26
Li Zefanba77c9e2009-11-20 15:53:25 +080027static int alloc_lines = -1;
28static int caller_lines = -1;
29
Li Zefan7707b6b2009-11-24 13:25:48 +080030static bool raw_ip;
31
Li Zefan7d0d3942009-11-24 13:26:31 +080032static int *cpunode_map;
33static int max_cpu_num;
34
Li Zefanba77c9e2009-11-20 15:53:25 +080035struct alloc_stat {
Li Zefan079d3f62009-11-24 13:26:55 +080036 u64 call_site;
37 u64 ptr;
Li Zefanba77c9e2009-11-20 15:53:25 +080038 u64 bytes_req;
39 u64 bytes_alloc;
40 u32 hit;
Li Zefan079d3f62009-11-24 13:26:55 +080041 u32 pingpong;
42
43 short alloc_cpu;
Li Zefanba77c9e2009-11-20 15:53:25 +080044
45 struct rb_node node;
46};
47
48static struct rb_root root_alloc_stat;
49static struct rb_root root_alloc_sorted;
50static struct rb_root root_caller_stat;
51static struct rb_root root_caller_sorted;
52
53static unsigned long total_requested, total_allocated;
Li Zefan7d0d3942009-11-24 13:26:31 +080054static unsigned long nr_allocs, nr_cross_allocs;
Li Zefanba77c9e2009-11-20 15:53:25 +080055
Li Zefan7d0d3942009-11-24 13:26:31 +080056#define PATH_SYS_NODE "/sys/devices/system/node"
57
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030058static int init_cpunode_map(void)
Li Zefan7d0d3942009-11-24 13:26:31 +080059{
60 FILE *fp;
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030061 int i, err = -1;
Li Zefan7d0d3942009-11-24 13:26:31 +080062
63 fp = fopen("/sys/devices/system/cpu/kernel_max", "r");
64 if (!fp) {
65 max_cpu_num = 4096;
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030066 return 0;
Li Zefan7d0d3942009-11-24 13:26:31 +080067 }
68
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030069 if (fscanf(fp, "%d", &max_cpu_num) < 1) {
70 pr_err("Failed to read 'kernel_max' from sysfs");
71 goto out_close;
72 }
73
Li Zefan7d0d3942009-11-24 13:26:31 +080074 max_cpu_num++;
75
76 cpunode_map = calloc(max_cpu_num, sizeof(int));
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030077 if (!cpunode_map) {
78 pr_err("%s: calloc failed\n", __func__);
79 goto out_close;
80 }
81
Li Zefan7d0d3942009-11-24 13:26:31 +080082 for (i = 0; i < max_cpu_num; i++)
83 cpunode_map[i] = -1;
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030084
85 err = 0;
86out_close:
Li Zefan7d0d3942009-11-24 13:26:31 +080087 fclose(fp);
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030088 return err;
Li Zefan7d0d3942009-11-24 13:26:31 +080089}
90
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030091static int setup_cpunode_map(void)
Li Zefan7d0d3942009-11-24 13:26:31 +080092{
93 struct dirent *dent1, *dent2;
94 DIR *dir1, *dir2;
95 unsigned int cpu, mem;
96 char buf[PATH_MAX];
97
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -030098 if (init_cpunode_map())
99 return -1;
Li Zefan7d0d3942009-11-24 13:26:31 +0800100
101 dir1 = opendir(PATH_SYS_NODE);
102 if (!dir1)
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300103 return -1;
Li Zefan7d0d3942009-11-24 13:26:31 +0800104
Ulrich Drepper659d8cf2009-12-19 16:40:28 -0500105 while ((dent1 = readdir(dir1)) != NULL) {
106 if (dent1->d_type != DT_DIR ||
107 sscanf(dent1->d_name, "node%u", &mem) < 1)
Li Zefan7d0d3942009-11-24 13:26:31 +0800108 continue;
109
110 snprintf(buf, PATH_MAX, "%s/%s", PATH_SYS_NODE, dent1->d_name);
111 dir2 = opendir(buf);
112 if (!dir2)
113 continue;
Ulrich Drepper659d8cf2009-12-19 16:40:28 -0500114 while ((dent2 = readdir(dir2)) != NULL) {
115 if (dent2->d_type != DT_LNK ||
116 sscanf(dent2->d_name, "cpu%u", &cpu) < 1)
Li Zefan7d0d3942009-11-24 13:26:31 +0800117 continue;
118 cpunode_map[cpu] = mem;
119 }
Namhyung Kim8442da12012-01-08 02:25:28 +0900120 closedir(dir2);
Li Zefan7d0d3942009-11-24 13:26:31 +0800121 }
Namhyung Kim8442da12012-01-08 02:25:28 +0900122 closedir(dir1);
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300123 return 0;
Li Zefan7d0d3942009-11-24 13:26:31 +0800124}
125
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300126static int insert_alloc_stat(unsigned long call_site, unsigned long ptr,
127 int bytes_req, int bytes_alloc, int cpu)
Li Zefanba77c9e2009-11-20 15:53:25 +0800128{
129 struct rb_node **node = &root_alloc_stat.rb_node;
130 struct rb_node *parent = NULL;
131 struct alloc_stat *data = NULL;
132
Li Zefanba77c9e2009-11-20 15:53:25 +0800133 while (*node) {
134 parent = *node;
135 data = rb_entry(*node, struct alloc_stat, node);
136
137 if (ptr > data->ptr)
138 node = &(*node)->rb_right;
139 else if (ptr < data->ptr)
140 node = &(*node)->rb_left;
141 else
142 break;
143 }
144
145 if (data && data->ptr == ptr) {
146 data->hit++;
147 data->bytes_req += bytes_req;
Wenji Huang4efb5292009-12-21 17:52:55 +0800148 data->bytes_alloc += bytes_alloc;
Li Zefanba77c9e2009-11-20 15:53:25 +0800149 } else {
150 data = malloc(sizeof(*data));
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300151 if (!data) {
152 pr_err("%s: malloc failed\n", __func__);
153 return -1;
154 }
Li Zefanba77c9e2009-11-20 15:53:25 +0800155 data->ptr = ptr;
Li Zefan079d3f62009-11-24 13:26:55 +0800156 data->pingpong = 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800157 data->hit = 1;
158 data->bytes_req = bytes_req;
159 data->bytes_alloc = bytes_alloc;
160
161 rb_link_node(&data->node, parent, node);
162 rb_insert_color(&data->node, &root_alloc_stat);
163 }
Li Zefan079d3f62009-11-24 13:26:55 +0800164 data->call_site = call_site;
165 data->alloc_cpu = cpu;
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300166 return 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800167}
168
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300169static int insert_caller_stat(unsigned long call_site,
Li Zefanba77c9e2009-11-20 15:53:25 +0800170 int bytes_req, int bytes_alloc)
171{
172 struct rb_node **node = &root_caller_stat.rb_node;
173 struct rb_node *parent = NULL;
174 struct alloc_stat *data = NULL;
175
Li Zefanba77c9e2009-11-20 15:53:25 +0800176 while (*node) {
177 parent = *node;
178 data = rb_entry(*node, struct alloc_stat, node);
179
180 if (call_site > data->call_site)
181 node = &(*node)->rb_right;
182 else if (call_site < data->call_site)
183 node = &(*node)->rb_left;
184 else
185 break;
186 }
187
188 if (data && data->call_site == call_site) {
189 data->hit++;
190 data->bytes_req += bytes_req;
Wenji Huang4efb5292009-12-21 17:52:55 +0800191 data->bytes_alloc += bytes_alloc;
Li Zefanba77c9e2009-11-20 15:53:25 +0800192 } else {
193 data = malloc(sizeof(*data));
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300194 if (!data) {
195 pr_err("%s: malloc failed\n", __func__);
196 return -1;
197 }
Li Zefanba77c9e2009-11-20 15:53:25 +0800198 data->call_site = call_site;
Li Zefan079d3f62009-11-24 13:26:55 +0800199 data->pingpong = 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800200 data->hit = 1;
201 data->bytes_req = bytes_req;
202 data->bytes_alloc = bytes_alloc;
203
204 rb_link_node(&data->node, parent, node);
205 rb_insert_color(&data->node, &root_caller_stat);
206 }
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300207
208 return 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800209}
210
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300211static int perf_evsel__process_alloc_event(struct perf_evsel *evsel,
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300212 struct perf_sample *sample)
Li Zefanba77c9e2009-11-20 15:53:25 +0800213{
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300214 unsigned long ptr = perf_evsel__intval(evsel, sample, "ptr"),
215 call_site = perf_evsel__intval(evsel, sample, "call_site");
216 int bytes_req = perf_evsel__intval(evsel, sample, "bytes_req"),
217 bytes_alloc = perf_evsel__intval(evsel, sample, "bytes_alloc");
Li Zefanba77c9e2009-11-20 15:53:25 +0800218
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300219 if (insert_alloc_stat(call_site, ptr, bytes_req, bytes_alloc, sample->cpu) ||
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300220 insert_caller_stat(call_site, bytes_req, bytes_alloc))
221 return -1;
Li Zefanba77c9e2009-11-20 15:53:25 +0800222
223 total_requested += bytes_req;
224 total_allocated += bytes_alloc;
Li Zefan7d0d3942009-11-24 13:26:31 +0800225
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300226 nr_allocs++;
227 return 0;
228}
229
230static int perf_evsel__process_alloc_node_event(struct perf_evsel *evsel,
231 struct perf_sample *sample)
232{
233 int ret = perf_evsel__process_alloc_event(evsel, sample);
234
235 if (!ret) {
236 int node1 = cpunode_map[sample->cpu],
237 node2 = perf_evsel__intval(evsel, sample, "node");
238
Li Zefan7d0d3942009-11-24 13:26:31 +0800239 if (node1 != node2)
240 nr_cross_allocs++;
241 }
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300242
243 return ret;
Li Zefanba77c9e2009-11-20 15:53:25 +0800244}
245
Li Zefan079d3f62009-11-24 13:26:55 +0800246static int ptr_cmp(struct alloc_stat *, struct alloc_stat *);
247static int callsite_cmp(struct alloc_stat *, struct alloc_stat *);
248
249static struct alloc_stat *search_alloc_stat(unsigned long ptr,
250 unsigned long call_site,
251 struct rb_root *root,
252 sort_fn_t sort_fn)
253{
254 struct rb_node *node = root->rb_node;
255 struct alloc_stat key = { .ptr = ptr, .call_site = call_site };
256
257 while (node) {
258 struct alloc_stat *data;
259 int cmp;
260
261 data = rb_entry(node, struct alloc_stat, node);
262
263 cmp = sort_fn(&key, data);
264 if (cmp < 0)
265 node = node->rb_left;
266 else if (cmp > 0)
267 node = node->rb_right;
268 else
269 return data;
270 }
271 return NULL;
272}
273
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300274static int perf_evsel__process_free_event(struct perf_evsel *evsel,
275 struct perf_sample *sample)
Li Zefanba77c9e2009-11-20 15:53:25 +0800276{
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300277 unsigned long ptr = perf_evsel__intval(evsel, sample, "ptr");
Li Zefan079d3f62009-11-24 13:26:55 +0800278 struct alloc_stat *s_alloc, *s_caller;
279
Li Zefan079d3f62009-11-24 13:26:55 +0800280 s_alloc = search_alloc_stat(ptr, 0, &root_alloc_stat, ptr_cmp);
281 if (!s_alloc)
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300282 return 0;
Li Zefan079d3f62009-11-24 13:26:55 +0800283
Arnaldo Carvalho de Melo22ad7982012-08-07 10:56:43 -0300284 if ((short)sample->cpu != s_alloc->alloc_cpu) {
Li Zefan079d3f62009-11-24 13:26:55 +0800285 s_alloc->pingpong++;
286
287 s_caller = search_alloc_stat(0, s_alloc->call_site,
288 &root_caller_stat, callsite_cmp);
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300289 if (!s_caller)
290 return -1;
Li Zefan079d3f62009-11-24 13:26:55 +0800291 s_caller->pingpong++;
292 }
293 s_alloc->alloc_cpu = -1;
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300294
295 return 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800296}
297
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300298typedef int (*tracepoint_handler)(struct perf_evsel *evsel,
299 struct perf_sample *sample);
Li Zefanba77c9e2009-11-20 15:53:25 +0800300
Irina Tirdea1d037ca2012-09-11 01:15:03 +0300301static int process_sample_event(struct perf_tool *tool __maybe_unused,
Arnaldo Carvalho de Melod20deb62011-11-25 08:19:45 -0200302 union perf_event *event,
Arnaldo Carvalho de Melo8115d602011-01-29 14:01:45 -0200303 struct perf_sample *sample,
Arnaldo Carvalho de Melofcf65bf2012-08-07 09:58:03 -0300304 struct perf_evsel *evsel,
Arnaldo Carvalho de Melo743eb862011-11-28 07:56:39 -0200305 struct machine *machine)
Li Zefanba77c9e2009-11-20 15:53:25 +0800306{
Arnaldo Carvalho de Melo743eb862011-11-28 07:56:39 -0200307 struct thread *thread = machine__findnew_thread(machine, event->ip.pid);
Li Zefanba77c9e2009-11-20 15:53:25 +0800308
Li Zefanba77c9e2009-11-20 15:53:25 +0800309 if (thread == NULL) {
310 pr_debug("problem processing %d event, skipping it.\n",
311 event->header.type);
312 return -1;
313 }
314
315 dump_printf(" ... thread: %s:%d\n", thread->comm, thread->pid);
316
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300317 if (evsel->handler.func != NULL) {
318 tracepoint_handler f = evsel->handler.func;
319 return f(evsel, sample);
320 }
321
322 return 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800323}
324
Arnaldo Carvalho de Melofcf65bf2012-08-07 09:58:03 -0300325static struct perf_tool perf_kmem = {
326 .sample = process_sample_event,
327 .comm = perf_event__process_comm,
328 .ordered_samples = true,
Li Zefanba77c9e2009-11-20 15:53:25 +0800329};
330
Li Zefanba77c9e2009-11-20 15:53:25 +0800331static double fragmentation(unsigned long n_req, unsigned long n_alloc)
332{
333 if (n_alloc == 0)
334 return 0.0;
335 else
336 return 100.0 - (100.0 * n_req / n_alloc);
337}
338
Arnaldo Carvalho de Melo4aa65632009-12-13 19:50:29 -0200339static void __print_result(struct rb_root *root, struct perf_session *session,
340 int n_lines, int is_caller)
Li Zefanba77c9e2009-11-20 15:53:25 +0800341{
342 struct rb_node *next;
Arnaldo Carvalho de Melo23346f22010-04-27 21:17:50 -0300343 struct machine *machine;
Li Zefanba77c9e2009-11-20 15:53:25 +0800344
Li Zefan079d3f62009-11-24 13:26:55 +0800345 printf("%.102s\n", graph_dotted_line);
346 printf(" %-34s |", is_caller ? "Callsite": "Alloc Ptr");
Pekka Enberg47103272010-01-19 19:23:23 +0200347 printf(" Total_alloc/Per | Total_req/Per | Hit | Ping-pong | Frag\n");
Li Zefan079d3f62009-11-24 13:26:55 +0800348 printf("%.102s\n", graph_dotted_line);
Li Zefanba77c9e2009-11-20 15:53:25 +0800349
350 next = rb_first(root);
351
Arnaldo Carvalho de Melo23346f22010-04-27 21:17:50 -0300352 machine = perf_session__find_host_machine(session);
353 if (!machine) {
Zhang, Yanmina1645ce2010-04-19 13:32:50 +0800354 pr_err("__print_result: couldn't find kernel information\n");
355 return;
356 }
Li Zefanba77c9e2009-11-20 15:53:25 +0800357 while (next && n_lines--) {
Arnaldo Carvalho de Melo1b145ae2009-11-23 17:51:09 -0200358 struct alloc_stat *data = rb_entry(next, struct alloc_stat,
359 node);
360 struct symbol *sym = NULL;
Arnaldo Carvalho de Melo71cf8b82010-04-01 21:24:38 -0300361 struct map *map;
Li Zefan079d3f62009-11-24 13:26:55 +0800362 char buf[BUFSIZ];
Arnaldo Carvalho de Melo1b145ae2009-11-23 17:51:09 -0200363 u64 addr;
Li Zefanba77c9e2009-11-20 15:53:25 +0800364
Arnaldo Carvalho de Melo1b145ae2009-11-23 17:51:09 -0200365 if (is_caller) {
366 addr = data->call_site;
Li Zefan7707b6b2009-11-24 13:25:48 +0800367 if (!raw_ip)
Arnaldo Carvalho de Melo5c0541d2010-04-29 15:25:23 -0300368 sym = machine__find_kernel_function(machine, addr, &map, NULL);
Arnaldo Carvalho de Melo1b145ae2009-11-23 17:51:09 -0200369 } else
370 addr = data->ptr;
Li Zefanba77c9e2009-11-20 15:53:25 +0800371
Arnaldo Carvalho de Melo1b145ae2009-11-23 17:51:09 -0200372 if (sym != NULL)
Arnaldo Carvalho de Melo9486aa32011-01-22 20:37:02 -0200373 snprintf(buf, sizeof(buf), "%s+%" PRIx64 "", sym->name,
Arnaldo Carvalho de Melo71cf8b82010-04-01 21:24:38 -0300374 addr - map->unmap_ip(map, sym->start));
Arnaldo Carvalho de Melo1b145ae2009-11-23 17:51:09 -0200375 else
Arnaldo Carvalho de Melo9486aa32011-01-22 20:37:02 -0200376 snprintf(buf, sizeof(buf), "%#" PRIx64 "", addr);
Li Zefan079d3f62009-11-24 13:26:55 +0800377 printf(" %-34s |", buf);
Arnaldo Carvalho de Melo1b145ae2009-11-23 17:51:09 -0200378
Pekka Enberg47103272010-01-19 19:23:23 +0200379 printf(" %9llu/%-5lu | %9llu/%-5lu | %8lu | %8lu | %6.3f%%\n",
Li Zefan079d3f62009-11-24 13:26:55 +0800380 (unsigned long long)data->bytes_alloc,
Li Zefanba77c9e2009-11-20 15:53:25 +0800381 (unsigned long)data->bytes_alloc / data->hit,
382 (unsigned long long)data->bytes_req,
383 (unsigned long)data->bytes_req / data->hit,
384 (unsigned long)data->hit,
Li Zefan079d3f62009-11-24 13:26:55 +0800385 (unsigned long)data->pingpong,
Li Zefanba77c9e2009-11-20 15:53:25 +0800386 fragmentation(data->bytes_req, data->bytes_alloc));
387
388 next = rb_next(next);
389 }
390
391 if (n_lines == -1)
Li Zefan079d3f62009-11-24 13:26:55 +0800392 printf(" ... | ... | ... | ... | ... | ... \n");
Li Zefanba77c9e2009-11-20 15:53:25 +0800393
Li Zefan079d3f62009-11-24 13:26:55 +0800394 printf("%.102s\n", graph_dotted_line);
Li Zefanba77c9e2009-11-20 15:53:25 +0800395}
396
397static void print_summary(void)
398{
399 printf("\nSUMMARY\n=======\n");
400 printf("Total bytes requested: %lu\n", total_requested);
401 printf("Total bytes allocated: %lu\n", total_allocated);
402 printf("Total bytes wasted on internal fragmentation: %lu\n",
403 total_allocated - total_requested);
404 printf("Internal fragmentation: %f%%\n",
405 fragmentation(total_requested, total_allocated));
Li Zefan7d0d3942009-11-24 13:26:31 +0800406 printf("Cross CPU allocations: %lu/%lu\n", nr_cross_allocs, nr_allocs);
Li Zefanba77c9e2009-11-20 15:53:25 +0800407}
408
Arnaldo Carvalho de Melo4aa65632009-12-13 19:50:29 -0200409static void print_result(struct perf_session *session)
Li Zefanba77c9e2009-11-20 15:53:25 +0800410{
411 if (caller_flag)
Arnaldo Carvalho de Melo4aa65632009-12-13 19:50:29 -0200412 __print_result(&root_caller_sorted, session, caller_lines, 1);
Li Zefanba77c9e2009-11-20 15:53:25 +0800413 if (alloc_flag)
Arnaldo Carvalho de Melo4aa65632009-12-13 19:50:29 -0200414 __print_result(&root_alloc_sorted, session, alloc_lines, 0);
Li Zefanba77c9e2009-11-20 15:53:25 +0800415 print_summary();
416}
417
Li Zefan29b3e152009-11-24 13:26:10 +0800418struct sort_dimension {
419 const char name[20];
420 sort_fn_t cmp;
421 struct list_head list;
422};
423
424static LIST_HEAD(caller_sort);
425static LIST_HEAD(alloc_sort);
426
Li Zefanba77c9e2009-11-20 15:53:25 +0800427static void sort_insert(struct rb_root *root, struct alloc_stat *data,
Li Zefan29b3e152009-11-24 13:26:10 +0800428 struct list_head *sort_list)
Li Zefanba77c9e2009-11-20 15:53:25 +0800429{
430 struct rb_node **new = &(root->rb_node);
431 struct rb_node *parent = NULL;
Li Zefan29b3e152009-11-24 13:26:10 +0800432 struct sort_dimension *sort;
Li Zefanba77c9e2009-11-20 15:53:25 +0800433
434 while (*new) {
435 struct alloc_stat *this;
Li Zefan29b3e152009-11-24 13:26:10 +0800436 int cmp = 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800437
438 this = rb_entry(*new, struct alloc_stat, node);
439 parent = *new;
440
Li Zefan29b3e152009-11-24 13:26:10 +0800441 list_for_each_entry(sort, sort_list, list) {
442 cmp = sort->cmp(data, this);
443 if (cmp)
444 break;
445 }
Li Zefanba77c9e2009-11-20 15:53:25 +0800446
447 if (cmp > 0)
448 new = &((*new)->rb_left);
449 else
450 new = &((*new)->rb_right);
451 }
452
453 rb_link_node(&data->node, parent, new);
454 rb_insert_color(&data->node, root);
455}
456
457static void __sort_result(struct rb_root *root, struct rb_root *root_sorted,
Li Zefan29b3e152009-11-24 13:26:10 +0800458 struct list_head *sort_list)
Li Zefanba77c9e2009-11-20 15:53:25 +0800459{
460 struct rb_node *node;
461 struct alloc_stat *data;
462
463 for (;;) {
464 node = rb_first(root);
465 if (!node)
466 break;
467
468 rb_erase(node, root);
469 data = rb_entry(node, struct alloc_stat, node);
Li Zefan29b3e152009-11-24 13:26:10 +0800470 sort_insert(root_sorted, data, sort_list);
Li Zefanba77c9e2009-11-20 15:53:25 +0800471 }
472}
473
474static void sort_result(void)
475{
Li Zefan29b3e152009-11-24 13:26:10 +0800476 __sort_result(&root_alloc_stat, &root_alloc_sorted, &alloc_sort);
477 __sort_result(&root_caller_stat, &root_caller_sorted, &caller_sort);
Li Zefanba77c9e2009-11-20 15:53:25 +0800478}
479
Feng Tang70cb4e92012-10-30 11:56:02 +0800480static int __cmd_kmem(void)
Li Zefanba77c9e2009-11-20 15:53:25 +0800481{
Arnaldo Carvalho de Melod549c7692009-12-27 21:37:02 -0200482 int err = -EINVAL;
Arnaldo Carvalho de Meloda378962012-06-27 13:08:42 -0300483 struct perf_session *session;
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300484 const struct perf_evsel_str_handler kmem_tracepoints[] = {
485 { "kmem:kmalloc", perf_evsel__process_alloc_event, },
486 { "kmem:kmem_cache_alloc", perf_evsel__process_alloc_event, },
487 { "kmem:kmalloc_node", perf_evsel__process_alloc_node_event, },
488 { "kmem:kmem_cache_alloc_node", perf_evsel__process_alloc_node_event, },
489 { "kmem:kfree", perf_evsel__process_free_event, },
490 { "kmem:kmem_cache_free", perf_evsel__process_free_event, },
491 };
Arnaldo Carvalho de Meloda378962012-06-27 13:08:42 -0300492
Arnaldo Carvalho de Melofcf65bf2012-08-07 09:58:03 -0300493 session = perf_session__new(input_name, O_RDONLY, 0, false, &perf_kmem);
Arnaldo Carvalho de Melo4aa65632009-12-13 19:50:29 -0200494 if (session == NULL)
495 return -ENOMEM;
Li Zefanba77c9e2009-11-20 15:53:25 +0800496
Arnaldo Carvalho de Meloe727ca72010-04-01 19:12:13 -0300497 if (perf_session__create_kernel_maps(session) < 0)
498 goto out_delete;
499
Arnaldo Carvalho de Melod549c7692009-12-27 21:37:02 -0200500 if (!perf_session__has_traces(session, "kmem record"))
501 goto out_delete;
502
Arnaldo Carvalho de Melo0f7d2f12012-09-24 10:46:54 -0300503 if (perf_session__set_tracepoints_handlers(session, kmem_tracepoints)) {
504 pr_err("Initializing perf session tracepoint handlers failed\n");
505 return -1;
506 }
507
Arnaldo Carvalho de Melo4aa65632009-12-13 19:50:29 -0200508 setup_pager();
Arnaldo Carvalho de Melofcf65bf2012-08-07 09:58:03 -0300509 err = perf_session__process_events(session, &perf_kmem);
Arnaldo Carvalho de Melo4aa65632009-12-13 19:50:29 -0200510 if (err != 0)
511 goto out_delete;
512 sort_result();
513 print_result(session);
514out_delete:
515 perf_session__delete(session);
516 return err;
Li Zefanba77c9e2009-11-20 15:53:25 +0800517}
518
Li Zefanba77c9e2009-11-20 15:53:25 +0800519static int ptr_cmp(struct alloc_stat *l, struct alloc_stat *r)
520{
521 if (l->ptr < r->ptr)
522 return -1;
523 else if (l->ptr > r->ptr)
524 return 1;
525 return 0;
526}
527
Li Zefan29b3e152009-11-24 13:26:10 +0800528static struct sort_dimension ptr_sort_dimension = {
529 .name = "ptr",
530 .cmp = ptr_cmp,
531};
532
Li Zefanba77c9e2009-11-20 15:53:25 +0800533static int callsite_cmp(struct alloc_stat *l, struct alloc_stat *r)
534{
535 if (l->call_site < r->call_site)
536 return -1;
537 else if (l->call_site > r->call_site)
538 return 1;
539 return 0;
540}
541
Li Zefan29b3e152009-11-24 13:26:10 +0800542static struct sort_dimension callsite_sort_dimension = {
543 .name = "callsite",
544 .cmp = callsite_cmp,
545};
546
Pekka Enbergf3ced7c2009-11-22 11:58:00 +0200547static int hit_cmp(struct alloc_stat *l, struct alloc_stat *r)
548{
549 if (l->hit < r->hit)
550 return -1;
551 else if (l->hit > r->hit)
552 return 1;
553 return 0;
554}
555
Li Zefan29b3e152009-11-24 13:26:10 +0800556static struct sort_dimension hit_sort_dimension = {
557 .name = "hit",
558 .cmp = hit_cmp,
559};
560
Li Zefanba77c9e2009-11-20 15:53:25 +0800561static int bytes_cmp(struct alloc_stat *l, struct alloc_stat *r)
562{
563 if (l->bytes_alloc < r->bytes_alloc)
564 return -1;
565 else if (l->bytes_alloc > r->bytes_alloc)
566 return 1;
567 return 0;
568}
569
Li Zefan29b3e152009-11-24 13:26:10 +0800570static struct sort_dimension bytes_sort_dimension = {
571 .name = "bytes",
572 .cmp = bytes_cmp,
573};
574
Pekka Enbergf3ced7c2009-11-22 11:58:00 +0200575static int frag_cmp(struct alloc_stat *l, struct alloc_stat *r)
576{
577 double x, y;
578
579 x = fragmentation(l->bytes_req, l->bytes_alloc);
580 y = fragmentation(r->bytes_req, r->bytes_alloc);
581
582 if (x < y)
583 return -1;
584 else if (x > y)
585 return 1;
586 return 0;
587}
588
Li Zefan29b3e152009-11-24 13:26:10 +0800589static struct sort_dimension frag_sort_dimension = {
590 .name = "frag",
591 .cmp = frag_cmp,
592};
593
Li Zefan079d3f62009-11-24 13:26:55 +0800594static int pingpong_cmp(struct alloc_stat *l, struct alloc_stat *r)
595{
596 if (l->pingpong < r->pingpong)
597 return -1;
598 else if (l->pingpong > r->pingpong)
599 return 1;
600 return 0;
601}
602
603static struct sort_dimension pingpong_sort_dimension = {
604 .name = "pingpong",
605 .cmp = pingpong_cmp,
606};
607
Li Zefan29b3e152009-11-24 13:26:10 +0800608static struct sort_dimension *avail_sorts[] = {
609 &ptr_sort_dimension,
610 &callsite_sort_dimension,
611 &hit_sort_dimension,
612 &bytes_sort_dimension,
613 &frag_sort_dimension,
Li Zefan079d3f62009-11-24 13:26:55 +0800614 &pingpong_sort_dimension,
Li Zefan29b3e152009-11-24 13:26:10 +0800615};
616
617#define NUM_AVAIL_SORTS \
618 (int)(sizeof(avail_sorts) / sizeof(struct sort_dimension *))
619
620static int sort_dimension__add(const char *tok, struct list_head *list)
621{
622 struct sort_dimension *sort;
623 int i;
624
625 for (i = 0; i < NUM_AVAIL_SORTS; i++) {
626 if (!strcmp(avail_sorts[i]->name, tok)) {
627 sort = malloc(sizeof(*sort));
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300628 if (!sort) {
629 pr_err("%s: malloc failed\n", __func__);
630 return -1;
631 }
Li Zefan29b3e152009-11-24 13:26:10 +0800632 memcpy(sort, avail_sorts[i], sizeof(*sort));
633 list_add_tail(&sort->list, list);
634 return 0;
635 }
636 }
637
638 return -1;
639}
640
641static int setup_sorting(struct list_head *sort_list, const char *arg)
642{
643 char *tok;
644 char *str = strdup(arg);
645
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300646 if (!str) {
647 pr_err("%s: strdup failed\n", __func__);
648 return -1;
649 }
Li Zefan29b3e152009-11-24 13:26:10 +0800650
651 while (true) {
652 tok = strsep(&str, ",");
653 if (!tok)
654 break;
655 if (sort_dimension__add(tok, sort_list) < 0) {
656 error("Unknown --sort key: '%s'", tok);
Namhyung Kim1b228592012-01-08 02:25:29 +0900657 free(str);
Li Zefan29b3e152009-11-24 13:26:10 +0800658 return -1;
659 }
660 }
661
662 free(str);
663 return 0;
664}
665
Irina Tirdea1d037ca2012-09-11 01:15:03 +0300666static int parse_sort_opt(const struct option *opt __maybe_unused,
667 const char *arg, int unset __maybe_unused)
Li Zefanba77c9e2009-11-20 15:53:25 +0800668{
Li Zefanba77c9e2009-11-20 15:53:25 +0800669 if (!arg)
670 return -1;
671
Li Zefanba77c9e2009-11-20 15:53:25 +0800672 if (caller_flag > alloc_flag)
Li Zefan29b3e152009-11-24 13:26:10 +0800673 return setup_sorting(&caller_sort, arg);
Li Zefanba77c9e2009-11-20 15:53:25 +0800674 else
Li Zefan29b3e152009-11-24 13:26:10 +0800675 return setup_sorting(&alloc_sort, arg);
Li Zefanba77c9e2009-11-20 15:53:25 +0800676
677 return 0;
678}
679
Irina Tirdea1d037ca2012-09-11 01:15:03 +0300680static int parse_caller_opt(const struct option *opt __maybe_unused,
681 const char *arg __maybe_unused,
682 int unset __maybe_unused)
Li Zefanba77c9e2009-11-20 15:53:25 +0800683{
Li Zefan90b86a92009-12-10 15:21:57 +0800684 caller_flag = (alloc_flag + 1);
685 return 0;
686}
Li Zefanba77c9e2009-11-20 15:53:25 +0800687
Irina Tirdea1d037ca2012-09-11 01:15:03 +0300688static int parse_alloc_opt(const struct option *opt __maybe_unused,
689 const char *arg __maybe_unused,
690 int unset __maybe_unused)
Li Zefan90b86a92009-12-10 15:21:57 +0800691{
692 alloc_flag = (caller_flag + 1);
Li Zefanba77c9e2009-11-20 15:53:25 +0800693 return 0;
694}
695
Irina Tirdea1d037ca2012-09-11 01:15:03 +0300696static int parse_line_opt(const struct option *opt __maybe_unused,
697 const char *arg, int unset __maybe_unused)
Li Zefanba77c9e2009-11-20 15:53:25 +0800698{
699 int lines;
700
701 if (!arg)
702 return -1;
703
704 lines = strtoul(arg, NULL, 10);
705
706 if (caller_flag > alloc_flag)
707 caller_lines = lines;
708 else
709 alloc_lines = lines;
710
711 return 0;
712}
713
Arnaldo Carvalho de Melo0433ffb2012-10-01 15:20:58 -0300714static int __cmd_record(int argc, const char **argv)
715{
716 const char * const record_args[] = {
717 "record", "-a", "-R", "-f", "-c", "1",
Li Zefanba77c9e2009-11-20 15:53:25 +0800718 "-e", "kmem:kmalloc",
719 "-e", "kmem:kmalloc_node",
720 "-e", "kmem:kfree",
721 "-e", "kmem:kmem_cache_alloc",
722 "-e", "kmem:kmem_cache_alloc_node",
723 "-e", "kmem:kmem_cache_free",
Arnaldo Carvalho de Melo0433ffb2012-10-01 15:20:58 -0300724 };
Li Zefanba77c9e2009-11-20 15:53:25 +0800725 unsigned int rec_argc, i, j;
726 const char **rec_argv;
727
728 rec_argc = ARRAY_SIZE(record_args) + argc - 1;
729 rec_argv = calloc(rec_argc + 1, sizeof(char *));
730
Chris Samuelce47dc52010-11-13 13:35:06 +1100731 if (rec_argv == NULL)
732 return -ENOMEM;
733
Li Zefanba77c9e2009-11-20 15:53:25 +0800734 for (i = 0; i < ARRAY_SIZE(record_args); i++)
735 rec_argv[i] = strdup(record_args[i]);
736
737 for (j = 1; j < (unsigned int)argc; j++, i++)
738 rec_argv[i] = argv[j];
739
740 return cmd_record(i, rec_argv, NULL);
741}
742
Irina Tirdea1d037ca2012-09-11 01:15:03 +0300743int cmd_kmem(int argc, const char **argv, const char *prefix __maybe_unused)
Li Zefanba77c9e2009-11-20 15:53:25 +0800744{
Arnaldo Carvalho de Melo0433ffb2012-10-01 15:20:58 -0300745 const char * const default_sort_order = "frag,hit,bytes";
Arnaldo Carvalho de Melo0433ffb2012-10-01 15:20:58 -0300746 const struct option kmem_options[] = {
747 OPT_STRING('i', "input", &input_name, "file", "input file name"),
748 OPT_CALLBACK_NOOPT(0, "caller", NULL, NULL,
749 "show per-callsite statistics", parse_caller_opt),
750 OPT_CALLBACK_NOOPT(0, "alloc", NULL, NULL,
751 "show per-allocation statistics", parse_alloc_opt),
752 OPT_CALLBACK('s', "sort", NULL, "key[,key2...]",
753 "sort by keys: ptr, call_site, bytes, hit, pingpong, frag",
754 parse_sort_opt),
755 OPT_CALLBACK('l', "line", NULL, "num", "show n lines", parse_line_opt),
756 OPT_BOOLEAN(0, "raw-ip", &raw_ip, "show raw ip instead of symbol"),
757 OPT_END()
758 };
759 const char * const kmem_usage[] = {
760 "perf kmem [<options>] {record|stat}",
761 NULL
762 };
Li Zefanba77c9e2009-11-20 15:53:25 +0800763 argc = parse_options(argc, argv, kmem_options, kmem_usage, 0);
764
Li Zefan90b86a92009-12-10 15:21:57 +0800765 if (!argc)
Li Zefanba77c9e2009-11-20 15:53:25 +0800766 usage_with_options(kmem_usage, kmem_options);
767
Arnaldo Carvalho de Melo655000e2009-12-15 20:04:40 -0200768 symbol__init();
769
Li Zefan90b86a92009-12-10 15:21:57 +0800770 if (!strncmp(argv[0], "rec", 3)) {
771 return __cmd_record(argc, argv);
772 } else if (!strcmp(argv[0], "stat")) {
Arnaldo Carvalho de Melo2814eb02012-09-08 22:53:06 -0300773 if (setup_cpunode_map())
774 return -1;
Li Zefanba77c9e2009-11-20 15:53:25 +0800775
Li Zefan90b86a92009-12-10 15:21:57 +0800776 if (list_empty(&caller_sort))
777 setup_sorting(&caller_sort, default_sort_order);
778 if (list_empty(&alloc_sort))
779 setup_sorting(&alloc_sort, default_sort_order);
Li Zefan7d0d3942009-11-24 13:26:31 +0800780
Feng Tang70cb4e92012-10-30 11:56:02 +0800781 return __cmd_kmem();
Pekka Enbergb00eca82010-01-19 19:26:11 +0200782 } else
783 usage_with_options(kmem_usage, kmem_options);
Li Zefan90b86a92009-12-10 15:21:57 +0800784
785 return 0;
Li Zefanba77c9e2009-11-20 15:53:25 +0800786}
787