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
 22#include "ulist.h"
 23#include "delayed-ref.h"
 24
 25/*
 26 * Record a dirty extent, and info qgroup to update quota on it
 27 * TODO: Use kmem cache to alloc it.
 28 */
 29struct btrfs_qgroup_extent_record {
 30	struct rb_node node;
 31	u64 bytenr;
 32	u64 num_bytes;
 33	struct ulist *old_roots;
 34};
 35
 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
 43int btrfs_quota_enable(struct btrfs_trans_handle *trans,
 44		       struct btrfs_fs_info *fs_info);
 45int btrfs_quota_disable(struct btrfs_trans_handle *trans,
 46			struct btrfs_fs_info *fs_info);
 47int btrfs_qgroup_rescan(struct btrfs_fs_info *fs_info);
 48void btrfs_qgroup_rescan_resume(struct btrfs_fs_info *fs_info);
 49int btrfs_qgroup_wait_for_completion(struct btrfs_fs_info *fs_info);
 50int btrfs_add_qgroup_relation(struct btrfs_trans_handle *trans,
 51			      struct btrfs_fs_info *fs_info, u64 src, u64 dst);
 52int btrfs_del_qgroup_relation(struct btrfs_trans_handle *trans,
 53			      struct btrfs_fs_info *fs_info, u64 src, u64 dst);
 54int btrfs_create_qgroup(struct btrfs_trans_handle *trans,
 55			struct btrfs_fs_info *fs_info, u64 qgroupid);
 56int btrfs_remove_qgroup(struct btrfs_trans_handle *trans,
 57			      struct btrfs_fs_info *fs_info, u64 qgroupid);
 58int btrfs_limit_qgroup(struct btrfs_trans_handle *trans,
 59		       struct btrfs_fs_info *fs_info, u64 qgroupid,
 60		       struct btrfs_qgroup_limit *limit);
 61int btrfs_read_qgroup_config(struct btrfs_fs_info *fs_info);
 62void btrfs_free_qgroup_config(struct btrfs_fs_info *fs_info);
 63struct btrfs_delayed_extent_op;
 64int btrfs_qgroup_prepare_account_extents(struct btrfs_trans_handle *trans,
 65					 struct btrfs_fs_info *fs_info);
 66struct btrfs_qgroup_extent_record
 67*btrfs_qgroup_insert_dirty_extent(struct btrfs_delayed_ref_root *delayed_refs,
 68				  struct btrfs_qgroup_extent_record *record);
 69int
 70btrfs_qgroup_account_extent(struct btrfs_trans_handle *trans,
 71			    struct btrfs_fs_info *fs_info,
 72			    u64 bytenr, u64 num_bytes,
 73			    struct ulist *old_roots, struct ulist *new_roots);
 74int btrfs_qgroup_account_extents(struct btrfs_trans_handle *trans,
 75				 struct btrfs_fs_info *fs_info);
 76int btrfs_run_qgroups(struct btrfs_trans_handle *trans,
 77		      struct btrfs_fs_info *fs_info);
 78int btrfs_qgroup_inherit(struct btrfs_trans_handle *trans,
 79			 struct btrfs_fs_info *fs_info, u64 srcid, u64 objectid,
 80			 struct btrfs_qgroup_inherit *inherit);
 81void btrfs_qgroup_free_refroot(struct btrfs_fs_info *fs_info,
 82			       u64 ref_root, u64 num_bytes);
 83/*
 84 * TODO: Add proper trace point for it, as btrfs_qgroup_free() is
 85 * called by everywhere, can't provide good trace for delayed ref case.
 86 */
 87static inline void btrfs_qgroup_free_delayed_ref(struct btrfs_fs_info *fs_info,
 88						 u64 ref_root, u64 num_bytes)
 89{
 90	btrfs_qgroup_free_refroot(fs_info, ref_root, num_bytes);
 91	trace_btrfs_qgroup_free_delayed_ref(ref_root, num_bytes);
 92}
 93void assert_qgroups_uptodate(struct btrfs_trans_handle *trans);
 94
 95#ifdef CONFIG_BTRFS_FS_RUN_SANITY_TESTS
 96int btrfs_verify_qgroup_counts(struct btrfs_fs_info *fs_info, u64 qgroupid,
 97			       u64 rfer, u64 excl);
 98#endif
 99
100/* New io_tree based accurate qgroup reserve API */
101int btrfs_qgroup_reserve_data(struct inode *inode, u64 start, u64 len);
102int btrfs_qgroup_release_data(struct inode *inode, u64 start, u64 len);
103int btrfs_qgroup_free_data(struct inode *inode, u64 start, u64 len);
104
105int btrfs_qgroup_reserve_meta(struct btrfs_root *root, int num_bytes);
106void btrfs_qgroup_free_meta_all(struct btrfs_root *root);
107void btrfs_qgroup_free_meta(struct btrfs_root *root, int num_bytes);
108void btrfs_qgroup_check_reserved_leak(struct inode *inode);
109#endif /* __BTRFS_QGROUP__ */