1 // SPDX-License-Identifier: GPL-2.0
2 #include "util/debug.h"
3 #include "util/event.h"
4 #include <subcmd/parse-options.h>
5 #include "util/parse-branch-options.h"
6 #include <stdlib.h>
7 #include <string.h>
8 
9 #define BRANCH_OPT(n, m) \
10 	{ .name = n, .mode = (m) }
11 
12 #define BRANCH_END { .name = NULL }
13 
14 struct branch_mode {
15 	const char *name;
16 	int mode;
17 };
18 
19 static const struct branch_mode branch_modes[] = {
20 	BRANCH_OPT("u", PERF_SAMPLE_BRANCH_USER),
21 	BRANCH_OPT("k", PERF_SAMPLE_BRANCH_KERNEL),
22 	BRANCH_OPT("hv", PERF_SAMPLE_BRANCH_HV),
23 	BRANCH_OPT("any", PERF_SAMPLE_BRANCH_ANY),
24 	BRANCH_OPT("any_call", PERF_SAMPLE_BRANCH_ANY_CALL),
25 	BRANCH_OPT("any_ret", PERF_SAMPLE_BRANCH_ANY_RETURN),
26 	BRANCH_OPT("ind_call", PERF_SAMPLE_BRANCH_IND_CALL),
27 	BRANCH_OPT("abort_tx", PERF_SAMPLE_BRANCH_ABORT_TX),
28 	BRANCH_OPT("in_tx", PERF_SAMPLE_BRANCH_IN_TX),
29 	BRANCH_OPT("no_tx", PERF_SAMPLE_BRANCH_NO_TX),
30 	BRANCH_OPT("cond", PERF_SAMPLE_BRANCH_COND),
31 	BRANCH_OPT("ind_jmp", PERF_SAMPLE_BRANCH_IND_JUMP),
32 	BRANCH_OPT("call", PERF_SAMPLE_BRANCH_CALL),
33 	BRANCH_OPT("no_flags", PERF_SAMPLE_BRANCH_NO_FLAGS),
34 	BRANCH_OPT("no_cycles", PERF_SAMPLE_BRANCH_NO_CYCLES),
35 	BRANCH_OPT("save_type", PERF_SAMPLE_BRANCH_TYPE_SAVE),
36 	BRANCH_OPT("stack", PERF_SAMPLE_BRANCH_CALL_STACK),
37 	BRANCH_OPT("hw_index", PERF_SAMPLE_BRANCH_HW_INDEX),
38 	BRANCH_OPT("priv", PERF_SAMPLE_BRANCH_PRIV_SAVE),
39 	BRANCH_END
40 };
41 
42 int parse_branch_str(const char *str, __u64 *mode)
43 {
44 #define ONLY_PLM \
45 	(PERF_SAMPLE_BRANCH_USER	|\
46 	 PERF_SAMPLE_BRANCH_KERNEL	|\
47 	 PERF_SAMPLE_BRANCH_HV)
48 
49 	int ret = 0;
50 	char *p, *s;
51 	char *os = NULL;
52 	const struct branch_mode *br;
53 
54 	if (str == NULL) {
55 		*mode = PERF_SAMPLE_BRANCH_ANY;
56 		return 0;
57 	}
58 
59 	/* because str is read-only */
60 	s = os = strdup(str);
61 	if (!s)
62 		return -1;
63 
64 	for (;;) {
65 		p = strchr(s, ',');
66 		if (p)
67 			*p = '\0';
68 
69 		for (br = branch_modes; br->name; br++) {
70 			if (!strcasecmp(s, br->name))
71 				break;
72 		}
73 		if (!br->name) {
74 			ret = -1;
75 			pr_warning("unknown branch filter %s,"
76 				    " check man page\n", s);
77 			goto error;
78 		}
79 
80 		*mode |= br->mode;
81 
82 		if (!p)
83 			break;
84 
85 		s = p + 1;
86 	}
87 
88 	/* default to any branch */
89 	if ((*mode & ~ONLY_PLM) == 0) {
90 		*mode = PERF_SAMPLE_BRANCH_ANY;
91 	}
92 error:
93 	free(os);
94 	return ret;
95 }
96 
97 int
98 parse_branch_stack(const struct option *opt, const char *str, int unset)
99 {
100 	__u64 *mode = (__u64 *)opt->value;
101 
102 	if (unset)
103 		return 0;
104 
105 	/*
106 	 * cannot set it twice, -b + --branch-filter for instance
107 	 */
108 	if (*mode) {
109 		pr_err("Error: Can't use --branch-any (-b) with --branch-filter (-j).\n");
110 		return -1;
111 	}
112 
113 	return parse_branch_str(str, mode);
114 }
115