qgroup.c 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590
  1. /*
  2. * Copyright (C) 2011 STRATO. 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. #include <linux/sched.h>
  19. #include <linux/pagemap.h>
  20. #include <linux/writeback.h>
  21. #include <linux/blkdev.h>
  22. #include <linux/rbtree.h>
  23. #include <linux/slab.h>
  24. #include <linux/workqueue.h>
  25. #include "ctree.h"
  26. #include "transaction.h"
  27. #include "disk-io.h"
  28. #include "locking.h"
  29. #include "ulist.h"
  30. #include "ioctl.h"
  31. #include "backref.h"
  32. /* TODO XXX FIXME
  33. * - subvol delete -> delete when ref goes to 0? delete limits also?
  34. * - reorganize keys
  35. * - compressed
  36. * - sync
  37. * - rescan
  38. * - copy also limits on subvol creation
  39. * - limit
  40. * - caches fuer ulists
  41. * - performance benchmarks
  42. * - check all ioctl parameters
  43. */
  44. /*
  45. * one struct for each qgroup, organized in fs_info->qgroup_tree.
  46. */
  47. struct btrfs_qgroup {
  48. u64 qgroupid;
  49. /*
  50. * state
  51. */
  52. u64 rfer; /* referenced */
  53. u64 rfer_cmpr; /* referenced compressed */
  54. u64 excl; /* exclusive */
  55. u64 excl_cmpr; /* exclusive compressed */
  56. /*
  57. * limits
  58. */
  59. u64 lim_flags; /* which limits are set */
  60. u64 max_rfer;
  61. u64 max_excl;
  62. u64 rsv_rfer;
  63. u64 rsv_excl;
  64. /*
  65. * reservation tracking
  66. */
  67. u64 reserved;
  68. /*
  69. * lists
  70. */
  71. struct list_head groups; /* groups this group is member of */
  72. struct list_head members; /* groups that are members of this group */
  73. struct list_head dirty; /* dirty groups */
  74. struct rb_node node; /* tree of qgroups */
  75. /*
  76. * temp variables for accounting operations
  77. */
  78. u64 tag;
  79. u64 refcnt;
  80. };
  81. /*
  82. * glue structure to represent the relations between qgroups.
  83. */
  84. struct btrfs_qgroup_list {
  85. struct list_head next_group;
  86. struct list_head next_member;
  87. struct btrfs_qgroup *group;
  88. struct btrfs_qgroup *member;
  89. };
  90. /* must be called with qgroup_lock held */
  91. static struct btrfs_qgroup *find_qgroup_rb(struct btrfs_fs_info *fs_info,
  92. u64 qgroupid)
  93. {
  94. struct rb_node *n = fs_info->qgroup_tree.rb_node;
  95. struct btrfs_qgroup *qgroup;
  96. while (n) {
  97. qgroup = rb_entry(n, struct btrfs_qgroup, node);
  98. if (qgroup->qgroupid < qgroupid)
  99. n = n->rb_left;
  100. else if (qgroup->qgroupid > qgroupid)
  101. n = n->rb_right;
  102. else
  103. return qgroup;
  104. }
  105. return NULL;
  106. }
  107. /* must be called with qgroup_lock held */
  108. static struct btrfs_qgroup *add_qgroup_rb(struct btrfs_fs_info *fs_info,
  109. u64 qgroupid)
  110. {
  111. struct rb_node **p = &fs_info->qgroup_tree.rb_node;
  112. struct rb_node *parent = NULL;
  113. struct btrfs_qgroup *qgroup;
  114. while (*p) {
  115. parent = *p;
  116. qgroup = rb_entry(parent, struct btrfs_qgroup, node);
  117. if (qgroup->qgroupid < qgroupid)
  118. p = &(*p)->rb_left;
  119. else if (qgroup->qgroupid > qgroupid)
  120. p = &(*p)->rb_right;
  121. else
  122. return qgroup;
  123. }
  124. qgroup = kzalloc(sizeof(*qgroup), GFP_ATOMIC);
  125. if (!qgroup)
  126. return ERR_PTR(-ENOMEM);
  127. qgroup->qgroupid = qgroupid;
  128. INIT_LIST_HEAD(&qgroup->groups);
  129. INIT_LIST_HEAD(&qgroup->members);
  130. INIT_LIST_HEAD(&qgroup->dirty);
  131. rb_link_node(&qgroup->node, parent, p);
  132. rb_insert_color(&qgroup->node, &fs_info->qgroup_tree);
  133. return qgroup;
  134. }
  135. /* must be called with qgroup_lock held */
  136. static int del_qgroup_rb(struct btrfs_fs_info *fs_info, u64 qgroupid)
  137. {
  138. struct btrfs_qgroup *qgroup = find_qgroup_rb(fs_info, qgroupid);
  139. struct btrfs_qgroup_list *list;
  140. if (!qgroup)
  141. return -ENOENT;
  142. rb_erase(&qgroup->node, &fs_info->qgroup_tree);
  143. list_del(&qgroup->dirty);
  144. while (!list_empty(&qgroup->groups)) {
  145. list = list_first_entry(&qgroup->groups,
  146. struct btrfs_qgroup_list, next_group);
  147. list_del(&list->next_group);
  148. list_del(&list->next_member);
  149. kfree(list);
  150. }
  151. while (!list_empty(&qgroup->members)) {
  152. list = list_first_entry(&qgroup->members,
  153. struct btrfs_qgroup_list, next_member);
  154. list_del(&list->next_group);
  155. list_del(&list->next_member);
  156. kfree(list);
  157. }
  158. kfree(qgroup);
  159. return 0;
  160. }
  161. /* must be called with qgroup_lock held */
  162. static int add_relation_rb(struct btrfs_fs_info *fs_info,
  163. u64 memberid, u64 parentid)
  164. {
  165. struct btrfs_qgroup *member;
  166. struct btrfs_qgroup *parent;
  167. struct btrfs_qgroup_list *list;
  168. member = find_qgroup_rb(fs_info, memberid);
  169. parent = find_qgroup_rb(fs_info, parentid);
  170. if (!member || !parent)
  171. return -ENOENT;
  172. list = kzalloc(sizeof(*list), GFP_ATOMIC);
  173. if (!list)
  174. return -ENOMEM;
  175. list->group = parent;
  176. list->member = member;
  177. list_add_tail(&list->next_group, &member->groups);
  178. list_add_tail(&list->next_member, &parent->members);
  179. return 0;
  180. }
  181. /* must be called with qgroup_lock held */
  182. static int del_relation_rb(struct btrfs_fs_info *fs_info,
  183. u64 memberid, u64 parentid)
  184. {
  185. struct btrfs_qgroup *member;
  186. struct btrfs_qgroup *parent;
  187. struct btrfs_qgroup_list *list;
  188. member = find_qgroup_rb(fs_info, memberid);
  189. parent = find_qgroup_rb(fs_info, parentid);
  190. if (!member || !parent)
  191. return -ENOENT;
  192. list_for_each_entry(list, &member->groups, next_group) {
  193. if (list->group == parent) {
  194. list_del(&list->next_group);
  195. list_del(&list->next_member);
  196. kfree(list);
  197. return 0;
  198. }
  199. }
  200. return -ENOENT;
  201. }
  202. /*
  203. * The full config is read in one go, only called from open_ctree()
  204. * It doesn't use any locking, as at this point we're still single-threaded
  205. */
  206. int btrfs_read_qgroup_config(struct btrfs_fs_info *fs_info)
  207. {
  208. struct btrfs_key key;
  209. struct btrfs_key found_key;
  210. struct btrfs_root *quota_root = fs_info->quota_root;
  211. struct btrfs_path *path = NULL;
  212. struct extent_buffer *l;
  213. int slot;
  214. int ret = 0;
  215. u64 flags = 0;
  216. if (!fs_info->quota_enabled)
  217. return 0;
  218. path = btrfs_alloc_path();
  219. if (!path) {
  220. ret = -ENOMEM;
  221. goto out;
  222. }
  223. /* default this to quota off, in case no status key is found */
  224. fs_info->qgroup_flags = 0;
  225. /*
  226. * pass 1: read status, all qgroup infos and limits
  227. */
  228. key.objectid = 0;
  229. key.type = 0;
  230. key.offset = 0;
  231. ret = btrfs_search_slot_for_read(quota_root, &key, path, 1, 1);
  232. if (ret)
  233. goto out;
  234. while (1) {
  235. struct btrfs_qgroup *qgroup;
  236. slot = path->slots[0];
  237. l = path->nodes[0];
  238. btrfs_item_key_to_cpu(l, &found_key, slot);
  239. if (found_key.type == BTRFS_QGROUP_STATUS_KEY) {
  240. struct btrfs_qgroup_status_item *ptr;
  241. ptr = btrfs_item_ptr(l, slot,
  242. struct btrfs_qgroup_status_item);
  243. if (btrfs_qgroup_status_version(l, ptr) !=
  244. BTRFS_QGROUP_STATUS_VERSION) {
  245. printk(KERN_ERR
  246. "btrfs: old qgroup version, quota disabled\n");
  247. goto out;
  248. }
  249. if (btrfs_qgroup_status_generation(l, ptr) !=
  250. fs_info->generation) {
  251. flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  252. printk(KERN_ERR
  253. "btrfs: qgroup generation mismatch, "
  254. "marked as inconsistent\n");
  255. }
  256. fs_info->qgroup_flags = btrfs_qgroup_status_flags(l,
  257. ptr);
  258. /* FIXME read scan element */
  259. goto next1;
  260. }
  261. if (found_key.type != BTRFS_QGROUP_INFO_KEY &&
  262. found_key.type != BTRFS_QGROUP_LIMIT_KEY)
  263. goto next1;
  264. qgroup = find_qgroup_rb(fs_info, found_key.offset);
  265. if ((qgroup && found_key.type == BTRFS_QGROUP_INFO_KEY) ||
  266. (!qgroup && found_key.type == BTRFS_QGROUP_LIMIT_KEY)) {
  267. printk(KERN_ERR "btrfs: inconsitent qgroup config\n");
  268. flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  269. }
  270. if (!qgroup) {
  271. qgroup = add_qgroup_rb(fs_info, found_key.offset);
  272. if (IS_ERR(qgroup)) {
  273. ret = PTR_ERR(qgroup);
  274. goto out;
  275. }
  276. }
  277. switch (found_key.type) {
  278. case BTRFS_QGROUP_INFO_KEY: {
  279. struct btrfs_qgroup_info_item *ptr;
  280. ptr = btrfs_item_ptr(l, slot,
  281. struct btrfs_qgroup_info_item);
  282. qgroup->rfer = btrfs_qgroup_info_rfer(l, ptr);
  283. qgroup->rfer_cmpr = btrfs_qgroup_info_rfer_cmpr(l, ptr);
  284. qgroup->excl = btrfs_qgroup_info_excl(l, ptr);
  285. qgroup->excl_cmpr = btrfs_qgroup_info_excl_cmpr(l, ptr);
  286. /* generation currently unused */
  287. break;
  288. }
  289. case BTRFS_QGROUP_LIMIT_KEY: {
  290. struct btrfs_qgroup_limit_item *ptr;
  291. ptr = btrfs_item_ptr(l, slot,
  292. struct btrfs_qgroup_limit_item);
  293. qgroup->lim_flags = btrfs_qgroup_limit_flags(l, ptr);
  294. qgroup->max_rfer = btrfs_qgroup_limit_max_rfer(l, ptr);
  295. qgroup->max_excl = btrfs_qgroup_limit_max_excl(l, ptr);
  296. qgroup->rsv_rfer = btrfs_qgroup_limit_rsv_rfer(l, ptr);
  297. qgroup->rsv_excl = btrfs_qgroup_limit_rsv_excl(l, ptr);
  298. break;
  299. }
  300. }
  301. next1:
  302. ret = btrfs_next_item(quota_root, path);
  303. if (ret < 0)
  304. goto out;
  305. if (ret)
  306. break;
  307. }
  308. btrfs_release_path(path);
  309. /*
  310. * pass 2: read all qgroup relations
  311. */
  312. key.objectid = 0;
  313. key.type = BTRFS_QGROUP_RELATION_KEY;
  314. key.offset = 0;
  315. ret = btrfs_search_slot_for_read(quota_root, &key, path, 1, 0);
  316. if (ret)
  317. goto out;
  318. while (1) {
  319. slot = path->slots[0];
  320. l = path->nodes[0];
  321. btrfs_item_key_to_cpu(l, &found_key, slot);
  322. if (found_key.type != BTRFS_QGROUP_RELATION_KEY)
  323. goto next2;
  324. if (found_key.objectid > found_key.offset) {
  325. /* parent <- member, not needed to build config */
  326. /* FIXME should we omit the key completely? */
  327. goto next2;
  328. }
  329. ret = add_relation_rb(fs_info, found_key.objectid,
  330. found_key.offset);
  331. if (ret)
  332. goto out;
  333. next2:
  334. ret = btrfs_next_item(quota_root, path);
  335. if (ret < 0)
  336. goto out;
  337. if (ret)
  338. break;
  339. }
  340. out:
  341. fs_info->qgroup_flags |= flags;
  342. if (!(fs_info->qgroup_flags & BTRFS_QGROUP_STATUS_FLAG_ON)) {
  343. fs_info->quota_enabled = 0;
  344. fs_info->pending_quota_state = 0;
  345. }
  346. btrfs_free_path(path);
  347. return ret < 0 ? ret : 0;
  348. }
  349. /*
  350. * This is only called from close_ctree() or open_ctree(), both in single-
  351. * treaded paths. Clean up the in-memory structures. No locking needed.
  352. */
  353. void btrfs_free_qgroup_config(struct btrfs_fs_info *fs_info)
  354. {
  355. struct rb_node *n;
  356. struct btrfs_qgroup *qgroup;
  357. struct btrfs_qgroup_list *list;
  358. while ((n = rb_first(&fs_info->qgroup_tree))) {
  359. qgroup = rb_entry(n, struct btrfs_qgroup, node);
  360. rb_erase(n, &fs_info->qgroup_tree);
  361. WARN_ON(!list_empty(&qgroup->dirty));
  362. while (!list_empty(&qgroup->groups)) {
  363. list = list_first_entry(&qgroup->groups,
  364. struct btrfs_qgroup_list,
  365. next_group);
  366. list_del(&list->next_group);
  367. list_del(&list->next_member);
  368. kfree(list);
  369. }
  370. while (!list_empty(&qgroup->members)) {
  371. list = list_first_entry(&qgroup->members,
  372. struct btrfs_qgroup_list,
  373. next_member);
  374. list_del(&list->next_group);
  375. list_del(&list->next_member);
  376. kfree(list);
  377. }
  378. kfree(qgroup);
  379. }
  380. }
  381. static int add_qgroup_relation_item(struct btrfs_trans_handle *trans,
  382. struct btrfs_root *quota_root,
  383. u64 src, u64 dst)
  384. {
  385. int ret;
  386. struct btrfs_path *path;
  387. struct btrfs_key key;
  388. path = btrfs_alloc_path();
  389. if (!path)
  390. return -ENOMEM;
  391. key.objectid = src;
  392. key.type = BTRFS_QGROUP_RELATION_KEY;
  393. key.offset = dst;
  394. ret = btrfs_insert_empty_item(trans, quota_root, path, &key, 0);
  395. btrfs_mark_buffer_dirty(path->nodes[0]);
  396. btrfs_free_path(path);
  397. return ret;
  398. }
  399. static int del_qgroup_relation_item(struct btrfs_trans_handle *trans,
  400. struct btrfs_root *quota_root,
  401. u64 src, u64 dst)
  402. {
  403. int ret;
  404. struct btrfs_path *path;
  405. struct btrfs_key key;
  406. path = btrfs_alloc_path();
  407. if (!path)
  408. return -ENOMEM;
  409. key.objectid = src;
  410. key.type = BTRFS_QGROUP_RELATION_KEY;
  411. key.offset = dst;
  412. ret = btrfs_search_slot(trans, quota_root, &key, path, -1, 1);
  413. if (ret < 0)
  414. goto out;
  415. if (ret > 0) {
  416. ret = -ENOENT;
  417. goto out;
  418. }
  419. ret = btrfs_del_item(trans, quota_root, path);
  420. out:
  421. btrfs_free_path(path);
  422. return ret;
  423. }
  424. static int add_qgroup_item(struct btrfs_trans_handle *trans,
  425. struct btrfs_root *quota_root, u64 qgroupid)
  426. {
  427. int ret;
  428. struct btrfs_path *path;
  429. struct btrfs_qgroup_info_item *qgroup_info;
  430. struct btrfs_qgroup_limit_item *qgroup_limit;
  431. struct extent_buffer *leaf;
  432. struct btrfs_key key;
  433. path = btrfs_alloc_path();
  434. if (!path)
  435. return -ENOMEM;
  436. key.objectid = 0;
  437. key.type = BTRFS_QGROUP_INFO_KEY;
  438. key.offset = qgroupid;
  439. ret = btrfs_insert_empty_item(trans, quota_root, path, &key,
  440. sizeof(*qgroup_info));
  441. if (ret)
  442. goto out;
  443. leaf = path->nodes[0];
  444. qgroup_info = btrfs_item_ptr(leaf, path->slots[0],
  445. struct btrfs_qgroup_info_item);
  446. btrfs_set_qgroup_info_generation(leaf, qgroup_info, trans->transid);
  447. btrfs_set_qgroup_info_rfer(leaf, qgroup_info, 0);
  448. btrfs_set_qgroup_info_rfer_cmpr(leaf, qgroup_info, 0);
  449. btrfs_set_qgroup_info_excl(leaf, qgroup_info, 0);
  450. btrfs_set_qgroup_info_excl_cmpr(leaf, qgroup_info, 0);
  451. btrfs_mark_buffer_dirty(leaf);
  452. btrfs_release_path(path);
  453. key.type = BTRFS_QGROUP_LIMIT_KEY;
  454. ret = btrfs_insert_empty_item(trans, quota_root, path, &key,
  455. sizeof(*qgroup_limit));
  456. if (ret)
  457. goto out;
  458. leaf = path->nodes[0];
  459. qgroup_limit = btrfs_item_ptr(leaf, path->slots[0],
  460. struct btrfs_qgroup_limit_item);
  461. btrfs_set_qgroup_limit_flags(leaf, qgroup_limit, 0);
  462. btrfs_set_qgroup_limit_max_rfer(leaf, qgroup_limit, 0);
  463. btrfs_set_qgroup_limit_max_excl(leaf, qgroup_limit, 0);
  464. btrfs_set_qgroup_limit_rsv_rfer(leaf, qgroup_limit, 0);
  465. btrfs_set_qgroup_limit_rsv_excl(leaf, qgroup_limit, 0);
  466. btrfs_mark_buffer_dirty(leaf);
  467. ret = 0;
  468. out:
  469. btrfs_free_path(path);
  470. return ret;
  471. }
  472. static int del_qgroup_item(struct btrfs_trans_handle *trans,
  473. struct btrfs_root *quota_root, u64 qgroupid)
  474. {
  475. int ret;
  476. struct btrfs_path *path;
  477. struct btrfs_key key;
  478. path = btrfs_alloc_path();
  479. if (!path)
  480. return -ENOMEM;
  481. key.objectid = 0;
  482. key.type = BTRFS_QGROUP_INFO_KEY;
  483. key.offset = qgroupid;
  484. ret = btrfs_search_slot(trans, quota_root, &key, path, -1, 1);
  485. if (ret < 0)
  486. goto out;
  487. if (ret > 0) {
  488. ret = -ENOENT;
  489. goto out;
  490. }
  491. ret = btrfs_del_item(trans, quota_root, path);
  492. if (ret)
  493. goto out;
  494. btrfs_release_path(path);
  495. key.type = BTRFS_QGROUP_LIMIT_KEY;
  496. ret = btrfs_search_slot(trans, quota_root, &key, path, -1, 1);
  497. if (ret < 0)
  498. goto out;
  499. if (ret > 0) {
  500. ret = -ENOENT;
  501. goto out;
  502. }
  503. ret = btrfs_del_item(trans, quota_root, path);
  504. out:
  505. btrfs_free_path(path);
  506. return ret;
  507. }
  508. static int update_qgroup_limit_item(struct btrfs_trans_handle *trans,
  509. struct btrfs_root *root, u64 qgroupid,
  510. u64 flags, u64 max_rfer, u64 max_excl,
  511. u64 rsv_rfer, u64 rsv_excl)
  512. {
  513. struct btrfs_path *path;
  514. struct btrfs_key key;
  515. struct extent_buffer *l;
  516. struct btrfs_qgroup_limit_item *qgroup_limit;
  517. int ret;
  518. int slot;
  519. key.objectid = 0;
  520. key.type = BTRFS_QGROUP_LIMIT_KEY;
  521. key.offset = qgroupid;
  522. path = btrfs_alloc_path();
  523. BUG_ON(!path);
  524. ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
  525. if (ret > 0)
  526. ret = -ENOENT;
  527. if (ret)
  528. goto out;
  529. l = path->nodes[0];
  530. slot = path->slots[0];
  531. qgroup_limit = btrfs_item_ptr(l, path->slots[0],
  532. struct btrfs_qgroup_limit_item);
  533. btrfs_set_qgroup_limit_flags(l, qgroup_limit, flags);
  534. btrfs_set_qgroup_limit_max_rfer(l, qgroup_limit, max_rfer);
  535. btrfs_set_qgroup_limit_max_excl(l, qgroup_limit, max_excl);
  536. btrfs_set_qgroup_limit_rsv_rfer(l, qgroup_limit, rsv_rfer);
  537. btrfs_set_qgroup_limit_rsv_excl(l, qgroup_limit, rsv_excl);
  538. btrfs_mark_buffer_dirty(l);
  539. out:
  540. btrfs_free_path(path);
  541. return ret;
  542. }
  543. static int update_qgroup_info_item(struct btrfs_trans_handle *trans,
  544. struct btrfs_root *root,
  545. struct btrfs_qgroup *qgroup)
  546. {
  547. struct btrfs_path *path;
  548. struct btrfs_key key;
  549. struct extent_buffer *l;
  550. struct btrfs_qgroup_info_item *qgroup_info;
  551. int ret;
  552. int slot;
  553. key.objectid = 0;
  554. key.type = BTRFS_QGROUP_INFO_KEY;
  555. key.offset = qgroup->qgroupid;
  556. path = btrfs_alloc_path();
  557. BUG_ON(!path);
  558. ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
  559. if (ret > 0)
  560. ret = -ENOENT;
  561. if (ret)
  562. goto out;
  563. l = path->nodes[0];
  564. slot = path->slots[0];
  565. qgroup_info = btrfs_item_ptr(l, path->slots[0],
  566. struct btrfs_qgroup_info_item);
  567. btrfs_set_qgroup_info_generation(l, qgroup_info, trans->transid);
  568. btrfs_set_qgroup_info_rfer(l, qgroup_info, qgroup->rfer);
  569. btrfs_set_qgroup_info_rfer_cmpr(l, qgroup_info, qgroup->rfer_cmpr);
  570. btrfs_set_qgroup_info_excl(l, qgroup_info, qgroup->excl);
  571. btrfs_set_qgroup_info_excl_cmpr(l, qgroup_info, qgroup->excl_cmpr);
  572. btrfs_mark_buffer_dirty(l);
  573. out:
  574. btrfs_free_path(path);
  575. return ret;
  576. }
  577. static int update_qgroup_status_item(struct btrfs_trans_handle *trans,
  578. struct btrfs_fs_info *fs_info,
  579. struct btrfs_root *root)
  580. {
  581. struct btrfs_path *path;
  582. struct btrfs_key key;
  583. struct extent_buffer *l;
  584. struct btrfs_qgroup_status_item *ptr;
  585. int ret;
  586. int slot;
  587. key.objectid = 0;
  588. key.type = BTRFS_QGROUP_STATUS_KEY;
  589. key.offset = 0;
  590. path = btrfs_alloc_path();
  591. BUG_ON(!path);
  592. ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
  593. if (ret > 0)
  594. ret = -ENOENT;
  595. if (ret)
  596. goto out;
  597. l = path->nodes[0];
  598. slot = path->slots[0];
  599. ptr = btrfs_item_ptr(l, slot, struct btrfs_qgroup_status_item);
  600. btrfs_set_qgroup_status_flags(l, ptr, fs_info->qgroup_flags);
  601. btrfs_set_qgroup_status_generation(l, ptr, trans->transid);
  602. /* XXX scan */
  603. btrfs_mark_buffer_dirty(l);
  604. out:
  605. btrfs_free_path(path);
  606. return ret;
  607. }
  608. /*
  609. * called with qgroup_lock held
  610. */
  611. static int btrfs_clean_quota_tree(struct btrfs_trans_handle *trans,
  612. struct btrfs_root *root)
  613. {
  614. struct btrfs_path *path;
  615. struct btrfs_key key;
  616. int ret;
  617. if (!root)
  618. return -EINVAL;
  619. path = btrfs_alloc_path();
  620. if (!path)
  621. return -ENOMEM;
  622. while (1) {
  623. key.objectid = 0;
  624. key.offset = 0;
  625. key.type = 0;
  626. path->leave_spinning = 1;
  627. ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
  628. if (ret > 0) {
  629. if (path->slots[0] == 0)
  630. break;
  631. path->slots[0]--;
  632. } else if (ret < 0) {
  633. break;
  634. }
  635. ret = btrfs_del_item(trans, root, path);
  636. if (ret)
  637. goto out;
  638. btrfs_release_path(path);
  639. }
  640. ret = 0;
  641. out:
  642. root->fs_info->pending_quota_state = 0;
  643. btrfs_free_path(path);
  644. return ret;
  645. }
  646. int btrfs_quota_enable(struct btrfs_trans_handle *trans,
  647. struct btrfs_fs_info *fs_info)
  648. {
  649. struct btrfs_root *quota_root;
  650. struct btrfs_path *path = NULL;
  651. struct btrfs_qgroup_status_item *ptr;
  652. struct extent_buffer *leaf;
  653. struct btrfs_key key;
  654. int ret = 0;
  655. spin_lock(&fs_info->qgroup_lock);
  656. if (fs_info->quota_root) {
  657. fs_info->pending_quota_state = 1;
  658. spin_unlock(&fs_info->qgroup_lock);
  659. goto out;
  660. }
  661. spin_unlock(&fs_info->qgroup_lock);
  662. /*
  663. * initially create the quota tree
  664. */
  665. quota_root = btrfs_create_tree(trans, fs_info,
  666. BTRFS_QUOTA_TREE_OBJECTID);
  667. if (IS_ERR(quota_root)) {
  668. ret = PTR_ERR(quota_root);
  669. goto out;
  670. }
  671. path = btrfs_alloc_path();
  672. if (!path) {
  673. ret = -ENOMEM;
  674. goto out_free_root;
  675. }
  676. key.objectid = 0;
  677. key.type = BTRFS_QGROUP_STATUS_KEY;
  678. key.offset = 0;
  679. ret = btrfs_insert_empty_item(trans, quota_root, path, &key,
  680. sizeof(*ptr));
  681. if (ret)
  682. goto out_free_path;
  683. leaf = path->nodes[0];
  684. ptr = btrfs_item_ptr(leaf, path->slots[0],
  685. struct btrfs_qgroup_status_item);
  686. btrfs_set_qgroup_status_generation(leaf, ptr, trans->transid);
  687. btrfs_set_qgroup_status_version(leaf, ptr, BTRFS_QGROUP_STATUS_VERSION);
  688. fs_info->qgroup_flags = BTRFS_QGROUP_STATUS_FLAG_ON |
  689. BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  690. btrfs_set_qgroup_status_flags(leaf, ptr, fs_info->qgroup_flags);
  691. btrfs_set_qgroup_status_scan(leaf, ptr, 0);
  692. btrfs_mark_buffer_dirty(leaf);
  693. spin_lock(&fs_info->qgroup_lock);
  694. fs_info->quota_root = quota_root;
  695. fs_info->pending_quota_state = 1;
  696. spin_unlock(&fs_info->qgroup_lock);
  697. out_free_path:
  698. btrfs_free_path(path);
  699. out_free_root:
  700. if (ret) {
  701. free_extent_buffer(quota_root->node);
  702. free_extent_buffer(quota_root->commit_root);
  703. kfree(quota_root);
  704. }
  705. out:
  706. return ret;
  707. }
  708. int btrfs_quota_disable(struct btrfs_trans_handle *trans,
  709. struct btrfs_fs_info *fs_info)
  710. {
  711. struct btrfs_root *tree_root = fs_info->tree_root;
  712. struct btrfs_root *quota_root;
  713. int ret = 0;
  714. spin_lock(&fs_info->qgroup_lock);
  715. fs_info->quota_enabled = 0;
  716. fs_info->pending_quota_state = 0;
  717. quota_root = fs_info->quota_root;
  718. fs_info->quota_root = NULL;
  719. btrfs_free_qgroup_config(fs_info);
  720. spin_unlock(&fs_info->qgroup_lock);
  721. if (!quota_root)
  722. return -EINVAL;
  723. ret = btrfs_clean_quota_tree(trans, quota_root);
  724. if (ret)
  725. goto out;
  726. ret = btrfs_del_root(trans, tree_root, &quota_root->root_key);
  727. if (ret)
  728. goto out;
  729. list_del(&quota_root->dirty_list);
  730. btrfs_tree_lock(quota_root->node);
  731. clean_tree_block(trans, tree_root, quota_root->node);
  732. btrfs_tree_unlock(quota_root->node);
  733. btrfs_free_tree_block(trans, quota_root, quota_root->node, 0, 1);
  734. free_extent_buffer(quota_root->node);
  735. free_extent_buffer(quota_root->commit_root);
  736. kfree(quota_root);
  737. out:
  738. return ret;
  739. }
  740. int btrfs_quota_rescan(struct btrfs_fs_info *fs_info)
  741. {
  742. /* FIXME */
  743. return 0;
  744. }
  745. int btrfs_add_qgroup_relation(struct btrfs_trans_handle *trans,
  746. struct btrfs_fs_info *fs_info, u64 src, u64 dst)
  747. {
  748. struct btrfs_root *quota_root;
  749. int ret = 0;
  750. quota_root = fs_info->quota_root;
  751. if (!quota_root)
  752. return -EINVAL;
  753. ret = add_qgroup_relation_item(trans, quota_root, src, dst);
  754. if (ret)
  755. return ret;
  756. ret = add_qgroup_relation_item(trans, quota_root, dst, src);
  757. if (ret) {
  758. del_qgroup_relation_item(trans, quota_root, src, dst);
  759. return ret;
  760. }
  761. spin_lock(&fs_info->qgroup_lock);
  762. ret = add_relation_rb(quota_root->fs_info, src, dst);
  763. spin_unlock(&fs_info->qgroup_lock);
  764. return ret;
  765. }
  766. int btrfs_del_qgroup_relation(struct btrfs_trans_handle *trans,
  767. struct btrfs_fs_info *fs_info, u64 src, u64 dst)
  768. {
  769. struct btrfs_root *quota_root;
  770. int ret = 0;
  771. int err;
  772. quota_root = fs_info->quota_root;
  773. if (!quota_root)
  774. return -EINVAL;
  775. ret = del_qgroup_relation_item(trans, quota_root, src, dst);
  776. err = del_qgroup_relation_item(trans, quota_root, dst, src);
  777. if (err && !ret)
  778. ret = err;
  779. spin_lock(&fs_info->qgroup_lock);
  780. del_relation_rb(fs_info, src, dst);
  781. spin_unlock(&fs_info->qgroup_lock);
  782. return ret;
  783. }
  784. int btrfs_create_qgroup(struct btrfs_trans_handle *trans,
  785. struct btrfs_fs_info *fs_info, u64 qgroupid, char *name)
  786. {
  787. struct btrfs_root *quota_root;
  788. struct btrfs_qgroup *qgroup;
  789. int ret = 0;
  790. quota_root = fs_info->quota_root;
  791. if (!quota_root)
  792. return -EINVAL;
  793. ret = add_qgroup_item(trans, quota_root, qgroupid);
  794. spin_lock(&fs_info->qgroup_lock);
  795. qgroup = add_qgroup_rb(fs_info, qgroupid);
  796. spin_unlock(&fs_info->qgroup_lock);
  797. if (IS_ERR(qgroup))
  798. ret = PTR_ERR(qgroup);
  799. return ret;
  800. }
  801. int btrfs_remove_qgroup(struct btrfs_trans_handle *trans,
  802. struct btrfs_fs_info *fs_info, u64 qgroupid)
  803. {
  804. struct btrfs_root *quota_root;
  805. int ret = 0;
  806. quota_root = fs_info->quota_root;
  807. if (!quota_root)
  808. return -EINVAL;
  809. ret = del_qgroup_item(trans, quota_root, qgroupid);
  810. spin_lock(&fs_info->qgroup_lock);
  811. del_qgroup_rb(quota_root->fs_info, qgroupid);
  812. spin_unlock(&fs_info->qgroup_lock);
  813. return ret;
  814. }
  815. int btrfs_limit_qgroup(struct btrfs_trans_handle *trans,
  816. struct btrfs_fs_info *fs_info, u64 qgroupid,
  817. struct btrfs_qgroup_limit *limit)
  818. {
  819. struct btrfs_root *quota_root = fs_info->quota_root;
  820. struct btrfs_qgroup *qgroup;
  821. int ret = 0;
  822. if (!quota_root)
  823. return -EINVAL;
  824. ret = update_qgroup_limit_item(trans, quota_root, qgroupid,
  825. limit->flags, limit->max_rfer,
  826. limit->max_excl, limit->rsv_rfer,
  827. limit->rsv_excl);
  828. if (ret) {
  829. fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  830. printk(KERN_INFO "unable to update quota limit for %llu\n",
  831. (unsigned long long)qgroupid);
  832. }
  833. spin_lock(&fs_info->qgroup_lock);
  834. qgroup = find_qgroup_rb(fs_info, qgroupid);
  835. if (!qgroup) {
  836. ret = -ENOENT;
  837. goto unlock;
  838. }
  839. qgroup->lim_flags = limit->flags;
  840. qgroup->max_rfer = limit->max_rfer;
  841. qgroup->max_excl = limit->max_excl;
  842. qgroup->rsv_rfer = limit->rsv_rfer;
  843. qgroup->rsv_excl = limit->rsv_excl;
  844. unlock:
  845. spin_unlock(&fs_info->qgroup_lock);
  846. return ret;
  847. }
  848. static void qgroup_dirty(struct btrfs_fs_info *fs_info,
  849. struct btrfs_qgroup *qgroup)
  850. {
  851. if (list_empty(&qgroup->dirty))
  852. list_add(&qgroup->dirty, &fs_info->dirty_qgroups);
  853. }
  854. /*
  855. * btrfs_qgroup_record_ref is called when the ref is added or deleted. it puts
  856. * the modification into a list that's later used by btrfs_end_transaction to
  857. * pass the recorded modifications on to btrfs_qgroup_account_ref.
  858. */
  859. int btrfs_qgroup_record_ref(struct btrfs_trans_handle *trans,
  860. struct btrfs_delayed_ref_node *node,
  861. struct btrfs_delayed_extent_op *extent_op)
  862. {
  863. struct qgroup_update *u;
  864. BUG_ON(!trans->delayed_ref_elem.seq);
  865. u = kmalloc(sizeof(*u), GFP_NOFS);
  866. if (!u)
  867. return -ENOMEM;
  868. u->node = node;
  869. u->extent_op = extent_op;
  870. list_add_tail(&u->list, &trans->qgroup_ref_list);
  871. return 0;
  872. }
  873. /*
  874. * btrfs_qgroup_account_ref is called for every ref that is added to or deleted
  875. * from the fs. First, all roots referencing the extent are searched, and
  876. * then the space is accounted accordingly to the different roots. The
  877. * accounting algorithm works in 3 steps documented inline.
  878. */
  879. int btrfs_qgroup_account_ref(struct btrfs_trans_handle *trans,
  880. struct btrfs_fs_info *fs_info,
  881. struct btrfs_delayed_ref_node *node,
  882. struct btrfs_delayed_extent_op *extent_op)
  883. {
  884. struct btrfs_key ins;
  885. struct btrfs_root *quota_root;
  886. u64 ref_root;
  887. struct btrfs_qgroup *qgroup;
  888. struct ulist_node *unode;
  889. struct ulist *roots = NULL;
  890. struct ulist *tmp = NULL;
  891. struct ulist_iterator uiter;
  892. u64 seq;
  893. int ret = 0;
  894. int sgn;
  895. if (!fs_info->quota_enabled)
  896. return 0;
  897. BUG_ON(!fs_info->quota_root);
  898. ins.objectid = node->bytenr;
  899. ins.offset = node->num_bytes;
  900. ins.type = BTRFS_EXTENT_ITEM_KEY;
  901. if (node->type == BTRFS_TREE_BLOCK_REF_KEY ||
  902. node->type == BTRFS_SHARED_BLOCK_REF_KEY) {
  903. struct btrfs_delayed_tree_ref *ref;
  904. ref = btrfs_delayed_node_to_tree_ref(node);
  905. ref_root = ref->root;
  906. } else if (node->type == BTRFS_EXTENT_DATA_REF_KEY ||
  907. node->type == BTRFS_SHARED_DATA_REF_KEY) {
  908. struct btrfs_delayed_data_ref *ref;
  909. ref = btrfs_delayed_node_to_data_ref(node);
  910. ref_root = ref->root;
  911. } else {
  912. BUG();
  913. }
  914. if (!is_fstree(ref_root)) {
  915. /*
  916. * non-fs-trees are not being accounted
  917. */
  918. return 0;
  919. }
  920. switch (node->action) {
  921. case BTRFS_ADD_DELAYED_REF:
  922. case BTRFS_ADD_DELAYED_EXTENT:
  923. sgn = 1;
  924. break;
  925. case BTRFS_DROP_DELAYED_REF:
  926. sgn = -1;
  927. break;
  928. case BTRFS_UPDATE_DELAYED_HEAD:
  929. return 0;
  930. default:
  931. BUG();
  932. }
  933. /*
  934. * the delayed ref sequence number we pass depends on the direction of
  935. * the operation. for add operations, we pass (node->seq - 1) to skip
  936. * the delayed ref's current sequence number, because we need the state
  937. * of the tree before the add operation. for delete operations, we pass
  938. * (node->seq) to include the delayed ref's current sequence number,
  939. * because we need the state of the tree after the delete operation.
  940. */
  941. ret = btrfs_find_all_roots(trans, fs_info, node->bytenr,
  942. sgn > 0 ? node->seq - 1 : node->seq, &roots);
  943. if (ret < 0)
  944. goto out;
  945. spin_lock(&fs_info->qgroup_lock);
  946. quota_root = fs_info->quota_root;
  947. if (!quota_root)
  948. goto unlock;
  949. qgroup = find_qgroup_rb(fs_info, ref_root);
  950. if (!qgroup)
  951. goto unlock;
  952. /*
  953. * step 1: for each old ref, visit all nodes once and inc refcnt
  954. */
  955. tmp = ulist_alloc(GFP_ATOMIC);
  956. if (!tmp) {
  957. ret = -ENOMEM;
  958. goto unlock;
  959. }
  960. seq = fs_info->qgroup_seq;
  961. fs_info->qgroup_seq += roots->nnodes + 1; /* max refcnt */
  962. ULIST_ITER_INIT(&uiter);
  963. while ((unode = ulist_next(roots, &uiter))) {
  964. struct ulist_node *tmp_unode;
  965. struct ulist_iterator tmp_uiter;
  966. struct btrfs_qgroup *qg;
  967. qg = find_qgroup_rb(fs_info, unode->val);
  968. if (!qg)
  969. continue;
  970. ulist_reinit(tmp);
  971. /* XXX id not needed */
  972. ulist_add(tmp, qg->qgroupid, (u64)(uintptr_t)qg, GFP_ATOMIC);
  973. ULIST_ITER_INIT(&tmp_uiter);
  974. while ((tmp_unode = ulist_next(tmp, &tmp_uiter))) {
  975. struct btrfs_qgroup_list *glist;
  976. qg = (struct btrfs_qgroup *)(uintptr_t)tmp_unode->aux;
  977. if (qg->refcnt < seq)
  978. qg->refcnt = seq + 1;
  979. else
  980. ++qg->refcnt;
  981. list_for_each_entry(glist, &qg->groups, next_group) {
  982. ulist_add(tmp, glist->group->qgroupid,
  983. (u64)(uintptr_t)glist->group,
  984. GFP_ATOMIC);
  985. }
  986. }
  987. }
  988. /*
  989. * step 2: walk from the new root
  990. */
  991. ulist_reinit(tmp);
  992. ulist_add(tmp, qgroup->qgroupid, (uintptr_t)qgroup, GFP_ATOMIC);
  993. ULIST_ITER_INIT(&uiter);
  994. while ((unode = ulist_next(tmp, &uiter))) {
  995. struct btrfs_qgroup *qg;
  996. struct btrfs_qgroup_list *glist;
  997. qg = (struct btrfs_qgroup *)(uintptr_t)unode->aux;
  998. if (qg->refcnt < seq) {
  999. /* not visited by step 1 */
  1000. qg->rfer += sgn * node->num_bytes;
  1001. qg->rfer_cmpr += sgn * node->num_bytes;
  1002. if (roots->nnodes == 0) {
  1003. qg->excl += sgn * node->num_bytes;
  1004. qg->excl_cmpr += sgn * node->num_bytes;
  1005. }
  1006. qgroup_dirty(fs_info, qg);
  1007. }
  1008. WARN_ON(qg->tag >= seq);
  1009. qg->tag = seq;
  1010. list_for_each_entry(glist, &qg->groups, next_group) {
  1011. ulist_add(tmp, glist->group->qgroupid,
  1012. (uintptr_t)glist->group, GFP_ATOMIC);
  1013. }
  1014. }
  1015. /*
  1016. * step 3: walk again from old refs
  1017. */
  1018. ULIST_ITER_INIT(&uiter);
  1019. while ((unode = ulist_next(roots, &uiter))) {
  1020. struct btrfs_qgroup *qg;
  1021. struct ulist_node *tmp_unode;
  1022. struct ulist_iterator tmp_uiter;
  1023. qg = find_qgroup_rb(fs_info, unode->val);
  1024. if (!qg)
  1025. continue;
  1026. ulist_reinit(tmp);
  1027. ulist_add(tmp, qg->qgroupid, (uintptr_t)qg, GFP_ATOMIC);
  1028. ULIST_ITER_INIT(&tmp_uiter);
  1029. while ((tmp_unode = ulist_next(tmp, &tmp_uiter))) {
  1030. struct btrfs_qgroup_list *glist;
  1031. qg = (struct btrfs_qgroup *)(uintptr_t)tmp_unode->aux;
  1032. if (qg->tag == seq)
  1033. continue;
  1034. if (qg->refcnt - seq == roots->nnodes) {
  1035. qg->excl -= sgn * node->num_bytes;
  1036. qg->excl_cmpr -= sgn * node->num_bytes;
  1037. qgroup_dirty(fs_info, qg);
  1038. }
  1039. list_for_each_entry(glist, &qg->groups, next_group) {
  1040. ulist_add(tmp, glist->group->qgroupid,
  1041. (uintptr_t)glist->group,
  1042. GFP_ATOMIC);
  1043. }
  1044. }
  1045. }
  1046. ret = 0;
  1047. unlock:
  1048. spin_unlock(&fs_info->qgroup_lock);
  1049. out:
  1050. ulist_free(roots);
  1051. ulist_free(tmp);
  1052. return ret;
  1053. }
  1054. /*
  1055. * called from commit_transaction. Writes all changed qgroups to disk.
  1056. */
  1057. int btrfs_run_qgroups(struct btrfs_trans_handle *trans,
  1058. struct btrfs_fs_info *fs_info)
  1059. {
  1060. struct btrfs_root *quota_root = fs_info->quota_root;
  1061. int ret = 0;
  1062. if (!quota_root)
  1063. goto out;
  1064. fs_info->quota_enabled = fs_info->pending_quota_state;
  1065. spin_lock(&fs_info->qgroup_lock);
  1066. while (!list_empty(&fs_info->dirty_qgroups)) {
  1067. struct btrfs_qgroup *qgroup;
  1068. qgroup = list_first_entry(&fs_info->dirty_qgroups,
  1069. struct btrfs_qgroup, dirty);
  1070. list_del_init(&qgroup->dirty);
  1071. spin_unlock(&fs_info->qgroup_lock);
  1072. ret = update_qgroup_info_item(trans, quota_root, qgroup);
  1073. if (ret)
  1074. fs_info->qgroup_flags |=
  1075. BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  1076. spin_lock(&fs_info->qgroup_lock);
  1077. }
  1078. if (fs_info->quota_enabled)
  1079. fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_ON;
  1080. else
  1081. fs_info->qgroup_flags &= ~BTRFS_QGROUP_STATUS_FLAG_ON;
  1082. spin_unlock(&fs_info->qgroup_lock);
  1083. ret = update_qgroup_status_item(trans, fs_info, quota_root);
  1084. if (ret)
  1085. fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  1086. out:
  1087. return ret;
  1088. }
  1089. /*
  1090. * copy the acounting information between qgroups. This is necessary when a
  1091. * snapshot or a subvolume is created
  1092. */
  1093. int btrfs_qgroup_inherit(struct btrfs_trans_handle *trans,
  1094. struct btrfs_fs_info *fs_info, u64 srcid, u64 objectid,
  1095. struct btrfs_qgroup_inherit *inherit)
  1096. {
  1097. int ret = 0;
  1098. int i;
  1099. u64 *i_qgroups;
  1100. struct btrfs_root *quota_root = fs_info->quota_root;
  1101. struct btrfs_qgroup *srcgroup;
  1102. struct btrfs_qgroup *dstgroup;
  1103. u32 level_size = 0;
  1104. if (!fs_info->quota_enabled)
  1105. return 0;
  1106. if (!quota_root)
  1107. return -EINVAL;
  1108. /*
  1109. * create a tracking group for the subvol itself
  1110. */
  1111. ret = add_qgroup_item(trans, quota_root, objectid);
  1112. if (ret)
  1113. goto out;
  1114. if (inherit && inherit->flags & BTRFS_QGROUP_INHERIT_SET_LIMITS) {
  1115. ret = update_qgroup_limit_item(trans, quota_root, objectid,
  1116. inherit->lim.flags,
  1117. inherit->lim.max_rfer,
  1118. inherit->lim.max_excl,
  1119. inherit->lim.rsv_rfer,
  1120. inherit->lim.rsv_excl);
  1121. if (ret)
  1122. goto out;
  1123. }
  1124. if (srcid) {
  1125. struct btrfs_root *srcroot;
  1126. struct btrfs_key srckey;
  1127. int srcroot_level;
  1128. srckey.objectid = srcid;
  1129. srckey.type = BTRFS_ROOT_ITEM_KEY;
  1130. srckey.offset = (u64)-1;
  1131. srcroot = btrfs_read_fs_root_no_name(fs_info, &srckey);
  1132. if (IS_ERR(srcroot)) {
  1133. ret = PTR_ERR(srcroot);
  1134. goto out;
  1135. }
  1136. rcu_read_lock();
  1137. srcroot_level = btrfs_header_level(srcroot->node);
  1138. level_size = btrfs_level_size(srcroot, srcroot_level);
  1139. rcu_read_unlock();
  1140. }
  1141. /*
  1142. * add qgroup to all inherited groups
  1143. */
  1144. if (inherit) {
  1145. i_qgroups = (u64 *)(inherit + 1);
  1146. for (i = 0; i < inherit->num_qgroups; ++i) {
  1147. ret = add_qgroup_relation_item(trans, quota_root,
  1148. objectid, *i_qgroups);
  1149. if (ret)
  1150. goto out;
  1151. ret = add_qgroup_relation_item(trans, quota_root,
  1152. *i_qgroups, objectid);
  1153. if (ret)
  1154. goto out;
  1155. ++i_qgroups;
  1156. }
  1157. }
  1158. spin_lock(&fs_info->qgroup_lock);
  1159. dstgroup = add_qgroup_rb(fs_info, objectid);
  1160. if (IS_ERR(dstgroup)) {
  1161. ret = PTR_ERR(dstgroup);
  1162. goto unlock;
  1163. }
  1164. if (srcid) {
  1165. srcgroup = find_qgroup_rb(fs_info, srcid);
  1166. if (!srcgroup)
  1167. goto unlock;
  1168. dstgroup->rfer = srcgroup->rfer - level_size;
  1169. dstgroup->rfer_cmpr = srcgroup->rfer_cmpr - level_size;
  1170. srcgroup->excl = level_size;
  1171. srcgroup->excl_cmpr = level_size;
  1172. qgroup_dirty(fs_info, dstgroup);
  1173. qgroup_dirty(fs_info, srcgroup);
  1174. }
  1175. if (!inherit)
  1176. goto unlock;
  1177. i_qgroups = (u64 *)(inherit + 1);
  1178. for (i = 0; i < inherit->num_qgroups; ++i) {
  1179. ret = add_relation_rb(quota_root->fs_info, objectid,
  1180. *i_qgroups);
  1181. if (ret)
  1182. goto unlock;
  1183. ++i_qgroups;
  1184. }
  1185. for (i = 0; i < inherit->num_ref_copies; ++i) {
  1186. struct btrfs_qgroup *src;
  1187. struct btrfs_qgroup *dst;
  1188. src = find_qgroup_rb(fs_info, i_qgroups[0]);
  1189. dst = find_qgroup_rb(fs_info, i_qgroups[1]);
  1190. if (!src || !dst) {
  1191. ret = -EINVAL;
  1192. goto unlock;
  1193. }
  1194. dst->rfer = src->rfer - level_size;
  1195. dst->rfer_cmpr = src->rfer_cmpr - level_size;
  1196. i_qgroups += 2;
  1197. }
  1198. for (i = 0; i < inherit->num_excl_copies; ++i) {
  1199. struct btrfs_qgroup *src;
  1200. struct btrfs_qgroup *dst;
  1201. src = find_qgroup_rb(fs_info, i_qgroups[0]);
  1202. dst = find_qgroup_rb(fs_info, i_qgroups[1]);
  1203. if (!src || !dst) {
  1204. ret = -EINVAL;
  1205. goto unlock;
  1206. }
  1207. dst->excl = src->excl + level_size;
  1208. dst->excl_cmpr = src->excl_cmpr + level_size;
  1209. i_qgroups += 2;
  1210. }
  1211. unlock:
  1212. spin_unlock(&fs_info->qgroup_lock);
  1213. out:
  1214. return ret;
  1215. }
  1216. /*
  1217. * reserve some space for a qgroup and all its parents. The reservation takes
  1218. * place with start_transaction or dealloc_reserve, similar to ENOSPC
  1219. * accounting. If not enough space is available, EDQUOT is returned.
  1220. * We assume that the requested space is new for all qgroups.
  1221. */
  1222. int btrfs_qgroup_reserve(struct btrfs_root *root, u64 num_bytes)
  1223. {
  1224. struct btrfs_root *quota_root;
  1225. struct btrfs_qgroup *qgroup;
  1226. struct btrfs_fs_info *fs_info = root->fs_info;
  1227. u64 ref_root = root->root_key.objectid;
  1228. int ret = 0;
  1229. struct ulist *ulist = NULL;
  1230. struct ulist_node *unode;
  1231. struct ulist_iterator uiter;
  1232. if (!is_fstree(ref_root))
  1233. return 0;
  1234. if (num_bytes == 0)
  1235. return 0;
  1236. spin_lock(&fs_info->qgroup_lock);
  1237. quota_root = fs_info->quota_root;
  1238. if (!quota_root)
  1239. goto out;
  1240. qgroup = find_qgroup_rb(fs_info, ref_root);
  1241. if (!qgroup)
  1242. goto out;
  1243. /*
  1244. * in a first step, we check all affected qgroups if any limits would
  1245. * be exceeded
  1246. */
  1247. ulist = ulist_alloc(GFP_ATOMIC);
  1248. if (!ulist) {
  1249. ret = -ENOMEM;
  1250. goto out;
  1251. }
  1252. ulist_add(ulist, qgroup->qgroupid, (uintptr_t)qgroup, GFP_ATOMIC);
  1253. ULIST_ITER_INIT(&uiter);
  1254. while ((unode = ulist_next(ulist, &uiter))) {
  1255. struct btrfs_qgroup *qg;
  1256. struct btrfs_qgroup_list *glist;
  1257. qg = (struct btrfs_qgroup *)(uintptr_t)unode->aux;
  1258. if ((qg->lim_flags & BTRFS_QGROUP_LIMIT_MAX_RFER) &&
  1259. qg->reserved + qg->rfer + num_bytes >
  1260. qg->max_rfer)
  1261. ret = -EDQUOT;
  1262. if ((qg->lim_flags & BTRFS_QGROUP_LIMIT_MAX_EXCL) &&
  1263. qg->reserved + qg->excl + num_bytes >
  1264. qg->max_excl)
  1265. ret = -EDQUOT;
  1266. list_for_each_entry(glist, &qg->groups, next_group) {
  1267. ulist_add(ulist, glist->group->qgroupid,
  1268. (uintptr_t)glist->group, GFP_ATOMIC);
  1269. }
  1270. }
  1271. if (ret)
  1272. goto out;
  1273. /*
  1274. * no limits exceeded, now record the reservation into all qgroups
  1275. */
  1276. ULIST_ITER_INIT(&uiter);
  1277. while ((unode = ulist_next(ulist, &uiter))) {
  1278. struct btrfs_qgroup *qg;
  1279. qg = (struct btrfs_qgroup *)(uintptr_t)unode->aux;
  1280. qg->reserved += num_bytes;
  1281. }
  1282. out:
  1283. spin_unlock(&fs_info->qgroup_lock);
  1284. ulist_free(ulist);
  1285. return ret;
  1286. }
  1287. void btrfs_qgroup_free(struct btrfs_root *root, u64 num_bytes)
  1288. {
  1289. struct btrfs_root *quota_root;
  1290. struct btrfs_qgroup *qgroup;
  1291. struct btrfs_fs_info *fs_info = root->fs_info;
  1292. struct ulist *ulist = NULL;
  1293. struct ulist_node *unode;
  1294. struct ulist_iterator uiter;
  1295. u64 ref_root = root->root_key.objectid;
  1296. if (!is_fstree(ref_root))
  1297. return;
  1298. if (num_bytes == 0)
  1299. return;
  1300. spin_lock(&fs_info->qgroup_lock);
  1301. quota_root = fs_info->quota_root;
  1302. if (!quota_root)
  1303. goto out;
  1304. qgroup = find_qgroup_rb(fs_info, ref_root);
  1305. if (!qgroup)
  1306. goto out;
  1307. ulist = ulist_alloc(GFP_ATOMIC);
  1308. if (!ulist) {
  1309. btrfs_std_error(fs_info, -ENOMEM);
  1310. goto out;
  1311. }
  1312. ulist_add(ulist, qgroup->qgroupid, (uintptr_t)qgroup, GFP_ATOMIC);
  1313. ULIST_ITER_INIT(&uiter);
  1314. while ((unode = ulist_next(ulist, &uiter))) {
  1315. struct btrfs_qgroup *qg;
  1316. struct btrfs_qgroup_list *glist;
  1317. qg = (struct btrfs_qgroup *)(uintptr_t)unode->aux;
  1318. qg->reserved -= num_bytes;
  1319. list_for_each_entry(glist, &qg->groups, next_group) {
  1320. ulist_add(ulist, glist->group->qgroupid,
  1321. (uintptr_t)glist->group, GFP_ATOMIC);
  1322. }
  1323. }
  1324. out:
  1325. spin_unlock(&fs_info->qgroup_lock);
  1326. ulist_free(ulist);
  1327. }
  1328. void assert_qgroups_uptodate(struct btrfs_trans_handle *trans)
  1329. {
  1330. if (list_empty(&trans->qgroup_ref_list) && !trans->delayed_ref_elem.seq)
  1331. return;
  1332. printk(KERN_ERR "btrfs: qgroups not uptodate in trans handle %p: list is%s empty, seq is %llu\n",
  1333. trans, list_empty(&trans->qgroup_ref_list) ? "" : " not",
  1334. trans->delayed_ref_elem.seq);
  1335. BUG();
  1336. }