ref_tracker.h 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  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. struct ref_tracker;
  9. struct ref_tracker_dir {
  10. #ifdef CONFIG_REF_TRACKER
  11. spinlock_t lock;
  12. unsigned int quarantine_avail;
  13. refcount_t untracked;
  14. refcount_t no_tracker;
  15. bool dead;
  16. struct list_head list; /* List of active trackers */
  17. struct list_head quarantine; /* List of dead trackers */
  18. #endif
  19. };
  20. #ifdef CONFIG_REF_TRACKER
  21. static inline void ref_tracker_dir_init(struct ref_tracker_dir *dir,
  22. unsigned int quarantine_count)
  23. {
  24. INIT_LIST_HEAD(&dir->list);
  25. INIT_LIST_HEAD(&dir->quarantine);
  26. spin_lock_init(&dir->lock);
  27. dir->quarantine_avail = quarantine_count;
  28. dir->dead = false;
  29. refcount_set(&dir->untracked, 1);
  30. refcount_set(&dir->no_tracker, 1);
  31. stack_depot_init();
  32. }
  33. void ref_tracker_dir_exit(struct ref_tracker_dir *dir);
  34. void ref_tracker_dir_print(struct ref_tracker_dir *dir,
  35. unsigned int display_limit);
  36. int ref_tracker_alloc(struct ref_tracker_dir *dir,
  37. struct ref_tracker **trackerp, gfp_t gfp);
  38. int ref_tracker_free(struct ref_tracker_dir *dir,
  39. struct ref_tracker **trackerp);
  40. #else /* CONFIG_REF_TRACKER */
  41. static inline void ref_tracker_dir_init(struct ref_tracker_dir *dir,
  42. unsigned int quarantine_count)
  43. {
  44. }
  45. static inline void ref_tracker_dir_exit(struct ref_tracker_dir *dir)
  46. {
  47. }
  48. static inline void ref_tracker_dir_print(struct ref_tracker_dir *dir,
  49. unsigned int display_limit)
  50. {
  51. }
  52. static inline int ref_tracker_alloc(struct ref_tracker_dir *dir,
  53. struct ref_tracker **trackerp,
  54. gfp_t gfp)
  55. {
  56. return 0;
  57. }
  58. static inline int ref_tracker_free(struct ref_tracker_dir *dir,
  59. struct ref_tracker **trackerp)
  60. {
  61. return 0;
  62. }
  63. #endif
  64. #endif /* _LINUX_REF_TRACKER_H */