file-item.c 22 KB

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