12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571 |
- /*
- * Copyright (C) 2011 STRATO. 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.
- */
- #include <linux/sched.h>
- #include <linux/pagemap.h>
- #include <linux/writeback.h>
- #include <linux/blkdev.h>
- #include <linux/rbtree.h>
- #include <linux/slab.h>
- #include <linux/workqueue.h>
- #include "ctree.h"
- #include "transaction.h"
- #include "disk-io.h"
- #include "locking.h"
- #include "ulist.h"
- #include "ioctl.h"
- #include "backref.h"
- /* TODO XXX FIXME
- * - subvol delete -> delete when ref goes to 0? delete limits also?
- * - reorganize keys
- * - compressed
- * - sync
- * - rescan
- * - copy also limits on subvol creation
- * - limit
- * - caches fuer ulists
- * - performance benchmarks
- * - check all ioctl parameters
- */
- /*
- * one struct for each qgroup, organized in fs_info->qgroup_tree.
- */
- struct btrfs_qgroup {
- u64 qgroupid;
- /*
- * state
- */
- u64 rfer; /* referenced */
- u64 rfer_cmpr; /* referenced compressed */
- u64 excl; /* exclusive */
- u64 excl_cmpr; /* exclusive compressed */
- /*
- * limits
- */
- u64 lim_flags; /* which limits are set */
- u64 max_rfer;
- u64 max_excl;
- u64 rsv_rfer;
- u64 rsv_excl;
- /*
- * reservation tracking
- */
- u64 reserved;
- /*
- * lists
- */
- struct list_head groups; /* groups this group is member of */
- struct list_head members; /* groups that are members of this group */
- struct list_head dirty; /* dirty groups */
- struct rb_node node; /* tree of qgroups */
- /*
- * temp variables for accounting operations
- */
- u64 tag;
- u64 refcnt;
- };
- /*
- * glue structure to represent the relations between qgroups.
- */
- struct btrfs_qgroup_list {
- struct list_head next_group;
- struct list_head next_member;
- struct btrfs_qgroup *group;
- struct btrfs_qgroup *member;
- };
- /* must be called with qgroup_lock held */
- static struct btrfs_qgroup *find_qgroup_rb(struct btrfs_fs_info *fs_info,
- u64 qgroupid)
- {
- struct rb_node *n = fs_info->qgroup_tree.rb_node;
- struct btrfs_qgroup *qgroup;
- while (n) {
- qgroup = rb_entry(n, struct btrfs_qgroup, node);
- if (qgroup->qgroupid < qgroupid)
- n = n->rb_left;
- else if (qgroup->qgroupid > qgroupid)
- n = n->rb_right;
- else
- return qgroup;
- }
- return NULL;
- }
- /* must be called with qgroup_lock held */
- static struct btrfs_qgroup *add_qgroup_rb(struct btrfs_fs_info *fs_info,
- u64 qgroupid)
- {
- struct rb_node **p = &fs_info->qgroup_tree.rb_node;
- struct rb_node *parent = NULL;
- struct btrfs_qgroup *qgroup;
- while (*p) {
- parent = *p;
- qgroup = rb_entry(parent, struct btrfs_qgroup, node);
- if (qgroup->qgroupid < qgroupid)
- p = &(*p)->rb_left;
- else if (qgroup->qgroupid > qgroupid)
- p = &(*p)->rb_right;
- else
- return qgroup;
- }
- qgroup = kzalloc(sizeof(*qgroup), GFP_ATOMIC);
- if (!qgroup)
- return ERR_PTR(-ENOMEM);
- qgroup->qgroupid = qgroupid;
- INIT_LIST_HEAD(&qgroup->groups);
- INIT_LIST_HEAD(&qgroup->members);
- INIT_LIST_HEAD(&qgroup->dirty);
- rb_link_node(&qgroup->node, parent, p);
- rb_insert_color(&qgroup->node, &fs_info->qgroup_tree);
- return qgroup;
- }
- /* must be called with qgroup_lock held */
- static int del_qgroup_rb(struct btrfs_fs_info *fs_info, u64 qgroupid)
- {
- struct btrfs_qgroup *qgroup = find_qgroup_rb(fs_info, qgroupid);
- struct btrfs_qgroup_list *list;
- if (!qgroup)
- return -ENOENT;
- rb_erase(&qgroup->node, &fs_info->qgroup_tree);
- list_del(&qgroup->dirty);
- while (!list_empty(&qgroup->groups)) {
- list = list_first_entry(&qgroup->groups,
- struct btrfs_qgroup_list, next_group);
- list_del(&list->next_group);
- list_del(&list->next_member);
- kfree(list);
- }
- while (!list_empty(&qgroup->members)) {
- list = list_first_entry(&qgroup->members,
- struct btrfs_qgroup_list, next_member);
- list_del(&list->next_group);
- list_del(&list->next_member);
- kfree(list);
- }
- kfree(qgroup);
- return 0;
- }
- /* must be called with qgroup_lock held */
- static int add_relation_rb(struct btrfs_fs_info *fs_info,
- u64 memberid, u64 parentid)
- {
- struct btrfs_qgroup *member;
- struct btrfs_qgroup *parent;
- struct btrfs_qgroup_list *list;
- member = find_qgroup_rb(fs_info, memberid);
- parent = find_qgroup_rb(fs_info, parentid);
- if (!member || !parent)
- return -ENOENT;
- list = kzalloc(sizeof(*list), GFP_ATOMIC);
- if (!list)
- return -ENOMEM;
- list->group = parent;
- list->member = member;
- list_add_tail(&list->next_group, &member->groups);
- list_add_tail(&list->next_member, &parent->members);
- return 0;
- }
- /* must be called with qgroup_lock held */
- static int del_relation_rb(struct btrfs_fs_info *fs_info,
- u64 memberid, u64 parentid)
- {
- struct btrfs_qgroup *member;
- struct btrfs_qgroup *parent;
- struct btrfs_qgroup_list *list;
- member = find_qgroup_rb(fs_info, memberid);
- parent = find_qgroup_rb(fs_info, parentid);
- if (!member || !parent)
- return -ENOENT;
- list_for_each_entry(list, &member->groups, next_group) {
- if (list->group == parent) {
- list_del(&list->next_group);
- list_del(&list->next_member);
- kfree(list);
- return 0;
- }
- }
- return -ENOENT;
- }
- /*
- * The full config is read in one go, only called from open_ctree()
- * It doesn't use any locking, as at this point we're still single-threaded
- */
- int btrfs_read_qgroup_config(struct btrfs_fs_info *fs_info)
- {
- struct btrfs_key key;
- struct btrfs_key found_key;
- struct btrfs_root *quota_root = fs_info->quota_root;
- struct btrfs_path *path = NULL;
- struct extent_buffer *l;
- int slot;
- int ret = 0;
- u64 flags = 0;
- if (!fs_info->quota_enabled)
- return 0;
- path = btrfs_alloc_path();
- if (!path) {
- ret = -ENOMEM;
- goto out;
- }
- /* default this to quota off, in case no status key is found */
- fs_info->qgroup_flags = 0;
- /*
- * pass 1: read status, all qgroup infos and limits
- */
- key.objectid = 0;
- key.type = 0;
- key.offset = 0;
- ret = btrfs_search_slot_for_read(quota_root, &key, path, 1, 1);
- if (ret)
- goto out;
- while (1) {
- struct btrfs_qgroup *qgroup;
- slot = path->slots[0];
- l = path->nodes[0];
- btrfs_item_key_to_cpu(l, &found_key, slot);
- if (found_key.type == BTRFS_QGROUP_STATUS_KEY) {
- struct btrfs_qgroup_status_item *ptr;
- ptr = btrfs_item_ptr(l, slot,
- struct btrfs_qgroup_status_item);
- if (btrfs_qgroup_status_version(l, ptr) !=
- BTRFS_QGROUP_STATUS_VERSION) {
- printk(KERN_ERR
- "btrfs: old qgroup version, quota disabled\n");
- goto out;
- }
- if (btrfs_qgroup_status_generation(l, ptr) !=
- fs_info->generation) {
- flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
- printk(KERN_ERR
- "btrfs: qgroup generation mismatch, "
- "marked as inconsistent\n");
- }
- fs_info->qgroup_flags = btrfs_qgroup_status_flags(l,
- ptr);
- /* FIXME read scan element */
- goto next1;
- }
- if (found_key.type != BTRFS_QGROUP_INFO_KEY &&
- found_key.type != BTRFS_QGROUP_LIMIT_KEY)
- goto next1;
- qgroup = find_qgroup_rb(fs_info, found_key.offset);
- if ((qgroup && found_key.type == BTRFS_QGROUP_INFO_KEY) ||
- (!qgroup && found_key.type == BTRFS_QGROUP_LIMIT_KEY)) {
- printk(KERN_ERR "btrfs: inconsitent qgroup config\n");
- flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
- }
- if (!qgroup) {
- qgroup = add_qgroup_rb(fs_info, found_key.offset);
- if (IS_ERR(qgroup)) {
- ret = PTR_ERR(qgroup);
- goto out;
- }
- }
- switch (found_key.type) {
- case BTRFS_QGROUP_INFO_KEY: {
- struct btrfs_qgroup_info_item *ptr;
- ptr = btrfs_item_ptr(l, slot,
- struct btrfs_qgroup_info_item);
- qgroup->rfer = btrfs_qgroup_info_rfer(l, ptr);
- qgroup->rfer_cmpr = btrfs_qgroup_info_rfer_cmpr(l, ptr);
- qgroup->excl = btrfs_qgroup_info_excl(l, ptr);
- qgroup->excl_cmpr = btrfs_qgroup_info_excl_cmpr(l, ptr);
- /* generation currently unused */
- break;
- }
- case BTRFS_QGROUP_LIMIT_KEY: {
- struct btrfs_qgroup_limit_item *ptr;
- ptr = btrfs_item_ptr(l, slot,
- struct btrfs_qgroup_limit_item);
- qgroup->lim_flags = btrfs_qgroup_limit_flags(l, ptr);
- qgroup->max_rfer = btrfs_qgroup_limit_max_rfer(l, ptr);
- qgroup->max_excl = btrfs_qgroup_limit_max_excl(l, ptr);
- qgroup->rsv_rfer = btrfs_qgroup_limit_rsv_rfer(l, ptr);
- qgroup->rsv_excl = btrfs_qgroup_limit_rsv_excl(l, ptr);
- break;
- }
- }
- next1:
- ret = btrfs_next_item(quota_root, path);
- if (ret < 0)
- goto out;
- if (ret)
- break;
- }
- btrfs_release_path(path);
- /*
- * pass 2: read all qgroup relations
- */
- key.objectid = 0;
- key.type = BTRFS_QGROUP_RELATION_KEY;
- key.offset = 0;
- ret = btrfs_search_slot_for_read(quota_root, &key, path, 1, 0);
- if (ret)
- goto out;
- while (1) {
- slot = path->slots[0];
- l = path->nodes[0];
- btrfs_item_key_to_cpu(l, &found_key, slot);
- if (found_key.type != BTRFS_QGROUP_RELATION_KEY)
- goto next2;
- if (found_key.objectid > found_key.offset) {
- /* parent <- member, not needed to build config */
- /* FIXME should we omit the key completely? */
- goto next2;
- }
- ret = add_relation_rb(fs_info, found_key.objectid,
- found_key.offset);
- if (ret)
- goto out;
- next2:
- ret = btrfs_next_item(quota_root, path);
- if (ret < 0)
- goto out;
- if (ret)
- break;
- }
- out:
- fs_info->qgroup_flags |= flags;
- if (!(fs_info->qgroup_flags & BTRFS_QGROUP_STATUS_FLAG_ON)) {
- fs_info->quota_enabled = 0;
- fs_info->pending_quota_state = 0;
- }
- btrfs_free_path(path);
- return ret < 0 ? ret : 0;
- }
- /*
- * This is only called from close_ctree() or open_ctree(), both in single-
- * treaded paths. Clean up the in-memory structures. No locking needed.
- */
- void btrfs_free_qgroup_config(struct btrfs_fs_info *fs_info)
- {
- struct rb_node *n;
- struct btrfs_qgroup *qgroup;
- struct btrfs_qgroup_list *list;
- while ((n = rb_first(&fs_info->qgroup_tree))) {
- qgroup = rb_entry(n, struct btrfs_qgroup, node);
- rb_erase(n, &fs_info->qgroup_tree);
- WARN_ON(!list_empty(&qgroup->dirty));
- while (!list_empty(&qgroup->groups)) {
- list = list_first_entry(&qgroup->groups,
- struct btrfs_qgroup_list,
- next_group);
- list_del(&list->next_group);
- list_del(&list->next_member);
- kfree(list);
- }
- while (!list_empty(&qgroup->members)) {
- list = list_first_entry(&qgroup->members,
- struct btrfs_qgroup_list,
- next_member);
- list_del(&list->next_group);
- list_del(&list->next_member);
- kfree(list);
- }
- kfree(qgroup);
- }
- }
- static int add_qgroup_relation_item(struct btrfs_trans_handle *trans,
- struct btrfs_root *quota_root,
- u64 src, u64 dst)
- {
- int ret;
- struct btrfs_path *path;
- struct btrfs_key key;
- path = btrfs_alloc_path();
- if (!path)
- return -ENOMEM;
- key.objectid = src;
- key.type = BTRFS_QGROUP_RELATION_KEY;
- key.offset = dst;
- ret = btrfs_insert_empty_item(trans, quota_root, path, &key, 0);
- btrfs_mark_buffer_dirty(path->nodes[0]);
- btrfs_free_path(path);
- return ret;
- }
- static int del_qgroup_relation_item(struct btrfs_trans_handle *trans,
- struct btrfs_root *quota_root,
- u64 src, u64 dst)
- {
- int ret;
- struct btrfs_path *path;
- struct btrfs_key key;
- path = btrfs_alloc_path();
- if (!path)
- return -ENOMEM;
- key.objectid = src;
- key.type = BTRFS_QGROUP_RELATION_KEY;
- key.offset = dst;
- ret = btrfs_search_slot(trans, quota_root, &key, path, -1, 1);
- if (ret < 0)
- goto out;
- if (ret > 0) {
- ret = -ENOENT;
- goto out;
- }
- ret = btrfs_del_item(trans, quota_root, path);
- out:
- btrfs_free_path(path);
- return ret;
- }
- static int add_qgroup_item(struct btrfs_trans_handle *trans,
- struct btrfs_root *quota_root, u64 qgroupid)
- {
- int ret;
- struct btrfs_path *path;
- struct btrfs_qgroup_info_item *qgroup_info;
- struct btrfs_qgroup_limit_item *qgroup_limit;
- struct extent_buffer *leaf;
- struct btrfs_key key;
- path = btrfs_alloc_path();
- if (!path)
- return -ENOMEM;
- key.objectid = 0;
- key.type = BTRFS_QGROUP_INFO_KEY;
- key.offset = qgroupid;
- ret = btrfs_insert_empty_item(trans, quota_root, path, &key,
- sizeof(*qgroup_info));
- if (ret)
- goto out;
- leaf = path->nodes[0];
- qgroup_info = btrfs_item_ptr(leaf, path->slots[0],
- struct btrfs_qgroup_info_item);
- btrfs_set_qgroup_info_generation(leaf, qgroup_info, trans->transid);
- btrfs_set_qgroup_info_rfer(leaf, qgroup_info, 0);
- btrfs_set_qgroup_info_rfer_cmpr(leaf, qgroup_info, 0);
- btrfs_set_qgroup_info_excl(leaf, qgroup_info, 0);
- btrfs_set_qgroup_info_excl_cmpr(leaf, qgroup_info, 0);
- btrfs_mark_buffer_dirty(leaf);
- btrfs_release_path(path);
- key.type = BTRFS_QGROUP_LIMIT_KEY;
- ret = btrfs_insert_empty_item(trans, quota_root, path, &key,
- sizeof(*qgroup_limit));
- if (ret)
- goto out;
- leaf = path->nodes[0];
- qgroup_limit = btrfs_item_ptr(leaf, path->slots[0],
- struct btrfs_qgroup_limit_item);
- btrfs_set_qgroup_limit_flags(leaf, qgroup_limit, 0);
- btrfs_set_qgroup_limit_max_rfer(leaf, qgroup_limit, 0);
- btrfs_set_qgroup_limit_max_excl(leaf, qgroup_limit, 0);
- btrfs_set_qgroup_limit_rsv_rfer(leaf, qgroup_limit, 0);
- btrfs_set_qgroup_limit_rsv_excl(leaf, qgroup_limit, 0);
- btrfs_mark_buffer_dirty(leaf);
- ret = 0;
- out:
- btrfs_free_path(path);
- return ret;
- }
- static int del_qgroup_item(struct btrfs_trans_handle *trans,
- struct btrfs_root *quota_root, u64 qgroupid)
- {
- int ret;
- struct btrfs_path *path;
- struct btrfs_key key;
- path = btrfs_alloc_path();
- if (!path)
- return -ENOMEM;
- key.objectid = 0;
- key.type = BTRFS_QGROUP_INFO_KEY;
- key.offset = qgroupid;
- ret = btrfs_search_slot(trans, quota_root, &key, path, -1, 1);
- if (ret < 0)
- goto out;
- if (ret > 0) {
- ret = -ENOENT;
- goto out;
- }
- ret = btrfs_del_item(trans, quota_root, path);
- if (ret)
- goto out;
- btrfs_release_path(path);
- key.type = BTRFS_QGROUP_LIMIT_KEY;
- ret = btrfs_search_slot(trans, quota_root, &key, path, -1, 1);
- if (ret < 0)
- goto out;
- if (ret > 0) {
- ret = -ENOENT;
- goto out;
- }
- ret = btrfs_del_item(trans, quota_root, path);
- out:
- btrfs_free_path(path);
- return ret;
- }
- static int update_qgroup_limit_item(struct btrfs_trans_handle *trans,
- struct btrfs_root *root, u64 qgroupid,
- u64 flags, u64 max_rfer, u64 max_excl,
- u64 rsv_rfer, u64 rsv_excl)
- {
- struct btrfs_path *path;
- struct btrfs_key key;
- struct extent_buffer *l;
- struct btrfs_qgroup_limit_item *qgroup_limit;
- int ret;
- int slot;
- key.objectid = 0;
- key.type = BTRFS_QGROUP_LIMIT_KEY;
- key.offset = qgroupid;
- path = btrfs_alloc_path();
- BUG_ON(!path);
- ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
- if (ret > 0)
- ret = -ENOENT;
- if (ret)
- goto out;
- l = path->nodes[0];
- slot = path->slots[0];
- qgroup_limit = btrfs_item_ptr(l, path->slots[0],
- struct btrfs_qgroup_limit_item);
- btrfs_set_qgroup_limit_flags(l, qgroup_limit, flags);
- btrfs_set_qgroup_limit_max_rfer(l, qgroup_limit, max_rfer);
- btrfs_set_qgroup_limit_max_excl(l, qgroup_limit, max_excl);
- btrfs_set_qgroup_limit_rsv_rfer(l, qgroup_limit, rsv_rfer);
- btrfs_set_qgroup_limit_rsv_excl(l, qgroup_limit, rsv_excl);
- btrfs_mark_buffer_dirty(l);
- out:
- btrfs_free_path(path);
- return ret;
- }
- static int update_qgroup_info_item(struct btrfs_trans_handle *trans,
- struct btrfs_root *root,
- struct btrfs_qgroup *qgroup)
- {
- struct btrfs_path *path;
- struct btrfs_key key;
- struct extent_buffer *l;
- struct btrfs_qgroup_info_item *qgroup_info;
- int ret;
- int slot;
- key.objectid = 0;
- key.type = BTRFS_QGROUP_INFO_KEY;
- key.offset = qgroup->qgroupid;
- path = btrfs_alloc_path();
- BUG_ON(!path);
- ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
- if (ret > 0)
- ret = -ENOENT;
- if (ret)
- goto out;
- l = path->nodes[0];
- slot = path->slots[0];
- qgroup_info = btrfs_item_ptr(l, path->slots[0],
- struct btrfs_qgroup_info_item);
- btrfs_set_qgroup_info_generation(l, qgroup_info, trans->transid);
- btrfs_set_qgroup_info_rfer(l, qgroup_info, qgroup->rfer);
- btrfs_set_qgroup_info_rfer_cmpr(l, qgroup_info, qgroup->rfer_cmpr);
- btrfs_set_qgroup_info_excl(l, qgroup_info, qgroup->excl);
- btrfs_set_qgroup_info_excl_cmpr(l, qgroup_info, qgroup->excl_cmpr);
- btrfs_mark_buffer_dirty(l);
- out:
- btrfs_free_path(path);
- return ret;
- }
- static int update_qgroup_status_item(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info,
- struct btrfs_root *root)
- {
- struct btrfs_path *path;
- struct btrfs_key key;
- struct extent_buffer *l;
- struct btrfs_qgroup_status_item *ptr;
- int ret;
- int slot;
- key.objectid = 0;
- key.type = BTRFS_QGROUP_STATUS_KEY;
- key.offset = 0;
- path = btrfs_alloc_path();
- BUG_ON(!path);
- ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
- if (ret > 0)
- ret = -ENOENT;
- if (ret)
- goto out;
- l = path->nodes[0];
- slot = path->slots[0];
- ptr = btrfs_item_ptr(l, slot, struct btrfs_qgroup_status_item);
- btrfs_set_qgroup_status_flags(l, ptr, fs_info->qgroup_flags);
- btrfs_set_qgroup_status_generation(l, ptr, trans->transid);
- /* XXX scan */
- btrfs_mark_buffer_dirty(l);
- out:
- btrfs_free_path(path);
- return ret;
- }
- /*
- * called with qgroup_lock held
- */
- static int btrfs_clean_quota_tree(struct btrfs_trans_handle *trans,
- struct btrfs_root *root)
- {
- struct btrfs_path *path;
- struct btrfs_key key;
- int ret;
- if (!root)
- return -EINVAL;
- path = btrfs_alloc_path();
- if (!path)
- return -ENOMEM;
- while (1) {
- key.objectid = 0;
- key.offset = 0;
- key.type = 0;
- path->leave_spinning = 1;
- ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
- if (ret > 0) {
- if (path->slots[0] == 0)
- break;
- path->slots[0]--;
- } else if (ret < 0) {
- break;
- }
- ret = btrfs_del_item(trans, root, path);
- if (ret)
- goto out;
- btrfs_release_path(path);
- }
- ret = 0;
- out:
- root->fs_info->pending_quota_state = 0;
- btrfs_free_path(path);
- return ret;
- }
- int btrfs_quota_enable(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info)
- {
- struct btrfs_root *quota_root;
- struct btrfs_path *path = NULL;
- struct btrfs_qgroup_status_item *ptr;
- struct extent_buffer *leaf;
- struct btrfs_key key;
- int ret = 0;
- spin_lock(&fs_info->qgroup_lock);
- if (fs_info->quota_root) {
- fs_info->pending_quota_state = 1;
- spin_unlock(&fs_info->qgroup_lock);
- goto out;
- }
- spin_unlock(&fs_info->qgroup_lock);
- /*
- * initially create the quota tree
- */
- quota_root = btrfs_create_tree(trans, fs_info,
- BTRFS_QUOTA_TREE_OBJECTID);
- if (IS_ERR(quota_root)) {
- ret = PTR_ERR(quota_root);
- goto out;
- }
- path = btrfs_alloc_path();
- if (!path)
- return -ENOMEM;
- key.objectid = 0;
- key.type = BTRFS_QGROUP_STATUS_KEY;
- key.offset = 0;
- ret = btrfs_insert_empty_item(trans, quota_root, path, &key,
- sizeof(*ptr));
- if (ret)
- goto out;
- leaf = path->nodes[0];
- ptr = btrfs_item_ptr(leaf, path->slots[0],
- struct btrfs_qgroup_status_item);
- btrfs_set_qgroup_status_generation(leaf, ptr, trans->transid);
- btrfs_set_qgroup_status_version(leaf, ptr, BTRFS_QGROUP_STATUS_VERSION);
- fs_info->qgroup_flags = BTRFS_QGROUP_STATUS_FLAG_ON |
- BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
- btrfs_set_qgroup_status_flags(leaf, ptr, fs_info->qgroup_flags);
- btrfs_set_qgroup_status_scan(leaf, ptr, 0);
- btrfs_mark_buffer_dirty(leaf);
- spin_lock(&fs_info->qgroup_lock);
- fs_info->quota_root = quota_root;
- fs_info->pending_quota_state = 1;
- spin_unlock(&fs_info->qgroup_lock);
- out:
- btrfs_free_path(path);
- return ret;
- }
- int btrfs_quota_disable(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info)
- {
- struct btrfs_root *tree_root = fs_info->tree_root;
- struct btrfs_root *quota_root;
- int ret = 0;
- spin_lock(&fs_info->qgroup_lock);
- fs_info->quota_enabled = 0;
- fs_info->pending_quota_state = 0;
- quota_root = fs_info->quota_root;
- fs_info->quota_root = NULL;
- btrfs_free_qgroup_config(fs_info);
- spin_unlock(&fs_info->qgroup_lock);
- if (!quota_root)
- return -EINVAL;
- ret = btrfs_clean_quota_tree(trans, quota_root);
- if (ret)
- goto out;
- ret = btrfs_del_root(trans, tree_root, "a_root->root_key);
- if (ret)
- goto out;
- list_del("a_root->dirty_list);
- btrfs_tree_lock(quota_root->node);
- clean_tree_block(trans, tree_root, quota_root->node);
- btrfs_tree_unlock(quota_root->node);
- btrfs_free_tree_block(trans, quota_root, quota_root->node, 0, 1);
- free_extent_buffer(quota_root->node);
- free_extent_buffer(quota_root->commit_root);
- kfree(quota_root);
- out:
- return ret;
- }
- int btrfs_quota_rescan(struct btrfs_fs_info *fs_info)
- {
- /* FIXME */
- return 0;
- }
- int btrfs_add_qgroup_relation(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info, u64 src, u64 dst)
- {
- struct btrfs_root *quota_root;
- int ret = 0;
- quota_root = fs_info->quota_root;
- if (!quota_root)
- return -EINVAL;
- ret = add_qgroup_relation_item(trans, quota_root, src, dst);
- if (ret)
- return ret;
- ret = add_qgroup_relation_item(trans, quota_root, dst, src);
- if (ret) {
- del_qgroup_relation_item(trans, quota_root, src, dst);
- return ret;
- }
- spin_lock(&fs_info->qgroup_lock);
- ret = add_relation_rb(quota_root->fs_info, src, dst);
- spin_unlock(&fs_info->qgroup_lock);
- return ret;
- }
- int btrfs_del_qgroup_relation(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info, u64 src, u64 dst)
- {
- struct btrfs_root *quota_root;
- int ret = 0;
- int err;
- quota_root = fs_info->quota_root;
- if (!quota_root)
- return -EINVAL;
- ret = del_qgroup_relation_item(trans, quota_root, src, dst);
- err = del_qgroup_relation_item(trans, quota_root, dst, src);
- if (err && !ret)
- ret = err;
- spin_lock(&fs_info->qgroup_lock);
- del_relation_rb(fs_info, src, dst);
- spin_unlock(&fs_info->qgroup_lock);
- return ret;
- }
- int btrfs_create_qgroup(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info, u64 qgroupid, char *name)
- {
- struct btrfs_root *quota_root;
- struct btrfs_qgroup *qgroup;
- int ret = 0;
- quota_root = fs_info->quota_root;
- if (!quota_root)
- return -EINVAL;
- ret = add_qgroup_item(trans, quota_root, qgroupid);
- spin_lock(&fs_info->qgroup_lock);
- qgroup = add_qgroup_rb(fs_info, qgroupid);
- spin_unlock(&fs_info->qgroup_lock);
- if (IS_ERR(qgroup))
- ret = PTR_ERR(qgroup);
- return ret;
- }
- int btrfs_remove_qgroup(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info, u64 qgroupid)
- {
- struct btrfs_root *quota_root;
- int ret = 0;
- quota_root = fs_info->quota_root;
- if (!quota_root)
- return -EINVAL;
- ret = del_qgroup_item(trans, quota_root, qgroupid);
- spin_lock(&fs_info->qgroup_lock);
- del_qgroup_rb(quota_root->fs_info, qgroupid);
- spin_unlock(&fs_info->qgroup_lock);
- return ret;
- }
- int btrfs_limit_qgroup(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info, u64 qgroupid,
- struct btrfs_qgroup_limit *limit)
- {
- struct btrfs_root *quota_root = fs_info->quota_root;
- struct btrfs_qgroup *qgroup;
- int ret = 0;
- if (!quota_root)
- return -EINVAL;
- ret = update_qgroup_limit_item(trans, quota_root, qgroupid,
- limit->flags, limit->max_rfer,
- limit->max_excl, limit->rsv_rfer,
- limit->rsv_excl);
- if (ret) {
- fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
- printk(KERN_INFO "unable to update quota limit for %llu\n",
- (unsigned long long)qgroupid);
- }
- spin_lock(&fs_info->qgroup_lock);
- qgroup = find_qgroup_rb(fs_info, qgroupid);
- if (!qgroup) {
- ret = -ENOENT;
- goto unlock;
- }
- qgroup->lim_flags = limit->flags;
- qgroup->max_rfer = limit->max_rfer;
- qgroup->max_excl = limit->max_excl;
- qgroup->rsv_rfer = limit->rsv_rfer;
- qgroup->rsv_excl = limit->rsv_excl;
- unlock:
- spin_unlock(&fs_info->qgroup_lock);
- return ret;
- }
- static void qgroup_dirty(struct btrfs_fs_info *fs_info,
- struct btrfs_qgroup *qgroup)
- {
- if (list_empty(&qgroup->dirty))
- list_add(&qgroup->dirty, &fs_info->dirty_qgroups);
- }
- /*
- * btrfs_qgroup_record_ref is called when the ref is added or deleted. it puts
- * the modification into a list that's later used by btrfs_end_transaction to
- * pass the recorded modifications on to btrfs_qgroup_account_ref.
- */
- int btrfs_qgroup_record_ref(struct btrfs_trans_handle *trans,
- struct btrfs_delayed_ref_node *node,
- struct btrfs_delayed_extent_op *extent_op)
- {
- struct qgroup_update *u;
- BUG_ON(!trans->delayed_ref_elem.seq);
- u = kmalloc(sizeof(*u), GFP_NOFS);
- if (!u)
- return -ENOMEM;
- u->node = node;
- u->extent_op = extent_op;
- list_add_tail(&u->list, &trans->qgroup_ref_list);
- return 0;
- }
- /*
- * btrfs_qgroup_account_ref is called for every ref that is added to or deleted
- * from the fs. First, all roots referencing the extent are searched, and
- * then the space is accounted accordingly to the different roots. The
- * accounting algorithm works in 3 steps documented inline.
- */
- int btrfs_qgroup_account_ref(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info,
- struct btrfs_delayed_ref_node *node,
- struct btrfs_delayed_extent_op *extent_op)
- {
- struct btrfs_key ins;
- struct btrfs_root *quota_root;
- u64 ref_root;
- struct btrfs_qgroup *qgroup;
- struct ulist_node *unode;
- struct ulist *roots = NULL;
- struct ulist *tmp = NULL;
- struct ulist_iterator uiter;
- u64 seq;
- int ret = 0;
- int sgn;
- if (!fs_info->quota_enabled)
- return 0;
- BUG_ON(!fs_info->quota_root);
- ins.objectid = node->bytenr;
- ins.offset = node->num_bytes;
- ins.type = BTRFS_EXTENT_ITEM_KEY;
- if (node->type == BTRFS_TREE_BLOCK_REF_KEY ||
- node->type == BTRFS_SHARED_BLOCK_REF_KEY) {
- struct btrfs_delayed_tree_ref *ref;
- ref = btrfs_delayed_node_to_tree_ref(node);
- ref_root = ref->root;
- } else if (node->type == BTRFS_EXTENT_DATA_REF_KEY ||
- node->type == BTRFS_SHARED_DATA_REF_KEY) {
- struct btrfs_delayed_data_ref *ref;
- ref = btrfs_delayed_node_to_data_ref(node);
- ref_root = ref->root;
- } else {
- BUG();
- }
- if (!is_fstree(ref_root)) {
- /*
- * non-fs-trees are not being accounted
- */
- return 0;
- }
- switch (node->action) {
- case BTRFS_ADD_DELAYED_REF:
- case BTRFS_ADD_DELAYED_EXTENT:
- sgn = 1;
- break;
- case BTRFS_DROP_DELAYED_REF:
- sgn = -1;
- break;
- case BTRFS_UPDATE_DELAYED_HEAD:
- return 0;
- default:
- BUG();
- }
- /*
- * the delayed ref sequence number we pass depends on the direction of
- * the operation. for add operations, we pass (node->seq - 1) to skip
- * the delayed ref's current sequence number, because we need the state
- * of the tree before the add operation. for delete operations, we pass
- * (node->seq) to include the delayed ref's current sequence number,
- * because we need the state of the tree after the delete operation.
- */
- ret = btrfs_find_all_roots(trans, fs_info, node->bytenr,
- sgn > 0 ? node->seq - 1 : node->seq, &roots);
- if (ret < 0)
- goto out;
- spin_lock(&fs_info->qgroup_lock);
- quota_root = fs_info->quota_root;
- if (!quota_root)
- goto unlock;
- qgroup = find_qgroup_rb(fs_info, ref_root);
- if (!qgroup)
- goto unlock;
- /*
- * step 1: for each old ref, visit all nodes once and inc refcnt
- */
- tmp = ulist_alloc(GFP_ATOMIC);
- if (!tmp) {
- ret = -ENOMEM;
- goto unlock;
- }
- seq = fs_info->qgroup_seq;
- fs_info->qgroup_seq += roots->nnodes + 1; /* max refcnt */
- ULIST_ITER_INIT(&uiter);
- while ((unode = ulist_next(roots, &uiter))) {
- struct ulist_node *tmp_unode;
- struct ulist_iterator tmp_uiter;
- struct btrfs_qgroup *qg;
- qg = find_qgroup_rb(fs_info, unode->val);
- if (!qg)
- continue;
- ulist_reinit(tmp);
- /* XXX id not needed */
- ulist_add(tmp, qg->qgroupid, (unsigned long)qg, GFP_ATOMIC);
- ULIST_ITER_INIT(&tmp_uiter);
- while ((tmp_unode = ulist_next(tmp, &tmp_uiter))) {
- struct btrfs_qgroup_list *glist;
- qg = (struct btrfs_qgroup *)tmp_unode->aux;
- if (qg->refcnt < seq)
- qg->refcnt = seq + 1;
- else
- ++qg->refcnt;
- list_for_each_entry(glist, &qg->groups, next_group) {
- ulist_add(tmp, glist->group->qgroupid,
- (unsigned long)glist->group,
- GFP_ATOMIC);
- }
- }
- }
- /*
- * step 2: walk from the new root
- */
- ulist_reinit(tmp);
- ulist_add(tmp, qgroup->qgroupid, (unsigned long)qgroup, GFP_ATOMIC);
- ULIST_ITER_INIT(&uiter);
- while ((unode = ulist_next(tmp, &uiter))) {
- struct btrfs_qgroup *qg;
- struct btrfs_qgroup_list *glist;
- qg = (struct btrfs_qgroup *)unode->aux;
- if (qg->refcnt < seq) {
- /* not visited by step 1 */
- qg->rfer += sgn * node->num_bytes;
- qg->rfer_cmpr += sgn * node->num_bytes;
- if (roots->nnodes == 0) {
- qg->excl += sgn * node->num_bytes;
- qg->excl_cmpr += sgn * node->num_bytes;
- }
- qgroup_dirty(fs_info, qg);
- }
- WARN_ON(qg->tag >= seq);
- qg->tag = seq;
- list_for_each_entry(glist, &qg->groups, next_group) {
- ulist_add(tmp, glist->group->qgroupid,
- (unsigned long)glist->group, GFP_ATOMIC);
- }
- }
- /*
- * step 3: walk again from old refs
- */
- ULIST_ITER_INIT(&uiter);
- while ((unode = ulist_next(roots, &uiter))) {
- struct btrfs_qgroup *qg;
- struct ulist_node *tmp_unode;
- struct ulist_iterator tmp_uiter;
- qg = find_qgroup_rb(fs_info, unode->val);
- if (!qg)
- continue;
- ulist_reinit(tmp);
- ulist_add(tmp, qg->qgroupid, (unsigned long)qg, GFP_ATOMIC);
- ULIST_ITER_INIT(&tmp_uiter);
- while ((tmp_unode = ulist_next(tmp, &tmp_uiter))) {
- struct btrfs_qgroup_list *glist;
- qg = (struct btrfs_qgroup *)tmp_unode->aux;
- if (qg->tag == seq)
- continue;
- if (qg->refcnt - seq == roots->nnodes) {
- qg->excl -= sgn * node->num_bytes;
- qg->excl_cmpr -= sgn * node->num_bytes;
- qgroup_dirty(fs_info, qg);
- }
- list_for_each_entry(glist, &qg->groups, next_group) {
- ulist_add(tmp, glist->group->qgroupid,
- (unsigned long)glist->group,
- GFP_ATOMIC);
- }
- }
- }
- ret = 0;
- unlock:
- spin_unlock(&fs_info->qgroup_lock);
- out:
- ulist_free(roots);
- ulist_free(tmp);
- return ret;
- }
- /*
- * called from commit_transaction. Writes all changed qgroups to disk.
- */
- int btrfs_run_qgroups(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info)
- {
- struct btrfs_root *quota_root = fs_info->quota_root;
- int ret = 0;
- if (!quota_root)
- goto out;
- fs_info->quota_enabled = fs_info->pending_quota_state;
- spin_lock(&fs_info->qgroup_lock);
- while (!list_empty(&fs_info->dirty_qgroups)) {
- struct btrfs_qgroup *qgroup;
- qgroup = list_first_entry(&fs_info->dirty_qgroups,
- struct btrfs_qgroup, dirty);
- list_del_init(&qgroup->dirty);
- spin_unlock(&fs_info->qgroup_lock);
- ret = update_qgroup_info_item(trans, quota_root, qgroup);
- if (ret)
- fs_info->qgroup_flags |=
- BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
- spin_lock(&fs_info->qgroup_lock);
- }
- if (fs_info->quota_enabled)
- fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_ON;
- else
- fs_info->qgroup_flags &= ~BTRFS_QGROUP_STATUS_FLAG_ON;
- spin_unlock(&fs_info->qgroup_lock);
- ret = update_qgroup_status_item(trans, fs_info, quota_root);
- if (ret)
- fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
- out:
- return ret;
- }
- /*
- * copy the acounting information between qgroups. This is necessary when a
- * snapshot or a subvolume is created
- */
- int btrfs_qgroup_inherit(struct btrfs_trans_handle *trans,
- struct btrfs_fs_info *fs_info, u64 srcid, u64 objectid,
- struct btrfs_qgroup_inherit *inherit)
- {
- int ret = 0;
- int i;
- u64 *i_qgroups;
- struct btrfs_root *quota_root = fs_info->quota_root;
- struct btrfs_qgroup *srcgroup;
- struct btrfs_qgroup *dstgroup;
- u32 level_size = 0;
- if (!fs_info->quota_enabled)
- return 0;
- if (!quota_root)
- return -EINVAL;
- /*
- * create a tracking group for the subvol itself
- */
- ret = add_qgroup_item(trans, quota_root, objectid);
- if (ret)
- goto out;
- if (inherit && inherit->flags & BTRFS_QGROUP_INHERIT_SET_LIMITS) {
- ret = update_qgroup_limit_item(trans, quota_root, objectid,
- inherit->lim.flags,
- inherit->lim.max_rfer,
- inherit->lim.max_excl,
- inherit->lim.rsv_rfer,
- inherit->lim.rsv_excl);
- if (ret)
- goto out;
- }
- if (srcid) {
- struct btrfs_root *srcroot;
- struct btrfs_key srckey;
- int srcroot_level;
- srckey.objectid = srcid;
- srckey.type = BTRFS_ROOT_ITEM_KEY;
- srckey.offset = (u64)-1;
- srcroot = btrfs_read_fs_root_no_name(fs_info, &srckey);
- if (IS_ERR(srcroot)) {
- ret = PTR_ERR(srcroot);
- goto out;
- }
- rcu_read_lock();
- srcroot_level = btrfs_header_level(srcroot->node);
- level_size = btrfs_level_size(srcroot, srcroot_level);
- rcu_read_unlock();
- }
- /*
- * add qgroup to all inherited groups
- */
- if (inherit) {
- i_qgroups = (u64 *)(inherit + 1);
- for (i = 0; i < inherit->num_qgroups; ++i) {
- ret = add_qgroup_relation_item(trans, quota_root,
- objectid, *i_qgroups);
- if (ret)
- goto out;
- ret = add_qgroup_relation_item(trans, quota_root,
- *i_qgroups, objectid);
- if (ret)
- goto out;
- ++i_qgroups;
- }
- }
- spin_lock(&fs_info->qgroup_lock);
- dstgroup = add_qgroup_rb(fs_info, objectid);
- if (!dstgroup)
- goto unlock;
- if (srcid) {
- srcgroup = find_qgroup_rb(fs_info, srcid);
- if (!srcgroup)
- goto unlock;
- dstgroup->rfer = srcgroup->rfer - level_size;
- dstgroup->rfer_cmpr = srcgroup->rfer_cmpr - level_size;
- srcgroup->excl = level_size;
- srcgroup->excl_cmpr = level_size;
- qgroup_dirty(fs_info, dstgroup);
- qgroup_dirty(fs_info, srcgroup);
- }
- if (!inherit)
- goto unlock;
- i_qgroups = (u64 *)(inherit + 1);
- for (i = 0; i < inherit->num_qgroups; ++i) {
- ret = add_relation_rb(quota_root->fs_info, objectid,
- *i_qgroups);
- if (ret)
- goto unlock;
- ++i_qgroups;
- }
- for (i = 0; i < inherit->num_ref_copies; ++i) {
- struct btrfs_qgroup *src;
- struct btrfs_qgroup *dst;
- src = find_qgroup_rb(fs_info, i_qgroups[0]);
- dst = find_qgroup_rb(fs_info, i_qgroups[1]);
- if (!src || !dst) {
- ret = -EINVAL;
- goto unlock;
- }
- dst->rfer = src->rfer - level_size;
- dst->rfer_cmpr = src->rfer_cmpr - level_size;
- i_qgroups += 2;
- }
- for (i = 0; i < inherit->num_excl_copies; ++i) {
- struct btrfs_qgroup *src;
- struct btrfs_qgroup *dst;
- src = find_qgroup_rb(fs_info, i_qgroups[0]);
- dst = find_qgroup_rb(fs_info, i_qgroups[1]);
- if (!src || !dst) {
- ret = -EINVAL;
- goto unlock;
- }
- dst->excl = src->excl + level_size;
- dst->excl_cmpr = src->excl_cmpr + level_size;
- i_qgroups += 2;
- }
- unlock:
- spin_unlock(&fs_info->qgroup_lock);
- out:
- return ret;
- }
- /*
- * reserve some space for a qgroup and all its parents. The reservation takes
- * place with start_transaction or dealloc_reserve, similar to ENOSPC
- * accounting. If not enough space is available, EDQUOT is returned.
- * We assume that the requested space is new for all qgroups.
- */
- int btrfs_qgroup_reserve(struct btrfs_root *root, u64 num_bytes)
- {
- struct btrfs_root *quota_root;
- struct btrfs_qgroup *qgroup;
- struct btrfs_fs_info *fs_info = root->fs_info;
- u64 ref_root = root->root_key.objectid;
- int ret = 0;
- struct ulist *ulist = NULL;
- struct ulist_node *unode;
- struct ulist_iterator uiter;
- if (!is_fstree(ref_root))
- return 0;
- if (num_bytes == 0)
- return 0;
- spin_lock(&fs_info->qgroup_lock);
- quota_root = fs_info->quota_root;
- if (!quota_root)
- goto out;
- qgroup = find_qgroup_rb(fs_info, ref_root);
- if (!qgroup)
- goto out;
- /*
- * in a first step, we check all affected qgroups if any limits would
- * be exceeded
- */
- ulist = ulist_alloc(GFP_ATOMIC);
- ulist_add(ulist, qgroup->qgroupid, (unsigned long)qgroup, GFP_ATOMIC);
- ULIST_ITER_INIT(&uiter);
- while ((unode = ulist_next(ulist, &uiter))) {
- struct btrfs_qgroup *qg;
- struct btrfs_qgroup_list *glist;
- qg = (struct btrfs_qgroup *)unode->aux;
- if ((qg->lim_flags & BTRFS_QGROUP_LIMIT_MAX_RFER) &&
- qg->reserved + qg->rfer + num_bytes >
- qg->max_rfer)
- ret = -EDQUOT;
- if ((qg->lim_flags & BTRFS_QGROUP_LIMIT_MAX_EXCL) &&
- qg->reserved + qg->excl + num_bytes >
- qg->max_excl)
- ret = -EDQUOT;
- list_for_each_entry(glist, &qg->groups, next_group) {
- ulist_add(ulist, glist->group->qgroupid,
- (unsigned long)glist->group, GFP_ATOMIC);
- }
- }
- if (ret)
- goto out;
- /*
- * no limits exceeded, now record the reservation into all qgroups
- */
- ULIST_ITER_INIT(&uiter);
- while ((unode = ulist_next(ulist, &uiter))) {
- struct btrfs_qgroup *qg;
- qg = (struct btrfs_qgroup *)unode->aux;
- qg->reserved += num_bytes;
- }
- out:
- spin_unlock(&fs_info->qgroup_lock);
- ulist_free(ulist);
- return ret;
- }
- void btrfs_qgroup_free(struct btrfs_root *root, u64 num_bytes)
- {
- struct btrfs_root *quota_root;
- struct btrfs_qgroup *qgroup;
- struct btrfs_fs_info *fs_info = root->fs_info;
- struct ulist *ulist = NULL;
- struct ulist_node *unode;
- struct ulist_iterator uiter;
- u64 ref_root = root->root_key.objectid;
- if (!is_fstree(ref_root))
- return;
- if (num_bytes == 0)
- return;
- spin_lock(&fs_info->qgroup_lock);
- quota_root = fs_info->quota_root;
- if (!quota_root)
- goto out;
- qgroup = find_qgroup_rb(fs_info, ref_root);
- if (!qgroup)
- goto out;
- ulist = ulist_alloc(GFP_ATOMIC);
- ulist_add(ulist, qgroup->qgroupid, (unsigned long)qgroup, GFP_ATOMIC);
- ULIST_ITER_INIT(&uiter);
- while ((unode = ulist_next(ulist, &uiter))) {
- struct btrfs_qgroup *qg;
- struct btrfs_qgroup_list *glist;
- qg = (struct btrfs_qgroup *)unode->aux;
- qg->reserved -= num_bytes;
- list_for_each_entry(glist, &qg->groups, next_group) {
- ulist_add(ulist, glist->group->qgroupid,
- (unsigned long)glist->group, GFP_ATOMIC);
- }
- }
- out:
- spin_unlock(&fs_info->qgroup_lock);
- ulist_free(ulist);
- }
- void assert_qgroups_uptodate(struct btrfs_trans_handle *trans)
- {
- if (list_empty(&trans->qgroup_ref_list) && !trans->delayed_ref_elem.seq)
- return;
- printk(KERN_ERR "btrfs: qgroups not uptodate in trans handle %p: list is%s empty, seq is %llu\n",
- trans, list_empty(&trans->qgroup_ref_list) ? "" : " not",
- trans->delayed_ref_elem.seq);
- BUG();
- }
|