fs/btrfs/delayed-ref.h - linux - Git at Google

 /* 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_cached 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_cached 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_delayed_ref_root *delayed_refs,
 			   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_delayed_ref_root *delayed_refs);

 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
	/* 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_cached 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_cached 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_delayed_ref_root *delayed_refs,
	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_delayed_ref_root *delayed_refs);

	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