fs/btrfs/qgroup.h

   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  */
  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
  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
  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);
  49 int btrfs_qgroup_wait_for_completion(struct btrfs_fs_info *fs_info,
  50                                      bool interruptible);
  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,
  56                         struct btrfs_fs_info *fs_info, u64 qgroupid);
  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;
  65 int btrfs_qgroup_prepare_account_extents(struct btrfs_trans_handle *trans,
  66                                          struct btrfs_fs_info *fs_info);
  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
  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);
 101 int btrfs_qgroup_account_extents(struct btrfs_trans_handle *trans,
 102                                  struct btrfs_fs_info *fs_info);
 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);
 108 void btrfs_qgroup_free_refroot(struct btrfs_fs_info *fs_info,
 109                                u64 ref_root, u64 num_bytes);
 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);
 118         trace_btrfs_qgroup_free_delayed_ref(fs_info, ref_root, num_bytes);
 119 }
 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
 127 /* New io_tree based accurate qgroup reserve API */
 128 int btrfs_qgroup_reserve_data(struct inode *inode, u64 start, u64 len);
 129 int btrfs_qgroup_release_data(struct inode *inode, u64 start, u64 len);
 130 int btrfs_qgroup_free_data(struct inode *inode, u64 start, u64 len);
 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);
 135 void btrfs_qgroup_check_reserved_leak(struct inode *inode);
 136 #endif /* __BTRFS_QGROUP__ */