Chris Mason | 2e635a2 | 2007-03-21 11:12:56 -0400 | [diff] [blame] | 1 | #include <linux/module.h> |
Chris Mason | 1e1d270 | 2007-03-15 19:03:33 -0400 | [diff] [blame] | 2 | #include "ctree.h" |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 3 | #include "disk-io.h" |
Chris Mason | 9f5fae2 | 2007-03-20 14:38:32 -0400 | [diff] [blame] | 4 | #include "transaction.h" |
Chris Mason | 1e1d270 | 2007-03-15 19:03:33 -0400 | [diff] [blame] | 5 | |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 6 | int btrfs_alloc_file_extent(struct btrfs_trans_handle *trans, |
| 7 | struct btrfs_root *root, |
| 8 | u64 objectid, u64 offset, |
| 9 | u64 num_blocks, u64 hint_block, |
| 10 | u64 *result) |
Chris Mason | 9f5fae2 | 2007-03-20 14:38:32 -0400 | [diff] [blame] | 11 | { |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 12 | struct btrfs_key ins; |
| 13 | int ret = 0; |
| 14 | struct btrfs_file_extent_item *item; |
| 15 | struct btrfs_key file_key; |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 16 | struct btrfs_path *path; |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 17 | |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 18 | path = btrfs_alloc_path(); |
| 19 | BUG_ON(!path); |
| 20 | btrfs_init_path(path); |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 21 | ret = btrfs_alloc_extent(trans, root, num_blocks, hint_block, |
Chris Mason | d0dbc62 | 2007-04-10 12:36:36 -0400 | [diff] [blame^] | 22 | (u64)-1, &ins); |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 23 | BUG_ON(ret); |
| 24 | file_key.objectid = objectid; |
| 25 | file_key.offset = offset; |
| 26 | file_key.flags = 0; |
| 27 | btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); |
| 28 | |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 29 | ret = btrfs_insert_empty_item(trans, root, path, &file_key, |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 30 | sizeof(*item)); |
Chris Mason | 9773a78 | 2007-03-27 11:26:26 -0400 | [diff] [blame] | 31 | BUG_ON(ret); |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 32 | item = btrfs_item_ptr(btrfs_buffer_leaf(path->nodes[0]), path->slots[0], |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 33 | struct btrfs_file_extent_item); |
| 34 | btrfs_set_file_extent_disk_blocknr(item, ins.objectid); |
| 35 | btrfs_set_file_extent_disk_num_blocks(item, ins.offset); |
| 36 | btrfs_set_file_extent_offset(item, 0); |
| 37 | btrfs_set_file_extent_num_blocks(item, ins.offset); |
Chris Mason | 71951f3 | 2007-03-27 09:16:29 -0400 | [diff] [blame] | 38 | btrfs_set_file_extent_generation(item, trans->transid); |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 39 | btrfs_mark_buffer_dirty(path->nodes[0]); |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 40 | *result = ins.objectid; |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 41 | btrfs_release_path(root, path); |
| 42 | btrfs_free_path(path); |
Chris Mason | 9f5fae2 | 2007-03-20 14:38:32 -0400 | [diff] [blame] | 43 | return 0; |
| 44 | } |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 45 | |
| 46 | int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans, |
| 47 | struct btrfs_root *root, |
| 48 | struct btrfs_path *path, u64 objectid, |
Chris Mason | 9773a78 | 2007-03-27 11:26:26 -0400 | [diff] [blame] | 49 | u64 offset, int mod) |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 50 | { |
| 51 | int ret; |
| 52 | struct btrfs_key file_key; |
| 53 | int ins_len = mod < 0 ? -1 : 0; |
| 54 | int cow = mod != 0; |
| 55 | |
| 56 | file_key.objectid = objectid; |
Chris Mason | 9773a78 | 2007-03-27 11:26:26 -0400 | [diff] [blame] | 57 | file_key.offset = offset; |
Chris Mason | dee26a9 | 2007-03-26 16:00:06 -0400 | [diff] [blame] | 58 | file_key.flags = 0; |
| 59 | btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); |
| 60 | ret = btrfs_search_slot(trans, root, &file_key, path, ins_len, cow); |
| 61 | return ret; |
| 62 | } |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 63 | |
| 64 | int btrfs_csum_file_block(struct btrfs_trans_handle *trans, |
| 65 | struct btrfs_root *root, |
| 66 | u64 objectid, u64 offset, |
| 67 | char *data, size_t len) |
| 68 | { |
| 69 | int ret; |
| 70 | struct btrfs_key file_key; |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 71 | struct btrfs_path *path; |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 72 | struct btrfs_csum_item *item; |
| 73 | |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 74 | path = btrfs_alloc_path(); |
| 75 | BUG_ON(!path); |
| 76 | btrfs_init_path(path); |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 77 | file_key.objectid = objectid; |
| 78 | file_key.offset = offset; |
| 79 | file_key.flags = 0; |
| 80 | btrfs_set_key_type(&file_key, BTRFS_CSUM_ITEM_KEY); |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 81 | ret = btrfs_insert_empty_item(trans, root, path, &file_key, |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 82 | BTRFS_CSUM_SIZE); |
| 83 | if (ret != 0 && ret != -EEXIST) |
| 84 | goto fail; |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 85 | item = btrfs_item_ptr(btrfs_buffer_leaf(path->nodes[0]), path->slots[0], |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 86 | struct btrfs_csum_item); |
| 87 | ret = 0; |
| 88 | ret = btrfs_csum_data(root, data, len, item->csum); |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 89 | btrfs_mark_buffer_dirty(path->nodes[0]); |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 90 | fail: |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 91 | btrfs_release_path(root, path); |
| 92 | btrfs_free_path(path); |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 93 | return ret; |
| 94 | } |
| 95 | |
| 96 | int btrfs_csum_verify_file_block(struct btrfs_root *root, |
| 97 | u64 objectid, u64 offset, |
| 98 | char *data, size_t len) |
| 99 | { |
| 100 | int ret; |
| 101 | struct btrfs_key file_key; |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 102 | struct btrfs_path *path; |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 103 | struct btrfs_csum_item *item; |
| 104 | char result[BTRFS_CSUM_SIZE]; |
| 105 | |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 106 | path = btrfs_alloc_path(); |
| 107 | BUG_ON(!path); |
| 108 | btrfs_init_path(path); |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 109 | file_key.objectid = objectid; |
| 110 | file_key.offset = offset; |
| 111 | file_key.flags = 0; |
| 112 | btrfs_set_key_type(&file_key, BTRFS_CSUM_ITEM_KEY); |
Chris Mason | 2da566e | 2007-04-02 15:43:21 -0400 | [diff] [blame] | 113 | mutex_lock(&root->fs_info->fs_mutex); |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 114 | ret = btrfs_search_slot(NULL, root, &file_key, path, 0, 0); |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 115 | if (ret) |
| 116 | goto fail; |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 117 | item = btrfs_item_ptr(btrfs_buffer_leaf(path->nodes[0]), path->slots[0], |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 118 | struct btrfs_csum_item); |
| 119 | ret = 0; |
| 120 | ret = btrfs_csum_data(root, data, len, result); |
| 121 | WARN_ON(ret); |
| 122 | if (memcmp(result, item->csum, BTRFS_CSUM_SIZE)) |
| 123 | ret = 1; |
| 124 | fail: |
Chris Mason | 5caf2a0 | 2007-04-02 11:20:42 -0400 | [diff] [blame] | 125 | btrfs_release_path(root, path); |
| 126 | btrfs_free_path(path); |
Chris Mason | 2da566e | 2007-04-02 15:43:21 -0400 | [diff] [blame] | 127 | mutex_unlock(&root->fs_info->fs_mutex); |
Chris Mason | f254e52 | 2007-03-29 15:15:27 -0400 | [diff] [blame] | 128 | return ret; |
| 129 | } |
| 130 | |