file-item.c 22 KB

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