[linux-2.6-block.git] / fs / btrfs / qgroup.h

/*
 * Copyright (C) 2014 Facebook.  All rights reserved.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public
 * License v2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public
 * License along with this program; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 021110-1307, USA.
 */

#ifndef __BTRFS_QGROUP__
#define __BTRFS_QGROUP__

#include "ulist.h"
#include "delayed-ref.h"

/*
 * Record a dirty extent, and info qgroup to update quota on it
 * TODO: Use kmem cache to alloc it.
 */
struct btrfs_qgroup_extent_record {
	struct rb_node node;
	u64 bytenr;
	u64 num_bytes;
	struct ulist *old_roots;
};

/*
 * For qgroup event trace points only
 */
#define QGROUP_RESERVE		(1<<0)
#define QGROUP_RELEASE		(1<<1)
#define QGROUP_FREE		(1<<2)

int btrfs_quota_enable(struct btrfs_trans_handle *trans,
		       struct btrfs_fs_info *fs_info);
int btrfs_quota_disable(struct btrfs_trans_handle *trans,
			struct btrfs_fs_info *fs_info);
int btrfs_qgroup_rescan(struct btrfs_fs_info *fs_info);
void btrfs_qgroup_rescan_resume(struct btrfs_fs_info *fs_info);
int btrfs_qgroup_wait_for_completion(struct btrfs_fs_info *fs_info,
				     bool interruptible);
int btrfs_add_qgroup_relation(struct btrfs_trans_handle *trans,
			      struct btrfs_fs_info *fs_info, u64 src, u64 dst);
int btrfs_del_qgroup_relation(struct btrfs_trans_handle *trans,
			      struct btrfs_fs_info *fs_info, u64 src, u64 dst);
int btrfs_create_qgroup(struct btrfs_trans_handle *trans,
			struct btrfs_fs_info *fs_info, u64 qgroupid);
int btrfs_remove_qgroup(struct btrfs_trans_handle *trans,
			      struct btrfs_fs_info *fs_info, u64 qgroupid);
int btrfs_limit_qgroup(struct btrfs_trans_handle *trans,
		       struct btrfs_fs_info *fs_info, u64 qgroupid,
		       struct btrfs_qgroup_limit *limit);
int btrfs_read_qgroup_config(struct btrfs_fs_info *fs_info);
void btrfs_free_qgroup_config(struct btrfs_fs_info *fs_info);
struct btrfs_delayed_extent_op;
int btrfs_qgroup_prepare_account_extents(struct btrfs_trans_handle *trans,
					 struct btrfs_fs_info *fs_info);
/*
 * Insert one dirty extent record into @delayed_refs, informing qgroup to
 * account that extent at commit trans time.
 *
 * No lock version, caller must acquire delayed ref lock and allocate memory.
 *
 * Return 0 for success insert
 * Return >0 for existing record, caller can free @record safely.
 * Error is not possible
 */
int btrfs_qgroup_insert_dirty_extent_nolock(
		struct btrfs_fs_info *fs_info,
		struct btrfs_delayed_ref_root *delayed_refs,
		struct btrfs_qgroup_extent_record *record);

/*
 * Insert one dirty extent record into @delayed_refs, informing qgroup to
 * account that extent at commit trans time.
 *
 * Better encapsulated version.
 *
 * Return 0 if the operation is done.
 * Return <0 for error, like memory allocation failure or invalid parameter
 * (NULL trans)
 */
int btrfs_qgroup_insert_dirty_extent(struct btrfs_trans_handle *trans,
		struct btrfs_fs_info *fs_info, u64 bytenr, u64 num_bytes,
		gfp_t gfp_flag);

int
btrfs_qgroup_account_extent(struct btrfs_trans_handle *trans,
			    struct btrfs_fs_info *fs_info,
			    u64 bytenr, u64 num_bytes,
			    struct ulist *old_roots, struct ulist *new_roots);
int btrfs_qgroup_account_extents(struct btrfs_trans_handle *trans,
				 struct btrfs_fs_info *fs_info);
int btrfs_run_qgroups(struct btrfs_trans_handle *trans,
		      struct btrfs_fs_info *fs_info);
int btrfs_qgroup_inherit(struct btrfs_trans_handle *trans,
			 struct btrfs_fs_info *fs_info, u64 srcid, u64 objectid,
			 struct btrfs_qgroup_inherit *inherit);
void btrfs_qgroup_free_refroot(struct btrfs_fs_info *fs_info,
			       u64 ref_root, u64 num_bytes);
