1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/compiler.h>
3 #include <sys/types.h>
4 #include <regex.h>
5 
6 struct arm64_annotate {
7 	regex_t call_insn,
8 		jump_insn;
9 };
10 
11 static struct ins_ops *arm64__associate_instruction_ops(struct arch *arch, const char *name)
12 {
13 	struct arm64_annotate *arm = arch->priv;
14 	struct ins_ops *ops;
15 	regmatch_t match[2];
16 
17 	if (!regexec(&arm->jump_insn, name, 2, match, 0))
18 		ops = &jump_ops;
19 	else if (!regexec(&arm->call_insn, name, 2, match, 0))
20 		ops = &call_ops;
21 	else if (!strcmp(name, "ret"))
22 		ops = &ret_ops;
23 	else
24 		return NULL;
25 
26 	arch__associate_ins_ops(arch, name, ops);
27 	return ops;
28 }
29 
30 static int arm64__annotate_init(struct arch *arch, char *cpuid __maybe_unused)
31 {
32 	struct arm64_annotate *arm;
33 	int err;
34 
35 	if (arch->initialized)
36 		return 0;
37 
38 	arm = zalloc(sizeof(*arm));
39 	if (!arm)
40 		return -1;
41 
42 	/* bl, blr */
43 	err = regcomp(&arm->call_insn, "^blr?$", REG_EXTENDED);
44 	if (err)
45 		goto out_free_arm;
46 	/* b, b.cond, br, cbz/cbnz, tbz/tbnz */
47 	err = regcomp(&arm->jump_insn, "^[ct]?br?\\.?(cc|cs|eq|ge|gt|hi|le|ls|lt|mi|ne|pl)?n?z?$",
48 		      REG_EXTENDED);
49 	if (err)
50 		goto out_free_call;
51 
52 	arch->initialized = true;
53 	arch->priv	  = arm;
54 	arch->associate_instruction_ops   = arm64__associate_instruction_ops;
55 	arch->objdump.comment_char	  = '/';
56 	arch->objdump.skip_functions_char = '+';
57 	return 0;
58 
59 out_free_call:
60 	regfree(&arm->call_insn);
61 out_free_arm:
62 	free(arm);
63 	return -1;
64 }
65