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