refcount.h 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. #ifndef _LINUX_REFCOUNT_H
  2. #define _LINUX_REFCOUNT_H
  3. #include <linux/atomic.h>
  4. #include <linux/mutex.h>
  5. #include <linux/spinlock.h>
  6. #include <linux/kernel.h>
  7. typedef struct refcount_struct {
  8. atomic_t refs;
  9. } refcount_t;
  10. #define REFCOUNT_INIT(n) { .refs = ATOMIC_INIT(n), }
  11. static inline void refcount_set(refcount_t *r, unsigned int n)
  12. {
  13. atomic_set(&r->refs, n);
  14. }
  15. static inline unsigned int refcount_read(const refcount_t *r)
  16. {
  17. return atomic_read(&r->refs);
  18. }
  19. extern __must_check bool refcount_add_not_zero(unsigned int i, refcount_t *r);
  20. extern void refcount_add(unsigned int i, refcount_t *r);
  21. extern __must_check bool refcount_inc_not_zero(refcount_t *r);
  22. extern void refcount_inc(refcount_t *r);
  23. extern __must_check bool refcount_sub_and_test(unsigned int i, refcount_t *r);
  24. extern void refcount_sub(unsigned int i, refcount_t *r);
  25. extern __must_check bool refcount_dec_and_test(refcount_t *r);
  26. extern void refcount_dec(refcount_t *r);
  27. extern __must_check bool refcount_dec_if_one(refcount_t *r);
  28. extern __must_check bool refcount_dec_not_one(refcount_t *r);
  29. extern __must_check bool refcount_dec_and_mutex_lock(refcount_t *r, struct mutex *lock);
  30. extern __must_check bool refcount_dec_and_lock(refcount_t *r, spinlock_t *lock);
  31. #endif /* _LINUX_REFCOUNT_H */