/*
 * TODO: Add proper trace point for it, as btrfs_qgroup_free() is
 * called by everywhere, can't provide good trace for delayed ref case.
 */
static inline void btrfs_qgroup_free_delayed_ref(struct btrfs_fs_info *fs_info,
						 u64 ref_root, u64 num_bytes)
{
	btrfs_qgroup_free_refroot(fs_info, ref_root, num_bytes);
	trace_btrfs_qgroup_free_delayed_ref(fs_info, ref_root, num_bytes);
}
void assert_qgroups_uptodate(struct btrfs_trans_handle *trans);

#ifdef CONFIG_BTRFS_FS_RUN_SANITY_TESTS
int btrfs_verify_qgroup_counts(struct btrfs_fs_info *fs_info, u64 qgroupid,
			       u64 rfer, u64 excl);
#endif

/* New io_tree based accurate qgroup reserve API */
int btrfs_qgroup_reserve_data(struct inode *inode, u64 start, u64 len);
int btrfs_qgroup_release_data(struct inode *inode, u64 start, u64 len);
int btrfs_qgroup_free_data(struct inode *inode, u64 start, u64 len);

int btrfs_qgroup_reserve_meta(struct btrfs_root *root, int num_bytes);
void btrfs_qgroup_free_meta_all(struct btrfs_root *root);
void btrfs_qgroup_free_meta(struct btrfs_root *root, int num_bytes);
void btrfs_qgroup_check_reserved_leak(struct inode *inode);
#endif /* __BTRFS_QGROUP__ */
Commit	Line	Data
fcebe456 JB	1	/*
	2	* Copyright (C) 2014 Facebook. All rights reserved.
	3	*
	4	* This program is free software; you can redistribute it and/or
	5	* modify it under the terms of the GNU General Public
	6	* License v2 as published by the Free Software Foundation.
	7	*
	8	* This program is distributed in the hope that it will be useful,
	9	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	10	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	11	* General Public License for more details.
	12	*
	13	* You should have received a copy of the GNU General Public
	14	* License along with this program; if not, write to the
	15	* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
	16	* Boston, MA 021110-1307, USA.
	17	*/
	18
	19	#ifndef __BTRFS_QGROUP__
	20	#define __BTRFS_QGROUP__
	21
3368d001 QW	22	#include "ulist.h"
	23	#include "delayed-ref.h"
	24
3368d001 QW	25	/*
	26	* Record a dirty extent, and info qgroup to update quota on it
	27	* TODO: Use kmem cache to alloc it.
	28	*/
	29	struct btrfs_qgroup_extent_record {
	30	struct rb_node node;
	31	u64 bytenr;
	32	u64 num_bytes;
	33	struct ulist *old_roots;
	34	};
	35
81fb6f77 QW	36	/*
	37	* For qgroup event trace points only
	38	*/
	39	#define QGROUP_RESERVE (1<<0)
	40	#define QGROUP_RELEASE (1<<1)
	41	#define QGROUP_FREE (1<<2)
	42
fcebe456 JB	43	int btrfs_quota_enable(struct btrfs_trans_handle *trans,
	44	struct btrfs_fs_info *fs_info);
	45	int btrfs_quota_disable(struct btrfs_trans_handle *trans,
	46	struct btrfs_fs_info *fs_info);
	47	int btrfs_qgroup_rescan(struct btrfs_fs_info *fs_info);
	48	void btrfs_qgroup_rescan_resume(struct btrfs_fs_info *fs_info);
d06f23d6 JM	49	int btrfs_qgroup_wait_for_completion(struct btrfs_fs_info *fs_info,
d06f23d6 JM	50	bool interruptible);
fcebe456 JB	51	int btrfs_add_qgroup_relation(struct btrfs_trans_handle *trans,
	52	struct btrfs_fs_info *fs_info, u64 src, u64 dst);
	53	int btrfs_del_qgroup_relation(struct btrfs_trans_handle *trans,
	54	struct btrfs_fs_info *fs_info, u64 src, u64 dst);
	55	int btrfs_create_qgroup(struct btrfs_trans_handle *trans,
4087cf24	56	struct btrfs_fs_info *fs_info, u64 qgroupid);
fcebe456 JB	57	int btrfs_remove_qgroup(struct btrfs_trans_handle *trans,
	58	struct btrfs_fs_info *fs_info, u64 qgroupid);
	59	int btrfs_limit_qgroup(struct btrfs_trans_handle *trans,
	60	struct btrfs_fs_info *fs_info, u64 qgroupid,
	61	struct btrfs_qgroup_limit *limit);
	62	int btrfs_read_qgroup_config(struct btrfs_fs_info *fs_info);
	63	void btrfs_free_qgroup_config(struct btrfs_fs_info *fs_info);
	64	struct btrfs_delayed_extent_op;
