qgroup.c 37 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571
  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. return -ENOMEM;
  674. key.objectid = 0;
  675. key.type = BTRFS_QGROUP_STATUS_KEY;
  676. key.offset = 0;
  677. ret = btrfs_insert_empty_item(trans, quota_root, path, &key,
  678. sizeof(*ptr));
  679. if (ret)
  680. goto out;
  681. leaf = path->nodes[0];
  682. ptr = btrfs_item_ptr(leaf, path->slots[0],
  683. struct btrfs_qgroup_status_item);
  684. btrfs_set_qgroup_status_generation(leaf, ptr, trans->transid);
  685. btrfs_set_qgroup_status_version(leaf, ptr, BTRFS_QGROUP_STATUS_VERSION);
  686. fs_info->qgroup_flags = BTRFS_QGROUP_STATUS_FLAG_ON |
  687. BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  688. btrfs_set_qgroup_status_flags(leaf, ptr, fs_info->qgroup_flags);
  689. btrfs_set_qgroup_status_scan(leaf, ptr, 0);
  690. btrfs_mark_buffer_dirty(leaf);
  691. spin_lock(&fs_info->qgroup_lock);
  692. fs_info->quota_root = quota_root;
  693. fs_info->pending_quota_state = 1;
  694. spin_unlock(&fs_info->qgroup_lock);
  695. out:
  696. btrfs_free_path(path);
  697. return ret;
  698. }
  699. int btrfs_quota_disable(struct btrfs_trans_handle *trans,
  700. struct btrfs_fs_info *fs_info)
  701. {
  702. struct btrfs_root *tree_root = fs_info->tree_root;
  703. struct btrfs_root *quota_root;
  704. int ret = 0;
  705. spin_lock(&fs_info->qgroup_lock);
  706. fs_info->quota_enabled = 0;
  707. fs_info->pending_quota_state = 0;
  708. quota_root = fs_info->quota_root;
  709. fs_info->quota_root = NULL;
  710. btrfs_free_qgroup_config(fs_info);
  711. spin_unlock(&fs_info->qgroup_lock);
  712. if (!quota_root)
  713. return -EINVAL;
  714. ret = btrfs_clean_quota_tree(trans, quota_root);
  715. if (ret)
  716. goto out;
  717. ret = btrfs_del_root(trans, tree_root, &quota_root->root_key);
  718. if (ret)
  719. goto out;
  720. list_del(&quota_root->dirty_list);
  721. btrfs_tree_lock(quota_root->node);
  722. clean_tree_block(trans, tree_root, quota_root->node);
  723. btrfs_tree_unlock(quota_root->node);
  724. btrfs_free_tree_block(trans, quota_root, quota_root->node, 0, 1);
  725. free_extent_buffer(quota_root->node);
  726. free_extent_buffer(quota_root->commit_root);
  727. kfree(quota_root);
  728. out:
  729. return ret;
  730. }
  731. int btrfs_quota_rescan(struct btrfs_fs_info *fs_info)
  732. {
  733. /* FIXME */
  734. return 0;
  735. }
  736. int btrfs_add_qgroup_relation(struct btrfs_trans_handle *trans,
  737. struct btrfs_fs_info *fs_info, u64 src, u64 dst)
  738. {
  739. struct btrfs_root *quota_root;
  740. int ret = 0;
  741. quota_root = fs_info->quota_root;
  742. if (!quota_root)
  743. return -EINVAL;
  744. ret = add_qgroup_relation_item(trans, quota_root, src, dst);
  745. if (ret)
  746. return ret;
  747. ret = add_qgroup_relation_item(trans, quota_root, dst, src);
  748. if (ret) {
  749. del_qgroup_relation_item(trans, quota_root, src, dst);
  750. return ret;
  751. }
  752. spin_lock(&fs_info->qgroup_lock);
  753. ret = add_relation_rb(quota_root->fs_info, src, dst);
  754. spin_unlock(&fs_info->qgroup_lock);
  755. return ret;
  756. }
  757. int btrfs_del_qgroup_relation(struct btrfs_trans_handle *trans,
  758. struct btrfs_fs_info *fs_info, u64 src, u64 dst)
  759. {
  760. struct btrfs_root *quota_root;
  761. int ret = 0;
  762. int err;
  763. quota_root = fs_info->quota_root;
  764. if (!quota_root)
  765. return -EINVAL;
  766. ret = del_qgroup_relation_item(trans, quota_root, src, dst);
  767. err = del_qgroup_relation_item(trans, quota_root, dst, src);
  768. if (err && !ret)
  769. ret = err;
  770. spin_lock(&fs_info->qgroup_lock);
  771. del_relation_rb(fs_info, src, dst);
  772. spin_unlock(&fs_info->qgroup_lock);
  773. return ret;
  774. }
  775. int btrfs_create_qgroup(struct btrfs_trans_handle *trans,
  776. struct btrfs_fs_info *fs_info, u64 qgroupid, char *name)
  777. {
  778. struct btrfs_root *quota_root;
  779. struct btrfs_qgroup *qgroup;
  780. int ret = 0;
  781. quota_root = fs_info->quota_root;
  782. if (!quota_root)
  783. return -EINVAL;
  784. ret = add_qgroup_item(trans, quota_root, qgroupid);
  785. spin_lock(&fs_info->qgroup_lock);
  786. qgroup = add_qgroup_rb(fs_info, qgroupid);
  787. spin_unlock(&fs_info->qgroup_lock);
  788. if (IS_ERR(qgroup))
  789. ret = PTR_ERR(qgroup);
  790. return ret;
  791. }
  792. int btrfs_remove_qgroup(struct btrfs_trans_handle *trans,
  793. struct btrfs_fs_info *fs_info, u64 qgroupid)
  794. {
  795. struct btrfs_root *quota_root;
  796. int ret = 0;
  797. quota_root = fs_info->quota_root;
  798. if (!quota_root)
  799. return -EINVAL;
  800. ret = del_qgroup_item(trans, quota_root, qgroupid);
  801. spin_lock(&fs_info->qgroup_lock);
  802. del_qgroup_rb(quota_root->fs_info, qgroupid);
  803. spin_unlock(&fs_info->qgroup_lock);
  804. return ret;
  805. }
  806. int btrfs_limit_qgroup(struct btrfs_trans_handle *trans,
  807. struct btrfs_fs_info *fs_info, u64 qgroupid,
  808. struct btrfs_qgroup_limit *limit)
  809. {
  810. struct btrfs_root *quota_root = fs_info->quota_root;
  811. struct btrfs_qgroup *qgroup;
  812. int ret = 0;
  813. if (!quota_root)
  814. return -EINVAL;
  815. ret = update_qgroup_limit_item(trans, quota_root, qgroupid,
  816. limit->flags, limit->max_rfer,
  817. limit->max_excl, limit->rsv_rfer,
  818. limit->rsv_excl);
  819. if (ret) {
  820. fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  821. printk(KERN_INFO "unable to update quota limit for %llu\n",
  822. (unsigned long long)qgroupid);
  823. }
  824. spin_lock(&fs_info->qgroup_lock);
  825. qgroup = find_qgroup_rb(fs_info, qgroupid);
  826. if (!qgroup) {
  827. ret = -ENOENT;
  828. goto unlock;
  829. }
  830. qgroup->lim_flags = limit->flags;
  831. qgroup->max_rfer = limit->max_rfer;
  832. qgroup->max_excl = limit->max_excl;
  833. qgroup->rsv_rfer = limit->rsv_rfer;
  834. qgroup->rsv_excl = limit->rsv_excl;
  835. unlock:
  836. spin_unlock(&fs_info->qgroup_lock);
  837. return ret;
  838. }
  839. static void qgroup_dirty(struct btrfs_fs_info *fs_info,
  840. struct btrfs_qgroup *qgroup)
  841. {
  842. if (list_empty(&qgroup->dirty))
  843. list_add(&qgroup->dirty, &fs_info->dirty_qgroups);
  844. }
  845. /*
  846. * btrfs_qgroup_record_ref is called when the ref is added or deleted. it puts
  847. * the modification into a list that's later used by btrfs_end_transaction to
  848. * pass the recorded modifications on to btrfs_qgroup_account_ref.
  849. */
  850. int btrfs_qgroup_record_ref(struct btrfs_trans_handle *trans,
  851. struct btrfs_delayed_ref_node *node,
  852. struct btrfs_delayed_extent_op *extent_op)
  853. {
  854. struct qgroup_update *u;
  855. BUG_ON(!trans->delayed_ref_elem.seq);
  856. u = kmalloc(sizeof(*u), GFP_NOFS);
  857. if (!u)
  858. return -ENOMEM;
  859. u->node = node;
  860. u->extent_op = extent_op;
  861. list_add_tail(&u->list, &trans->qgroup_ref_list);
  862. return 0;
  863. }
  864. /*
  865. * btrfs_qgroup_account_ref is called for every ref that is added to or deleted
  866. * from the fs. First, all roots referencing the extent are searched, and
  867. * then the space is accounted accordingly to the different roots. The
  868. * accounting algorithm works in 3 steps documented inline.
  869. */
  870. int btrfs_qgroup_account_ref(struct btrfs_trans_handle *trans,
  871. struct btrfs_fs_info *fs_info,
  872. struct btrfs_delayed_ref_node *node,
  873. struct btrfs_delayed_extent_op *extent_op)
  874. {
  875. struct btrfs_key ins;
  876. struct btrfs_root *quota_root;
  877. u64 ref_root;
  878. struct btrfs_qgroup *qgroup;
  879. struct ulist_node *unode;
  880. struct ulist *roots = NULL;
  881. struct ulist *tmp = NULL;
  882. struct ulist_iterator uiter;
  883. u64 seq;
  884. int ret = 0;
  885. int sgn;
  886. if (!fs_info->quota_enabled)
  887. return 0;
  888. BUG_ON(!fs_info->quota_root);
  889. ins.objectid = node->bytenr;
  890. ins.offset = node->num_bytes;
  891. ins.type = BTRFS_EXTENT_ITEM_KEY;
  892. if (node->type == BTRFS_TREE_BLOCK_REF_KEY ||
  893. node->type == BTRFS_SHARED_BLOCK_REF_KEY) {
  894. struct btrfs_delayed_tree_ref *ref;
  895. ref = btrfs_delayed_node_to_tree_ref(node);
  896. ref_root = ref->root;
  897. } else if (node->type == BTRFS_EXTENT_DATA_REF_KEY ||
  898. node->type == BTRFS_SHARED_DATA_REF_KEY) {
  899. struct btrfs_delayed_data_ref *ref;
  900. ref = btrfs_delayed_node_to_data_ref(node);
  901. ref_root = ref->root;
  902. } else {
  903. BUG();
  904. }
  905. if (!is_fstree(ref_root)) {
  906. /*
  907. * non-fs-trees are not being accounted
  908. */
  909. return 0;
  910. }
  911. switch (node->action) {
  912. case BTRFS_ADD_DELAYED_REF:
  913. case BTRFS_ADD_DELAYED_EXTENT:
  914. sgn = 1;
  915. break;
  916. case BTRFS_DROP_DELAYED_REF:
  917. sgn = -1;
  918. break;
  919. case BTRFS_UPDATE_DELAYED_HEAD:
  920. return 0;
  921. default:
  922. BUG();
  923. }
  924. /*
  925. * the delayed ref sequence number we pass depends on the direction of
  926. * the operation. for add operations, we pass (node->seq - 1) to skip
  927. * the delayed ref's current sequence number, because we need the state
  928. * of the tree before the add operation. for delete operations, we pass
  929. * (node->seq) to include the delayed ref's current sequence number,
  930. * because we need the state of the tree after the delete operation.
  931. */
  932. ret = btrfs_find_all_roots(trans, fs_info, node->bytenr,
  933. sgn > 0 ? node->seq - 1 : node->seq, &roots);
  934. if (ret < 0)
  935. goto out;
  936. spin_lock(&fs_info->qgroup_lock);
  937. quota_root = fs_info->quota_root;
  938. if (!quota_root)
  939. goto unlock;
  940. qgroup = find_qgroup_rb(fs_info, ref_root);
  941. if (!qgroup)
  942. goto unlock;
  943. /*
  944. * step 1: for each old ref, visit all nodes once and inc refcnt
  945. */
  946. tmp = ulist_alloc(GFP_ATOMIC);
  947. if (!tmp) {
  948. ret = -ENOMEM;
  949. goto unlock;
  950. }
  951. seq = fs_info->qgroup_seq;
  952. fs_info->qgroup_seq += roots->nnodes + 1; /* max refcnt */
  953. ULIST_ITER_INIT(&uiter);
  954. while ((unode = ulist_next(roots, &uiter))) {
  955. struct ulist_node *tmp_unode;
  956. struct ulist_iterator tmp_uiter;
  957. struct btrfs_qgroup *qg;
  958. qg = find_qgroup_rb(fs_info, unode->val);
  959. if (!qg)
  960. continue;
  961. ulist_reinit(tmp);
  962. /* XXX id not needed */
  963. ulist_add(tmp, qg->qgroupid, (unsigned long)qg, GFP_ATOMIC);
  964. ULIST_ITER_INIT(&tmp_uiter);
  965. while ((tmp_unode = ulist_next(tmp, &tmp_uiter))) {
  966. struct btrfs_qgroup_list *glist;
  967. qg = (struct btrfs_qgroup *)tmp_unode->aux;
  968. if (qg->refcnt < seq)
  969. qg->refcnt = seq + 1;
  970. else
  971. ++qg->refcnt;
  972. list_for_each_entry(glist, &qg->groups, next_group) {
  973. ulist_add(tmp, glist->group->qgroupid,
  974. (unsigned long)glist->group,
  975. GFP_ATOMIC);
  976. }
  977. }
  978. }
  979. /*
  980. * step 2: walk from the new root
  981. */
  982. ulist_reinit(tmp);
  983. ulist_add(tmp, qgroup->qgroupid, (unsigned long)qgroup, GFP_ATOMIC);
  984. ULIST_ITER_INIT(&uiter);
  985. while ((unode = ulist_next(tmp, &uiter))) {
  986. struct btrfs_qgroup *qg;
  987. struct btrfs_qgroup_list *glist;
  988. qg = (struct btrfs_qgroup *)unode->aux;
  989. if (qg->refcnt < seq) {
  990. /* not visited by step 1 */
  991. qg->rfer += sgn * node->num_bytes;
  992. qg->rfer_cmpr += sgn * node->num_bytes;
  993. if (roots->nnodes == 0) {
  994. qg->excl += sgn * node->num_bytes;
  995. qg->excl_cmpr += sgn * node->num_bytes;
  996. }
  997. qgroup_dirty(fs_info, qg);
  998. }
  999. WARN_ON(qg->tag >= seq);
  1000. qg->tag = seq;
  1001. list_for_each_entry(glist, &qg->groups, next_group) {
  1002. ulist_add(tmp, glist->group->qgroupid,
  1003. (unsigned long)glist->group, GFP_ATOMIC);
  1004. }
  1005. }
  1006. /*
  1007. * step 3: walk again from old refs
  1008. */
  1009. ULIST_ITER_INIT(&uiter);
  1010. while ((unode = ulist_next(roots, &uiter))) {
  1011. struct btrfs_qgroup *qg;
  1012. struct ulist_node *tmp_unode;
  1013. struct ulist_iterator tmp_uiter;
  1014. qg = find_qgroup_rb(fs_info, unode->val);
  1015. if (!qg)
  1016. continue;
  1017. ulist_reinit(tmp);
  1018. ulist_add(tmp, qg->qgroupid, (unsigned long)qg, GFP_ATOMIC);
  1019. ULIST_ITER_INIT(&tmp_uiter);
  1020. while ((tmp_unode = ulist_next(tmp, &tmp_uiter))) {
  1021. struct btrfs_qgroup_list *glist;
  1022. qg = (struct btrfs_qgroup *)tmp_unode->aux;
  1023. if (qg->tag == seq)
  1024. continue;
  1025. if (qg->refcnt - seq == roots->nnodes) {
  1026. qg->excl -= sgn * node->num_bytes;
  1027. qg->excl_cmpr -= sgn * node->num_bytes;
  1028. qgroup_dirty(fs_info, qg);
  1029. }
  1030. list_for_each_entry(glist, &qg->groups, next_group) {
  1031. ulist_add(tmp, glist->group->qgroupid,
  1032. (unsigned long)glist->group,
  1033. GFP_ATOMIC);
  1034. }
  1035. }
  1036. }
  1037. ret = 0;
  1038. unlock:
  1039. spin_unlock(&fs_info->qgroup_lock);
  1040. out:
  1041. ulist_free(roots);
  1042. ulist_free(tmp);
  1043. return ret;
  1044. }
  1045. /*
  1046. * called from commit_transaction. Writes all changed qgroups to disk.
  1047. */
  1048. int btrfs_run_qgroups(struct btrfs_trans_handle *trans,
  1049. struct btrfs_fs_info *fs_info)
  1050. {
  1051. struct btrfs_root *quota_root = fs_info->quota_root;
  1052. int ret = 0;
  1053. if (!quota_root)
  1054. goto out;
  1055. fs_info->quota_enabled = fs_info->pending_quota_state;
  1056. spin_lock(&fs_info->qgroup_lock);
  1057. while (!list_empty(&fs_info->dirty_qgroups)) {
  1058. struct btrfs_qgroup *qgroup;
  1059. qgroup = list_first_entry(&fs_info->dirty_qgroups,
  1060. struct btrfs_qgroup, dirty);
  1061. list_del_init(&qgroup->dirty);
  1062. spin_unlock(&fs_info->qgroup_lock);
  1063. ret = update_qgroup_info_item(trans, quota_root, qgroup);
  1064. if (ret)
  1065. fs_info->qgroup_flags |=
  1066. BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  1067. spin_lock(&fs_info->qgroup_lock);
  1068. }
  1069. if (fs_info->quota_enabled)
  1070. fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_ON;
  1071. else
  1072. fs_info->qgroup_flags &= ~BTRFS_QGROUP_STATUS_FLAG_ON;
  1073. spin_unlock(&fs_info->qgroup_lock);
  1074. ret = update_qgroup_status_item(trans, fs_info, quota_root);
  1075. if (ret)
  1076. fs_info->qgroup_flags |= BTRFS_QGROUP_STATUS_FLAG_INCONSISTENT;
  1077. out:
  1078. return ret;
  1079. }
  1080. /*
  1081. * copy the acounting information between qgroups. This is necessary when a
  1082. * snapshot or a subvolume is created
  1083. */
  1084. int btrfs_qgroup_inherit(struct btrfs_trans_handle *trans,
  1085. struct btrfs_fs_info *fs_info, u64 srcid, u64 objectid,
  1086. struct btrfs_qgroup_inherit *inherit)
  1087. {
  1088. int ret = 0;
  1089. int i;
  1090. u64 *i_qgroups;
  1091. struct btrfs_root *quota_root = fs_info->quota_root;
  1092. struct btrfs_qgroup *srcgroup;
  1093. struct btrfs_qgroup *dstgroup;
  1094. u32 level_size = 0;
  1095. if (!fs_info->quota_enabled)
  1096. return 0;
  1097. if (!quota_root)
  1098. return -EINVAL;
  1099. /*
  1100. * create a tracking group for the subvol itself
  1101. */
  1102. ret = add_qgroup_item(trans, quota_root, objectid);
  1103. if (ret)
  1104. goto out;
  1105. if (inherit && inherit->flags & BTRFS_QGROUP_INHERIT_SET_LIMITS) {
  1106. ret = update_qgroup_limit_item(trans, quota_root, objectid,
  1107. inherit->lim.flags,
  1108. inherit->lim.max_rfer,
  1109. inherit->lim.max_excl,
  1110. inherit->lim.rsv_rfer,
  1111. inherit->lim.rsv_excl);
  1112. if (ret)
  1113. goto out;
  1114. }
  1115. if (srcid) {
  1116. struct btrfs_root *srcroot;
  1117. struct btrfs_key srckey;
  1118. int srcroot_level;
  1119. srckey.objectid = srcid;
  1120. srckey.type = BTRFS_ROOT_ITEM_KEY;
  1121. srckey.offset = (u64)-1;
  1122. srcroot = btrfs_read_fs_root_no_name(fs_info, &srckey);
  1123. if (IS_ERR(srcroot)) {
  1124. ret = PTR_ERR(srcroot);
  1125. goto out;
  1126. }
  1127. rcu_read_lock();
  1128. srcroot_level = btrfs_header_level(srcroot->node);
  1129. level_size = btrfs_level_size(srcroot, srcroot_level);
  1130. rcu_read_unlock();
  1131. }
  1132. /*
  1133. * add qgroup to all inherited groups
  1134. */
  1135. if (inherit) {
  1136. i_qgroups = (u64 *)(inherit + 1);
  1137. for (i = 0; i < inherit->num_qgroups; ++i) {
  1138. ret = add_qgroup_relation_item(trans, quota_root,
  1139. objectid, *i_qgroups);
  1140. if (ret)
  1141. goto out;
  1142. ret = add_qgroup_relation_item(trans, quota_root,
  1143. *i_qgroups, objectid);
  1144. if (ret)
  1145. goto out;
  1146. ++i_qgroups;
  1147. }
  1148. }
  1149. spin_lock(&fs_info->qgroup_lock);
  1150. dstgroup = add_qgroup_rb(fs_info, objectid);
  1151. if (!dstgroup)
  1152. goto unlock;
  1153. if (srcid) {
  1154. srcgroup = find_qgroup_rb(fs_info, srcid);
  1155. if (!srcgroup)
  1156. goto unlock;
  1157. dstgroup->rfer = srcgroup->rfer - level_size;
  1158. dstgroup->rfer_cmpr = srcgroup->rfer_cmpr - level_size;
  1159. srcgroup->excl = level_size;
  1160. srcgroup->excl_cmpr = level_size;
  1161. qgroup_dirty(fs_info, dstgroup);
  1162. qgroup_dirty(fs_info, srcgroup);
  1163. }
  1164. if (!inherit)
  1165. goto unlock;
  1166. i_qgroups = (u64 *)(inherit + 1);
  1167. for (i = 0; i < inherit->num_qgroups; ++i) {
  1168. ret = add_relation_rb(quota_root->fs_info, objectid,
  1169. *i_qgroups);
  1170. if (ret)
  1171. goto unlock;
  1172. ++i_qgroups;
  1173. }
  1174. for (i = 0; i < inherit->num_ref_copies; ++i) {
  1175. struct btrfs_qgroup *src;
  1176. struct btrfs_qgroup *dst;
  1177. src = find_qgroup_rb(fs_info, i_qgroups[0]);
  1178. dst = find_qgroup_rb(fs_info, i_qgroups[1]);
  1179. if (!src || !dst) {
  1180. ret = -EINVAL;
  1181. goto unlock;
  1182. }
  1183. dst->rfer = src->rfer - level_size;
  1184. dst->rfer_cmpr = src->rfer_cmpr - level_size;
  1185. i_qgroups += 2;
  1186. }
  1187. for (i = 0; i < inherit->num_excl_copies; ++i) {
  1188. struct btrfs_qgroup *src;
  1189. struct btrfs_qgroup *dst;
  1190. src = find_qgroup_rb(fs_info, i_qgroups[0]);
  1191. dst = find_qgroup_rb(fs_info, i_qgroups[1]);
  1192. if (!src || !dst) {
  1193. ret = -EINVAL;
  1194. goto unlock;
  1195. }
  1196. dst->excl = src->excl + level_size;
  1197. dst->excl_cmpr = src->excl_cmpr + level_size;
  1198. i_qgroups += 2;
  1199. }
  1200. unlock:
  1201. spin_unlock(&fs_info->qgroup_lock);
  1202. out:
  1203. return ret;
  1204. }
  1205. /*
  1206. * reserve some space for a qgroup and all its parents. The reservation takes
  1207. * place with start_transaction or dealloc_reserve, similar to ENOSPC
  1208. * accounting. If not enough space is available, EDQUOT is returned.
  1209. * We assume that the requested space is new for all qgroups.
  1210. */
  1211. int btrfs_qgroup_reserve(struct btrfs_root *root, u64 num_bytes)
  1212. {
  1213. struct btrfs_root *quota_root;
  1214. struct btrfs_qgroup *qgroup;
  1215. struct btrfs_fs_info *fs_info = root->fs_info;
  1216. u64 ref_root = root->root_key.objectid;
  1217. int ret = 0;
  1218. struct ulist *ulist = NULL;
  1219. struct ulist_node *unode;
  1220. struct ulist_iterator uiter;
  1221. if (!is_fstree(ref_root))
  1222. return 0;
  1223. if (num_bytes == 0)
  1224. return 0;
  1225. spin_lock(&fs_info->qgroup_lock);
  1226. quota_root = fs_info->quota_root;
  1227. if (!quota_root)
  1228. goto out;
  1229. qgroup = find_qgroup_rb(fs_info, ref_root);
  1230. if (!qgroup)
  1231. goto out;
  1232. /*
  1233. * in a first step, we check all affected qgroups if any limits would
  1234. * be exceeded
  1235. */
  1236. ulist = ulist_alloc(GFP_ATOMIC);
  1237. ulist_add(ulist, qgroup->qgroupid, (unsigned long)qgroup, GFP_ATOMIC);
  1238. ULIST_ITER_INIT(&uiter);
  1239. while ((unode = ulist_next(ulist, &uiter))) {
  1240. struct btrfs_qgroup *qg;
  1241. struct btrfs_qgroup_list *glist;
  1242. qg = (struct btrfs_qgroup *)unode->aux;
  1243. if ((qg->lim_flags & BTRFS_QGROUP_LIMIT_MAX_RFER) &&
  1244. qg->reserved + qg->rfer + num_bytes >
  1245. qg->max_rfer)
  1246. ret = -EDQUOT;
  1247. if ((qg->lim_flags & BTRFS_QGROUP_LIMIT_MAX_EXCL) &&
  1248. qg->reserved + qg->excl + num_bytes >
  1249. qg->max_excl)
  1250. ret = -EDQUOT;
  1251. list_for_each_entry(glist, &qg->groups, next_group) {
  1252. ulist_add(ulist, glist->group->qgroupid,
  1253. (unsigned long)glist->group, GFP_ATOMIC);
  1254. }
  1255. }
  1256. if (ret)
  1257. goto out;
  1258. /*
  1259. * no limits exceeded, now record the reservation into all qgroups
  1260. */
  1261. ULIST_ITER_INIT(&uiter);
  1262. while ((unode = ulist_next(ulist, &uiter))) {
  1263. struct btrfs_qgroup *qg;
  1264. qg = (struct btrfs_qgroup *)unode->aux;
  1265. qg->reserved += num_bytes;
  1266. }
  1267. out:
  1268. spin_unlock(&fs_info->qgroup_lock);
  1269. ulist_free(ulist);
  1270. return ret;
  1271. }
  1272. void btrfs_qgroup_free(struct btrfs_root *root, u64 num_bytes)
  1273. {
  1274. struct btrfs_root *quota_root;
  1275. struct btrfs_qgroup *qgroup;
  1276. struct btrfs_fs_info *fs_info = root->fs_info;
  1277. struct ulist *ulist = NULL;
  1278. struct ulist_node *unode;
  1279. struct ulist_iterator uiter;
  1280. u64 ref_root = root->root_key.objectid;
  1281. if (!is_fstree(ref_root))
  1282. return;
  1283. if (num_bytes == 0)
  1284. return;
  1285. spin_lock(&fs_info->qgroup_lock);
  1286. quota_root = fs_info->quota_root;
  1287. if (!quota_root)
  1288. goto out;
  1289. qgroup = find_qgroup_rb(fs_info, ref_root);
  1290. if (!qgroup)
  1291. goto out;
  1292. ulist = ulist_alloc(GFP_ATOMIC);
  1293. ulist_add(ulist, qgroup->qgroupid, (unsigned long)qgroup, GFP_ATOMIC);
  1294. ULIST_ITER_INIT(&uiter);
  1295. while ((unode = ulist_next(ulist, &uiter))) {
  1296. struct btrfs_qgroup *qg;
  1297. struct btrfs_qgroup_list *glist;
  1298. qg = (struct btrfs_qgroup *)unode->aux;
  1299. qg->reserved -= num_bytes;
  1300. list_for_each_entry(glist, &qg->groups, next_group) {
  1301. ulist_add(ulist, glist->group->qgroupid,
  1302. (unsigned long)glist->group, GFP_ATOMIC);
  1303. }
  1304. }
  1305. out:
  1306. spin_unlock(&fs_info->qgroup_lock);
  1307. ulist_free(ulist);
  1308. }
  1309. void assert_qgroups_uptodate(struct btrfs_trans_handle *trans)
  1310. {
  1311. if (list_empty(&trans->qgroup_ref_list) && !trans->delayed_ref_elem.seq)
  1312. return;
  1313. printk(KERN_ERR "btrfs: qgroups not uptodate in trans handle %p: list is%s empty, seq is %llu\n",
  1314. trans, list_empty(&trans->qgroup_ref_list) ? "" : " not",
  1315. trans->delayed_ref_elem.seq);
  1316. BUG();
  1317. }