xref: /linux-6.15/include/linux/stackdepot.h (revision 36aa1e67)
1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 /*
3  * A generic stack depot implementation
4  *
5  * Author: Alexander Potapenko <[email protected]>
6  * Copyright (C) 2016 Google, Inc.
7  *
8  * Based on code by Dmitry Chernenkov.
9  */
10 
11 #ifndef _LINUX_STACKDEPOT_H
12 #define _LINUX_STACKDEPOT_H
13 
14 #include <linux/gfp.h>
15 
16 typedef u32 depot_stack_handle_t;
17 
18 /*
19  * Number of bits in the handle that stack depot doesn't use. Users may store
20  * information in them.
21  */
22 #define STACK_DEPOT_EXTRA_BITS 5
23 
24 /*
25  * Every user of stack depot has to call stack_depot_init() during its own init
26  * when it's decided that it will be calling stack_depot_save() later. This is
27  * recommended for e.g. modules initialized later in the boot process, when
28  * slab_is_available() is true.
29  *
30  * The alternative is to select STACKDEPOT_ALWAYS_INIT to have stack depot
31  * enabled as part of mm_init(), for subsystems where it's known at compile time
32  * that stack depot will be used.
33  *
34  * Another alternative is to call stack_depot_request_early_init(), when the
35  * decision to use stack depot is taken e.g. when evaluating kernel boot
36  * parameters, which precedes the enablement point in mm_init().
37  *
38  * stack_depot_init() and stack_depot_request_early_init() can be called
39  * regardless of CONFIG_STACKDEPOT and are no-op when disabled. The actual
40  * save/fetch/print functions should only be called from code that makes sure
41  * CONFIG_STACKDEPOT is enabled.
42  */
43 #ifdef CONFIG_STACKDEPOT
44 int stack_depot_init(void);
45 
46 void __init stack_depot_request_early_init(void);
47 
48 /* This is supposed to be called only from mm_init() */
49 int __init stack_depot_early_init(void);
50 #else
51 static inline int stack_depot_init(void) { return 0; }
52 
53 static inline void stack_depot_request_early_init(void) { }
54 
55 static inline int stack_depot_early_init(void)	{ return 0; }
56 #endif
57 
58 depot_stack_handle_t __stack_depot_save(unsigned long *entries,
59 					unsigned int nr_entries,
60 					gfp_t gfp_flags, bool can_alloc);
61 
62 depot_stack_handle_t stack_depot_save(unsigned long *entries,
63 				      unsigned int nr_entries, gfp_t gfp_flags);
64 
65 unsigned int stack_depot_fetch(depot_stack_handle_t handle,
66 			       unsigned long **entries);
67 
68 void stack_depot_print(depot_stack_handle_t stack);
69 
70 int stack_depot_snprint(depot_stack_handle_t handle, char *buf, size_t size,
71 		       int spaces);
72 
73 depot_stack_handle_t __must_check stack_depot_set_extra_bits(
74 			depot_stack_handle_t handle, unsigned int extra_bits);
75 
76 unsigned int stack_depot_get_extra_bits(depot_stack_handle_t handle);
77 
78 #endif
79