1/* SPDX-License-Identifier: GPL-2.0 */
 2#ifndef _BCACHEFS_NOCOW_LOCKING_H
 3#define _BCACHEFS_NOCOW_LOCKING_H
 4
 5#include "bcachefs.h"
 6#include "alloc_background.h"
 7#include "nocow_locking_types.h"
 8
 9#include <linux/hash.h>
10
11static inline struct nocow_lock_bucket *bucket_nocow_lock(struct bucket_nocow_lock_table *t,
12							  u64 dev_bucket)
13{
14	unsigned h = hash_64(dev_bucket, BUCKET_NOCOW_LOCKS_BITS);
15
16	return t->l + (h & (BUCKET_NOCOW_LOCKS - 1));
17}
18
19#define BUCKET_NOCOW_LOCK_UPDATE	(1 << 0)
20
21bool bch2_bucket_nocow_is_locked(struct bucket_nocow_lock_table *, struct bpos);
22void bch2_bucket_nocow_unlock(struct bucket_nocow_lock_table *, struct bpos, int);
23bool __bch2_bucket_nocow_trylock(struct nocow_lock_bucket *, u64, int);
24void __bch2_bucket_nocow_lock(struct bucket_nocow_lock_table *,
25			      struct nocow_lock_bucket *, u64, int);
26
27static inline void bch2_bucket_nocow_lock(struct bucket_nocow_lock_table *t,
28					  struct bpos bucket, int flags)
29{
30	u64 dev_bucket = bucket_to_u64(bucket);
31	struct nocow_lock_bucket *l = bucket_nocow_lock(t, dev_bucket);
32
33	__bch2_bucket_nocow_lock(t, l, dev_bucket, flags);
34}
35
36static inline bool bch2_bucket_nocow_trylock(struct bucket_nocow_lock_table *t,
37					  struct bpos bucket, int flags)
38{
39	u64 dev_bucket = bucket_to_u64(bucket);
40	struct nocow_lock_bucket *l = bucket_nocow_lock(t, dev_bucket);
41
42	return __bch2_bucket_nocow_trylock(l, dev_bucket, flags);
43}
44
45void bch2_nocow_locks_to_text(struct printbuf *, struct bucket_nocow_lock_table *);
46
47void bch2_fs_nocow_locking_exit(struct bch_fs *);
48int bch2_fs_nocow_locking_init(struct bch_fs *);
49
50#endif /* _BCACHEFS_NOCOW_LOCKING_H */