| /* SPDX-License-Identifier: GPL-2.0 */ | 
 | /* | 
 |  * Copyright (C) 2008 Oracle.  All rights reserved. | 
 |  */ | 
 |  | 
 | #ifndef BTRFS_DELAYED_REF_H | 
 | #define BTRFS_DELAYED_REF_H | 
 |  | 
 | #include <linux/refcount.h> | 
 |  | 
 | /* these are the possible values of struct btrfs_delayed_ref_node->action */ | 
 | #define BTRFS_ADD_DELAYED_REF    1 /* add one backref to the tree */ | 
 | #define BTRFS_DROP_DELAYED_REF   2 /* delete one backref from the tree */ | 
 | #define BTRFS_ADD_DELAYED_EXTENT 3 /* record a full extent allocation */ | 
 | #define BTRFS_UPDATE_DELAYED_HEAD 4 /* not changing ref count on head ref */ | 
 |  | 
 | struct btrfs_delayed_ref_node { | 
 | 	struct rb_node ref_node; | 
 | 	/* | 
 | 	 * If action is BTRFS_ADD_DELAYED_REF, also link this node to | 
 | 	 * ref_head->ref_add_list, then we do not need to iterate the | 
 | 	 * whole ref_head->ref_list to find BTRFS_ADD_DELAYED_REF nodes. | 
 | 	 */ | 
 | 	struct list_head add_list; | 
 |  | 
 | 	/* the starting bytenr of the extent */ | 
 | 	u64 bytenr; | 
 |  | 
 | 	/* the size of the extent */ | 
 | 	u64 num_bytes; | 
 |  | 
 | 	/* seq number to keep track of insertion order */ | 
 | 	u64 seq; | 
 |  | 
 | 	/* ref count on this data structure */ | 
 | 	refcount_t refs; | 
 |  | 
 | 	/* | 
 | 	 * how many refs is this entry adding or deleting.  For | 
 | 	 * head refs, this may be a negative number because it is keeping | 
 | 	 * track of the total mods done to the reference count. | 
 | 	 * For individual refs, this will always be a positive number | 
 | 	 * | 
 | 	 * It may be more than one, since it is possible for a single | 
 | 	 * parent to have more than one ref on an extent | 
 | 	 */ | 
 | 	int ref_mod; | 
 |  | 
 | 	unsigned int action:8; | 
 | 	unsigned int type:8; | 
 | 	/* is this node still in the rbtree? */ | 
 | 	unsigned int is_head:1; | 
 | 	unsigned int in_tree:1; | 
 | }; | 
 |  | 
 | struct btrfs_delayed_extent_op { | 
 | 	struct btrfs_disk_key key; | 
 | 	u8 level; | 
 | 	bool update_key; | 
 | 	bool update_flags; | 
 | 	bool is_data; | 
 | 	u64 flags_to_set; | 
 | }; | 
 |  | 
 | /* | 
 |  * the head refs are used to hold a lock on a given extent, which allows us | 
 |  * to make sure that only one process is running the delayed refs | 
 |  * at a time for a single extent.  They also store the sum of all the | 
 |  * reference count modifications we've queued up. | 
 |  */ | 
 | struct btrfs_delayed_ref_head { | 
 | 	u64 bytenr; | 
 | 	u64 num_bytes; | 
 | 	refcount_t refs; | 
 | 	/* | 
 | 	 * the mutex is held while running the refs, and it is also | 
 | 	 * held when checking the sum of reference modifications. | 
 | 	 */ | 
 | 	struct mutex mutex; | 
 |  | 
 | 	spinlock_t lock; | 
 | 	struct rb_root ref_tree; | 
 | 	/* accumulate add BTRFS_ADD_DELAYED_REF nodes to this ref_add_list. */ | 
 | 	struct list_head ref_add_list; | 
 |  | 
 | 	struct rb_node href_node; | 
 |  | 
 | 	struct btrfs_delayed_extent_op *extent_op; | 
 |  | 
 | 	/* | 
 | 	 * This is used to track the final ref_mod from all the refs associated | 
 | 	 * with this head ref, this is not adjusted as delayed refs are run, | 
 | 	 * this is meant to track if we need to do the csum accounting or not. | 
 | 	 */ | 
 | 	int total_ref_mod; | 
 |  | 
 | 	/* | 
 | 	 * This is the current outstanding mod references for this bytenr.  This | 
 | 	 * is used with lookup_extent_info to get an accurate reference count | 
 | 	 * for a bytenr, so it is adjusted as delayed refs are run so that any | 
 | 	 * on disk reference count + ref_mod is accurate. | 
 | 	 */ | 
 | 	int ref_mod; | 
 |  | 
 | 	/* | 
 | 	 * For qgroup reserved space freeing. | 
 | 	 * | 
 | 	 * ref_root and reserved will be recorded after | 
 | 	 * BTRFS_ADD_DELAYED_EXTENT is called. | 
 | 	 * And will be used to free reserved qgroup space at | 
 | 	 * run_delayed_refs() time. | 
 | 	 */ | 
 | 	u64 qgroup_ref_root; | 
 | 	u64 qgroup_reserved; | 
 |  | 
 | 	/* | 
 | 	 * when a new extent is allocated, it is just reserved in memory | 
 | 	 * The actual extent isn't inserted into the extent allocation tree | 
 | 	 * until the delayed ref is processed.  must_insert_reserved is | 
 | 	 * used to flag a delayed ref so the accounting can be updated | 
 | 	 * when a full insert is done. | 
 | 	 * | 
 | 	 * It is possible the extent will be freed before it is ever | 
 | 	 * inserted into the extent allocation tree.  In this case | 
 | 	 * we need to update the in ram accounting to properly reflect | 
 | 	 * the free has happened. | 
 | 	 */ | 
 | 	unsigned int must_insert_reserved:1; | 
 | 	unsigned int is_data:1; | 
 | 	unsigned int is_system:1; | 
 | 	unsigned int processing:1; | 
 | }; | 
 |  | 
 | struct btrfs_delayed_tree_ref { | 
 | 	struct btrfs_delayed_ref_node node; | 
 | 	u64 root; | 
 | 	u64 parent; | 
 | 	int level; | 
 | }; | 
 |  | 
 | struct btrfs_delayed_data_ref { | 
 | 	struct btrfs_delayed_ref_node node; | 
 | 	u64 root; | 
 | 	u64 parent; | 
 | 	u64 objectid; | 
 | 	u64 offset; | 
 | }; | 
 |  | 
 | struct btrfs_delayed_ref_root { | 
 | 	/* head ref rbtree */ | 
 | 	struct rb_root href_root; | 
 |  | 
 | 	/* dirty extent records */ | 
 | 	struct rb_root dirty_extent_root; | 
 |  | 
 | 	/* this spin lock protects the rbtree and the entries inside */ | 
 | 	spinlock_t lock; | 
 |  | 
 | 	/* how many delayed ref updates we've queued, used by the | 
 | 	 * throttling code | 
 | 	 */ | 
 | 	atomic_t num_entries; | 
 |  | 
 | 	/* total number of head nodes in tree */ | 
 | 	unsigned long num_heads; | 
 |  | 
 | 	/* total number of head nodes ready for processing */ | 
 | 	unsigned long num_heads_ready; | 
 |  | 
 | 	u64 pending_csums; | 
 |  | 
 | 	/* | 
 | 	 * set when the tree is flushing before a transaction commit, | 
 | 	 * used by the throttling code to decide if new updates need | 
 | 	 * to be run right away | 
 | 	 */ | 
 | 	int flushing; | 
 |  | 
 | 	u64 run_delayed_start; | 
 |  | 
 | 	/* | 
 | 	 * To make qgroup to skip given root. | 
 | 	 * This is for snapshot, as btrfs_qgroup_inherit() will manually | 
 | 	 * modify counters for snapshot and its source, so we should skip | 
 | 	 * the snapshot in new_root/old_roots or it will get calculated twice | 
 | 	 */ | 
 | 	u64 qgroup_to_skip; | 
 | }; | 
 |  | 
 | extern struct kmem_cache *btrfs_delayed_ref_head_cachep; | 
 | extern struct kmem_cache *btrfs_delayed_tree_ref_cachep; | 
 | extern struct kmem_cache *btrfs_delayed_data_ref_cachep; | 
 | extern struct kmem_cache *btrfs_delayed_extent_op_cachep; | 
 |  | 
 | int __init btrfs_delayed_ref_init(void); | 
 | void __cold btrfs_delayed_ref_exit(void); | 
 |  | 
 | static inline struct btrfs_delayed_extent_op * | 
 | btrfs_alloc_delayed_extent_op(void) | 
 | { | 
 | 	return kmem_cache_alloc(btrfs_delayed_extent_op_cachep, GFP_NOFS); | 
 | } | 
 |  | 
 | static inline void | 
 | btrfs_free_delayed_extent_op(struct btrfs_delayed_extent_op *op) | 
 | { | 
 | 	if (op) | 
 | 		kmem_cache_free(btrfs_delayed_extent_op_cachep, op); | 
 | } | 
 |  | 
 | static inline void btrfs_put_delayed_ref(struct btrfs_delayed_ref_node *ref) | 
 | { | 
 | 	WARN_ON(refcount_read(&ref->refs) == 0); | 
 | 	if (refcount_dec_and_test(&ref->refs)) { | 
 | 		WARN_ON(ref->in_tree); | 
 | 		switch (ref->type) { | 
 | 		case BTRFS_TREE_BLOCK_REF_KEY: | 
 | 		case BTRFS_SHARED_BLOCK_REF_KEY: | 
 | 			kmem_cache_free(btrfs_delayed_tree_ref_cachep, ref); | 
 | 			break; | 
 | 		case BTRFS_EXTENT_DATA_REF_KEY: | 
 | 		case BTRFS_SHARED_DATA_REF_KEY: | 
 | 			kmem_cache_free(btrfs_delayed_data_ref_cachep, ref); | 
 | 			break; | 
 | 		default: | 
 | 			BUG(); | 
 | 		} | 
 | 	} | 
 | } | 
 |  | 
 | static inline void btrfs_put_delayed_ref_head(struct btrfs_delayed_ref_head *head) | 
 | { | 
 | 	if (refcount_dec_and_test(&head->refs)) | 
 | 		kmem_cache_free(btrfs_delayed_ref_head_cachep, head); | 
 | } | 
 |  | 
 | int btrfs_add_delayed_tree_ref(struct btrfs_trans_handle *trans, | 
 | 			       u64 bytenr, u64 num_bytes, u64 parent, | 
 | 			       u64 ref_root, int level, int action, | 
 | 			       struct btrfs_delayed_extent_op *extent_op, | 
 | 			       int *old_ref_mod, int *new_ref_mod); | 
 | int btrfs_add_delayed_data_ref(struct btrfs_trans_handle *trans, | 
 | 			       u64 bytenr, u64 num_bytes, | 
 | 			       u64 parent, u64 ref_root, | 
 | 			       u64 owner, u64 offset, u64 reserved, int action, | 
 | 			       int *old_ref_mod, int *new_ref_mod); | 
 | int btrfs_add_delayed_extent_op(struct btrfs_fs_info *fs_info, | 
 | 				struct btrfs_trans_handle *trans, | 
 | 				u64 bytenr, u64 num_bytes, | 
 | 				struct btrfs_delayed_extent_op *extent_op); | 
 | void btrfs_merge_delayed_refs(struct btrfs_trans_handle *trans, | 
 | 			      struct btrfs_delayed_ref_root *delayed_refs, | 
 | 			      struct btrfs_delayed_ref_head *head); | 
 |  | 
 | struct btrfs_delayed_ref_head * | 
 | btrfs_find_delayed_ref_head(struct btrfs_delayed_ref_root *delayed_refs, | 
 | 			    u64 bytenr); | 
 | int btrfs_delayed_ref_lock(struct btrfs_trans_handle *trans, | 
 | 			   struct btrfs_delayed_ref_head *head); | 
 | static inline void btrfs_delayed_ref_unlock(struct btrfs_delayed_ref_head *head) | 
 | { | 
 | 	mutex_unlock(&head->mutex); | 
 | } | 
 |  | 
 |  | 
 | struct btrfs_delayed_ref_head * | 
 | btrfs_select_ref_head(struct btrfs_trans_handle *trans); | 
 |  | 
 | int btrfs_check_delayed_seq(struct btrfs_fs_info *fs_info, u64 seq); | 
 |  | 
 | /* | 
 |  * helper functions to cast a node into its container | 
 |  */ | 
 | static inline struct btrfs_delayed_tree_ref * | 
 | btrfs_delayed_node_to_tree_ref(struct btrfs_delayed_ref_node *node) | 
 | { | 
 | 	return container_of(node, struct btrfs_delayed_tree_ref, node); | 
 | } | 
 |  | 
 | static inline struct btrfs_delayed_data_ref * | 
 | btrfs_delayed_node_to_data_ref(struct btrfs_delayed_ref_node *node) | 
 | { | 
 | 	return container_of(node, struct btrfs_delayed_data_ref, node); | 
 | } | 
 |  | 
 | #endif |