1 | // SPDX-License-Identifier: GPL-2.0-or-later |
2 | #ifndef _LINUX_REF_TRACKER_H |
3 | #define _LINUX_REF_TRACKER_H |
4 | #include <linux/refcount.h> |
5 | #include <linux/types.h> |
6 | #include <linux/spinlock.h> |
7 | #include <linux/stackdepot.h> |
8 | |
9 | struct ref_tracker; |
10 | |
11 | struct ref_tracker_dir { |
12 | #ifdef CONFIG_REF_TRACKER |
13 | spinlock_t lock; |
14 | unsigned int quarantine_avail; |
15 | refcount_t untracked; |
16 | refcount_t no_tracker; |
17 | bool dead; |
18 | struct list_head list; /* List of active trackers */ |
19 | struct list_head quarantine; /* List of dead trackers */ |
20 | char name[32]; |
21 | #endif |
22 | }; |
23 | |
24 | #ifdef CONFIG_REF_TRACKER |
25 | |
26 | static inline void ref_tracker_dir_init(struct ref_tracker_dir *dir, |
27 | unsigned int quarantine_count, |
28 | const char *name) |
29 | { |
30 | INIT_LIST_HEAD(list: &dir->list); |
31 | INIT_LIST_HEAD(list: &dir->quarantine); |
32 | spin_lock_init(&dir->lock); |
33 | dir->quarantine_avail = quarantine_count; |
34 | dir->dead = false; |
35 | refcount_set(r: &dir->untracked, n: 1); |
36 | refcount_set(r: &dir->no_tracker, n: 1); |
37 | strscpy(dir->name, name, sizeof(dir->name)); |
38 | stack_depot_init(); |
39 | } |
40 | |
41 | void ref_tracker_dir_exit(struct ref_tracker_dir *dir); |
42 | |
43 | void ref_tracker_dir_print_locked(struct ref_tracker_dir *dir, |
44 | unsigned int display_limit); |
45 | |
46 | void ref_tracker_dir_print(struct ref_tracker_dir *dir, |
47 | unsigned int display_limit); |
48 | |
49 | int ref_tracker_dir_snprint(struct ref_tracker_dir *dir, char *buf, size_t size); |
50 | |
51 | int ref_tracker_alloc(struct ref_tracker_dir *dir, |
52 | struct ref_tracker **trackerp, gfp_t gfp); |
53 | |
54 | int ref_tracker_free(struct ref_tracker_dir *dir, |
55 | struct ref_tracker **trackerp); |
56 | |
57 | #else /* CONFIG_REF_TRACKER */ |
58 | |
59 | static inline void ref_tracker_dir_init(struct ref_tracker_dir *dir, |
60 | unsigned int quarantine_count, |
61 | const char *name) |
62 | { |
63 | } |
64 | |
65 | static inline void ref_tracker_dir_exit(struct ref_tracker_dir *dir) |
66 | { |
67 | } |
68 | |
69 | static inline void ref_tracker_dir_print_locked(struct ref_tracker_dir *dir, |
70 | unsigned int display_limit) |
71 | { |
72 | } |
73 | |
74 | static inline void ref_tracker_dir_print(struct ref_tracker_dir *dir, |
75 | unsigned int display_limit) |
76 | { |
77 | } |
78 | |
79 | static inline int ref_tracker_dir_snprint(struct ref_tracker_dir *dir, |
80 | char *buf, size_t size) |
81 | { |
82 | return 0; |
83 | } |
84 | |
85 | static inline int ref_tracker_alloc(struct ref_tracker_dir *dir, |
86 | struct ref_tracker **trackerp, |
87 | gfp_t gfp) |
88 | { |
89 | return 0; |
90 | } |
91 | |
92 | static inline int ref_tracker_free(struct ref_tracker_dir *dir, |
93 | struct ref_tracker **trackerp) |
94 | { |
95 | return 0; |
96 | } |
97 | |
98 | #endif |
99 | |
100 | #endif /* _LINUX_REF_TRACKER_H */ |
101 | |