3b7d00f9 QW	65	int btrfs_qgroup_prepare_account_extents(struct btrfs_trans_handle *trans,
3b7d00f9 QW	66	struct btrfs_fs_info *fs_info);
cb93b52c QW	67	/*
	68	* Insert one dirty extent record into @delayed_refs, informing qgroup to
	69	* account that extent at commit trans time.
	70	*
	71	* No lock version, caller must acquire delayed ref lock and allocate memory.
	72	*
	73	* Return 0 for success insert
	74	* Return >0 for existing record, caller can free @record safely.
	75	* Error is not possible
	76	*/
	77	int btrfs_qgroup_insert_dirty_extent_nolock(
	78	struct btrfs_fs_info *fs_info,
	79	struct btrfs_delayed_ref_root *delayed_refs,
	80	struct btrfs_qgroup_extent_record *record);
	81
	82	/*
	83	* Insert one dirty extent record into @delayed_refs, informing qgroup to
	84	* account that extent at commit trans time.
	85	*
	86	* Better encapsulated version.
	87	*
	88	* Return 0 if the operation is done.
	89	* Return <0 for error, like memory allocation failure or invalid parameter
	90	* (NULL trans)
	91	*/
	92	int btrfs_qgroup_insert_dirty_extent(struct btrfs_trans_handle *trans,
	93	struct btrfs_fs_info *fs_info, u64 bytenr, u64 num_bytes,
	94	gfp_t gfp_flag);
	95
442244c9 QW	96	int
	97	btrfs_qgroup_account_extent(struct btrfs_trans_handle *trans,
	98	struct btrfs_fs_info *fs_info,
	99	u64 bytenr, u64 num_bytes,
	100	struct ulist old_roots, struct ulist new_roots);
550d7a2e QW	101	int btrfs_qgroup_account_extents(struct btrfs_trans_handle *trans,
550d7a2e QW	102	struct btrfs_fs_info *fs_info);
fcebe456 JB	103	int btrfs_run_qgroups(struct btrfs_trans_handle *trans,
	104	struct btrfs_fs_info *fs_info);
	105	int btrfs_qgroup_inherit(struct btrfs_trans_handle *trans,
	106	struct btrfs_fs_info *fs_info, u64 srcid, u64 objectid,
	107	struct btrfs_qgroup_inherit *inherit);
297d750b QW	108	void btrfs_qgroup_free_refroot(struct btrfs_fs_info *fs_info,
297d750b QW	109	u64 ref_root, u64 num_bytes);
297d750b QW	110	/*
	111	* TODO: Add proper trace point for it, as btrfs_qgroup_free() is
	112	* called by everywhere, can't provide good trace for delayed ref case.
	113	*/
	114	static inline void btrfs_qgroup_free_delayed_ref(struct btrfs_fs_info *fs_info,
	115	u64 ref_root, u64 num_bytes)
	116	{
	117	btrfs_qgroup_free_refroot(fs_info, ref_root, num_bytes);
bc074524	118	trace_btrfs_qgroup_free_delayed_ref(fs_info, ref_root, num_bytes);
297d750b	119	}
fcebe456 JB	120	void assert_qgroups_uptodate(struct btrfs_trans_handle *trans);
	121
	122	#ifdef CONFIG_BTRFS_FS_RUN_SANITY_TESTS
	123	int btrfs_verify_qgroup_counts(struct btrfs_fs_info *fs_info, u64 qgroupid,
	124	u64 rfer, u64 excl);
	125	#endif
	126
52472553 QW	127	/* New io_tree based accurate qgroup reserve API */
52472553 QW	128	int btrfs_qgroup_reserve_data(struct inode *inode, u64 start, u64 len);
f695fdce QW	129	int btrfs_qgroup_release_data(struct inode *inode, u64 start, u64 len);
f695fdce QW	130	int btrfs_qgroup_free_data(struct inode *inode, u64 start, u64 len);
55eeaf05 QW	131
	132	int btrfs_qgroup_reserve_meta(struct btrfs_root *root, int num_bytes);
	133	void btrfs_qgroup_free_meta_all(struct btrfs_root *root);
	134	void btrfs_qgroup_free_meta(struct btrfs_root *root, int num_bytes);
56fa9d07	135	void btrfs_qgroup_check_reserved_leak(struct inode *inode);
fcebe456	136	#endif /* __BTRFS_QGROUP__ */