extent-tree.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760
  1. #include <linux/module.h>
  2. #include "ctree.h"
  3. #include "disk-io.h"
  4. #include "print-tree.h"
  5. #include "transaction.h"
  6. static int find_free_extent(struct btrfs_trans_handle *trans, struct btrfs_root
  7. *orig_root, u64 num_blocks, u64 search_start, u64
  8. search_end, struct btrfs_key *ins);
  9. static int finish_current_insert(struct btrfs_trans_handle *trans, struct
  10. btrfs_root *extent_root);
  11. static int del_pending_extents(struct btrfs_trans_handle *trans, struct
  12. btrfs_root *extent_root);
  13. int btrfs_inc_extent_ref(struct btrfs_trans_handle *trans,
  14. struct btrfs_root *root,
  15. u64 blocknr, u64 num_blocks)
  16. {
  17. struct btrfs_path *path;
  18. int ret;
  19. struct btrfs_key key;
  20. struct btrfs_leaf *l;
  21. struct btrfs_extent_item *item;
  22. struct btrfs_key ins;
  23. u32 refs;
  24. find_free_extent(trans, root->fs_info->extent_root, 0, 0, (u64)-1,
  25. &ins);
  26. path = btrfs_alloc_path();
  27. BUG_ON(!path);
  28. btrfs_init_path(path);
  29. key.objectid = blocknr;
  30. key.flags = 0;
  31. btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY);
  32. key.offset = num_blocks;
  33. ret = btrfs_search_slot(trans, root->fs_info->extent_root, &key, path,
  34. 0, 1);
  35. if (ret != 0) {
  36. printk("can't find block %Lu %Lu\n", blocknr, num_blocks);
  37. BUG();
  38. }
  39. BUG_ON(ret != 0);
  40. l = btrfs_buffer_leaf(path->nodes[0]);
  41. item = btrfs_item_ptr(l, path->slots[0], struct btrfs_extent_item);
  42. refs = btrfs_extent_refs(item);
  43. btrfs_set_extent_refs(item, refs + 1);
  44. btrfs_mark_buffer_dirty(path->nodes[0]);
  45. btrfs_release_path(root->fs_info->extent_root, path);
  46. btrfs_free_path(path);
  47. finish_current_insert(trans, root->fs_info->extent_root);
  48. del_pending_extents(trans, root->fs_info->extent_root);
  49. return 0;
  50. }
  51. static int lookup_extent_ref(struct btrfs_trans_handle *trans,
  52. struct btrfs_root *root, u64 blocknr,
  53. u64 num_blocks, u32 *refs)
  54. {
  55. struct btrfs_path *path;
  56. int ret;
  57. struct btrfs_key key;
  58. struct btrfs_leaf *l;
  59. struct btrfs_extent_item *item;
  60. path = btrfs_alloc_path();
  61. btrfs_init_path(path);
  62. key.objectid = blocknr;
  63. key.offset = num_blocks;
  64. key.flags = 0;
  65. btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY);
  66. ret = btrfs_search_slot(trans, root->fs_info->extent_root, &key, path,
  67. 0, 0);
  68. if (ret != 0)
  69. BUG();
  70. l = btrfs_buffer_leaf(path->nodes[0]);
  71. item = btrfs_item_ptr(l, path->slots[0], struct btrfs_extent_item);
  72. *refs = btrfs_extent_refs(item);
  73. btrfs_release_path(root->fs_info->extent_root, path);
  74. btrfs_free_path(path);
  75. return 0;
  76. }
  77. int btrfs_inc_root_ref(struct btrfs_trans_handle *trans,
  78. struct btrfs_root *root)
  79. {
  80. return btrfs_inc_extent_ref(trans, root, bh_blocknr(root->node), 1);
  81. }
  82. int btrfs_inc_ref(struct btrfs_trans_handle *trans, struct btrfs_root *root,
  83. struct buffer_head *buf)
  84. {
  85. u64 blocknr;
  86. struct btrfs_node *buf_node;
  87. struct btrfs_leaf *buf_leaf;
  88. struct btrfs_disk_key *key;
  89. struct btrfs_file_extent_item *fi;
  90. int i;
  91. int leaf;
  92. int ret;
  93. if (!root->ref_cows)
  94. return 0;
  95. buf_node = btrfs_buffer_node(buf);
  96. leaf = btrfs_is_leaf(buf_node);
  97. buf_leaf = btrfs_buffer_leaf(buf);
  98. for (i = 0; i < btrfs_header_nritems(&buf_node->header); i++) {
  99. if (leaf) {
  100. key = &buf_leaf->items[i].key;
  101. if (btrfs_disk_key_type(key) != BTRFS_EXTENT_DATA_KEY)
  102. continue;
  103. fi = btrfs_item_ptr(buf_leaf, i,
  104. struct btrfs_file_extent_item);
  105. if (btrfs_file_extent_type(fi) ==
  106. BTRFS_FILE_EXTENT_INLINE)
  107. continue;
  108. ret = btrfs_inc_extent_ref(trans, root,
  109. btrfs_file_extent_disk_blocknr(fi),
  110. btrfs_file_extent_disk_num_blocks(fi));
  111. BUG_ON(ret);
  112. } else {
  113. blocknr = btrfs_node_blockptr(buf_node, i);
  114. ret = btrfs_inc_extent_ref(trans, root, blocknr, 1);
  115. BUG_ON(ret);
  116. }
  117. }
  118. return 0;
  119. }
  120. int btrfs_finish_extent_commit(struct btrfs_trans_handle *trans, struct
  121. btrfs_root *root)
  122. {
  123. unsigned long gang[8];
  124. u64 first = 0;
  125. int ret;
  126. int i;
  127. struct radix_tree_root *pinned_radix = &root->fs_info->pinned_radix;
  128. while(1) {
  129. ret = find_first_radix_bit(pinned_radix, gang,
  130. ARRAY_SIZE(gang));
  131. if (!ret)
  132. break;
  133. if (!first)
  134. first = gang[0];
  135. for (i = 0; i < ret; i++) {
  136. clear_radix_bit(pinned_radix, gang[i]);
  137. }
  138. }
  139. if (root->fs_info->last_insert.objectid > first)
  140. root->fs_info->last_insert.objectid = first;
  141. root->fs_info->last_insert.offset = 0;
  142. return 0;
  143. }
  144. static int finish_current_insert(struct btrfs_trans_handle *trans, struct
  145. btrfs_root *extent_root)
  146. {
  147. struct btrfs_key ins;
  148. struct btrfs_extent_item extent_item;
  149. int i;
  150. int ret;
  151. u64 super_blocks_used;
  152. struct btrfs_fs_info *info = extent_root->fs_info;
  153. btrfs_set_extent_refs(&extent_item, 1);
  154. ins.offset = 1;
  155. ins.flags = 0;
  156. btrfs_set_key_type(&ins, BTRFS_EXTENT_ITEM_KEY);
  157. btrfs_set_extent_owner(&extent_item, extent_root->root_key.objectid);
  158. for (i = 0; i < extent_root->fs_info->extent_tree_insert_nr; i++) {
  159. ins.objectid = extent_root->fs_info->extent_tree_insert[i];
  160. super_blocks_used = btrfs_super_blocks_used(info->disk_super);
  161. btrfs_set_super_blocks_used(info->disk_super,
  162. super_blocks_used + 1);
  163. ret = btrfs_insert_item(trans, extent_root, &ins, &extent_item,
  164. sizeof(extent_item));
  165. BUG_ON(ret);
  166. }
  167. extent_root->fs_info->extent_tree_insert_nr = 0;
  168. extent_root->fs_info->extent_tree_prealloc_nr = 0;
  169. return 0;
  170. }
  171. static int pin_down_block(struct btrfs_root *root, u64 blocknr, int pending)
  172. {
  173. int err;
  174. struct btrfs_header *header;
  175. struct buffer_head *bh;
  176. if (!pending) {
  177. bh = btrfs_find_tree_block(root, blocknr);
  178. if (bh) {
  179. if (buffer_uptodate(bh)) {
  180. u64 transid =
  181. root->fs_info->running_transaction->transid;
  182. header = btrfs_buffer_header(bh);
  183. if (btrfs_header_generation(header) ==
  184. transid) {
  185. btrfs_block_release(root, bh);
  186. return 0;
  187. }
  188. }
  189. btrfs_block_release(root, bh);
  190. }
  191. err = set_radix_bit(&root->fs_info->pinned_radix, blocknr);
  192. } else {
  193. err = set_radix_bit(&root->fs_info->pending_del_radix, blocknr);
  194. }
  195. BUG_ON(err);
  196. return 0;
  197. }
  198. /*
  199. * remove an extent from the root, returns 0 on success
  200. */
  201. static int __free_extent(struct btrfs_trans_handle *trans, struct btrfs_root
  202. *root, u64 blocknr, u64 num_blocks, int pin)
  203. {
  204. struct btrfs_path *path;
  205. struct btrfs_key key;
  206. struct btrfs_fs_info *info = root->fs_info;
  207. struct btrfs_root *extent_root = info->extent_root;
  208. int ret;
  209. struct btrfs_extent_item *ei;
  210. struct btrfs_key ins;
  211. u32 refs;
  212. key.objectid = blocknr;
  213. key.flags = 0;
  214. btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY);
  215. key.offset = num_blocks;
  216. find_free_extent(trans, root, 0, 0, (u64)-1, &ins);
  217. path = btrfs_alloc_path();
  218. BUG_ON(!path);
  219. btrfs_init_path(path);
  220. ret = btrfs_search_slot(trans, extent_root, &key, path, -1, 1);
  221. if (ret) {
  222. printk("failed to find %Lu\n", key.objectid);
  223. btrfs_print_tree(extent_root, extent_root->node);
  224. printk("failed to find %Lu\n", key.objectid);
  225. BUG();
  226. }
  227. ei = btrfs_item_ptr(btrfs_buffer_leaf(path->nodes[0]), path->slots[0],
  228. struct btrfs_extent_item);
  229. BUG_ON(ei->refs == 0);
  230. refs = btrfs_extent_refs(ei) - 1;
  231. btrfs_set_extent_refs(ei, refs);
  232. btrfs_mark_buffer_dirty(path->nodes[0]);
  233. if (refs == 0) {
  234. u64 super_blocks_used;
  235. if (pin) {
  236. ret = pin_down_block(root, blocknr, 0);
  237. BUG_ON(ret);
  238. }
  239. super_blocks_used = btrfs_super_blocks_used(info->disk_super);
  240. btrfs_set_super_blocks_used(info->disk_super,
  241. super_blocks_used - num_blocks);
  242. ret = btrfs_del_item(trans, extent_root, path);
  243. if (ret)
  244. BUG();
  245. }
  246. btrfs_release_path(extent_root, path);
  247. btrfs_free_path(path);
  248. finish_current_insert(trans, extent_root);
  249. return ret;
  250. }
  251. /*
  252. * find all the blocks marked as pending in the radix tree and remove
  253. * them from the extent map
  254. */
  255. static int del_pending_extents(struct btrfs_trans_handle *trans, struct
  256. btrfs_root *extent_root)
  257. {
  258. int ret;
  259. int wret;
  260. int err = 0;
  261. unsigned long gang[4];
  262. int i;
  263. struct radix_tree_root *pending_radix;
  264. struct radix_tree_root *pinned_radix;
  265. pending_radix = &extent_root->fs_info->pending_del_radix;
  266. pinned_radix = &extent_root->fs_info->pinned_radix;
  267. while(1) {
  268. ret = find_first_radix_bit(pending_radix, gang,
  269. ARRAY_SIZE(gang));
  270. if (!ret)
  271. break;
  272. for (i = 0; i < ret; i++) {
  273. wret = set_radix_bit(pinned_radix, gang[i]);
  274. BUG_ON(wret);
  275. wret = clear_radix_bit(pending_radix, gang[i]);
  276. BUG_ON(wret);
  277. wret = __free_extent(trans, extent_root,
  278. gang[i], 1, 0);
  279. if (wret)
  280. err = wret;
  281. }
  282. }
  283. return err;
  284. }
  285. /*
  286. * remove an extent from the root, returns 0 on success
  287. */
  288. int btrfs_free_extent(struct btrfs_trans_handle *trans, struct btrfs_root
  289. *root, u64 blocknr, u64 num_blocks, int pin)
  290. {
  291. struct btrfs_root *extent_root = root->fs_info->extent_root;
  292. int pending_ret;
  293. int ret;
  294. if (root == extent_root) {
  295. pin_down_block(root, blocknr, 1);
  296. return 0;
  297. }
  298. ret = __free_extent(trans, root, blocknr, num_blocks, pin);
  299. pending_ret = del_pending_extents(trans, root->fs_info->extent_root);
  300. return ret ? ret : pending_ret;
  301. }
  302. /*
  303. * walks the btree of allocated extents and find a hole of a given size.
  304. * The key ins is changed to record the hole:
  305. * ins->objectid == block start
  306. * ins->flags = BTRFS_EXTENT_ITEM_KEY
  307. * ins->offset == number of blocks
  308. * Any available blocks before search_start are skipped.
  309. */
  310. static int find_free_extent(struct btrfs_trans_handle *trans, struct btrfs_root
  311. *orig_root, u64 num_blocks, u64 search_start, u64
  312. search_end, struct btrfs_key *ins)
  313. {
  314. struct btrfs_path *path;
  315. struct btrfs_key key;
  316. int ret;
  317. u64 hole_size = 0;
  318. int slot = 0;
  319. u64 last_block = 0;
  320. u64 test_block;
  321. int start_found;
  322. struct btrfs_leaf *l;
  323. struct btrfs_root * root = orig_root->fs_info->extent_root;
  324. struct btrfs_fs_info *info = root->fs_info;
  325. int total_needed = num_blocks;
  326. int total_found = 0;
  327. int fill_prealloc = 0;
  328. int level;
  329. path = btrfs_alloc_path();
  330. ins->flags = 0;
  331. btrfs_set_key_type(ins, BTRFS_EXTENT_ITEM_KEY);
  332. level = btrfs_header_level(btrfs_buffer_header(root->node));
  333. if (num_blocks == 0) {
  334. fill_prealloc = 1;
  335. num_blocks = 1;
  336. total_needed = min(level + 2, BTRFS_MAX_LEVEL) * 3;
  337. }
  338. if (info->last_insert.objectid == 0 && search_end == (u64)-1) {
  339. struct btrfs_disk_key *last_key;
  340. btrfs_init_path(path);
  341. ins->objectid = (u64)-1;
  342. ins->offset = (u64)-1;
  343. ret = btrfs_search_slot(trans, root, ins, path, 0, 0);
  344. if (ret < 0)
  345. goto error;
  346. BUG_ON(ret == 0);
  347. if (path->slots[0] > 0)
  348. path->slots[0]--;
  349. l = btrfs_buffer_leaf(path->nodes[0]);
  350. last_key = &l->items[path->slots[0]].key;
  351. search_start = btrfs_disk_key_objectid(last_key);
  352. }
  353. if (info->last_insert.objectid > search_start)
  354. search_start = info->last_insert.objectid;
  355. check_failed:
  356. btrfs_init_path(path);
  357. ins->objectid = search_start;
  358. ins->offset = 0;
  359. start_found = 0;
  360. ret = btrfs_search_slot(trans, root, ins, path, 0, 0);
  361. if (ret < 0)
  362. goto error;
  363. if (path->slots[0] > 0)
  364. path->slots[0]--;
  365. while (1) {
  366. l = btrfs_buffer_leaf(path->nodes[0]);
  367. slot = path->slots[0];
  368. if (slot >= btrfs_header_nritems(&l->header)) {
  369. if (fill_prealloc) {
  370. info->extent_tree_prealloc_nr = 0;
  371. total_found = 0;
  372. }
  373. ret = btrfs_next_leaf(root, path);
  374. if (ret == 0)
  375. continue;
  376. if (ret < 0)
  377. goto error;
  378. if (!start_found) {
  379. ins->objectid = search_start;
  380. ins->offset = (u64)-1 - search_start;
  381. start_found = 1;
  382. goto check_pending;
  383. }
  384. ins->objectid = last_block > search_start ?
  385. last_block : search_start;
  386. ins->offset = (u64)-1 - ins->objectid;
  387. goto check_pending;
  388. }
  389. btrfs_disk_key_to_cpu(&key, &l->items[slot].key);
  390. if (key.objectid >= search_start) {
  391. if (start_found) {
  392. if (last_block < search_start)
  393. last_block = search_start;
  394. hole_size = key.objectid - last_block;
  395. if (hole_size > num_blocks) {
  396. ins->objectid = last_block;
  397. ins->offset = hole_size;
  398. goto check_pending;
  399. }
  400. }
  401. }
  402. start_found = 1;
  403. last_block = key.objectid + key.offset;
  404. path->slots[0]++;
  405. }
  406. // FIXME -ENOSPC
  407. check_pending:
  408. /* we have to make sure we didn't find an extent that has already
  409. * been allocated by the map tree or the original allocation
  410. */
  411. btrfs_release_path(root, path);
  412. BUG_ON(ins->objectid < search_start);
  413. for (test_block = ins->objectid;
  414. test_block < ins->objectid + num_blocks; test_block++) {
  415. if (test_radix_bit(&info->pinned_radix, test_block)) {
  416. search_start = test_block + 1;
  417. goto check_failed;
  418. }
  419. }
  420. if (!fill_prealloc && info->extent_tree_insert_nr) {
  421. u64 last =
  422. info->extent_tree_insert[info->extent_tree_insert_nr - 1];
  423. if (ins->objectid + num_blocks >
  424. info->extent_tree_insert[0] &&
  425. ins->objectid <= last) {
  426. search_start = last + 1;
  427. WARN_ON(1);
  428. goto check_failed;
  429. }
  430. }
  431. if (!fill_prealloc && info->extent_tree_prealloc_nr) {
  432. u64 first =
  433. info->extent_tree_prealloc[info->extent_tree_prealloc_nr - 1];
  434. if (ins->objectid + num_blocks > first &&
  435. ins->objectid <= info->extent_tree_prealloc[0]) {
  436. search_start = info->extent_tree_prealloc[0] + 1;
  437. WARN_ON(1);
  438. goto check_failed;
  439. }
  440. }
  441. if (fill_prealloc) {
  442. int nr;
  443. test_block = ins->objectid;
  444. while(test_block < ins->objectid + ins->offset &&
  445. total_found < total_needed) {
  446. nr = total_needed - total_found - 1;
  447. BUG_ON(nr < 0);
  448. root->fs_info->extent_tree_prealloc[nr] =
  449. test_block;
  450. total_found++;
  451. test_block++;
  452. }
  453. if (total_found < total_needed) {
  454. search_start = test_block;
  455. goto check_failed;
  456. }
  457. root->fs_info->extent_tree_prealloc_nr = total_found;
  458. }
  459. root->fs_info->last_insert.objectid = ins->objectid;
  460. ins->offset = num_blocks;
  461. btrfs_free_path(path);
  462. return 0;
  463. error:
  464. btrfs_release_path(root, path);
  465. btrfs_free_path(path);
  466. return ret;
  467. }
  468. /*
  469. * finds a free extent and does all the dirty work required for allocation
  470. * returns the key for the extent through ins, and a tree buffer for
  471. * the first block of the extent through buf.
  472. *
  473. * returns 0 if everything worked, non-zero otherwise.
  474. */
  475. int btrfs_alloc_extent(struct btrfs_trans_handle *trans,
  476. struct btrfs_root *root, u64 owner,
  477. u64 num_blocks, u64 search_start,
  478. u64 search_end, struct btrfs_key *ins)
  479. {
  480. int ret;
  481. int pending_ret;
  482. u64 super_blocks_used;
  483. struct btrfs_fs_info *info = root->fs_info;
  484. struct btrfs_root *extent_root = info->extent_root;
  485. struct btrfs_extent_item extent_item;
  486. struct btrfs_key prealloc_key;
  487. btrfs_set_extent_refs(&extent_item, 1);
  488. btrfs_set_extent_owner(&extent_item, owner);
  489. if (root == extent_root) {
  490. int nr;
  491. BUG_ON(info->extent_tree_prealloc_nr == 0);
  492. BUG_ON(num_blocks != 1);
  493. ins->offset = 1;
  494. info->extent_tree_prealloc_nr--;
  495. nr = info->extent_tree_prealloc_nr;
  496. ins->objectid = info->extent_tree_prealloc[nr];
  497. info->extent_tree_insert[info->extent_tree_insert_nr++] =
  498. ins->objectid;
  499. return 0;
  500. }
  501. /* do the real allocation */
  502. ret = find_free_extent(trans, root, num_blocks, search_start,
  503. search_end, ins);
  504. if (ret)
  505. return ret;
  506. /* then do prealloc for the extent tree */
  507. ret = find_free_extent(trans, root, 0, ins->objectid + ins->offset,
  508. search_end, &prealloc_key);
  509. if (ret)
  510. return ret;
  511. super_blocks_used = btrfs_super_blocks_used(info->disk_super);
  512. btrfs_set_super_blocks_used(info->disk_super, super_blocks_used +
  513. num_blocks);
  514. ret = btrfs_insert_item(trans, extent_root, ins, &extent_item,
  515. sizeof(extent_item));
  516. finish_current_insert(trans, extent_root);
  517. pending_ret = del_pending_extents(trans, extent_root);
  518. if (ret)
  519. return ret;
  520. if (pending_ret)
  521. return pending_ret;
  522. return 0;
  523. }
  524. /*
  525. * helper function to allocate a block for a given tree
  526. * returns the tree buffer or NULL.
  527. */
  528. struct buffer_head *btrfs_alloc_free_block(struct btrfs_trans_handle *trans,
  529. struct btrfs_root *root)
  530. {
  531. struct btrfs_key ins;
  532. int ret;
  533. struct buffer_head *buf;
  534. ret = btrfs_alloc_extent(trans, root, root->root_key.objectid,
  535. 1, 0, (unsigned long)-1, &ins);
  536. if (ret) {
  537. BUG();
  538. return NULL;
  539. }
  540. buf = btrfs_find_create_tree_block(root, ins.objectid);
  541. set_buffer_uptodate(buf);
  542. return buf;
  543. }
  544. static int drop_leaf_ref(struct btrfs_trans_handle *trans,
  545. struct btrfs_root *root, struct buffer_head *cur)
  546. {
  547. struct btrfs_disk_key *key;
  548. struct btrfs_leaf *leaf;
  549. struct btrfs_file_extent_item *fi;
  550. int i;
  551. int nritems;
  552. int ret;
  553. BUG_ON(!btrfs_is_leaf(btrfs_buffer_node(cur)));
  554. leaf = btrfs_buffer_leaf(cur);
  555. nritems = btrfs_header_nritems(&leaf->header);
  556. for (i = 0; i < nritems; i++) {
  557. key = &leaf->items[i].key;
  558. if (btrfs_disk_key_type(key) != BTRFS_EXTENT_DATA_KEY)
  559. continue;
  560. fi = btrfs_item_ptr(leaf, i, struct btrfs_file_extent_item);
  561. if (btrfs_file_extent_type(fi) == BTRFS_FILE_EXTENT_INLINE)
  562. continue;
  563. /*
  564. * FIXME make sure to insert a trans record that
  565. * repeats the snapshot del on crash
  566. */
  567. ret = btrfs_free_extent(trans, root,
  568. btrfs_file_extent_disk_blocknr(fi),
  569. btrfs_file_extent_disk_num_blocks(fi),
  570. 0);
  571. BUG_ON(ret);
  572. }
  573. return 0;
  574. }
  575. /*
  576. * helper function for drop_snapshot, this walks down the tree dropping ref
  577. * counts as it goes.
  578. */
  579. static int walk_down_tree(struct btrfs_trans_handle *trans, struct btrfs_root
  580. *root, struct btrfs_path *path, int *level)
  581. {
  582. struct buffer_head *next;
  583. struct buffer_head *cur;
  584. u64 blocknr;
  585. int ret;
  586. u32 refs;
  587. WARN_ON(*level < 0);
  588. WARN_ON(*level >= BTRFS_MAX_LEVEL);
  589. ret = lookup_extent_ref(trans, root, bh_blocknr(path->nodes[*level]),
  590. 1, &refs);
  591. BUG_ON(ret);
  592. if (refs > 1)
  593. goto out;
  594. /*
  595. * walk down to the last node level and free all the leaves
  596. */
  597. while(*level >= 0) {
  598. WARN_ON(*level < 0);
  599. WARN_ON(*level >= BTRFS_MAX_LEVEL);
  600. cur = path->nodes[*level];
  601. if (btrfs_header_level(btrfs_buffer_header(cur)) != *level)
  602. WARN_ON(1);
  603. if (path->slots[*level] >=
  604. btrfs_header_nritems(btrfs_buffer_header(cur)))
  605. break;
  606. if (*level == 0) {
  607. ret = drop_leaf_ref(trans, root, cur);
  608. BUG_ON(ret);
  609. break;
  610. }
  611. blocknr = btrfs_node_blockptr(btrfs_buffer_node(cur),
  612. path->slots[*level]);
  613. ret = lookup_extent_ref(trans, root, blocknr, 1, &refs);
  614. BUG_ON(ret);
  615. if (refs != 1) {
  616. path->slots[*level]++;
  617. ret = btrfs_free_extent(trans, root, blocknr, 1, 1);
  618. BUG_ON(ret);
  619. continue;
  620. }
  621. next = read_tree_block(root, blocknr);
  622. WARN_ON(*level <= 0);
  623. if (path->nodes[*level-1])
  624. btrfs_block_release(root, path->nodes[*level-1]);
  625. path->nodes[*level-1] = next;
  626. *level = btrfs_header_level(btrfs_buffer_header(next));
  627. path->slots[*level] = 0;
  628. }
  629. out:
  630. WARN_ON(*level < 0);
  631. WARN_ON(*level >= BTRFS_MAX_LEVEL);
  632. ret = btrfs_free_extent(trans, root,
  633. bh_blocknr(path->nodes[*level]), 1, 1);
  634. btrfs_block_release(root, path->nodes[*level]);
  635. path->nodes[*level] = NULL;
  636. *level += 1;
  637. BUG_ON(ret);
  638. return 0;
  639. }
  640. /*
  641. * helper for dropping snapshots. This walks back up the tree in the path
  642. * to find the first node higher up where we haven't yet gone through
  643. * all the slots
  644. */
  645. static int walk_up_tree(struct btrfs_trans_handle *trans, struct btrfs_root
  646. *root, struct btrfs_path *path, int *level)
  647. {
  648. int i;
  649. int slot;
  650. int ret;
  651. for(i = *level; i < BTRFS_MAX_LEVEL - 1 && path->nodes[i]; i++) {
  652. slot = path->slots[i];
  653. if (slot < btrfs_header_nritems(
  654. btrfs_buffer_header(path->nodes[i])) - 1) {
  655. path->slots[i]++;
  656. *level = i;
  657. return 0;
  658. } else {
  659. ret = btrfs_free_extent(trans, root,
  660. bh_blocknr(path->nodes[*level]),
  661. 1, 1);
  662. BUG_ON(ret);
  663. btrfs_block_release(root, path->nodes[*level]);
  664. path->nodes[*level] = NULL;
  665. *level = i + 1;
  666. }
  667. }
  668. return 1;
  669. }
  670. /*
  671. * drop the reference count on the tree rooted at 'snap'. This traverses
  672. * the tree freeing any blocks that have a ref count of zero after being
  673. * decremented.
  674. */
  675. int btrfs_drop_snapshot(struct btrfs_trans_handle *trans, struct btrfs_root
  676. *root, struct buffer_head *snap)
  677. {
  678. int ret = 0;
  679. int wret;
  680. int level;
  681. struct btrfs_path *path;
  682. int i;
  683. int orig_level;
  684. path = btrfs_alloc_path();
  685. BUG_ON(!path);
  686. btrfs_init_path(path);
  687. level = btrfs_header_level(btrfs_buffer_header(snap));
  688. orig_level = level;
  689. path->nodes[level] = snap;
  690. path->slots[level] = 0;
  691. while(1) {
  692. wret = walk_down_tree(trans, root, path, &level);
  693. if (wret > 0)
  694. break;
  695. if (wret < 0)
  696. ret = wret;
  697. wret = walk_up_tree(trans, root, path, &level);
  698. if (wret > 0)
  699. break;
  700. if (wret < 0)
  701. ret = wret;
  702. }
  703. for (i = 0; i <= orig_level; i++) {
  704. if (path->nodes[i]) {
  705. btrfs_block_release(root, path->nodes[i]);
  706. }
  707. }
  708. btrfs_free_path(path);
  709. return ret;
  710. }