xref: /linux/tools/perf/builtin-kmem.c (revision 8f8d5745bb520c76b81abef4a2cb3023d0313bfd)
1 // SPDX-License-Identifier: GPL-2.0
2 #include "builtin.h"
3 #include "perf.h"
4 
5 #include "util/evlist.h"
6 #include "util/evsel.h"
7 #include "util/util.h"
8 #include "util/config.h"
9 #include "util/map.h"
10 #include "util/symbol.h"
11 #include "util/thread.h"
12 #include "util/header.h"
13 #include "util/session.h"
14 #include "util/tool.h"
15 #include "util/callchain.h"
16 #include "util/time-utils.h"
17 
18 #include <subcmd/parse-options.h>
19 #include "util/trace-event.h"
20 #include "util/data.h"
21 #include "util/cpumap.h"
22 
23 #include "util/debug.h"
24 
25 #include <linux/kernel.h>
26 #include <linux/rbtree.h>
27 #include <linux/string.h>
28 #include <errno.h>
29 #include <inttypes.h>
30 #include <locale.h>
31 #include <regex.h>
32 
33 #include "sane_ctype.h"
34 
35 static int	kmem_slab;
36 static int	kmem_page;
37 
38 static long	kmem_page_size;
39 static enum {
40 	KMEM_SLAB,
41 	KMEM_PAGE,
42 } kmem_default = KMEM_SLAB;  /* for backward compatibility */
43 
44 struct alloc_stat;
45 typedef int (*sort_fn_t)(void *, void *);
46 
47 static int			alloc_flag;
48 static int			caller_flag;
49 
50 static int			alloc_lines = -1;
51 static int			caller_lines = -1;
52 
53 static bool			raw_ip;
54 
55 struct alloc_stat {
56 	u64	call_site;
57 	u64	ptr;
58 	u64	bytes_req;
59 	u64	bytes_alloc;
60 	u64	last_alloc;
61 	u32	hit;
62 	u32	pingpong;
63 
64 	short	alloc_cpu;
65 
66 	struct rb_node node;
67 };
68 
69 static struct rb_root root_alloc_stat;
70 static struct rb_root root_alloc_sorted;
71 static struct rb_root root_caller_stat;
72 static struct rb_root root_caller_sorted;
73 
74 static unsigned long total_requested, total_allocated, total_freed;
75 static unsigned long nr_allocs, nr_cross_allocs;
76 
77 /* filters for controlling start and stop of time of analysis */
78 static struct perf_time_interval ptime;
79 const char *time_str;
80 
81 static int insert_alloc_stat(unsigned long call_site, unsigned long ptr,
82 			     int bytes_req, int bytes_alloc, int cpu)
83 {
84 	struct rb_node **node = &root_alloc_stat.rb_node;
85 	struct rb_node *parent = NULL;
86 	struct alloc_stat *data = NULL;
87 
88 	while (*node) {
89 		parent = *node;
90 		data = rb_entry(*node, struct alloc_stat, node);
91 
92 		if (ptr > data->ptr)
93 			node = &(*node)->rb_right;
94 		else if (ptr < data->ptr)
95 			node = &(*node)->rb_left;
96 		else
97 			break;
98 	}
99 
100 	if (data && data->ptr == ptr) {
101 		data->hit++;
102 		data->bytes_req += bytes_req;
103 		data->bytes_alloc += bytes_alloc;
104 	} else {
105 		data = malloc(sizeof(*data));
106 		if (!data) {
107 			pr_err("%s: malloc failed\n", __func__);
108 			return -1;
109 		}
110 		data->ptr = ptr;
111 		data->pingpong = 0;
112 		data->hit = 1;
113 		data->bytes_req = bytes_req;
114 		data->bytes_alloc = bytes_alloc;
115 
116 		rb_link_node(&data->node, parent, node);
117 		rb_insert_color(&data->node, &root_alloc_stat);
118 	}
119 	data->call_site = call_site;
120 	data->alloc_cpu = cpu;
121 	data->last_alloc = bytes_alloc;
122 
123 	return 0;
124 }
125 
126 static int insert_caller_stat(unsigned long call_site,
127 			      int bytes_req, int bytes_alloc)
128 {
129 	struct rb_node **node = &root_caller_stat.rb_node;
130 	struct rb_node *parent = NULL;
131 	struct alloc_stat *data = NULL;
132 
133 	while (*node) {
134 		parent = *node;
135 		data = rb_entry(*node, struct alloc_stat, node);
136 
137 		if (call_site > data->call_site)
138 			node = &(*node)->rb_right;
139 		else if (call_site < data->call_site)
140 			node = &(*node)->rb_left;
141 		else
142 			break;
143 	}
144 
145 	if (data && data->call_site == call_site) {
146 		data->hit++;
147 		data->bytes_req += bytes_req;
148 		data->bytes_alloc += bytes_alloc;
149 	} else {
150 		data = malloc(sizeof(*data));
151 		if (!data) {
152 			pr_err("%s: malloc failed\n", __func__);
153 			return -1;
154 		}
155 		data->call_site = call_site;
156 		data->pingpong = 0;
157 		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_caller_stat);
163 	}
164 
165 	return 0;
166 }
167 
168 static int perf_evsel__process_alloc_event(struct perf_evsel *evsel,
169 					   struct perf_sample *sample)
170 {
171 	unsigned long ptr = perf_evsel__intval(evsel, sample, "ptr"),
172 		      call_site = perf_evsel__intval(evsel, sample, "call_site");
173 	int bytes_req = perf_evsel__intval(evsel, sample, "bytes_req"),
174 	    bytes_alloc = perf_evsel__intval(evsel, sample, "bytes_alloc");
175 
176 	if (insert_alloc_stat(call_site, ptr, bytes_req, bytes_alloc, sample->cpu) ||
177 	    insert_caller_stat(call_site, bytes_req, bytes_alloc))
178 		return -1;
179 
180 	total_requested += bytes_req;
181 	total_allocated += bytes_alloc;
182 
183 	nr_allocs++;
184 	return 0;
185 }
186 
187 static int perf_evsel__process_alloc_node_event(struct perf_evsel *evsel,
188 						struct perf_sample *sample)
189 {
190 	int ret = perf_evsel__process_alloc_event(evsel, sample);
191 
192 	if (!ret) {
193 		int node1 = cpu__get_node(sample->cpu),
194 		    node2 = perf_evsel__intval(evsel, sample, "node");
195 
196 		if (node1 != node2)
197 			nr_cross_allocs++;
198 	}
199 
200 	return ret;
201 }
202 
203 static int ptr_cmp(void *, void *);
204 static int slab_callsite_cmp(void *, void *);
205 
206 static struct alloc_stat *search_alloc_stat(unsigned long ptr,
207 					    unsigned long call_site,
208 					    struct rb_root *root,
209 					    sort_fn_t sort_fn)
210 {
211 	struct rb_node *node = root->rb_node;
212 	struct alloc_stat key = { .ptr = ptr, .call_site = call_site };
213 
214 	while (node) {
215 		struct alloc_stat *data;
216 		int cmp;
217 
218 		data = rb_entry(node, struct alloc_stat, node);
219 
220 		cmp = sort_fn(&key, data);
221 		if (cmp < 0)
222 			node = node->rb_left;
223 		else if (cmp > 0)
224 			node = node->rb_right;
225 		else
226 			return data;
227 	}
228 	return NULL;
229 }
230 
231 static int perf_evsel__process_free_event(struct perf_evsel *evsel,
232 					  struct perf_sample *sample)
233 {
234 	unsigned long ptr = perf_evsel__intval(evsel, sample, "ptr");
235 	struct alloc_stat *s_alloc, *s_caller;
236 
237 	s_alloc = search_alloc_stat(ptr, 0, &root_alloc_stat, ptr_cmp);
238 	if (!s_alloc)
239 		return 0;
240 
241 	total_freed += s_alloc->last_alloc;
242 
243 	if ((short)sample->cpu != s_alloc->alloc_cpu) {
244 		s_alloc->pingpong++;
245 
246 		s_caller = search_alloc_stat(0, s_alloc->call_site,
247 					     &root_caller_stat,
248 					     slab_callsite_cmp);
249 		if (!s_caller)
250 			return -1;
251 		s_caller->pingpong++;
252 	}
253 	s_alloc->alloc_cpu = -1;
254 
255 	return 0;
256 }
257 
258 static u64 total_page_alloc_bytes;
259 static u64 total_page_free_bytes;
260 static u64 total_page_nomatch_bytes;
261 static u64 total_page_fail_bytes;
262 static unsigned long nr_page_allocs;
263 static unsigned long nr_page_frees;
264 static unsigned long nr_page_fails;
265 static unsigned long nr_page_nomatch;
266 
267 static bool use_pfn;
268 static bool live_page;
269 static struct perf_session *kmem_session;
270 
271 #define MAX_MIGRATE_TYPES  6
272 #define MAX_PAGE_ORDER     11
273 
274 static int order_stats[MAX_PAGE_ORDER][MAX_MIGRATE_TYPES];
275 
276 struct page_stat {
277 	struct rb_node 	node;
278 	u64 		page;
279 	u64 		callsite;
280 	int 		order;
281 	unsigned 	gfp_flags;
282 	unsigned 	migrate_type;
283 	u64		alloc_bytes;
284 	u64 		free_bytes;
285 	int 		nr_alloc;
286 	int 		nr_free;
287 };
288 
289 static struct rb_root page_live_tree;
290 static struct rb_root page_alloc_tree;
291 static struct rb_root page_alloc_sorted;
292 static struct rb_root page_caller_tree;
293 static struct rb_root page_caller_sorted;
294 
295 struct alloc_func {
296 	u64 start;
297 	u64 end;
298 	char *name;
299 };
300 
301 static int nr_alloc_funcs;
302 static struct alloc_func *alloc_func_list;
303 
304 static int funcmp(const void *a, const void *b)
305 {
306 	const struct alloc_func *fa = a;
307 	const struct alloc_func *fb = b;
308 
309 	if (fa->start > fb->start)
310 		return 1;
311 	else
312 		return -1;
313 }
314 
315 static int callcmp(const void *a, const void *b)
316 {
317 	const struct alloc_func *fa = a;
318 	const struct alloc_func *fb = b;
319 
320 	if (fb->start <= fa->start && fa->end < fb->end)
321 		return 0;
322 
323 	if (fa->start > fb->start)
324 		return 1;
325 	else
326 		return -1;
327 }
328 
329 static int build_alloc_func_list(void)
330 {
331 	int ret;
332 	struct map *kernel_map;
333 	struct symbol *sym;
334 	struct rb_node *node;
335 	struct alloc_func *func;
336 	struct machine *machine = &kmem_session->machines.host;
337 	regex_t alloc_func_regex;
338 	static const char pattern[] = "^_?_?(alloc|get_free|get_zeroed)_pages?";
339 
340 	ret = regcomp(&alloc_func_regex, pattern, REG_EXTENDED);
341 	if (ret) {
342 		char err[BUFSIZ];
343 
344 		regerror(ret, &alloc_func_regex, err, sizeof(err));
345 		pr_err("Invalid regex: %s\n%s", pattern, err);
346 		return -EINVAL;
347 	}
348 
349 	kernel_map = machine__kernel_map(machine);
350 	if (map__load(kernel_map) < 0) {
351 		pr_err("cannot load kernel map\n");
352 		return -ENOENT;
353 	}
354 
355 	map__for_each_symbol(kernel_map, sym, node) {
356 		if (regexec(&alloc_func_regex, sym->name, 0, NULL, 0))
357 			continue;
358 
359 		func = realloc(alloc_func_list,
360 			       (nr_alloc_funcs + 1) * sizeof(*func));
361 		if (func == NULL)
362 			return -ENOMEM;
363 
364 		pr_debug("alloc func: %s\n", sym->name);
365 		func[nr_alloc_funcs].start = sym->start;
366 		func[nr_alloc_funcs].end   = sym->end;
367 		func[nr_alloc_funcs].name  = sym->name;
368 
369 		alloc_func_list = func;
370 		nr_alloc_funcs++;
371 	}
372 
373 	qsort(alloc_func_list, nr_alloc_funcs, sizeof(*func), funcmp);
374 
375 	regfree(&alloc_func_regex);
376 	return 0;
377 }
378 
379 /*
380  * Find first non-memory allocation function from callchain.
381  * The allocation functions are in the 'alloc_func_list'.
382  */
383 static u64 find_callsite(struct perf_evsel *evsel, struct perf_sample *sample)
384 {
385 	struct addr_location al;
386 	struct machine *machine = &kmem_session->machines.host;
387 	struct callchain_cursor_node *node;
388 
389 	if (alloc_func_list == NULL) {
390 		if (build_alloc_func_list() < 0)
391 			goto out;
392 	}
393 
394 	al.thread = machine__findnew_thread(machine, sample->pid, sample->tid);
395 	sample__resolve_callchain(sample, &callchain_cursor, NULL, evsel, &al, 16);
396 
397 	callchain_cursor_commit(&callchain_cursor);
398 	while (true) {
399 		struct alloc_func key, *caller;
400 		u64 addr;
401 
402 		node = callchain_cursor_current(&callchain_cursor);
403 		if (node == NULL)
404 			break;
405 
406 		key.start = key.end = node->ip;
407 		caller = bsearch(&key, alloc_func_list, nr_alloc_funcs,
408 				 sizeof(key), callcmp);
409 		if (!caller) {
410 			/* found */
411 			if (node->map)
412 				addr = map__unmap_ip(node->map, node->ip);
413 			else
414 				addr = node->ip;
415 
416 			return addr;
417 		} else
418 			pr_debug3("skipping alloc function: %s\n", caller->name);
419 
420 		callchain_cursor_advance(&callchain_cursor);
421 	}
422 
423 out:
424 	pr_debug2("unknown callsite: %"PRIx64 "\n", sample->ip);
425 	return sample->ip;
426 }
427 
428 struct sort_dimension {
429 	const char		name[20];
430 	sort_fn_t		cmp;
431 	struct list_head	list;
432 };
433 
434 static LIST_HEAD(page_alloc_sort_input);
435 static LIST_HEAD(page_caller_sort_input);
436 
437 static struct page_stat *
438 __page_stat__findnew_page(struct page_stat *pstat, bool create)
439 {
440 	struct rb_node **node = &page_live_tree.rb_node;
441 	struct rb_node *parent = NULL;
442 	struct page_stat *data;
443 
444 	while (*node) {
445 		s64 cmp;
446 
447 		parent = *node;
448 		data = rb_entry(*node, struct page_stat, node);
449 
450 		cmp = data->page - pstat->page;
451 		if (cmp < 0)
452 			node = &parent->rb_left;
453 		else if (cmp > 0)
454 			node = &parent->rb_right;
455 		else
456 			return data;
457 	}
458 
459 	if (!create)
460 		return NULL;
461 
462 	data = zalloc(sizeof(*data));
463 	if (data != NULL) {
464 		data->page = pstat->page;
465 		data->order = pstat->order;
466 		data->gfp_flags = pstat->gfp_flags;
467 		data->migrate_type = pstat->migrate_type;
468 
469 		rb_link_node(&data->node, parent, node);
470 		rb_insert_color(&data->node, &page_live_tree);
471 	}
472 
473 	return data;
474 }
475 
476 static struct page_stat *page_stat__find_page(struct page_stat *pstat)
477 {
478 	return __page_stat__findnew_page(pstat, false);
479 }
480 
481 static struct page_stat *page_stat__findnew_page(struct page_stat *pstat)
482 {
483 	return __page_stat__findnew_page(pstat, true);
484 }
485 
486 static struct page_stat *
487 __page_stat__findnew_alloc(struct page_stat *pstat, bool create)
488 {
489 	struct rb_node **node = &page_alloc_tree.rb_node;
490 	struct rb_node *parent = NULL;
491 	struct page_stat *data;
492 	struct sort_dimension *sort;
493 
494 	while (*node) {
495 		int cmp = 0;
496 
497 		parent = *node;
498 		data = rb_entry(*node, struct page_stat, node);
499 
500 		list_for_each_entry(sort, &page_alloc_sort_input, list) {
501 			cmp = sort->cmp(pstat, data);
502 			if (cmp)
503 				break;
504 		}
505 
506 		if (cmp < 0)
507 			node = &parent->rb_left;
508 		else if (cmp > 0)
509 			node = &parent->rb_right;
510 		else
511 			return data;
512 	}
513 
514 	if (!create)
515 		return NULL;
516 
517 	data = zalloc(sizeof(*data));
518 	if (data != NULL) {
519 		data->page = pstat->page;
520 		data->order = pstat->order;
521 		data->gfp_flags = pstat->gfp_flags;
522 		data->migrate_type = pstat->migrate_type;
523 
524 		rb_link_node(&data->node, parent, node);
525 		rb_insert_color(&data->node, &page_alloc_tree);
526 	}
527 
528 	return data;
529 }
530 
531 static struct page_stat *page_stat__find_alloc(struct page_stat *pstat)
532 {
533 	return __page_stat__findnew_alloc(pstat, false);
534 }
535 
536 static struct page_stat *page_stat__findnew_alloc(struct page_stat *pstat)
537 {
538 	return __page_stat__findnew_alloc(pstat, true);
539 }
540 
541 static struct page_stat *
542 __page_stat__findnew_caller(struct page_stat *pstat, bool create)
543 {
544 	struct rb_node **node = &page_caller_tree.rb_node;
545 	struct rb_node *parent = NULL;
546 	struct page_stat *data;
547 	struct sort_dimension *sort;
548 
549 	while (*node) {
550 		int cmp = 0;
551 
552 		parent = *node;
553 		data = rb_entry(*node, struct page_stat, node);
554 
555 		list_for_each_entry(sort, &page_caller_sort_input, list) {
556 			cmp = sort->cmp(pstat, data);
557 			if (cmp)
558 				break;
559 		}
560 
561 		if (cmp < 0)
562 			node = &parent->rb_left;
563 		else if (cmp > 0)
564 			node = &parent->rb_right;
565 		else
566 			return data;
567 	}
568 
569 	if (!create)
570 		return NULL;
571 
572 	data = zalloc(sizeof(*data));
573 	if (data != NULL) {
574 		data->callsite = pstat->callsite;
575 		data->order = pstat->order;
576 		data->gfp_flags = pstat->gfp_flags;
577 		data->migrate_type = pstat->migrate_type;
578 
579 		rb_link_node(&data->node, parent, node);
580 		rb_insert_color(&data->node, &page_caller_tree);
581 	}
582 
583 	return data;
584 }
585 
586 static struct page_stat *page_stat__find_caller(struct page_stat *pstat)
587 {
588 	return __page_stat__findnew_caller(pstat, false);
589 }
590 
591 static struct page_stat *page_stat__findnew_caller(struct page_stat *pstat)
592 {
593 	return __page_stat__findnew_caller(pstat, true);
594 }
595 
596 static bool valid_page(u64 pfn_or_page)
597 {
598 	if (use_pfn && pfn_or_page == -1UL)
599 		return false;
600 	if (!use_pfn && pfn_or_page == 0)
601 		return false;
602 	return true;
603 }
604 
605 struct gfp_flag {
606 	unsigned int flags;
607 	char *compact_str;
608 	char *human_readable;
609 };
610 
611 static struct gfp_flag *gfps;
612 static int nr_gfps;
613 
614 static int gfpcmp(const void *a, const void *b)
615 {
616 	const struct gfp_flag *fa = a;
617 	const struct gfp_flag *fb = b;
618 
619 	return fa->flags - fb->flags;
620 }
621 
622 /* see include/trace/events/mmflags.h */
623 static const struct {
624 	const char *original;
625 	const char *compact;
626 } gfp_compact_table[] = {
627 	{ "GFP_TRANSHUGE",		"THP" },
628 	{ "GFP_TRANSHUGE_LIGHT",	"THL" },
629 	{ "GFP_HIGHUSER_MOVABLE",	"HUM" },
630 	{ "GFP_HIGHUSER",		"HU" },
631 	{ "GFP_USER",			"U" },
632 	{ "GFP_KERNEL_ACCOUNT",		"KAC" },
633 	{ "GFP_KERNEL",			"K" },
634 	{ "GFP_NOFS",			"NF" },
635 	{ "GFP_ATOMIC",			"A" },
636 	{ "GFP_NOIO",			"NI" },
637 	{ "GFP_NOWAIT",			"NW" },
638 	{ "GFP_DMA",			"D" },
639 	{ "__GFP_HIGHMEM",		"HM" },
640 	{ "GFP_DMA32",			"D32" },
641 	{ "__GFP_HIGH",			"H" },
642 	{ "__GFP_ATOMIC",		"_A" },
643 	{ "__GFP_IO",			"I" },
644 	{ "__GFP_FS",			"F" },
645 	{ "__GFP_NOWARN",		"NWR" },
646 	{ "__GFP_RETRY_MAYFAIL",	"R" },
647 	{ "__GFP_NOFAIL",		"NF" },
648 	{ "__GFP_NORETRY",		"NR" },
649 	{ "__GFP_COMP",			"C" },
650 	{ "__GFP_ZERO",			"Z" },
651 	{ "__GFP_NOMEMALLOC",		"NMA" },
652 	{ "__GFP_MEMALLOC",		"MA" },
653 	{ "__GFP_HARDWALL",		"HW" },
654 	{ "__GFP_THISNODE",		"TN" },
655 	{ "__GFP_RECLAIMABLE",		"RC" },
656 	{ "__GFP_MOVABLE",		"M" },
657 	{ "__GFP_ACCOUNT",		"AC" },
658 	{ "__GFP_WRITE",		"WR" },
659 	{ "__GFP_RECLAIM",		"R" },
660 	{ "__GFP_DIRECT_RECLAIM",	"DR" },
661 	{ "__GFP_KSWAPD_RECLAIM",	"KR" },
662 };
663 
664 static size_t max_gfp_len;
665 
666 static char *compact_gfp_flags(char *gfp_flags)
667 {
668 	char *orig_flags = strdup(gfp_flags);
669 	char *new_flags = NULL;
670 	char *str, *pos = NULL;
671 	size_t len = 0;
672 
673 	if (orig_flags == NULL)
674 		return NULL;
675 
676 	str = strtok_r(orig_flags, "|", &pos);
677 	while (str) {
678 		size_t i;
679 		char *new;
680 		const char *cpt;
681 
682 		for (i = 0; i < ARRAY_SIZE(gfp_compact_table); i++) {
683 			if (strcmp(gfp_compact_table[i].original, str))
684 				continue;
685 
686 			cpt = gfp_compact_table[i].compact;
687 			new = realloc(new_flags, len + strlen(cpt) + 2);
688 			if (new == NULL) {
689 				free(new_flags);
690 				return NULL;
691 			}
692 
693 			new_flags = new;
694 
695 			if (!len) {
696 				strcpy(new_flags, cpt);
697 			} else {
698 				strcat(new_flags, "|");
699 				strcat(new_flags, cpt);
700 				len++;
701 			}
702 
703 			len += strlen(cpt);
704 		}
705 
706 		str = strtok_r(NULL, "|", &pos);
707 	}
708 
709 	if (max_gfp_len < len)
710 		max_gfp_len = len;
711 
712 	free(orig_flags);
713 	return new_flags;
714 }
715 
716 static char *compact_gfp_string(unsigned long gfp_flags)
717 {
718 	struct gfp_flag key = {
719 		.flags = gfp_flags,
720 	};
721 	struct gfp_flag *gfp;
722 
723 	gfp = bsearch(&key, gfps, nr_gfps, sizeof(*gfps), gfpcmp);
724 	if (gfp)
725 		return gfp->compact_str;
726 
727 	return NULL;
728 }
729 
730 static int parse_gfp_flags(struct perf_evsel *evsel, struct perf_sample *sample,
731 			   unsigned int gfp_flags)
732 {
733 	struct tep_record record = {
734 		.cpu = sample->cpu,
735 		.data = sample->raw_data,
736 		.size = sample->raw_size,
737 	};
738 	struct trace_seq seq;
739 	char *str, *pos = NULL;
740 
741 	if (nr_gfps) {
742 		struct gfp_flag key = {
743 			.flags = gfp_flags,
744 		};
745 
746 		if (bsearch(&key, gfps, nr_gfps, sizeof(*gfps), gfpcmp))
747 			return 0;
748 	}
749 
750 	trace_seq_init(&seq);
751 	tep_event_info(&seq, evsel->tp_format, &record);
752 
753 	str = strtok_r(seq.buffer, " ", &pos);
754 	while (str) {
755 		if (!strncmp(str, "gfp_flags=", 10)) {
756 			struct gfp_flag *new;
757 
758 			new = realloc(gfps, (nr_gfps + 1) * sizeof(*gfps));
759 			if (new == NULL)
760 				return -ENOMEM;
761 
762 			gfps = new;
763 			new += nr_gfps++;
764 
765 			new->flags = gfp_flags;
766 			new->human_readable = strdup(str + 10);
767 			new->compact_str = compact_gfp_flags(str + 10);
768 			if (!new->human_readable || !new->compact_str)
769 				return -ENOMEM;
770 
771 			qsort(gfps, nr_gfps, sizeof(*gfps), gfpcmp);
772 		}
773 
774 		str = strtok_r(NULL, " ", &pos);
775 	}
776 
777 	trace_seq_destroy(&seq);
778 	return 0;
779 }
780 
781 static int perf_evsel__process_page_alloc_event(struct perf_evsel *evsel,
782 						struct perf_sample *sample)
783 {
784 	u64 page;
785 	unsigned int order = perf_evsel__intval(evsel, sample, "order");
786 	unsigned int gfp_flags = perf_evsel__intval(evsel, sample, "gfp_flags");
787 	unsigned int migrate_type = perf_evsel__intval(evsel, sample,
788 						       "migratetype");
789 	u64 bytes = kmem_page_size << order;
790 	u64 callsite;
791 	struct page_stat *pstat;
792 	struct page_stat this = {
793 		.order = order,
794 		.gfp_flags = gfp_flags,
795 		.migrate_type = migrate_type,
796 	};
797 
798 	if (use_pfn)
799 		page = perf_evsel__intval(evsel, sample, "pfn");
800 	else
801 		page = perf_evsel__intval(evsel, sample, "page");
802 
803 	nr_page_allocs++;
804 	total_page_alloc_bytes += bytes;
805 
806 	if (!valid_page(page)) {
807 		nr_page_fails++;
808 		total_page_fail_bytes += bytes;
809 
810 		return 0;
811 	}
812 
813 	if (parse_gfp_flags(evsel, sample, gfp_flags) < 0)
814 		return -1;
815 
816 	callsite = find_callsite(evsel, sample);
817 
818 	/*
819 	 * This is to find the current page (with correct gfp flags and
820 	 * migrate type) at free event.
821 	 */
822 	this.page = page;
823 	pstat = page_stat__findnew_page(&this);
824 	if (pstat == NULL)
825 		return -ENOMEM;
826 
827 	pstat->nr_alloc++;
828 	pstat->alloc_bytes += bytes;
829 	pstat->callsite = callsite;
830 
831 	if (!live_page) {
832 		pstat = page_stat__findnew_alloc(&this);
833 		if (pstat == NULL)
834 			return -ENOMEM;
835 
836 		pstat->nr_alloc++;
837 		pstat->alloc_bytes += bytes;
838 		pstat->callsite = callsite;
839 	}
840 
841 	this.callsite = callsite;
842 	pstat = page_stat__findnew_caller(&this);
843 	if (pstat == NULL)
844 		return -ENOMEM;
845 
846 	pstat->nr_alloc++;
847 	pstat->alloc_bytes += bytes;
848 
849 	order_stats[order][migrate_type]++;
850 
851 	return 0;
852 }
853 
854 static int perf_evsel__process_page_free_event(struct perf_evsel *evsel,
855 						struct perf_sample *sample)
856 {
857 	u64 page;
858 	unsigned int order = perf_evsel__intval(evsel, sample, "order");
859 	u64 bytes = kmem_page_size << order;
860 	struct page_stat *pstat;
861 	struct page_stat this = {
862 		.order = order,
863 	};
864 
865 	if (use_pfn)
866 		page = perf_evsel__intval(evsel, sample, "pfn");
867 	else
868 		page = perf_evsel__intval(evsel, sample, "page");
869 
870 	nr_page_frees++;
871 	total_page_free_bytes += bytes;
872 
873 	this.page = page;
874 	pstat = page_stat__find_page(&this);
875 	if (pstat == NULL) {
876 		pr_debug2("missing free at page %"PRIx64" (order: %d)\n",
877 			  page, order);
878 
879 		nr_page_nomatch++;
880 		total_page_nomatch_bytes += bytes;
881 
882 		return 0;
883 	}
884 
885 	this.gfp_flags = pstat->gfp_flags;
886 	this.migrate_type = pstat->migrate_type;
887 	this.callsite = pstat->callsite;
888 
889 	rb_erase(&pstat->node, &page_live_tree);
890 	free(pstat);
891 
892 	if (live_page) {
893 		order_stats[this.order][this.migrate_type]--;
894 	} else {
895 		pstat = page_stat__find_alloc(&this);
896 		if (pstat == NULL)
897 			return -ENOMEM;
898 
899 		pstat->nr_free++;
900 		pstat->free_bytes += bytes;
901 	}
902 
903 	pstat = page_stat__find_caller(&this);
904 	if (pstat == NULL)
905 		return -ENOENT;
906 
907 	pstat->nr_free++;
908 	pstat->free_bytes += bytes;
909 
910 	if (live_page) {
911 		pstat->nr_alloc--;
912 		pstat->alloc_bytes -= bytes;
913 
914 		if (pstat->nr_alloc == 0) {
915 			rb_erase(&pstat->node, &page_caller_tree);
916 			free(pstat);
917 		}
918 	}
919 
920 	return 0;
921 }
922 
923 static bool perf_kmem__skip_sample(struct perf_sample *sample)
924 {
925 	/* skip sample based on time? */
926 	if (perf_time__skip_sample(&ptime, sample->time))
927 		return true;
928 
929 	return false;
930 }
931 
932 typedef int (*tracepoint_handler)(struct perf_evsel *evsel,
933 				  struct perf_sample *sample);
934 
935 static int process_sample_event(struct perf_tool *tool __maybe_unused,
936 				union perf_event *event,
937 				struct perf_sample *sample,
938 				struct perf_evsel *evsel,
939 				struct machine *machine)
940 {
941 	int err = 0;
942 	struct thread *thread = machine__findnew_thread(machine, sample->pid,
943 							sample->tid);
944 
945 	if (thread == NULL) {
946 		pr_debug("problem processing %d event, skipping it.\n",
947 			 event->header.type);
948 		return -1;
949 	}
950 
951 	if (perf_kmem__skip_sample(sample))
952 		return 0;
953 
954 	dump_printf(" ... thread: %s:%d\n", thread__comm_str(thread), thread->tid);
955 
956 	if (evsel->handler != NULL) {
957 		tracepoint_handler f = evsel->handler;
958 		err = f(evsel, sample);
959 	}
960 
961 	thread__put(thread);
962 
963 	return err;
964 }
965 
966 static struct perf_tool perf_kmem = {
967 	.sample		 = process_sample_event,
968 	.comm		 = perf_event__process_comm,
969 	.mmap		 = perf_event__process_mmap,
970 	.mmap2		 = perf_event__process_mmap2,
971 	.namespaces	 = perf_event__process_namespaces,
972 	.ordered_events	 = true,
973 };
974 
975 static double fragmentation(unsigned long n_req, unsigned long n_alloc)
976 {
977 	if (n_alloc == 0)
978 		return 0.0;
979 	else
980 		return 100.0 - (100.0 * n_req / n_alloc);
981 }
982 
983 static void __print_slab_result(struct rb_root *root,
984 				struct perf_session *session,
985 				int n_lines, int is_caller)
986 {
987 	struct rb_node *next;
988 	struct machine *machine = &session->machines.host;
989 
990 	printf("%.105s\n", graph_dotted_line);
991 	printf(" %-34s |",  is_caller ? "Callsite": "Alloc Ptr");
992 	printf(" Total_alloc/Per | Total_req/Per   | Hit      | Ping-pong | Frag\n");
993 	printf("%.105s\n", graph_dotted_line);
994 
995 	next = rb_first(root);
996 
997 	while (next && n_lines--) {
998 		struct alloc_stat *data = rb_entry(next, struct alloc_stat,
999 						   node);
1000 		struct symbol *sym = NULL;
1001 		struct map *map;
1002 		char buf[BUFSIZ];
1003 		u64 addr;
1004 
1005 		if (is_caller) {
1006 			addr = data->call_site;
1007 			if (!raw_ip)
1008 				sym = machine__find_kernel_symbol(machine, addr, &map);
1009 		} else
1010 			addr = data->ptr;
1011 
1012 		if (sym != NULL)
1013 			snprintf(buf, sizeof(buf), "%s+%" PRIx64 "", sym->name,
1014 				 addr - map->unmap_ip(map, sym->start));
1015 		else
1016 			snprintf(buf, sizeof(buf), "%#" PRIx64 "", addr);
1017 		printf(" %-34s |", buf);
1018 
1019 		printf(" %9llu/%-5lu | %9llu/%-5lu | %8lu | %9lu | %6.3f%%\n",
1020 		       (unsigned long long)data->bytes_alloc,
1021 		       (unsigned long)data->bytes_alloc / data->hit,
1022 		       (unsigned long long)data->bytes_req,
1023 		       (unsigned long)data->bytes_req / data->hit,
1024 		       (unsigned long)data->hit,
1025 		       (unsigned long)data->pingpong,
1026 		       fragmentation(data->bytes_req, data->bytes_alloc));
1027 
1028 		next = rb_next(next);
1029 	}
1030 
1031 	if (n_lines == -1)
1032 		printf(" ...                                | ...             | ...             | ...      | ...       | ...   \n");
1033 
1034 	printf("%.105s\n", graph_dotted_line);
1035 }
1036 
1037 static const char * const migrate_type_str[] = {
1038 	"UNMOVABL",
1039 	"RECLAIM",
1040 	"MOVABLE",
1041 	"RESERVED",
1042 	"CMA/ISLT",
1043 	"UNKNOWN",
1044 };
1045 
1046 static void __print_page_alloc_result(struct perf_session *session, int n_lines)
1047 {
1048 	struct rb_node *next = rb_first(&page_alloc_sorted);
1049 	struct machine *machine = &session->machines.host;
1050 	const char *format;
1051 	int gfp_len = max(strlen("GFP flags"), max_gfp_len);
1052 
1053 	printf("\n%.105s\n", graph_dotted_line);
1054 	printf(" %-16s | %5s alloc (KB) | Hits      | Order | Mig.type | %-*s | Callsite\n",
1055 	       use_pfn ? "PFN" : "Page", live_page ? "Live" : "Total",
1056 	       gfp_len, "GFP flags");
1057 	printf("%.105s\n", graph_dotted_line);
1058 
1059 	if (use_pfn)
1060 		format = " %16llu | %'16llu | %'9d | %5d | %8s | %-*s | %s\n";
1061 	else
1062 		format = " %016llx | %'16llu | %'9d | %5d | %8s | %-*s | %s\n";
1063 
1064 	while (next && n_lines--) {
1065 		struct page_stat *data;
1066 		struct symbol *sym;
1067 		struct map *map;
1068 		char buf[32];
1069 		char *caller = buf;
1070 
1071 		data = rb_entry(next, struct page_stat, node);
1072 		sym = machine__find_kernel_symbol(machine, data->callsite, &map);
1073 		if (sym)
1074 			caller = sym->name;
1075 		else
1076 			scnprintf(buf, sizeof(buf), "%"PRIx64, data->callsite);
1077 
1078 		printf(format, (unsigned long long)data->page,
1079 		       (unsigned long long)data->alloc_bytes / 1024,
1080 		       data->nr_alloc, data->order,
1081 		       migrate_type_str[data->migrate_type],
1082 		       gfp_len, compact_gfp_string(data->gfp_flags), caller);
1083 
1084 		next = rb_next(next);
1085 	}
1086 
1087 	if (n_lines == -1) {
1088 		printf(" ...              | ...              | ...       | ...   | ...      | %-*s | ...\n",
1089 		       gfp_len, "...");
1090 	}
1091 
1092 	printf("%.105s\n", graph_dotted_line);
1093 }
1094 
1095 static void __print_page_caller_result(struct perf_session *session, int n_lines)
1096 {
1097 	struct rb_node *next = rb_first(&page_caller_sorted);
1098 	struct machine *machine = &session->machines.host;
1099 	int gfp_len = max(strlen("GFP flags"), max_gfp_len);
1100 
1101 	printf("\n%.105s\n", graph_dotted_line);
1102 	printf(" %5s alloc (KB) | Hits      | Order | Mig.type | %-*s | Callsite\n",
1103 	       live_page ? "Live" : "Total", gfp_len, "GFP flags");
1104 	printf("%.105s\n", graph_dotted_line);
1105 
1106 	while (next && n_lines--) {
1107 		struct page_stat *data;
1108 		struct symbol *sym;
1109 		struct map *map;
1110 		char buf[32];
1111 		char *caller = buf;
1112 
1113 		data = rb_entry(next, struct page_stat, node);
1114 		sym = machine__find_kernel_symbol(machine, data->callsite, &map);
1115 		if (sym)
1116 			caller = sym->name;
1117 		else
1118 			scnprintf(buf, sizeof(buf), "%"PRIx64, data->callsite);
1119 
1120 		printf(" %'16llu | %'9d | %5d | %8s | %-*s | %s\n",
1121 		       (unsigned long long)data->alloc_bytes / 1024,
1122 		       data->nr_alloc, data->order,
1123 		       migrate_type_str[data->migrate_type],
1124 		       gfp_len, compact_gfp_string(data->gfp_flags), caller);
1125 
1126 		next = rb_next(next);
1127 	}
1128 
1129 	if (n_lines == -1) {
1130 		printf(" ...              | ...       | ...   | ...      | %-*s | ...\n",
1131 		       gfp_len, "...");
1132 	}
1133 
1134 	printf("%.105s\n", graph_dotted_line);
1135 }
1136 
1137 static void print_gfp_flags(void)
1138 {
1139 	int i;
1140 
1141 	printf("#\n");
1142 	printf("# GFP flags\n");
1143 	printf("# ---------\n");
1144 	for (i = 0; i < nr_gfps; i++) {
1145 		printf("# %08x: %*s: %s\n", gfps[i].flags,
1146 		       (int) max_gfp_len, gfps[i].compact_str,
1147 		       gfps[i].human_readable);
1148 	}
1149 }
1150 
1151 static void print_slab_summary(void)
1152 {
1153 	printf("\nSUMMARY (SLAB allocator)");
1154 	printf("\n========================\n");
1155 	printf("Total bytes requested: %'lu\n", total_requested);
1156 	printf("Total bytes allocated: %'lu\n", total_allocated);
1157 	printf("Total bytes freed:     %'lu\n", total_freed);
1158 	if (total_allocated > total_freed) {
1159 		printf("Net total bytes allocated: %'lu\n",
1160 		total_allocated - total_freed);
1161 	}
1162 	printf("Total bytes wasted on internal fragmentation: %'lu\n",
1163 	       total_allocated - total_requested);
1164 	printf("Internal fragmentation: %f%%\n",
1165 	       fragmentation(total_requested, total_allocated));
1166 	printf("Cross CPU allocations: %'lu/%'lu\n", nr_cross_allocs, nr_allocs);
1167 }
1168 
1169 static void print_page_summary(void)
1170 {
1171 	int o, m;
1172 	u64 nr_alloc_freed = nr_page_frees - nr_page_nomatch;
1173 	u64 total_alloc_freed_bytes = total_page_free_bytes - total_page_nomatch_bytes;
1174 
1175 	printf("\nSUMMARY (page allocator)");
1176 	printf("\n========================\n");
1177 	printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total allocation requests",
1178 	       nr_page_allocs, total_page_alloc_bytes / 1024);
1179 	printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total free requests",
1180 	       nr_page_frees, total_page_free_bytes / 1024);
1181 	printf("\n");
1182 
1183 	printf("%-30s: %'16"PRIu64"   [ %'16"PRIu64" KB ]\n", "Total alloc+freed requests",
1184 	       nr_alloc_freed, (total_alloc_freed_bytes) / 1024);
1185 	printf("%-30s: %'16"PRIu64"   [ %'16"PRIu64" KB ]\n", "Total alloc-only requests",
1186 	       nr_page_allocs - nr_alloc_freed,
1187 	       (total_page_alloc_bytes - total_alloc_freed_bytes) / 1024);
1188 	printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total free-only requests",
1189 	       nr_page_nomatch, total_page_nomatch_bytes / 1024);
1190 	printf("\n");
1191 
1192 	printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total allocation failures",
1193 	       nr_page_fails, total_page_fail_bytes / 1024);
1194 	printf("\n");
1195 
1196 	printf("%5s  %12s  %12s  %12s  %12s  %12s\n", "Order",  "Unmovable",
1197 	       "Reclaimable", "Movable", "Reserved", "CMA/Isolated");
1198 	printf("%.5s  %.12s  %.12s  %.12s  %.12s  %.12s\n", graph_dotted_line,
1199 	       graph_dotted_line, graph_dotted_line, graph_dotted_line,
1200 	       graph_dotted_line, graph_dotted_line);
1201 
1202 	for (o = 0; o < MAX_PAGE_ORDER; o++) {
1203 		printf("%5d", o);
1204 		for (m = 0; m < MAX_MIGRATE_TYPES - 1; m++) {
1205 			if (order_stats[o][m])
1206 				printf("  %'12d", order_stats[o][m]);
1207 			else
1208 				printf("  %12c", '.');
1209 		}
1210 		printf("\n");
1211 	}
1212 }
1213 
1214 static void print_slab_result(struct perf_session *session)
1215 {
1216 	if (caller_flag)
1217 		__print_slab_result(&root_caller_sorted, session, caller_lines, 1);
1218 	if (alloc_flag)
1219 		__print_slab_result(&root_alloc_sorted, session, alloc_lines, 0);
1220 	print_slab_summary();
1221 }
1222 
1223 static void print_page_result(struct perf_session *session)
1224 {
1225 	if (caller_flag || alloc_flag)
1226 		print_gfp_flags();
1227 	if (caller_flag)
1228 		__print_page_caller_result(session, caller_lines);
1229 	if (alloc_flag)
1230 		__print_page_alloc_result(session, alloc_lines);
1231 	print_page_summary();
1232 }
1233 
1234 static void print_result(struct perf_session *session)
1235 {
1236 	if (kmem_slab)
1237 		print_slab_result(session);
1238 	if (kmem_page)
1239 		print_page_result(session);
1240 }
1241 
1242 static LIST_HEAD(slab_caller_sort);
1243 static LIST_HEAD(slab_alloc_sort);
1244 static LIST_HEAD(page_caller_sort);
1245 static LIST_HEAD(page_alloc_sort);
1246 
1247 static void sort_slab_insert(struct rb_root *root, struct alloc_stat *data,
1248 			     struct list_head *sort_list)
1249 {
1250 	struct rb_node **new = &(root->rb_node);
1251 	struct rb_node *parent = NULL;
1252 	struct sort_dimension *sort;
1253 
1254 	while (*new) {
1255 		struct alloc_stat *this;
1256 		int cmp = 0;
1257 
1258 		this = rb_entry(*new, struct alloc_stat, node);
1259 		parent = *new;
1260 
1261 		list_for_each_entry(sort, sort_list, list) {
1262 			cmp = sort->cmp(data, this);
1263 			if (cmp)
1264 				break;
1265 		}
1266 
1267 		if (cmp > 0)
1268 			new = &((*new)->rb_left);
1269 		else
1270 			new = &((*new)->rb_right);
1271 	}
1272 
1273 	rb_link_node(&data->node, parent, new);
1274 	rb_insert_color(&data->node, root);
1275 }
1276 
1277 static void __sort_slab_result(struct rb_root *root, struct rb_root *root_sorted,
1278 			       struct list_head *sort_list)
1279 {
1280 	struct rb_node *node;
1281 	struct alloc_stat *data;
1282 
1283 	for (;;) {
1284 		node = rb_first(root);
1285 		if (!node)
1286 			break;
1287 
1288 		rb_erase(node, root);
1289 		data = rb_entry(node, struct alloc_stat, node);
1290 		sort_slab_insert(root_sorted, data, sort_list);
1291 	}
1292 }
1293 
1294 static void sort_page_insert(struct rb_root *root, struct page_stat *data,
1295 			     struct list_head *sort_list)
1296 {
1297 	struct rb_node **new = &root->rb_node;
1298 	struct rb_node *parent = NULL;
1299 	struct sort_dimension *sort;
1300 
1301 	while (*new) {
1302 		struct page_stat *this;
1303 		int cmp = 0;
1304 
1305 		this = rb_entry(*new, struct page_stat, node);
1306 		parent = *new;
1307 
1308 		list_for_each_entry(sort, sort_list, list) {
1309 			cmp = sort->cmp(data, this);
1310 			if (cmp)
1311 				break;
1312 		}
1313 
1314 		if (cmp > 0)
1315 			new = &parent->rb_left;
1316 		else
1317 			new = &parent->rb_right;
1318 	}
1319 
1320 	rb_link_node(&data->node, parent, new);
1321 	rb_insert_color(&data->node, root);
1322 }
1323 
1324 static void __sort_page_result(struct rb_root *root, struct rb_root *root_sorted,
1325 			       struct list_head *sort_list)
1326 {
1327 	struct rb_node *node;
1328 	struct page_stat *data;
1329 
1330 	for (;;) {
1331 		node = rb_first(root);
1332 		if (!node)
1333 			break;
1334 
1335 		rb_erase(node, root);
1336 		data = rb_entry(node, struct page_stat, node);
1337 		sort_page_insert(root_sorted, data, sort_list);
1338 	}
1339 }
1340 
1341 static void sort_result(void)
1342 {
1343 	if (kmem_slab) {
1344 		__sort_slab_result(&root_alloc_stat, &root_alloc_sorted,
1345 				   &slab_alloc_sort);
1346 		__sort_slab_result(&root_caller_stat, &root_caller_sorted,
1347 				   &slab_caller_sort);
1348 	}
1349 	if (kmem_page) {
1350 		if (live_page)
1351 			__sort_page_result(&page_live_tree, &page_alloc_sorted,
1352 					   &page_alloc_sort);
1353 		else
1354 			__sort_page_result(&page_alloc_tree, &page_alloc_sorted,
1355 					   &page_alloc_sort);
1356 
1357 		__sort_page_result(&page_caller_tree, &page_caller_sorted,
1358 				   &page_caller_sort);
1359 	}
1360 }
1361 
1362 static int __cmd_kmem(struct perf_session *session)
1363 {
1364 	int err = -EINVAL;
1365 	struct perf_evsel *evsel;
1366 	const struct perf_evsel_str_handler kmem_tracepoints[] = {
1367 		/* slab allocator */
1368 		{ "kmem:kmalloc",		perf_evsel__process_alloc_event, },
1369     		{ "kmem:kmem_cache_alloc",	perf_evsel__process_alloc_event, },
1370 		{ "kmem:kmalloc_node",		perf_evsel__process_alloc_node_event, },
1371     		{ "kmem:kmem_cache_alloc_node", perf_evsel__process_alloc_node_event, },
1372 		{ "kmem:kfree",			perf_evsel__process_free_event, },
1373     		{ "kmem:kmem_cache_free",	perf_evsel__process_free_event, },
1374 		/* page allocator */
1375 		{ "kmem:mm_page_alloc",		perf_evsel__process_page_alloc_event, },
1376 		{ "kmem:mm_page_free",		perf_evsel__process_page_free_event, },
1377 	};
1378 
1379 	if (!perf_session__has_traces(session, "kmem record"))
1380 		goto out;
1381 
1382 	if (perf_session__set_tracepoints_handlers(session, kmem_tracepoints)) {
1383 		pr_err("Initializing perf session tracepoint handlers failed\n");
1384 		goto out;
1385 	}
1386 
1387 	evlist__for_each_entry(session->evlist, evsel) {
1388 		if (!strcmp(perf_evsel__name(evsel), "kmem:mm_page_alloc") &&
1389 		    perf_evsel__field(evsel, "pfn")) {
1390 			use_pfn = true;
1391 			break;
1392 		}
1393 	}
1394 
1395 	setup_pager();
1396 	err = perf_session__process_events(session);
1397 	if (err != 0) {
1398 		pr_err("error during process events: %d\n", err);
1399 		goto out;
1400 	}
1401 	sort_result();
1402 	print_result(session);
1403 out:
1404 	return err;
1405 }
1406 
1407 /* slab sort keys */
1408 static int ptr_cmp(void *a, void *b)
1409 {
1410 	struct alloc_stat *l = a;
1411 	struct alloc_stat *r = b;
1412 
1413 	if (l->ptr < r->ptr)
1414 		return -1;
1415 	else if (l->ptr > r->ptr)
1416 		return 1;
1417 	return 0;
1418 }
1419 
1420 static struct sort_dimension ptr_sort_dimension = {
1421 	.name	= "ptr",
1422 	.cmp	= ptr_cmp,
1423 };
1424 
1425 static int slab_callsite_cmp(void *a, void *b)
1426 {
1427 	struct alloc_stat *l = a;
1428 	struct alloc_stat *r = b;
1429 
1430 	if (l->call_site < r->call_site)
1431 		return -1;
1432 	else if (l->call_site > r->call_site)
1433 		return 1;
1434 	return 0;
1435 }
1436 
1437 static struct sort_dimension callsite_sort_dimension = {
1438 	.name	= "callsite",
1439 	.cmp	= slab_callsite_cmp,
1440 };
1441 
1442 static int hit_cmp(void *a, void *b)
1443 {
1444 	struct alloc_stat *l = a;
1445 	struct alloc_stat *r = b;
1446 
1447 	if (l->hit < r->hit)
1448 		return -1;
1449 	else if (l->hit > r->hit)
1450 		return 1;
1451 	return 0;
1452 }
1453 
1454 static struct sort_dimension hit_sort_dimension = {
1455 	.name	= "hit",
1456 	.cmp	= hit_cmp,
1457 };
1458 
1459 static int bytes_cmp(void *a, void *b)
1460 {
1461 	struct alloc_stat *l = a;
1462 	struct alloc_stat *r = b;
1463 
1464 	if (l->bytes_alloc < r->bytes_alloc)
1465 		return -1;
1466 	else if (l->bytes_alloc > r->bytes_alloc)
1467 		return 1;
1468 	return 0;
1469 }
1470 
1471 static struct sort_dimension bytes_sort_dimension = {
1472 	.name	= "bytes",
1473 	.cmp	= bytes_cmp,
1474 };
1475 
1476 static int frag_cmp(void *a, void *b)
1477 {
1478 	double x, y;
1479 	struct alloc_stat *l = a;
1480 	struct alloc_stat *r = b;
1481 
1482 	x = fragmentation(l->bytes_req, l->bytes_alloc);
1483 	y = fragmentation(r->bytes_req, r->bytes_alloc);
1484 
1485 	if (x < y)
1486 		return -1;
1487 	else if (x > y)
1488 		return 1;
1489 	return 0;
1490 }
1491 
1492 static struct sort_dimension frag_sort_dimension = {
1493 	.name	= "frag",
1494 	.cmp	= frag_cmp,
1495 };
1496 
1497 static int pingpong_cmp(void *a, void *b)
1498 {
1499 	struct alloc_stat *l = a;
1500 	struct alloc_stat *r = b;
1501 
1502 	if (l->pingpong < r->pingpong)
1503 		return -1;
1504 	else if (l->pingpong > r->pingpong)
1505 		return 1;
1506 	return 0;
1507 }
1508 
1509 static struct sort_dimension pingpong_sort_dimension = {
1510 	.name	= "pingpong",
1511 	.cmp	= pingpong_cmp,
1512 };
1513 
1514 /* page sort keys */
1515 static int page_cmp(void *a, void *b)
1516 {
1517 	struct page_stat *l = a;
1518 	struct page_stat *r = b;
1519 
1520 	if (l->page < r->page)
1521 		return -1;
1522 	else if (l->page > r->page)
1523 		return 1;
1524 	return 0;
1525 }
1526 
1527 static struct sort_dimension page_sort_dimension = {
1528 	.name	= "page",
1529 	.cmp	= page_cmp,
1530 };
1531 
1532 static int page_callsite_cmp(void *a, void *b)
1533 {
1534 	struct page_stat *l = a;
1535 	struct page_stat *r = b;
1536 
1537 	if (l->callsite < r->callsite)
1538 		return -1;
1539 	else if (l->callsite > r->callsite)
1540 		return 1;
1541 	return 0;
1542 }
1543 
1544 static struct sort_dimension page_callsite_sort_dimension = {
1545 	.name	= "callsite",
1546 	.cmp	= page_callsite_cmp,
1547 };
1548 
1549 static int page_hit_cmp(void *a, void *b)
1550 {
1551 	struct page_stat *l = a;
1552 	struct page_stat *r = b;
1553 
1554 	if (l->nr_alloc < r->nr_alloc)
1555 		return -1;
1556 	else if (l->nr_alloc > r->nr_alloc)
1557 		return 1;
1558 	return 0;
1559 }
1560 
1561 static struct sort_dimension page_hit_sort_dimension = {
1562 	.name	= "hit",
1563 	.cmp	= page_hit_cmp,
1564 };
1565 
1566 static int page_bytes_cmp(void *a, void *b)
1567 {
1568 	struct page_stat *l = a;
1569 	struct page_stat *r = b;
1570 
1571 	if (l->alloc_bytes < r->alloc_bytes)
1572 		return -1;
1573 	else if (l->alloc_bytes > r->alloc_bytes)
1574 		return 1;
1575 	return 0;
1576 }
1577 
1578 static struct sort_dimension page_bytes_sort_dimension = {
1579 	.name	= "bytes",
1580 	.cmp	= page_bytes_cmp,
1581 };
1582 
1583 static int page_order_cmp(void *a, void *b)
1584 {
1585 	struct page_stat *l = a;
1586 	struct page_stat *r = b;
1587 
1588 	if (l->order < r->order)
1589 		return -1;
1590 	else if (l->order > r->order)
1591 		return 1;
1592 	return 0;
1593 }
1594 
1595 static struct sort_dimension page_order_sort_dimension = {
1596 	.name	= "order",
1597 	.cmp	= page_order_cmp,
1598 };
1599 
1600 static int migrate_type_cmp(void *a, void *b)
1601 {
1602 	struct page_stat *l = a;
1603 	struct page_stat *r = b;
1604 
1605 	/* for internal use to find free'd page */
1606 	if (l->migrate_type == -1U)
1607 		return 0;
1608 
1609 	if (l->migrate_type < r->migrate_type)
1610 		return -1;
1611 	else if (l->migrate_type > r->migrate_type)
1612 		return 1;
1613 	return 0;
1614 }
1615 
1616 static struct sort_dimension migrate_type_sort_dimension = {
1617 	.name	= "migtype",
1618 	.cmp	= migrate_type_cmp,
1619 };
1620 
1621 static int gfp_flags_cmp(void *a, void *b)
1622 {
1623 	struct page_stat *l = a;
1624 	struct page_stat *r = b;
1625 
1626 	/* for internal use to find free'd page */
1627 	if (l->gfp_flags == -1U)
1628 		return 0;
1629 
1630 	if (l->gfp_flags < r->gfp_flags)
1631 		return -1;
1632 	else if (l->gfp_flags > r->gfp_flags)
1633 		return 1;
1634 	return 0;
1635 }
1636 
1637 static struct sort_dimension gfp_flags_sort_dimension = {
1638 	.name	= "gfp",
1639 	.cmp	= gfp_flags_cmp,
1640 };
1641 
1642 static struct sort_dimension *slab_sorts[] = {
1643 	&ptr_sort_dimension,
1644 	&callsite_sort_dimension,
1645 	&hit_sort_dimension,
1646 	&bytes_sort_dimension,
1647 	&frag_sort_dimension,
1648 	&pingpong_sort_dimension,
1649 };
1650 
1651 static struct sort_dimension *page_sorts[] = {
1652 	&page_sort_dimension,
1653 	&page_callsite_sort_dimension,
1654 	&page_hit_sort_dimension,
1655 	&page_bytes_sort_dimension,
1656 	&page_order_sort_dimension,
1657 	&migrate_type_sort_dimension,
1658 	&gfp_flags_sort_dimension,
1659 };
1660 
1661 static int slab_sort_dimension__add(const char *tok, struct list_head *list)
1662 {
1663 	struct sort_dimension *sort;
1664 	int i;
1665 
1666 	for (i = 0; i < (int)ARRAY_SIZE(slab_sorts); i++) {
1667 		if (!strcmp(slab_sorts[i]->name, tok)) {
1668 			sort = memdup(slab_sorts[i], sizeof(*slab_sorts[i]));
1669 			if (!sort) {
1670 				pr_err("%s: memdup failed\n", __func__);
1671 				return -1;
1672 			}
1673 			list_add_tail(&sort->list, list);
1674 			return 0;
1675 		}
1676 	}
1677 
1678 	return -1;
1679 }
1680 
1681 static int page_sort_dimension__add(const char *tok, struct list_head *list)
1682 {
1683 	struct sort_dimension *sort;
1684 	int i;
1685 
1686 	for (i = 0; i < (int)ARRAY_SIZE(page_sorts); i++) {
1687 		if (!strcmp(page_sorts[i]->name, tok)) {
1688 			sort = memdup(page_sorts[i], sizeof(*page_sorts[i]));
1689 			if (!sort) {
1690 				pr_err("%s: memdup failed\n", __func__);
1691 				return -1;
1692 			}
1693 			list_add_tail(&sort->list, list);
1694 			return 0;
1695 		}
1696 	}
1697 
1698 	return -1;
1699 }
1700 
1701 static int setup_slab_sorting(struct list_head *sort_list, const char *arg)
1702 {
1703 	char *tok;
1704 	char *str = strdup(arg);
1705 	char *pos = str;
1706 
1707 	if (!str) {
1708 		pr_err("%s: strdup failed\n", __func__);
1709 		return -1;
1710 	}
1711 
1712 	while (true) {
1713 		tok = strsep(&pos, ",");
1714 		if (!tok)
1715 			break;
1716 		if (slab_sort_dimension__add(tok, sort_list) < 0) {
1717 			pr_err("Unknown slab --sort key: '%s'", tok);
1718 			free(str);
1719 			return -1;
1720 		}
1721 	}
1722 
1723 	free(str);
1724 	return 0;
1725 }
1726 
1727 static int setup_page_sorting(struct list_head *sort_list, const char *arg)
1728 {
1729 	char *tok;
1730 	char *str = strdup(arg);
1731 	char *pos = str;
1732 
1733 	if (!str) {
1734 		pr_err("%s: strdup failed\n", __func__);
1735 		return -1;
1736 	}
1737 
1738 	while (true) {
1739 		tok = strsep(&pos, ",");
1740 		if (!tok)
1741 			break;
1742 		if (page_sort_dimension__add(tok, sort_list) < 0) {
1743 			pr_err("Unknown page --sort key: '%s'", tok);
1744 			free(str);
1745 			return -1;
1746 		}
1747 	}
1748 
1749 	free(str);
1750 	return 0;
1751 }
1752 
1753 static int parse_sort_opt(const struct option *opt __maybe_unused,
1754 			  const char *arg, int unset __maybe_unused)
1755 {
1756 	if (!arg)
1757 		return -1;
1758 
1759 	if (kmem_page > kmem_slab ||
1760 	    (kmem_page == 0 && kmem_slab == 0 && kmem_default == KMEM_PAGE)) {
1761 		if (caller_flag > alloc_flag)
1762 			return setup_page_sorting(&page_caller_sort, arg);
1763 		else
1764 			return setup_page_sorting(&page_alloc_sort, arg);
1765 	} else {
1766 		if (caller_flag > alloc_flag)
1767 			return setup_slab_sorting(&slab_caller_sort, arg);
1768 		else
1769 			return setup_slab_sorting(&slab_alloc_sort, arg);
1770 	}
1771 
1772 	return 0;
1773 }
1774 
1775 static int parse_caller_opt(const struct option *opt __maybe_unused,
1776 			    const char *arg __maybe_unused,
1777 			    int unset __maybe_unused)
1778 {
1779 	caller_flag = (alloc_flag + 1);
1780 	return 0;
1781 }
1782 
1783 static int parse_alloc_opt(const struct option *opt __maybe_unused,
1784 			   const char *arg __maybe_unused,
1785 			   int unset __maybe_unused)
1786 {
1787 	alloc_flag = (caller_flag + 1);
1788 	return 0;
1789 }
1790 
1791 static int parse_slab_opt(const struct option *opt __maybe_unused,
1792 			  const char *arg __maybe_unused,
1793 			  int unset __maybe_unused)
1794 {
1795 	kmem_slab = (kmem_page + 1);
1796 	return 0;
1797 }
1798 
1799 static int parse_page_opt(const struct option *opt __maybe_unused,
1800 			  const char *arg __maybe_unused,
1801 			  int unset __maybe_unused)
1802 {
1803 	kmem_page = (kmem_slab + 1);
1804 	return 0;
1805 }
1806 
1807 static int parse_line_opt(const struct option *opt __maybe_unused,
1808 			  const char *arg, int unset __maybe_unused)
1809 {
1810 	int lines;
1811 
1812 	if (!arg)
1813 		return -1;
1814 
1815 	lines = strtoul(arg, NULL, 10);
1816 
1817 	if (caller_flag > alloc_flag)
1818 		caller_lines = lines;
1819 	else
1820 		alloc_lines = lines;
1821 
1822 	return 0;
1823 }
1824 
1825 static int __cmd_record(int argc, const char **argv)
1826 {
1827 	const char * const record_args[] = {
1828 	"record", "-a", "-R", "-c", "1",
1829 	};
1830 	const char * const slab_events[] = {
1831 	"-e", "kmem:kmalloc",
1832 	"-e", "kmem:kmalloc_node",
1833 	"-e", "kmem:kfree",
1834 	"-e", "kmem:kmem_cache_alloc",
1835 	"-e", "kmem:kmem_cache_alloc_node",
1836 	"-e", "kmem:kmem_cache_free",
1837 	};
1838 	const char * const page_events[] = {
1839 	"-e", "kmem:mm_page_alloc",
1840 	"-e", "kmem:mm_page_free",
1841 	};
1842 	unsigned int rec_argc, i, j;
1843 	const char **rec_argv;
1844 
1845 	rec_argc = ARRAY_SIZE(record_args) + argc - 1;
1846 	if (kmem_slab)
1847 		rec_argc += ARRAY_SIZE(slab_events);
1848 	if (kmem_page)
1849 		rec_argc += ARRAY_SIZE(page_events) + 1; /* for -g */
1850 
1851 	rec_argv = calloc(rec_argc + 1, sizeof(char *));
1852 
1853 	if (rec_argv == NULL)
1854 		return -ENOMEM;
1855 
1856 	for (i = 0; i < ARRAY_SIZE(record_args); i++)
1857 		rec_argv[i] = strdup(record_args[i]);
1858 
1859 	if (kmem_slab) {
1860 		for (j = 0; j < ARRAY_SIZE(slab_events); j++, i++)
1861 			rec_argv[i] = strdup(slab_events[j]);
1862 	}
1863 	if (kmem_page) {
1864 		rec_argv[i++] = strdup("-g");
1865 
1866 		for (j = 0; j < ARRAY_SIZE(page_events); j++, i++)
1867 			rec_argv[i] = strdup(page_events[j]);
1868 	}
1869 
1870 	for (j = 1; j < (unsigned int)argc; j++, i++)
1871 		rec_argv[i] = argv[j];
1872 
1873 	return cmd_record(i, rec_argv);
1874 }
1875 
1876 static int kmem_config(const char *var, const char *value, void *cb __maybe_unused)
1877 {
1878 	if (!strcmp(var, "kmem.default")) {
1879 		if (!strcmp(value, "slab"))
1880 			kmem_default = KMEM_SLAB;
1881 		else if (!strcmp(value, "page"))
1882 			kmem_default = KMEM_PAGE;
1883 		else
1884 			pr_err("invalid default value ('slab' or 'page' required): %s\n",
1885 			       value);
1886 		return 0;
1887 	}
1888 
1889 	return 0;
1890 }
1891 
1892 int cmd_kmem(int argc, const char **argv)
1893 {
1894 	const char * const default_slab_sort = "frag,hit,bytes";
1895 	const char * const default_page_sort = "bytes,hit";
1896 	struct perf_data data = {
1897 		.mode = PERF_DATA_MODE_READ,
1898 	};
1899 	const struct option kmem_options[] = {
1900 	OPT_STRING('i', "input", &input_name, "file", "input file name"),
1901 	OPT_INCR('v', "verbose", &verbose,
1902 		    "be more verbose (show symbol address, etc)"),
1903 	OPT_CALLBACK_NOOPT(0, "caller", NULL, NULL,
1904 			   "show per-callsite statistics", parse_caller_opt),
1905 	OPT_CALLBACK_NOOPT(0, "alloc", NULL, NULL,
1906 			   "show per-allocation statistics", parse_alloc_opt),
1907 	OPT_CALLBACK('s', "sort", NULL, "key[,key2...]",
1908 		     "sort by keys: ptr, callsite, bytes, hit, pingpong, frag, "
1909 		     "page, order, migtype, gfp", parse_sort_opt),
1910 	OPT_CALLBACK('l', "line", NULL, "num", "show n lines", parse_line_opt),
1911 	OPT_BOOLEAN(0, "raw-ip", &raw_ip, "show raw ip instead of symbol"),
1912 	OPT_BOOLEAN('f', "force", &data.force, "don't complain, do it"),
1913 	OPT_CALLBACK_NOOPT(0, "slab", NULL, NULL, "Analyze slab allocator",
1914 			   parse_slab_opt),
1915 	OPT_CALLBACK_NOOPT(0, "page", NULL, NULL, "Analyze page allocator",
1916 			   parse_page_opt),
1917 	OPT_BOOLEAN(0, "live", &live_page, "Show live page stat"),
1918 	OPT_STRING(0, "time", &time_str, "str",
1919 		   "Time span of interest (start,stop)"),
1920 	OPT_END()
1921 	};
1922 	const char *const kmem_subcommands[] = { "record", "stat", NULL };
1923 	const char *kmem_usage[] = {
1924 		NULL,
1925 		NULL
1926 	};
1927 	struct perf_session *session;
1928 	static const char errmsg[] = "No %s allocation events found.  Have you run 'perf kmem record --%s'?\n";
1929 	int ret = perf_config(kmem_config, NULL);
1930 
1931 	if (ret)
1932 		return ret;
1933 
1934 	argc = parse_options_subcommand(argc, argv, kmem_options,
1935 					kmem_subcommands, kmem_usage, 0);
1936 
1937 	if (!argc)
1938 		usage_with_options(kmem_usage, kmem_options);
1939 
1940 	if (kmem_slab == 0 && kmem_page == 0) {
1941 		if (kmem_default == KMEM_SLAB)
1942 			kmem_slab = 1;
1943 		else
1944 			kmem_page = 1;
1945 	}
1946 
1947 	if (!strncmp(argv[0], "rec", 3)) {
1948 		symbol__init(NULL);
1949 		return __cmd_record(argc, argv);
1950 	}
1951 
1952 	data.path = input_name;
1953 
1954 	kmem_session = session = perf_session__new(&data, false, &perf_kmem);
1955 	if (session == NULL)
1956 		return -1;
1957 
1958 	ret = -1;
1959 
1960 	if (kmem_slab) {
1961 		if (!perf_evlist__find_tracepoint_by_name(session->evlist,
1962 							  "kmem:kmalloc")) {
1963 			pr_err(errmsg, "slab", "slab");
1964 			goto out_delete;
1965 		}
1966 	}
1967 
1968 	if (kmem_page) {
1969 		struct perf_evsel *evsel;
1970 
1971 		evsel = perf_evlist__find_tracepoint_by_name(session->evlist,
1972 							     "kmem:mm_page_alloc");
1973 		if (evsel == NULL) {
1974 			pr_err(errmsg, "page", "page");
1975 			goto out_delete;
1976 		}
1977 
1978 		kmem_page_size = tep_get_page_size(evsel->tp_format->pevent);
1979 		symbol_conf.use_callchain = true;
1980 	}
1981 
1982 	symbol__init(&session->header.env);
1983 
1984 	if (perf_time__parse_str(&ptime, time_str) != 0) {
1985 		pr_err("Invalid time string\n");
1986 		ret = -EINVAL;
1987 		goto out_delete;
1988 	}
1989 
1990 	if (!strcmp(argv[0], "stat")) {
1991 		setlocale(LC_ALL, "");
1992 
1993 		if (cpu__setup_cpunode_map())
1994 			goto out_delete;
1995 
1996 		if (list_empty(&slab_caller_sort))
1997 			setup_slab_sorting(&slab_caller_sort, default_slab_sort);
1998 		if (list_empty(&slab_alloc_sort))
1999 			setup_slab_sorting(&slab_alloc_sort, default_slab_sort);
2000 		if (list_empty(&page_caller_sort))
2001 			setup_page_sorting(&page_caller_sort, default_page_sort);
2002 		if (list_empty(&page_alloc_sort))
2003 			setup_page_sorting(&page_alloc_sort, default_page_sort);
2004 
2005 		if (kmem_page) {
2006 			setup_page_sorting(&page_alloc_sort_input,
2007 					   "page,order,migtype,gfp");
2008 			setup_page_sorting(&page_caller_sort_input,
2009 					   "callsite,order,migtype,gfp");
2010 		}
2011 		ret = __cmd_kmem(session);
2012 	} else
2013 		usage_with_options(kmem_usage, kmem_options);
2014 
2015 out_delete:
2016 	perf_session__delete(session);
2017 
2018 	return ret;
2019 }
2020 
2021