xref: /linux/fs/btrfs/delayed-ref.h (revision a1a9b71ebd6582c94090ee5b70fb3eabf7088f22)
1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * Copyright (C) 2008 Oracle.  All rights reserved.
4  */
5 
6 #ifndef BTRFS_DELAYED_REF_H
7 #define BTRFS_DELAYED_REF_H
8 
9 #include <linux/refcount.h>
10 
11 /* these are the possible values of struct btrfs_delayed_ref_node->action */
12 #define BTRFS_ADD_DELAYED_REF    1 /* add one backref to the tree */
13 #define BTRFS_DROP_DELAYED_REF   2 /* delete one backref from the tree */
14 #define BTRFS_ADD_DELAYED_EXTENT 3 /* record a full extent allocation */
15 #define BTRFS_UPDATE_DELAYED_HEAD 4 /* not changing ref count on head ref */
16 
17 struct btrfs_delayed_ref_node {
18 	struct rb_node ref_node;
19 	/*
20 	 * If action is BTRFS_ADD_DELAYED_REF, also link this node to
21 	 * ref_head->ref_add_list, then we do not need to iterate the
22 	 * whole ref_head->ref_list to find BTRFS_ADD_DELAYED_REF nodes.
23 	 */
24 	struct list_head add_list;
25 
26 	/* the starting bytenr of the extent */
27 	u64 bytenr;
28 
29 	/* the size of the extent */
30 	u64 num_bytes;
31 
32 	/* seq number to keep track of insertion order */
33 	u64 seq;
34 
35 	/* ref count on this data structure */
36 	refcount_t refs;
37 
38 	/*
39 	 * how many refs is this entry adding or deleting.  For
40 	 * head refs, this may be a negative number because it is keeping
41 	 * track of the total mods done to the reference count.
42 	 * For individual refs, this will always be a positive number
43 	 *
44 	 * It may be more than one, since it is possible for a single
45 	 * parent to have more than one ref on an extent
46 	 */
47 	int ref_mod;
48 
49 	unsigned int action:8;
50 	unsigned int type:8;
51 	/* is this node still in the rbtree? */
52 	unsigned int is_head:1;
53 	unsigned int in_tree:1;
54 };
55 
56 struct btrfs_delayed_extent_op {
57 	struct btrfs_disk_key key;
58 	u8 level;
59 	bool update_key;
60 	bool update_flags;
61 	u64 flags_to_set;
62 };
63 
64 /*
65  * the head refs are used to hold a lock on a given extent, which allows us
66  * to make sure that only one process is running the delayed refs
67  * at a time for a single extent.  They also store the sum of all the
68  * reference count modifications we've queued up.
69  */
70 struct btrfs_delayed_ref_head {
71 	u64 bytenr;
72 	u64 num_bytes;
73 	refcount_t refs;
74 	/*
75 	 * the mutex is held while running the refs, and it is also
76 	 * held when checking the sum of reference modifications.
77 	 */
78 	struct mutex mutex;
79 
80 	spinlock_t lock;
81 	struct rb_root_cached ref_tree;
82 	/* accumulate add BTRFS_ADD_DELAYED_REF nodes to this ref_add_list. */
83 	struct list_head ref_add_list;
84 
85 	struct rb_node href_node;
86 
87 	struct btrfs_delayed_extent_op *extent_op;
88 
89 	/*
90 	 * This is used to track the final ref_mod from all the refs associated
91 	 * with this head ref, this is not adjusted as delayed refs are run,
92 	 * this is meant to track if we need to do the csum accounting or not.
93 	 */
94 	int total_ref_mod;
95 
96 	/*
97 	 * This is the current outstanding mod references for this bytenr.  This
98 	 * is used with lookup_extent_info to get an accurate reference count
99 	 * for a bytenr, so it is adjusted as delayed refs are run so that any
100 	 * on disk reference count + ref_mod is accurate.
101 	 */
102 	int ref_mod;
103 
104 	/*
105 	 * when a new extent is allocated, it is just reserved in memory
106 	 * The actual extent isn't inserted into the extent allocation tree
107 	 * until the delayed ref is processed.  must_insert_reserved is
108 	 * used to flag a delayed ref so the accounting can be updated
109 	 * when a full insert is done.
110 	 *
111 	 * It is possible the extent will be freed before it is ever
112 	 * inserted into the extent allocation tree.  In this case
113 	 * we need to update the in ram accounting to properly reflect
114 	 * the free has happened.
115 	 */
116 	unsigned int must_insert_reserved:1;
117 	unsigned int is_data:1;
118 	unsigned int is_system:1;
119 	unsigned int processing:1;
120 };
121 
122 struct btrfs_delayed_tree_ref {
123 	struct btrfs_delayed_ref_node node;
124 	u64 root;
125 	u64 parent;
126 	int level;
127 };
128 
129 struct btrfs_delayed_data_ref {
130 	struct btrfs_delayed_ref_node node;
131 	u64 root;
132 	u64 parent;
133 	u64 objectid;
134 	u64 offset;
135 };
136 
137 enum btrfs_delayed_ref_flags {
138 	/* Indicate that we are flushing delayed refs for the commit */
139 	BTRFS_DELAYED_REFS_FLUSHING,
140 };
141 
142 struct btrfs_delayed_ref_root {
143 	/* head ref rbtree */
144 	struct rb_root_cached href_root;
145 
146 	/* dirty extent records */
147 	struct rb_root dirty_extent_root;
148 
149 	/* this spin lock protects the rbtree and the entries inside */
150 	spinlock_t lock;
151 
152 	/* how many delayed ref updates we've queued, used by the
153 	 * throttling code
154 	 */
155 	atomic_t num_entries;
156 
157 	/* total number of head nodes in tree */
158 	unsigned long num_heads;
159 
160 	/* total number of head nodes ready for processing */
161 	unsigned long num_heads_ready;
162 
163 	u64 pending_csums;
164 
165 	unsigned long flags;
166 
167 	u64 run_delayed_start;
168 
169 	/*
170 	 * To make qgroup to skip given root.
171 	 * This is for snapshot, as btrfs_qgroup_inherit() will manually
172 	 * modify counters for snapshot and its source, so we should skip
173 	 * the snapshot in new_root/old_roots or it will get calculated twice
174 	 */
175 	u64 qgroup_to_skip;
176 };
177 
178 enum btrfs_ref_type {
179 	BTRFS_REF_NOT_SET,
180 	BTRFS_REF_DATA,
181 	BTRFS_REF_METADATA,
182 	BTRFS_REF_LAST,
183 };
184 
185 struct btrfs_data_ref {
186 	/* For EXTENT_DATA_REF */
187 
188 	/* Original root this data extent belongs to */
189 	u64 owning_root;
190 
191 	/* Inode which refers to this data extent */
192 	u64 ino;
193 
194 	/*
195 	 * file_offset - extent_offset
196 	 *
197 	 * file_offset is the key.offset of the EXTENT_DATA key.
198 	 * extent_offset is btrfs_file_extent_offset() of the EXTENT_DATA data.
199 	 */
200 	u64 offset;
201 };
202 
203 struct btrfs_tree_ref {
204 	/*
205 	 * Level of this tree block
206 	 *
207 	 * Shared for skinny (TREE_BLOCK_REF) and normal tree ref.
208 	 */
209 	int level;
210 
211 	/*
212 	 * Root which owns this tree block.
213 	 *
214 	 * For TREE_BLOCK_REF (skinny metadata, either inline or keyed)
215 	 */
216 	u64 owning_root;
217 
218 	/* For non-skinny metadata, no special member needed */
219 };
220 
221 struct btrfs_ref {
222 	enum btrfs_ref_type type;
223 	int action;
224 
225 	/*
226 	 * Whether this extent should go through qgroup record.
227 	 *
228 	 * Normally false, but for certain cases like delayed subtree scan,
229 	 * setting this flag can hugely reduce qgroup overhead.
230 	 */
231 	bool skip_qgroup;
232 
233 #ifdef CONFIG_BTRFS_FS_REF_VERIFY
234 	/* Through which root is this modification. */
235 	u64 real_root;
236 #endif
237 	u64 bytenr;
238 	u64 len;
239 
240 	/* Bytenr of the parent tree block */
241 	u64 parent;
242 	union {
243 		struct btrfs_data_ref data_ref;
244 		struct btrfs_tree_ref tree_ref;
245 	};
246 };
247 
248 extern struct kmem_cache *btrfs_delayed_ref_head_cachep;
249 extern struct kmem_cache *btrfs_delayed_tree_ref_cachep;
250 extern struct kmem_cache *btrfs_delayed_data_ref_cachep;
251 extern struct kmem_cache *btrfs_delayed_extent_op_cachep;
252 
253 int __init btrfs_delayed_ref_init(void);
254 void __cold btrfs_delayed_ref_exit(void);
255 
256 static inline void btrfs_init_generic_ref(struct btrfs_ref *generic_ref,
257 				int action, u64 bytenr, u64 len, u64 parent)
258 {
259 	generic_ref->action = action;
260 	generic_ref->bytenr = bytenr;
261 	generic_ref->len = len;
262 	generic_ref->parent = parent;
263 }
264 
265 static inline void btrfs_init_tree_ref(struct btrfs_ref *generic_ref,
266 				int level, u64 root, u64 mod_root, bool skip_qgroup)
267 {
268 #ifdef CONFIG_BTRFS_FS_REF_VERIFY
269 	/* If @real_root not set, use @root as fallback */
270 	generic_ref->real_root = mod_root ?: root;
271 #endif
272 	generic_ref->tree_ref.level = level;
273 	generic_ref->tree_ref.owning_root = root;
274 	generic_ref->type = BTRFS_REF_METADATA;
275 	if (skip_qgroup || !(is_fstree(root) &&
276 			     (!mod_root || is_fstree(mod_root))))
277 		generic_ref->skip_qgroup = true;
278 	else
279 		generic_ref->skip_qgroup = false;
280 
281 }
282 
283 static inline void btrfs_init_data_ref(struct btrfs_ref *generic_ref,
284 				u64 ref_root, u64 ino, u64 offset, u64 mod_root,
285 				bool skip_qgroup)
286 {
287 #ifdef CONFIG_BTRFS_FS_REF_VERIFY
288 	/* If @real_root not set, use @root as fallback */
289 	generic_ref->real_root = mod_root ?: ref_root;
290 #endif
291 	generic_ref->data_ref.owning_root = ref_root;
292 	generic_ref->data_ref.ino = ino;
293 	generic_ref->data_ref.offset = offset;
294 	generic_ref->type = BTRFS_REF_DATA;
295 	if (skip_qgroup || !(is_fstree(ref_root) &&
296 			     (!mod_root || is_fstree(mod_root))))
297 		generic_ref->skip_qgroup = true;
298 	else
299 		generic_ref->skip_qgroup = false;
300 }
301 
302 static inline struct btrfs_delayed_extent_op *
303 btrfs_alloc_delayed_extent_op(void)
304 {
305 	return kmem_cache_alloc(btrfs_delayed_extent_op_cachep, GFP_NOFS);
306 }
307 
308 static inline void
309 btrfs_free_delayed_extent_op(struct btrfs_delayed_extent_op *op)
310 {
311 	if (op)
312 		kmem_cache_free(btrfs_delayed_extent_op_cachep, op);
313 }
314 
315 static inline void btrfs_put_delayed_ref(struct btrfs_delayed_ref_node *ref)
316 {
317 	WARN_ON(refcount_read(&ref->refs) == 0);
318 	if (refcount_dec_and_test(&ref->refs)) {
319 		WARN_ON(ref->in_tree);
320 		switch (ref->type) {
321 		case BTRFS_TREE_BLOCK_REF_KEY:
322 		case BTRFS_SHARED_BLOCK_REF_KEY:
323 			kmem_cache_free(btrfs_delayed_tree_ref_cachep, ref);
324 			break;
325 		case BTRFS_EXTENT_DATA_REF_KEY:
326 		case BTRFS_SHARED_DATA_REF_KEY:
327 			kmem_cache_free(btrfs_delayed_data_ref_cachep, ref);
328 			break;
329 		default:
330 			BUG();
331 		}
332 	}
333 }
334 
335 static inline u64 btrfs_ref_head_to_space_flags(
336 				struct btrfs_delayed_ref_head *head_ref)
337 {
338 	if (head_ref->is_data)
339 		return BTRFS_BLOCK_GROUP_DATA;
340 	else if (head_ref->is_system)
341 		return BTRFS_BLOCK_GROUP_SYSTEM;
342 	return BTRFS_BLOCK_GROUP_METADATA;
343 }
344 
345 static inline void btrfs_put_delayed_ref_head(struct btrfs_delayed_ref_head *head)
346 {
347 	if (refcount_dec_and_test(&head->refs))
348 		kmem_cache_free(btrfs_delayed_ref_head_cachep, head);
349 }
350 
351 int btrfs_add_delayed_tree_ref(struct btrfs_trans_handle *trans,
352 			       struct btrfs_ref *generic_ref,
353 			       struct btrfs_delayed_extent_op *extent_op);
354 int btrfs_add_delayed_data_ref(struct btrfs_trans_handle *trans,
355 			       struct btrfs_ref *generic_ref,
356 			       u64 reserved);
357 int btrfs_add_delayed_extent_op(struct btrfs_trans_handle *trans,
358 				u64 bytenr, u64 num_bytes,
359 				struct btrfs_delayed_extent_op *extent_op);
360 void btrfs_merge_delayed_refs(struct btrfs_trans_handle *trans,
361 			      struct btrfs_delayed_ref_root *delayed_refs,
362 			      struct btrfs_delayed_ref_head *head);
363 
364 struct btrfs_delayed_ref_head *
365 btrfs_find_delayed_ref_head(struct btrfs_delayed_ref_root *delayed_refs,
366 			    u64 bytenr);
367 int btrfs_delayed_ref_lock(struct btrfs_delayed_ref_root *delayed_refs,
368 			   struct btrfs_delayed_ref_head *head);
369 static inline void btrfs_delayed_ref_unlock(struct btrfs_delayed_ref_head *head)
370 {
371 	mutex_unlock(&head->mutex);
372 }
373 void btrfs_delete_ref_head(struct btrfs_delayed_ref_root *delayed_refs,
374 			   struct btrfs_delayed_ref_head *head);
375 
376 struct btrfs_delayed_ref_head *btrfs_select_ref_head(
377 		struct btrfs_delayed_ref_root *delayed_refs);
378 
379 int btrfs_check_delayed_seq(struct btrfs_fs_info *fs_info, u64 seq);
380 
381 void btrfs_delayed_refs_rsv_release(struct btrfs_fs_info *fs_info, int nr);
382 void btrfs_update_delayed_refs_rsv(struct btrfs_trans_handle *trans);
383 int btrfs_delayed_refs_rsv_refill(struct btrfs_fs_info *fs_info,
384 				  enum btrfs_reserve_flush_enum flush);
385 void btrfs_migrate_to_delayed_refs_rsv(struct btrfs_fs_info *fs_info,
386 				       struct btrfs_block_rsv *src,
387 				       u64 num_bytes);
388 int btrfs_should_throttle_delayed_refs(struct btrfs_trans_handle *trans);
389 bool btrfs_check_space_for_delayed_refs(struct btrfs_fs_info *fs_info);
390 
391 /*
392  * helper functions to cast a node into its container
393  */
394 static inline struct btrfs_delayed_tree_ref *
395 btrfs_delayed_node_to_tree_ref(struct btrfs_delayed_ref_node *node)
396 {
397 	return container_of(node, struct btrfs_delayed_tree_ref, node);
398 }
399 
400 static inline struct btrfs_delayed_data_ref *
401 btrfs_delayed_node_to_data_ref(struct btrfs_delayed_ref_node *node)
402 {
403 	return container_of(node, struct btrfs_delayed_data_ref, node);
404 }
405 
406 #endif
407