xref: /openbmc/linux/tools/perf/builtin-lock.c (revision 3477f079)
1 // SPDX-License-Identifier: GPL-2.0
2 #include <errno.h>
3 #include <inttypes.h>
4 #include "builtin.h"
5 #include "perf.h"
6 
7 #include "util/evlist.h" // for struct evsel_str_handler
8 #include "util/evsel.h"
9 #include "util/symbol.h"
10 #include "util/thread.h"
11 #include "util/header.h"
12 #include "util/target.h"
13 #include "util/callchain.h"
14 #include "util/lock-contention.h"
15 #include "util/bpf_skel/lock_data.h"
16 
17 #include <subcmd/pager.h>
18 #include <subcmd/parse-options.h>
19 #include "util/trace-event.h"
20 #include "util/tracepoint.h"
21 
22 #include "util/debug.h"
23 #include "util/session.h"
24 #include "util/tool.h"
25 #include "util/data.h"
26 #include "util/string2.h"
27 #include "util/map.h"
28 #include "util/util.h"
29 
30 #include <sys/types.h>
31 #include <sys/prctl.h>
32 #include <semaphore.h>
33 #include <math.h>
34 #include <limits.h>
35 #include <ctype.h>
36 
37 #include <linux/list.h>
38 #include <linux/hash.h>
39 #include <linux/kernel.h>
40 #include <linux/zalloc.h>
41 #include <linux/err.h>
42 #include <linux/stringify.h>
43 
44 static struct perf_session *session;
45 static struct target target;
46 
47 /* based on kernel/lockdep.c */
48 #define LOCKHASH_BITS		12
49 #define LOCKHASH_SIZE		(1UL << LOCKHASH_BITS)
50 
51 static struct hlist_head lockhash_table[LOCKHASH_SIZE];
52 
53 #define __lockhashfn(key)	hash_long((unsigned long)key, LOCKHASH_BITS)
54 #define lockhashentry(key)	(lockhash_table + __lockhashfn((key)))
55 
56 static struct rb_root		thread_stats;
57 
58 static bool combine_locks;
59 static bool show_thread_stats;
60 static bool show_lock_addrs;
61 static bool show_lock_owner;
62 static bool use_bpf;
63 static unsigned long bpf_map_entries = 10240;
64 static int max_stack_depth = CONTENTION_STACK_DEPTH;
65 static int stack_skip = CONTENTION_STACK_SKIP;
66 static int print_nr_entries = INT_MAX / 2;
67 static LIST_HEAD(callstack_filters);
68 
69 struct callstack_filter {
70 	struct list_head list;
71 	char name[];
72 };
73 
74 static struct lock_filter filters;
75 
76 static enum lock_aggr_mode aggr_mode = LOCK_AGGR_ADDR;
77 
78 static bool needs_callstack(void)
79 {
80 	return verbose > 0 || !list_empty(&callstack_filters);
81 }
82 
83 static struct thread_stat *thread_stat_find(u32 tid)
84 {
85 	struct rb_node *node;
86 	struct thread_stat *st;
87 
88 	node = thread_stats.rb_node;
89 	while (node) {
90 		st = container_of(node, struct thread_stat, rb);
91 		if (st->tid == tid)
92 			return st;
93 		else if (tid < st->tid)
94 			node = node->rb_left;
95 		else
96 			node = node->rb_right;
97 	}
98 
99 	return NULL;
100 }
101 
102 static void thread_stat_insert(struct thread_stat *new)
103 {
104 	struct rb_node **rb = &thread_stats.rb_node;
105 	struct rb_node *parent = NULL;
106 	struct thread_stat *p;
107 
108 	while (*rb) {
109 		p = container_of(*rb, struct thread_stat, rb);
110 		parent = *rb;
111 
112 		if (new->tid < p->tid)
113 			rb = &(*rb)->rb_left;
114 		else if (new->tid > p->tid)
115 			rb = &(*rb)->rb_right;
116 		else
117 			BUG_ON("inserting invalid thread_stat\n");
118 	}
119 
120 	rb_link_node(&new->rb, parent, rb);
121 	rb_insert_color(&new->rb, &thread_stats);
122 }
123 
124 static struct thread_stat *thread_stat_findnew_after_first(u32 tid)
125 {
126 	struct thread_stat *st;
127 
128 	st = thread_stat_find(tid);
129 	if (st)
130 		return st;
131 
132 	st = zalloc(sizeof(struct thread_stat));
133 	if (!st) {
134 		pr_err("memory allocation failed\n");
135 		return NULL;
136 	}
137 
138 	st->tid = tid;
139 	INIT_LIST_HEAD(&st->seq_list);
140 
141 	thread_stat_insert(st);
142 
143 	return st;
144 }
145 
146 static struct thread_stat *thread_stat_findnew_first(u32 tid);
147 static struct thread_stat *(*thread_stat_findnew)(u32 tid) =
148 	thread_stat_findnew_first;
149 
150 static struct thread_stat *thread_stat_findnew_first(u32 tid)
151 {
152 	struct thread_stat *st;
153 
154 	st = zalloc(sizeof(struct thread_stat));
155 	if (!st) {
156 		pr_err("memory allocation failed\n");
157 		return NULL;
158 	}
159 	st->tid = tid;
160 	INIT_LIST_HEAD(&st->seq_list);
161 
162 	rb_link_node(&st->rb, NULL, &thread_stats.rb_node);
163 	rb_insert_color(&st->rb, &thread_stats);
164 
165 	thread_stat_findnew = thread_stat_findnew_after_first;
166 	return st;
167 }
168 
169 /* build simple key function one is bigger than two */
170 #define SINGLE_KEY(member)						\
171 	static int lock_stat_key_ ## member(struct lock_stat *one,	\
172 					 struct lock_stat *two)		\
173 	{								\
174 		return one->member > two->member;			\
175 	}
176 
177 SINGLE_KEY(nr_acquired)
178 SINGLE_KEY(nr_contended)
179 SINGLE_KEY(avg_wait_time)
180 SINGLE_KEY(wait_time_total)
181 SINGLE_KEY(wait_time_max)
182 
183 static int lock_stat_key_wait_time_min(struct lock_stat *one,
184 					struct lock_stat *two)
185 {
186 	u64 s1 = one->wait_time_min;
187 	u64 s2 = two->wait_time_min;
188 	if (s1 == ULLONG_MAX)
189 		s1 = 0;
190 	if (s2 == ULLONG_MAX)
191 		s2 = 0;
192 	return s1 > s2;
193 }
194 
195 struct lock_key {
196 	/*
197 	 * name: the value for specify by user
198 	 * this should be simpler than raw name of member
199 	 * e.g. nr_acquired -> acquired, wait_time_total -> wait_total
200 	 */
201 	const char		*name;
202 	/* header: the string printed on the header line */
203 	const char		*header;
204 	/* len: the printing width of the field */
205 	int			len;
206 	/* key: a pointer to function to compare two lock stats for sorting */
207 	int			(*key)(struct lock_stat*, struct lock_stat*);
208 	/* print: a pointer to function to print a given lock stats */
209 	void			(*print)(struct lock_key*, struct lock_stat*);
210 	/* list: list entry to link this */
211 	struct list_head	list;
212 };
213 
214 static void lock_stat_key_print_time(unsigned long long nsec, int len)
215 {
216 	static const struct {
217 		float base;
218 		const char *unit;
219 	} table[] = {
220 		{ 1e9 * 3600, "h " },
221 		{ 1e9 * 60, "m " },
222 		{ 1e9, "s " },
223 		{ 1e6, "ms" },
224 		{ 1e3, "us" },
225 		{ 0, NULL },
226 	};
227 
228 	for (int i = 0; table[i].unit; i++) {
229 		if (nsec < table[i].base)
230 			continue;
231 
232 		pr_info("%*.2f %s", len - 3, nsec / table[i].base, table[i].unit);
233 		return;
234 	}
235 
236 	pr_info("%*llu %s", len - 3, nsec, "ns");
237 }
238 
239 #define PRINT_KEY(member)						\
240 static void lock_stat_key_print_ ## member(struct lock_key *key,	\
241 					   struct lock_stat *ls)	\
242 {									\
243 	pr_info("%*llu", key->len, (unsigned long long)ls->member);	\
244 }
245 
246 #define PRINT_TIME(member)						\
247 static void lock_stat_key_print_ ## member(struct lock_key *key,	\
248 					   struct lock_stat *ls)	\
249 {									\
250 	lock_stat_key_print_time((unsigned long long)ls->member, key->len);	\
251 }
252 
253 PRINT_KEY(nr_acquired)
254 PRINT_KEY(nr_contended)
255 PRINT_TIME(avg_wait_time)
256 PRINT_TIME(wait_time_total)
257 PRINT_TIME(wait_time_max)
258 
259 static void lock_stat_key_print_wait_time_min(struct lock_key *key,
260 					      struct lock_stat *ls)
261 {
262 	u64 wait_time = ls->wait_time_min;
263 
264 	if (wait_time == ULLONG_MAX)
265 		wait_time = 0;
266 
267 	lock_stat_key_print_time(wait_time, key->len);
268 }
269 
270 
271 static const char		*sort_key = "acquired";
272 
273 static int			(*compare)(struct lock_stat *, struct lock_stat *);
274 
275 static struct rb_root		sorted; /* place to store intermediate data */
276 static struct rb_root		result;	/* place to store sorted data */
277 
278 static LIST_HEAD(lock_keys);
279 static const char		*output_fields;
280 
281 #define DEF_KEY_LOCK(name, header, fn_suffix, len)			\
282 	{ #name, header, len, lock_stat_key_ ## fn_suffix, lock_stat_key_print_ ## fn_suffix, {} }
283 static struct lock_key report_keys[] = {
284 	DEF_KEY_LOCK(acquired, "acquired", nr_acquired, 10),
285 	DEF_KEY_LOCK(contended, "contended", nr_contended, 10),
286 	DEF_KEY_LOCK(avg_wait, "avg wait", avg_wait_time, 12),
287 	DEF_KEY_LOCK(wait_total, "total wait", wait_time_total, 12),
288 	DEF_KEY_LOCK(wait_max, "max wait", wait_time_max, 12),
289 	DEF_KEY_LOCK(wait_min, "min wait", wait_time_min, 12),
290 
291 	/* extra comparisons much complicated should be here */
292 	{ }
293 };
294 
295 static struct lock_key contention_keys[] = {
296 	DEF_KEY_LOCK(contended, "contended", nr_contended, 10),
297 	DEF_KEY_LOCK(wait_total, "total wait", wait_time_total, 12),
298 	DEF_KEY_LOCK(wait_max, "max wait", wait_time_max, 12),
299 	DEF_KEY_LOCK(wait_min, "min wait", wait_time_min, 12),
300 	DEF_KEY_LOCK(avg_wait, "avg wait", avg_wait_time, 12),
301 
302 	/* extra comparisons much complicated should be here */
303 	{ }
304 };
305 
306 static int select_key(bool contention)
307 {
308 	int i;
309 	struct lock_key *keys = report_keys;
310 
311 	if (contention)
312 		keys = contention_keys;
313 
314 	for (i = 0; keys[i].name; i++) {
315 		if (!strcmp(keys[i].name, sort_key)) {
316 			compare = keys[i].key;
317 
318 			/* selected key should be in the output fields */
319 			if (list_empty(&keys[i].list))
320 				list_add_tail(&keys[i].list, &lock_keys);
321 
322 			return 0;
323 		}
324 	}
325 
326 	pr_err("Unknown compare key: %s\n", sort_key);
327 	return -1;
328 }
329 
330 static int add_output_field(bool contention, char *name)
331 {
332 	int i;
333 	struct lock_key *keys = report_keys;
334 
335 	if (contention)
336 		keys = contention_keys;
337 
338 	for (i = 0; keys[i].name; i++) {
339 		if (strcmp(keys[i].name, name))
340 			continue;
341 
342 		/* prevent double link */
343 		if (list_empty(&keys[i].list))
344 			list_add_tail(&keys[i].list, &lock_keys);
345 
346 		return 0;
347 	}
348 
349 	pr_err("Unknown output field: %s\n", name);
350 	return -1;
351 }
352 
353 static int setup_output_field(bool contention, const char *str)
354 {
355 	char *tok, *tmp, *orig;
356 	int i, ret = 0;
357 	struct lock_key *keys = report_keys;
358 
359 	if (contention)
360 		keys = contention_keys;
361 
362 	/* no output field given: use all of them */
363 	if (str == NULL) {
364 		for (i = 0; keys[i].name; i++)
365 			list_add_tail(&keys[i].list, &lock_keys);
366 		return 0;
367 	}
368 
369 	for (i = 0; keys[i].name; i++)
370 		INIT_LIST_HEAD(&keys[i].list);
371 
372 	orig = tmp = strdup(str);
373 	if (orig == NULL)
374 		return -ENOMEM;
375 
376 	while ((tok = strsep(&tmp, ",")) != NULL){
377 		ret = add_output_field(contention, tok);
378 		if (ret < 0)
379 			break;
380 	}
381 	free(orig);
382 
383 	return ret;
384 }
385 
386 static void combine_lock_stats(struct lock_stat *st)
387 {
388 	struct rb_node **rb = &sorted.rb_node;
389 	struct rb_node *parent = NULL;
390 	struct lock_stat *p;
391 	int ret;
392 
393 	while (*rb) {
394 		p = container_of(*rb, struct lock_stat, rb);
395 		parent = *rb;
396 
397 		if (st->name && p->name)
398 			ret = strcmp(st->name, p->name);
399 		else
400 			ret = !!st->name - !!p->name;
401 
402 		if (ret == 0) {
403 			p->nr_acquired += st->nr_acquired;
404 			p->nr_contended += st->nr_contended;
405 			p->wait_time_total += st->wait_time_total;
406 
407 			if (p->nr_contended)
408 				p->avg_wait_time = p->wait_time_total / p->nr_contended;
409 
410 			if (p->wait_time_min > st->wait_time_min)
411 				p->wait_time_min = st->wait_time_min;
412 			if (p->wait_time_max < st->wait_time_max)
413 				p->wait_time_max = st->wait_time_max;
414 
415 			p->broken |= st->broken;
416 			st->combined = 1;
417 			return;
418 		}
419 
420 		if (ret < 0)
421 			rb = &(*rb)->rb_left;
422 		else
423 			rb = &(*rb)->rb_right;
424 	}
425 
426 	rb_link_node(&st->rb, parent, rb);
427 	rb_insert_color(&st->rb, &sorted);
428 }
429 
430 static void insert_to_result(struct lock_stat *st,
431 			     int (*bigger)(struct lock_stat *, struct lock_stat *))
432 {
433 	struct rb_node **rb = &result.rb_node;
434 	struct rb_node *parent = NULL;
435 	struct lock_stat *p;
436 
437 	if (combine_locks && st->combined)
438 		return;
439 
440 	while (*rb) {
441 		p = container_of(*rb, struct lock_stat, rb);
442 		parent = *rb;
443 
444 		if (bigger(st, p))
445 			rb = &(*rb)->rb_left;
446 		else
447 			rb = &(*rb)->rb_right;
448 	}
449 
450 	rb_link_node(&st->rb, parent, rb);
451 	rb_insert_color(&st->rb, &result);
452 }
453 
454 /* returns left most element of result, and erase it */
455 static struct lock_stat *pop_from_result(void)
456 {
457 	struct rb_node *node = result.rb_node;
458 
459 	if (!node)
460 		return NULL;
461 
462 	while (node->rb_left)
463 		node = node->rb_left;
464 
465 	rb_erase(node, &result);
466 	return container_of(node, struct lock_stat, rb);
467 }
468 
469 struct lock_stat *lock_stat_find(u64 addr)
470 {
471 	struct hlist_head *entry = lockhashentry(addr);
472 	struct lock_stat *ret;
473 
474 	hlist_for_each_entry(ret, entry, hash_entry) {
475 		if (ret->addr == addr)
476 			return ret;
477 	}
478 	return NULL;
479 }
480 
481 struct lock_stat *lock_stat_findnew(u64 addr, const char *name, int flags)
482 {
483 	struct hlist_head *entry = lockhashentry(addr);
484 	struct lock_stat *ret, *new;
485 
486 	hlist_for_each_entry(ret, entry, hash_entry) {
487 		if (ret->addr == addr)
488 			return ret;
489 	}
490 
491 	new = zalloc(sizeof(struct lock_stat));
492 	if (!new)
493 		goto alloc_failed;
494 
495 	new->addr = addr;
496 	new->name = strdup(name);
497 	if (!new->name) {
498 		free(new);
499 		goto alloc_failed;
500 	}
501 
502 	new->flags = flags;
503 	new->wait_time_min = ULLONG_MAX;
504 
505 	hlist_add_head(&new->hash_entry, entry);
506 	return new;
507 
508 alloc_failed:
509 	pr_err("memory allocation failed\n");
510 	return NULL;
511 }
512 
513 bool match_callstack_filter(struct machine *machine, u64 *callstack)
514 {
515 	struct map *kmap;
516 	struct symbol *sym;
517 	u64 ip;
518 
519 	if (list_empty(&callstack_filters))
520 		return true;
521 
522 	for (int i = 0; i < max_stack_depth; i++) {
523 		struct callstack_filter *filter;
524 
525 		if (!callstack || !callstack[i])
526 			break;
527 
528 		ip = callstack[i];
529 		sym = machine__find_kernel_symbol(machine, ip, &kmap);
530 		if (sym == NULL)
531 			continue;
532 
533 		list_for_each_entry(filter, &callstack_filters, list) {
534 			if (strstr(sym->name, filter->name))
535 				return true;
536 		}
537 	}
538 	return false;
539 }
540 
541 struct trace_lock_handler {
542 	/* it's used on CONFIG_LOCKDEP */
543 	int (*acquire_event)(struct evsel *evsel,
544 			     struct perf_sample *sample);
545 
546 	/* it's used on CONFIG_LOCKDEP && CONFIG_LOCK_STAT */
547 	int (*acquired_event)(struct evsel *evsel,
548 			      struct perf_sample *sample);
549 
550 	/* it's used on CONFIG_LOCKDEP && CONFIG_LOCK_STAT */
551 	int (*contended_event)(struct evsel *evsel,
552 			       struct perf_sample *sample);
553 
554 	/* it's used on CONFIG_LOCKDEP */
555 	int (*release_event)(struct evsel *evsel,
556 			     struct perf_sample *sample);
557 
558 	/* it's used when CONFIG_LOCKDEP is off */
559 	int (*contention_begin_event)(struct evsel *evsel,
560 				      struct perf_sample *sample);
561 
562 	/* it's used when CONFIG_LOCKDEP is off */
563 	int (*contention_end_event)(struct evsel *evsel,
564 				    struct perf_sample *sample);
565 };
566 
567 static struct lock_seq_stat *get_seq(struct thread_stat *ts, u64 addr)
568 {
569 	struct lock_seq_stat *seq;
570 
571 	list_for_each_entry(seq, &ts->seq_list, list) {
572 		if (seq->addr == addr)
573 			return seq;
574 	}
575 
576 	seq = zalloc(sizeof(struct lock_seq_stat));
577 	if (!seq) {
578 		pr_err("memory allocation failed\n");
579 		return NULL;
580 	}
581 	seq->state = SEQ_STATE_UNINITIALIZED;
582 	seq->addr = addr;
583 
584 	list_add(&seq->list, &ts->seq_list);
585 	return seq;
586 }
587 
588 enum broken_state {
589 	BROKEN_ACQUIRE,
590 	BROKEN_ACQUIRED,
591 	BROKEN_CONTENDED,
592 	BROKEN_RELEASE,
593 	BROKEN_MAX,
594 };
595 
596 static int bad_hist[BROKEN_MAX];
597 
598 enum acquire_flags {
599 	TRY_LOCK = 1,
600 	READ_LOCK = 2,
601 };
602 
603 static int get_key_by_aggr_mode_simple(u64 *key, u64 addr, u32 tid)
604 {
605 	switch (aggr_mode) {
606 	case LOCK_AGGR_ADDR:
607 		*key = addr;
608 		break;
609 	case LOCK_AGGR_TASK:
610 		*key = tid;
611 		break;
612 	case LOCK_AGGR_CALLER:
613 	default:
614 		pr_err("Invalid aggregation mode: %d\n", aggr_mode);
615 		return -EINVAL;
616 	}
617 	return 0;
618 }
619 
620 static u64 callchain_id(struct evsel *evsel, struct perf_sample *sample);
621 
622 static int get_key_by_aggr_mode(u64 *key, u64 addr, struct evsel *evsel,
623 				 struct perf_sample *sample)
624 {
625 	if (aggr_mode == LOCK_AGGR_CALLER) {
626 		*key = callchain_id(evsel, sample);
627 		return 0;
628 	}
629 	return get_key_by_aggr_mode_simple(key, addr, sample->tid);
630 }
631 
632 static int report_lock_acquire_event(struct evsel *evsel,
633 				     struct perf_sample *sample)
634 {
635 	struct lock_stat *ls;
636 	struct thread_stat *ts;
637 	struct lock_seq_stat *seq;
638 	const char *name = evsel__strval(evsel, sample, "name");
639 	u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
640 	int flag = evsel__intval(evsel, sample, "flags");
641 	u64 key;
642 	int ret;
643 
644 	ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
645 	if (ret < 0)
646 		return ret;
647 
648 	ls = lock_stat_findnew(key, name, 0);
649 	if (!ls)
650 		return -ENOMEM;
651 
652 	ts = thread_stat_findnew(sample->tid);
653 	if (!ts)
654 		return -ENOMEM;
655 
656 	seq = get_seq(ts, addr);
657 	if (!seq)
658 		return -ENOMEM;
659 
660 	switch (seq->state) {
661 	case SEQ_STATE_UNINITIALIZED:
662 	case SEQ_STATE_RELEASED:
663 		if (!flag) {
664 			seq->state = SEQ_STATE_ACQUIRING;
665 		} else {
666 			if (flag & TRY_LOCK)
667 				ls->nr_trylock++;
668 			if (flag & READ_LOCK)
669 				ls->nr_readlock++;
670 			seq->state = SEQ_STATE_READ_ACQUIRED;
671 			seq->read_count = 1;
672 			ls->nr_acquired++;
673 		}
674 		break;
675 	case SEQ_STATE_READ_ACQUIRED:
676 		if (flag & READ_LOCK) {
677 			seq->read_count++;
678 			ls->nr_acquired++;
679 			goto end;
680 		} else {
681 			goto broken;
682 		}
683 		break;
684 	case SEQ_STATE_ACQUIRED:
685 	case SEQ_STATE_ACQUIRING:
686 	case SEQ_STATE_CONTENDED:
687 broken:
688 		/* broken lock sequence */
689 		if (!ls->broken) {
690 			ls->broken = 1;
691 			bad_hist[BROKEN_ACQUIRE]++;
692 		}
693 		list_del_init(&seq->list);
694 		free(seq);
695 		goto end;
696 	default:
697 		BUG_ON("Unknown state of lock sequence found!\n");
698 		break;
699 	}
700 
701 	ls->nr_acquire++;
702 	seq->prev_event_time = sample->time;
703 end:
704 	return 0;
705 }
706 
707 static int report_lock_acquired_event(struct evsel *evsel,
708 				      struct perf_sample *sample)
709 {
710 	struct lock_stat *ls;
711 	struct thread_stat *ts;
712 	struct lock_seq_stat *seq;
713 	u64 contended_term;
714 	const char *name = evsel__strval(evsel, sample, "name");
715 	u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
716 	u64 key;
717 	int ret;
718 
719 	ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
720 	if (ret < 0)
721 		return ret;
722 
723 	ls = lock_stat_findnew(key, name, 0);
724 	if (!ls)
725 		return -ENOMEM;
726 
727 	ts = thread_stat_findnew(sample->tid);
728 	if (!ts)
729 		return -ENOMEM;
730 
731 	seq = get_seq(ts, addr);
732 	if (!seq)
733 		return -ENOMEM;
734 
735 	switch (seq->state) {
736 	case SEQ_STATE_UNINITIALIZED:
737 		/* orphan event, do nothing */
738 		return 0;
739 	case SEQ_STATE_ACQUIRING:
740 		break;
741 	case SEQ_STATE_CONTENDED:
742 		contended_term = sample->time - seq->prev_event_time;
743 		ls->wait_time_total += contended_term;
744 		if (contended_term < ls->wait_time_min)
745 			ls->wait_time_min = contended_term;
746 		if (ls->wait_time_max < contended_term)
747 			ls->wait_time_max = contended_term;
748 		break;
749 	case SEQ_STATE_RELEASED:
750 	case SEQ_STATE_ACQUIRED:
751 	case SEQ_STATE_READ_ACQUIRED:
752 		/* broken lock sequence */
753 		if (!ls->broken) {
754 			ls->broken = 1;
755 			bad_hist[BROKEN_ACQUIRED]++;
756 		}
757 		list_del_init(&seq->list);
758 		free(seq);
759 		goto end;
760 	default:
761 		BUG_ON("Unknown state of lock sequence found!\n");
762 		break;
763 	}
764 
765 	seq->state = SEQ_STATE_ACQUIRED;
766 	ls->nr_acquired++;
767 	ls->avg_wait_time = ls->nr_contended ? ls->wait_time_total/ls->nr_contended : 0;
768 	seq->prev_event_time = sample->time;
769 end:
770 	return 0;
771 }
772 
773 static int report_lock_contended_event(struct evsel *evsel,
774 				       struct perf_sample *sample)
775 {
776 	struct lock_stat *ls;
777 	struct thread_stat *ts;
778 	struct lock_seq_stat *seq;
779 	const char *name = evsel__strval(evsel, sample, "name");
780 	u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
781 	u64 key;
782 	int ret;
783 
784 	ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
785 	if (ret < 0)
786 		return ret;
787 
788 	ls = lock_stat_findnew(key, name, 0);
789 	if (!ls)
790 		return -ENOMEM;
791 
792 	ts = thread_stat_findnew(sample->tid);
793 	if (!ts)
794 		return -ENOMEM;
795 
796 	seq = get_seq(ts, addr);
797 	if (!seq)
798 		return -ENOMEM;
799 
800 	switch (seq->state) {
801 	case SEQ_STATE_UNINITIALIZED:
802 		/* orphan event, do nothing */
803 		return 0;
804 	case SEQ_STATE_ACQUIRING:
805 		break;
806 	case SEQ_STATE_RELEASED:
807 	case SEQ_STATE_ACQUIRED:
808 	case SEQ_STATE_READ_ACQUIRED:
809 	case SEQ_STATE_CONTENDED:
810 		/* broken lock sequence */
811 		if (!ls->broken) {
812 			ls->broken = 1;
813 			bad_hist[BROKEN_CONTENDED]++;
814 		}
815 		list_del_init(&seq->list);
816 		free(seq);
817 		goto end;
818 	default:
819 		BUG_ON("Unknown state of lock sequence found!\n");
820 		break;
821 	}
822 
823 	seq->state = SEQ_STATE_CONTENDED;
824 	ls->nr_contended++;
825 	ls->avg_wait_time = ls->wait_time_total/ls->nr_contended;
826 	seq->prev_event_time = sample->time;
827 end:
828 	return 0;
829 }
830 
831 static int report_lock_release_event(struct evsel *evsel,
832 				     struct perf_sample *sample)
833 {
834 	struct lock_stat *ls;
835 	struct thread_stat *ts;
836 	struct lock_seq_stat *seq;
837 	const char *name = evsel__strval(evsel, sample, "name");
838 	u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
839 	u64 key;
840 	int ret;
841 
842 	ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
843 	if (ret < 0)
844 		return ret;
845 
846 	ls = lock_stat_findnew(key, name, 0);
847 	if (!ls)
848 		return -ENOMEM;
849 
850 	ts = thread_stat_findnew(sample->tid);
851 	if (!ts)
852 		return -ENOMEM;
853 
854 	seq = get_seq(ts, addr);
855 	if (!seq)
856 		return -ENOMEM;
857 
858 	switch (seq->state) {
859 	case SEQ_STATE_UNINITIALIZED:
860 		goto end;
861 	case SEQ_STATE_ACQUIRED:
862 		break;
863 	case SEQ_STATE_READ_ACQUIRED:
864 		seq->read_count--;
865 		BUG_ON(seq->read_count < 0);
866 		if (seq->read_count) {
867 			ls->nr_release++;
868 			goto end;
869 		}
870 		break;
871 	case SEQ_STATE_ACQUIRING:
872 	case SEQ_STATE_CONTENDED:
873 	case SEQ_STATE_RELEASED:
874 		/* broken lock sequence */
875 		if (!ls->broken) {
876 			ls->broken = 1;
877 			bad_hist[BROKEN_RELEASE]++;
878 		}
879 		goto free_seq;
880 	default:
881 		BUG_ON("Unknown state of lock sequence found!\n");
882 		break;
883 	}
884 
885 	ls->nr_release++;
886 free_seq:
887 	list_del_init(&seq->list);
888 	free(seq);
889 end:
890 	return 0;
891 }
892 
893 static int get_symbol_name_offset(struct map *map, struct symbol *sym, u64 ip,
894 				  char *buf, int size)
895 {
896 	u64 offset;
897 
898 	if (map == NULL || sym == NULL) {
899 		buf[0] = '\0';
900 		return 0;
901 	}
902 
903 	offset = map->map_ip(map, ip) - sym->start;
904 
905 	if (offset)
906 		return scnprintf(buf, size, "%s+%#lx", sym->name, offset);
907 	else
908 		return strlcpy(buf, sym->name, size);
909 }
910 static int lock_contention_caller(struct evsel *evsel, struct perf_sample *sample,
911 				  char *buf, int size)
912 {
913 	struct thread *thread;
914 	struct callchain_cursor *cursor = &callchain_cursor;
915 	struct machine *machine = &session->machines.host;
916 	struct symbol *sym;
917 	int skip = 0;
918 	int ret;
919 
920 	/* lock names will be replaced to task name later */
921 	if (show_thread_stats)
922 		return -1;
923 
924 	thread = machine__findnew_thread(machine, -1, sample->pid);
925 	if (thread == NULL)
926 		return -1;
927 
928 	/* use caller function name from the callchain */
929 	ret = thread__resolve_callchain(thread, cursor, evsel, sample,
930 					NULL, NULL, max_stack_depth);
931 	if (ret != 0) {
932 		thread__put(thread);
933 		return -1;
934 	}
935 
936 	callchain_cursor_commit(cursor);
937 	thread__put(thread);
938 
939 	while (true) {
940 		struct callchain_cursor_node *node;
941 
942 		node = callchain_cursor_current(cursor);
943 		if (node == NULL)
944 			break;
945 
946 		/* skip first few entries - for lock functions */
947 		if (++skip <= stack_skip)
948 			goto next;
949 
950 		sym = node->ms.sym;
951 		if (sym && !machine__is_lock_function(machine, node->ip)) {
952 			get_symbol_name_offset(node->ms.map, sym, node->ip,
953 					       buf, size);
954 			return 0;
955 		}
956 
957 next:
958 		callchain_cursor_advance(cursor);
959 	}
960 	return -1;
961 }
962 
963 static u64 callchain_id(struct evsel *evsel, struct perf_sample *sample)
964 {
965 	struct callchain_cursor *cursor = &callchain_cursor;
966 	struct machine *machine = &session->machines.host;
967 	struct thread *thread;
968 	u64 hash = 0;
969 	int skip = 0;
970 	int ret;
971 
972 	thread = machine__findnew_thread(machine, -1, sample->pid);
973 	if (thread == NULL)
974 		return -1;
975 
976 	/* use caller function name from the callchain */
977 	ret = thread__resolve_callchain(thread, cursor, evsel, sample,
978 					NULL, NULL, max_stack_depth);
979 	thread__put(thread);
980 
981 	if (ret != 0)
982 		return -1;
983 
984 	callchain_cursor_commit(cursor);
985 
986 	while (true) {
987 		struct callchain_cursor_node *node;
988 
989 		node = callchain_cursor_current(cursor);
990 		if (node == NULL)
991 			break;
992 
993 		/* skip first few entries - for lock functions */
994 		if (++skip <= stack_skip)
995 			goto next;
996 
997 		if (node->ms.sym && machine__is_lock_function(machine, node->ip))
998 			goto next;
999 
1000 		hash ^= hash_long((unsigned long)node->ip, 64);
1001 
1002 next:
1003 		callchain_cursor_advance(cursor);
1004 	}
1005 	return hash;
1006 }
1007 
1008 static u64 *get_callstack(struct perf_sample *sample, int max_stack)
1009 {
1010 	u64 *callstack;
1011 	u64 i;
1012 	int c;
1013 
1014 	callstack = calloc(max_stack, sizeof(*callstack));
1015 	if (callstack == NULL)
1016 		return NULL;
1017 
1018 	for (i = 0, c = 0; i < sample->callchain->nr && c < max_stack; i++) {
1019 		u64 ip = sample->callchain->ips[i];
1020 
1021 		if (ip >= PERF_CONTEXT_MAX)
1022 			continue;
1023 
1024 		callstack[c++] = ip;
1025 	}
1026 	return callstack;
1027 }
1028 
1029 static int report_lock_contention_begin_event(struct evsel *evsel,
1030 					      struct perf_sample *sample)
1031 {
1032 	struct lock_stat *ls;
1033 	struct thread_stat *ts;
1034 	struct lock_seq_stat *seq;
1035 	u64 addr = evsel__intval(evsel, sample, "lock_addr");
1036 	unsigned int flags = evsel__intval(evsel, sample, "flags");
1037 	u64 key;
1038 	int i, ret;
1039 	static bool kmap_loaded;
1040 	struct machine *machine = &session->machines.host;
1041 	struct map *kmap;
1042 	struct symbol *sym;
1043 
1044 	ret = get_key_by_aggr_mode(&key, addr, evsel, sample);
1045 	if (ret < 0)
1046 		return ret;
1047 
1048 	if (!kmap_loaded) {
1049 		unsigned long *addrs;
1050 
1051 		/* make sure it loads the kernel map to find lock symbols */
1052 		map__load(machine__kernel_map(machine));
1053 		kmap_loaded = true;
1054 
1055 		/* convert (kernel) symbols to addresses */
1056 		for (i = 0; i < filters.nr_syms; i++) {
1057 			sym = machine__find_kernel_symbol_by_name(machine,
1058 								  filters.syms[i],
1059 								  &kmap);
1060 			if (sym == NULL) {
1061 				pr_warning("ignore unknown symbol: %s\n",
1062 					   filters.syms[i]);
1063 				continue;
1064 			}
1065 
1066 			addrs = realloc(filters.addrs,
1067 					(filters.nr_addrs + 1) * sizeof(*addrs));
1068 			if (addrs == NULL) {
1069 				pr_warning("memory allocation failure\n");
1070 				return -ENOMEM;
1071 			}
1072 
1073 			addrs[filters.nr_addrs++] = kmap->unmap_ip(kmap, sym->start);
1074 			filters.addrs = addrs;
1075 		}
1076 	}
1077 
1078 	ls = lock_stat_find(key);
1079 	if (!ls) {
1080 		char buf[128];
1081 		const char *name = "";
1082 
1083 		switch (aggr_mode) {
1084 		case LOCK_AGGR_ADDR:
1085 			sym = machine__find_kernel_symbol(machine, key, &kmap);
1086 			if (sym)
1087 				name = sym->name;
1088 			break;
1089 		case LOCK_AGGR_CALLER:
1090 			name = buf;
1091 			if (lock_contention_caller(evsel, sample, buf, sizeof(buf)) < 0)
1092 				name = "Unknown";
1093 			break;
1094 		case LOCK_AGGR_TASK:
1095 		default:
1096 			break;
1097 		}
1098 
1099 		ls = lock_stat_findnew(key, name, flags);
1100 		if (!ls)
1101 			return -ENOMEM;
1102 	}
1103 
1104 	if (filters.nr_types) {
1105 		bool found = false;
1106 
1107 		for (i = 0; i < filters.nr_types; i++) {
1108 			if (flags == filters.types[i]) {
1109 				found = true;
1110 				break;
1111 			}
1112 		}
1113 
1114 		if (!found)
1115 			return 0;
1116 	}
1117 
1118 	if (filters.nr_addrs) {
1119 		bool found = false;
1120 
1121 		for (i = 0; i < filters.nr_addrs; i++) {
1122 			if (addr == filters.addrs[i]) {
1123 				found = true;
1124 				break;
1125 			}
1126 		}
1127 
1128 		if (!found)
1129 			return 0;
1130 	}
1131 
1132 	if (needs_callstack()) {
1133 		u64 *callstack = get_callstack(sample, max_stack_depth);
1134 		if (callstack == NULL)
1135 			return -ENOMEM;
1136 
1137 		if (!match_callstack_filter(machine, callstack)) {
1138 			free(callstack);
1139 			return 0;
1140 		}
1141 
1142 		if (ls->callstack == NULL)
1143 			ls->callstack = callstack;
1144 		else
1145 			free(callstack);
1146 	}
1147 
1148 	ts = thread_stat_findnew(sample->tid);
1149 	if (!ts)
1150 		return -ENOMEM;
1151 
1152 	seq = get_seq(ts, addr);
1153 	if (!seq)
1154 		return -ENOMEM;
1155 
1156 	switch (seq->state) {
1157 	case SEQ_STATE_UNINITIALIZED:
1158 	case SEQ_STATE_ACQUIRED:
1159 		break;
1160 	case SEQ_STATE_CONTENDED:
1161 		/*
1162 		 * It can have nested contention begin with mutex spinning,
1163 		 * then we would use the original contention begin event and
1164 		 * ignore the second one.
1165 		 */
1166 		goto end;
1167 	case SEQ_STATE_ACQUIRING:
1168 	case SEQ_STATE_READ_ACQUIRED:
1169 	case SEQ_STATE_RELEASED:
1170 		/* broken lock sequence */
1171 		if (!ls->broken) {
1172 			ls->broken = 1;
1173 			bad_hist[BROKEN_CONTENDED]++;
1174 		}
1175 		list_del_init(&seq->list);
1176 		free(seq);
1177 		goto end;
1178 	default:
1179 		BUG_ON("Unknown state of lock sequence found!\n");
1180 		break;
1181 	}
1182 
1183 	if (seq->state != SEQ_STATE_CONTENDED) {
1184 		seq->state = SEQ_STATE_CONTENDED;
1185 		seq->prev_event_time = sample->time;
1186 		ls->nr_contended++;
1187 	}
1188 end:
1189 	return 0;
1190 }
1191 
1192 static int report_lock_contention_end_event(struct evsel *evsel,
1193 					    struct perf_sample *sample)
1194 {
1195 	struct lock_stat *ls;
1196 	struct thread_stat *ts;
1197 	struct lock_seq_stat *seq;
1198 	u64 contended_term;
1199 	u64 addr = evsel__intval(evsel, sample, "lock_addr");
1200 	u64 key;
1201 	int ret;
1202 
1203 	ret = get_key_by_aggr_mode(&key, addr, evsel, sample);
1204 	if (ret < 0)
1205 		return ret;
1206 
1207 	ls = lock_stat_find(key);
1208 	if (!ls)
1209 		return 0;
1210 
1211 	ts = thread_stat_find(sample->tid);
1212 	if (!ts)
1213 		return 0;
1214 
1215 	seq = get_seq(ts, addr);
1216 	if (!seq)
1217 		return -ENOMEM;
1218 
1219 	switch (seq->state) {
1220 	case SEQ_STATE_UNINITIALIZED:
1221 		goto end;
1222 	case SEQ_STATE_CONTENDED:
1223 		contended_term = sample->time - seq->prev_event_time;
1224 		ls->wait_time_total += contended_term;
1225 		if (contended_term < ls->wait_time_min)
1226 			ls->wait_time_min = contended_term;
1227 		if (ls->wait_time_max < contended_term)
1228 			ls->wait_time_max = contended_term;
1229 		break;
1230 	case SEQ_STATE_ACQUIRING:
1231 	case SEQ_STATE_ACQUIRED:
1232 	case SEQ_STATE_READ_ACQUIRED:
1233 	case SEQ_STATE_RELEASED:
1234 		/* broken lock sequence */
1235 		if (!ls->broken) {
1236 			ls->broken = 1;
1237 			bad_hist[BROKEN_ACQUIRED]++;
1238 		}
1239 		list_del_init(&seq->list);
1240 		free(seq);
1241 		goto end;
1242 	default:
1243 		BUG_ON("Unknown state of lock sequence found!\n");
1244 		break;
1245 	}
1246 
1247 	seq->state = SEQ_STATE_ACQUIRED;
1248 	ls->nr_acquired++;
1249 	ls->avg_wait_time = ls->wait_time_total/ls->nr_acquired;
1250 end:
1251 	return 0;
1252 }
1253 
1254 /* lock oriented handlers */
1255 /* TODO: handlers for CPU oriented, thread oriented */
1256 static struct trace_lock_handler report_lock_ops  = {
1257 	.acquire_event		= report_lock_acquire_event,
1258 	.acquired_event		= report_lock_acquired_event,
1259 	.contended_event	= report_lock_contended_event,
1260 	.release_event		= report_lock_release_event,
1261 	.contention_begin_event	= report_lock_contention_begin_event,
1262 	.contention_end_event	= report_lock_contention_end_event,
1263 };
1264 
1265 static struct trace_lock_handler contention_lock_ops  = {
1266 	.contention_begin_event	= report_lock_contention_begin_event,
1267 	.contention_end_event	= report_lock_contention_end_event,
1268 };
1269 
1270 
1271 static struct trace_lock_handler *trace_handler;
1272 
1273 static int evsel__process_lock_acquire(struct evsel *evsel, struct perf_sample *sample)
1274 {
1275 	if (trace_handler->acquire_event)
1276 		return trace_handler->acquire_event(evsel, sample);
1277 	return 0;
1278 }
1279 
1280 static int evsel__process_lock_acquired(struct evsel *evsel, struct perf_sample *sample)
1281 {
1282 	if (trace_handler->acquired_event)
1283 		return trace_handler->acquired_event(evsel, sample);
1284 	return 0;
1285 }
1286 
1287 static int evsel__process_lock_contended(struct evsel *evsel, struct perf_sample *sample)
1288 {
1289 	if (trace_handler->contended_event)
1290 		return trace_handler->contended_event(evsel, sample);
1291 	return 0;
1292 }
1293 
1294 static int evsel__process_lock_release(struct evsel *evsel, struct perf_sample *sample)
1295 {
1296 	if (trace_handler->release_event)
1297 		return trace_handler->release_event(evsel, sample);
1298 	return 0;
1299 }
1300 
1301 static int evsel__process_contention_begin(struct evsel *evsel, struct perf_sample *sample)
1302 {
1303 	if (trace_handler->contention_begin_event)
1304 		return trace_handler->contention_begin_event(evsel, sample);
1305 	return 0;
1306 }
1307 
1308 static int evsel__process_contention_end(struct evsel *evsel, struct perf_sample *sample)
1309 {
1310 	if (trace_handler->contention_end_event)
1311 		return trace_handler->contention_end_event(evsel, sample);
1312 	return 0;
1313 }
1314 
1315 static void print_bad_events(int bad, int total)
1316 {
1317 	/* Output for debug, this have to be removed */
1318 	int i;
1319 	int broken = 0;
1320 	const char *name[4] =
1321 		{ "acquire", "acquired", "contended", "release" };
1322 
1323 	for (i = 0; i < BROKEN_MAX; i++)
1324 		broken += bad_hist[i];
1325 
1326 	if (quiet || (broken == 0 && verbose <= 0))
1327 		return;
1328 
1329 	pr_info("\n=== output for debug===\n\n");
1330 	pr_info("bad: %d, total: %d\n", bad, total);
1331 	pr_info("bad rate: %.2f %%\n", (double)bad / (double)total * 100);
1332 	pr_info("histogram of events caused bad sequence\n");
1333 	for (i = 0; i < BROKEN_MAX; i++)
1334 		pr_info(" %10s: %d\n", name[i], bad_hist[i]);
1335 }
1336 
1337 /* TODO: various way to print, coloring, nano or milli sec */
1338 static void print_result(void)
1339 {
1340 	struct lock_stat *st;
1341 	struct lock_key *key;
1342 	char cut_name[20];
1343 	int bad, total, printed;
1344 
1345 	if (!quiet) {
1346 		pr_info("%20s ", "Name");
1347 		list_for_each_entry(key, &lock_keys, list)
1348 			pr_info("%*s ", key->len, key->header);
1349 		pr_info("\n\n");
1350 	}
1351 
1352 	bad = total = printed = 0;
1353 	while ((st = pop_from_result())) {
1354 		total++;
1355 		if (st->broken)
1356 			bad++;
1357 		if (!st->nr_acquired)
1358 			continue;
1359 
1360 		bzero(cut_name, 20);
1361 
1362 		if (strlen(st->name) < 20) {
1363 			/* output raw name */
1364 			const char *name = st->name;
1365 
1366 			if (show_thread_stats) {
1367 				struct thread *t;
1368 
1369 				/* st->addr contains tid of thread */
1370 				t = perf_session__findnew(session, st->addr);
1371 				name = thread__comm_str(t);
1372 			}
1373 
1374 			pr_info("%20s ", name);
1375 		} else {
1376 			strncpy(cut_name, st->name, 16);
1377 			cut_name[16] = '.';
1378 			cut_name[17] = '.';
1379 			cut_name[18] = '.';
1380 			cut_name[19] = '\0';
1381 			/* cut off name for saving output style */
1382 			pr_info("%20s ", cut_name);
1383 		}
1384 
1385 		list_for_each_entry(key, &lock_keys, list) {
1386 			key->print(key, st);
1387 			pr_info(" ");
1388 		}
1389 		pr_info("\n");
1390 
1391 		if (++printed >= print_nr_entries)
1392 			break;
1393 	}
1394 
1395 	print_bad_events(bad, total);
1396 }
1397 
1398 static bool info_threads, info_map;
1399 
1400 static void dump_threads(void)
1401 {
1402 	struct thread_stat *st;
1403 	struct rb_node *node;
1404 	struct thread *t;
1405 
1406 	pr_info("%10s: comm\n", "Thread ID");
1407 
1408 	node = rb_first(&thread_stats);
1409 	while (node) {
1410 		st = container_of(node, struct thread_stat, rb);
1411 		t = perf_session__findnew(session, st->tid);
1412 		pr_info("%10d: %s\n", st->tid, thread__comm_str(t));
1413 		node = rb_next(node);
1414 		thread__put(t);
1415 	}
1416 }
1417 
1418 static int compare_maps(struct lock_stat *a, struct lock_stat *b)
1419 {
1420 	int ret;
1421 
1422 	if (a->name && b->name)
1423 		ret = strcmp(a->name, b->name);
1424 	else
1425 		ret = !!a->name - !!b->name;
1426 
1427 	if (!ret)
1428 		return a->addr < b->addr;
1429 	else
1430 		return ret < 0;
1431 }
1432 
1433 static void dump_map(void)
1434 {
1435 	unsigned int i;
1436 	struct lock_stat *st;
1437 
1438 	pr_info("Address of instance: name of class\n");
1439 	for (i = 0; i < LOCKHASH_SIZE; i++) {
1440 		hlist_for_each_entry(st, &lockhash_table[i], hash_entry) {
1441 			insert_to_result(st, compare_maps);
1442 		}
1443 	}
1444 
1445 	while ((st = pop_from_result()))
1446 		pr_info(" %#llx: %s\n", (unsigned long long)st->addr, st->name);
1447 }
1448 
1449 static int dump_info(void)
1450 {
1451 	int rc = 0;
1452 
1453 	if (info_threads)
1454 		dump_threads();
1455 	else if (info_map)
1456 		dump_map();
1457 	else {
1458 		rc = -1;
1459 		pr_err("Unknown type of information\n");
1460 	}
1461 
1462 	return rc;
1463 }
1464 
1465 static const struct evsel_str_handler lock_tracepoints[] = {
1466 	{ "lock:lock_acquire",	 evsel__process_lock_acquire,   }, /* CONFIG_LOCKDEP */
1467 	{ "lock:lock_acquired",	 evsel__process_lock_acquired,  }, /* CONFIG_LOCKDEP, CONFIG_LOCK_STAT */
1468 	{ "lock:lock_contended", evsel__process_lock_contended, }, /* CONFIG_LOCKDEP, CONFIG_LOCK_STAT */
1469 	{ "lock:lock_release",	 evsel__process_lock_release,   }, /* CONFIG_LOCKDEP */
1470 };
1471 
1472 static const struct evsel_str_handler contention_tracepoints[] = {
1473 	{ "lock:contention_begin", evsel__process_contention_begin, },
1474 	{ "lock:contention_end",   evsel__process_contention_end,   },
1475 };
1476 
1477 static int process_event_update(struct perf_tool *tool,
1478 				union perf_event *event,
1479 				struct evlist **pevlist)
1480 {
1481 	int ret;
1482 
1483 	ret = perf_event__process_event_update(tool, event, pevlist);
1484 	if (ret < 0)
1485 		return ret;
1486 
1487 	/* this can return -EEXIST since we call it for each evsel */
1488 	perf_session__set_tracepoints_handlers(session, lock_tracepoints);
1489 	perf_session__set_tracepoints_handlers(session, contention_tracepoints);
1490 	return 0;
1491 }
1492 
1493 typedef int (*tracepoint_handler)(struct evsel *evsel,
1494 				  struct perf_sample *sample);
1495 
1496 static int process_sample_event(struct perf_tool *tool __maybe_unused,
1497 				union perf_event *event,
1498 				struct perf_sample *sample,
1499 				struct evsel *evsel,
1500 				struct machine *machine)
1501 {
1502 	int err = 0;
1503 	struct thread *thread = machine__findnew_thread(machine, sample->pid,
1504 							sample->tid);
1505 
1506 	if (thread == NULL) {
1507 		pr_debug("problem processing %d event, skipping it.\n",
1508 			event->header.type);
1509 		return -1;
1510 	}
1511 
1512 	if (evsel->handler != NULL) {
1513 		tracepoint_handler f = evsel->handler;
1514 		err = f(evsel, sample);
1515 	}
1516 
1517 	thread__put(thread);
1518 
1519 	return err;
1520 }
1521 
1522 static void combine_result(void)
1523 {
1524 	unsigned int i;
1525 	struct lock_stat *st;
1526 
1527 	if (!combine_locks)
1528 		return;
1529 
1530 	for (i = 0; i < LOCKHASH_SIZE; i++) {
1531 		hlist_for_each_entry(st, &lockhash_table[i], hash_entry) {
1532 			combine_lock_stats(st);
1533 		}
1534 	}
1535 }
1536 
1537 static void sort_result(void)
1538 {
1539 	unsigned int i;
1540 	struct lock_stat *st;
1541 
1542 	for (i = 0; i < LOCKHASH_SIZE; i++) {
1543 		hlist_for_each_entry(st, &lockhash_table[i], hash_entry) {
1544 			insert_to_result(st, compare);
1545 		}
1546 	}
1547 }
1548 
1549 static const struct {
1550 	unsigned int flags;
1551 	const char *name;
1552 } lock_type_table[] = {
1553 	{ 0,				"semaphore" },
1554 	{ LCB_F_SPIN,			"spinlock" },
1555 	{ LCB_F_SPIN | LCB_F_READ,	"rwlock:R" },
1556 	{ LCB_F_SPIN | LCB_F_WRITE,	"rwlock:W"},
1557 	{ LCB_F_READ,			"rwsem:R" },
1558 	{ LCB_F_WRITE,			"rwsem:W" },
1559 	{ LCB_F_RT,			"rtmutex" },
1560 	{ LCB_F_RT | LCB_F_READ,	"rwlock-rt:R" },
1561 	{ LCB_F_RT | LCB_F_WRITE,	"rwlock-rt:W"},
1562 	{ LCB_F_PERCPU | LCB_F_READ,	"pcpu-sem:R" },
1563 	{ LCB_F_PERCPU | LCB_F_WRITE,	"pcpu-sem:W" },
1564 	{ LCB_F_MUTEX,			"mutex" },
1565 	{ LCB_F_MUTEX | LCB_F_SPIN,	"mutex" },
1566 	/* alias for get_type_flag() */
1567 	{ LCB_F_MUTEX | LCB_F_SPIN,	"mutex-spin" },
1568 };
1569 
1570 static const char *get_type_str(unsigned int flags)
1571 {
1572 	for (unsigned int i = 0; i < ARRAY_SIZE(lock_type_table); i++) {
1573 		if (lock_type_table[i].flags == flags)
1574 			return lock_type_table[i].name;
1575 	}
1576 	return "unknown";
1577 }
1578 
1579 static unsigned int get_type_flag(const char *str)
1580 {
1581 	for (unsigned int i = 0; i < ARRAY_SIZE(lock_type_table); i++) {
1582 		if (!strcmp(lock_type_table[i].name, str))
1583 			return lock_type_table[i].flags;
1584 	}
1585 	return UINT_MAX;
1586 }
1587 
1588 static void lock_filter_finish(void)
1589 {
1590 	zfree(&filters.types);
1591 	filters.nr_types = 0;
1592 
1593 	zfree(&filters.addrs);
1594 	filters.nr_addrs = 0;
1595 
1596 	for (int i = 0; i < filters.nr_syms; i++)
1597 		free(filters.syms[i]);
1598 
1599 	zfree(&filters.syms);
1600 	filters.nr_syms = 0;
1601 }
1602 
1603 static void sort_contention_result(void)
1604 {
1605 	sort_result();
1606 }
1607 
1608 static void print_contention_result(struct lock_contention *con)
1609 {
1610 	struct lock_stat *st;
1611 	struct lock_key *key;
1612 	int bad, total, printed;
1613 
1614 	if (!quiet) {
1615 		list_for_each_entry(key, &lock_keys, list)
1616 			pr_info("%*s ", key->len, key->header);
1617 
1618 		switch (aggr_mode) {
1619 		case LOCK_AGGR_TASK:
1620 			pr_info("  %10s   %s\n\n", "pid",
1621 				show_lock_owner ? "owner" : "comm");
1622 			break;
1623 		case LOCK_AGGR_CALLER:
1624 			pr_info("  %10s   %s\n\n", "type", "caller");
1625 			break;
1626 		case LOCK_AGGR_ADDR:
1627 			pr_info("  %16s   %s\n\n", "address", "symbol");
1628 			break;
1629 		default:
1630 			break;
1631 		}
1632 	}
1633 
1634 	bad = total = printed = 0;
1635 	if (use_bpf)
1636 		bad = bad_hist[BROKEN_CONTENDED];
1637 
1638 	while ((st = pop_from_result())) {
1639 		struct thread *t;
1640 		int pid;
1641 
1642 		total += use_bpf ? st->nr_contended : 1;
1643 		if (st->broken)
1644 			bad++;
1645 
1646 		if (!st->wait_time_total)
1647 			continue;
1648 
1649 		list_for_each_entry(key, &lock_keys, list) {
1650 			key->print(key, st);
1651 			pr_info(" ");
1652 		}
1653 
1654 		switch (aggr_mode) {
1655 		case LOCK_AGGR_CALLER:
1656 			pr_info("  %10s   %s\n", get_type_str(st->flags), st->name);
1657 			break;
1658 		case LOCK_AGGR_TASK:
1659 			pid = st->addr;
1660 			t = perf_session__findnew(session, pid);
1661 			pr_info("  %10d   %s\n",
1662 				pid, pid == -1 ? "Unknown" : thread__comm_str(t));
1663 			break;
1664 		case LOCK_AGGR_ADDR:
1665 			pr_info("  %016llx   %s\n", (unsigned long long)st->addr,
1666 				st->name ? : "");
1667 			break;
1668 		default:
1669 			break;
1670 		}
1671 
1672 		if (aggr_mode == LOCK_AGGR_CALLER && verbose > 0) {
1673 			struct map *kmap;
1674 			struct symbol *sym;
1675 			char buf[128];
1676 			u64 ip;
1677 
1678 			for (int i = 0; i < max_stack_depth; i++) {
1679 				if (!st->callstack || !st->callstack[i])
1680 					break;
1681 
1682 				ip = st->callstack[i];
1683 				sym = machine__find_kernel_symbol(con->machine, ip, &kmap);
1684 				get_symbol_name_offset(kmap, sym, ip, buf, sizeof(buf));
1685 				pr_info("\t\t\t%#lx  %s\n", (unsigned long)ip, buf);
1686 			}
1687 		}
1688 
1689 		if (++printed >= print_nr_entries)
1690 			break;
1691 	}
1692 
1693 	print_bad_events(bad, total);
1694 }
1695 
1696 static bool force;
1697 
1698 static int __cmd_report(bool display_info)
1699 {
1700 	int err = -EINVAL;
1701 	struct perf_tool eops = {
1702 		.attr		 = perf_event__process_attr,
1703 		.event_update	 = process_event_update,
1704 		.sample		 = process_sample_event,
1705 		.comm		 = perf_event__process_comm,
1706 		.mmap		 = perf_event__process_mmap,
1707 		.namespaces	 = perf_event__process_namespaces,
1708 		.tracing_data	 = perf_event__process_tracing_data,
1709 		.ordered_events	 = true,
1710 	};
1711 	struct perf_data data = {
1712 		.path  = input_name,
1713 		.mode  = PERF_DATA_MODE_READ,
1714 		.force = force,
1715 	};
1716 
1717 	session = perf_session__new(&data, &eops);
1718 	if (IS_ERR(session)) {
1719 		pr_err("Initializing perf session failed\n");
1720 		return PTR_ERR(session);
1721 	}
1722 
1723 	/* for lock function check */
1724 	symbol_conf.sort_by_name = true;
1725 	symbol_conf.allow_aliases = true;
1726 	symbol__init(&session->header.env);
1727 
1728 	if (!data.is_pipe) {
1729 		if (!perf_session__has_traces(session, "lock record"))
1730 			goto out_delete;
1731 
1732 		if (perf_session__set_tracepoints_handlers(session, lock_tracepoints)) {
1733 			pr_err("Initializing perf session tracepoint handlers failed\n");
1734 			goto out_delete;
1735 		}
1736 
1737 		if (perf_session__set_tracepoints_handlers(session, contention_tracepoints)) {
1738 			pr_err("Initializing perf session tracepoint handlers failed\n");
1739 			goto out_delete;
1740 		}
1741 	}
1742 
1743 	if (setup_output_field(false, output_fields))
1744 		goto out_delete;
1745 
1746 	if (select_key(false))
1747 		goto out_delete;
1748 
1749 	if (show_thread_stats)
1750 		aggr_mode = LOCK_AGGR_TASK;
1751 
1752 	err = perf_session__process_events(session);
1753 	if (err)
1754 		goto out_delete;
1755 
1756 	setup_pager();
1757 	if (display_info) /* used for info subcommand */
1758 		err = dump_info();
1759 	else {
1760 		combine_result();
1761 		sort_result();
1762 		print_result();
1763 	}
1764 
1765 out_delete:
1766 	perf_session__delete(session);
1767 	return err;
1768 }
1769 
1770 static void sighandler(int sig __maybe_unused)
1771 {
1772 }
1773 
1774 static int check_lock_contention_options(const struct option *options,
1775 					 const char * const *usage)
1776 
1777 {
1778 	if (show_thread_stats && show_lock_addrs) {
1779 		pr_err("Cannot use thread and addr mode together\n");
1780 		parse_options_usage(usage, options, "threads", 0);
1781 		parse_options_usage(NULL, options, "lock-addr", 0);
1782 		return -1;
1783 	}
1784 
1785 	if (show_lock_owner && !use_bpf) {
1786 		pr_err("Lock owners are available only with BPF\n");
1787 		parse_options_usage(usage, options, "lock-owner", 0);
1788 		parse_options_usage(NULL, options, "use-bpf", 0);
1789 		return -1;
1790 	}
1791 
1792 	if (show_lock_owner && show_lock_addrs) {
1793 		pr_err("Cannot use owner and addr mode together\n");
1794 		parse_options_usage(usage, options, "lock-owner", 0);
1795 		parse_options_usage(NULL, options, "lock-addr", 0);
1796 		return -1;
1797 	}
1798 
1799 	if (show_lock_owner)
1800 		show_thread_stats = true;
1801 
1802 	return 0;
1803 }
1804 
1805 static int __cmd_contention(int argc, const char **argv)
1806 {
1807 	int err = -EINVAL;
1808 	struct perf_tool eops = {
1809 		.attr		 = perf_event__process_attr,
1810 		.event_update	 = process_event_update,
1811 		.sample		 = process_sample_event,
1812 		.comm		 = perf_event__process_comm,
1813 		.mmap		 = perf_event__process_mmap,
1814 		.tracing_data	 = perf_event__process_tracing_data,
1815 		.ordered_events	 = true,
1816 	};
1817 	struct perf_data data = {
1818 		.path  = input_name,
1819 		.mode  = PERF_DATA_MODE_READ,
1820 		.force = force,
1821 	};
1822 	struct lock_contention con = {
1823 		.target = &target,
1824 		.result = &lockhash_table[0],
1825 		.map_nr_entries = bpf_map_entries,
1826 		.max_stack = max_stack_depth,
1827 		.stack_skip = stack_skip,
1828 		.filters = &filters,
1829 		.save_callstack = needs_callstack(),
1830 		.owner = show_lock_owner,
1831 	};
1832 
1833 	session = perf_session__new(use_bpf ? NULL : &data, &eops);
1834 	if (IS_ERR(session)) {
1835 		pr_err("Initializing perf session failed\n");
1836 		return PTR_ERR(session);
1837 	}
1838 
1839 	con.machine = &session->machines.host;
1840 
1841 	con.aggr_mode = aggr_mode = show_thread_stats ? LOCK_AGGR_TASK :
1842 		show_lock_addrs ? LOCK_AGGR_ADDR : LOCK_AGGR_CALLER;
1843 
1844 	if (con.aggr_mode == LOCK_AGGR_CALLER)
1845 		con.save_callstack = true;
1846 
1847 	/* for lock function check */
1848 	symbol_conf.sort_by_name = true;
1849 	symbol_conf.allow_aliases = true;
1850 	symbol__init(&session->header.env);
1851 
1852 	if (use_bpf) {
1853 		err = target__validate(&target);
1854 		if (err) {
1855 			char errbuf[512];
1856 
1857 			target__strerror(&target, err, errbuf, 512);
1858 			pr_err("%s\n", errbuf);
1859 			goto out_delete;
1860 		}
1861 
1862 		signal(SIGINT, sighandler);
1863 		signal(SIGCHLD, sighandler);
1864 		signal(SIGTERM, sighandler);
1865 
1866 		con.evlist = evlist__new();
1867 		if (con.evlist == NULL) {
1868 			err = -ENOMEM;
1869 			goto out_delete;
1870 		}
1871 
1872 		err = evlist__create_maps(con.evlist, &target);
1873 		if (err < 0)
1874 			goto out_delete;
1875 
1876 		if (argc) {
1877 			err = evlist__prepare_workload(con.evlist, &target,
1878 						       argv, false, NULL);
1879 			if (err < 0)
1880 				goto out_delete;
1881 		}
1882 
1883 		if (lock_contention_prepare(&con) < 0) {
1884 			pr_err("lock contention BPF setup failed\n");
1885 			goto out_delete;
1886 		}
1887 	} else if (!data.is_pipe) {
1888 		if (!perf_session__has_traces(session, "lock record"))
1889 			goto out_delete;
1890 
1891 		if (!evlist__find_evsel_by_str(session->evlist,
1892 					       "lock:contention_begin")) {
1893 			pr_err("lock contention evsel not found\n");
1894 			goto out_delete;
1895 		}
1896 
1897 		if (perf_session__set_tracepoints_handlers(session,
1898 						contention_tracepoints)) {
1899 			pr_err("Initializing perf session tracepoint handlers failed\n");
1900 			goto out_delete;
1901 		}
1902 	}
1903 
1904 	if (setup_output_field(true, output_fields))
1905 		goto out_delete;
1906 
1907 	if (select_key(true))
1908 		goto out_delete;
1909 
1910 	if (use_bpf) {
1911 		lock_contention_start();
1912 		if (argc)
1913 			evlist__start_workload(con.evlist);
1914 
1915 		/* wait for signal */
1916 		pause();
1917 
1918 		lock_contention_stop();
1919 		lock_contention_read(&con);
1920 
1921 		/* abuse bad hist stats for lost entries */
1922 		bad_hist[BROKEN_CONTENDED] = con.lost;
1923 	} else {
1924 		err = perf_session__process_events(session);
1925 		if (err)
1926 			goto out_delete;
1927 	}
1928 
1929 	setup_pager();
1930 
1931 	sort_contention_result();
1932 	print_contention_result(&con);
1933 
1934 out_delete:
1935 	lock_filter_finish();
1936 	evlist__delete(con.evlist);
1937 	lock_contention_finish();
1938 	perf_session__delete(session);
1939 	return err;
1940 }
1941 
1942 
1943 static int __cmd_record(int argc, const char **argv)
1944 {
1945 	const char *record_args[] = {
1946 		"record", "-R", "-m", "1024", "-c", "1", "--synth", "task",
1947 	};
1948 	const char *callgraph_args[] = {
1949 		"--call-graph", "fp," __stringify(CONTENTION_STACK_DEPTH),
1950 	};
1951 	unsigned int rec_argc, i, j, ret;
1952 	unsigned int nr_tracepoints;
1953 	unsigned int nr_callgraph_args = 0;
1954 	const char **rec_argv;
1955 	bool has_lock_stat = true;
1956 
1957 	for (i = 0; i < ARRAY_SIZE(lock_tracepoints); i++) {
1958 		if (!is_valid_tracepoint(lock_tracepoints[i].name)) {
1959 			pr_debug("tracepoint %s is not enabled. "
1960 				 "Are CONFIG_LOCKDEP and CONFIG_LOCK_STAT enabled?\n",
1961 				 lock_tracepoints[i].name);
1962 			has_lock_stat = false;
1963 			break;
1964 		}
1965 	}
1966 
1967 	if (has_lock_stat)
1968 		goto setup_args;
1969 
1970 	for (i = 0; i < ARRAY_SIZE(contention_tracepoints); i++) {
1971 		if (!is_valid_tracepoint(contention_tracepoints[i].name)) {
1972 			pr_err("tracepoint %s is not enabled.\n",
1973 			       contention_tracepoints[i].name);
1974 			return 1;
1975 		}
1976 	}
1977 
1978 	nr_callgraph_args = ARRAY_SIZE(callgraph_args);
1979 
1980 setup_args:
1981 	rec_argc = ARRAY_SIZE(record_args) + nr_callgraph_args + argc - 1;
1982 
1983 	if (has_lock_stat)
1984 		nr_tracepoints = ARRAY_SIZE(lock_tracepoints);
1985 	else
1986 		nr_tracepoints = ARRAY_SIZE(contention_tracepoints);
1987 
1988 	/* factor of 2 is for -e in front of each tracepoint */
1989 	rec_argc += 2 * nr_tracepoints;
1990 
1991 	rec_argv = calloc(rec_argc + 1, sizeof(char *));
1992 	if (!rec_argv)
1993 		return -ENOMEM;
1994 
1995 	for (i = 0; i < ARRAY_SIZE(record_args); i++)
1996 		rec_argv[i] = strdup(record_args[i]);
1997 
1998 	for (j = 0; j < nr_tracepoints; j++) {
1999 		const char *ev_name;
2000 
2001 		if (has_lock_stat)
2002 			ev_name = strdup(lock_tracepoints[j].name);
2003 		else
2004 			ev_name = strdup(contention_tracepoints[j].name);
2005 
2006 		if (!ev_name)
2007 			return -ENOMEM;
2008 
2009 		rec_argv[i++] = "-e";
2010 		rec_argv[i++] = ev_name;
2011 	}
2012 
2013 	for (j = 0; j < nr_callgraph_args; j++, i++)
2014 		rec_argv[i] = callgraph_args[j];
2015 
2016 	for (j = 1; j < (unsigned int)argc; j++, i++)
2017 		rec_argv[i] = argv[j];
2018 
2019 	BUG_ON(i != rec_argc);
2020 
2021 	ret = cmd_record(i, rec_argv);
2022 	free(rec_argv);
2023 	return ret;
2024 }
2025 
2026 static int parse_map_entry(const struct option *opt, const char *str,
2027 			    int unset __maybe_unused)
2028 {
2029 	unsigned long *len = (unsigned long *)opt->value;
2030 	unsigned long val;
2031 	char *endptr;
2032 
2033 	errno = 0;
2034 	val = strtoul(str, &endptr, 0);
2035 	if (*endptr != '\0' || errno != 0) {
2036 		pr_err("invalid BPF map length: %s\n", str);
2037 		return -1;
2038 	}
2039 
2040 	*len = val;
2041 	return 0;
2042 }
2043 
2044 static int parse_max_stack(const struct option *opt, const char *str,
2045 			   int unset __maybe_unused)
2046 {
2047 	unsigned long *len = (unsigned long *)opt->value;
2048 	long val;
2049 	char *endptr;
2050 
2051 	errno = 0;
2052 	val = strtol(str, &endptr, 0);
2053 	if (*endptr != '\0' || errno != 0) {
2054 		pr_err("invalid max stack depth: %s\n", str);
2055 		return -1;
2056 	}
2057 
2058 	if (val < 0 || val > sysctl__max_stack()) {
2059 		pr_err("invalid max stack depth: %ld\n", val);
2060 		return -1;
2061 	}
2062 
2063 	*len = val;
2064 	return 0;
2065 }
2066 
2067 static bool add_lock_type(unsigned int flags)
2068 {
2069 	unsigned int *tmp;
2070 
2071 	tmp = realloc(filters.types, (filters.nr_types + 1) * sizeof(*filters.types));
2072 	if (tmp == NULL)
2073 		return false;
2074 
2075 	tmp[filters.nr_types++] = flags;
2076 	filters.types = tmp;
2077 	return true;
2078 }
2079 
2080 static int parse_lock_type(const struct option *opt __maybe_unused, const char *str,
2081 			   int unset __maybe_unused)
2082 {
2083 	char *s, *tmp, *tok;
2084 	int ret = 0;
2085 
2086 	s = strdup(str);
2087 	if (s == NULL)
2088 		return -1;
2089 
2090 	for (tok = strtok_r(s, ", ", &tmp); tok; tok = strtok_r(NULL, ", ", &tmp)) {
2091 		unsigned int flags = get_type_flag(tok);
2092 
2093 		if (flags == -1U) {
2094 			char buf[32];
2095 
2096 			if (strchr(tok, ':'))
2097 			    continue;
2098 
2099 			/* try :R and :W suffixes for rwlock, rwsem, ... */
2100 			scnprintf(buf, sizeof(buf), "%s:R", tok);
2101 			flags = get_type_flag(buf);
2102 			if (flags != UINT_MAX) {
2103 				if (!add_lock_type(flags)) {
2104 					ret = -1;
2105 					break;
2106 				}
2107 			}
2108 
2109 			scnprintf(buf, sizeof(buf), "%s:W", tok);
2110 			flags = get_type_flag(buf);
2111 			if (flags != UINT_MAX) {
2112 				if (!add_lock_type(flags)) {
2113 					ret = -1;
2114 					break;
2115 				}
2116 			}
2117 			continue;
2118 		}
2119 
2120 		if (!add_lock_type(flags)) {
2121 			ret = -1;
2122 			break;
2123 		}
2124 
2125 		if (!strcmp(tok, "mutex")) {
2126 			flags = get_type_flag("mutex-spin");
2127 			if (flags != UINT_MAX) {
2128 				if (!add_lock_type(flags)) {
2129 					ret = -1;
2130 					break;
2131 				}
2132 			}
2133 		}
2134 	}
2135 
2136 	free(s);
2137 	return ret;
2138 }
2139 
2140 static bool add_lock_addr(unsigned long addr)
2141 {
2142 	unsigned long *tmp;
2143 
2144 	tmp = realloc(filters.addrs, (filters.nr_addrs + 1) * sizeof(*filters.addrs));
2145 	if (tmp == NULL) {
2146 		pr_err("Memory allocation failure\n");
2147 		return false;
2148 	}
2149 
2150 	tmp[filters.nr_addrs++] = addr;
2151 	filters.addrs = tmp;
2152 	return true;
2153 }
2154 
2155 static bool add_lock_sym(char *name)
2156 {
2157 	char **tmp;
2158 	char *sym = strdup(name);
2159 
2160 	if (sym == NULL) {
2161 		pr_err("Memory allocation failure\n");
2162 		return false;
2163 	}
2164 
2165 	tmp = realloc(filters.syms, (filters.nr_syms + 1) * sizeof(*filters.syms));
2166 	if (tmp == NULL) {
2167 		pr_err("Memory allocation failure\n");
2168 		free(sym);
2169 		return false;
2170 	}
2171 
2172 	tmp[filters.nr_syms++] = sym;
2173 	filters.syms = tmp;
2174 	return true;
2175 }
2176 
2177 static int parse_lock_addr(const struct option *opt __maybe_unused, const char *str,
2178 			   int unset __maybe_unused)
2179 {
2180 	char *s, *tmp, *tok;
2181 	int ret = 0;
2182 	u64 addr;
2183 
2184 	s = strdup(str);
2185 	if (s == NULL)
2186 		return -1;
2187 
2188 	for (tok = strtok_r(s, ", ", &tmp); tok; tok = strtok_r(NULL, ", ", &tmp)) {
2189 		char *end;
2190 
2191 		addr = strtoul(tok, &end, 16);
2192 		if (*end == '\0') {
2193 			if (!add_lock_addr(addr)) {
2194 				ret = -1;
2195 				break;
2196 			}
2197 			continue;
2198 		}
2199 
2200 		/*
2201 		 * At this moment, we don't have kernel symbols.  Save the symbols
2202 		 * in a separate list and resolve them to addresses later.
2203 		 */
2204 		if (!add_lock_sym(tok)) {
2205 			ret = -1;
2206 			break;
2207 		}
2208 	}
2209 
2210 	free(s);
2211 	return ret;
2212 }
2213 
2214 static int parse_call_stack(const struct option *opt __maybe_unused, const char *str,
2215 			   int unset __maybe_unused)
2216 {
2217 	char *s, *tmp, *tok;
2218 	int ret = 0;
2219 
2220 	s = strdup(str);
2221 	if (s == NULL)
2222 		return -1;
2223 
2224 	for (tok = strtok_r(s, ", ", &tmp); tok; tok = strtok_r(NULL, ", ", &tmp)) {
2225 		struct callstack_filter *entry;
2226 
2227 		entry = malloc(sizeof(*entry) + strlen(tok) + 1);
2228 		if (entry == NULL) {
2229 			pr_err("Memory allocation failure\n");
2230 			return -1;
2231 		}
2232 
2233 		strcpy(entry->name, tok);
2234 		list_add_tail(&entry->list, &callstack_filters);
2235 	}
2236 
2237 	free(s);
2238 	return ret;
2239 }
2240 
2241 int cmd_lock(int argc, const char **argv)
2242 {
2243 	const struct option lock_options[] = {
2244 	OPT_STRING('i', "input", &input_name, "file", "input file name"),
2245 	OPT_INCR('v', "verbose", &verbose, "be more verbose (show symbol address, etc)"),
2246 	OPT_BOOLEAN('D', "dump-raw-trace", &dump_trace, "dump raw trace in ASCII"),
2247 	OPT_BOOLEAN('f', "force", &force, "don't complain, do it"),
2248 	OPT_STRING(0, "vmlinux", &symbol_conf.vmlinux_name,
2249 		   "file", "vmlinux pathname"),
2250 	OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name,
2251 		   "file", "kallsyms pathname"),
2252 	OPT_BOOLEAN('q', "quiet", &quiet, "Do not show any warnings or messages"),
2253 	OPT_END()
2254 	};
2255 
2256 	const struct option info_options[] = {
2257 	OPT_BOOLEAN('t', "threads", &info_threads,
2258 		    "dump thread list in perf.data"),
2259 	OPT_BOOLEAN('m', "map", &info_map,
2260 		    "map of lock instances (address:name table)"),
2261 	OPT_PARENT(lock_options)
2262 	};
2263 
2264 	const struct option report_options[] = {
2265 	OPT_STRING('k', "key", &sort_key, "acquired",
2266 		    "key for sorting (acquired / contended / avg_wait / wait_total / wait_max / wait_min)"),
2267 	OPT_STRING('F', "field", &output_fields, NULL,
2268 		    "output fields (acquired / contended / avg_wait / wait_total / wait_max / wait_min)"),
2269 	/* TODO: type */
2270 	OPT_BOOLEAN('c', "combine-locks", &combine_locks,
2271 		    "combine locks in the same class"),
2272 	OPT_BOOLEAN('t', "threads", &show_thread_stats,
2273 		    "show per-thread lock stats"),
2274 	OPT_INTEGER('E', "entries", &print_nr_entries, "display this many functions"),
2275 	OPT_PARENT(lock_options)
2276 	};
2277 
2278 	struct option contention_options[] = {
2279 	OPT_STRING('k', "key", &sort_key, "wait_total",
2280 		    "key for sorting (contended / wait_total / wait_max / wait_min / avg_wait)"),
2281 	OPT_STRING('F', "field", &output_fields, "contended,wait_total,wait_max,avg_wait",
2282 		    "output fields (contended / wait_total / wait_max / wait_min / avg_wait)"),
2283 	OPT_BOOLEAN('t', "threads", &show_thread_stats,
2284 		    "show per-thread lock stats"),
2285 	OPT_BOOLEAN('b', "use-bpf", &use_bpf, "use BPF program to collect lock contention stats"),
2286 	OPT_BOOLEAN('a', "all-cpus", &target.system_wide,
2287 		    "System-wide collection from all CPUs"),
2288 	OPT_STRING('C', "cpu", &target.cpu_list, "cpu",
2289 		    "List of cpus to monitor"),
2290 	OPT_STRING('p', "pid", &target.pid, "pid",
2291 		   "Trace on existing process id"),
2292 	OPT_STRING(0, "tid", &target.tid, "tid",
2293 		   "Trace on existing thread id (exclusive to --pid)"),
2294 	OPT_CALLBACK(0, "map-nr-entries", &bpf_map_entries, "num",
2295 		     "Max number of BPF map entries", parse_map_entry),
2296 	OPT_CALLBACK(0, "max-stack", &max_stack_depth, "num",
2297 		     "Set the maximum stack depth when collecting lopck contention, "
2298 		     "Default: " __stringify(CONTENTION_STACK_DEPTH), parse_max_stack),
2299 	OPT_INTEGER(0, "stack-skip", &stack_skip,
2300 		    "Set the number of stack depth to skip when finding a lock caller, "
2301 		    "Default: " __stringify(CONTENTION_STACK_SKIP)),
2302 	OPT_INTEGER('E', "entries", &print_nr_entries, "display this many functions"),
2303 	OPT_BOOLEAN('l', "lock-addr", &show_lock_addrs, "show lock stats by address"),
2304 	OPT_CALLBACK('Y', "type-filter", NULL, "FLAGS",
2305 		     "Filter specific type of locks", parse_lock_type),
2306 	OPT_CALLBACK('L', "lock-filter", NULL, "ADDRS/NAMES",
2307 		     "Filter specific address/symbol of locks", parse_lock_addr),
2308 	OPT_CALLBACK('S', "callstack-filter", NULL, "NAMES",
2309 		     "Filter specific function in the callstack", parse_call_stack),
2310 	OPT_BOOLEAN('o', "lock-owner", &show_lock_owner, "show lock owners instead of waiters"),
2311 	OPT_PARENT(lock_options)
2312 	};
2313 
2314 	const char * const info_usage[] = {
2315 		"perf lock info [<options>]",
2316 		NULL
2317 	};
2318 	const char *const lock_subcommands[] = { "record", "report", "script",
2319 						 "info", "contention", NULL };
2320 	const char *lock_usage[] = {
2321 		NULL,
2322 		NULL
2323 	};
2324 	const char * const report_usage[] = {
2325 		"perf lock report [<options>]",
2326 		NULL
2327 	};
2328 	const char * const contention_usage[] = {
2329 		"perf lock contention [<options>]",
2330 		NULL
2331 	};
2332 	unsigned int i;
2333 	int rc = 0;
2334 
2335 	for (i = 0; i < LOCKHASH_SIZE; i++)
2336 		INIT_HLIST_HEAD(lockhash_table + i);
2337 
2338 	argc = parse_options_subcommand(argc, argv, lock_options, lock_subcommands,
2339 					lock_usage, PARSE_OPT_STOP_AT_NON_OPTION);
2340 	if (!argc)
2341 		usage_with_options(lock_usage, lock_options);
2342 
2343 	if (strlen(argv[0]) > 2 && strstarts("record", argv[0])) {
2344 		return __cmd_record(argc, argv);
2345 	} else if (strlen(argv[0]) > 2 && strstarts("report", argv[0])) {
2346 		trace_handler = &report_lock_ops;
2347 		if (argc) {
2348 			argc = parse_options(argc, argv,
2349 					     report_options, report_usage, 0);
2350 			if (argc)
2351 				usage_with_options(report_usage, report_options);
2352 		}
2353 		rc = __cmd_report(false);
2354 	} else if (!strcmp(argv[0], "script")) {
2355 		/* Aliased to 'perf script' */
2356 		return cmd_script(argc, argv);
2357 	} else if (!strcmp(argv[0], "info")) {
2358 		if (argc) {
2359 			argc = parse_options(argc, argv,
2360 					     info_options, info_usage, 0);
2361 			if (argc)
2362 				usage_with_options(info_usage, info_options);
2363 		}
2364 		/* recycling report_lock_ops */
2365 		trace_handler = &report_lock_ops;
2366 		rc = __cmd_report(true);
2367 	} else if (strlen(argv[0]) > 2 && strstarts("contention", argv[0])) {
2368 		trace_handler = &contention_lock_ops;
2369 		sort_key = "wait_total";
2370 		output_fields = "contended,wait_total,wait_max,avg_wait";
2371 
2372 #ifndef HAVE_BPF_SKEL
2373 		set_option_nobuild(contention_options, 'b', "use-bpf",
2374 				   "no BUILD_BPF_SKEL=1", false);
2375 #endif
2376 		if (argc) {
2377 			argc = parse_options(argc, argv, contention_options,
2378 					     contention_usage, 0);
2379 		}
2380 
2381 		if (check_lock_contention_options(contention_options,
2382 						  contention_usage) < 0)
2383 			return -1;
2384 
2385 		rc = __cmd_contention(argc, argv);
2386 	} else {
2387 		usage_with_options(lock_usage, lock_options);
2388 	}
2389 
2390 	return rc;
2391 }
2392