file-item.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860
  1. /*
  2. * Copyright (C) 2007 Oracle. 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/bio.h>
  19. #include <linux/slab.h>
  20. #include <linux/pagemap.h>
  21. #include <linux/highmem.h>
  22. #include "ctree.h"
  23. #include "disk-io.h"
  24. #include "transaction.h"
  25. #include "print-tree.h"
  26. #define MAX_CSUM_ITEMS(r, size) ((((BTRFS_LEAF_DATA_SIZE(r) - \
  27. sizeof(struct btrfs_item) * 2) / \
  28. size) - 1))
  29. #define MAX_ORDERED_SUM_BYTES(r) ((PAGE_SIZE - \
  30. sizeof(struct btrfs_ordered_sum)) / \
  31. sizeof(struct btrfs_sector_sum) * \
  32. (r)->sectorsize - (r)->sectorsize)
  33. int btrfs_insert_file_extent(struct btrfs_trans_handle *trans,
  34. struct btrfs_root *root,
  35. u64 objectid, u64 pos,
  36. u64 disk_offset, u64 disk_num_bytes,
  37. u64 num_bytes, u64 offset, u64 ram_bytes,
  38. u8 compression, u8 encryption, u16 other_encoding)
  39. {
  40. int ret = 0;
  41. struct btrfs_file_extent_item *item;
  42. struct btrfs_key file_key;
  43. struct btrfs_path *path;
  44. struct extent_buffer *leaf;
  45. path = btrfs_alloc_path();
  46. BUG_ON(!path);
  47. file_key.objectid = objectid;
  48. file_key.offset = pos;
  49. btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY);
  50. path->leave_spinning = 1;
  51. ret = btrfs_insert_empty_item(trans, root, path, &file_key,
  52. sizeof(*item));
  53. if (ret < 0)
  54. goto out;
  55. BUG_ON(ret);
  56. leaf = path->nodes[0];
  57. item = btrfs_item_ptr(leaf, path->slots[0],
  58. struct btrfs_file_extent_item);
  59. btrfs_set_file_extent_disk_bytenr(leaf, item, disk_offset);
  60. btrfs_set_file_extent_disk_num_bytes(leaf, item, disk_num_bytes);
  61. btrfs_set_file_extent_offset(leaf, item, offset);
  62. btrfs_set_file_extent_num_bytes(leaf, item, num_bytes);
  63. btrfs_set_file_extent_ram_bytes(leaf, item, ram_bytes);
  64. btrfs_set_file_extent_generation(leaf, item, trans->transid);
  65. btrfs_set_file_extent_type(leaf, item, BTRFS_FILE_EXTENT_REG);
  66. btrfs_set_file_extent_compression(leaf, item, compression);
  67. btrfs_set_file_extent_encryption(leaf, item, encryption);
  68. btrfs_set_file_extent_other_encoding(leaf, item, other_encoding);
  69. btrfs_mark_buffer_dirty(leaf);
  70. out:
  71. btrfs_free_path(path);
  72. return ret;
  73. }
  74. struct btrfs_csum_item *btrfs_lookup_csum(struct btrfs_trans_handle *trans,
  75. struct btrfs_root *root,
  76. struct btrfs_path *path,
  77. u64 bytenr, int cow)
  78. {
  79. int ret;
  80. struct btrfs_key file_key;
  81. struct btrfs_key found_key;
  82. struct btrfs_csum_item *item;
  83. struct extent_buffer *leaf;
  84. u64 csum_offset = 0;
  85. u16 csum_size =
  86. btrfs_super_csum_size(&root->fs_info->super_copy);
  87. int csums_in_item;
  88. file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
  89. file_key.offset = bytenr;
  90. btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY);
  91. ret = btrfs_search_slot(trans, root, &file_key, path, 0, cow);
  92. if (ret < 0)
  93. goto fail;
  94. leaf = path->nodes[0];
  95. if (ret > 0) {
  96. ret = 1;
  97. if (path->slots[0] == 0)
  98. goto fail;
  99. path->slots[0]--;
  100. btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
  101. if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY)
  102. goto fail;
  103. csum_offset = (bytenr - found_key.offset) >>
  104. root->fs_info->sb->s_blocksize_bits;
  105. csums_in_item = btrfs_item_size_nr(leaf, path->slots[0]);
  106. csums_in_item /= csum_size;
  107. if (csum_offset >= csums_in_item) {
  108. ret = -EFBIG;
  109. goto fail;
  110. }
  111. }
  112. item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
  113. item = (struct btrfs_csum_item *)((unsigned char *)item +
  114. csum_offset * csum_size);
  115. return item;
  116. fail:
  117. if (ret > 0)
  118. ret = -ENOENT;
  119. return ERR_PTR(ret);
  120. }
  121. int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans,
  122. struct btrfs_root *root,
  123. struct btrfs_path *path, u64 objectid,
  124. u64 offset, int mod)
  125. {
  126. int ret;
  127. struct btrfs_key file_key;
  128. int ins_len = mod < 0 ? -1 : 0;
  129. int cow = mod != 0;
  130. file_key.objectid = objectid;
  131. file_key.offset = offset;
  132. btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY);
  133. ret = btrfs_search_slot(trans, root, &file_key, path, ins_len, cow);
  134. return ret;
  135. }
  136. static int __btrfs_lookup_bio_sums(struct btrfs_root *root,
  137. struct inode *inode, struct bio *bio,
  138. u64 logical_offset, u32 *dst, int dio)
  139. {
  140. u32 sum;
  141. struct bio_vec *bvec = bio->bi_io_vec;
  142. int bio_index = 0;
  143. u64 offset = 0;
  144. u64 item_start_offset = 0;
  145. u64 item_last_offset = 0;
  146. u64 disk_bytenr;
  147. u32 diff;
  148. u16 csum_size =
  149. btrfs_super_csum_size(&root->fs_info->super_copy);
  150. int ret;
  151. struct btrfs_path *path;
  152. struct btrfs_csum_item *item = NULL;
  153. struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
  154. path = btrfs_alloc_path();
  155. if (bio->bi_size > PAGE_CACHE_SIZE * 8)
  156. path->reada = 2;
  157. WARN_ON(bio->bi_vcnt <= 0);
  158. disk_bytenr = (u64)bio->bi_sector << 9;
  159. if (dio)
  160. offset = logical_offset;
  161. while (bio_index < bio->bi_vcnt) {
  162. if (!dio)
  163. offset = page_offset(bvec->bv_page) + bvec->bv_offset;
  164. ret = btrfs_find_ordered_sum(inode, offset, disk_bytenr, &sum);
  165. if (ret == 0)
  166. goto found;
  167. if (!item || disk_bytenr < item_start_offset ||
  168. disk_bytenr >= item_last_offset) {
  169. struct btrfs_key found_key;
  170. u32 item_size;
  171. if (item)
  172. btrfs_release_path(root, path);
  173. item = btrfs_lookup_csum(NULL, root->fs_info->csum_root,
  174. path, disk_bytenr, 0);
  175. if (IS_ERR(item)) {
  176. ret = PTR_ERR(item);
  177. if (ret == -ENOENT || ret == -EFBIG)
  178. ret = 0;
  179. sum = 0;
  180. if (BTRFS_I(inode)->root->root_key.objectid ==
  181. BTRFS_DATA_RELOC_TREE_OBJECTID) {
  182. set_extent_bits(io_tree, offset,
  183. offset + bvec->bv_len - 1,
  184. EXTENT_NODATASUM, GFP_NOFS);
  185. } else {
  186. printk(KERN_INFO "btrfs no csum found "
  187. "for inode %lu start %llu\n",
  188. inode->i_ino,
  189. (unsigned long long)offset);
  190. }
  191. item = NULL;
  192. btrfs_release_path(root, path);
  193. goto found;
  194. }
  195. btrfs_item_key_to_cpu(path->nodes[0], &found_key,
  196. path->slots[0]);
  197. item_start_offset = found_key.offset;
  198. item_size = btrfs_item_size_nr(path->nodes[0],
  199. path->slots[0]);
  200. item_last_offset = item_start_offset +
  201. (item_size / csum_size) *
  202. root->sectorsize;
  203. item = btrfs_item_ptr(path->nodes[0], path->slots[0],
  204. struct btrfs_csum_item);
  205. }
  206. /*
  207. * this byte range must be able to fit inside
  208. * a single leaf so it will also fit inside a u32
  209. */
  210. diff = disk_bytenr - item_start_offset;
  211. diff = diff / root->sectorsize;
  212. diff = diff * csum_size;
  213. read_extent_buffer(path->nodes[0], &sum,
  214. ((unsigned long)item) + diff,
  215. csum_size);
  216. found:
  217. if (dst)
  218. *dst++ = sum;
  219. else
  220. set_state_private(io_tree, offset, sum);
  221. disk_bytenr += bvec->bv_len;
  222. offset += bvec->bv_len;
  223. bio_index++;
  224. bvec++;
  225. }
  226. btrfs_free_path(path);
  227. return 0;
  228. }
  229. int btrfs_lookup_bio_sums(struct btrfs_root *root, struct inode *inode,
  230. struct bio *bio, u32 *dst)
  231. {
  232. return __btrfs_lookup_bio_sums(root, inode, bio, 0, dst, 0);
  233. }
  234. int btrfs_lookup_bio_sums_dio(struct btrfs_root *root, struct inode *inode,
  235. struct bio *bio, u64 offset, u32 *dst)
  236. {
  237. return __btrfs_lookup_bio_sums(root, inode, bio, offset, dst, 1);
  238. }
  239. int btrfs_lookup_csums_range(struct btrfs_root *root, u64 start, u64 end,
  240. struct list_head *list)
  241. {
  242. struct btrfs_key key;
  243. struct btrfs_path *path;
  244. struct extent_buffer *leaf;
  245. struct btrfs_ordered_sum *sums;
  246. struct btrfs_sector_sum *sector_sum;
  247. struct btrfs_csum_item *item;
  248. unsigned long offset;
  249. int ret;
  250. size_t size;
  251. u64 csum_end;
  252. u16 csum_size = btrfs_super_csum_size(&root->fs_info->super_copy);
  253. path = btrfs_alloc_path();
  254. BUG_ON(!path);
  255. key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
  256. key.offset = start;
  257. key.type = BTRFS_EXTENT_CSUM_KEY;
  258. ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
  259. if (ret < 0)
  260. goto fail;
  261. if (ret > 0 && path->slots[0] > 0) {
  262. leaf = path->nodes[0];
  263. btrfs_item_key_to_cpu(leaf, &key, path->slots[0] - 1);
  264. if (key.objectid == BTRFS_EXTENT_CSUM_OBJECTID &&
  265. key.type == BTRFS_EXTENT_CSUM_KEY) {
  266. offset = (start - key.offset) >>
  267. root->fs_info->sb->s_blocksize_bits;
  268. if (offset * csum_size <
  269. btrfs_item_size_nr(leaf, path->slots[0] - 1))
  270. path->slots[0]--;
  271. }
  272. }
  273. while (start <= end) {
  274. leaf = path->nodes[0];
  275. if (path->slots[0] >= btrfs_header_nritems(leaf)) {
  276. ret = btrfs_next_leaf(root, path);
  277. if (ret < 0)
  278. goto fail;
  279. if (ret > 0)
  280. break;
  281. leaf = path->nodes[0];
  282. }
  283. btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
  284. if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
  285. key.type != BTRFS_EXTENT_CSUM_KEY)
  286. break;
  287. btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
  288. if (key.offset > end)
  289. break;
  290. if (key.offset > start)
  291. start = key.offset;
  292. size = btrfs_item_size_nr(leaf, path->slots[0]);
  293. csum_end = key.offset + (size / csum_size) * root->sectorsize;
  294. if (csum_end <= start) {
  295. path->slots[0]++;
  296. continue;
  297. }
  298. csum_end = min(csum_end, end + 1);
  299. item = btrfs_item_ptr(path->nodes[0], path->slots[0],
  300. struct btrfs_csum_item);
  301. while (start < csum_end) {
  302. size = min_t(size_t, csum_end - start,
  303. MAX_ORDERED_SUM_BYTES(root));
  304. sums = kzalloc(btrfs_ordered_sum_size(root, size),
  305. GFP_NOFS);
  306. BUG_ON(!sums);
  307. sector_sum = sums->sums;
  308. sums->bytenr = start;
  309. sums->len = size;
  310. offset = (start - key.offset) >>
  311. root->fs_info->sb->s_blocksize_bits;
  312. offset *= csum_size;
  313. while (size > 0) {
  314. read_extent_buffer(path->nodes[0],
  315. &sector_sum->sum,
  316. ((unsigned long)item) +
  317. offset, csum_size);
  318. sector_sum->bytenr = start;
  319. size -= root->sectorsize;
  320. start += root->sectorsize;
  321. offset += csum_size;
  322. sector_sum++;
  323. }
  324. list_add_tail(&sums->list, list);
  325. }
  326. path->slots[0]++;
  327. }
  328. ret = 0;
  329. fail:
  330. btrfs_free_path(path);
  331. return ret;
  332. }
  333. int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode,
  334. struct bio *bio, u64 file_start, int contig)
  335. {
  336. struct btrfs_ordered_sum *sums;
  337. struct btrfs_sector_sum *sector_sum;
  338. struct btrfs_ordered_extent *ordered;
  339. char *data;
  340. struct bio_vec *bvec = bio->bi_io_vec;
  341. int bio_index = 0;
  342. unsigned long total_bytes = 0;
  343. unsigned long this_sum_bytes = 0;
  344. u64 offset;
  345. u64 disk_bytenr;
  346. WARN_ON(bio->bi_vcnt <= 0);
  347. sums = kzalloc(btrfs_ordered_sum_size(root, bio->bi_size), GFP_NOFS);
  348. if (!sums)
  349. return -ENOMEM;
  350. sector_sum = sums->sums;
  351. disk_bytenr = (u64)bio->bi_sector << 9;
  352. sums->len = bio->bi_size;
  353. INIT_LIST_HEAD(&sums->list);
  354. if (contig)
  355. offset = file_start;
  356. else
  357. offset = page_offset(bvec->bv_page) + bvec->bv_offset;
  358. ordered = btrfs_lookup_ordered_extent(inode, offset);
  359. BUG_ON(!ordered);
  360. sums->bytenr = ordered->start;
  361. while (bio_index < bio->bi_vcnt) {
  362. if (!contig)
  363. offset = page_offset(bvec->bv_page) + bvec->bv_offset;
  364. if (!contig && (offset >= ordered->file_offset + ordered->len ||
  365. offset < ordered->file_offset)) {
  366. unsigned long bytes_left;
  367. sums->len = this_sum_bytes;
  368. this_sum_bytes = 0;
  369. btrfs_add_ordered_sum(inode, ordered, sums);
  370. btrfs_put_ordered_extent(ordered);
  371. bytes_left = bio->bi_size - total_bytes;
  372. sums = kzalloc(btrfs_ordered_sum_size(root, bytes_left),
  373. GFP_NOFS);
  374. BUG_ON(!sums);
  375. sector_sum = sums->sums;
  376. sums->len = bytes_left;
  377. ordered = btrfs_lookup_ordered_extent(inode, offset);
  378. BUG_ON(!ordered);
  379. sums->bytenr = ordered->start;
  380. }
  381. data = kmap_atomic(bvec->bv_page, KM_USER0);
  382. sector_sum->sum = ~(u32)0;
  383. sector_sum->sum = btrfs_csum_data(root,
  384. data + bvec->bv_offset,
  385. sector_sum->sum,
  386. bvec->bv_len);
  387. kunmap_atomic(data, KM_USER0);
  388. btrfs_csum_final(sector_sum->sum,
  389. (char *)&sector_sum->sum);
  390. sector_sum->bytenr = disk_bytenr;
  391. sector_sum++;
  392. bio_index++;
  393. total_bytes += bvec->bv_len;
  394. this_sum_bytes += bvec->bv_len;
  395. disk_bytenr += bvec->bv_len;
  396. offset += bvec->bv_len;
  397. bvec++;
  398. }
  399. this_sum_bytes = 0;
  400. btrfs_add_ordered_sum(inode, ordered, sums);
  401. btrfs_put_ordered_extent(ordered);
  402. return 0;
  403. }
  404. /*
  405. * helper function for csum removal, this expects the
  406. * key to describe the csum pointed to by the path, and it expects
  407. * the csum to overlap the range [bytenr, len]
  408. *
  409. * The csum should not be entirely contained in the range and the
  410. * range should not be entirely contained in the csum.
  411. *
  412. * This calls btrfs_truncate_item with the correct args based on the
  413. * overlap, and fixes up the key as required.
  414. */
  415. static noinline int truncate_one_csum(struct btrfs_trans_handle *trans,
  416. struct btrfs_root *root,
  417. struct btrfs_path *path,
  418. struct btrfs_key *key,
  419. u64 bytenr, u64 len)
  420. {
  421. struct extent_buffer *leaf;
  422. u16 csum_size =
  423. btrfs_super_csum_size(&root->fs_info->super_copy);
  424. u64 csum_end;
  425. u64 end_byte = bytenr + len;
  426. u32 blocksize_bits = root->fs_info->sb->s_blocksize_bits;
  427. int ret;
  428. leaf = path->nodes[0];
  429. csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size;
  430. csum_end <<= root->fs_info->sb->s_blocksize_bits;
  431. csum_end += key->offset;
  432. if (key->offset < bytenr && csum_end <= end_byte) {
  433. /*
  434. * [ bytenr - len ]
  435. * [ ]
  436. * [csum ]
  437. * A simple truncate off the end of the item
  438. */
  439. u32 new_size = (bytenr - key->offset) >> blocksize_bits;
  440. new_size *= csum_size;
  441. ret = btrfs_truncate_item(trans, root, path, new_size, 1);
  442. BUG_ON(ret);
  443. } else if (key->offset >= bytenr && csum_end > end_byte &&
  444. end_byte > key->offset) {
  445. /*
  446. * [ bytenr - len ]
  447. * [ ]
  448. * [csum ]
  449. * we need to truncate from the beginning of the csum
  450. */
  451. u32 new_size = (csum_end - end_byte) >> blocksize_bits;
  452. new_size *= csum_size;
  453. ret = btrfs_truncate_item(trans, root, path, new_size, 0);
  454. BUG_ON(ret);
  455. key->offset = end_byte;
  456. ret = btrfs_set_item_key_safe(trans, root, path, key);
  457. BUG_ON(ret);
  458. } else {
  459. BUG();
  460. }
  461. return 0;
  462. }
  463. /*
  464. * deletes the csum items from the csum tree for a given
  465. * range of bytes.
  466. */
  467. int btrfs_del_csums(struct btrfs_trans_handle *trans,
  468. struct btrfs_root *root, u64 bytenr, u64 len)
  469. {
  470. struct btrfs_path *path;
  471. struct btrfs_key key;
  472. u64 end_byte = bytenr + len;
  473. u64 csum_end;
  474. struct extent_buffer *leaf;
  475. int ret;
  476. u16 csum_size =
  477. btrfs_super_csum_size(&root->fs_info->super_copy);
  478. int blocksize_bits = root->fs_info->sb->s_blocksize_bits;
  479. root = root->fs_info->csum_root;
  480. path = btrfs_alloc_path();
  481. if (!path)
  482. return -ENOMEM;
  483. while (1) {
  484. key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
  485. key.offset = end_byte - 1;
  486. key.type = BTRFS_EXTENT_CSUM_KEY;
  487. path->leave_spinning = 1;
  488. ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
  489. if (ret > 0) {
  490. if (path->slots[0] == 0)
  491. goto out;
  492. path->slots[0]--;
  493. } else if (ret < 0) {
  494. goto out;
  495. }
  496. leaf = path->nodes[0];
  497. btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
  498. if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
  499. key.type != BTRFS_EXTENT_CSUM_KEY) {
  500. break;
  501. }
  502. if (key.offset >= end_byte)
  503. break;
  504. csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size;
  505. csum_end <<= blocksize_bits;
  506. csum_end += key.offset;
  507. /* this csum ends before we start, we're done */
  508. if (csum_end <= bytenr)
  509. break;
  510. /* delete the entire item, it is inside our range */
  511. if (key.offset >= bytenr && csum_end <= end_byte) {
  512. ret = btrfs_del_item(trans, root, path);
  513. BUG_ON(ret);
  514. if (key.offset == bytenr)
  515. break;
  516. } else if (key.offset < bytenr && csum_end > end_byte) {
  517. unsigned long offset;
  518. unsigned long shift_len;
  519. unsigned long item_offset;
  520. /*
  521. * [ bytenr - len ]
  522. * [csum ]
  523. *
  524. * Our bytes are in the middle of the csum,
  525. * we need to split this item and insert a new one.
  526. *
  527. * But we can't drop the path because the
  528. * csum could change, get removed, extended etc.
  529. *
  530. * The trick here is the max size of a csum item leaves
  531. * enough room in the tree block for a single
  532. * item header. So, we split the item in place,
  533. * adding a new header pointing to the existing
  534. * bytes. Then we loop around again and we have
  535. * a nicely formed csum item that we can neatly
  536. * truncate.
  537. */
  538. offset = (bytenr - key.offset) >> blocksize_bits;
  539. offset *= csum_size;
  540. shift_len = (len >> blocksize_bits) * csum_size;
  541. item_offset = btrfs_item_ptr_offset(leaf,
  542. path->slots[0]);
  543. memset_extent_buffer(leaf, 0, item_offset + offset,
  544. shift_len);
  545. key.offset = bytenr;
  546. /*
  547. * btrfs_split_item returns -EAGAIN when the
  548. * item changed size or key
  549. */
  550. ret = btrfs_split_item(trans, root, path, &key, offset);
  551. BUG_ON(ret && ret != -EAGAIN);
  552. key.offset = end_byte - 1;
  553. } else {
  554. ret = truncate_one_csum(trans, root, path,
  555. &key, bytenr, len);
  556. BUG_ON(ret);
  557. if (key.offset < bytenr)
  558. break;
  559. }
  560. btrfs_release_path(root, path);
  561. }
  562. out:
  563. btrfs_free_path(path);
  564. return 0;
  565. }
  566. int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans,
  567. struct btrfs_root *root,
  568. struct btrfs_ordered_sum *sums)
  569. {
  570. u64 bytenr;
  571. int ret;
  572. struct btrfs_key file_key;
  573. struct btrfs_key found_key;
  574. u64 next_offset;
  575. u64 total_bytes = 0;
  576. int found_next;
  577. struct btrfs_path *path;
  578. struct btrfs_csum_item *item;
  579. struct btrfs_csum_item *item_end;
  580. struct extent_buffer *leaf = NULL;
  581. u64 csum_offset;
  582. struct btrfs_sector_sum *sector_sum;
  583. u32 nritems;
  584. u32 ins_size;
  585. char *eb_map;
  586. char *eb_token;
  587. unsigned long map_len;
  588. unsigned long map_start;
  589. u16 csum_size =
  590. btrfs_super_csum_size(&root->fs_info->super_copy);
  591. path = btrfs_alloc_path();
  592. BUG_ON(!path);
  593. sector_sum = sums->sums;
  594. again:
  595. next_offset = (u64)-1;
  596. found_next = 0;
  597. file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
  598. file_key.offset = sector_sum->bytenr;
  599. bytenr = sector_sum->bytenr;
  600. btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY);
  601. item = btrfs_lookup_csum(trans, root, path, sector_sum->bytenr, 1);
  602. if (!IS_ERR(item)) {
  603. leaf = path->nodes[0];
  604. ret = 0;
  605. goto found;
  606. }
  607. ret = PTR_ERR(item);
  608. if (ret != -EFBIG && ret != -ENOENT)
  609. goto fail_unlock;
  610. if (ret == -EFBIG) {
  611. u32 item_size;
  612. /* we found one, but it isn't big enough yet */
  613. leaf = path->nodes[0];
  614. item_size = btrfs_item_size_nr(leaf, path->slots[0]);
  615. if ((item_size / csum_size) >=
  616. MAX_CSUM_ITEMS(root, csum_size)) {
  617. /* already at max size, make a new one */
  618. goto insert;
  619. }
  620. } else {
  621. int slot = path->slots[0] + 1;
  622. /* we didn't find a csum item, insert one */
  623. nritems = btrfs_header_nritems(path->nodes[0]);
  624. if (path->slots[0] >= nritems - 1) {
  625. ret = btrfs_next_leaf(root, path);
  626. if (ret == 1)
  627. found_next = 1;
  628. if (ret != 0)
  629. goto insert;
  630. slot = 0;
  631. }
  632. btrfs_item_key_to_cpu(path->nodes[0], &found_key, slot);
  633. if (found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
  634. found_key.type != BTRFS_EXTENT_CSUM_KEY) {
  635. found_next = 1;
  636. goto insert;
  637. }
  638. next_offset = found_key.offset;
  639. found_next = 1;
  640. goto insert;
  641. }
  642. /*
  643. * at this point, we know the tree has an item, but it isn't big
  644. * enough yet to put our csum in. Grow it
  645. */
  646. btrfs_release_path(root, path);
  647. ret = btrfs_search_slot(trans, root, &file_key, path,
  648. csum_size, 1);
  649. if (ret < 0)
  650. goto fail_unlock;
  651. if (ret > 0) {
  652. if (path->slots[0] == 0)
  653. goto insert;
  654. path->slots[0]--;
  655. }
  656. leaf = path->nodes[0];
  657. btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
  658. csum_offset = (bytenr - found_key.offset) >>
  659. root->fs_info->sb->s_blocksize_bits;
  660. if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY ||
  661. found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
  662. csum_offset >= MAX_CSUM_ITEMS(root, csum_size)) {
  663. goto insert;
  664. }
  665. if (csum_offset >= btrfs_item_size_nr(leaf, path->slots[0]) /
  666. csum_size) {
  667. u32 diff = (csum_offset + 1) * csum_size;
  668. /*
  669. * is the item big enough already? we dropped our lock
  670. * before and need to recheck
  671. */
  672. if (diff < btrfs_item_size_nr(leaf, path->slots[0]))
  673. goto csum;
  674. diff = diff - btrfs_item_size_nr(leaf, path->slots[0]);
  675. if (diff != csum_size)
  676. goto insert;
  677. ret = btrfs_extend_item(trans, root, path, diff);
  678. BUG_ON(ret);
  679. goto csum;
  680. }
  681. insert:
  682. btrfs_release_path(root, path);
  683. csum_offset = 0;
  684. if (found_next) {
  685. u64 tmp = total_bytes + root->sectorsize;
  686. u64 next_sector = sector_sum->bytenr;
  687. struct btrfs_sector_sum *next = sector_sum + 1;
  688. while (tmp < sums->len) {
  689. if (next_sector + root->sectorsize != next->bytenr)
  690. break;
  691. tmp += root->sectorsize;
  692. next_sector = next->bytenr;
  693. next++;
  694. }
  695. tmp = min(tmp, next_offset - file_key.offset);
  696. tmp >>= root->fs_info->sb->s_blocksize_bits;
  697. tmp = max((u64)1, tmp);
  698. tmp = min(tmp, (u64)MAX_CSUM_ITEMS(root, csum_size));
  699. ins_size = csum_size * tmp;
  700. } else {
  701. ins_size = csum_size;
  702. }
  703. path->leave_spinning = 1;
  704. ret = btrfs_insert_empty_item(trans, root, path, &file_key,
  705. ins_size);
  706. path->leave_spinning = 0;
  707. if (ret < 0)
  708. goto fail_unlock;
  709. if (ret != 0) {
  710. WARN_ON(1);
  711. goto fail_unlock;
  712. }
  713. csum:
  714. leaf = path->nodes[0];
  715. item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
  716. ret = 0;
  717. item = (struct btrfs_csum_item *)((unsigned char *)item +
  718. csum_offset * csum_size);
  719. found:
  720. item_end = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
  721. item_end = (struct btrfs_csum_item *)((unsigned char *)item_end +
  722. btrfs_item_size_nr(leaf, path->slots[0]));
  723. eb_token = NULL;
  724. next_sector:
  725. if (!eb_token ||
  726. (unsigned long)item + csum_size >= map_start + map_len) {
  727. int err;
  728. if (eb_token)
  729. unmap_extent_buffer(leaf, eb_token, KM_USER1);
  730. eb_token = NULL;
  731. err = map_private_extent_buffer(leaf, (unsigned long)item,
  732. csum_size,
  733. &eb_token, &eb_map,
  734. &map_start, &map_len, KM_USER1);
  735. if (err)
  736. eb_token = NULL;
  737. }
  738. if (eb_token) {
  739. memcpy(eb_token + ((unsigned long)item & (PAGE_CACHE_SIZE - 1)),
  740. &sector_sum->sum, csum_size);
  741. } else {
  742. write_extent_buffer(leaf, &sector_sum->sum,
  743. (unsigned long)item, csum_size);
  744. }
  745. total_bytes += root->sectorsize;
  746. sector_sum++;
  747. if (total_bytes < sums->len) {
  748. item = (struct btrfs_csum_item *)((char *)item +
  749. csum_size);
  750. if (item < item_end && bytenr + PAGE_CACHE_SIZE ==
  751. sector_sum->bytenr) {
  752. bytenr = sector_sum->bytenr;
  753. goto next_sector;
  754. }
  755. }
  756. if (eb_token) {
  757. unmap_extent_buffer(leaf, eb_token, KM_USER1);
  758. eb_token = NULL;
  759. }
  760. btrfs_mark_buffer_dirty(path->nodes[0]);
  761. if (total_bytes < sums->len) {
  762. btrfs_release_path(root, path);
  763. cond_resched();
  764. goto again;
  765. }
  766. out:
  767. btrfs_free_path(path);
  768. return ret;
  769. fail_unlock:
  770. goto out;
  771. }