12e635a27SChris Mason #include <linux/module.h> 21e1d2701SChris Mason #include "ctree.h" 3dee26a9fSChris Mason #include "disk-io.h" 49f5fae2fSChris Mason #include "transaction.h" 51e1d2701SChris Mason 66567e837SChris Mason #define MAX_CSUM_ITEMS(r) ((((BTRFS_LEAF_DATA_SIZE(r) - \ 76567e837SChris Mason sizeof(struct btrfs_item)) / \ 86567e837SChris Mason sizeof(struct btrfs_csum_item)) - 1)) 9dee26a9fSChris Mason int btrfs_alloc_file_extent(struct btrfs_trans_handle *trans, 10dee26a9fSChris Mason struct btrfs_root *root, 11dee26a9fSChris Mason u64 objectid, u64 offset, 12dee26a9fSChris Mason u64 num_blocks, u64 hint_block, 13dee26a9fSChris Mason u64 *result) 149f5fae2fSChris Mason { 15dee26a9fSChris Mason struct btrfs_key ins; 16dee26a9fSChris Mason int ret = 0; 17dee26a9fSChris Mason struct btrfs_file_extent_item *item; 18dee26a9fSChris Mason struct btrfs_key file_key; 195caf2a00SChris Mason struct btrfs_path *path; 20dee26a9fSChris Mason 215caf2a00SChris Mason path = btrfs_alloc_path(); 225caf2a00SChris Mason BUG_ON(!path); 235caf2a00SChris Mason btrfs_init_path(path); 24dee26a9fSChris Mason ret = btrfs_alloc_extent(trans, root, num_blocks, hint_block, 25d0dbc624SChris Mason (u64)-1, &ins); 26dee26a9fSChris Mason BUG_ON(ret); 27dee26a9fSChris Mason file_key.objectid = objectid; 28dee26a9fSChris Mason file_key.offset = offset; 29dee26a9fSChris Mason file_key.flags = 0; 30dee26a9fSChris Mason btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); 31dee26a9fSChris Mason 325caf2a00SChris Mason ret = btrfs_insert_empty_item(trans, root, path, &file_key, 33dee26a9fSChris Mason sizeof(*item)); 349773a788SChris Mason BUG_ON(ret); 355caf2a00SChris Mason item = btrfs_item_ptr(btrfs_buffer_leaf(path->nodes[0]), path->slots[0], 36dee26a9fSChris Mason struct btrfs_file_extent_item); 37dee26a9fSChris Mason btrfs_set_file_extent_disk_blocknr(item, ins.objectid); 38dee26a9fSChris Mason btrfs_set_file_extent_disk_num_blocks(item, ins.offset); 39dee26a9fSChris Mason btrfs_set_file_extent_offset(item, 0); 40dee26a9fSChris Mason btrfs_set_file_extent_num_blocks(item, ins.offset); 4171951f35SChris Mason btrfs_set_file_extent_generation(item, trans->transid); 425caf2a00SChris Mason btrfs_mark_buffer_dirty(path->nodes[0]); 43dee26a9fSChris Mason *result = ins.objectid; 445caf2a00SChris Mason btrfs_release_path(root, path); 455caf2a00SChris Mason btrfs_free_path(path); 469f5fae2fSChris Mason return 0; 479f5fae2fSChris Mason } 48dee26a9fSChris Mason 496567e837SChris Mason static struct btrfs_csum_item *__lookup_csum_item(struct btrfs_root *root, 506567e837SChris Mason struct btrfs_path *path, 516567e837SChris Mason u64 objectid, u64 offset) 526567e837SChris Mason { 536567e837SChris Mason int ret; 546567e837SChris Mason struct btrfs_key file_key; 556567e837SChris Mason struct btrfs_key found_key; 566567e837SChris Mason struct btrfs_csum_item *item; 576567e837SChris Mason struct btrfs_leaf *leaf; 586567e837SChris Mason u64 csum_offset = 0; 596567e837SChris Mason 606567e837SChris Mason file_key.objectid = objectid; 616567e837SChris Mason file_key.offset = offset; 626567e837SChris Mason file_key.flags = 0; 636567e837SChris Mason btrfs_set_key_type(&file_key, BTRFS_CSUM_ITEM_KEY); 646567e837SChris Mason ret = btrfs_search_slot(NULL, root, &file_key, path, 0, 0); 656567e837SChris Mason if (ret < 0) 666567e837SChris Mason goto fail; 676567e837SChris Mason leaf = btrfs_buffer_leaf(path->nodes[0]); 686567e837SChris Mason if (ret > 0) { 696567e837SChris Mason ret = 1; 706567e837SChris Mason if (path->slots[0] == 0) 716567e837SChris Mason goto fail; 726567e837SChris Mason path->slots[0]--; 736567e837SChris Mason btrfs_disk_key_to_cpu(&found_key, 746567e837SChris Mason &leaf->items[path->slots[0]].key); 756567e837SChris Mason if (btrfs_key_type(&found_key) != BTRFS_CSUM_ITEM_KEY || 766567e837SChris Mason found_key.objectid != objectid) { 776567e837SChris Mason goto fail; 786567e837SChris Mason } 796567e837SChris Mason csum_offset = (offset - found_key.offset) >> 806567e837SChris Mason root->fs_info->sb->s_blocksize_bits; 816567e837SChris Mason if (csum_offset >= 826567e837SChris Mason btrfs_item_size(leaf->items + path->slots[0]) / 836567e837SChris Mason sizeof(struct btrfs_csum_item)) { 846567e837SChris Mason goto fail; 856567e837SChris Mason } 866567e837SChris Mason } 876567e837SChris Mason item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item); 886567e837SChris Mason item += csum_offset; 896567e837SChris Mason return item; 906567e837SChris Mason fail: 916567e837SChris Mason if (ret > 0) 926567e837SChris Mason ret = -EIO; 936567e837SChris Mason return ERR_PTR(ret); 946567e837SChris Mason } 956567e837SChris Mason 966567e837SChris Mason 97dee26a9fSChris Mason int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans, 98dee26a9fSChris Mason struct btrfs_root *root, 99dee26a9fSChris Mason struct btrfs_path *path, u64 objectid, 1009773a788SChris Mason u64 offset, int mod) 101dee26a9fSChris Mason { 102dee26a9fSChris Mason int ret; 103dee26a9fSChris Mason struct btrfs_key file_key; 104dee26a9fSChris Mason int ins_len = mod < 0 ? -1 : 0; 105dee26a9fSChris Mason int cow = mod != 0; 1066567e837SChris Mason struct btrfs_csum_item *csum_item; 107dee26a9fSChris Mason 1086567e837SChris Mason csum_item = __lookup_csum_item(root, path, objectid, offset); 1096567e837SChris Mason if (IS_ERR(csum_item)) 1106567e837SChris Mason return PTR_ERR(csum_item); 111dee26a9fSChris Mason file_key.objectid = objectid; 1126567e837SChris Mason file_key.offset = btrfs_csum_extent_offset(csum_item); 113dee26a9fSChris Mason file_key.flags = 0; 114dee26a9fSChris Mason btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); 1156567e837SChris Mason btrfs_release_path(root, path); 116dee26a9fSChris Mason ret = btrfs_search_slot(trans, root, &file_key, path, ins_len, cow); 117dee26a9fSChris Mason return ret; 118dee26a9fSChris Mason } 119f254e52cSChris Mason 120f254e52cSChris Mason int btrfs_csum_file_block(struct btrfs_trans_handle *trans, 121f254e52cSChris Mason struct btrfs_root *root, 122f254e52cSChris Mason u64 objectid, u64 offset, 1236567e837SChris Mason u64 extent_offset, 124f254e52cSChris Mason char *data, size_t len) 125f254e52cSChris Mason { 126f254e52cSChris Mason int ret; 127f254e52cSChris Mason struct btrfs_key file_key; 1286567e837SChris Mason struct btrfs_key found_key; 1295caf2a00SChris Mason struct btrfs_path *path; 130f254e52cSChris Mason struct btrfs_csum_item *item; 1316567e837SChris Mason struct btrfs_leaf *leaf; 1326567e837SChris Mason u64 csum_offset; 133f254e52cSChris Mason 1345caf2a00SChris Mason path = btrfs_alloc_path(); 1355caf2a00SChris Mason BUG_ON(!path); 1365caf2a00SChris Mason btrfs_init_path(path); 137f254e52cSChris Mason file_key.objectid = objectid; 138f254e52cSChris Mason file_key.offset = offset; 139f254e52cSChris Mason file_key.flags = 0; 140f254e52cSChris Mason btrfs_set_key_type(&file_key, BTRFS_CSUM_ITEM_KEY); 1416567e837SChris Mason ret = btrfs_search_slot(trans, root, &file_key, path, 1426567e837SChris Mason sizeof(struct btrfs_csum_item), 1); 1436567e837SChris Mason if (ret < 0) 1446567e837SChris Mason goto fail; 1456567e837SChris Mason if (ret == 0) { 1466567e837SChris Mason csum_offset = 0; 1476567e837SChris Mason goto csum; 1486567e837SChris Mason } 1496567e837SChris Mason if (path->slots[0] == 0) { 1506567e837SChris Mason btrfs_release_path(root, path); 1516567e837SChris Mason goto insert; 1526567e837SChris Mason } 1536567e837SChris Mason path->slots[0]--; 1546567e837SChris Mason leaf = btrfs_buffer_leaf(path->nodes[0]); 1556567e837SChris Mason btrfs_disk_key_to_cpu(&found_key, &leaf->items[path->slots[0]].key); 1566567e837SChris Mason csum_offset = (offset - found_key.offset) >> 1576567e837SChris Mason root->fs_info->sb->s_blocksize_bits; 1586567e837SChris Mason if (btrfs_key_type(&found_key) != BTRFS_CSUM_ITEM_KEY || 1596567e837SChris Mason found_key.objectid != objectid || 1606567e837SChris Mason csum_offset >= MAX_CSUM_ITEMS(root)) { 1616567e837SChris Mason btrfs_release_path(root, path); 1626567e837SChris Mason goto insert; 1636567e837SChris Mason } 1646567e837SChris Mason if (csum_offset >= btrfs_item_size(leaf->items + path->slots[0]) / 1656567e837SChris Mason sizeof(struct btrfs_csum_item)) { 1666567e837SChris Mason ret = btrfs_extend_item(trans, root, path, 1676567e837SChris Mason sizeof(struct btrfs_csum_item)); 1686567e837SChris Mason BUG_ON(ret); 1696567e837SChris Mason goto csum; 1706567e837SChris Mason } 1716567e837SChris Mason 1726567e837SChris Mason insert: 1736567e837SChris Mason csum_offset = 0; 1745caf2a00SChris Mason ret = btrfs_insert_empty_item(trans, root, path, &file_key, 1756567e837SChris Mason sizeof(struct btrfs_csum_item)); 176f254e52cSChris Mason if (ret != 0 && ret != -EEXIST) 177f254e52cSChris Mason goto fail; 1786567e837SChris Mason csum: 1795caf2a00SChris Mason item = btrfs_item_ptr(btrfs_buffer_leaf(path->nodes[0]), path->slots[0], 180f254e52cSChris Mason struct btrfs_csum_item); 181f254e52cSChris Mason ret = 0; 1826567e837SChris Mason item += csum_offset; 183f254e52cSChris Mason ret = btrfs_csum_data(root, data, len, item->csum); 1846567e837SChris Mason btrfs_set_csum_extent_offset(item, extent_offset); 1855caf2a00SChris Mason btrfs_mark_buffer_dirty(path->nodes[0]); 186f254e52cSChris Mason fail: 1875caf2a00SChris Mason btrfs_release_path(root, path); 1885caf2a00SChris Mason btrfs_free_path(path); 189f254e52cSChris Mason return ret; 190f254e52cSChris Mason } 191f254e52cSChris Mason 192f254e52cSChris Mason int btrfs_csum_verify_file_block(struct btrfs_root *root, 193f254e52cSChris Mason u64 objectid, u64 offset, 194f254e52cSChris Mason char *data, size_t len) 195f254e52cSChris Mason { 196f254e52cSChris Mason int ret; 197f254e52cSChris Mason struct btrfs_key file_key; 1985caf2a00SChris Mason struct btrfs_path *path; 199f254e52cSChris Mason struct btrfs_csum_item *item; 200f254e52cSChris Mason char result[BTRFS_CSUM_SIZE]; 201f254e52cSChris Mason 2025caf2a00SChris Mason path = btrfs_alloc_path(); 2035caf2a00SChris Mason BUG_ON(!path); 2045caf2a00SChris Mason btrfs_init_path(path); 205f254e52cSChris Mason file_key.objectid = objectid; 206f254e52cSChris Mason file_key.offset = offset; 207f254e52cSChris Mason file_key.flags = 0; 208f254e52cSChris Mason btrfs_set_key_type(&file_key, BTRFS_CSUM_ITEM_KEY); 2092da566edSChris Mason mutex_lock(&root->fs_info->fs_mutex); 2106567e837SChris Mason 2116567e837SChris Mason item = __lookup_csum_item(root, path, objectid, offset); 2126567e837SChris Mason if (IS_ERR(item)) { 2136567e837SChris Mason ret = PTR_ERR(item); 214f254e52cSChris Mason goto fail; 2156567e837SChris Mason } 2166567e837SChris Mason 217f254e52cSChris Mason ret = btrfs_csum_data(root, data, len, result); 218f254e52cSChris Mason WARN_ON(ret); 219f254e52cSChris Mason if (memcmp(result, item->csum, BTRFS_CSUM_SIZE)) 220f254e52cSChris Mason ret = 1; 221f254e52cSChris Mason fail: 2225caf2a00SChris Mason btrfs_release_path(root, path); 2235caf2a00SChris Mason btrfs_free_path(path); 2242da566edSChris Mason mutex_unlock(&root->fs_info->fs_mutex); 225f254e52cSChris Mason return ret; 226f254e52cSChris Mason } 227f254e52cSChris Mason 228