]>
Commit | Line | Data |
---|---|---|
f405df5d PZ |
1 | #ifndef _LINUX_REFCOUNT_H |
2 | #define _LINUX_REFCOUNT_H | |
3 | ||
f405df5d | 4 | #include <linux/atomic.h> |
f405df5d PZ |
5 | #include <linux/mutex.h> |
6 | #include <linux/spinlock.h> | |
318b1ded | 7 | #include <linux/kernel.h> |
f405df5d | 8 | |
bd174169 DW |
9 | /** |
10 | * refcount_t - variant of atomic_t specialized for reference counts | |
11 | * @refs: atomic_t counter field | |
12 | * | |
13 | * The counter saturates at UINT_MAX and will not move once | |
14 | * there. This avoids wrapping the counter and causing 'spurious' | |
15 | * use-after-free bugs. | |
16 | */ | |
f405df5d PZ |
17 | typedef struct refcount_struct { |
18 | atomic_t refs; | |
19 | } refcount_t; | |
20 | ||
21 | #define REFCOUNT_INIT(n) { .refs = ATOMIC_INIT(n), } | |
22 | ||
bd174169 DW |
23 | /** |
24 | * refcount_set - set a refcount's value | |
25 | * @r: the refcount | |
26 | * @n: value to which the refcount will be set | |
27 | */ | |
f405df5d PZ |
28 | static inline void refcount_set(refcount_t *r, unsigned int n) |
29 | { | |
30 | atomic_set(&r->refs, n); | |
31 | } | |
32 | ||
bd174169 DW |
33 | /** |
34 | * refcount_read - get a refcount's value | |
35 | * @r: the refcount | |
36 | * | |
37 | * Return: the refcount's value | |
38 | */ | |
f405df5d PZ |
39 | static inline unsigned int refcount_read(const refcount_t *r) |
40 | { | |
41 | return atomic_read(&r->refs); | |
42 | } | |
43 | ||
29dee3c0 PZ |
44 | extern __must_check bool refcount_add_not_zero(unsigned int i, refcount_t *r); |
45 | extern void refcount_add(unsigned int i, refcount_t *r); | |
f405df5d | 46 | |
29dee3c0 PZ |
47 | extern __must_check bool refcount_inc_not_zero(refcount_t *r); |
48 | extern void refcount_inc(refcount_t *r); | |
f405df5d | 49 | |
29dee3c0 PZ |
50 | extern __must_check bool refcount_sub_and_test(unsigned int i, refcount_t *r); |
51 | extern void refcount_sub(unsigned int i, refcount_t *r); | |
f405df5d | 52 | |
29dee3c0 PZ |
53 | extern __must_check bool refcount_dec_and_test(refcount_t *r); |
54 | extern void refcount_dec(refcount_t *r); | |
f405df5d | 55 | |
29dee3c0 PZ |
56 | extern __must_check bool refcount_dec_if_one(refcount_t *r); |
57 | extern __must_check bool refcount_dec_not_one(refcount_t *r); | |
58 | extern __must_check bool refcount_dec_and_mutex_lock(refcount_t *r, struct mutex *lock); | |
59 | extern __must_check bool refcount_dec_and_lock(refcount_t *r, spinlock_t *lock); | |
f405df5d PZ |
60 | |
61 | #endif /* _LINUX_REFCOUNT_H */ |