1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * This code come from arch/arm64/kernel/return_address.c
4  *
5  * Copyright (C) 2023 SiFive.
6  */
7 
8 #include <linux/export.h>
9 #include <linux/kprobes.h>
10 #include <linux/stacktrace.h>
11 
12 struct return_address_data {
13 	unsigned int level;
14 	void *addr;
15 };
16 
17 static bool save_return_addr(void *d, unsigned long pc)
18 {
19 	struct return_address_data *data = d;
20 
21 	if (!data->level) {
22 		data->addr = (void *)pc;
23 		return false;
24 	}
25 
26 	--data->level;
27 
28 	return true;
29 }
30 NOKPROBE_SYMBOL(save_return_addr);
31 
32 noinline void *return_address(unsigned int level)
33 {
34 	struct return_address_data data;
35 
36 	data.level = level + 3;
37 	data.addr = NULL;
38 
39 	arch_stack_walk(save_return_addr, &data, current, NULL);
40 
41 	if (!data.level)
42 		return data.addr;
43 	else
44 		return NULL;
45 
46 }
47 EXPORT_SYMBOL_GPL(return_address);
48 NOKPROBE_SYMBOL(return_address);
49