file-item.c 22 KB

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