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 #endif 21 }; 22 23 #ifdef CONFIG_REF_TRACKER 24 static inline void ref_tracker_dir_init(struct ref_tracker_dir *dir, 25 unsigned int quarantine_count) 26 { 27 INIT_LIST_HEAD(&dir->list); 28 INIT_LIST_HEAD(&dir->quarantine); 29 spin_lock_init(&dir->lock); 30 dir->quarantine_avail = quarantine_count; 31 dir->dead = false; 32 refcount_set(&dir->untracked, 1); 33 refcount_set(&dir->no_tracker, 1); 34 stack_depot_init(); 35 } 36 37 void ref_tracker_dir_exit(struct ref_tracker_dir *dir); 38 39 void ref_tracker_dir_print(struct ref_tracker_dir *dir, 40 unsigned int display_limit); 41 42 int ref_tracker_alloc(struct ref_tracker_dir *dir, 43 struct ref_tracker **trackerp, gfp_t gfp); 44 45 int ref_tracker_free(struct ref_tracker_dir *dir, 46 struct ref_tracker **trackerp); 47 48 #else /* CONFIG_REF_TRACKER */ 49 50 static inline void ref_tracker_dir_init(struct ref_tracker_dir *dir, 51 unsigned int quarantine_count) 52 { 53 } 54 55 static inline void ref_tracker_dir_exit(struct ref_tracker_dir *dir) 56 { 57 } 58 59 static inline void ref_tracker_dir_print(struct ref_tracker_dir *dir, 60 unsigned int display_limit) 61 { 62 } 63 64 static inline int ref_tracker_alloc(struct ref_tracker_dir *dir, 65 struct ref_tracker **trackerp, 66 gfp_t gfp) 67 { 68 return 0; 69 } 70 71 static inline int ref_tracker_free(struct ref_tracker_dir *dir, 72 struct ref_tracker **trackerp) 73 { 74 return 0; 75 } 76 77 #endif 78 79 #endif /* _LINUX_REF_TRACKER_H */ 80