Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 1 | #define _XOPEN_SOURCE 500 |
| 2 | #include <stdio.h> |
| 3 | #include <stdlib.h> |
| 4 | #include <sys/types.h> |
| 5 | #include <sys/stat.h> |
| 6 | #include <fcntl.h> |
| 7 | #include <unistd.h> |
| 8 | #include "kerncompat.h" |
| 9 | #include "radix-tree.h" |
| 10 | #include "ctree.h" |
| 11 | #include "disk-io.h" |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 12 | #include "transaction.h" |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 13 | |
| 14 | static int allocated_blocks = 0; |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 15 | int cache_max = 10000; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 16 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 17 | static int check_tree_block(struct btrfs_root *root, struct btrfs_buffer *buf) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 18 | { |
Chris Mason | 7518a23 | 2007-03-12 12:01:18 -0400 | [diff] [blame] | 19 | if (buf->blocknr != btrfs_header_blocknr(&buf->node.header)) |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 20 | BUG(); |
Chris Mason | 7518a23 | 2007-03-12 12:01:18 -0400 | [diff] [blame] | 21 | if (root->node && btrfs_header_parentid(&buf->node.header) != |
| 22 | btrfs_header_parentid(&root->node->node.header)) |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 23 | BUG(); |
| 24 | return 0; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 25 | } |
| 26 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 27 | static int free_some_buffers(struct btrfs_root *root) |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 28 | { |
| 29 | struct list_head *node, *next; |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 30 | struct btrfs_buffer *b; |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 31 | if (root->cache_size < cache_max) |
| 32 | return 0; |
| 33 | list_for_each_safe(node, next, &root->cache) { |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 34 | b = list_entry(node, struct btrfs_buffer, cache); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 35 | if (b->count == 1) { |
| 36 | BUG_ON(!list_empty(&b->dirty)); |
| 37 | list_del_init(&b->cache); |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 38 | btrfs_block_release(root, b); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 39 | if (root->cache_size < cache_max) |
Chris Mason | 77ce684 | 2007-03-02 10:06:43 -0500 | [diff] [blame] | 40 | break; |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 41 | } |
| 42 | } |
| 43 | return 0; |
| 44 | } |
| 45 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 46 | struct btrfs_buffer *alloc_tree_block(struct btrfs_root *root, u64 blocknr) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 47 | { |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 48 | struct btrfs_buffer *buf; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 49 | int ret; |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 50 | |
| 51 | buf = malloc(sizeof(struct btrfs_buffer) + root->blocksize); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 52 | if (!buf) |
| 53 | return buf; |
| 54 | allocated_blocks++; |
| 55 | buf->blocknr = blocknr; |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 56 | buf->count = 2; |
| 57 | INIT_LIST_HEAD(&buf->dirty); |
| 58 | free_some_buffers(root); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 59 | radix_tree_preload(GFP_KERNEL); |
| 60 | ret = radix_tree_insert(&root->cache_radix, blocknr, buf); |
| 61 | radix_tree_preload_end(); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 62 | list_add_tail(&buf->cache, &root->cache); |
| 63 | root->cache_size++; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 64 | if (ret) { |
| 65 | free(buf); |
| 66 | return NULL; |
| 67 | } |
| 68 | return buf; |
| 69 | } |
| 70 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 71 | struct btrfs_buffer *find_tree_block(struct btrfs_root *root, u64 blocknr) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 72 | { |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 73 | struct btrfs_buffer *buf; |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 74 | buf = radix_tree_lookup(&root->cache_radix, blocknr); |
| 75 | if (buf) { |
| 76 | buf->count++; |
| 77 | } else { |
| 78 | buf = alloc_tree_block(root, blocknr); |
| 79 | if (!buf) { |
| 80 | BUG(); |
| 81 | return NULL; |
| 82 | } |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 83 | } |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 84 | return buf; |
| 85 | } |
| 86 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 87 | struct btrfs_buffer *read_tree_block(struct btrfs_root *root, u64 blocknr) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 88 | { |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 89 | loff_t offset = blocknr * root->blocksize; |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 90 | struct btrfs_buffer *buf; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 91 | int ret; |
| 92 | |
| 93 | buf = radix_tree_lookup(&root->cache_radix, blocknr); |
| 94 | if (buf) { |
| 95 | buf->count++; |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 96 | } else { |
| 97 | buf = alloc_tree_block(root, blocknr); |
| 98 | if (!buf) |
| 99 | return NULL; |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 100 | ret = pread(root->fp, &buf->node, root->blocksize, offset); |
| 101 | if (ret != root->blocksize) { |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 102 | free(buf); |
| 103 | return NULL; |
| 104 | } |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 105 | } |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 106 | if (check_tree_block(root, buf)) |
Chris Mason | cfaa729 | 2007-02-21 17:04:57 -0500 | [diff] [blame] | 107 | BUG(); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 108 | return buf; |
| 109 | } |
| 110 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 111 | int dirty_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root, |
| 112 | struct btrfs_buffer *buf) |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 113 | { |
| 114 | if (!list_empty(&buf->dirty)) |
| 115 | return 0; |
| 116 | list_add_tail(&buf->dirty, &root->trans); |
| 117 | buf->count++; |
| 118 | return 0; |
| 119 | } |
| 120 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 121 | int clean_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root, |
| 122 | struct btrfs_buffer *buf) |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 123 | { |
| 124 | if (!list_empty(&buf->dirty)) { |
| 125 | list_del_init(&buf->dirty); |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 126 | btrfs_block_release(root, buf); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 127 | } |
| 128 | return 0; |
| 129 | } |
| 130 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 131 | int write_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root, |
| 132 | struct btrfs_buffer *buf) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 133 | { |
| 134 | u64 blocknr = buf->blocknr; |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 135 | loff_t offset = blocknr * root->blocksize; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 136 | int ret; |
| 137 | |
Chris Mason | 7518a23 | 2007-03-12 12:01:18 -0400 | [diff] [blame] | 138 | if (buf->blocknr != btrfs_header_blocknr(&buf->node.header)) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 139 | BUG(); |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 140 | ret = pwrite(root->fp, &buf->node, root->blocksize, offset); |
| 141 | if (ret != root->blocksize) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 142 | return ret; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 143 | return 0; |
| 144 | } |
| 145 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 146 | static int __commit_transaction(struct btrfs_trans_handle *trans, struct |
| 147 | btrfs_root *root) |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 148 | { |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 149 | struct btrfs_buffer *b; |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 150 | int ret = 0; |
| 151 | int wret; |
| 152 | while(!list_empty(&root->trans)) { |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 153 | b = list_entry(root->trans.next, struct btrfs_buffer, dirty); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 154 | list_del_init(&b->dirty); |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 155 | wret = write_tree_block(trans, root, b); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 156 | if (wret) |
| 157 | ret = wret; |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 158 | btrfs_block_release(root, b); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 159 | } |
| 160 | return ret; |
| 161 | } |
| 162 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 163 | static int commit_extent_and_tree_roots(struct btrfs_trans_handle *trans, |
| 164 | struct btrfs_root *tree_root, struct |
| 165 | btrfs_root *extent_root) |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 166 | { |
| 167 | int ret; |
| 168 | u64 old_extent_block; |
| 169 | |
| 170 | while(1) { |
| 171 | old_extent_block = btrfs_root_blocknr(&extent_root->root_item); |
| 172 | if (old_extent_block == extent_root->node->blocknr) |
| 173 | break; |
| 174 | btrfs_set_root_blocknr(&extent_root->root_item, |
| 175 | extent_root->node->blocknr); |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 176 | ret = btrfs_update_root(trans, tree_root, |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 177 | &extent_root->root_key, |
| 178 | &extent_root->root_item); |
| 179 | BUG_ON(ret); |
| 180 | } |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 181 | __commit_transaction(trans, extent_root); |
| 182 | __commit_transaction(trans, tree_root); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 183 | return 0; |
| 184 | } |
| 185 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 186 | int btrfs_commit_transaction(struct btrfs_trans_handle *trans, struct |
| 187 | btrfs_root *root, struct btrfs_super_block *s) |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 188 | { |
Chris Mason | a28ec19 | 2007-03-06 20:08:01 -0500 | [diff] [blame] | 189 | int ret = 0; |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 190 | struct btrfs_buffer *snap = root->commit_root; |
| 191 | struct btrfs_key snap_key; |
Chris Mason | a28ec19 | 2007-03-06 20:08:01 -0500 | [diff] [blame] | 192 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 193 | ret = __commit_transaction(trans, root); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 194 | BUG_ON(ret); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 195 | |
| 196 | if (root->commit_root == root->node) |
| 197 | return 0; |
| 198 | |
| 199 | memcpy(&snap_key, &root->root_key, sizeof(snap_key)); |
| 200 | root->root_key.offset++; |
| 201 | |
| 202 | btrfs_set_root_blocknr(&root->root_item, root->node->blocknr); |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 203 | ret = btrfs_insert_root(trans, root->tree_root, &root->root_key, |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 204 | &root->root_item); |
| 205 | BUG_ON(ret); |
| 206 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 207 | ret = commit_extent_and_tree_roots(trans, root->tree_root, |
| 208 | root->extent_root); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 209 | BUG_ON(ret); |
| 210 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 211 | write_ctree_super(trans, root, s); |
| 212 | btrfs_finish_extent_commit(trans, root->extent_root); |
| 213 | btrfs_finish_extent_commit(trans, root->tree_root); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 214 | |
| 215 | root->commit_root = root->node; |
| 216 | root->node->count++; |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 217 | ret = btrfs_drop_snapshot(trans, root, snap); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 218 | BUG_ON(ret); |
| 219 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 220 | ret = btrfs_del_root(trans, root->tree_root, &snap_key); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 221 | BUG_ON(ret); |
| 222 | |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 223 | return ret; |
| 224 | } |
| 225 | |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 226 | static int __setup_root(struct btrfs_super_block *super, |
| 227 | struct btrfs_root *root, u64 objectid, int fp) |
Chris Mason | d97e63b | 2007-02-20 16:40:44 -0500 | [diff] [blame] | 228 | { |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 229 | INIT_LIST_HEAD(&root->trans); |
| 230 | INIT_LIST_HEAD(&root->cache); |
Chris Mason | a28ec19 | 2007-03-06 20:08:01 -0500 | [diff] [blame] | 231 | root->cache_size = 0; |
Chris Mason | d97e63b | 2007-02-20 16:40:44 -0500 | [diff] [blame] | 232 | root->fp = fp; |
Chris Mason | cfaa729 | 2007-02-21 17:04:57 -0500 | [diff] [blame] | 233 | root->node = NULL; |
Chris Mason | a28ec19 | 2007-03-06 20:08:01 -0500 | [diff] [blame] | 234 | root->commit_root = NULL; |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 235 | root->blocksize = btrfs_super_blocksize(super); |
| 236 | root->ref_cows = 0; |
Chris Mason | a28ec19 | 2007-03-06 20:08:01 -0500 | [diff] [blame] | 237 | memset(&root->current_insert, 0, sizeof(root->current_insert)); |
Chris Mason | 0579da4 | 2007-03-07 16:15:30 -0500 | [diff] [blame] | 238 | memset(&root->last_insert, 0, sizeof(root->last_insert)); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 239 | memset(&root->root_key, 0, sizeof(root->root_key)); |
| 240 | memset(&root->root_item, 0, sizeof(root->root_item)); |
| 241 | return 0; |
| 242 | } |
| 243 | |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 244 | static int find_and_setup_root(struct btrfs_super_block *super, |
| 245 | struct btrfs_root *tree_root, u64 objectid, |
| 246 | struct btrfs_root *root, int fp) |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 247 | { |
| 248 | int ret; |
| 249 | |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 250 | __setup_root(super, root, objectid, fp); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 251 | ret = btrfs_find_last_root(tree_root, objectid, |
| 252 | &root->root_item, &root->root_key); |
| 253 | BUG_ON(ret); |
| 254 | |
| 255 | root->node = read_tree_block(root, |
| 256 | btrfs_root_blocknr(&root->root_item)); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 257 | BUG_ON(!root->node); |
Chris Mason | d97e63b | 2007-02-20 16:40:44 -0500 | [diff] [blame] | 258 | return 0; |
| 259 | } |
| 260 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 261 | struct btrfs_root *open_ctree(char *filename, struct btrfs_super_block *super) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 262 | { |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 263 | struct btrfs_root *root = malloc(sizeof(struct btrfs_root)); |
| 264 | struct btrfs_root *extent_root = malloc(sizeof(struct btrfs_root)); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 265 | struct btrfs_root *tree_root = malloc(sizeof(struct btrfs_root)); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 266 | int fp; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 267 | int ret; |
| 268 | |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 269 | root->extent_root = extent_root; |
| 270 | root->tree_root = tree_root; |
| 271 | |
| 272 | extent_root->extent_root = extent_root; |
| 273 | extent_root->tree_root = tree_root; |
| 274 | |
| 275 | tree_root->extent_root = extent_root; |
| 276 | tree_root->tree_root = tree_root; |
| 277 | |
Chris Mason | c673024 | 2007-02-26 10:46:55 -0500 | [diff] [blame] | 278 | fp = open(filename, O_CREAT | O_RDWR, 0600); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 279 | if (fp < 0) { |
| 280 | free(root); |
| 281 | return NULL; |
| 282 | } |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 283 | INIT_RADIX_TREE(&root->cache_radix, GFP_KERNEL); |
Chris Mason | a28ec19 | 2007-03-06 20:08:01 -0500 | [diff] [blame] | 284 | INIT_RADIX_TREE(&root->pinned_radix, GFP_KERNEL); |
| 285 | INIT_RADIX_TREE(&extent_root->pinned_radix, GFP_KERNEL); |
Chris Mason | 9a8dd15 | 2007-02-23 08:38:36 -0500 | [diff] [blame] | 286 | INIT_RADIX_TREE(&extent_root->cache_radix, GFP_KERNEL); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 287 | INIT_RADIX_TREE(&tree_root->pinned_radix, GFP_KERNEL); |
| 288 | INIT_RADIX_TREE(&tree_root->cache_radix, GFP_KERNEL); |
| 289 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 290 | ret = pread(fp, super, sizeof(struct btrfs_super_block), |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 291 | BTRFS_SUPER_INFO_OFFSET); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 292 | if (ret == 0 || btrfs_super_root(super) == 0) { |
Chris Mason | 5c680ed | 2007-02-22 11:39:13 -0500 | [diff] [blame] | 293 | printf("making new FS!\n"); |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 294 | ret = mkfs(fp, 0, 1024); |
Chris Mason | d97e63b | 2007-02-20 16:40:44 -0500 | [diff] [blame] | 295 | if (ret) |
| 296 | return NULL; |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 297 | ret = pread(fp, super, sizeof(struct btrfs_super_block), |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 298 | BTRFS_SUPER_INFO_OFFSET); |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 299 | if (ret != sizeof(struct btrfs_super_block)) |
Chris Mason | d97e63b | 2007-02-20 16:40:44 -0500 | [diff] [blame] | 300 | return NULL; |
| 301 | } |
| 302 | BUG_ON(ret < 0); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 303 | |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 304 | __setup_root(super, tree_root, BTRFS_ROOT_TREE_OBJECTID, fp); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 305 | tree_root->node = read_tree_block(tree_root, btrfs_super_root(super)); |
| 306 | BUG_ON(!tree_root->node); |
| 307 | |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 308 | ret = find_and_setup_root(super, tree_root, BTRFS_EXTENT_TREE_OBJECTID, |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 309 | extent_root, fp); |
| 310 | BUG_ON(ret); |
| 311 | |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 312 | ret = find_and_setup_root(super, tree_root, BTRFS_FS_TREE_OBJECTID, |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 313 | root, fp); |
| 314 | BUG_ON(ret); |
| 315 | |
Chris Mason | a28ec19 | 2007-03-06 20:08:01 -0500 | [diff] [blame] | 316 | root->commit_root = root->node; |
| 317 | root->node->count++; |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 318 | root->ref_cows = 1; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 319 | return root; |
| 320 | } |
| 321 | |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 322 | int write_ctree_super(struct btrfs_trans_handle *trans, struct btrfs_root |
| 323 | *root, struct btrfs_super_block *s) |
Chris Mason | cfaa729 | 2007-02-21 17:04:57 -0500 | [diff] [blame] | 324 | { |
| 325 | int ret; |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 326 | btrfs_set_super_root(s, root->tree_root->node->blocknr); |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 327 | ret = pwrite(root->fp, s, sizeof(*s), |
Chris Mason | 123abc8 | 2007-03-14 14:14:43 -0400 | [diff] [blame] | 328 | BTRFS_SUPER_INFO_OFFSET); |
Chris Mason | cfaa729 | 2007-02-21 17:04:57 -0500 | [diff] [blame] | 329 | if (ret != sizeof(*s)) { |
| 330 | fprintf(stderr, "failed to write new super block err %d\n", ret); |
| 331 | return ret; |
| 332 | } |
| 333 | return 0; |
| 334 | } |
| 335 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 336 | static int drop_cache(struct btrfs_root *root) |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 337 | { |
| 338 | while(!list_empty(&root->cache)) { |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 339 | struct btrfs_buffer *b = list_entry(root->cache.next, |
| 340 | struct btrfs_buffer, cache); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 341 | list_del_init(&b->cache); |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 342 | btrfs_block_release(root, b); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 343 | } |
| 344 | return 0; |
| 345 | } |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 346 | int close_ctree(struct btrfs_root *root, struct btrfs_super_block *s) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 347 | { |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 348 | int ret; |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 349 | struct btrfs_trans_handle *trans; |
| 350 | |
| 351 | trans = root->running_transaction; |
| 352 | btrfs_commit_transaction(trans, root, s); |
| 353 | ret = commit_extent_and_tree_roots(trans, root->tree_root, |
| 354 | root->extent_root); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 355 | BUG_ON(ret); |
Chris Mason | e089f05 | 2007-03-16 16:20:31 -0400 | [diff] [blame^] | 356 | write_ctree_super(trans, root, s); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 357 | drop_cache(root->extent_root); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 358 | drop_cache(root->tree_root); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 359 | drop_cache(root); |
| 360 | BUG_ON(!list_empty(&root->trans)); |
| 361 | BUG_ON(!list_empty(&root->extent_root->trans)); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 362 | BUG_ON(!list_empty(&root->tree_root->trans)); |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 363 | |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 364 | close(root->fp); |
| 365 | if (root->node) |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 366 | btrfs_block_release(root, root->node); |
Chris Mason | cfaa729 | 2007-02-21 17:04:57 -0500 | [diff] [blame] | 367 | if (root->extent_root->node) |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 368 | btrfs_block_release(root->extent_root, root->extent_root->node); |
Chris Mason | 3768f36 | 2007-03-13 16:47:54 -0400 | [diff] [blame] | 369 | if (root->tree_root->node) |
| 370 | btrfs_block_release(root->tree_root, root->tree_root->node); |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 371 | btrfs_block_release(root, root->commit_root); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 372 | free(root); |
| 373 | printf("on close %d blocks are allocated\n", allocated_blocks); |
| 374 | return 0; |
| 375 | } |
| 376 | |
Chris Mason | 234b63a | 2007-03-13 10:46:10 -0400 | [diff] [blame] | 377 | void btrfs_block_release(struct btrfs_root *root, struct btrfs_buffer *buf) |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 378 | { |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 379 | buf->count--; |
Chris Mason | cfaa729 | 2007-02-21 17:04:57 -0500 | [diff] [blame] | 380 | if (buf->count < 0) |
| 381 | BUG(); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 382 | if (buf->count == 0) { |
Chris Mason | 02217ed | 2007-03-02 16:08:05 -0500 | [diff] [blame] | 383 | BUG_ON(!list_empty(&buf->cache)); |
| 384 | BUG_ON(!list_empty(&buf->dirty)); |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 385 | if (!radix_tree_lookup(&root->cache_radix, buf->blocknr)) |
| 386 | BUG(); |
| 387 | radix_tree_delete(&root->cache_radix, buf->blocknr); |
| 388 | memset(buf, 0, sizeof(*buf)); |
| 389 | free(buf); |
| 390 | BUG_ON(allocated_blocks == 0); |
| 391 | allocated_blocks--; |
Chris Mason | ed2ff2c | 2007-03-01 18:59:40 -0500 | [diff] [blame] | 392 | BUG_ON(root->cache_size == 0); |
| 393 | root->cache_size--; |
Chris Mason | eb60cea | 2007-02-02 09:18:22 -0500 | [diff] [blame] | 394 | } |
| 395 | } |
| 396 | |