16cbd5570SChris Mason /* 26cbd5570SChris Mason * Copyright (C) 2007 Oracle. All rights reserved. 36cbd5570SChris Mason * 46cbd5570SChris Mason * This program is free software; you can redistribute it and/or 56cbd5570SChris Mason * modify it under the terms of the GNU General Public 66cbd5570SChris Mason * License v2 as published by the Free Software Foundation. 76cbd5570SChris Mason * 86cbd5570SChris Mason * This program is distributed in the hope that it will be useful, 96cbd5570SChris Mason * but WITHOUT ANY WARRANTY; without even the implied warranty of 106cbd5570SChris Mason * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 116cbd5570SChris Mason * General Public License for more details. 126cbd5570SChris Mason * 136cbd5570SChris Mason * You should have received a copy of the GNU General Public 146cbd5570SChris Mason * License along with this program; if not, write to the 156cbd5570SChris Mason * Free Software Foundation, Inc., 59 Temple Place - Suite 330, 166cbd5570SChris Mason * Boston, MA 021110-1307, USA. 176cbd5570SChris Mason */ 186cbd5570SChris Mason 19065631f6SChris Mason #include <linux/bio.h> 20065631f6SChris Mason #include <linux/pagemap.h> 21065631f6SChris Mason #include <linux/highmem.h> 221e1d2701SChris Mason #include "ctree.h" 23dee26a9fSChris Mason #include "disk-io.h" 249f5fae2fSChris Mason #include "transaction.h" 251de037a4SChris Mason #include "print-tree.h" 261e1d2701SChris Mason 27607d432dSJosef Bacik #define MAX_CSUM_ITEMS(r, size) ((((BTRFS_LEAF_DATA_SIZE(r) - \ 28a429e513SChris Mason sizeof(struct btrfs_item) * 2) / \ 29607d432dSJosef Bacik size) - 1)) 30b18c6685SChris Mason int btrfs_insert_file_extent(struct btrfs_trans_handle *trans, 31dee26a9fSChris Mason struct btrfs_root *root, 32b18c6685SChris Mason u64 objectid, u64 pos, 33f2eb0a24SSage Weil u64 disk_offset, u64 disk_num_bytes, 34c8b97818SChris Mason u64 num_bytes, u64 offset, u64 ram_bytes, 35c8b97818SChris Mason u8 compression, u8 encryption, u16 other_encoding) 369f5fae2fSChris Mason { 37dee26a9fSChris Mason int ret = 0; 38dee26a9fSChris Mason struct btrfs_file_extent_item *item; 39dee26a9fSChris Mason struct btrfs_key file_key; 405caf2a00SChris Mason struct btrfs_path *path; 415f39d397SChris Mason struct extent_buffer *leaf; 42dee26a9fSChris Mason 435caf2a00SChris Mason path = btrfs_alloc_path(); 445caf2a00SChris Mason BUG_ON(!path); 45dee26a9fSChris Mason file_key.objectid = objectid; 46b18c6685SChris Mason file_key.offset = pos; 47dee26a9fSChris Mason btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); 48dee26a9fSChris Mason 495caf2a00SChris Mason ret = btrfs_insert_empty_item(trans, root, path, &file_key, 50dee26a9fSChris Mason sizeof(*item)); 5154aa1f4dSChris Mason if (ret < 0) 5254aa1f4dSChris Mason goto out; 539773a788SChris Mason BUG_ON(ret); 545f39d397SChris Mason leaf = path->nodes[0]; 555f39d397SChris Mason item = btrfs_item_ptr(leaf, path->slots[0], 56dee26a9fSChris Mason struct btrfs_file_extent_item); 57f2eb0a24SSage Weil btrfs_set_file_extent_disk_bytenr(leaf, item, disk_offset); 58db94535dSChris Mason btrfs_set_file_extent_disk_num_bytes(leaf, item, disk_num_bytes); 59f2eb0a24SSage Weil btrfs_set_file_extent_offset(leaf, item, offset); 60db94535dSChris Mason btrfs_set_file_extent_num_bytes(leaf, item, num_bytes); 61c8b97818SChris Mason btrfs_set_file_extent_ram_bytes(leaf, item, ram_bytes); 625f39d397SChris Mason btrfs_set_file_extent_generation(leaf, item, trans->transid); 635f39d397SChris Mason btrfs_set_file_extent_type(leaf, item, BTRFS_FILE_EXTENT_REG); 64c8b97818SChris Mason btrfs_set_file_extent_compression(leaf, item, compression); 65c8b97818SChris Mason btrfs_set_file_extent_encryption(leaf, item, encryption); 66c8b97818SChris Mason btrfs_set_file_extent_other_encoding(leaf, item, other_encoding); 67c8b97818SChris Mason 685f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 6954aa1f4dSChris Mason out: 705caf2a00SChris Mason btrfs_free_path(path); 7154aa1f4dSChris Mason return ret; 729f5fae2fSChris Mason } 73dee26a9fSChris Mason 74b18c6685SChris Mason struct btrfs_csum_item *btrfs_lookup_csum(struct btrfs_trans_handle *trans, 75b18c6685SChris Mason struct btrfs_root *root, 766567e837SChris Mason struct btrfs_path *path, 77d20f7043SChris Mason u64 bytenr, int cow) 786567e837SChris Mason { 796567e837SChris Mason int ret; 806567e837SChris Mason struct btrfs_key file_key; 816567e837SChris Mason struct btrfs_key found_key; 826567e837SChris Mason struct btrfs_csum_item *item; 835f39d397SChris Mason struct extent_buffer *leaf; 846567e837SChris Mason u64 csum_offset = 0; 85607d432dSJosef Bacik u16 csum_size = 86607d432dSJosef Bacik btrfs_super_csum_size(&root->fs_info->super_copy); 87a429e513SChris Mason int csums_in_item; 886567e837SChris Mason 89d20f7043SChris Mason file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; 90d20f7043SChris Mason file_key.offset = bytenr; 91d20f7043SChris Mason btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY); 92b18c6685SChris Mason ret = btrfs_search_slot(trans, root, &file_key, path, 0, cow); 936567e837SChris Mason if (ret < 0) 946567e837SChris Mason goto fail; 955f39d397SChris Mason leaf = path->nodes[0]; 966567e837SChris Mason if (ret > 0) { 976567e837SChris Mason ret = 1; 9870b2befdSChris Mason if (path->slots[0] == 0) 996567e837SChris Mason goto fail; 1006567e837SChris Mason path->slots[0]--; 1015f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 102d20f7043SChris Mason if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY) 1036567e837SChris Mason goto fail; 104d20f7043SChris Mason 105d20f7043SChris Mason csum_offset = (bytenr - found_key.offset) >> 1066567e837SChris Mason root->fs_info->sb->s_blocksize_bits; 1075f39d397SChris Mason csums_in_item = btrfs_item_size_nr(leaf, path->slots[0]); 108607d432dSJosef Bacik csums_in_item /= csum_size; 109a429e513SChris Mason 110a429e513SChris Mason if (csum_offset >= csums_in_item) { 111a429e513SChris Mason ret = -EFBIG; 1126567e837SChris Mason goto fail; 1136567e837SChris Mason } 1146567e837SChris Mason } 1156567e837SChris Mason item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item); 116509659cdSChris Mason item = (struct btrfs_csum_item *)((unsigned char *)item + 117607d432dSJosef Bacik csum_offset * csum_size); 1186567e837SChris Mason return item; 1196567e837SChris Mason fail: 1206567e837SChris Mason if (ret > 0) 121b18c6685SChris Mason ret = -ENOENT; 1226567e837SChris Mason return ERR_PTR(ret); 1236567e837SChris Mason } 1246567e837SChris Mason 1256567e837SChris Mason 126dee26a9fSChris Mason int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans, 127dee26a9fSChris Mason struct btrfs_root *root, 128dee26a9fSChris Mason struct btrfs_path *path, u64 objectid, 1299773a788SChris Mason u64 offset, int mod) 130dee26a9fSChris Mason { 131dee26a9fSChris Mason int ret; 132dee26a9fSChris Mason struct btrfs_key file_key; 133dee26a9fSChris Mason int ins_len = mod < 0 ? -1 : 0; 134dee26a9fSChris Mason int cow = mod != 0; 135dee26a9fSChris Mason 136dee26a9fSChris Mason file_key.objectid = objectid; 13770b2befdSChris Mason file_key.offset = offset; 138dee26a9fSChris Mason btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); 139dee26a9fSChris Mason ret = btrfs_search_slot(trans, root, &file_key, path, ins_len, cow); 140dee26a9fSChris Mason return ret; 141dee26a9fSChris Mason } 142f254e52cSChris Mason 14317d217feSYan Zheng 14461b49440SChris Mason int btrfs_lookup_bio_sums(struct btrfs_root *root, struct inode *inode, 145d20f7043SChris Mason struct bio *bio, u32 *dst) 14661b49440SChris Mason { 14761b49440SChris Mason u32 sum; 14861b49440SChris Mason struct bio_vec *bvec = bio->bi_io_vec; 14961b49440SChris Mason int bio_index = 0; 15061b49440SChris Mason u64 offset; 15161b49440SChris Mason u64 item_start_offset = 0; 15261b49440SChris Mason u64 item_last_offset = 0; 153d20f7043SChris Mason u64 disk_bytenr; 15461b49440SChris Mason u32 diff; 155607d432dSJosef Bacik u16 csum_size = 156607d432dSJosef Bacik btrfs_super_csum_size(&root->fs_info->super_copy); 15761b49440SChris Mason int ret; 15861b49440SChris Mason struct btrfs_path *path; 15961b49440SChris Mason struct btrfs_csum_item *item = NULL; 16061b49440SChris Mason struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; 16161b49440SChris Mason 16261b49440SChris Mason path = btrfs_alloc_path(); 1634d1b5fb4SChris Mason if (bio->bi_size > PAGE_CACHE_SIZE * 8) 1644d1b5fb4SChris Mason path->reada = 2; 16561b49440SChris Mason 16661b49440SChris Mason WARN_ON(bio->bi_vcnt <= 0); 16761b49440SChris Mason 168d20f7043SChris Mason disk_bytenr = (u64)bio->bi_sector << 9; 16961b49440SChris Mason while (bio_index < bio->bi_vcnt) { 17061b49440SChris Mason offset = page_offset(bvec->bv_page) + bvec->bv_offset; 171d20f7043SChris Mason ret = btrfs_find_ordered_sum(inode, offset, disk_bytenr, &sum); 17261b49440SChris Mason if (ret == 0) 17361b49440SChris Mason goto found; 17461b49440SChris Mason 175d20f7043SChris Mason if (!item || disk_bytenr < item_start_offset || 176d20f7043SChris Mason disk_bytenr >= item_last_offset) { 17761b49440SChris Mason struct btrfs_key found_key; 17861b49440SChris Mason u32 item_size; 17961b49440SChris Mason 18061b49440SChris Mason if (item) 18161b49440SChris Mason btrfs_release_path(root, path); 182d20f7043SChris Mason item = btrfs_lookup_csum(NULL, root->fs_info->csum_root, 183d20f7043SChris Mason path, disk_bytenr, 0); 18461b49440SChris Mason if (IS_ERR(item)) { 18561b49440SChris Mason ret = PTR_ERR(item); 18661b49440SChris Mason if (ret == -ENOENT || ret == -EFBIG) 18761b49440SChris Mason ret = 0; 18861b49440SChris Mason sum = 0; 18917d217feSYan Zheng if (BTRFS_I(inode)->root->root_key.objectid == 19017d217feSYan Zheng BTRFS_DATA_RELOC_TREE_OBJECTID) { 19117d217feSYan Zheng set_extent_bits(io_tree, offset, 19217d217feSYan Zheng offset + bvec->bv_len - 1, 19317d217feSYan Zheng EXTENT_NODATASUM, GFP_NOFS); 19417d217feSYan Zheng } else { 195*d397712bSChris Mason printk(KERN_INFO "btrfs no csum found " 196*d397712bSChris Mason "for inode %lu start %llu\n", 197*d397712bSChris Mason inode->i_ino, 19861b49440SChris Mason (unsigned long long)offset); 19917d217feSYan Zheng } 2006dab8157SChris Mason item = NULL; 20139be25cdSChris Mason btrfs_release_path(root, path); 20261b49440SChris Mason goto found; 20361b49440SChris Mason } 20461b49440SChris Mason btrfs_item_key_to_cpu(path->nodes[0], &found_key, 20561b49440SChris Mason path->slots[0]); 20661b49440SChris Mason 20761b49440SChris Mason item_start_offset = found_key.offset; 20861b49440SChris Mason item_size = btrfs_item_size_nr(path->nodes[0], 20961b49440SChris Mason path->slots[0]); 21061b49440SChris Mason item_last_offset = item_start_offset + 211607d432dSJosef Bacik (item_size / csum_size) * 21261b49440SChris Mason root->sectorsize; 21361b49440SChris Mason item = btrfs_item_ptr(path->nodes[0], path->slots[0], 21461b49440SChris Mason struct btrfs_csum_item); 21561b49440SChris Mason } 21661b49440SChris Mason /* 21761b49440SChris Mason * this byte range must be able to fit inside 21861b49440SChris Mason * a single leaf so it will also fit inside a u32 21961b49440SChris Mason */ 220d20f7043SChris Mason diff = disk_bytenr - item_start_offset; 22161b49440SChris Mason diff = diff / root->sectorsize; 222607d432dSJosef Bacik diff = diff * csum_size; 22361b49440SChris Mason 22461b49440SChris Mason read_extent_buffer(path->nodes[0], &sum, 2253de9d6b6SChris Mason ((unsigned long)item) + diff, 226607d432dSJosef Bacik csum_size); 22761b49440SChris Mason found: 228d20f7043SChris Mason if (dst) 229d20f7043SChris Mason *dst++ = sum; 230d20f7043SChris Mason else 23161b49440SChris Mason set_state_private(io_tree, offset, sum); 232d20f7043SChris Mason disk_bytenr += bvec->bv_len; 23361b49440SChris Mason bio_index++; 23461b49440SChris Mason bvec++; 23561b49440SChris Mason } 23661b49440SChris Mason btrfs_free_path(path); 23761b49440SChris Mason return 0; 23861b49440SChris Mason } 23961b49440SChris Mason 24017d217feSYan Zheng int btrfs_lookup_csums_range(struct btrfs_root *root, u64 start, u64 end, 24117d217feSYan Zheng struct list_head *list) 24217d217feSYan Zheng { 24317d217feSYan Zheng struct btrfs_key key; 24417d217feSYan Zheng struct btrfs_path *path; 24517d217feSYan Zheng struct extent_buffer *leaf; 24617d217feSYan Zheng struct btrfs_ordered_sum *sums; 24717d217feSYan Zheng struct btrfs_sector_sum *sector_sum; 24817d217feSYan Zheng struct btrfs_csum_item *item; 24917d217feSYan Zheng unsigned long offset; 25017d217feSYan Zheng int ret; 25117d217feSYan Zheng size_t size; 25217d217feSYan Zheng u64 csum_end; 25317d217feSYan Zheng u16 csum_size = btrfs_super_csum_size(&root->fs_info->super_copy); 25417d217feSYan Zheng 25517d217feSYan Zheng path = btrfs_alloc_path(); 25617d217feSYan Zheng BUG_ON(!path); 25717d217feSYan Zheng 25817d217feSYan Zheng key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; 25917d217feSYan Zheng key.offset = start; 26017d217feSYan Zheng key.type = BTRFS_EXTENT_CSUM_KEY; 26117d217feSYan Zheng 26217d217feSYan Zheng ret = btrfs_search_slot(NULL, root->fs_info->csum_root, 26317d217feSYan Zheng &key, path, 0, 0); 26417d217feSYan Zheng if (ret < 0) 26517d217feSYan Zheng goto fail; 26617d217feSYan Zheng if (ret > 0 && path->slots[0] > 0) { 26717d217feSYan Zheng leaf = path->nodes[0]; 26817d217feSYan Zheng btrfs_item_key_to_cpu(leaf, &key, path->slots[0] - 1); 26917d217feSYan Zheng if (key.objectid == BTRFS_EXTENT_CSUM_OBJECTID && 27017d217feSYan Zheng key.type == BTRFS_EXTENT_CSUM_KEY) { 27117d217feSYan Zheng offset = (start - key.offset) >> 27217d217feSYan Zheng root->fs_info->sb->s_blocksize_bits; 27317d217feSYan Zheng if (offset * csum_size < 27417d217feSYan Zheng btrfs_item_size_nr(leaf, path->slots[0] - 1)) 27517d217feSYan Zheng path->slots[0]--; 27617d217feSYan Zheng } 27717d217feSYan Zheng } 27817d217feSYan Zheng 27917d217feSYan Zheng while (start <= end) { 28017d217feSYan Zheng leaf = path->nodes[0]; 28117d217feSYan Zheng if (path->slots[0] >= btrfs_header_nritems(leaf)) { 28217d217feSYan Zheng ret = btrfs_next_leaf(root->fs_info->csum_root, path); 28317d217feSYan Zheng if (ret < 0) 28417d217feSYan Zheng goto fail; 28517d217feSYan Zheng if (ret > 0) 28617d217feSYan Zheng break; 28717d217feSYan Zheng leaf = path->nodes[0]; 28817d217feSYan Zheng } 28917d217feSYan Zheng 29017d217feSYan Zheng btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); 29117d217feSYan Zheng if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || 29217d217feSYan Zheng key.type != BTRFS_EXTENT_CSUM_KEY) 29317d217feSYan Zheng break; 29417d217feSYan Zheng 29517d217feSYan Zheng btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); 29617d217feSYan Zheng if (key.offset > end) 29717d217feSYan Zheng break; 29817d217feSYan Zheng 29917d217feSYan Zheng if (key.offset > start) 30017d217feSYan Zheng start = key.offset; 30117d217feSYan Zheng 30217d217feSYan Zheng size = btrfs_item_size_nr(leaf, path->slots[0]); 30317d217feSYan Zheng csum_end = key.offset + (size / csum_size) * root->sectorsize; 30487b29b20SYan Zheng if (csum_end <= start) { 30587b29b20SYan Zheng path->slots[0]++; 30687b29b20SYan Zheng continue; 30787b29b20SYan Zheng } 30817d217feSYan Zheng 30917d217feSYan Zheng size = min(csum_end, end + 1) - start; 31017d217feSYan Zheng sums = kzalloc(btrfs_ordered_sum_size(root, size), GFP_NOFS); 31117d217feSYan Zheng BUG_ON(!sums); 31217d217feSYan Zheng 31317d217feSYan Zheng sector_sum = sums->sums; 31417d217feSYan Zheng sums->bytenr = start; 31517d217feSYan Zheng sums->len = size; 31617d217feSYan Zheng 31717d217feSYan Zheng offset = (start - key.offset) >> 31817d217feSYan Zheng root->fs_info->sb->s_blocksize_bits; 31917d217feSYan Zheng offset *= csum_size; 32017d217feSYan Zheng 32117d217feSYan Zheng item = btrfs_item_ptr(path->nodes[0], path->slots[0], 32217d217feSYan Zheng struct btrfs_csum_item); 32317d217feSYan Zheng while (size > 0) { 32417d217feSYan Zheng read_extent_buffer(path->nodes[0], §or_sum->sum, 32517d217feSYan Zheng ((unsigned long)item) + offset, 32617d217feSYan Zheng csum_size); 32717d217feSYan Zheng sector_sum->bytenr = start; 32817d217feSYan Zheng 32917d217feSYan Zheng size -= root->sectorsize; 33017d217feSYan Zheng start += root->sectorsize; 33117d217feSYan Zheng offset += csum_size; 33217d217feSYan Zheng sector_sum++; 33317d217feSYan Zheng } 33417d217feSYan Zheng list_add_tail(&sums->list, list); 33517d217feSYan Zheng 33617d217feSYan Zheng path->slots[0]++; 33717d217feSYan Zheng } 33817d217feSYan Zheng ret = 0; 33917d217feSYan Zheng fail: 34017d217feSYan Zheng btrfs_free_path(path); 34117d217feSYan Zheng return ret; 34217d217feSYan Zheng } 34317d217feSYan Zheng 3443edf7d33SChris Mason int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode, 345d20f7043SChris Mason struct bio *bio, u64 file_start, int contig) 346e015640fSChris Mason { 347e6dcd2dcSChris Mason struct btrfs_ordered_sum *sums; 348e6dcd2dcSChris Mason struct btrfs_sector_sum *sector_sum; 3493edf7d33SChris Mason struct btrfs_ordered_extent *ordered; 350e015640fSChris Mason char *data; 351e015640fSChris Mason struct bio_vec *bvec = bio->bi_io_vec; 352e015640fSChris Mason int bio_index = 0; 3533edf7d33SChris Mason unsigned long total_bytes = 0; 3543edf7d33SChris Mason unsigned long this_sum_bytes = 0; 3553edf7d33SChris Mason u64 offset; 356d20f7043SChris Mason u64 disk_bytenr; 357e015640fSChris Mason 358e6dcd2dcSChris Mason WARN_ON(bio->bi_vcnt <= 0); 359e6dcd2dcSChris Mason sums = kzalloc(btrfs_ordered_sum_size(root, bio->bi_size), GFP_NOFS); 360e015640fSChris Mason if (!sums) 361e015640fSChris Mason return -ENOMEM; 3623edf7d33SChris Mason 363ed98b56aSChris Mason sector_sum = sums->sums; 364d20f7043SChris Mason disk_bytenr = (u64)bio->bi_sector << 9; 365e6dcd2dcSChris Mason sums->len = bio->bi_size; 366e6dcd2dcSChris Mason INIT_LIST_HEAD(&sums->list); 367d20f7043SChris Mason 368d20f7043SChris Mason if (contig) 369d20f7043SChris Mason offset = file_start; 370d20f7043SChris Mason else 371d20f7043SChris Mason offset = page_offset(bvec->bv_page) + bvec->bv_offset; 372d20f7043SChris Mason 373d20f7043SChris Mason ordered = btrfs_lookup_ordered_extent(inode, offset); 3743edf7d33SChris Mason BUG_ON(!ordered); 375d20f7043SChris Mason sums->bytenr = ordered->start; 376e015640fSChris Mason 377e015640fSChris Mason while (bio_index < bio->bi_vcnt) { 378d20f7043SChris Mason if (!contig) 3793edf7d33SChris Mason offset = page_offset(bvec->bv_page) + bvec->bv_offset; 380d20f7043SChris Mason 381d20f7043SChris Mason if (!contig && (offset >= ordered->file_offset + ordered->len || 382d20f7043SChris Mason offset < ordered->file_offset)) { 3833edf7d33SChris Mason unsigned long bytes_left; 3843edf7d33SChris Mason sums->len = this_sum_bytes; 3853edf7d33SChris Mason this_sum_bytes = 0; 3863edf7d33SChris Mason btrfs_add_ordered_sum(inode, ordered, sums); 3873edf7d33SChris Mason btrfs_put_ordered_extent(ordered); 3883edf7d33SChris Mason 3893edf7d33SChris Mason bytes_left = bio->bi_size - total_bytes; 3903edf7d33SChris Mason 3913edf7d33SChris Mason sums = kzalloc(btrfs_ordered_sum_size(root, bytes_left), 3923edf7d33SChris Mason GFP_NOFS); 3933edf7d33SChris Mason BUG_ON(!sums); 394ed98b56aSChris Mason sector_sum = sums->sums; 3953edf7d33SChris Mason sums->len = bytes_left; 396d20f7043SChris Mason ordered = btrfs_lookup_ordered_extent(inode, offset); 3973edf7d33SChris Mason BUG_ON(!ordered); 398d20f7043SChris Mason sums->bytenr = ordered->start; 3993edf7d33SChris Mason } 4003edf7d33SChris Mason 401e015640fSChris Mason data = kmap_atomic(bvec->bv_page, KM_USER0); 402e6dcd2dcSChris Mason sector_sum->sum = ~(u32)0; 403e6dcd2dcSChris Mason sector_sum->sum = btrfs_csum_data(root, 404e6dcd2dcSChris Mason data + bvec->bv_offset, 405e6dcd2dcSChris Mason sector_sum->sum, 406e6dcd2dcSChris Mason bvec->bv_len); 407e015640fSChris Mason kunmap_atomic(data, KM_USER0); 408e6dcd2dcSChris Mason btrfs_csum_final(sector_sum->sum, 409e6dcd2dcSChris Mason (char *)§or_sum->sum); 410d20f7043SChris Mason sector_sum->bytenr = disk_bytenr; 411ed98b56aSChris Mason 412e6dcd2dcSChris Mason sector_sum++; 413e015640fSChris Mason bio_index++; 4143edf7d33SChris Mason total_bytes += bvec->bv_len; 4153edf7d33SChris Mason this_sum_bytes += bvec->bv_len; 416d20f7043SChris Mason disk_bytenr += bvec->bv_len; 417d20f7043SChris Mason offset += bvec->bv_len; 418e015640fSChris Mason bvec++; 419e015640fSChris Mason } 420ed98b56aSChris Mason this_sum_bytes = 0; 4213edf7d33SChris Mason btrfs_add_ordered_sum(inode, ordered, sums); 4223edf7d33SChris Mason btrfs_put_ordered_extent(ordered); 423e015640fSChris Mason return 0; 424e015640fSChris Mason } 425e015640fSChris Mason 426459931ecSChris Mason /* 427459931ecSChris Mason * helper function for csum removal, this expects the 428459931ecSChris Mason * key to describe the csum pointed to by the path, and it expects 429459931ecSChris Mason * the csum to overlap the range [bytenr, len] 430459931ecSChris Mason * 431459931ecSChris Mason * The csum should not be entirely contained in the range and the 432459931ecSChris Mason * range should not be entirely contained in the csum. 433459931ecSChris Mason * 434459931ecSChris Mason * This calls btrfs_truncate_item with the correct args based on the 435459931ecSChris Mason * overlap, and fixes up the key as required. 436459931ecSChris Mason */ 437459931ecSChris Mason static noinline int truncate_one_csum(struct btrfs_trans_handle *trans, 438459931ecSChris Mason struct btrfs_root *root, 439459931ecSChris Mason struct btrfs_path *path, 440459931ecSChris Mason struct btrfs_key *key, 441459931ecSChris Mason u64 bytenr, u64 len) 442459931ecSChris Mason { 443459931ecSChris Mason struct extent_buffer *leaf; 444459931ecSChris Mason u16 csum_size = 445459931ecSChris Mason btrfs_super_csum_size(&root->fs_info->super_copy); 446459931ecSChris Mason u64 csum_end; 447459931ecSChris Mason u64 end_byte = bytenr + len; 448459931ecSChris Mason u32 blocksize_bits = root->fs_info->sb->s_blocksize_bits; 449459931ecSChris Mason int ret; 450459931ecSChris Mason 451459931ecSChris Mason leaf = path->nodes[0]; 452459931ecSChris Mason csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size; 453459931ecSChris Mason csum_end <<= root->fs_info->sb->s_blocksize_bits; 454459931ecSChris Mason csum_end += key->offset; 455459931ecSChris Mason 456459931ecSChris Mason if (key->offset < bytenr && csum_end <= end_byte) { 457459931ecSChris Mason /* 458459931ecSChris Mason * [ bytenr - len ] 459459931ecSChris Mason * [ ] 460459931ecSChris Mason * [csum ] 461459931ecSChris Mason * A simple truncate off the end of the item 462459931ecSChris Mason */ 463459931ecSChris Mason u32 new_size = (bytenr - key->offset) >> blocksize_bits; 464459931ecSChris Mason new_size *= csum_size; 465459931ecSChris Mason ret = btrfs_truncate_item(trans, root, path, new_size, 1); 466459931ecSChris Mason BUG_ON(ret); 467459931ecSChris Mason } else if (key->offset >= bytenr && csum_end > end_byte && 468459931ecSChris Mason end_byte > key->offset) { 469459931ecSChris Mason /* 470459931ecSChris Mason * [ bytenr - len ] 471459931ecSChris Mason * [ ] 472459931ecSChris Mason * [csum ] 473459931ecSChris Mason * we need to truncate from the beginning of the csum 474459931ecSChris Mason */ 475459931ecSChris Mason u32 new_size = (csum_end - end_byte) >> blocksize_bits; 476459931ecSChris Mason new_size *= csum_size; 477459931ecSChris Mason 478459931ecSChris Mason ret = btrfs_truncate_item(trans, root, path, new_size, 0); 479459931ecSChris Mason BUG_ON(ret); 480459931ecSChris Mason 481459931ecSChris Mason key->offset = end_byte; 482459931ecSChris Mason ret = btrfs_set_item_key_safe(trans, root, path, key); 483459931ecSChris Mason BUG_ON(ret); 484459931ecSChris Mason } else { 485459931ecSChris Mason BUG(); 486459931ecSChris Mason } 487459931ecSChris Mason return 0; 488459931ecSChris Mason } 489459931ecSChris Mason 490459931ecSChris Mason /* 491459931ecSChris Mason * deletes the csum items from the csum tree for a given 492459931ecSChris Mason * range of bytes. 493459931ecSChris Mason */ 494459931ecSChris Mason int btrfs_del_csums(struct btrfs_trans_handle *trans, 495459931ecSChris Mason struct btrfs_root *root, u64 bytenr, u64 len) 496459931ecSChris Mason { 497459931ecSChris Mason struct btrfs_path *path; 498459931ecSChris Mason struct btrfs_key key; 499459931ecSChris Mason u64 end_byte = bytenr + len; 500459931ecSChris Mason u64 csum_end; 501459931ecSChris Mason struct extent_buffer *leaf; 502459931ecSChris Mason int ret; 503459931ecSChris Mason u16 csum_size = 504459931ecSChris Mason btrfs_super_csum_size(&root->fs_info->super_copy); 505459931ecSChris Mason int blocksize_bits = root->fs_info->sb->s_blocksize_bits; 506459931ecSChris Mason 507459931ecSChris Mason root = root->fs_info->csum_root; 508459931ecSChris Mason 509459931ecSChris Mason path = btrfs_alloc_path(); 510459931ecSChris Mason 511459931ecSChris Mason while (1) { 512459931ecSChris Mason key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; 513459931ecSChris Mason key.offset = end_byte - 1; 514459931ecSChris Mason key.type = BTRFS_EXTENT_CSUM_KEY; 515459931ecSChris Mason 516459931ecSChris Mason ret = btrfs_search_slot(trans, root, &key, path, -1, 1); 517459931ecSChris Mason if (ret > 0) { 518459931ecSChris Mason if (path->slots[0] == 0) 519459931ecSChris Mason goto out; 520459931ecSChris Mason path->slots[0]--; 521459931ecSChris Mason } 522459931ecSChris Mason leaf = path->nodes[0]; 523459931ecSChris Mason btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); 524459931ecSChris Mason 525459931ecSChris Mason if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || 526459931ecSChris Mason key.type != BTRFS_EXTENT_CSUM_KEY) { 527459931ecSChris Mason break; 528459931ecSChris Mason } 529459931ecSChris Mason 530459931ecSChris Mason if (key.offset >= end_byte) 531459931ecSChris Mason break; 532459931ecSChris Mason 533459931ecSChris Mason csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size; 534459931ecSChris Mason csum_end <<= blocksize_bits; 535459931ecSChris Mason csum_end += key.offset; 536459931ecSChris Mason 537459931ecSChris Mason /* this csum ends before we start, we're done */ 538459931ecSChris Mason if (csum_end <= bytenr) 539459931ecSChris Mason break; 540459931ecSChris Mason 541459931ecSChris Mason /* delete the entire item, it is inside our range */ 542459931ecSChris Mason if (key.offset >= bytenr && csum_end <= end_byte) { 543459931ecSChris Mason ret = btrfs_del_item(trans, root, path); 544459931ecSChris Mason BUG_ON(ret); 545dcbdd4dcSChris Mason if (key.offset == bytenr) 546dcbdd4dcSChris Mason break; 547459931ecSChris Mason } else if (key.offset < bytenr && csum_end > end_byte) { 548459931ecSChris Mason unsigned long offset; 549459931ecSChris Mason unsigned long shift_len; 550459931ecSChris Mason unsigned long item_offset; 551459931ecSChris Mason /* 552459931ecSChris Mason * [ bytenr - len ] 553459931ecSChris Mason * [csum ] 554459931ecSChris Mason * 555459931ecSChris Mason * Our bytes are in the middle of the csum, 556459931ecSChris Mason * we need to split this item and insert a new one. 557459931ecSChris Mason * 558459931ecSChris Mason * But we can't drop the path because the 559459931ecSChris Mason * csum could change, get removed, extended etc. 560459931ecSChris Mason * 561459931ecSChris Mason * The trick here is the max size of a csum item leaves 562459931ecSChris Mason * enough room in the tree block for a single 563459931ecSChris Mason * item header. So, we split the item in place, 564459931ecSChris Mason * adding a new header pointing to the existing 565459931ecSChris Mason * bytes. Then we loop around again and we have 566459931ecSChris Mason * a nicely formed csum item that we can neatly 567459931ecSChris Mason * truncate. 568459931ecSChris Mason */ 569459931ecSChris Mason offset = (bytenr - key.offset) >> blocksize_bits; 570459931ecSChris Mason offset *= csum_size; 571459931ecSChris Mason 572459931ecSChris Mason shift_len = (len >> blocksize_bits) * csum_size; 573459931ecSChris Mason 574459931ecSChris Mason item_offset = btrfs_item_ptr_offset(leaf, 575459931ecSChris Mason path->slots[0]); 576459931ecSChris Mason 577459931ecSChris Mason memset_extent_buffer(leaf, 0, item_offset + offset, 578459931ecSChris Mason shift_len); 579459931ecSChris Mason key.offset = bytenr; 580459931ecSChris Mason 581459931ecSChris Mason /* 582459931ecSChris Mason * btrfs_split_item returns -EAGAIN when the 583459931ecSChris Mason * item changed size or key 584459931ecSChris Mason */ 585459931ecSChris Mason ret = btrfs_split_item(trans, root, path, &key, offset); 586459931ecSChris Mason BUG_ON(ret && ret != -EAGAIN); 587459931ecSChris Mason 588459931ecSChris Mason key.offset = end_byte - 1; 589459931ecSChris Mason } else { 590459931ecSChris Mason ret = truncate_one_csum(trans, root, path, 591459931ecSChris Mason &key, bytenr, len); 592459931ecSChris Mason BUG_ON(ret); 593dcbdd4dcSChris Mason if (key.offset < bytenr) 594dcbdd4dcSChris Mason break; 595459931ecSChris Mason } 596459931ecSChris Mason btrfs_release_path(root, path); 597459931ecSChris Mason } 598459931ecSChris Mason out: 599459931ecSChris Mason btrfs_free_path(path); 600459931ecSChris Mason return 0; 601459931ecSChris Mason } 602459931ecSChris Mason 603065631f6SChris Mason int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans, 604d20f7043SChris Mason struct btrfs_root *root, 605e6dcd2dcSChris Mason struct btrfs_ordered_sum *sums) 606f254e52cSChris Mason { 607d20f7043SChris Mason u64 bytenr; 608f254e52cSChris Mason int ret; 609f254e52cSChris Mason struct btrfs_key file_key; 6106567e837SChris Mason struct btrfs_key found_key; 611065631f6SChris Mason u64 next_offset; 612e6dcd2dcSChris Mason u64 total_bytes = 0; 613065631f6SChris Mason int found_next; 6145caf2a00SChris Mason struct btrfs_path *path; 615f254e52cSChris Mason struct btrfs_csum_item *item; 616065631f6SChris Mason struct btrfs_csum_item *item_end; 617ff79f819SChris Mason struct extent_buffer *leaf = NULL; 6186567e837SChris Mason u64 csum_offset; 619e6dcd2dcSChris Mason struct btrfs_sector_sum *sector_sum; 620f578d4bdSChris Mason u32 nritems; 621f578d4bdSChris Mason u32 ins_size; 6226e92f5e6SChris Mason char *eb_map; 6236e92f5e6SChris Mason char *eb_token; 6246e92f5e6SChris Mason unsigned long map_len; 6256e92f5e6SChris Mason unsigned long map_start; 626607d432dSJosef Bacik u16 csum_size = 627607d432dSJosef Bacik btrfs_super_csum_size(&root->fs_info->super_copy); 6286e92f5e6SChris Mason 6295caf2a00SChris Mason path = btrfs_alloc_path(); 6305caf2a00SChris Mason BUG_ON(!path); 631ed98b56aSChris Mason sector_sum = sums->sums; 632065631f6SChris Mason again: 633065631f6SChris Mason next_offset = (u64)-1; 634065631f6SChris Mason found_next = 0; 635d20f7043SChris Mason file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; 636d20f7043SChris Mason file_key.offset = sector_sum->bytenr; 637d20f7043SChris Mason bytenr = sector_sum->bytenr; 638d20f7043SChris Mason btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY); 639a429e513SChris Mason 640d20f7043SChris Mason item = btrfs_lookup_csum(trans, root, path, sector_sum->bytenr, 1); 641ff79f819SChris Mason if (!IS_ERR(item)) { 642ff79f819SChris Mason leaf = path->nodes[0]; 643639cb586SChris Mason ret = 0; 644a429e513SChris Mason goto found; 645ff79f819SChris Mason } 646a429e513SChris Mason ret = PTR_ERR(item); 647a429e513SChris Mason if (ret == -EFBIG) { 648a429e513SChris Mason u32 item_size; 649a429e513SChris Mason /* we found one, but it isn't big enough yet */ 6505f39d397SChris Mason leaf = path->nodes[0]; 6515f39d397SChris Mason item_size = btrfs_item_size_nr(leaf, path->slots[0]); 652607d432dSJosef Bacik if ((item_size / csum_size) >= 653607d432dSJosef Bacik MAX_CSUM_ITEMS(root, csum_size)) { 654a429e513SChris Mason /* already at max size, make a new one */ 655a429e513SChris Mason goto insert; 656a429e513SChris Mason } 657a429e513SChris Mason } else { 658f578d4bdSChris Mason int slot = path->slots[0] + 1; 659a429e513SChris Mason /* we didn't find a csum item, insert one */ 660f578d4bdSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 661f578d4bdSChris Mason if (path->slots[0] >= nritems - 1) { 662f578d4bdSChris Mason ret = btrfs_next_leaf(root, path); 663b56baf5bSYan if (ret == 1) 664f578d4bdSChris Mason found_next = 1; 665b56baf5bSYan if (ret != 0) 666f578d4bdSChris Mason goto insert; 667b56baf5bSYan slot = 0; 668f578d4bdSChris Mason } 669f578d4bdSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &found_key, slot); 670d20f7043SChris Mason if (found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || 671d20f7043SChris Mason found_key.type != BTRFS_EXTENT_CSUM_KEY) { 672f578d4bdSChris Mason found_next = 1; 673f578d4bdSChris Mason goto insert; 674f578d4bdSChris Mason } 675f578d4bdSChris Mason next_offset = found_key.offset; 676f578d4bdSChris Mason found_next = 1; 677a429e513SChris Mason goto insert; 678a429e513SChris Mason } 679a429e513SChris Mason 680a429e513SChris Mason /* 681a429e513SChris Mason * at this point, we know the tree has an item, but it isn't big 682a429e513SChris Mason * enough yet to put our csum in. Grow it 683a429e513SChris Mason */ 684a429e513SChris Mason btrfs_release_path(root, path); 6856567e837SChris Mason ret = btrfs_search_slot(trans, root, &file_key, path, 686607d432dSJosef Bacik csum_size, 1); 6876567e837SChris Mason if (ret < 0) 68853863232SChris Mason goto fail_unlock; 689459931ecSChris Mason 690459931ecSChris Mason if (ret > 0) { 691459931ecSChris Mason if (path->slots[0] == 0) 6926567e837SChris Mason goto insert; 6936567e837SChris Mason path->slots[0]--; 694459931ecSChris Mason } 695459931ecSChris Mason 6965f39d397SChris Mason leaf = path->nodes[0]; 6975f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 698d20f7043SChris Mason csum_offset = (bytenr - found_key.offset) >> 6996567e837SChris Mason root->fs_info->sb->s_blocksize_bits; 700459931ecSChris Mason 701d20f7043SChris Mason if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY || 702d20f7043SChris Mason found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || 703607d432dSJosef Bacik csum_offset >= MAX_CSUM_ITEMS(root, csum_size)) { 7046567e837SChris Mason goto insert; 7056567e837SChris Mason } 706459931ecSChris Mason 7075f39d397SChris Mason if (csum_offset >= btrfs_item_size_nr(leaf, path->slots[0]) / 708607d432dSJosef Bacik csum_size) { 709607d432dSJosef Bacik u32 diff = (csum_offset + 1) * csum_size; 710459931ecSChris Mason 711459931ecSChris Mason /* 712459931ecSChris Mason * is the item big enough already? we dropped our lock 713459931ecSChris Mason * before and need to recheck 714459931ecSChris Mason */ 715459931ecSChris Mason if (diff < btrfs_item_size_nr(leaf, path->slots[0])) 716459931ecSChris Mason goto csum; 717459931ecSChris Mason 7185f39d397SChris Mason diff = diff - btrfs_item_size_nr(leaf, path->slots[0]); 719*d397712bSChris Mason if (diff != csum_size) 7203a686375SChris Mason goto insert; 721459931ecSChris Mason 722a429e513SChris Mason ret = btrfs_extend_item(trans, root, path, diff); 7236567e837SChris Mason BUG_ON(ret); 7246567e837SChris Mason goto csum; 7256567e837SChris Mason } 7266567e837SChris Mason 7276567e837SChris Mason insert: 728a429e513SChris Mason btrfs_release_path(root, path); 7296567e837SChris Mason csum_offset = 0; 730f578d4bdSChris Mason if (found_next) { 731d20f7043SChris Mason u64 tmp = total_bytes + root->sectorsize; 732d20f7043SChris Mason u64 next_sector = sector_sum->bytenr; 733d20f7043SChris Mason struct btrfs_sector_sum *next = sector_sum + 1; 734d20f7043SChris Mason 735d20f7043SChris Mason while (tmp < sums->len) { 736d20f7043SChris Mason if (next_sector + root->sectorsize != next->bytenr) 737d20f7043SChris Mason break; 738d20f7043SChris Mason tmp += root->sectorsize; 739d20f7043SChris Mason next_sector = next->bytenr; 740d20f7043SChris Mason next++; 741d20f7043SChris Mason } 742d20f7043SChris Mason tmp = min(tmp, next_offset - file_key.offset); 743f578d4bdSChris Mason tmp >>= root->fs_info->sb->s_blocksize_bits; 744f578d4bdSChris Mason tmp = max((u64)1, tmp); 745607d432dSJosef Bacik tmp = min(tmp, (u64)MAX_CSUM_ITEMS(root, csum_size)); 746607d432dSJosef Bacik ins_size = csum_size * tmp; 747f578d4bdSChris Mason } else { 748607d432dSJosef Bacik ins_size = csum_size; 749f578d4bdSChris Mason } 7505caf2a00SChris Mason ret = btrfs_insert_empty_item(trans, root, path, &file_key, 751f578d4bdSChris Mason ins_size); 75254aa1f4dSChris Mason if (ret < 0) 75353863232SChris Mason goto fail_unlock; 754a429e513SChris Mason if (ret != 0) { 755a429e513SChris Mason WARN_ON(1); 75653863232SChris Mason goto fail_unlock; 757a429e513SChris Mason } 7586567e837SChris Mason csum: 7595f39d397SChris Mason leaf = path->nodes[0]; 7605f39d397SChris Mason item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item); 761f254e52cSChris Mason ret = 0; 762509659cdSChris Mason item = (struct btrfs_csum_item *)((unsigned char *)item + 763607d432dSJosef Bacik csum_offset * csum_size); 764b18c6685SChris Mason found: 765065631f6SChris Mason item_end = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item); 766065631f6SChris Mason item_end = (struct btrfs_csum_item *)((unsigned char *)item_end + 767065631f6SChris Mason btrfs_item_size_nr(leaf, path->slots[0])); 7686e92f5e6SChris Mason eb_token = NULL; 76953863232SChris Mason cond_resched(); 770e6dcd2dcSChris Mason next_sector: 771aadfeb6eSChris Mason 7726e92f5e6SChris Mason if (!eb_token || 773607d432dSJosef Bacik (unsigned long)item + csum_size >= map_start + map_len) { 7746e92f5e6SChris Mason int err; 7756e92f5e6SChris Mason 7766e92f5e6SChris Mason if (eb_token) 777eb20978fSChris Mason unmap_extent_buffer(leaf, eb_token, KM_USER1); 7786e92f5e6SChris Mason eb_token = NULL; 7796e92f5e6SChris Mason err = map_private_extent_buffer(leaf, (unsigned long)item, 780607d432dSJosef Bacik csum_size, 7816e92f5e6SChris Mason &eb_token, &eb_map, 782eb20978fSChris Mason &map_start, &map_len, KM_USER1); 7836e92f5e6SChris Mason if (err) 7846e92f5e6SChris Mason eb_token = NULL; 7856e92f5e6SChris Mason } 7866e92f5e6SChris Mason if (eb_token) { 7876e92f5e6SChris Mason memcpy(eb_token + ((unsigned long)item & (PAGE_CACHE_SIZE - 1)), 788607d432dSJosef Bacik §or_sum->sum, csum_size); 7896e92f5e6SChris Mason } else { 790e6dcd2dcSChris Mason write_extent_buffer(leaf, §or_sum->sum, 791607d432dSJosef Bacik (unsigned long)item, csum_size); 7926e92f5e6SChris Mason } 7937f3c74fbSChris Mason 794e6dcd2dcSChris Mason total_bytes += root->sectorsize; 795e6dcd2dcSChris Mason sector_sum++; 796e6dcd2dcSChris Mason if (total_bytes < sums->len) { 7976e92f5e6SChris Mason item = (struct btrfs_csum_item *)((char *)item + 798607d432dSJosef Bacik csum_size); 799d20f7043SChris Mason if (item < item_end && bytenr + PAGE_CACHE_SIZE == 800d20f7043SChris Mason sector_sum->bytenr) { 801d20f7043SChris Mason bytenr = sector_sum->bytenr; 802e6dcd2dcSChris Mason goto next_sector; 803065631f6SChris Mason } 8042e1a992eSChris Mason } 8056e92f5e6SChris Mason if (eb_token) { 806eb20978fSChris Mason unmap_extent_buffer(leaf, eb_token, KM_USER1); 8076e92f5e6SChris Mason eb_token = NULL; 8086e92f5e6SChris Mason } 8095caf2a00SChris Mason btrfs_mark_buffer_dirty(path->nodes[0]); 81053863232SChris Mason cond_resched(); 811e6dcd2dcSChris Mason if (total_bytes < sums->len) { 8125caf2a00SChris Mason btrfs_release_path(root, path); 813065631f6SChris Mason goto again; 814065631f6SChris Mason } 81553863232SChris Mason out: 8165caf2a00SChris Mason btrfs_free_path(path); 817f254e52cSChris Mason return ret; 81853863232SChris Mason 81953863232SChris Mason fail_unlock: 82053863232SChris Mason goto out; 821f254e52cSChris Mason } 822