disk-io.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742
  1. #include <linux/module.h>
  2. #include <linux/fs.h>
  3. #include <linux/blkdev.h>
  4. #include <linux/crypto.h>
  5. #include <linux/scatterlist.h>
  6. #include <linux/swap.h>
  7. #include <linux/radix-tree.h>
  8. #include "ctree.h"
  9. #include "disk-io.h"
  10. #include "transaction.h"
  11. #include "btrfs_inode.h"
  12. struct dev_lookup {
  13. u64 block_start;
  14. u64 num_blocks;
  15. u64 device_id;
  16. struct block_device *bdev;
  17. };
  18. int btrfs_insert_dev_radix(struct btrfs_root *root,
  19. struct block_device *bdev,
  20. u64 device_id,
  21. u64 block_start,
  22. u64 num_blocks)
  23. {
  24. struct dev_lookup *lookup;
  25. int ret;
  26. lookup = kmalloc(sizeof(*lookup), GFP_NOFS);
  27. if (!lookup)
  28. return -ENOMEM;
  29. lookup->block_start = block_start;
  30. lookup->num_blocks = num_blocks;
  31. lookup->bdev = bdev;
  32. lookup->device_id = device_id;
  33. ret = radix_tree_insert(&root->fs_info->dev_radix, block_start +
  34. num_blocks - 1, lookup);
  35. return ret;
  36. }
  37. u64 bh_blocknr(struct buffer_head *bh)
  38. {
  39. int blkbits = bh->b_page->mapping->host->i_blkbits;
  40. u64 blocknr = bh->b_page->index << (PAGE_CACHE_SHIFT - blkbits);
  41. unsigned long offset;
  42. if (PageHighMem(bh->b_page))
  43. offset = (unsigned long)bh->b_data;
  44. else
  45. offset = bh->b_data - (char *)page_address(bh->b_page);
  46. blocknr += offset >> (PAGE_CACHE_SHIFT - blkbits);
  47. return blocknr;
  48. }
  49. static int check_tree_block(struct btrfs_root *root, struct buffer_head *buf)
  50. {
  51. struct btrfs_node *node = btrfs_buffer_node(buf);
  52. if (bh_blocknr(buf) != btrfs_header_blocknr(&node->header)) {
  53. printk(KERN_CRIT "bh_blocknr(buf) is %Lu, header is %Lu\n",
  54. bh_blocknr(buf), btrfs_header_blocknr(&node->header));
  55. BUG();
  56. }
  57. return 0;
  58. }
  59. struct buffer_head *btrfs_find_tree_block(struct btrfs_root *root, u64 blocknr)
  60. {
  61. struct address_space *mapping = root->fs_info->btree_inode->i_mapping;
  62. int blockbits = root->fs_info->sb->s_blocksize_bits;
  63. unsigned long index = blocknr >> (PAGE_CACHE_SHIFT - blockbits);
  64. struct page *page;
  65. struct buffer_head *bh;
  66. struct buffer_head *head;
  67. struct buffer_head *ret = NULL;
  68. page = find_lock_page(mapping, index);
  69. if (!page)
  70. return NULL;
  71. if (!page_has_buffers(page))
  72. goto out_unlock;
  73. head = page_buffers(page);
  74. bh = head;
  75. do {
  76. if (buffer_mapped(bh) && bh_blocknr(bh) == blocknr) {
  77. ret = bh;
  78. get_bh(bh);
  79. goto out_unlock;
  80. }
  81. bh = bh->b_this_page;
  82. } while (bh != head);
  83. out_unlock:
  84. unlock_page(page);
  85. if (ret) {
  86. touch_buffer(ret);
  87. }
  88. page_cache_release(page);
  89. return ret;
  90. }
  91. int btrfs_map_bh_to_logical(struct btrfs_root *root, struct buffer_head *bh,
  92. u64 logical)
  93. {
  94. struct dev_lookup *lookup[2];
  95. int ret;
  96. if (logical == 0) {
  97. bh->b_bdev = NULL;
  98. bh->b_blocknr = 0;
  99. set_buffer_mapped(bh);
  100. return 0;
  101. }
  102. root = root->fs_info->dev_root;
  103. ret = radix_tree_gang_lookup(&root->fs_info->dev_radix,
  104. (void **)lookup,
  105. (unsigned long)logical,
  106. ARRAY_SIZE(lookup));
  107. if (ret == 0 || lookup[0]->block_start > logical ||
  108. lookup[0]->block_start + lookup[0]->num_blocks <= logical) {
  109. ret = -ENOENT;
  110. goto out;
  111. }
  112. bh->b_bdev = lookup[0]->bdev;
  113. bh->b_blocknr = logical - lookup[0]->block_start;
  114. set_buffer_mapped(bh);
  115. ret = 0;
  116. out:
  117. return ret;
  118. }
  119. struct buffer_head *btrfs_find_create_tree_block(struct btrfs_root *root,
  120. u64 blocknr)
  121. {
  122. struct address_space *mapping = root->fs_info->btree_inode->i_mapping;
  123. int blockbits = root->fs_info->sb->s_blocksize_bits;
  124. unsigned long index = blocknr >> (PAGE_CACHE_SHIFT - blockbits);
  125. struct page *page;
  126. struct buffer_head *bh;
  127. struct buffer_head *head;
  128. struct buffer_head *ret = NULL;
  129. int err;
  130. u64 first_block = index << (PAGE_CACHE_SHIFT - blockbits);
  131. page = grab_cache_page(mapping, index);
  132. if (!page)
  133. return NULL;
  134. if (!page_has_buffers(page))
  135. create_empty_buffers(page, root->fs_info->sb->s_blocksize, 0);
  136. head = page_buffers(page);
  137. bh = head;
  138. do {
  139. if (!buffer_mapped(bh)) {
  140. err = btrfs_map_bh_to_logical(root, bh, first_block);
  141. BUG_ON(err);
  142. }
  143. if (bh_blocknr(bh) == blocknr) {
  144. ret = bh;
  145. get_bh(bh);
  146. goto out_unlock;
  147. }
  148. bh = bh->b_this_page;
  149. first_block++;
  150. } while (bh != head);
  151. out_unlock:
  152. unlock_page(page);
  153. if (ret)
  154. touch_buffer(ret);
  155. page_cache_release(page);
  156. return ret;
  157. }
  158. static int btree_get_block(struct inode *inode, sector_t iblock,
  159. struct buffer_head *bh, int create)
  160. {
  161. int err;
  162. struct btrfs_root *root = BTRFS_I(bh->b_page->mapping->host)->root;
  163. err = btrfs_map_bh_to_logical(root, bh, iblock);
  164. return err;
  165. }
  166. int btrfs_csum_data(struct btrfs_root * root, char *data, size_t len,
  167. char *result)
  168. {
  169. struct scatterlist sg;
  170. struct crypto_hash *tfm = root->fs_info->hash_tfm;
  171. struct hash_desc desc;
  172. int ret;
  173. desc.tfm = tfm;
  174. desc.flags = 0;
  175. sg_init_one(&sg, data, len);
  176. spin_lock(&root->fs_info->hash_lock);
  177. ret = crypto_hash_digest(&desc, &sg, 1, result);
  178. spin_unlock(&root->fs_info->hash_lock);
  179. if (ret) {
  180. printk("sha256 digest failed\n");
  181. }
  182. return ret;
  183. }
  184. static int csum_tree_block(struct btrfs_root *root, struct buffer_head *bh,
  185. int verify)
  186. {
  187. char result[BTRFS_CSUM_SIZE];
  188. int ret;
  189. struct btrfs_node *node;
  190. ret = btrfs_csum_data(root, bh->b_data + BTRFS_CSUM_SIZE,
  191. bh->b_size - BTRFS_CSUM_SIZE, result);
  192. if (ret)
  193. return ret;
  194. if (verify) {
  195. if (memcmp(bh->b_data, result, BTRFS_CSUM_SIZE)) {
  196. printk("checksum verify failed on %Lu\n",
  197. bh_blocknr(bh));
  198. return 1;
  199. }
  200. } else {
  201. node = btrfs_buffer_node(bh);
  202. memcpy(node->header.csum, result, BTRFS_CSUM_SIZE);
  203. }
  204. return 0;
  205. }
  206. static int btree_writepage(struct page *page, struct writeback_control *wbc)
  207. {
  208. struct buffer_head *bh;
  209. struct btrfs_root *root = BTRFS_I(page->mapping->host)->root;
  210. struct buffer_head *head;
  211. if (!page_has_buffers(page)) {
  212. create_empty_buffers(page, root->fs_info->sb->s_blocksize,
  213. (1 << BH_Dirty)|(1 << BH_Uptodate));
  214. }
  215. head = page_buffers(page);
  216. bh = head;
  217. do {
  218. if (buffer_dirty(bh))
  219. csum_tree_block(root, bh, 0);
  220. bh = bh->b_this_page;
  221. } while (bh != head);
  222. return block_write_full_page(page, btree_get_block, wbc);
  223. }
  224. static int btree_readpage(struct file * file, struct page * page)
  225. {
  226. return block_read_full_page(page, btree_get_block);
  227. }
  228. static struct address_space_operations btree_aops = {
  229. .readpage = btree_readpage,
  230. .writepage = btree_writepage,
  231. .sync_page = block_sync_page,
  232. };
  233. struct buffer_head *read_tree_block(struct btrfs_root *root, u64 blocknr)
  234. {
  235. struct buffer_head *bh = NULL;
  236. bh = btrfs_find_create_tree_block(root, blocknr);
  237. if (!bh)
  238. return bh;
  239. if (buffer_uptodate(bh))
  240. goto uptodate;
  241. lock_buffer(bh);
  242. if (!buffer_uptodate(bh)) {
  243. get_bh(bh);
  244. bh->b_end_io = end_buffer_read_sync;
  245. submit_bh(READ, bh);
  246. wait_on_buffer(bh);
  247. if (!buffer_uptodate(bh))
  248. goto fail;
  249. csum_tree_block(root, bh, 1);
  250. } else {
  251. unlock_buffer(bh);
  252. }
  253. uptodate:
  254. if (check_tree_block(root, bh))
  255. BUG();
  256. return bh;
  257. fail:
  258. brelse(bh);
  259. return NULL;
  260. }
  261. int dirty_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root,
  262. struct buffer_head *buf)
  263. {
  264. WARN_ON(atomic_read(&buf->b_count) == 0);
  265. mark_buffer_dirty(buf);
  266. return 0;
  267. }
  268. int clean_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root,
  269. struct buffer_head *buf)
  270. {
  271. WARN_ON(atomic_read(&buf->b_count) == 0);
  272. clear_buffer_dirty(buf);
  273. return 0;
  274. }
  275. static int __setup_root(int blocksize,
  276. struct btrfs_root *root,
  277. struct btrfs_fs_info *fs_info,
  278. u64 objectid)
  279. {
  280. root->node = NULL;
  281. root->inode = NULL;
  282. root->commit_root = NULL;
  283. root->blocksize = blocksize;
  284. root->ref_cows = 0;
  285. root->fs_info = fs_info;
  286. root->objectid = objectid;
  287. root->last_trans = 0;
  288. root->highest_inode = 0;
  289. root->last_inode_alloc = 0;
  290. memset(&root->root_key, 0, sizeof(root->root_key));
  291. memset(&root->root_item, 0, sizeof(root->root_item));
  292. return 0;
  293. }
  294. static int find_and_setup_root(int blocksize,
  295. struct btrfs_root *tree_root,
  296. struct btrfs_fs_info *fs_info,
  297. u64 objectid,
  298. struct btrfs_root *root)
  299. {
  300. int ret;
  301. __setup_root(blocksize, root, fs_info, objectid);
  302. ret = btrfs_find_last_root(tree_root, objectid,
  303. &root->root_item, &root->root_key);
  304. BUG_ON(ret);
  305. root->node = read_tree_block(root,
  306. btrfs_root_blocknr(&root->root_item));
  307. BUG_ON(!root->node);
  308. return 0;
  309. }
  310. struct btrfs_root *btrfs_read_fs_root(struct btrfs_fs_info *fs_info,
  311. struct btrfs_key *location)
  312. {
  313. struct btrfs_root *root;
  314. struct btrfs_root *tree_root = fs_info->tree_root;
  315. struct btrfs_path *path;
  316. struct btrfs_leaf *l;
  317. u64 highest_inode;
  318. int ret = 0;
  319. printk("read_fs_root looking for %Lu %Lu %u\n", location->objectid, location->offset, location->flags);
  320. root = radix_tree_lookup(&fs_info->fs_roots_radix,
  321. (unsigned long)location->objectid);
  322. if (root) {
  323. printk("found %p in cache\n", root);
  324. return root;
  325. }
  326. root = kmalloc(sizeof(*root), GFP_NOFS);
  327. if (!root) {
  328. printk("failed1\n");
  329. return ERR_PTR(-ENOMEM);
  330. }
  331. if (location->offset == (u64)-1) {
  332. ret = find_and_setup_root(fs_info->sb->s_blocksize,
  333. fs_info->tree_root, fs_info,
  334. location->objectid, root);
  335. if (ret) {
  336. printk("failed2\n");
  337. kfree(root);
  338. return ERR_PTR(ret);
  339. }
  340. goto insert;
  341. }
  342. __setup_root(fs_info->sb->s_blocksize, root, fs_info,
  343. location->objectid);
  344. path = btrfs_alloc_path();
  345. BUG_ON(!path);
  346. ret = btrfs_search_slot(NULL, tree_root, location, path, 0, 0);
  347. if (ret != 0) {
  348. printk("internal search_slot gives us %d\n", ret);
  349. if (ret > 0)
  350. ret = -ENOENT;
  351. goto out;
  352. }
  353. l = btrfs_buffer_leaf(path->nodes[0]);
  354. memcpy(&root->root_item,
  355. btrfs_item_ptr(l, path->slots[0], struct btrfs_root_item),
  356. sizeof(root->root_item));
  357. memcpy(&root->root_key, location, sizeof(*location));
  358. ret = 0;
  359. out:
  360. btrfs_release_path(root, path);
  361. btrfs_free_path(path);
  362. if (ret) {
  363. kfree(root);
  364. return ERR_PTR(ret);
  365. }
  366. root->node = read_tree_block(root,
  367. btrfs_root_blocknr(&root->root_item));
  368. BUG_ON(!root->node);
  369. insert:
  370. printk("inserting %p\n", root);
  371. root->ref_cows = 1;
  372. ret = radix_tree_insert(&fs_info->fs_roots_radix,
  373. (unsigned long)root->root_key.objectid,
  374. root);
  375. if (ret) {
  376. printk("radix_tree_insert gives us %d\n", ret);
  377. brelse(root->node);
  378. kfree(root);
  379. return ERR_PTR(ret);
  380. }
  381. ret = btrfs_find_highest_inode(root, &highest_inode);
  382. if (ret == 0) {
  383. root->highest_inode = highest_inode;
  384. root->last_inode_alloc = highest_inode;
  385. printk("highest inode is %Lu\n", highest_inode);
  386. }
  387. printk("all worked\n");
  388. return root;
  389. }
  390. static int btrfs_open_disk(struct btrfs_root *root, u64 device_id,
  391. u64 block_start, u64 num_blocks,
  392. char *filename, int name_len)
  393. {
  394. char *null_filename;
  395. struct block_device *bdev;
  396. int ret;
  397. null_filename = kmalloc(name_len + 1, GFP_NOFS);
  398. if (!null_filename)
  399. return -ENOMEM;
  400. memcpy(null_filename, filename, name_len);
  401. null_filename[name_len] = '\0';
  402. bdev = open_bdev_excl(null_filename, O_RDWR, root->fs_info->sb);
  403. if (IS_ERR(bdev)) {
  404. ret = PTR_ERR(bdev);
  405. goto out;
  406. }
  407. set_blocksize(bdev, root->fs_info->sb->s_blocksize);
  408. ret = btrfs_insert_dev_radix(root, bdev, device_id,
  409. block_start, num_blocks);
  410. BUG_ON(ret);
  411. ret = 0;
  412. out:
  413. kfree(null_filename);
  414. return ret;
  415. }
  416. static int read_device_info(struct btrfs_root *root)
  417. {
  418. struct btrfs_path *path;
  419. int ret;
  420. struct btrfs_key key;
  421. struct btrfs_leaf *leaf;
  422. struct btrfs_device_item *dev_item;
  423. int nritems;
  424. int slot;
  425. root = root->fs_info->dev_root;
  426. path = btrfs_alloc_path();
  427. if (!path)
  428. return -ENOMEM;
  429. key.objectid = 0;
  430. key.offset = 0;
  431. key.flags = 0;
  432. btrfs_set_key_type(&key, BTRFS_DEV_ITEM_KEY);
  433. mutex_lock(&root->fs_info->fs_mutex);
  434. ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
  435. leaf = btrfs_buffer_leaf(path->nodes[0]);
  436. nritems = btrfs_header_nritems(&leaf->header);
  437. while(1) {
  438. slot = path->slots[0];
  439. if (slot >= nritems) {
  440. ret = btrfs_next_leaf(root, path);
  441. if (ret)
  442. break;
  443. leaf = btrfs_buffer_leaf(path->nodes[0]);
  444. nritems = btrfs_header_nritems(&leaf->header);
  445. slot = path->slots[0];
  446. }
  447. btrfs_disk_key_to_cpu(&key, &leaf->items[slot].key);
  448. if (btrfs_key_type(&key) != BTRFS_DEV_ITEM_KEY) {
  449. path->slots[0]++;
  450. continue;
  451. }
  452. dev_item = btrfs_item_ptr(leaf, slot, struct btrfs_device_item);
  453. printk("found key %Lu %Lu\n", key.objectid, key.offset);
  454. if (btrfs_device_id(dev_item) !=
  455. btrfs_super_device_id(root->fs_info->disk_super)) {
  456. ret = btrfs_open_disk(root, btrfs_device_id(dev_item),
  457. key.objectid, key.offset,
  458. (char *)(dev_item + 1),
  459. btrfs_device_pathlen(dev_item));
  460. BUG_ON(ret);
  461. }
  462. path->slots[0]++;
  463. }
  464. btrfs_free_path(path);
  465. mutex_unlock(&root->fs_info->fs_mutex);
  466. return 0;
  467. }
  468. struct btrfs_root *open_ctree(struct super_block *sb)
  469. {
  470. struct btrfs_root *extent_root = kmalloc(sizeof(struct btrfs_root),
  471. GFP_NOFS);
  472. struct btrfs_root *dev_root = kmalloc(sizeof(struct btrfs_root),
  473. GFP_NOFS);
  474. struct btrfs_root *tree_root = kmalloc(sizeof(struct btrfs_root),
  475. GFP_NOFS);
  476. struct btrfs_fs_info *fs_info = kmalloc(sizeof(*fs_info),
  477. GFP_NOFS);
  478. int ret;
  479. struct btrfs_super_block *disk_super;
  480. struct dev_lookup *dev_lookup;
  481. init_bit_radix(&fs_info->pinned_radix);
  482. init_bit_radix(&fs_info->pending_del_radix);
  483. INIT_RADIX_TREE(&fs_info->fs_roots_radix, GFP_NOFS);
  484. INIT_RADIX_TREE(&fs_info->dev_radix, GFP_NOFS);
  485. sb_set_blocksize(sb, 4096);
  486. fs_info->running_transaction = NULL;
  487. fs_info->tree_root = tree_root;
  488. fs_info->extent_root = extent_root;
  489. fs_info->dev_root = dev_root;
  490. fs_info->sb = sb;
  491. fs_info->btree_inode = new_inode(sb);
  492. fs_info->btree_inode->i_ino = 1;
  493. fs_info->btree_inode->i_nlink = 1;
  494. fs_info->btree_inode->i_size = sb->s_bdev->bd_inode->i_size;
  495. fs_info->btree_inode->i_mapping->a_ops = &btree_aops;
  496. BTRFS_I(fs_info->btree_inode)->root = tree_root;
  497. memset(&BTRFS_I(fs_info->btree_inode)->location, 0,
  498. sizeof(struct btrfs_key));
  499. insert_inode_hash(fs_info->btree_inode);
  500. mapping_set_gfp_mask(fs_info->btree_inode->i_mapping, GFP_NOFS);
  501. fs_info->hash_tfm = crypto_alloc_hash("sha256", 0, CRYPTO_ALG_ASYNC);
  502. spin_lock_init(&fs_info->hash_lock);
  503. if (!fs_info->hash_tfm || IS_ERR(fs_info->hash_tfm)) {
  504. printk("failed to allocate sha256 hash\n");
  505. return NULL;
  506. }
  507. mutex_init(&fs_info->trans_mutex);
  508. mutex_init(&fs_info->fs_mutex);
  509. memset(&fs_info->current_insert, 0, sizeof(fs_info->current_insert));
  510. memset(&fs_info->last_insert, 0, sizeof(fs_info->last_insert));
  511. __setup_root(sb->s_blocksize, dev_root,
  512. fs_info, BTRFS_DEV_TREE_OBJECTID);
  513. __setup_root(sb->s_blocksize, tree_root,
  514. fs_info, BTRFS_ROOT_TREE_OBJECTID);
  515. dev_lookup = kmalloc(sizeof(*dev_lookup), GFP_NOFS);
  516. dev_lookup->block_start = 0;
  517. dev_lookup->num_blocks = (u32)-2;
  518. dev_lookup->bdev = sb->s_bdev;
  519. dev_lookup->device_id = 0;
  520. ret = radix_tree_insert(&fs_info->dev_radix, (u32)-2, dev_lookup);
  521. BUG_ON(ret);
  522. fs_info->sb_buffer = read_tree_block(tree_root,
  523. BTRFS_SUPER_INFO_OFFSET /
  524. sb->s_blocksize);
  525. if (!fs_info->sb_buffer)
  526. return NULL;
  527. disk_super = (struct btrfs_super_block *)fs_info->sb_buffer->b_data;
  528. if (!btrfs_super_root(disk_super))
  529. return NULL;
  530. i_size_write(fs_info->btree_inode,
  531. btrfs_super_total_blocks(disk_super) <<
  532. fs_info->btree_inode->i_blkbits);
  533. radix_tree_delete(&fs_info->dev_radix, (u32)-2);
  534. dev_lookup->block_start = btrfs_super_device_block_start(disk_super);
  535. dev_lookup->num_blocks = btrfs_super_device_num_blocks(disk_super);
  536. dev_lookup->device_id = btrfs_super_device_id(disk_super);
  537. ret = radix_tree_insert(&fs_info->dev_radix,
  538. dev_lookup->block_start +
  539. dev_lookup->num_blocks - 1, dev_lookup);
  540. BUG_ON(ret);
  541. fs_info->disk_super = disk_super;
  542. dev_root->node = read_tree_block(tree_root,
  543. btrfs_super_device_root(disk_super));
  544. ret = read_device_info(dev_root);
  545. BUG_ON(ret);
  546. tree_root->node = read_tree_block(tree_root,
  547. btrfs_super_root(disk_super));
  548. BUG_ON(!tree_root->node);
  549. mutex_lock(&fs_info->fs_mutex);
  550. ret = find_and_setup_root(sb->s_blocksize, tree_root, fs_info,
  551. BTRFS_EXTENT_TREE_OBJECTID, extent_root);
  552. BUG_ON(ret);
  553. fs_info->generation = btrfs_super_generation(disk_super) + 1;
  554. memset(&fs_info->kobj, 0, sizeof(fs_info->kobj));
  555. kobj_set_kset_s(fs_info, btrfs_subsys);
  556. kobject_set_name(&fs_info->kobj, "%s", sb->s_id);
  557. kobject_register(&fs_info->kobj);
  558. mutex_unlock(&fs_info->fs_mutex);
  559. return tree_root;
  560. }
  561. int write_ctree_super(struct btrfs_trans_handle *trans, struct btrfs_root
  562. *root)
  563. {
  564. struct buffer_head *bh = root->fs_info->sb_buffer;
  565. btrfs_set_super_root(root->fs_info->disk_super,
  566. bh_blocknr(root->fs_info->tree_root->node));
  567. lock_buffer(bh);
  568. WARN_ON(atomic_read(&bh->b_count) < 1);
  569. clear_buffer_dirty(bh);
  570. csum_tree_block(root, bh, 0);
  571. bh->b_end_io = end_buffer_write_sync;
  572. get_bh(bh);
  573. submit_bh(WRITE, bh);
  574. wait_on_buffer(bh);
  575. if (!buffer_uptodate(bh)) {
  576. WARN_ON(1);
  577. return -EIO;
  578. }
  579. return 0;
  580. }
  581. static int free_fs_root(struct btrfs_fs_info *fs_info, struct btrfs_root *root)
  582. {
  583. radix_tree_delete(&fs_info->fs_roots_radix,
  584. (unsigned long)root->root_key.objectid);
  585. if (root->inode)
  586. iput(root->inode);
  587. if (root->node)
  588. brelse(root->node);
  589. if (root->commit_root)
  590. brelse(root->commit_root);
  591. kfree(root);
  592. return 0;
  593. }
  594. int del_fs_roots(struct btrfs_fs_info *fs_info)
  595. {
  596. int ret;
  597. struct btrfs_root *gang[8];
  598. int i;
  599. while(1) {
  600. ret = radix_tree_gang_lookup(&fs_info->fs_roots_radix,
  601. (void **)gang, 0,
  602. ARRAY_SIZE(gang));
  603. if (!ret)
  604. break;
  605. for (i = 0; i < ret; i++)
  606. free_fs_root(fs_info, gang[i]);
  607. }
  608. return 0;
  609. }
  610. static int free_dev_radix(struct btrfs_fs_info *fs_info)
  611. {
  612. struct dev_lookup *lookup[8];
  613. struct block_device *super_bdev = fs_info->sb->s_bdev;
  614. int ret;
  615. int i;
  616. while(1) {
  617. ret = radix_tree_gang_lookup(&fs_info->dev_radix,
  618. (void **)lookup, 0,
  619. ARRAY_SIZE(lookup));
  620. if (!ret)
  621. break;
  622. for (i = 0; i < ret; i++) {
  623. if (lookup[i]->bdev != super_bdev)
  624. close_bdev_excl(lookup[i]->bdev);
  625. radix_tree_delete(&fs_info->dev_radix,
  626. lookup[i]->block_start +
  627. lookup[i]->num_blocks - 1);
  628. kfree(lookup[i]);
  629. }
  630. }
  631. return 0;
  632. }
  633. int close_ctree(struct btrfs_root *root)
  634. {
  635. int ret;
  636. struct btrfs_trans_handle *trans;
  637. struct btrfs_fs_info *fs_info = root->fs_info;
  638. mutex_lock(&fs_info->fs_mutex);
  639. trans = btrfs_start_transaction(root, 1);
  640. btrfs_commit_transaction(trans, root);
  641. /* run commit again to drop the original snapshot */
  642. trans = btrfs_start_transaction(root, 1);
  643. btrfs_commit_transaction(trans, root);
  644. ret = btrfs_write_and_wait_transaction(NULL, root);
  645. BUG_ON(ret);
  646. write_ctree_super(NULL, root);
  647. mutex_unlock(&fs_info->fs_mutex);
  648. if (fs_info->extent_root->node)
  649. btrfs_block_release(fs_info->extent_root,
  650. fs_info->extent_root->node);
  651. if (fs_info->dev_root->node)
  652. btrfs_block_release(fs_info->dev_root,
  653. fs_info->dev_root->node);
  654. if (fs_info->tree_root->node)
  655. btrfs_block_release(fs_info->tree_root,
  656. fs_info->tree_root->node);
  657. btrfs_block_release(root, fs_info->sb_buffer);
  658. crypto_free_hash(fs_info->hash_tfm);
  659. truncate_inode_pages(fs_info->btree_inode->i_mapping, 0);
  660. iput(fs_info->btree_inode);
  661. free_dev_radix(fs_info);
  662. del_fs_roots(fs_info);
  663. kfree(fs_info->extent_root);
  664. kfree(fs_info->tree_root);
  665. kobject_unregister(&fs_info->kobj);
  666. return 0;
  667. }
  668. void btrfs_block_release(struct btrfs_root *root, struct buffer_head *buf)
  669. {
  670. brelse(buf);
  671. }