diff options
| author | Ingo Molnar <mingo@elte.hu> | 2009-01-12 05:32:03 -0500 |
|---|---|---|
| committer | Ingo Molnar <mingo@elte.hu> | 2009-01-12 05:32:03 -0500 |
| commit | e3ee1e123183ca9847e74b7b8e2694c9e3b817a6 (patch) | |
| tree | 652a84674ed05eaa46a813de2223af0bd0168a5a /fs/btrfs | |
| parent | 5762ba1873b0bb9faa631aaa02f533c2b9837f82 (diff) | |
| parent | c59765042f53a79a7a65585042ff463b69cb248c (diff) | |
Merge commit 'v2.6.29-rc1' into timers/hrtimers
Conflicts:
kernel/time/tick-common.c
Diffstat (limited to 'fs/btrfs')
55 files changed, 42385 insertions, 0 deletions
diff --git a/fs/btrfs/Makefile b/fs/btrfs/Makefile new file mode 100644 index 000000000000..d2cf5a54a4b8 --- /dev/null +++ b/fs/btrfs/Makefile | |||
| @@ -0,0 +1,25 @@ | |||
| 1 | ifneq ($(KERNELRELEASE),) | ||
| 2 | # kbuild part of makefile | ||
| 3 | |||
| 4 | obj-$(CONFIG_BTRFS_FS) := btrfs.o | ||
| 5 | btrfs-y := super.o ctree.o extent-tree.o print-tree.o root-tree.o dir-item.o \ | ||
| 6 | file-item.o inode-item.o inode-map.o disk-io.o \ | ||
| 7 | transaction.o inode.o file.o tree-defrag.o \ | ||
| 8 | extent_map.o sysfs.o struct-funcs.o xattr.o ordered-data.o \ | ||
| 9 | extent_io.o volumes.o async-thread.o ioctl.o locking.o orphan.o \ | ||
| 10 | ref-cache.o export.o tree-log.o acl.o free-space-cache.o zlib.o \ | ||
| 11 | compression.o | ||
| 12 | else | ||
| 13 | |||
| 14 | # Normal Makefile | ||
| 15 | |||
| 16 | KERNELDIR := /lib/modules/`uname -r`/build | ||
| 17 | all: | ||
| 18 | $(MAKE) -C $(KERNELDIR) M=`pwd` CONFIG_BTRFS_FS=m modules | ||
| 19 | |||
| 20 | modules_install: | ||
| 21 | $(MAKE) -C $(KERNELDIR) M=`pwd` modules_install | ||
| 22 | clean: | ||
| 23 | $(MAKE) -C $(KERNELDIR) M=`pwd` clean | ||
| 24 | |||
| 25 | endif | ||
diff --git a/fs/btrfs/acl.c b/fs/btrfs/acl.c new file mode 100644 index 000000000000..1d53b62dbba5 --- /dev/null +++ b/fs/btrfs/acl.c | |||
| @@ -0,0 +1,351 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2007 Red Hat. 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 | |||
| 19 | #include <linux/fs.h> | ||
| 20 | #include <linux/string.h> | ||
| 21 | #include <linux/xattr.h> | ||
| 22 | #include <linux/posix_acl_xattr.h> | ||
| 23 | #include <linux/posix_acl.h> | ||
| 24 | #include <linux/sched.h> | ||
| 25 | |||
| 26 | #include "ctree.h" | ||
| 27 | #include "btrfs_inode.h" | ||
| 28 | #include "xattr.h" | ||
| 29 | |||
| 30 | #ifdef CONFIG_FS_POSIX_ACL | ||
| 31 | |||
| 32 | static void btrfs_update_cached_acl(struct inode *inode, | ||
| 33 | struct posix_acl **p_acl, | ||
| 34 | struct posix_acl *acl) | ||
| 35 | { | ||
| 36 | spin_lock(&inode->i_lock); | ||
| 37 | if (*p_acl && *p_acl != BTRFS_ACL_NOT_CACHED) | ||
| 38 | posix_acl_release(*p_acl); | ||
| 39 | *p_acl = posix_acl_dup(acl); | ||
| 40 | spin_unlock(&inode->i_lock); | ||
| 41 | } | ||
| 42 | |||
| 43 | static struct posix_acl *btrfs_get_acl(struct inode *inode, int type) | ||
| 44 | { | ||
| 45 | int size; | ||
| 46 | const char *name; | ||
| 47 | char *value = NULL; | ||
| 48 | struct posix_acl *acl = NULL, **p_acl; | ||
| 49 | |||
| 50 | switch (type) { | ||
| 51 | case ACL_TYPE_ACCESS: | ||
| 52 | name = POSIX_ACL_XATTR_ACCESS; | ||
| 53 | p_acl = &BTRFS_I(inode)->i_acl; | ||
| 54 | break; | ||
| 55 | case ACL_TYPE_DEFAULT: | ||
| 56 | name = POSIX_ACL_XATTR_DEFAULT; | ||
| 57 | p_acl = &BTRFS_I(inode)->i_default_acl; | ||
| 58 | break; | ||
| 59 | default: | ||
| 60 | return ERR_PTR(-EINVAL); | ||
| 61 | } | ||
| 62 | |||
| 63 | spin_lock(&inode->i_lock); | ||
| 64 | if (*p_acl != BTRFS_ACL_NOT_CACHED) | ||
| 65 | acl = posix_acl_dup(*p_acl); | ||
| 66 | spin_unlock(&inode->i_lock); | ||
| 67 | |||
| 68 | if (acl) | ||
| 69 | return acl; | ||
| 70 | |||
| 71 | |||
| 72 | size = __btrfs_getxattr(inode, name, "", 0); | ||
| 73 | if (size > 0) { | ||
| 74 | value = kzalloc(size, GFP_NOFS); | ||
| 75 | if (!value) | ||
| 76 | return ERR_PTR(-ENOMEM); | ||
| 77 | size = __btrfs_getxattr(inode, name, value, size); | ||
| 78 | if (size > 0) { | ||
| 79 | acl = posix_acl_from_xattr(value, size); | ||
| 80 | btrfs_update_cached_acl(inode, p_acl, acl); | ||
| 81 | } | ||
| 82 | kfree(value); | ||
| 83 | } else if (size == -ENOENT) { | ||
| 84 | acl = NULL; | ||
| 85 | btrfs_update_cached_acl(inode, p_acl, acl); | ||
| 86 | } | ||
| 87 | |||
| 88 | return acl; | ||
| 89 | } | ||
| 90 | |||
| 91 | static int btrfs_xattr_get_acl(struct inode *inode, int type, | ||
| 92 | void *value, size_t size) | ||
| 93 | { | ||
| 94 | struct posix_acl *acl; | ||
| 95 | int ret = 0; | ||
| 96 | |||
| 97 | acl = btrfs_get_acl(inode, type); | ||
| 98 | |||
| 99 | if (IS_ERR(acl)) | ||
| 100 | return PTR_ERR(acl); | ||
| 101 | if (acl == NULL) | ||
| 102 | return -ENODATA; | ||
| 103 | ret = posix_acl_to_xattr(acl, value, size); | ||
| 104 | posix_acl_release(acl); | ||
| 105 | |||
| 106 | return ret; | ||
| 107 | } | ||
| 108 | |||
| 109 | /* | ||
| 110 | * Needs to be called with fs_mutex held | ||
| 111 | */ | ||
| 112 | static int btrfs_set_acl(struct inode *inode, struct posix_acl *acl, int type) | ||
| 113 | { | ||
| 114 | int ret, size = 0; | ||
| 115 | const char *name; | ||
| 116 | struct posix_acl **p_acl; | ||
| 117 | char *value = NULL; | ||
| 118 | mode_t mode; | ||
| 119 | |||
| 120 | if (acl) { | ||
| 121 | ret = posix_acl_valid(acl); | ||
| 122 | if (ret < 0) | ||
| 123 | return ret; | ||
| 124 | ret = 0; | ||
| 125 | } | ||
| 126 | |||
| 127 | switch (type) { | ||
| 128 | case ACL_TYPE_ACCESS: | ||
| 129 | mode = inode->i_mode; | ||
| 130 | ret = posix_acl_equiv_mode(acl, &mode); | ||
| 131 | if (ret < 0) | ||
| 132 | return ret; | ||
| 133 | ret = 0; | ||
| 134 | inode->i_mode = mode; | ||
| 135 | name = POSIX_ACL_XATTR_ACCESS; | ||
| 136 | p_acl = &BTRFS_I(inode)->i_acl; | ||
| 137 | break; | ||
| 138 | case ACL_TYPE_DEFAULT: | ||
| 139 | if (!S_ISDIR(inode->i_mode)) | ||
| 140 | return acl ? -EINVAL : 0; | ||
| 141 | name = POSIX_ACL_XATTR_DEFAULT; | ||
| 142 | p_acl = &BTRFS_I(inode)->i_default_acl; | ||
| 143 | break; | ||
| 144 | default: | ||
| 145 | return -EINVAL; | ||
| 146 | } | ||
| 147 | |||
| 148 | if (acl) { | ||
| 149 | size = posix_acl_xattr_size(acl->a_count); | ||
| 150 | value = kmalloc(size, GFP_NOFS); | ||
| 151 | if (!value) { | ||
| 152 | ret = -ENOMEM; | ||
| 153 | goto out; | ||
| 154 | } | ||
| 155 | |||
| 156 | ret = posix_acl_to_xattr(acl, value, size); | ||
| 157 | if (ret < 0) | ||
| 158 | goto out; | ||
| 159 | } | ||
| 160 | |||
| 161 | ret = __btrfs_setxattr(inode, name, value, size, 0); | ||
| 162 | |||
| 163 | out: | ||
| 164 | kfree(value); | ||
| 165 | |||
| 166 | if (!ret) | ||
| 167 | btrfs_update_cached_acl(inode, p_acl, acl); | ||
| 168 | |||
| 169 | return ret; | ||
| 170 | } | ||
| 171 | |||
| 172 | static int btrfs_xattr_set_acl(struct inode *inode, int type, | ||
| 173 | const void *value, size_t size) | ||
| 174 | { | ||
| 175 | int ret = 0; | ||
| 176 | struct posix_acl *acl = NULL; | ||
| 177 | |||
| 178 | if (value) { | ||
| 179 | acl = posix_acl_from_xattr(value, size); | ||
| 180 | if (acl == NULL) { | ||
| 181 | value = NULL; | ||
| 182 | size = 0; | ||
| 183 | } else if (IS_ERR(acl)) { | ||
| 184 | return PTR_ERR(acl); | ||
| 185 | } | ||
| 186 | } | ||
| 187 | |||
| 188 | ret = btrfs_set_acl(inode, acl, type); | ||
| 189 | |||
| 190 | posix_acl_release(acl); | ||
| 191 | |||
| 192 | return ret; | ||
| 193 | } | ||
| 194 | |||
| 195 | |||
| 196 | static int btrfs_xattr_acl_access_get(struct inode *inode, const char *name, | ||
| 197 | void *value, size_t size) | ||
| 198 | { | ||
| 199 | return btrfs_xattr_get_acl(inode, ACL_TYPE_ACCESS, value, size); | ||
| 200 | } | ||
| 201 | |||
| 202 | static int btrfs_xattr_acl_access_set(struct inode *inode, const char *name, | ||
| 203 | const void *value, size_t size, int flags) | ||
| 204 | { | ||
| 205 | return btrfs_xattr_set_acl(inode, ACL_TYPE_ACCESS, value, size); | ||
| 206 | } | ||
| 207 | |||
| 208 | static int btrfs_xattr_acl_default_get(struct inode *inode, const char *name, | ||
| 209 | void *value, size_t size) | ||
| 210 | { | ||
| 211 | return btrfs_xattr_get_acl(inode, ACL_TYPE_DEFAULT, value, size); | ||
| 212 | } | ||
| 213 | |||
| 214 | static int btrfs_xattr_acl_default_set(struct inode *inode, const char *name, | ||
| 215 | const void *value, size_t size, int flags) | ||
| 216 | { | ||
| 217 | return btrfs_xattr_set_acl(inode, ACL_TYPE_DEFAULT, value, size); | ||
| 218 | } | ||
| 219 | |||
| 220 | int btrfs_check_acl(struct inode *inode, int mask) | ||
| 221 | { | ||
| 222 | struct posix_acl *acl; | ||
| 223 | int error = -EAGAIN; | ||
| 224 | |||
| 225 | acl = btrfs_get_acl(inode, ACL_TYPE_ACCESS); | ||
| 226 | |||
| 227 | if (IS_ERR(acl)) | ||
| 228 | return PTR_ERR(acl); | ||
| 229 | if (acl) { | ||
| 230 | error = posix_acl_permission(inode, acl, mask); | ||
| 231 | posix_acl_release(acl); | ||
| 232 | } | ||
| 233 | |||
| 234 | return error; | ||
| 235 | } | ||
| 236 | |||
| 237 | /* | ||
| 238 | * btrfs_init_acl is already generally called under fs_mutex, so the locking | ||
| 239 | * stuff has been fixed to work with that. If the locking stuff changes, we | ||
| 240 | * need to re-evaluate the acl locking stuff. | ||
| 241 | */ | ||
| 242 | int btrfs_init_acl(struct inode *inode, struct inode *dir) | ||
| 243 | { | ||
| 244 | struct posix_acl *acl = NULL; | ||
| 245 | int ret = 0; | ||
| 246 | |||
| 247 | /* this happens with subvols */ | ||
| 248 | if (!dir) | ||
| 249 | return 0; | ||
| 250 | |||
| 251 | if (!S_ISLNK(inode->i_mode)) { | ||
| 252 | if (IS_POSIXACL(dir)) { | ||
| 253 | acl = btrfs_get_acl(dir, ACL_TYPE_DEFAULT); | ||
| 254 | if (IS_ERR(acl)) | ||
| 255 | return PTR_ERR(acl); | ||
| 256 | } | ||
| 257 | |||
| 258 | if (!acl) | ||
| 259 | inode->i_mode &= ~current->fs->umask; | ||
| 260 | } | ||
| 261 | |||
| 262 | if (IS_POSIXACL(dir) && acl) { | ||
| 263 | struct posix_acl *clone; | ||
| 264 | mode_t mode; | ||
| 265 | |||
| 266 | if (S_ISDIR(inode->i_mode)) { | ||
| 267 | ret = btrfs_set_acl(inode, acl, ACL_TYPE_DEFAULT); | ||
| 268 | if (ret) | ||
| 269 | goto failed; | ||
| 270 | } | ||
| 271 | clone = posix_acl_clone(acl, GFP_NOFS); | ||
| 272 | ret = -ENOMEM; | ||
| 273 | if (!clone) | ||
| 274 | goto failed; | ||
| 275 | |||
| 276 | mode = inode->i_mode; | ||
| 277 | ret = posix_acl_create_masq(clone, &mode); | ||
| 278 | if (ret >= 0) { | ||
| 279 | inode->i_mode = mode; | ||
| 280 | if (ret > 0) { | ||
| 281 | /* we need an acl */ | ||
| 282 | ret = btrfs_set_acl(inode, clone, | ||
| 283 | ACL_TYPE_ACCESS); | ||
| 284 | } | ||
| 285 | } | ||
| 286 | } | ||
| 287 | failed: | ||
| 288 | posix_acl_release(acl); | ||
| 289 | |||
| 290 | return ret; | ||
| 291 | } | ||
| 292 | |||
| 293 | int btrfs_acl_chmod(struct inode *inode) | ||
| 294 | { | ||
| 295 | struct posix_acl *acl, *clone; | ||
| 296 | int ret = 0; | ||
| 297 | |||
| 298 | if (S_ISLNK(inode->i_mode)) | ||
| 299 | return -EOPNOTSUPP; | ||
| 300 | |||
| 301 | if (!IS_POSIXACL(inode)) | ||
| 302 | return 0; | ||
| 303 | |||
| 304 | acl = btrfs_get_acl(inode, ACL_TYPE_ACCESS); | ||
| 305 | if (IS_ERR(acl) || !acl) | ||
| 306 | return PTR_ERR(acl); | ||
| 307 | |||
| 308 | clone = posix_acl_clone(acl, GFP_KERNEL); | ||
| 309 | posix_acl_release(acl); | ||
| 310 | if (!clone) | ||
| 311 | return -ENOMEM; | ||
| 312 | |||
| 313 | ret = posix_acl_chmod_masq(clone, inode->i_mode); | ||
| 314 | if (!ret) | ||
| 315 | ret = btrfs_set_acl(inode, clone, ACL_TYPE_ACCESS); | ||
| 316 | |||
| 317 | posix_acl_release(clone); | ||
| 318 | |||
| 319 | return ret; | ||
| 320 | } | ||
| 321 | |||
| 322 | struct xattr_handler btrfs_xattr_acl_default_handler = { | ||
| 323 | .prefix = POSIX_ACL_XATTR_DEFAULT, | ||
| 324 | .get = btrfs_xattr_acl_default_get, | ||
| 325 | .set = btrfs_xattr_acl_default_set, | ||
| 326 | }; | ||
| 327 | |||
| 328 | struct xattr_handler btrfs_xattr_acl_access_handler = { | ||
| 329 | .prefix = POSIX_ACL_XATTR_ACCESS, | ||
| 330 | .get = btrfs_xattr_acl_access_get, | ||
| 331 | .set = btrfs_xattr_acl_access_set, | ||
| 332 | }; | ||
| 333 | |||
| 334 | #else /* CONFIG_FS_POSIX_ACL */ | ||
| 335 | |||
| 336 | int btrfs_acl_chmod(struct inode *inode) | ||
| 337 | { | ||
| 338 | return 0; | ||
| 339 | } | ||
| 340 | |||
| 341 | int btrfs_init_acl(struct inode *inode, struct inode *dir) | ||
| 342 | { | ||
| 343 | return 0; | ||
| 344 | } | ||
| 345 | |||
| 346 | int btrfs_check_acl(struct inode *inode, int mask) | ||
| 347 | { | ||
| 348 | return 0; | ||
| 349 | } | ||
| 350 | |||
| 351 | #endif /* CONFIG_FS_POSIX_ACL */ | ||
diff --git a/fs/btrfs/async-thread.c b/fs/btrfs/async-thread.c new file mode 100644 index 000000000000..8e2fec05dbe0 --- /dev/null +++ b/fs/btrfs/async-thread.c | |||
| @@ -0,0 +1,419 @@ | |||
| 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 | |||
| 19 | #include <linux/version.h> | ||
| 20 | #include <linux/kthread.h> | ||
| 21 | #include <linux/list.h> | ||
| 22 | #include <linux/spinlock.h> | ||
| 23 | # include <linux/freezer.h> | ||
| 24 | #include "async-thread.h" | ||
| 25 | |||
| 26 | #define WORK_QUEUED_BIT 0 | ||
| 27 | #define WORK_DONE_BIT 1 | ||
| 28 | #define WORK_ORDER_DONE_BIT 2 | ||
| 29 | |||
| 30 | /* | ||
| 31 | * container for the kthread task pointer and the list of pending work | ||
| 32 | * One of these is allocated per thread. | ||
| 33 | */ | ||
| 34 | struct btrfs_worker_thread { | ||
| 35 | /* pool we belong to */ | ||
| 36 | struct btrfs_workers *workers; | ||
| 37 | |||
| 38 | /* list of struct btrfs_work that are waiting for service */ | ||
| 39 | struct list_head pending; | ||
| 40 | |||
| 41 | /* list of worker threads from struct btrfs_workers */ | ||
| 42 | struct list_head worker_list; | ||
| 43 | |||
| 44 | /* kthread */ | ||
| 45 | struct task_struct *task; | ||
| 46 | |||
| 47 | /* number of things on the pending list */ | ||
| 48 | atomic_t num_pending; | ||
| 49 | |||
| 50 | unsigned long sequence; | ||
| 51 | |||
| 52 | /* protects the pending list. */ | ||
| 53 | spinlock_t lock; | ||
| 54 | |||
| 55 | /* set to non-zero when this thread is already awake and kicking */ | ||
| 56 | int working; | ||
| 57 | |||
| 58 | /* are we currently idle */ | ||
| 59 | int idle; | ||
| 60 | }; | ||
| 61 | |||
| 62 | /* | ||
| 63 | * helper function to move a thread onto the idle list after it | ||
| 64 | * has finished some requests. | ||
| 65 | */ | ||
| 66 | static void check_idle_worker(struct btrfs_worker_thread *worker) | ||
| 67 | { | ||
| 68 | if (!worker->idle && atomic_read(&worker->num_pending) < | ||
| 69 | worker->workers->idle_thresh / 2) { | ||
| 70 | unsigned long flags; | ||
| 71 | spin_lock_irqsave(&worker->workers->lock, flags); | ||
| 72 | worker->idle = 1; | ||
| 73 | list_move(&worker->worker_list, &worker->workers->idle_list); | ||
| 74 | spin_unlock_irqrestore(&worker->workers->lock, flags); | ||
| 75 | } | ||
| 76 | } | ||
| 77 | |||
| 78 | /* | ||
| 79 | * helper function to move a thread off the idle list after new | ||
| 80 | * pending work is added. | ||
| 81 | */ | ||
| 82 | static void check_busy_worker(struct btrfs_worker_thread *worker) | ||
| 83 | { | ||
| 84 | if (worker->idle && atomic_read(&worker->num_pending) >= | ||
| 85 | worker->workers->idle_thresh) { | ||
| 86 | unsigned long flags; | ||
| 87 | spin_lock_irqsave(&worker->workers->lock, flags); | ||
| 88 | worker->idle = 0; | ||
| 89 | list_move_tail(&worker->worker_list, | ||
| 90 | &worker->workers->worker_list); | ||
| 91 | spin_unlock_irqrestore(&worker->workers->lock, flags); | ||
| 92 | } | ||
| 93 | } | ||
| 94 | |||
| 95 | static noinline int run_ordered_completions(struct btrfs_workers *workers, | ||
| 96 | struct btrfs_work *work) | ||
| 97 | { | ||
| 98 | unsigned long flags; | ||
| 99 | |||
| 100 | if (!workers->ordered) | ||
| 101 | return 0; | ||
| 102 | |||
| 103 | set_bit(WORK_DONE_BIT, &work->flags); | ||
| 104 | |||
| 105 | spin_lock_irqsave(&workers->lock, flags); | ||
| 106 | |||
| 107 | while (!list_empty(&workers->order_list)) { | ||
| 108 | work = list_entry(workers->order_list.next, | ||
| 109 | struct btrfs_work, order_list); | ||
| 110 | |||
| 111 | if (!test_bit(WORK_DONE_BIT, &work->flags)) | ||
| 112 | break; | ||
| 113 | |||
| 114 | /* we are going to call the ordered done function, but | ||
| 115 | * we leave the work item on the list as a barrier so | ||
| 116 | * that later work items that are done don't have their | ||
| 117 | * functions called before this one returns | ||
| 118 | */ | ||
| 119 | if (test_and_set_bit(WORK_ORDER_DONE_BIT, &work->flags)) | ||
| 120 | break; | ||
| 121 | |||
| 122 | spin_unlock_irqrestore(&workers->lock, flags); | ||
| 123 | |||
| 124 | work->ordered_func(work); | ||
| 125 | |||
| 126 | /* now take the lock again and call the freeing code */ | ||
| 127 | spin_lock_irqsave(&workers->lock, flags); | ||
| 128 | list_del(&work->order_list); | ||
| 129 | work->ordered_free(work); | ||
| 130 | } | ||
| 131 | |||
| 132 | spin_unlock_irqrestore(&workers->lock, flags); | ||
| 133 | return 0; | ||
| 134 | } | ||
| 135 | |||
| 136 | /* | ||
| 137 | * main loop for servicing work items | ||
| 138 | */ | ||
| 139 | static int worker_loop(void *arg) | ||
| 140 | { | ||
| 141 | struct btrfs_worker_thread *worker = arg; | ||
| 142 | struct list_head *cur; | ||
| 143 | struct btrfs_work *work; | ||
| 144 | do { | ||
| 145 | spin_lock_irq(&worker->lock); | ||
| 146 | while (!list_empty(&worker->pending)) { | ||
| 147 | cur = worker->pending.next; | ||
| 148 | work = list_entry(cur, struct btrfs_work, list); | ||
| 149 | list_del(&work->list); | ||
| 150 | clear_bit(WORK_QUEUED_BIT, &work->flags); | ||
| 151 | |||
| 152 | work->worker = worker; | ||
| 153 | spin_unlock_irq(&worker->lock); | ||
| 154 | |||
| 155 | work->func(work); | ||
| 156 | |||
| 157 | atomic_dec(&worker->num_pending); | ||
| 158 | /* | ||
| 159 | * unless this is an ordered work queue, | ||
| 160 | * 'work' was probably freed by func above. | ||
| 161 | */ | ||
| 162 | run_ordered_completions(worker->workers, work); | ||
| 163 | |||
| 164 | spin_lock_irq(&worker->lock); | ||
| 165 | check_idle_worker(worker); | ||
| 166 | |||
| 167 | } | ||
| 168 | worker->working = 0; | ||
| 169 | if (freezing(current)) { | ||
| 170 | refrigerator(); | ||
| 171 | } else { | ||
| 172 | set_current_state(TASK_INTERRUPTIBLE); | ||
| 173 | spin_unlock_irq(&worker->lock); | ||
| 174 | if (!kthread_should_stop()) | ||
| 175 | schedule(); | ||
| 176 | __set_current_state(TASK_RUNNING); | ||
| 177 | } | ||
| 178 | } while (!kthread_should_stop()); | ||
| 179 | return 0; | ||
| 180 | } | ||
| 181 | |||
| 182 | /* | ||
| 183 | * this will wait for all the worker threads to shutdown | ||
| 184 | */ | ||
| 185 | int btrfs_stop_workers(struct btrfs_workers *workers) | ||
| 186 | { | ||
| 187 | struct list_head *cur; | ||
| 188 | struct btrfs_worker_thread *worker; | ||
| 189 | |||
| 190 | list_splice_init(&workers->idle_list, &workers->worker_list); | ||
| 191 | while (!list_empty(&workers->worker_list)) { | ||
| 192 | cur = workers->worker_list.next; | ||
| 193 | worker = list_entry(cur, struct btrfs_worker_thread, | ||
| 194 | worker_list); | ||
| 195 | kthread_stop(worker->task); | ||
| 196 | list_del(&worker->worker_list); | ||
| 197 | kfree(worker); | ||
| 198 | } | ||
| 199 | return 0; | ||
| 200 | } | ||
| 201 | |||
| 202 | /* | ||
| 203 | * simple init on struct btrfs_workers | ||
| 204 | */ | ||
| 205 | void btrfs_init_workers(struct btrfs_workers *workers, char *name, int max) | ||
| 206 | { | ||
| 207 | workers->num_workers = 0; | ||
| 208 | INIT_LIST_HEAD(&workers->worker_list); | ||
| 209 | INIT_LIST_HEAD(&workers->idle_list); | ||
| 210 | INIT_LIST_HEAD(&workers->order_list); | ||
| 211 | spin_lock_init(&workers->lock); | ||
| 212 | workers->max_workers = max; | ||
| 213 | workers->idle_thresh = 32; | ||
| 214 | workers->name = name; | ||
| 215 | workers->ordered = 0; | ||
| 216 | } | ||
| 217 | |||
| 218 | /* | ||
| 219 | * starts new worker threads. This does not enforce the max worker | ||
| 220 | * count in case you need to temporarily go past it. | ||
| 221 | */ | ||
| 222 | int btrfs_start_workers(struct btrfs_workers *workers, int num_workers) | ||
| 223 | { | ||
| 224 | struct btrfs_worker_thread *worker; | ||
| 225 | int ret = 0; | ||
| 226 | int i; | ||
| 227 | |||
| 228 | for (i = 0; i < num_workers; i++) { | ||
| 229 | worker = kzalloc(sizeof(*worker), GFP_NOFS); | ||
| 230 | if (!worker) { | ||
| 231 | ret = -ENOMEM; | ||
| 232 | goto fail; | ||
| 233 | } | ||
| 234 | |||
| 235 | INIT_LIST_HEAD(&worker->pending); | ||
| 236 | INIT_LIST_HEAD(&worker->worker_list); | ||
| 237 | spin_lock_init(&worker->lock); | ||
| 238 | atomic_set(&worker->num_pending, 0); | ||
| 239 | worker->task = kthread_run(worker_loop, worker, | ||
| 240 | "btrfs-%s-%d", workers->name, | ||
| 241 | workers->num_workers + i); | ||
| 242 | worker->workers = workers; | ||
| 243 | if (IS_ERR(worker->task)) { | ||
| 244 | kfree(worker); | ||
| 245 | ret = PTR_ERR(worker->task); | ||
| 246 | goto fail; | ||
| 247 | } | ||
| 248 | |||
| 249 | spin_lock_irq(&workers->lock); | ||
| 250 | list_add_tail(&worker->worker_list, &workers->idle_list); | ||
| 251 | worker->idle = 1; | ||
| 252 | workers->num_workers++; | ||
| 253 | spin_unlock_irq(&workers->lock); | ||
| 254 | } | ||
| 255 | return 0; | ||
| 256 | fail: | ||
| 257 | btrfs_stop_workers(workers); | ||
| 258 | return ret; | ||
| 259 | } | ||
| 260 | |||
| 261 | /* | ||
| 262 | * run through the list and find a worker thread that doesn't have a lot | ||
| 263 | * to do right now. This can return null if we aren't yet at the thread | ||
| 264 | * count limit and all of the threads are busy. | ||
| 265 | */ | ||
| 266 | static struct btrfs_worker_thread *next_worker(struct btrfs_workers *workers) | ||
| 267 | { | ||
| 268 | struct btrfs_worker_thread *worker; | ||
| 269 | struct list_head *next; | ||
| 270 | int enforce_min = workers->num_workers < workers->max_workers; | ||
| 271 | |||
| 272 | /* | ||
| 273 | * if we find an idle thread, don't move it to the end of the | ||
| 274 | * idle list. This improves the chance that the next submission | ||
| 275 | * will reuse the same thread, and maybe catch it while it is still | ||
| 276 | * working | ||
| 277 | */ | ||
| 278 | if (!list_empty(&workers->idle_list)) { | ||
| 279 | next = workers->idle_list.next; | ||
| 280 | worker = list_entry(next, struct btrfs_worker_thread, | ||
| 281 | worker_list); | ||
| 282 | return worker; | ||
| 283 | } | ||
| 284 | if (enforce_min || list_empty(&workers->worker_list)) | ||
| 285 | return NULL; | ||
| 286 | |||
| 287 | /* | ||
| 288 | * if we pick a busy task, move the task to the end of the list. | ||
| 289 | * hopefully this will keep things somewhat evenly balanced. | ||
| 290 | * Do the move in batches based on the sequence number. This groups | ||
| 291 | * requests submitted at roughly the same time onto the same worker. | ||
| 292 | */ | ||
| 293 | next = workers->worker_list.next; | ||
| 294 | worker = list_entry(next, struct btrfs_worker_thread, worker_list); | ||
| 295 | atomic_inc(&worker->num_pending); | ||
| 296 | worker->sequence++; | ||
| 297 | |||
| 298 | if (worker->sequence % workers->idle_thresh == 0) | ||
| 299 | list_move_tail(next, &workers->worker_list); | ||
| 300 | return worker; | ||
| 301 | } | ||
| 302 | |||
| 303 | /* | ||
| 304 | * selects a worker thread to take the next job. This will either find | ||
| 305 | * an idle worker, start a new worker up to the max count, or just return | ||
| 306 | * one of the existing busy workers. | ||
| 307 | */ | ||
| 308 | static struct btrfs_worker_thread *find_worker(struct btrfs_workers *workers) | ||
| 309 | { | ||
| 310 | struct btrfs_worker_thread *worker; | ||
| 311 | unsigned long flags; | ||
| 312 | |||
| 313 | again: | ||
| 314 | spin_lock_irqsave(&workers->lock, flags); | ||
| 315 | worker = next_worker(workers); | ||
| 316 | spin_unlock_irqrestore(&workers->lock, flags); | ||
| 317 | |||
| 318 | if (!worker) { | ||
| 319 | spin_lock_irqsave(&workers->lock, flags); | ||
| 320 | if (workers->num_workers >= workers->max_workers) { | ||
| 321 | struct list_head *fallback = NULL; | ||
| 322 | /* | ||
| 323 | * we have failed to find any workers, just | ||
| 324 | * return the force one | ||
| 325 | */ | ||
| 326 | if (!list_empty(&workers->worker_list)) | ||
| 327 | fallback = workers->worker_list.next; | ||
| 328 | if (!list_empty(&workers->idle_list)) | ||
| 329 | fallback = workers->idle_list.next; | ||
| 330 | BUG_ON(!fallback); | ||
| 331 | worker = list_entry(fallback, | ||
| 332 | struct btrfs_worker_thread, worker_list); | ||
| 333 | spin_unlock_irqrestore(&workers->lock, flags); | ||
| 334 | } else { | ||
| 335 | spin_unlock_irqrestore(&workers->lock, flags); | ||
| 336 | /* we're below the limit, start another worker */ | ||
| 337 | btrfs_start_workers(workers, 1); | ||
| 338 | goto again; | ||
| 339 | } | ||
| 340 | } | ||
| 341 | return worker; | ||
| 342 | } | ||
| 343 | |||
| 344 | /* | ||
| 345 | * btrfs_requeue_work just puts the work item back on the tail of the list | ||
| 346 | * it was taken from. It is intended for use with long running work functions | ||
| 347 | * that make some progress and want to give the cpu up for others. | ||
| 348 | */ | ||
| 349 | int btrfs_requeue_work(struct btrfs_work *work) | ||
| 350 | { | ||
| 351 | struct btrfs_worker_thread *worker = work->worker; | ||
| 352 | unsigned long flags; | ||
| 353 | |||
| 354 | if (test_and_set_bit(WORK_QUEUED_BIT, &work->flags)) | ||
| 355 | goto out; | ||
| 356 | |||
| 357 | spin_lock_irqsave(&worker->lock, flags); | ||
| 358 | atomic_inc(&worker->num_pending); | ||
| 359 | list_add_tail(&work->list, &worker->pending); | ||
| 360 | |||
| 361 | /* by definition we're busy, take ourselves off the idle | ||
| 362 | * list | ||
| 363 | */ | ||
| 364 | if (worker->idle) { | ||
| 365 | spin_lock_irqsave(&worker->workers->lock, flags); | ||
| 366 | worker->idle = 0; | ||
| 367 | list_move_tail(&worker->worker_list, | ||
| 368 | &worker->workers->worker_list); | ||
| 369 | spin_unlock_irqrestore(&worker->workers->lock, flags); | ||
| 370 | } | ||
| 371 | |||
| 372 | spin_unlock_irqrestore(&worker->lock, flags); | ||
| 373 | |||
| 374 | out: | ||
| 375 | return 0; | ||
| 376 | } | ||
| 377 | |||
| 378 | /* | ||
| 379 | * places a struct btrfs_work into the pending queue of one of the kthreads | ||
| 380 | */ | ||
| 381 | int btrfs_queue_worker(struct btrfs_workers *workers, struct btrfs_work *work) | ||
| 382 | { | ||
| 383 | struct btrfs_worker_thread *worker; | ||
| 384 | unsigned long flags; | ||
| 385 | int wake = 0; | ||
| 386 | |||
| 387 | /* don't requeue something already on a list */ | ||
| 388 | if (test_and_set_bit(WORK_QUEUED_BIT, &work->flags)) | ||
| 389 | goto out; | ||
| 390 | |||
| 391 | worker = find_worker(workers); | ||
| 392 | if (workers->ordered) { | ||
| 393 | spin_lock_irqsave(&workers->lock, flags); | ||
| 394 | list_add_tail(&work->order_list, &workers->order_list); | ||
| 395 | spin_unlock_irqrestore(&workers->lock, flags); | ||
| 396 | } else { | ||
| 397 | INIT_LIST_HEAD(&work->order_list); | ||
| 398 | } | ||
| 399 | |||
| 400 | spin_lock_irqsave(&worker->lock, flags); | ||
| 401 | atomic_inc(&worker->num_pending); | ||
| 402 | check_busy_worker(worker); | ||
| 403 | list_add_tail(&work->list, &worker->pending); | ||
| 404 | |||
| 405 | /* | ||
| 406 | * avoid calling into wake_up_process if this thread has already | ||
| 407 | * been kicked | ||
| 408 | */ | ||
| 409 | if (!worker->working) | ||
| 410 | wake = 1; | ||
| 411 | worker->working = 1; | ||
| 412 | |||
| 413 | spin_unlock_irqrestore(&worker->lock, flags); | ||
| 414 | |||
| 415 | if (wake) | ||
| 416 | wake_up_process(worker->task); | ||
| 417 | out: | ||
| 418 | return 0; | ||
| 419 | } | ||
diff --git a/fs/btrfs/async-thread.h b/fs/btrfs/async-thread.h new file mode 100644 index 000000000000..31be4ed8b63e --- /dev/null +++ b/fs/btrfs/async-thread.h | |||
| @@ -0,0 +1,101 @@ | |||
| 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 | |||
| 19 | #ifndef __BTRFS_ASYNC_THREAD_ | ||
| 20 | #define __BTRFS_ASYNC_THREAD_ | ||
| 21 | |||
| 22 | struct btrfs_worker_thread; | ||
| 23 | |||
| 24 | /* | ||
| 25 | * This is similar to a workqueue, but it is meant to spread the operations | ||
| 26 | * across all available cpus instead of just the CPU that was used to | ||
| 27 | * queue the work. There is also some batching introduced to try and | ||
| 28 | * cut down on context switches. | ||
| 29 | * | ||
| 30 | * By default threads are added on demand up to 2 * the number of cpus. | ||
| 31 | * Changing struct btrfs_workers->max_workers is one way to prevent | ||
| 32 | * demand creation of kthreads. | ||
| 33 | * | ||
| 34 | * the basic model of these worker threads is to embed a btrfs_work | ||
| 35 | * structure in your own data struct, and use container_of in a | ||
| 36 | * work function to get back to your data struct. | ||
| 37 | */ | ||
| 38 | struct btrfs_work { | ||
| 39 | /* | ||
| 40 | * func should be set to the function you want called | ||
| 41 | * your work struct is passed as the only arg | ||
| 42 | * | ||
| 43 | * ordered_func must be set for work sent to an ordered work queue, | ||
| 44 | * and it is called to complete a given work item in the same | ||
| 45 | * order they were sent to the queue. | ||
| 46 | */ | ||
| 47 | void (*func)(struct btrfs_work *work); | ||
| 48 | void (*ordered_func)(struct btrfs_work *work); | ||
| 49 | void (*ordered_free)(struct btrfs_work *work); | ||
| 50 | |||
| 51 | /* | ||
| 52 | * flags should be set to zero. It is used to make sure the | ||
| 53 | * struct is only inserted once into the list. | ||
| 54 | */ | ||
| 55 | unsigned long flags; | ||
| 56 | |||
| 57 | /* don't touch these */ | ||
| 58 | struct btrfs_worker_thread *worker; | ||
| 59 | struct list_head list; | ||
| 60 | struct list_head order_list; | ||
| 61 | }; | ||
| 62 | |||
| 63 | struct btrfs_workers { | ||
| 64 | /* current number of running workers */ | ||
| 65 | int num_workers; | ||
| 66 | |||
| 67 | /* max number of workers allowed. changed by btrfs_start_workers */ | ||
| 68 | int max_workers; | ||
| 69 | |||
| 70 | /* once a worker has this many requests or fewer, it is idle */ | ||
| 71 | int idle_thresh; | ||
| 72 | |||
| 73 | /* force completions in the order they were queued */ | ||
| 74 | int ordered; | ||
| 75 | |||
| 76 | /* list with all the work threads. The workers on the idle thread | ||
| 77 | * may be actively servicing jobs, but they haven't yet hit the | ||
| 78 | * idle thresh limit above. | ||
| 79 | */ | ||
| 80 | struct list_head worker_list; | ||
| 81 | struct list_head idle_list; | ||
| 82 | |||
| 83 | /* | ||
| 84 | * when operating in ordered mode, this maintains the list | ||
| 85 | * of work items waiting for completion | ||
| 86 | */ | ||
| 87 | struct list_head order_list; | ||
| 88 | |||
| 89 | /* lock for finding the next worker thread to queue on */ | ||
| 90 | spinlock_t lock; | ||
| 91 | |||
| 92 | /* extra name for this worker, used for current->name */ | ||
| 93 | char *name; | ||
| 94 | }; | ||
| 95 | |||
| 96 | int btrfs_queue_worker(struct btrfs_workers *workers, struct btrfs_work *work); | ||
| 97 | int btrfs_start_workers(struct btrfs_workers *workers, int num_workers); | ||
| 98 | int btrfs_stop_workers(struct btrfs_workers *workers); | ||
| 99 | void btrfs_init_workers(struct btrfs_workers *workers, char *name, int max); | ||
| 100 | int btrfs_requeue_work(struct btrfs_work *work); | ||
| 101 | #endif | ||
diff --git a/fs/btrfs/btrfs_inode.h b/fs/btrfs/btrfs_inode.h new file mode 100644 index 000000000000..a8c9693b75ac --- /dev/null +++ b/fs/btrfs/btrfs_inode.h | |||
| @@ -0,0 +1,131 @@ | |||
| 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 | |||
| 19 | #ifndef __BTRFS_I__ | ||
| 20 | #define __BTRFS_I__ | ||
| 21 | |||
| 22 | #include "extent_map.h" | ||
| 23 | #include "extent_io.h" | ||
| 24 | #include "ordered-data.h" | ||
| 25 | |||
| 26 | /* in memory btrfs inode */ | ||
| 27 | struct btrfs_inode { | ||
| 28 | /* which subvolume this inode belongs to */ | ||
| 29 | struct btrfs_root *root; | ||
| 30 | |||
| 31 | /* key used to find this inode on disk. This is used by the code | ||
| 32 | * to read in roots of subvolumes | ||
| 33 | */ | ||
| 34 | struct btrfs_key location; | ||
| 35 | |||
| 36 | /* the extent_tree has caches of all the extent mappings to disk */ | ||
| 37 | struct extent_map_tree extent_tree; | ||
| 38 | |||
| 39 | /* the io_tree does range state (DIRTY, LOCKED etc) */ | ||
| 40 | struct extent_io_tree io_tree; | ||
| 41 | |||
| 42 | /* special utility tree used to record which mirrors have already been | ||
| 43 | * tried when checksums fail for a given block | ||
| 44 | */ | ||
| 45 | struct extent_io_tree io_failure_tree; | ||
| 46 | |||
| 47 | /* held while inesrting or deleting extents from files */ | ||
| 48 | struct mutex extent_mutex; | ||
| 49 | |||
| 50 | /* held while logging the inode in tree-log.c */ | ||
| 51 | struct mutex log_mutex; | ||
| 52 | |||
| 53 | /* used to order data wrt metadata */ | ||
| 54 | struct btrfs_ordered_inode_tree ordered_tree; | ||
| 55 | |||
| 56 | /* standard acl pointers */ | ||
| 57 | struct posix_acl *i_acl; | ||
| 58 | struct posix_acl *i_default_acl; | ||
| 59 | |||
| 60 | /* for keeping track of orphaned inodes */ | ||
| 61 | struct list_head i_orphan; | ||
| 62 | |||
| 63 | /* list of all the delalloc inodes in the FS. There are times we need | ||
| 64 | * to write all the delalloc pages to disk, and this list is used | ||
| 65 | * to walk them all. | ||
| 66 | */ | ||
| 67 | struct list_head delalloc_inodes; | ||
| 68 | |||
| 69 | /* full 64 bit generation number, struct vfs_inode doesn't have a big | ||
| 70 | * enough field for this. | ||
| 71 | */ | ||
| 72 | u64 generation; | ||
| 73 | |||
| 74 | /* sequence number for NFS changes */ | ||
| 75 | u64 sequence; | ||
| 76 | |||
| 77 | /* | ||
| 78 | * transid of the trans_handle that last modified this inode | ||
| 79 | */ | ||
| 80 | u64 last_trans; | ||
| 81 | /* | ||
| 82 | * transid that last logged this inode | ||
| 83 | */ | ||
| 84 | u64 logged_trans; | ||
| 85 | |||
| 86 | /* | ||
| 87 | * trans that last made a change that should be fully fsync'd. This | ||
| 88 | * gets reset to zero each time the inode is logged | ||
| 89 | */ | ||
| 90 | u64 log_dirty_trans; | ||
| 91 | |||
| 92 | /* total number of bytes pending delalloc, used by stat to calc the | ||
| 93 | * real block usage of the file | ||
| 94 | */ | ||
| 95 | u64 delalloc_bytes; | ||
| 96 | |||
| 97 | /* | ||
| 98 | * the size of the file stored in the metadata on disk. data=ordered | ||
| 99 | * means the in-memory i_size might be larger than the size on disk | ||
| 100 | * because not all the blocks are written yet. | ||
| 101 | */ | ||
| 102 | u64 disk_i_size; | ||
| 103 | |||
| 104 | /* flags field from the on disk inode */ | ||
| 105 | u32 flags; | ||
| 106 | |||
| 107 | /* | ||
| 108 | * if this is a directory then index_cnt is the counter for the index | ||
| 109 | * number for new files that are created | ||
| 110 | */ | ||
| 111 | u64 index_cnt; | ||
| 112 | |||
| 113 | /* the start of block group preferred for allocations. */ | ||
| 114 | u64 block_group; | ||
| 115 | |||
| 116 | struct inode vfs_inode; | ||
| 117 | }; | ||
| 118 | |||
| 119 | static inline struct btrfs_inode *BTRFS_I(struct inode *inode) | ||
| 120 | { | ||
| 121 | return container_of(inode, struct btrfs_inode, vfs_inode); | ||
| 122 | } | ||
| 123 | |||
| 124 | static inline void btrfs_i_size_write(struct inode *inode, u64 size) | ||
| 125 | { | ||
| 126 | inode->i_size = size; | ||
| 127 | BTRFS_I(inode)->disk_i_size = size; | ||
| 128 | } | ||
| 129 | |||
| 130 | |||
| 131 | #endif | ||
diff --git a/fs/btrfs/compat.h b/fs/btrfs/compat.h new file mode 100644 index 000000000000..7c4503ef6efd --- /dev/null +++ b/fs/btrfs/compat.h | |||
| @@ -0,0 +1,7 @@ | |||
| 1 | #ifndef _COMPAT_H_ | ||
| 2 | #define _COMPAT_H_ | ||
| 3 | |||
| 4 | #define btrfs_drop_nlink(inode) drop_nlink(inode) | ||
| 5 | #define btrfs_inc_nlink(inode) inc_nlink(inode) | ||
| 6 | |||
| 7 | #endif /* _COMPAT_H_ */ | ||
diff --git a/fs/btrfs/compression.c b/fs/btrfs/compression.c new file mode 100644 index 000000000000..ee848d8585d9 --- /dev/null +++ b/fs/btrfs/compression.c | |||
| @@ -0,0 +1,709 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | |||
| 19 | #include <linux/kernel.h> | ||
| 20 | #include <linux/bio.h> | ||
| 21 | #include <linux/buffer_head.h> | ||
| 22 | #include <linux/file.h> | ||
| 23 | #include <linux/fs.h> | ||
| 24 | #include <linux/pagemap.h> | ||
| 25 | #include <linux/highmem.h> | ||
| 26 | #include <linux/time.h> | ||
| 27 | #include <linux/init.h> | ||
| 28 | #include <linux/string.h> | ||
| 29 | #include <linux/smp_lock.h> | ||
| 30 | #include <linux/backing-dev.h> | ||
| 31 | #include <linux/mpage.h> | ||
| 32 | #include <linux/swap.h> | ||
| 33 | #include <linux/writeback.h> | ||
| 34 | #include <linux/bit_spinlock.h> | ||
| 35 | #include <linux/version.h> | ||
| 36 | #include <linux/pagevec.h> | ||
| 37 | #include "compat.h" | ||
| 38 | #include "ctree.h" | ||
| 39 | #include "disk-io.h" | ||
| 40 | #include "transaction.h" | ||
| 41 | #include "btrfs_inode.h" | ||
| 42 | #include "volumes.h" | ||
| 43 | #include "ordered-data.h" | ||
| 44 | #include "compression.h" | ||
| 45 | #include "extent_io.h" | ||
| 46 | #include "extent_map.h" | ||
| 47 | |||
| 48 | struct compressed_bio { | ||
| 49 | /* number of bios pending for this compressed extent */ | ||
| 50 | atomic_t pending_bios; | ||
| 51 | |||
| 52 | /* the pages with the compressed data on them */ | ||
| 53 | struct page **compressed_pages; | ||
| 54 | |||
| 55 | /* inode that owns this data */ | ||
| 56 | struct inode *inode; | ||
| 57 | |||
| 58 | /* starting offset in the inode for our pages */ | ||
| 59 | u64 start; | ||
| 60 | |||
| 61 | /* number of bytes in the inode we're working on */ | ||
| 62 | unsigned long len; | ||
| 63 | |||
| 64 | /* number of bytes on disk */ | ||
| 65 | unsigned long compressed_len; | ||
| 66 | |||
| 67 | /* number of compressed pages in the array */ | ||
| 68 | unsigned long nr_pages; | ||
| 69 | |||
| 70 | /* IO errors */ | ||
| 71 | int errors; | ||
| 72 | int mirror_num; | ||
| 73 | |||
| 74 | /* for reads, this is the bio we are copying the data into */ | ||
| 75 | struct bio *orig_bio; | ||
| 76 | |||
| 77 | /* | ||
| 78 | * the start of a variable length array of checksums only | ||
| 79 | * used by reads | ||
| 80 | */ | ||
| 81 | u32 sums; | ||
| 82 | }; | ||
| 83 | |||
| 84 | static inline int compressed_bio_size(struct btrfs_root *root, | ||
| 85 | unsigned long disk_size) | ||
| 86 | { | ||
| 87 | u16 csum_size = btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 88 | return sizeof(struct compressed_bio) + | ||
| 89 | ((disk_size + root->sectorsize - 1) / root->sectorsize) * | ||
| 90 | csum_size; | ||
| 91 | } | ||
| 92 | |||
| 93 | static struct bio *compressed_bio_alloc(struct block_device *bdev, | ||
| 94 | u64 first_byte, gfp_t gfp_flags) | ||
| 95 | { | ||
| 96 | struct bio *bio; | ||
| 97 | int nr_vecs; | ||
| 98 | |||
| 99 | nr_vecs = bio_get_nr_vecs(bdev); | ||
| 100 | bio = bio_alloc(gfp_flags, nr_vecs); | ||
| 101 | |||
| 102 | if (bio == NULL && (current->flags & PF_MEMALLOC)) { | ||
| 103 | while (!bio && (nr_vecs /= 2)) | ||
| 104 | bio = bio_alloc(gfp_flags, nr_vecs); | ||
| 105 | } | ||
| 106 | |||
| 107 | if (bio) { | ||
| 108 | bio->bi_size = 0; | ||
| 109 | bio->bi_bdev = bdev; | ||
| 110 | bio->bi_sector = first_byte >> 9; | ||
| 111 | } | ||
| 112 | return bio; | ||
| 113 | } | ||
| 114 | |||
| 115 | static int check_compressed_csum(struct inode *inode, | ||
| 116 | struct compressed_bio *cb, | ||
| 117 | u64 disk_start) | ||
| 118 | { | ||
| 119 | int ret; | ||
| 120 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 121 | struct page *page; | ||
| 122 | unsigned long i; | ||
| 123 | char *kaddr; | ||
| 124 | u32 csum; | ||
| 125 | u32 *cb_sum = &cb->sums; | ||
| 126 | |||
| 127 | if (btrfs_test_flag(inode, NODATASUM)) | ||
| 128 | return 0; | ||
| 129 | |||
| 130 | for (i = 0; i < cb->nr_pages; i++) { | ||
| 131 | page = cb->compressed_pages[i]; | ||
| 132 | csum = ~(u32)0; | ||
| 133 | |||
| 134 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 135 | csum = btrfs_csum_data(root, kaddr, csum, PAGE_CACHE_SIZE); | ||
| 136 | btrfs_csum_final(csum, (char *)&csum); | ||
| 137 | kunmap_atomic(kaddr, KM_USER0); | ||
| 138 | |||
| 139 | if (csum != *cb_sum) { | ||
| 140 | printk(KERN_INFO "btrfs csum failed ino %lu " | ||
| 141 | "extent %llu csum %u " | ||
| 142 | "wanted %u mirror %d\n", inode->i_ino, | ||
| 143 | (unsigned long long)disk_start, | ||
| 144 | csum, *cb_sum, cb->mirror_num); | ||
| 145 | ret = -EIO; | ||
| 146 | goto fail; | ||
| 147 | } | ||
| 148 | cb_sum++; | ||
| 149 | |||
| 150 | } | ||
| 151 | ret = 0; | ||
| 152 | fail: | ||
| 153 | return ret; | ||
| 154 | } | ||
| 155 | |||
| 156 | /* when we finish reading compressed pages from the disk, we | ||
| 157 | * decompress them and then run the bio end_io routines on the | ||
| 158 | * decompressed pages (in the inode address space). | ||
| 159 | * | ||
| 160 | * This allows the checksumming and other IO error handling routines | ||
| 161 | * to work normally | ||
| 162 | * | ||
| 163 | * The compressed pages are freed here, and it must be run | ||
| 164 | * in process context | ||
| 165 | */ | ||
| 166 | static void end_compressed_bio_read(struct bio *bio, int err) | ||
| 167 | { | ||
| 168 | struct extent_io_tree *tree; | ||
| 169 | struct compressed_bio *cb = bio->bi_private; | ||
| 170 | struct inode *inode; | ||
| 171 | struct page *page; | ||
| 172 | unsigned long index; | ||
| 173 | int ret; | ||
| 174 | |||
| 175 | if (err) | ||
| 176 | cb->errors = 1; | ||
| 177 | |||
| 178 | /* if there are more bios still pending for this compressed | ||
| 179 | * extent, just exit | ||
| 180 | */ | ||
| 181 | if (!atomic_dec_and_test(&cb->pending_bios)) | ||
| 182 | goto out; | ||
| 183 | |||
| 184 | inode = cb->inode; | ||
| 185 | ret = check_compressed_csum(inode, cb, (u64)bio->bi_sector << 9); | ||
| 186 | if (ret) | ||
| 187 | goto csum_failed; | ||
| 188 | |||
| 189 | /* ok, we're the last bio for this extent, lets start | ||
| 190 | * the decompression. | ||
| 191 | */ | ||
| 192 | tree = &BTRFS_I(inode)->io_tree; | ||
| 193 | ret = btrfs_zlib_decompress_biovec(cb->compressed_pages, | ||
| 194 | cb->start, | ||
| 195 | cb->orig_bio->bi_io_vec, | ||
| 196 | cb->orig_bio->bi_vcnt, | ||
| 197 | cb->compressed_len); | ||
| 198 | csum_failed: | ||
| 199 | if (ret) | ||
| 200 | cb->errors = 1; | ||
| 201 | |||
| 202 | /* release the compressed pages */ | ||
| 203 | index = 0; | ||
| 204 | for (index = 0; index < cb->nr_pages; index++) { | ||
| 205 | page = cb->compressed_pages[index]; | ||
| 206 | page->mapping = NULL; | ||
| 207 | page_cache_release(page); | ||
| 208 | } | ||
| 209 | |||
| 210 | /* do io completion on the original bio */ | ||
| 211 | if (cb->errors) { | ||
| 212 | bio_io_error(cb->orig_bio); | ||
| 213 | } else { | ||
| 214 | int bio_index = 0; | ||
| 215 | struct bio_vec *bvec = cb->orig_bio->bi_io_vec; | ||
| 216 | |||
| 217 | /* | ||
| 218 | * we have verified the checksum already, set page | ||
| 219 | * checked so the end_io handlers know about it | ||
| 220 | */ | ||
| 221 | while (bio_index < cb->orig_bio->bi_vcnt) { | ||
| 222 | SetPageChecked(bvec->bv_page); | ||
| 223 | bvec++; | ||
| 224 | bio_index++; | ||
| 225 | } | ||
| 226 | bio_endio(cb->orig_bio, 0); | ||
| 227 | } | ||
| 228 | |||
| 229 | /* finally free the cb struct */ | ||
| 230 | kfree(cb->compressed_pages); | ||
| 231 | kfree(cb); | ||
| 232 | out: | ||
| 233 | bio_put(bio); | ||
| 234 | } | ||
| 235 | |||
| 236 | /* | ||
| 237 | * Clear the writeback bits on all of the file | ||
| 238 | * pages for a compressed write | ||
| 239 | */ | ||
| 240 | static noinline int end_compressed_writeback(struct inode *inode, u64 start, | ||
| 241 | unsigned long ram_size) | ||
| 242 | { | ||
| 243 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 244 | unsigned long end_index = (start + ram_size - 1) >> PAGE_CACHE_SHIFT; | ||
| 245 | struct page *pages[16]; | ||
| 246 | unsigned long nr_pages = end_index - index + 1; | ||
| 247 | int i; | ||
| 248 | int ret; | ||
| 249 | |||
| 250 | while (nr_pages > 0) { | ||
| 251 | ret = find_get_pages_contig(inode->i_mapping, index, | ||
| 252 | min_t(unsigned long, | ||
| 253 | nr_pages, ARRAY_SIZE(pages)), pages); | ||
| 254 | if (ret == 0) { | ||
| 255 | nr_pages -= 1; | ||
| 256 | index += 1; | ||
| 257 | continue; | ||
| 258 | } | ||
| 259 | for (i = 0; i < ret; i++) { | ||
| 260 | end_page_writeback(pages[i]); | ||
| 261 | page_cache_release(pages[i]); | ||
| 262 | } | ||
| 263 | nr_pages -= ret; | ||
| 264 | index += ret; | ||
| 265 | } | ||
| 266 | /* the inode may be gone now */ | ||
| 267 | return 0; | ||
| 268 | } | ||
| 269 | |||
| 270 | /* | ||
| 271 | * do the cleanup once all the compressed pages hit the disk. | ||
| 272 | * This will clear writeback on the file pages and free the compressed | ||
| 273 | * pages. | ||
| 274 | * | ||
| 275 | * This also calls the writeback end hooks for the file pages so that | ||
| 276 | * metadata and checksums can be updated in the file. | ||
| 277 | */ | ||
| 278 | static void end_compressed_bio_write(struct bio *bio, int err) | ||
| 279 | { | ||
| 280 | struct extent_io_tree *tree; | ||
| 281 | struct compressed_bio *cb = bio->bi_private; | ||
| 282 | struct inode *inode; | ||
| 283 | struct page *page; | ||
| 284 | unsigned long index; | ||
| 285 | |||
| 286 | if (err) | ||
| 287 | cb->errors = 1; | ||
| 288 | |||
| 289 | /* if there are more bios still pending for this compressed | ||
| 290 | * extent, just exit | ||
| 291 | */ | ||
| 292 | if (!atomic_dec_and_test(&cb->pending_bios)) | ||
| 293 | goto out; | ||
| 294 | |||
| 295 | /* ok, we're the last bio for this extent, step one is to | ||
| 296 | * call back into the FS and do all the end_io operations | ||
| 297 | */ | ||
| 298 | inode = cb->inode; | ||
| 299 | tree = &BTRFS_I(inode)->io_tree; | ||
| 300 | cb->compressed_pages[0]->mapping = cb->inode->i_mapping; | ||
| 301 | tree->ops->writepage_end_io_hook(cb->compressed_pages[0], | ||
| 302 | cb->start, | ||
| 303 | cb->start + cb->len - 1, | ||
| 304 | NULL, 1); | ||
| 305 | cb->compressed_pages[0]->mapping = NULL; | ||
| 306 | |||
| 307 | end_compressed_writeback(inode, cb->start, cb->len); | ||
| 308 | /* note, our inode could be gone now */ | ||
| 309 | |||
| 310 | /* | ||
| 311 | * release the compressed pages, these came from alloc_page and | ||
| 312 | * are not attached to the inode at all | ||
| 313 | */ | ||
| 314 | index = 0; | ||
| 315 | for (index = 0; index < cb->nr_pages; index++) { | ||
| 316 | page = cb->compressed_pages[index]; | ||
| 317 | page->mapping = NULL; | ||
| 318 | page_cache_release(page); | ||
| 319 | } | ||
| 320 | |||
| 321 | /* finally free the cb struct */ | ||
| 322 | kfree(cb->compressed_pages); | ||
| 323 | kfree(cb); | ||
| 324 | out: | ||
| 325 | bio_put(bio); | ||
| 326 | } | ||
| 327 | |||
| 328 | /* | ||
| 329 | * worker function to build and submit bios for previously compressed pages. | ||
| 330 | * The corresponding pages in the inode should be marked for writeback | ||
| 331 | * and the compressed pages should have a reference on them for dropping | ||
| 332 | * when the IO is complete. | ||
| 333 | * | ||
| 334 | * This also checksums the file bytes and gets things ready for | ||
| 335 | * the end io hooks. | ||
| 336 | */ | ||
| 337 | int btrfs_submit_compressed_write(struct inode *inode, u64 start, | ||
| 338 | unsigned long len, u64 disk_start, | ||
| 339 | unsigned long compressed_len, | ||
| 340 | struct page **compressed_pages, | ||
| 341 | unsigned long nr_pages) | ||
| 342 | { | ||
| 343 | struct bio *bio = NULL; | ||
| 344 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 345 | struct compressed_bio *cb; | ||
| 346 | unsigned long bytes_left; | ||
| 347 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 348 | int page_index = 0; | ||
| 349 | struct page *page; | ||
| 350 | u64 first_byte = disk_start; | ||
| 351 | struct block_device *bdev; | ||
| 352 | int ret; | ||
| 353 | |||
| 354 | WARN_ON(start & ((u64)PAGE_CACHE_SIZE - 1)); | ||
| 355 | cb = kmalloc(compressed_bio_size(root, compressed_len), GFP_NOFS); | ||
| 356 | atomic_set(&cb->pending_bios, 0); | ||
| 357 | cb->errors = 0; | ||
| 358 | cb->inode = inode; | ||
| 359 | cb->start = start; | ||
| 360 | cb->len = len; | ||
| 361 | cb->mirror_num = 0; | ||
| 362 | cb->compressed_pages = compressed_pages; | ||
| 363 | cb->compressed_len = compressed_len; | ||
| 364 | cb->orig_bio = NULL; | ||
| 365 | cb->nr_pages = nr_pages; | ||
| 366 | |||
| 367 | bdev = BTRFS_I(inode)->root->fs_info->fs_devices->latest_bdev; | ||
| 368 | |||
| 369 | bio = compressed_bio_alloc(bdev, first_byte, GFP_NOFS); | ||
| 370 | bio->bi_private = cb; | ||
| 371 | bio->bi_end_io = end_compressed_bio_write; | ||
| 372 | atomic_inc(&cb->pending_bios); | ||
| 373 | |||
| 374 | /* create and submit bios for the compressed pages */ | ||
| 375 | bytes_left = compressed_len; | ||
| 376 | for (page_index = 0; page_index < cb->nr_pages; page_index++) { | ||
| 377 | page = compressed_pages[page_index]; | ||
| 378 | page->mapping = inode->i_mapping; | ||
| 379 | if (bio->bi_size) | ||
| 380 | ret = io_tree->ops->merge_bio_hook(page, 0, | ||
| 381 | PAGE_CACHE_SIZE, | ||
| 382 | bio, 0); | ||
| 383 | else | ||
| 384 | ret = 0; | ||
| 385 | |||
| 386 | page->mapping = NULL; | ||
| 387 | if (ret || bio_add_page(bio, page, PAGE_CACHE_SIZE, 0) < | ||
| 388 | PAGE_CACHE_SIZE) { | ||
| 389 | bio_get(bio); | ||
| 390 | |||
| 391 | /* | ||
| 392 | * inc the count before we submit the bio so | ||
| 393 | * we know the end IO handler won't happen before | ||
| 394 | * we inc the count. Otherwise, the cb might get | ||
| 395 | * freed before we're done setting it up | ||
| 396 | */ | ||
| 397 | atomic_inc(&cb->pending_bios); | ||
| 398 | ret = btrfs_bio_wq_end_io(root->fs_info, bio, 0); | ||
| 399 | BUG_ON(ret); | ||
| 400 | |||
| 401 | ret = btrfs_csum_one_bio(root, inode, bio, start, 1); | ||
| 402 | BUG_ON(ret); | ||
| 403 | |||
| 404 | ret = btrfs_map_bio(root, WRITE, bio, 0, 1); | ||
| 405 | BUG_ON(ret); | ||
| 406 | |||
| 407 | bio_put(bio); | ||
| 408 | |||
| 409 | bio = compressed_bio_alloc(bdev, first_byte, GFP_NOFS); | ||
| 410 | bio->bi_private = cb; | ||
| 411 | bio->bi_end_io = end_compressed_bio_write; | ||
| 412 | bio_add_page(bio, page, PAGE_CACHE_SIZE, 0); | ||
| 413 | } | ||
| 414 | if (bytes_left < PAGE_CACHE_SIZE) { | ||
| 415 | printk("bytes left %lu compress len %lu nr %lu\n", | ||
| 416 | bytes_left, cb->compressed_len, cb->nr_pages); | ||
| 417 | } | ||
| 418 | bytes_left -= PAGE_CACHE_SIZE; | ||
| 419 | first_byte += PAGE_CACHE_SIZE; | ||
| 420 | cond_resched(); | ||
| 421 | } | ||
| 422 | bio_get(bio); | ||
| 423 | |||
| 424 | ret = btrfs_bio_wq_end_io(root->fs_info, bio, 0); | ||
| 425 | BUG_ON(ret); | ||
| 426 | |||
| 427 | ret = btrfs_csum_one_bio(root, inode, bio, start, 1); | ||
| 428 | BUG_ON(ret); | ||
| 429 | |||
| 430 | ret = btrfs_map_bio(root, WRITE, bio, 0, 1); | ||
| 431 | BUG_ON(ret); | ||
| 432 | |||
| 433 | bio_put(bio); | ||
| 434 | return 0; | ||
| 435 | } | ||
| 436 | |||
| 437 | static noinline int add_ra_bio_pages(struct inode *inode, | ||
| 438 | u64 compressed_end, | ||
| 439 | struct compressed_bio *cb) | ||
| 440 | { | ||
| 441 | unsigned long end_index; | ||
| 442 | unsigned long page_index; | ||
| 443 | u64 last_offset; | ||
| 444 | u64 isize = i_size_read(inode); | ||
| 445 | int ret; | ||
| 446 | struct page *page; | ||
| 447 | unsigned long nr_pages = 0; | ||
| 448 | struct extent_map *em; | ||
| 449 | struct address_space *mapping = inode->i_mapping; | ||
| 450 | struct pagevec pvec; | ||
| 451 | struct extent_map_tree *em_tree; | ||
| 452 | struct extent_io_tree *tree; | ||
| 453 | u64 end; | ||
| 454 | int misses = 0; | ||
| 455 | |||
| 456 | page = cb->orig_bio->bi_io_vec[cb->orig_bio->bi_vcnt - 1].bv_page; | ||
| 457 | last_offset = (page_offset(page) + PAGE_CACHE_SIZE); | ||
| 458 | em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 459 | tree = &BTRFS_I(inode)->io_tree; | ||
| 460 | |||
| 461 | if (isize == 0) | ||
| 462 | return 0; | ||
| 463 | |||
| 464 | end_index = (i_size_read(inode) - 1) >> PAGE_CACHE_SHIFT; | ||
| 465 | |||
| 466 | pagevec_init(&pvec, 0); | ||
| 467 | while (last_offset < compressed_end) { | ||
| 468 | page_index = last_offset >> PAGE_CACHE_SHIFT; | ||
| 469 | |||
| 470 | if (page_index > end_index) | ||
| 471 | break; | ||
| 472 | |||
| 473 | rcu_read_lock(); | ||
| 474 | page = radix_tree_lookup(&mapping->page_tree, page_index); | ||
| 475 | rcu_read_unlock(); | ||
| 476 | if (page) { | ||
| 477 | misses++; | ||
| 478 | if (misses > 4) | ||
| 479 | break; | ||
| 480 | goto next; | ||
| 481 | } | ||
| 482 | |||
| 483 | page = alloc_page(mapping_gfp_mask(mapping) | GFP_NOFS); | ||
| 484 | if (!page) | ||
| 485 | break; | ||
| 486 | |||
| 487 | page->index = page_index; | ||
| 488 | /* | ||
| 489 | * what we want to do here is call add_to_page_cache_lru, | ||
| 490 | * but that isn't exported, so we reproduce it here | ||
| 491 | */ | ||
| 492 | if (add_to_page_cache(page, mapping, | ||
| 493 | page->index, GFP_NOFS)) { | ||
| 494 | page_cache_release(page); | ||
| 495 | goto next; | ||
| 496 | } | ||
| 497 | |||
| 498 | /* open coding of lru_cache_add, also not exported */ | ||
| 499 | page_cache_get(page); | ||
| 500 | if (!pagevec_add(&pvec, page)) | ||
| 501 | __pagevec_lru_add_file(&pvec); | ||
| 502 | |||
| 503 | end = last_offset + PAGE_CACHE_SIZE - 1; | ||
| 504 | /* | ||
| 505 | * at this point, we have a locked page in the page cache | ||
| 506 | * for these bytes in the file. But, we have to make | ||
| 507 | * sure they map to this compressed extent on disk. | ||
| 508 | */ | ||
| 509 | set_page_extent_mapped(page); | ||
| 510 | lock_extent(tree, last_offset, end, GFP_NOFS); | ||
| 511 | spin_lock(&em_tree->lock); | ||
| 512 | em = lookup_extent_mapping(em_tree, last_offset, | ||
| 513 | PAGE_CACHE_SIZE); | ||
| 514 | spin_unlock(&em_tree->lock); | ||
| 515 | |||
| 516 | if (!em || last_offset < em->start || | ||
| 517 | (last_offset + PAGE_CACHE_SIZE > extent_map_end(em)) || | ||
| 518 | (em->block_start >> 9) != cb->orig_bio->bi_sector) { | ||
| 519 | free_extent_map(em); | ||
| 520 | unlock_extent(tree, last_offset, end, GFP_NOFS); | ||
| 521 | unlock_page(page); | ||
| 522 | page_cache_release(page); | ||
| 523 | break; | ||
| 524 | } | ||
| 525 | free_extent_map(em); | ||
| 526 | |||
| 527 | if (page->index == end_index) { | ||
| 528 | char *userpage; | ||
| 529 | size_t zero_offset = isize & (PAGE_CACHE_SIZE - 1); | ||
| 530 | |||
| 531 | if (zero_offset) { | ||
| 532 | int zeros; | ||
| 533 | zeros = PAGE_CACHE_SIZE - zero_offset; | ||
| 534 | userpage = kmap_atomic(page, KM_USER0); | ||
| 535 | memset(userpage + zero_offset, 0, zeros); | ||
| 536 | flush_dcache_page(page); | ||
| 537 | kunmap_atomic(userpage, KM_USER0); | ||
| 538 | } | ||
| 539 | } | ||
| 540 | |||
| 541 | ret = bio_add_page(cb->orig_bio, page, | ||
| 542 | PAGE_CACHE_SIZE, 0); | ||
| 543 | |||
| 544 | if (ret == PAGE_CACHE_SIZE) { | ||
| 545 | nr_pages++; | ||
| 546 | page_cache_release(page); | ||
| 547 | } else { | ||
| 548 | unlock_extent(tree, last_offset, end, GFP_NOFS); | ||
| 549 | unlock_page(page); | ||
| 550 | page_cache_release(page); | ||
| 551 | break; | ||
| 552 | } | ||
| 553 | next: | ||
| 554 | last_offset += PAGE_CACHE_SIZE; | ||
| 555 | } | ||
| 556 | if (pagevec_count(&pvec)) | ||
| 557 | __pagevec_lru_add_file(&pvec); | ||
| 558 | return 0; | ||
| 559 | } | ||
| 560 | |||
| 561 | /* | ||
| 562 | * for a compressed read, the bio we get passed has all the inode pages | ||
| 563 | * in it. We don't actually do IO on those pages but allocate new ones | ||
| 564 | * to hold the compressed pages on disk. | ||
| 565 | * | ||
| 566 | * bio->bi_sector points to the compressed extent on disk | ||
| 567 | * bio->bi_io_vec points to all of the inode pages | ||
| 568 | * bio->bi_vcnt is a count of pages | ||
| 569 | * | ||
| 570 | * After the compressed pages are read, we copy the bytes into the | ||
| 571 | * bio we were passed and then call the bio end_io calls | ||
| 572 | */ | ||
| 573 | int btrfs_submit_compressed_read(struct inode *inode, struct bio *bio, | ||
| 574 | int mirror_num, unsigned long bio_flags) | ||
| 575 | { | ||
| 576 | struct extent_io_tree *tree; | ||
| 577 | struct extent_map_tree *em_tree; | ||
| 578 | struct compressed_bio *cb; | ||
| 579 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 580 | unsigned long uncompressed_len = bio->bi_vcnt * PAGE_CACHE_SIZE; | ||
| 581 | unsigned long compressed_len; | ||
| 582 | unsigned long nr_pages; | ||
| 583 | unsigned long page_index; | ||
| 584 | struct page *page; | ||
| 585 | struct block_device *bdev; | ||
| 586 | struct bio *comp_bio; | ||
| 587 | u64 cur_disk_byte = (u64)bio->bi_sector << 9; | ||
| 588 | u64 em_len; | ||
| 589 | u64 em_start; | ||
| 590 | struct extent_map *em; | ||
| 591 | int ret; | ||
| 592 | u32 *sums; | ||
| 593 | |||
| 594 | tree = &BTRFS_I(inode)->io_tree; | ||
| 595 | em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 596 | |||
| 597 | /* we need the actual starting offset of this extent in the file */ | ||
| 598 | spin_lock(&em_tree->lock); | ||
| 599 | em = lookup_extent_mapping(em_tree, | ||
| 600 | page_offset(bio->bi_io_vec->bv_page), | ||
| 601 | PAGE_CACHE_SIZE); | ||
| 602 | spin_unlock(&em_tree->lock); | ||
| 603 | |||
| 604 | compressed_len = em->block_len; | ||
| 605 | cb = kmalloc(compressed_bio_size(root, compressed_len), GFP_NOFS); | ||
| 606 | atomic_set(&cb->pending_bios, 0); | ||
| 607 | cb->errors = 0; | ||
| 608 | cb->inode = inode; | ||
| 609 | cb->mirror_num = mirror_num; | ||
| 610 | sums = &cb->sums; | ||
| 611 | |||
| 612 | cb->start = em->orig_start; | ||
| 613 | em_len = em->len; | ||
| 614 | em_start = em->start; | ||
| 615 | |||
| 616 | free_extent_map(em); | ||
| 617 | em = NULL; | ||
| 618 | |||
| 619 | cb->len = uncompressed_len; | ||
| 620 | cb->compressed_len = compressed_len; | ||
| 621 | cb->orig_bio = bio; | ||
| 622 | |||
| 623 | nr_pages = (compressed_len + PAGE_CACHE_SIZE - 1) / | ||
| 624 | PAGE_CACHE_SIZE; | ||
| 625 | cb->compressed_pages = kmalloc(sizeof(struct page *) * nr_pages, | ||
| 626 | GFP_NOFS); | ||
| 627 | bdev = BTRFS_I(inode)->root->fs_info->fs_devices->latest_bdev; | ||
| 628 | |||
| 629 | for (page_index = 0; page_index < nr_pages; page_index++) { | ||
| 630 | cb->compressed_pages[page_index] = alloc_page(GFP_NOFS | | ||
| 631 | __GFP_HIGHMEM); | ||
| 632 | } | ||
| 633 | cb->nr_pages = nr_pages; | ||
| 634 | |||
| 635 | add_ra_bio_pages(inode, em_start + em_len, cb); | ||
| 636 | |||
| 637 | /* include any pages we added in add_ra-bio_pages */ | ||
| 638 | uncompressed_len = bio->bi_vcnt * PAGE_CACHE_SIZE; | ||
| 639 | cb->len = uncompressed_len; | ||
| 640 | |||
| 641 | comp_bio = compressed_bio_alloc(bdev, cur_disk_byte, GFP_NOFS); | ||
| 642 | comp_bio->bi_private = cb; | ||
| 643 | comp_bio->bi_end_io = end_compressed_bio_read; | ||
| 644 | atomic_inc(&cb->pending_bios); | ||
| 645 | |||
| 646 | for (page_index = 0; page_index < nr_pages; page_index++) { | ||
| 647 | page = cb->compressed_pages[page_index]; | ||
| 648 | page->mapping = inode->i_mapping; | ||
| 649 | page->index = em_start >> PAGE_CACHE_SHIFT; | ||
| 650 | |||
| 651 | if (comp_bio->bi_size) | ||
| 652 | ret = tree->ops->merge_bio_hook(page, 0, | ||
| 653 | PAGE_CACHE_SIZE, | ||
| 654 | comp_bio, 0); | ||
| 655 | else | ||
| 656 | ret = 0; | ||
| 657 | |||
| 658 | page->mapping = NULL; | ||
| 659 | if (ret || bio_add_page(comp_bio, page, PAGE_CACHE_SIZE, 0) < | ||
| 660 | PAGE_CACHE_SIZE) { | ||
| 661 | bio_get(comp_bio); | ||
| 662 | |||
| 663 | ret = btrfs_bio_wq_end_io(root->fs_info, comp_bio, 0); | ||
| 664 | BUG_ON(ret); | ||
| 665 | |||
| 666 | /* | ||
| 667 | * inc the count before we submit the bio so | ||
| 668 | * we know the end IO handler won't happen before | ||
| 669 | * we inc the count. Otherwise, the cb might get | ||
| 670 | * freed before we're done setting it up | ||
| 671 | */ | ||
| 672 | atomic_inc(&cb->pending_bios); | ||
| 673 | |||
| 674 | if (!btrfs_test_flag(inode, NODATASUM)) { | ||
| 675 | btrfs_lookup_bio_sums(root, inode, comp_bio, | ||
| 676 | sums); | ||
| 677 | } | ||
| 678 | sums += (comp_bio->bi_size + root->sectorsize - 1) / | ||
| 679 | root->sectorsize; | ||
| 680 | |||
| 681 | ret = btrfs_map_bio(root, READ, comp_bio, | ||
| 682 | mirror_num, 0); | ||
| 683 | BUG_ON(ret); | ||
| 684 | |||
| 685 | bio_put(comp_bio); | ||
| 686 | |||
| 687 | comp_bio = compressed_bio_alloc(bdev, cur_disk_byte, | ||
| 688 | GFP_NOFS); | ||
| 689 | comp_bio->bi_private = cb; | ||
| 690 | comp_bio->bi_end_io = end_compressed_bio_read; | ||
| 691 | |||
| 692 | bio_add_page(comp_bio, page, PAGE_CACHE_SIZE, 0); | ||
| 693 | } | ||
| 694 | cur_disk_byte += PAGE_CACHE_SIZE; | ||
| 695 | } | ||
| 696 | bio_get(comp_bio); | ||
| 697 | |||
| 698 | ret = btrfs_bio_wq_end_io(root->fs_info, comp_bio, 0); | ||
| 699 | BUG_ON(ret); | ||
| 700 | |||
| 701 | if (!btrfs_test_flag(inode, NODATASUM)) | ||
| 702 | btrfs_lookup_bio_sums(root, inode, comp_bio, sums); | ||
| 703 | |||
| 704 | ret = btrfs_map_bio(root, READ, comp_bio, mirror_num, 0); | ||
| 705 | BUG_ON(ret); | ||
| 706 | |||
| 707 | bio_put(comp_bio); | ||
| 708 | return 0; | ||
| 709 | } | ||
diff --git a/fs/btrfs/compression.h b/fs/btrfs/compression.h new file mode 100644 index 000000000000..421f5b4aa715 --- /dev/null +++ b/fs/btrfs/compression.h | |||
| @@ -0,0 +1,47 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | |||
| 19 | #ifndef __BTRFS_COMPRESSION_ | ||
| 20 | #define __BTRFS_COMPRESSION_ | ||
| 21 | |||
| 22 | int btrfs_zlib_decompress(unsigned char *data_in, | ||
| 23 | struct page *dest_page, | ||
| 24 | unsigned long start_byte, | ||
| 25 | size_t srclen, size_t destlen); | ||
| 26 | int btrfs_zlib_compress_pages(struct address_space *mapping, | ||
| 27 | u64 start, unsigned long len, | ||
| 28 | struct page **pages, | ||
| 29 | unsigned long nr_dest_pages, | ||
| 30 | unsigned long *out_pages, | ||
| 31 | unsigned long *total_in, | ||
| 32 | unsigned long *total_out, | ||
| 33 | unsigned long max_out); | ||
| 34 | int btrfs_zlib_decompress_biovec(struct page **pages_in, | ||
| 35 | u64 disk_start, | ||
| 36 | struct bio_vec *bvec, | ||
| 37 | int vcnt, | ||
| 38 | size_t srclen); | ||
| 39 | void btrfs_zlib_exit(void); | ||
| 40 | int btrfs_submit_compressed_write(struct inode *inode, u64 start, | ||
| 41 | unsigned long len, u64 disk_start, | ||
| 42 | unsigned long compressed_len, | ||
| 43 | struct page **compressed_pages, | ||
| 44 | unsigned long nr_pages); | ||
| 45 | int btrfs_submit_compressed_read(struct inode *inode, struct bio *bio, | ||
| 46 | int mirror_num, unsigned long bio_flags); | ||
| 47 | #endif | ||
diff --git a/fs/btrfs/crc32c.h b/fs/btrfs/crc32c.h new file mode 100644 index 000000000000..6e1b3de36700 --- /dev/null +++ b/fs/btrfs/crc32c.h | |||
| @@ -0,0 +1,29 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | |||
| 19 | #ifndef __BTRFS_CRC32C__ | ||
| 20 | #define __BTRFS_CRC32C__ | ||
| 21 | #include <linux/crc32c.h> | ||
| 22 | |||
| 23 | /* | ||
| 24 | * this file used to do more for selecting the HW version of crc32c, | ||
| 25 | * perhaps it will one day again soon. | ||
| 26 | */ | ||
| 27 | #define btrfs_crc32c(seed, data, length) crc32c(seed, data, length) | ||
| 28 | #endif | ||
| 29 | |||
diff --git a/fs/btrfs/ctree.c b/fs/btrfs/ctree.c new file mode 100644 index 000000000000..9e46c0776816 --- /dev/null +++ b/fs/btrfs/ctree.c | |||
| @@ -0,0 +1,3953 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2007,2008 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 | |||
| 19 | #include <linux/sched.h> | ||
| 20 | #include "ctree.h" | ||
| 21 | #include "disk-io.h" | ||
| 22 | #include "transaction.h" | ||
| 23 | #include "print-tree.h" | ||
| 24 | #include "locking.h" | ||
| 25 | |||
| 26 | static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 27 | *root, struct btrfs_path *path, int level); | ||
| 28 | static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 29 | *root, struct btrfs_key *ins_key, | ||
| 30 | struct btrfs_path *path, int data_size, int extend); | ||
| 31 | static int push_node_left(struct btrfs_trans_handle *trans, | ||
| 32 | struct btrfs_root *root, struct extent_buffer *dst, | ||
| 33 | struct extent_buffer *src, int empty); | ||
| 34 | static int balance_node_right(struct btrfs_trans_handle *trans, | ||
| 35 | struct btrfs_root *root, | ||
| 36 | struct extent_buffer *dst_buf, | ||
| 37 | struct extent_buffer *src_buf); | ||
| 38 | static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 39 | struct btrfs_path *path, int level, int slot); | ||
| 40 | |||
| 41 | inline void btrfs_init_path(struct btrfs_path *p) | ||
| 42 | { | ||
| 43 | memset(p, 0, sizeof(*p)); | ||
| 44 | } | ||
| 45 | |||
| 46 | struct btrfs_path *btrfs_alloc_path(void) | ||
| 47 | { | ||
| 48 | struct btrfs_path *path; | ||
| 49 | path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS); | ||
| 50 | if (path) { | ||
| 51 | btrfs_init_path(path); | ||
| 52 | path->reada = 1; | ||
| 53 | } | ||
| 54 | return path; | ||
| 55 | } | ||
| 56 | |||
| 57 | /* this also releases the path */ | ||
| 58 | void btrfs_free_path(struct btrfs_path *p) | ||
| 59 | { | ||
| 60 | btrfs_release_path(NULL, p); | ||
| 61 | kmem_cache_free(btrfs_path_cachep, p); | ||
| 62 | } | ||
| 63 | |||
| 64 | /* | ||
| 65 | * path release drops references on the extent buffers in the path | ||
| 66 | * and it drops any locks held by this path | ||
| 67 | * | ||
| 68 | * It is safe to call this on paths that no locks or extent buffers held. | ||
| 69 | */ | ||
| 70 | noinline void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p) | ||
| 71 | { | ||
| 72 | int i; | ||
| 73 | |||
| 74 | for (i = 0; i < BTRFS_MAX_LEVEL; i++) { | ||
| 75 | p->slots[i] = 0; | ||
| 76 | if (!p->nodes[i]) | ||
| 77 | continue; | ||
| 78 | if (p->locks[i]) { | ||
| 79 | btrfs_tree_unlock(p->nodes[i]); | ||
| 80 | p->locks[i] = 0; | ||
| 81 | } | ||
| 82 | free_extent_buffer(p->nodes[i]); | ||
| 83 | p->nodes[i] = NULL; | ||
| 84 | } | ||
| 85 | } | ||
| 86 | |||
| 87 | /* | ||
| 88 | * safely gets a reference on the root node of a tree. A lock | ||
| 89 | * is not taken, so a concurrent writer may put a different node | ||
| 90 | * at the root of the tree. See btrfs_lock_root_node for the | ||
| 91 | * looping required. | ||
| 92 | * | ||
| 93 | * The extent buffer returned by this has a reference taken, so | ||
| 94 | * it won't disappear. It may stop being the root of the tree | ||
| 95 | * at any time because there are no locks held. | ||
| 96 | */ | ||
| 97 | struct extent_buffer *btrfs_root_node(struct btrfs_root *root) | ||
| 98 | { | ||
| 99 | struct extent_buffer *eb; | ||
| 100 | spin_lock(&root->node_lock); | ||
| 101 | eb = root->node; | ||
| 102 | extent_buffer_get(eb); | ||
| 103 | spin_unlock(&root->node_lock); | ||
| 104 | return eb; | ||
| 105 | } | ||
| 106 | |||
| 107 | /* loop around taking references on and locking the root node of the | ||
| 108 | * tree until you end up with a lock on the root. A locked buffer | ||
| 109 | * is returned, with a reference held. | ||
| 110 | */ | ||
| 111 | struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root) | ||
| 112 | { | ||
| 113 | struct extent_buffer *eb; | ||
| 114 | |||
| 115 | while (1) { | ||
| 116 | eb = btrfs_root_node(root); | ||
| 117 | btrfs_tree_lock(eb); | ||
| 118 | |||
| 119 | spin_lock(&root->node_lock); | ||
| 120 | if (eb == root->node) { | ||
| 121 | spin_unlock(&root->node_lock); | ||
| 122 | break; | ||
| 123 | } | ||
| 124 | spin_unlock(&root->node_lock); | ||
| 125 | |||
| 126 | btrfs_tree_unlock(eb); | ||
| 127 | free_extent_buffer(eb); | ||
| 128 | } | ||
| 129 | return eb; | ||
| 130 | } | ||
| 131 | |||
| 132 | /* cowonly root (everything not a reference counted cow subvolume), just get | ||
| 133 | * put onto a simple dirty list. transaction.c walks this to make sure they | ||
| 134 | * get properly updated on disk. | ||
| 135 | */ | ||
| 136 | static void add_root_to_dirty_list(struct btrfs_root *root) | ||
| 137 | { | ||
| 138 | if (root->track_dirty && list_empty(&root->dirty_list)) { | ||
| 139 | list_add(&root->dirty_list, | ||
| 140 | &root->fs_info->dirty_cowonly_roots); | ||
| 141 | } | ||
| 142 | } | ||
| 143 | |||
| 144 | /* | ||
| 145 | * used by snapshot creation to make a copy of a root for a tree with | ||
| 146 | * a given objectid. The buffer with the new root node is returned in | ||
| 147 | * cow_ret, and this func returns zero on success or a negative error code. | ||
| 148 | */ | ||
| 149 | int btrfs_copy_root(struct btrfs_trans_handle *trans, | ||
| 150 | struct btrfs_root *root, | ||
| 151 | struct extent_buffer *buf, | ||
| 152 | struct extent_buffer **cow_ret, u64 new_root_objectid) | ||
| 153 | { | ||
| 154 | struct extent_buffer *cow; | ||
| 155 | u32 nritems; | ||
| 156 | int ret = 0; | ||
| 157 | int level; | ||
| 158 | struct btrfs_root *new_root; | ||
| 159 | |||
| 160 | new_root = kmalloc(sizeof(*new_root), GFP_NOFS); | ||
| 161 | if (!new_root) | ||
| 162 | return -ENOMEM; | ||
| 163 | |||
| 164 | memcpy(new_root, root, sizeof(*new_root)); | ||
| 165 | new_root->root_key.objectid = new_root_objectid; | ||
| 166 | |||
| 167 | WARN_ON(root->ref_cows && trans->transid != | ||
| 168 | root->fs_info->running_transaction->transid); | ||
| 169 | WARN_ON(root->ref_cows && trans->transid != root->last_trans); | ||
| 170 | |||
| 171 | level = btrfs_header_level(buf); | ||
| 172 | nritems = btrfs_header_nritems(buf); | ||
| 173 | |||
| 174 | cow = btrfs_alloc_free_block(trans, new_root, buf->len, 0, | ||
| 175 | new_root_objectid, trans->transid, | ||
| 176 | level, buf->start, 0); | ||
| 177 | if (IS_ERR(cow)) { | ||
| 178 | kfree(new_root); | ||
| 179 | return PTR_ERR(cow); | ||
| 180 | } | ||
| 181 | |||
| 182 | copy_extent_buffer(cow, buf, 0, 0, cow->len); | ||
| 183 | btrfs_set_header_bytenr(cow, cow->start); | ||
| 184 | btrfs_set_header_generation(cow, trans->transid); | ||
| 185 | btrfs_set_header_owner(cow, new_root_objectid); | ||
| 186 | btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); | ||
| 187 | |||
| 188 | write_extent_buffer(cow, root->fs_info->fsid, | ||
| 189 | (unsigned long)btrfs_header_fsid(cow), | ||
| 190 | BTRFS_FSID_SIZE); | ||
| 191 | |||
| 192 | WARN_ON(btrfs_header_generation(buf) > trans->transid); | ||
| 193 | ret = btrfs_inc_ref(trans, new_root, buf, cow, NULL); | ||
| 194 | kfree(new_root); | ||
| 195 | |||
| 196 | if (ret) | ||
| 197 | return ret; | ||
| 198 | |||
| 199 | btrfs_mark_buffer_dirty(cow); | ||
| 200 | *cow_ret = cow; | ||
| 201 | return 0; | ||
| 202 | } | ||
| 203 | |||
| 204 | /* | ||
| 205 | * does the dirty work in cow of a single block. The parent block (if | ||
| 206 | * supplied) is updated to point to the new cow copy. The new buffer is marked | ||
| 207 | * dirty and returned locked. If you modify the block it needs to be marked | ||
| 208 | * dirty again. | ||
| 209 | * | ||
| 210 | * search_start -- an allocation hint for the new block | ||
| 211 | * | ||
| 212 | * empty_size -- a hint that you plan on doing more cow. This is the size in | ||
| 213 | * bytes the allocator should try to find free next to the block it returns. | ||
| 214 | * This is just a hint and may be ignored by the allocator. | ||
| 215 | * | ||
| 216 | * prealloc_dest -- if you have already reserved a destination for the cow, | ||
| 217 | * this uses that block instead of allocating a new one. | ||
| 218 | * btrfs_alloc_reserved_extent is used to finish the allocation. | ||
| 219 | */ | ||
| 220 | static noinline int __btrfs_cow_block(struct btrfs_trans_handle *trans, | ||
| 221 | struct btrfs_root *root, | ||
| 222 | struct extent_buffer *buf, | ||
| 223 | struct extent_buffer *parent, int parent_slot, | ||
| 224 | struct extent_buffer **cow_ret, | ||
| 225 | u64 search_start, u64 empty_size, | ||
| 226 | u64 prealloc_dest) | ||
| 227 | { | ||
| 228 | u64 parent_start; | ||
| 229 | struct extent_buffer *cow; | ||
| 230 | u32 nritems; | ||
| 231 | int ret = 0; | ||
| 232 | int level; | ||
| 233 | int unlock_orig = 0; | ||
| 234 | |||
| 235 | if (*cow_ret == buf) | ||
| 236 | unlock_orig = 1; | ||
| 237 | |||
| 238 | WARN_ON(!btrfs_tree_locked(buf)); | ||
| 239 | |||
| 240 | if (parent) | ||
| 241 | parent_start = parent->start; | ||
| 242 | else | ||
| 243 | parent_start = 0; | ||
| 244 | |||
| 245 | WARN_ON(root->ref_cows && trans->transid != | ||
| 246 | root->fs_info->running_transaction->transid); | ||
| 247 | WARN_ON(root->ref_cows && trans->transid != root->last_trans); | ||
| 248 | |||
| 249 | level = btrfs_header_level(buf); | ||
| 250 | nritems = btrfs_header_nritems(buf); | ||
| 251 | |||
| 252 | if (prealloc_dest) { | ||
| 253 | struct btrfs_key ins; | ||
| 254 | |||
| 255 | ins.objectid = prealloc_dest; | ||
| 256 | ins.offset = buf->len; | ||
| 257 | ins.type = BTRFS_EXTENT_ITEM_KEY; | ||
| 258 | |||
| 259 | ret = btrfs_alloc_reserved_extent(trans, root, parent_start, | ||
| 260 | root->root_key.objectid, | ||
| 261 | trans->transid, level, &ins); | ||
| 262 | BUG_ON(ret); | ||
| 263 | cow = btrfs_init_new_buffer(trans, root, prealloc_dest, | ||
| 264 | buf->len); | ||
| 265 | } else { | ||
| 266 | cow = btrfs_alloc_free_block(trans, root, buf->len, | ||
| 267 | parent_start, | ||
| 268 | root->root_key.objectid, | ||
| 269 | trans->transid, level, | ||
| 270 | search_start, empty_size); | ||
| 271 | } | ||
| 272 | if (IS_ERR(cow)) | ||
| 273 | return PTR_ERR(cow); | ||
| 274 | |||
| 275 | copy_extent_buffer(cow, buf, 0, 0, cow->len); | ||
| 276 | btrfs_set_header_bytenr(cow, cow->start); | ||
| 277 | btrfs_set_header_generation(cow, trans->transid); | ||
| 278 | btrfs_set_header_owner(cow, root->root_key.objectid); | ||
| 279 | btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); | ||
| 280 | |||
| 281 | write_extent_buffer(cow, root->fs_info->fsid, | ||
| 282 | (unsigned long)btrfs_header_fsid(cow), | ||
| 283 | BTRFS_FSID_SIZE); | ||
| 284 | |||
| 285 | WARN_ON(btrfs_header_generation(buf) > trans->transid); | ||
| 286 | if (btrfs_header_generation(buf) != trans->transid) { | ||
| 287 | u32 nr_extents; | ||
| 288 | ret = btrfs_inc_ref(trans, root, buf, cow, &nr_extents); | ||
| 289 | if (ret) | ||
| 290 | return ret; | ||
| 291 | |||
| 292 | ret = btrfs_cache_ref(trans, root, buf, nr_extents); | ||
| 293 | WARN_ON(ret); | ||
| 294 | } else if (btrfs_header_owner(buf) == BTRFS_TREE_RELOC_OBJECTID) { | ||
| 295 | /* | ||
| 296 | * There are only two places that can drop reference to | ||
| 297 | * tree blocks owned by living reloc trees, one is here, | ||
| 298 | * the other place is btrfs_drop_subtree. In both places, | ||
| 299 | * we check reference count while tree block is locked. | ||
| 300 | * Furthermore, if reference count is one, it won't get | ||
| 301 | * increased by someone else. | ||
| 302 | */ | ||
| 303 | u32 refs; | ||
| 304 | ret = btrfs_lookup_extent_ref(trans, root, buf->start, | ||
| 305 | buf->len, &refs); | ||
| 306 | BUG_ON(ret); | ||
| 307 | if (refs == 1) { | ||
| 308 | ret = btrfs_update_ref(trans, root, buf, cow, | ||
| 309 | 0, nritems); | ||
| 310 | clean_tree_block(trans, root, buf); | ||
| 311 | } else { | ||
| 312 | ret = btrfs_inc_ref(trans, root, buf, cow, NULL); | ||
| 313 | } | ||
| 314 | BUG_ON(ret); | ||
| 315 | } else { | ||
| 316 | ret = btrfs_update_ref(trans, root, buf, cow, 0, nritems); | ||
| 317 | if (ret) | ||
| 318 | return ret; | ||
| 319 | clean_tree_block(trans, root, buf); | ||
| 320 | } | ||
| 321 | |||
| 322 | if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) { | ||
| 323 | ret = btrfs_reloc_tree_cache_ref(trans, root, cow, buf->start); | ||
| 324 | WARN_ON(ret); | ||
| 325 | } | ||
| 326 | |||
| 327 | if (buf == root->node) { | ||
| 328 | WARN_ON(parent && parent != buf); | ||
| 329 | |||
| 330 | spin_lock(&root->node_lock); | ||
| 331 | root->node = cow; | ||
| 332 | extent_buffer_get(cow); | ||
| 333 | spin_unlock(&root->node_lock); | ||
| 334 | |||
| 335 | if (buf != root->commit_root) { | ||
| 336 | btrfs_free_extent(trans, root, buf->start, | ||
| 337 | buf->len, buf->start, | ||
| 338 | root->root_key.objectid, | ||
| 339 | btrfs_header_generation(buf), | ||
| 340 | level, 1); | ||
| 341 | } | ||
| 342 | free_extent_buffer(buf); | ||
| 343 | add_root_to_dirty_list(root); | ||
| 344 | } else { | ||
| 345 | btrfs_set_node_blockptr(parent, parent_slot, | ||
| 346 | cow->start); | ||
| 347 | WARN_ON(trans->transid == 0); | ||
| 348 | btrfs_set_node_ptr_generation(parent, parent_slot, | ||
| 349 | trans->transid); | ||
| 350 | btrfs_mark_buffer_dirty(parent); | ||
| 351 | WARN_ON(btrfs_header_generation(parent) != trans->transid); | ||
| 352 | btrfs_free_extent(trans, root, buf->start, buf->len, | ||
| 353 | parent_start, btrfs_header_owner(parent), | ||
| 354 | btrfs_header_generation(parent), level, 1); | ||
| 355 | } | ||
| 356 | if (unlock_orig) | ||
| 357 | btrfs_tree_unlock(buf); | ||
| 358 | free_extent_buffer(buf); | ||
| 359 | btrfs_mark_buffer_dirty(cow); | ||
| 360 | *cow_ret = cow; | ||
| 361 | return 0; | ||
| 362 | } | ||
| 363 | |||
| 364 | /* | ||
| 365 | * cows a single block, see __btrfs_cow_block for the real work. | ||
| 366 | * This version of it has extra checks so that a block isn't cow'd more than | ||
| 367 | * once per transaction, as long as it hasn't been written yet | ||
| 368 | */ | ||
| 369 | noinline int btrfs_cow_block(struct btrfs_trans_handle *trans, | ||
| 370 | struct btrfs_root *root, struct extent_buffer *buf, | ||
| 371 | struct extent_buffer *parent, int parent_slot, | ||
| 372 | struct extent_buffer **cow_ret, u64 prealloc_dest) | ||
| 373 | { | ||
| 374 | u64 search_start; | ||
| 375 | int ret; | ||
| 376 | |||
| 377 | if (trans->transaction != root->fs_info->running_transaction) { | ||
| 378 | printk(KERN_CRIT "trans %llu running %llu\n", | ||
| 379 | (unsigned long long)trans->transid, | ||
| 380 | (unsigned long long) | ||
| 381 | root->fs_info->running_transaction->transid); | ||
| 382 | WARN_ON(1); | ||
| 383 | } | ||
| 384 | if (trans->transid != root->fs_info->generation) { | ||
| 385 | printk(KERN_CRIT "trans %llu running %llu\n", | ||
| 386 | (unsigned long long)trans->transid, | ||
| 387 | (unsigned long long)root->fs_info->generation); | ||
| 388 | WARN_ON(1); | ||
| 389 | } | ||
| 390 | |||
| 391 | spin_lock(&root->fs_info->hash_lock); | ||
| 392 | if (btrfs_header_generation(buf) == trans->transid && | ||
| 393 | btrfs_header_owner(buf) == root->root_key.objectid && | ||
| 394 | !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) { | ||
| 395 | *cow_ret = buf; | ||
| 396 | spin_unlock(&root->fs_info->hash_lock); | ||
| 397 | WARN_ON(prealloc_dest); | ||
| 398 | return 0; | ||
| 399 | } | ||
| 400 | spin_unlock(&root->fs_info->hash_lock); | ||
| 401 | search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1); | ||
| 402 | ret = __btrfs_cow_block(trans, root, buf, parent, | ||
| 403 | parent_slot, cow_ret, search_start, 0, | ||
| 404 | prealloc_dest); | ||
| 405 | return ret; | ||
| 406 | } | ||
| 407 | |||
| 408 | /* | ||
| 409 | * helper function for defrag to decide if two blocks pointed to by a | ||
| 410 | * node are actually close by | ||
| 411 | */ | ||
| 412 | static int close_blocks(u64 blocknr, u64 other, u32 blocksize) | ||
| 413 | { | ||
| 414 | if (blocknr < other && other - (blocknr + blocksize) < 32768) | ||
| 415 | return 1; | ||
| 416 | if (blocknr > other && blocknr - (other + blocksize) < 32768) | ||
| 417 | return 1; | ||
| 418 | return 0; | ||
| 419 | } | ||
| 420 | |||
| 421 | /* | ||
| 422 | * compare two keys in a memcmp fashion | ||
| 423 | */ | ||
| 424 | static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) | ||
| 425 | { | ||
| 426 | struct btrfs_key k1; | ||
| 427 | |||
| 428 | btrfs_disk_key_to_cpu(&k1, disk); | ||
| 429 | |||
| 430 | if (k1.objectid > k2->objectid) | ||
| 431 | return 1; | ||
| 432 | if (k1.objectid < k2->objectid) | ||
| 433 | return -1; | ||
| 434 | if (k1.type > k2->type) | ||
| 435 | return 1; | ||
| 436 | if (k1.type < k2->type) | ||
| 437 | return -1; | ||
| 438 | if (k1.offset > k2->offset) | ||
| 439 | return 1; | ||
| 440 | if (k1.offset < k2->offset) | ||
| 441 | return -1; | ||
| 442 | return 0; | ||
| 443 | } | ||
| 444 | |||
| 445 | /* | ||
| 446 | * same as comp_keys only with two btrfs_key's | ||
| 447 | */ | ||
| 448 | static int comp_cpu_keys(struct btrfs_key *k1, struct btrfs_key *k2) | ||
| 449 | { | ||
| 450 | if (k1->objectid > k2->objectid) | ||
| 451 | return 1; | ||
| 452 | if (k1->objectid < k2->objectid) | ||
| 453 | return -1; | ||
| 454 | if (k1->type > k2->type) | ||
| 455 | return 1; | ||
| 456 | if (k1->type < k2->type) | ||
| 457 | return -1; | ||
| 458 | if (k1->offset > k2->offset) | ||
| 459 | return 1; | ||
| 460 | if (k1->offset < k2->offset) | ||
| 461 | return -1; | ||
| 462 | return 0; | ||
| 463 | } | ||
| 464 | |||
| 465 | /* | ||
| 466 | * this is used by the defrag code to go through all the | ||
| 467 | * leaves pointed to by a node and reallocate them so that | ||
| 468 | * disk order is close to key order | ||
| 469 | */ | ||
| 470 | int btrfs_realloc_node(struct btrfs_trans_handle *trans, | ||
| 471 | struct btrfs_root *root, struct extent_buffer *parent, | ||
| 472 | int start_slot, int cache_only, u64 *last_ret, | ||
| 473 | struct btrfs_key *progress) | ||
| 474 | { | ||
| 475 | struct extent_buffer *cur; | ||
| 476 | u64 blocknr; | ||
| 477 | u64 gen; | ||
| 478 | u64 search_start = *last_ret; | ||
| 479 | u64 last_block = 0; | ||
| 480 | u64 other; | ||
| 481 | u32 parent_nritems; | ||
| 482 | int end_slot; | ||
| 483 | int i; | ||
| 484 | int err = 0; | ||
| 485 | int parent_level; | ||
| 486 | int uptodate; | ||
| 487 | u32 blocksize; | ||
| 488 | int progress_passed = 0; | ||
| 489 | struct btrfs_disk_key disk_key; | ||
| 490 | |||
| 491 | parent_level = btrfs_header_level(parent); | ||
| 492 | if (cache_only && parent_level != 1) | ||
| 493 | return 0; | ||
| 494 | |||
| 495 | if (trans->transaction != root->fs_info->running_transaction) | ||
| 496 | WARN_ON(1); | ||
| 497 | if (trans->transid != root->fs_info->generation) | ||
| 498 | WARN_ON(1); | ||
| 499 | |||
| 500 | parent_nritems = btrfs_header_nritems(parent); | ||
| 501 | blocksize = btrfs_level_size(root, parent_level - 1); | ||
| 502 | end_slot = parent_nritems; | ||
| 503 | |||
| 504 | if (parent_nritems == 1) | ||
| 505 | return 0; | ||
| 506 | |||
| 507 | for (i = start_slot; i < end_slot; i++) { | ||
| 508 | int close = 1; | ||
| 509 | |||
| 510 | if (!parent->map_token) { | ||
| 511 | map_extent_buffer(parent, | ||
| 512 | btrfs_node_key_ptr_offset(i), | ||
| 513 | sizeof(struct btrfs_key_ptr), | ||
| 514 | &parent->map_token, &parent->kaddr, | ||
| 515 | &parent->map_start, &parent->map_len, | ||
| 516 | KM_USER1); | ||
| 517 | } | ||
| 518 | btrfs_node_key(parent, &disk_key, i); | ||
| 519 | if (!progress_passed && comp_keys(&disk_key, progress) < 0) | ||
| 520 | continue; | ||
| 521 | |||
| 522 | progress_passed = 1; | ||
| 523 | blocknr = btrfs_node_blockptr(parent, i); | ||
| 524 | gen = btrfs_node_ptr_generation(parent, i); | ||
| 525 | if (last_block == 0) | ||
| 526 | last_block = blocknr; | ||
| 527 | |||
| 528 | if (i > 0) { | ||
| 529 | other = btrfs_node_blockptr(parent, i - 1); | ||
| 530 | close = close_blocks(blocknr, other, blocksize); | ||
| 531 | } | ||
| 532 | if (!close && i < end_slot - 2) { | ||
| 533 | other = btrfs_node_blockptr(parent, i + 1); | ||
| 534 | close = close_blocks(blocknr, other, blocksize); | ||
| 535 | } | ||
| 536 | if (close) { | ||
| 537 | last_block = blocknr; | ||
| 538 | continue; | ||
| 539 | } | ||
| 540 | if (parent->map_token) { | ||
| 541 | unmap_extent_buffer(parent, parent->map_token, | ||
| 542 | KM_USER1); | ||
| 543 | parent->map_token = NULL; | ||
| 544 | } | ||
| 545 | |||
| 546 | cur = btrfs_find_tree_block(root, blocknr, blocksize); | ||
| 547 | if (cur) | ||
| 548 | uptodate = btrfs_buffer_uptodate(cur, gen); | ||
| 549 | else | ||
| 550 | uptodate = 0; | ||
| 551 | if (!cur || !uptodate) { | ||
| 552 | if (cache_only) { | ||
| 553 | free_extent_buffer(cur); | ||
| 554 | continue; | ||
| 555 | } | ||
| 556 | if (!cur) { | ||
| 557 | cur = read_tree_block(root, blocknr, | ||
| 558 | blocksize, gen); | ||
| 559 | } else if (!uptodate) { | ||
| 560 | btrfs_read_buffer(cur, gen); | ||
| 561 | } | ||
| 562 | } | ||
| 563 | if (search_start == 0) | ||
| 564 | search_start = last_block; | ||
| 565 | |||
| 566 | btrfs_tree_lock(cur); | ||
| 567 | err = __btrfs_cow_block(trans, root, cur, parent, i, | ||
| 568 | &cur, search_start, | ||
| 569 | min(16 * blocksize, | ||
| 570 | (end_slot - i) * blocksize), 0); | ||
| 571 | if (err) { | ||
| 572 | btrfs_tree_unlock(cur); | ||
| 573 | free_extent_buffer(cur); | ||
| 574 | break; | ||
| 575 | } | ||
| 576 | search_start = cur->start; | ||
| 577 | last_block = cur->start; | ||
| 578 | *last_ret = search_start; | ||
| 579 | btrfs_tree_unlock(cur); | ||
| 580 | free_extent_buffer(cur); | ||
| 581 | } | ||
| 582 | if (parent->map_token) { | ||
| 583 | unmap_extent_buffer(parent, parent->map_token, | ||
| 584 | KM_USER1); | ||
| 585 | parent->map_token = NULL; | ||
| 586 | } | ||
| 587 | return err; | ||
| 588 | } | ||
| 589 | |||
| 590 | /* | ||
| 591 | * The leaf data grows from end-to-front in the node. | ||
| 592 | * this returns the address of the start of the last item, | ||
| 593 | * which is the stop of the leaf data stack | ||
| 594 | */ | ||
| 595 | static inline unsigned int leaf_data_end(struct btrfs_root *root, | ||
| 596 | struct extent_buffer *leaf) | ||
| 597 | { | ||
| 598 | u32 nr = btrfs_header_nritems(leaf); | ||
| 599 | if (nr == 0) | ||
| 600 | return BTRFS_LEAF_DATA_SIZE(root); | ||
| 601 | return btrfs_item_offset_nr(leaf, nr - 1); | ||
| 602 | } | ||
| 603 | |||
| 604 | /* | ||
| 605 | * extra debugging checks to make sure all the items in a key are | ||
| 606 | * well formed and in the proper order | ||
| 607 | */ | ||
| 608 | static int check_node(struct btrfs_root *root, struct btrfs_path *path, | ||
| 609 | int level) | ||
| 610 | { | ||
| 611 | struct extent_buffer *parent = NULL; | ||
| 612 | struct extent_buffer *node = path->nodes[level]; | ||
| 613 | struct btrfs_disk_key parent_key; | ||
| 614 | struct btrfs_disk_key node_key; | ||
| 615 | int parent_slot; | ||
| 616 | int slot; | ||
| 617 | struct btrfs_key cpukey; | ||
| 618 | u32 nritems = btrfs_header_nritems(node); | ||
| 619 | |||
| 620 | if (path->nodes[level + 1]) | ||
| 621 | parent = path->nodes[level + 1]; | ||
| 622 | |||
| 623 | slot = path->slots[level]; | ||
| 624 | BUG_ON(nritems == 0); | ||
| 625 | if (parent) { | ||
| 626 | parent_slot = path->slots[level + 1]; | ||
| 627 | btrfs_node_key(parent, &parent_key, parent_slot); | ||
| 628 | btrfs_node_key(node, &node_key, 0); | ||
| 629 | BUG_ON(memcmp(&parent_key, &node_key, | ||
| 630 | sizeof(struct btrfs_disk_key))); | ||
| 631 | BUG_ON(btrfs_node_blockptr(parent, parent_slot) != | ||
| 632 | btrfs_header_bytenr(node)); | ||
| 633 | } | ||
| 634 | BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); | ||
| 635 | if (slot != 0) { | ||
| 636 | btrfs_node_key_to_cpu(node, &cpukey, slot - 1); | ||
| 637 | btrfs_node_key(node, &node_key, slot); | ||
| 638 | BUG_ON(comp_keys(&node_key, &cpukey) <= 0); | ||
| 639 | } | ||
| 640 | if (slot < nritems - 1) { | ||
| 641 | btrfs_node_key_to_cpu(node, &cpukey, slot + 1); | ||
| 642 | btrfs_node_key(node, &node_key, slot); | ||
| 643 | BUG_ON(comp_keys(&node_key, &cpukey) >= 0); | ||
| 644 | } | ||
| 645 | return 0; | ||
| 646 | } | ||
| 647 | |||
| 648 | /* | ||
| 649 | * extra checking to make sure all the items in a leaf are | ||
| 650 | * well formed and in the proper order | ||
| 651 | */ | ||
| 652 | static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, | ||
| 653 | int level) | ||
| 654 | { | ||
| 655 | struct extent_buffer *leaf = path->nodes[level]; | ||
| 656 | struct extent_buffer *parent = NULL; | ||
| 657 | int parent_slot; | ||
| 658 | struct btrfs_key cpukey; | ||
| 659 | struct btrfs_disk_key parent_key; | ||
| 660 | struct btrfs_disk_key leaf_key; | ||
| 661 | int slot = path->slots[0]; | ||
| 662 | |||
| 663 | u32 nritems = btrfs_header_nritems(leaf); | ||
| 664 | |||
| 665 | if (path->nodes[level + 1]) | ||
| 666 | parent = path->nodes[level + 1]; | ||
| 667 | |||
| 668 | if (nritems == 0) | ||
| 669 | return 0; | ||
| 670 | |||
| 671 | if (parent) { | ||
| 672 | parent_slot = path->slots[level + 1]; | ||
| 673 | btrfs_node_key(parent, &parent_key, parent_slot); | ||
| 674 | btrfs_item_key(leaf, &leaf_key, 0); | ||
| 675 | |||
| 676 | BUG_ON(memcmp(&parent_key, &leaf_key, | ||
| 677 | sizeof(struct btrfs_disk_key))); | ||
| 678 | BUG_ON(btrfs_node_blockptr(parent, parent_slot) != | ||
| 679 | btrfs_header_bytenr(leaf)); | ||
| 680 | } | ||
| 681 | if (slot != 0 && slot < nritems - 1) { | ||
| 682 | btrfs_item_key(leaf, &leaf_key, slot); | ||
| 683 | btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1); | ||
| 684 | if (comp_keys(&leaf_key, &cpukey) <= 0) { | ||
| 685 | btrfs_print_leaf(root, leaf); | ||
| 686 | printk(KERN_CRIT "slot %d offset bad key\n", slot); | ||
| 687 | BUG_ON(1); | ||
| 688 | } | ||
| 689 | if (btrfs_item_offset_nr(leaf, slot - 1) != | ||
| 690 | btrfs_item_end_nr(leaf, slot)) { | ||
| 691 | btrfs_print_leaf(root, leaf); | ||
| 692 | printk(KERN_CRIT "slot %d offset bad\n", slot); | ||
| 693 | BUG_ON(1); | ||
| 694 | } | ||
| 695 | } | ||
| 696 | if (slot < nritems - 1) { | ||
| 697 | btrfs_item_key(leaf, &leaf_key, slot); | ||
| 698 | btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1); | ||
| 699 | BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0); | ||
| 700 | if (btrfs_item_offset_nr(leaf, slot) != | ||
| 701 | btrfs_item_end_nr(leaf, slot + 1)) { | ||
| 702 | btrfs_print_leaf(root, leaf); | ||
| 703 | printk(KERN_CRIT "slot %d offset bad\n", slot); | ||
| 704 | BUG_ON(1); | ||
| 705 | } | ||
| 706 | } | ||
| 707 | BUG_ON(btrfs_item_offset_nr(leaf, 0) + | ||
| 708 | btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root)); | ||
| 709 | return 0; | ||
| 710 | } | ||
| 711 | |||
| 712 | static noinline int check_block(struct btrfs_root *root, | ||
| 713 | struct btrfs_path *path, int level) | ||
| 714 | { | ||
| 715 | return 0; | ||
| 716 | if (level == 0) | ||
| 717 | return check_leaf(root, path, level); | ||
| 718 | return check_node(root, path, level); | ||
| 719 | } | ||
| 720 | |||
| 721 | /* | ||
| 722 | * search for key in the extent_buffer. The items start at offset p, | ||
| 723 | * and they are item_size apart. There are 'max' items in p. | ||
| 724 | * | ||
| 725 | * the slot in the array is returned via slot, and it points to | ||
| 726 | * the place where you would insert key if it is not found in | ||
| 727 | * the array. | ||
| 728 | * | ||
| 729 | * slot may point to max if the key is bigger than all of the keys | ||
| 730 | */ | ||
| 731 | static noinline int generic_bin_search(struct extent_buffer *eb, | ||
| 732 | unsigned long p, | ||
| 733 | int item_size, struct btrfs_key *key, | ||
| 734 | int max, int *slot) | ||
| 735 | { | ||
| 736 | int low = 0; | ||
| 737 | int high = max; | ||
| 738 | int mid; | ||
| 739 | int ret; | ||
| 740 | struct btrfs_disk_key *tmp = NULL; | ||
| 741 | struct btrfs_disk_key unaligned; | ||
| 742 | unsigned long offset; | ||
| 743 | char *map_token = NULL; | ||
| 744 | char *kaddr = NULL; | ||
| 745 | unsigned long map_start = 0; | ||
| 746 | unsigned long map_len = 0; | ||
| 747 | int err; | ||
| 748 | |||
| 749 | while (low < high) { | ||
| 750 | mid = (low + high) / 2; | ||
| 751 | offset = p + mid * item_size; | ||
| 752 | |||
| 753 | if (!map_token || offset < map_start || | ||
| 754 | (offset + sizeof(struct btrfs_disk_key)) > | ||
| 755 | map_start + map_len) { | ||
| 756 | if (map_token) { | ||
| 757 | unmap_extent_buffer(eb, map_token, KM_USER0); | ||
| 758 | map_token = NULL; | ||
| 759 | } | ||
| 760 | |||
| 761 | err = map_private_extent_buffer(eb, offset, | ||
| 762 | sizeof(struct btrfs_disk_key), | ||
| 763 | &map_token, &kaddr, | ||
| 764 | &map_start, &map_len, KM_USER0); | ||
| 765 | |||
| 766 | if (!err) { | ||
| 767 | tmp = (struct btrfs_disk_key *)(kaddr + offset - | ||
| 768 | map_start); | ||
| 769 | } else { | ||
| 770 | read_extent_buffer(eb, &unaligned, | ||
| 771 | offset, sizeof(unaligned)); | ||
| 772 | tmp = &unaligned; | ||
| 773 | } | ||
| 774 | |||
| 775 | } else { | ||
| 776 | tmp = (struct btrfs_disk_key *)(kaddr + offset - | ||
| 777 | map_start); | ||
| 778 | } | ||
| 779 | ret = comp_keys(tmp, key); | ||
| 780 | |||
| 781 | if (ret < 0) | ||
| 782 | low = mid + 1; | ||
| 783 | else if (ret > 0) | ||
| 784 | high = mid; | ||
| 785 | else { | ||
| 786 | *slot = mid; | ||
| 787 | if (map_token) | ||
| 788 | unmap_extent_buffer(eb, map_token, KM_USER0); | ||
| 789 | return 0; | ||
| 790 | } | ||
| 791 | } | ||
| 792 | *slot = low; | ||
| 793 | if (map_token) | ||
| 794 | unmap_extent_buffer(eb, map_token, KM_USER0); | ||
| 795 | return 1; | ||
| 796 | } | ||
| 797 | |||
| 798 | /* | ||
| 799 | * simple bin_search frontend that does the right thing for | ||
| 800 | * leaves vs nodes | ||
| 801 | */ | ||
| 802 | static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, | ||
| 803 | int level, int *slot) | ||
| 804 | { | ||
| 805 | if (level == 0) { | ||
| 806 | return generic_bin_search(eb, | ||
| 807 | offsetof(struct btrfs_leaf, items), | ||
| 808 | sizeof(struct btrfs_item), | ||
| 809 | key, btrfs_header_nritems(eb), | ||
| 810 | slot); | ||
| 811 | } else { | ||
| 812 | return generic_bin_search(eb, | ||
| 813 | offsetof(struct btrfs_node, ptrs), | ||
| 814 | sizeof(struct btrfs_key_ptr), | ||
| 815 | key, btrfs_header_nritems(eb), | ||
| 816 | slot); | ||
| 817 | } | ||
| 818 | return -1; | ||
| 819 | } | ||
| 820 | |||
| 821 | /* given a node and slot number, this reads the blocks it points to. The | ||
| 822 | * extent buffer is returned with a reference taken (but unlocked). | ||
| 823 | * NULL is returned on error. | ||
| 824 | */ | ||
| 825 | static noinline struct extent_buffer *read_node_slot(struct btrfs_root *root, | ||
| 826 | struct extent_buffer *parent, int slot) | ||
| 827 | { | ||
| 828 | int level = btrfs_header_level(parent); | ||
| 829 | if (slot < 0) | ||
| 830 | return NULL; | ||
| 831 | if (slot >= btrfs_header_nritems(parent)) | ||
| 832 | return NULL; | ||
| 833 | |||
| 834 | BUG_ON(level == 0); | ||
| 835 | |||
| 836 | return read_tree_block(root, btrfs_node_blockptr(parent, slot), | ||
| 837 | btrfs_level_size(root, level - 1), | ||
| 838 | btrfs_node_ptr_generation(parent, slot)); | ||
| 839 | } | ||
| 840 | |||
| 841 | /* | ||
| 842 | * node level balancing, used to make sure nodes are in proper order for | ||
| 843 | * item deletion. We balance from the top down, so we have to make sure | ||
| 844 | * that a deletion won't leave an node completely empty later on. | ||
| 845 | */ | ||
| 846 | static noinline int balance_level(struct btrfs_trans_handle *trans, | ||
| 847 | struct btrfs_root *root, | ||
| 848 | struct btrfs_path *path, int level) | ||
| 849 | { | ||
| 850 | struct extent_buffer *right = NULL; | ||
| 851 | struct extent_buffer *mid; | ||
| 852 | struct extent_buffer *left = NULL; | ||
| 853 | struct extent_buffer *parent = NULL; | ||
| 854 | int ret = 0; | ||
| 855 | int wret; | ||
| 856 | int pslot; | ||
| 857 | int orig_slot = path->slots[level]; | ||
| 858 | int err_on_enospc = 0; | ||
| 859 | u64 orig_ptr; | ||
| 860 | |||
| 861 | if (level == 0) | ||
| 862 | return 0; | ||
| 863 | |||
| 864 | mid = path->nodes[level]; | ||
| 865 | WARN_ON(!path->locks[level]); | ||
| 866 | WARN_ON(btrfs_header_generation(mid) != trans->transid); | ||
| 867 | |||
| 868 | orig_ptr = btrfs_node_blockptr(mid, orig_slot); | ||
| 869 | |||
| 870 | if (level < BTRFS_MAX_LEVEL - 1) | ||
| 871 | parent = path->nodes[level + 1]; | ||
| 872 | pslot = path->slots[level + 1]; | ||
| 873 | |||
| 874 | /* | ||
| 875 | * deal with the case where there is only one pointer in the root | ||
| 876 | * by promoting the node below to a root | ||
| 877 | */ | ||
| 878 | if (!parent) { | ||
| 879 | struct extent_buffer *child; | ||
| 880 | |||
| 881 | if (btrfs_header_nritems(mid) != 1) | ||
| 882 | return 0; | ||
| 883 | |||
| 884 | /* promote the child to a root */ | ||
| 885 | child = read_node_slot(root, mid, 0); | ||
| 886 | btrfs_tree_lock(child); | ||
| 887 | BUG_ON(!child); | ||
| 888 | ret = btrfs_cow_block(trans, root, child, mid, 0, &child, 0); | ||
| 889 | BUG_ON(ret); | ||
| 890 | |||
| 891 | spin_lock(&root->node_lock); | ||
| 892 | root->node = child; | ||
| 893 | spin_unlock(&root->node_lock); | ||
| 894 | |||
| 895 | ret = btrfs_update_extent_ref(trans, root, child->start, | ||
| 896 | mid->start, child->start, | ||
| 897 | root->root_key.objectid, | ||
| 898 | trans->transid, level - 1); | ||
| 899 | BUG_ON(ret); | ||
| 900 | |||
| 901 | add_root_to_dirty_list(root); | ||
| 902 | btrfs_tree_unlock(child); | ||
| 903 | path->locks[level] = 0; | ||
| 904 | path->nodes[level] = NULL; | ||
| 905 | clean_tree_block(trans, root, mid); | ||
| 906 | btrfs_tree_unlock(mid); | ||
| 907 | /* once for the path */ | ||
| 908 | free_extent_buffer(mid); | ||
| 909 | ret = btrfs_free_extent(trans, root, mid->start, mid->len, | ||
| 910 | mid->start, root->root_key.objectid, | ||
| 911 | btrfs_header_generation(mid), | ||
| 912 | level, 1); | ||
| 913 | /* once for the root ptr */ | ||
| 914 | free_extent_buffer(mid); | ||
| 915 | return ret; | ||
| 916 | } | ||
| 917 | if (btrfs_header_nritems(mid) > | ||
| 918 | BTRFS_NODEPTRS_PER_BLOCK(root) / 4) | ||
| 919 | return 0; | ||
| 920 | |||
| 921 | if (btrfs_header_nritems(mid) < 2) | ||
| 922 | err_on_enospc = 1; | ||
| 923 | |||
| 924 | left = read_node_slot(root, parent, pslot - 1); | ||
| 925 | if (left) { | ||
| 926 | btrfs_tree_lock(left); | ||
| 927 | wret = btrfs_cow_block(trans, root, left, | ||
| 928 | parent, pslot - 1, &left, 0); | ||
| 929 | if (wret) { | ||
| 930 | ret = wret; | ||
| 931 | goto enospc; | ||
| 932 | } | ||
| 933 | } | ||
| 934 | right = read_node_slot(root, parent, pslot + 1); | ||
| 935 | if (right) { | ||
| 936 | btrfs_tree_lock(right); | ||
| 937 | wret = btrfs_cow_block(trans, root, right, | ||
| 938 | parent, pslot + 1, &right, 0); | ||
| 939 | if (wret) { | ||
| 940 | ret = wret; | ||
| 941 | goto enospc; | ||
| 942 | } | ||
| 943 | } | ||
| 944 | |||
| 945 | /* first, try to make some room in the middle buffer */ | ||
| 946 | if (left) { | ||
| 947 | orig_slot += btrfs_header_nritems(left); | ||
| 948 | wret = push_node_left(trans, root, left, mid, 1); | ||
| 949 | if (wret < 0) | ||
| 950 | ret = wret; | ||
| 951 | if (btrfs_header_nritems(mid) < 2) | ||
| 952 | err_on_enospc = 1; | ||
| 953 | } | ||
| 954 | |||
| 955 | /* | ||
| 956 | * then try to empty the right most buffer into the middle | ||
| 957 | */ | ||
| 958 | if (right) { | ||
| 959 | wret = push_node_left(trans, root, mid, right, 1); | ||
| 960 | if (wret < 0 && wret != -ENOSPC) | ||
| 961 | ret = wret; | ||
| 962 | if (btrfs_header_nritems(right) == 0) { | ||
| 963 | u64 bytenr = right->start; | ||
| 964 | u64 generation = btrfs_header_generation(parent); | ||
| 965 | u32 blocksize = right->len; | ||
| 966 | |||
| 967 | clean_tree_block(trans, root, right); | ||
| 968 | btrfs_tree_unlock(right); | ||
| 969 | free_extent_buffer(right); | ||
| 970 | right = NULL; | ||
| 971 | wret = del_ptr(trans, root, path, level + 1, pslot + | ||
| 972 | 1); | ||
| 973 | if (wret) | ||
| 974 | ret = wret; | ||
| 975 | wret = btrfs_free_extent(trans, root, bytenr, | ||
| 976 | blocksize, parent->start, | ||
| 977 | btrfs_header_owner(parent), | ||
| 978 | generation, level, 1); | ||
| 979 | if (wret) | ||
| 980 | ret = wret; | ||
| 981 | } else { | ||
| 982 | struct btrfs_disk_key right_key; | ||
| 983 | btrfs_node_key(right, &right_key, 0); | ||
| 984 | btrfs_set_node_key(parent, &right_key, pslot + 1); | ||
| 985 | btrfs_mark_buffer_dirty(parent); | ||
| 986 | } | ||
| 987 | } | ||
| 988 | if (btrfs_header_nritems(mid) == 1) { | ||
| 989 | /* | ||
| 990 | * we're not allowed to leave a node with one item in the | ||
| 991 | * tree during a delete. A deletion from lower in the tree | ||
| 992 | * could try to delete the only pointer in this node. | ||
| 993 | * So, pull some keys from the left. | ||
| 994 | * There has to be a left pointer at this point because | ||
| 995 | * otherwise we would have pulled some pointers from the | ||
| 996 | * right | ||
| 997 | */ | ||
| 998 | BUG_ON(!left); | ||
| 999 | wret = balance_node_right(trans, root, mid, left); | ||
| 1000 | if (wret < 0) { | ||
| 1001 | ret = wret; | ||
| 1002 | goto enospc; | ||
| 1003 | } | ||
| 1004 | if (wret == 1) { | ||
| 1005 | wret = push_node_left(trans, root, left, mid, 1); | ||
| 1006 | if (wret < 0) | ||
| 1007 | ret = wret; | ||
| 1008 | } | ||
| 1009 | BUG_ON(wret == 1); | ||
| 1010 | } | ||
| 1011 | if (btrfs_header_nritems(mid) == 0) { | ||
| 1012 | /* we've managed to empty the middle node, drop it */ | ||
| 1013 | u64 root_gen = btrfs_header_generation(parent); | ||
| 1014 | u64 bytenr = mid->start; | ||
| 1015 | u32 blocksize = mid->len; | ||
| 1016 | |||
| 1017 | clean_tree_block(trans, root, mid); | ||
| 1018 | btrfs_tree_unlock(mid); | ||
| 1019 | free_extent_buffer(mid); | ||
| 1020 | mid = NULL; | ||
| 1021 | wret = del_ptr(trans, root, path, level + 1, pslot); | ||
| 1022 | if (wret) | ||
| 1023 | ret = wret; | ||
| 1024 | wret = btrfs_free_extent(trans, root, bytenr, blocksize, | ||
| 1025 | parent->start, | ||
| 1026 | btrfs_header_owner(parent), | ||
| 1027 | root_gen, level, 1); | ||
| 1028 | if (wret) | ||
| 1029 | ret = wret; | ||
| 1030 | } else { | ||
| 1031 | /* update the parent key to reflect our changes */ | ||
| 1032 | struct btrfs_disk_key mid_key; | ||
| 1033 | btrfs_node_key(mid, &mid_key, 0); | ||
| 1034 | btrfs_set_node_key(parent, &mid_key, pslot); | ||
| 1035 | btrfs_mark_buffer_dirty(parent); | ||
| 1036 | } | ||
| 1037 | |||
| 1038 | /* update the path */ | ||
| 1039 | if (left) { | ||
| 1040 | if (btrfs_header_nritems(left) > orig_slot) { | ||
| 1041 | extent_buffer_get(left); | ||
| 1042 | /* left was locked after cow */ | ||
| 1043 | path->nodes[level] = left; | ||
| 1044 | path->slots[level + 1] -= 1; | ||
| 1045 | path->slots[level] = orig_slot; | ||
| 1046 | if (mid) { | ||
| 1047 | btrfs_tree_unlock(mid); | ||
| 1048 | free_extent_buffer(mid); | ||
| 1049 | } | ||
| 1050 | } else { | ||
| 1051 | orig_slot -= btrfs_header_nritems(left); | ||
| 1052 | path->slots[level] = orig_slot; | ||
| 1053 | } | ||
| 1054 | } | ||
| 1055 | /* double check we haven't messed things up */ | ||
| 1056 | check_block(root, path, level); | ||
| 1057 | if (orig_ptr != | ||
| 1058 | btrfs_node_blockptr(path->nodes[level], path->slots[level])) | ||
| 1059 | BUG(); | ||
| 1060 | enospc: | ||
| 1061 | if (right) { | ||
| 1062 | btrfs_tree_unlock(right); | ||
| 1063 | free_extent_buffer(right); | ||
| 1064 | } | ||
| 1065 | if (left) { | ||
| 1066 | if (path->nodes[level] != left) | ||
| 1067 | btrfs_tree_unlock(left); | ||
| 1068 | free_extent_buffer(left); | ||
| 1069 | } | ||
| 1070 | return ret; | ||
| 1071 | } | ||
| 1072 | |||
| 1073 | /* Node balancing for insertion. Here we only split or push nodes around | ||
| 1074 | * when they are completely full. This is also done top down, so we | ||
| 1075 | * have to be pessimistic. | ||
| 1076 | */ | ||
| 1077 | static noinline int push_nodes_for_insert(struct btrfs_trans_handle *trans, | ||
| 1078 | struct btrfs_root *root, | ||
| 1079 | struct btrfs_path *path, int level) | ||
| 1080 | { | ||
| 1081 | struct extent_buffer *right = NULL; | ||
| 1082 | struct extent_buffer *mid; | ||
| 1083 | struct extent_buffer *left = NULL; | ||
| 1084 | struct extent_buffer *parent = NULL; | ||
| 1085 | int ret = 0; | ||
| 1086 | int wret; | ||
| 1087 | int pslot; | ||
| 1088 | int orig_slot = path->slots[level]; | ||
| 1089 | u64 orig_ptr; | ||
| 1090 | |||
| 1091 | if (level == 0) | ||
| 1092 | return 1; | ||
| 1093 | |||
| 1094 | mid = path->nodes[level]; | ||
| 1095 | WARN_ON(btrfs_header_generation(mid) != trans->transid); | ||
| 1096 | orig_ptr = btrfs_node_blockptr(mid, orig_slot); | ||
| 1097 | |||
| 1098 | if (level < BTRFS_MAX_LEVEL - 1) | ||
| 1099 | parent = path->nodes[level + 1]; | ||
| 1100 | pslot = path->slots[level + 1]; | ||
| 1101 | |||
| 1102 | if (!parent) | ||
| 1103 | return 1; | ||
| 1104 | |||
| 1105 | left = read_node_slot(root, parent, pslot - 1); | ||
| 1106 | |||
| 1107 | /* first, try to make some room in the middle buffer */ | ||
| 1108 | if (left) { | ||
| 1109 | u32 left_nr; | ||
| 1110 | |||
| 1111 | btrfs_tree_lock(left); | ||
| 1112 | left_nr = btrfs_header_nritems(left); | ||
| 1113 | if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { | ||
| 1114 | wret = 1; | ||
| 1115 | } else { | ||
| 1116 | ret = btrfs_cow_block(trans, root, left, parent, | ||
| 1117 | pslot - 1, &left, 0); | ||
| 1118 | if (ret) | ||
| 1119 | wret = 1; | ||
| 1120 | else { | ||
| 1121 | wret = push_node_left(trans, root, | ||
| 1122 | left, mid, 0); | ||
| 1123 | } | ||
| 1124 | } | ||
| 1125 | if (wret < 0) | ||
| 1126 | ret = wret; | ||
| 1127 | if (wret == 0) { | ||
| 1128 | struct btrfs_disk_key disk_key; | ||
| 1129 | orig_slot += left_nr; | ||
| 1130 | btrfs_node_key(mid, &disk_key, 0); | ||
| 1131 | btrfs_set_node_key(parent, &disk_key, pslot); | ||
| 1132 | btrfs_mark_buffer_dirty(parent); | ||
| 1133 | if (btrfs_header_nritems(left) > orig_slot) { | ||
| 1134 | path->nodes[level] = left; | ||
| 1135 | path->slots[level + 1] -= 1; | ||
| 1136 | path->slots[level] = orig_slot; | ||
| 1137 | btrfs_tree_unlock(mid); | ||
| 1138 | free_extent_buffer(mid); | ||
| 1139 | } else { | ||
| 1140 | orig_slot -= | ||
| 1141 | btrfs_header_nritems(left); | ||
| 1142 | path->slots[level] = orig_slot; | ||
| 1143 | btrfs_tree_unlock(left); | ||
| 1144 | free_extent_buffer(left); | ||
| 1145 | } | ||
| 1146 | return 0; | ||
| 1147 | } | ||
| 1148 | btrfs_tree_unlock(left); | ||
| 1149 | free_extent_buffer(left); | ||
| 1150 | } | ||
| 1151 | right = read_node_slot(root, parent, pslot + 1); | ||
| 1152 | |||
| 1153 | /* | ||
| 1154 | * then try to empty the right most buffer into the middle | ||
| 1155 | */ | ||
| 1156 | if (right) { | ||
| 1157 | u32 right_nr; | ||
| 1158 | btrfs_tree_lock(right); | ||
| 1159 | right_nr = btrfs_header_nritems(right); | ||
| 1160 | if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { | ||
| 1161 | wret = 1; | ||
| 1162 | } else { | ||
| 1163 | ret = btrfs_cow_block(trans, root, right, | ||
| 1164 | parent, pslot + 1, | ||
| 1165 | &right, 0); | ||
| 1166 | if (ret) | ||
| 1167 | wret = 1; | ||
| 1168 | else { | ||
| 1169 | wret = balance_node_right(trans, root, | ||
| 1170 | right, mid); | ||
| 1171 | } | ||
| 1172 | } | ||
| 1173 | if (wret < 0) | ||
| 1174 | ret = wret; | ||
| 1175 | if (wret == 0) { | ||
| 1176 | struct btrfs_disk_key disk_key; | ||
| 1177 | |||
| 1178 | btrfs_node_key(right, &disk_key, 0); | ||
| 1179 | btrfs_set_node_key(parent, &disk_key, pslot + 1); | ||
| 1180 | btrfs_mark_buffer_dirty(parent); | ||
| 1181 | |||
| 1182 | if (btrfs_header_nritems(mid) <= orig_slot) { | ||
| 1183 | path->nodes[level] = right; | ||
| 1184 | path->slots[level + 1] += 1; | ||
| 1185 | path->slots[level] = orig_slot - | ||
| 1186 | btrfs_header_nritems(mid); | ||
| 1187 | btrfs_tree_unlock(mid); | ||
| 1188 | free_extent_buffer(mid); | ||
| 1189 | } else { | ||
| 1190 | btrfs_tree_unlock(right); | ||
| 1191 | free_extent_buffer(right); | ||
| 1192 | } | ||
| 1193 | return 0; | ||
| 1194 | } | ||
| 1195 | btrfs_tree_unlock(right); | ||
| 1196 | free_extent_buffer(right); | ||
| 1197 | } | ||
| 1198 | return 1; | ||
| 1199 | } | ||
| 1200 | |||
| 1201 | /* | ||
| 1202 | * readahead one full node of leaves, finding things that are close | ||
| 1203 | * to the block in 'slot', and triggering ra on them. | ||
| 1204 | */ | ||
| 1205 | static noinline void reada_for_search(struct btrfs_root *root, | ||
| 1206 | struct btrfs_path *path, | ||
| 1207 | int level, int slot, u64 objectid) | ||
| 1208 | { | ||
| 1209 | struct extent_buffer *node; | ||
| 1210 | struct btrfs_disk_key disk_key; | ||
| 1211 | u32 nritems; | ||
| 1212 | u64 search; | ||
| 1213 | u64 lowest_read; | ||
| 1214 | u64 highest_read; | ||
| 1215 | u64 nread = 0; | ||
| 1216 | int direction = path->reada; | ||
| 1217 | struct extent_buffer *eb; | ||
| 1218 | u32 nr; | ||
| 1219 | u32 blocksize; | ||
| 1220 | u32 nscan = 0; | ||
| 1221 | |||
| 1222 | if (level != 1) | ||
| 1223 | return; | ||
| 1224 | |||
| 1225 | if (!path->nodes[level]) | ||
| 1226 | return; | ||
| 1227 | |||
| 1228 | node = path->nodes[level]; | ||
| 1229 | |||
| 1230 | search = btrfs_node_blockptr(node, slot); | ||
| 1231 | blocksize = btrfs_level_size(root, level - 1); | ||
| 1232 | eb = btrfs_find_tree_block(root, search, blocksize); | ||
| 1233 | if (eb) { | ||
| 1234 | free_extent_buffer(eb); | ||
| 1235 | return; | ||
| 1236 | } | ||
| 1237 | |||
| 1238 | highest_read = search; | ||
| 1239 | lowest_read = search; | ||
| 1240 | |||
| 1241 | nritems = btrfs_header_nritems(node); | ||
| 1242 | nr = slot; | ||
| 1243 | while (1) { | ||
| 1244 | if (direction < 0) { | ||
| 1245 | if (nr == 0) | ||
| 1246 | break; | ||
| 1247 | nr--; | ||
| 1248 | } else if (direction > 0) { | ||
| 1249 | nr++; | ||
| 1250 | if (nr >= nritems) | ||
| 1251 | break; | ||
| 1252 | } | ||
| 1253 | if (path->reada < 0 && objectid) { | ||
| 1254 | btrfs_node_key(node, &disk_key, nr); | ||
| 1255 | if (btrfs_disk_key_objectid(&disk_key) != objectid) | ||
| 1256 | break; | ||
| 1257 | } | ||
| 1258 | search = btrfs_node_blockptr(node, nr); | ||
| 1259 | if ((search >= lowest_read && search <= highest_read) || | ||
| 1260 | (search < lowest_read && lowest_read - search <= 16384) || | ||
| 1261 | (search > highest_read && search - highest_read <= 16384)) { | ||
| 1262 | readahead_tree_block(root, search, blocksize, | ||
| 1263 | btrfs_node_ptr_generation(node, nr)); | ||
| 1264 | nread += blocksize; | ||
| 1265 | } | ||
| 1266 | nscan++; | ||
| 1267 | if (path->reada < 2 && (nread > (64 * 1024) || nscan > 32)) | ||
| 1268 | break; | ||
| 1269 | |||
| 1270 | if (nread > (256 * 1024) || nscan > 128) | ||
| 1271 | break; | ||
| 1272 | |||
| 1273 | if (search < lowest_read) | ||
| 1274 | lowest_read = search; | ||
| 1275 | if (search > highest_read) | ||
| 1276 | highest_read = search; | ||
| 1277 | } | ||
| 1278 | } | ||
| 1279 | |||
| 1280 | /* | ||
| 1281 | * when we walk down the tree, it is usually safe to unlock the higher layers | ||
| 1282 | * in the tree. The exceptions are when our path goes through slot 0, because | ||
| 1283 | * operations on the tree might require changing key pointers higher up in the | ||
| 1284 | * tree. | ||
| 1285 | * | ||
| 1286 | * callers might also have set path->keep_locks, which tells this code to keep | ||
| 1287 | * the lock if the path points to the last slot in the block. This is part of | ||
| 1288 | * walking through the tree, and selecting the next slot in the higher block. | ||
| 1289 | * | ||
| 1290 | * lowest_unlock sets the lowest level in the tree we're allowed to unlock. so | ||
| 1291 | * if lowest_unlock is 1, level 0 won't be unlocked | ||
| 1292 | */ | ||
| 1293 | static noinline void unlock_up(struct btrfs_path *path, int level, | ||
| 1294 | int lowest_unlock) | ||
| 1295 | { | ||
| 1296 | int i; | ||
| 1297 | int skip_level = level; | ||
| 1298 | int no_skips = 0; | ||
| 1299 | struct extent_buffer *t; | ||
| 1300 | |||
| 1301 | for (i = level; i < BTRFS_MAX_LEVEL; i++) { | ||
| 1302 | if (!path->nodes[i]) | ||
| 1303 | break; | ||
| 1304 | if (!path->locks[i]) | ||
| 1305 | break; | ||
| 1306 | if (!no_skips && path->slots[i] == 0) { | ||
| 1307 | skip_level = i + 1; | ||
| 1308 | continue; | ||
| 1309 | } | ||
| 1310 | if (!no_skips && path->keep_locks) { | ||
| 1311 | u32 nritems; | ||
| 1312 | t = path->nodes[i]; | ||
| 1313 | nritems = btrfs_header_nritems(t); | ||
| 1314 | if (nritems < 1 || path->slots[i] >= nritems - 1) { | ||
| 1315 | skip_level = i + 1; | ||
| 1316 | continue; | ||
| 1317 | } | ||
| 1318 | } | ||
| 1319 | if (skip_level < i && i >= lowest_unlock) | ||
| 1320 | no_skips = 1; | ||
| 1321 | |||
| 1322 | t = path->nodes[i]; | ||
| 1323 | if (i >= lowest_unlock && i > skip_level && path->locks[i]) { | ||
| 1324 | btrfs_tree_unlock(t); | ||
| 1325 | path->locks[i] = 0; | ||
| 1326 | } | ||
| 1327 | } | ||
| 1328 | } | ||
| 1329 | |||
| 1330 | /* | ||
| 1331 | * look for key in the tree. path is filled in with nodes along the way | ||
| 1332 | * if key is found, we return zero and you can find the item in the leaf | ||
| 1333 | * level of the path (level 0) | ||
| 1334 | * | ||
| 1335 | * If the key isn't found, the path points to the slot where it should | ||
| 1336 | * be inserted, and 1 is returned. If there are other errors during the | ||
| 1337 | * search a negative error number is returned. | ||
| 1338 | * | ||
| 1339 | * if ins_len > 0, nodes and leaves will be split as we walk down the | ||
| 1340 | * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if | ||
| 1341 | * possible) | ||
| 1342 | */ | ||
| 1343 | int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1344 | *root, struct btrfs_key *key, struct btrfs_path *p, int | ||
| 1345 | ins_len, int cow) | ||
| 1346 | { | ||
| 1347 | struct extent_buffer *b; | ||
| 1348 | struct extent_buffer *tmp; | ||
| 1349 | int slot; | ||
| 1350 | int ret; | ||
| 1351 | int level; | ||
| 1352 | int should_reada = p->reada; | ||
| 1353 | int lowest_unlock = 1; | ||
| 1354 | int blocksize; | ||
| 1355 | u8 lowest_level = 0; | ||
| 1356 | u64 blocknr; | ||
| 1357 | u64 gen; | ||
| 1358 | struct btrfs_key prealloc_block; | ||
| 1359 | |||
| 1360 | lowest_level = p->lowest_level; | ||
| 1361 | WARN_ON(lowest_level && ins_len > 0); | ||
| 1362 | WARN_ON(p->nodes[0] != NULL); | ||
| 1363 | |||
| 1364 | if (ins_len < 0) | ||
| 1365 | lowest_unlock = 2; | ||
| 1366 | |||
| 1367 | prealloc_block.objectid = 0; | ||
| 1368 | |||
| 1369 | again: | ||
| 1370 | if (p->skip_locking) | ||
| 1371 | b = btrfs_root_node(root); | ||
| 1372 | else | ||
| 1373 | b = btrfs_lock_root_node(root); | ||
| 1374 | |||
| 1375 | while (b) { | ||
| 1376 | level = btrfs_header_level(b); | ||
| 1377 | |||
| 1378 | /* | ||
| 1379 | * setup the path here so we can release it under lock | ||
| 1380 | * contention with the cow code | ||
| 1381 | */ | ||
| 1382 | p->nodes[level] = b; | ||
| 1383 | if (!p->skip_locking) | ||
| 1384 | p->locks[level] = 1; | ||
| 1385 | |||
| 1386 | if (cow) { | ||
| 1387 | int wret; | ||
| 1388 | |||
| 1389 | /* is a cow on this block not required */ | ||
| 1390 | spin_lock(&root->fs_info->hash_lock); | ||
| 1391 | if (btrfs_header_generation(b) == trans->transid && | ||
| 1392 | btrfs_header_owner(b) == root->root_key.objectid && | ||
| 1393 | !btrfs_header_flag(b, BTRFS_HEADER_FLAG_WRITTEN)) { | ||
| 1394 | spin_unlock(&root->fs_info->hash_lock); | ||
| 1395 | goto cow_done; | ||
| 1396 | } | ||
| 1397 | spin_unlock(&root->fs_info->hash_lock); | ||
| 1398 | |||
| 1399 | /* ok, we have to cow, is our old prealloc the right | ||
| 1400 | * size? | ||
| 1401 | */ | ||
| 1402 | if (prealloc_block.objectid && | ||
| 1403 | prealloc_block.offset != b->len) { | ||
| 1404 | btrfs_free_reserved_extent(root, | ||
| 1405 | prealloc_block.objectid, | ||
| 1406 | prealloc_block.offset); | ||
| 1407 | prealloc_block.objectid = 0; | ||
| 1408 | } | ||
| 1409 | |||
| 1410 | /* | ||
| 1411 | * for higher level blocks, try not to allocate blocks | ||
| 1412 | * with the block and the parent locks held. | ||
| 1413 | */ | ||
| 1414 | if (level > 1 && !prealloc_block.objectid && | ||
| 1415 | btrfs_path_lock_waiting(p, level)) { | ||
| 1416 | u32 size = b->len; | ||
| 1417 | u64 hint = b->start; | ||
| 1418 | |||
| 1419 | btrfs_release_path(root, p); | ||
| 1420 | ret = btrfs_reserve_extent(trans, root, | ||
| 1421 | size, size, 0, | ||
| 1422 | hint, (u64)-1, | ||
| 1423 | &prealloc_block, 0); | ||
| 1424 | BUG_ON(ret); | ||
| 1425 | goto again; | ||
| 1426 | } | ||
| 1427 | |||
| 1428 | wret = btrfs_cow_block(trans, root, b, | ||
| 1429 | p->nodes[level + 1], | ||
| 1430 | p->slots[level + 1], | ||
| 1431 | &b, prealloc_block.objectid); | ||
| 1432 | prealloc_block.objectid = 0; | ||
| 1433 | if (wret) { | ||
| 1434 | free_extent_buffer(b); | ||
| 1435 | ret = wret; | ||
| 1436 | goto done; | ||
| 1437 | } | ||
| 1438 | } | ||
| 1439 | cow_done: | ||
| 1440 | BUG_ON(!cow && ins_len); | ||
| 1441 | if (level != btrfs_header_level(b)) | ||
| 1442 | WARN_ON(1); | ||
| 1443 | level = btrfs_header_level(b); | ||
| 1444 | |||
| 1445 | p->nodes[level] = b; | ||
| 1446 | if (!p->skip_locking) | ||
| 1447 | p->locks[level] = 1; | ||
| 1448 | |||
| 1449 | ret = check_block(root, p, level); | ||
| 1450 | if (ret) { | ||
| 1451 | ret = -1; | ||
| 1452 | goto done; | ||
| 1453 | } | ||
| 1454 | |||
| 1455 | ret = bin_search(b, key, level, &slot); | ||
| 1456 | if (level != 0) { | ||
| 1457 | if (ret && slot > 0) | ||
| 1458 | slot -= 1; | ||
| 1459 | p->slots[level] = slot; | ||
| 1460 | if ((p->search_for_split || ins_len > 0) && | ||
| 1461 | btrfs_header_nritems(b) >= | ||
| 1462 | BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { | ||
| 1463 | int sret = split_node(trans, root, p, level); | ||
| 1464 | BUG_ON(sret > 0); | ||
| 1465 | if (sret) { | ||
| 1466 | ret = sret; | ||
| 1467 | goto done; | ||
| 1468 | } | ||
| 1469 | b = p->nodes[level]; | ||
| 1470 | slot = p->slots[level]; | ||
| 1471 | } else if (ins_len < 0) { | ||
| 1472 | int sret = balance_level(trans, root, p, | ||
| 1473 | level); | ||
| 1474 | if (sret) { | ||
| 1475 | ret = sret; | ||
| 1476 | goto done; | ||
| 1477 | } | ||
| 1478 | b = p->nodes[level]; | ||
| 1479 | if (!b) { | ||
| 1480 | btrfs_release_path(NULL, p); | ||
| 1481 | goto again; | ||
| 1482 | } | ||
| 1483 | slot = p->slots[level]; | ||
| 1484 | BUG_ON(btrfs_header_nritems(b) == 1); | ||
| 1485 | } | ||
| 1486 | unlock_up(p, level, lowest_unlock); | ||
| 1487 | |||
| 1488 | /* this is only true while dropping a snapshot */ | ||
| 1489 | if (level == lowest_level) { | ||
| 1490 | ret = 0; | ||
| 1491 | goto done; | ||
| 1492 | } | ||
| 1493 | |||
| 1494 | blocknr = btrfs_node_blockptr(b, slot); | ||
| 1495 | gen = btrfs_node_ptr_generation(b, slot); | ||
| 1496 | blocksize = btrfs_level_size(root, level - 1); | ||
| 1497 | |||
| 1498 | tmp = btrfs_find_tree_block(root, blocknr, blocksize); | ||
| 1499 | if (tmp && btrfs_buffer_uptodate(tmp, gen)) { | ||
| 1500 | b = tmp; | ||
| 1501 | } else { | ||
| 1502 | /* | ||
| 1503 | * reduce lock contention at high levels | ||
| 1504 | * of the btree by dropping locks before | ||
| 1505 | * we read. | ||
| 1506 | */ | ||
| 1507 | if (level > 1) { | ||
| 1508 | btrfs_release_path(NULL, p); | ||
| 1509 | if (tmp) | ||
| 1510 | free_extent_buffer(tmp); | ||
| 1511 | if (should_reada) | ||
| 1512 | reada_for_search(root, p, | ||
| 1513 | level, slot, | ||
| 1514 | key->objectid); | ||
| 1515 | |||
| 1516 | tmp = read_tree_block(root, blocknr, | ||
| 1517 | blocksize, gen); | ||
| 1518 | if (tmp) | ||
| 1519 | free_extent_buffer(tmp); | ||
| 1520 | goto again; | ||
| 1521 | } else { | ||
| 1522 | if (tmp) | ||
| 1523 | free_extent_buffer(tmp); | ||
| 1524 | if (should_reada) | ||
| 1525 | reada_for_search(root, p, | ||
| 1526 | level, slot, | ||
| 1527 | key->objectid); | ||
| 1528 | b = read_node_slot(root, b, slot); | ||
| 1529 | } | ||
| 1530 | } | ||
| 1531 | if (!p->skip_locking) | ||
| 1532 | btrfs_tree_lock(b); | ||
| 1533 | } else { | ||
| 1534 | p->slots[level] = slot; | ||
| 1535 | if (ins_len > 0 && | ||
| 1536 | btrfs_leaf_free_space(root, b) < ins_len) { | ||
| 1537 | int sret = split_leaf(trans, root, key, | ||
| 1538 | p, ins_len, ret == 0); | ||
| 1539 | BUG_ON(sret > 0); | ||
| 1540 | if (sret) { | ||
| 1541 | ret = sret; | ||
| 1542 | goto done; | ||
| 1543 | } | ||
| 1544 | } | ||
| 1545 | if (!p->search_for_split) | ||
| 1546 | unlock_up(p, level, lowest_unlock); | ||
| 1547 | goto done; | ||
| 1548 | } | ||
| 1549 | } | ||
| 1550 | ret = 1; | ||
| 1551 | done: | ||
| 1552 | if (prealloc_block.objectid) { | ||
| 1553 | btrfs_free_reserved_extent(root, | ||
| 1554 | prealloc_block.objectid, | ||
| 1555 | prealloc_block.offset); | ||
| 1556 | } | ||
| 1557 | |||
| 1558 | return ret; | ||
| 1559 | } | ||
| 1560 | |||
| 1561 | int btrfs_merge_path(struct btrfs_trans_handle *trans, | ||
| 1562 | struct btrfs_root *root, | ||
| 1563 | struct btrfs_key *node_keys, | ||
| 1564 | u64 *nodes, int lowest_level) | ||
| 1565 | { | ||
| 1566 | struct extent_buffer *eb; | ||
| 1567 | struct extent_buffer *parent; | ||
| 1568 | struct btrfs_key key; | ||
| 1569 | u64 bytenr; | ||
| 1570 | u64 generation; | ||
| 1571 | u32 blocksize; | ||
| 1572 | int level; | ||
| 1573 | int slot; | ||
| 1574 | int key_match; | ||
| 1575 | int ret; | ||
| 1576 | |||
| 1577 | eb = btrfs_lock_root_node(root); | ||
| 1578 | ret = btrfs_cow_block(trans, root, eb, NULL, 0, &eb, 0); | ||
| 1579 | BUG_ON(ret); | ||
| 1580 | |||
| 1581 | parent = eb; | ||
| 1582 | while (1) { | ||
| 1583 | level = btrfs_header_level(parent); | ||
| 1584 | if (level == 0 || level <= lowest_level) | ||
| 1585 | break; | ||
| 1586 | |||
| 1587 | ret = bin_search(parent, &node_keys[lowest_level], level, | ||
| 1588 | &slot); | ||
| 1589 | if (ret && slot > 0) | ||
| 1590 | slot--; | ||
| 1591 | |||
| 1592 | bytenr = btrfs_node_blockptr(parent, slot); | ||
| 1593 | if (nodes[level - 1] == bytenr) | ||
| 1594 | break; | ||
| 1595 | |||
| 1596 | blocksize = btrfs_level_size(root, level - 1); | ||
| 1597 | generation = btrfs_node_ptr_generation(parent, slot); | ||
| 1598 | btrfs_node_key_to_cpu(eb, &key, slot); | ||
| 1599 | key_match = !memcmp(&key, &node_keys[level - 1], sizeof(key)); | ||
| 1600 | |||
| 1601 | if (generation == trans->transid) { | ||
| 1602 | eb = read_tree_block(root, bytenr, blocksize, | ||
| 1603 | generation); | ||
| 1604 | btrfs_tree_lock(eb); | ||
| 1605 | } | ||
| 1606 | |||
| 1607 | /* | ||
| 1608 | * if node keys match and node pointer hasn't been modified | ||
| 1609 | * in the running transaction, we can merge the path. for | ||
| 1610 | * blocks owened by reloc trees, the node pointer check is | ||
| 1611 | * skipped, this is because these blocks are fully controlled | ||
| 1612 | * by the space balance code, no one else can modify them. | ||
| 1613 | */ | ||
| 1614 | if (!nodes[level - 1] || !key_match || | ||
| 1615 | (generation == trans->transid && | ||
| 1616 | btrfs_header_owner(eb) != BTRFS_TREE_RELOC_OBJECTID)) { | ||
| 1617 | if (level == 1 || level == lowest_level + 1) { | ||
| 1618 | if (generation == trans->transid) { | ||
| 1619 | btrfs_tree_unlock(eb); | ||
| 1620 | free_extent_buffer(eb); | ||
| 1621 | } | ||
| 1622 | break; | ||
| 1623 | } | ||
| 1624 | |||
| 1625 | if (generation != trans->transid) { | ||
| 1626 | eb = read_tree_block(root, bytenr, blocksize, | ||
| 1627 | generation); | ||
| 1628 | btrfs_tree_lock(eb); | ||
| 1629 | } | ||
| 1630 | |||
| 1631 | ret = btrfs_cow_block(trans, root, eb, parent, slot, | ||
| 1632 | &eb, 0); | ||
| 1633 | BUG_ON(ret); | ||
| 1634 | |||
| 1635 | if (root->root_key.objectid == | ||
| 1636 | BTRFS_TREE_RELOC_OBJECTID) { | ||
| 1637 | if (!nodes[level - 1]) { | ||
| 1638 | nodes[level - 1] = eb->start; | ||
| 1639 | memcpy(&node_keys[level - 1], &key, | ||
| 1640 | sizeof(node_keys[0])); | ||
| 1641 | } else { | ||
| 1642 | WARN_ON(1); | ||
| 1643 | } | ||
| 1644 | } | ||
| 1645 | |||
| 1646 | btrfs_tree_unlock(parent); | ||
| 1647 | free_extent_buffer(parent); | ||
| 1648 | parent = eb; | ||
| 1649 | continue; | ||
| 1650 | } | ||
| 1651 | |||
| 1652 | btrfs_set_node_blockptr(parent, slot, nodes[level - 1]); | ||
| 1653 | btrfs_set_node_ptr_generation(parent, slot, trans->transid); | ||
| 1654 | btrfs_mark_buffer_dirty(parent); | ||
| 1655 | |||
| 1656 | ret = btrfs_inc_extent_ref(trans, root, | ||
| 1657 | nodes[level - 1], | ||
| 1658 | blocksize, parent->start, | ||
| 1659 | btrfs_header_owner(parent), | ||
| 1660 | btrfs_header_generation(parent), | ||
| 1661 | level - 1); | ||
| 1662 | BUG_ON(ret); | ||
| 1663 | |||
| 1664 | /* | ||
| 1665 | * If the block was created in the running transaction, | ||
| 1666 | * it's possible this is the last reference to it, so we | ||
| 1667 | * should drop the subtree. | ||
| 1668 | */ | ||
| 1669 | if (generation == trans->transid) { | ||
| 1670 | ret = btrfs_drop_subtree(trans, root, eb, parent); | ||
| 1671 | BUG_ON(ret); | ||
| 1672 | btrfs_tree_unlock(eb); | ||
| 1673 | free_extent_buffer(eb); | ||
| 1674 | } else { | ||
| 1675 | ret = btrfs_free_extent(trans, root, bytenr, | ||
| 1676 | blocksize, parent->start, | ||
| 1677 | btrfs_header_owner(parent), | ||
| 1678 | btrfs_header_generation(parent), | ||
| 1679 | level - 1, 1); | ||
| 1680 | BUG_ON(ret); | ||
| 1681 | } | ||
| 1682 | break; | ||
| 1683 | } | ||
| 1684 | btrfs_tree_unlock(parent); | ||
| 1685 | free_extent_buffer(parent); | ||
| 1686 | return 0; | ||
| 1687 | } | ||
| 1688 | |||
| 1689 | /* | ||
| 1690 | * adjust the pointers going up the tree, starting at level | ||
| 1691 | * making sure the right key of each node is points to 'key'. | ||
| 1692 | * This is used after shifting pointers to the left, so it stops | ||
| 1693 | * fixing up pointers when a given leaf/node is not in slot 0 of the | ||
| 1694 | * higher levels | ||
| 1695 | * | ||
| 1696 | * If this fails to write a tree block, it returns -1, but continues | ||
| 1697 | * fixing up the blocks in ram so the tree is consistent. | ||
| 1698 | */ | ||
| 1699 | static int fixup_low_keys(struct btrfs_trans_handle *trans, | ||
| 1700 | struct btrfs_root *root, struct btrfs_path *path, | ||
| 1701 | struct btrfs_disk_key *key, int level) | ||
| 1702 | { | ||
| 1703 | int i; | ||
| 1704 | int ret = 0; | ||
| 1705 | struct extent_buffer *t; | ||
| 1706 | |||
| 1707 | for (i = level; i < BTRFS_MAX_LEVEL; i++) { | ||
| 1708 | int tslot = path->slots[i]; | ||
| 1709 | if (!path->nodes[i]) | ||
| 1710 | break; | ||
| 1711 | t = path->nodes[i]; | ||
| 1712 | btrfs_set_node_key(t, key, tslot); | ||
| 1713 | btrfs_mark_buffer_dirty(path->nodes[i]); | ||
| 1714 | if (tslot != 0) | ||
| 1715 | break; | ||
| 1716 | } | ||
| 1717 | return ret; | ||
| 1718 | } | ||
| 1719 | |||
| 1720 | /* | ||
| 1721 | * update item key. | ||
| 1722 | * | ||
| 1723 | * This function isn't completely safe. It's the caller's responsibility | ||
| 1724 | * that the new key won't break the order | ||
| 1725 | */ | ||
| 1726 | int btrfs_set_item_key_safe(struct btrfs_trans_handle *trans, | ||
| 1727 | struct btrfs_root *root, struct btrfs_path *path, | ||
| 1728 | struct btrfs_key *new_key) | ||
| 1729 | { | ||
| 1730 | struct btrfs_disk_key disk_key; | ||
| 1731 | struct extent_buffer *eb; | ||
| 1732 | int slot; | ||
| 1733 | |||
| 1734 | eb = path->nodes[0]; | ||
| 1735 | slot = path->slots[0]; | ||
| 1736 | if (slot > 0) { | ||
| 1737 | btrfs_item_key(eb, &disk_key, slot - 1); | ||
| 1738 | if (comp_keys(&disk_key, new_key) >= 0) | ||
| 1739 | return -1; | ||
| 1740 | } | ||
| 1741 | if (slot < btrfs_header_nritems(eb) - 1) { | ||
| 1742 | btrfs_item_key(eb, &disk_key, slot + 1); | ||
| 1743 | if (comp_keys(&disk_key, new_key) <= 0) | ||
| 1744 | return -1; | ||
| 1745 | } | ||
| 1746 | |||
| 1747 | btrfs_cpu_key_to_disk(&disk_key, new_key); | ||
| 1748 | btrfs_set_item_key(eb, &disk_key, slot); | ||
| 1749 | btrfs_mark_buffer_dirty(eb); | ||
| 1750 | if (slot == 0) | ||
| 1751 | fixup_low_keys(trans, root, path, &disk_key, 1); | ||
| 1752 | return 0; | ||
| 1753 | } | ||
| 1754 | |||
| 1755 | /* | ||
| 1756 | * try to push data from one node into the next node left in the | ||
| 1757 | * tree. | ||
| 1758 | * | ||
| 1759 | * returns 0 if some ptrs were pushed left, < 0 if there was some horrible | ||
| 1760 | * error, and > 0 if there was no room in the left hand block. | ||
| 1761 | */ | ||
| 1762 | static int push_node_left(struct btrfs_trans_handle *trans, | ||
| 1763 | struct btrfs_root *root, struct extent_buffer *dst, | ||
| 1764 | struct extent_buffer *src, int empty) | ||
| 1765 | { | ||
| 1766 | int push_items = 0; | ||
| 1767 | int src_nritems; | ||
| 1768 | int dst_nritems; | ||
| 1769 | int ret = 0; | ||
| 1770 | |||
| 1771 | src_nritems = btrfs_header_nritems(src); | ||
| 1772 | dst_nritems = btrfs_header_nritems(dst); | ||
| 1773 | push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; | ||
| 1774 | WARN_ON(btrfs_header_generation(src) != trans->transid); | ||
| 1775 | WARN_ON(btrfs_header_generation(dst) != trans->transid); | ||
| 1776 | |||
| 1777 | if (!empty && src_nritems <= 8) | ||
| 1778 | return 1; | ||
| 1779 | |||
| 1780 | if (push_items <= 0) | ||
| 1781 | return 1; | ||
| 1782 | |||
| 1783 | if (empty) { | ||
| 1784 | push_items = min(src_nritems, push_items); | ||
| 1785 | if (push_items < src_nritems) { | ||
| 1786 | /* leave at least 8 pointers in the node if | ||
| 1787 | * we aren't going to empty it | ||
| 1788 | */ | ||
| 1789 | if (src_nritems - push_items < 8) { | ||
| 1790 | if (push_items <= 8) | ||
| 1791 | return 1; | ||
| 1792 | push_items -= 8; | ||
| 1793 | } | ||
| 1794 | } | ||
| 1795 | } else | ||
| 1796 | push_items = min(src_nritems - 8, push_items); | ||
| 1797 | |||
| 1798 | copy_extent_buffer(dst, src, | ||
| 1799 | btrfs_node_key_ptr_offset(dst_nritems), | ||
| 1800 | btrfs_node_key_ptr_offset(0), | ||
| 1801 | push_items * sizeof(struct btrfs_key_ptr)); | ||
| 1802 | |||
| 1803 | if (push_items < src_nritems) { | ||
| 1804 | memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), | ||
| 1805 | btrfs_node_key_ptr_offset(push_items), | ||
| 1806 | (src_nritems - push_items) * | ||
| 1807 | sizeof(struct btrfs_key_ptr)); | ||
| 1808 | } | ||
| 1809 | btrfs_set_header_nritems(src, src_nritems - push_items); | ||
| 1810 | btrfs_set_header_nritems(dst, dst_nritems + push_items); | ||
| 1811 | btrfs_mark_buffer_dirty(src); | ||
| 1812 | btrfs_mark_buffer_dirty(dst); | ||
| 1813 | |||
| 1814 | ret = btrfs_update_ref(trans, root, src, dst, dst_nritems, push_items); | ||
| 1815 | BUG_ON(ret); | ||
| 1816 | |||
| 1817 | return ret; | ||
| 1818 | } | ||
| 1819 | |||
| 1820 | /* | ||
| 1821 | * try to push data from one node into the next node right in the | ||
| 1822 | * tree. | ||
| 1823 | * | ||
| 1824 | * returns 0 if some ptrs were pushed, < 0 if there was some horrible | ||
| 1825 | * error, and > 0 if there was no room in the right hand block. | ||
| 1826 | * | ||
| 1827 | * this will only push up to 1/2 the contents of the left node over | ||
| 1828 | */ | ||
| 1829 | static int balance_node_right(struct btrfs_trans_handle *trans, | ||
| 1830 | struct btrfs_root *root, | ||
| 1831 | struct extent_buffer *dst, | ||
| 1832 | struct extent_buffer *src) | ||
| 1833 | { | ||
| 1834 | int push_items = 0; | ||
| 1835 | int max_push; | ||
| 1836 | int src_nritems; | ||
| 1837 | int dst_nritems; | ||
| 1838 | int ret = 0; | ||
| 1839 | |||
| 1840 | WARN_ON(btrfs_header_generation(src) != trans->transid); | ||
| 1841 | WARN_ON(btrfs_header_generation(dst) != trans->transid); | ||
| 1842 | |||
| 1843 | src_nritems = btrfs_header_nritems(src); | ||
| 1844 | dst_nritems = btrfs_header_nritems(dst); | ||
| 1845 | push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; | ||
| 1846 | if (push_items <= 0) | ||
| 1847 | return 1; | ||
| 1848 | |||
| 1849 | if (src_nritems < 4) | ||
| 1850 | return 1; | ||
| 1851 | |||
| 1852 | max_push = src_nritems / 2 + 1; | ||
| 1853 | /* don't try to empty the node */ | ||
| 1854 | if (max_push >= src_nritems) | ||
| 1855 | return 1; | ||
| 1856 | |||
| 1857 | if (max_push < push_items) | ||
| 1858 | push_items = max_push; | ||
| 1859 | |||
| 1860 | memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), | ||
| 1861 | btrfs_node_key_ptr_offset(0), | ||
| 1862 | (dst_nritems) * | ||
| 1863 | sizeof(struct btrfs_key_ptr)); | ||
| 1864 | |||
| 1865 | copy_extent_buffer(dst, src, | ||
| 1866 | btrfs_node_key_ptr_offset(0), | ||
| 1867 | btrfs_node_key_ptr_offset(src_nritems - push_items), | ||
| 1868 | push_items * sizeof(struct btrfs_key_ptr)); | ||
| 1869 | |||
| 1870 | btrfs_set_header_nritems(src, src_nritems - push_items); | ||
| 1871 | btrfs_set_header_nritems(dst, dst_nritems + push_items); | ||
| 1872 | |||
| 1873 | btrfs_mark_buffer_dirty(src); | ||
| 1874 | btrfs_mark_buffer_dirty(dst); | ||
| 1875 | |||
| 1876 | ret = btrfs_update_ref(trans, root, src, dst, 0, push_items); | ||
| 1877 | BUG_ON(ret); | ||
| 1878 | |||
| 1879 | return ret; | ||
| 1880 | } | ||
| 1881 | |||
| 1882 | /* | ||
| 1883 | * helper function to insert a new root level in the tree. | ||
| 1884 | * A new node is allocated, and a single item is inserted to | ||
| 1885 | * point to the existing root | ||
| 1886 | * | ||
| 1887 | * returns zero on success or < 0 on failure. | ||
| 1888 | */ | ||
| 1889 | static noinline int insert_new_root(struct btrfs_trans_handle *trans, | ||
| 1890 | struct btrfs_root *root, | ||
| 1891 | struct btrfs_path *path, int level) | ||
| 1892 | { | ||
| 1893 | u64 lower_gen; | ||
| 1894 | struct extent_buffer *lower; | ||
| 1895 | struct extent_buffer *c; | ||
| 1896 | struct extent_buffer *old; | ||
| 1897 | struct btrfs_disk_key lower_key; | ||
| 1898 | int ret; | ||
| 1899 | |||
| 1900 | BUG_ON(path->nodes[level]); | ||
| 1901 | BUG_ON(path->nodes[level-1] != root->node); | ||
| 1902 | |||
| 1903 | lower = path->nodes[level-1]; | ||
| 1904 | if (level == 1) | ||
| 1905 | btrfs_item_key(lower, &lower_key, 0); | ||
| 1906 | else | ||
| 1907 | btrfs_node_key(lower, &lower_key, 0); | ||
| 1908 | |||
| 1909 | c = btrfs_alloc_free_block(trans, root, root->nodesize, 0, | ||
| 1910 | root->root_key.objectid, trans->transid, | ||
| 1911 | level, root->node->start, 0); | ||
| 1912 | if (IS_ERR(c)) | ||
| 1913 | return PTR_ERR(c); | ||
| 1914 | |||
| 1915 | memset_extent_buffer(c, 0, 0, root->nodesize); | ||
| 1916 | btrfs_set_header_nritems(c, 1); | ||
| 1917 | btrfs_set_header_level(c, level); | ||
| 1918 | btrfs_set_header_bytenr(c, c->start); | ||
| 1919 | btrfs_set_header_generation(c, trans->transid); | ||
| 1920 | btrfs_set_header_owner(c, root->root_key.objectid); | ||
| 1921 | |||
| 1922 | write_extent_buffer(c, root->fs_info->fsid, | ||
| 1923 | (unsigned long)btrfs_header_fsid(c), | ||
| 1924 | BTRFS_FSID_SIZE); | ||
| 1925 | |||
| 1926 | write_extent_buffer(c, root->fs_info->chunk_tree_uuid, | ||
| 1927 | (unsigned long)btrfs_header_chunk_tree_uuid(c), | ||
| 1928 | BTRFS_UUID_SIZE); | ||
| 1929 | |||
| 1930 | btrfs_set_node_key(c, &lower_key, 0); | ||
| 1931 | btrfs_set_node_blockptr(c, 0, lower->start); | ||
| 1932 | lower_gen = btrfs_header_generation(lower); | ||
| 1933 | WARN_ON(lower_gen != trans->transid); | ||
| 1934 | |||
| 1935 | btrfs_set_node_ptr_generation(c, 0, lower_gen); | ||
| 1936 | |||
| 1937 | btrfs_mark_buffer_dirty(c); | ||
| 1938 | |||
| 1939 | spin_lock(&root->node_lock); | ||
| 1940 | old = root->node; | ||
| 1941 | root->node = c; | ||
| 1942 | spin_unlock(&root->node_lock); | ||
| 1943 | |||
| 1944 | ret = btrfs_update_extent_ref(trans, root, lower->start, | ||
| 1945 | lower->start, c->start, | ||
| 1946 | root->root_key.objectid, | ||
| 1947 | trans->transid, level - 1); | ||
| 1948 | BUG_ON(ret); | ||
| 1949 | |||
| 1950 | /* the super has an extra ref to root->node */ | ||
| 1951 | free_extent_buffer(old); | ||
| 1952 | |||
| 1953 | add_root_to_dirty_list(root); | ||
| 1954 | extent_buffer_get(c); | ||
| 1955 | path->nodes[level] = c; | ||
| 1956 | path->locks[level] = 1; | ||
| 1957 | path->slots[level] = 0; | ||
| 1958 | return 0; | ||
| 1959 | } | ||
| 1960 | |||
| 1961 | /* | ||
| 1962 | * worker function to insert a single pointer in a node. | ||
| 1963 | * the node should have enough room for the pointer already | ||
| 1964 | * | ||
| 1965 | * slot and level indicate where you want the key to go, and | ||
| 1966 | * blocknr is the block the key points to. | ||
| 1967 | * | ||
| 1968 | * returns zero on success and < 0 on any error | ||
| 1969 | */ | ||
| 1970 | static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1971 | *root, struct btrfs_path *path, struct btrfs_disk_key | ||
| 1972 | *key, u64 bytenr, int slot, int level) | ||
| 1973 | { | ||
| 1974 | struct extent_buffer *lower; | ||
| 1975 | int nritems; | ||
| 1976 | |||
| 1977 | BUG_ON(!path->nodes[level]); | ||
| 1978 | lower = path->nodes[level]; | ||
| 1979 | nritems = btrfs_header_nritems(lower); | ||
| 1980 | if (slot > nritems) | ||
| 1981 | BUG(); | ||
| 1982 | if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) | ||
| 1983 | BUG(); | ||
| 1984 | if (slot != nritems) { | ||
| 1985 | memmove_extent_buffer(lower, | ||
| 1986 | btrfs_node_key_ptr_offset(slot + 1), | ||
| 1987 | btrfs_node_key_ptr_offset(slot), | ||
| 1988 | (nritems - slot) * sizeof(struct btrfs_key_ptr)); | ||
| 1989 | } | ||
| 1990 | btrfs_set_node_key(lower, key, slot); | ||
| 1991 | btrfs_set_node_blockptr(lower, slot, bytenr); | ||
| 1992 | WARN_ON(trans->transid == 0); | ||
| 1993 | btrfs_set_node_ptr_generation(lower, slot, trans->transid); | ||
| 1994 | btrfs_set_header_nritems(lower, nritems + 1); | ||
| 1995 | btrfs_mark_buffer_dirty(lower); | ||
| 1996 | return 0; | ||
| 1997 | } | ||
| 1998 | |||
| 1999 | /* | ||
| 2000 | * split the node at the specified level in path in two. | ||
| 2001 | * The path is corrected to point to the appropriate node after the split | ||
| 2002 | * | ||
| 2003 | * Before splitting this tries to make some room in the node by pushing | ||
| 2004 | * left and right, if either one works, it returns right away. | ||
| 2005 | * | ||
| 2006 | * returns 0 on success and < 0 on failure | ||
| 2007 | */ | ||
| 2008 | static noinline int split_node(struct btrfs_trans_handle *trans, | ||
| 2009 | struct btrfs_root *root, | ||
| 2010 | struct btrfs_path *path, int level) | ||
| 2011 | { | ||
| 2012 | struct extent_buffer *c; | ||
| 2013 | struct extent_buffer *split; | ||
| 2014 | struct btrfs_disk_key disk_key; | ||
| 2015 | int mid; | ||
| 2016 | int ret; | ||
| 2017 | int wret; | ||
| 2018 | u32 c_nritems; | ||
| 2019 | |||
| 2020 | c = path->nodes[level]; | ||
| 2021 | WARN_ON(btrfs_header_generation(c) != trans->transid); | ||
| 2022 | if (c == root->node) { | ||
| 2023 | /* trying to split the root, lets make a new one */ | ||
| 2024 | ret = insert_new_root(trans, root, path, level + 1); | ||
| 2025 | if (ret) | ||
| 2026 | return ret; | ||
| 2027 | } else { | ||
| 2028 | ret = push_nodes_for_insert(trans, root, path, level); | ||
| 2029 | c = path->nodes[level]; | ||
| 2030 | if (!ret && btrfs_header_nritems(c) < | ||
| 2031 | BTRFS_NODEPTRS_PER_BLOCK(root) - 3) | ||
| 2032 | return 0; | ||
| 2033 | if (ret < 0) | ||
| 2034 | return ret; | ||
| 2035 | } | ||
| 2036 | |||
| 2037 | c_nritems = btrfs_header_nritems(c); | ||
| 2038 | |||
| 2039 | split = btrfs_alloc_free_block(trans, root, root->nodesize, | ||
| 2040 | path->nodes[level + 1]->start, | ||
| 2041 | root->root_key.objectid, | ||
| 2042 | trans->transid, level, c->start, 0); | ||
| 2043 | if (IS_ERR(split)) | ||
| 2044 | return PTR_ERR(split); | ||
| 2045 | |||
| 2046 | btrfs_set_header_flags(split, btrfs_header_flags(c)); | ||
| 2047 | btrfs_set_header_level(split, btrfs_header_level(c)); | ||
| 2048 | btrfs_set_header_bytenr(split, split->start); | ||
| 2049 | btrfs_set_header_generation(split, trans->transid); | ||
| 2050 | btrfs_set_header_owner(split, root->root_key.objectid); | ||
| 2051 | btrfs_set_header_flags(split, 0); | ||
| 2052 | write_extent_buffer(split, root->fs_info->fsid, | ||
| 2053 | (unsigned long)btrfs_header_fsid(split), | ||
| 2054 | BTRFS_FSID_SIZE); | ||
| 2055 | write_extent_buffer(split, root->fs_info->chunk_tree_uuid, | ||
| 2056 | (unsigned long)btrfs_header_chunk_tree_uuid(split), | ||
| 2057 | BTRFS_UUID_SIZE); | ||
| 2058 | |||
| 2059 | mid = (c_nritems + 1) / 2; | ||
| 2060 | |||
| 2061 | copy_extent_buffer(split, c, | ||
| 2062 | btrfs_node_key_ptr_offset(0), | ||
| 2063 | btrfs_node_key_ptr_offset(mid), | ||
| 2064 | (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); | ||
| 2065 | btrfs_set_header_nritems(split, c_nritems - mid); | ||
| 2066 | btrfs_set_header_nritems(c, mid); | ||
| 2067 | ret = 0; | ||
| 2068 | |||
| 2069 | btrfs_mark_buffer_dirty(c); | ||
| 2070 | btrfs_mark_buffer_dirty(split); | ||
| 2071 | |||
| 2072 | btrfs_node_key(split, &disk_key, 0); | ||
| 2073 | wret = insert_ptr(trans, root, path, &disk_key, split->start, | ||
| 2074 | path->slots[level + 1] + 1, | ||
| 2075 | level + 1); | ||
| 2076 | if (wret) | ||
| 2077 | ret = wret; | ||
| 2078 | |||
| 2079 | ret = btrfs_update_ref(trans, root, c, split, 0, c_nritems - mid); | ||
| 2080 | BUG_ON(ret); | ||
| 2081 | |||
| 2082 | if (path->slots[level] >= mid) { | ||
| 2083 | path->slots[level] -= mid; | ||
| 2084 | btrfs_tree_unlock(c); | ||
| 2085 | free_extent_buffer(c); | ||
| 2086 | path->nodes[level] = split; | ||
| 2087 | path->slots[level + 1] += 1; | ||
| 2088 | } else { | ||
| 2089 | btrfs_tree_unlock(split); | ||
| 2090 | free_extent_buffer(split); | ||
| 2091 | } | ||
| 2092 | return ret; | ||
| 2093 | } | ||
| 2094 | |||
| 2095 | /* | ||
| 2096 | * how many bytes are required to store the items in a leaf. start | ||
| 2097 | * and nr indicate which items in the leaf to check. This totals up the | ||
| 2098 | * space used both by the item structs and the item data | ||
| 2099 | */ | ||
| 2100 | static int leaf_space_used(struct extent_buffer *l, int start, int nr) | ||
| 2101 | { | ||
| 2102 | int data_len; | ||
| 2103 | int nritems = btrfs_header_nritems(l); | ||
| 2104 | int end = min(nritems, start + nr) - 1; | ||
| 2105 | |||
| 2106 | if (!nr) | ||
| 2107 | return 0; | ||
| 2108 | data_len = btrfs_item_end_nr(l, start); | ||
| 2109 | data_len = data_len - btrfs_item_offset_nr(l, end); | ||
| 2110 | data_len += sizeof(struct btrfs_item) * nr; | ||
| 2111 | WARN_ON(data_len < 0); | ||
| 2112 | return data_len; | ||
| 2113 | } | ||
| 2114 | |||
| 2115 | /* | ||
| 2116 | * The space between the end of the leaf items and | ||
| 2117 | * the start of the leaf data. IOW, how much room | ||
| 2118 | * the leaf has left for both items and data | ||
| 2119 | */ | ||
| 2120 | noinline int btrfs_leaf_free_space(struct btrfs_root *root, | ||
| 2121 | struct extent_buffer *leaf) | ||
| 2122 | { | ||
| 2123 | int nritems = btrfs_header_nritems(leaf); | ||
| 2124 | int ret; | ||
| 2125 | ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); | ||
| 2126 | if (ret < 0) { | ||
| 2127 | printk(KERN_CRIT "leaf free space ret %d, leaf data size %lu, " | ||
| 2128 | "used %d nritems %d\n", | ||
| 2129 | ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), | ||
| 2130 | leaf_space_used(leaf, 0, nritems), nritems); | ||
| 2131 | } | ||
| 2132 | return ret; | ||
| 2133 | } | ||
| 2134 | |||
| 2135 | /* | ||
| 2136 | * push some data in the path leaf to the right, trying to free up at | ||
| 2137 | * least data_size bytes. returns zero if the push worked, nonzero otherwise | ||
| 2138 | * | ||
| 2139 | * returns 1 if the push failed because the other node didn't have enough | ||
| 2140 | * room, 0 if everything worked out and < 0 if there were major errors. | ||
| 2141 | */ | ||
| 2142 | static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 2143 | *root, struct btrfs_path *path, int data_size, | ||
| 2144 | int empty) | ||
| 2145 | { | ||
| 2146 | struct extent_buffer *left = path->nodes[0]; | ||
| 2147 | struct extent_buffer *right; | ||
| 2148 | struct extent_buffer *upper; | ||
| 2149 | struct btrfs_disk_key disk_key; | ||
| 2150 | int slot; | ||
| 2151 | u32 i; | ||
| 2152 | int free_space; | ||
| 2153 | int push_space = 0; | ||
| 2154 | int push_items = 0; | ||
| 2155 | struct btrfs_item *item; | ||
| 2156 | u32 left_nritems; | ||
| 2157 | u32 nr; | ||
| 2158 | u32 right_nritems; | ||
| 2159 | u32 data_end; | ||
| 2160 | u32 this_item_size; | ||
| 2161 | int ret; | ||
| 2162 | |||
| 2163 | slot = path->slots[1]; | ||
| 2164 | if (!path->nodes[1]) | ||
| 2165 | return 1; | ||
| 2166 | |||
| 2167 | upper = path->nodes[1]; | ||
| 2168 | if (slot >= btrfs_header_nritems(upper) - 1) | ||
| 2169 | return 1; | ||
| 2170 | |||
| 2171 | WARN_ON(!btrfs_tree_locked(path->nodes[1])); | ||
| 2172 | |||
| 2173 | right = read_node_slot(root, upper, slot + 1); | ||
| 2174 | btrfs_tree_lock(right); | ||
| 2175 | free_space = btrfs_leaf_free_space(root, right); | ||
| 2176 | if (free_space < data_size) | ||
| 2177 | goto out_unlock; | ||
| 2178 | |||
| 2179 | /* cow and double check */ | ||
| 2180 | ret = btrfs_cow_block(trans, root, right, upper, | ||
| 2181 | slot + 1, &right, 0); | ||
| 2182 | if (ret) | ||
| 2183 | goto out_unlock; | ||
| 2184 | |||
| 2185 | free_space = btrfs_leaf_free_space(root, right); | ||
| 2186 | if (free_space < data_size) | ||
| 2187 | goto out_unlock; | ||
| 2188 | |||
| 2189 | left_nritems = btrfs_header_nritems(left); | ||
| 2190 | if (left_nritems == 0) | ||
| 2191 | goto out_unlock; | ||
| 2192 | |||
| 2193 | if (empty) | ||
| 2194 | nr = 0; | ||
| 2195 | else | ||
| 2196 | nr = 1; | ||
| 2197 | |||
| 2198 | if (path->slots[0] >= left_nritems) | ||
| 2199 | push_space += data_size; | ||
| 2200 | |||
| 2201 | i = left_nritems - 1; | ||
| 2202 | while (i >= nr) { | ||
| 2203 | item = btrfs_item_nr(left, i); | ||
| 2204 | |||
| 2205 | if (!empty && push_items > 0) { | ||
| 2206 | if (path->slots[0] > i) | ||
| 2207 | break; | ||
| 2208 | if (path->slots[0] == i) { | ||
| 2209 | int space = btrfs_leaf_free_space(root, left); | ||
| 2210 | if (space + push_space * 2 > free_space) | ||
| 2211 | break; | ||
| 2212 | } | ||
| 2213 | } | ||
| 2214 | |||
| 2215 | if (path->slots[0] == i) | ||
| 2216 | push_space += data_size; | ||
| 2217 | |||
| 2218 | if (!left->map_token) { | ||
| 2219 | map_extent_buffer(left, (unsigned long)item, | ||
| 2220 | sizeof(struct btrfs_item), | ||
| 2221 | &left->map_token, &left->kaddr, | ||
| 2222 | &left->map_start, &left->map_len, | ||
| 2223 | KM_USER1); | ||
| 2224 | } | ||
| 2225 | |||
| 2226 | this_item_size = btrfs_item_size(left, item); | ||
| 2227 | if (this_item_size + sizeof(*item) + push_space > free_space) | ||
| 2228 | break; | ||
| 2229 | |||
| 2230 | push_items++; | ||
| 2231 | push_space += this_item_size + sizeof(*item); | ||
| 2232 | if (i == 0) | ||
| 2233 | break; | ||
| 2234 | i--; | ||
| 2235 | } | ||
| 2236 | if (left->map_token) { | ||
| 2237 | unmap_extent_buffer(left, left->map_token, KM_USER1); | ||
| 2238 | left->map_token = NULL; | ||
| 2239 | } | ||
| 2240 | |||
| 2241 | if (push_items == 0) | ||
| 2242 | goto out_unlock; | ||
| 2243 | |||
| 2244 | if (!empty && push_items == left_nritems) | ||
| 2245 | WARN_ON(1); | ||
| 2246 | |||
| 2247 | /* push left to right */ | ||
| 2248 | right_nritems = btrfs_header_nritems(right); | ||
| 2249 | |||
| 2250 | push_space = btrfs_item_end_nr(left, left_nritems - push_items); | ||
| 2251 | push_space -= leaf_data_end(root, left); | ||
| 2252 | |||
| 2253 | /* make room in the right data area */ | ||
| 2254 | data_end = leaf_data_end(root, right); | ||
| 2255 | memmove_extent_buffer(right, | ||
| 2256 | btrfs_leaf_data(right) + data_end - push_space, | ||
| 2257 | btrfs_leaf_data(right) + data_end, | ||
| 2258 | BTRFS_LEAF_DATA_SIZE(root) - data_end); | ||
| 2259 | |||
| 2260 | /* copy from the left data area */ | ||
| 2261 | copy_extent_buffer(right, left, btrfs_leaf_data(right) + | ||
| 2262 | BTRFS_LEAF_DATA_SIZE(root) - push_space, | ||
| 2263 | btrfs_leaf_data(left) + leaf_data_end(root, left), | ||
| 2264 | push_space); | ||
| 2265 | |||
| 2266 | memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), | ||
| 2267 | btrfs_item_nr_offset(0), | ||
| 2268 | right_nritems * sizeof(struct btrfs_item)); | ||
| 2269 | |||
| 2270 | /* copy the items from left to right */ | ||
| 2271 | copy_extent_buffer(right, left, btrfs_item_nr_offset(0), | ||
| 2272 | btrfs_item_nr_offset(left_nritems - push_items), | ||
| 2273 | push_items * sizeof(struct btrfs_item)); | ||
| 2274 | |||
| 2275 | /* update the item pointers */ | ||
| 2276 | right_nritems += push_items; | ||
| 2277 | btrfs_set_header_nritems(right, right_nritems); | ||
| 2278 | push_space = BTRFS_LEAF_DATA_SIZE(root); | ||
| 2279 | for (i = 0; i < right_nritems; i++) { | ||
| 2280 | item = btrfs_item_nr(right, i); | ||
| 2281 | if (!right->map_token) { | ||
| 2282 | map_extent_buffer(right, (unsigned long)item, | ||
| 2283 | sizeof(struct btrfs_item), | ||
| 2284 | &right->map_token, &right->kaddr, | ||
| 2285 | &right->map_start, &right->map_len, | ||
| 2286 | KM_USER1); | ||
| 2287 | } | ||
| 2288 | push_space -= btrfs_item_size(right, item); | ||
| 2289 | btrfs_set_item_offset(right, item, push_space); | ||
| 2290 | } | ||
| 2291 | |||
| 2292 | if (right->map_token) { | ||
| 2293 | unmap_extent_buffer(right, right->map_token, KM_USER1); | ||
| 2294 | right->map_token = NULL; | ||
| 2295 | } | ||
| 2296 | left_nritems -= push_items; | ||
| 2297 | btrfs_set_header_nritems(left, left_nritems); | ||
| 2298 | |||
| 2299 | if (left_nritems) | ||
| 2300 | btrfs_mark_buffer_dirty(left); | ||
| 2301 | btrfs_mark_buffer_dirty(right); | ||
| 2302 | |||
| 2303 | ret = btrfs_update_ref(trans, root, left, right, 0, push_items); | ||
| 2304 | BUG_ON(ret); | ||
| 2305 | |||
| 2306 | btrfs_item_key(right, &disk_key, 0); | ||
| 2307 | btrfs_set_node_key(upper, &disk_key, slot + 1); | ||
| 2308 | btrfs_mark_buffer_dirty(upper); | ||
| 2309 | |||
| 2310 | /* then fixup the leaf pointer in the path */ | ||
| 2311 | if (path->slots[0] >= left_nritems) { | ||
| 2312 | path->slots[0] -= left_nritems; | ||
| 2313 | if (btrfs_header_nritems(path->nodes[0]) == 0) | ||
| 2314 | clean_tree_block(trans, root, path->nodes[0]); | ||
| 2315 | btrfs_tree_unlock(path->nodes[0]); | ||
| 2316 | free_extent_buffer(path->nodes[0]); | ||
| 2317 | path->nodes[0] = right; | ||
| 2318 | path->slots[1] += 1; | ||
| 2319 | } else { | ||
| 2320 | btrfs_tree_unlock(right); | ||
| 2321 | free_extent_buffer(right); | ||
| 2322 | } | ||
| 2323 | return 0; | ||
| 2324 | |||
| 2325 | out_unlock: | ||
| 2326 | btrfs_tree_unlock(right); | ||
| 2327 | free_extent_buffer(right); | ||
| 2328 | return 1; | ||
| 2329 | } | ||
| 2330 | |||
| 2331 | /* | ||
| 2332 | * push some data in the path leaf to the left, trying to free up at | ||
| 2333 | * least data_size bytes. returns zero if the push worked, nonzero otherwise | ||
| 2334 | */ | ||
| 2335 | static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 2336 | *root, struct btrfs_path *path, int data_size, | ||
| 2337 | int empty) | ||
| 2338 | { | ||
| 2339 | struct btrfs_disk_key disk_key; | ||
| 2340 | struct extent_buffer *right = path->nodes[0]; | ||
| 2341 | struct extent_buffer *left; | ||
| 2342 | int slot; | ||
| 2343 | int i; | ||
| 2344 | int free_space; | ||
| 2345 | int push_space = 0; | ||
| 2346 | int push_items = 0; | ||
| 2347 | struct btrfs_item *item; | ||
| 2348 | u32 old_left_nritems; | ||
| 2349 | u32 right_nritems; | ||
| 2350 | u32 nr; | ||
| 2351 | int ret = 0; | ||
| 2352 | int wret; | ||
| 2353 | u32 this_item_size; | ||
| 2354 | u32 old_left_item_size; | ||
| 2355 | |||
| 2356 | slot = path->slots[1]; | ||
| 2357 | if (slot == 0) | ||
| 2358 | return 1; | ||
| 2359 | if (!path->nodes[1]) | ||
| 2360 | return 1; | ||
| 2361 | |||
| 2362 | right_nritems = btrfs_header_nritems(right); | ||
| 2363 | if (right_nritems == 0) | ||
| 2364 | return 1; | ||
| 2365 | |||
| 2366 | WARN_ON(!btrfs_tree_locked(path->nodes[1])); | ||
| 2367 | |||
| 2368 | left = read_node_slot(root, path->nodes[1], slot - 1); | ||
| 2369 | btrfs_tree_lock(left); | ||
| 2370 | free_space = btrfs_leaf_free_space(root, left); | ||
| 2371 | if (free_space < data_size) { | ||
| 2372 | ret = 1; | ||
| 2373 | goto out; | ||
| 2374 | } | ||
| 2375 | |||
| 2376 | /* cow and double check */ | ||
| 2377 | ret = btrfs_cow_block(trans, root, left, | ||
| 2378 | path->nodes[1], slot - 1, &left, 0); | ||
| 2379 | if (ret) { | ||
| 2380 | /* we hit -ENOSPC, but it isn't fatal here */ | ||
| 2381 | ret = 1; | ||
| 2382 | goto out; | ||
| 2383 | } | ||
| 2384 | |||
| 2385 | free_space = btrfs_leaf_free_space(root, left); | ||
| 2386 | if (free_space < data_size) { | ||
| 2387 | ret = 1; | ||
| 2388 | goto out; | ||
| 2389 | } | ||
| 2390 | |||
| 2391 | if (empty) | ||
| 2392 | nr = right_nritems; | ||
| 2393 | else | ||
| 2394 | nr = right_nritems - 1; | ||
| 2395 | |||
| 2396 | for (i = 0; i < nr; i++) { | ||
| 2397 | item = btrfs_item_nr(right, i); | ||
| 2398 | if (!right->map_token) { | ||
| 2399 | map_extent_buffer(right, (unsigned long)item, | ||
| 2400 | sizeof(struct btrfs_item), | ||
| 2401 | &right->map_token, &right->kaddr, | ||
| 2402 | &right->map_start, &right->map_len, | ||
| 2403 | KM_USER1); | ||
| 2404 | } | ||
| 2405 | |||
| 2406 | if (!empty && push_items > 0) { | ||
| 2407 | if (path->slots[0] < i) | ||
| 2408 | break; | ||
| 2409 | if (path->slots[0] == i) { | ||
| 2410 | int space = btrfs_leaf_free_space(root, right); | ||
| 2411 | if (space + push_space * 2 > free_space) | ||
| 2412 | break; | ||
| 2413 | } | ||
| 2414 | } | ||
| 2415 | |||
| 2416 | if (path->slots[0] == i) | ||
| 2417 | push_space += data_size; | ||
| 2418 | |||
| 2419 | this_item_size = btrfs_item_size(right, item); | ||
| 2420 | if (this_item_size + sizeof(*item) + push_space > free_space) | ||
| 2421 | break; | ||
| 2422 | |||
| 2423 | push_items++; | ||
| 2424 | push_space += this_item_size + sizeof(*item); | ||
| 2425 | } | ||
| 2426 | |||
| 2427 | if (right->map_token) { | ||
| 2428 | unmap_extent_buffer(right, right->map_token, KM_USER1); | ||
| 2429 | right->map_token = NULL; | ||
| 2430 | } | ||
| 2431 | |||
| 2432 | if (push_items == 0) { | ||
| 2433 | ret = 1; | ||
| 2434 | goto out; | ||
| 2435 | } | ||
| 2436 | if (!empty && push_items == btrfs_header_nritems(right)) | ||
| 2437 | WARN_ON(1); | ||
| 2438 | |||
| 2439 | /* push data from right to left */ | ||
| 2440 | copy_extent_buffer(left, right, | ||
| 2441 | btrfs_item_nr_offset(btrfs_header_nritems(left)), | ||
| 2442 | btrfs_item_nr_offset(0), | ||
| 2443 | push_items * sizeof(struct btrfs_item)); | ||
| 2444 | |||
| 2445 | push_space = BTRFS_LEAF_DATA_SIZE(root) - | ||
| 2446 | btrfs_item_offset_nr(right, push_items - 1); | ||
| 2447 | |||
| 2448 | copy_extent_buffer(left, right, btrfs_leaf_data(left) + | ||
| 2449 | leaf_data_end(root, left) - push_space, | ||
| 2450 | btrfs_leaf_data(right) + | ||
| 2451 | btrfs_item_offset_nr(right, push_items - 1), | ||
| 2452 | push_space); | ||
| 2453 | old_left_nritems = btrfs_header_nritems(left); | ||
| 2454 | BUG_ON(old_left_nritems <= 0); | ||
| 2455 | |||
| 2456 | old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); | ||
| 2457 | for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { | ||
| 2458 | u32 ioff; | ||
| 2459 | |||
| 2460 | item = btrfs_item_nr(left, i); | ||
| 2461 | if (!left->map_token) { | ||
| 2462 | map_extent_buffer(left, (unsigned long)item, | ||
| 2463 | sizeof(struct btrfs_item), | ||
| 2464 | &left->map_token, &left->kaddr, | ||
| 2465 | &left->map_start, &left->map_len, | ||
| 2466 | KM_USER1); | ||
| 2467 | } | ||
| 2468 | |||
| 2469 | ioff = btrfs_item_offset(left, item); | ||
| 2470 | btrfs_set_item_offset(left, item, | ||
| 2471 | ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); | ||
| 2472 | } | ||
| 2473 | btrfs_set_header_nritems(left, old_left_nritems + push_items); | ||
| 2474 | if (left->map_token) { | ||
| 2475 | unmap_extent_buffer(left, left->map_token, KM_USER1); | ||
| 2476 | left->map_token = NULL; | ||
| 2477 | } | ||
| 2478 | |||
| 2479 | /* fixup right node */ | ||
| 2480 | if (push_items > right_nritems) { | ||
| 2481 | printk(KERN_CRIT "push items %d nr %u\n", push_items, | ||
| 2482 | right_nritems); | ||
| 2483 | WARN_ON(1); | ||
| 2484 | } | ||
| 2485 | |||
| 2486 | if (push_items < right_nritems) { | ||
| 2487 | push_space = btrfs_item_offset_nr(right, push_items - 1) - | ||
| 2488 | leaf_data_end(root, right); | ||
| 2489 | memmove_extent_buffer(right, btrfs_leaf_data(right) + | ||
| 2490 | BTRFS_LEAF_DATA_SIZE(root) - push_space, | ||
| 2491 | btrfs_leaf_data(right) + | ||
| 2492 | leaf_data_end(root, right), push_space); | ||
| 2493 | |||
| 2494 | memmove_extent_buffer(right, btrfs_item_nr_offset(0), | ||
| 2495 | btrfs_item_nr_offset(push_items), | ||
| 2496 | (btrfs_header_nritems(right) - push_items) * | ||
| 2497 | sizeof(struct btrfs_item)); | ||
| 2498 | } | ||
| 2499 | right_nritems -= push_items; | ||
| 2500 | btrfs_set_header_nritems(right, right_nritems); | ||
| 2501 | push_space = BTRFS_LEAF_DATA_SIZE(root); | ||
| 2502 | for (i = 0; i < right_nritems; i++) { | ||
| 2503 | item = btrfs_item_nr(right, i); | ||
| 2504 | |||
| 2505 | if (!right->map_token) { | ||
| 2506 | map_extent_buffer(right, (unsigned long)item, | ||
| 2507 | sizeof(struct btrfs_item), | ||
| 2508 | &right->map_token, &right->kaddr, | ||
| 2509 | &right->map_start, &right->map_len, | ||
| 2510 | KM_USER1); | ||
| 2511 | } | ||
| 2512 | |||
| 2513 | push_space = push_space - btrfs_item_size(right, item); | ||
| 2514 | btrfs_set_item_offset(right, item, push_space); | ||
| 2515 | } | ||
| 2516 | if (right->map_token) { | ||
| 2517 | unmap_extent_buffer(right, right->map_token, KM_USER1); | ||
| 2518 | right->map_token = NULL; | ||
| 2519 | } | ||
| 2520 | |||
| 2521 | btrfs_mark_buffer_dirty(left); | ||
| 2522 | if (right_nritems) | ||
| 2523 | btrfs_mark_buffer_dirty(right); | ||
| 2524 | |||
| 2525 | ret = btrfs_update_ref(trans, root, right, left, | ||
| 2526 | old_left_nritems, push_items); | ||
| 2527 | BUG_ON(ret); | ||
| 2528 | |||
| 2529 | btrfs_item_key(right, &disk_key, 0); | ||
| 2530 | wret = fixup_low_keys(trans, root, path, &disk_key, 1); | ||
| 2531 | if (wret) | ||
| 2532 | ret = wret; | ||
| 2533 | |||
| 2534 | /* then fixup the leaf pointer in the path */ | ||
| 2535 | if (path->slots[0] < push_items) { | ||
| 2536 | path->slots[0] += old_left_nritems; | ||
| 2537 | if (btrfs_header_nritems(path->nodes[0]) == 0) | ||
| 2538 | clean_tree_block(trans, root, path->nodes[0]); | ||
| 2539 | btrfs_tree_unlock(path->nodes[0]); | ||
| 2540 | free_extent_buffer(path->nodes[0]); | ||
| 2541 | path->nodes[0] = left; | ||
| 2542 | path->slots[1] -= 1; | ||
| 2543 | } else { | ||
| 2544 | btrfs_tree_unlock(left); | ||
| 2545 | free_extent_buffer(left); | ||
| 2546 | path->slots[0] -= push_items; | ||
| 2547 | } | ||
| 2548 | BUG_ON(path->slots[0] < 0); | ||
| 2549 | return ret; | ||
| 2550 | out: | ||
| 2551 | btrfs_tree_unlock(left); | ||
| 2552 | free_extent_buffer(left); | ||
| 2553 | return ret; | ||
| 2554 | } | ||
| 2555 | |||
| 2556 | /* | ||
| 2557 | * split the path's leaf in two, making sure there is at least data_size | ||
| 2558 | * available for the resulting leaf level of the path. | ||
| 2559 | * | ||
| 2560 | * returns 0 if all went well and < 0 on failure. | ||
| 2561 | */ | ||
| 2562 | static noinline int split_leaf(struct btrfs_trans_handle *trans, | ||
| 2563 | struct btrfs_root *root, | ||
| 2564 | struct btrfs_key *ins_key, | ||
| 2565 | struct btrfs_path *path, int data_size, | ||
| 2566 | int extend) | ||
| 2567 | { | ||
| 2568 | struct extent_buffer *l; | ||
| 2569 | u32 nritems; | ||
| 2570 | int mid; | ||
| 2571 | int slot; | ||
| 2572 | struct extent_buffer *right; | ||
| 2573 | int data_copy_size; | ||
| 2574 | int rt_data_off; | ||
| 2575 | int i; | ||
| 2576 | int ret = 0; | ||
| 2577 | int wret; | ||
| 2578 | int double_split; | ||
| 2579 | int num_doubles = 0; | ||
| 2580 | struct btrfs_disk_key disk_key; | ||
| 2581 | |||
| 2582 | /* first try to make some room by pushing left and right */ | ||
| 2583 | if (data_size && ins_key->type != BTRFS_DIR_ITEM_KEY) { | ||
| 2584 | wret = push_leaf_right(trans, root, path, data_size, 0); | ||
| 2585 | if (wret < 0) | ||
| 2586 | return wret; | ||
| 2587 | if (wret) { | ||
| 2588 | wret = push_leaf_left(trans, root, path, data_size, 0); | ||
| 2589 | if (wret < 0) | ||
| 2590 | return wret; | ||
| 2591 | } | ||
| 2592 | l = path->nodes[0]; | ||
| 2593 | |||
| 2594 | /* did the pushes work? */ | ||
| 2595 | if (btrfs_leaf_free_space(root, l) >= data_size) | ||
| 2596 | return 0; | ||
| 2597 | } | ||
| 2598 | |||
| 2599 | if (!path->nodes[1]) { | ||
| 2600 | ret = insert_new_root(trans, root, path, 1); | ||
| 2601 | if (ret) | ||
| 2602 | return ret; | ||
| 2603 | } | ||
| 2604 | again: | ||
| 2605 | double_split = 0; | ||
| 2606 | l = path->nodes[0]; | ||
| 2607 | slot = path->slots[0]; | ||
| 2608 | nritems = btrfs_header_nritems(l); | ||
| 2609 | mid = (nritems + 1) / 2; | ||
| 2610 | |||
| 2611 | right = btrfs_alloc_free_block(trans, root, root->leafsize, | ||
| 2612 | path->nodes[1]->start, | ||
| 2613 | root->root_key.objectid, | ||
| 2614 | trans->transid, 0, l->start, 0); | ||
| 2615 | if (IS_ERR(right)) { | ||
| 2616 | BUG_ON(1); | ||
| 2617 | return PTR_ERR(right); | ||
| 2618 | } | ||
| 2619 | |||
| 2620 | memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); | ||
| 2621 | btrfs_set_header_bytenr(right, right->start); | ||
| 2622 | btrfs_set_header_generation(right, trans->transid); | ||
| 2623 | btrfs_set_header_owner(right, root->root_key.objectid); | ||
| 2624 | btrfs_set_header_level(right, 0); | ||
| 2625 | write_extent_buffer(right, root->fs_info->fsid, | ||
| 2626 | (unsigned long)btrfs_header_fsid(right), | ||
| 2627 | BTRFS_FSID_SIZE); | ||
| 2628 | |||
| 2629 | write_extent_buffer(right, root->fs_info->chunk_tree_uuid, | ||
| 2630 | (unsigned long)btrfs_header_chunk_tree_uuid(right), | ||
| 2631 | BTRFS_UUID_SIZE); | ||
| 2632 | if (mid <= slot) { | ||
| 2633 | if (nritems == 1 || | ||
| 2634 | leaf_space_used(l, mid, nritems - mid) + data_size > | ||
| 2635 | BTRFS_LEAF_DATA_SIZE(root)) { | ||
| 2636 | if (slot >= nritems) { | ||
| 2637 | btrfs_cpu_key_to_disk(&disk_key, ins_key); | ||
| 2638 | btrfs_set_header_nritems(right, 0); | ||
| 2639 | wret = insert_ptr(trans, root, path, | ||
| 2640 | &disk_key, right->start, | ||
| 2641 | path->slots[1] + 1, 1); | ||
| 2642 | if (wret) | ||
| 2643 | ret = wret; | ||
| 2644 | |||
| 2645 | btrfs_tree_unlock(path->nodes[0]); | ||
| 2646 | free_extent_buffer(path->nodes[0]); | ||
| 2647 | path->nodes[0] = right; | ||
| 2648 | path->slots[0] = 0; | ||
| 2649 | path->slots[1] += 1; | ||
| 2650 | btrfs_mark_buffer_dirty(right); | ||
| 2651 | return ret; | ||
| 2652 | } | ||
| 2653 | mid = slot; | ||
| 2654 | if (mid != nritems && | ||
| 2655 | leaf_space_used(l, mid, nritems - mid) + | ||
| 2656 | data_size > BTRFS_LEAF_DATA_SIZE(root)) { | ||
| 2657 | double_split = 1; | ||
| 2658 | } | ||
| 2659 | } | ||
| 2660 | } else { | ||
| 2661 | if (leaf_space_used(l, 0, mid) + data_size > | ||
| 2662 | BTRFS_LEAF_DATA_SIZE(root)) { | ||
| 2663 | if (!extend && data_size && slot == 0) { | ||
| 2664 | btrfs_cpu_key_to_disk(&disk_key, ins_key); | ||
| 2665 | btrfs_set_header_nritems(right, 0); | ||
| 2666 | wret = insert_ptr(trans, root, path, | ||
| 2667 | &disk_key, | ||
| 2668 | right->start, | ||
| 2669 | path->slots[1], 1); | ||
| 2670 | if (wret) | ||
| 2671 | ret = wret; | ||
| 2672 | btrfs_tree_unlock(path->nodes[0]); | ||
| 2673 | free_extent_buffer(path->nodes[0]); | ||
| 2674 | path->nodes[0] = right; | ||
| 2675 | path->slots[0] = 0; | ||
| 2676 | if (path->slots[1] == 0) { | ||
| 2677 | wret = fixup_low_keys(trans, root, | ||
| 2678 | path, &disk_key, 1); | ||
| 2679 | if (wret) | ||
| 2680 | ret = wret; | ||
| 2681 | } | ||
| 2682 | btrfs_mark_buffer_dirty(right); | ||
| 2683 | return ret; | ||
| 2684 | } else if ((extend || !data_size) && slot == 0) { | ||
| 2685 | mid = 1; | ||
| 2686 | } else { | ||
| 2687 | mid = slot; | ||
| 2688 | if (mid != nritems && | ||
| 2689 | leaf_space_used(l, mid, nritems - mid) + | ||
| 2690 | data_size > BTRFS_LEAF_DATA_SIZE(root)) { | ||
| 2691 | double_split = 1; | ||
| 2692 | } | ||
| 2693 | } | ||
| 2694 | } | ||
| 2695 | } | ||
| 2696 | nritems = nritems - mid; | ||
| 2697 | btrfs_set_header_nritems(right, nritems); | ||
| 2698 | data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); | ||
| 2699 | |||
| 2700 | copy_extent_buffer(right, l, btrfs_item_nr_offset(0), | ||
| 2701 | btrfs_item_nr_offset(mid), | ||
| 2702 | nritems * sizeof(struct btrfs_item)); | ||
| 2703 | |||
| 2704 | copy_extent_buffer(right, l, | ||
| 2705 | btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - | ||
| 2706 | data_copy_size, btrfs_leaf_data(l) + | ||
| 2707 | leaf_data_end(root, l), data_copy_size); | ||
| 2708 | |||
| 2709 | rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - | ||
| 2710 | btrfs_item_end_nr(l, mid); | ||
| 2711 | |||
| 2712 | for (i = 0; i < nritems; i++) { | ||
| 2713 | struct btrfs_item *item = btrfs_item_nr(right, i); | ||
| 2714 | u32 ioff; | ||
| 2715 | |||
| 2716 | if (!right->map_token) { | ||
| 2717 | map_extent_buffer(right, (unsigned long)item, | ||
| 2718 | sizeof(struct btrfs_item), | ||
| 2719 | &right->map_token, &right->kaddr, | ||
| 2720 | &right->map_start, &right->map_len, | ||
| 2721 | KM_USER1); | ||
| 2722 | } | ||
| 2723 | |||
| 2724 | ioff = btrfs_item_offset(right, item); | ||
| 2725 | btrfs_set_item_offset(right, item, ioff + rt_data_off); | ||
| 2726 | } | ||
| 2727 | |||
| 2728 | if (right->map_token) { | ||
| 2729 | unmap_extent_buffer(right, right->map_token, KM_USER1); | ||
| 2730 | right->map_token = NULL; | ||
| 2731 | } | ||
| 2732 | |||
| 2733 | btrfs_set_header_nritems(l, mid); | ||
| 2734 | ret = 0; | ||
| 2735 | btrfs_item_key(right, &disk_key, 0); | ||
| 2736 | wret = insert_ptr(trans, root, path, &disk_key, right->start, | ||
| 2737 | path->slots[1] + 1, 1); | ||
| 2738 | if (wret) | ||
| 2739 | ret = wret; | ||
| 2740 | |||
| 2741 | btrfs_mark_buffer_dirty(right); | ||
| 2742 | btrfs_mark_buffer_dirty(l); | ||
| 2743 | BUG_ON(path->slots[0] != slot); | ||
| 2744 | |||
| 2745 | ret = btrfs_update_ref(trans, root, l, right, 0, nritems); | ||
| 2746 | BUG_ON(ret); | ||
| 2747 | |||
| 2748 | if (mid <= slot) { | ||
| 2749 | btrfs_tree_unlock(path->nodes[0]); | ||
| 2750 | free_extent_buffer(path->nodes[0]); | ||
| 2751 | path->nodes[0] = right; | ||
| 2752 | path->slots[0] -= mid; | ||
| 2753 | path->slots[1] += 1; | ||
| 2754 | } else { | ||
| 2755 | btrfs_tree_unlock(right); | ||
| 2756 | free_extent_buffer(right); | ||
| 2757 | } | ||
| 2758 | |||
| 2759 | BUG_ON(path->slots[0] < 0); | ||
| 2760 | |||
| 2761 | if (double_split) { | ||
| 2762 | BUG_ON(num_doubles != 0); | ||
| 2763 | num_doubles++; | ||
| 2764 | goto again; | ||
| 2765 | } | ||
| 2766 | return ret; | ||
| 2767 | } | ||
| 2768 | |||
| 2769 | /* | ||
| 2770 | * This function splits a single item into two items, | ||
| 2771 | * giving 'new_key' to the new item and splitting the | ||
| 2772 | * old one at split_offset (from the start of the item). | ||
| 2773 | * | ||
| 2774 | * The path may be released by this operation. After | ||
| 2775 | * the split, the path is pointing to the old item. The | ||
| 2776 | * new item is going to be in the same node as the old one. | ||
| 2777 | * | ||
| 2778 | * Note, the item being split must be smaller enough to live alone on | ||
| 2779 | * a tree block with room for one extra struct btrfs_item | ||
| 2780 | * | ||
| 2781 | * This allows us to split the item in place, keeping a lock on the | ||
| 2782 | * leaf the entire time. | ||
| 2783 | */ | ||
| 2784 | int btrfs_split_item(struct btrfs_trans_handle *trans, | ||
| 2785 | struct btrfs_root *root, | ||
| 2786 | struct btrfs_path *path, | ||
| 2787 | struct btrfs_key *new_key, | ||
| 2788 | unsigned long split_offset) | ||
| 2789 | { | ||
| 2790 | u32 item_size; | ||
| 2791 | struct extent_buffer *leaf; | ||
| 2792 | struct btrfs_key orig_key; | ||
| 2793 | struct btrfs_item *item; | ||
| 2794 | struct btrfs_item *new_item; | ||
| 2795 | int ret = 0; | ||
| 2796 | int slot; | ||
| 2797 | u32 nritems; | ||
| 2798 | u32 orig_offset; | ||
| 2799 | struct btrfs_disk_key disk_key; | ||
| 2800 | char *buf; | ||
| 2801 | |||
| 2802 | leaf = path->nodes[0]; | ||
| 2803 | btrfs_item_key_to_cpu(leaf, &orig_key, path->slots[0]); | ||
| 2804 | if (btrfs_leaf_free_space(root, leaf) >= sizeof(struct btrfs_item)) | ||
| 2805 | goto split; | ||
| 2806 | |||
| 2807 | item_size = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 2808 | btrfs_release_path(root, path); | ||
| 2809 | |||
| 2810 | path->search_for_split = 1; | ||
| 2811 | path->keep_locks = 1; | ||
| 2812 | |||
| 2813 | ret = btrfs_search_slot(trans, root, &orig_key, path, 0, 1); | ||
| 2814 | path->search_for_split = 0; | ||
| 2815 | |||
| 2816 | /* if our item isn't there or got smaller, return now */ | ||
| 2817 | if (ret != 0 || item_size != btrfs_item_size_nr(path->nodes[0], | ||
| 2818 | path->slots[0])) { | ||
| 2819 | path->keep_locks = 0; | ||
| 2820 | return -EAGAIN; | ||
| 2821 | } | ||
| 2822 | |||
| 2823 | ret = split_leaf(trans, root, &orig_key, path, | ||
| 2824 | sizeof(struct btrfs_item), 1); | ||
| 2825 | path->keep_locks = 0; | ||
| 2826 | BUG_ON(ret); | ||
| 2827 | |||
| 2828 | leaf = path->nodes[0]; | ||
| 2829 | BUG_ON(btrfs_leaf_free_space(root, leaf) < sizeof(struct btrfs_item)); | ||
| 2830 | |||
| 2831 | split: | ||
| 2832 | item = btrfs_item_nr(leaf, path->slots[0]); | ||
| 2833 | orig_offset = btrfs_item_offset(leaf, item); | ||
| 2834 | item_size = btrfs_item_size(leaf, item); | ||
| 2835 | |||
| 2836 | |||
| 2837 | buf = kmalloc(item_size, GFP_NOFS); | ||
| 2838 | read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf, | ||
| 2839 | path->slots[0]), item_size); | ||
| 2840 | slot = path->slots[0] + 1; | ||
| 2841 | leaf = path->nodes[0]; | ||
| 2842 | |||
| 2843 | nritems = btrfs_header_nritems(leaf); | ||
| 2844 | |||
| 2845 | if (slot != nritems) { | ||
| 2846 | /* shift the items */ | ||
| 2847 | memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1), | ||
| 2848 | btrfs_item_nr_offset(slot), | ||
| 2849 | (nritems - slot) * sizeof(struct btrfs_item)); | ||
| 2850 | |||
| 2851 | } | ||
| 2852 | |||
| 2853 | btrfs_cpu_key_to_disk(&disk_key, new_key); | ||
| 2854 | btrfs_set_item_key(leaf, &disk_key, slot); | ||
| 2855 | |||
| 2856 | new_item = btrfs_item_nr(leaf, slot); | ||
| 2857 | |||
| 2858 | btrfs_set_item_offset(leaf, new_item, orig_offset); | ||
| 2859 | btrfs_set_item_size(leaf, new_item, item_size - split_offset); | ||
| 2860 | |||
| 2861 | btrfs_set_item_offset(leaf, item, | ||
| 2862 | orig_offset + item_size - split_offset); | ||
| 2863 | btrfs_set_item_size(leaf, item, split_offset); | ||
| 2864 | |||
| 2865 | btrfs_set_header_nritems(leaf, nritems + 1); | ||
| 2866 | |||
| 2867 | /* write the data for the start of the original item */ | ||
| 2868 | write_extent_buffer(leaf, buf, | ||
| 2869 | btrfs_item_ptr_offset(leaf, path->slots[0]), | ||
| 2870 | split_offset); | ||
| 2871 | |||
| 2872 | /* write the data for the new item */ | ||
| 2873 | write_extent_buffer(leaf, buf + split_offset, | ||
| 2874 | btrfs_item_ptr_offset(leaf, slot), | ||
| 2875 | item_size - split_offset); | ||
| 2876 | btrfs_mark_buffer_dirty(leaf); | ||
| 2877 | |||
| 2878 | ret = 0; | ||
| 2879 | if (btrfs_leaf_free_space(root, leaf) < 0) { | ||
| 2880 | btrfs_print_leaf(root, leaf); | ||
| 2881 | BUG(); | ||
| 2882 | } | ||
| 2883 | kfree(buf); | ||
| 2884 | return ret; | ||
| 2885 | } | ||
| 2886 | |||
| 2887 | /* | ||
| 2888 | * make the item pointed to by the path smaller. new_size indicates | ||
| 2889 | * how small to make it, and from_end tells us if we just chop bytes | ||
| 2890 | * off the end of the item or if we shift the item to chop bytes off | ||
| 2891 | * the front. | ||
| 2892 | */ | ||
| 2893 | int btrfs_truncate_item(struct btrfs_trans_handle *trans, | ||
| 2894 | struct btrfs_root *root, | ||
| 2895 | struct btrfs_path *path, | ||
| 2896 | u32 new_size, int from_end) | ||
| 2897 | { | ||
| 2898 | int ret = 0; | ||
| 2899 | int slot; | ||
| 2900 | int slot_orig; | ||
| 2901 | struct extent_buffer *leaf; | ||
| 2902 | struct btrfs_item *item; | ||
| 2903 | u32 nritems; | ||
| 2904 | unsigned int data_end; | ||
| 2905 | unsigned int old_data_start; | ||
| 2906 | unsigned int old_size; | ||
| 2907 | unsigned int size_diff; | ||
| 2908 | int i; | ||
| 2909 | |||
| 2910 | slot_orig = path->slots[0]; | ||
| 2911 | leaf = path->nodes[0]; | ||
| 2912 | slot = path->slots[0]; | ||
| 2913 | |||
| 2914 | old_size = btrfs_item_size_nr(leaf, slot); | ||
| 2915 | if (old_size == new_size) | ||
| 2916 | return 0; | ||
| 2917 | |||
| 2918 | nritems = btrfs_header_nritems(leaf); | ||
| 2919 | data_end = leaf_data_end(root, leaf); | ||
| 2920 | |||
| 2921 | old_data_start = btrfs_item_offset_nr(leaf, slot); | ||
| 2922 | |||
| 2923 | size_diff = old_size - new_size; | ||
| 2924 | |||
| 2925 | BUG_ON(slot < 0); | ||
| 2926 | BUG_ON(slot >= nritems); | ||
| 2927 | |||
| 2928 | /* | ||
| 2929 | * item0..itemN ... dataN.offset..dataN.size .. data0.size | ||
| 2930 | */ | ||
| 2931 | /* first correct the data pointers */ | ||
| 2932 | for (i = slot; i < nritems; i++) { | ||
| 2933 | u32 ioff; | ||
| 2934 | item = btrfs_item_nr(leaf, i); | ||
| 2935 | |||
| 2936 | if (!leaf->map_token) { | ||
| 2937 | map_extent_buffer(leaf, (unsigned long)item, | ||
| 2938 | sizeof(struct btrfs_item), | ||
| 2939 | &leaf->map_token, &leaf->kaddr, | ||
| 2940 | &leaf->map_start, &leaf->map_len, | ||
| 2941 | KM_USER1); | ||
| 2942 | } | ||
| 2943 | |||
| 2944 | ioff = btrfs_item_offset(leaf, item); | ||
| 2945 | btrfs_set_item_offset(leaf, item, ioff + size_diff); | ||
| 2946 | } | ||
| 2947 | |||
| 2948 | if (leaf->map_token) { | ||
| 2949 | unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); | ||
| 2950 | leaf->map_token = NULL; | ||
| 2951 | } | ||
| 2952 | |||
| 2953 | /* shift the data */ | ||
| 2954 | if (from_end) { | ||
| 2955 | memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + | ||
| 2956 | data_end + size_diff, btrfs_leaf_data(leaf) + | ||
| 2957 | data_end, old_data_start + new_size - data_end); | ||
| 2958 | } else { | ||
| 2959 | struct btrfs_disk_key disk_key; | ||
| 2960 | u64 offset; | ||
| 2961 | |||
| 2962 | btrfs_item_key(leaf, &disk_key, slot); | ||
| 2963 | |||
| 2964 | if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { | ||
| 2965 | unsigned long ptr; | ||
| 2966 | struct btrfs_file_extent_item *fi; | ||
| 2967 | |||
| 2968 | fi = btrfs_item_ptr(leaf, slot, | ||
| 2969 | struct btrfs_file_extent_item); | ||
| 2970 | fi = (struct btrfs_file_extent_item *)( | ||
| 2971 | (unsigned long)fi - size_diff); | ||
| 2972 | |||
| 2973 | if (btrfs_file_extent_type(leaf, fi) == | ||
| 2974 | BTRFS_FILE_EXTENT_INLINE) { | ||
| 2975 | ptr = btrfs_item_ptr_offset(leaf, slot); | ||
| 2976 | memmove_extent_buffer(leaf, ptr, | ||
| 2977 | (unsigned long)fi, | ||
| 2978 | offsetof(struct btrfs_file_extent_item, | ||
| 2979 | disk_bytenr)); | ||
| 2980 | } | ||
| 2981 | } | ||
| 2982 | |||
| 2983 | memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + | ||
| 2984 | data_end + size_diff, btrfs_leaf_data(leaf) + | ||
| 2985 | data_end, old_data_start - data_end); | ||
| 2986 | |||
| 2987 | offset = btrfs_disk_key_offset(&disk_key); | ||
| 2988 | btrfs_set_disk_key_offset(&disk_key, offset + size_diff); | ||
| 2989 | btrfs_set_item_key(leaf, &disk_key, slot); | ||
| 2990 | if (slot == 0) | ||
| 2991 | fixup_low_keys(trans, root, path, &disk_key, 1); | ||
| 2992 | } | ||
| 2993 | |||
| 2994 | item = btrfs_item_nr(leaf, slot); | ||
| 2995 | btrfs_set_item_size(leaf, item, new_size); | ||
| 2996 | btrfs_mark_buffer_dirty(leaf); | ||
| 2997 | |||
| 2998 | ret = 0; | ||
| 2999 | if (btrfs_leaf_free_space(root, leaf) < 0) { | ||
| 3000 | btrfs_print_leaf(root, leaf); | ||
| 3001 | BUG(); | ||
| 3002 | } | ||
| 3003 | return ret; | ||
| 3004 | } | ||
| 3005 | |||
| 3006 | /* | ||
| 3007 | * make the item pointed to by the path bigger, data_size is the new size. | ||
| 3008 | */ | ||
| 3009 | int btrfs_extend_item(struct btrfs_trans_handle *trans, | ||
| 3010 | struct btrfs_root *root, struct btrfs_path *path, | ||
| 3011 | u32 data_size) | ||
| 3012 | { | ||
| 3013 | int ret = 0; | ||
| 3014 | int slot; | ||
| 3015 | int slot_orig; | ||
| 3016 | struct extent_buffer *leaf; | ||
| 3017 | struct btrfs_item *item; | ||
| 3018 | u32 nritems; | ||
| 3019 | unsigned int data_end; | ||
| 3020 | unsigned int old_data; | ||
| 3021 | unsigned int old_size; | ||
| 3022 | int i; | ||
| 3023 | |||
| 3024 | slot_orig = path->slots[0]; | ||
| 3025 | leaf = path->nodes[0]; | ||
| 3026 | |||
| 3027 | nritems = btrfs_header_nritems(leaf); | ||
| 3028 | data_end = leaf_data_end(root, leaf); | ||
| 3029 | |||
| 3030 | if (btrfs_leaf_free_space(root, leaf) < data_size) { | ||
| 3031 | btrfs_print_leaf(root, leaf); | ||
| 3032 | BUG(); | ||
| 3033 | } | ||
| 3034 | slot = path->slots[0]; | ||
| 3035 | old_data = btrfs_item_end_nr(leaf, slot); | ||
| 3036 | |||
| 3037 | BUG_ON(slot < 0); | ||
| 3038 | if (slot >= nritems) { | ||
| 3039 | btrfs_print_leaf(root, leaf); | ||
| 3040 | printk(KERN_CRIT "slot %d too large, nritems %d\n", | ||
| 3041 | slot, nritems); | ||
| 3042 | BUG_ON(1); | ||
| 3043 | } | ||
| 3044 | |||
| 3045 | /* | ||
| 3046 | * item0..itemN ... dataN.offset..dataN.size .. data0.size | ||
| 3047 | */ | ||
| 3048 | /* first correct the data pointers */ | ||
| 3049 | for (i = slot; i < nritems; i++) { | ||
| 3050 | u32 ioff; | ||
| 3051 | item = btrfs_item_nr(leaf, i); | ||
| 3052 | |||
| 3053 | if (!leaf->map_token) { | ||
| 3054 | map_extent_buffer(leaf, (unsigned long)item, | ||
| 3055 | sizeof(struct btrfs_item), | ||
| 3056 | &leaf->map_token, &leaf->kaddr, | ||
| 3057 | &leaf->map_start, &leaf->map_len, | ||
| 3058 | KM_USER1); | ||
| 3059 | } | ||
| 3060 | ioff = btrfs_item_offset(leaf, item); | ||
| 3061 | btrfs_set_item_offset(leaf, item, ioff - data_size); | ||
| 3062 | } | ||
| 3063 | |||
| 3064 | if (leaf->map_token) { | ||
| 3065 | unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); | ||
| 3066 | leaf->map_token = NULL; | ||
| 3067 | } | ||
| 3068 | |||
| 3069 | /* shift the data */ | ||
| 3070 | memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + | ||
| 3071 | data_end - data_size, btrfs_leaf_data(leaf) + | ||
| 3072 | data_end, old_data - data_end); | ||
| 3073 | |||
| 3074 | data_end = old_data; | ||
| 3075 | old_size = btrfs_item_size_nr(leaf, slot); | ||
| 3076 | item = btrfs_item_nr(leaf, slot); | ||
| 3077 | btrfs_set_item_size(leaf, item, old_size + data_size); | ||
| 3078 | btrfs_mark_buffer_dirty(leaf); | ||
| 3079 | |||
| 3080 | ret = 0; | ||
| 3081 | if (btrfs_leaf_free_space(root, leaf) < 0) { | ||
| 3082 | btrfs_print_leaf(root, leaf); | ||
| 3083 | BUG(); | ||
| 3084 | } | ||
| 3085 | return ret; | ||
| 3086 | } | ||
| 3087 | |||
| 3088 | /* | ||
| 3089 | * Given a key and some data, insert items into the tree. | ||
| 3090 | * This does all the path init required, making room in the tree if needed. | ||
| 3091 | * Returns the number of keys that were inserted. | ||
| 3092 | */ | ||
| 3093 | int btrfs_insert_some_items(struct btrfs_trans_handle *trans, | ||
| 3094 | struct btrfs_root *root, | ||
| 3095 | struct btrfs_path *path, | ||
| 3096 | struct btrfs_key *cpu_key, u32 *data_size, | ||
| 3097 | int nr) | ||
| 3098 | { | ||
| 3099 | struct extent_buffer *leaf; | ||
| 3100 | struct btrfs_item *item; | ||
| 3101 | int ret = 0; | ||
| 3102 | int slot; | ||
| 3103 | int i; | ||
| 3104 | u32 nritems; | ||
| 3105 | u32 total_data = 0; | ||
| 3106 | u32 total_size = 0; | ||
| 3107 | unsigned int data_end; | ||
| 3108 | struct btrfs_disk_key disk_key; | ||
| 3109 | struct btrfs_key found_key; | ||
| 3110 | |||
| 3111 | for (i = 0; i < nr; i++) { | ||
| 3112 | if (total_size + data_size[i] + sizeof(struct btrfs_item) > | ||
| 3113 | BTRFS_LEAF_DATA_SIZE(root)) { | ||
| 3114 | break; | ||
| 3115 | nr = i; | ||
| 3116 | } | ||
| 3117 | total_data += data_size[i]; | ||
| 3118 | total_size += data_size[i] + sizeof(struct btrfs_item); | ||
| 3119 | } | ||
| 3120 | BUG_ON(nr == 0); | ||
| 3121 | |||
| 3122 | ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); | ||
| 3123 | if (ret == 0) | ||
| 3124 | return -EEXIST; | ||
| 3125 | if (ret < 0) | ||
| 3126 | goto out; | ||
| 3127 | |||
| 3128 | leaf = path->nodes[0]; | ||
| 3129 | |||
| 3130 | nritems = btrfs_header_nritems(leaf); | ||
| 3131 | data_end = leaf_data_end(root, leaf); | ||
| 3132 | |||
| 3133 | if (btrfs_leaf_free_space(root, leaf) < total_size) { | ||
| 3134 | for (i = nr; i >= 0; i--) { | ||
| 3135 | total_data -= data_size[i]; | ||
| 3136 | total_size -= data_size[i] + sizeof(struct btrfs_item); | ||
| 3137 | if (total_size < btrfs_leaf_free_space(root, leaf)) | ||
| 3138 | break; | ||
| 3139 | } | ||
| 3140 | nr = i; | ||
| 3141 | } | ||
| 3142 | |||
| 3143 | slot = path->slots[0]; | ||
| 3144 | BUG_ON(slot < 0); | ||
| 3145 | |||
| 3146 | if (slot != nritems) { | ||
| 3147 | unsigned int old_data = btrfs_item_end_nr(leaf, slot); | ||
| 3148 | |||
| 3149 | item = btrfs_item_nr(leaf, slot); | ||
| 3150 | btrfs_item_key_to_cpu(leaf, &found_key, slot); | ||
| 3151 | |||
| 3152 | /* figure out how many keys we can insert in here */ | ||
| 3153 | total_data = data_size[0]; | ||
| 3154 | for (i = 1; i < nr; i++) { | ||
| 3155 | if (comp_cpu_keys(&found_key, cpu_key + i) <= 0) | ||
| 3156 | break; | ||
| 3157 | total_data += data_size[i]; | ||
| 3158 | } | ||
| 3159 | nr = i; | ||
| 3160 | |||
| 3161 | if (old_data < data_end) { | ||
| 3162 | btrfs_print_leaf(root, leaf); | ||
| 3163 | printk(KERN_CRIT "slot %d old_data %d data_end %d\n", | ||
| 3164 | slot, old_data, data_end); | ||
| 3165 | BUG_ON(1); | ||
| 3166 | } | ||
| 3167 | /* | ||
| 3168 | * item0..itemN ... dataN.offset..dataN.size .. data0.size | ||
| 3169 | */ | ||
| 3170 | /* first correct the data pointers */ | ||
| 3171 | WARN_ON(leaf->map_token); | ||
| 3172 | for (i = slot; i < nritems; i++) { | ||
| 3173 | u32 ioff; | ||
| 3174 | |||
| 3175 | item = btrfs_item_nr(leaf, i); | ||
| 3176 | if (!leaf->map_token) { | ||
| 3177 | map_extent_buffer(leaf, (unsigned long)item, | ||
| 3178 | sizeof(struct btrfs_item), | ||
| 3179 | &leaf->map_token, &leaf->kaddr, | ||
| 3180 | &leaf->map_start, &leaf->map_len, | ||
| 3181 | KM_USER1); | ||
| 3182 | } | ||
| 3183 | |||
| 3184 | ioff = btrfs_item_offset(leaf, item); | ||
| 3185 | btrfs_set_item_offset(leaf, item, ioff - total_data); | ||
| 3186 | } | ||
| 3187 | if (leaf->map_token) { | ||
| 3188 | unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); | ||
| 3189 | leaf->map_token = NULL; | ||
| 3190 | } | ||
| 3191 | |||
| 3192 | /* shift the items */ | ||
| 3193 | memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), | ||
| 3194 | btrfs_item_nr_offset(slot), | ||
| 3195 | (nritems - slot) * sizeof(struct btrfs_item)); | ||
| 3196 | |||
| 3197 | /* shift the data */ | ||
| 3198 | memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + | ||
| 3199 | data_end - total_data, btrfs_leaf_data(leaf) + | ||
| 3200 | data_end, old_data - data_end); | ||
| 3201 | data_end = old_data; | ||
| 3202 | } else { | ||
| 3203 | /* | ||
| 3204 | * this sucks but it has to be done, if we are inserting at | ||
| 3205 | * the end of the leaf only insert 1 of the items, since we | ||
| 3206 | * have no way of knowing whats on the next leaf and we'd have | ||
| 3207 | * to drop our current locks to figure it out | ||
| 3208 | */ | ||
| 3209 | nr = 1; | ||
| 3210 | } | ||
| 3211 | |||
| 3212 | /* setup the item for the new data */ | ||
| 3213 | for (i = 0; i < nr; i++) { | ||
| 3214 | btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); | ||
| 3215 | btrfs_set_item_key(leaf, &disk_key, slot + i); | ||
| 3216 | item = btrfs_item_nr(leaf, slot + i); | ||
| 3217 | btrfs_set_item_offset(leaf, item, data_end - data_size[i]); | ||
| 3218 | data_end -= data_size[i]; | ||
| 3219 | btrfs_set_item_size(leaf, item, data_size[i]); | ||
| 3220 | } | ||
| 3221 | btrfs_set_header_nritems(leaf, nritems + nr); | ||
| 3222 | btrfs_mark_buffer_dirty(leaf); | ||
| 3223 | |||
| 3224 | ret = 0; | ||
| 3225 | if (slot == 0) { | ||
| 3226 | btrfs_cpu_key_to_disk(&disk_key, cpu_key); | ||
| 3227 | ret = fixup_low_keys(trans, root, path, &disk_key, 1); | ||
| 3228 | } | ||
| 3229 | |||
| 3230 | if (btrfs_leaf_free_space(root, leaf) < 0) { | ||
| 3231 | btrfs_print_leaf(root, leaf); | ||
| 3232 | BUG(); | ||
| 3233 | } | ||
| 3234 | out: | ||
| 3235 | if (!ret) | ||
| 3236 | ret = nr; | ||
| 3237 | return ret; | ||
| 3238 | } | ||
| 3239 | |||
| 3240 | /* | ||
| 3241 | * Given a key and some data, insert items into the tree. | ||
| 3242 | * This does all the path init required, making room in the tree if needed. | ||
| 3243 | */ | ||
| 3244 | int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, | ||
| 3245 | struct btrfs_root *root, | ||
| 3246 | struct btrfs_path *path, | ||
| 3247 | struct btrfs_key *cpu_key, u32 *data_size, | ||
| 3248 | int nr) | ||
| 3249 | { | ||
| 3250 | struct extent_buffer *leaf; | ||
| 3251 | struct btrfs_item *item; | ||
| 3252 | int ret = 0; | ||
| 3253 | int slot; | ||
| 3254 | int slot_orig; | ||
| 3255 | int i; | ||
| 3256 | u32 nritems; | ||
| 3257 | u32 total_size = 0; | ||
| 3258 | u32 total_data = 0; | ||
| 3259 | unsigned int data_end; | ||
| 3260 | struct btrfs_disk_key disk_key; | ||
| 3261 | |||
| 3262 | for (i = 0; i < nr; i++) | ||
| 3263 | total_data += data_size[i]; | ||
| 3264 | |||
| 3265 | total_size = total_data + (nr * sizeof(struct btrfs_item)); | ||
| 3266 | ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); | ||
| 3267 | if (ret == 0) | ||
| 3268 | return -EEXIST; | ||
| 3269 | if (ret < 0) | ||
| 3270 | goto out; | ||
| 3271 | |||
| 3272 | slot_orig = path->slots[0]; | ||
| 3273 | leaf = path->nodes[0]; | ||
| 3274 | |||
| 3275 | nritems = btrfs_header_nritems(leaf); | ||
| 3276 | data_end = leaf_data_end(root, leaf); | ||
| 3277 | |||
| 3278 | if (btrfs_leaf_free_space(root, leaf) < total_size) { | ||
| 3279 | btrfs_print_leaf(root, leaf); | ||
| 3280 | printk(KERN_CRIT "not enough freespace need %u have %d\n", | ||
| 3281 | total_size, btrfs_leaf_free_space(root, leaf)); | ||
| 3282 | BUG(); | ||
| 3283 | } | ||
| 3284 | |||
| 3285 | slot = path->slots[0]; | ||
| 3286 | BUG_ON(slot < 0); | ||
| 3287 | |||
| 3288 | if (slot != nritems) { | ||
| 3289 | unsigned int old_data = btrfs_item_end_nr(leaf, slot); | ||
| 3290 | |||
| 3291 | if (old_data < data_end) { | ||
| 3292 | btrfs_print_leaf(root, leaf); | ||
| 3293 | printk(KERN_CRIT "slot %d old_data %d data_end %d\n", | ||
| 3294 | slot, old_data, data_end); | ||
| 3295 | BUG_ON(1); | ||
| 3296 | } | ||
| 3297 | /* | ||
| 3298 | * item0..itemN ... dataN.offset..dataN.size .. data0.size | ||
| 3299 | */ | ||
| 3300 | /* first correct the data pointers */ | ||
| 3301 | WARN_ON(leaf->map_token); | ||
| 3302 | for (i = slot; i < nritems; i++) { | ||
| 3303 | u32 ioff; | ||
| 3304 | |||
| 3305 | item = btrfs_item_nr(leaf, i); | ||
| 3306 | if (!leaf->map_token) { | ||
| 3307 | map_extent_buffer(leaf, (unsigned long)item, | ||
| 3308 | sizeof(struct btrfs_item), | ||
| 3309 | &leaf->map_token, &leaf->kaddr, | ||
| 3310 | &leaf->map_start, &leaf->map_len, | ||
| 3311 | KM_USER1); | ||
| 3312 | } | ||
| 3313 | |||
| 3314 | ioff = btrfs_item_offset(leaf, item); | ||
| 3315 | btrfs_set_item_offset(leaf, item, ioff - total_data); | ||
| 3316 | } | ||
| 3317 | if (leaf->map_token) { | ||
| 3318 | unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); | ||
| 3319 | leaf->map_token = NULL; | ||
| 3320 | } | ||
| 3321 | |||
| 3322 | /* shift the items */ | ||
| 3323 | memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), | ||
| 3324 | btrfs_item_nr_offset(slot), | ||
| 3325 | (nritems - slot) * sizeof(struct btrfs_item)); | ||
| 3326 | |||
| 3327 | /* shift the data */ | ||
| 3328 | memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + | ||
| 3329 | data_end - total_data, btrfs_leaf_data(leaf) + | ||
| 3330 | data_end, old_data - data_end); | ||
| 3331 | data_end = old_data; | ||
| 3332 | } | ||
| 3333 | |||
| 3334 | /* setup the item for the new data */ | ||
| 3335 | for (i = 0; i < nr; i++) { | ||
| 3336 | btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); | ||
| 3337 | btrfs_set_item_key(leaf, &disk_key, slot + i); | ||
| 3338 | item = btrfs_item_nr(leaf, slot + i); | ||
| 3339 | btrfs_set_item_offset(leaf, item, data_end - data_size[i]); | ||
| 3340 | data_end -= data_size[i]; | ||
| 3341 | btrfs_set_item_size(leaf, item, data_size[i]); | ||
| 3342 | } | ||
| 3343 | btrfs_set_header_nritems(leaf, nritems + nr); | ||
| 3344 | btrfs_mark_buffer_dirty(leaf); | ||
| 3345 | |||
| 3346 | ret = 0; | ||
| 3347 | if (slot == 0) { | ||
| 3348 | btrfs_cpu_key_to_disk(&disk_key, cpu_key); | ||
| 3349 | ret = fixup_low_keys(trans, root, path, &disk_key, 1); | ||
| 3350 | } | ||
| 3351 | |||
| 3352 | if (btrfs_leaf_free_space(root, leaf) < 0) { | ||
| 3353 | btrfs_print_leaf(root, leaf); | ||
| 3354 | BUG(); | ||
| 3355 | } | ||
| 3356 | out: | ||
| 3357 | return ret; | ||
| 3358 | } | ||
| 3359 | |||
| 3360 | /* | ||
| 3361 | * Given a key and some data, insert an item into the tree. | ||
| 3362 | * This does all the path init required, making room in the tree if needed. | ||
| 3363 | */ | ||
| 3364 | int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 3365 | *root, struct btrfs_key *cpu_key, void *data, u32 | ||
| 3366 | data_size) | ||
| 3367 | { | ||
| 3368 | int ret = 0; | ||
| 3369 | struct btrfs_path *path; | ||
| 3370 | struct extent_buffer *leaf; | ||
| 3371 | unsigned long ptr; | ||
| 3372 | |||
| 3373 | path = btrfs_alloc_path(); | ||
| 3374 | BUG_ON(!path); | ||
| 3375 | ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); | ||
| 3376 | if (!ret) { | ||
| 3377 | leaf = path->nodes[0]; | ||
| 3378 | ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); | ||
| 3379 | write_extent_buffer(leaf, data, ptr, data_size); | ||
| 3380 | btrfs_mark_buffer_dirty(leaf); | ||
| 3381 | } | ||
| 3382 | btrfs_free_path(path); | ||
| 3383 | return ret; | ||
| 3384 | } | ||
| 3385 | |||
| 3386 | /* | ||
| 3387 | * delete the pointer from a given node. | ||
| 3388 | * | ||
| 3389 | * the tree should have been previously balanced so the deletion does not | ||
| 3390 | * empty a node. | ||
| 3391 | */ | ||
| 3392 | static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 3393 | struct btrfs_path *path, int level, int slot) | ||
| 3394 | { | ||
| 3395 | struct extent_buffer *parent = path->nodes[level]; | ||
| 3396 | u32 nritems; | ||
| 3397 | int ret = 0; | ||
| 3398 | int wret; | ||
| 3399 | |||
| 3400 | nritems = btrfs_header_nritems(parent); | ||
| 3401 | if (slot != nritems - 1) { | ||
| 3402 | memmove_extent_buffer(parent, | ||
| 3403 | btrfs_node_key_ptr_offset(slot), | ||
| 3404 | btrfs_node_key_ptr_offset(slot + 1), | ||
| 3405 | sizeof(struct btrfs_key_ptr) * | ||
| 3406 | (nritems - slot - 1)); | ||
| 3407 | } | ||
| 3408 | nritems--; | ||
| 3409 | btrfs_set_header_nritems(parent, nritems); | ||
| 3410 | if (nritems == 0 && parent == root->node) { | ||
| 3411 | BUG_ON(btrfs_header_level(root->node) != 1); | ||
| 3412 | /* just turn the root into a leaf and break */ | ||
| 3413 | btrfs_set_header_level(root->node, 0); | ||
| 3414 | } else if (slot == 0) { | ||
| 3415 | struct btrfs_disk_key disk_key; | ||
| 3416 | |||
| 3417 | btrfs_node_key(parent, &disk_key, 0); | ||
| 3418 | wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); | ||
| 3419 | if (wret) | ||
| 3420 | ret = wret; | ||
| 3421 | } | ||
| 3422 | btrfs_mark_buffer_dirty(parent); | ||
| 3423 | return ret; | ||
| 3424 | } | ||
| 3425 | |||
| 3426 | /* | ||
| 3427 | * a helper function to delete the leaf pointed to by path->slots[1] and | ||
| 3428 | * path->nodes[1]. bytenr is the node block pointer, but since the callers | ||
| 3429 | * already know it, it is faster to have them pass it down than to | ||
| 3430 | * read it out of the node again. | ||
| 3431 | * | ||
| 3432 | * This deletes the pointer in path->nodes[1] and frees the leaf | ||
| 3433 | * block extent. zero is returned if it all worked out, < 0 otherwise. | ||
| 3434 | * | ||
| 3435 | * The path must have already been setup for deleting the leaf, including | ||
| 3436 | * all the proper balancing. path->nodes[1] must be locked. | ||
| 3437 | */ | ||
| 3438 | noinline int btrfs_del_leaf(struct btrfs_trans_handle *trans, | ||
| 3439 | struct btrfs_root *root, | ||
| 3440 | struct btrfs_path *path, u64 bytenr) | ||
| 3441 | { | ||
| 3442 | int ret; | ||
| 3443 | u64 root_gen = btrfs_header_generation(path->nodes[1]); | ||
| 3444 | |||
| 3445 | ret = del_ptr(trans, root, path, 1, path->slots[1]); | ||
| 3446 | if (ret) | ||
| 3447 | return ret; | ||
| 3448 | |||
| 3449 | ret = btrfs_free_extent(trans, root, bytenr, | ||
| 3450 | btrfs_level_size(root, 0), | ||
| 3451 | path->nodes[1]->start, | ||
| 3452 | btrfs_header_owner(path->nodes[1]), | ||
| 3453 | root_gen, 0, 1); | ||
| 3454 | return ret; | ||
| 3455 | } | ||
| 3456 | /* | ||
| 3457 | * delete the item at the leaf level in path. If that empties | ||
| 3458 | * the leaf, remove it from the tree | ||
| 3459 | */ | ||
| 3460 | int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 3461 | struct btrfs_path *path, int slot, int nr) | ||
| 3462 | { | ||
| 3463 | struct extent_buffer *leaf; | ||
| 3464 | struct btrfs_item *item; | ||
| 3465 | int last_off; | ||
| 3466 | int dsize = 0; | ||
| 3467 | int ret = 0; | ||
| 3468 | int wret; | ||
| 3469 | int i; | ||
| 3470 | u32 nritems; | ||
| 3471 | |||
| 3472 | leaf = path->nodes[0]; | ||
| 3473 | last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); | ||
| 3474 | |||
| 3475 | for (i = 0; i < nr; i++) | ||
| 3476 | dsize += btrfs_item_size_nr(leaf, slot + i); | ||
| 3477 | |||
| 3478 | nritems = btrfs_header_nritems(leaf); | ||
| 3479 | |||
| 3480 | if (slot + nr != nritems) { | ||
| 3481 | int data_end = leaf_data_end(root, leaf); | ||
| 3482 | |||
| 3483 | memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + | ||
| 3484 | data_end + dsize, | ||
| 3485 | btrfs_leaf_data(leaf) + data_end, | ||
| 3486 | last_off - data_end); | ||
| 3487 | |||
| 3488 | for (i = slot + nr; i < nritems; i++) { | ||
| 3489 | u32 ioff; | ||
| 3490 | |||
| 3491 | item = btrfs_item_nr(leaf, i); | ||
| 3492 | if (!leaf->map_token) { | ||
| 3493 | map_extent_buffer(leaf, (unsigned long)item, | ||
| 3494 | sizeof(struct btrfs_item), | ||
| 3495 | &leaf->map_token, &leaf->kaddr, | ||
| 3496 | &leaf->map_start, &leaf->map_len, | ||
| 3497 | KM_USER1); | ||
| 3498 | } | ||
| 3499 | ioff = btrfs_item_offset(leaf, item); | ||
| 3500 | btrfs_set_item_offset(leaf, item, ioff + dsize); | ||
| 3501 | } | ||
| 3502 | |||
| 3503 | if (leaf->map_token) { | ||
| 3504 | unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); | ||
| 3505 | leaf->map_token = NULL; | ||
| 3506 | } | ||
| 3507 | |||
| 3508 | memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), | ||
| 3509 | btrfs_item_nr_offset(slot + nr), | ||
| 3510 | sizeof(struct btrfs_item) * | ||
| 3511 | (nritems - slot - nr)); | ||
| 3512 | } | ||
| 3513 | btrfs_set_header_nritems(leaf, nritems - nr); | ||
| 3514 | nritems -= nr; | ||
| 3515 | |||
| 3516 | /* delete the leaf if we've emptied it */ | ||
| 3517 | if (nritems == 0) { | ||
| 3518 | if (leaf == root->node) { | ||
| 3519 | btrfs_set_header_level(leaf, 0); | ||
| 3520 | } else { | ||
| 3521 | ret = btrfs_del_leaf(trans, root, path, leaf->start); | ||
| 3522 | BUG_ON(ret); | ||
| 3523 | } | ||
| 3524 | } else { | ||
| 3525 | int used = leaf_space_used(leaf, 0, nritems); | ||
| 3526 | if (slot == 0) { | ||
| 3527 | struct btrfs_disk_key disk_key; | ||
| 3528 | |||
| 3529 | btrfs_item_key(leaf, &disk_key, 0); | ||
| 3530 | wret = fixup_low_keys(trans, root, path, | ||
| 3531 | &disk_key, 1); | ||
| 3532 | if (wret) | ||
| 3533 | ret = wret; | ||
| 3534 | } | ||
| 3535 | |||
| 3536 | /* delete the leaf if it is mostly empty */ | ||
| 3537 | if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) { | ||
| 3538 | /* push_leaf_left fixes the path. | ||
| 3539 | * make sure the path still points to our leaf | ||
| 3540 | * for possible call to del_ptr below | ||
| 3541 | */ | ||
| 3542 | slot = path->slots[1]; | ||
| 3543 | extent_buffer_get(leaf); | ||
| 3544 | |||
| 3545 | wret = push_leaf_left(trans, root, path, 1, 1); | ||
| 3546 | if (wret < 0 && wret != -ENOSPC) | ||
| 3547 | ret = wret; | ||
| 3548 | |||
| 3549 | if (path->nodes[0] == leaf && | ||
| 3550 | btrfs_header_nritems(leaf)) { | ||
| 3551 | wret = push_leaf_right(trans, root, path, 1, 1); | ||
| 3552 | if (wret < 0 && wret != -ENOSPC) | ||
| 3553 | ret = wret; | ||
| 3554 | } | ||
| 3555 | |||
| 3556 | if (btrfs_header_nritems(leaf) == 0) { | ||
| 3557 | path->slots[1] = slot; | ||
| 3558 | ret = btrfs_del_leaf(trans, root, path, | ||
| 3559 | leaf->start); | ||
| 3560 | BUG_ON(ret); | ||
| 3561 | free_extent_buffer(leaf); | ||
| 3562 | } else { | ||
| 3563 | /* if we're still in the path, make sure | ||
| 3564 | * we're dirty. Otherwise, one of the | ||
| 3565 | * push_leaf functions must have already | ||
| 3566 | * dirtied this buffer | ||
| 3567 | */ | ||
| 3568 | if (path->nodes[0] == leaf) | ||
| 3569 | btrfs_mark_buffer_dirty(leaf); | ||
| 3570 | free_extent_buffer(leaf); | ||
| 3571 | } | ||
| 3572 | } else { | ||
| 3573 | btrfs_mark_buffer_dirty(leaf); | ||
| 3574 | } | ||
| 3575 | } | ||
| 3576 | return ret; | ||
| 3577 | } | ||
| 3578 | |||
| 3579 | /* | ||
| 3580 | * search the tree again to find a leaf with lesser keys | ||
| 3581 | * returns 0 if it found something or 1 if there are no lesser leaves. | ||
| 3582 | * returns < 0 on io errors. | ||
| 3583 | * | ||
| 3584 | * This may release the path, and so you may lose any locks held at the | ||
| 3585 | * time you call it. | ||
| 3586 | */ | ||
| 3587 | int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) | ||
| 3588 | { | ||
| 3589 | struct btrfs_key key; | ||
| 3590 | struct btrfs_disk_key found_key; | ||
| 3591 | int ret; | ||
| 3592 | |||
| 3593 | btrfs_item_key_to_cpu(path->nodes[0], &key, 0); | ||
| 3594 | |||
| 3595 | if (key.offset > 0) | ||
| 3596 | key.offset--; | ||
| 3597 | else if (key.type > 0) | ||
| 3598 | key.type--; | ||
| 3599 | else if (key.objectid > 0) | ||
| 3600 | key.objectid--; | ||
| 3601 | else | ||
| 3602 | return 1; | ||
| 3603 | |||
| 3604 | btrfs_release_path(root, path); | ||
| 3605 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 3606 | if (ret < 0) | ||
| 3607 | return ret; | ||
| 3608 | btrfs_item_key(path->nodes[0], &found_key, 0); | ||
| 3609 | ret = comp_keys(&found_key, &key); | ||
| 3610 | if (ret < 0) | ||
| 3611 | return 0; | ||
| 3612 | return 1; | ||
| 3613 | } | ||
| 3614 | |||
| 3615 | /* | ||
| 3616 | * A helper function to walk down the tree starting at min_key, and looking | ||
| 3617 | * for nodes or leaves that are either in cache or have a minimum | ||
| 3618 | * transaction id. This is used by the btree defrag code, and tree logging | ||
| 3619 | * | ||
| 3620 | * This does not cow, but it does stuff the starting key it finds back | ||
| 3621 | * into min_key, so you can call btrfs_search_slot with cow=1 on the | ||
| 3622 | * key and get a writable path. | ||
| 3623 | * | ||
| 3624 | * This does lock as it descends, and path->keep_locks should be set | ||
| 3625 | * to 1 by the caller. | ||
| 3626 | * | ||
| 3627 | * This honors path->lowest_level to prevent descent past a given level | ||
| 3628 | * of the tree. | ||
| 3629 | * | ||
| 3630 | * min_trans indicates the oldest transaction that you are interested | ||
| 3631 | * in walking through. Any nodes or leaves older than min_trans are | ||
| 3632 | * skipped over (without reading them). | ||
| 3633 | * | ||
| 3634 | * returns zero if something useful was found, < 0 on error and 1 if there | ||
| 3635 | * was nothing in the tree that matched the search criteria. | ||
| 3636 | */ | ||
| 3637 | int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key, | ||
| 3638 | struct btrfs_key *max_key, | ||
| 3639 | struct btrfs_path *path, int cache_only, | ||
| 3640 | u64 min_trans) | ||
| 3641 | { | ||
| 3642 | struct extent_buffer *cur; | ||
| 3643 | struct btrfs_key found_key; | ||
| 3644 | int slot; | ||
| 3645 | int sret; | ||
| 3646 | u32 nritems; | ||
| 3647 | int level; | ||
| 3648 | int ret = 1; | ||
| 3649 | |||
| 3650 | WARN_ON(!path->keep_locks); | ||
| 3651 | again: | ||
| 3652 | cur = btrfs_lock_root_node(root); | ||
| 3653 | level = btrfs_header_level(cur); | ||
| 3654 | WARN_ON(path->nodes[level]); | ||
| 3655 | path->nodes[level] = cur; | ||
| 3656 | path->locks[level] = 1; | ||
| 3657 | |||
| 3658 | if (btrfs_header_generation(cur) < min_trans) { | ||
| 3659 | ret = 1; | ||
| 3660 | goto out; | ||
| 3661 | } | ||
| 3662 | while (1) { | ||
| 3663 | nritems = btrfs_header_nritems(cur); | ||
| 3664 | level = btrfs_header_level(cur); | ||
| 3665 | sret = bin_search(cur, min_key, level, &slot); | ||
| 3666 | |||
| 3667 | /* at the lowest level, we're done, setup the path and exit */ | ||
| 3668 | if (level == path->lowest_level) { | ||
| 3669 | if (slot >= nritems) | ||
| 3670 | goto find_next_key; | ||
| 3671 | ret = 0; | ||
| 3672 | path->slots[level] = slot; | ||
| 3673 | btrfs_item_key_to_cpu(cur, &found_key, slot); | ||
| 3674 | goto out; | ||
| 3675 | } | ||
| 3676 | if (sret && slot > 0) | ||
| 3677 | slot--; | ||
| 3678 | /* | ||
| 3679 | * check this node pointer against the cache_only and | ||
| 3680 | * min_trans parameters. If it isn't in cache or is too | ||
| 3681 | * old, skip to the next one. | ||
| 3682 | */ | ||
| 3683 | while (slot < nritems) { | ||
| 3684 | u64 blockptr; | ||
| 3685 | u64 gen; | ||
| 3686 | struct extent_buffer *tmp; | ||
| 3687 | struct btrfs_disk_key disk_key; | ||
| 3688 | |||
| 3689 | blockptr = btrfs_node_blockptr(cur, slot); | ||
| 3690 | gen = btrfs_node_ptr_generation(cur, slot); | ||
| 3691 | if (gen < min_trans) { | ||
| 3692 | slot++; | ||
| 3693 | continue; | ||
| 3694 | } | ||
| 3695 | if (!cache_only) | ||
| 3696 | break; | ||
| 3697 | |||
| 3698 | if (max_key) { | ||
| 3699 | btrfs_node_key(cur, &disk_key, slot); | ||
| 3700 | if (comp_keys(&disk_key, max_key) >= 0) { | ||
| 3701 | ret = 1; | ||
| 3702 | goto out; | ||
| 3703 | } | ||
| 3704 | } | ||
| 3705 | |||
| 3706 | tmp = btrfs_find_tree_block(root, blockptr, | ||
| 3707 | btrfs_level_size(root, level - 1)); | ||
| 3708 | |||
| 3709 | if (tmp && btrfs_buffer_uptodate(tmp, gen)) { | ||
| 3710 | free_extent_buffer(tmp); | ||
| 3711 | break; | ||
| 3712 | } | ||
| 3713 | if (tmp) | ||
| 3714 | free_extent_buffer(tmp); | ||
| 3715 | slot++; | ||
| 3716 | } | ||
| 3717 | find_next_key: | ||
| 3718 | /* | ||
| 3719 | * we didn't find a candidate key in this node, walk forward | ||
| 3720 | * and find another one | ||
| 3721 | */ | ||
| 3722 | if (slot >= nritems) { | ||
| 3723 | path->slots[level] = slot; | ||
| 3724 | sret = btrfs_find_next_key(root, path, min_key, level, | ||
| 3725 | cache_only, min_trans); | ||
| 3726 | if (sret == 0) { | ||
| 3727 | btrfs_release_path(root, path); | ||
| 3728 | goto again; | ||
| 3729 | } else { | ||
| 3730 | goto out; | ||
| 3731 | } | ||
| 3732 | } | ||
| 3733 | /* save our key for returning back */ | ||
| 3734 | btrfs_node_key_to_cpu(cur, &found_key, slot); | ||
| 3735 | path->slots[level] = slot; | ||
| 3736 | if (level == path->lowest_level) { | ||
| 3737 | ret = 0; | ||
| 3738 | unlock_up(path, level, 1); | ||
| 3739 | goto out; | ||
| 3740 | } | ||
| 3741 | cur = read_node_slot(root, cur, slot); | ||
| 3742 | |||
| 3743 | btrfs_tree_lock(cur); | ||
| 3744 | path->locks[level - 1] = 1; | ||
| 3745 | path->nodes[level - 1] = cur; | ||
| 3746 | unlock_up(path, level, 1); | ||
| 3747 | } | ||
| 3748 | out: | ||
| 3749 | if (ret == 0) | ||
| 3750 | memcpy(min_key, &found_key, sizeof(found_key)); | ||
| 3751 | return ret; | ||
| 3752 | } | ||
| 3753 | |||
| 3754 | /* | ||
| 3755 | * this is similar to btrfs_next_leaf, but does not try to preserve | ||
| 3756 | * and fixup the path. It looks for and returns the next key in the | ||
| 3757 | * tree based on the current path and the cache_only and min_trans | ||
| 3758 | * parameters. | ||
| 3759 | * | ||
| 3760 | * 0 is returned if another key is found, < 0 if there are any errors | ||
| 3761 | * and 1 is returned if there are no higher keys in the tree | ||
| 3762 | * | ||
| 3763 | * path->keep_locks should be set to 1 on the search made before | ||
| 3764 | * calling this function. | ||
| 3765 | */ | ||
| 3766 | int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path, | ||
| 3767 | struct btrfs_key *key, int lowest_level, | ||
| 3768 | int cache_only, u64 min_trans) | ||
| 3769 | { | ||
| 3770 | int level = lowest_level; | ||
| 3771 | int slot; | ||
| 3772 | struct extent_buffer *c; | ||
| 3773 | |||
| 3774 | WARN_ON(!path->keep_locks); | ||
| 3775 | while (level < BTRFS_MAX_LEVEL) { | ||
| 3776 | if (!path->nodes[level]) | ||
| 3777 | return 1; | ||
| 3778 | |||
| 3779 | slot = path->slots[level] + 1; | ||
| 3780 | c = path->nodes[level]; | ||
| 3781 | next: | ||
| 3782 | if (slot >= btrfs_header_nritems(c)) { | ||
| 3783 | level++; | ||
| 3784 | if (level == BTRFS_MAX_LEVEL) | ||
| 3785 | return 1; | ||
| 3786 | continue; | ||
| 3787 | } | ||
| 3788 | if (level == 0) | ||
| 3789 | btrfs_item_key_to_cpu(c, key, slot); | ||
| 3790 | else { | ||
| 3791 | u64 blockptr = btrfs_node_blockptr(c, slot); | ||
| 3792 | u64 gen = btrfs_node_ptr_generation(c, slot); | ||
| 3793 | |||
| 3794 | if (cache_only) { | ||
| 3795 | struct extent_buffer *cur; | ||
| 3796 | cur = btrfs_find_tree_block(root, blockptr, | ||
| 3797 | btrfs_level_size(root, level - 1)); | ||
| 3798 | if (!cur || !btrfs_buffer_uptodate(cur, gen)) { | ||
| 3799 | slot++; | ||
| 3800 | if (cur) | ||
| 3801 | free_extent_buffer(cur); | ||
| 3802 | goto next; | ||
| 3803 | } | ||
| 3804 | free_extent_buffer(cur); | ||
| 3805 | } | ||
| 3806 | if (gen < min_trans) { | ||
| 3807 | slot++; | ||
| 3808 | goto next; | ||
| 3809 | } | ||
| 3810 | btrfs_node_key_to_cpu(c, key, slot); | ||
| 3811 | } | ||
| 3812 | return 0; | ||
| 3813 | } | ||
| 3814 | return 1; | ||
| 3815 | } | ||
| 3816 | |||
| 3817 | /* | ||
| 3818 | * search the tree again to find a leaf with greater keys | ||
| 3819 | * returns 0 if it found something or 1 if there are no greater leaves. | ||
| 3820 | * returns < 0 on io errors. | ||
| 3821 | */ | ||
| 3822 | int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) | ||
| 3823 | { | ||
| 3824 | int slot; | ||
| 3825 | int level = 1; | ||
| 3826 | struct extent_buffer *c; | ||
| 3827 | struct extent_buffer *next = NULL; | ||
| 3828 | struct btrfs_key key; | ||
| 3829 | u32 nritems; | ||
| 3830 | int ret; | ||
| 3831 | |||
| 3832 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 3833 | if (nritems == 0) | ||
| 3834 | return 1; | ||
| 3835 | |||
| 3836 | btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1); | ||
| 3837 | |||
| 3838 | btrfs_release_path(root, path); | ||
| 3839 | path->keep_locks = 1; | ||
| 3840 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 3841 | path->keep_locks = 0; | ||
| 3842 | |||
| 3843 | if (ret < 0) | ||
| 3844 | return ret; | ||
| 3845 | |||
| 3846 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 3847 | /* | ||
| 3848 | * by releasing the path above we dropped all our locks. A balance | ||
| 3849 | * could have added more items next to the key that used to be | ||
| 3850 | * at the very end of the block. So, check again here and | ||
| 3851 | * advance the path if there are now more items available. | ||
| 3852 | */ | ||
| 3853 | if (nritems > 0 && path->slots[0] < nritems - 1) { | ||
| 3854 | path->slots[0]++; | ||
| 3855 | goto done; | ||
| 3856 | } | ||
| 3857 | |||
| 3858 | while (level < BTRFS_MAX_LEVEL) { | ||
| 3859 | if (!path->nodes[level]) | ||
| 3860 | return 1; | ||
| 3861 | |||
| 3862 | slot = path->slots[level] + 1; | ||
| 3863 | c = path->nodes[level]; | ||
| 3864 | if (slot >= btrfs_header_nritems(c)) { | ||
| 3865 | level++; | ||
| 3866 | if (level == BTRFS_MAX_LEVEL) | ||
| 3867 | return 1; | ||
| 3868 | continue; | ||
| 3869 | } | ||
| 3870 | |||
| 3871 | if (next) { | ||
| 3872 | btrfs_tree_unlock(next); | ||
| 3873 | free_extent_buffer(next); | ||
| 3874 | } | ||
| 3875 | |||
| 3876 | if (level == 1 && (path->locks[1] || path->skip_locking) && | ||
| 3877 | path->reada) | ||
| 3878 | reada_for_search(root, path, level, slot, 0); | ||
| 3879 | |||
| 3880 | next = read_node_slot(root, c, slot); | ||
| 3881 | if (!path->skip_locking) { | ||
| 3882 | WARN_ON(!btrfs_tree_locked(c)); | ||
| 3883 | btrfs_tree_lock(next); | ||
| 3884 | } | ||
| 3885 | break; | ||
| 3886 | } | ||
| 3887 | path->slots[level] = slot; | ||
| 3888 | while (1) { | ||
| 3889 | level--; | ||
| 3890 | c = path->nodes[level]; | ||
| 3891 | if (path->locks[level]) | ||
| 3892 | btrfs_tree_unlock(c); | ||
| 3893 | free_extent_buffer(c); | ||
| 3894 | path->nodes[level] = next; | ||
| 3895 | path->slots[level] = 0; | ||
| 3896 | if (!path->skip_locking) | ||
| 3897 | path->locks[level] = 1; | ||
| 3898 | if (!level) | ||
| 3899 | break; | ||
| 3900 | if (level == 1 && path->locks[1] && path->reada) | ||
| 3901 | reada_for_search(root, path, level, slot, 0); | ||
| 3902 | next = read_node_slot(root, next, 0); | ||
| 3903 | if (!path->skip_locking) { | ||
| 3904 | WARN_ON(!btrfs_tree_locked(path->nodes[level])); | ||
| 3905 | btrfs_tree_lock(next); | ||
| 3906 | } | ||
| 3907 | } | ||
| 3908 | done: | ||
| 3909 | unlock_up(path, 0, 1); | ||
| 3910 | return 0; | ||
| 3911 | } | ||
| 3912 | |||
| 3913 | /* | ||
| 3914 | * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps | ||
| 3915 | * searching until it gets past min_objectid or finds an item of 'type' | ||
| 3916 | * | ||
| 3917 | * returns 0 if something is found, 1 if nothing was found and < 0 on error | ||
| 3918 | */ | ||
| 3919 | int btrfs_previous_item(struct btrfs_root *root, | ||
| 3920 | struct btrfs_path *path, u64 min_objectid, | ||
| 3921 | int type) | ||
| 3922 | { | ||
| 3923 | struct btrfs_key found_key; | ||
| 3924 | struct extent_buffer *leaf; | ||
| 3925 | u32 nritems; | ||
| 3926 | int ret; | ||
| 3927 | |||
| 3928 | while (1) { | ||
| 3929 | if (path->slots[0] == 0) { | ||
| 3930 | ret = btrfs_prev_leaf(root, path); | ||
| 3931 | if (ret != 0) | ||
| 3932 | return ret; | ||
| 3933 | } else { | ||
| 3934 | path->slots[0]--; | ||
| 3935 | } | ||
| 3936 | leaf = path->nodes[0]; | ||
| 3937 | nritems = btrfs_header_nritems(leaf); | ||
| 3938 | if (nritems == 0) | ||
| 3939 | return 1; | ||
| 3940 | if (path->slots[0] == nritems) | ||
| 3941 | path->slots[0]--; | ||
| 3942 | |||
| 3943 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 3944 | if (found_key.type == type) | ||
| 3945 | return 0; | ||
| 3946 | if (found_key.objectid < min_objectid) | ||
| 3947 | break; | ||
| 3948 | if (found_key.objectid == min_objectid && | ||
| 3949 | found_key.type < type) | ||
| 3950 | break; | ||
| 3951 | } | ||
| 3952 | return 1; | ||
| 3953 | } | ||
diff --git a/fs/btrfs/ctree.h b/fs/btrfs/ctree.h new file mode 100644 index 000000000000..eee060f88113 --- /dev/null +++ b/fs/btrfs/ctree.h | |||
| @@ -0,0 +1,2129 @@ | |||
| 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 | |||
| 19 | #ifndef __BTRFS_CTREE__ | ||
| 20 | #define __BTRFS_CTREE__ | ||
| 21 | |||
| 22 | #include <linux/version.h> | ||
| 23 | #include <linux/mm.h> | ||
| 24 | #include <linux/highmem.h> | ||
| 25 | #include <linux/fs.h> | ||
| 26 | #include <linux/completion.h> | ||
| 27 | #include <linux/backing-dev.h> | ||
| 28 | #include <linux/wait.h> | ||
| 29 | #include <asm/kmap_types.h> | ||
| 30 | #include "extent_io.h" | ||
| 31 | #include "extent_map.h" | ||
| 32 | #include "async-thread.h" | ||
| 33 | |||
| 34 | struct btrfs_trans_handle; | ||
| 35 | struct btrfs_transaction; | ||
| 36 | extern struct kmem_cache *btrfs_trans_handle_cachep; | ||
| 37 | extern struct kmem_cache *btrfs_transaction_cachep; | ||
| 38 | extern struct kmem_cache *btrfs_bit_radix_cachep; | ||
| 39 | extern struct kmem_cache *btrfs_path_cachep; | ||
| 40 | struct btrfs_ordered_sum; | ||
| 41 | |||
| 42 | #define BTRFS_MAGIC "_BHRfS_M" | ||
| 43 | |||
| 44 | #define BTRFS_ACL_NOT_CACHED ((void *)-1) | ||
| 45 | |||
| 46 | #ifdef CONFIG_LOCKDEP | ||
| 47 | # define BTRFS_MAX_LEVEL 7 | ||
| 48 | #else | ||
| 49 | # define BTRFS_MAX_LEVEL 8 | ||
| 50 | #endif | ||
| 51 | |||
| 52 | /* holds pointers to all of the tree roots */ | ||
| 53 | #define BTRFS_ROOT_TREE_OBJECTID 1ULL | ||
| 54 | |||
| 55 | /* stores information about which extents are in use, and reference counts */ | ||
| 56 | #define BTRFS_EXTENT_TREE_OBJECTID 2ULL | ||
| 57 | |||
| 58 | /* | ||
| 59 | * chunk tree stores translations from logical -> physical block numbering | ||
| 60 | * the super block points to the chunk tree | ||
| 61 | */ | ||
| 62 | #define BTRFS_CHUNK_TREE_OBJECTID 3ULL | ||
| 63 | |||
| 64 | /* | ||
| 65 | * stores information about which areas of a given device are in use. | ||
| 66 | * one per device. The tree of tree roots points to the device tree | ||
| 67 | */ | ||
| 68 | #define BTRFS_DEV_TREE_OBJECTID 4ULL | ||
| 69 | |||
| 70 | /* one per subvolume, storing files and directories */ | ||
| 71 | #define BTRFS_FS_TREE_OBJECTID 5ULL | ||
| 72 | |||
| 73 | /* directory objectid inside the root tree */ | ||
| 74 | #define BTRFS_ROOT_TREE_DIR_OBJECTID 6ULL | ||
| 75 | |||
| 76 | /* holds checksums of all the data extents */ | ||
| 77 | #define BTRFS_CSUM_TREE_OBJECTID 7ULL | ||
| 78 | |||
| 79 | /* orhpan objectid for tracking unlinked/truncated files */ | ||
| 80 | #define BTRFS_ORPHAN_OBJECTID -5ULL | ||
| 81 | |||
| 82 | /* does write ahead logging to speed up fsyncs */ | ||
| 83 | #define BTRFS_TREE_LOG_OBJECTID -6ULL | ||
| 84 | #define BTRFS_TREE_LOG_FIXUP_OBJECTID -7ULL | ||
| 85 | |||
| 86 | /* for space balancing */ | ||
| 87 | #define BTRFS_TREE_RELOC_OBJECTID -8ULL | ||
| 88 | #define BTRFS_DATA_RELOC_TREE_OBJECTID -9ULL | ||
| 89 | |||
| 90 | /* | ||
| 91 | * extent checksums all have this objectid | ||
| 92 | * this allows them to share the logging tree | ||
| 93 | * for fsyncs | ||
| 94 | */ | ||
| 95 | #define BTRFS_EXTENT_CSUM_OBJECTID -10ULL | ||
| 96 | |||
| 97 | /* dummy objectid represents multiple objectids */ | ||
| 98 | #define BTRFS_MULTIPLE_OBJECTIDS -255ULL | ||
| 99 | |||
| 100 | /* | ||
| 101 | * All files have objectids in this range. | ||
| 102 | */ | ||
| 103 | #define BTRFS_FIRST_FREE_OBJECTID 256ULL | ||
| 104 | #define BTRFS_LAST_FREE_OBJECTID -256ULL | ||
| 105 | #define BTRFS_FIRST_CHUNK_TREE_OBJECTID 256ULL | ||
| 106 | |||
| 107 | |||
| 108 | /* | ||
| 109 | * the device items go into the chunk tree. The key is in the form | ||
| 110 | * [ 1 BTRFS_DEV_ITEM_KEY device_id ] | ||
| 111 | */ | ||
| 112 | #define BTRFS_DEV_ITEMS_OBJECTID 1ULL | ||
| 113 | |||
| 114 | /* | ||
| 115 | * we can actually store much bigger names, but lets not confuse the rest | ||
| 116 | * of linux | ||
| 117 | */ | ||
| 118 | #define BTRFS_NAME_LEN 255 | ||
| 119 | |||
| 120 | /* 32 bytes in various csum fields */ | ||
| 121 | #define BTRFS_CSUM_SIZE 32 | ||
| 122 | |||
| 123 | /* csum types */ | ||
| 124 | #define BTRFS_CSUM_TYPE_CRC32 0 | ||
| 125 | |||
| 126 | static int btrfs_csum_sizes[] = { 4, 0 }; | ||
| 127 | |||
| 128 | /* four bytes for CRC32 */ | ||
| 129 | #define BTRFS_EMPTY_DIR_SIZE 0 | ||
| 130 | |||
| 131 | #define BTRFS_FT_UNKNOWN 0 | ||
| 132 | #define BTRFS_FT_REG_FILE 1 | ||
| 133 | #define BTRFS_FT_DIR 2 | ||
| 134 | #define BTRFS_FT_CHRDEV 3 | ||
| 135 | #define BTRFS_FT_BLKDEV 4 | ||
| 136 | #define BTRFS_FT_FIFO 5 | ||
| 137 | #define BTRFS_FT_SOCK 6 | ||
| 138 | #define BTRFS_FT_SYMLINK 7 | ||
| 139 | #define BTRFS_FT_XATTR 8 | ||
| 140 | #define BTRFS_FT_MAX 9 | ||
| 141 | |||
| 142 | /* | ||
| 143 | * the key defines the order in the tree, and so it also defines (optimal) | ||
| 144 | * block layout. objectid corresonds to the inode number. The flags | ||
| 145 | * tells us things about the object, and is a kind of stream selector. | ||
| 146 | * so for a given inode, keys with flags of 1 might refer to the inode | ||
| 147 | * data, flags of 2 may point to file data in the btree and flags == 3 | ||
| 148 | * may point to extents. | ||
| 149 | * | ||
| 150 | * offset is the starting byte offset for this key in the stream. | ||
| 151 | * | ||
| 152 | * btrfs_disk_key is in disk byte order. struct btrfs_key is always | ||
| 153 | * in cpu native order. Otherwise they are identical and their sizes | ||
| 154 | * should be the same (ie both packed) | ||
| 155 | */ | ||
| 156 | struct btrfs_disk_key { | ||
| 157 | __le64 objectid; | ||
| 158 | u8 type; | ||
| 159 | __le64 offset; | ||
| 160 | } __attribute__ ((__packed__)); | ||
| 161 | |||
| 162 | struct btrfs_key { | ||
| 163 | u64 objectid; | ||
| 164 | u8 type; | ||
| 165 | u64 offset; | ||
| 166 | } __attribute__ ((__packed__)); | ||
| 167 | |||
| 168 | struct btrfs_mapping_tree { | ||
| 169 | struct extent_map_tree map_tree; | ||
| 170 | }; | ||
| 171 | |||
| 172 | #define BTRFS_UUID_SIZE 16 | ||
| 173 | struct btrfs_dev_item { | ||
| 174 | /* the internal btrfs device id */ | ||
| 175 | __le64 devid; | ||
| 176 | |||
| 177 | /* size of the device */ | ||
| 178 | __le64 total_bytes; | ||
| 179 | |||
| 180 | /* bytes used */ | ||
| 181 | __le64 bytes_used; | ||
| 182 | |||
| 183 | /* optimal io alignment for this device */ | ||
| 184 | __le32 io_align; | ||
| 185 | |||
| 186 | /* optimal io width for this device */ | ||
| 187 | __le32 io_width; | ||
| 188 | |||
| 189 | /* minimal io size for this device */ | ||
| 190 | __le32 sector_size; | ||
| 191 | |||
| 192 | /* type and info about this device */ | ||
| 193 | __le64 type; | ||
| 194 | |||
| 195 | /* expected generation for this device */ | ||
| 196 | __le64 generation; | ||
| 197 | |||
| 198 | /* | ||
| 199 | * starting byte of this partition on the device, | ||
| 200 | * to allowr for stripe alignment in the future | ||
| 201 | */ | ||
| 202 | __le64 start_offset; | ||
| 203 | |||
| 204 | /* grouping information for allocation decisions */ | ||
| 205 | __le32 dev_group; | ||
| 206 | |||
| 207 | /* seek speed 0-100 where 100 is fastest */ | ||
| 208 | u8 seek_speed; | ||
| 209 | |||
| 210 | /* bandwidth 0-100 where 100 is fastest */ | ||
| 211 | u8 bandwidth; | ||
| 212 | |||
| 213 | /* btrfs generated uuid for this device */ | ||
| 214 | u8 uuid[BTRFS_UUID_SIZE]; | ||
| 215 | |||
| 216 | /* uuid of FS who owns this device */ | ||
| 217 | u8 fsid[BTRFS_UUID_SIZE]; | ||
| 218 | } __attribute__ ((__packed__)); | ||
| 219 | |||
| 220 | struct btrfs_stripe { | ||
| 221 | __le64 devid; | ||
| 222 | __le64 offset; | ||
| 223 | u8 dev_uuid[BTRFS_UUID_SIZE]; | ||
| 224 | } __attribute__ ((__packed__)); | ||
| 225 | |||
| 226 | struct btrfs_chunk { | ||
| 227 | /* size of this chunk in bytes */ | ||
| 228 | __le64 length; | ||
| 229 | |||
| 230 | /* objectid of the root referencing this chunk */ | ||
| 231 | __le64 owner; | ||
| 232 | |||
| 233 | __le64 stripe_len; | ||
| 234 | __le64 type; | ||
| 235 | |||
| 236 | /* optimal io alignment for this chunk */ | ||
| 237 | __le32 io_align; | ||
| 238 | |||
| 239 | /* optimal io width for this chunk */ | ||
| 240 | __le32 io_width; | ||
| 241 | |||
| 242 | /* minimal io size for this chunk */ | ||
| 243 | __le32 sector_size; | ||
| 244 | |||
| 245 | /* 2^16 stripes is quite a lot, a second limit is the size of a single | ||
| 246 | * item in the btree | ||
| 247 | */ | ||
| 248 | __le16 num_stripes; | ||
| 249 | |||
| 250 | /* sub stripes only matter for raid10 */ | ||
| 251 | __le16 sub_stripes; | ||
| 252 | struct btrfs_stripe stripe; | ||
| 253 | /* additional stripes go here */ | ||
| 254 | } __attribute__ ((__packed__)); | ||
| 255 | |||
| 256 | static inline unsigned long btrfs_chunk_item_size(int num_stripes) | ||
| 257 | { | ||
| 258 | BUG_ON(num_stripes == 0); | ||
| 259 | return sizeof(struct btrfs_chunk) + | ||
| 260 | sizeof(struct btrfs_stripe) * (num_stripes - 1); | ||
| 261 | } | ||
| 262 | |||
| 263 | #define BTRFS_FSID_SIZE 16 | ||
| 264 | #define BTRFS_HEADER_FLAG_WRITTEN (1 << 0) | ||
| 265 | |||
| 266 | /* | ||
| 267 | * every tree block (leaf or node) starts with this header. | ||
| 268 | */ | ||
| 269 | struct btrfs_header { | ||
| 270 | /* these first four must match the super block */ | ||
| 271 | u8 csum[BTRFS_CSUM_SIZE]; | ||
| 272 | u8 fsid[BTRFS_FSID_SIZE]; /* FS specific uuid */ | ||
| 273 | __le64 bytenr; /* which block this node is supposed to live in */ | ||
| 274 | __le64 flags; | ||
| 275 | |||
| 276 | /* allowed to be different from the super from here on down */ | ||
| 277 | u8 chunk_tree_uuid[BTRFS_UUID_SIZE]; | ||
| 278 | __le64 generation; | ||
| 279 | __le64 owner; | ||
| 280 | __le32 nritems; | ||
| 281 | u8 level; | ||
| 282 | } __attribute__ ((__packed__)); | ||
| 283 | |||
| 284 | #define BTRFS_NODEPTRS_PER_BLOCK(r) (((r)->nodesize - \ | ||
| 285 | sizeof(struct btrfs_header)) / \ | ||
| 286 | sizeof(struct btrfs_key_ptr)) | ||
| 287 | #define __BTRFS_LEAF_DATA_SIZE(bs) ((bs) - sizeof(struct btrfs_header)) | ||
| 288 | #define BTRFS_LEAF_DATA_SIZE(r) (__BTRFS_LEAF_DATA_SIZE(r->leafsize)) | ||
| 289 | #define BTRFS_MAX_INLINE_DATA_SIZE(r) (BTRFS_LEAF_DATA_SIZE(r) - \ | ||
| 290 | sizeof(struct btrfs_item) - \ | ||
| 291 | sizeof(struct btrfs_file_extent_item)) | ||
| 292 | |||
| 293 | #define BTRFS_SUPER_FLAG_SEEDING (1ULL << 32) | ||
| 294 | |||
| 295 | /* | ||
| 296 | * this is a very generous portion of the super block, giving us | ||
| 297 | * room to translate 14 chunks with 3 stripes each. | ||
| 298 | */ | ||
| 299 | #define BTRFS_SYSTEM_CHUNK_ARRAY_SIZE 2048 | ||
| 300 | #define BTRFS_LABEL_SIZE 256 | ||
| 301 | |||
| 302 | /* | ||
| 303 | * the super block basically lists the main trees of the FS | ||
| 304 | * it currently lacks any block count etc etc | ||
| 305 | */ | ||
| 306 | struct btrfs_super_block { | ||
| 307 | u8 csum[BTRFS_CSUM_SIZE]; | ||
| 308 | /* the first 4 fields must match struct btrfs_header */ | ||
| 309 | u8 fsid[BTRFS_FSID_SIZE]; /* FS specific uuid */ | ||
| 310 | __le64 bytenr; /* this block number */ | ||
| 311 | __le64 flags; | ||
| 312 | |||
| 313 | /* allowed to be different from the btrfs_header from here own down */ | ||
| 314 | __le64 magic; | ||
| 315 | __le64 generation; | ||
| 316 | __le64 root; | ||
| 317 | __le64 chunk_root; | ||
| 318 | __le64 log_root; | ||
| 319 | |||
| 320 | /* this will help find the new super based on the log root */ | ||
| 321 | __le64 log_root_transid; | ||
| 322 | __le64 total_bytes; | ||
| 323 | __le64 bytes_used; | ||
| 324 | __le64 root_dir_objectid; | ||
| 325 | __le64 num_devices; | ||
| 326 | __le32 sectorsize; | ||
| 327 | __le32 nodesize; | ||
| 328 | __le32 leafsize; | ||
| 329 | __le32 stripesize; | ||
| 330 | __le32 sys_chunk_array_size; | ||
| 331 | __le64 chunk_root_generation; | ||
| 332 | __le64 compat_flags; | ||
| 333 | __le64 compat_ro_flags; | ||
| 334 | __le64 incompat_flags; | ||
| 335 | __le16 csum_type; | ||
| 336 | u8 root_level; | ||
| 337 | u8 chunk_root_level; | ||
| 338 | u8 log_root_level; | ||
| 339 | struct btrfs_dev_item dev_item; | ||
| 340 | |||
| 341 | char label[BTRFS_LABEL_SIZE]; | ||
| 342 | |||
| 343 | /* future expansion */ | ||
| 344 | __le64 reserved[32]; | ||
| 345 | u8 sys_chunk_array[BTRFS_SYSTEM_CHUNK_ARRAY_SIZE]; | ||
| 346 | } __attribute__ ((__packed__)); | ||
| 347 | |||
| 348 | /* | ||
| 349 | * Compat flags that we support. If any incompat flags are set other than the | ||
| 350 | * ones specified below then we will fail to mount | ||
| 351 | */ | ||
| 352 | #define BTRFS_FEATURE_COMPAT_SUPP 0x0 | ||
| 353 | #define BTRFS_FEATURE_COMPAT_RO_SUPP 0x0 | ||
| 354 | #define BTRFS_FEATURE_INCOMPAT_SUPP 0x0 | ||
| 355 | |||
| 356 | /* | ||
| 357 | * A leaf is full of items. offset and size tell us where to find | ||
| 358 | * the item in the leaf (relative to the start of the data area) | ||
| 359 | */ | ||
| 360 | struct btrfs_item { | ||
| 361 | struct btrfs_disk_key key; | ||
| 362 | __le32 offset; | ||
| 363 | __le32 size; | ||
| 364 | } __attribute__ ((__packed__)); | ||
| 365 | |||
| 366 | /* | ||
| 367 | * leaves have an item area and a data area: | ||
| 368 | * [item0, item1....itemN] [free space] [dataN...data1, data0] | ||
| 369 | * | ||
| 370 | * The data is separate from the items to get the keys closer together | ||
| 371 | * during searches. | ||
| 372 | */ | ||
| 373 | struct btrfs_leaf { | ||
| 374 | struct btrfs_header header; | ||
| 375 | struct btrfs_item items[]; | ||
| 376 | } __attribute__ ((__packed__)); | ||
| 377 | |||
| 378 | /* | ||
| 379 | * all non-leaf blocks are nodes, they hold only keys and pointers to | ||
| 380 | * other blocks | ||
| 381 | */ | ||
| 382 | struct btrfs_key_ptr { | ||
| 383 | struct btrfs_disk_key key; | ||
| 384 | __le64 blockptr; | ||
| 385 | __le64 generation; | ||
| 386 | } __attribute__ ((__packed__)); | ||
| 387 | |||
| 388 | struct btrfs_node { | ||
| 389 | struct btrfs_header header; | ||
| 390 | struct btrfs_key_ptr ptrs[]; | ||
| 391 | } __attribute__ ((__packed__)); | ||
| 392 | |||
| 393 | /* | ||
| 394 | * btrfs_paths remember the path taken from the root down to the leaf. | ||
| 395 | * level 0 is always the leaf, and nodes[1...BTRFS_MAX_LEVEL] will point | ||
| 396 | * to any other levels that are present. | ||
| 397 | * | ||
| 398 | * The slots array records the index of the item or block pointer | ||
| 399 | * used while walking the tree. | ||
| 400 | */ | ||
| 401 | struct btrfs_path { | ||
| 402 | struct extent_buffer *nodes[BTRFS_MAX_LEVEL]; | ||
| 403 | int slots[BTRFS_MAX_LEVEL]; | ||
| 404 | /* if there is real range locking, this locks field will change */ | ||
| 405 | int locks[BTRFS_MAX_LEVEL]; | ||
| 406 | int reada; | ||
| 407 | /* keep some upper locks as we walk down */ | ||
| 408 | int keep_locks; | ||
| 409 | int skip_locking; | ||
| 410 | int lowest_level; | ||
| 411 | |||
| 412 | /* | ||
| 413 | * set by btrfs_split_item, tells search_slot to keep all locks | ||
| 414 | * and to force calls to keep space in the nodes | ||
| 415 | */ | ||
| 416 | int search_for_split; | ||
| 417 | }; | ||
| 418 | |||
| 419 | /* | ||
| 420 | * items in the extent btree are used to record the objectid of the | ||
| 421 | * owner of the block and the number of references | ||
| 422 | */ | ||
| 423 | struct btrfs_extent_item { | ||
| 424 | __le32 refs; | ||
| 425 | } __attribute__ ((__packed__)); | ||
| 426 | |||
| 427 | struct btrfs_extent_ref { | ||
| 428 | __le64 root; | ||
| 429 | __le64 generation; | ||
| 430 | __le64 objectid; | ||
| 431 | __le32 num_refs; | ||
| 432 | } __attribute__ ((__packed__)); | ||
| 433 | |||
| 434 | /* dev extents record free space on individual devices. The owner | ||
| 435 | * field points back to the chunk allocation mapping tree that allocated | ||
| 436 | * the extent. The chunk tree uuid field is a way to double check the owner | ||
| 437 | */ | ||
| 438 | struct btrfs_dev_extent { | ||
| 439 | __le64 chunk_tree; | ||
| 440 | __le64 chunk_objectid; | ||
| 441 | __le64 chunk_offset; | ||
| 442 | __le64 length; | ||
| 443 | u8 chunk_tree_uuid[BTRFS_UUID_SIZE]; | ||
| 444 | } __attribute__ ((__packed__)); | ||
| 445 | |||
| 446 | struct btrfs_inode_ref { | ||
| 447 | __le64 index; | ||
| 448 | __le16 name_len; | ||
| 449 | /* name goes here */ | ||
| 450 | } __attribute__ ((__packed__)); | ||
| 451 | |||
| 452 | struct btrfs_timespec { | ||
| 453 | __le64 sec; | ||
| 454 | __le32 nsec; | ||
| 455 | } __attribute__ ((__packed__)); | ||
| 456 | |||
| 457 | typedef enum { | ||
| 458 | BTRFS_COMPRESS_NONE = 0, | ||
| 459 | BTRFS_COMPRESS_ZLIB = 1, | ||
| 460 | BTRFS_COMPRESS_LAST = 2, | ||
| 461 | } btrfs_compression_type; | ||
| 462 | |||
| 463 | /* we don't understand any encryption methods right now */ | ||
| 464 | typedef enum { | ||
| 465 | BTRFS_ENCRYPTION_NONE = 0, | ||
| 466 | BTRFS_ENCRYPTION_LAST = 1, | ||
| 467 | } btrfs_encryption_type; | ||
| 468 | |||
| 469 | struct btrfs_inode_item { | ||
| 470 | /* nfs style generation number */ | ||
| 471 | __le64 generation; | ||
| 472 | /* transid that last touched this inode */ | ||
| 473 | __le64 transid; | ||
| 474 | __le64 size; | ||
| 475 | __le64 nbytes; | ||
| 476 | __le64 block_group; | ||
| 477 | __le32 nlink; | ||
| 478 | __le32 uid; | ||
| 479 | __le32 gid; | ||
| 480 | __le32 mode; | ||
| 481 | __le64 rdev; | ||
| 482 | __le64 flags; | ||
| 483 | |||
| 484 | /* modification sequence number for NFS */ | ||
| 485 | __le64 sequence; | ||
| 486 | |||
| 487 | /* | ||
| 488 | * a little future expansion, for more than this we can | ||
| 489 | * just grow the inode item and version it | ||
| 490 | */ | ||
| 491 | __le64 reserved[4]; | ||
| 492 | struct btrfs_timespec atime; | ||
| 493 | struct btrfs_timespec ctime; | ||
| 494 | struct btrfs_timespec mtime; | ||
| 495 | struct btrfs_timespec otime; | ||
| 496 | } __attribute__ ((__packed__)); | ||
| 497 | |||
| 498 | struct btrfs_dir_log_item { | ||
| 499 | __le64 end; | ||
| 500 | } __attribute__ ((__packed__)); | ||
| 501 | |||
| 502 | struct btrfs_dir_item { | ||
| 503 | struct btrfs_disk_key location; | ||
| 504 | __le64 transid; | ||
| 505 | __le16 data_len; | ||
| 506 | __le16 name_len; | ||
| 507 | u8 type; | ||
| 508 | } __attribute__ ((__packed__)); | ||
| 509 | |||
| 510 | struct btrfs_root_item { | ||
| 511 | struct btrfs_inode_item inode; | ||
| 512 | __le64 generation; | ||
| 513 | __le64 root_dirid; | ||
| 514 | __le64 bytenr; | ||
| 515 | __le64 byte_limit; | ||
| 516 | __le64 bytes_used; | ||
| 517 | __le64 last_snapshot; | ||
| 518 | __le64 flags; | ||
| 519 | __le32 refs; | ||
| 520 | struct btrfs_disk_key drop_progress; | ||
| 521 | u8 drop_level; | ||
| 522 | u8 level; | ||
| 523 | } __attribute__ ((__packed__)); | ||
| 524 | |||
| 525 | /* | ||
| 526 | * this is used for both forward and backward root refs | ||
| 527 | */ | ||
| 528 | struct btrfs_root_ref { | ||
| 529 | __le64 dirid; | ||
| 530 | __le64 sequence; | ||
| 531 | __le16 name_len; | ||
| 532 | } __attribute__ ((__packed__)); | ||
| 533 | |||
| 534 | #define BTRFS_FILE_EXTENT_INLINE 0 | ||
| 535 | #define BTRFS_FILE_EXTENT_REG 1 | ||
| 536 | #define BTRFS_FILE_EXTENT_PREALLOC 2 | ||
| 537 | |||
| 538 | struct btrfs_file_extent_item { | ||
| 539 | /* | ||
| 540 | * transaction id that created this extent | ||
| 541 | */ | ||
| 542 | __le64 generation; | ||
| 543 | /* | ||
| 544 | * max number of bytes to hold this extent in ram | ||
| 545 | * when we split a compressed extent we can't know how big | ||
| 546 | * each of the resulting pieces will be. So, this is | ||
| 547 | * an upper limit on the size of the extent in ram instead of | ||
| 548 | * an exact limit. | ||
| 549 | */ | ||
| 550 | __le64 ram_bytes; | ||
| 551 | |||
| 552 | /* | ||
| 553 | * 32 bits for the various ways we might encode the data, | ||
| 554 | * including compression and encryption. If any of these | ||
| 555 | * are set to something a given disk format doesn't understand | ||
| 556 | * it is treated like an incompat flag for reading and writing, | ||
| 557 | * but not for stat. | ||
| 558 | */ | ||
| 559 | u8 compression; | ||
| 560 | u8 encryption; | ||
| 561 | __le16 other_encoding; /* spare for later use */ | ||
| 562 | |||
| 563 | /* are we inline data or a real extent? */ | ||
| 564 | u8 type; | ||
| 565 | |||
| 566 | /* | ||
| 567 | * disk space consumed by the extent, checksum blocks are included | ||
| 568 | * in these numbers | ||
| 569 | */ | ||
| 570 | __le64 disk_bytenr; | ||
| 571 | __le64 disk_num_bytes; | ||
| 572 | /* | ||
| 573 | * the logical offset in file blocks (no csums) | ||
| 574 | * this extent record is for. This allows a file extent to point | ||
| 575 | * into the middle of an existing extent on disk, sharing it | ||
| 576 | * between two snapshots (useful if some bytes in the middle of the | ||
| 577 | * extent have changed | ||
| 578 | */ | ||
| 579 | __le64 offset; | ||
| 580 | /* | ||
| 581 | * the logical number of file blocks (no csums included). This | ||
| 582 | * always reflects the size uncompressed and without encoding. | ||
| 583 | */ | ||
| 584 | __le64 num_bytes; | ||
| 585 | |||
| 586 | } __attribute__ ((__packed__)); | ||
| 587 | |||
| 588 | struct btrfs_csum_item { | ||
| 589 | u8 csum; | ||
| 590 | } __attribute__ ((__packed__)); | ||
| 591 | |||
| 592 | /* different types of block groups (and chunks) */ | ||
| 593 | #define BTRFS_BLOCK_GROUP_DATA (1 << 0) | ||
| 594 | #define BTRFS_BLOCK_GROUP_SYSTEM (1 << 1) | ||
| 595 | #define BTRFS_BLOCK_GROUP_METADATA (1 << 2) | ||
| 596 | #define BTRFS_BLOCK_GROUP_RAID0 (1 << 3) | ||
| 597 | #define BTRFS_BLOCK_GROUP_RAID1 (1 << 4) | ||
| 598 | #define BTRFS_BLOCK_GROUP_DUP (1 << 5) | ||
| 599 | #define BTRFS_BLOCK_GROUP_RAID10 (1 << 6) | ||
| 600 | |||
| 601 | struct btrfs_block_group_item { | ||
| 602 | __le64 used; | ||
| 603 | __le64 chunk_objectid; | ||
| 604 | __le64 flags; | ||
| 605 | } __attribute__ ((__packed__)); | ||
| 606 | |||
| 607 | struct btrfs_space_info { | ||
| 608 | u64 flags; | ||
| 609 | u64 total_bytes; | ||
| 610 | u64 bytes_used; | ||
| 611 | u64 bytes_pinned; | ||
| 612 | u64 bytes_reserved; | ||
| 613 | u64 bytes_readonly; | ||
| 614 | int full; | ||
| 615 | int force_alloc; | ||
| 616 | struct list_head list; | ||
| 617 | |||
| 618 | /* for block groups in our same type */ | ||
| 619 | struct list_head block_groups; | ||
| 620 | spinlock_t lock; | ||
| 621 | struct rw_semaphore groups_sem; | ||
| 622 | }; | ||
| 623 | |||
| 624 | struct btrfs_free_space { | ||
| 625 | struct rb_node bytes_index; | ||
| 626 | struct rb_node offset_index; | ||
| 627 | u64 offset; | ||
| 628 | u64 bytes; | ||
| 629 | }; | ||
| 630 | |||
| 631 | struct btrfs_block_group_cache { | ||
| 632 | struct btrfs_key key; | ||
| 633 | struct btrfs_block_group_item item; | ||
| 634 | spinlock_t lock; | ||
| 635 | struct mutex alloc_mutex; | ||
| 636 | struct mutex cache_mutex; | ||
| 637 | u64 pinned; | ||
| 638 | u64 reserved; | ||
| 639 | u64 flags; | ||
| 640 | int cached; | ||
| 641 | int ro; | ||
| 642 | int dirty; | ||
| 643 | |||
| 644 | struct btrfs_space_info *space_info; | ||
| 645 | |||
| 646 | /* free space cache stuff */ | ||
| 647 | struct rb_root free_space_bytes; | ||
| 648 | struct rb_root free_space_offset; | ||
| 649 | |||
| 650 | /* block group cache stuff */ | ||
| 651 | struct rb_node cache_node; | ||
| 652 | |||
| 653 | /* for block groups in the same raid type */ | ||
| 654 | struct list_head list; | ||
| 655 | |||
| 656 | /* usage count */ | ||
| 657 | atomic_t count; | ||
| 658 | }; | ||
| 659 | |||
| 660 | struct btrfs_leaf_ref_tree { | ||
| 661 | struct rb_root root; | ||
| 662 | struct list_head list; | ||
| 663 | spinlock_t lock; | ||
| 664 | }; | ||
| 665 | |||
| 666 | struct btrfs_device; | ||
| 667 | struct btrfs_fs_devices; | ||
| 668 | struct btrfs_fs_info { | ||
| 669 | u8 fsid[BTRFS_FSID_SIZE]; | ||
| 670 | u8 chunk_tree_uuid[BTRFS_UUID_SIZE]; | ||
| 671 | struct btrfs_root *extent_root; | ||
| 672 | struct btrfs_root *tree_root; | ||
| 673 | struct btrfs_root *chunk_root; | ||
| 674 | struct btrfs_root *dev_root; | ||
| 675 | struct btrfs_root *fs_root; | ||
| 676 | struct btrfs_root *csum_root; | ||
| 677 | |||
| 678 | /* the log root tree is a directory of all the other log roots */ | ||
| 679 | struct btrfs_root *log_root_tree; | ||
| 680 | struct radix_tree_root fs_roots_radix; | ||
| 681 | |||
| 682 | /* block group cache stuff */ | ||
| 683 | spinlock_t block_group_cache_lock; | ||
| 684 | struct rb_root block_group_cache_tree; | ||
| 685 | |||
| 686 | struct extent_io_tree pinned_extents; | ||
| 687 | struct extent_io_tree pending_del; | ||
| 688 | struct extent_io_tree extent_ins; | ||
| 689 | |||
| 690 | /* logical->physical extent mapping */ | ||
| 691 | struct btrfs_mapping_tree mapping_tree; | ||
| 692 | |||
| 693 | u64 generation; | ||
| 694 | u64 last_trans_committed; | ||
| 695 | u64 last_trans_new_blockgroup; | ||
| 696 | u64 open_ioctl_trans; | ||
| 697 | unsigned long mount_opt; | ||
| 698 | u64 max_extent; | ||
| 699 | u64 max_inline; | ||
| 700 | u64 alloc_start; | ||
| 701 | struct btrfs_transaction *running_transaction; | ||
| 702 | wait_queue_head_t transaction_throttle; | ||
| 703 | wait_queue_head_t transaction_wait; | ||
| 704 | |||
| 705 | wait_queue_head_t async_submit_wait; | ||
| 706 | wait_queue_head_t tree_log_wait; | ||
| 707 | |||
| 708 | struct btrfs_super_block super_copy; | ||
| 709 | struct btrfs_super_block super_for_commit; | ||
| 710 | struct block_device *__bdev; | ||
| 711 | struct super_block *sb; | ||
| 712 | struct inode *btree_inode; | ||
| 713 | struct backing_dev_info bdi; | ||
| 714 | spinlock_t hash_lock; | ||
| 715 | struct mutex trans_mutex; | ||
| 716 | struct mutex tree_log_mutex; | ||
| 717 | struct mutex transaction_kthread_mutex; | ||
| 718 | struct mutex cleaner_mutex; | ||
| 719 | struct mutex extent_ins_mutex; | ||
| 720 | struct mutex pinned_mutex; | ||
| 721 | struct mutex chunk_mutex; | ||
| 722 | struct mutex drop_mutex; | ||
| 723 | struct mutex volume_mutex; | ||
| 724 | struct mutex tree_reloc_mutex; | ||
| 725 | struct list_head trans_list; | ||
| 726 | struct list_head hashers; | ||
| 727 | struct list_head dead_roots; | ||
| 728 | |||
| 729 | atomic_t nr_async_submits; | ||
| 730 | atomic_t async_submit_draining; | ||
| 731 | atomic_t nr_async_bios; | ||
| 732 | atomic_t async_delalloc_pages; | ||
| 733 | atomic_t tree_log_writers; | ||
| 734 | atomic_t tree_log_commit; | ||
| 735 | unsigned long tree_log_batch; | ||
| 736 | u64 tree_log_transid; | ||
| 737 | |||
| 738 | /* | ||
| 739 | * this is used by the balancing code to wait for all the pending | ||
| 740 | * ordered extents | ||
| 741 | */ | ||
| 742 | spinlock_t ordered_extent_lock; | ||
| 743 | struct list_head ordered_extents; | ||
| 744 | struct list_head delalloc_inodes; | ||
| 745 | |||
| 746 | /* | ||
| 747 | * there is a pool of worker threads for checksumming during writes | ||
| 748 | * and a pool for checksumming after reads. This is because readers | ||
| 749 | * can run with FS locks held, and the writers may be waiting for | ||
| 750 | * those locks. We don't want ordering in the pending list to cause | ||
| 751 | * deadlocks, and so the two are serviced separately. | ||
| 752 | * | ||
| 753 | * A third pool does submit_bio to avoid deadlocking with the other | ||
| 754 | * two | ||
| 755 | */ | ||
| 756 | struct btrfs_workers workers; | ||
| 757 | struct btrfs_workers delalloc_workers; | ||
| 758 | struct btrfs_workers endio_workers; | ||
| 759 | struct btrfs_workers endio_meta_workers; | ||
| 760 | struct btrfs_workers endio_meta_write_workers; | ||
| 761 | struct btrfs_workers endio_write_workers; | ||
| 762 | struct btrfs_workers submit_workers; | ||
| 763 | /* | ||
| 764 | * fixup workers take dirty pages that didn't properly go through | ||
| 765 | * the cow mechanism and make them safe to write. It happens | ||
| 766 | * for the sys_munmap function call path | ||
| 767 | */ | ||
| 768 | struct btrfs_workers fixup_workers; | ||
| 769 | struct task_struct *transaction_kthread; | ||
| 770 | struct task_struct *cleaner_kthread; | ||
| 771 | int thread_pool_size; | ||
| 772 | |||
| 773 | /* tree relocation relocated fields */ | ||
| 774 | struct list_head dead_reloc_roots; | ||
| 775 | struct btrfs_leaf_ref_tree reloc_ref_tree; | ||
| 776 | struct btrfs_leaf_ref_tree shared_ref_tree; | ||
| 777 | |||
| 778 | struct kobject super_kobj; | ||
| 779 | struct completion kobj_unregister; | ||
| 780 | int do_barriers; | ||
| 781 | int closing; | ||
| 782 | int log_root_recovering; | ||
| 783 | atomic_t throttles; | ||
| 784 | atomic_t throttle_gen; | ||
| 785 | |||
| 786 | u64 total_pinned; | ||
| 787 | struct list_head dirty_cowonly_roots; | ||
| 788 | |||
| 789 | struct btrfs_fs_devices *fs_devices; | ||
| 790 | struct list_head space_info; | ||
| 791 | spinlock_t delalloc_lock; | ||
| 792 | spinlock_t new_trans_lock; | ||
| 793 | u64 delalloc_bytes; | ||
| 794 | u64 last_alloc; | ||
| 795 | u64 last_data_alloc; | ||
| 796 | |||
| 797 | spinlock_t ref_cache_lock; | ||
| 798 | u64 total_ref_cache_size; | ||
| 799 | |||
| 800 | u64 avail_data_alloc_bits; | ||
| 801 | u64 avail_metadata_alloc_bits; | ||
| 802 | u64 avail_system_alloc_bits; | ||
| 803 | u64 data_alloc_profile; | ||
| 804 | u64 metadata_alloc_profile; | ||
| 805 | u64 system_alloc_profile; | ||
| 806 | |||
| 807 | void *bdev_holder; | ||
| 808 | }; | ||
| 809 | |||
| 810 | /* | ||
| 811 | * in ram representation of the tree. extent_root is used for all allocations | ||
| 812 | * and for the extent tree extent_root root. | ||
| 813 | */ | ||
| 814 | struct btrfs_dirty_root; | ||
| 815 | struct btrfs_root { | ||
| 816 | struct extent_buffer *node; | ||
| 817 | |||
| 818 | /* the node lock is held while changing the node pointer */ | ||
| 819 | spinlock_t node_lock; | ||
| 820 | |||
| 821 | struct extent_buffer *commit_root; | ||
| 822 | struct btrfs_leaf_ref_tree *ref_tree; | ||
| 823 | struct btrfs_leaf_ref_tree ref_tree_struct; | ||
| 824 | struct btrfs_dirty_root *dirty_root; | ||
| 825 | struct btrfs_root *log_root; | ||
| 826 | struct btrfs_root *reloc_root; | ||
| 827 | |||
| 828 | struct btrfs_root_item root_item; | ||
| 829 | struct btrfs_key root_key; | ||
| 830 | struct btrfs_fs_info *fs_info; | ||
| 831 | struct extent_io_tree dirty_log_pages; | ||
| 832 | |||
| 833 | struct kobject root_kobj; | ||
| 834 | struct completion kobj_unregister; | ||
| 835 | struct mutex objectid_mutex; | ||
| 836 | struct mutex log_mutex; | ||
| 837 | |||
| 838 | u64 objectid; | ||
| 839 | u64 last_trans; | ||
| 840 | |||
| 841 | /* data allocations are done in sectorsize units */ | ||
| 842 | u32 sectorsize; | ||
| 843 | |||
| 844 | /* node allocations are done in nodesize units */ | ||
| 845 | u32 nodesize; | ||
| 846 | |||
| 847 | /* leaf allocations are done in leafsize units */ | ||
| 848 | u32 leafsize; | ||
| 849 | |||
| 850 | u32 stripesize; | ||
| 851 | |||
| 852 | u32 type; | ||
| 853 | u64 highest_inode; | ||
| 854 | u64 last_inode_alloc; | ||
| 855 | int ref_cows; | ||
| 856 | int track_dirty; | ||
| 857 | u64 defrag_trans_start; | ||
| 858 | struct btrfs_key defrag_progress; | ||
| 859 | struct btrfs_key defrag_max; | ||
| 860 | int defrag_running; | ||
| 861 | int defrag_level; | ||
| 862 | char *name; | ||
| 863 | int in_sysfs; | ||
| 864 | |||
| 865 | /* the dirty list is only used by non-reference counted roots */ | ||
| 866 | struct list_head dirty_list; | ||
| 867 | |||
| 868 | spinlock_t list_lock; | ||
| 869 | struct list_head dead_list; | ||
| 870 | struct list_head orphan_list; | ||
| 871 | |||
| 872 | /* | ||
| 873 | * right now this just gets used so that a root has its own devid | ||
| 874 | * for stat. It may be used for more later | ||
| 875 | */ | ||
| 876 | struct super_block anon_super; | ||
| 877 | }; | ||
| 878 | |||
| 879 | /* | ||
| 880 | |||
| 881 | * inode items have the data typically returned from stat and store other | ||
| 882 | * info about object characteristics. There is one for every file and dir in | ||
| 883 | * the FS | ||
| 884 | */ | ||
| 885 | #define BTRFS_INODE_ITEM_KEY 1 | ||
| 886 | #define BTRFS_INODE_REF_KEY 12 | ||
| 887 | #define BTRFS_XATTR_ITEM_KEY 24 | ||
| 888 | #define BTRFS_ORPHAN_ITEM_KEY 48 | ||
| 889 | /* reserve 2-15 close to the inode for later flexibility */ | ||
| 890 | |||
| 891 | /* | ||
| 892 | * dir items are the name -> inode pointers in a directory. There is one | ||
| 893 | * for every name in a directory. | ||
| 894 | */ | ||
| 895 | #define BTRFS_DIR_LOG_ITEM_KEY 60 | ||
| 896 | #define BTRFS_DIR_LOG_INDEX_KEY 72 | ||
| 897 | #define BTRFS_DIR_ITEM_KEY 84 | ||
| 898 | #define BTRFS_DIR_INDEX_KEY 96 | ||
| 899 | /* | ||
| 900 | * extent data is for file data | ||
| 901 | */ | ||
| 902 | #define BTRFS_EXTENT_DATA_KEY 108 | ||
| 903 | |||
| 904 | /* | ||
| 905 | * extent csums are stored in a separate tree and hold csums for | ||
| 906 | * an entire extent on disk. | ||
| 907 | */ | ||
| 908 | #define BTRFS_EXTENT_CSUM_KEY 128 | ||
| 909 | |||
| 910 | /* | ||
| 911 | * root items point to tree roots. There are typically in the root | ||
| 912 | * tree used by the super block to find all the other trees | ||
| 913 | */ | ||
| 914 | #define BTRFS_ROOT_ITEM_KEY 132 | ||
| 915 | |||
| 916 | /* | ||
| 917 | * root backrefs tie subvols and snapshots to the directory entries that | ||
| 918 | * reference them | ||
| 919 | */ | ||
| 920 | #define BTRFS_ROOT_BACKREF_KEY 144 | ||
| 921 | |||
| 922 | /* | ||
| 923 | * root refs make a fast index for listing all of the snapshots and | ||
| 924 | * subvolumes referenced by a given root. They point directly to the | ||
| 925 | * directory item in the root that references the subvol | ||
| 926 | */ | ||
| 927 | #define BTRFS_ROOT_REF_KEY 156 | ||
| 928 | |||
| 929 | /* | ||
| 930 | * extent items are in the extent map tree. These record which blocks | ||
| 931 | * are used, and how many references there are to each block | ||
| 932 | */ | ||
| 933 | #define BTRFS_EXTENT_ITEM_KEY 168 | ||
| 934 | #define BTRFS_EXTENT_REF_KEY 180 | ||
| 935 | |||
| 936 | /* | ||
| 937 | * block groups give us hints into the extent allocation trees. Which | ||
| 938 | * blocks are free etc etc | ||
| 939 | */ | ||
| 940 | #define BTRFS_BLOCK_GROUP_ITEM_KEY 192 | ||
| 941 | |||
| 942 | #define BTRFS_DEV_EXTENT_KEY 204 | ||
| 943 | #define BTRFS_DEV_ITEM_KEY 216 | ||
| 944 | #define BTRFS_CHUNK_ITEM_KEY 228 | ||
| 945 | |||
| 946 | /* | ||
| 947 | * string items are for debugging. They just store a short string of | ||
| 948 | * data in the FS | ||
| 949 | */ | ||
| 950 | #define BTRFS_STRING_ITEM_KEY 253 | ||
| 951 | |||
| 952 | #define BTRFS_MOUNT_NODATASUM (1 << 0) | ||
| 953 | #define BTRFS_MOUNT_NODATACOW (1 << 1) | ||
| 954 | #define BTRFS_MOUNT_NOBARRIER (1 << 2) | ||
| 955 | #define BTRFS_MOUNT_SSD (1 << 3) | ||
| 956 | #define BTRFS_MOUNT_DEGRADED (1 << 4) | ||
| 957 | #define BTRFS_MOUNT_COMPRESS (1 << 5) | ||
| 958 | |||
| 959 | #define btrfs_clear_opt(o, opt) ((o) &= ~BTRFS_MOUNT_##opt) | ||
| 960 | #define btrfs_set_opt(o, opt) ((o) |= BTRFS_MOUNT_##opt) | ||
| 961 | #define btrfs_test_opt(root, opt) ((root)->fs_info->mount_opt & \ | ||
| 962 | BTRFS_MOUNT_##opt) | ||
| 963 | /* | ||
| 964 | * Inode flags | ||
| 965 | */ | ||
| 966 | #define BTRFS_INODE_NODATASUM (1 << 0) | ||
| 967 | #define BTRFS_INODE_NODATACOW (1 << 1) | ||
| 968 | #define BTRFS_INODE_READONLY (1 << 2) | ||
| 969 | #define BTRFS_INODE_NOCOMPRESS (1 << 3) | ||
| 970 | #define BTRFS_INODE_PREALLOC (1 << 4) | ||
| 971 | #define btrfs_clear_flag(inode, flag) (BTRFS_I(inode)->flags &= \ | ||
| 972 | ~BTRFS_INODE_##flag) | ||
| 973 | #define btrfs_set_flag(inode, flag) (BTRFS_I(inode)->flags |= \ | ||
| 974 | BTRFS_INODE_##flag) | ||
| 975 | #define btrfs_test_flag(inode, flag) (BTRFS_I(inode)->flags & \ | ||
| 976 | BTRFS_INODE_##flag) | ||
| 977 | /* some macros to generate set/get funcs for the struct fields. This | ||
| 978 | * assumes there is a lefoo_to_cpu for every type, so lets make a simple | ||
| 979 | * one for u8: | ||
| 980 | */ | ||
| 981 | #define le8_to_cpu(v) (v) | ||
| 982 | #define cpu_to_le8(v) (v) | ||
| 983 | #define __le8 u8 | ||
| 984 | |||
| 985 | #define read_eb_member(eb, ptr, type, member, result) ( \ | ||
| 986 | read_extent_buffer(eb, (char *)(result), \ | ||
| 987 | ((unsigned long)(ptr)) + \ | ||
| 988 | offsetof(type, member), \ | ||
| 989 | sizeof(((type *)0)->member))) | ||
| 990 | |||
| 991 | #define write_eb_member(eb, ptr, type, member, result) ( \ | ||
| 992 | write_extent_buffer(eb, (char *)(result), \ | ||
| 993 | ((unsigned long)(ptr)) + \ | ||
| 994 | offsetof(type, member), \ | ||
| 995 | sizeof(((type *)0)->member))) | ||
| 996 | |||
| 997 | #ifndef BTRFS_SETGET_FUNCS | ||
| 998 | #define BTRFS_SETGET_FUNCS(name, type, member, bits) \ | ||
| 999 | u##bits btrfs_##name(struct extent_buffer *eb, type *s); \ | ||
| 1000 | void btrfs_set_##name(struct extent_buffer *eb, type *s, u##bits val); | ||
| 1001 | #endif | ||
| 1002 | |||
| 1003 | #define BTRFS_SETGET_HEADER_FUNCS(name, type, member, bits) \ | ||
| 1004 | static inline u##bits btrfs_##name(struct extent_buffer *eb) \ | ||
| 1005 | { \ | ||
| 1006 | type *p = kmap_atomic(eb->first_page, KM_USER0); \ | ||
| 1007 | u##bits res = le##bits##_to_cpu(p->member); \ | ||
| 1008 | kunmap_atomic(p, KM_USER0); \ | ||
| 1009 | return res; \ | ||
| 1010 | } \ | ||
| 1011 | static inline void btrfs_set_##name(struct extent_buffer *eb, \ | ||
| 1012 | u##bits val) \ | ||
| 1013 | { \ | ||
| 1014 | type *p = kmap_atomic(eb->first_page, KM_USER0); \ | ||
| 1015 | p->member = cpu_to_le##bits(val); \ | ||
| 1016 | kunmap_atomic(p, KM_USER0); \ | ||
| 1017 | } | ||
| 1018 | |||
| 1019 | #define BTRFS_SETGET_STACK_FUNCS(name, type, member, bits) \ | ||
| 1020 | static inline u##bits btrfs_##name(type *s) \ | ||
| 1021 | { \ | ||
| 1022 | return le##bits##_to_cpu(s->member); \ | ||
| 1023 | } \ | ||
| 1024 | static inline void btrfs_set_##name(type *s, u##bits val) \ | ||
| 1025 | { \ | ||
| 1026 | s->member = cpu_to_le##bits(val); \ | ||
| 1027 | } | ||
| 1028 | |||
| 1029 | BTRFS_SETGET_FUNCS(device_type, struct btrfs_dev_item, type, 64); | ||
| 1030 | BTRFS_SETGET_FUNCS(device_total_bytes, struct btrfs_dev_item, total_bytes, 64); | ||
| 1031 | BTRFS_SETGET_FUNCS(device_bytes_used, struct btrfs_dev_item, bytes_used, 64); | ||
| 1032 | BTRFS_SETGET_FUNCS(device_io_align, struct btrfs_dev_item, io_align, 32); | ||
| 1033 | BTRFS_SETGET_FUNCS(device_io_width, struct btrfs_dev_item, io_width, 32); | ||
| 1034 | BTRFS_SETGET_FUNCS(device_start_offset, struct btrfs_dev_item, | ||
| 1035 | start_offset, 64); | ||
| 1036 | BTRFS_SETGET_FUNCS(device_sector_size, struct btrfs_dev_item, sector_size, 32); | ||
| 1037 | BTRFS_SETGET_FUNCS(device_id, struct btrfs_dev_item, devid, 64); | ||
| 1038 | BTRFS_SETGET_FUNCS(device_group, struct btrfs_dev_item, dev_group, 32); | ||
| 1039 | BTRFS_SETGET_FUNCS(device_seek_speed, struct btrfs_dev_item, seek_speed, 8); | ||
| 1040 | BTRFS_SETGET_FUNCS(device_bandwidth, struct btrfs_dev_item, bandwidth, 8); | ||
| 1041 | BTRFS_SETGET_FUNCS(device_generation, struct btrfs_dev_item, generation, 64); | ||
| 1042 | |||
| 1043 | BTRFS_SETGET_STACK_FUNCS(stack_device_type, struct btrfs_dev_item, type, 64); | ||
| 1044 | BTRFS_SETGET_STACK_FUNCS(stack_device_total_bytes, struct btrfs_dev_item, | ||
| 1045 | total_bytes, 64); | ||
| 1046 | BTRFS_SETGET_STACK_FUNCS(stack_device_bytes_used, struct btrfs_dev_item, | ||
| 1047 | bytes_used, 64); | ||
| 1048 | BTRFS_SETGET_STACK_FUNCS(stack_device_io_align, struct btrfs_dev_item, | ||
| 1049 | io_align, 32); | ||
| 1050 | BTRFS_SETGET_STACK_FUNCS(stack_device_io_width, struct btrfs_dev_item, | ||
| 1051 | io_width, 32); | ||
| 1052 | BTRFS_SETGET_STACK_FUNCS(stack_device_sector_size, struct btrfs_dev_item, | ||
| 1053 | sector_size, 32); | ||
| 1054 | BTRFS_SETGET_STACK_FUNCS(stack_device_id, struct btrfs_dev_item, devid, 64); | ||
| 1055 | BTRFS_SETGET_STACK_FUNCS(stack_device_group, struct btrfs_dev_item, | ||
| 1056 | dev_group, 32); | ||
| 1057 | BTRFS_SETGET_STACK_FUNCS(stack_device_seek_speed, struct btrfs_dev_item, | ||
| 1058 | seek_speed, 8); | ||
| 1059 | BTRFS_SETGET_STACK_FUNCS(stack_device_bandwidth, struct btrfs_dev_item, | ||
| 1060 | bandwidth, 8); | ||
| 1061 | BTRFS_SETGET_STACK_FUNCS(stack_device_generation, struct btrfs_dev_item, | ||
| 1062 | generation, 64); | ||
| 1063 | |||
| 1064 | static inline char *btrfs_device_uuid(struct btrfs_dev_item *d) | ||
| 1065 | { | ||
| 1066 | return (char *)d + offsetof(struct btrfs_dev_item, uuid); | ||
| 1067 | } | ||
| 1068 | |||
| 1069 | static inline char *btrfs_device_fsid(struct btrfs_dev_item *d) | ||
| 1070 | { | ||
| 1071 | return (char *)d + offsetof(struct btrfs_dev_item, fsid); | ||
| 1072 | } | ||
| 1073 | |||
| 1074 | BTRFS_SETGET_FUNCS(chunk_length, struct btrfs_chunk, length, 64); | ||
| 1075 | BTRFS_SETGET_FUNCS(chunk_owner, struct btrfs_chunk, owner, 64); | ||
| 1076 | BTRFS_SETGET_FUNCS(chunk_stripe_len, struct btrfs_chunk, stripe_len, 64); | ||
| 1077 | BTRFS_SETGET_FUNCS(chunk_io_align, struct btrfs_chunk, io_align, 32); | ||
| 1078 | BTRFS_SETGET_FUNCS(chunk_io_width, struct btrfs_chunk, io_width, 32); | ||
| 1079 | BTRFS_SETGET_FUNCS(chunk_sector_size, struct btrfs_chunk, sector_size, 32); | ||
| 1080 | BTRFS_SETGET_FUNCS(chunk_type, struct btrfs_chunk, type, 64); | ||
| 1081 | BTRFS_SETGET_FUNCS(chunk_num_stripes, struct btrfs_chunk, num_stripes, 16); | ||
| 1082 | BTRFS_SETGET_FUNCS(chunk_sub_stripes, struct btrfs_chunk, sub_stripes, 16); | ||
| 1083 | BTRFS_SETGET_FUNCS(stripe_devid, struct btrfs_stripe, devid, 64); | ||
| 1084 | BTRFS_SETGET_FUNCS(stripe_offset, struct btrfs_stripe, offset, 64); | ||
| 1085 | |||
| 1086 | static inline char *btrfs_stripe_dev_uuid(struct btrfs_stripe *s) | ||
| 1087 | { | ||
| 1088 | return (char *)s + offsetof(struct btrfs_stripe, dev_uuid); | ||
| 1089 | } | ||
| 1090 | |||
| 1091 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_length, struct btrfs_chunk, length, 64); | ||
| 1092 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_owner, struct btrfs_chunk, owner, 64); | ||
| 1093 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_stripe_len, struct btrfs_chunk, | ||
| 1094 | stripe_len, 64); | ||
| 1095 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_io_align, struct btrfs_chunk, | ||
| 1096 | io_align, 32); | ||
| 1097 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_io_width, struct btrfs_chunk, | ||
| 1098 | io_width, 32); | ||
| 1099 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_sector_size, struct btrfs_chunk, | ||
| 1100 | sector_size, 32); | ||
| 1101 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_type, struct btrfs_chunk, type, 64); | ||
| 1102 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_num_stripes, struct btrfs_chunk, | ||
| 1103 | num_stripes, 16); | ||
| 1104 | BTRFS_SETGET_STACK_FUNCS(stack_chunk_sub_stripes, struct btrfs_chunk, | ||
| 1105 | sub_stripes, 16); | ||
| 1106 | BTRFS_SETGET_STACK_FUNCS(stack_stripe_devid, struct btrfs_stripe, devid, 64); | ||
| 1107 | BTRFS_SETGET_STACK_FUNCS(stack_stripe_offset, struct btrfs_stripe, offset, 64); | ||
| 1108 | |||
| 1109 | static inline struct btrfs_stripe *btrfs_stripe_nr(struct btrfs_chunk *c, | ||
| 1110 | int nr) | ||
| 1111 | { | ||
| 1112 | unsigned long offset = (unsigned long)c; | ||
| 1113 | offset += offsetof(struct btrfs_chunk, stripe); | ||
| 1114 | offset += nr * sizeof(struct btrfs_stripe); | ||
| 1115 | return (struct btrfs_stripe *)offset; | ||
| 1116 | } | ||
| 1117 | |||
| 1118 | static inline char *btrfs_stripe_dev_uuid_nr(struct btrfs_chunk *c, int nr) | ||
| 1119 | { | ||
| 1120 | return btrfs_stripe_dev_uuid(btrfs_stripe_nr(c, nr)); | ||
| 1121 | } | ||
| 1122 | |||
| 1123 | static inline u64 btrfs_stripe_offset_nr(struct extent_buffer *eb, | ||
| 1124 | struct btrfs_chunk *c, int nr) | ||
| 1125 | { | ||
| 1126 | return btrfs_stripe_offset(eb, btrfs_stripe_nr(c, nr)); | ||
| 1127 | } | ||
| 1128 | |||
| 1129 | static inline void btrfs_set_stripe_offset_nr(struct extent_buffer *eb, | ||
| 1130 | struct btrfs_chunk *c, int nr, | ||
| 1131 | u64 val) | ||
| 1132 | { | ||
| 1133 | btrfs_set_stripe_offset(eb, btrfs_stripe_nr(c, nr), val); | ||
| 1134 | } | ||
| 1135 | |||
| 1136 | static inline u64 btrfs_stripe_devid_nr(struct extent_buffer *eb, | ||
| 1137 | struct btrfs_chunk *c, int nr) | ||
| 1138 | { | ||
| 1139 | return btrfs_stripe_devid(eb, btrfs_stripe_nr(c, nr)); | ||
| 1140 | } | ||
| 1141 | |||
| 1142 | static inline void btrfs_set_stripe_devid_nr(struct extent_buffer *eb, | ||
| 1143 | struct btrfs_chunk *c, int nr, | ||
| 1144 | u64 val) | ||
| 1145 | { | ||
| 1146 | btrfs_set_stripe_devid(eb, btrfs_stripe_nr(c, nr), val); | ||
| 1147 | } | ||
| 1148 | |||
| 1149 | /* struct btrfs_block_group_item */ | ||
| 1150 | BTRFS_SETGET_STACK_FUNCS(block_group_used, struct btrfs_block_group_item, | ||
| 1151 | used, 64); | ||
| 1152 | BTRFS_SETGET_FUNCS(disk_block_group_used, struct btrfs_block_group_item, | ||
| 1153 | used, 64); | ||
| 1154 | BTRFS_SETGET_STACK_FUNCS(block_group_chunk_objectid, | ||
| 1155 | struct btrfs_block_group_item, chunk_objectid, 64); | ||
| 1156 | |||
| 1157 | BTRFS_SETGET_FUNCS(disk_block_group_chunk_objectid, | ||
| 1158 | struct btrfs_block_group_item, chunk_objectid, 64); | ||
| 1159 | BTRFS_SETGET_FUNCS(disk_block_group_flags, | ||
| 1160 | struct btrfs_block_group_item, flags, 64); | ||
| 1161 | BTRFS_SETGET_STACK_FUNCS(block_group_flags, | ||
| 1162 | struct btrfs_block_group_item, flags, 64); | ||
| 1163 | |||
| 1164 | /* struct btrfs_inode_ref */ | ||
| 1165 | BTRFS_SETGET_FUNCS(inode_ref_name_len, struct btrfs_inode_ref, name_len, 16); | ||
| 1166 | BTRFS_SETGET_FUNCS(inode_ref_index, struct btrfs_inode_ref, index, 64); | ||
| 1167 | |||
| 1168 | /* struct btrfs_inode_item */ | ||
| 1169 | BTRFS_SETGET_FUNCS(inode_generation, struct btrfs_inode_item, generation, 64); | ||
| 1170 | BTRFS_SETGET_FUNCS(inode_sequence, struct btrfs_inode_item, sequence, 64); | ||
| 1171 | BTRFS_SETGET_FUNCS(inode_transid, struct btrfs_inode_item, transid, 64); | ||
| 1172 | BTRFS_SETGET_FUNCS(inode_size, struct btrfs_inode_item, size, 64); | ||
| 1173 | BTRFS_SETGET_FUNCS(inode_nbytes, struct btrfs_inode_item, nbytes, 64); | ||
| 1174 | BTRFS_SETGET_FUNCS(inode_block_group, struct btrfs_inode_item, block_group, 64); | ||
| 1175 | BTRFS_SETGET_FUNCS(inode_nlink, struct btrfs_inode_item, nlink, 32); | ||
| 1176 | BTRFS_SETGET_FUNCS(inode_uid, struct btrfs_inode_item, uid, 32); | ||
| 1177 | BTRFS_SETGET_FUNCS(inode_gid, struct btrfs_inode_item, gid, 32); | ||
| 1178 | BTRFS_SETGET_FUNCS(inode_mode, struct btrfs_inode_item, mode, 32); | ||
| 1179 | BTRFS_SETGET_FUNCS(inode_rdev, struct btrfs_inode_item, rdev, 64); | ||
| 1180 | BTRFS_SETGET_FUNCS(inode_flags, struct btrfs_inode_item, flags, 64); | ||
| 1181 | |||
| 1182 | static inline struct btrfs_timespec * | ||
| 1183 | btrfs_inode_atime(struct btrfs_inode_item *inode_item) | ||
| 1184 | { | ||
| 1185 | unsigned long ptr = (unsigned long)inode_item; | ||
| 1186 | ptr += offsetof(struct btrfs_inode_item, atime); | ||
| 1187 | return (struct btrfs_timespec *)ptr; | ||
| 1188 | } | ||
| 1189 | |||
| 1190 | static inline struct btrfs_timespec * | ||
| 1191 | btrfs_inode_mtime(struct btrfs_inode_item *inode_item) | ||
| 1192 | { | ||
| 1193 | unsigned long ptr = (unsigned long)inode_item; | ||
| 1194 | ptr += offsetof(struct btrfs_inode_item, mtime); | ||
| 1195 | return (struct btrfs_timespec *)ptr; | ||
| 1196 | } | ||
| 1197 | |||
| 1198 | static inline struct btrfs_timespec * | ||
| 1199 | btrfs_inode_ctime(struct btrfs_inode_item *inode_item) | ||
| 1200 | { | ||
| 1201 | unsigned long ptr = (unsigned long)inode_item; | ||
| 1202 | ptr += offsetof(struct btrfs_inode_item, ctime); | ||
| 1203 | return (struct btrfs_timespec *)ptr; | ||
| 1204 | } | ||
| 1205 | |||
| 1206 | static inline struct btrfs_timespec * | ||
| 1207 | btrfs_inode_otime(struct btrfs_inode_item *inode_item) | ||
| 1208 | { | ||
| 1209 | unsigned long ptr = (unsigned long)inode_item; | ||
| 1210 | ptr += offsetof(struct btrfs_inode_item, otime); | ||
| 1211 | return (struct btrfs_timespec *)ptr; | ||
| 1212 | } | ||
| 1213 | |||
| 1214 | BTRFS_SETGET_FUNCS(timespec_sec, struct btrfs_timespec, sec, 64); | ||
| 1215 | BTRFS_SETGET_FUNCS(timespec_nsec, struct btrfs_timespec, nsec, 32); | ||
| 1216 | |||
| 1217 | /* struct btrfs_dev_extent */ | ||
| 1218 | BTRFS_SETGET_FUNCS(dev_extent_chunk_tree, struct btrfs_dev_extent, | ||
| 1219 | chunk_tree, 64); | ||
| 1220 | BTRFS_SETGET_FUNCS(dev_extent_chunk_objectid, struct btrfs_dev_extent, | ||
| 1221 | chunk_objectid, 64); | ||
| 1222 | BTRFS_SETGET_FUNCS(dev_extent_chunk_offset, struct btrfs_dev_extent, | ||
| 1223 | chunk_offset, 64); | ||
| 1224 | BTRFS_SETGET_FUNCS(dev_extent_length, struct btrfs_dev_extent, length, 64); | ||
| 1225 | |||
| 1226 | static inline u8 *btrfs_dev_extent_chunk_tree_uuid(struct btrfs_dev_extent *dev) | ||
| 1227 | { | ||
| 1228 | unsigned long ptr = offsetof(struct btrfs_dev_extent, chunk_tree_uuid); | ||
| 1229 | return (u8 *)((unsigned long)dev + ptr); | ||
| 1230 | } | ||
| 1231 | |||
| 1232 | /* struct btrfs_extent_ref */ | ||
| 1233 | BTRFS_SETGET_FUNCS(ref_root, struct btrfs_extent_ref, root, 64); | ||
| 1234 | BTRFS_SETGET_FUNCS(ref_generation, struct btrfs_extent_ref, generation, 64); | ||
| 1235 | BTRFS_SETGET_FUNCS(ref_objectid, struct btrfs_extent_ref, objectid, 64); | ||
| 1236 | BTRFS_SETGET_FUNCS(ref_num_refs, struct btrfs_extent_ref, num_refs, 32); | ||
| 1237 | |||
| 1238 | BTRFS_SETGET_STACK_FUNCS(stack_ref_root, struct btrfs_extent_ref, root, 64); | ||
| 1239 | BTRFS_SETGET_STACK_FUNCS(stack_ref_generation, struct btrfs_extent_ref, | ||
| 1240 | generation, 64); | ||
| 1241 | BTRFS_SETGET_STACK_FUNCS(stack_ref_objectid, struct btrfs_extent_ref, | ||
| 1242 | objectid, 64); | ||
| 1243 | BTRFS_SETGET_STACK_FUNCS(stack_ref_num_refs, struct btrfs_extent_ref, | ||
| 1244 | num_refs, 32); | ||
| 1245 | |||
| 1246 | /* struct btrfs_extent_item */ | ||
| 1247 | BTRFS_SETGET_FUNCS(extent_refs, struct btrfs_extent_item, refs, 32); | ||
| 1248 | BTRFS_SETGET_STACK_FUNCS(stack_extent_refs, struct btrfs_extent_item, | ||
| 1249 | refs, 32); | ||
| 1250 | |||
| 1251 | /* struct btrfs_node */ | ||
| 1252 | BTRFS_SETGET_FUNCS(key_blockptr, struct btrfs_key_ptr, blockptr, 64); | ||
| 1253 | BTRFS_SETGET_FUNCS(key_generation, struct btrfs_key_ptr, generation, 64); | ||
| 1254 | |||
| 1255 | static inline u64 btrfs_node_blockptr(struct extent_buffer *eb, int nr) | ||
| 1256 | { | ||
| 1257 | unsigned long ptr; | ||
| 1258 | ptr = offsetof(struct btrfs_node, ptrs) + | ||
| 1259 | sizeof(struct btrfs_key_ptr) * nr; | ||
| 1260 | return btrfs_key_blockptr(eb, (struct btrfs_key_ptr *)ptr); | ||
| 1261 | } | ||
| 1262 | |||
| 1263 | static inline void btrfs_set_node_blockptr(struct extent_buffer *eb, | ||
| 1264 | int nr, u64 val) | ||
| 1265 | { | ||
| 1266 | unsigned long ptr; | ||
| 1267 | ptr = offsetof(struct btrfs_node, ptrs) + | ||
| 1268 | sizeof(struct btrfs_key_ptr) * nr; | ||
| 1269 | btrfs_set_key_blockptr(eb, (struct btrfs_key_ptr *)ptr, val); | ||
| 1270 | } | ||
| 1271 | |||
| 1272 | static inline u64 btrfs_node_ptr_generation(struct extent_buffer *eb, int nr) | ||
| 1273 | { | ||
| 1274 | unsigned long ptr; | ||
| 1275 | ptr = offsetof(struct btrfs_node, ptrs) + | ||
| 1276 | sizeof(struct btrfs_key_ptr) * nr; | ||
| 1277 | return btrfs_key_generation(eb, (struct btrfs_key_ptr *)ptr); | ||
| 1278 | } | ||
| 1279 | |||
| 1280 | static inline void btrfs_set_node_ptr_generation(struct extent_buffer *eb, | ||
| 1281 | int nr, u64 val) | ||
| 1282 | { | ||
| 1283 | unsigned long ptr; | ||
| 1284 | ptr = offsetof(struct btrfs_node, ptrs) + | ||
| 1285 | sizeof(struct btrfs_key_ptr) * nr; | ||
| 1286 | btrfs_set_key_generation(eb, (struct btrfs_key_ptr *)ptr, val); | ||
| 1287 | } | ||
| 1288 | |||
| 1289 | static inline unsigned long btrfs_node_key_ptr_offset(int nr) | ||
| 1290 | { | ||
| 1291 | return offsetof(struct btrfs_node, ptrs) + | ||
| 1292 | sizeof(struct btrfs_key_ptr) * nr; | ||
| 1293 | } | ||
| 1294 | |||
| 1295 | void btrfs_node_key(struct extent_buffer *eb, | ||
| 1296 | struct btrfs_disk_key *disk_key, int nr); | ||
| 1297 | |||
| 1298 | static inline void btrfs_set_node_key(struct extent_buffer *eb, | ||
| 1299 | struct btrfs_disk_key *disk_key, int nr) | ||
| 1300 | { | ||
| 1301 | unsigned long ptr; | ||
| 1302 | ptr = btrfs_node_key_ptr_offset(nr); | ||
| 1303 | write_eb_member(eb, (struct btrfs_key_ptr *)ptr, | ||
| 1304 | struct btrfs_key_ptr, key, disk_key); | ||
| 1305 | } | ||
| 1306 | |||
| 1307 | /* struct btrfs_item */ | ||
| 1308 | BTRFS_SETGET_FUNCS(item_offset, struct btrfs_item, offset, 32); | ||
| 1309 | BTRFS_SETGET_FUNCS(item_size, struct btrfs_item, size, 32); | ||
| 1310 | |||
| 1311 | static inline unsigned long btrfs_item_nr_offset(int nr) | ||
| 1312 | { | ||
| 1313 | return offsetof(struct btrfs_leaf, items) + | ||
| 1314 | sizeof(struct btrfs_item) * nr; | ||
| 1315 | } | ||
| 1316 | |||
| 1317 | static inline struct btrfs_item *btrfs_item_nr(struct extent_buffer *eb, | ||
| 1318 | int nr) | ||
| 1319 | { | ||
| 1320 | return (struct btrfs_item *)btrfs_item_nr_offset(nr); | ||
| 1321 | } | ||
| 1322 | |||
| 1323 | static inline u32 btrfs_item_end(struct extent_buffer *eb, | ||
| 1324 | struct btrfs_item *item) | ||
| 1325 | { | ||
| 1326 | return btrfs_item_offset(eb, item) + btrfs_item_size(eb, item); | ||
| 1327 | } | ||
| 1328 | |||
| 1329 | static inline u32 btrfs_item_end_nr(struct extent_buffer *eb, int nr) | ||
| 1330 | { | ||
| 1331 | return btrfs_item_end(eb, btrfs_item_nr(eb, nr)); | ||
| 1332 | } | ||
| 1333 | |||
| 1334 | static inline u32 btrfs_item_offset_nr(struct extent_buffer *eb, int nr) | ||
| 1335 | { | ||
| 1336 | return btrfs_item_offset(eb, btrfs_item_nr(eb, nr)); | ||
| 1337 | } | ||
| 1338 | |||
| 1339 | static inline u32 btrfs_item_size_nr(struct extent_buffer *eb, int nr) | ||
| 1340 | { | ||
| 1341 | return btrfs_item_size(eb, btrfs_item_nr(eb, nr)); | ||
| 1342 | } | ||
| 1343 | |||
| 1344 | static inline void btrfs_item_key(struct extent_buffer *eb, | ||
| 1345 | struct btrfs_disk_key *disk_key, int nr) | ||
| 1346 | { | ||
| 1347 | struct btrfs_item *item = btrfs_item_nr(eb, nr); | ||
| 1348 | read_eb_member(eb, item, struct btrfs_item, key, disk_key); | ||
| 1349 | } | ||
| 1350 | |||
| 1351 | static inline void btrfs_set_item_key(struct extent_buffer *eb, | ||
| 1352 | struct btrfs_disk_key *disk_key, int nr) | ||
| 1353 | { | ||
| 1354 | struct btrfs_item *item = btrfs_item_nr(eb, nr); | ||
| 1355 | write_eb_member(eb, item, struct btrfs_item, key, disk_key); | ||
| 1356 | } | ||
| 1357 | |||
| 1358 | BTRFS_SETGET_FUNCS(dir_log_end, struct btrfs_dir_log_item, end, 64); | ||
| 1359 | |||
| 1360 | /* | ||
| 1361 | * struct btrfs_root_ref | ||
| 1362 | */ | ||
| 1363 | BTRFS_SETGET_FUNCS(root_ref_dirid, struct btrfs_root_ref, dirid, 64); | ||
| 1364 | BTRFS_SETGET_FUNCS(root_ref_sequence, struct btrfs_root_ref, sequence, 64); | ||
| 1365 | BTRFS_SETGET_FUNCS(root_ref_name_len, struct btrfs_root_ref, name_len, 16); | ||
| 1366 | |||
| 1367 | /* struct btrfs_dir_item */ | ||
| 1368 | BTRFS_SETGET_FUNCS(dir_data_len, struct btrfs_dir_item, data_len, 16); | ||
| 1369 | BTRFS_SETGET_FUNCS(dir_type, struct btrfs_dir_item, type, 8); | ||
| 1370 | BTRFS_SETGET_FUNCS(dir_name_len, struct btrfs_dir_item, name_len, 16); | ||
| 1371 | BTRFS_SETGET_FUNCS(dir_transid, struct btrfs_dir_item, transid, 64); | ||
| 1372 | |||
| 1373 | static inline void btrfs_dir_item_key(struct extent_buffer *eb, | ||
| 1374 | struct btrfs_dir_item *item, | ||
| 1375 | struct btrfs_disk_key *key) | ||
| 1376 | { | ||
| 1377 | read_eb_member(eb, item, struct btrfs_dir_item, location, key); | ||
| 1378 | } | ||
| 1379 | |||
| 1380 | static inline void btrfs_set_dir_item_key(struct extent_buffer *eb, | ||
| 1381 | struct btrfs_dir_item *item, | ||
| 1382 | struct btrfs_disk_key *key) | ||
| 1383 | { | ||
| 1384 | write_eb_member(eb, item, struct btrfs_dir_item, location, key); | ||
| 1385 | } | ||
| 1386 | |||
| 1387 | /* struct btrfs_disk_key */ | ||
| 1388 | BTRFS_SETGET_STACK_FUNCS(disk_key_objectid, struct btrfs_disk_key, | ||
| 1389 | objectid, 64); | ||
| 1390 | BTRFS_SETGET_STACK_FUNCS(disk_key_offset, struct btrfs_disk_key, offset, 64); | ||
| 1391 | BTRFS_SETGET_STACK_FUNCS(disk_key_type, struct btrfs_disk_key, type, 8); | ||
| 1392 | |||
| 1393 | static inline void btrfs_disk_key_to_cpu(struct btrfs_key *cpu, | ||
| 1394 | struct btrfs_disk_key *disk) | ||
| 1395 | { | ||
| 1396 | cpu->offset = le64_to_cpu(disk->offset); | ||
| 1397 | cpu->type = disk->type; | ||
| 1398 | cpu->objectid = le64_to_cpu(disk->objectid); | ||
| 1399 | } | ||
| 1400 | |||
| 1401 | static inline void btrfs_cpu_key_to_disk(struct btrfs_disk_key *disk, | ||
| 1402 | struct btrfs_key *cpu) | ||
| 1403 | { | ||
| 1404 | disk->offset = cpu_to_le64(cpu->offset); | ||
| 1405 | disk->type = cpu->type; | ||
| 1406 | disk->objectid = cpu_to_le64(cpu->objectid); | ||
| 1407 | } | ||
| 1408 | |||
| 1409 | static inline void btrfs_node_key_to_cpu(struct extent_buffer *eb, | ||
| 1410 | struct btrfs_key *key, int nr) | ||
| 1411 | { | ||
| 1412 | struct btrfs_disk_key disk_key; | ||
| 1413 | btrfs_node_key(eb, &disk_key, nr); | ||
| 1414 | btrfs_disk_key_to_cpu(key, &disk_key); | ||
| 1415 | } | ||
| 1416 | |||
| 1417 | static inline void btrfs_item_key_to_cpu(struct extent_buffer *eb, | ||
| 1418 | struct btrfs_key *key, int nr) | ||
| 1419 | { | ||
| 1420 | struct btrfs_disk_key disk_key; | ||
| 1421 | btrfs_item_key(eb, &disk_key, nr); | ||
| 1422 | btrfs_disk_key_to_cpu(key, &disk_key); | ||
| 1423 | } | ||
| 1424 | |||
| 1425 | static inline void btrfs_dir_item_key_to_cpu(struct extent_buffer *eb, | ||
| 1426 | struct btrfs_dir_item *item, | ||
| 1427 | struct btrfs_key *key) | ||
| 1428 | { | ||
| 1429 | struct btrfs_disk_key disk_key; | ||
| 1430 | btrfs_dir_item_key(eb, item, &disk_key); | ||
| 1431 | btrfs_disk_key_to_cpu(key, &disk_key); | ||
| 1432 | } | ||
| 1433 | |||
| 1434 | |||
| 1435 | static inline u8 btrfs_key_type(struct btrfs_key *key) | ||
| 1436 | { | ||
| 1437 | return key->type; | ||
| 1438 | } | ||
| 1439 | |||
| 1440 | static inline void btrfs_set_key_type(struct btrfs_key *key, u8 val) | ||
| 1441 | { | ||
| 1442 | key->type = val; | ||
| 1443 | } | ||
| 1444 | |||
| 1445 | /* struct btrfs_header */ | ||
| 1446 | BTRFS_SETGET_HEADER_FUNCS(header_bytenr, struct btrfs_header, bytenr, 64); | ||
| 1447 | BTRFS_SETGET_HEADER_FUNCS(header_generation, struct btrfs_header, | ||
| 1448 | generation, 64); | ||
| 1449 | BTRFS_SETGET_HEADER_FUNCS(header_owner, struct btrfs_header, owner, 64); | ||
| 1450 | BTRFS_SETGET_HEADER_FUNCS(header_nritems, struct btrfs_header, nritems, 32); | ||
| 1451 | BTRFS_SETGET_HEADER_FUNCS(header_flags, struct btrfs_header, flags, 64); | ||
| 1452 | BTRFS_SETGET_HEADER_FUNCS(header_level, struct btrfs_header, level, 8); | ||
| 1453 | |||
| 1454 | static inline int btrfs_header_flag(struct extent_buffer *eb, u64 flag) | ||
| 1455 | { | ||
| 1456 | return (btrfs_header_flags(eb) & flag) == flag; | ||
| 1457 | } | ||
| 1458 | |||
| 1459 | static inline int btrfs_set_header_flag(struct extent_buffer *eb, u64 flag) | ||
| 1460 | { | ||
| 1461 | u64 flags = btrfs_header_flags(eb); | ||
| 1462 | btrfs_set_header_flags(eb, flags | flag); | ||
| 1463 | return (flags & flag) == flag; | ||
| 1464 | } | ||
| 1465 | |||
| 1466 | static inline int btrfs_clear_header_flag(struct extent_buffer *eb, u64 flag) | ||
| 1467 | { | ||
| 1468 | u64 flags = btrfs_header_flags(eb); | ||
| 1469 | btrfs_set_header_flags(eb, flags & ~flag); | ||
| 1470 | return (flags & flag) == flag; | ||
| 1471 | } | ||
| 1472 | |||
| 1473 | static inline u8 *btrfs_header_fsid(struct extent_buffer *eb) | ||
| 1474 | { | ||
| 1475 | unsigned long ptr = offsetof(struct btrfs_header, fsid); | ||
| 1476 | return (u8 *)ptr; | ||
| 1477 | } | ||
| 1478 | |||
| 1479 | static inline u8 *btrfs_header_chunk_tree_uuid(struct extent_buffer *eb) | ||
| 1480 | { | ||
| 1481 | unsigned long ptr = offsetof(struct btrfs_header, chunk_tree_uuid); | ||
| 1482 | return (u8 *)ptr; | ||
| 1483 | } | ||
| 1484 | |||
| 1485 | static inline u8 *btrfs_super_fsid(struct extent_buffer *eb) | ||
| 1486 | { | ||
| 1487 | unsigned long ptr = offsetof(struct btrfs_super_block, fsid); | ||
| 1488 | return (u8 *)ptr; | ||
| 1489 | } | ||
| 1490 | |||
| 1491 | static inline u8 *btrfs_header_csum(struct extent_buffer *eb) | ||
| 1492 | { | ||
| 1493 | unsigned long ptr = offsetof(struct btrfs_header, csum); | ||
| 1494 | return (u8 *)ptr; | ||
| 1495 | } | ||
| 1496 | |||
| 1497 | static inline struct btrfs_node *btrfs_buffer_node(struct extent_buffer *eb) | ||
| 1498 | { | ||
| 1499 | return NULL; | ||
| 1500 | } | ||
| 1501 | |||
| 1502 | static inline struct btrfs_leaf *btrfs_buffer_leaf(struct extent_buffer *eb) | ||
| 1503 | { | ||
| 1504 | return NULL; | ||
| 1505 | } | ||
| 1506 | |||
| 1507 | static inline struct btrfs_header *btrfs_buffer_header(struct extent_buffer *eb) | ||
| 1508 | { | ||
| 1509 | return NULL; | ||
| 1510 | } | ||
| 1511 | |||
| 1512 | static inline int btrfs_is_leaf(struct extent_buffer *eb) | ||
| 1513 | { | ||
| 1514 | return btrfs_header_level(eb) == 0; | ||
| 1515 | } | ||
| 1516 | |||
| 1517 | /* struct btrfs_root_item */ | ||
| 1518 | BTRFS_SETGET_FUNCS(disk_root_generation, struct btrfs_root_item, | ||
| 1519 | generation, 64); | ||
| 1520 | BTRFS_SETGET_FUNCS(disk_root_refs, struct btrfs_root_item, refs, 32); | ||
| 1521 | BTRFS_SETGET_FUNCS(disk_root_bytenr, struct btrfs_root_item, bytenr, 64); | ||
| 1522 | BTRFS_SETGET_FUNCS(disk_root_level, struct btrfs_root_item, level, 8); | ||
| 1523 | |||
| 1524 | BTRFS_SETGET_STACK_FUNCS(root_generation, struct btrfs_root_item, | ||
| 1525 | generation, 64); | ||
| 1526 | BTRFS_SETGET_STACK_FUNCS(root_bytenr, struct btrfs_root_item, bytenr, 64); | ||
| 1527 | BTRFS_SETGET_STACK_FUNCS(root_level, struct btrfs_root_item, level, 8); | ||
| 1528 | BTRFS_SETGET_STACK_FUNCS(root_dirid, struct btrfs_root_item, root_dirid, 64); | ||
| 1529 | BTRFS_SETGET_STACK_FUNCS(root_refs, struct btrfs_root_item, refs, 32); | ||
| 1530 | BTRFS_SETGET_STACK_FUNCS(root_flags, struct btrfs_root_item, flags, 64); | ||
| 1531 | BTRFS_SETGET_STACK_FUNCS(root_used, struct btrfs_root_item, bytes_used, 64); | ||
| 1532 | BTRFS_SETGET_STACK_FUNCS(root_limit, struct btrfs_root_item, byte_limit, 64); | ||
| 1533 | BTRFS_SETGET_STACK_FUNCS(root_last_snapshot, struct btrfs_root_item, | ||
| 1534 | last_snapshot, 64); | ||
| 1535 | |||
| 1536 | /* struct btrfs_super_block */ | ||
| 1537 | |||
| 1538 | BTRFS_SETGET_STACK_FUNCS(super_bytenr, struct btrfs_super_block, bytenr, 64); | ||
| 1539 | BTRFS_SETGET_STACK_FUNCS(super_flags, struct btrfs_super_block, flags, 64); | ||
| 1540 | BTRFS_SETGET_STACK_FUNCS(super_generation, struct btrfs_super_block, | ||
| 1541 | generation, 64); | ||
| 1542 | BTRFS_SETGET_STACK_FUNCS(super_root, struct btrfs_super_block, root, 64); | ||
| 1543 | BTRFS_SETGET_STACK_FUNCS(super_sys_array_size, | ||
| 1544 | struct btrfs_super_block, sys_chunk_array_size, 32); | ||
| 1545 | BTRFS_SETGET_STACK_FUNCS(super_chunk_root_generation, | ||
| 1546 | struct btrfs_super_block, chunk_root_generation, 64); | ||
| 1547 | BTRFS_SETGET_STACK_FUNCS(super_root_level, struct btrfs_super_block, | ||
| 1548 | root_level, 8); | ||
| 1549 | BTRFS_SETGET_STACK_FUNCS(super_chunk_root, struct btrfs_super_block, | ||
| 1550 | chunk_root, 64); | ||
| 1551 | BTRFS_SETGET_STACK_FUNCS(super_chunk_root_level, struct btrfs_super_block, | ||
| 1552 | chunk_root_level, 8); | ||
| 1553 | BTRFS_SETGET_STACK_FUNCS(super_log_root, struct btrfs_super_block, | ||
| 1554 | log_root, 64); | ||
| 1555 | BTRFS_SETGET_STACK_FUNCS(super_log_root_transid, struct btrfs_super_block, | ||
| 1556 | log_root_transid, 64); | ||
| 1557 | BTRFS_SETGET_STACK_FUNCS(super_log_root_level, struct btrfs_super_block, | ||
| 1558 | log_root_level, 8); | ||
| 1559 | BTRFS_SETGET_STACK_FUNCS(super_total_bytes, struct btrfs_super_block, | ||
| 1560 | total_bytes, 64); | ||
| 1561 | BTRFS_SETGET_STACK_FUNCS(super_bytes_used, struct btrfs_super_block, | ||
| 1562 | bytes_used, 64); | ||
| 1563 | BTRFS_SETGET_STACK_FUNCS(super_sectorsize, struct btrfs_super_block, | ||
| 1564 | sectorsize, 32); | ||
| 1565 | BTRFS_SETGET_STACK_FUNCS(super_nodesize, struct btrfs_super_block, | ||
| 1566 | nodesize, 32); | ||
| 1567 | BTRFS_SETGET_STACK_FUNCS(super_leafsize, struct btrfs_super_block, | ||
| 1568 | leafsize, 32); | ||
| 1569 | BTRFS_SETGET_STACK_FUNCS(super_stripesize, struct btrfs_super_block, | ||
| 1570 | stripesize, 32); | ||
| 1571 | BTRFS_SETGET_STACK_FUNCS(super_root_dir, struct btrfs_super_block, | ||
| 1572 | root_dir_objectid, 64); | ||
| 1573 | BTRFS_SETGET_STACK_FUNCS(super_num_devices, struct btrfs_super_block, | ||
| 1574 | num_devices, 64); | ||
| 1575 | BTRFS_SETGET_STACK_FUNCS(super_compat_flags, struct btrfs_super_block, | ||
| 1576 | compat_flags, 64); | ||
| 1577 | BTRFS_SETGET_STACK_FUNCS(super_compat_ro_flags, struct btrfs_super_block, | ||
| 1578 | compat_flags, 64); | ||
| 1579 | BTRFS_SETGET_STACK_FUNCS(super_incompat_flags, struct btrfs_super_block, | ||
| 1580 | incompat_flags, 64); | ||
| 1581 | BTRFS_SETGET_STACK_FUNCS(super_csum_type, struct btrfs_super_block, | ||
| 1582 | csum_type, 16); | ||
| 1583 | |||
| 1584 | static inline int btrfs_super_csum_size(struct btrfs_super_block *s) | ||
| 1585 | { | ||
| 1586 | int t = btrfs_super_csum_type(s); | ||
| 1587 | BUG_ON(t >= ARRAY_SIZE(btrfs_csum_sizes)); | ||
| 1588 | return btrfs_csum_sizes[t]; | ||
| 1589 | } | ||
| 1590 | |||
| 1591 | static inline unsigned long btrfs_leaf_data(struct extent_buffer *l) | ||
| 1592 | { | ||
| 1593 | return offsetof(struct btrfs_leaf, items); | ||
| 1594 | } | ||
| 1595 | |||
| 1596 | /* struct btrfs_file_extent_item */ | ||
| 1597 | BTRFS_SETGET_FUNCS(file_extent_type, struct btrfs_file_extent_item, type, 8); | ||
| 1598 | |||
| 1599 | static inline unsigned long | ||
| 1600 | btrfs_file_extent_inline_start(struct btrfs_file_extent_item *e) | ||
| 1601 | { | ||
| 1602 | unsigned long offset = (unsigned long)e; | ||
| 1603 | offset += offsetof(struct btrfs_file_extent_item, disk_bytenr); | ||
| 1604 | return offset; | ||
| 1605 | } | ||
| 1606 | |||
| 1607 | static inline u32 btrfs_file_extent_calc_inline_size(u32 datasize) | ||
| 1608 | { | ||
| 1609 | return offsetof(struct btrfs_file_extent_item, disk_bytenr) + datasize; | ||
| 1610 | } | ||
| 1611 | |||
| 1612 | BTRFS_SETGET_FUNCS(file_extent_disk_bytenr, struct btrfs_file_extent_item, | ||
| 1613 | disk_bytenr, 64); | ||
| 1614 | BTRFS_SETGET_FUNCS(file_extent_generation, struct btrfs_file_extent_item, | ||
| 1615 | generation, 64); | ||
| 1616 | BTRFS_SETGET_FUNCS(file_extent_disk_num_bytes, struct btrfs_file_extent_item, | ||
| 1617 | disk_num_bytes, 64); | ||
| 1618 | BTRFS_SETGET_FUNCS(file_extent_offset, struct btrfs_file_extent_item, | ||
| 1619 | offset, 64); | ||
| 1620 | BTRFS_SETGET_FUNCS(file_extent_num_bytes, struct btrfs_file_extent_item, | ||
| 1621 | num_bytes, 64); | ||
| 1622 | BTRFS_SETGET_FUNCS(file_extent_ram_bytes, struct btrfs_file_extent_item, | ||
| 1623 | ram_bytes, 64); | ||
| 1624 | BTRFS_SETGET_FUNCS(file_extent_compression, struct btrfs_file_extent_item, | ||
| 1625 | compression, 8); | ||
| 1626 | BTRFS_SETGET_FUNCS(file_extent_encryption, struct btrfs_file_extent_item, | ||
| 1627 | encryption, 8); | ||
| 1628 | BTRFS_SETGET_FUNCS(file_extent_other_encoding, struct btrfs_file_extent_item, | ||
| 1629 | other_encoding, 16); | ||
| 1630 | |||
| 1631 | /* this returns the number of file bytes represented by the inline item. | ||
| 1632 | * If an item is compressed, this is the uncompressed size | ||
| 1633 | */ | ||
| 1634 | static inline u32 btrfs_file_extent_inline_len(struct extent_buffer *eb, | ||
| 1635 | struct btrfs_file_extent_item *e) | ||
| 1636 | { | ||
| 1637 | return btrfs_file_extent_ram_bytes(eb, e); | ||
| 1638 | } | ||
| 1639 | |||
| 1640 | /* | ||
| 1641 | * this returns the number of bytes used by the item on disk, minus the | ||
| 1642 | * size of any extent headers. If a file is compressed on disk, this is | ||
| 1643 | * the compressed size | ||
| 1644 | */ | ||
| 1645 | static inline u32 btrfs_file_extent_inline_item_len(struct extent_buffer *eb, | ||
| 1646 | struct btrfs_item *e) | ||
| 1647 | { | ||
| 1648 | unsigned long offset; | ||
| 1649 | offset = offsetof(struct btrfs_file_extent_item, disk_bytenr); | ||
| 1650 | return btrfs_item_size(eb, e) - offset; | ||
| 1651 | } | ||
| 1652 | |||
| 1653 | static inline struct btrfs_root *btrfs_sb(struct super_block *sb) | ||
| 1654 | { | ||
| 1655 | return sb->s_fs_info; | ||
| 1656 | } | ||
| 1657 | |||
| 1658 | static inline int btrfs_set_root_name(struct btrfs_root *root, | ||
| 1659 | const char *name, int len) | ||
| 1660 | { | ||
| 1661 | /* if we already have a name just free it */ | ||
| 1662 | kfree(root->name); | ||
| 1663 | |||
| 1664 | root->name = kmalloc(len+1, GFP_KERNEL); | ||
| 1665 | if (!root->name) | ||
| 1666 | return -ENOMEM; | ||
| 1667 | |||
| 1668 | memcpy(root->name, name, len); | ||
| 1669 | root->name[len] = '\0'; | ||
| 1670 | |||
| 1671 | return 0; | ||
| 1672 | } | ||
| 1673 | |||
| 1674 | static inline u32 btrfs_level_size(struct btrfs_root *root, int level) | ||
| 1675 | { | ||
| 1676 | if (level == 0) | ||
| 1677 | return root->leafsize; | ||
| 1678 | return root->nodesize; | ||
| 1679 | } | ||
| 1680 | |||
| 1681 | /* helper function to cast into the data area of the leaf. */ | ||
| 1682 | #define btrfs_item_ptr(leaf, slot, type) \ | ||
| 1683 | ((type *)(btrfs_leaf_data(leaf) + \ | ||
| 1684 | btrfs_item_offset_nr(leaf, slot))) | ||
| 1685 | |||
| 1686 | #define btrfs_item_ptr_offset(leaf, slot) \ | ||
| 1687 | ((unsigned long)(btrfs_leaf_data(leaf) + \ | ||
| 1688 | btrfs_item_offset_nr(leaf, slot))) | ||
| 1689 | |||
| 1690 | static inline struct dentry *fdentry(struct file *file) | ||
| 1691 | { | ||
| 1692 | return file->f_path.dentry; | ||
| 1693 | } | ||
| 1694 | |||
| 1695 | /* extent-tree.c */ | ||
| 1696 | int btrfs_lookup_extent(struct btrfs_root *root, u64 start, u64 len); | ||
| 1697 | int btrfs_lookup_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1698 | struct btrfs_root *root, u64 bytenr, | ||
| 1699 | u64 num_bytes, u32 *refs); | ||
| 1700 | int btrfs_update_pinned_extents(struct btrfs_root *root, | ||
| 1701 | u64 bytenr, u64 num, int pin); | ||
| 1702 | int btrfs_drop_leaf_ref(struct btrfs_trans_handle *trans, | ||
| 1703 | struct btrfs_root *root, struct extent_buffer *leaf); | ||
| 1704 | int btrfs_cross_ref_exist(struct btrfs_trans_handle *trans, | ||
| 1705 | struct btrfs_root *root, u64 objectid, u64 bytenr); | ||
| 1706 | int btrfs_extent_post_op(struct btrfs_trans_handle *trans, | ||
| 1707 | struct btrfs_root *root); | ||
| 1708 | int btrfs_copy_pinned(struct btrfs_root *root, struct extent_io_tree *copy); | ||
| 1709 | struct btrfs_block_group_cache *btrfs_lookup_block_group( | ||
| 1710 | struct btrfs_fs_info *info, | ||
| 1711 | u64 bytenr); | ||
| 1712 | u64 btrfs_find_block_group(struct btrfs_root *root, | ||
| 1713 | u64 search_start, u64 search_hint, int owner); | ||
| 1714 | struct extent_buffer *btrfs_alloc_free_block(struct btrfs_trans_handle *trans, | ||
| 1715 | struct btrfs_root *root, | ||
| 1716 | u32 blocksize, u64 parent, | ||
| 1717 | u64 root_objectid, | ||
| 1718 | u64 ref_generation, | ||
| 1719 | int level, | ||
| 1720 | u64 hint, | ||
| 1721 | u64 empty_size); | ||
| 1722 | struct extent_buffer *btrfs_init_new_buffer(struct btrfs_trans_handle *trans, | ||
| 1723 | struct btrfs_root *root, | ||
| 1724 | u64 bytenr, u32 blocksize); | ||
| 1725 | int btrfs_alloc_extent(struct btrfs_trans_handle *trans, | ||
| 1726 | struct btrfs_root *root, | ||
| 1727 | u64 num_bytes, u64 parent, u64 min_bytes, | ||
| 1728 | u64 root_objectid, u64 ref_generation, | ||
| 1729 | u64 owner, u64 empty_size, u64 hint_byte, | ||
| 1730 | u64 search_end, struct btrfs_key *ins, u64 data); | ||
| 1731 | int btrfs_alloc_reserved_extent(struct btrfs_trans_handle *trans, | ||
| 1732 | struct btrfs_root *root, u64 parent, | ||
| 1733 | u64 root_objectid, u64 ref_generation, | ||
| 1734 | u64 owner, struct btrfs_key *ins); | ||
| 1735 | int btrfs_alloc_logged_extent(struct btrfs_trans_handle *trans, | ||
| 1736 | struct btrfs_root *root, u64 parent, | ||
| 1737 | u64 root_objectid, u64 ref_generation, | ||
| 1738 | u64 owner, struct btrfs_key *ins); | ||
| 1739 | int btrfs_reserve_extent(struct btrfs_trans_handle *trans, | ||
| 1740 | struct btrfs_root *root, | ||
| 1741 | u64 num_bytes, u64 min_alloc_size, | ||
| 1742 | u64 empty_size, u64 hint_byte, | ||
| 1743 | u64 search_end, struct btrfs_key *ins, | ||
| 1744 | u64 data); | ||
| 1745 | int btrfs_inc_ref(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 1746 | struct extent_buffer *orig_buf, struct extent_buffer *buf, | ||
| 1747 | u32 *nr_extents); | ||
| 1748 | int btrfs_cache_ref(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 1749 | struct extent_buffer *buf, u32 nr_extents); | ||
| 1750 | int btrfs_update_ref(struct btrfs_trans_handle *trans, | ||
| 1751 | struct btrfs_root *root, struct extent_buffer *orig_buf, | ||
| 1752 | struct extent_buffer *buf, int start_slot, int nr); | ||
| 1753 | int btrfs_free_extent(struct btrfs_trans_handle *trans, | ||
| 1754 | struct btrfs_root *root, | ||
| 1755 | u64 bytenr, u64 num_bytes, u64 parent, | ||
| 1756 | u64 root_objectid, u64 ref_generation, | ||
| 1757 | u64 owner_objectid, int pin); | ||
| 1758 | int btrfs_free_reserved_extent(struct btrfs_root *root, u64 start, u64 len); | ||
| 1759 | int btrfs_finish_extent_commit(struct btrfs_trans_handle *trans, | ||
| 1760 | struct btrfs_root *root, | ||
| 1761 | struct extent_io_tree *unpin); | ||
| 1762 | int btrfs_inc_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1763 | struct btrfs_root *root, | ||
| 1764 | u64 bytenr, u64 num_bytes, u64 parent, | ||
| 1765 | u64 root_objectid, u64 ref_generation, | ||
| 1766 | u64 owner_objectid); | ||
| 1767 | int btrfs_update_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1768 | struct btrfs_root *root, u64 bytenr, | ||
| 1769 | u64 orig_parent, u64 parent, | ||
| 1770 | u64 root_objectid, u64 ref_generation, | ||
| 1771 | u64 owner_objectid); | ||
| 1772 | int btrfs_write_dirty_block_groups(struct btrfs_trans_handle *trans, | ||
| 1773 | struct btrfs_root *root); | ||
| 1774 | int btrfs_extent_readonly(struct btrfs_root *root, u64 bytenr); | ||
| 1775 | int btrfs_free_block_groups(struct btrfs_fs_info *info); | ||
| 1776 | int btrfs_read_block_groups(struct btrfs_root *root); | ||
| 1777 | int btrfs_make_block_group(struct btrfs_trans_handle *trans, | ||
| 1778 | struct btrfs_root *root, u64 bytes_used, | ||
| 1779 | u64 type, u64 chunk_objectid, u64 chunk_offset, | ||
| 1780 | u64 size); | ||
| 1781 | int btrfs_remove_block_group(struct btrfs_trans_handle *trans, | ||
| 1782 | struct btrfs_root *root, u64 group_start); | ||
| 1783 | int btrfs_relocate_block_group(struct btrfs_root *root, u64 group_start); | ||
| 1784 | int btrfs_free_reloc_root(struct btrfs_trans_handle *trans, | ||
| 1785 | struct btrfs_root *root); | ||
| 1786 | int btrfs_drop_dead_reloc_roots(struct btrfs_root *root); | ||
| 1787 | int btrfs_reloc_tree_cache_ref(struct btrfs_trans_handle *trans, | ||
| 1788 | struct btrfs_root *root, | ||
| 1789 | struct extent_buffer *buf, u64 orig_start); | ||
| 1790 | int btrfs_add_dead_reloc_root(struct btrfs_root *root); | ||
| 1791 | int btrfs_cleanup_reloc_trees(struct btrfs_root *root); | ||
| 1792 | int btrfs_reloc_clone_csums(struct inode *inode, u64 file_pos, u64 len); | ||
| 1793 | u64 btrfs_reduce_alloc_profile(struct btrfs_root *root, u64 flags); | ||
| 1794 | /* ctree.c */ | ||
| 1795 | int btrfs_previous_item(struct btrfs_root *root, | ||
| 1796 | struct btrfs_path *path, u64 min_objectid, | ||
| 1797 | int type); | ||
| 1798 | int btrfs_merge_path(struct btrfs_trans_handle *trans, | ||
| 1799 | struct btrfs_root *root, | ||
| 1800 | struct btrfs_key *node_keys, | ||
| 1801 | u64 *nodes, int lowest_level); | ||
| 1802 | int btrfs_set_item_key_safe(struct btrfs_trans_handle *trans, | ||
| 1803 | struct btrfs_root *root, struct btrfs_path *path, | ||
| 1804 | struct btrfs_key *new_key); | ||
| 1805 | struct extent_buffer *btrfs_root_node(struct btrfs_root *root); | ||
| 1806 | struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root); | ||
| 1807 | int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path, | ||
| 1808 | struct btrfs_key *key, int lowest_level, | ||
| 1809 | int cache_only, u64 min_trans); | ||
| 1810 | int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key, | ||
| 1811 | struct btrfs_key *max_key, | ||
| 1812 | struct btrfs_path *path, int cache_only, | ||
| 1813 | u64 min_trans); | ||
| 1814 | int btrfs_cow_block(struct btrfs_trans_handle *trans, | ||
| 1815 | struct btrfs_root *root, struct extent_buffer *buf, | ||
| 1816 | struct extent_buffer *parent, int parent_slot, | ||
| 1817 | struct extent_buffer **cow_ret, u64 prealloc_dest); | ||
| 1818 | int btrfs_copy_root(struct btrfs_trans_handle *trans, | ||
| 1819 | struct btrfs_root *root, | ||
| 1820 | struct extent_buffer *buf, | ||
| 1821 | struct extent_buffer **cow_ret, u64 new_root_objectid); | ||
| 1822 | int btrfs_extend_item(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1823 | *root, struct btrfs_path *path, u32 data_size); | ||
| 1824 | int btrfs_truncate_item(struct btrfs_trans_handle *trans, | ||
| 1825 | struct btrfs_root *root, | ||
| 1826 | struct btrfs_path *path, | ||
| 1827 | u32 new_size, int from_end); | ||
| 1828 | int btrfs_split_item(struct btrfs_trans_handle *trans, | ||
| 1829 | struct btrfs_root *root, | ||
| 1830 | struct btrfs_path *path, | ||
| 1831 | struct btrfs_key *new_key, | ||
| 1832 | unsigned long split_offset); | ||
| 1833 | int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1834 | *root, struct btrfs_key *key, struct btrfs_path *p, int | ||
| 1835 | ins_len, int cow); | ||
| 1836 | int btrfs_realloc_node(struct btrfs_trans_handle *trans, | ||
| 1837 | struct btrfs_root *root, struct extent_buffer *parent, | ||
| 1838 | int start_slot, int cache_only, u64 *last_ret, | ||
| 1839 | struct btrfs_key *progress); | ||
| 1840 | void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p); | ||
| 1841 | struct btrfs_path *btrfs_alloc_path(void); | ||
| 1842 | void btrfs_free_path(struct btrfs_path *p); | ||
| 1843 | void btrfs_init_path(struct btrfs_path *p); | ||
| 1844 | int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 1845 | struct btrfs_path *path, int slot, int nr); | ||
| 1846 | int btrfs_del_leaf(struct btrfs_trans_handle *trans, | ||
| 1847 | struct btrfs_root *root, | ||
| 1848 | struct btrfs_path *path, u64 bytenr); | ||
| 1849 | static inline int btrfs_del_item(struct btrfs_trans_handle *trans, | ||
| 1850 | struct btrfs_root *root, | ||
| 1851 | struct btrfs_path *path) | ||
| 1852 | { | ||
| 1853 | return btrfs_del_items(trans, root, path, path->slots[0], 1); | ||
| 1854 | } | ||
| 1855 | |||
| 1856 | int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1857 | *root, struct btrfs_key *key, void *data, u32 data_size); | ||
| 1858 | int btrfs_insert_some_items(struct btrfs_trans_handle *trans, | ||
| 1859 | struct btrfs_root *root, | ||
| 1860 | struct btrfs_path *path, | ||
| 1861 | struct btrfs_key *cpu_key, u32 *data_size, | ||
| 1862 | int nr); | ||
| 1863 | int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, | ||
| 1864 | struct btrfs_root *root, | ||
| 1865 | struct btrfs_path *path, | ||
| 1866 | struct btrfs_key *cpu_key, u32 *data_size, int nr); | ||
| 1867 | |||
| 1868 | static inline int btrfs_insert_empty_item(struct btrfs_trans_handle *trans, | ||
| 1869 | struct btrfs_root *root, | ||
| 1870 | struct btrfs_path *path, | ||
| 1871 | struct btrfs_key *key, | ||
| 1872 | u32 data_size) | ||
| 1873 | { | ||
| 1874 | return btrfs_insert_empty_items(trans, root, path, key, &data_size, 1); | ||
| 1875 | } | ||
| 1876 | |||
| 1877 | int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path); | ||
| 1878 | int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path); | ||
| 1879 | int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf); | ||
| 1880 | int btrfs_drop_snapshot(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1881 | *root); | ||
| 1882 | int btrfs_drop_subtree(struct btrfs_trans_handle *trans, | ||
| 1883 | struct btrfs_root *root, | ||
| 1884 | struct extent_buffer *node, | ||
| 1885 | struct extent_buffer *parent); | ||
| 1886 | /* root-item.c */ | ||
| 1887 | int btrfs_find_root_ref(struct btrfs_root *tree_root, | ||
| 1888 | struct btrfs_path *path, | ||
| 1889 | u64 root_id, u64 ref_id); | ||
| 1890 | int btrfs_add_root_ref(struct btrfs_trans_handle *trans, | ||
| 1891 | struct btrfs_root *tree_root, | ||
| 1892 | u64 root_id, u8 type, u64 ref_id, | ||
| 1893 | u64 dirid, u64 sequence, | ||
| 1894 | const char *name, int name_len); | ||
| 1895 | int btrfs_del_root(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 1896 | struct btrfs_key *key); | ||
| 1897 | int btrfs_insert_root(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1898 | *root, struct btrfs_key *key, struct btrfs_root_item | ||
| 1899 | *item); | ||
| 1900 | int btrfs_update_root(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1901 | *root, struct btrfs_key *key, struct btrfs_root_item | ||
| 1902 | *item); | ||
| 1903 | int btrfs_find_last_root(struct btrfs_root *root, u64 objectid, struct | ||
| 1904 | btrfs_root_item *item, struct btrfs_key *key); | ||
| 1905 | int btrfs_search_root(struct btrfs_root *root, u64 search_start, | ||
| 1906 | u64 *found_objectid); | ||
| 1907 | int btrfs_find_dead_roots(struct btrfs_root *root, u64 objectid, | ||
| 1908 | struct btrfs_root *latest_root); | ||
| 1909 | /* dir-item.c */ | ||
| 1910 | int btrfs_insert_dir_item(struct btrfs_trans_handle *trans, | ||
| 1911 | struct btrfs_root *root, const char *name, | ||
| 1912 | int name_len, u64 dir, | ||
| 1913 | struct btrfs_key *location, u8 type, u64 index); | ||
| 1914 | struct btrfs_dir_item *btrfs_lookup_dir_item(struct btrfs_trans_handle *trans, | ||
| 1915 | struct btrfs_root *root, | ||
| 1916 | struct btrfs_path *path, u64 dir, | ||
| 1917 | const char *name, int name_len, | ||
| 1918 | int mod); | ||
| 1919 | struct btrfs_dir_item * | ||
| 1920 | btrfs_lookup_dir_index_item(struct btrfs_trans_handle *trans, | ||
| 1921 | struct btrfs_root *root, | ||
| 1922 | struct btrfs_path *path, u64 dir, | ||
| 1923 | u64 objectid, const char *name, int name_len, | ||
| 1924 | int mod); | ||
| 1925 | struct btrfs_dir_item *btrfs_match_dir_item_name(struct btrfs_root *root, | ||
| 1926 | struct btrfs_path *path, | ||
| 1927 | const char *name, int name_len); | ||
| 1928 | int btrfs_delete_one_dir_name(struct btrfs_trans_handle *trans, | ||
| 1929 | struct btrfs_root *root, | ||
| 1930 | struct btrfs_path *path, | ||
| 1931 | struct btrfs_dir_item *di); | ||
| 1932 | int btrfs_insert_xattr_item(struct btrfs_trans_handle *trans, | ||
| 1933 | struct btrfs_root *root, const char *name, | ||
| 1934 | u16 name_len, const void *data, u16 data_len, | ||
| 1935 | u64 dir); | ||
| 1936 | struct btrfs_dir_item *btrfs_lookup_xattr(struct btrfs_trans_handle *trans, | ||
| 1937 | struct btrfs_root *root, | ||
| 1938 | struct btrfs_path *path, u64 dir, | ||
| 1939 | const char *name, u16 name_len, | ||
| 1940 | int mod); | ||
| 1941 | |||
| 1942 | /* orphan.c */ | ||
| 1943 | int btrfs_insert_orphan_item(struct btrfs_trans_handle *trans, | ||
| 1944 | struct btrfs_root *root, u64 offset); | ||
| 1945 | int btrfs_del_orphan_item(struct btrfs_trans_handle *trans, | ||
| 1946 | struct btrfs_root *root, u64 offset); | ||
| 1947 | |||
| 1948 | /* inode-map.c */ | ||
| 1949 | int btrfs_find_free_objectid(struct btrfs_trans_handle *trans, | ||
| 1950 | struct btrfs_root *fs_root, | ||
| 1951 | u64 dirid, u64 *objectid); | ||
| 1952 | int btrfs_find_highest_inode(struct btrfs_root *fs_root, u64 *objectid); | ||
| 1953 | |||
| 1954 | /* inode-item.c */ | ||
| 1955 | int btrfs_insert_inode_ref(struct btrfs_trans_handle *trans, | ||
| 1956 | struct btrfs_root *root, | ||
| 1957 | const char *name, int name_len, | ||
| 1958 | u64 inode_objectid, u64 ref_objectid, u64 index); | ||
| 1959 | int btrfs_del_inode_ref(struct btrfs_trans_handle *trans, | ||
| 1960 | struct btrfs_root *root, | ||
| 1961 | const char *name, int name_len, | ||
| 1962 | u64 inode_objectid, u64 ref_objectid, u64 *index); | ||
| 1963 | int btrfs_insert_empty_inode(struct btrfs_trans_handle *trans, | ||
| 1964 | struct btrfs_root *root, | ||
| 1965 | struct btrfs_path *path, u64 objectid); | ||
| 1966 | int btrfs_lookup_inode(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 1967 | *root, struct btrfs_path *path, | ||
| 1968 | struct btrfs_key *location, int mod); | ||
| 1969 | |||
| 1970 | /* file-item.c */ | ||
| 1971 | int btrfs_del_csums(struct btrfs_trans_handle *trans, | ||
| 1972 | struct btrfs_root *root, u64 bytenr, u64 len); | ||
| 1973 | int btrfs_lookup_bio_sums(struct btrfs_root *root, struct inode *inode, | ||
| 1974 | struct bio *bio, u32 *dst); | ||
| 1975 | int btrfs_insert_file_extent(struct btrfs_trans_handle *trans, | ||
| 1976 | struct btrfs_root *root, | ||
| 1977 | u64 objectid, u64 pos, | ||
| 1978 | u64 disk_offset, u64 disk_num_bytes, | ||
| 1979 | u64 num_bytes, u64 offset, u64 ram_bytes, | ||
| 1980 | u8 compression, u8 encryption, u16 other_encoding); | ||
| 1981 | int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans, | ||
| 1982 | struct btrfs_root *root, | ||
| 1983 | struct btrfs_path *path, u64 objectid, | ||
| 1984 | u64 bytenr, int mod); | ||
| 1985 | int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans, | ||
| 1986 | struct btrfs_root *root, | ||
| 1987 | struct btrfs_ordered_sum *sums); | ||
| 1988 | int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode, | ||
| 1989 | struct bio *bio, u64 file_start, int contig); | ||
| 1990 | int btrfs_csum_file_bytes(struct btrfs_root *root, struct inode *inode, | ||
| 1991 | u64 start, unsigned long len); | ||
| 1992 | struct btrfs_csum_item *btrfs_lookup_csum(struct btrfs_trans_handle *trans, | ||
| 1993 | struct btrfs_root *root, | ||
| 1994 | struct btrfs_path *path, | ||
| 1995 | u64 bytenr, int cow); | ||
| 1996 | int btrfs_csum_truncate(struct btrfs_trans_handle *trans, | ||
| 1997 | struct btrfs_root *root, struct btrfs_path *path, | ||
| 1998 | u64 isize); | ||
| 1999 | int btrfs_lookup_csums_range(struct btrfs_root *root, u64 start, | ||
| 2000 | u64 end, struct list_head *list); | ||
| 2001 | /* inode.c */ | ||
| 2002 | |||
| 2003 | /* RHEL and EL kernels have a patch that renames PG_checked to FsMisc */ | ||
| 2004 | #if defined(ClearPageFsMisc) && !defined(ClearPageChecked) | ||
| 2005 | #define ClearPageChecked ClearPageFsMisc | ||
| 2006 | #define SetPageChecked SetPageFsMisc | ||
| 2007 | #define PageChecked PageFsMisc | ||
| 2008 | #endif | ||
| 2009 | |||
| 2010 | struct inode *btrfs_lookup_dentry(struct inode *dir, struct dentry *dentry); | ||
| 2011 | int btrfs_set_inode_index(struct inode *dir, u64 *index); | ||
| 2012 | int btrfs_unlink_inode(struct btrfs_trans_handle *trans, | ||
| 2013 | struct btrfs_root *root, | ||
| 2014 | struct inode *dir, struct inode *inode, | ||
| 2015 | const char *name, int name_len); | ||
| 2016 | int btrfs_add_link(struct btrfs_trans_handle *trans, | ||
| 2017 | struct inode *parent_inode, struct inode *inode, | ||
| 2018 | const char *name, int name_len, int add_backref, u64 index); | ||
| 2019 | int btrfs_truncate_inode_items(struct btrfs_trans_handle *trans, | ||
| 2020 | struct btrfs_root *root, | ||
| 2021 | struct inode *inode, u64 new_size, | ||
| 2022 | u32 min_type); | ||
| 2023 | |||
| 2024 | int btrfs_start_delalloc_inodes(struct btrfs_root *root); | ||
| 2025 | int btrfs_set_extent_delalloc(struct inode *inode, u64 start, u64 end); | ||
| 2026 | int btrfs_writepages(struct address_space *mapping, | ||
| 2027 | struct writeback_control *wbc); | ||
| 2028 | int btrfs_create_subvol_root(struct btrfs_trans_handle *trans, | ||
| 2029 | struct btrfs_root *new_root, struct dentry *dentry, | ||
| 2030 | u64 new_dirid, u64 alloc_hint); | ||
| 2031 | int btrfs_merge_bio_hook(struct page *page, unsigned long offset, | ||
| 2032 | size_t size, struct bio *bio, unsigned long bio_flags); | ||
| 2033 | |||
| 2034 | unsigned long btrfs_force_ra(struct address_space *mapping, | ||
| 2035 | struct file_ra_state *ra, struct file *file, | ||
| 2036 | pgoff_t offset, pgoff_t last_index); | ||
| 2037 | int btrfs_check_free_space(struct btrfs_root *root, u64 num_required, | ||
| 2038 | int for_del); | ||
| 2039 | int btrfs_page_mkwrite(struct vm_area_struct *vma, struct page *page); | ||
| 2040 | int btrfs_readpage(struct file *file, struct page *page); | ||
| 2041 | void btrfs_delete_inode(struct inode *inode); | ||
| 2042 | void btrfs_put_inode(struct inode *inode); | ||
| 2043 | void btrfs_read_locked_inode(struct inode *inode); | ||
| 2044 | int btrfs_write_inode(struct inode *inode, int wait); | ||
| 2045 | void btrfs_dirty_inode(struct inode *inode); | ||
| 2046 | struct inode *btrfs_alloc_inode(struct super_block *sb); | ||
| 2047 | void btrfs_destroy_inode(struct inode *inode); | ||
| 2048 | int btrfs_init_cachep(void); | ||
| 2049 | void btrfs_destroy_cachep(void); | ||
| 2050 | long btrfs_ioctl_trans_end(struct file *file); | ||
| 2051 | struct inode *btrfs_ilookup(struct super_block *s, u64 objectid, | ||
| 2052 | struct btrfs_root *root, int wait); | ||
| 2053 | struct inode *btrfs_iget_locked(struct super_block *s, u64 objectid, | ||
| 2054 | struct btrfs_root *root); | ||
| 2055 | struct inode *btrfs_iget(struct super_block *s, struct btrfs_key *location, | ||
| 2056 | struct btrfs_root *root, int *is_new); | ||
| 2057 | int btrfs_commit_write(struct file *file, struct page *page, | ||
| 2058 | unsigned from, unsigned to); | ||
| 2059 | struct extent_map *btrfs_get_extent(struct inode *inode, struct page *page, | ||
| 2060 | size_t page_offset, u64 start, u64 end, | ||
| 2061 | int create); | ||
| 2062 | int btrfs_update_inode(struct btrfs_trans_handle *trans, | ||
| 2063 | struct btrfs_root *root, | ||
| 2064 | struct inode *inode); | ||
| 2065 | int btrfs_orphan_add(struct btrfs_trans_handle *trans, struct inode *inode); | ||
| 2066 | int btrfs_orphan_del(struct btrfs_trans_handle *trans, struct inode *inode); | ||
| 2067 | void btrfs_orphan_cleanup(struct btrfs_root *root); | ||
| 2068 | int btrfs_cont_expand(struct inode *inode, loff_t size); | ||
| 2069 | |||
| 2070 | /* ioctl.c */ | ||
| 2071 | long btrfs_ioctl(struct file *file, unsigned int cmd, unsigned long arg); | ||
| 2072 | |||
| 2073 | /* file.c */ | ||
| 2074 | int btrfs_sync_file(struct file *file, struct dentry *dentry, int datasync); | ||
| 2075 | int btrfs_drop_extent_cache(struct inode *inode, u64 start, u64 end, | ||
| 2076 | int skip_pinned); | ||
| 2077 | int btrfs_check_file(struct btrfs_root *root, struct inode *inode); | ||
| 2078 | extern struct file_operations btrfs_file_operations; | ||
| 2079 | int btrfs_drop_extents(struct btrfs_trans_handle *trans, | ||
| 2080 | struct btrfs_root *root, struct inode *inode, | ||
| 2081 | u64 start, u64 end, u64 inline_limit, u64 *hint_block); | ||
| 2082 | int btrfs_mark_extent_written(struct btrfs_trans_handle *trans, | ||
| 2083 | struct btrfs_root *root, | ||
| 2084 | struct inode *inode, u64 start, u64 end); | ||
| 2085 | int btrfs_release_file(struct inode *inode, struct file *file); | ||
| 2086 | |||
| 2087 | /* tree-defrag.c */ | ||
| 2088 | int btrfs_defrag_leaves(struct btrfs_trans_handle *trans, | ||
| 2089 | struct btrfs_root *root, int cache_only); | ||
| 2090 | |||
| 2091 | /* sysfs.c */ | ||
| 2092 | int btrfs_init_sysfs(void); | ||
| 2093 | void btrfs_exit_sysfs(void); | ||
| 2094 | int btrfs_sysfs_add_super(struct btrfs_fs_info *fs); | ||
| 2095 | int btrfs_sysfs_add_root(struct btrfs_root *root); | ||
| 2096 | void btrfs_sysfs_del_root(struct btrfs_root *root); | ||
| 2097 | void btrfs_sysfs_del_super(struct btrfs_fs_info *root); | ||
| 2098 | |||
| 2099 | /* xattr.c */ | ||
| 2100 | ssize_t btrfs_listxattr(struct dentry *dentry, char *buffer, size_t size); | ||
| 2101 | |||
| 2102 | /* super.c */ | ||
| 2103 | u64 btrfs_parse_size(char *str); | ||
| 2104 | int btrfs_parse_options(struct btrfs_root *root, char *options); | ||
| 2105 | int btrfs_sync_fs(struct super_block *sb, int wait); | ||
| 2106 | |||
| 2107 | /* acl.c */ | ||
| 2108 | int btrfs_check_acl(struct inode *inode, int mask); | ||
| 2109 | int btrfs_init_acl(struct inode *inode, struct inode *dir); | ||
| 2110 | int btrfs_acl_chmod(struct inode *inode); | ||
| 2111 | |||
| 2112 | /* free-space-cache.c */ | ||
| 2113 | int btrfs_add_free_space(struct btrfs_block_group_cache *block_group, | ||
| 2114 | u64 bytenr, u64 size); | ||
| 2115 | int btrfs_add_free_space_lock(struct btrfs_block_group_cache *block_group, | ||
| 2116 | u64 offset, u64 bytes); | ||
| 2117 | int btrfs_remove_free_space(struct btrfs_block_group_cache *block_group, | ||
| 2118 | u64 bytenr, u64 size); | ||
| 2119 | int btrfs_remove_free_space_lock(struct btrfs_block_group_cache *block_group, | ||
| 2120 | u64 offset, u64 bytes); | ||
| 2121 | void btrfs_remove_free_space_cache(struct btrfs_block_group_cache | ||
| 2122 | *block_group); | ||
| 2123 | struct btrfs_free_space *btrfs_find_free_space(struct btrfs_block_group_cache | ||
| 2124 | *block_group, u64 offset, | ||
| 2125 | u64 bytes); | ||
| 2126 | void btrfs_dump_free_space(struct btrfs_block_group_cache *block_group, | ||
| 2127 | u64 bytes); | ||
| 2128 | u64 btrfs_block_group_free_space(struct btrfs_block_group_cache *block_group); | ||
| 2129 | #endif | ||
diff --git a/fs/btrfs/dir-item.c b/fs/btrfs/dir-item.c new file mode 100644 index 000000000000..926a0b287a7d --- /dev/null +++ b/fs/btrfs/dir-item.c | |||
| @@ -0,0 +1,386 @@ | |||
| 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 | |||
| 19 | #include "ctree.h" | ||
| 20 | #include "disk-io.h" | ||
| 21 | #include "hash.h" | ||
| 22 | #include "transaction.h" | ||
| 23 | |||
| 24 | /* | ||
| 25 | * insert a name into a directory, doing overflow properly if there is a hash | ||
| 26 | * collision. data_size indicates how big the item inserted should be. On | ||
| 27 | * success a struct btrfs_dir_item pointer is returned, otherwise it is | ||
| 28 | * an ERR_PTR. | ||
| 29 | * | ||
| 30 | * The name is not copied into the dir item, you have to do that yourself. | ||
| 31 | */ | ||
| 32 | static struct btrfs_dir_item *insert_with_overflow(struct btrfs_trans_handle | ||
| 33 | *trans, | ||
| 34 | struct btrfs_root *root, | ||
| 35 | struct btrfs_path *path, | ||
| 36 | struct btrfs_key *cpu_key, | ||
| 37 | u32 data_size, | ||
| 38 | const char *name, | ||
| 39 | int name_len) | ||
| 40 | { | ||
| 41 | int ret; | ||
| 42 | char *ptr; | ||
| 43 | struct btrfs_item *item; | ||
| 44 | struct extent_buffer *leaf; | ||
| 45 | |||
| 46 | ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); | ||
| 47 | if (ret == -EEXIST) { | ||
| 48 | struct btrfs_dir_item *di; | ||
| 49 | di = btrfs_match_dir_item_name(root, path, name, name_len); | ||
| 50 | if (di) | ||
| 51 | return ERR_PTR(-EEXIST); | ||
| 52 | ret = btrfs_extend_item(trans, root, path, data_size); | ||
| 53 | WARN_ON(ret > 0); | ||
| 54 | } | ||
| 55 | if (ret < 0) | ||
| 56 | return ERR_PTR(ret); | ||
| 57 | WARN_ON(ret > 0); | ||
| 58 | leaf = path->nodes[0]; | ||
| 59 | item = btrfs_item_nr(leaf, path->slots[0]); | ||
| 60 | ptr = btrfs_item_ptr(leaf, path->slots[0], char); | ||
| 61 | BUG_ON(data_size > btrfs_item_size(leaf, item)); | ||
| 62 | ptr += btrfs_item_size(leaf, item) - data_size; | ||
| 63 | return (struct btrfs_dir_item *)ptr; | ||
| 64 | } | ||
| 65 | |||
| 66 | /* | ||
| 67 | * xattrs work a lot like directories, this inserts an xattr item | ||
| 68 | * into the tree | ||
| 69 | */ | ||
| 70 | int btrfs_insert_xattr_item(struct btrfs_trans_handle *trans, | ||
| 71 | struct btrfs_root *root, const char *name, | ||
| 72 | u16 name_len, const void *data, u16 data_len, | ||
| 73 | u64 dir) | ||
| 74 | { | ||
| 75 | int ret = 0; | ||
| 76 | struct btrfs_path *path; | ||
| 77 | struct btrfs_dir_item *dir_item; | ||
| 78 | unsigned long name_ptr, data_ptr; | ||
| 79 | struct btrfs_key key, location; | ||
| 80 | struct btrfs_disk_key disk_key; | ||
| 81 | struct extent_buffer *leaf; | ||
| 82 | u32 data_size; | ||
| 83 | |||
| 84 | key.objectid = dir; | ||
| 85 | btrfs_set_key_type(&key, BTRFS_XATTR_ITEM_KEY); | ||
| 86 | key.offset = btrfs_name_hash(name, name_len); | ||
| 87 | path = btrfs_alloc_path(); | ||
| 88 | if (!path) | ||
| 89 | return -ENOMEM; | ||
| 90 | if (name_len + data_len + sizeof(struct btrfs_dir_item) > | ||
| 91 | BTRFS_LEAF_DATA_SIZE(root) - sizeof(struct btrfs_item)) | ||
| 92 | return -ENOSPC; | ||
| 93 | |||
| 94 | data_size = sizeof(*dir_item) + name_len + data_len; | ||
| 95 | dir_item = insert_with_overflow(trans, root, path, &key, data_size, | ||
| 96 | name, name_len); | ||
| 97 | /* | ||
| 98 | * FIXME: at some point we should handle xattr's that are larger than | ||
| 99 | * what we can fit in our leaf. We set location to NULL b/c we arent | ||
| 100 | * pointing at anything else, that will change if we store the xattr | ||
| 101 | * data in a separate inode. | ||
| 102 | */ | ||
| 103 | BUG_ON(IS_ERR(dir_item)); | ||
| 104 | memset(&location, 0, sizeof(location)); | ||
| 105 | |||
| 106 | leaf = path->nodes[0]; | ||
| 107 | btrfs_cpu_key_to_disk(&disk_key, &location); | ||
| 108 | btrfs_set_dir_item_key(leaf, dir_item, &disk_key); | ||
| 109 | btrfs_set_dir_type(leaf, dir_item, BTRFS_FT_XATTR); | ||
| 110 | btrfs_set_dir_name_len(leaf, dir_item, name_len); | ||
| 111 | btrfs_set_dir_transid(leaf, dir_item, trans->transid); | ||
| 112 | btrfs_set_dir_data_len(leaf, dir_item, data_len); | ||
| 113 | name_ptr = (unsigned long)(dir_item + 1); | ||
| 114 | data_ptr = (unsigned long)((char *)name_ptr + name_len); | ||
| 115 | |||
| 116 | write_extent_buffer(leaf, name, name_ptr, name_len); | ||
| 117 | write_extent_buffer(leaf, data, data_ptr, data_len); | ||
| 118 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 119 | |||
| 120 | btrfs_free_path(path); | ||
| 121 | return ret; | ||
| 122 | } | ||
| 123 | |||
| 124 | /* | ||
| 125 | * insert a directory item in the tree, doing all the magic for | ||
| 126 | * both indexes. 'dir' indicates which objectid to insert it into, | ||
| 127 | * 'location' is the key to stuff into the directory item, 'type' is the | ||
| 128 | * type of the inode we're pointing to, and 'index' is the sequence number | ||
| 129 | * to use for the second index (if one is created). | ||
| 130 | */ | ||
| 131 | int btrfs_insert_dir_item(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 132 | *root, const char *name, int name_len, u64 dir, | ||
| 133 | struct btrfs_key *location, u8 type, u64 index) | ||
| 134 | { | ||
| 135 | int ret = 0; | ||
| 136 | int ret2 = 0; | ||
| 137 | struct btrfs_path *path; | ||
| 138 | struct btrfs_dir_item *dir_item; | ||
| 139 | struct extent_buffer *leaf; | ||
| 140 | unsigned long name_ptr; | ||
| 141 | struct btrfs_key key; | ||
| 142 | struct btrfs_disk_key disk_key; | ||
| 143 | u32 data_size; | ||
| 144 | |||
| 145 | key.objectid = dir; | ||
| 146 | btrfs_set_key_type(&key, BTRFS_DIR_ITEM_KEY); | ||
| 147 | key.offset = btrfs_name_hash(name, name_len); | ||
| 148 | path = btrfs_alloc_path(); | ||
| 149 | data_size = sizeof(*dir_item) + name_len; | ||
| 150 | dir_item = insert_with_overflow(trans, root, path, &key, data_size, | ||
| 151 | name, name_len); | ||
| 152 | if (IS_ERR(dir_item)) { | ||
| 153 | ret = PTR_ERR(dir_item); | ||
| 154 | if (ret == -EEXIST) | ||
| 155 | goto second_insert; | ||
| 156 | goto out; | ||
| 157 | } | ||
| 158 | |||
| 159 | leaf = path->nodes[0]; | ||
| 160 | btrfs_cpu_key_to_disk(&disk_key, location); | ||
| 161 | btrfs_set_dir_item_key(leaf, dir_item, &disk_key); | ||
| 162 | btrfs_set_dir_type(leaf, dir_item, type); | ||
| 163 | btrfs_set_dir_data_len(leaf, dir_item, 0); | ||
| 164 | btrfs_set_dir_name_len(leaf, dir_item, name_len); | ||
| 165 | btrfs_set_dir_transid(leaf, dir_item, trans->transid); | ||
| 166 | name_ptr = (unsigned long)(dir_item + 1); | ||
| 167 | |||
| 168 | write_extent_buffer(leaf, name, name_ptr, name_len); | ||
| 169 | btrfs_mark_buffer_dirty(leaf); | ||
| 170 | |||
| 171 | second_insert: | ||
| 172 | /* FIXME, use some real flag for selecting the extra index */ | ||
| 173 | if (root == root->fs_info->tree_root) { | ||
| 174 | ret = 0; | ||
| 175 | goto out; | ||
| 176 | } | ||
| 177 | btrfs_release_path(root, path); | ||
| 178 | |||
| 179 | btrfs_set_key_type(&key, BTRFS_DIR_INDEX_KEY); | ||
| 180 | key.offset = index; | ||
| 181 | dir_item = insert_with_overflow(trans, root, path, &key, data_size, | ||
| 182 | name, name_len); | ||
| 183 | if (IS_ERR(dir_item)) { | ||
| 184 | ret2 = PTR_ERR(dir_item); | ||
| 185 | goto out; | ||
| 186 | } | ||
| 187 | leaf = path->nodes[0]; | ||
| 188 | btrfs_cpu_key_to_disk(&disk_key, location); | ||
| 189 | btrfs_set_dir_item_key(leaf, dir_item, &disk_key); | ||
| 190 | btrfs_set_dir_type(leaf, dir_item, type); | ||
| 191 | btrfs_set_dir_data_len(leaf, dir_item, 0); | ||
| 192 | btrfs_set_dir_name_len(leaf, dir_item, name_len); | ||
| 193 | btrfs_set_dir_transid(leaf, dir_item, trans->transid); | ||
| 194 | name_ptr = (unsigned long)(dir_item + 1); | ||
| 195 | write_extent_buffer(leaf, name, name_ptr, name_len); | ||
| 196 | btrfs_mark_buffer_dirty(leaf); | ||
| 197 | out: | ||
| 198 | btrfs_free_path(path); | ||
| 199 | if (ret) | ||
| 200 | return ret; | ||
| 201 | if (ret2) | ||
| 202 | return ret2; | ||
| 203 | return 0; | ||
| 204 | } | ||
| 205 | |||
| 206 | /* | ||
| 207 | * lookup a directory item based on name. 'dir' is the objectid | ||
| 208 | * we're searching in, and 'mod' tells us if you plan on deleting the | ||
| 209 | * item (use mod < 0) or changing the options (use mod > 0) | ||
| 210 | */ | ||
| 211 | struct btrfs_dir_item *btrfs_lookup_dir_item(struct btrfs_trans_handle *trans, | ||
| 212 | struct btrfs_root *root, | ||
| 213 | struct btrfs_path *path, u64 dir, | ||
| 214 | const char *name, int name_len, | ||
| 215 | int mod) | ||
| 216 | { | ||
| 217 | int ret; | ||
| 218 | struct btrfs_key key; | ||
| 219 | int ins_len = mod < 0 ? -1 : 0; | ||
| 220 | int cow = mod != 0; | ||
| 221 | struct btrfs_key found_key; | ||
| 222 | struct extent_buffer *leaf; | ||
| 223 | |||
| 224 | key.objectid = dir; | ||
| 225 | btrfs_set_key_type(&key, BTRFS_DIR_ITEM_KEY); | ||
| 226 | |||
| 227 | key.offset = btrfs_name_hash(name, name_len); | ||
| 228 | |||
| 229 | ret = btrfs_search_slot(trans, root, &key, path, ins_len, cow); | ||
| 230 | if (ret < 0) | ||
| 231 | return ERR_PTR(ret); | ||
| 232 | if (ret > 0) { | ||
| 233 | if (path->slots[0] == 0) | ||
| 234 | return NULL; | ||
| 235 | path->slots[0]--; | ||
| 236 | } | ||
| 237 | |||
| 238 | leaf = path->nodes[0]; | ||
| 239 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 240 | |||
| 241 | if (found_key.objectid != dir || | ||
| 242 | btrfs_key_type(&found_key) != BTRFS_DIR_ITEM_KEY || | ||
| 243 | found_key.offset != key.offset) | ||
| 244 | return NULL; | ||
| 245 | |||
| 246 | return btrfs_match_dir_item_name(root, path, name, name_len); | ||
| 247 | } | ||
| 248 | |||
| 249 | /* | ||
| 250 | * lookup a directory item based on index. 'dir' is the objectid | ||
| 251 | * we're searching in, and 'mod' tells us if you plan on deleting the | ||
| 252 | * item (use mod < 0) or changing the options (use mod > 0) | ||
| 253 | * | ||
| 254 | * The name is used to make sure the index really points to the name you were | ||
| 255 | * looking for. | ||
| 256 | */ | ||
| 257 | struct btrfs_dir_item * | ||
| 258 | btrfs_lookup_dir_index_item(struct btrfs_trans_handle *trans, | ||
| 259 | struct btrfs_root *root, | ||
| 260 | struct btrfs_path *path, u64 dir, | ||
| 261 | u64 objectid, const char *name, int name_len, | ||
| 262 | int mod) | ||
| 263 | { | ||
| 264 | int ret; | ||
| 265 | struct btrfs_key key; | ||
| 266 | int ins_len = mod < 0 ? -1 : 0; | ||
| 267 | int cow = mod != 0; | ||
| 268 | |||
| 269 | key.objectid = dir; | ||
| 270 | btrfs_set_key_type(&key, BTRFS_DIR_INDEX_KEY); | ||
| 271 | key.offset = objectid; | ||
| 272 | |||
| 273 | ret = btrfs_search_slot(trans, root, &key, path, ins_len, cow); | ||
| 274 | if (ret < 0) | ||
| 275 | return ERR_PTR(ret); | ||
| 276 | if (ret > 0) | ||
| 277 | return ERR_PTR(-ENOENT); | ||
| 278 | return btrfs_match_dir_item_name(root, path, name, name_len); | ||
| 279 | } | ||
| 280 | |||
| 281 | struct btrfs_dir_item *btrfs_lookup_xattr(struct btrfs_trans_handle *trans, | ||
| 282 | struct btrfs_root *root, | ||
| 283 | struct btrfs_path *path, u64 dir, | ||
| 284 | const char *name, u16 name_len, | ||
| 285 | int mod) | ||
| 286 | { | ||
| 287 | int ret; | ||
| 288 | struct btrfs_key key; | ||
| 289 | int ins_len = mod < 0 ? -1 : 0; | ||
| 290 | int cow = mod != 0; | ||
| 291 | struct btrfs_key found_key; | ||
| 292 | struct extent_buffer *leaf; | ||
| 293 | |||
| 294 | key.objectid = dir; | ||
| 295 | btrfs_set_key_type(&key, BTRFS_XATTR_ITEM_KEY); | ||
| 296 | key.offset = btrfs_name_hash(name, name_len); | ||
| 297 | ret = btrfs_search_slot(trans, root, &key, path, ins_len, cow); | ||
| 298 | if (ret < 0) | ||
| 299 | return ERR_PTR(ret); | ||
| 300 | if (ret > 0) { | ||
| 301 | if (path->slots[0] == 0) | ||
| 302 | return NULL; | ||
| 303 | path->slots[0]--; | ||
| 304 | } | ||
| 305 | |||
| 306 | leaf = path->nodes[0]; | ||
| 307 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 308 | |||
| 309 | if (found_key.objectid != dir || | ||
| 310 | btrfs_key_type(&found_key) != BTRFS_XATTR_ITEM_KEY || | ||
| 311 | found_key.offset != key.offset) | ||
| 312 | return NULL; | ||
| 313 | |||
| 314 | return btrfs_match_dir_item_name(root, path, name, name_len); | ||
| 315 | } | ||
| 316 | |||
| 317 | /* | ||
| 318 | * helper function to look at the directory item pointed to by 'path' | ||
| 319 | * this walks through all the entries in a dir item and finds one | ||
| 320 | * for a specific name. | ||
| 321 | */ | ||
| 322 | struct btrfs_dir_item *btrfs_match_dir_item_name(struct btrfs_root *root, | ||
| 323 | struct btrfs_path *path, | ||
| 324 | const char *name, int name_len) | ||
| 325 | { | ||
| 326 | struct btrfs_dir_item *dir_item; | ||
| 327 | unsigned long name_ptr; | ||
| 328 | u32 total_len; | ||
| 329 | u32 cur = 0; | ||
| 330 | u32 this_len; | ||
| 331 | struct extent_buffer *leaf; | ||
| 332 | |||
| 333 | leaf = path->nodes[0]; | ||
| 334 | dir_item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_dir_item); | ||
| 335 | total_len = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 336 | while (cur < total_len) { | ||
| 337 | this_len = sizeof(*dir_item) + | ||
| 338 | btrfs_dir_name_len(leaf, dir_item) + | ||
| 339 | btrfs_dir_data_len(leaf, dir_item); | ||
| 340 | name_ptr = (unsigned long)(dir_item + 1); | ||
| 341 | |||
| 342 | if (btrfs_dir_name_len(leaf, dir_item) == name_len && | ||
| 343 | memcmp_extent_buffer(leaf, name, name_ptr, name_len) == 0) | ||
| 344 | return dir_item; | ||
| 345 | |||
| 346 | cur += this_len; | ||
| 347 | dir_item = (struct btrfs_dir_item *)((char *)dir_item + | ||
| 348 | this_len); | ||
| 349 | } | ||
| 350 | return NULL; | ||
| 351 | } | ||
| 352 | |||
| 353 | /* | ||
| 354 | * given a pointer into a directory item, delete it. This | ||
| 355 | * handles items that have more than one entry in them. | ||
| 356 | */ | ||
| 357 | int btrfs_delete_one_dir_name(struct btrfs_trans_handle *trans, | ||
| 358 | struct btrfs_root *root, | ||
| 359 | struct btrfs_path *path, | ||
| 360 | struct btrfs_dir_item *di) | ||
| 361 | { | ||
| 362 | |||
| 363 | struct extent_buffer *leaf; | ||
| 364 | u32 sub_item_len; | ||
| 365 | u32 item_len; | ||
| 366 | int ret = 0; | ||
| 367 | |||
| 368 | leaf = path->nodes[0]; | ||
| 369 | sub_item_len = sizeof(*di) + btrfs_dir_name_len(leaf, di) + | ||
| 370 | btrfs_dir_data_len(leaf, di); | ||
| 371 | item_len = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 372 | if (sub_item_len == item_len) { | ||
| 373 | ret = btrfs_del_item(trans, root, path); | ||
| 374 | } else { | ||
| 375 | /* MARKER */ | ||
| 376 | unsigned long ptr = (unsigned long)di; | ||
| 377 | unsigned long start; | ||
| 378 | |||
| 379 | start = btrfs_item_ptr_offset(leaf, path->slots[0]); | ||
| 380 | memmove_extent_buffer(leaf, ptr, ptr + sub_item_len, | ||
| 381 | item_len - (ptr + sub_item_len - start)); | ||
| 382 | ret = btrfs_truncate_item(trans, root, path, | ||
| 383 | item_len - sub_item_len, 1); | ||
| 384 | } | ||
| 385 | return 0; | ||
| 386 | } | ||
diff --git a/fs/btrfs/disk-io.c b/fs/btrfs/disk-io.c new file mode 100644 index 000000000000..81a313874ae5 --- /dev/null +++ b/fs/btrfs/disk-io.c | |||
| @@ -0,0 +1,2343 @@ | |||
| 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 | |||
| 19 | #include <linux/version.h> | ||
| 20 | #include <linux/fs.h> | ||
| 21 | #include <linux/blkdev.h> | ||
| 22 | #include <linux/scatterlist.h> | ||
| 23 | #include <linux/swap.h> | ||
| 24 | #include <linux/radix-tree.h> | ||
| 25 | #include <linux/writeback.h> | ||
| 26 | #include <linux/buffer_head.h> | ||
| 27 | #include <linux/workqueue.h> | ||
| 28 | #include <linux/kthread.h> | ||
| 29 | #include <linux/freezer.h> | ||
| 30 | #include "compat.h" | ||
| 31 | #include "crc32c.h" | ||
| 32 | #include "ctree.h" | ||
| 33 | #include "disk-io.h" | ||
| 34 | #include "transaction.h" | ||
| 35 | #include "btrfs_inode.h" | ||
| 36 | #include "volumes.h" | ||
| 37 | #include "print-tree.h" | ||
| 38 | #include "async-thread.h" | ||
| 39 | #include "locking.h" | ||
| 40 | #include "ref-cache.h" | ||
| 41 | #include "tree-log.h" | ||
| 42 | |||
| 43 | static struct extent_io_ops btree_extent_io_ops; | ||
| 44 | static void end_workqueue_fn(struct btrfs_work *work); | ||
| 45 | |||
| 46 | /* | ||
| 47 | * end_io_wq structs are used to do processing in task context when an IO is | ||
| 48 | * complete. This is used during reads to verify checksums, and it is used | ||
| 49 | * by writes to insert metadata for new file extents after IO is complete. | ||
| 50 | */ | ||
| 51 | struct end_io_wq { | ||
| 52 | struct bio *bio; | ||
| 53 | bio_end_io_t *end_io; | ||
| 54 | void *private; | ||
| 55 | struct btrfs_fs_info *info; | ||
| 56 | int error; | ||
| 57 | int metadata; | ||
| 58 | struct list_head list; | ||
| 59 | struct btrfs_work work; | ||
| 60 | }; | ||
| 61 | |||
| 62 | /* | ||
| 63 | * async submit bios are used to offload expensive checksumming | ||
| 64 | * onto the worker threads. They checksum file and metadata bios | ||
| 65 | * just before they are sent down the IO stack. | ||
| 66 | */ | ||
| 67 | struct async_submit_bio { | ||
| 68 | struct inode *inode; | ||
| 69 | struct bio *bio; | ||
| 70 | struct list_head list; | ||
| 71 | extent_submit_bio_hook_t *submit_bio_start; | ||
| 72 | extent_submit_bio_hook_t *submit_bio_done; | ||
| 73 | int rw; | ||
| 74 | int mirror_num; | ||
| 75 | unsigned long bio_flags; | ||
| 76 | struct btrfs_work work; | ||
| 77 | }; | ||
| 78 | |||
| 79 | /* | ||
| 80 | * extents on the btree inode are pretty simple, there's one extent | ||
| 81 | * that covers the entire device | ||
| 82 | */ | ||
| 83 | static struct extent_map *btree_get_extent(struct inode *inode, | ||
| 84 | struct page *page, size_t page_offset, u64 start, u64 len, | ||
| 85 | int create) | ||
| 86 | { | ||
| 87 | struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 88 | struct extent_map *em; | ||
| 89 | int ret; | ||
| 90 | |||
| 91 | spin_lock(&em_tree->lock); | ||
| 92 | em = lookup_extent_mapping(em_tree, start, len); | ||
| 93 | if (em) { | ||
| 94 | em->bdev = | ||
| 95 | BTRFS_I(inode)->root->fs_info->fs_devices->latest_bdev; | ||
| 96 | spin_unlock(&em_tree->lock); | ||
| 97 | goto out; | ||
| 98 | } | ||
| 99 | spin_unlock(&em_tree->lock); | ||
| 100 | |||
| 101 | em = alloc_extent_map(GFP_NOFS); | ||
| 102 | if (!em) { | ||
| 103 | em = ERR_PTR(-ENOMEM); | ||
| 104 | goto out; | ||
| 105 | } | ||
| 106 | em->start = 0; | ||
| 107 | em->len = (u64)-1; | ||
| 108 | em->block_len = (u64)-1; | ||
| 109 | em->block_start = 0; | ||
| 110 | em->bdev = BTRFS_I(inode)->root->fs_info->fs_devices->latest_bdev; | ||
| 111 | |||
| 112 | spin_lock(&em_tree->lock); | ||
| 113 | ret = add_extent_mapping(em_tree, em); | ||
| 114 | if (ret == -EEXIST) { | ||
| 115 | u64 failed_start = em->start; | ||
| 116 | u64 failed_len = em->len; | ||
| 117 | |||
| 118 | free_extent_map(em); | ||
| 119 | em = lookup_extent_mapping(em_tree, start, len); | ||
| 120 | if (em) { | ||
| 121 | ret = 0; | ||
| 122 | } else { | ||
| 123 | em = lookup_extent_mapping(em_tree, failed_start, | ||
| 124 | failed_len); | ||
| 125 | ret = -EIO; | ||
| 126 | } | ||
| 127 | } else if (ret) { | ||
| 128 | free_extent_map(em); | ||
| 129 | em = NULL; | ||
| 130 | } | ||
| 131 | spin_unlock(&em_tree->lock); | ||
| 132 | |||
| 133 | if (ret) | ||
| 134 | em = ERR_PTR(ret); | ||
| 135 | out: | ||
| 136 | return em; | ||
| 137 | } | ||
| 138 | |||
| 139 | u32 btrfs_csum_data(struct btrfs_root *root, char *data, u32 seed, size_t len) | ||
| 140 | { | ||
| 141 | return btrfs_crc32c(seed, data, len); | ||
| 142 | } | ||
| 143 | |||
| 144 | void btrfs_csum_final(u32 crc, char *result) | ||
| 145 | { | ||
| 146 | *(__le32 *)result = ~cpu_to_le32(crc); | ||
| 147 | } | ||
| 148 | |||
| 149 | /* | ||
| 150 | * compute the csum for a btree block, and either verify it or write it | ||
| 151 | * into the csum field of the block. | ||
| 152 | */ | ||
| 153 | static int csum_tree_block(struct btrfs_root *root, struct extent_buffer *buf, | ||
| 154 | int verify) | ||
| 155 | { | ||
| 156 | u16 csum_size = | ||
| 157 | btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 158 | char *result = NULL; | ||
| 159 | unsigned long len; | ||
| 160 | unsigned long cur_len; | ||
| 161 | unsigned long offset = BTRFS_CSUM_SIZE; | ||
| 162 | char *map_token = NULL; | ||
| 163 | char *kaddr; | ||
| 164 | unsigned long map_start; | ||
| 165 | unsigned long map_len; | ||
| 166 | int err; | ||
| 167 | u32 crc = ~(u32)0; | ||
| 168 | unsigned long inline_result; | ||
| 169 | |||
| 170 | len = buf->len - offset; | ||
| 171 | while (len > 0) { | ||
| 172 | err = map_private_extent_buffer(buf, offset, 32, | ||
| 173 | &map_token, &kaddr, | ||
| 174 | &map_start, &map_len, KM_USER0); | ||
| 175 | if (err) | ||
| 176 | return 1; | ||
| 177 | cur_len = min(len, map_len - (offset - map_start)); | ||
| 178 | crc = btrfs_csum_data(root, kaddr + offset - map_start, | ||
| 179 | crc, cur_len); | ||
| 180 | len -= cur_len; | ||
| 181 | offset += cur_len; | ||
| 182 | unmap_extent_buffer(buf, map_token, KM_USER0); | ||
| 183 | } | ||
| 184 | if (csum_size > sizeof(inline_result)) { | ||
| 185 | result = kzalloc(csum_size * sizeof(char), GFP_NOFS); | ||
| 186 | if (!result) | ||
| 187 | return 1; | ||
| 188 | } else { | ||
| 189 | result = (char *)&inline_result; | ||
| 190 | } | ||
| 191 | |||
| 192 | btrfs_csum_final(crc, result); | ||
| 193 | |||
| 194 | if (verify) { | ||
| 195 | if (memcmp_extent_buffer(buf, result, 0, csum_size)) { | ||
| 196 | u32 val; | ||
| 197 | u32 found = 0; | ||
| 198 | memcpy(&found, result, csum_size); | ||
| 199 | |||
| 200 | read_extent_buffer(buf, &val, 0, csum_size); | ||
| 201 | printk(KERN_INFO "btrfs: %s checksum verify failed " | ||
| 202 | "on %llu wanted %X found %X level %d\n", | ||
| 203 | root->fs_info->sb->s_id, | ||
| 204 | buf->start, val, found, btrfs_header_level(buf)); | ||
| 205 | if (result != (char *)&inline_result) | ||
| 206 | kfree(result); | ||
| 207 | return 1; | ||
| 208 | } | ||
| 209 | } else { | ||
| 210 | write_extent_buffer(buf, result, 0, csum_size); | ||
| 211 | } | ||
| 212 | if (result != (char *)&inline_result) | ||
| 213 | kfree(result); | ||
| 214 | return 0; | ||
| 215 | } | ||
| 216 | |||
| 217 | /* | ||
| 218 | * we can't consider a given block up to date unless the transid of the | ||
| 219 | * block matches the transid in the parent node's pointer. This is how we | ||
| 220 | * detect blocks that either didn't get written at all or got written | ||
| 221 | * in the wrong place. | ||
| 222 | */ | ||
| 223 | static int verify_parent_transid(struct extent_io_tree *io_tree, | ||
| 224 | struct extent_buffer *eb, u64 parent_transid) | ||
| 225 | { | ||
| 226 | int ret; | ||
| 227 | |||
| 228 | if (!parent_transid || btrfs_header_generation(eb) == parent_transid) | ||
| 229 | return 0; | ||
| 230 | |||
| 231 | lock_extent(io_tree, eb->start, eb->start + eb->len - 1, GFP_NOFS); | ||
| 232 | if (extent_buffer_uptodate(io_tree, eb) && | ||
| 233 | btrfs_header_generation(eb) == parent_transid) { | ||
| 234 | ret = 0; | ||
| 235 | goto out; | ||
| 236 | } | ||
| 237 | printk("parent transid verify failed on %llu wanted %llu found %llu\n", | ||
| 238 | (unsigned long long)eb->start, | ||
| 239 | (unsigned long long)parent_transid, | ||
| 240 | (unsigned long long)btrfs_header_generation(eb)); | ||
| 241 | ret = 1; | ||
| 242 | clear_extent_buffer_uptodate(io_tree, eb); | ||
| 243 | out: | ||
| 244 | unlock_extent(io_tree, eb->start, eb->start + eb->len - 1, | ||
| 245 | GFP_NOFS); | ||
| 246 | return ret; | ||
| 247 | } | ||
| 248 | |||
| 249 | /* | ||
| 250 | * helper to read a given tree block, doing retries as required when | ||
| 251 | * the checksums don't match and we have alternate mirrors to try. | ||
| 252 | */ | ||
| 253 | static int btree_read_extent_buffer_pages(struct btrfs_root *root, | ||
| 254 | struct extent_buffer *eb, | ||
| 255 | u64 start, u64 parent_transid) | ||
| 256 | { | ||
| 257 | struct extent_io_tree *io_tree; | ||
| 258 | int ret; | ||
| 259 | int num_copies = 0; | ||
| 260 | int mirror_num = 0; | ||
| 261 | |||
| 262 | io_tree = &BTRFS_I(root->fs_info->btree_inode)->io_tree; | ||
| 263 | while (1) { | ||
| 264 | ret = read_extent_buffer_pages(io_tree, eb, start, 1, | ||
| 265 | btree_get_extent, mirror_num); | ||
| 266 | if (!ret && | ||
| 267 | !verify_parent_transid(io_tree, eb, parent_transid)) | ||
| 268 | return ret; | ||
| 269 | |||
| 270 | num_copies = btrfs_num_copies(&root->fs_info->mapping_tree, | ||
| 271 | eb->start, eb->len); | ||
| 272 | if (num_copies == 1) | ||
| 273 | return ret; | ||
| 274 | |||
| 275 | mirror_num++; | ||
| 276 | if (mirror_num > num_copies) | ||
| 277 | return ret; | ||
| 278 | } | ||
| 279 | return -EIO; | ||
| 280 | } | ||
| 281 | |||
| 282 | /* | ||
| 283 | * checksum a dirty tree block before IO. This has extra checks to make sure | ||
| 284 | * we only fill in the checksum field in the first page of a multi-page block | ||
| 285 | */ | ||
| 286 | |||
| 287 | static int csum_dirty_buffer(struct btrfs_root *root, struct page *page) | ||
| 288 | { | ||
| 289 | struct extent_io_tree *tree; | ||
| 290 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 291 | u64 found_start; | ||
| 292 | int found_level; | ||
| 293 | unsigned long len; | ||
| 294 | struct extent_buffer *eb; | ||
| 295 | int ret; | ||
| 296 | |||
| 297 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 298 | |||
| 299 | if (page->private == EXTENT_PAGE_PRIVATE) | ||
| 300 | goto out; | ||
| 301 | if (!page->private) | ||
| 302 | goto out; | ||
| 303 | len = page->private >> 2; | ||
| 304 | WARN_ON(len == 0); | ||
| 305 | |||
| 306 | eb = alloc_extent_buffer(tree, start, len, page, GFP_NOFS); | ||
| 307 | ret = btree_read_extent_buffer_pages(root, eb, start + PAGE_CACHE_SIZE, | ||
| 308 | btrfs_header_generation(eb)); | ||
| 309 | BUG_ON(ret); | ||
| 310 | found_start = btrfs_header_bytenr(eb); | ||
| 311 | if (found_start != start) { | ||
| 312 | WARN_ON(1); | ||
| 313 | goto err; | ||
| 314 | } | ||
| 315 | if (eb->first_page != page) { | ||
| 316 | WARN_ON(1); | ||
| 317 | goto err; | ||
| 318 | } | ||
| 319 | if (!PageUptodate(page)) { | ||
| 320 | WARN_ON(1); | ||
| 321 | goto err; | ||
| 322 | } | ||
| 323 | found_level = btrfs_header_level(eb); | ||
| 324 | |||
| 325 | csum_tree_block(root, eb, 0); | ||
| 326 | err: | ||
| 327 | free_extent_buffer(eb); | ||
| 328 | out: | ||
| 329 | return 0; | ||
| 330 | } | ||
| 331 | |||
| 332 | static int check_tree_block_fsid(struct btrfs_root *root, | ||
| 333 | struct extent_buffer *eb) | ||
| 334 | { | ||
| 335 | struct btrfs_fs_devices *fs_devices = root->fs_info->fs_devices; | ||
| 336 | u8 fsid[BTRFS_UUID_SIZE]; | ||
| 337 | int ret = 1; | ||
| 338 | |||
| 339 | read_extent_buffer(eb, fsid, (unsigned long)btrfs_header_fsid(eb), | ||
| 340 | BTRFS_FSID_SIZE); | ||
| 341 | while (fs_devices) { | ||
| 342 | if (!memcmp(fsid, fs_devices->fsid, BTRFS_FSID_SIZE)) { | ||
| 343 | ret = 0; | ||
| 344 | break; | ||
| 345 | } | ||
| 346 | fs_devices = fs_devices->seed; | ||
| 347 | } | ||
| 348 | return ret; | ||
| 349 | } | ||
| 350 | |||
| 351 | static int btree_readpage_end_io_hook(struct page *page, u64 start, u64 end, | ||
| 352 | struct extent_state *state) | ||
| 353 | { | ||
| 354 | struct extent_io_tree *tree; | ||
| 355 | u64 found_start; | ||
| 356 | int found_level; | ||
| 357 | unsigned long len; | ||
| 358 | struct extent_buffer *eb; | ||
| 359 | struct btrfs_root *root = BTRFS_I(page->mapping->host)->root; | ||
| 360 | int ret = 0; | ||
| 361 | |||
| 362 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 363 | if (page->private == EXTENT_PAGE_PRIVATE) | ||
| 364 | goto out; | ||
| 365 | if (!page->private) | ||
| 366 | goto out; | ||
| 367 | |||
| 368 | len = page->private >> 2; | ||
| 369 | WARN_ON(len == 0); | ||
| 370 | |||
| 371 | eb = alloc_extent_buffer(tree, start, len, page, GFP_NOFS); | ||
| 372 | |||
| 373 | found_start = btrfs_header_bytenr(eb); | ||
| 374 | if (found_start != start) { | ||
| 375 | printk(KERN_INFO "btrfs bad tree block start %llu %llu\n", | ||
| 376 | (unsigned long long)found_start, | ||
| 377 | (unsigned long long)eb->start); | ||
| 378 | ret = -EIO; | ||
| 379 | goto err; | ||
| 380 | } | ||
| 381 | if (eb->first_page != page) { | ||
| 382 | printk(KERN_INFO "btrfs bad first page %lu %lu\n", | ||
| 383 | eb->first_page->index, page->index); | ||
| 384 | WARN_ON(1); | ||
| 385 | ret = -EIO; | ||
| 386 | goto err; | ||
| 387 | } | ||
| 388 | if (check_tree_block_fsid(root, eb)) { | ||
| 389 | printk(KERN_INFO "btrfs bad fsid on block %llu\n", | ||
| 390 | (unsigned long long)eb->start); | ||
| 391 | ret = -EIO; | ||
| 392 | goto err; | ||
| 393 | } | ||
| 394 | found_level = btrfs_header_level(eb); | ||
| 395 | |||
| 396 | ret = csum_tree_block(root, eb, 1); | ||
| 397 | if (ret) | ||
| 398 | ret = -EIO; | ||
| 399 | |||
| 400 | end = min_t(u64, eb->len, PAGE_CACHE_SIZE); | ||
| 401 | end = eb->start + end - 1; | ||
| 402 | err: | ||
| 403 | free_extent_buffer(eb); | ||
| 404 | out: | ||
| 405 | return ret; | ||
| 406 | } | ||
| 407 | |||
| 408 | static void end_workqueue_bio(struct bio *bio, int err) | ||
| 409 | { | ||
| 410 | struct end_io_wq *end_io_wq = bio->bi_private; | ||
| 411 | struct btrfs_fs_info *fs_info; | ||
| 412 | |||
| 413 | fs_info = end_io_wq->info; | ||
| 414 | end_io_wq->error = err; | ||
| 415 | end_io_wq->work.func = end_workqueue_fn; | ||
| 416 | end_io_wq->work.flags = 0; | ||
| 417 | |||
| 418 | if (bio->bi_rw & (1 << BIO_RW)) { | ||
| 419 | if (end_io_wq->metadata) | ||
| 420 | btrfs_queue_worker(&fs_info->endio_meta_write_workers, | ||
| 421 | &end_io_wq->work); | ||
| 422 | else | ||
| 423 | btrfs_queue_worker(&fs_info->endio_write_workers, | ||
| 424 | &end_io_wq->work); | ||
| 425 | } else { | ||
| 426 | if (end_io_wq->metadata) | ||
| 427 | btrfs_queue_worker(&fs_info->endio_meta_workers, | ||
| 428 | &end_io_wq->work); | ||
| 429 | else | ||
| 430 | btrfs_queue_worker(&fs_info->endio_workers, | ||
| 431 | &end_io_wq->work); | ||
| 432 | } | ||
| 433 | } | ||
| 434 | |||
| 435 | int btrfs_bio_wq_end_io(struct btrfs_fs_info *info, struct bio *bio, | ||
| 436 | int metadata) | ||
| 437 | { | ||
| 438 | struct end_io_wq *end_io_wq; | ||
| 439 | end_io_wq = kmalloc(sizeof(*end_io_wq), GFP_NOFS); | ||
| 440 | if (!end_io_wq) | ||
| 441 | return -ENOMEM; | ||
| 442 | |||
| 443 | end_io_wq->private = bio->bi_private; | ||
| 444 | end_io_wq->end_io = bio->bi_end_io; | ||
| 445 | end_io_wq->info = info; | ||
| 446 | end_io_wq->error = 0; | ||
| 447 | end_io_wq->bio = bio; | ||
| 448 | end_io_wq->metadata = metadata; | ||
| 449 | |||
| 450 | bio->bi_private = end_io_wq; | ||
| 451 | bio->bi_end_io = end_workqueue_bio; | ||
| 452 | return 0; | ||
| 453 | } | ||
| 454 | |||
| 455 | unsigned long btrfs_async_submit_limit(struct btrfs_fs_info *info) | ||
| 456 | { | ||
| 457 | unsigned long limit = min_t(unsigned long, | ||
| 458 | info->workers.max_workers, | ||
| 459 | info->fs_devices->open_devices); | ||
| 460 | return 256 * limit; | ||
| 461 | } | ||
| 462 | |||
| 463 | int btrfs_congested_async(struct btrfs_fs_info *info, int iodone) | ||
| 464 | { | ||
| 465 | return atomic_read(&info->nr_async_bios) > | ||
| 466 | btrfs_async_submit_limit(info); | ||
| 467 | } | ||
| 468 | |||
| 469 | static void run_one_async_start(struct btrfs_work *work) | ||
| 470 | { | ||
| 471 | struct btrfs_fs_info *fs_info; | ||
| 472 | struct async_submit_bio *async; | ||
| 473 | |||
| 474 | async = container_of(work, struct async_submit_bio, work); | ||
| 475 | fs_info = BTRFS_I(async->inode)->root->fs_info; | ||
| 476 | async->submit_bio_start(async->inode, async->rw, async->bio, | ||
| 477 | async->mirror_num, async->bio_flags); | ||
| 478 | } | ||
| 479 | |||
| 480 | static void run_one_async_done(struct btrfs_work *work) | ||
| 481 | { | ||
| 482 | struct btrfs_fs_info *fs_info; | ||
| 483 | struct async_submit_bio *async; | ||
| 484 | int limit; | ||
| 485 | |||
| 486 | async = container_of(work, struct async_submit_bio, work); | ||
| 487 | fs_info = BTRFS_I(async->inode)->root->fs_info; | ||
| 488 | |||
| 489 | limit = btrfs_async_submit_limit(fs_info); | ||
| 490 | limit = limit * 2 / 3; | ||
| 491 | |||
| 492 | atomic_dec(&fs_info->nr_async_submits); | ||
| 493 | |||
| 494 | if (atomic_read(&fs_info->nr_async_submits) < limit && | ||
| 495 | waitqueue_active(&fs_info->async_submit_wait)) | ||
| 496 | wake_up(&fs_info->async_submit_wait); | ||
| 497 | |||
| 498 | async->submit_bio_done(async->inode, async->rw, async->bio, | ||
| 499 | async->mirror_num, async->bio_flags); | ||
| 500 | } | ||
| 501 | |||
| 502 | static void run_one_async_free(struct btrfs_work *work) | ||
| 503 | { | ||
| 504 | struct async_submit_bio *async; | ||
| 505 | |||
| 506 | async = container_of(work, struct async_submit_bio, work); | ||
| 507 | kfree(async); | ||
| 508 | } | ||
| 509 | |||
| 510 | int btrfs_wq_submit_bio(struct btrfs_fs_info *fs_info, struct inode *inode, | ||
| 511 | int rw, struct bio *bio, int mirror_num, | ||
| 512 | unsigned long bio_flags, | ||
| 513 | extent_submit_bio_hook_t *submit_bio_start, | ||
| 514 | extent_submit_bio_hook_t *submit_bio_done) | ||
| 515 | { | ||
| 516 | struct async_submit_bio *async; | ||
| 517 | |||
| 518 | async = kmalloc(sizeof(*async), GFP_NOFS); | ||
| 519 | if (!async) | ||
| 520 | return -ENOMEM; | ||
| 521 | |||
| 522 | async->inode = inode; | ||
| 523 | async->rw = rw; | ||
| 524 | async->bio = bio; | ||
| 525 | async->mirror_num = mirror_num; | ||
| 526 | async->submit_bio_start = submit_bio_start; | ||
| 527 | async->submit_bio_done = submit_bio_done; | ||
| 528 | |||
| 529 | async->work.func = run_one_async_start; | ||
| 530 | async->work.ordered_func = run_one_async_done; | ||
| 531 | async->work.ordered_free = run_one_async_free; | ||
| 532 | |||
| 533 | async->work.flags = 0; | ||
| 534 | async->bio_flags = bio_flags; | ||
| 535 | |||
| 536 | atomic_inc(&fs_info->nr_async_submits); | ||
| 537 | btrfs_queue_worker(&fs_info->workers, &async->work); | ||
| 538 | #if 0 | ||
| 539 | int limit = btrfs_async_submit_limit(fs_info); | ||
| 540 | if (atomic_read(&fs_info->nr_async_submits) > limit) { | ||
| 541 | wait_event_timeout(fs_info->async_submit_wait, | ||
| 542 | (atomic_read(&fs_info->nr_async_submits) < limit), | ||
| 543 | HZ/10); | ||
| 544 | |||
| 545 | wait_event_timeout(fs_info->async_submit_wait, | ||
| 546 | (atomic_read(&fs_info->nr_async_bios) < limit), | ||
| 547 | HZ/10); | ||
| 548 | } | ||
| 549 | #endif | ||
| 550 | while (atomic_read(&fs_info->async_submit_draining) && | ||
| 551 | atomic_read(&fs_info->nr_async_submits)) { | ||
| 552 | wait_event(fs_info->async_submit_wait, | ||
| 553 | (atomic_read(&fs_info->nr_async_submits) == 0)); | ||
| 554 | } | ||
| 555 | |||
| 556 | return 0; | ||
| 557 | } | ||
| 558 | |||
| 559 | static int btree_csum_one_bio(struct bio *bio) | ||
| 560 | { | ||
| 561 | struct bio_vec *bvec = bio->bi_io_vec; | ||
| 562 | int bio_index = 0; | ||
| 563 | struct btrfs_root *root; | ||
| 564 | |||
| 565 | WARN_ON(bio->bi_vcnt <= 0); | ||
| 566 | while (bio_index < bio->bi_vcnt) { | ||
| 567 | root = BTRFS_I(bvec->bv_page->mapping->host)->root; | ||
| 568 | csum_dirty_buffer(root, bvec->bv_page); | ||
| 569 | bio_index++; | ||
| 570 | bvec++; | ||
| 571 | } | ||
| 572 | return 0; | ||
| 573 | } | ||
| 574 | |||
| 575 | static int __btree_submit_bio_start(struct inode *inode, int rw, | ||
| 576 | struct bio *bio, int mirror_num, | ||
| 577 | unsigned long bio_flags) | ||
| 578 | { | ||
| 579 | /* | ||
| 580 | * when we're called for a write, we're already in the async | ||
| 581 | * submission context. Just jump into btrfs_map_bio | ||
| 582 | */ | ||
| 583 | btree_csum_one_bio(bio); | ||
| 584 | return 0; | ||
| 585 | } | ||
| 586 | |||
| 587 | static int __btree_submit_bio_done(struct inode *inode, int rw, struct bio *bio, | ||
| 588 | int mirror_num, unsigned long bio_flags) | ||
| 589 | { | ||
| 590 | /* | ||
| 591 | * when we're called for a write, we're already in the async | ||
| 592 | * submission context. Just jump into btrfs_map_bio | ||
| 593 | */ | ||
| 594 | return btrfs_map_bio(BTRFS_I(inode)->root, rw, bio, mirror_num, 1); | ||
| 595 | } | ||
| 596 | |||
| 597 | static int btree_submit_bio_hook(struct inode *inode, int rw, struct bio *bio, | ||
| 598 | int mirror_num, unsigned long bio_flags) | ||
| 599 | { | ||
| 600 | int ret; | ||
| 601 | |||
| 602 | ret = btrfs_bio_wq_end_io(BTRFS_I(inode)->root->fs_info, | ||
| 603 | bio, 1); | ||
| 604 | BUG_ON(ret); | ||
| 605 | |||
| 606 | if (!(rw & (1 << BIO_RW))) { | ||
| 607 | /* | ||
| 608 | * called for a read, do the setup so that checksum validation | ||
| 609 | * can happen in the async kernel threads | ||
| 610 | */ | ||
| 611 | return btrfs_map_bio(BTRFS_I(inode)->root, rw, bio, | ||
| 612 | mirror_num, 0); | ||
| 613 | } | ||
| 614 | /* | ||
| 615 | * kthread helpers are used to submit writes so that checksumming | ||
| 616 | * can happen in parallel across all CPUs | ||
| 617 | */ | ||
| 618 | return btrfs_wq_submit_bio(BTRFS_I(inode)->root->fs_info, | ||
| 619 | inode, rw, bio, mirror_num, 0, | ||
| 620 | __btree_submit_bio_start, | ||
| 621 | __btree_submit_bio_done); | ||
| 622 | } | ||
| 623 | |||
| 624 | static int btree_writepage(struct page *page, struct writeback_control *wbc) | ||
| 625 | { | ||
| 626 | struct extent_io_tree *tree; | ||
| 627 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 628 | |||
| 629 | if (current->flags & PF_MEMALLOC) { | ||
| 630 | redirty_page_for_writepage(wbc, page); | ||
| 631 | unlock_page(page); | ||
| 632 | return 0; | ||
| 633 | } | ||
| 634 | return extent_write_full_page(tree, page, btree_get_extent, wbc); | ||
| 635 | } | ||
| 636 | |||
| 637 | static int btree_writepages(struct address_space *mapping, | ||
| 638 | struct writeback_control *wbc) | ||
| 639 | { | ||
| 640 | struct extent_io_tree *tree; | ||
| 641 | tree = &BTRFS_I(mapping->host)->io_tree; | ||
| 642 | if (wbc->sync_mode == WB_SYNC_NONE) { | ||
| 643 | u64 num_dirty; | ||
| 644 | u64 start = 0; | ||
| 645 | unsigned long thresh = 32 * 1024 * 1024; | ||
| 646 | |||
| 647 | if (wbc->for_kupdate) | ||
| 648 | return 0; | ||
| 649 | |||
| 650 | num_dirty = count_range_bits(tree, &start, (u64)-1, | ||
| 651 | thresh, EXTENT_DIRTY); | ||
| 652 | if (num_dirty < thresh) | ||
| 653 | return 0; | ||
| 654 | } | ||
| 655 | return extent_writepages(tree, mapping, btree_get_extent, wbc); | ||
| 656 | } | ||
| 657 | |||
| 658 | static int btree_readpage(struct file *file, struct page *page) | ||
| 659 | { | ||
| 660 | struct extent_io_tree *tree; | ||
| 661 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 662 | return extent_read_full_page(tree, page, btree_get_extent); | ||
| 663 | } | ||
| 664 | |||
| 665 | static int btree_releasepage(struct page *page, gfp_t gfp_flags) | ||
| 666 | { | ||
| 667 | struct extent_io_tree *tree; | ||
| 668 | struct extent_map_tree *map; | ||
| 669 | int ret; | ||
| 670 | |||
| 671 | if (PageWriteback(page) || PageDirty(page)) | ||
| 672 | return 0; | ||
| 673 | |||
| 674 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 675 | map = &BTRFS_I(page->mapping->host)->extent_tree; | ||
| 676 | |||
| 677 | ret = try_release_extent_state(map, tree, page, gfp_flags); | ||
| 678 | if (!ret) | ||
| 679 | return 0; | ||
| 680 | |||
| 681 | ret = try_release_extent_buffer(tree, page); | ||
| 682 | if (ret == 1) { | ||
| 683 | ClearPagePrivate(page); | ||
| 684 | set_page_private(page, 0); | ||
| 685 | page_cache_release(page); | ||
| 686 | } | ||
| 687 | |||
| 688 | return ret; | ||
| 689 | } | ||
| 690 | |||
| 691 | static void btree_invalidatepage(struct page *page, unsigned long offset) | ||
| 692 | { | ||
| 693 | struct extent_io_tree *tree; | ||
| 694 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 695 | extent_invalidatepage(tree, page, offset); | ||
| 696 | btree_releasepage(page, GFP_NOFS); | ||
| 697 | if (PagePrivate(page)) { | ||
| 698 | printk(KERN_WARNING "btrfs warning page private not zero " | ||
| 699 | "on page %llu\n", (unsigned long long)page_offset(page)); | ||
| 700 | ClearPagePrivate(page); | ||
| 701 | set_page_private(page, 0); | ||
| 702 | page_cache_release(page); | ||
| 703 | } | ||
| 704 | } | ||
| 705 | |||
| 706 | #if 0 | ||
| 707 | static int btree_writepage(struct page *page, struct writeback_control *wbc) | ||
| 708 | { | ||
| 709 | struct buffer_head *bh; | ||
| 710 | struct btrfs_root *root = BTRFS_I(page->mapping->host)->root; | ||
| 711 | struct buffer_head *head; | ||
| 712 | if (!page_has_buffers(page)) { | ||
| 713 | create_empty_buffers(page, root->fs_info->sb->s_blocksize, | ||
| 714 | (1 << BH_Dirty)|(1 << BH_Uptodate)); | ||
| 715 | } | ||
| 716 | head = page_buffers(page); | ||
| 717 | bh = head; | ||
| 718 | do { | ||
| 719 | if (buffer_dirty(bh)) | ||
| 720 | csum_tree_block(root, bh, 0); | ||
| 721 | bh = bh->b_this_page; | ||
| 722 | } while (bh != head); | ||
| 723 | return block_write_full_page(page, btree_get_block, wbc); | ||
| 724 | } | ||
| 725 | #endif | ||
| 726 | |||
| 727 | static struct address_space_operations btree_aops = { | ||
| 728 | .readpage = btree_readpage, | ||
| 729 | .writepage = btree_writepage, | ||
| 730 | .writepages = btree_writepages, | ||
| 731 | .releasepage = btree_releasepage, | ||
| 732 | .invalidatepage = btree_invalidatepage, | ||
| 733 | .sync_page = block_sync_page, | ||
| 734 | }; | ||
| 735 | |||
| 736 | int readahead_tree_block(struct btrfs_root *root, u64 bytenr, u32 blocksize, | ||
| 737 | u64 parent_transid) | ||
| 738 | { | ||
| 739 | struct extent_buffer *buf = NULL; | ||
| 740 | struct inode *btree_inode = root->fs_info->btree_inode; | ||
| 741 | int ret = 0; | ||
| 742 | |||
| 743 | buf = btrfs_find_create_tree_block(root, bytenr, blocksize); | ||
| 744 | if (!buf) | ||
| 745 | return 0; | ||
| 746 | read_extent_buffer_pages(&BTRFS_I(btree_inode)->io_tree, | ||
| 747 | buf, 0, 0, btree_get_extent, 0); | ||
| 748 | free_extent_buffer(buf); | ||
| 749 | return ret; | ||
| 750 | } | ||
| 751 | |||
| 752 | struct extent_buffer *btrfs_find_tree_block(struct btrfs_root *root, | ||
| 753 | u64 bytenr, u32 blocksize) | ||
| 754 | { | ||
| 755 | struct inode *btree_inode = root->fs_info->btree_inode; | ||
| 756 | struct extent_buffer *eb; | ||
| 757 | eb = find_extent_buffer(&BTRFS_I(btree_inode)->io_tree, | ||
| 758 | bytenr, blocksize, GFP_NOFS); | ||
| 759 | return eb; | ||
| 760 | } | ||
| 761 | |||
| 762 | struct extent_buffer *btrfs_find_create_tree_block(struct btrfs_root *root, | ||
| 763 | u64 bytenr, u32 blocksize) | ||
| 764 | { | ||
| 765 | struct inode *btree_inode = root->fs_info->btree_inode; | ||
| 766 | struct extent_buffer *eb; | ||
| 767 | |||
| 768 | eb = alloc_extent_buffer(&BTRFS_I(btree_inode)->io_tree, | ||
| 769 | bytenr, blocksize, NULL, GFP_NOFS); | ||
| 770 | return eb; | ||
| 771 | } | ||
| 772 | |||
| 773 | |||
| 774 | int btrfs_write_tree_block(struct extent_buffer *buf) | ||
| 775 | { | ||
| 776 | return btrfs_fdatawrite_range(buf->first_page->mapping, buf->start, | ||
| 777 | buf->start + buf->len - 1, WB_SYNC_ALL); | ||
| 778 | } | ||
| 779 | |||
| 780 | int btrfs_wait_tree_block_writeback(struct extent_buffer *buf) | ||
| 781 | { | ||
| 782 | return btrfs_wait_on_page_writeback_range(buf->first_page->mapping, | ||
| 783 | buf->start, buf->start + buf->len - 1); | ||
| 784 | } | ||
| 785 | |||
| 786 | struct extent_buffer *read_tree_block(struct btrfs_root *root, u64 bytenr, | ||
| 787 | u32 blocksize, u64 parent_transid) | ||
| 788 | { | ||
| 789 | struct extent_buffer *buf = NULL; | ||
| 790 | struct inode *btree_inode = root->fs_info->btree_inode; | ||
| 791 | struct extent_io_tree *io_tree; | ||
| 792 | int ret; | ||
| 793 | |||
| 794 | io_tree = &BTRFS_I(btree_inode)->io_tree; | ||
| 795 | |||
| 796 | buf = btrfs_find_create_tree_block(root, bytenr, blocksize); | ||
| 797 | if (!buf) | ||
| 798 | return NULL; | ||
| 799 | |||
| 800 | ret = btree_read_extent_buffer_pages(root, buf, 0, parent_transid); | ||
| 801 | |||
| 802 | if (ret == 0) | ||
| 803 | buf->flags |= EXTENT_UPTODATE; | ||
| 804 | else | ||
| 805 | WARN_ON(1); | ||
| 806 | return buf; | ||
| 807 | |||
| 808 | } | ||
| 809 | |||
| 810 | int clean_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 811 | struct extent_buffer *buf) | ||
| 812 | { | ||
| 813 | struct inode *btree_inode = root->fs_info->btree_inode; | ||
| 814 | if (btrfs_header_generation(buf) == | ||
| 815 | root->fs_info->running_transaction->transid) { | ||
| 816 | WARN_ON(!btrfs_tree_locked(buf)); | ||
| 817 | clear_extent_buffer_dirty(&BTRFS_I(btree_inode)->io_tree, | ||
| 818 | buf); | ||
| 819 | } | ||
| 820 | return 0; | ||
| 821 | } | ||
| 822 | |||
| 823 | static int __setup_root(u32 nodesize, u32 leafsize, u32 sectorsize, | ||
| 824 | u32 stripesize, struct btrfs_root *root, | ||
| 825 | struct btrfs_fs_info *fs_info, | ||
| 826 | u64 objectid) | ||
| 827 | { | ||
| 828 | root->node = NULL; | ||
| 829 | root->commit_root = NULL; | ||
| 830 | root->ref_tree = NULL; | ||
| 831 | root->sectorsize = sectorsize; | ||
| 832 | root->nodesize = nodesize; | ||
| 833 | root->leafsize = leafsize; | ||
| 834 | root->stripesize = stripesize; | ||
| 835 | root->ref_cows = 0; | ||
| 836 | root->track_dirty = 0; | ||
| 837 | |||
| 838 | root->fs_info = fs_info; | ||
| 839 | root->objectid = objectid; | ||
| 840 | root->last_trans = 0; | ||
| 841 | root->highest_inode = 0; | ||
| 842 | root->last_inode_alloc = 0; | ||
| 843 | root->name = NULL; | ||
| 844 | root->in_sysfs = 0; | ||
| 845 | |||
| 846 | INIT_LIST_HEAD(&root->dirty_list); | ||
| 847 | INIT_LIST_HEAD(&root->orphan_list); | ||
| 848 | INIT_LIST_HEAD(&root->dead_list); | ||
| 849 | spin_lock_init(&root->node_lock); | ||
| 850 | spin_lock_init(&root->list_lock); | ||
| 851 | mutex_init(&root->objectid_mutex); | ||
| 852 | mutex_init(&root->log_mutex); | ||
| 853 | extent_io_tree_init(&root->dirty_log_pages, | ||
| 854 | fs_info->btree_inode->i_mapping, GFP_NOFS); | ||
| 855 | |||
| 856 | btrfs_leaf_ref_tree_init(&root->ref_tree_struct); | ||
| 857 | root->ref_tree = &root->ref_tree_struct; | ||
| 858 | |||
| 859 | memset(&root->root_key, 0, sizeof(root->root_key)); | ||
| 860 | memset(&root->root_item, 0, sizeof(root->root_item)); | ||
| 861 | memset(&root->defrag_progress, 0, sizeof(root->defrag_progress)); | ||
| 862 | memset(&root->root_kobj, 0, sizeof(root->root_kobj)); | ||
| 863 | root->defrag_trans_start = fs_info->generation; | ||
| 864 | init_completion(&root->kobj_unregister); | ||
| 865 | root->defrag_running = 0; | ||
| 866 | root->defrag_level = 0; | ||
| 867 | root->root_key.objectid = objectid; | ||
| 868 | root->anon_super.s_root = NULL; | ||
| 869 | root->anon_super.s_dev = 0; | ||
| 870 | INIT_LIST_HEAD(&root->anon_super.s_list); | ||
| 871 | INIT_LIST_HEAD(&root->anon_super.s_instances); | ||
| 872 | init_rwsem(&root->anon_super.s_umount); | ||
| 873 | |||
| 874 | return 0; | ||
| 875 | } | ||
| 876 | |||
| 877 | static int find_and_setup_root(struct btrfs_root *tree_root, | ||
| 878 | struct btrfs_fs_info *fs_info, | ||
| 879 | u64 objectid, | ||
| 880 | struct btrfs_root *root) | ||
| 881 | { | ||
| 882 | int ret; | ||
| 883 | u32 blocksize; | ||
| 884 | u64 generation; | ||
| 885 | |||
| 886 | __setup_root(tree_root->nodesize, tree_root->leafsize, | ||
| 887 | tree_root->sectorsize, tree_root->stripesize, | ||
| 888 | root, fs_info, objectid); | ||
| 889 | ret = btrfs_find_last_root(tree_root, objectid, | ||
| 890 | &root->root_item, &root->root_key); | ||
| 891 | BUG_ON(ret); | ||
| 892 | |||
| 893 | generation = btrfs_root_generation(&root->root_item); | ||
| 894 | blocksize = btrfs_level_size(root, btrfs_root_level(&root->root_item)); | ||
| 895 | root->node = read_tree_block(root, btrfs_root_bytenr(&root->root_item), | ||
| 896 | blocksize, generation); | ||
| 897 | BUG_ON(!root->node); | ||
| 898 | return 0; | ||
| 899 | } | ||
| 900 | |||
| 901 | int btrfs_free_log_root_tree(struct btrfs_trans_handle *trans, | ||
| 902 | struct btrfs_fs_info *fs_info) | ||
| 903 | { | ||
| 904 | struct extent_buffer *eb; | ||
| 905 | struct btrfs_root *log_root_tree = fs_info->log_root_tree; | ||
| 906 | u64 start = 0; | ||
| 907 | u64 end = 0; | ||
| 908 | int ret; | ||
| 909 | |||
| 910 | if (!log_root_tree) | ||
| 911 | return 0; | ||
| 912 | |||
| 913 | while (1) { | ||
| 914 | ret = find_first_extent_bit(&log_root_tree->dirty_log_pages, | ||
| 915 | 0, &start, &end, EXTENT_DIRTY); | ||
| 916 | if (ret) | ||
| 917 | break; | ||
| 918 | |||
| 919 | clear_extent_dirty(&log_root_tree->dirty_log_pages, | ||
| 920 | start, end, GFP_NOFS); | ||
| 921 | } | ||
| 922 | eb = fs_info->log_root_tree->node; | ||
| 923 | |||
| 924 | WARN_ON(btrfs_header_level(eb) != 0); | ||
| 925 | WARN_ON(btrfs_header_nritems(eb) != 0); | ||
| 926 | |||
| 927 | ret = btrfs_free_reserved_extent(fs_info->tree_root, | ||
| 928 | eb->start, eb->len); | ||
| 929 | BUG_ON(ret); | ||
| 930 | |||
| 931 | free_extent_buffer(eb); | ||
| 932 | kfree(fs_info->log_root_tree); | ||
| 933 | fs_info->log_root_tree = NULL; | ||
| 934 | return 0; | ||
| 935 | } | ||
| 936 | |||
| 937 | int btrfs_init_log_root_tree(struct btrfs_trans_handle *trans, | ||
| 938 | struct btrfs_fs_info *fs_info) | ||
| 939 | { | ||
| 940 | struct btrfs_root *root; | ||
| 941 | struct btrfs_root *tree_root = fs_info->tree_root; | ||
| 942 | |||
| 943 | root = kzalloc(sizeof(*root), GFP_NOFS); | ||
| 944 | if (!root) | ||
| 945 | return -ENOMEM; | ||
| 946 | |||
| 947 | __setup_root(tree_root->nodesize, tree_root->leafsize, | ||
| 948 | tree_root->sectorsize, tree_root->stripesize, | ||
| 949 | root, fs_info, BTRFS_TREE_LOG_OBJECTID); | ||
| 950 | |||
| 951 | root->root_key.objectid = BTRFS_TREE_LOG_OBJECTID; | ||
| 952 | root->root_key.type = BTRFS_ROOT_ITEM_KEY; | ||
| 953 | root->root_key.offset = BTRFS_TREE_LOG_OBJECTID; | ||
| 954 | root->ref_cows = 0; | ||
| 955 | |||
| 956 | root->node = btrfs_alloc_free_block(trans, root, root->leafsize, | ||
| 957 | 0, BTRFS_TREE_LOG_OBJECTID, | ||
| 958 | trans->transid, 0, 0, 0); | ||
| 959 | |||
| 960 | btrfs_set_header_nritems(root->node, 0); | ||
| 961 | btrfs_set_header_level(root->node, 0); | ||
| 962 | btrfs_set_header_bytenr(root->node, root->node->start); | ||
| 963 | btrfs_set_header_generation(root->node, trans->transid); | ||
| 964 | btrfs_set_header_owner(root->node, BTRFS_TREE_LOG_OBJECTID); | ||
| 965 | |||
| 966 | write_extent_buffer(root->node, root->fs_info->fsid, | ||
| 967 | (unsigned long)btrfs_header_fsid(root->node), | ||
| 968 | BTRFS_FSID_SIZE); | ||
| 969 | btrfs_mark_buffer_dirty(root->node); | ||
| 970 | btrfs_tree_unlock(root->node); | ||
| 971 | fs_info->log_root_tree = root; | ||
| 972 | return 0; | ||
| 973 | } | ||
| 974 | |||
| 975 | struct btrfs_root *btrfs_read_fs_root_no_radix(struct btrfs_root *tree_root, | ||
| 976 | struct btrfs_key *location) | ||
| 977 | { | ||
| 978 | struct btrfs_root *root; | ||
| 979 | struct btrfs_fs_info *fs_info = tree_root->fs_info; | ||
| 980 | struct btrfs_path *path; | ||
| 981 | struct extent_buffer *l; | ||
| 982 | u64 highest_inode; | ||
| 983 | u64 generation; | ||
| 984 | u32 blocksize; | ||
| 985 | int ret = 0; | ||
| 986 | |||
| 987 | root = kzalloc(sizeof(*root), GFP_NOFS); | ||
| 988 | if (!root) | ||
| 989 | return ERR_PTR(-ENOMEM); | ||
| 990 | if (location->offset == (u64)-1) { | ||
| 991 | ret = find_and_setup_root(tree_root, fs_info, | ||
| 992 | location->objectid, root); | ||
| 993 | if (ret) { | ||
| 994 | kfree(root); | ||
| 995 | return ERR_PTR(ret); | ||
| 996 | } | ||
| 997 | goto insert; | ||
| 998 | } | ||
| 999 | |||
| 1000 | __setup_root(tree_root->nodesize, tree_root->leafsize, | ||
| 1001 | tree_root->sectorsize, tree_root->stripesize, | ||
| 1002 | root, fs_info, location->objectid); | ||
| 1003 | |||
| 1004 | path = btrfs_alloc_path(); | ||
| 1005 | BUG_ON(!path); | ||
| 1006 | ret = btrfs_search_slot(NULL, tree_root, location, path, 0, 0); | ||
| 1007 | if (ret != 0) { | ||
| 1008 | if (ret > 0) | ||
| 1009 | ret = -ENOENT; | ||
| 1010 | goto out; | ||
| 1011 | } | ||
| 1012 | l = path->nodes[0]; | ||
| 1013 | read_extent_buffer(l, &root->root_item, | ||
| 1014 | btrfs_item_ptr_offset(l, path->slots[0]), | ||
| 1015 | sizeof(root->root_item)); | ||
| 1016 | memcpy(&root->root_key, location, sizeof(*location)); | ||
| 1017 | ret = 0; | ||
| 1018 | out: | ||
| 1019 | btrfs_release_path(root, path); | ||
| 1020 | btrfs_free_path(path); | ||
| 1021 | if (ret) { | ||
| 1022 | kfree(root); | ||
| 1023 | return ERR_PTR(ret); | ||
| 1024 | } | ||
| 1025 | generation = btrfs_root_generation(&root->root_item); | ||
| 1026 | blocksize = btrfs_level_size(root, btrfs_root_level(&root->root_item)); | ||
| 1027 | root->node = read_tree_block(root, btrfs_root_bytenr(&root->root_item), | ||
| 1028 | blocksize, generation); | ||
| 1029 | BUG_ON(!root->node); | ||
| 1030 | insert: | ||
| 1031 | if (location->objectid != BTRFS_TREE_LOG_OBJECTID) { | ||
| 1032 | root->ref_cows = 1; | ||
| 1033 | ret = btrfs_find_highest_inode(root, &highest_inode); | ||
| 1034 | if (ret == 0) { | ||
| 1035 | root->highest_inode = highest_inode; | ||
| 1036 | root->last_inode_alloc = highest_inode; | ||
| 1037 | } | ||
| 1038 | } | ||
| 1039 | return root; | ||
| 1040 | } | ||
| 1041 | |||
| 1042 | struct btrfs_root *btrfs_lookup_fs_root(struct btrfs_fs_info *fs_info, | ||
| 1043 | u64 root_objectid) | ||
| 1044 | { | ||
| 1045 | struct btrfs_root *root; | ||
| 1046 | |||
| 1047 | if (root_objectid == BTRFS_ROOT_TREE_OBJECTID) | ||
| 1048 | return fs_info->tree_root; | ||
| 1049 | if (root_objectid == BTRFS_EXTENT_TREE_OBJECTID) | ||
| 1050 | return fs_info->extent_root; | ||
| 1051 | |||
| 1052 | root = radix_tree_lookup(&fs_info->fs_roots_radix, | ||
| 1053 | (unsigned long)root_objectid); | ||
| 1054 | return root; | ||
| 1055 | } | ||
| 1056 | |||
| 1057 | struct btrfs_root *btrfs_read_fs_root_no_name(struct btrfs_fs_info *fs_info, | ||
| 1058 | struct btrfs_key *location) | ||
| 1059 | { | ||
| 1060 | struct btrfs_root *root; | ||
| 1061 | int ret; | ||
| 1062 | |||
| 1063 | if (location->objectid == BTRFS_ROOT_TREE_OBJECTID) | ||
| 1064 | return fs_info->tree_root; | ||
| 1065 | if (location->objectid == BTRFS_EXTENT_TREE_OBJECTID) | ||
| 1066 | return fs_info->extent_root; | ||
| 1067 | if (location->objectid == BTRFS_CHUNK_TREE_OBJECTID) | ||
| 1068 | return fs_info->chunk_root; | ||
| 1069 | if (location->objectid == BTRFS_DEV_TREE_OBJECTID) | ||
| 1070 | return fs_info->dev_root; | ||
| 1071 | if (location->objectid == BTRFS_CSUM_TREE_OBJECTID) | ||
| 1072 | return fs_info->csum_root; | ||
| 1073 | |||
| 1074 | root = radix_tree_lookup(&fs_info->fs_roots_radix, | ||
| 1075 | (unsigned long)location->objectid); | ||
| 1076 | if (root) | ||
| 1077 | return root; | ||
| 1078 | |||
| 1079 | root = btrfs_read_fs_root_no_radix(fs_info->tree_root, location); | ||
| 1080 | if (IS_ERR(root)) | ||
| 1081 | return root; | ||
| 1082 | |||
| 1083 | set_anon_super(&root->anon_super, NULL); | ||
| 1084 | |||
| 1085 | ret = radix_tree_insert(&fs_info->fs_roots_radix, | ||
| 1086 | (unsigned long)root->root_key.objectid, | ||
| 1087 | root); | ||
| 1088 | if (ret) { | ||
| 1089 | free_extent_buffer(root->node); | ||
| 1090 | kfree(root); | ||
| 1091 | return ERR_PTR(ret); | ||
| 1092 | } | ||
| 1093 | if (!(fs_info->sb->s_flags & MS_RDONLY)) { | ||
| 1094 | ret = btrfs_find_dead_roots(fs_info->tree_root, | ||
| 1095 | root->root_key.objectid, root); | ||
| 1096 | BUG_ON(ret); | ||
| 1097 | btrfs_orphan_cleanup(root); | ||
| 1098 | } | ||
| 1099 | return root; | ||
| 1100 | } | ||
| 1101 | |||
| 1102 | struct btrfs_root *btrfs_read_fs_root(struct btrfs_fs_info *fs_info, | ||
| 1103 | struct btrfs_key *location, | ||
| 1104 | const char *name, int namelen) | ||
| 1105 | { | ||
| 1106 | struct btrfs_root *root; | ||
| 1107 | int ret; | ||
| 1108 | |||
| 1109 | root = btrfs_read_fs_root_no_name(fs_info, location); | ||
| 1110 | if (!root) | ||
| 1111 | return NULL; | ||
| 1112 | |||
| 1113 | if (root->in_sysfs) | ||
| 1114 | return root; | ||
| 1115 | |||
| 1116 | ret = btrfs_set_root_name(root, name, namelen); | ||
| 1117 | if (ret) { | ||
| 1118 | free_extent_buffer(root->node); | ||
| 1119 | kfree(root); | ||
| 1120 | return ERR_PTR(ret); | ||
| 1121 | } | ||
| 1122 | #if 0 | ||
| 1123 | ret = btrfs_sysfs_add_root(root); | ||
| 1124 | if (ret) { | ||
| 1125 | free_extent_buffer(root->node); | ||
| 1126 | kfree(root->name); | ||
| 1127 | kfree(root); | ||
| 1128 | return ERR_PTR(ret); | ||
| 1129 | } | ||
| 1130 | #endif | ||
| 1131 | root->in_sysfs = 1; | ||
| 1132 | return root; | ||
| 1133 | } | ||
| 1134 | |||
| 1135 | static int btrfs_congested_fn(void *congested_data, int bdi_bits) | ||
| 1136 | { | ||
| 1137 | struct btrfs_fs_info *info = (struct btrfs_fs_info *)congested_data; | ||
| 1138 | int ret = 0; | ||
| 1139 | struct list_head *cur; | ||
| 1140 | struct btrfs_device *device; | ||
| 1141 | struct backing_dev_info *bdi; | ||
| 1142 | #if 0 | ||
| 1143 | if ((bdi_bits & (1 << BDI_write_congested)) && | ||
| 1144 | btrfs_congested_async(info, 0)) | ||
| 1145 | return 1; | ||
| 1146 | #endif | ||
| 1147 | list_for_each(cur, &info->fs_devices->devices) { | ||
| 1148 | device = list_entry(cur, struct btrfs_device, dev_list); | ||
| 1149 | if (!device->bdev) | ||
| 1150 | continue; | ||
| 1151 | bdi = blk_get_backing_dev_info(device->bdev); | ||
| 1152 | if (bdi && bdi_congested(bdi, bdi_bits)) { | ||
| 1153 | ret = 1; | ||
| 1154 | break; | ||
| 1155 | } | ||
| 1156 | } | ||
| 1157 | return ret; | ||
| 1158 | } | ||
| 1159 | |||
| 1160 | /* | ||
| 1161 | * this unplugs every device on the box, and it is only used when page | ||
| 1162 | * is null | ||
| 1163 | */ | ||
| 1164 | static void __unplug_io_fn(struct backing_dev_info *bdi, struct page *page) | ||
| 1165 | { | ||
| 1166 | struct list_head *cur; | ||
| 1167 | struct btrfs_device *device; | ||
| 1168 | struct btrfs_fs_info *info; | ||
| 1169 | |||
| 1170 | info = (struct btrfs_fs_info *)bdi->unplug_io_data; | ||
| 1171 | list_for_each(cur, &info->fs_devices->devices) { | ||
| 1172 | device = list_entry(cur, struct btrfs_device, dev_list); | ||
| 1173 | if (!device->bdev) | ||
| 1174 | continue; | ||
| 1175 | |||
| 1176 | bdi = blk_get_backing_dev_info(device->bdev); | ||
| 1177 | if (bdi->unplug_io_fn) | ||
| 1178 | bdi->unplug_io_fn(bdi, page); | ||
| 1179 | } | ||
| 1180 | } | ||
| 1181 | |||
| 1182 | static void btrfs_unplug_io_fn(struct backing_dev_info *bdi, struct page *page) | ||
| 1183 | { | ||
| 1184 | struct inode *inode; | ||
| 1185 | struct extent_map_tree *em_tree; | ||
| 1186 | struct extent_map *em; | ||
| 1187 | struct address_space *mapping; | ||
| 1188 | u64 offset; | ||
| 1189 | |||
| 1190 | /* the generic O_DIRECT read code does this */ | ||
| 1191 | if (1 || !page) { | ||
| 1192 | __unplug_io_fn(bdi, page); | ||
| 1193 | return; | ||
| 1194 | } | ||
| 1195 | |||
| 1196 | /* | ||
| 1197 | * page->mapping may change at any time. Get a consistent copy | ||
| 1198 | * and use that for everything below | ||
| 1199 | */ | ||
| 1200 | smp_mb(); | ||
| 1201 | mapping = page->mapping; | ||
| 1202 | if (!mapping) | ||
| 1203 | return; | ||
| 1204 | |||
| 1205 | inode = mapping->host; | ||
| 1206 | |||
| 1207 | /* | ||
| 1208 | * don't do the expensive searching for a small number of | ||
| 1209 | * devices | ||
| 1210 | */ | ||
| 1211 | if (BTRFS_I(inode)->root->fs_info->fs_devices->open_devices <= 2) { | ||
| 1212 | __unplug_io_fn(bdi, page); | ||
| 1213 | return; | ||
| 1214 | } | ||
| 1215 | |||
| 1216 | offset = page_offset(page); | ||
| 1217 | |||
| 1218 | em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 1219 | spin_lock(&em_tree->lock); | ||
| 1220 | em = lookup_extent_mapping(em_tree, offset, PAGE_CACHE_SIZE); | ||
| 1221 | spin_unlock(&em_tree->lock); | ||
| 1222 | if (!em) { | ||
| 1223 | __unplug_io_fn(bdi, page); | ||
| 1224 | return; | ||
| 1225 | } | ||
| 1226 | |||
| 1227 | if (em->block_start >= EXTENT_MAP_LAST_BYTE) { | ||
| 1228 | free_extent_map(em); | ||
| 1229 | __unplug_io_fn(bdi, page); | ||
| 1230 | return; | ||
| 1231 | } | ||
| 1232 | offset = offset - em->start; | ||
| 1233 | btrfs_unplug_page(&BTRFS_I(inode)->root->fs_info->mapping_tree, | ||
| 1234 | em->block_start + offset, page); | ||
| 1235 | free_extent_map(em); | ||
| 1236 | } | ||
| 1237 | |||
| 1238 | static int setup_bdi(struct btrfs_fs_info *info, struct backing_dev_info *bdi) | ||
| 1239 | { | ||
| 1240 | bdi_init(bdi); | ||
| 1241 | bdi->ra_pages = default_backing_dev_info.ra_pages; | ||
| 1242 | bdi->state = 0; | ||
| 1243 | bdi->capabilities = default_backing_dev_info.capabilities; | ||
| 1244 | bdi->unplug_io_fn = btrfs_unplug_io_fn; | ||
| 1245 | bdi->unplug_io_data = info; | ||
| 1246 | bdi->congested_fn = btrfs_congested_fn; | ||
| 1247 | bdi->congested_data = info; | ||
| 1248 | return 0; | ||
| 1249 | } | ||
| 1250 | |||
| 1251 | static int bio_ready_for_csum(struct bio *bio) | ||
| 1252 | { | ||
| 1253 | u64 length = 0; | ||
| 1254 | u64 buf_len = 0; | ||
| 1255 | u64 start = 0; | ||
| 1256 | struct page *page; | ||
| 1257 | struct extent_io_tree *io_tree = NULL; | ||
| 1258 | struct btrfs_fs_info *info = NULL; | ||
| 1259 | struct bio_vec *bvec; | ||
| 1260 | int i; | ||
| 1261 | int ret; | ||
| 1262 | |||
| 1263 | bio_for_each_segment(bvec, bio, i) { | ||
| 1264 | page = bvec->bv_page; | ||
| 1265 | if (page->private == EXTENT_PAGE_PRIVATE) { | ||
| 1266 | length += bvec->bv_len; | ||
| 1267 | continue; | ||
| 1268 | } | ||
| 1269 | if (!page->private) { | ||
| 1270 | length += bvec->bv_len; | ||
| 1271 | continue; | ||
| 1272 | } | ||
| 1273 | length = bvec->bv_len; | ||
| 1274 | buf_len = page->private >> 2; | ||
| 1275 | start = page_offset(page) + bvec->bv_offset; | ||
| 1276 | io_tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 1277 | info = BTRFS_I(page->mapping->host)->root->fs_info; | ||
| 1278 | } | ||
| 1279 | /* are we fully contained in this bio? */ | ||
| 1280 | if (buf_len <= length) | ||
| 1281 | return 1; | ||
| 1282 | |||
| 1283 | ret = extent_range_uptodate(io_tree, start + length, | ||
| 1284 | start + buf_len - 1); | ||
| 1285 | if (ret == 1) | ||
| 1286 | return ret; | ||
| 1287 | return ret; | ||
| 1288 | } | ||
| 1289 | |||
| 1290 | /* | ||
| 1291 | * called by the kthread helper functions to finally call the bio end_io | ||
| 1292 | * functions. This is where read checksum verification actually happens | ||
| 1293 | */ | ||
| 1294 | static void end_workqueue_fn(struct btrfs_work *work) | ||
| 1295 | { | ||
| 1296 | struct bio *bio; | ||
| 1297 | struct end_io_wq *end_io_wq; | ||
| 1298 | struct btrfs_fs_info *fs_info; | ||
| 1299 | int error; | ||
| 1300 | |||
| 1301 | end_io_wq = container_of(work, struct end_io_wq, work); | ||
| 1302 | bio = end_io_wq->bio; | ||
| 1303 | fs_info = end_io_wq->info; | ||
| 1304 | |||
| 1305 | /* metadata bio reads are special because the whole tree block must | ||
| 1306 | * be checksummed at once. This makes sure the entire block is in | ||
| 1307 | * ram and up to date before trying to verify things. For | ||
| 1308 | * blocksize <= pagesize, it is basically a noop | ||
| 1309 | */ | ||
| 1310 | if (!(bio->bi_rw & (1 << BIO_RW)) && end_io_wq->metadata && | ||
| 1311 | !bio_ready_for_csum(bio)) { | ||
| 1312 | btrfs_queue_worker(&fs_info->endio_meta_workers, | ||
| 1313 | &end_io_wq->work); | ||
| 1314 | return; | ||
| 1315 | } | ||
| 1316 | error = end_io_wq->error; | ||
| 1317 | bio->bi_private = end_io_wq->private; | ||
| 1318 | bio->bi_end_io = end_io_wq->end_io; | ||
| 1319 | kfree(end_io_wq); | ||
| 1320 | bio_endio(bio, error); | ||
| 1321 | } | ||
| 1322 | |||
| 1323 | static int cleaner_kthread(void *arg) | ||
| 1324 | { | ||
| 1325 | struct btrfs_root *root = arg; | ||
| 1326 | |||
| 1327 | do { | ||
| 1328 | smp_mb(); | ||
| 1329 | if (root->fs_info->closing) | ||
| 1330 | break; | ||
| 1331 | |||
| 1332 | vfs_check_frozen(root->fs_info->sb, SB_FREEZE_WRITE); | ||
| 1333 | mutex_lock(&root->fs_info->cleaner_mutex); | ||
| 1334 | btrfs_clean_old_snapshots(root); | ||
| 1335 | mutex_unlock(&root->fs_info->cleaner_mutex); | ||
| 1336 | |||
| 1337 | if (freezing(current)) { | ||
| 1338 | refrigerator(); | ||
| 1339 | } else { | ||
| 1340 | smp_mb(); | ||
| 1341 | if (root->fs_info->closing) | ||
| 1342 | break; | ||
| 1343 | set_current_state(TASK_INTERRUPTIBLE); | ||
| 1344 | schedule(); | ||
| 1345 | __set_current_state(TASK_RUNNING); | ||
| 1346 | } | ||
| 1347 | } while (!kthread_should_stop()); | ||
| 1348 | return 0; | ||
| 1349 | } | ||
| 1350 | |||
| 1351 | static int transaction_kthread(void *arg) | ||
| 1352 | { | ||
| 1353 | struct btrfs_root *root = arg; | ||
| 1354 | struct btrfs_trans_handle *trans; | ||
| 1355 | struct btrfs_transaction *cur; | ||
| 1356 | unsigned long now; | ||
| 1357 | unsigned long delay; | ||
| 1358 | int ret; | ||
| 1359 | |||
| 1360 | do { | ||
| 1361 | smp_mb(); | ||
| 1362 | if (root->fs_info->closing) | ||
| 1363 | break; | ||
| 1364 | |||
| 1365 | delay = HZ * 30; | ||
| 1366 | vfs_check_frozen(root->fs_info->sb, SB_FREEZE_WRITE); | ||
| 1367 | mutex_lock(&root->fs_info->transaction_kthread_mutex); | ||
| 1368 | |||
| 1369 | if (root->fs_info->total_ref_cache_size > 20 * 1024 * 1024) { | ||
| 1370 | printk(KERN_INFO "btrfs: total reference cache " | ||
| 1371 | "size %llu\n", | ||
| 1372 | root->fs_info->total_ref_cache_size); | ||
| 1373 | } | ||
| 1374 | |||
| 1375 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 1376 | cur = root->fs_info->running_transaction; | ||
| 1377 | if (!cur) { | ||
| 1378 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1379 | goto sleep; | ||
| 1380 | } | ||
| 1381 | |||
| 1382 | now = get_seconds(); | ||
| 1383 | if (now < cur->start_time || now - cur->start_time < 30) { | ||
| 1384 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1385 | delay = HZ * 5; | ||
| 1386 | goto sleep; | ||
| 1387 | } | ||
| 1388 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1389 | trans = btrfs_start_transaction(root, 1); | ||
| 1390 | ret = btrfs_commit_transaction(trans, root); | ||
| 1391 | sleep: | ||
| 1392 | wake_up_process(root->fs_info->cleaner_kthread); | ||
| 1393 | mutex_unlock(&root->fs_info->transaction_kthread_mutex); | ||
| 1394 | |||
| 1395 | if (freezing(current)) { | ||
| 1396 | refrigerator(); | ||
| 1397 | } else { | ||
| 1398 | if (root->fs_info->closing) | ||
| 1399 | break; | ||
| 1400 | set_current_state(TASK_INTERRUPTIBLE); | ||
| 1401 | schedule_timeout(delay); | ||
| 1402 | __set_current_state(TASK_RUNNING); | ||
| 1403 | } | ||
| 1404 | } while (!kthread_should_stop()); | ||
| 1405 | return 0; | ||
| 1406 | } | ||
| 1407 | |||
| 1408 | struct btrfs_root *open_ctree(struct super_block *sb, | ||
| 1409 | struct btrfs_fs_devices *fs_devices, | ||
| 1410 | char *options) | ||
| 1411 | { | ||
| 1412 | u32 sectorsize; | ||
| 1413 | u32 nodesize; | ||
| 1414 | u32 leafsize; | ||
| 1415 | u32 blocksize; | ||
| 1416 | u32 stripesize; | ||
| 1417 | u64 generation; | ||
| 1418 | u64 features; | ||
| 1419 | struct btrfs_key location; | ||
| 1420 | struct buffer_head *bh; | ||
| 1421 | struct btrfs_root *extent_root = kzalloc(sizeof(struct btrfs_root), | ||
| 1422 | GFP_NOFS); | ||
| 1423 | struct btrfs_root *csum_root = kzalloc(sizeof(struct btrfs_root), | ||
| 1424 | GFP_NOFS); | ||
| 1425 | struct btrfs_root *tree_root = kzalloc(sizeof(struct btrfs_root), | ||
| 1426 | GFP_NOFS); | ||
| 1427 | struct btrfs_fs_info *fs_info = kzalloc(sizeof(*fs_info), | ||
| 1428 | GFP_NOFS); | ||
| 1429 | struct btrfs_root *chunk_root = kzalloc(sizeof(struct btrfs_root), | ||
| 1430 | GFP_NOFS); | ||
| 1431 | struct btrfs_root *dev_root = kzalloc(sizeof(struct btrfs_root), | ||
| 1432 | GFP_NOFS); | ||
| 1433 | struct btrfs_root *log_tree_root; | ||
| 1434 | |||
| 1435 | int ret; | ||
| 1436 | int err = -EINVAL; | ||
| 1437 | |||
| 1438 | struct btrfs_super_block *disk_super; | ||
| 1439 | |||
| 1440 | if (!extent_root || !tree_root || !fs_info || | ||
| 1441 | !chunk_root || !dev_root || !csum_root) { | ||
| 1442 | err = -ENOMEM; | ||
| 1443 | goto fail; | ||
| 1444 | } | ||
| 1445 | INIT_RADIX_TREE(&fs_info->fs_roots_radix, GFP_NOFS); | ||
| 1446 | INIT_LIST_HEAD(&fs_info->trans_list); | ||
| 1447 | INIT_LIST_HEAD(&fs_info->dead_roots); | ||
| 1448 | INIT_LIST_HEAD(&fs_info->hashers); | ||
| 1449 | INIT_LIST_HEAD(&fs_info->delalloc_inodes); | ||
| 1450 | spin_lock_init(&fs_info->hash_lock); | ||
| 1451 | spin_lock_init(&fs_info->delalloc_lock); | ||
| 1452 | spin_lock_init(&fs_info->new_trans_lock); | ||
| 1453 | spin_lock_init(&fs_info->ref_cache_lock); | ||
| 1454 | |||
| 1455 | init_completion(&fs_info->kobj_unregister); | ||
| 1456 | fs_info->tree_root = tree_root; | ||
| 1457 | fs_info->extent_root = extent_root; | ||
| 1458 | fs_info->csum_root = csum_root; | ||
| 1459 | fs_info->chunk_root = chunk_root; | ||
| 1460 | fs_info->dev_root = dev_root; | ||
| 1461 | fs_info->fs_devices = fs_devices; | ||
| 1462 | INIT_LIST_HEAD(&fs_info->dirty_cowonly_roots); | ||
| 1463 | INIT_LIST_HEAD(&fs_info->space_info); | ||
| 1464 | btrfs_mapping_init(&fs_info->mapping_tree); | ||
| 1465 | atomic_set(&fs_info->nr_async_submits, 0); | ||
| 1466 | atomic_set(&fs_info->async_delalloc_pages, 0); | ||
| 1467 | atomic_set(&fs_info->async_submit_draining, 0); | ||
| 1468 | atomic_set(&fs_info->nr_async_bios, 0); | ||
| 1469 | atomic_set(&fs_info->throttles, 0); | ||
| 1470 | atomic_set(&fs_info->throttle_gen, 0); | ||
| 1471 | fs_info->sb = sb; | ||
| 1472 | fs_info->max_extent = (u64)-1; | ||
| 1473 | fs_info->max_inline = 8192 * 1024; | ||
| 1474 | setup_bdi(fs_info, &fs_info->bdi); | ||
| 1475 | fs_info->btree_inode = new_inode(sb); | ||
| 1476 | fs_info->btree_inode->i_ino = 1; | ||
| 1477 | fs_info->btree_inode->i_nlink = 1; | ||
| 1478 | |||
| 1479 | fs_info->thread_pool_size = min_t(unsigned long, | ||
| 1480 | num_online_cpus() + 2, 8); | ||
| 1481 | |||
| 1482 | INIT_LIST_HEAD(&fs_info->ordered_extents); | ||
| 1483 | spin_lock_init(&fs_info->ordered_extent_lock); | ||
| 1484 | |||
| 1485 | sb->s_blocksize = 4096; | ||
| 1486 | sb->s_blocksize_bits = blksize_bits(4096); | ||
| 1487 | |||
| 1488 | /* | ||
| 1489 | * we set the i_size on the btree inode to the max possible int. | ||
| 1490 | * the real end of the address space is determined by all of | ||
| 1491 | * the devices in the system | ||
| 1492 | */ | ||
| 1493 | fs_info->btree_inode->i_size = OFFSET_MAX; | ||
| 1494 | fs_info->btree_inode->i_mapping->a_ops = &btree_aops; | ||
| 1495 | fs_info->btree_inode->i_mapping->backing_dev_info = &fs_info->bdi; | ||
| 1496 | |||
| 1497 | extent_io_tree_init(&BTRFS_I(fs_info->btree_inode)->io_tree, | ||
| 1498 | fs_info->btree_inode->i_mapping, | ||
| 1499 | GFP_NOFS); | ||
| 1500 | extent_map_tree_init(&BTRFS_I(fs_info->btree_inode)->extent_tree, | ||
| 1501 | GFP_NOFS); | ||
| 1502 | |||
| 1503 | BTRFS_I(fs_info->btree_inode)->io_tree.ops = &btree_extent_io_ops; | ||
| 1504 | |||
| 1505 | spin_lock_init(&fs_info->block_group_cache_lock); | ||
| 1506 | fs_info->block_group_cache_tree.rb_node = NULL; | ||
| 1507 | |||
| 1508 | extent_io_tree_init(&fs_info->pinned_extents, | ||
| 1509 | fs_info->btree_inode->i_mapping, GFP_NOFS); | ||
| 1510 | extent_io_tree_init(&fs_info->pending_del, | ||
| 1511 | fs_info->btree_inode->i_mapping, GFP_NOFS); | ||
| 1512 | extent_io_tree_init(&fs_info->extent_ins, | ||
| 1513 | fs_info->btree_inode->i_mapping, GFP_NOFS); | ||
| 1514 | fs_info->do_barriers = 1; | ||
| 1515 | |||
| 1516 | INIT_LIST_HEAD(&fs_info->dead_reloc_roots); | ||
| 1517 | btrfs_leaf_ref_tree_init(&fs_info->reloc_ref_tree); | ||
| 1518 | btrfs_leaf_ref_tree_init(&fs_info->shared_ref_tree); | ||
| 1519 | |||
| 1520 | BTRFS_I(fs_info->btree_inode)->root = tree_root; | ||
| 1521 | memset(&BTRFS_I(fs_info->btree_inode)->location, 0, | ||
| 1522 | sizeof(struct btrfs_key)); | ||
| 1523 | insert_inode_hash(fs_info->btree_inode); | ||
| 1524 | |||
| 1525 | mutex_init(&fs_info->trans_mutex); | ||
| 1526 | mutex_init(&fs_info->tree_log_mutex); | ||
| 1527 | mutex_init(&fs_info->drop_mutex); | ||
| 1528 | mutex_init(&fs_info->extent_ins_mutex); | ||
| 1529 | mutex_init(&fs_info->pinned_mutex); | ||
| 1530 | mutex_init(&fs_info->chunk_mutex); | ||
| 1531 | mutex_init(&fs_info->transaction_kthread_mutex); | ||
| 1532 | mutex_init(&fs_info->cleaner_mutex); | ||
| 1533 | mutex_init(&fs_info->volume_mutex); | ||
| 1534 | mutex_init(&fs_info->tree_reloc_mutex); | ||
| 1535 | init_waitqueue_head(&fs_info->transaction_throttle); | ||
| 1536 | init_waitqueue_head(&fs_info->transaction_wait); | ||
| 1537 | init_waitqueue_head(&fs_info->async_submit_wait); | ||
| 1538 | init_waitqueue_head(&fs_info->tree_log_wait); | ||
| 1539 | atomic_set(&fs_info->tree_log_commit, 0); | ||
| 1540 | atomic_set(&fs_info->tree_log_writers, 0); | ||
| 1541 | fs_info->tree_log_transid = 0; | ||
| 1542 | |||
| 1543 | __setup_root(4096, 4096, 4096, 4096, tree_root, | ||
| 1544 | fs_info, BTRFS_ROOT_TREE_OBJECTID); | ||
| 1545 | |||
| 1546 | |||
| 1547 | bh = btrfs_read_dev_super(fs_devices->latest_bdev); | ||
| 1548 | if (!bh) | ||
| 1549 | goto fail_iput; | ||
| 1550 | |||
| 1551 | memcpy(&fs_info->super_copy, bh->b_data, sizeof(fs_info->super_copy)); | ||
| 1552 | memcpy(&fs_info->super_for_commit, &fs_info->super_copy, | ||
| 1553 | sizeof(fs_info->super_for_commit)); | ||
| 1554 | brelse(bh); | ||
| 1555 | |||
| 1556 | memcpy(fs_info->fsid, fs_info->super_copy.fsid, BTRFS_FSID_SIZE); | ||
| 1557 | |||
| 1558 | disk_super = &fs_info->super_copy; | ||
| 1559 | if (!btrfs_super_root(disk_super)) | ||
| 1560 | goto fail_iput; | ||
| 1561 | |||
| 1562 | ret = btrfs_parse_options(tree_root, options); | ||
| 1563 | if (ret) { | ||
| 1564 | err = ret; | ||
| 1565 | goto fail_iput; | ||
| 1566 | } | ||
| 1567 | |||
| 1568 | features = btrfs_super_incompat_flags(disk_super) & | ||
| 1569 | ~BTRFS_FEATURE_INCOMPAT_SUPP; | ||
| 1570 | if (features) { | ||
| 1571 | printk(KERN_ERR "BTRFS: couldn't mount because of " | ||
| 1572 | "unsupported optional features (%Lx).\n", | ||
| 1573 | features); | ||
| 1574 | err = -EINVAL; | ||
| 1575 | goto fail_iput; | ||
| 1576 | } | ||
| 1577 | |||
| 1578 | features = btrfs_super_compat_ro_flags(disk_super) & | ||
| 1579 | ~BTRFS_FEATURE_COMPAT_RO_SUPP; | ||
| 1580 | if (!(sb->s_flags & MS_RDONLY) && features) { | ||
| 1581 | printk(KERN_ERR "BTRFS: couldn't mount RDWR because of " | ||
| 1582 | "unsupported option features (%Lx).\n", | ||
| 1583 | features); | ||
| 1584 | err = -EINVAL; | ||
| 1585 | goto fail_iput; | ||
| 1586 | } | ||
| 1587 | |||
| 1588 | /* | ||
| 1589 | * we need to start all the end_io workers up front because the | ||
| 1590 | * queue work function gets called at interrupt time, and so it | ||
| 1591 | * cannot dynamically grow. | ||
| 1592 | */ | ||
| 1593 | btrfs_init_workers(&fs_info->workers, "worker", | ||
| 1594 | fs_info->thread_pool_size); | ||
| 1595 | |||
| 1596 | btrfs_init_workers(&fs_info->delalloc_workers, "delalloc", | ||
| 1597 | fs_info->thread_pool_size); | ||
| 1598 | |||
| 1599 | btrfs_init_workers(&fs_info->submit_workers, "submit", | ||
| 1600 | min_t(u64, fs_devices->num_devices, | ||
| 1601 | fs_info->thread_pool_size)); | ||
| 1602 | |||
| 1603 | /* a higher idle thresh on the submit workers makes it much more | ||
| 1604 | * likely that bios will be send down in a sane order to the | ||
| 1605 | * devices | ||
| 1606 | */ | ||
| 1607 | fs_info->submit_workers.idle_thresh = 64; | ||
| 1608 | |||
| 1609 | fs_info->workers.idle_thresh = 16; | ||
| 1610 | fs_info->workers.ordered = 1; | ||
| 1611 | |||
| 1612 | fs_info->delalloc_workers.idle_thresh = 2; | ||
| 1613 | fs_info->delalloc_workers.ordered = 1; | ||
| 1614 | |||
| 1615 | btrfs_init_workers(&fs_info->fixup_workers, "fixup", 1); | ||
| 1616 | btrfs_init_workers(&fs_info->endio_workers, "endio", | ||
| 1617 | fs_info->thread_pool_size); | ||
| 1618 | btrfs_init_workers(&fs_info->endio_meta_workers, "endio-meta", | ||
| 1619 | fs_info->thread_pool_size); | ||
| 1620 | btrfs_init_workers(&fs_info->endio_meta_write_workers, | ||
| 1621 | "endio-meta-write", fs_info->thread_pool_size); | ||
| 1622 | btrfs_init_workers(&fs_info->endio_write_workers, "endio-write", | ||
| 1623 | fs_info->thread_pool_size); | ||
| 1624 | |||
| 1625 | /* | ||
| 1626 | * endios are largely parallel and should have a very | ||
| 1627 | * low idle thresh | ||
| 1628 | */ | ||
| 1629 | fs_info->endio_workers.idle_thresh = 4; | ||
| 1630 | fs_info->endio_write_workers.idle_thresh = 64; | ||
| 1631 | fs_info->endio_meta_write_workers.idle_thresh = 64; | ||
| 1632 | |||
| 1633 | btrfs_start_workers(&fs_info->workers, 1); | ||
| 1634 | btrfs_start_workers(&fs_info->submit_workers, 1); | ||
| 1635 | btrfs_start_workers(&fs_info->delalloc_workers, 1); | ||
| 1636 | btrfs_start_workers(&fs_info->fixup_workers, 1); | ||
| 1637 | btrfs_start_workers(&fs_info->endio_workers, fs_info->thread_pool_size); | ||
| 1638 | btrfs_start_workers(&fs_info->endio_meta_workers, | ||
| 1639 | fs_info->thread_pool_size); | ||
| 1640 | btrfs_start_workers(&fs_info->endio_meta_write_workers, | ||
| 1641 | fs_info->thread_pool_size); | ||
| 1642 | btrfs_start_workers(&fs_info->endio_write_workers, | ||
| 1643 | fs_info->thread_pool_size); | ||
| 1644 | |||
| 1645 | fs_info->bdi.ra_pages *= btrfs_super_num_devices(disk_super); | ||
| 1646 | fs_info->bdi.ra_pages = max(fs_info->bdi.ra_pages, | ||
| 1647 | 4 * 1024 * 1024 / PAGE_CACHE_SIZE); | ||
| 1648 | |||
| 1649 | nodesize = btrfs_super_nodesize(disk_super); | ||
| 1650 | leafsize = btrfs_super_leafsize(disk_super); | ||
| 1651 | sectorsize = btrfs_super_sectorsize(disk_super); | ||
| 1652 | stripesize = btrfs_super_stripesize(disk_super); | ||
| 1653 | tree_root->nodesize = nodesize; | ||
| 1654 | tree_root->leafsize = leafsize; | ||
| 1655 | tree_root->sectorsize = sectorsize; | ||
| 1656 | tree_root->stripesize = stripesize; | ||
| 1657 | |||
| 1658 | sb->s_blocksize = sectorsize; | ||
| 1659 | sb->s_blocksize_bits = blksize_bits(sectorsize); | ||
| 1660 | |||
| 1661 | if (strncmp((char *)(&disk_super->magic), BTRFS_MAGIC, | ||
| 1662 | sizeof(disk_super->magic))) { | ||
| 1663 | printk(KERN_INFO "btrfs: valid FS not found on %s\n", sb->s_id); | ||
| 1664 | goto fail_sb_buffer; | ||
| 1665 | } | ||
| 1666 | |||
| 1667 | mutex_lock(&fs_info->chunk_mutex); | ||
| 1668 | ret = btrfs_read_sys_array(tree_root); | ||
| 1669 | mutex_unlock(&fs_info->chunk_mutex); | ||
| 1670 | if (ret) { | ||
| 1671 | printk(KERN_WARNING "btrfs: failed to read the system " | ||
| 1672 | "array on %s\n", sb->s_id); | ||
| 1673 | goto fail_sys_array; | ||
| 1674 | } | ||
| 1675 | |||
| 1676 | blocksize = btrfs_level_size(tree_root, | ||
| 1677 | btrfs_super_chunk_root_level(disk_super)); | ||
| 1678 | generation = btrfs_super_chunk_root_generation(disk_super); | ||
| 1679 | |||
| 1680 | __setup_root(nodesize, leafsize, sectorsize, stripesize, | ||
| 1681 | chunk_root, fs_info, BTRFS_CHUNK_TREE_OBJECTID); | ||
| 1682 | |||
| 1683 | chunk_root->node = read_tree_block(chunk_root, | ||
| 1684 | btrfs_super_chunk_root(disk_super), | ||
| 1685 | blocksize, generation); | ||
| 1686 | BUG_ON(!chunk_root->node); | ||
| 1687 | |||
| 1688 | read_extent_buffer(chunk_root->node, fs_info->chunk_tree_uuid, | ||
| 1689 | (unsigned long)btrfs_header_chunk_tree_uuid(chunk_root->node), | ||
| 1690 | BTRFS_UUID_SIZE); | ||
| 1691 | |||
| 1692 | mutex_lock(&fs_info->chunk_mutex); | ||
| 1693 | ret = btrfs_read_chunk_tree(chunk_root); | ||
| 1694 | mutex_unlock(&fs_info->chunk_mutex); | ||
| 1695 | if (ret) { | ||
| 1696 | printk(KERN_WARNING "btrfs: failed to read chunk tree on %s\n", | ||
| 1697 | sb->s_id); | ||
| 1698 | goto fail_chunk_root; | ||
| 1699 | } | ||
| 1700 | |||
| 1701 | btrfs_close_extra_devices(fs_devices); | ||
| 1702 | |||
| 1703 | blocksize = btrfs_level_size(tree_root, | ||
| 1704 | btrfs_super_root_level(disk_super)); | ||
| 1705 | generation = btrfs_super_generation(disk_super); | ||
| 1706 | |||
| 1707 | tree_root->node = read_tree_block(tree_root, | ||
| 1708 | btrfs_super_root(disk_super), | ||
| 1709 | blocksize, generation); | ||
| 1710 | if (!tree_root->node) | ||
| 1711 | goto fail_chunk_root; | ||
| 1712 | |||
| 1713 | |||
| 1714 | ret = find_and_setup_root(tree_root, fs_info, | ||
| 1715 | BTRFS_EXTENT_TREE_OBJECTID, extent_root); | ||
| 1716 | if (ret) | ||
| 1717 | goto fail_tree_root; | ||
| 1718 | extent_root->track_dirty = 1; | ||
| 1719 | |||
| 1720 | ret = find_and_setup_root(tree_root, fs_info, | ||
| 1721 | BTRFS_DEV_TREE_OBJECTID, dev_root); | ||
| 1722 | dev_root->track_dirty = 1; | ||
| 1723 | |||
| 1724 | if (ret) | ||
| 1725 | goto fail_extent_root; | ||
| 1726 | |||
| 1727 | ret = find_and_setup_root(tree_root, fs_info, | ||
| 1728 | BTRFS_CSUM_TREE_OBJECTID, csum_root); | ||
| 1729 | if (ret) | ||
| 1730 | goto fail_extent_root; | ||
| 1731 | |||
| 1732 | csum_root->track_dirty = 1; | ||
| 1733 | |||
| 1734 | btrfs_read_block_groups(extent_root); | ||
| 1735 | |||
| 1736 | fs_info->generation = generation; | ||
| 1737 | fs_info->last_trans_committed = generation; | ||
| 1738 | fs_info->data_alloc_profile = (u64)-1; | ||
| 1739 | fs_info->metadata_alloc_profile = (u64)-1; | ||
| 1740 | fs_info->system_alloc_profile = fs_info->metadata_alloc_profile; | ||
| 1741 | fs_info->cleaner_kthread = kthread_run(cleaner_kthread, tree_root, | ||
| 1742 | "btrfs-cleaner"); | ||
| 1743 | if (!fs_info->cleaner_kthread) | ||
| 1744 | goto fail_csum_root; | ||
| 1745 | |||
| 1746 | fs_info->transaction_kthread = kthread_run(transaction_kthread, | ||
| 1747 | tree_root, | ||
| 1748 | "btrfs-transaction"); | ||
| 1749 | if (!fs_info->transaction_kthread) | ||
| 1750 | goto fail_cleaner; | ||
| 1751 | |||
| 1752 | if (btrfs_super_log_root(disk_super) != 0) { | ||
| 1753 | u64 bytenr = btrfs_super_log_root(disk_super); | ||
| 1754 | |||
| 1755 | if (fs_devices->rw_devices == 0) { | ||
| 1756 | printk(KERN_WARNING "Btrfs log replay required " | ||
| 1757 | "on RO media\n"); | ||
| 1758 | err = -EIO; | ||
| 1759 | goto fail_trans_kthread; | ||
| 1760 | } | ||
| 1761 | blocksize = | ||
| 1762 | btrfs_level_size(tree_root, | ||
| 1763 | btrfs_super_log_root_level(disk_super)); | ||
| 1764 | |||
| 1765 | log_tree_root = kzalloc(sizeof(struct btrfs_root), | ||
| 1766 | GFP_NOFS); | ||
| 1767 | |||
| 1768 | __setup_root(nodesize, leafsize, sectorsize, stripesize, | ||
| 1769 | log_tree_root, fs_info, BTRFS_TREE_LOG_OBJECTID); | ||
| 1770 | |||
| 1771 | log_tree_root->node = read_tree_block(tree_root, bytenr, | ||
| 1772 | blocksize, | ||
| 1773 | generation + 1); | ||
| 1774 | ret = btrfs_recover_log_trees(log_tree_root); | ||
| 1775 | BUG_ON(ret); | ||
| 1776 | |||
| 1777 | if (sb->s_flags & MS_RDONLY) { | ||
| 1778 | ret = btrfs_commit_super(tree_root); | ||
| 1779 | BUG_ON(ret); | ||
| 1780 | } | ||
| 1781 | } | ||
| 1782 | |||
| 1783 | if (!(sb->s_flags & MS_RDONLY)) { | ||
| 1784 | ret = btrfs_cleanup_reloc_trees(tree_root); | ||
| 1785 | BUG_ON(ret); | ||
| 1786 | } | ||
| 1787 | |||
| 1788 | location.objectid = BTRFS_FS_TREE_OBJECTID; | ||
| 1789 | location.type = BTRFS_ROOT_ITEM_KEY; | ||
| 1790 | location.offset = (u64)-1; | ||
| 1791 | |||
| 1792 | fs_info->fs_root = btrfs_read_fs_root_no_name(fs_info, &location); | ||
| 1793 | if (!fs_info->fs_root) | ||
| 1794 | goto fail_trans_kthread; | ||
| 1795 | return tree_root; | ||
| 1796 | |||
| 1797 | fail_trans_kthread: | ||
| 1798 | kthread_stop(fs_info->transaction_kthread); | ||
| 1799 | fail_cleaner: | ||
| 1800 | kthread_stop(fs_info->cleaner_kthread); | ||
| 1801 | |||
| 1802 | /* | ||
| 1803 | * make sure we're done with the btree inode before we stop our | ||
| 1804 | * kthreads | ||
| 1805 | */ | ||
| 1806 | filemap_write_and_wait(fs_info->btree_inode->i_mapping); | ||
| 1807 | invalidate_inode_pages2(fs_info->btree_inode->i_mapping); | ||
| 1808 | |||
| 1809 | fail_csum_root: | ||
| 1810 | free_extent_buffer(csum_root->node); | ||
| 1811 | fail_extent_root: | ||
| 1812 | free_extent_buffer(extent_root->node); | ||
| 1813 | fail_tree_root: | ||
| 1814 | free_extent_buffer(tree_root->node); | ||
| 1815 | fail_chunk_root: | ||
| 1816 | free_extent_buffer(chunk_root->node); | ||
| 1817 | fail_sys_array: | ||
| 1818 | free_extent_buffer(dev_root->node); | ||
| 1819 | fail_sb_buffer: | ||
| 1820 | btrfs_stop_workers(&fs_info->fixup_workers); | ||
| 1821 | btrfs_stop_workers(&fs_info->delalloc_workers); | ||
| 1822 | btrfs_stop_workers(&fs_info->workers); | ||
| 1823 | btrfs_stop_workers(&fs_info->endio_workers); | ||
| 1824 | btrfs_stop_workers(&fs_info->endio_meta_workers); | ||
| 1825 | btrfs_stop_workers(&fs_info->endio_meta_write_workers); | ||
| 1826 | btrfs_stop_workers(&fs_info->endio_write_workers); | ||
| 1827 | btrfs_stop_workers(&fs_info->submit_workers); | ||
| 1828 | fail_iput: | ||
| 1829 | invalidate_inode_pages2(fs_info->btree_inode->i_mapping); | ||
| 1830 | iput(fs_info->btree_inode); | ||
| 1831 | fail: | ||
| 1832 | btrfs_close_devices(fs_info->fs_devices); | ||
| 1833 | btrfs_mapping_tree_free(&fs_info->mapping_tree); | ||
| 1834 | |||
| 1835 | kfree(extent_root); | ||
| 1836 | kfree(tree_root); | ||
| 1837 | bdi_destroy(&fs_info->bdi); | ||
| 1838 | kfree(fs_info); | ||
| 1839 | kfree(chunk_root); | ||
| 1840 | kfree(dev_root); | ||
| 1841 | kfree(csum_root); | ||
| 1842 | return ERR_PTR(err); | ||
| 1843 | } | ||
| 1844 | |||
| 1845 | static void btrfs_end_buffer_write_sync(struct buffer_head *bh, int uptodate) | ||
| 1846 | { | ||
| 1847 | char b[BDEVNAME_SIZE]; | ||
| 1848 | |||
| 1849 | if (uptodate) { | ||
| 1850 | set_buffer_uptodate(bh); | ||
| 1851 | } else { | ||
| 1852 | if (!buffer_eopnotsupp(bh) && printk_ratelimit()) { | ||
| 1853 | printk(KERN_WARNING "lost page write due to " | ||
| 1854 | "I/O error on %s\n", | ||
| 1855 | bdevname(bh->b_bdev, b)); | ||
| 1856 | } | ||
| 1857 | /* note, we dont' set_buffer_write_io_error because we have | ||
| 1858 | * our own ways of dealing with the IO errors | ||
| 1859 | */ | ||
| 1860 | clear_buffer_uptodate(bh); | ||
| 1861 | } | ||
| 1862 | unlock_buffer(bh); | ||
| 1863 | put_bh(bh); | ||
| 1864 | } | ||
| 1865 | |||
| 1866 | struct buffer_head *btrfs_read_dev_super(struct block_device *bdev) | ||
| 1867 | { | ||
| 1868 | struct buffer_head *bh; | ||
| 1869 | struct buffer_head *latest = NULL; | ||
| 1870 | struct btrfs_super_block *super; | ||
| 1871 | int i; | ||
| 1872 | u64 transid = 0; | ||
| 1873 | u64 bytenr; | ||
| 1874 | |||
| 1875 | /* we would like to check all the supers, but that would make | ||
| 1876 | * a btrfs mount succeed after a mkfs from a different FS. | ||
| 1877 | * So, we need to add a special mount option to scan for | ||
| 1878 | * later supers, using BTRFS_SUPER_MIRROR_MAX instead | ||
| 1879 | */ | ||
| 1880 | for (i = 0; i < 1; i++) { | ||
| 1881 | bytenr = btrfs_sb_offset(i); | ||
| 1882 | if (bytenr + 4096 >= i_size_read(bdev->bd_inode)) | ||
| 1883 | break; | ||
| 1884 | bh = __bread(bdev, bytenr / 4096, 4096); | ||
| 1885 | if (!bh) | ||
| 1886 | continue; | ||
| 1887 | |||
| 1888 | super = (struct btrfs_super_block *)bh->b_data; | ||
| 1889 | if (btrfs_super_bytenr(super) != bytenr || | ||
| 1890 | strncmp((char *)(&super->magic), BTRFS_MAGIC, | ||
| 1891 | sizeof(super->magic))) { | ||
| 1892 | brelse(bh); | ||
| 1893 | continue; | ||
| 1894 | } | ||
| 1895 | |||
| 1896 | if (!latest || btrfs_super_generation(super) > transid) { | ||
| 1897 | brelse(latest); | ||
| 1898 | latest = bh; | ||
| 1899 | transid = btrfs_super_generation(super); | ||
| 1900 | } else { | ||
| 1901 | brelse(bh); | ||
| 1902 | } | ||
| 1903 | } | ||
| 1904 | return latest; | ||
| 1905 | } | ||
| 1906 | |||
| 1907 | static int write_dev_supers(struct btrfs_device *device, | ||
| 1908 | struct btrfs_super_block *sb, | ||
| 1909 | int do_barriers, int wait, int max_mirrors) | ||
| 1910 | { | ||
| 1911 | struct buffer_head *bh; | ||
| 1912 | int i; | ||
| 1913 | int ret; | ||
| 1914 | int errors = 0; | ||
| 1915 | u32 crc; | ||
| 1916 | u64 bytenr; | ||
| 1917 | int last_barrier = 0; | ||
| 1918 | |||
| 1919 | if (max_mirrors == 0) | ||
| 1920 | max_mirrors = BTRFS_SUPER_MIRROR_MAX; | ||
| 1921 | |||
| 1922 | /* make sure only the last submit_bh does a barrier */ | ||
| 1923 | if (do_barriers) { | ||
| 1924 | for (i = 0; i < max_mirrors; i++) { | ||
| 1925 | bytenr = btrfs_sb_offset(i); | ||
| 1926 | if (bytenr + BTRFS_SUPER_INFO_SIZE >= | ||
| 1927 | device->total_bytes) | ||
| 1928 | break; | ||
| 1929 | last_barrier = i; | ||
| 1930 | } | ||
| 1931 | } | ||
| 1932 | |||
| 1933 | for (i = 0; i < max_mirrors; i++) { | ||
| 1934 | bytenr = btrfs_sb_offset(i); | ||
| 1935 | if (bytenr + BTRFS_SUPER_INFO_SIZE >= device->total_bytes) | ||
| 1936 | break; | ||
| 1937 | |||
| 1938 | if (wait) { | ||
| 1939 | bh = __find_get_block(device->bdev, bytenr / 4096, | ||
| 1940 | BTRFS_SUPER_INFO_SIZE); | ||
| 1941 | BUG_ON(!bh); | ||
| 1942 | brelse(bh); | ||
| 1943 | wait_on_buffer(bh); | ||
| 1944 | if (buffer_uptodate(bh)) { | ||
| 1945 | brelse(bh); | ||
| 1946 | continue; | ||
| 1947 | } | ||
| 1948 | } else { | ||
| 1949 | btrfs_set_super_bytenr(sb, bytenr); | ||
| 1950 | |||
| 1951 | crc = ~(u32)0; | ||
| 1952 | crc = btrfs_csum_data(NULL, (char *)sb + | ||
| 1953 | BTRFS_CSUM_SIZE, crc, | ||
| 1954 | BTRFS_SUPER_INFO_SIZE - | ||
| 1955 | BTRFS_CSUM_SIZE); | ||
| 1956 | btrfs_csum_final(crc, sb->csum); | ||
| 1957 | |||
| 1958 | bh = __getblk(device->bdev, bytenr / 4096, | ||
| 1959 | BTRFS_SUPER_INFO_SIZE); | ||
| 1960 | memcpy(bh->b_data, sb, BTRFS_SUPER_INFO_SIZE); | ||
| 1961 | |||
| 1962 | set_buffer_uptodate(bh); | ||
| 1963 | get_bh(bh); | ||
| 1964 | lock_buffer(bh); | ||
| 1965 | bh->b_end_io = btrfs_end_buffer_write_sync; | ||
| 1966 | } | ||
| 1967 | |||
| 1968 | if (i == last_barrier && do_barriers && device->barriers) { | ||
| 1969 | ret = submit_bh(WRITE_BARRIER, bh); | ||
| 1970 | if (ret == -EOPNOTSUPP) { | ||
| 1971 | printk("btrfs: disabling barriers on dev %s\n", | ||
| 1972 | device->name); | ||
| 1973 | set_buffer_uptodate(bh); | ||
| 1974 | device->barriers = 0; | ||
| 1975 | get_bh(bh); | ||
| 1976 | lock_buffer(bh); | ||
| 1977 | ret = submit_bh(WRITE, bh); | ||
| 1978 | } | ||
| 1979 | } else { | ||
| 1980 | ret = submit_bh(WRITE, bh); | ||
| 1981 | } | ||
| 1982 | |||
| 1983 | if (!ret && wait) { | ||
| 1984 | wait_on_buffer(bh); | ||
| 1985 | if (!buffer_uptodate(bh)) | ||
| 1986 | errors++; | ||
| 1987 | } else if (ret) { | ||
| 1988 | errors++; | ||
| 1989 | } | ||
| 1990 | if (wait) | ||
| 1991 | brelse(bh); | ||
| 1992 | } | ||
| 1993 | return errors < i ? 0 : -1; | ||
| 1994 | } | ||
| 1995 | |||
| 1996 | int write_all_supers(struct btrfs_root *root, int max_mirrors) | ||
| 1997 | { | ||
| 1998 | struct list_head *cur; | ||
| 1999 | struct list_head *head = &root->fs_info->fs_devices->devices; | ||
| 2000 | struct btrfs_device *dev; | ||
| 2001 | struct btrfs_super_block *sb; | ||
| 2002 | struct btrfs_dev_item *dev_item; | ||
| 2003 | int ret; | ||
| 2004 | int do_barriers; | ||
| 2005 | int max_errors; | ||
| 2006 | int total_errors = 0; | ||
| 2007 | u64 flags; | ||
| 2008 | |||
| 2009 | max_errors = btrfs_super_num_devices(&root->fs_info->super_copy) - 1; | ||
| 2010 | do_barriers = !btrfs_test_opt(root, NOBARRIER); | ||
| 2011 | |||
| 2012 | sb = &root->fs_info->super_for_commit; | ||
| 2013 | dev_item = &sb->dev_item; | ||
| 2014 | list_for_each(cur, head) { | ||
| 2015 | dev = list_entry(cur, struct btrfs_device, dev_list); | ||
| 2016 | if (!dev->bdev) { | ||
| 2017 | total_errors++; | ||
| 2018 | continue; | ||
| 2019 | } | ||
| 2020 | if (!dev->in_fs_metadata || !dev->writeable) | ||
| 2021 | continue; | ||
| 2022 | |||
| 2023 | btrfs_set_stack_device_generation(dev_item, 0); | ||
| 2024 | btrfs_set_stack_device_type(dev_item, dev->type); | ||
| 2025 | btrfs_set_stack_device_id(dev_item, dev->devid); | ||
| 2026 | btrfs_set_stack_device_total_bytes(dev_item, dev->total_bytes); | ||
| 2027 | btrfs_set_stack_device_bytes_used(dev_item, dev->bytes_used); | ||
| 2028 | btrfs_set_stack_device_io_align(dev_item, dev->io_align); | ||
| 2029 | btrfs_set_stack_device_io_width(dev_item, dev->io_width); | ||
| 2030 | btrfs_set_stack_device_sector_size(dev_item, dev->sector_size); | ||
| 2031 | memcpy(dev_item->uuid, dev->uuid, BTRFS_UUID_SIZE); | ||
| 2032 | memcpy(dev_item->fsid, dev->fs_devices->fsid, BTRFS_UUID_SIZE); | ||
| 2033 | |||
| 2034 | flags = btrfs_super_flags(sb); | ||
| 2035 | btrfs_set_super_flags(sb, flags | BTRFS_HEADER_FLAG_WRITTEN); | ||
| 2036 | |||
| 2037 | ret = write_dev_supers(dev, sb, do_barriers, 0, max_mirrors); | ||
| 2038 | if (ret) | ||
| 2039 | total_errors++; | ||
| 2040 | } | ||
| 2041 | if (total_errors > max_errors) { | ||
| 2042 | printk(KERN_ERR "btrfs: %d errors while writing supers\n", | ||
| 2043 | total_errors); | ||
| 2044 | BUG(); | ||
| 2045 | } | ||
| 2046 | |||
| 2047 | total_errors = 0; | ||
| 2048 | list_for_each(cur, head) { | ||
| 2049 | dev = list_entry(cur, struct btrfs_device, dev_list); | ||
| 2050 | if (!dev->bdev) | ||
| 2051 | continue; | ||
| 2052 | if (!dev->in_fs_metadata || !dev->writeable) | ||
| 2053 | continue; | ||
| 2054 | |||
| 2055 | ret = write_dev_supers(dev, sb, do_barriers, 1, max_mirrors); | ||
| 2056 | if (ret) | ||
| 2057 | total_errors++; | ||
| 2058 | } | ||
| 2059 | if (total_errors > max_errors) { | ||
| 2060 | printk(KERN_ERR "btrfs: %d errors while writing supers\n", | ||
| 2061 | total_errors); | ||
| 2062 | BUG(); | ||
| 2063 | } | ||
| 2064 | return 0; | ||
| 2065 | } | ||
| 2066 | |||
| 2067 | int write_ctree_super(struct btrfs_trans_handle *trans, | ||
| 2068 | struct btrfs_root *root, int max_mirrors) | ||
| 2069 | { | ||
| 2070 | int ret; | ||
| 2071 | |||
| 2072 | ret = write_all_supers(root, max_mirrors); | ||
| 2073 | return ret; | ||
| 2074 | } | ||
| 2075 | |||
| 2076 | int btrfs_free_fs_root(struct btrfs_fs_info *fs_info, struct btrfs_root *root) | ||
| 2077 | { | ||
| 2078 | radix_tree_delete(&fs_info->fs_roots_radix, | ||
| 2079 | (unsigned long)root->root_key.objectid); | ||
| 2080 | if (root->anon_super.s_dev) { | ||
| 2081 | down_write(&root->anon_super.s_umount); | ||
| 2082 | kill_anon_super(&root->anon_super); | ||
| 2083 | } | ||
| 2084 | if (root->node) | ||
| 2085 | free_extent_buffer(root->node); | ||
| 2086 | if (root->commit_root) | ||
| 2087 | free_extent_buffer(root->commit_root); | ||
| 2088 | kfree(root->name); | ||
| 2089 | kfree(root); | ||
| 2090 | return 0; | ||
| 2091 | } | ||
| 2092 | |||
| 2093 | static int del_fs_roots(struct btrfs_fs_info *fs_info) | ||
| 2094 | { | ||
| 2095 | int ret; | ||
| 2096 | struct btrfs_root *gang[8]; | ||
| 2097 | int i; | ||
| 2098 | |||
| 2099 | while (1) { | ||
| 2100 | ret = radix_tree_gang_lookup(&fs_info->fs_roots_radix, | ||
| 2101 | (void **)gang, 0, | ||
| 2102 | ARRAY_SIZE(gang)); | ||
| 2103 | if (!ret) | ||
| 2104 | break; | ||
| 2105 | for (i = 0; i < ret; i++) | ||
| 2106 | btrfs_free_fs_root(fs_info, gang[i]); | ||
| 2107 | } | ||
| 2108 | return 0; | ||
| 2109 | } | ||
| 2110 | |||
| 2111 | int btrfs_cleanup_fs_roots(struct btrfs_fs_info *fs_info) | ||
| 2112 | { | ||
| 2113 | u64 root_objectid = 0; | ||
| 2114 | struct btrfs_root *gang[8]; | ||
| 2115 | int i; | ||
| 2116 | int ret; | ||
| 2117 | |||
| 2118 | while (1) { | ||
| 2119 | ret = radix_tree_gang_lookup(&fs_info->fs_roots_radix, | ||
| 2120 | (void **)gang, root_objectid, | ||
| 2121 | ARRAY_SIZE(gang)); | ||
| 2122 | if (!ret) | ||
| 2123 | break; | ||
| 2124 | for (i = 0; i < ret; i++) { | ||
| 2125 | root_objectid = gang[i]->root_key.objectid; | ||
| 2126 | ret = btrfs_find_dead_roots(fs_info->tree_root, | ||
| 2127 | root_objectid, gang[i]); | ||
| 2128 | BUG_ON(ret); | ||
| 2129 | btrfs_orphan_cleanup(gang[i]); | ||
| 2130 | } | ||
| 2131 | root_objectid++; | ||
| 2132 | } | ||
| 2133 | return 0; | ||
| 2134 | } | ||
| 2135 | |||
| 2136 | int btrfs_commit_super(struct btrfs_root *root) | ||
| 2137 | { | ||
| 2138 | struct btrfs_trans_handle *trans; | ||
| 2139 | int ret; | ||
| 2140 | |||
| 2141 | mutex_lock(&root->fs_info->cleaner_mutex); | ||
| 2142 | btrfs_clean_old_snapshots(root); | ||
| 2143 | mutex_unlock(&root->fs_info->cleaner_mutex); | ||
| 2144 | trans = btrfs_start_transaction(root, 1); | ||
| 2145 | ret = btrfs_commit_transaction(trans, root); | ||
| 2146 | BUG_ON(ret); | ||
| 2147 | /* run commit again to drop the original snapshot */ | ||
| 2148 | trans = btrfs_start_transaction(root, 1); | ||
| 2149 | btrfs_commit_transaction(trans, root); | ||
| 2150 | ret = btrfs_write_and_wait_transaction(NULL, root); | ||
| 2151 | BUG_ON(ret); | ||
| 2152 | |||
| 2153 | ret = write_ctree_super(NULL, root, 0); | ||
| 2154 | return ret; | ||
| 2155 | } | ||
| 2156 | |||
| 2157 | int close_ctree(struct btrfs_root *root) | ||
| 2158 | { | ||
| 2159 | struct btrfs_fs_info *fs_info = root->fs_info; | ||
| 2160 | int ret; | ||
| 2161 | |||
| 2162 | fs_info->closing = 1; | ||
| 2163 | smp_mb(); | ||
| 2164 | |||
| 2165 | kthread_stop(root->fs_info->transaction_kthread); | ||
| 2166 | kthread_stop(root->fs_info->cleaner_kthread); | ||
| 2167 | |||
| 2168 | if (!(fs_info->sb->s_flags & MS_RDONLY)) { | ||
| 2169 | ret = btrfs_commit_super(root); | ||
| 2170 | if (ret) | ||
| 2171 | printk(KERN_ERR "btrfs: commit super ret %d\n", ret); | ||
| 2172 | } | ||
| 2173 | |||
| 2174 | if (fs_info->delalloc_bytes) { | ||
| 2175 | printk(KERN_INFO "btrfs: at unmount delalloc count %llu\n", | ||
| 2176 | fs_info->delalloc_bytes); | ||
| 2177 | } | ||
| 2178 | if (fs_info->total_ref_cache_size) { | ||
| 2179 | printk(KERN_INFO "btrfs: at umount reference cache size %llu\n", | ||
| 2180 | (unsigned long long)fs_info->total_ref_cache_size); | ||
| 2181 | } | ||
| 2182 | |||
| 2183 | if (fs_info->extent_root->node) | ||
| 2184 | free_extent_buffer(fs_info->extent_root->node); | ||
| 2185 | |||
| 2186 | if (fs_info->tree_root->node) | ||
| 2187 | free_extent_buffer(fs_info->tree_root->node); | ||
| 2188 | |||
| 2189 | if (root->fs_info->chunk_root->node) | ||
| 2190 | free_extent_buffer(root->fs_info->chunk_root->node); | ||
| 2191 | |||
| 2192 | if (root->fs_info->dev_root->node) | ||
| 2193 | free_extent_buffer(root->fs_info->dev_root->node); | ||
| 2194 | |||
| 2195 | if (root->fs_info->csum_root->node) | ||
| 2196 | free_extent_buffer(root->fs_info->csum_root->node); | ||
| 2197 | |||
| 2198 | btrfs_free_block_groups(root->fs_info); | ||
| 2199 | |||
| 2200 | del_fs_roots(fs_info); | ||
| 2201 | |||
| 2202 | iput(fs_info->btree_inode); | ||
| 2203 | |||
| 2204 | btrfs_stop_workers(&fs_info->fixup_workers); | ||
| 2205 | btrfs_stop_workers(&fs_info->delalloc_workers); | ||
| 2206 | btrfs_stop_workers(&fs_info->workers); | ||
| 2207 | btrfs_stop_workers(&fs_info->endio_workers); | ||
| 2208 | btrfs_stop_workers(&fs_info->endio_meta_workers); | ||
| 2209 | btrfs_stop_workers(&fs_info->endio_meta_write_workers); | ||
| 2210 | btrfs_stop_workers(&fs_info->endio_write_workers); | ||
| 2211 | btrfs_stop_workers(&fs_info->submit_workers); | ||
| 2212 | |||
| 2213 | #if 0 | ||
| 2214 | while (!list_empty(&fs_info->hashers)) { | ||
| 2215 | struct btrfs_hasher *hasher; | ||
| 2216 | hasher = list_entry(fs_info->hashers.next, struct btrfs_hasher, | ||
| 2217 | hashers); | ||
| 2218 | list_del(&hasher->hashers); | ||
| 2219 | crypto_free_hash(&fs_info->hash_tfm); | ||
| 2220 | kfree(hasher); | ||
| 2221 | } | ||
| 2222 | #endif | ||
| 2223 | btrfs_close_devices(fs_info->fs_devices); | ||
| 2224 | btrfs_mapping_tree_free(&fs_info->mapping_tree); | ||
| 2225 | |||
| 2226 | bdi_destroy(&fs_info->bdi); | ||
| 2227 | |||
| 2228 | kfree(fs_info->extent_root); | ||
| 2229 | kfree(fs_info->tree_root); | ||
| 2230 | kfree(fs_info->chunk_root); | ||
| 2231 | kfree(fs_info->dev_root); | ||
| 2232 | kfree(fs_info->csum_root); | ||
| 2233 | return 0; | ||
| 2234 | } | ||
| 2235 | |||
| 2236 | int btrfs_buffer_uptodate(struct extent_buffer *buf, u64 parent_transid) | ||
| 2237 | { | ||
| 2238 | int ret; | ||
| 2239 | struct inode *btree_inode = buf->first_page->mapping->host; | ||
| 2240 | |||
| 2241 | ret = extent_buffer_uptodate(&BTRFS_I(btree_inode)->io_tree, buf); | ||
| 2242 | if (!ret) | ||
| 2243 | return ret; | ||
| 2244 | |||
| 2245 | ret = verify_parent_transid(&BTRFS_I(btree_inode)->io_tree, buf, | ||
| 2246 | parent_transid); | ||
| 2247 | return !ret; | ||
| 2248 | } | ||
| 2249 | |||
| 2250 | int btrfs_set_buffer_uptodate(struct extent_buffer *buf) | ||
| 2251 | { | ||
| 2252 | struct inode *btree_inode = buf->first_page->mapping->host; | ||
| 2253 | return set_extent_buffer_uptodate(&BTRFS_I(btree_inode)->io_tree, | ||
| 2254 | buf); | ||
| 2255 | } | ||
| 2256 | |||
| 2257 | void btrfs_mark_buffer_dirty(struct extent_buffer *buf) | ||
| 2258 | { | ||
| 2259 | struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root; | ||
| 2260 | u64 transid = btrfs_header_generation(buf); | ||
| 2261 | struct inode *btree_inode = root->fs_info->btree_inode; | ||
| 2262 | |||
| 2263 | WARN_ON(!btrfs_tree_locked(buf)); | ||
| 2264 | if (transid != root->fs_info->generation) { | ||
| 2265 | printk(KERN_CRIT "btrfs transid mismatch buffer %llu, " | ||
| 2266 | "found %llu running %llu\n", | ||
| 2267 | (unsigned long long)buf->start, | ||
| 2268 | (unsigned long long)transid, | ||
| 2269 | (unsigned long long)root->fs_info->generation); | ||
| 2270 | WARN_ON(1); | ||
| 2271 | } | ||
| 2272 | set_extent_buffer_dirty(&BTRFS_I(btree_inode)->io_tree, buf); | ||
| 2273 | } | ||
| 2274 | |||
| 2275 | void btrfs_btree_balance_dirty(struct btrfs_root *root, unsigned long nr) | ||
| 2276 | { | ||
| 2277 | /* | ||
| 2278 | * looks as though older kernels can get into trouble with | ||
| 2279 | * this code, they end up stuck in balance_dirty_pages forever | ||
| 2280 | */ | ||
| 2281 | struct extent_io_tree *tree; | ||
| 2282 | u64 num_dirty; | ||
| 2283 | u64 start = 0; | ||
| 2284 | unsigned long thresh = 32 * 1024 * 1024; | ||
| 2285 | tree = &BTRFS_I(root->fs_info->btree_inode)->io_tree; | ||
| 2286 | |||
| 2287 | if (current_is_pdflush() || current->flags & PF_MEMALLOC) | ||
| 2288 | return; | ||
| 2289 | |||
| 2290 | num_dirty = count_range_bits(tree, &start, (u64)-1, | ||
| 2291 | thresh, EXTENT_DIRTY); | ||
| 2292 | if (num_dirty > thresh) { | ||
| 2293 | balance_dirty_pages_ratelimited_nr( | ||
| 2294 | root->fs_info->btree_inode->i_mapping, 1); | ||
| 2295 | } | ||
| 2296 | return; | ||
| 2297 | } | ||
| 2298 | |||
| 2299 | int btrfs_read_buffer(struct extent_buffer *buf, u64 parent_transid) | ||
| 2300 | { | ||
| 2301 | struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root; | ||
| 2302 | int ret; | ||
| 2303 | ret = btree_read_extent_buffer_pages(root, buf, 0, parent_transid); | ||
| 2304 | if (ret == 0) | ||
| 2305 | buf->flags |= EXTENT_UPTODATE; | ||
| 2306 | return ret; | ||
| 2307 | } | ||
| 2308 | |||
| 2309 | int btree_lock_page_hook(struct page *page) | ||
| 2310 | { | ||
| 2311 | struct inode *inode = page->mapping->host; | ||
| 2312 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 2313 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 2314 | struct extent_buffer *eb; | ||
| 2315 | unsigned long len; | ||
| 2316 | u64 bytenr = page_offset(page); | ||
| 2317 | |||
| 2318 | if (page->private == EXTENT_PAGE_PRIVATE) | ||
| 2319 | goto out; | ||
| 2320 | |||
| 2321 | len = page->private >> 2; | ||
| 2322 | eb = find_extent_buffer(io_tree, bytenr, len, GFP_NOFS); | ||
| 2323 | if (!eb) | ||
| 2324 | goto out; | ||
| 2325 | |||
| 2326 | btrfs_tree_lock(eb); | ||
| 2327 | spin_lock(&root->fs_info->hash_lock); | ||
| 2328 | btrfs_set_header_flag(eb, BTRFS_HEADER_FLAG_WRITTEN); | ||
| 2329 | spin_unlock(&root->fs_info->hash_lock); | ||
| 2330 | btrfs_tree_unlock(eb); | ||
| 2331 | free_extent_buffer(eb); | ||
| 2332 | out: | ||
| 2333 | lock_page(page); | ||
| 2334 | return 0; | ||
| 2335 | } | ||
| 2336 | |||
| 2337 | static struct extent_io_ops btree_extent_io_ops = { | ||
| 2338 | .write_cache_pages_lock_hook = btree_lock_page_hook, | ||
| 2339 | .readpage_end_io_hook = btree_readpage_end_io_hook, | ||
| 2340 | .submit_bio_hook = btree_submit_bio_hook, | ||
| 2341 | /* note we're sharing with inode.c for the merge bio hook */ | ||
| 2342 | .merge_bio_hook = btrfs_merge_bio_hook, | ||
| 2343 | }; | ||
diff --git a/fs/btrfs/disk-io.h b/fs/btrfs/disk-io.h new file mode 100644 index 000000000000..c0ff404c31b7 --- /dev/null +++ b/fs/btrfs/disk-io.h | |||
| @@ -0,0 +1,102 @@ | |||
| 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 | |||
| 19 | #ifndef __DISKIO__ | ||
| 20 | #define __DISKIO__ | ||
| 21 | |||
| 22 | #define BTRFS_SUPER_INFO_OFFSET (64 * 1024) | ||
| 23 | #define BTRFS_SUPER_INFO_SIZE 4096 | ||
| 24 | |||
| 25 | #define BTRFS_SUPER_MIRROR_MAX 3 | ||
| 26 | #define BTRFS_SUPER_MIRROR_SHIFT 12 | ||
| 27 | |||
| 28 | static inline u64 btrfs_sb_offset(int mirror) | ||
| 29 | { | ||
| 30 | u64 start = 16 * 1024; | ||
| 31 | if (mirror) | ||
| 32 | return start << (BTRFS_SUPER_MIRROR_SHIFT * mirror); | ||
| 33 | return BTRFS_SUPER_INFO_OFFSET; | ||
| 34 | } | ||
| 35 | |||
| 36 | struct btrfs_device; | ||
| 37 | struct btrfs_fs_devices; | ||
| 38 | |||
| 39 | struct extent_buffer *read_tree_block(struct btrfs_root *root, u64 bytenr, | ||
| 40 | u32 blocksize, u64 parent_transid); | ||
| 41 | int readahead_tree_block(struct btrfs_root *root, u64 bytenr, u32 blocksize, | ||
| 42 | u64 parent_transid); | ||
| 43 | struct extent_buffer *btrfs_find_create_tree_block(struct btrfs_root *root, | ||
| 44 | u64 bytenr, u32 blocksize); | ||
| 45 | int clean_tree_block(struct btrfs_trans_handle *trans, | ||
| 46 | struct btrfs_root *root, struct extent_buffer *buf); | ||
| 47 | struct btrfs_root *open_ctree(struct super_block *sb, | ||
| 48 | struct btrfs_fs_devices *fs_devices, | ||
| 49 | char *options); | ||
| 50 | int close_ctree(struct btrfs_root *root); | ||
| 51 | int write_ctree_super(struct btrfs_trans_handle *trans, | ||
| 52 | struct btrfs_root *root, int max_mirrors); | ||
| 53 | struct buffer_head *btrfs_read_dev_super(struct block_device *bdev); | ||
| 54 | int btrfs_commit_super(struct btrfs_root *root); | ||
| 55 | struct extent_buffer *btrfs_find_tree_block(struct btrfs_root *root, | ||
| 56 | u64 bytenr, u32 blocksize); | ||
| 57 | struct btrfs_root *btrfs_lookup_fs_root(struct btrfs_fs_info *fs_info, | ||
| 58 | u64 root_objectid); | ||
| 59 | struct btrfs_root *btrfs_read_fs_root(struct btrfs_fs_info *fs_info, | ||
| 60 | struct btrfs_key *location, | ||
| 61 | const char *name, int namelen); | ||
| 62 | struct btrfs_root *btrfs_read_fs_root_no_radix(struct btrfs_root *tree_root, | ||
| 63 | struct btrfs_key *location); | ||
| 64 | struct btrfs_root *btrfs_read_fs_root_no_name(struct btrfs_fs_info *fs_info, | ||
| 65 | struct btrfs_key *location); | ||
| 66 | int btrfs_cleanup_fs_roots(struct btrfs_fs_info *fs_info); | ||
| 67 | int btrfs_insert_dev_radix(struct btrfs_root *root, | ||
| 68 | struct block_device *bdev, | ||
| 69 | u64 device_id, | ||
| 70 | u64 block_start, | ||
| 71 | u64 num_blocks); | ||
| 72 | void btrfs_btree_balance_dirty(struct btrfs_root *root, unsigned long nr); | ||
| 73 | int btrfs_free_fs_root(struct btrfs_fs_info *fs_info, struct btrfs_root *root); | ||
| 74 | void btrfs_mark_buffer_dirty(struct extent_buffer *buf); | ||
| 75 | int btrfs_buffer_uptodate(struct extent_buffer *buf, u64 parent_transid); | ||
| 76 | int btrfs_set_buffer_uptodate(struct extent_buffer *buf); | ||
| 77 | int wait_on_tree_block_writeback(struct btrfs_root *root, | ||
| 78 | struct extent_buffer *buf); | ||
| 79 | int btrfs_read_buffer(struct extent_buffer *buf, u64 parent_transid); | ||
| 80 | u32 btrfs_csum_data(struct btrfs_root *root, char *data, u32 seed, size_t len); | ||
| 81 | void btrfs_csum_final(u32 crc, char *result); | ||
| 82 | int btrfs_open_device(struct btrfs_device *dev); | ||
| 83 | int btrfs_verify_block_csum(struct btrfs_root *root, | ||
| 84 | struct extent_buffer *buf); | ||
| 85 | int btrfs_bio_wq_end_io(struct btrfs_fs_info *info, struct bio *bio, | ||
| 86 | int metadata); | ||
| 87 | int btrfs_wq_submit_bio(struct btrfs_fs_info *fs_info, struct inode *inode, | ||
| 88 | int rw, struct bio *bio, int mirror_num, | ||
| 89 | unsigned long bio_flags, | ||
| 90 | extent_submit_bio_hook_t *submit_bio_start, | ||
| 91 | extent_submit_bio_hook_t *submit_bio_done); | ||
| 92 | |||
| 93 | int btrfs_congested_async(struct btrfs_fs_info *info, int iodone); | ||
| 94 | unsigned long btrfs_async_submit_limit(struct btrfs_fs_info *info); | ||
| 95 | int btrfs_write_tree_block(struct extent_buffer *buf); | ||
| 96 | int btrfs_wait_tree_block_writeback(struct extent_buffer *buf); | ||
| 97 | int btrfs_free_log_root_tree(struct btrfs_trans_handle *trans, | ||
| 98 | struct btrfs_fs_info *fs_info); | ||
| 99 | int btrfs_init_log_root_tree(struct btrfs_trans_handle *trans, | ||
| 100 | struct btrfs_fs_info *fs_info); | ||
| 101 | int btree_lock_page_hook(struct page *page); | ||
| 102 | #endif | ||
diff --git a/fs/btrfs/export.c b/fs/btrfs/export.c new file mode 100644 index 000000000000..85315d2c90de --- /dev/null +++ b/fs/btrfs/export.c | |||
| @@ -0,0 +1,203 @@ | |||
| 1 | #include <linux/fs.h> | ||
| 2 | #include <linux/types.h> | ||
| 3 | #include "ctree.h" | ||
| 4 | #include "disk-io.h" | ||
| 5 | #include "btrfs_inode.h" | ||
| 6 | #include "print-tree.h" | ||
| 7 | #include "export.h" | ||
| 8 | #include "compat.h" | ||
| 9 | |||
| 10 | #define BTRFS_FID_SIZE_NON_CONNECTABLE (offsetof(struct btrfs_fid, \ | ||
| 11 | parent_objectid) / 4) | ||
| 12 | #define BTRFS_FID_SIZE_CONNECTABLE (offsetof(struct btrfs_fid, \ | ||
| 13 | parent_root_objectid) / 4) | ||
| 14 | #define BTRFS_FID_SIZE_CONNECTABLE_ROOT (sizeof(struct btrfs_fid) / 4) | ||
| 15 | |||
| 16 | static int btrfs_encode_fh(struct dentry *dentry, u32 *fh, int *max_len, | ||
| 17 | int connectable) | ||
| 18 | { | ||
| 19 | struct btrfs_fid *fid = (struct btrfs_fid *)fh; | ||
| 20 | struct inode *inode = dentry->d_inode; | ||
| 21 | int len = *max_len; | ||
| 22 | int type; | ||
| 23 | |||
| 24 | if ((len < BTRFS_FID_SIZE_NON_CONNECTABLE) || | ||
| 25 | (connectable && len < BTRFS_FID_SIZE_CONNECTABLE)) | ||
| 26 | return 255; | ||
| 27 | |||
| 28 | len = BTRFS_FID_SIZE_NON_CONNECTABLE; | ||
| 29 | type = FILEID_BTRFS_WITHOUT_PARENT; | ||
| 30 | |||
| 31 | fid->objectid = BTRFS_I(inode)->location.objectid; | ||
| 32 | fid->root_objectid = BTRFS_I(inode)->root->objectid; | ||
| 33 | fid->gen = inode->i_generation; | ||
| 34 | |||
| 35 | if (connectable && !S_ISDIR(inode->i_mode)) { | ||
| 36 | struct inode *parent; | ||
| 37 | u64 parent_root_id; | ||
| 38 | |||
| 39 | spin_lock(&dentry->d_lock); | ||
| 40 | |||
| 41 | parent = dentry->d_parent->d_inode; | ||
| 42 | fid->parent_objectid = BTRFS_I(parent)->location.objectid; | ||
| 43 | fid->parent_gen = parent->i_generation; | ||
| 44 | parent_root_id = BTRFS_I(parent)->root->objectid; | ||
| 45 | |||
| 46 | spin_unlock(&dentry->d_lock); | ||
| 47 | |||
| 48 | if (parent_root_id != fid->root_objectid) { | ||
| 49 | fid->parent_root_objectid = parent_root_id; | ||
| 50 | len = BTRFS_FID_SIZE_CONNECTABLE_ROOT; | ||
| 51 | type = FILEID_BTRFS_WITH_PARENT_ROOT; | ||
| 52 | } else { | ||
| 53 | len = BTRFS_FID_SIZE_CONNECTABLE; | ||
| 54 | type = FILEID_BTRFS_WITH_PARENT; | ||
| 55 | } | ||
| 56 | } | ||
| 57 | |||
| 58 | *max_len = len; | ||
| 59 | return type; | ||
| 60 | } | ||
| 61 | |||
| 62 | static struct dentry *btrfs_get_dentry(struct super_block *sb, u64 objectid, | ||
| 63 | u64 root_objectid, u32 generation) | ||
| 64 | { | ||
| 65 | struct btrfs_root *root; | ||
| 66 | struct inode *inode; | ||
| 67 | struct btrfs_key key; | ||
| 68 | |||
| 69 | key.objectid = root_objectid; | ||
| 70 | btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY); | ||
| 71 | key.offset = (u64)-1; | ||
| 72 | |||
| 73 | root = btrfs_read_fs_root_no_name(btrfs_sb(sb)->fs_info, &key); | ||
| 74 | if (IS_ERR(root)) | ||
| 75 | return ERR_CAST(root); | ||
| 76 | |||
| 77 | key.objectid = objectid; | ||
| 78 | btrfs_set_key_type(&key, BTRFS_INODE_ITEM_KEY); | ||
| 79 | key.offset = 0; | ||
| 80 | |||
| 81 | inode = btrfs_iget(sb, &key, root, NULL); | ||
| 82 | if (IS_ERR(inode)) | ||
| 83 | return (void *)inode; | ||
| 84 | |||
| 85 | if (generation != inode->i_generation) { | ||
| 86 | iput(inode); | ||
| 87 | return ERR_PTR(-ESTALE); | ||
| 88 | } | ||
| 89 | |||
| 90 | return d_obtain_alias(inode); | ||
| 91 | } | ||
| 92 | |||
| 93 | static struct dentry *btrfs_fh_to_parent(struct super_block *sb, struct fid *fh, | ||
| 94 | int fh_len, int fh_type) | ||
| 95 | { | ||
| 96 | struct btrfs_fid *fid = (struct btrfs_fid *) fh; | ||
| 97 | u64 objectid, root_objectid; | ||
| 98 | u32 generation; | ||
| 99 | |||
| 100 | if (fh_type == FILEID_BTRFS_WITH_PARENT) { | ||
| 101 | if (fh_len != BTRFS_FID_SIZE_CONNECTABLE) | ||
| 102 | return NULL; | ||
| 103 | root_objectid = fid->root_objectid; | ||
| 104 | } else if (fh_type == FILEID_BTRFS_WITH_PARENT_ROOT) { | ||
| 105 | if (fh_len != BTRFS_FID_SIZE_CONNECTABLE_ROOT) | ||
| 106 | return NULL; | ||
| 107 | root_objectid = fid->parent_root_objectid; | ||
| 108 | } else | ||
| 109 | return NULL; | ||
| 110 | |||
| 111 | objectid = fid->parent_objectid; | ||
| 112 | generation = fid->parent_gen; | ||
| 113 | |||
| 114 | return btrfs_get_dentry(sb, objectid, root_objectid, generation); | ||
| 115 | } | ||
| 116 | |||
| 117 | static struct dentry *btrfs_fh_to_dentry(struct super_block *sb, struct fid *fh, | ||
| 118 | int fh_len, int fh_type) | ||
| 119 | { | ||
| 120 | struct btrfs_fid *fid = (struct btrfs_fid *) fh; | ||
| 121 | u64 objectid, root_objectid; | ||
| 122 | u32 generation; | ||
| 123 | |||
| 124 | if ((fh_type != FILEID_BTRFS_WITH_PARENT || | ||
| 125 | fh_len != BTRFS_FID_SIZE_CONNECTABLE) && | ||
| 126 | (fh_type != FILEID_BTRFS_WITH_PARENT_ROOT || | ||
| 127 | fh_len != BTRFS_FID_SIZE_CONNECTABLE_ROOT) && | ||
| 128 | (fh_type != FILEID_BTRFS_WITHOUT_PARENT || | ||
| 129 | fh_len != BTRFS_FID_SIZE_NON_CONNECTABLE)) | ||
| 130 | return NULL; | ||
| 131 | |||
| 132 | objectid = fid->objectid; | ||
| 133 | root_objectid = fid->root_objectid; | ||
| 134 | generation = fid->gen; | ||
| 135 | |||
| 136 | return btrfs_get_dentry(sb, objectid, root_objectid, generation); | ||
| 137 | } | ||
| 138 | |||
| 139 | static struct dentry *btrfs_get_parent(struct dentry *child) | ||
| 140 | { | ||
| 141 | struct inode *dir = child->d_inode; | ||
| 142 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 143 | struct btrfs_key key; | ||
| 144 | struct btrfs_path *path; | ||
| 145 | struct extent_buffer *leaf; | ||
| 146 | int slot; | ||
| 147 | u64 objectid; | ||
| 148 | int ret; | ||
| 149 | |||
| 150 | path = btrfs_alloc_path(); | ||
| 151 | |||
| 152 | key.objectid = dir->i_ino; | ||
| 153 | btrfs_set_key_type(&key, BTRFS_INODE_REF_KEY); | ||
| 154 | key.offset = (u64)-1; | ||
| 155 | |||
| 156 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 157 | if (ret < 0) { | ||
| 158 | /* Error */ | ||
| 159 | btrfs_free_path(path); | ||
| 160 | return ERR_PTR(ret); | ||
| 161 | } | ||
| 162 | leaf = path->nodes[0]; | ||
| 163 | slot = path->slots[0]; | ||
| 164 | if (ret) { | ||
| 165 | /* btrfs_search_slot() returns the slot where we'd want to | ||
| 166 | insert a backref for parent inode #0xFFFFFFFFFFFFFFFF. | ||
| 167 | The _real_ backref, telling us what the parent inode | ||
| 168 | _actually_ is, will be in the slot _before_ the one | ||
| 169 | that btrfs_search_slot() returns. */ | ||
| 170 | if (!slot) { | ||
| 171 | /* Unless there is _no_ key in the tree before... */ | ||
| 172 | btrfs_free_path(path); | ||
| 173 | return ERR_PTR(-EIO); | ||
| 174 | } | ||
| 175 | slot--; | ||
| 176 | } | ||
| 177 | |||
| 178 | btrfs_item_key_to_cpu(leaf, &key, slot); | ||
| 179 | btrfs_free_path(path); | ||
| 180 | |||
| 181 | if (key.objectid != dir->i_ino || key.type != BTRFS_INODE_REF_KEY) | ||
| 182 | return ERR_PTR(-EINVAL); | ||
| 183 | |||
| 184 | objectid = key.offset; | ||
| 185 | |||
| 186 | /* If we are already at the root of a subvol, return the real root */ | ||
| 187 | if (objectid == dir->i_ino) | ||
| 188 | return dget(dir->i_sb->s_root); | ||
| 189 | |||
| 190 | /* Build a new key for the inode item */ | ||
| 191 | key.objectid = objectid; | ||
| 192 | btrfs_set_key_type(&key, BTRFS_INODE_ITEM_KEY); | ||
| 193 | key.offset = 0; | ||
| 194 | |||
| 195 | return d_obtain_alias(btrfs_iget(root->fs_info->sb, &key, root, NULL)); | ||
| 196 | } | ||
| 197 | |||
| 198 | const struct export_operations btrfs_export_ops = { | ||
| 199 | .encode_fh = btrfs_encode_fh, | ||
| 200 | .fh_to_dentry = btrfs_fh_to_dentry, | ||
| 201 | .fh_to_parent = btrfs_fh_to_parent, | ||
| 202 | .get_parent = btrfs_get_parent, | ||
| 203 | }; | ||
diff --git a/fs/btrfs/export.h b/fs/btrfs/export.h new file mode 100644 index 000000000000..074348a95841 --- /dev/null +++ b/fs/btrfs/export.h | |||
| @@ -0,0 +1,19 @@ | |||
| 1 | #ifndef BTRFS_EXPORT_H | ||
| 2 | #define BTRFS_EXPORT_H | ||
| 3 | |||
| 4 | #include <linux/exportfs.h> | ||
| 5 | |||
| 6 | extern const struct export_operations btrfs_export_ops; | ||
| 7 | |||
| 8 | struct btrfs_fid { | ||
| 9 | u64 objectid; | ||
| 10 | u64 root_objectid; | ||
| 11 | u32 gen; | ||
| 12 | |||
| 13 | u64 parent_objectid; | ||
| 14 | u32 parent_gen; | ||
| 15 | |||
| 16 | u64 parent_root_objectid; | ||
| 17 | } __attribute__ ((packed)); | ||
| 18 | |||
| 19 | #endif | ||
diff --git a/fs/btrfs/extent-tree.c b/fs/btrfs/extent-tree.c new file mode 100644 index 000000000000..293da650873f --- /dev/null +++ b/fs/btrfs/extent-tree.c | |||
| @@ -0,0 +1,5986 @@ | |||
| 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/sched.h> | ||
| 19 | #include <linux/pagemap.h> | ||
| 20 | #include <linux/writeback.h> | ||
| 21 | #include <linux/blkdev.h> | ||
| 22 | #include <linux/version.h> | ||
| 23 | #include "compat.h" | ||
| 24 | #include "hash.h" | ||
| 25 | #include "crc32c.h" | ||
| 26 | #include "ctree.h" | ||
| 27 | #include "disk-io.h" | ||
| 28 | #include "print-tree.h" | ||
| 29 | #include "transaction.h" | ||
| 30 | #include "volumes.h" | ||
| 31 | #include "locking.h" | ||
| 32 | #include "ref-cache.h" | ||
| 33 | #include "compat.h" | ||
| 34 | |||
| 35 | #define PENDING_EXTENT_INSERT 0 | ||
| 36 | #define PENDING_EXTENT_DELETE 1 | ||
| 37 | #define PENDING_BACKREF_UPDATE 2 | ||
| 38 | |||
| 39 | struct pending_extent_op { | ||
| 40 | int type; | ||
| 41 | u64 bytenr; | ||
| 42 | u64 num_bytes; | ||
| 43 | u64 parent; | ||
| 44 | u64 orig_parent; | ||
| 45 | u64 generation; | ||
| 46 | u64 orig_generation; | ||
| 47 | int level; | ||
| 48 | struct list_head list; | ||
| 49 | int del; | ||
| 50 | }; | ||
| 51 | |||
| 52 | static int finish_current_insert(struct btrfs_trans_handle *trans, | ||
| 53 | struct btrfs_root *extent_root, int all); | ||
| 54 | static int del_pending_extents(struct btrfs_trans_handle *trans, | ||
| 55 | struct btrfs_root *extent_root, int all); | ||
| 56 | static int pin_down_bytes(struct btrfs_trans_handle *trans, | ||
| 57 | struct btrfs_root *root, | ||
| 58 | u64 bytenr, u64 num_bytes, int is_data); | ||
| 59 | static int update_block_group(struct btrfs_trans_handle *trans, | ||
| 60 | struct btrfs_root *root, | ||
| 61 | u64 bytenr, u64 num_bytes, int alloc, | ||
| 62 | int mark_free); | ||
| 63 | |||
| 64 | static int block_group_bits(struct btrfs_block_group_cache *cache, u64 bits) | ||
| 65 | { | ||
| 66 | return (cache->flags & bits) == bits; | ||
| 67 | } | ||
| 68 | |||
| 69 | /* | ||
| 70 | * this adds the block group to the fs_info rb tree for the block group | ||
| 71 | * cache | ||
| 72 | */ | ||
| 73 | static int btrfs_add_block_group_cache(struct btrfs_fs_info *info, | ||
| 74 | struct btrfs_block_group_cache *block_group) | ||
| 75 | { | ||
| 76 | struct rb_node **p; | ||
| 77 | struct rb_node *parent = NULL; | ||
| 78 | struct btrfs_block_group_cache *cache; | ||
| 79 | |||
| 80 | spin_lock(&info->block_group_cache_lock); | ||
| 81 | p = &info->block_group_cache_tree.rb_node; | ||
| 82 | |||
| 83 | while (*p) { | ||
| 84 | parent = *p; | ||
| 85 | cache = rb_entry(parent, struct btrfs_block_group_cache, | ||
| 86 | cache_node); | ||
| 87 | if (block_group->key.objectid < cache->key.objectid) { | ||
| 88 | p = &(*p)->rb_left; | ||
| 89 | } else if (block_group->key.objectid > cache->key.objectid) { | ||
| 90 | p = &(*p)->rb_right; | ||
| 91 | } else { | ||
| 92 | spin_unlock(&info->block_group_cache_lock); | ||
| 93 | return -EEXIST; | ||
| 94 | } | ||
| 95 | } | ||
| 96 | |||
| 97 | rb_link_node(&block_group->cache_node, parent, p); | ||
| 98 | rb_insert_color(&block_group->cache_node, | ||
| 99 | &info->block_group_cache_tree); | ||
| 100 | spin_unlock(&info->block_group_cache_lock); | ||
| 101 | |||
| 102 | return 0; | ||
| 103 | } | ||
| 104 | |||
| 105 | /* | ||
| 106 | * This will return the block group at or after bytenr if contains is 0, else | ||
| 107 | * it will return the block group that contains the bytenr | ||
| 108 | */ | ||
| 109 | static struct btrfs_block_group_cache * | ||
| 110 | block_group_cache_tree_search(struct btrfs_fs_info *info, u64 bytenr, | ||
| 111 | int contains) | ||
| 112 | { | ||
| 113 | struct btrfs_block_group_cache *cache, *ret = NULL; | ||
| 114 | struct rb_node *n; | ||
| 115 | u64 end, start; | ||
| 116 | |||
| 117 | spin_lock(&info->block_group_cache_lock); | ||
| 118 | n = info->block_group_cache_tree.rb_node; | ||
| 119 | |||
| 120 | while (n) { | ||
| 121 | cache = rb_entry(n, struct btrfs_block_group_cache, | ||
| 122 | cache_node); | ||
| 123 | end = cache->key.objectid + cache->key.offset - 1; | ||
| 124 | start = cache->key.objectid; | ||
| 125 | |||
| 126 | if (bytenr < start) { | ||
| 127 | if (!contains && (!ret || start < ret->key.objectid)) | ||
| 128 | ret = cache; | ||
| 129 | n = n->rb_left; | ||
| 130 | } else if (bytenr > start) { | ||
| 131 | if (contains && bytenr <= end) { | ||
| 132 | ret = cache; | ||
| 133 | break; | ||
| 134 | } | ||
| 135 | n = n->rb_right; | ||
| 136 | } else { | ||
| 137 | ret = cache; | ||
| 138 | break; | ||
| 139 | } | ||
| 140 | } | ||
| 141 | if (ret) | ||
| 142 | atomic_inc(&ret->count); | ||
| 143 | spin_unlock(&info->block_group_cache_lock); | ||
| 144 | |||
| 145 | return ret; | ||
| 146 | } | ||
| 147 | |||
| 148 | /* | ||
| 149 | * this is only called by cache_block_group, since we could have freed extents | ||
| 150 | * we need to check the pinned_extents for any extents that can't be used yet | ||
| 151 | * since their free space will be released as soon as the transaction commits. | ||
| 152 | */ | ||
| 153 | static int add_new_free_space(struct btrfs_block_group_cache *block_group, | ||
| 154 | struct btrfs_fs_info *info, u64 start, u64 end) | ||
| 155 | { | ||
| 156 | u64 extent_start, extent_end, size; | ||
| 157 | int ret; | ||
| 158 | |||
| 159 | mutex_lock(&info->pinned_mutex); | ||
| 160 | while (start < end) { | ||
| 161 | ret = find_first_extent_bit(&info->pinned_extents, start, | ||
| 162 | &extent_start, &extent_end, | ||
| 163 | EXTENT_DIRTY); | ||
| 164 | if (ret) | ||
| 165 | break; | ||
| 166 | |||
| 167 | if (extent_start == start) { | ||
| 168 | start = extent_end + 1; | ||
| 169 | } else if (extent_start > start && extent_start < end) { | ||
| 170 | size = extent_start - start; | ||
| 171 | ret = btrfs_add_free_space(block_group, start, | ||
| 172 | size); | ||
| 173 | BUG_ON(ret); | ||
| 174 | start = extent_end + 1; | ||
| 175 | } else { | ||
| 176 | break; | ||
| 177 | } | ||
| 178 | } | ||
| 179 | |||
| 180 | if (start < end) { | ||
| 181 | size = end - start; | ||
| 182 | ret = btrfs_add_free_space(block_group, start, size); | ||
| 183 | BUG_ON(ret); | ||
| 184 | } | ||
| 185 | mutex_unlock(&info->pinned_mutex); | ||
| 186 | |||
| 187 | return 0; | ||
| 188 | } | ||
| 189 | |||
| 190 | static int remove_sb_from_cache(struct btrfs_root *root, | ||
| 191 | struct btrfs_block_group_cache *cache) | ||
| 192 | { | ||
| 193 | u64 bytenr; | ||
| 194 | u64 *logical; | ||
| 195 | int stripe_len; | ||
| 196 | int i, nr, ret; | ||
| 197 | |||
| 198 | for (i = 0; i < BTRFS_SUPER_MIRROR_MAX; i++) { | ||
| 199 | bytenr = btrfs_sb_offset(i); | ||
| 200 | ret = btrfs_rmap_block(&root->fs_info->mapping_tree, | ||
| 201 | cache->key.objectid, bytenr, 0, | ||
| 202 | &logical, &nr, &stripe_len); | ||
| 203 | BUG_ON(ret); | ||
| 204 | while (nr--) { | ||
| 205 | btrfs_remove_free_space(cache, logical[nr], | ||
| 206 | stripe_len); | ||
| 207 | } | ||
| 208 | kfree(logical); | ||
| 209 | } | ||
| 210 | return 0; | ||
| 211 | } | ||
| 212 | |||
| 213 | static int cache_block_group(struct btrfs_root *root, | ||
| 214 | struct btrfs_block_group_cache *block_group) | ||
| 215 | { | ||
| 216 | struct btrfs_path *path; | ||
| 217 | int ret = 0; | ||
| 218 | struct btrfs_key key; | ||
| 219 | struct extent_buffer *leaf; | ||
| 220 | int slot; | ||
| 221 | u64 last; | ||
| 222 | |||
| 223 | if (!block_group) | ||
| 224 | return 0; | ||
| 225 | |||
| 226 | root = root->fs_info->extent_root; | ||
| 227 | |||
| 228 | if (block_group->cached) | ||
| 229 | return 0; | ||
| 230 | |||
| 231 | path = btrfs_alloc_path(); | ||
| 232 | if (!path) | ||
| 233 | return -ENOMEM; | ||
| 234 | |||
| 235 | path->reada = 2; | ||
| 236 | /* | ||
| 237 | * we get into deadlocks with paths held by callers of this function. | ||
| 238 | * since the alloc_mutex is protecting things right now, just | ||
| 239 | * skip the locking here | ||
| 240 | */ | ||
| 241 | path->skip_locking = 1; | ||
| 242 | last = max_t(u64, block_group->key.objectid, BTRFS_SUPER_INFO_OFFSET); | ||
| 243 | key.objectid = last; | ||
| 244 | key.offset = 0; | ||
| 245 | btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY); | ||
| 246 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 247 | if (ret < 0) | ||
| 248 | goto err; | ||
| 249 | |||
| 250 | while (1) { | ||
| 251 | leaf = path->nodes[0]; | ||
| 252 | slot = path->slots[0]; | ||
| 253 | if (slot >= btrfs_header_nritems(leaf)) { | ||
| 254 | ret = btrfs_next_leaf(root, path); | ||
| 255 | if (ret < 0) | ||
| 256 | goto err; | ||
| 257 | if (ret == 0) | ||
| 258 | continue; | ||
| 259 | else | ||
| 260 | break; | ||
| 261 | } | ||
| 262 | btrfs_item_key_to_cpu(leaf, &key, slot); | ||
| 263 | if (key.objectid < block_group->key.objectid) | ||
| 264 | goto next; | ||
| 265 | |||
| 266 | if (key.objectid >= block_group->key.objectid + | ||
| 267 | block_group->key.offset) | ||
| 268 | break; | ||
| 269 | |||
| 270 | if (btrfs_key_type(&key) == BTRFS_EXTENT_ITEM_KEY) { | ||
| 271 | add_new_free_space(block_group, root->fs_info, last, | ||
| 272 | key.objectid); | ||
| 273 | |||
| 274 | last = key.objectid + key.offset; | ||
| 275 | } | ||
| 276 | next: | ||
| 277 | path->slots[0]++; | ||
| 278 | } | ||
| 279 | |||
| 280 | add_new_free_space(block_group, root->fs_info, last, | ||
| 281 | block_group->key.objectid + | ||
| 282 | block_group->key.offset); | ||
| 283 | |||
| 284 | remove_sb_from_cache(root, block_group); | ||
| 285 | block_group->cached = 1; | ||
| 286 | ret = 0; | ||
| 287 | err: | ||
| 288 | btrfs_free_path(path); | ||
| 289 | return ret; | ||
| 290 | } | ||
| 291 | |||
| 292 | /* | ||
| 293 | * return the block group that starts at or after bytenr | ||
| 294 | */ | ||
| 295 | static struct btrfs_block_group_cache * | ||
| 296 | btrfs_lookup_first_block_group(struct btrfs_fs_info *info, u64 bytenr) | ||
| 297 | { | ||
| 298 | struct btrfs_block_group_cache *cache; | ||
| 299 | |||
| 300 | cache = block_group_cache_tree_search(info, bytenr, 0); | ||
| 301 | |||
| 302 | return cache; | ||
| 303 | } | ||
| 304 | |||
| 305 | /* | ||
| 306 | * return the block group that contains teh given bytenr | ||
| 307 | */ | ||
| 308 | struct btrfs_block_group_cache *btrfs_lookup_block_group( | ||
| 309 | struct btrfs_fs_info *info, | ||
| 310 | u64 bytenr) | ||
| 311 | { | ||
| 312 | struct btrfs_block_group_cache *cache; | ||
| 313 | |||
| 314 | cache = block_group_cache_tree_search(info, bytenr, 1); | ||
| 315 | |||
| 316 | return cache; | ||
| 317 | } | ||
| 318 | |||
| 319 | static inline void put_block_group(struct btrfs_block_group_cache *cache) | ||
| 320 | { | ||
| 321 | if (atomic_dec_and_test(&cache->count)) | ||
| 322 | kfree(cache); | ||
| 323 | } | ||
| 324 | |||
| 325 | static struct btrfs_space_info *__find_space_info(struct btrfs_fs_info *info, | ||
| 326 | u64 flags) | ||
| 327 | { | ||
| 328 | struct list_head *head = &info->space_info; | ||
| 329 | struct list_head *cur; | ||
| 330 | struct btrfs_space_info *found; | ||
| 331 | list_for_each(cur, head) { | ||
| 332 | found = list_entry(cur, struct btrfs_space_info, list); | ||
| 333 | if (found->flags == flags) | ||
| 334 | return found; | ||
| 335 | } | ||
| 336 | return NULL; | ||
| 337 | } | ||
| 338 | |||
| 339 | static u64 div_factor(u64 num, int factor) | ||
| 340 | { | ||
| 341 | if (factor == 10) | ||
| 342 | return num; | ||
| 343 | num *= factor; | ||
| 344 | do_div(num, 10); | ||
| 345 | return num; | ||
| 346 | } | ||
| 347 | |||
| 348 | u64 btrfs_find_block_group(struct btrfs_root *root, | ||
| 349 | u64 search_start, u64 search_hint, int owner) | ||
| 350 | { | ||
| 351 | struct btrfs_block_group_cache *cache; | ||
| 352 | u64 used; | ||
| 353 | u64 last = max(search_hint, search_start); | ||
| 354 | u64 group_start = 0; | ||
| 355 | int full_search = 0; | ||
| 356 | int factor = 9; | ||
| 357 | int wrapped = 0; | ||
| 358 | again: | ||
| 359 | while (1) { | ||
| 360 | cache = btrfs_lookup_first_block_group(root->fs_info, last); | ||
| 361 | if (!cache) | ||
| 362 | break; | ||
| 363 | |||
| 364 | spin_lock(&cache->lock); | ||
| 365 | last = cache->key.objectid + cache->key.offset; | ||
| 366 | used = btrfs_block_group_used(&cache->item); | ||
| 367 | |||
| 368 | if ((full_search || !cache->ro) && | ||
| 369 | block_group_bits(cache, BTRFS_BLOCK_GROUP_METADATA)) { | ||
| 370 | if (used + cache->pinned + cache->reserved < | ||
| 371 | div_factor(cache->key.offset, factor)) { | ||
| 372 | group_start = cache->key.objectid; | ||
| 373 | spin_unlock(&cache->lock); | ||
| 374 | put_block_group(cache); | ||
| 375 | goto found; | ||
| 376 | } | ||
| 377 | } | ||
| 378 | spin_unlock(&cache->lock); | ||
| 379 | put_block_group(cache); | ||
| 380 | cond_resched(); | ||
| 381 | } | ||
| 382 | if (!wrapped) { | ||
| 383 | last = search_start; | ||
| 384 | wrapped = 1; | ||
| 385 | goto again; | ||
| 386 | } | ||
| 387 | if (!full_search && factor < 10) { | ||
| 388 | last = search_start; | ||
| 389 | full_search = 1; | ||
| 390 | factor = 10; | ||
| 391 | goto again; | ||
| 392 | } | ||
| 393 | found: | ||
| 394 | return group_start; | ||
| 395 | } | ||
| 396 | |||
| 397 | /* simple helper to search for an existing extent at a given offset */ | ||
| 398 | int btrfs_lookup_extent(struct btrfs_root *root, u64 start, u64 len) | ||
| 399 | { | ||
| 400 | int ret; | ||
| 401 | struct btrfs_key key; | ||
| 402 | struct btrfs_path *path; | ||
| 403 | |||
| 404 | path = btrfs_alloc_path(); | ||
| 405 | BUG_ON(!path); | ||
| 406 | key.objectid = start; | ||
| 407 | key.offset = len; | ||
| 408 | btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY); | ||
| 409 | ret = btrfs_search_slot(NULL, root->fs_info->extent_root, &key, path, | ||
| 410 | 0, 0); | ||
| 411 | btrfs_free_path(path); | ||
| 412 | return ret; | ||
| 413 | } | ||
| 414 | |||
| 415 | /* | ||
| 416 | * Back reference rules. Back refs have three main goals: | ||
| 417 | * | ||
| 418 | * 1) differentiate between all holders of references to an extent so that | ||
| 419 | * when a reference is dropped we can make sure it was a valid reference | ||
| 420 | * before freeing the extent. | ||
| 421 | * | ||
| 422 | * 2) Provide enough information to quickly find the holders of an extent | ||
| 423 | * if we notice a given block is corrupted or bad. | ||
| 424 | * | ||
| 425 | * 3) Make it easy to migrate blocks for FS shrinking or storage pool | ||
| 426 | * maintenance. This is actually the same as #2, but with a slightly | ||
| 427 | * different use case. | ||
| 428 | * | ||
| 429 | * File extents can be referenced by: | ||
| 430 | * | ||
| 431 | * - multiple snapshots, subvolumes, or different generations in one subvol | ||
| 432 | * - different files inside a single subvolume | ||
| 433 | * - different offsets inside a file (bookend extents in file.c) | ||
| 434 | * | ||
| 435 | * The extent ref structure has fields for: | ||
| 436 | * | ||
| 437 | * - Objectid of the subvolume root | ||
| 438 | * - Generation number of the tree holding the reference | ||
| 439 | * - objectid of the file holding the reference | ||
| 440 | * - number of references holding by parent node (alway 1 for tree blocks) | ||
| 441 | * | ||
| 442 | * Btree leaf may hold multiple references to a file extent. In most cases, | ||
| 443 | * these references are from same file and the corresponding offsets inside | ||
| 444 | * the file are close together. | ||
| 445 | * | ||
| 446 | * When a file extent is allocated the fields are filled in: | ||
| 447 | * (root_key.objectid, trans->transid, inode objectid, 1) | ||
| 448 | * | ||
| 449 | * When a leaf is cow'd new references are added for every file extent found | ||
| 450 | * in the leaf. It looks similar to the create case, but trans->transid will | ||
| 451 | * be different when the block is cow'd. | ||
| 452 | * | ||
| 453 | * (root_key.objectid, trans->transid, inode objectid, | ||
| 454 | * number of references in the leaf) | ||
| 455 | * | ||
| 456 | * When a file extent is removed either during snapshot deletion or | ||
| 457 | * file truncation, we find the corresponding back reference and check | ||
| 458 | * the following fields: | ||
| 459 | * | ||
| 460 | * (btrfs_header_owner(leaf), btrfs_header_generation(leaf), | ||
| 461 | * inode objectid) | ||
| 462 | * | ||
| 463 | * Btree extents can be referenced by: | ||
| 464 | * | ||
| 465 | * - Different subvolumes | ||
| 466 | * - Different generations of the same subvolume | ||
| 467 | * | ||
| 468 | * When a tree block is created, back references are inserted: | ||
| 469 | * | ||
| 470 | * (root->root_key.objectid, trans->transid, level, 1) | ||
| 471 | * | ||
| 472 | * When a tree block is cow'd, new back references are added for all the | ||
| 473 | * blocks it points to. If the tree block isn't in reference counted root, | ||
| 474 | * the old back references are removed. These new back references are of | ||
| 475 | * the form (trans->transid will have increased since creation): | ||
| 476 | * | ||
| 477 | * (root->root_key.objectid, trans->transid, level, 1) | ||
| 478 | * | ||
| 479 | * When a backref is in deleting, the following fields are checked: | ||
| 480 | * | ||
| 481 | * if backref was for a tree root: | ||
| 482 | * (btrfs_header_owner(itself), btrfs_header_generation(itself), level) | ||
| 483 | * else | ||
| 484 | * (btrfs_header_owner(parent), btrfs_header_generation(parent), level) | ||
| 485 | * | ||
| 486 | * Back Reference Key composing: | ||
| 487 | * | ||
| 488 | * The key objectid corresponds to the first byte in the extent, the key | ||
| 489 | * type is set to BTRFS_EXTENT_REF_KEY, and the key offset is the first | ||
| 490 | * byte of parent extent. If a extent is tree root, the key offset is set | ||
| 491 | * to the key objectid. | ||
| 492 | */ | ||
| 493 | |||
| 494 | static noinline int lookup_extent_backref(struct btrfs_trans_handle *trans, | ||
| 495 | struct btrfs_root *root, | ||
| 496 | struct btrfs_path *path, | ||
| 497 | u64 bytenr, u64 parent, | ||
| 498 | u64 ref_root, u64 ref_generation, | ||
| 499 | u64 owner_objectid, int del) | ||
| 500 | { | ||
| 501 | struct btrfs_key key; | ||
| 502 | struct btrfs_extent_ref *ref; | ||
| 503 | struct extent_buffer *leaf; | ||
| 504 | u64 ref_objectid; | ||
| 505 | int ret; | ||
| 506 | |||
| 507 | key.objectid = bytenr; | ||
| 508 | key.type = BTRFS_EXTENT_REF_KEY; | ||
| 509 | key.offset = parent; | ||
| 510 | |||
| 511 | ret = btrfs_search_slot(trans, root, &key, path, del ? -1 : 0, 1); | ||
| 512 | if (ret < 0) | ||
| 513 | goto out; | ||
| 514 | if (ret > 0) { | ||
| 515 | ret = -ENOENT; | ||
| 516 | goto out; | ||
| 517 | } | ||
| 518 | |||
| 519 | leaf = path->nodes[0]; | ||
| 520 | ref = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_extent_ref); | ||
| 521 | ref_objectid = btrfs_ref_objectid(leaf, ref); | ||
| 522 | if (btrfs_ref_root(leaf, ref) != ref_root || | ||
| 523 | btrfs_ref_generation(leaf, ref) != ref_generation || | ||
| 524 | (ref_objectid != owner_objectid && | ||
| 525 | ref_objectid != BTRFS_MULTIPLE_OBJECTIDS)) { | ||
| 526 | ret = -EIO; | ||
| 527 | WARN_ON(1); | ||
| 528 | goto out; | ||
| 529 | } | ||
| 530 | ret = 0; | ||
| 531 | out: | ||
| 532 | return ret; | ||
| 533 | } | ||
| 534 | |||
| 535 | /* | ||
| 536 | * updates all the backrefs that are pending on update_list for the | ||
| 537 | * extent_root | ||
| 538 | */ | ||
| 539 | static noinline int update_backrefs(struct btrfs_trans_handle *trans, | ||
| 540 | struct btrfs_root *extent_root, | ||
| 541 | struct btrfs_path *path, | ||
| 542 | struct list_head *update_list) | ||
| 543 | { | ||
| 544 | struct btrfs_key key; | ||
| 545 | struct btrfs_extent_ref *ref; | ||
| 546 | struct btrfs_fs_info *info = extent_root->fs_info; | ||
| 547 | struct pending_extent_op *op; | ||
| 548 | struct extent_buffer *leaf; | ||
| 549 | int ret = 0; | ||
| 550 | struct list_head *cur = update_list->next; | ||
| 551 | u64 ref_objectid; | ||
| 552 | u64 ref_root = extent_root->root_key.objectid; | ||
| 553 | |||
| 554 | op = list_entry(cur, struct pending_extent_op, list); | ||
| 555 | |||
| 556 | search: | ||
| 557 | key.objectid = op->bytenr; | ||
| 558 | key.type = BTRFS_EXTENT_REF_KEY; | ||
| 559 | key.offset = op->orig_parent; | ||
| 560 | |||
| 561 | ret = btrfs_search_slot(trans, extent_root, &key, path, 0, 1); | ||
| 562 | BUG_ON(ret); | ||
| 563 | |||
| 564 | leaf = path->nodes[0]; | ||
| 565 | |||
| 566 | loop: | ||
| 567 | ref = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_extent_ref); | ||
| 568 | |||
| 569 | ref_objectid = btrfs_ref_objectid(leaf, ref); | ||
| 570 | |||
| 571 | if (btrfs_ref_root(leaf, ref) != ref_root || | ||
| 572 | btrfs_ref_generation(leaf, ref) != op->orig_generation || | ||
| 573 | (ref_objectid != op->level && | ||
| 574 | ref_objectid != BTRFS_MULTIPLE_OBJECTIDS)) { | ||
| 575 | printk(KERN_ERR "btrfs couldn't find %llu, parent %llu, " | ||
| 576 | "root %llu, owner %u\n", | ||
| 577 | (unsigned long long)op->bytenr, | ||
| 578 | (unsigned long long)op->orig_parent, | ||
| 579 | (unsigned long long)ref_root, op->level); | ||
| 580 | btrfs_print_leaf(extent_root, leaf); | ||
| 581 | BUG(); | ||
| 582 | } | ||
| 583 | |||
| 584 | key.objectid = op->bytenr; | ||
| 585 | key.offset = op->parent; | ||
| 586 | key.type = BTRFS_EXTENT_REF_KEY; | ||
| 587 | ret = btrfs_set_item_key_safe(trans, extent_root, path, &key); | ||
| 588 | BUG_ON(ret); | ||
| 589 | ref = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_extent_ref); | ||
| 590 | btrfs_set_ref_generation(leaf, ref, op->generation); | ||
| 591 | |||
| 592 | cur = cur->next; | ||
| 593 | |||
| 594 | list_del_init(&op->list); | ||
| 595 | unlock_extent(&info->extent_ins, op->bytenr, | ||
| 596 | op->bytenr + op->num_bytes - 1, GFP_NOFS); | ||
| 597 | kfree(op); | ||
| 598 | |||
| 599 | if (cur == update_list) { | ||
| 600 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 601 | btrfs_release_path(extent_root, path); | ||
| 602 | goto out; | ||
| 603 | } | ||
| 604 | |||
| 605 | op = list_entry(cur, struct pending_extent_op, list); | ||
| 606 | |||
| 607 | path->slots[0]++; | ||
| 608 | while (path->slots[0] < btrfs_header_nritems(leaf)) { | ||
| 609 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 610 | if (key.objectid == op->bytenr && | ||
| 611 | key.type == BTRFS_EXTENT_REF_KEY) | ||
| 612 | goto loop; | ||
| 613 | path->slots[0]++; | ||
| 614 | } | ||
| 615 | |||
| 616 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 617 | btrfs_release_path(extent_root, path); | ||
| 618 | goto search; | ||
| 619 | |||
| 620 | out: | ||
| 621 | return 0; | ||
| 622 | } | ||
| 623 | |||
| 624 | static noinline int insert_extents(struct btrfs_trans_handle *trans, | ||
| 625 | struct btrfs_root *extent_root, | ||
| 626 | struct btrfs_path *path, | ||
| 627 | struct list_head *insert_list, int nr) | ||
| 628 | { | ||
| 629 | struct btrfs_key *keys; | ||
| 630 | u32 *data_size; | ||
| 631 | struct pending_extent_op *op; | ||
| 632 | struct extent_buffer *leaf; | ||
| 633 | struct list_head *cur = insert_list->next; | ||
| 634 | struct btrfs_fs_info *info = extent_root->fs_info; | ||
| 635 | u64 ref_root = extent_root->root_key.objectid; | ||
| 636 | int i = 0, last = 0, ret; | ||
| 637 | int total = nr * 2; | ||
| 638 | |||
| 639 | if (!nr) | ||
| 640 | return 0; | ||
| 641 | |||
| 642 | keys = kzalloc(total * sizeof(struct btrfs_key), GFP_NOFS); | ||
| 643 | if (!keys) | ||
| 644 | return -ENOMEM; | ||
| 645 | |||
| 646 | data_size = kzalloc(total * sizeof(u32), GFP_NOFS); | ||
| 647 | if (!data_size) { | ||
| 648 | kfree(keys); | ||
| 649 | return -ENOMEM; | ||
| 650 | } | ||
| 651 | |||
| 652 | list_for_each_entry(op, insert_list, list) { | ||
| 653 | keys[i].objectid = op->bytenr; | ||
| 654 | keys[i].offset = op->num_bytes; | ||
| 655 | keys[i].type = BTRFS_EXTENT_ITEM_KEY; | ||
| 656 | data_size[i] = sizeof(struct btrfs_extent_item); | ||
| 657 | i++; | ||
| 658 | |||
| 659 | keys[i].objectid = op->bytenr; | ||
| 660 | keys[i].offset = op->parent; | ||
| 661 | keys[i].type = BTRFS_EXTENT_REF_KEY; | ||
| 662 | data_size[i] = sizeof(struct btrfs_extent_ref); | ||
| 663 | i++; | ||
| 664 | } | ||
| 665 | |||
| 666 | op = list_entry(cur, struct pending_extent_op, list); | ||
| 667 | i = 0; | ||
| 668 | while (i < total) { | ||
| 669 | int c; | ||
| 670 | ret = btrfs_insert_some_items(trans, extent_root, path, | ||
| 671 | keys+i, data_size+i, total-i); | ||
| 672 | BUG_ON(ret < 0); | ||
| 673 | |||
| 674 | if (last && ret > 1) | ||
| 675 | BUG(); | ||
| 676 | |||
| 677 | leaf = path->nodes[0]; | ||
| 678 | for (c = 0; c < ret; c++) { | ||
| 679 | int ref_first = keys[i].type == BTRFS_EXTENT_REF_KEY; | ||
| 680 | |||
| 681 | /* | ||
| 682 | * if the first item we inserted was a backref, then | ||
| 683 | * the EXTENT_ITEM will be the odd c's, else it will | ||
| 684 | * be the even c's | ||
| 685 | */ | ||
| 686 | if ((ref_first && (c % 2)) || | ||
| 687 | (!ref_first && !(c % 2))) { | ||
| 688 | struct btrfs_extent_item *itm; | ||
| 689 | |||
| 690 | itm = btrfs_item_ptr(leaf, path->slots[0] + c, | ||
| 691 | struct btrfs_extent_item); | ||
| 692 | btrfs_set_extent_refs(path->nodes[0], itm, 1); | ||
| 693 | op->del++; | ||
| 694 | } else { | ||
| 695 | struct btrfs_extent_ref *ref; | ||
| 696 | |||
| 697 | ref = btrfs_item_ptr(leaf, path->slots[0] + c, | ||
| 698 | struct btrfs_extent_ref); | ||
| 699 | btrfs_set_ref_root(leaf, ref, ref_root); | ||
| 700 | btrfs_set_ref_generation(leaf, ref, | ||
| 701 | op->generation); | ||
| 702 | btrfs_set_ref_objectid(leaf, ref, op->level); | ||
| 703 | btrfs_set_ref_num_refs(leaf, ref, 1); | ||
| 704 | op->del++; | ||
| 705 | } | ||
| 706 | |||
| 707 | /* | ||
| 708 | * using del to see when its ok to free up the | ||
| 709 | * pending_extent_op. In the case where we insert the | ||
| 710 | * last item on the list in order to help do batching | ||
| 711 | * we need to not free the extent op until we actually | ||
| 712 | * insert the extent_item | ||
| 713 | */ | ||
| 714 | if (op->del == 2) { | ||
| 715 | unlock_extent(&info->extent_ins, op->bytenr, | ||
| 716 | op->bytenr + op->num_bytes - 1, | ||
| 717 | GFP_NOFS); | ||
| 718 | cur = cur->next; | ||
| 719 | list_del_init(&op->list); | ||
| 720 | kfree(op); | ||
| 721 | if (cur != insert_list) | ||
| 722 | op = list_entry(cur, | ||
| 723 | struct pending_extent_op, | ||
| 724 | list); | ||
| 725 | } | ||
| 726 | } | ||
| 727 | btrfs_mark_buffer_dirty(leaf); | ||
| 728 | btrfs_release_path(extent_root, path); | ||
| 729 | |||
| 730 | /* | ||
| 731 | * Ok backref's and items usually go right next to eachother, | ||
| 732 | * but if we could only insert 1 item that means that we | ||
| 733 | * inserted on the end of a leaf, and we have no idea what may | ||
| 734 | * be on the next leaf so we just play it safe. In order to | ||
| 735 | * try and help this case we insert the last thing on our | ||
| 736 | * insert list so hopefully it will end up being the last | ||
| 737 | * thing on the leaf and everything else will be before it, | ||
| 738 | * which will let us insert a whole bunch of items at the same | ||
| 739 | * time. | ||
| 740 | */ | ||
| 741 | if (ret == 1 && !last && (i + ret < total)) { | ||
| 742 | /* | ||
| 743 | * last: where we will pick up the next time around | ||
| 744 | * i: our current key to insert, will be total - 1 | ||
| 745 | * cur: the current op we are screwing with | ||
| 746 | * op: duh | ||
| 747 | */ | ||
| 748 | last = i + ret; | ||
| 749 | i = total - 1; | ||
| 750 | cur = insert_list->prev; | ||
| 751 | op = list_entry(cur, struct pending_extent_op, list); | ||
| 752 | } else if (last) { | ||
| 753 | /* | ||
| 754 | * ok we successfully inserted the last item on the | ||
| 755 | * list, lets reset everything | ||
| 756 | * | ||
| 757 | * i: our current key to insert, so where we left off | ||
| 758 | * last time | ||
| 759 | * last: done with this | ||
| 760 | * cur: the op we are messing with | ||
| 761 | * op: duh | ||
| 762 | * total: since we inserted the last key, we need to | ||
| 763 | * decrement total so we dont overflow | ||
| 764 | */ | ||
| 765 | i = last; | ||
| 766 | last = 0; | ||
| 767 | total--; | ||
| 768 | if (i < total) { | ||
| 769 | cur = insert_list->next; | ||
| 770 | op = list_entry(cur, struct pending_extent_op, | ||
| 771 | list); | ||
| 772 | } | ||
| 773 | } else { | ||
| 774 | i += ret; | ||
| 775 | } | ||
| 776 | |||
| 777 | cond_resched(); | ||
| 778 | } | ||
| 779 | ret = 0; | ||
| 780 | kfree(keys); | ||
| 781 | kfree(data_size); | ||
| 782 | return ret; | ||
| 783 | } | ||
| 784 | |||
| 785 | static noinline int insert_extent_backref(struct btrfs_trans_handle *trans, | ||
| 786 | struct btrfs_root *root, | ||
| 787 | struct btrfs_path *path, | ||
| 788 | u64 bytenr, u64 parent, | ||
| 789 | u64 ref_root, u64 ref_generation, | ||
| 790 | u64 owner_objectid) | ||
| 791 | { | ||
| 792 | struct btrfs_key key; | ||
| 793 | struct extent_buffer *leaf; | ||
| 794 | struct btrfs_extent_ref *ref; | ||
| 795 | u32 num_refs; | ||
| 796 | int ret; | ||
| 797 | |||
| 798 | key.objectid = bytenr; | ||
| 799 | key.type = BTRFS_EXTENT_REF_KEY; | ||
| 800 | key.offset = parent; | ||
| 801 | |||
| 802 | ret = btrfs_insert_empty_item(trans, root, path, &key, sizeof(*ref)); | ||
| 803 | if (ret == 0) { | ||
| 804 | leaf = path->nodes[0]; | ||
| 805 | ref = btrfs_item_ptr(leaf, path->slots[0], | ||
| 806 | struct btrfs_extent_ref); | ||
| 807 | btrfs_set_ref_root(leaf, ref, ref_root); | ||
| 808 | btrfs_set_ref_generation(leaf, ref, ref_generation); | ||
| 809 | btrfs_set_ref_objectid(leaf, ref, owner_objectid); | ||
| 810 | btrfs_set_ref_num_refs(leaf, ref, 1); | ||
| 811 | } else if (ret == -EEXIST) { | ||
| 812 | u64 existing_owner; | ||
| 813 | BUG_ON(owner_objectid < BTRFS_FIRST_FREE_OBJECTID); | ||
| 814 | leaf = path->nodes[0]; | ||
| 815 | ref = btrfs_item_ptr(leaf, path->slots[0], | ||
| 816 | struct btrfs_extent_ref); | ||
| 817 | if (btrfs_ref_root(leaf, ref) != ref_root || | ||
| 818 | btrfs_ref_generation(leaf, ref) != ref_generation) { | ||
| 819 | ret = -EIO; | ||
| 820 | WARN_ON(1); | ||
| 821 | goto out; | ||
| 822 | } | ||
| 823 | |||
| 824 | num_refs = btrfs_ref_num_refs(leaf, ref); | ||
| 825 | BUG_ON(num_refs == 0); | ||
| 826 | btrfs_set_ref_num_refs(leaf, ref, num_refs + 1); | ||
| 827 | |||
| 828 | existing_owner = btrfs_ref_objectid(leaf, ref); | ||
| 829 | if (existing_owner != owner_objectid && | ||
| 830 | existing_owner != BTRFS_MULTIPLE_OBJECTIDS) { | ||
| 831 | btrfs_set_ref_objectid(leaf, ref, | ||
| 832 | BTRFS_MULTIPLE_OBJECTIDS); | ||
| 833 | } | ||
| 834 | ret = 0; | ||
| 835 | } else { | ||
| 836 | goto out; | ||
| 837 | } | ||
| 838 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 839 | out: | ||
| 840 | btrfs_release_path(root, path); | ||
| 841 | return ret; | ||
| 842 | } | ||
| 843 | |||
| 844 | static noinline int remove_extent_backref(struct btrfs_trans_handle *trans, | ||
| 845 | struct btrfs_root *root, | ||
| 846 | struct btrfs_path *path) | ||
| 847 | { | ||
| 848 | struct extent_buffer *leaf; | ||
| 849 | struct btrfs_extent_ref *ref; | ||
| 850 | u32 num_refs; | ||
| 851 | int ret = 0; | ||
| 852 | |||
| 853 | leaf = path->nodes[0]; | ||
| 854 | ref = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_extent_ref); | ||
| 855 | num_refs = btrfs_ref_num_refs(leaf, ref); | ||
| 856 | BUG_ON(num_refs == 0); | ||
| 857 | num_refs -= 1; | ||
| 858 | if (num_refs == 0) { | ||
| 859 | ret = btrfs_del_item(trans, root, path); | ||
| 860 | } else { | ||
| 861 | btrfs_set_ref_num_refs(leaf, ref, num_refs); | ||
| 862 | btrfs_mark_buffer_dirty(leaf); | ||
| 863 | } | ||
| 864 | btrfs_release_path(root, path); | ||
| 865 | return ret; | ||
| 866 | } | ||
| 867 | |||
| 868 | #ifdef BIO_RW_DISCARD | ||
| 869 | static void btrfs_issue_discard(struct block_device *bdev, | ||
| 870 | u64 start, u64 len) | ||
| 871 | { | ||
| 872 | blkdev_issue_discard(bdev, start >> 9, len >> 9, GFP_KERNEL); | ||
| 873 | } | ||
| 874 | #endif | ||
| 875 | |||
| 876 | static int btrfs_discard_extent(struct btrfs_root *root, u64 bytenr, | ||
| 877 | u64 num_bytes) | ||
| 878 | { | ||
| 879 | #ifdef BIO_RW_DISCARD | ||
| 880 | int ret; | ||
| 881 | u64 map_length = num_bytes; | ||
| 882 | struct btrfs_multi_bio *multi = NULL; | ||
| 883 | |||
| 884 | /* Tell the block device(s) that the sectors can be discarded */ | ||
| 885 | ret = btrfs_map_block(&root->fs_info->mapping_tree, READ, | ||
| 886 | bytenr, &map_length, &multi, 0); | ||
| 887 | if (!ret) { | ||
| 888 | struct btrfs_bio_stripe *stripe = multi->stripes; | ||
| 889 | int i; | ||
| 890 | |||
| 891 | if (map_length > num_bytes) | ||
| 892 | map_length = num_bytes; | ||
| 893 | |||
| 894 | for (i = 0; i < multi->num_stripes; i++, stripe++) { | ||
| 895 | btrfs_issue_discard(stripe->dev->bdev, | ||
| 896 | stripe->physical, | ||
| 897 | map_length); | ||
| 898 | } | ||
| 899 | kfree(multi); | ||
| 900 | } | ||
| 901 | |||
| 902 | return ret; | ||
| 903 | #else | ||
| 904 | return 0; | ||
| 905 | #endif | ||
| 906 | } | ||
| 907 | |||
| 908 | static noinline int free_extents(struct btrfs_trans_handle *trans, | ||
| 909 | struct btrfs_root *extent_root, | ||
| 910 | struct list_head *del_list) | ||
| 911 | { | ||
| 912 | struct btrfs_fs_info *info = extent_root->fs_info; | ||
| 913 | struct btrfs_path *path; | ||
| 914 | struct btrfs_key key, found_key; | ||
| 915 | struct extent_buffer *leaf; | ||
| 916 | struct list_head *cur; | ||
| 917 | struct pending_extent_op *op; | ||
| 918 | struct btrfs_extent_item *ei; | ||
| 919 | int ret, num_to_del, extent_slot = 0, found_extent = 0; | ||
| 920 | u32 refs; | ||
| 921 | u64 bytes_freed = 0; | ||
| 922 | |||
| 923 | path = btrfs_alloc_path(); | ||
| 924 | if (!path) | ||
| 925 | return -ENOMEM; | ||
| 926 | path->reada = 1; | ||
| 927 | |||
| 928 | search: | ||
| 929 | /* search for the backref for the current ref we want to delete */ | ||
| 930 | cur = del_list->next; | ||
| 931 | op = list_entry(cur, struct pending_extent_op, list); | ||
| 932 | ret = lookup_extent_backref(trans, extent_root, path, op->bytenr, | ||
| 933 | op->orig_parent, | ||
| 934 | extent_root->root_key.objectid, | ||
| 935 | op->orig_generation, op->level, 1); | ||
| 936 | if (ret) { | ||
| 937 | printk(KERN_ERR "btrfs unable to find backref byte nr %llu " | ||
| 938 | "root %llu gen %llu owner %u\n", | ||
| 939 | (unsigned long long)op->bytenr, | ||
| 940 | (unsigned long long)extent_root->root_key.objectid, | ||
| 941 | (unsigned long long)op->orig_generation, op->level); | ||
| 942 | btrfs_print_leaf(extent_root, path->nodes[0]); | ||
| 943 | WARN_ON(1); | ||
| 944 | goto out; | ||
| 945 | } | ||
| 946 | |||
| 947 | extent_slot = path->slots[0]; | ||
| 948 | num_to_del = 1; | ||
| 949 | found_extent = 0; | ||
| 950 | |||
| 951 | /* | ||
| 952 | * if we aren't the first item on the leaf we can move back one and see | ||
| 953 | * if our ref is right next to our extent item | ||
| 954 | */ | ||
| 955 | if (likely(extent_slot)) { | ||
| 956 | extent_slot--; | ||
| 957 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 958 | extent_slot); | ||
| 959 | if (found_key.objectid == op->bytenr && | ||
| 960 | found_key.type == BTRFS_EXTENT_ITEM_KEY && | ||
| 961 | found_key.offset == op->num_bytes) { | ||
| 962 | num_to_del++; | ||
| 963 | found_extent = 1; | ||
| 964 | } | ||
| 965 | } | ||
| 966 | |||
| 967 | /* | ||
| 968 | * if we didn't find the extent we need to delete the backref and then | ||
| 969 | * search for the extent item key so we can update its ref count | ||
| 970 | */ | ||
| 971 | if (!found_extent) { | ||
| 972 | key.objectid = op->bytenr; | ||
| 973 | key.type = BTRFS_EXTENT_ITEM_KEY; | ||
| 974 | key.offset = op->num_bytes; | ||
| 975 | |||
| 976 | ret = remove_extent_backref(trans, extent_root, path); | ||
| 977 | BUG_ON(ret); | ||
| 978 | btrfs_release_path(extent_root, path); | ||
| 979 | ret = btrfs_search_slot(trans, extent_root, &key, path, -1, 1); | ||
| 980 | BUG_ON(ret); | ||
| 981 | extent_slot = path->slots[0]; | ||
| 982 | } | ||
| 983 | |||
| 984 | /* this is where we update the ref count for the extent */ | ||
| 985 | leaf = path->nodes[0]; | ||
| 986 | ei = btrfs_item_ptr(leaf, extent_slot, struct btrfs_extent_item); | ||
| 987 | refs = btrfs_extent_refs(leaf, ei); | ||
| 988 | BUG_ON(refs == 0); | ||
| 989 | refs--; | ||
| 990 | btrfs_set_extent_refs(leaf, ei, refs); | ||
| 991 | |||
| 992 | btrfs_mark_buffer_dirty(leaf); | ||
| 993 | |||
| 994 | /* | ||
| 995 | * This extent needs deleting. The reason cur_slot is extent_slot + | ||
| 996 | * num_to_del is because extent_slot points to the slot where the extent | ||
| 997 | * is, and if the backref was not right next to the extent we will be | ||
| 998 | * deleting at least 1 item, and will want to start searching at the | ||
| 999 | * slot directly next to extent_slot. However if we did find the | ||
| 1000 | * backref next to the extent item them we will be deleting at least 2 | ||
| 1001 | * items and will want to start searching directly after the ref slot | ||
| 1002 | */ | ||
| 1003 | if (!refs) { | ||
| 1004 | struct list_head *pos, *n, *end; | ||
| 1005 | int cur_slot = extent_slot+num_to_del; | ||
| 1006 | u64 super_used; | ||
| 1007 | u64 root_used; | ||
| 1008 | |||
| 1009 | path->slots[0] = extent_slot; | ||
| 1010 | bytes_freed = op->num_bytes; | ||
| 1011 | |||
| 1012 | mutex_lock(&info->pinned_mutex); | ||
| 1013 | ret = pin_down_bytes(trans, extent_root, op->bytenr, | ||
| 1014 | op->num_bytes, op->level >= | ||
| 1015 | BTRFS_FIRST_FREE_OBJECTID); | ||
| 1016 | mutex_unlock(&info->pinned_mutex); | ||
| 1017 | BUG_ON(ret < 0); | ||
| 1018 | op->del = ret; | ||
| 1019 | |||
| 1020 | /* | ||
| 1021 | * we need to see if we can delete multiple things at once, so | ||
| 1022 | * start looping through the list of extents we are wanting to | ||
| 1023 | * delete and see if their extent/backref's are right next to | ||
| 1024 | * eachother and the extents only have 1 ref | ||
| 1025 | */ | ||
| 1026 | for (pos = cur->next; pos != del_list; pos = pos->next) { | ||
| 1027 | struct pending_extent_op *tmp; | ||
| 1028 | |||
| 1029 | tmp = list_entry(pos, struct pending_extent_op, list); | ||
| 1030 | |||
| 1031 | /* we only want to delete extent+ref at this stage */ | ||
| 1032 | if (cur_slot >= btrfs_header_nritems(leaf) - 1) | ||
| 1033 | break; | ||
| 1034 | |||
| 1035 | btrfs_item_key_to_cpu(leaf, &found_key, cur_slot); | ||
| 1036 | if (found_key.objectid != tmp->bytenr || | ||
| 1037 | found_key.type != BTRFS_EXTENT_ITEM_KEY || | ||
| 1038 | found_key.offset != tmp->num_bytes) | ||
| 1039 | break; | ||
| 1040 | |||
| 1041 | /* check to make sure this extent only has one ref */ | ||
| 1042 | ei = btrfs_item_ptr(leaf, cur_slot, | ||
| 1043 | struct btrfs_extent_item); | ||
| 1044 | if (btrfs_extent_refs(leaf, ei) != 1) | ||
| 1045 | break; | ||
| 1046 | |||
| 1047 | btrfs_item_key_to_cpu(leaf, &found_key, cur_slot+1); | ||
| 1048 | if (found_key.objectid != tmp->bytenr || | ||
| 1049 | found_key.type != BTRFS_EXTENT_REF_KEY || | ||
| 1050 | found_key.offset != tmp->orig_parent) | ||
| 1051 | break; | ||
| 1052 | |||
| 1053 | /* | ||
| 1054 | * the ref is right next to the extent, we can set the | ||
| 1055 | * ref count to 0 since we will delete them both now | ||
| 1056 | */ | ||
| 1057 | btrfs_set_extent_refs(leaf, ei, 0); | ||
| 1058 | |||
| 1059 | /* pin down the bytes for this extent */ | ||
| 1060 | mutex_lock(&info->pinned_mutex); | ||
| 1061 | ret = pin_down_bytes(trans, extent_root, tmp->bytenr, | ||
| 1062 | tmp->num_bytes, tmp->level >= | ||
| 1063 | BTRFS_FIRST_FREE_OBJECTID); | ||
| 1064 | mutex_unlock(&info->pinned_mutex); | ||
| 1065 | BUG_ON(ret < 0); | ||
| 1066 | |||
| 1067 | /* | ||
| 1068 | * use the del field to tell if we need to go ahead and | ||
| 1069 | * free up the extent when we delete the item or not. | ||
| 1070 | */ | ||
| 1071 | tmp->del = ret; | ||
| 1072 | bytes_freed += tmp->num_bytes; | ||
| 1073 | |||
| 1074 | num_to_del += 2; | ||
| 1075 | cur_slot += 2; | ||
| 1076 | } | ||
| 1077 | end = pos; | ||
| 1078 | |||
| 1079 | /* update the free space counters */ | ||
| 1080 | spin_lock(&info->delalloc_lock); | ||
| 1081 | super_used = btrfs_super_bytes_used(&info->super_copy); | ||
| 1082 | btrfs_set_super_bytes_used(&info->super_copy, | ||
| 1083 | super_used - bytes_freed); | ||
| 1084 | |||
| 1085 | root_used = btrfs_root_used(&extent_root->root_item); | ||
| 1086 | btrfs_set_root_used(&extent_root->root_item, | ||
| 1087 | root_used - bytes_freed); | ||
| 1088 | spin_unlock(&info->delalloc_lock); | ||
| 1089 | |||
| 1090 | /* delete the items */ | ||
| 1091 | ret = btrfs_del_items(trans, extent_root, path, | ||
| 1092 | path->slots[0], num_to_del); | ||
| 1093 | BUG_ON(ret); | ||
| 1094 | |||
| 1095 | /* | ||
| 1096 | * loop through the extents we deleted and do the cleanup work | ||
| 1097 | * on them | ||
| 1098 | */ | ||
| 1099 | for (pos = cur, n = pos->next; pos != end; | ||
| 1100 | pos = n, n = pos->next) { | ||
| 1101 | struct pending_extent_op *tmp; | ||
| 1102 | tmp = list_entry(pos, struct pending_extent_op, list); | ||
| 1103 | |||
| 1104 | /* | ||
| 1105 | * remember tmp->del tells us wether or not we pinned | ||
| 1106 | * down the extent | ||
| 1107 | */ | ||
| 1108 | ret = update_block_group(trans, extent_root, | ||
| 1109 | tmp->bytenr, tmp->num_bytes, 0, | ||
| 1110 | tmp->del); | ||
| 1111 | BUG_ON(ret); | ||
| 1112 | |||
| 1113 | list_del_init(&tmp->list); | ||
| 1114 | unlock_extent(&info->extent_ins, tmp->bytenr, | ||
| 1115 | tmp->bytenr + tmp->num_bytes - 1, | ||
| 1116 | GFP_NOFS); | ||
| 1117 | kfree(tmp); | ||
| 1118 | } | ||
| 1119 | } else if (refs && found_extent) { | ||
| 1120 | /* | ||
| 1121 | * the ref and extent were right next to eachother, but the | ||
| 1122 | * extent still has a ref, so just free the backref and keep | ||
| 1123 | * going | ||
| 1124 | */ | ||
| 1125 | ret = remove_extent_backref(trans, extent_root, path); | ||
| 1126 | BUG_ON(ret); | ||
| 1127 | |||
| 1128 | list_del_init(&op->list); | ||
| 1129 | unlock_extent(&info->extent_ins, op->bytenr, | ||
| 1130 | op->bytenr + op->num_bytes - 1, GFP_NOFS); | ||
| 1131 | kfree(op); | ||
| 1132 | } else { | ||
| 1133 | /* | ||
| 1134 | * the extent has multiple refs and the backref we were looking | ||
| 1135 | * for was not right next to it, so just unlock and go next, | ||
| 1136 | * we're good to go | ||
| 1137 | */ | ||
| 1138 | list_del_init(&op->list); | ||
| 1139 | unlock_extent(&info->extent_ins, op->bytenr, | ||
| 1140 | op->bytenr + op->num_bytes - 1, GFP_NOFS); | ||
| 1141 | kfree(op); | ||
| 1142 | } | ||
| 1143 | |||
| 1144 | btrfs_release_path(extent_root, path); | ||
| 1145 | if (!list_empty(del_list)) | ||
| 1146 | goto search; | ||
| 1147 | |||
| 1148 | out: | ||
| 1149 | btrfs_free_path(path); | ||
| 1150 | return ret; | ||
| 1151 | } | ||
| 1152 | |||
| 1153 | static int __btrfs_update_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1154 | struct btrfs_root *root, u64 bytenr, | ||
| 1155 | u64 orig_parent, u64 parent, | ||
| 1156 | u64 orig_root, u64 ref_root, | ||
| 1157 | u64 orig_generation, u64 ref_generation, | ||
| 1158 | u64 owner_objectid) | ||
| 1159 | { | ||
| 1160 | int ret; | ||
| 1161 | struct btrfs_root *extent_root = root->fs_info->extent_root; | ||
| 1162 | struct btrfs_path *path; | ||
| 1163 | |||
| 1164 | if (root == root->fs_info->extent_root) { | ||
| 1165 | struct pending_extent_op *extent_op; | ||
| 1166 | u64 num_bytes; | ||
| 1167 | |||
| 1168 | BUG_ON(owner_objectid >= BTRFS_MAX_LEVEL); | ||
| 1169 | num_bytes = btrfs_level_size(root, (int)owner_objectid); | ||
| 1170 | mutex_lock(&root->fs_info->extent_ins_mutex); | ||
| 1171 | if (test_range_bit(&root->fs_info->extent_ins, bytenr, | ||
| 1172 | bytenr + num_bytes - 1, EXTENT_WRITEBACK, 0)) { | ||
| 1173 | u64 priv; | ||
| 1174 | ret = get_state_private(&root->fs_info->extent_ins, | ||
| 1175 | bytenr, &priv); | ||
| 1176 | BUG_ON(ret); | ||
| 1177 | extent_op = (struct pending_extent_op *) | ||
| 1178 | (unsigned long)priv; | ||
| 1179 | BUG_ON(extent_op->parent != orig_parent); | ||
| 1180 | BUG_ON(extent_op->generation != orig_generation); | ||
| 1181 | |||
| 1182 | extent_op->parent = parent; | ||
| 1183 | extent_op->generation = ref_generation; | ||
| 1184 | } else { | ||
| 1185 | extent_op = kmalloc(sizeof(*extent_op), GFP_NOFS); | ||
| 1186 | BUG_ON(!extent_op); | ||
| 1187 | |||
| 1188 | extent_op->type = PENDING_BACKREF_UPDATE; | ||
| 1189 | extent_op->bytenr = bytenr; | ||
| 1190 | extent_op->num_bytes = num_bytes; | ||
| 1191 | extent_op->parent = parent; | ||
| 1192 | extent_op->orig_parent = orig_parent; | ||
| 1193 | extent_op->generation = ref_generation; | ||
| 1194 | extent_op->orig_generation = orig_generation; | ||
| 1195 | extent_op->level = (int)owner_objectid; | ||
| 1196 | INIT_LIST_HEAD(&extent_op->list); | ||
| 1197 | extent_op->del = 0; | ||
| 1198 | |||
| 1199 | set_extent_bits(&root->fs_info->extent_ins, | ||
| 1200 | bytenr, bytenr + num_bytes - 1, | ||
| 1201 | EXTENT_WRITEBACK, GFP_NOFS); | ||
| 1202 | set_state_private(&root->fs_info->extent_ins, | ||
| 1203 | bytenr, (unsigned long)extent_op); | ||
| 1204 | } | ||
| 1205 | mutex_unlock(&root->fs_info->extent_ins_mutex); | ||
| 1206 | return 0; | ||
| 1207 | } | ||
| 1208 | |||
| 1209 | path = btrfs_alloc_path(); | ||
| 1210 | if (!path) | ||
| 1211 | return -ENOMEM; | ||
| 1212 | ret = lookup_extent_backref(trans, extent_root, path, | ||
| 1213 | bytenr, orig_parent, orig_root, | ||
| 1214 | orig_generation, owner_objectid, 1); | ||
| 1215 | if (ret) | ||
| 1216 | goto out; | ||
| 1217 | ret = remove_extent_backref(trans, extent_root, path); | ||
| 1218 | if (ret) | ||
| 1219 | goto out; | ||
| 1220 | ret = insert_extent_backref(trans, extent_root, path, bytenr, | ||
| 1221 | parent, ref_root, ref_generation, | ||
| 1222 | owner_objectid); | ||
| 1223 | BUG_ON(ret); | ||
| 1224 | finish_current_insert(trans, extent_root, 0); | ||
| 1225 | del_pending_extents(trans, extent_root, 0); | ||
| 1226 | out: | ||
| 1227 | btrfs_free_path(path); | ||
| 1228 | return ret; | ||
| 1229 | } | ||
| 1230 | |||
| 1231 | int btrfs_update_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1232 | struct btrfs_root *root, u64 bytenr, | ||
| 1233 | u64 orig_parent, u64 parent, | ||
| 1234 | u64 ref_root, u64 ref_generation, | ||
| 1235 | u64 owner_objectid) | ||
| 1236 | { | ||
| 1237 | int ret; | ||
| 1238 | if (ref_root == BTRFS_TREE_LOG_OBJECTID && | ||
| 1239 | owner_objectid < BTRFS_FIRST_FREE_OBJECTID) | ||
| 1240 | return 0; | ||
| 1241 | ret = __btrfs_update_extent_ref(trans, root, bytenr, orig_parent, | ||
| 1242 | parent, ref_root, ref_root, | ||
| 1243 | ref_generation, ref_generation, | ||
| 1244 | owner_objectid); | ||
| 1245 | return ret; | ||
| 1246 | } | ||
| 1247 | |||
| 1248 | static int __btrfs_inc_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1249 | struct btrfs_root *root, u64 bytenr, | ||
| 1250 | u64 orig_parent, u64 parent, | ||
| 1251 | u64 orig_root, u64 ref_root, | ||
| 1252 | u64 orig_generation, u64 ref_generation, | ||
| 1253 | u64 owner_objectid) | ||
| 1254 | { | ||
| 1255 | struct btrfs_path *path; | ||
| 1256 | int ret; | ||
| 1257 | struct btrfs_key key; | ||
| 1258 | struct extent_buffer *l; | ||
| 1259 | struct btrfs_extent_item *item; | ||
| 1260 | u32 refs; | ||
| 1261 | |||
| 1262 | path = btrfs_alloc_path(); | ||
| 1263 | if (!path) | ||
| 1264 | return -ENOMEM; | ||
| 1265 | |||
| 1266 | path->reada = 1; | ||
| 1267 | key.objectid = bytenr; | ||
| 1268 | key.type = BTRFS_EXTENT_ITEM_KEY; | ||
| 1269 | key.offset = (u64)-1; | ||
| 1270 | |||
| 1271 | ret = btrfs_search_slot(trans, root->fs_info->extent_root, &key, path, | ||
| 1272 | 0, 1); | ||
| 1273 | if (ret < 0) | ||
| 1274 | return ret; | ||
| 1275 | BUG_ON(ret == 0 || path->slots[0] == 0); | ||
| 1276 | |||
| 1277 | path->slots[0]--; | ||
| 1278 | l = path->nodes[0]; | ||
| 1279 | |||
| 1280 | btrfs_item_key_to_cpu(l, &key, path->slots[0]); | ||
| 1281 | if (key.objectid != bytenr) { | ||
| 1282 | btrfs_print_leaf(root->fs_info->extent_root, path->nodes[0]); | ||
| 1283 | printk(KERN_ERR "btrfs wanted %llu found %llu\n", | ||
| 1284 | (unsigned long long)bytenr, | ||
| 1285 | (unsigned long long)key.objectid); | ||
| 1286 | BUG(); | ||
| 1287 | } | ||
| 1288 | BUG_ON(key.type != BTRFS_EXTENT_ITEM_KEY); | ||
| 1289 | |||
| 1290 | item = btrfs_item_ptr(l, path->slots[0], struct btrfs_extent_item); | ||
| 1291 | refs = btrfs_extent_refs(l, item); | ||
| 1292 | btrfs_set_extent_refs(l, item, refs + 1); | ||
| 1293 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 1294 | |||
| 1295 | btrfs_release_path(root->fs_info->extent_root, path); | ||
| 1296 | |||
| 1297 | path->reada = 1; | ||
| 1298 | ret = insert_extent_backref(trans, root->fs_info->extent_root, | ||
| 1299 | path, bytenr, parent, | ||
| 1300 | ref_root, ref_generation, | ||
| 1301 | owner_objectid); | ||
| 1302 | BUG_ON(ret); | ||
| 1303 | finish_current_insert(trans, root->fs_info->extent_root, 0); | ||
| 1304 | del_pending_extents(trans, root->fs_info->extent_root, 0); | ||
| 1305 | |||
| 1306 | btrfs_free_path(path); | ||
| 1307 | return 0; | ||
| 1308 | } | ||
| 1309 | |||
| 1310 | int btrfs_inc_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1311 | struct btrfs_root *root, | ||
| 1312 | u64 bytenr, u64 num_bytes, u64 parent, | ||
| 1313 | u64 ref_root, u64 ref_generation, | ||
| 1314 | u64 owner_objectid) | ||
| 1315 | { | ||
| 1316 | int ret; | ||
| 1317 | if (ref_root == BTRFS_TREE_LOG_OBJECTID && | ||
| 1318 | owner_objectid < BTRFS_FIRST_FREE_OBJECTID) | ||
| 1319 | return 0; | ||
| 1320 | ret = __btrfs_inc_extent_ref(trans, root, bytenr, 0, parent, | ||
| 1321 | 0, ref_root, 0, ref_generation, | ||
| 1322 | owner_objectid); | ||
| 1323 | return ret; | ||
| 1324 | } | ||
| 1325 | |||
| 1326 | int btrfs_extent_post_op(struct btrfs_trans_handle *trans, | ||
| 1327 | struct btrfs_root *root) | ||
| 1328 | { | ||
| 1329 | finish_current_insert(trans, root->fs_info->extent_root, 1); | ||
| 1330 | del_pending_extents(trans, root->fs_info->extent_root, 1); | ||
| 1331 | return 0; | ||
| 1332 | } | ||
| 1333 | |||
| 1334 | int btrfs_lookup_extent_ref(struct btrfs_trans_handle *trans, | ||
| 1335 | struct btrfs_root *root, u64 bytenr, | ||
| 1336 | u64 num_bytes, u32 *refs) | ||
| 1337 | { | ||
| 1338 | struct btrfs_path *path; | ||
| 1339 | int ret; | ||
| 1340 | struct btrfs_key key; | ||
| 1341 | struct extent_buffer *l; | ||
| 1342 | struct btrfs_extent_item *item; | ||
| 1343 | |||
| 1344 | WARN_ON(num_bytes < root->sectorsize); | ||
| 1345 | path = btrfs_alloc_path(); | ||
| 1346 | path->reada = 1; | ||
| 1347 | key.objectid = bytenr; | ||
| 1348 | key.offset = num_bytes; | ||
| 1349 | btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY); | ||
| 1350 | ret = btrfs_search_slot(trans, root->fs_info->extent_root, &key, path, | ||
| 1351 | 0, 0); | ||
| 1352 | if (ret < 0) | ||
| 1353 | goto out; | ||
| 1354 | if (ret != 0) { | ||
| 1355 | btrfs_print_leaf(root, path->nodes[0]); | ||
| 1356 | printk(KERN_INFO "btrfs failed to find block number %llu\n", | ||
| 1357 | (unsigned long long)bytenr); | ||
| 1358 | BUG(); | ||
| 1359 | } | ||
| 1360 | l = path->nodes[0]; | ||
| 1361 | item = btrfs_item_ptr(l, path->slots[0], struct btrfs_extent_item); | ||
| 1362 | *refs = btrfs_extent_refs(l, item); | ||
| 1363 | out: | ||
| 1364 | btrfs_free_path(path); | ||
| 1365 | return 0; | ||
| 1366 | } | ||
| 1367 | |||
| 1368 | int btrfs_cross_ref_exist(struct btrfs_trans_handle *trans, | ||
| 1369 | struct btrfs_root *root, u64 objectid, u64 bytenr) | ||
| 1370 | { | ||
| 1371 | struct btrfs_root *extent_root = root->fs_info->extent_root; | ||
| 1372 | struct btrfs_path *path; | ||
| 1373 | struct extent_buffer *leaf; | ||
| 1374 | struct btrfs_extent_ref *ref_item; | ||
| 1375 | struct btrfs_key key; | ||
| 1376 | struct btrfs_key found_key; | ||
| 1377 | u64 ref_root; | ||
| 1378 | u64 last_snapshot; | ||
| 1379 | u32 nritems; | ||
| 1380 | int ret; | ||
| 1381 | |||
| 1382 | key.objectid = bytenr; | ||
| 1383 | key.offset = (u64)-1; | ||
| 1384 | key.type = BTRFS_EXTENT_ITEM_KEY; | ||
| 1385 | |||
| 1386 | path = btrfs_alloc_path(); | ||
| 1387 | ret = btrfs_search_slot(NULL, extent_root, &key, path, 0, 0); | ||
| 1388 | if (ret < 0) | ||
| 1389 | goto out; | ||
| 1390 | BUG_ON(ret == 0); | ||
| 1391 | |||
| 1392 | ret = -ENOENT; | ||
| 1393 | if (path->slots[0] == 0) | ||
| 1394 | goto out; | ||
| 1395 | |||
| 1396 | path->slots[0]--; | ||
| 1397 | leaf = path->nodes[0]; | ||
| 1398 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 1399 | |||
| 1400 | if (found_key.objectid != bytenr || | ||
| 1401 | found_key.type != BTRFS_EXTENT_ITEM_KEY) | ||
| 1402 | goto out; | ||
| 1403 | |||
| 1404 | last_snapshot = btrfs_root_last_snapshot(&root->root_item); | ||
| 1405 | while (1) { | ||
| 1406 | leaf = path->nodes[0]; | ||
| 1407 | nritems = btrfs_header_nritems(leaf); | ||
| 1408 | if (path->slots[0] >= nritems) { | ||
| 1409 | ret = btrfs_next_leaf(extent_root, path); | ||
| 1410 | if (ret < 0) | ||
| 1411 | goto out; | ||
| 1412 | if (ret == 0) | ||
| 1413 | continue; | ||
| 1414 | break; | ||
| 1415 | } | ||
| 1416 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 1417 | if (found_key.objectid != bytenr) | ||
| 1418 | break; | ||
| 1419 | |||
| 1420 | if (found_key.type != BTRFS_EXTENT_REF_KEY) { | ||
| 1421 | path->slots[0]++; | ||
| 1422 | continue; | ||
| 1423 | } | ||
| 1424 | |||
| 1425 | ref_item = btrfs_item_ptr(leaf, path->slots[0], | ||
| 1426 | struct btrfs_extent_ref); | ||
| 1427 | ref_root = btrfs_ref_root(leaf, ref_item); | ||
| 1428 | if ((ref_root != root->root_key.objectid && | ||
| 1429 | ref_root != BTRFS_TREE_LOG_OBJECTID) || | ||
| 1430 | objectid != btrfs_ref_objectid(leaf, ref_item)) { | ||
| 1431 | ret = 1; | ||
| 1432 | goto out; | ||
| 1433 | } | ||
| 1434 | if (btrfs_ref_generation(leaf, ref_item) <= last_snapshot) { | ||
| 1435 | ret = 1; | ||
| 1436 | goto out; | ||
| 1437 | } | ||
| 1438 | |||
| 1439 | path->slots[0]++; | ||
| 1440 | } | ||
| 1441 | ret = 0; | ||
| 1442 | out: | ||
| 1443 | btrfs_free_path(path); | ||
| 1444 | return ret; | ||
| 1445 | } | ||
| 1446 | |||
| 1447 | int btrfs_cache_ref(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 1448 | struct extent_buffer *buf, u32 nr_extents) | ||
| 1449 | { | ||
| 1450 | struct btrfs_key key; | ||
| 1451 | struct btrfs_file_extent_item *fi; | ||
| 1452 | u64 root_gen; | ||
| 1453 | u32 nritems; | ||
| 1454 | int i; | ||
| 1455 | int level; | ||
| 1456 | int ret = 0; | ||
| 1457 | int shared = 0; | ||
| 1458 | |||
| 1459 | if (!root->ref_cows) | ||
| 1460 | return 0; | ||
| 1461 | |||
| 1462 | if (root->root_key.objectid != BTRFS_TREE_RELOC_OBJECTID) { | ||
| 1463 | shared = 0; | ||
| 1464 | root_gen = root->root_key.offset; | ||
| 1465 | } else { | ||
| 1466 | shared = 1; | ||
| 1467 | root_gen = trans->transid - 1; | ||
| 1468 | } | ||
| 1469 | |||
| 1470 | level = btrfs_header_level(buf); | ||
| 1471 | nritems = btrfs_header_nritems(buf); | ||
| 1472 | |||
| 1473 | if (level == 0) { | ||
| 1474 | struct btrfs_leaf_ref *ref; | ||
| 1475 | struct btrfs_extent_info *info; | ||
| 1476 | |||
| 1477 | ref = btrfs_alloc_leaf_ref(root, nr_extents); | ||
| 1478 | if (!ref) { | ||
| 1479 | ret = -ENOMEM; | ||
| 1480 | goto out; | ||
| 1481 | } | ||
| 1482 | |||
| 1483 | ref->root_gen = root_gen; | ||
| 1484 | ref->bytenr = buf->start; | ||
| 1485 | ref->owner = btrfs_header_owner(buf); | ||
| 1486 | ref->generation = btrfs_header_generation(buf); | ||
| 1487 | ref->nritems = nr_extents; | ||
| 1488 | info = ref->extents; | ||
| 1489 | |||
| 1490 | for (i = 0; nr_extents > 0 && i < nritems; i++) { | ||
| 1491 | u64 disk_bytenr; | ||
| 1492 | btrfs_item_key_to_cpu(buf, &key, i); | ||
| 1493 | if (btrfs_key_type(&key) != BTRFS_EXTENT_DATA_KEY) | ||
| 1494 | continue; | ||
| 1495 | fi = btrfs_item_ptr(buf, i, | ||
| 1496 | struct btrfs_file_extent_item); | ||
| 1497 | if (btrfs_file_extent_type(buf, fi) == | ||
| 1498 | BTRFS_FILE_EXTENT_INLINE) | ||
| 1499 | continue; | ||
| 1500 | disk_bytenr = btrfs_file_extent_disk_bytenr(buf, fi); | ||
| 1501 | if (disk_bytenr == 0) | ||
| 1502 | continue; | ||
| 1503 | |||
| 1504 | info->bytenr = disk_bytenr; | ||
| 1505 | info->num_bytes = | ||
| 1506 | btrfs_file_extent_disk_num_bytes(buf, fi); | ||
| 1507 | info->objectid = key.objectid; | ||
| 1508 | info->offset = key.offset; | ||
| 1509 | info++; | ||
| 1510 | } | ||
| 1511 | |||
| 1512 | ret = btrfs_add_leaf_ref(root, ref, shared); | ||
| 1513 | if (ret == -EEXIST && shared) { | ||
| 1514 | struct btrfs_leaf_ref *old; | ||
| 1515 | old = btrfs_lookup_leaf_ref(root, ref->bytenr); | ||
| 1516 | BUG_ON(!old); | ||
| 1517 | btrfs_remove_leaf_ref(root, old); | ||
| 1518 | btrfs_free_leaf_ref(root, old); | ||
| 1519 | ret = btrfs_add_leaf_ref(root, ref, shared); | ||
| 1520 | } | ||
| 1521 | WARN_ON(ret); | ||
| 1522 | btrfs_free_leaf_ref(root, ref); | ||
| 1523 | } | ||
| 1524 | out: | ||
| 1525 | return ret; | ||
| 1526 | } | ||
| 1527 | |||
| 1528 | int btrfs_inc_ref(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 1529 | struct extent_buffer *orig_buf, struct extent_buffer *buf, | ||
| 1530 | u32 *nr_extents) | ||
| 1531 | { | ||
| 1532 | u64 bytenr; | ||
| 1533 | u64 ref_root; | ||
| 1534 | u64 orig_root; | ||
| 1535 | u64 ref_generation; | ||
| 1536 | u64 orig_generation; | ||
| 1537 | u32 nritems; | ||
| 1538 | u32 nr_file_extents = 0; | ||
| 1539 | struct btrfs_key key; | ||
| 1540 | struct btrfs_file_extent_item *fi; | ||
| 1541 | int i; | ||
| 1542 | int level; | ||
| 1543 | int ret = 0; | ||
| 1544 | int faili = 0; | ||
| 1545 | int (*process_func)(struct btrfs_trans_handle *, struct btrfs_root *, | ||
| 1546 | u64, u64, u64, u64, u64, u64, u64, u64); | ||
| 1547 | |||
| 1548 | ref_root = btrfs_header_owner(buf); | ||
| 1549 | ref_generation = btrfs_header_generation(buf); | ||
| 1550 | orig_root = btrfs_header_owner(orig_buf); | ||
| 1551 | orig_generation = btrfs_header_generation(orig_buf); | ||
| 1552 | |||
| 1553 | nritems = btrfs_header_nritems(buf); | ||
| 1554 | level = btrfs_header_level(buf); | ||
| 1555 | |||
| 1556 | if (root->ref_cows) { | ||
| 1557 | process_func = __btrfs_inc_extent_ref; | ||
| 1558 | } else { | ||
| 1559 | if (level == 0 && | ||
| 1560 | root->root_key.objectid != BTRFS_TREE_LOG_OBJECTID) | ||
| 1561 | goto out; | ||
| 1562 | if (level != 0 && | ||
| 1563 | root->root_key.objectid == BTRFS_TREE_LOG_OBJECTID) | ||
| 1564 | goto out; | ||
| 1565 | process_func = __btrfs_update_extent_ref; | ||
| 1566 | } | ||
| 1567 | |||
| 1568 | for (i = 0; i < nritems; i++) { | ||
| 1569 | cond_resched(); | ||
| 1570 | if (level == 0) { | ||
| 1571 | btrfs_item_key_to_cpu(buf, &key, i); | ||
| 1572 | if (btrfs_key_type(&key) != BTRFS_EXTENT_DATA_KEY) | ||
| 1573 | continue; | ||
| 1574 | fi = btrfs_item_ptr(buf, i, | ||
| 1575 | struct btrfs_file_extent_item); | ||
| 1576 | if (btrfs_file_extent_type(buf, fi) == | ||
| 1577 | BTRFS_FILE_EXTENT_INLINE) | ||
| 1578 | continue; | ||
| 1579 | bytenr = btrfs_file_extent_disk_bytenr(buf, fi); | ||
| 1580 | if (bytenr == 0) | ||
| 1581 | continue; | ||
| 1582 | |||
| 1583 | nr_file_extents++; | ||
| 1584 | |||
| 1585 | ret = process_func(trans, root, bytenr, | ||
| 1586 | orig_buf->start, buf->start, | ||
| 1587 | orig_root, ref_root, | ||
| 1588 | orig_generation, ref_generation, | ||
| 1589 | key.objectid); | ||
| 1590 | |||
| 1591 | if (ret) { | ||
| 1592 | faili = i; | ||
| 1593 | WARN_ON(1); | ||
| 1594 | goto fail; | ||
| 1595 | } | ||
| 1596 | } else { | ||
| 1597 | bytenr = btrfs_node_blockptr(buf, i); | ||
| 1598 | ret = process_func(trans, root, bytenr, | ||
| 1599 | orig_buf->start, buf->start, | ||
| 1600 | orig_root, ref_root, | ||
| 1601 | orig_generation, ref_generation, | ||
| 1602 | level - 1); | ||
| 1603 | if (ret) { | ||
| 1604 | faili = i; | ||
| 1605 | WARN_ON(1); | ||
| 1606 | goto fail; | ||
| 1607 | } | ||
| 1608 | } | ||
| 1609 | } | ||
| 1610 | out: | ||
| 1611 | if (nr_extents) { | ||
| 1612 | if (level == 0) | ||
| 1613 | *nr_extents = nr_file_extents; | ||
| 1614 | else | ||
| 1615 | *nr_extents = nritems; | ||
| 1616 | } | ||
| 1617 | return 0; | ||
| 1618 | fail: | ||
| 1619 | WARN_ON(1); | ||
| 1620 | return ret; | ||
| 1621 | } | ||
| 1622 | |||
| 1623 | int btrfs_update_ref(struct btrfs_trans_handle *trans, | ||
| 1624 | struct btrfs_root *root, struct extent_buffer *orig_buf, | ||
| 1625 | struct extent_buffer *buf, int start_slot, int nr) | ||
| 1626 | |||
| 1627 | { | ||
| 1628 | u64 bytenr; | ||
| 1629 | u64 ref_root; | ||
| 1630 | u64 orig_root; | ||
| 1631 | u64 ref_generation; | ||
| 1632 | u64 orig_generation; | ||
| 1633 | struct btrfs_key key; | ||
| 1634 | struct btrfs_file_extent_item *fi; | ||
| 1635 | int i; | ||
| 1636 | int ret; | ||
| 1637 | int slot; | ||
| 1638 | int level; | ||
| 1639 | |||
| 1640 | BUG_ON(start_slot < 0); | ||
| 1641 | BUG_ON(start_slot + nr > btrfs_header_nritems(buf)); | ||
| 1642 | |||
| 1643 | ref_root = btrfs_header_owner(buf); | ||
| 1644 | ref_generation = btrfs_header_generation(buf); | ||
| 1645 | orig_root = btrfs_header_owner(orig_buf); | ||
| 1646 | orig_generation = btrfs_header_generation(orig_buf); | ||
| 1647 | level = btrfs_header_level(buf); | ||
| 1648 | |||
| 1649 | if (!root->ref_cows) { | ||
| 1650 | if (level == 0 && | ||
| 1651 | root->root_key.objectid != BTRFS_TREE_LOG_OBJECTID) | ||
| 1652 | return 0; | ||
| 1653 | if (level != 0 && | ||
| 1654 | root->root_key.objectid == BTRFS_TREE_LOG_OBJECTID) | ||
| 1655 | return 0; | ||
| 1656 | } | ||
| 1657 | |||
| 1658 | for (i = 0, slot = start_slot; i < nr; i++, slot++) { | ||
| 1659 | cond_resched(); | ||
| 1660 | if (level == 0) { | ||
| 1661 | btrfs_item_key_to_cpu(buf, &key, slot); | ||
| 1662 | if (btrfs_key_type(&key) != BTRFS_EXTENT_DATA_KEY) | ||
| 1663 | continue; | ||
| 1664 | fi = btrfs_item_ptr(buf, slot, | ||
| 1665 | struct btrfs_file_extent_item); | ||
| 1666 | if (btrfs_file_extent_type(buf, fi) == | ||
| 1667 | BTRFS_FILE_EXTENT_INLINE) | ||
| 1668 | continue; | ||
| 1669 | bytenr = btrfs_file_extent_disk_bytenr(buf, fi); | ||
| 1670 | if (bytenr == 0) | ||
| 1671 | continue; | ||
| 1672 | ret = __btrfs_update_extent_ref(trans, root, bytenr, | ||
| 1673 | orig_buf->start, buf->start, | ||
| 1674 | orig_root, ref_root, | ||
| 1675 | orig_generation, ref_generation, | ||
| 1676 | key.objectid); | ||
| 1677 | if (ret) | ||
| 1678 | goto fail; | ||
| 1679 | } else { | ||
| 1680 | bytenr = btrfs_node_blockptr(buf, slot); | ||
| 1681 | ret = __btrfs_update_extent_ref(trans, root, bytenr, | ||
| 1682 | orig_buf->start, buf->start, | ||
| 1683 | orig_root, ref_root, | ||
| 1684 | orig_generation, ref_generation, | ||
| 1685 | level - 1); | ||
| 1686 | if (ret) | ||
| 1687 | goto fail; | ||
| 1688 | } | ||
| 1689 | } | ||
| 1690 | return 0; | ||
| 1691 | fail: | ||
| 1692 | WARN_ON(1); | ||
| 1693 | return -1; | ||
| 1694 | } | ||
| 1695 | |||
| 1696 | static int write_one_cache_group(struct btrfs_trans_handle *trans, | ||
| 1697 | struct btrfs_root *root, | ||
| 1698 | struct btrfs_path *path, | ||
| 1699 | struct btrfs_block_group_cache *cache) | ||
| 1700 | { | ||
| 1701 | int ret; | ||
| 1702 | int pending_ret; | ||
| 1703 | struct btrfs_root *extent_root = root->fs_info->extent_root; | ||
| 1704 | unsigned long bi; | ||
| 1705 | struct extent_buffer *leaf; | ||
| 1706 | |||
| 1707 | ret = btrfs_search_slot(trans, extent_root, &cache->key, path, 0, 1); | ||
| 1708 | if (ret < 0) | ||
| 1709 | goto fail; | ||
| 1710 | BUG_ON(ret); | ||
| 1711 | |||
| 1712 | leaf = path->nodes[0]; | ||
| 1713 | bi = btrfs_item_ptr_offset(leaf, path->slots[0]); | ||
| 1714 | write_extent_buffer(leaf, &cache->item, bi, sizeof(cache->item)); | ||
| 1715 | btrfs_mark_buffer_dirty(leaf); | ||
| 1716 | btrfs_release_path(extent_root, path); | ||
| 1717 | fail: | ||
| 1718 | finish_current_insert(trans, extent_root, 0); | ||
| 1719 | pending_ret = del_pending_extents(trans, extent_root, 0); | ||
| 1720 | if (ret) | ||
| 1721 | return ret; | ||
| 1722 | if (pending_ret) | ||
| 1723 | return pending_ret; | ||
| 1724 | return 0; | ||
| 1725 | |||
| 1726 | } | ||
| 1727 | |||
| 1728 | int btrfs_write_dirty_block_groups(struct btrfs_trans_handle *trans, | ||
| 1729 | struct btrfs_root *root) | ||
| 1730 | { | ||
| 1731 | struct btrfs_block_group_cache *cache, *entry; | ||
| 1732 | struct rb_node *n; | ||
| 1733 | int err = 0; | ||
| 1734 | int werr = 0; | ||
| 1735 | struct btrfs_path *path; | ||
| 1736 | u64 last = 0; | ||
| 1737 | |||
| 1738 | path = btrfs_alloc_path(); | ||
| 1739 | if (!path) | ||
| 1740 | return -ENOMEM; | ||
| 1741 | |||
| 1742 | while (1) { | ||
| 1743 | cache = NULL; | ||
| 1744 | spin_lock(&root->fs_info->block_group_cache_lock); | ||
| 1745 | for (n = rb_first(&root->fs_info->block_group_cache_tree); | ||
| 1746 | n; n = rb_next(n)) { | ||
| 1747 | entry = rb_entry(n, struct btrfs_block_group_cache, | ||
| 1748 | cache_node); | ||
| 1749 | if (entry->dirty) { | ||
| 1750 | cache = entry; | ||
| 1751 | break; | ||
| 1752 | } | ||
| 1753 | } | ||
| 1754 | spin_unlock(&root->fs_info->block_group_cache_lock); | ||
| 1755 | |||
| 1756 | if (!cache) | ||
| 1757 | break; | ||
| 1758 | |||
| 1759 | cache->dirty = 0; | ||
| 1760 | last += cache->key.offset; | ||
| 1761 | |||
| 1762 | err = write_one_cache_group(trans, root, | ||
| 1763 | path, cache); | ||
| 1764 | /* | ||
| 1765 | * if we fail to write the cache group, we want | ||
| 1766 | * to keep it marked dirty in hopes that a later | ||
| 1767 | * write will work | ||
| 1768 | */ | ||
| 1769 | if (err) { | ||
| 1770 | werr = err; | ||
| 1771 | continue; | ||
| 1772 | } | ||
| 1773 | } | ||
| 1774 | btrfs_free_path(path); | ||
| 1775 | return werr; | ||
| 1776 | } | ||
| 1777 | |||
| 1778 | int btrfs_extent_readonly(struct btrfs_root *root, u64 bytenr) | ||
| 1779 | { | ||
| 1780 | struct btrfs_block_group_cache *block_group; | ||
| 1781 | int readonly = 0; | ||
| 1782 | |||
| 1783 | block_group = btrfs_lookup_block_group(root->fs_info, bytenr); | ||
| 1784 | if (!block_group || block_group->ro) | ||
| 1785 | readonly = 1; | ||
| 1786 | if (block_group) | ||
| 1787 | put_block_group(block_group); | ||
| 1788 | return readonly; | ||
| 1789 | } | ||
| 1790 | |||
| 1791 | static int update_space_info(struct btrfs_fs_info *info, u64 flags, | ||
| 1792 | u64 total_bytes, u64 bytes_used, | ||
| 1793 | struct btrfs_space_info **space_info) | ||
| 1794 | { | ||
| 1795 | struct btrfs_space_info *found; | ||
| 1796 | |||
| 1797 | found = __find_space_info(info, flags); | ||
| 1798 | if (found) { | ||
| 1799 | spin_lock(&found->lock); | ||
| 1800 | found->total_bytes += total_bytes; | ||
| 1801 | found->bytes_used += bytes_used; | ||
| 1802 | found->full = 0; | ||
| 1803 | spin_unlock(&found->lock); | ||
| 1804 | *space_info = found; | ||
| 1805 | return 0; | ||
| 1806 | } | ||
| 1807 | found = kzalloc(sizeof(*found), GFP_NOFS); | ||
| 1808 | if (!found) | ||
| 1809 | return -ENOMEM; | ||
| 1810 | |||
| 1811 | list_add(&found->list, &info->space_info); | ||
| 1812 | INIT_LIST_HEAD(&found->block_groups); | ||
| 1813 | init_rwsem(&found->groups_sem); | ||
| 1814 | spin_lock_init(&found->lock); | ||
| 1815 | found->flags = flags; | ||
| 1816 | found->total_bytes = total_bytes; | ||
| 1817 | found->bytes_used = bytes_used; | ||
| 1818 | found->bytes_pinned = 0; | ||
| 1819 | found->bytes_reserved = 0; | ||
| 1820 | found->bytes_readonly = 0; | ||
| 1821 | found->full = 0; | ||
| 1822 | found->force_alloc = 0; | ||
| 1823 | *space_info = found; | ||
| 1824 | return 0; | ||
| 1825 | } | ||
| 1826 | |||
| 1827 | static void set_avail_alloc_bits(struct btrfs_fs_info *fs_info, u64 flags) | ||
| 1828 | { | ||
| 1829 | u64 extra_flags = flags & (BTRFS_BLOCK_GROUP_RAID0 | | ||
| 1830 | BTRFS_BLOCK_GROUP_RAID1 | | ||
| 1831 | BTRFS_BLOCK_GROUP_RAID10 | | ||
| 1832 | BTRFS_BLOCK_GROUP_DUP); | ||
| 1833 | if (extra_flags) { | ||
| 1834 | if (flags & BTRFS_BLOCK_GROUP_DATA) | ||
| 1835 | fs_info->avail_data_alloc_bits |= extra_flags; | ||
| 1836 | if (flags & BTRFS_BLOCK_GROUP_METADATA) | ||
| 1837 | fs_info->avail_metadata_alloc_bits |= extra_flags; | ||
| 1838 | if (flags & BTRFS_BLOCK_GROUP_SYSTEM) | ||
| 1839 | fs_info->avail_system_alloc_bits |= extra_flags; | ||
| 1840 | } | ||
| 1841 | } | ||
| 1842 | |||
| 1843 | static void set_block_group_readonly(struct btrfs_block_group_cache *cache) | ||
| 1844 | { | ||
| 1845 | spin_lock(&cache->space_info->lock); | ||
| 1846 | spin_lock(&cache->lock); | ||
| 1847 | if (!cache->ro) { | ||
| 1848 | cache->space_info->bytes_readonly += cache->key.offset - | ||
| 1849 | btrfs_block_group_used(&cache->item); | ||
| 1850 | cache->ro = 1; | ||
| 1851 | } | ||
| 1852 | spin_unlock(&cache->lock); | ||
| 1853 | spin_unlock(&cache->space_info->lock); | ||
| 1854 | } | ||
| 1855 | |||
| 1856 | u64 btrfs_reduce_alloc_profile(struct btrfs_root *root, u64 flags) | ||
| 1857 | { | ||
| 1858 | u64 num_devices = root->fs_info->fs_devices->rw_devices; | ||
| 1859 | |||
| 1860 | if (num_devices == 1) | ||
| 1861 | flags &= ~(BTRFS_BLOCK_GROUP_RAID1 | BTRFS_BLOCK_GROUP_RAID0); | ||
| 1862 | if (num_devices < 4) | ||
| 1863 | flags &= ~BTRFS_BLOCK_GROUP_RAID10; | ||
| 1864 | |||
| 1865 | if ((flags & BTRFS_BLOCK_GROUP_DUP) && | ||
| 1866 | (flags & (BTRFS_BLOCK_GROUP_RAID1 | | ||
| 1867 | BTRFS_BLOCK_GROUP_RAID10))) { | ||
| 1868 | flags &= ~BTRFS_BLOCK_GROUP_DUP; | ||
| 1869 | } | ||
| 1870 | |||
| 1871 | if ((flags & BTRFS_BLOCK_GROUP_RAID1) && | ||
| 1872 | (flags & BTRFS_BLOCK_GROUP_RAID10)) { | ||
| 1873 | flags &= ~BTRFS_BLOCK_GROUP_RAID1; | ||
| 1874 | } | ||
| 1875 | |||
| 1876 | if ((flags & BTRFS_BLOCK_GROUP_RAID0) && | ||
| 1877 | ((flags & BTRFS_BLOCK_GROUP_RAID1) | | ||
| 1878 | (flags & BTRFS_BLOCK_GROUP_RAID10) | | ||
| 1879 | (flags & BTRFS_BLOCK_GROUP_DUP))) | ||
| 1880 | flags &= ~BTRFS_BLOCK_GROUP_RAID0; | ||
| 1881 | return flags; | ||
| 1882 | } | ||
| 1883 | |||
| 1884 | static int do_chunk_alloc(struct btrfs_trans_handle *trans, | ||
| 1885 | struct btrfs_root *extent_root, u64 alloc_bytes, | ||
| 1886 | u64 flags, int force) | ||
| 1887 | { | ||
| 1888 | struct btrfs_space_info *space_info; | ||
| 1889 | u64 thresh; | ||
| 1890 | int ret = 0; | ||
| 1891 | |||
| 1892 | mutex_lock(&extent_root->fs_info->chunk_mutex); | ||
| 1893 | |||
| 1894 | flags = btrfs_reduce_alloc_profile(extent_root, flags); | ||
| 1895 | |||
| 1896 | space_info = __find_space_info(extent_root->fs_info, flags); | ||
| 1897 | if (!space_info) { | ||
| 1898 | ret = update_space_info(extent_root->fs_info, flags, | ||
| 1899 | 0, 0, &space_info); | ||
| 1900 | BUG_ON(ret); | ||
| 1901 | } | ||
| 1902 | BUG_ON(!space_info); | ||
| 1903 | |||
| 1904 | spin_lock(&space_info->lock); | ||
| 1905 | if (space_info->force_alloc) { | ||
| 1906 | force = 1; | ||
| 1907 | space_info->force_alloc = 0; | ||
| 1908 | } | ||
| 1909 | if (space_info->full) { | ||
| 1910 | spin_unlock(&space_info->lock); | ||
| 1911 | goto out; | ||
| 1912 | } | ||
| 1913 | |||
| 1914 | thresh = space_info->total_bytes - space_info->bytes_readonly; | ||
| 1915 | thresh = div_factor(thresh, 6); | ||
| 1916 | if (!force && | ||
| 1917 | (space_info->bytes_used + space_info->bytes_pinned + | ||
| 1918 | space_info->bytes_reserved + alloc_bytes) < thresh) { | ||
| 1919 | spin_unlock(&space_info->lock); | ||
| 1920 | goto out; | ||
| 1921 | } | ||
| 1922 | spin_unlock(&space_info->lock); | ||
| 1923 | |||
| 1924 | ret = btrfs_alloc_chunk(trans, extent_root, flags); | ||
| 1925 | if (ret) | ||
| 1926 | space_info->full = 1; | ||
| 1927 | out: | ||
| 1928 | mutex_unlock(&extent_root->fs_info->chunk_mutex); | ||
| 1929 | return ret; | ||
| 1930 | } | ||
| 1931 | |||
| 1932 | static int update_block_group(struct btrfs_trans_handle *trans, | ||
| 1933 | struct btrfs_root *root, | ||
| 1934 | u64 bytenr, u64 num_bytes, int alloc, | ||
| 1935 | int mark_free) | ||
| 1936 | { | ||
| 1937 | struct btrfs_block_group_cache *cache; | ||
| 1938 | struct btrfs_fs_info *info = root->fs_info; | ||
| 1939 | u64 total = num_bytes; | ||
| 1940 | u64 old_val; | ||
| 1941 | u64 byte_in_group; | ||
| 1942 | |||
| 1943 | while (total) { | ||
| 1944 | cache = btrfs_lookup_block_group(info, bytenr); | ||
| 1945 | if (!cache) | ||
| 1946 | return -1; | ||
| 1947 | byte_in_group = bytenr - cache->key.objectid; | ||
| 1948 | WARN_ON(byte_in_group > cache->key.offset); | ||
| 1949 | |||
| 1950 | spin_lock(&cache->space_info->lock); | ||
| 1951 | spin_lock(&cache->lock); | ||
| 1952 | cache->dirty = 1; | ||
| 1953 | old_val = btrfs_block_group_used(&cache->item); | ||
| 1954 | num_bytes = min(total, cache->key.offset - byte_in_group); | ||
| 1955 | if (alloc) { | ||
| 1956 | old_val += num_bytes; | ||
| 1957 | cache->space_info->bytes_used += num_bytes; | ||
| 1958 | if (cache->ro) | ||
| 1959 | cache->space_info->bytes_readonly -= num_bytes; | ||
| 1960 | btrfs_set_block_group_used(&cache->item, old_val); | ||
| 1961 | spin_unlock(&cache->lock); | ||
| 1962 | spin_unlock(&cache->space_info->lock); | ||
| 1963 | } else { | ||
| 1964 | old_val -= num_bytes; | ||
| 1965 | cache->space_info->bytes_used -= num_bytes; | ||
| 1966 | if (cache->ro) | ||
| 1967 | cache->space_info->bytes_readonly += num_bytes; | ||
| 1968 | btrfs_set_block_group_used(&cache->item, old_val); | ||
| 1969 | spin_unlock(&cache->lock); | ||
| 1970 | spin_unlock(&cache->space_info->lock); | ||
| 1971 | if (mark_free) { | ||
| 1972 | int ret; | ||
| 1973 | |||
| 1974 | ret = btrfs_discard_extent(root, bytenr, | ||
| 1975 | num_bytes); | ||
| 1976 | WARN_ON(ret); | ||
| 1977 | |||
| 1978 | ret = btrfs_add_free_space(cache, bytenr, | ||
| 1979 | num_bytes); | ||
| 1980 | WARN_ON(ret); | ||
| 1981 | } | ||
| 1982 | } | ||
| 1983 | put_block_group(cache); | ||
| 1984 | total -= num_bytes; | ||
| 1985 | bytenr += num_bytes; | ||
| 1986 | } | ||
| 1987 | return 0; | ||
| 1988 | } | ||
| 1989 | |||
| 1990 | static u64 first_logical_byte(struct btrfs_root *root, u64 search_start) | ||
| 1991 | { | ||
| 1992 | struct btrfs_block_group_cache *cache; | ||
| 1993 | u64 bytenr; | ||
| 1994 | |||
| 1995 | cache = btrfs_lookup_first_block_group(root->fs_info, search_start); | ||
| 1996 | if (!cache) | ||
| 1997 | return 0; | ||
| 1998 | |||
| 1999 | bytenr = cache->key.objectid; | ||
| 2000 | put_block_group(cache); | ||
| 2001 | |||
| 2002 | return bytenr; | ||
| 2003 | } | ||
| 2004 | |||
| 2005 | int btrfs_update_pinned_extents(struct btrfs_root *root, | ||
| 2006 | u64 bytenr, u64 num, int pin) | ||
| 2007 | { | ||
| 2008 | u64 len; | ||
| 2009 | struct btrfs_block_group_cache *cache; | ||
| 2010 | struct btrfs_fs_info *fs_info = root->fs_info; | ||
| 2011 | |||
| 2012 | WARN_ON(!mutex_is_locked(&root->fs_info->pinned_mutex)); | ||
| 2013 | if (pin) { | ||
| 2014 | set_extent_dirty(&fs_info->pinned_extents, | ||
| 2015 | bytenr, bytenr + num - 1, GFP_NOFS); | ||
| 2016 | } else { | ||
| 2017 | clear_extent_dirty(&fs_info->pinned_extents, | ||
| 2018 | bytenr, bytenr + num - 1, GFP_NOFS); | ||
| 2019 | } | ||
| 2020 | while (num > 0) { | ||
| 2021 | cache = btrfs_lookup_block_group(fs_info, bytenr); | ||
| 2022 | BUG_ON(!cache); | ||
| 2023 | len = min(num, cache->key.offset - | ||
| 2024 | (bytenr - cache->key.objectid)); | ||
| 2025 | if (pin) { | ||
| 2026 | spin_lock(&cache->space_info->lock); | ||
| 2027 | spin_lock(&cache->lock); | ||
| 2028 | cache->pinned += len; | ||
| 2029 | cache->space_info->bytes_pinned += len; | ||
| 2030 | spin_unlock(&cache->lock); | ||
| 2031 | spin_unlock(&cache->space_info->lock); | ||
| 2032 | fs_info->total_pinned += len; | ||
| 2033 | } else { | ||
| 2034 | spin_lock(&cache->space_info->lock); | ||
| 2035 | spin_lock(&cache->lock); | ||
| 2036 | cache->pinned -= len; | ||
| 2037 | cache->space_info->bytes_pinned -= len; | ||
| 2038 | spin_unlock(&cache->lock); | ||
| 2039 | spin_unlock(&cache->space_info->lock); | ||
| 2040 | fs_info->total_pinned -= len; | ||
| 2041 | if (cache->cached) | ||
| 2042 | btrfs_add_free_space(cache, bytenr, len); | ||
| 2043 | } | ||
| 2044 | put_block_group(cache); | ||
| 2045 | bytenr += len; | ||
| 2046 | num -= len; | ||
| 2047 | } | ||
| 2048 | return 0; | ||
| 2049 | } | ||
| 2050 | |||
| 2051 | static int update_reserved_extents(struct btrfs_root *root, | ||
| 2052 | u64 bytenr, u64 num, int reserve) | ||
| 2053 | { | ||
| 2054 | u64 len; | ||
| 2055 | struct btrfs_block_group_cache *cache; | ||
| 2056 | struct btrfs_fs_info *fs_info = root->fs_info; | ||
| 2057 | |||
| 2058 | while (num > 0) { | ||
| 2059 | cache = btrfs_lookup_block_group(fs_info, bytenr); | ||
| 2060 | BUG_ON(!cache); | ||
| 2061 | len = min(num, cache->key.offset - | ||
| 2062 | (bytenr - cache->key.objectid)); | ||
| 2063 | |||
| 2064 | spin_lock(&cache->space_info->lock); | ||
| 2065 | spin_lock(&cache->lock); | ||
| 2066 | if (reserve) { | ||
| 2067 | cache->reserved += len; | ||
| 2068 | cache->space_info->bytes_reserved += len; | ||
| 2069 | } else { | ||
| 2070 | cache->reserved -= len; | ||
| 2071 | cache->space_info->bytes_reserved -= len; | ||
| 2072 | } | ||
| 2073 | spin_unlock(&cache->lock); | ||
| 2074 | spin_unlock(&cache->space_info->lock); | ||
| 2075 | put_block_group(cache); | ||
| 2076 | bytenr += len; | ||
| 2077 | num -= len; | ||
| 2078 | } | ||
| 2079 | return 0; | ||
| 2080 | } | ||
| 2081 | |||
| 2082 | int btrfs_copy_pinned(struct btrfs_root *root, struct extent_io_tree *copy) | ||
| 2083 | { | ||
| 2084 | u64 last = 0; | ||
| 2085 | u64 start; | ||
| 2086 | u64 end; | ||
| 2087 | struct extent_io_tree *pinned_extents = &root->fs_info->pinned_extents; | ||
| 2088 | int ret; | ||
| 2089 | |||
| 2090 | mutex_lock(&root->fs_info->pinned_mutex); | ||
| 2091 | while (1) { | ||
| 2092 | ret = find_first_extent_bit(pinned_extents, last, | ||
| 2093 | &start, &end, EXTENT_DIRTY); | ||
| 2094 | if (ret) | ||
| 2095 | break; | ||
| 2096 | set_extent_dirty(copy, start, end, GFP_NOFS); | ||
| 2097 | last = end + 1; | ||
| 2098 | } | ||
| 2099 | mutex_unlock(&root->fs_info->pinned_mutex); | ||
| 2100 | return 0; | ||
| 2101 | } | ||
| 2102 | |||
| 2103 | int btrfs_finish_extent_commit(struct btrfs_trans_handle *trans, | ||
| 2104 | struct btrfs_root *root, | ||
| 2105 | struct extent_io_tree *unpin) | ||
| 2106 | { | ||
| 2107 | u64 start; | ||
| 2108 | u64 end; | ||
| 2109 | int ret; | ||
| 2110 | |||
| 2111 | mutex_lock(&root->fs_info->pinned_mutex); | ||
| 2112 | while (1) { | ||
| 2113 | ret = find_first_extent_bit(unpin, 0, &start, &end, | ||
| 2114 | EXTENT_DIRTY); | ||
| 2115 | if (ret) | ||
| 2116 | break; | ||
| 2117 | |||
| 2118 | ret = btrfs_discard_extent(root, start, end + 1 - start); | ||
| 2119 | |||
| 2120 | btrfs_update_pinned_extents(root, start, end + 1 - start, 0); | ||
| 2121 | clear_extent_dirty(unpin, start, end, GFP_NOFS); | ||
| 2122 | |||
| 2123 | if (need_resched()) { | ||
| 2124 | mutex_unlock(&root->fs_info->pinned_mutex); | ||
| 2125 | cond_resched(); | ||
| 2126 | mutex_lock(&root->fs_info->pinned_mutex); | ||
| 2127 | } | ||
| 2128 | } | ||
| 2129 | mutex_unlock(&root->fs_info->pinned_mutex); | ||
| 2130 | return ret; | ||
| 2131 | } | ||
| 2132 | |||
| 2133 | static int finish_current_insert(struct btrfs_trans_handle *trans, | ||
| 2134 | struct btrfs_root *extent_root, int all) | ||
| 2135 | { | ||
| 2136 | u64 start; | ||
| 2137 | u64 end; | ||
| 2138 | u64 priv; | ||
| 2139 | u64 search = 0; | ||
| 2140 | u64 skipped = 0; | ||
| 2141 | struct btrfs_fs_info *info = extent_root->fs_info; | ||
| 2142 | struct btrfs_path *path; | ||
| 2143 | struct pending_extent_op *extent_op, *tmp; | ||
| 2144 | struct list_head insert_list, update_list; | ||
| 2145 | int ret; | ||
| 2146 | int num_inserts = 0, max_inserts; | ||
| 2147 | |||
| 2148 | path = btrfs_alloc_path(); | ||
| 2149 | INIT_LIST_HEAD(&insert_list); | ||
| 2150 | INIT_LIST_HEAD(&update_list); | ||
| 2151 | |||
| 2152 | max_inserts = extent_root->leafsize / | ||
| 2153 | (2 * sizeof(struct btrfs_key) + 2 * sizeof(struct btrfs_item) + | ||
| 2154 | sizeof(struct btrfs_extent_ref) + | ||
| 2155 | sizeof(struct btrfs_extent_item)); | ||
| 2156 | again: | ||
| 2157 | mutex_lock(&info->extent_ins_mutex); | ||
| 2158 | while (1) { | ||
| 2159 | ret = find_first_extent_bit(&info->extent_ins, search, &start, | ||
| 2160 | &end, EXTENT_WRITEBACK); | ||
| 2161 | if (ret) { | ||
| 2162 | if (skipped && all && !num_inserts) { | ||
| 2163 | skipped = 0; | ||
| 2164 | search = 0; | ||
| 2165 | continue; | ||
| 2166 | } | ||
| 2167 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2168 | break; | ||
| 2169 | } | ||
| 2170 | |||
| 2171 | ret = try_lock_extent(&info->extent_ins, start, end, GFP_NOFS); | ||
| 2172 | if (!ret) { | ||
| 2173 | skipped = 1; | ||
| 2174 | search = end + 1; | ||
| 2175 | if (need_resched()) { | ||
| 2176 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2177 | cond_resched(); | ||
| 2178 | mutex_lock(&info->extent_ins_mutex); | ||
| 2179 | } | ||
| 2180 | continue; | ||
| 2181 | } | ||
| 2182 | |||
| 2183 | ret = get_state_private(&info->extent_ins, start, &priv); | ||
| 2184 | BUG_ON(ret); | ||
| 2185 | extent_op = (struct pending_extent_op *)(unsigned long) priv; | ||
| 2186 | |||
| 2187 | if (extent_op->type == PENDING_EXTENT_INSERT) { | ||
| 2188 | num_inserts++; | ||
| 2189 | list_add_tail(&extent_op->list, &insert_list); | ||
| 2190 | search = end + 1; | ||
| 2191 | if (num_inserts == max_inserts) { | ||
| 2192 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2193 | break; | ||
| 2194 | } | ||
| 2195 | } else if (extent_op->type == PENDING_BACKREF_UPDATE) { | ||
| 2196 | list_add_tail(&extent_op->list, &update_list); | ||
| 2197 | search = end + 1; | ||
| 2198 | } else { | ||
| 2199 | BUG(); | ||
| 2200 | } | ||
| 2201 | } | ||
| 2202 | |||
| 2203 | /* | ||
| 2204 | * process the update list, clear the writeback bit for it, and if | ||
| 2205 | * somebody marked this thing for deletion then just unlock it and be | ||
| 2206 | * done, the free_extents will handle it | ||
| 2207 | */ | ||
| 2208 | mutex_lock(&info->extent_ins_mutex); | ||
| 2209 | list_for_each_entry_safe(extent_op, tmp, &update_list, list) { | ||
| 2210 | clear_extent_bits(&info->extent_ins, extent_op->bytenr, | ||
| 2211 | extent_op->bytenr + extent_op->num_bytes - 1, | ||
| 2212 | EXTENT_WRITEBACK, GFP_NOFS); | ||
| 2213 | if (extent_op->del) { | ||
| 2214 | list_del_init(&extent_op->list); | ||
| 2215 | unlock_extent(&info->extent_ins, extent_op->bytenr, | ||
| 2216 | extent_op->bytenr + extent_op->num_bytes | ||
| 2217 | - 1, GFP_NOFS); | ||
| 2218 | kfree(extent_op); | ||
| 2219 | } | ||
| 2220 | } | ||
| 2221 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2222 | |||
| 2223 | /* | ||
| 2224 | * still have things left on the update list, go ahead an update | ||
| 2225 | * everything | ||
| 2226 | */ | ||
| 2227 | if (!list_empty(&update_list)) { | ||
| 2228 | ret = update_backrefs(trans, extent_root, path, &update_list); | ||
| 2229 | BUG_ON(ret); | ||
| 2230 | } | ||
| 2231 | |||
| 2232 | /* | ||
| 2233 | * if no inserts need to be done, but we skipped some extents and we | ||
| 2234 | * need to make sure everything is cleaned then reset everything and | ||
| 2235 | * go back to the beginning | ||
| 2236 | */ | ||
| 2237 | if (!num_inserts && all && skipped) { | ||
| 2238 | search = 0; | ||
| 2239 | skipped = 0; | ||
| 2240 | INIT_LIST_HEAD(&update_list); | ||
| 2241 | INIT_LIST_HEAD(&insert_list); | ||
| 2242 | goto again; | ||
| 2243 | } else if (!num_inserts) { | ||
| 2244 | goto out; | ||
| 2245 | } | ||
| 2246 | |||
| 2247 | /* | ||
| 2248 | * process the insert extents list. Again if we are deleting this | ||
| 2249 | * extent, then just unlock it, pin down the bytes if need be, and be | ||
| 2250 | * done with it. Saves us from having to actually insert the extent | ||
| 2251 | * into the tree and then subsequently come along and delete it | ||
| 2252 | */ | ||
| 2253 | mutex_lock(&info->extent_ins_mutex); | ||
| 2254 | list_for_each_entry_safe(extent_op, tmp, &insert_list, list) { | ||
| 2255 | clear_extent_bits(&info->extent_ins, extent_op->bytenr, | ||
| 2256 | extent_op->bytenr + extent_op->num_bytes - 1, | ||
| 2257 | EXTENT_WRITEBACK, GFP_NOFS); | ||
| 2258 | if (extent_op->del) { | ||
| 2259 | u64 used; | ||
| 2260 | list_del_init(&extent_op->list); | ||
| 2261 | unlock_extent(&info->extent_ins, extent_op->bytenr, | ||
| 2262 | extent_op->bytenr + extent_op->num_bytes | ||
| 2263 | - 1, GFP_NOFS); | ||
| 2264 | |||
| 2265 | mutex_lock(&extent_root->fs_info->pinned_mutex); | ||
| 2266 | ret = pin_down_bytes(trans, extent_root, | ||
| 2267 | extent_op->bytenr, | ||
| 2268 | extent_op->num_bytes, 0); | ||
| 2269 | mutex_unlock(&extent_root->fs_info->pinned_mutex); | ||
| 2270 | |||
| 2271 | spin_lock(&info->delalloc_lock); | ||
| 2272 | used = btrfs_super_bytes_used(&info->super_copy); | ||
| 2273 | btrfs_set_super_bytes_used(&info->super_copy, | ||
| 2274 | used - extent_op->num_bytes); | ||
| 2275 | used = btrfs_root_used(&extent_root->root_item); | ||
| 2276 | btrfs_set_root_used(&extent_root->root_item, | ||
| 2277 | used - extent_op->num_bytes); | ||
| 2278 | spin_unlock(&info->delalloc_lock); | ||
| 2279 | |||
| 2280 | ret = update_block_group(trans, extent_root, | ||
| 2281 | extent_op->bytenr, | ||
| 2282 | extent_op->num_bytes, | ||
| 2283 | 0, ret > 0); | ||
| 2284 | BUG_ON(ret); | ||
| 2285 | kfree(extent_op); | ||
| 2286 | num_inserts--; | ||
| 2287 | } | ||
| 2288 | } | ||
| 2289 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2290 | |||
| 2291 | ret = insert_extents(trans, extent_root, path, &insert_list, | ||
| 2292 | num_inserts); | ||
| 2293 | BUG_ON(ret); | ||
| 2294 | |||
| 2295 | /* | ||
| 2296 | * if we broke out of the loop in order to insert stuff because we hit | ||
| 2297 | * the maximum number of inserts at a time we can handle, then loop | ||
| 2298 | * back and pick up where we left off | ||
| 2299 | */ | ||
| 2300 | if (num_inserts == max_inserts) { | ||
| 2301 | INIT_LIST_HEAD(&insert_list); | ||
| 2302 | INIT_LIST_HEAD(&update_list); | ||
| 2303 | num_inserts = 0; | ||
| 2304 | goto again; | ||
| 2305 | } | ||
| 2306 | |||
| 2307 | /* | ||
| 2308 | * again, if we need to make absolutely sure there are no more pending | ||
| 2309 | * extent operations left and we know that we skipped some, go back to | ||
| 2310 | * the beginning and do it all again | ||
| 2311 | */ | ||
| 2312 | if (all && skipped) { | ||
| 2313 | INIT_LIST_HEAD(&insert_list); | ||
| 2314 | INIT_LIST_HEAD(&update_list); | ||
| 2315 | search = 0; | ||
| 2316 | skipped = 0; | ||
| 2317 | num_inserts = 0; | ||
| 2318 | goto again; | ||
| 2319 | } | ||
| 2320 | out: | ||
| 2321 | btrfs_free_path(path); | ||
| 2322 | return 0; | ||
| 2323 | } | ||
| 2324 | |||
| 2325 | static int pin_down_bytes(struct btrfs_trans_handle *trans, | ||
| 2326 | struct btrfs_root *root, | ||
| 2327 | u64 bytenr, u64 num_bytes, int is_data) | ||
| 2328 | { | ||
| 2329 | int err = 0; | ||
| 2330 | struct extent_buffer *buf; | ||
| 2331 | |||
| 2332 | if (is_data) | ||
| 2333 | goto pinit; | ||
| 2334 | |||
| 2335 | buf = btrfs_find_tree_block(root, bytenr, num_bytes); | ||
| 2336 | if (!buf) | ||
| 2337 | goto pinit; | ||
| 2338 | |||
| 2339 | /* we can reuse a block if it hasn't been written | ||
| 2340 | * and it is from this transaction. We can't | ||
| 2341 | * reuse anything from the tree log root because | ||
| 2342 | * it has tiny sub-transactions. | ||
| 2343 | */ | ||
| 2344 | if (btrfs_buffer_uptodate(buf, 0) && | ||
| 2345 | btrfs_try_tree_lock(buf)) { | ||
| 2346 | u64 header_owner = btrfs_header_owner(buf); | ||
| 2347 | u64 header_transid = btrfs_header_generation(buf); | ||
| 2348 | if (header_owner != BTRFS_TREE_LOG_OBJECTID && | ||
| 2349 | header_owner != BTRFS_TREE_RELOC_OBJECTID && | ||
| 2350 | header_transid == trans->transid && | ||
| 2351 | !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) { | ||
| 2352 | clean_tree_block(NULL, root, buf); | ||
| 2353 | btrfs_tree_unlock(buf); | ||
| 2354 | free_extent_buffer(buf); | ||
| 2355 | return 1; | ||
| 2356 | } | ||
| 2357 | btrfs_tree_unlock(buf); | ||
| 2358 | } | ||
| 2359 | free_extent_buffer(buf); | ||
| 2360 | pinit: | ||
| 2361 | btrfs_update_pinned_extents(root, bytenr, num_bytes, 1); | ||
| 2362 | |||
| 2363 | BUG_ON(err < 0); | ||
| 2364 | return 0; | ||
| 2365 | } | ||
| 2366 | |||
| 2367 | /* | ||
| 2368 | * remove an extent from the root, returns 0 on success | ||
| 2369 | */ | ||
| 2370 | static int __free_extent(struct btrfs_trans_handle *trans, | ||
| 2371 | struct btrfs_root *root, | ||
| 2372 | u64 bytenr, u64 num_bytes, u64 parent, | ||
| 2373 | u64 root_objectid, u64 ref_generation, | ||
| 2374 | u64 owner_objectid, int pin, int mark_free) | ||
| 2375 | { | ||
| 2376 | struct btrfs_path *path; | ||
| 2377 | struct btrfs_key key; | ||
| 2378 | struct btrfs_fs_info *info = root->fs_info; | ||
| 2379 | struct btrfs_root *extent_root = info->extent_root; | ||
| 2380 | struct extent_buffer *leaf; | ||
| 2381 | int ret; | ||
| 2382 | int extent_slot = 0; | ||
| 2383 | int found_extent = 0; | ||
| 2384 | int num_to_del = 1; | ||
| 2385 | struct btrfs_extent_item *ei; | ||
| 2386 | u32 refs; | ||
| 2387 | |||
| 2388 | key.objectid = bytenr; | ||
| 2389 | btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY); | ||
| 2390 | key.offset = num_bytes; | ||
| 2391 | path = btrfs_alloc_path(); | ||
| 2392 | if (!path) | ||
| 2393 | return -ENOMEM; | ||
| 2394 | |||
| 2395 | path->reada = 1; | ||
| 2396 | ret = lookup_extent_backref(trans, extent_root, path, | ||
| 2397 | bytenr, parent, root_objectid, | ||
| 2398 | ref_generation, owner_objectid, 1); | ||
| 2399 | if (ret == 0) { | ||
| 2400 | struct btrfs_key found_key; | ||
| 2401 | extent_slot = path->slots[0]; | ||
| 2402 | while (extent_slot > 0) { | ||
| 2403 | extent_slot--; | ||
| 2404 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 2405 | extent_slot); | ||
| 2406 | if (found_key.objectid != bytenr) | ||
| 2407 | break; | ||
| 2408 | if (found_key.type == BTRFS_EXTENT_ITEM_KEY && | ||
| 2409 | found_key.offset == num_bytes) { | ||
| 2410 | found_extent = 1; | ||
| 2411 | break; | ||
| 2412 | } | ||
| 2413 | if (path->slots[0] - extent_slot > 5) | ||
| 2414 | break; | ||
| 2415 | } | ||
| 2416 | if (!found_extent) { | ||
| 2417 | ret = remove_extent_backref(trans, extent_root, path); | ||
| 2418 | BUG_ON(ret); | ||
| 2419 | btrfs_release_path(extent_root, path); | ||
| 2420 | ret = btrfs_search_slot(trans, extent_root, | ||
| 2421 | &key, path, -1, 1); | ||
| 2422 | if (ret) { | ||
| 2423 | printk(KERN_ERR "umm, got %d back from search" | ||
| 2424 | ", was looking for %llu\n", ret, | ||
| 2425 | (unsigned long long)bytenr); | ||
| 2426 | btrfs_print_leaf(extent_root, path->nodes[0]); | ||
| 2427 | } | ||
| 2428 | BUG_ON(ret); | ||
| 2429 | extent_slot = path->slots[0]; | ||
| 2430 | } | ||
| 2431 | } else { | ||
| 2432 | btrfs_print_leaf(extent_root, path->nodes[0]); | ||
| 2433 | WARN_ON(1); | ||
| 2434 | printk(KERN_ERR "btrfs unable to find ref byte nr %llu " | ||
| 2435 | "root %llu gen %llu owner %llu\n", | ||
| 2436 | (unsigned long long)bytenr, | ||
| 2437 | (unsigned long long)root_objectid, | ||
| 2438 | (unsigned long long)ref_generation, | ||
| 2439 | (unsigned long long)owner_objectid); | ||
| 2440 | } | ||
| 2441 | |||
| 2442 | leaf = path->nodes[0]; | ||
| 2443 | ei = btrfs_item_ptr(leaf, extent_slot, | ||
| 2444 | struct btrfs_extent_item); | ||
| 2445 | refs = btrfs_extent_refs(leaf, ei); | ||
| 2446 | BUG_ON(refs == 0); | ||
| 2447 | refs -= 1; | ||
| 2448 | btrfs_set_extent_refs(leaf, ei, refs); | ||
| 2449 | |||
| 2450 | btrfs_mark_buffer_dirty(leaf); | ||
| 2451 | |||
| 2452 | if (refs == 0 && found_extent && path->slots[0] == extent_slot + 1) { | ||
| 2453 | struct btrfs_extent_ref *ref; | ||
| 2454 | ref = btrfs_item_ptr(leaf, path->slots[0], | ||
| 2455 | struct btrfs_extent_ref); | ||
| 2456 | BUG_ON(btrfs_ref_num_refs(leaf, ref) != 1); | ||
| 2457 | /* if the back ref and the extent are next to each other | ||
| 2458 | * they get deleted below in one shot | ||
| 2459 | */ | ||
| 2460 | path->slots[0] = extent_slot; | ||
| 2461 | num_to_del = 2; | ||
| 2462 | } else if (found_extent) { | ||
| 2463 | /* otherwise delete the extent back ref */ | ||
| 2464 | ret = remove_extent_backref(trans, extent_root, path); | ||
| 2465 | BUG_ON(ret); | ||
| 2466 | /* if refs are 0, we need to setup the path for deletion */ | ||
| 2467 | if (refs == 0) { | ||
| 2468 | btrfs_release_path(extent_root, path); | ||
| 2469 | ret = btrfs_search_slot(trans, extent_root, &key, path, | ||
| 2470 | -1, 1); | ||
| 2471 | BUG_ON(ret); | ||
| 2472 | } | ||
| 2473 | } | ||
| 2474 | |||
| 2475 | if (refs == 0) { | ||
| 2476 | u64 super_used; | ||
| 2477 | u64 root_used; | ||
| 2478 | |||
| 2479 | if (pin) { | ||
| 2480 | mutex_lock(&root->fs_info->pinned_mutex); | ||
| 2481 | ret = pin_down_bytes(trans, root, bytenr, num_bytes, | ||
| 2482 | owner_objectid >= BTRFS_FIRST_FREE_OBJECTID); | ||
| 2483 | mutex_unlock(&root->fs_info->pinned_mutex); | ||
| 2484 | if (ret > 0) | ||
| 2485 | mark_free = 1; | ||
| 2486 | BUG_ON(ret < 0); | ||
| 2487 | } | ||
| 2488 | /* block accounting for super block */ | ||
| 2489 | spin_lock(&info->delalloc_lock); | ||
| 2490 | super_used = btrfs_super_bytes_used(&info->super_copy); | ||
| 2491 | btrfs_set_super_bytes_used(&info->super_copy, | ||
| 2492 | super_used - num_bytes); | ||
| 2493 | |||
| 2494 | /* block accounting for root item */ | ||
| 2495 | root_used = btrfs_root_used(&root->root_item); | ||
| 2496 | btrfs_set_root_used(&root->root_item, | ||
| 2497 | root_used - num_bytes); | ||
| 2498 | spin_unlock(&info->delalloc_lock); | ||
| 2499 | ret = btrfs_del_items(trans, extent_root, path, path->slots[0], | ||
| 2500 | num_to_del); | ||
| 2501 | BUG_ON(ret); | ||
| 2502 | btrfs_release_path(extent_root, path); | ||
| 2503 | |||
| 2504 | if (owner_objectid >= BTRFS_FIRST_FREE_OBJECTID) { | ||
| 2505 | ret = btrfs_del_csums(trans, root, bytenr, num_bytes); | ||
| 2506 | BUG_ON(ret); | ||
| 2507 | } | ||
| 2508 | |||
| 2509 | ret = update_block_group(trans, root, bytenr, num_bytes, 0, | ||
| 2510 | mark_free); | ||
| 2511 | BUG_ON(ret); | ||
| 2512 | } | ||
| 2513 | btrfs_free_path(path); | ||
| 2514 | finish_current_insert(trans, extent_root, 0); | ||
| 2515 | return ret; | ||
| 2516 | } | ||
| 2517 | |||
| 2518 | /* | ||
| 2519 | * find all the blocks marked as pending in the radix tree and remove | ||
| 2520 | * them from the extent map | ||
| 2521 | */ | ||
| 2522 | static int del_pending_extents(struct btrfs_trans_handle *trans, | ||
| 2523 | struct btrfs_root *extent_root, int all) | ||
| 2524 | { | ||
| 2525 | int ret; | ||
| 2526 | int err = 0; | ||
| 2527 | u64 start; | ||
| 2528 | u64 end; | ||
| 2529 | u64 priv; | ||
| 2530 | u64 search = 0; | ||
| 2531 | int nr = 0, skipped = 0; | ||
| 2532 | struct extent_io_tree *pending_del; | ||
| 2533 | struct extent_io_tree *extent_ins; | ||
| 2534 | struct pending_extent_op *extent_op; | ||
| 2535 | struct btrfs_fs_info *info = extent_root->fs_info; | ||
| 2536 | struct list_head delete_list; | ||
| 2537 | |||
| 2538 | INIT_LIST_HEAD(&delete_list); | ||
| 2539 | extent_ins = &extent_root->fs_info->extent_ins; | ||
| 2540 | pending_del = &extent_root->fs_info->pending_del; | ||
| 2541 | |||
| 2542 | again: | ||
| 2543 | mutex_lock(&info->extent_ins_mutex); | ||
| 2544 | while (1) { | ||
| 2545 | ret = find_first_extent_bit(pending_del, search, &start, &end, | ||
| 2546 | EXTENT_WRITEBACK); | ||
| 2547 | if (ret) { | ||
| 2548 | if (all && skipped && !nr) { | ||
| 2549 | search = 0; | ||
| 2550 | continue; | ||
| 2551 | } | ||
| 2552 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2553 | break; | ||
| 2554 | } | ||
| 2555 | |||
| 2556 | ret = try_lock_extent(extent_ins, start, end, GFP_NOFS); | ||
| 2557 | if (!ret) { | ||
| 2558 | search = end+1; | ||
| 2559 | skipped = 1; | ||
| 2560 | |||
| 2561 | if (need_resched()) { | ||
| 2562 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2563 | cond_resched(); | ||
| 2564 | mutex_lock(&info->extent_ins_mutex); | ||
| 2565 | } | ||
| 2566 | |||
| 2567 | continue; | ||
| 2568 | } | ||
| 2569 | BUG_ON(ret < 0); | ||
| 2570 | |||
| 2571 | ret = get_state_private(pending_del, start, &priv); | ||
| 2572 | BUG_ON(ret); | ||
| 2573 | extent_op = (struct pending_extent_op *)(unsigned long)priv; | ||
| 2574 | |||
| 2575 | clear_extent_bits(pending_del, start, end, EXTENT_WRITEBACK, | ||
| 2576 | GFP_NOFS); | ||
| 2577 | if (!test_range_bit(extent_ins, start, end, | ||
| 2578 | EXTENT_WRITEBACK, 0)) { | ||
| 2579 | list_add_tail(&extent_op->list, &delete_list); | ||
| 2580 | nr++; | ||
| 2581 | } else { | ||
| 2582 | kfree(extent_op); | ||
| 2583 | |||
| 2584 | ret = get_state_private(&info->extent_ins, start, | ||
| 2585 | &priv); | ||
| 2586 | BUG_ON(ret); | ||
| 2587 | extent_op = (struct pending_extent_op *) | ||
| 2588 | (unsigned long)priv; | ||
| 2589 | |||
| 2590 | clear_extent_bits(&info->extent_ins, start, end, | ||
| 2591 | EXTENT_WRITEBACK, GFP_NOFS); | ||
| 2592 | |||
| 2593 | if (extent_op->type == PENDING_BACKREF_UPDATE) { | ||
| 2594 | list_add_tail(&extent_op->list, &delete_list); | ||
| 2595 | search = end + 1; | ||
| 2596 | nr++; | ||
| 2597 | continue; | ||
| 2598 | } | ||
| 2599 | |||
| 2600 | mutex_lock(&extent_root->fs_info->pinned_mutex); | ||
| 2601 | ret = pin_down_bytes(trans, extent_root, start, | ||
| 2602 | end + 1 - start, 0); | ||
| 2603 | mutex_unlock(&extent_root->fs_info->pinned_mutex); | ||
| 2604 | |||
| 2605 | ret = update_block_group(trans, extent_root, start, | ||
| 2606 | end + 1 - start, 0, ret > 0); | ||
| 2607 | |||
| 2608 | unlock_extent(extent_ins, start, end, GFP_NOFS); | ||
| 2609 | BUG_ON(ret); | ||
| 2610 | kfree(extent_op); | ||
| 2611 | } | ||
| 2612 | if (ret) | ||
| 2613 | err = ret; | ||
| 2614 | |||
| 2615 | search = end + 1; | ||
| 2616 | |||
| 2617 | if (need_resched()) { | ||
| 2618 | mutex_unlock(&info->extent_ins_mutex); | ||
| 2619 | cond_resched(); | ||
| 2620 | mutex_lock(&info->extent_ins_mutex); | ||
| 2621 | } | ||
| 2622 | } | ||
| 2623 | |||
| 2624 | if (nr) { | ||
| 2625 | ret = free_extents(trans, extent_root, &delete_list); | ||
| 2626 | BUG_ON(ret); | ||
| 2627 | } | ||
| 2628 | |||
| 2629 | if (all && skipped) { | ||
| 2630 | INIT_LIST_HEAD(&delete_list); | ||
| 2631 | search = 0; | ||
| 2632 | nr = 0; | ||
| 2633 | goto again; | ||
| 2634 | } | ||
| 2635 | |||
| 2636 | return err; | ||
| 2637 | } | ||
| 2638 | |||
| 2639 | /* | ||
| 2640 | * remove an extent from the root, returns 0 on success | ||
| 2641 | */ | ||
| 2642 | static int __btrfs_free_extent(struct btrfs_trans_handle *trans, | ||
| 2643 | struct btrfs_root *root, | ||
| 2644 | u64 bytenr, u64 num_bytes, u64 parent, | ||
| 2645 | u64 root_objectid, u64 ref_generation, | ||
| 2646 | u64 owner_objectid, int pin) | ||
| 2647 | { | ||
| 2648 | struct btrfs_root *extent_root = root->fs_info->extent_root; | ||
| 2649 | int pending_ret; | ||
| 2650 | int ret; | ||
| 2651 | |||
| 2652 | WARN_ON(num_bytes < root->sectorsize); | ||
| 2653 | if (root == extent_root) { | ||
| 2654 | struct pending_extent_op *extent_op = NULL; | ||
| 2655 | |||
| 2656 | mutex_lock(&root->fs_info->extent_ins_mutex); | ||
| 2657 | if (test_range_bit(&root->fs_info->extent_ins, bytenr, | ||
| 2658 | bytenr + num_bytes - 1, EXTENT_WRITEBACK, 0)) { | ||
| 2659 | u64 priv; | ||
| 2660 | ret = get_state_private(&root->fs_info->extent_ins, | ||
| 2661 | bytenr, &priv); | ||
| 2662 | BUG_ON(ret); | ||
| 2663 | extent_op = (struct pending_extent_op *) | ||
| 2664 | (unsigned long)priv; | ||
| 2665 | |||
| 2666 | extent_op->del = 1; | ||
| 2667 | if (extent_op->type == PENDING_EXTENT_INSERT) { | ||
| 2668 | mutex_unlock(&root->fs_info->extent_ins_mutex); | ||
| 2669 | return 0; | ||
| 2670 | } | ||
| 2671 | } | ||
| 2672 | |||
| 2673 | if (extent_op) { | ||
| 2674 | ref_generation = extent_op->orig_generation; | ||
| 2675 | parent = extent_op->orig_parent; | ||
| 2676 | } | ||
| 2677 | |||
| 2678 | extent_op = kmalloc(sizeof(*extent_op), GFP_NOFS); | ||
| 2679 | BUG_ON(!extent_op); | ||
| 2680 | |||
| 2681 | extent_op->type = PENDING_EXTENT_DELETE; | ||
| 2682 | extent_op->bytenr = bytenr; | ||
| 2683 | extent_op->num_bytes = num_bytes; | ||
| 2684 | extent_op->parent = parent; | ||
| 2685 | extent_op->orig_parent = parent; | ||
| 2686 | extent_op->generation = ref_generation; | ||
| 2687 | extent_op->orig_generation = ref_generation; | ||
| 2688 | extent_op->level = (int)owner_objectid; | ||
| 2689 | INIT_LIST_HEAD(&extent_op->list); | ||
| 2690 | extent_op->del = 0; | ||
| 2691 | |||
| 2692 | set_extent_bits(&root->fs_info->pending_del, | ||
| 2693 | bytenr, bytenr + num_bytes - 1, | ||
| 2694 | EXTENT_WRITEBACK, GFP_NOFS); | ||
| 2695 | set_state_private(&root->fs_info->pending_del, | ||
| 2696 | bytenr, (unsigned long)extent_op); | ||
| 2697 | mutex_unlock(&root->fs_info->extent_ins_mutex); | ||
| 2698 | return 0; | ||
| 2699 | } | ||
| 2700 | /* if metadata always pin */ | ||
| 2701 | if (owner_objectid < BTRFS_FIRST_FREE_OBJECTID) { | ||
| 2702 | if (root->root_key.objectid == BTRFS_TREE_LOG_OBJECTID) { | ||
| 2703 | struct btrfs_block_group_cache *cache; | ||
| 2704 | |||
| 2705 | /* btrfs_free_reserved_extent */ | ||
| 2706 | cache = btrfs_lookup_block_group(root->fs_info, bytenr); | ||
| 2707 | BUG_ON(!cache); | ||
| 2708 | btrfs_add_free_space(cache, bytenr, num_bytes); | ||
| 2709 | put_block_group(cache); | ||
| 2710 | update_reserved_extents(root, bytenr, num_bytes, 0); | ||
| 2711 | return 0; | ||
| 2712 | } | ||
| 2713 | pin = 1; | ||
| 2714 | } | ||
| 2715 | |||
| 2716 | /* if data pin when any transaction has committed this */ | ||
| 2717 | if (ref_generation != trans->transid) | ||
| 2718 | pin = 1; | ||
| 2719 | |||
| 2720 | ret = __free_extent(trans, root, bytenr, num_bytes, parent, | ||
| 2721 | root_objectid, ref_generation, | ||
| 2722 | owner_objectid, pin, pin == 0); | ||
| 2723 | |||
| 2724 | finish_current_insert(trans, root->fs_info->extent_root, 0); | ||
| 2725 | pending_ret = del_pending_extents(trans, root->fs_info->extent_root, 0); | ||
| 2726 | return ret ? ret : pending_ret; | ||
| 2727 | } | ||
| 2728 | |||
| 2729 | int btrfs_free_extent(struct btrfs_trans_handle *trans, | ||
| 2730 | struct btrfs_root *root, | ||
| 2731 | u64 bytenr, u64 num_bytes, u64 parent, | ||
| 2732 | u64 root_objectid, u64 ref_generation, | ||
| 2733 | u64 owner_objectid, int pin) | ||
| 2734 | { | ||
| 2735 | int ret; | ||
| 2736 | |||
| 2737 | ret = __btrfs_free_extent(trans, root, bytenr, num_bytes, parent, | ||
| 2738 | root_objectid, ref_generation, | ||
| 2739 | owner_objectid, pin); | ||
| 2740 | return ret; | ||
| 2741 | } | ||
| 2742 | |||
| 2743 | static u64 stripe_align(struct btrfs_root *root, u64 val) | ||
| 2744 | { | ||
| 2745 | u64 mask = ((u64)root->stripesize - 1); | ||
| 2746 | u64 ret = (val + mask) & ~mask; | ||
| 2747 | return ret; | ||
| 2748 | } | ||
| 2749 | |||
| 2750 | /* | ||
| 2751 | * walks the btree of allocated extents and find a hole of a given size. | ||
| 2752 | * The key ins is changed to record the hole: | ||
| 2753 | * ins->objectid == block start | ||
| 2754 | * ins->flags = BTRFS_EXTENT_ITEM_KEY | ||
| 2755 | * ins->offset == number of blocks | ||
| 2756 | * Any available blocks before search_start are skipped. | ||
| 2757 | */ | ||
| 2758 | static noinline int find_free_extent(struct btrfs_trans_handle *trans, | ||
| 2759 | struct btrfs_root *orig_root, | ||
| 2760 | u64 num_bytes, u64 empty_size, | ||
| 2761 | u64 search_start, u64 search_end, | ||
| 2762 | u64 hint_byte, struct btrfs_key *ins, | ||
| 2763 | u64 exclude_start, u64 exclude_nr, | ||
| 2764 | int data) | ||
| 2765 | { | ||
| 2766 | int ret = 0; | ||
| 2767 | struct btrfs_root *root = orig_root->fs_info->extent_root; | ||
| 2768 | u64 total_needed = num_bytes; | ||
| 2769 | u64 *last_ptr = NULL; | ||
| 2770 | u64 last_wanted = 0; | ||
| 2771 | struct btrfs_block_group_cache *block_group = NULL; | ||
| 2772 | int chunk_alloc_done = 0; | ||
| 2773 | int empty_cluster = 2 * 1024 * 1024; | ||
| 2774 | int allowed_chunk_alloc = 0; | ||
| 2775 | struct list_head *head = NULL, *cur = NULL; | ||
| 2776 | int loop = 0; | ||
| 2777 | int extra_loop = 0; | ||
| 2778 | struct btrfs_space_info *space_info; | ||
| 2779 | |||
| 2780 | WARN_ON(num_bytes < root->sectorsize); | ||
| 2781 | btrfs_set_key_type(ins, BTRFS_EXTENT_ITEM_KEY); | ||
| 2782 | ins->objectid = 0; | ||
| 2783 | ins->offset = 0; | ||
| 2784 | |||
| 2785 | if (orig_root->ref_cows || empty_size) | ||
| 2786 | allowed_chunk_alloc = 1; | ||
| 2787 | |||
| 2788 | if (data & BTRFS_BLOCK_GROUP_METADATA) { | ||
| 2789 | last_ptr = &root->fs_info->last_alloc; | ||
| 2790 | empty_cluster = 64 * 1024; | ||
| 2791 | } | ||
| 2792 | |||
| 2793 | if ((data & BTRFS_BLOCK_GROUP_DATA) && btrfs_test_opt(root, SSD)) | ||
| 2794 | last_ptr = &root->fs_info->last_data_alloc; | ||
| 2795 | |||
| 2796 | if (last_ptr) { | ||
| 2797 | if (*last_ptr) { | ||
| 2798 | hint_byte = *last_ptr; | ||
| 2799 | last_wanted = *last_ptr; | ||
| 2800 | } else | ||
| 2801 | empty_size += empty_cluster; | ||
| 2802 | } else { | ||
| 2803 | empty_cluster = 0; | ||
| 2804 | } | ||
| 2805 | search_start = max(search_start, first_logical_byte(root, 0)); | ||
| 2806 | search_start = max(search_start, hint_byte); | ||
| 2807 | |||
| 2808 | if (last_wanted && search_start != last_wanted) { | ||
| 2809 | last_wanted = 0; | ||
| 2810 | empty_size += empty_cluster; | ||
| 2811 | } | ||
| 2812 | |||
| 2813 | total_needed += empty_size; | ||
| 2814 | block_group = btrfs_lookup_block_group(root->fs_info, search_start); | ||
| 2815 | if (!block_group) | ||
| 2816 | block_group = btrfs_lookup_first_block_group(root->fs_info, | ||
| 2817 | search_start); | ||
| 2818 | space_info = __find_space_info(root->fs_info, data); | ||
| 2819 | |||
| 2820 | down_read(&space_info->groups_sem); | ||
| 2821 | while (1) { | ||
| 2822 | struct btrfs_free_space *free_space; | ||
| 2823 | /* | ||
| 2824 | * the only way this happens if our hint points to a block | ||
| 2825 | * group thats not of the proper type, while looping this | ||
| 2826 | * should never happen | ||
| 2827 | */ | ||
| 2828 | if (empty_size) | ||
| 2829 | extra_loop = 1; | ||
| 2830 | |||
| 2831 | if (!block_group) | ||
| 2832 | goto new_group_no_lock; | ||
| 2833 | |||
| 2834 | if (unlikely(!block_group->cached)) { | ||
| 2835 | mutex_lock(&block_group->cache_mutex); | ||
| 2836 | ret = cache_block_group(root, block_group); | ||
| 2837 | mutex_unlock(&block_group->cache_mutex); | ||
| 2838 | if (ret) | ||
| 2839 | break; | ||
| 2840 | } | ||
| 2841 | |||
| 2842 | mutex_lock(&block_group->alloc_mutex); | ||
| 2843 | if (unlikely(!block_group_bits(block_group, data))) | ||
| 2844 | goto new_group; | ||
| 2845 | |||
| 2846 | if (unlikely(block_group->ro)) | ||
| 2847 | goto new_group; | ||
| 2848 | |||
| 2849 | free_space = btrfs_find_free_space(block_group, search_start, | ||
| 2850 | total_needed); | ||
| 2851 | if (free_space) { | ||
| 2852 | u64 start = block_group->key.objectid; | ||
| 2853 | u64 end = block_group->key.objectid + | ||
| 2854 | block_group->key.offset; | ||
| 2855 | |||
| 2856 | search_start = stripe_align(root, free_space->offset); | ||
| 2857 | |||
| 2858 | /* move on to the next group */ | ||
| 2859 | if (search_start + num_bytes >= search_end) | ||
| 2860 | goto new_group; | ||
| 2861 | |||
| 2862 | /* move on to the next group */ | ||
| 2863 | if (search_start + num_bytes > end) | ||
| 2864 | goto new_group; | ||
| 2865 | |||
| 2866 | if (last_wanted && search_start != last_wanted) { | ||
| 2867 | total_needed += empty_cluster; | ||
| 2868 | empty_size += empty_cluster; | ||
| 2869 | last_wanted = 0; | ||
| 2870 | /* | ||
| 2871 | * if search_start is still in this block group | ||
| 2872 | * then we just re-search this block group | ||
| 2873 | */ | ||
| 2874 | if (search_start >= start && | ||
| 2875 | search_start < end) { | ||
| 2876 | mutex_unlock(&block_group->alloc_mutex); | ||
| 2877 | continue; | ||
| 2878 | } | ||
| 2879 | |||
| 2880 | /* else we go to the next block group */ | ||
| 2881 | goto new_group; | ||
| 2882 | } | ||
| 2883 | |||
| 2884 | if (exclude_nr > 0 && | ||
| 2885 | (search_start + num_bytes > exclude_start && | ||
| 2886 | search_start < exclude_start + exclude_nr)) { | ||
| 2887 | search_start = exclude_start + exclude_nr; | ||
| 2888 | /* | ||
| 2889 | * if search_start is still in this block group | ||
| 2890 | * then we just re-search this block group | ||
| 2891 | */ | ||
| 2892 | if (search_start >= start && | ||
| 2893 | search_start < end) { | ||
| 2894 | mutex_unlock(&block_group->alloc_mutex); | ||
| 2895 | last_wanted = 0; | ||
| 2896 | continue; | ||
| 2897 | } | ||
| 2898 | |||
| 2899 | /* else we go to the next block group */ | ||
| 2900 | goto new_group; | ||
| 2901 | } | ||
| 2902 | |||
| 2903 | ins->objectid = search_start; | ||
| 2904 | ins->offset = num_bytes; | ||
| 2905 | |||
| 2906 | btrfs_remove_free_space_lock(block_group, search_start, | ||
| 2907 | num_bytes); | ||
| 2908 | /* we are all good, lets return */ | ||
| 2909 | mutex_unlock(&block_group->alloc_mutex); | ||
| 2910 | break; | ||
| 2911 | } | ||
| 2912 | new_group: | ||
| 2913 | mutex_unlock(&block_group->alloc_mutex); | ||
| 2914 | put_block_group(block_group); | ||
| 2915 | block_group = NULL; | ||
| 2916 | new_group_no_lock: | ||
| 2917 | /* don't try to compare new allocations against the | ||
| 2918 | * last allocation any more | ||
| 2919 | */ | ||
| 2920 | last_wanted = 0; | ||
| 2921 | |||
| 2922 | /* | ||
| 2923 | * Here's how this works. | ||
| 2924 | * loop == 0: we were searching a block group via a hint | ||
| 2925 | * and didn't find anything, so we start at | ||
| 2926 | * the head of the block groups and keep searching | ||
| 2927 | * loop == 1: we're searching through all of the block groups | ||
| 2928 | * if we hit the head again we have searched | ||
| 2929 | * all of the block groups for this space and we | ||
| 2930 | * need to try and allocate, if we cant error out. | ||
| 2931 | * loop == 2: we allocated more space and are looping through | ||
| 2932 | * all of the block groups again. | ||
| 2933 | */ | ||
| 2934 | if (loop == 0) { | ||
| 2935 | head = &space_info->block_groups; | ||
| 2936 | cur = head->next; | ||
| 2937 | loop++; | ||
| 2938 | } else if (loop == 1 && cur == head) { | ||
| 2939 | int keep_going; | ||
| 2940 | |||
| 2941 | /* at this point we give up on the empty_size | ||
| 2942 | * allocations and just try to allocate the min | ||
| 2943 | * space. | ||
| 2944 | * | ||
| 2945 | * The extra_loop field was set if an empty_size | ||
| 2946 | * allocation was attempted above, and if this | ||
| 2947 | * is try we need to try the loop again without | ||
| 2948 | * the additional empty_size. | ||
| 2949 | */ | ||
| 2950 | total_needed -= empty_size; | ||
| 2951 | empty_size = 0; | ||
| 2952 | keep_going = extra_loop; | ||
| 2953 | loop++; | ||
| 2954 | |||
| 2955 | if (allowed_chunk_alloc && !chunk_alloc_done) { | ||
| 2956 | up_read(&space_info->groups_sem); | ||
| 2957 | ret = do_chunk_alloc(trans, root, num_bytes + | ||
| 2958 | 2 * 1024 * 1024, data, 1); | ||
| 2959 | down_read(&space_info->groups_sem); | ||
| 2960 | if (ret < 0) | ||
| 2961 | goto loop_check; | ||
| 2962 | head = &space_info->block_groups; | ||
| 2963 | /* | ||
| 2964 | * we've allocated a new chunk, keep | ||
| 2965 | * trying | ||
| 2966 | */ | ||
| 2967 | keep_going = 1; | ||
| 2968 | chunk_alloc_done = 1; | ||
| 2969 | } else if (!allowed_chunk_alloc) { | ||
| 2970 | space_info->force_alloc = 1; | ||
| 2971 | } | ||
| 2972 | loop_check: | ||
| 2973 | if (keep_going) { | ||
| 2974 | cur = head->next; | ||
| 2975 | extra_loop = 0; | ||
| 2976 | } else { | ||
| 2977 | break; | ||
| 2978 | } | ||
| 2979 | } else if (cur == head) { | ||
| 2980 | break; | ||
| 2981 | } | ||
| 2982 | |||
| 2983 | block_group = list_entry(cur, struct btrfs_block_group_cache, | ||
| 2984 | list); | ||
| 2985 | atomic_inc(&block_group->count); | ||
| 2986 | |||
| 2987 | search_start = block_group->key.objectid; | ||
| 2988 | cur = cur->next; | ||
| 2989 | } | ||
| 2990 | |||
| 2991 | /* we found what we needed */ | ||
| 2992 | if (ins->objectid) { | ||
| 2993 | if (!(data & BTRFS_BLOCK_GROUP_DATA)) | ||
| 2994 | trans->block_group = block_group->key.objectid; | ||
| 2995 | |||
| 2996 | if (last_ptr) | ||
| 2997 | *last_ptr = ins->objectid + ins->offset; | ||
| 2998 | ret = 0; | ||
| 2999 | } else if (!ret) { | ||
| 3000 | printk(KERN_ERR "btrfs searching for %llu bytes, " | ||
| 3001 | "num_bytes %llu, loop %d, allowed_alloc %d\n", | ||
| 3002 | (unsigned long long)total_needed, | ||
| 3003 | (unsigned long long)num_bytes, | ||
| 3004 | loop, allowed_chunk_alloc); | ||
| 3005 | ret = -ENOSPC; | ||
| 3006 | } | ||
| 3007 | if (block_group) | ||
| 3008 | put_block_group(block_group); | ||
| 3009 | |||
| 3010 | up_read(&space_info->groups_sem); | ||
| 3011 | return ret; | ||
| 3012 | } | ||
| 3013 | |||
| 3014 | static void dump_space_info(struct btrfs_space_info *info, u64 bytes) | ||
| 3015 | { | ||
| 3016 | struct btrfs_block_group_cache *cache; | ||
| 3017 | struct list_head *l; | ||
| 3018 | |||
| 3019 | printk(KERN_INFO "space_info has %llu free, is %sfull\n", | ||
| 3020 | (unsigned long long)(info->total_bytes - info->bytes_used - | ||
| 3021 | info->bytes_pinned - info->bytes_reserved), | ||
| 3022 | (info->full) ? "" : "not "); | ||
| 3023 | |||
| 3024 | down_read(&info->groups_sem); | ||
| 3025 | list_for_each(l, &info->block_groups) { | ||
| 3026 | cache = list_entry(l, struct btrfs_block_group_cache, list); | ||
| 3027 | spin_lock(&cache->lock); | ||
| 3028 | printk(KERN_INFO "block group %llu has %llu bytes, %llu used " | ||
| 3029 | "%llu pinned %llu reserved\n", | ||
| 3030 | (unsigned long long)cache->key.objectid, | ||
| 3031 | (unsigned long long)cache->key.offset, | ||
| 3032 | (unsigned long long)btrfs_block_group_used(&cache->item), | ||
| 3033 | (unsigned long long)cache->pinned, | ||
| 3034 | (unsigned long long)cache->reserved); | ||
| 3035 | btrfs_dump_free_space(cache, bytes); | ||
| 3036 | spin_unlock(&cache->lock); | ||
| 3037 | } | ||
| 3038 | up_read(&info->groups_sem); | ||
| 3039 | } | ||
| 3040 | |||
| 3041 | static int __btrfs_reserve_extent(struct btrfs_trans_handle *trans, | ||
| 3042 | struct btrfs_root *root, | ||
| 3043 | u64 num_bytes, u64 min_alloc_size, | ||
| 3044 | u64 empty_size, u64 hint_byte, | ||
| 3045 | u64 search_end, struct btrfs_key *ins, | ||
| 3046 | u64 data) | ||
| 3047 | { | ||
| 3048 | int ret; | ||
| 3049 | u64 search_start = 0; | ||
| 3050 | u64 alloc_profile; | ||
| 3051 | struct btrfs_fs_info *info = root->fs_info; | ||
| 3052 | |||
| 3053 | if (data) { | ||
| 3054 | alloc_profile = info->avail_data_alloc_bits & | ||
| 3055 | info->data_alloc_profile; | ||
| 3056 | data = BTRFS_BLOCK_GROUP_DATA | alloc_profile; | ||
| 3057 | } else if (root == root->fs_info->chunk_root) { | ||
| 3058 | alloc_profile = info->avail_system_alloc_bits & | ||
| 3059 | info->system_alloc_profile; | ||
| 3060 | data = BTRFS_BLOCK_GROUP_SYSTEM | alloc_profile; | ||
| 3061 | } else { | ||
| 3062 | alloc_profile = info->avail_metadata_alloc_bits & | ||
| 3063 | info->metadata_alloc_profile; | ||
| 3064 | data = BTRFS_BLOCK_GROUP_METADATA | alloc_profile; | ||
| 3065 | } | ||
| 3066 | again: | ||
| 3067 | data = btrfs_reduce_alloc_profile(root, data); | ||
| 3068 | /* | ||
| 3069 | * the only place that sets empty_size is btrfs_realloc_node, which | ||
| 3070 | * is not called recursively on allocations | ||
| 3071 | */ | ||
| 3072 | if (empty_size || root->ref_cows) { | ||
| 3073 | if (!(data & BTRFS_BLOCK_GROUP_METADATA)) { | ||
| 3074 | ret = do_chunk_alloc(trans, root->fs_info->extent_root, | ||
| 3075 | 2 * 1024 * 1024, | ||
| 3076 | BTRFS_BLOCK_GROUP_METADATA | | ||
| 3077 | (info->metadata_alloc_profile & | ||
| 3078 | info->avail_metadata_alloc_bits), 0); | ||
| 3079 | } | ||
| 3080 | ret = do_chunk_alloc(trans, root->fs_info->extent_root, | ||
| 3081 | num_bytes + 2 * 1024 * 1024, data, 0); | ||
| 3082 | } | ||
| 3083 | |||
| 3084 | WARN_ON(num_bytes < root->sectorsize); | ||
| 3085 | ret = find_free_extent(trans, root, num_bytes, empty_size, | ||
| 3086 | search_start, search_end, hint_byte, ins, | ||
| 3087 | trans->alloc_exclude_start, | ||
| 3088 | trans->alloc_exclude_nr, data); | ||
| 3089 | |||
| 3090 | if (ret == -ENOSPC && num_bytes > min_alloc_size) { | ||
| 3091 | num_bytes = num_bytes >> 1; | ||
| 3092 | num_bytes = num_bytes & ~(root->sectorsize - 1); | ||
| 3093 | num_bytes = max(num_bytes, min_alloc_size); | ||
| 3094 | do_chunk_alloc(trans, root->fs_info->extent_root, | ||
| 3095 | num_bytes, data, 1); | ||
| 3096 | goto again; | ||
| 3097 | } | ||
| 3098 | if (ret) { | ||
| 3099 | struct btrfs_space_info *sinfo; | ||
| 3100 | |||
| 3101 | sinfo = __find_space_info(root->fs_info, data); | ||
| 3102 | printk(KERN_ERR "btrfs allocation failed flags %llu, " | ||
| 3103 | "wanted %llu\n", (unsigned long long)data, | ||
| 3104 | (unsigned long long)num_bytes); | ||
| 3105 | dump_space_info(sinfo, num_bytes); | ||
| 3106 | BUG(); | ||
| 3107 | } | ||
| 3108 | |||
| 3109 | return ret; | ||
| 3110 | } | ||
| 3111 | |||
| 3112 | int btrfs_free_reserved_extent(struct btrfs_root *root, u64 start, u64 len) | ||
| 3113 | { | ||
| 3114 | struct btrfs_block_group_cache *cache; | ||
| 3115 | int ret = 0; | ||
| 3116 | |||
| 3117 | cache = btrfs_lookup_block_group(root->fs_info, start); | ||
| 3118 | if (!cache) { | ||
| 3119 | printk(KERN_ERR "Unable to find block group for %llu\n", | ||
| 3120 | (unsigned long long)start); | ||
| 3121 | return -ENOSPC; | ||
| 3122 | } | ||
| 3123 | |||
| 3124 | ret = btrfs_discard_extent(root, start, len); | ||
| 3125 | |||
| 3126 | btrfs_add_free_space(cache, start, len); | ||
| 3127 | put_block_group(cache); | ||
| 3128 | update_reserved_extents(root, start, len, 0); | ||
| 3129 | |||
| 3130 | return ret; | ||
| 3131 | } | ||
| 3132 | |||
| 3133 | int btrfs_reserve_extent(struct btrfs_trans_handle *trans, | ||
| 3134 | struct btrfs_root *root, | ||
| 3135 | u64 num_bytes, u64 min_alloc_size, | ||
| 3136 | u64 empty_size, u64 hint_byte, | ||
| 3137 | u64 search_end, struct btrfs_key *ins, | ||
| 3138 | u64 data) | ||
| 3139 | { | ||
| 3140 | int ret; | ||
| 3141 | ret = __btrfs_reserve_extent(trans, root, num_bytes, min_alloc_size, | ||
| 3142 | empty_size, hint_byte, search_end, ins, | ||
| 3143 | data); | ||
| 3144 | update_reserved_extents(root, ins->objectid, ins->offset, 1); | ||
| 3145 | return ret; | ||
| 3146 | } | ||
| 3147 | |||
| 3148 | static int __btrfs_alloc_reserved_extent(struct btrfs_trans_handle *trans, | ||
| 3149 | struct btrfs_root *root, u64 parent, | ||
| 3150 | u64 root_objectid, u64 ref_generation, | ||
| 3151 | u64 owner, struct btrfs_key *ins) | ||
| 3152 | { | ||
| 3153 | int ret; | ||
| 3154 | int pending_ret; | ||
| 3155 | u64 super_used; | ||
| 3156 | u64 root_used; | ||
| 3157 | u64 num_bytes = ins->offset; | ||
| 3158 | u32 sizes[2]; | ||
| 3159 | struct btrfs_fs_info *info = root->fs_info; | ||
| 3160 | struct btrfs_root *extent_root = info->extent_root; | ||
| 3161 | struct btrfs_extent_item *extent_item; | ||
| 3162 | struct btrfs_extent_ref *ref; | ||
| 3163 | struct btrfs_path *path; | ||
| 3164 | struct btrfs_key keys[2]; | ||
| 3165 | |||
| 3166 | if (parent == 0) | ||
| 3167 | parent = ins->objectid; | ||
| 3168 | |||
| 3169 | /* block accounting for super block */ | ||
| 3170 | spin_lock(&info->delalloc_lock); | ||
| 3171 | super_used = btrfs_super_bytes_used(&info->super_copy); | ||
| 3172 | btrfs_set_super_bytes_used(&info->super_copy, super_used + num_bytes); | ||
| 3173 | |||
| 3174 | /* block accounting for root item */ | ||
| 3175 | root_used = btrfs_root_used(&root->root_item); | ||
| 3176 | btrfs_set_root_used(&root->root_item, root_used + num_bytes); | ||
| 3177 | spin_unlock(&info->delalloc_lock); | ||
| 3178 | |||
| 3179 | if (root == extent_root) { | ||
| 3180 | struct pending_extent_op *extent_op; | ||
| 3181 | |||
| 3182 | extent_op = kmalloc(sizeof(*extent_op), GFP_NOFS); | ||
| 3183 | BUG_ON(!extent_op); | ||
| 3184 | |||
| 3185 | extent_op->type = PENDING_EXTENT_INSERT; | ||
| 3186 | extent_op->bytenr = ins->objectid; | ||
| 3187 | extent_op->num_bytes = ins->offset; | ||
| 3188 | extent_op->parent = parent; | ||
| 3189 | extent_op->orig_parent = 0; | ||
| 3190 | extent_op->generation = ref_generation; | ||
| 3191 | extent_op->orig_generation = 0; | ||
| 3192 | extent_op->level = (int)owner; | ||
| 3193 | INIT_LIST_HEAD(&extent_op->list); | ||
| 3194 | extent_op->del = 0; | ||
| 3195 | |||
| 3196 | mutex_lock(&root->fs_info->extent_ins_mutex); | ||
| 3197 | set_extent_bits(&root->fs_info->extent_ins, ins->objectid, | ||
| 3198 | ins->objectid + ins->offset - 1, | ||
| 3199 | EXTENT_WRITEBACK, GFP_NOFS); | ||
| 3200 | set_state_private(&root->fs_info->extent_ins, | ||
| 3201 | ins->objectid, (unsigned long)extent_op); | ||
| 3202 | mutex_unlock(&root->fs_info->extent_ins_mutex); | ||
| 3203 | goto update_block; | ||
| 3204 | } | ||
| 3205 | |||
| 3206 | memcpy(&keys[0], ins, sizeof(*ins)); | ||
| 3207 | keys[1].objectid = ins->objectid; | ||
| 3208 | keys[1].type = BTRFS_EXTENT_REF_KEY; | ||
| 3209 | keys[1].offset = parent; | ||
| 3210 | sizes[0] = sizeof(*extent_item); | ||
| 3211 | sizes[1] = sizeof(*ref); | ||
| 3212 | |||
| 3213 | path = btrfs_alloc_path(); | ||
| 3214 | BUG_ON(!path); | ||
| 3215 | |||
| 3216 | ret = btrfs_insert_empty_items(trans, extent_root, path, keys, | ||
| 3217 | sizes, 2); | ||
| 3218 | BUG_ON(ret); | ||
| 3219 | |||
| 3220 | extent_item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 3221 | struct btrfs_extent_item); | ||
| 3222 | btrfs_set_extent_refs(path->nodes[0], extent_item, 1); | ||
| 3223 | ref = btrfs_item_ptr(path->nodes[0], path->slots[0] + 1, | ||
| 3224 | struct btrfs_extent_ref); | ||
| 3225 | |||
| 3226 | btrfs_set_ref_root(path->nodes[0], ref, root_objectid); | ||
| 3227 | btrfs_set_ref_generation(path->nodes[0], ref, ref_generation); | ||
| 3228 | btrfs_set_ref_objectid(path->nodes[0], ref, owner); | ||
| 3229 | btrfs_set_ref_num_refs(path->nodes[0], ref, 1); | ||
| 3230 | |||
| 3231 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 3232 | |||
| 3233 | trans->alloc_exclude_start = 0; | ||
| 3234 | trans->alloc_exclude_nr = 0; | ||
| 3235 | btrfs_free_path(path); | ||
| 3236 | finish_current_insert(trans, extent_root, 0); | ||
| 3237 | pending_ret = del_pending_extents(trans, extent_root, 0); | ||
| 3238 | |||
| 3239 | if (ret) | ||
| 3240 | goto out; | ||
| 3241 | if (pending_ret) { | ||
| 3242 | ret = pending_ret; | ||
| 3243 | goto out; | ||
| 3244 | } | ||
| 3245 | |||
| 3246 | update_block: | ||
| 3247 | ret = update_block_group(trans, root, ins->objectid, | ||
| 3248 | ins->offset, 1, 0); | ||
| 3249 | if (ret) { | ||
| 3250 | printk(KERN_ERR "btrfs update block group failed for %llu " | ||
| 3251 | "%llu\n", (unsigned long long)ins->objectid, | ||
| 3252 | (unsigned long long)ins->offset); | ||
| 3253 | BUG(); | ||
| 3254 | } | ||
| 3255 | out: | ||
| 3256 | return ret; | ||
| 3257 | } | ||
| 3258 | |||
| 3259 | int btrfs_alloc_reserved_extent(struct btrfs_trans_handle *trans, | ||
| 3260 | struct btrfs_root *root, u64 parent, | ||
| 3261 | u64 root_objectid, u64 ref_generation, | ||
| 3262 | u64 owner, struct btrfs_key *ins) | ||
| 3263 | { | ||
| 3264 | int ret; | ||
| 3265 | |||
| 3266 | if (root_objectid == BTRFS_TREE_LOG_OBJECTID) | ||
| 3267 | return 0; | ||
| 3268 | ret = __btrfs_alloc_reserved_extent(trans, root, parent, root_objectid, | ||
| 3269 | ref_generation, owner, ins); | ||
| 3270 | update_reserved_extents(root, ins->objectid, ins->offset, 0); | ||
| 3271 | return ret; | ||
| 3272 | } | ||
| 3273 | |||
| 3274 | /* | ||
| 3275 | * this is used by the tree logging recovery code. It records that | ||
| 3276 | * an extent has been allocated and makes sure to clear the free | ||
| 3277 | * space cache bits as well | ||
| 3278 | */ | ||
| 3279 | int btrfs_alloc_logged_extent(struct btrfs_trans_handle *trans, | ||
| 3280 | struct btrfs_root *root, u64 parent, | ||
| 3281 | u64 root_objectid, u64 ref_generation, | ||
| 3282 | u64 owner, struct btrfs_key *ins) | ||
| 3283 | { | ||
| 3284 | int ret; | ||
| 3285 | struct btrfs_block_group_cache *block_group; | ||
| 3286 | |||
| 3287 | block_group = btrfs_lookup_block_group(root->fs_info, ins->objectid); | ||
| 3288 | mutex_lock(&block_group->cache_mutex); | ||
| 3289 | cache_block_group(root, block_group); | ||
| 3290 | mutex_unlock(&block_group->cache_mutex); | ||
| 3291 | |||
| 3292 | ret = btrfs_remove_free_space(block_group, ins->objectid, | ||
| 3293 | ins->offset); | ||
| 3294 | BUG_ON(ret); | ||
| 3295 | put_block_group(block_group); | ||
| 3296 | ret = __btrfs_alloc_reserved_extent(trans, root, parent, root_objectid, | ||
| 3297 | ref_generation, owner, ins); | ||
| 3298 | return ret; | ||
| 3299 | } | ||
| 3300 | |||
| 3301 | /* | ||
| 3302 | * finds a free extent and does all the dirty work required for allocation | ||
| 3303 | * returns the key for the extent through ins, and a tree buffer for | ||
| 3304 | * the first block of the extent through buf. | ||
| 3305 | * | ||
| 3306 | * returns 0 if everything worked, non-zero otherwise. | ||
| 3307 | */ | ||
| 3308 | int btrfs_alloc_extent(struct btrfs_trans_handle *trans, | ||
| 3309 | struct btrfs_root *root, | ||
| 3310 | u64 num_bytes, u64 parent, u64 min_alloc_size, | ||
| 3311 | u64 root_objectid, u64 ref_generation, | ||
| 3312 | u64 owner_objectid, u64 empty_size, u64 hint_byte, | ||
| 3313 | u64 search_end, struct btrfs_key *ins, u64 data) | ||
| 3314 | { | ||
| 3315 | int ret; | ||
| 3316 | |||
| 3317 | ret = __btrfs_reserve_extent(trans, root, num_bytes, | ||
| 3318 | min_alloc_size, empty_size, hint_byte, | ||
| 3319 | search_end, ins, data); | ||
| 3320 | BUG_ON(ret); | ||
| 3321 | if (root_objectid != BTRFS_TREE_LOG_OBJECTID) { | ||
| 3322 | ret = __btrfs_alloc_reserved_extent(trans, root, parent, | ||
| 3323 | root_objectid, ref_generation, | ||
| 3324 | owner_objectid, ins); | ||
| 3325 | BUG_ON(ret); | ||
| 3326 | |||
| 3327 | } else { | ||
| 3328 | update_reserved_extents(root, ins->objectid, ins->offset, 1); | ||
| 3329 | } | ||
| 3330 | return ret; | ||
| 3331 | } | ||
| 3332 | |||
| 3333 | struct extent_buffer *btrfs_init_new_buffer(struct btrfs_trans_handle *trans, | ||
| 3334 | struct btrfs_root *root, | ||
| 3335 | u64 bytenr, u32 blocksize) | ||
| 3336 | { | ||
| 3337 | struct extent_buffer *buf; | ||
| 3338 | |||
| 3339 | buf = btrfs_find_create_tree_block(root, bytenr, blocksize); | ||
| 3340 | if (!buf) | ||
| 3341 | return ERR_PTR(-ENOMEM); | ||
| 3342 | btrfs_set_header_generation(buf, trans->transid); | ||
| 3343 | btrfs_tree_lock(buf); | ||
| 3344 | clean_tree_block(trans, root, buf); | ||
| 3345 | btrfs_set_buffer_uptodate(buf); | ||
| 3346 | if (root->root_key.objectid == BTRFS_TREE_LOG_OBJECTID) { | ||
| 3347 | set_extent_dirty(&root->dirty_log_pages, buf->start, | ||
| 3348 | buf->start + buf->len - 1, GFP_NOFS); | ||
| 3349 | } else { | ||
| 3350 | set_extent_dirty(&trans->transaction->dirty_pages, buf->start, | ||
| 3351 | buf->start + buf->len - 1, GFP_NOFS); | ||
| 3352 | } | ||
| 3353 | trans->blocks_used++; | ||
| 3354 | return buf; | ||
| 3355 | } | ||
| 3356 | |||
| 3357 | /* | ||
| 3358 | * helper function to allocate a block for a given tree | ||
| 3359 | * returns the tree buffer or NULL. | ||
| 3360 | */ | ||
| 3361 | struct extent_buffer *btrfs_alloc_free_block(struct btrfs_trans_handle *trans, | ||
| 3362 | struct btrfs_root *root, | ||
| 3363 | u32 blocksize, u64 parent, | ||
| 3364 | u64 root_objectid, | ||
| 3365 | u64 ref_generation, | ||
| 3366 | int level, | ||
| 3367 | u64 hint, | ||
| 3368 | u64 empty_size) | ||
| 3369 | { | ||
| 3370 | struct btrfs_key ins; | ||
| 3371 | int ret; | ||
| 3372 | struct extent_buffer *buf; | ||
| 3373 | |||
| 3374 | ret = btrfs_alloc_extent(trans, root, blocksize, parent, blocksize, | ||
| 3375 | root_objectid, ref_generation, level, | ||
| 3376 | empty_size, hint, (u64)-1, &ins, 0); | ||
| 3377 | if (ret) { | ||
| 3378 | BUG_ON(ret > 0); | ||
| 3379 | return ERR_PTR(ret); | ||
| 3380 | } | ||
| 3381 | |||
| 3382 | buf = btrfs_init_new_buffer(trans, root, ins.objectid, blocksize); | ||
| 3383 | return buf; | ||
| 3384 | } | ||
| 3385 | |||
| 3386 | int btrfs_drop_leaf_ref(struct btrfs_trans_handle *trans, | ||
| 3387 | struct btrfs_root *root, struct extent_buffer *leaf) | ||
| 3388 | { | ||
| 3389 | u64 leaf_owner; | ||
| 3390 | u64 leaf_generation; | ||
| 3391 | struct btrfs_key key; | ||
| 3392 | struct btrfs_file_extent_item *fi; | ||
| 3393 | int i; | ||
| 3394 | int nritems; | ||
| 3395 | int ret; | ||
| 3396 | |||
| 3397 | BUG_ON(!btrfs_is_leaf(leaf)); | ||
| 3398 | nritems = btrfs_header_nritems(leaf); | ||
| 3399 | leaf_owner = btrfs_header_owner(leaf); | ||
| 3400 | leaf_generation = btrfs_header_generation(leaf); | ||
| 3401 | |||
| 3402 | for (i = 0; i < nritems; i++) { | ||
| 3403 | u64 disk_bytenr; | ||
| 3404 | cond_resched(); | ||
| 3405 | |||
| 3406 | btrfs_item_key_to_cpu(leaf, &key, i); | ||
| 3407 | if (btrfs_key_type(&key) != BTRFS_EXTENT_DATA_KEY) | ||
| 3408 | continue; | ||
| 3409 | fi = btrfs_item_ptr(leaf, i, struct btrfs_file_extent_item); | ||
| 3410 | if (btrfs_file_extent_type(leaf, fi) == | ||
| 3411 | BTRFS_FILE_EXTENT_INLINE) | ||
| 3412 | continue; | ||
| 3413 | /* | ||
| 3414 | * FIXME make sure to insert a trans record that | ||
| 3415 | * repeats the snapshot del on crash | ||
| 3416 | */ | ||
| 3417 | disk_bytenr = btrfs_file_extent_disk_bytenr(leaf, fi); | ||
| 3418 | if (disk_bytenr == 0) | ||
| 3419 | continue; | ||
| 3420 | |||
| 3421 | ret = __btrfs_free_extent(trans, root, disk_bytenr, | ||
| 3422 | btrfs_file_extent_disk_num_bytes(leaf, fi), | ||
| 3423 | leaf->start, leaf_owner, leaf_generation, | ||
| 3424 | key.objectid, 0); | ||
| 3425 | BUG_ON(ret); | ||
| 3426 | |||
| 3427 | atomic_inc(&root->fs_info->throttle_gen); | ||
| 3428 | wake_up(&root->fs_info->transaction_throttle); | ||
| 3429 | cond_resched(); | ||
| 3430 | } | ||
| 3431 | return 0; | ||
| 3432 | } | ||
| 3433 | |||
| 3434 | static noinline int cache_drop_leaf_ref(struct btrfs_trans_handle *trans, | ||
| 3435 | struct btrfs_root *root, | ||
| 3436 | struct btrfs_leaf_ref *ref) | ||
| 3437 | { | ||
| 3438 | int i; | ||
| 3439 | int ret; | ||
| 3440 | struct btrfs_extent_info *info = ref->extents; | ||
| 3441 | |||
| 3442 | for (i = 0; i < ref->nritems; i++) { | ||
| 3443 | ret = __btrfs_free_extent(trans, root, info->bytenr, | ||
| 3444 | info->num_bytes, ref->bytenr, | ||
| 3445 | ref->owner, ref->generation, | ||
| 3446 | info->objectid, 0); | ||
| 3447 | |||
| 3448 | atomic_inc(&root->fs_info->throttle_gen); | ||
| 3449 | wake_up(&root->fs_info->transaction_throttle); | ||
| 3450 | cond_resched(); | ||
| 3451 | |||
| 3452 | BUG_ON(ret); | ||
| 3453 | info++; | ||
| 3454 | } | ||
| 3455 | |||
| 3456 | return 0; | ||
| 3457 | } | ||
| 3458 | |||
| 3459 | static int drop_snap_lookup_refcount(struct btrfs_root *root, u64 start, | ||
| 3460 | u64 len, u32 *refs) | ||
| 3461 | { | ||
| 3462 | int ret; | ||
| 3463 | |||
| 3464 | ret = btrfs_lookup_extent_ref(NULL, root, start, len, refs); | ||
| 3465 | BUG_ON(ret); | ||
| 3466 | |||
| 3467 | #if 0 /* some debugging code in case we see problems here */ | ||
| 3468 | /* if the refs count is one, it won't get increased again. But | ||
| 3469 | * if the ref count is > 1, someone may be decreasing it at | ||
| 3470 | * the same time we are. | ||
| 3471 | */ | ||
| 3472 | if (*refs != 1) { | ||
| 3473 | struct extent_buffer *eb = NULL; | ||
| 3474 | eb = btrfs_find_create_tree_block(root, start, len); | ||
| 3475 | if (eb) | ||
| 3476 | btrfs_tree_lock(eb); | ||
| 3477 | |||
| 3478 | mutex_lock(&root->fs_info->alloc_mutex); | ||
| 3479 | ret = lookup_extent_ref(NULL, root, start, len, refs); | ||
| 3480 | BUG_ON(ret); | ||
| 3481 | mutex_unlock(&root->fs_info->alloc_mutex); | ||
| 3482 | |||
| 3483 | if (eb) { | ||
| 3484 | btrfs_tree_unlock(eb); | ||
| 3485 | free_extent_buffer(eb); | ||
| 3486 | } | ||
| 3487 | if (*refs == 1) { | ||
| 3488 | printk(KERN_ERR "btrfs block %llu went down to one " | ||
| 3489 | "during drop_snap\n", (unsigned long long)start); | ||
| 3490 | } | ||
| 3491 | |||
| 3492 | } | ||
| 3493 | #endif | ||
| 3494 | |||
| 3495 | cond_resched(); | ||
| 3496 | return ret; | ||
| 3497 | } | ||
| 3498 | |||
| 3499 | /* | ||
| 3500 | * helper function for drop_snapshot, this walks down the tree dropping ref | ||
| 3501 | * counts as it goes. | ||
| 3502 | */ | ||
| 3503 | static noinline int walk_down_tree(struct btrfs_trans_handle *trans, | ||
| 3504 | struct btrfs_root *root, | ||
| 3505 | struct btrfs_path *path, int *level) | ||
| 3506 | { | ||
| 3507 | u64 root_owner; | ||
| 3508 | u64 root_gen; | ||
| 3509 | u64 bytenr; | ||
| 3510 | u64 ptr_gen; | ||
| 3511 | struct extent_buffer *next; | ||
| 3512 | struct extent_buffer *cur; | ||
| 3513 | struct extent_buffer *parent; | ||
| 3514 | struct btrfs_leaf_ref *ref; | ||
| 3515 | u32 blocksize; | ||
| 3516 | int ret; | ||
| 3517 | u32 refs; | ||
| 3518 | |||
| 3519 | WARN_ON(*level < 0); | ||
| 3520 | WARN_ON(*level >= BTRFS_MAX_LEVEL); | ||
| 3521 | ret = drop_snap_lookup_refcount(root, path->nodes[*level]->start, | ||
| 3522 | path->nodes[*level]->len, &refs); | ||
| 3523 | BUG_ON(ret); | ||
| 3524 | if (refs > 1) | ||
| 3525 | goto out; | ||
| 3526 | |||
| 3527 | /* | ||
| 3528 | * walk down to the last node level and free all the leaves | ||
| 3529 | */ | ||
| 3530 | while (*level >= 0) { | ||
| 3531 | WARN_ON(*level < 0); | ||
| 3532 | WARN_ON(*level >= BTRFS_MAX_LEVEL); | ||
| 3533 | cur = path->nodes[*level]; | ||
| 3534 | |||
| 3535 | if (btrfs_header_level(cur) != *level) | ||
| 3536 | WARN_ON(1); | ||
| 3537 | |||
| 3538 | if (path->slots[*level] >= | ||
| 3539 | btrfs_header_nritems(cur)) | ||
| 3540 | break; | ||
| 3541 | if (*level == 0) { | ||
| 3542 | ret = btrfs_drop_leaf_ref(trans, root, cur); | ||
| 3543 | BUG_ON(ret); | ||
| 3544 | break; | ||
| 3545 | } | ||
| 3546 | bytenr = btrfs_node_blockptr(cur, path->slots[*level]); | ||
| 3547 | ptr_gen = btrfs_node_ptr_generation(cur, path->slots[*level]); | ||
| 3548 | blocksize = btrfs_level_size(root, *level - 1); | ||
| 3549 | |||
| 3550 | ret = drop_snap_lookup_refcount(root, bytenr, blocksize, &refs); | ||
| 3551 | BUG_ON(ret); | ||
| 3552 | if (refs != 1) { | ||
| 3553 | parent = path->nodes[*level]; | ||
| 3554 | root_owner = btrfs_header_owner(parent); | ||
| 3555 | root_gen = btrfs_header_generation(parent); | ||
| 3556 | path->slots[*level]++; | ||
| 3557 | |||
| 3558 | ret = __btrfs_free_extent(trans, root, bytenr, | ||
| 3559 | blocksize, parent->start, | ||
| 3560 | root_owner, root_gen, | ||
| 3561 | *level - 1, 1); | ||
| 3562 | BUG_ON(ret); | ||
| 3563 | |||
| 3564 | atomic_inc(&root->fs_info->throttle_gen); | ||
| 3565 | wake_up(&root->fs_info->transaction_throttle); | ||
| 3566 | cond_resched(); | ||
| 3567 | |||
| 3568 | continue; | ||
| 3569 | } | ||
| 3570 | /* | ||
| 3571 | * at this point, we have a single ref, and since the | ||
| 3572 | * only place referencing this extent is a dead root | ||
| 3573 | * the reference count should never go higher. | ||
| 3574 | * So, we don't need to check it again | ||
| 3575 | */ | ||
| 3576 | if (*level == 1) { | ||
| 3577 | ref = btrfs_lookup_leaf_ref(root, bytenr); | ||
| 3578 | if (ref && ref->generation != ptr_gen) { | ||
| 3579 | btrfs_free_leaf_ref(root, ref); | ||
| 3580 | ref = NULL; | ||
| 3581 | } | ||
| 3582 | if (ref) { | ||
| 3583 | ret = cache_drop_leaf_ref(trans, root, ref); | ||
| 3584 | BUG_ON(ret); | ||
| 3585 | btrfs_remove_leaf_ref(root, ref); | ||
| 3586 | btrfs_free_leaf_ref(root, ref); | ||
| 3587 | *level = 0; | ||
| 3588 | break; | ||
| 3589 | } | ||
| 3590 | } | ||
| 3591 | next = btrfs_find_tree_block(root, bytenr, blocksize); | ||
| 3592 | if (!next || !btrfs_buffer_uptodate(next, ptr_gen)) { | ||
| 3593 | free_extent_buffer(next); | ||
| 3594 | |||
| 3595 | next = read_tree_block(root, bytenr, blocksize, | ||
| 3596 | ptr_gen); | ||
| 3597 | cond_resched(); | ||
| 3598 | #if 0 | ||
| 3599 | /* | ||
| 3600 | * this is a debugging check and can go away | ||
| 3601 | * the ref should never go all the way down to 1 | ||
| 3602 | * at this point | ||
| 3603 | */ | ||
| 3604 | ret = lookup_extent_ref(NULL, root, bytenr, blocksize, | ||
| 3605 | &refs); | ||
| 3606 | BUG_ON(ret); | ||
| 3607 | WARN_ON(refs != 1); | ||
| 3608 | #endif | ||
| 3609 | } | ||
| 3610 | WARN_ON(*level <= 0); | ||
| 3611 | if (path->nodes[*level-1]) | ||
| 3612 | free_extent_buffer(path->nodes[*level-1]); | ||
| 3613 | path->nodes[*level-1] = next; | ||
| 3614 | *level = btrfs_header_level(next); | ||
| 3615 | path->slots[*level] = 0; | ||
| 3616 | cond_resched(); | ||
| 3617 | } | ||
| 3618 | out: | ||
| 3619 | WARN_ON(*level < 0); | ||
| 3620 | WARN_ON(*level >= BTRFS_MAX_LEVEL); | ||
| 3621 | |||
| 3622 | if (path->nodes[*level] == root->node) { | ||
| 3623 | parent = path->nodes[*level]; | ||
| 3624 | bytenr = path->nodes[*level]->start; | ||
| 3625 | } else { | ||
| 3626 | parent = path->nodes[*level + 1]; | ||
| 3627 | bytenr = btrfs_node_blockptr(parent, path->slots[*level + 1]); | ||
| 3628 | } | ||
| 3629 | |||
| 3630 | blocksize = btrfs_level_size(root, *level); | ||
| 3631 | root_owner = btrfs_header_owner(parent); | ||
| 3632 | root_gen = btrfs_header_generation(parent); | ||
| 3633 | |||
| 3634 | ret = __btrfs_free_extent(trans, root, bytenr, blocksize, | ||
| 3635 | parent->start, root_owner, root_gen, | ||
| 3636 | *level, 1); | ||
| 3637 | free_extent_buffer(path->nodes[*level]); | ||
| 3638 | path->nodes[*level] = NULL; | ||
| 3639 | *level += 1; | ||
| 3640 | BUG_ON(ret); | ||
| 3641 | |||
| 3642 | cond_resched(); | ||
| 3643 | return 0; | ||
| 3644 | } | ||
| 3645 | |||
| 3646 | /* | ||
| 3647 | * helper function for drop_subtree, this function is similar to | ||
| 3648 | * walk_down_tree. The main difference is that it checks reference | ||
| 3649 | * counts while tree blocks are locked. | ||
| 3650 | */ | ||
| 3651 | static noinline int walk_down_subtree(struct btrfs_trans_handle *trans, | ||
| 3652 | struct btrfs_root *root, | ||
| 3653 | struct btrfs_path *path, int *level) | ||
| 3654 | { | ||
| 3655 | struct extent_buffer *next; | ||
| 3656 | struct extent_buffer *cur; | ||
| 3657 | struct extent_buffer *parent; | ||
| 3658 | u64 bytenr; | ||
| 3659 | u64 ptr_gen; | ||
| 3660 | u32 blocksize; | ||
| 3661 | u32 refs; | ||
| 3662 | int ret; | ||
| 3663 | |||
| 3664 | cur = path->nodes[*level]; | ||
| 3665 | ret = btrfs_lookup_extent_ref(trans, root, cur->start, cur->len, | ||
| 3666 | &refs); | ||
| 3667 | BUG_ON(ret); | ||
| 3668 | if (refs > 1) | ||
| 3669 | goto out; | ||
| 3670 | |||
| 3671 | while (*level >= 0) { | ||
| 3672 | cur = path->nodes[*level]; | ||
| 3673 | if (*level == 0) { | ||
| 3674 | ret = btrfs_drop_leaf_ref(trans, root, cur); | ||
| 3675 | BUG_ON(ret); | ||
| 3676 | clean_tree_block(trans, root, cur); | ||
| 3677 | break; | ||
| 3678 | } | ||
| 3679 | if (path->slots[*level] >= btrfs_header_nritems(cur)) { | ||
| 3680 | clean_tree_block(trans, root, cur); | ||
| 3681 | break; | ||
| 3682 | } | ||
| 3683 | |||
| 3684 | bytenr = btrfs_node_blockptr(cur, path->slots[*level]); | ||
| 3685 | blocksize = btrfs_level_size(root, *level - 1); | ||
| 3686 | ptr_gen = btrfs_node_ptr_generation(cur, path->slots[*level]); | ||
| 3687 | |||
| 3688 | next = read_tree_block(root, bytenr, blocksize, ptr_gen); | ||
| 3689 | btrfs_tree_lock(next); | ||
| 3690 | |||
| 3691 | ret = btrfs_lookup_extent_ref(trans, root, bytenr, blocksize, | ||
| 3692 | &refs); | ||
| 3693 | BUG_ON(ret); | ||
| 3694 | if (refs > 1) { | ||
| 3695 | parent = path->nodes[*level]; | ||
| 3696 | ret = btrfs_free_extent(trans, root, bytenr, | ||
| 3697 | blocksize, parent->start, | ||
| 3698 | btrfs_header_owner(parent), | ||
| 3699 | btrfs_header_generation(parent), | ||
| 3700 | *level - 1, 1); | ||
| 3701 | BUG_ON(ret); | ||
| 3702 | path->slots[*level]++; | ||
| 3703 | btrfs_tree_unlock(next); | ||
| 3704 | free_extent_buffer(next); | ||
| 3705 | continue; | ||
| 3706 | } | ||
| 3707 | |||
| 3708 | *level = btrfs_header_level(next); | ||
| 3709 | path->nodes[*level] = next; | ||
| 3710 | path->slots[*level] = 0; | ||
| 3711 | path->locks[*level] = 1; | ||
| 3712 | cond_resched(); | ||
| 3713 | } | ||
| 3714 | out: | ||
| 3715 | parent = path->nodes[*level + 1]; | ||
| 3716 | bytenr = path->nodes[*level]->start; | ||
| 3717 | blocksize = path->nodes[*level]->len; | ||
| 3718 | |||
| 3719 | ret = btrfs_free_extent(trans, root, bytenr, blocksize, | ||
| 3720 | parent->start, btrfs_header_owner(parent), | ||
| 3721 | btrfs_header_generation(parent), *level, 1); | ||
| 3722 | BUG_ON(ret); | ||
| 3723 | |||
| 3724 | if (path->locks[*level]) { | ||
| 3725 | btrfs_tree_unlock(path->nodes[*level]); | ||
| 3726 | path->locks[*level] = 0; | ||
| 3727 | } | ||
| 3728 | free_extent_buffer(path->nodes[*level]); | ||
| 3729 | path->nodes[*level] = NULL; | ||
| 3730 | *level += 1; | ||
| 3731 | cond_resched(); | ||
| 3732 | return 0; | ||
| 3733 | } | ||
| 3734 | |||
| 3735 | /* | ||
| 3736 | * helper for dropping snapshots. This walks back up the tree in the path | ||
| 3737 | * to find the first node higher up where we haven't yet gone through | ||
| 3738 | * all the slots | ||
| 3739 | */ | ||
| 3740 | static noinline int walk_up_tree(struct btrfs_trans_handle *trans, | ||
| 3741 | struct btrfs_root *root, | ||
| 3742 | struct btrfs_path *path, | ||
| 3743 | int *level, int max_level) | ||
| 3744 | { | ||
| 3745 | u64 root_owner; | ||
| 3746 | u64 root_gen; | ||
| 3747 | struct btrfs_root_item *root_item = &root->root_item; | ||
| 3748 | int i; | ||
| 3749 | int slot; | ||
| 3750 | int ret; | ||
| 3751 | |||
| 3752 | for (i = *level; i < max_level && path->nodes[i]; i++) { | ||
| 3753 | slot = path->slots[i]; | ||
| 3754 | if (slot < btrfs_header_nritems(path->nodes[i]) - 1) { | ||
| 3755 | struct extent_buffer *node; | ||
| 3756 | struct btrfs_disk_key disk_key; | ||
| 3757 | node = path->nodes[i]; | ||
| 3758 | path->slots[i]++; | ||
| 3759 | *level = i; | ||
| 3760 | WARN_ON(*level == 0); | ||
| 3761 | btrfs_node_key(node, &disk_key, path->slots[i]); | ||
| 3762 | memcpy(&root_item->drop_progress, | ||
| 3763 | &disk_key, sizeof(disk_key)); | ||
| 3764 | root_item->drop_level = i; | ||
| 3765 | return 0; | ||
| 3766 | } else { | ||
| 3767 | struct extent_buffer *parent; | ||
| 3768 | if (path->nodes[*level] == root->node) | ||
| 3769 | parent = path->nodes[*level]; | ||
| 3770 | else | ||
| 3771 | parent = path->nodes[*level + 1]; | ||
| 3772 | |||
| 3773 | root_owner = btrfs_header_owner(parent); | ||
| 3774 | root_gen = btrfs_header_generation(parent); | ||
| 3775 | |||
| 3776 | clean_tree_block(trans, root, path->nodes[*level]); | ||
| 3777 | ret = btrfs_free_extent(trans, root, | ||
| 3778 | path->nodes[*level]->start, | ||
| 3779 | path->nodes[*level]->len, | ||
| 3780 | parent->start, root_owner, | ||
| 3781 | root_gen, *level, 1); | ||
| 3782 | BUG_ON(ret); | ||
| 3783 | if (path->locks[*level]) { | ||
| 3784 | btrfs_tree_unlock(path->nodes[*level]); | ||
| 3785 | path->locks[*level] = 0; | ||
| 3786 | } | ||
| 3787 | free_extent_buffer(path->nodes[*level]); | ||
| 3788 | path->nodes[*level] = NULL; | ||
| 3789 | *level = i + 1; | ||
| 3790 | } | ||
| 3791 | } | ||
| 3792 | return 1; | ||
| 3793 | } | ||
| 3794 | |||
| 3795 | /* | ||
| 3796 | * drop the reference count on the tree rooted at 'snap'. This traverses | ||
| 3797 | * the tree freeing any blocks that have a ref count of zero after being | ||
| 3798 | * decremented. | ||
| 3799 | */ | ||
| 3800 | int btrfs_drop_snapshot(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 3801 | *root) | ||
| 3802 | { | ||
| 3803 | int ret = 0; | ||
| 3804 | int wret; | ||
| 3805 | int level; | ||
| 3806 | struct btrfs_path *path; | ||
| 3807 | int i; | ||
| 3808 | int orig_level; | ||
| 3809 | struct btrfs_root_item *root_item = &root->root_item; | ||
| 3810 | |||
| 3811 | WARN_ON(!mutex_is_locked(&root->fs_info->drop_mutex)); | ||
| 3812 | path = btrfs_alloc_path(); | ||
| 3813 | BUG_ON(!path); | ||
| 3814 | |||
| 3815 | level = btrfs_header_level(root->node); | ||
| 3816 | orig_level = level; | ||
| 3817 | if (btrfs_disk_key_objectid(&root_item->drop_progress) == 0) { | ||
| 3818 | path->nodes[level] = root->node; | ||
| 3819 | extent_buffer_get(root->node); | ||
| 3820 | path->slots[level] = 0; | ||
| 3821 | } else { | ||
| 3822 | struct btrfs_key key; | ||
| 3823 | struct btrfs_disk_key found_key; | ||
| 3824 | struct extent_buffer *node; | ||
| 3825 | |||
| 3826 | btrfs_disk_key_to_cpu(&key, &root_item->drop_progress); | ||
| 3827 | level = root_item->drop_level; | ||
| 3828 | path->lowest_level = level; | ||
| 3829 | wret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 3830 | if (wret < 0) { | ||
| 3831 | ret = wret; | ||
| 3832 | goto out; | ||
| 3833 | } | ||
| 3834 | node = path->nodes[level]; | ||
| 3835 | btrfs_node_key(node, &found_key, path->slots[level]); | ||
| 3836 | WARN_ON(memcmp(&found_key, &root_item->drop_progress, | ||
| 3837 | sizeof(found_key))); | ||
| 3838 | /* | ||
| 3839 | * unlock our path, this is safe because only this | ||
| 3840 | * function is allowed to delete this snapshot | ||
| 3841 | */ | ||
| 3842 | for (i = 0; i < BTRFS_MAX_LEVEL; i++) { | ||
| 3843 | if (path->nodes[i] && path->locks[i]) { | ||
| 3844 | path->locks[i] = 0; | ||
| 3845 | btrfs_tree_unlock(path->nodes[i]); | ||
| 3846 | } | ||
| 3847 | } | ||
| 3848 | } | ||
| 3849 | while (1) { | ||
| 3850 | wret = walk_down_tree(trans, root, path, &level); | ||
| 3851 | if (wret > 0) | ||
| 3852 | break; | ||
| 3853 | if (wret < 0) | ||
| 3854 | ret = wret; | ||
| 3855 | |||
| 3856 | wret = walk_up_tree(trans, root, path, &level, | ||
| 3857 | BTRFS_MAX_LEVEL); | ||
| 3858 | if (wret > 0) | ||
| 3859 | break; | ||
| 3860 | if (wret < 0) | ||
| 3861 | ret = wret; | ||
| 3862 | if (trans->transaction->in_commit) { | ||
| 3863 | ret = -EAGAIN; | ||
| 3864 | break; | ||
| 3865 | } | ||
| 3866 | atomic_inc(&root->fs_info->throttle_gen); | ||
| 3867 | wake_up(&root->fs_info->transaction_throttle); | ||
| 3868 | } | ||
| 3869 | for (i = 0; i <= orig_level; i++) { | ||
| 3870 | if (path->nodes[i]) { | ||
| 3871 | free_extent_buffer(path->nodes[i]); | ||
| 3872 | path->nodes[i] = NULL; | ||
| 3873 | } | ||
| 3874 | } | ||
| 3875 | out: | ||
| 3876 | btrfs_free_path(path); | ||
| 3877 | return ret; | ||
| 3878 | } | ||
| 3879 | |||
| 3880 | int btrfs_drop_subtree(struct btrfs_trans_handle *trans, | ||
| 3881 | struct btrfs_root *root, | ||
| 3882 | struct extent_buffer *node, | ||
| 3883 | struct extent_buffer *parent) | ||
| 3884 | { | ||
| 3885 | struct btrfs_path *path; | ||
| 3886 | int level; | ||
| 3887 | int parent_level; | ||
| 3888 | int ret = 0; | ||
| 3889 | int wret; | ||
| 3890 | |||
| 3891 | path = btrfs_alloc_path(); | ||
| 3892 | BUG_ON(!path); | ||
| 3893 | |||
| 3894 | BUG_ON(!btrfs_tree_locked(parent)); | ||
| 3895 | parent_level = btrfs_header_level(parent); | ||
| 3896 | extent_buffer_get(parent); | ||
| 3897 | path->nodes[parent_level] = parent; | ||
| 3898 | path->slots[parent_level] = btrfs_header_nritems(parent); | ||
| 3899 | |||
| 3900 | BUG_ON(!btrfs_tree_locked(node)); | ||
| 3901 | level = btrfs_header_level(node); | ||
| 3902 | extent_buffer_get(node); | ||
| 3903 | path->nodes[level] = node; | ||
| 3904 | path->slots[level] = 0; | ||
| 3905 | |||
| 3906 | while (1) { | ||
| 3907 | wret = walk_down_subtree(trans, root, path, &level); | ||
| 3908 | if (wret < 0) | ||
| 3909 | ret = wret; | ||
| 3910 | if (wret != 0) | ||
| 3911 | break; | ||
| 3912 | |||
| 3913 | wret = walk_up_tree(trans, root, path, &level, parent_level); | ||
| 3914 | if (wret < 0) | ||
| 3915 | ret = wret; | ||
| 3916 | if (wret != 0) | ||
| 3917 | break; | ||
| 3918 | } | ||
| 3919 | |||
| 3920 | btrfs_free_path(path); | ||
| 3921 | return ret; | ||
| 3922 | } | ||
| 3923 | |||
| 3924 | static unsigned long calc_ra(unsigned long start, unsigned long last, | ||
| 3925 | unsigned long nr) | ||
| 3926 | { | ||
| 3927 | return min(last, start + nr - 1); | ||
| 3928 | } | ||
| 3929 | |||
| 3930 | static noinline int relocate_inode_pages(struct inode *inode, u64 start, | ||
| 3931 | u64 len) | ||
| 3932 | { | ||
| 3933 | u64 page_start; | ||
| 3934 | u64 page_end; | ||
| 3935 | unsigned long first_index; | ||
| 3936 | unsigned long last_index; | ||
| 3937 | unsigned long i; | ||
| 3938 | struct page *page; | ||
| 3939 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 3940 | struct file_ra_state *ra; | ||
| 3941 | struct btrfs_ordered_extent *ordered; | ||
| 3942 | unsigned int total_read = 0; | ||
| 3943 | unsigned int total_dirty = 0; | ||
| 3944 | int ret = 0; | ||
| 3945 | |||
| 3946 | ra = kzalloc(sizeof(*ra), GFP_NOFS); | ||
| 3947 | |||
| 3948 | mutex_lock(&inode->i_mutex); | ||
| 3949 | first_index = start >> PAGE_CACHE_SHIFT; | ||
| 3950 | last_index = (start + len - 1) >> PAGE_CACHE_SHIFT; | ||
| 3951 | |||
| 3952 | /* make sure the dirty trick played by the caller work */ | ||
| 3953 | ret = invalidate_inode_pages2_range(inode->i_mapping, | ||
| 3954 | first_index, last_index); | ||
| 3955 | if (ret) | ||
| 3956 | goto out_unlock; | ||
| 3957 | |||
| 3958 | file_ra_state_init(ra, inode->i_mapping); | ||
| 3959 | |||
| 3960 | for (i = first_index ; i <= last_index; i++) { | ||
| 3961 | if (total_read % ra->ra_pages == 0) { | ||
| 3962 | btrfs_force_ra(inode->i_mapping, ra, NULL, i, | ||
| 3963 | calc_ra(i, last_index, ra->ra_pages)); | ||
| 3964 | } | ||
| 3965 | total_read++; | ||
| 3966 | again: | ||
| 3967 | if (((u64)i << PAGE_CACHE_SHIFT) > i_size_read(inode)) | ||
| 3968 | BUG_ON(1); | ||
| 3969 | page = grab_cache_page(inode->i_mapping, i); | ||
| 3970 | if (!page) { | ||
| 3971 | ret = -ENOMEM; | ||
| 3972 | goto out_unlock; | ||
| 3973 | } | ||
| 3974 | if (!PageUptodate(page)) { | ||
| 3975 | btrfs_readpage(NULL, page); | ||
| 3976 | lock_page(page); | ||
| 3977 | if (!PageUptodate(page)) { | ||
| 3978 | unlock_page(page); | ||
| 3979 | page_cache_release(page); | ||
| 3980 | ret = -EIO; | ||
| 3981 | goto out_unlock; | ||
| 3982 | } | ||
| 3983 | } | ||
| 3984 | wait_on_page_writeback(page); | ||
| 3985 | |||
| 3986 | page_start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 3987 | page_end = page_start + PAGE_CACHE_SIZE - 1; | ||
| 3988 | lock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 3989 | |||
| 3990 | ordered = btrfs_lookup_ordered_extent(inode, page_start); | ||
| 3991 | if (ordered) { | ||
| 3992 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 3993 | unlock_page(page); | ||
| 3994 | page_cache_release(page); | ||
| 3995 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 3996 | btrfs_put_ordered_extent(ordered); | ||
| 3997 | goto again; | ||
| 3998 | } | ||
| 3999 | set_page_extent_mapped(page); | ||
| 4000 | |||
| 4001 | if (i == first_index) | ||
| 4002 | set_extent_bits(io_tree, page_start, page_end, | ||
| 4003 | EXTENT_BOUNDARY, GFP_NOFS); | ||
| 4004 | btrfs_set_extent_delalloc(inode, page_start, page_end); | ||
| 4005 | |||
| 4006 | set_page_dirty(page); | ||
| 4007 | total_dirty++; | ||
| 4008 | |||
| 4009 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 4010 | unlock_page(page); | ||
| 4011 | page_cache_release(page); | ||
| 4012 | } | ||
| 4013 | |||
| 4014 | out_unlock: | ||
| 4015 | kfree(ra); | ||
| 4016 | mutex_unlock(&inode->i_mutex); | ||
| 4017 | balance_dirty_pages_ratelimited_nr(inode->i_mapping, total_dirty); | ||
| 4018 | return ret; | ||
| 4019 | } | ||
| 4020 | |||
| 4021 | static noinline int relocate_data_extent(struct inode *reloc_inode, | ||
| 4022 | struct btrfs_key *extent_key, | ||
| 4023 | u64 offset) | ||
| 4024 | { | ||
| 4025 | struct btrfs_root *root = BTRFS_I(reloc_inode)->root; | ||
| 4026 | struct extent_map_tree *em_tree = &BTRFS_I(reloc_inode)->extent_tree; | ||
| 4027 | struct extent_map *em; | ||
| 4028 | u64 start = extent_key->objectid - offset; | ||
| 4029 | u64 end = start + extent_key->offset - 1; | ||
| 4030 | |||
| 4031 | em = alloc_extent_map(GFP_NOFS); | ||
| 4032 | BUG_ON(!em || IS_ERR(em)); | ||
| 4033 | |||
| 4034 | em->start = start; | ||
| 4035 | em->len = extent_key->offset; | ||
| 4036 | em->block_len = extent_key->offset; | ||
| 4037 | em->block_start = extent_key->objectid; | ||
| 4038 | em->bdev = root->fs_info->fs_devices->latest_bdev; | ||
| 4039 | set_bit(EXTENT_FLAG_PINNED, &em->flags); | ||
| 4040 | |||
| 4041 | /* setup extent map to cheat btrfs_readpage */ | ||
| 4042 | lock_extent(&BTRFS_I(reloc_inode)->io_tree, start, end, GFP_NOFS); | ||
| 4043 | while (1) { | ||
| 4044 | int ret; | ||
| 4045 | spin_lock(&em_tree->lock); | ||
| 4046 | ret = add_extent_mapping(em_tree, em); | ||
| 4047 | spin_unlock(&em_tree->lock); | ||
| 4048 | if (ret != -EEXIST) { | ||
| 4049 | free_extent_map(em); | ||
| 4050 | break; | ||
| 4051 | } | ||
| 4052 | btrfs_drop_extent_cache(reloc_inode, start, end, 0); | ||
| 4053 | } | ||
| 4054 | unlock_extent(&BTRFS_I(reloc_inode)->io_tree, start, end, GFP_NOFS); | ||
| 4055 | |||
| 4056 | return relocate_inode_pages(reloc_inode, start, extent_key->offset); | ||
| 4057 | } | ||
| 4058 | |||
| 4059 | struct btrfs_ref_path { | ||
| 4060 | u64 extent_start; | ||
| 4061 | u64 nodes[BTRFS_MAX_LEVEL]; | ||
| 4062 | u64 root_objectid; | ||
| 4063 | u64 root_generation; | ||
| 4064 | u64 owner_objectid; | ||
| 4065 | u32 num_refs; | ||
| 4066 | int lowest_level; | ||
| 4067 | int current_level; | ||
| 4068 | int shared_level; | ||
| 4069 | |||
| 4070 | struct btrfs_key node_keys[BTRFS_MAX_LEVEL]; | ||
| 4071 | u64 new_nodes[BTRFS_MAX_LEVEL]; | ||
| 4072 | }; | ||
| 4073 | |||
| 4074 | struct disk_extent { | ||
| 4075 | u64 ram_bytes; | ||
| 4076 | u64 disk_bytenr; | ||
| 4077 | u64 disk_num_bytes; | ||
| 4078 | u64 offset; | ||
| 4079 | u64 num_bytes; | ||
| 4080 | u8 compression; | ||
| 4081 | u8 encryption; | ||
| 4082 | u16 other_encoding; | ||
| 4083 | }; | ||
| 4084 | |||
| 4085 | static int is_cowonly_root(u64 root_objectid) | ||
| 4086 | { | ||
| 4087 | if (root_objectid == BTRFS_ROOT_TREE_OBJECTID || | ||
| 4088 | root_objectid == BTRFS_EXTENT_TREE_OBJECTID || | ||
| 4089 | root_objectid == BTRFS_CHUNK_TREE_OBJECTID || | ||
| 4090 | root_objectid == BTRFS_DEV_TREE_OBJECTID || | ||
| 4091 | root_objectid == BTRFS_TREE_LOG_OBJECTID || | ||
| 4092 | root_objectid == BTRFS_CSUM_TREE_OBJECTID) | ||
| 4093 | return 1; | ||
| 4094 | return 0; | ||
| 4095 | } | ||
| 4096 | |||
| 4097 | static noinline int __next_ref_path(struct btrfs_trans_handle *trans, | ||
| 4098 | struct btrfs_root *extent_root, | ||
| 4099 | struct btrfs_ref_path *ref_path, | ||
| 4100 | int first_time) | ||
| 4101 | { | ||
| 4102 | struct extent_buffer *leaf; | ||
| 4103 | struct btrfs_path *path; | ||
| 4104 | struct btrfs_extent_ref *ref; | ||
| 4105 | struct btrfs_key key; | ||
| 4106 | struct btrfs_key found_key; | ||
| 4107 | u64 bytenr; | ||
| 4108 | u32 nritems; | ||
| 4109 | int level; | ||
| 4110 | int ret = 1; | ||
| 4111 | |||
| 4112 | path = btrfs_alloc_path(); | ||
| 4113 | if (!path) | ||
| 4114 | return -ENOMEM; | ||
| 4115 | |||
| 4116 | if (first_time) { | ||
| 4117 | ref_path->lowest_level = -1; | ||
| 4118 | ref_path->current_level = -1; | ||
| 4119 | ref_path->shared_level = -1; | ||
| 4120 | goto walk_up; | ||
| 4121 | } | ||
| 4122 | walk_down: | ||
| 4123 | level = ref_path->current_level - 1; | ||
| 4124 | while (level >= -1) { | ||
| 4125 | u64 parent; | ||
| 4126 | if (level < ref_path->lowest_level) | ||
| 4127 | break; | ||
| 4128 | |||
| 4129 | if (level >= 0) | ||
| 4130 | bytenr = ref_path->nodes[level]; | ||
| 4131 | else | ||
| 4132 | bytenr = ref_path->extent_start; | ||
| 4133 | BUG_ON(bytenr == 0); | ||
| 4134 | |||
| 4135 | parent = ref_path->nodes[level + 1]; | ||
| 4136 | ref_path->nodes[level + 1] = 0; | ||
| 4137 | ref_path->current_level = level; | ||
| 4138 | BUG_ON(parent == 0); | ||
| 4139 | |||
| 4140 | key.objectid = bytenr; | ||
| 4141 | key.offset = parent + 1; | ||
| 4142 | key.type = BTRFS_EXTENT_REF_KEY; | ||
| 4143 | |||
| 4144 | ret = btrfs_search_slot(trans, extent_root, &key, path, 0, 0); | ||
| 4145 | if (ret < 0) | ||
| 4146 | goto out; | ||
| 4147 | BUG_ON(ret == 0); | ||
| 4148 | |||
| 4149 | leaf = path->nodes[0]; | ||
| 4150 | nritems = btrfs_header_nritems(leaf); | ||
| 4151 | if (path->slots[0] >= nritems) { | ||
| 4152 | ret = btrfs_next_leaf(extent_root, path); | ||
| 4153 | if (ret < 0) | ||
| 4154 | goto out; | ||
| 4155 | if (ret > 0) | ||
| 4156 | goto next; | ||
| 4157 | leaf = path->nodes[0]; | ||
| 4158 | } | ||
| 4159 | |||
| 4160 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 4161 | if (found_key.objectid == bytenr && | ||
| 4162 | found_key.type == BTRFS_EXTENT_REF_KEY) { | ||
| 4163 | if (level < ref_path->shared_level) | ||
| 4164 | ref_path->shared_level = level; | ||
| 4165 | goto found; | ||
| 4166 | } | ||
| 4167 | next: | ||
| 4168 | level--; | ||
| 4169 | btrfs_release_path(extent_root, path); | ||
| 4170 | cond_resched(); | ||
| 4171 | } | ||
| 4172 | /* reached lowest level */ | ||
| 4173 | ret = 1; | ||
| 4174 | goto out; | ||
| 4175 | walk_up: | ||
| 4176 | level = ref_path->current_level; | ||
| 4177 | while (level < BTRFS_MAX_LEVEL - 1) { | ||
| 4178 | u64 ref_objectid; | ||
| 4179 | |||
| 4180 | if (level >= 0) | ||
| 4181 | bytenr = ref_path->nodes[level]; | ||
| 4182 | else | ||
| 4183 | bytenr = ref_path->extent_start; | ||
| 4184 | |||
| 4185 | BUG_ON(bytenr == 0); | ||
| 4186 | |||
| 4187 | key.objectid = bytenr; | ||
| 4188 | key.offset = 0; | ||
| 4189 | key.type = BTRFS_EXTENT_REF_KEY; | ||
| 4190 | |||
| 4191 | ret = btrfs_search_slot(trans, extent_root, &key, path, 0, 0); | ||
| 4192 | if (ret < 0) | ||
| 4193 | goto out; | ||
| 4194 | |||
| 4195 | leaf = path->nodes[0]; | ||
| 4196 | nritems = btrfs_header_nritems(leaf); | ||
| 4197 | if (path->slots[0] >= nritems) { | ||
| 4198 | ret = btrfs_next_leaf(extent_root, path); | ||
| 4199 | if (ret < 0) | ||
| 4200 | goto out; | ||
| 4201 | if (ret > 0) { | ||
| 4202 | /* the extent was freed by someone */ | ||
| 4203 | if (ref_path->lowest_level == level) | ||
| 4204 | goto out; | ||
| 4205 | btrfs_release_path(extent_root, path); | ||
| 4206 | goto walk_down; | ||
| 4207 | } | ||
| 4208 | leaf = path->nodes[0]; | ||
| 4209 | } | ||
| 4210 | |||
| 4211 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 4212 | if (found_key.objectid != bytenr || | ||
| 4213 | found_key.type != BTRFS_EXTENT_REF_KEY) { | ||
| 4214 | /* the extent was freed by someone */ | ||
| 4215 | if (ref_path->lowest_level == level) { | ||
| 4216 | ret = 1; | ||
| 4217 | goto out; | ||
| 4218 | } | ||
| 4219 | btrfs_release_path(extent_root, path); | ||
| 4220 | goto walk_down; | ||
| 4221 | } | ||
| 4222 | found: | ||
| 4223 | ref = btrfs_item_ptr(leaf, path->slots[0], | ||
| 4224 | struct btrfs_extent_ref); | ||
| 4225 | ref_objectid = btrfs_ref_objectid(leaf, ref); | ||
| 4226 | if (ref_objectid < BTRFS_FIRST_FREE_OBJECTID) { | ||
| 4227 | if (first_time) { | ||
| 4228 | level = (int)ref_objectid; | ||
| 4229 | BUG_ON(level >= BTRFS_MAX_LEVEL); | ||
| 4230 | ref_path->lowest_level = level; | ||
| 4231 | ref_path->current_level = level; | ||
| 4232 | ref_path->nodes[level] = bytenr; | ||
| 4233 | } else { | ||
| 4234 | WARN_ON(ref_objectid != level); | ||
| 4235 | } | ||
| 4236 | } else { | ||
| 4237 | WARN_ON(level != -1); | ||
| 4238 | } | ||
| 4239 | first_time = 0; | ||
| 4240 | |||
| 4241 | if (ref_path->lowest_level == level) { | ||
| 4242 | ref_path->owner_objectid = ref_objectid; | ||
| 4243 | ref_path->num_refs = btrfs_ref_num_refs(leaf, ref); | ||
| 4244 | } | ||
| 4245 | |||
| 4246 | /* | ||
| 4247 | * the block is tree root or the block isn't in reference | ||
| 4248 | * counted tree. | ||
| 4249 | */ | ||
| 4250 | if (found_key.objectid == found_key.offset || | ||
| 4251 | is_cowonly_root(btrfs_ref_root(leaf, ref))) { | ||
| 4252 | ref_path->root_objectid = btrfs_ref_root(leaf, ref); | ||
| 4253 | ref_path->root_generation = | ||
| 4254 | btrfs_ref_generation(leaf, ref); | ||
| 4255 | if (level < 0) { | ||
| 4256 | /* special reference from the tree log */ | ||
| 4257 | ref_path->nodes[0] = found_key.offset; | ||
| 4258 | ref_path->current_level = 0; | ||
| 4259 | } | ||
| 4260 | ret = 0; | ||
| 4261 | goto out; | ||
| 4262 | } | ||
| 4263 | |||
| 4264 | level++; | ||
| 4265 | BUG_ON(ref_path->nodes[level] != 0); | ||
| 4266 | ref_path->nodes[level] = found_key.offset; | ||
| 4267 | ref_path->current_level = level; | ||
| 4268 | |||
| 4269 | /* | ||
| 4270 | * the reference was created in the running transaction, | ||
| 4271 | * no need to continue walking up. | ||
| 4272 | */ | ||
| 4273 | if (btrfs_ref_generation(leaf, ref) == trans->transid) { | ||
| 4274 | ref_path->root_objectid = btrfs_ref_root(leaf, ref); | ||
| 4275 | ref_path->root_generation = | ||
| 4276 | btrfs_ref_generation(leaf, ref); | ||
| 4277 | ret = 0; | ||
| 4278 | goto out; | ||
| 4279 | } | ||
| 4280 | |||
| 4281 | btrfs_release_path(extent_root, path); | ||
| 4282 | cond_resched(); | ||
| 4283 | } | ||
| 4284 | /* reached max tree level, but no tree root found. */ | ||
| 4285 | BUG(); | ||
| 4286 | out: | ||
| 4287 | btrfs_free_path(path); | ||
| 4288 | return ret; | ||
| 4289 | } | ||
| 4290 | |||
| 4291 | static int btrfs_first_ref_path(struct btrfs_trans_handle *trans, | ||
| 4292 | struct btrfs_root *extent_root, | ||
| 4293 | struct btrfs_ref_path *ref_path, | ||
| 4294 | u64 extent_start) | ||
| 4295 | { | ||
| 4296 | memset(ref_path, 0, sizeof(*ref_path)); | ||
| 4297 | ref_path->extent_start = extent_start; | ||
| 4298 | |||
| 4299 | return __next_ref_path(trans, extent_root, ref_path, 1); | ||
| 4300 | } | ||
| 4301 | |||
| 4302 | static int btrfs_next_ref_path(struct btrfs_trans_handle *trans, | ||
| 4303 | struct btrfs_root *extent_root, | ||
| 4304 | struct btrfs_ref_path *ref_path) | ||
| 4305 | { | ||
| 4306 | return __next_ref_path(trans, extent_root, ref_path, 0); | ||
| 4307 | } | ||
| 4308 | |||
| 4309 | static noinline int get_new_locations(struct inode *reloc_inode, | ||
| 4310 | struct btrfs_key *extent_key, | ||
| 4311 | u64 offset, int no_fragment, | ||
| 4312 | struct disk_extent **extents, | ||
| 4313 | int *nr_extents) | ||
| 4314 | { | ||
| 4315 | struct btrfs_root *root = BTRFS_I(reloc_inode)->root; | ||
| 4316 | struct btrfs_path *path; | ||
| 4317 | struct btrfs_file_extent_item *fi; | ||
| 4318 | struct extent_buffer *leaf; | ||
| 4319 | struct disk_extent *exts = *extents; | ||
| 4320 | struct btrfs_key found_key; | ||
| 4321 | u64 cur_pos; | ||
| 4322 | u64 last_byte; | ||
| 4323 | u32 nritems; | ||
| 4324 | int nr = 0; | ||
| 4325 | int max = *nr_extents; | ||
| 4326 | int ret; | ||
| 4327 | |||
| 4328 | WARN_ON(!no_fragment && *extents); | ||
| 4329 | if (!exts) { | ||
| 4330 | max = 1; | ||
| 4331 | exts = kmalloc(sizeof(*exts) * max, GFP_NOFS); | ||
| 4332 | if (!exts) | ||
| 4333 | return -ENOMEM; | ||
| 4334 | } | ||
| 4335 | |||
| 4336 | path = btrfs_alloc_path(); | ||
| 4337 | BUG_ON(!path); | ||
| 4338 | |||
| 4339 | cur_pos = extent_key->objectid - offset; | ||
| 4340 | last_byte = extent_key->objectid + extent_key->offset; | ||
| 4341 | ret = btrfs_lookup_file_extent(NULL, root, path, reloc_inode->i_ino, | ||
| 4342 | cur_pos, 0); | ||
| 4343 | if (ret < 0) | ||
| 4344 | goto out; | ||
| 4345 | if (ret > 0) { | ||
| 4346 | ret = -ENOENT; | ||
| 4347 | goto out; | ||
| 4348 | } | ||
| 4349 | |||
| 4350 | while (1) { | ||
| 4351 | leaf = path->nodes[0]; | ||
| 4352 | nritems = btrfs_header_nritems(leaf); | ||
| 4353 | if (path->slots[0] >= nritems) { | ||
| 4354 | ret = btrfs_next_leaf(root, path); | ||
| 4355 | if (ret < 0) | ||
| 4356 | goto out; | ||
| 4357 | if (ret > 0) | ||
| 4358 | break; | ||
| 4359 | leaf = path->nodes[0]; | ||
| 4360 | } | ||
| 4361 | |||
| 4362 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 4363 | if (found_key.offset != cur_pos || | ||
| 4364 | found_key.type != BTRFS_EXTENT_DATA_KEY || | ||
| 4365 | found_key.objectid != reloc_inode->i_ino) | ||
| 4366 | break; | ||
| 4367 | |||
| 4368 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 4369 | struct btrfs_file_extent_item); | ||
| 4370 | if (btrfs_file_extent_type(leaf, fi) != | ||
| 4371 | BTRFS_FILE_EXTENT_REG || | ||
| 4372 | btrfs_file_extent_disk_bytenr(leaf, fi) == 0) | ||
| 4373 | break; | ||
| 4374 | |||
| 4375 | if (nr == max) { | ||
| 4376 | struct disk_extent *old = exts; | ||
| 4377 | max *= 2; | ||
| 4378 | exts = kzalloc(sizeof(*exts) * max, GFP_NOFS); | ||
| 4379 | memcpy(exts, old, sizeof(*exts) * nr); | ||
| 4380 | if (old != *extents) | ||
| 4381 | kfree(old); | ||
| 4382 | } | ||
| 4383 | |||
| 4384 | exts[nr].disk_bytenr = | ||
| 4385 | btrfs_file_extent_disk_bytenr(leaf, fi); | ||
| 4386 | exts[nr].disk_num_bytes = | ||
| 4387 | btrfs_file_extent_disk_num_bytes(leaf, fi); | ||
| 4388 | exts[nr].offset = btrfs_file_extent_offset(leaf, fi); | ||
| 4389 | exts[nr].num_bytes = btrfs_file_extent_num_bytes(leaf, fi); | ||
| 4390 | exts[nr].ram_bytes = btrfs_file_extent_ram_bytes(leaf, fi); | ||
| 4391 | exts[nr].compression = btrfs_file_extent_compression(leaf, fi); | ||
| 4392 | exts[nr].encryption = btrfs_file_extent_encryption(leaf, fi); | ||
| 4393 | exts[nr].other_encoding = btrfs_file_extent_other_encoding(leaf, | ||
| 4394 | fi); | ||
| 4395 | BUG_ON(exts[nr].offset > 0); | ||
| 4396 | BUG_ON(exts[nr].compression || exts[nr].encryption); | ||
| 4397 | BUG_ON(exts[nr].num_bytes != exts[nr].disk_num_bytes); | ||
| 4398 | |||
| 4399 | cur_pos += exts[nr].num_bytes; | ||
| 4400 | nr++; | ||
| 4401 | |||
| 4402 | if (cur_pos + offset >= last_byte) | ||
| 4403 | break; | ||
| 4404 | |||
| 4405 | if (no_fragment) { | ||
| 4406 | ret = 1; | ||
| 4407 | goto out; | ||
| 4408 | } | ||
| 4409 | path->slots[0]++; | ||
| 4410 | } | ||
| 4411 | |||
| 4412 | BUG_ON(cur_pos + offset > last_byte); | ||
| 4413 | if (cur_pos + offset < last_byte) { | ||
| 4414 | ret = -ENOENT; | ||
| 4415 | goto out; | ||
| 4416 | } | ||
| 4417 | ret = 0; | ||
| 4418 | out: | ||
| 4419 | btrfs_free_path(path); | ||
| 4420 | if (ret) { | ||
| 4421 | if (exts != *extents) | ||
| 4422 | kfree(exts); | ||
| 4423 | } else { | ||
| 4424 | *extents = exts; | ||
| 4425 | *nr_extents = nr; | ||
| 4426 | } | ||
| 4427 | return ret; | ||
| 4428 | } | ||
| 4429 | |||
| 4430 | static noinline int replace_one_extent(struct btrfs_trans_handle *trans, | ||
| 4431 | struct btrfs_root *root, | ||
| 4432 | struct btrfs_path *path, | ||
| 4433 | struct btrfs_key *extent_key, | ||
| 4434 | struct btrfs_key *leaf_key, | ||
| 4435 | struct btrfs_ref_path *ref_path, | ||
| 4436 | struct disk_extent *new_extents, | ||
| 4437 | int nr_extents) | ||
| 4438 | { | ||
| 4439 | struct extent_buffer *leaf; | ||
| 4440 | struct btrfs_file_extent_item *fi; | ||
| 4441 | struct inode *inode = NULL; | ||
| 4442 | struct btrfs_key key; | ||
| 4443 | u64 lock_start = 0; | ||
| 4444 | u64 lock_end = 0; | ||
| 4445 | u64 num_bytes; | ||
| 4446 | u64 ext_offset; | ||
| 4447 | u64 first_pos; | ||
| 4448 | u32 nritems; | ||
| 4449 | int nr_scaned = 0; | ||
| 4450 | int extent_locked = 0; | ||
| 4451 | int extent_type; | ||
| 4452 | int ret; | ||
| 4453 | |||
| 4454 | memcpy(&key, leaf_key, sizeof(key)); | ||
| 4455 | first_pos = INT_LIMIT(loff_t) - extent_key->offset; | ||
| 4456 | if (ref_path->owner_objectid != BTRFS_MULTIPLE_OBJECTIDS) { | ||
| 4457 | if (key.objectid < ref_path->owner_objectid || | ||
| 4458 | (key.objectid == ref_path->owner_objectid && | ||
| 4459 | key.type < BTRFS_EXTENT_DATA_KEY)) { | ||
| 4460 | key.objectid = ref_path->owner_objectid; | ||
| 4461 | key.type = BTRFS_EXTENT_DATA_KEY; | ||
| 4462 | key.offset = 0; | ||
| 4463 | } | ||
| 4464 | } | ||
| 4465 | |||
| 4466 | while (1) { | ||
| 4467 | ret = btrfs_search_slot(trans, root, &key, path, 0, 1); | ||
| 4468 | if (ret < 0) | ||
| 4469 | goto out; | ||
| 4470 | |||
| 4471 | leaf = path->nodes[0]; | ||
| 4472 | nritems = btrfs_header_nritems(leaf); | ||
| 4473 | next: | ||
| 4474 | if (extent_locked && ret > 0) { | ||
| 4475 | /* | ||
| 4476 | * the file extent item was modified by someone | ||
| 4477 | * before the extent got locked. | ||
| 4478 | */ | ||
| 4479 | unlock_extent(&BTRFS_I(inode)->io_tree, lock_start, | ||
| 4480 | lock_end, GFP_NOFS); | ||
| 4481 | extent_locked = 0; | ||
| 4482 | } | ||
| 4483 | |||
| 4484 | if (path->slots[0] >= nritems) { | ||
| 4485 | if (++nr_scaned > 2) | ||
| 4486 | break; | ||
| 4487 | |||
| 4488 | BUG_ON(extent_locked); | ||
| 4489 | ret = btrfs_next_leaf(root, path); | ||
| 4490 | if (ret < 0) | ||
| 4491 | goto out; | ||
| 4492 | if (ret > 0) | ||
| 4493 | break; | ||
| 4494 | leaf = path->nodes[0]; | ||
| 4495 | nritems = btrfs_header_nritems(leaf); | ||
| 4496 | } | ||
| 4497 | |||
| 4498 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 4499 | |||
| 4500 | if (ref_path->owner_objectid != BTRFS_MULTIPLE_OBJECTIDS) { | ||
| 4501 | if ((key.objectid > ref_path->owner_objectid) || | ||
| 4502 | (key.objectid == ref_path->owner_objectid && | ||
| 4503 | key.type > BTRFS_EXTENT_DATA_KEY) || | ||
| 4504 | (key.offset >= first_pos + extent_key->offset)) | ||
| 4505 | break; | ||
| 4506 | } | ||
| 4507 | |||
| 4508 | if (inode && key.objectid != inode->i_ino) { | ||
| 4509 | BUG_ON(extent_locked); | ||
| 4510 | btrfs_release_path(root, path); | ||
| 4511 | mutex_unlock(&inode->i_mutex); | ||
| 4512 | iput(inode); | ||
| 4513 | inode = NULL; | ||
| 4514 | continue; | ||
| 4515 | } | ||
| 4516 | |||
| 4517 | if (key.type != BTRFS_EXTENT_DATA_KEY) { | ||
| 4518 | path->slots[0]++; | ||
| 4519 | ret = 1; | ||
| 4520 | goto next; | ||
| 4521 | } | ||
| 4522 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 4523 | struct btrfs_file_extent_item); | ||
| 4524 | extent_type = btrfs_file_extent_type(leaf, fi); | ||
| 4525 | if ((extent_type != BTRFS_FILE_EXTENT_REG && | ||
| 4526 | extent_type != BTRFS_FILE_EXTENT_PREALLOC) || | ||
| 4527 | (btrfs_file_extent_disk_bytenr(leaf, fi) != | ||
| 4528 | extent_key->objectid)) { | ||
| 4529 | path->slots[0]++; | ||
| 4530 | ret = 1; | ||
| 4531 | goto next; | ||
| 4532 | } | ||
| 4533 | |||
| 4534 | num_bytes = btrfs_file_extent_num_bytes(leaf, fi); | ||
| 4535 | ext_offset = btrfs_file_extent_offset(leaf, fi); | ||
| 4536 | |||
| 4537 | if (first_pos > key.offset - ext_offset) | ||
| 4538 | first_pos = key.offset - ext_offset; | ||
| 4539 | |||
| 4540 | if (!extent_locked) { | ||
| 4541 | lock_start = key.offset; | ||
| 4542 | lock_end = lock_start + num_bytes - 1; | ||
| 4543 | } else { | ||
| 4544 | if (lock_start > key.offset || | ||
| 4545 | lock_end + 1 < key.offset + num_bytes) { | ||
| 4546 | unlock_extent(&BTRFS_I(inode)->io_tree, | ||
| 4547 | lock_start, lock_end, GFP_NOFS); | ||
| 4548 | extent_locked = 0; | ||
| 4549 | } | ||
| 4550 | } | ||
| 4551 | |||
| 4552 | if (!inode) { | ||
| 4553 | btrfs_release_path(root, path); | ||
| 4554 | |||
| 4555 | inode = btrfs_iget_locked(root->fs_info->sb, | ||
| 4556 | key.objectid, root); | ||
| 4557 | if (inode->i_state & I_NEW) { | ||
| 4558 | BTRFS_I(inode)->root = root; | ||
| 4559 | BTRFS_I(inode)->location.objectid = | ||
| 4560 | key.objectid; | ||
| 4561 | BTRFS_I(inode)->location.type = | ||
| 4562 | BTRFS_INODE_ITEM_KEY; | ||
| 4563 | BTRFS_I(inode)->location.offset = 0; | ||
| 4564 | btrfs_read_locked_inode(inode); | ||
| 4565 | unlock_new_inode(inode); | ||
| 4566 | } | ||
| 4567 | /* | ||
| 4568 | * some code call btrfs_commit_transaction while | ||
| 4569 | * holding the i_mutex, so we can't use mutex_lock | ||
| 4570 | * here. | ||
| 4571 | */ | ||
| 4572 | if (is_bad_inode(inode) || | ||
| 4573 | !mutex_trylock(&inode->i_mutex)) { | ||
| 4574 | iput(inode); | ||
| 4575 | inode = NULL; | ||
| 4576 | key.offset = (u64)-1; | ||
| 4577 | goto skip; | ||
| 4578 | } | ||
| 4579 | } | ||
| 4580 | |||
| 4581 | if (!extent_locked) { | ||
| 4582 | struct btrfs_ordered_extent *ordered; | ||
| 4583 | |||
| 4584 | btrfs_release_path(root, path); | ||
| 4585 | |||
| 4586 | lock_extent(&BTRFS_I(inode)->io_tree, lock_start, | ||
| 4587 | lock_end, GFP_NOFS); | ||
| 4588 | ordered = btrfs_lookup_first_ordered_extent(inode, | ||
| 4589 | lock_end); | ||
| 4590 | if (ordered && | ||
| 4591 | ordered->file_offset <= lock_end && | ||
| 4592 | ordered->file_offset + ordered->len > lock_start) { | ||
| 4593 | unlock_extent(&BTRFS_I(inode)->io_tree, | ||
| 4594 | lock_start, lock_end, GFP_NOFS); | ||
| 4595 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 4596 | btrfs_put_ordered_extent(ordered); | ||
| 4597 | key.offset += num_bytes; | ||
| 4598 | goto skip; | ||
| 4599 | } | ||
| 4600 | if (ordered) | ||
| 4601 | btrfs_put_ordered_extent(ordered); | ||
| 4602 | |||
| 4603 | extent_locked = 1; | ||
| 4604 | continue; | ||
| 4605 | } | ||
| 4606 | |||
| 4607 | if (nr_extents == 1) { | ||
| 4608 | /* update extent pointer in place */ | ||
| 4609 | btrfs_set_file_extent_disk_bytenr(leaf, fi, | ||
| 4610 | new_extents[0].disk_bytenr); | ||
| 4611 | btrfs_set_file_extent_disk_num_bytes(leaf, fi, | ||
| 4612 | new_extents[0].disk_num_bytes); | ||
| 4613 | btrfs_mark_buffer_dirty(leaf); | ||
| 4614 | |||
| 4615 | btrfs_drop_extent_cache(inode, key.offset, | ||
| 4616 | key.offset + num_bytes - 1, 0); | ||
| 4617 | |||
| 4618 | ret = btrfs_inc_extent_ref(trans, root, | ||
| 4619 | new_extents[0].disk_bytenr, | ||
| 4620 | new_extents[0].disk_num_bytes, | ||
| 4621 | leaf->start, | ||
| 4622 | root->root_key.objectid, | ||
| 4623 | trans->transid, | ||
| 4624 | key.objectid); | ||
| 4625 | BUG_ON(ret); | ||
| 4626 | |||
| 4627 | ret = btrfs_free_extent(trans, root, | ||
| 4628 | extent_key->objectid, | ||
| 4629 | extent_key->offset, | ||
| 4630 | leaf->start, | ||
| 4631 | btrfs_header_owner(leaf), | ||
| 4632 | btrfs_header_generation(leaf), | ||
| 4633 | key.objectid, 0); | ||
| 4634 | BUG_ON(ret); | ||
| 4635 | |||
| 4636 | btrfs_release_path(root, path); | ||
| 4637 | key.offset += num_bytes; | ||
| 4638 | } else { | ||
| 4639 | BUG_ON(1); | ||
| 4640 | #if 0 | ||
| 4641 | u64 alloc_hint; | ||
| 4642 | u64 extent_len; | ||
| 4643 | int i; | ||
| 4644 | /* | ||
| 4645 | * drop old extent pointer at first, then insert the | ||
| 4646 | * new pointers one bye one | ||
| 4647 | */ | ||
| 4648 | btrfs_release_path(root, path); | ||
| 4649 | ret = btrfs_drop_extents(trans, root, inode, key.offset, | ||
| 4650 | key.offset + num_bytes, | ||
| 4651 | key.offset, &alloc_hint); | ||
| 4652 | BUG_ON(ret); | ||
| 4653 | |||
| 4654 | for (i = 0; i < nr_extents; i++) { | ||
| 4655 | if (ext_offset >= new_extents[i].num_bytes) { | ||
| 4656 | ext_offset -= new_extents[i].num_bytes; | ||
| 4657 | continue; | ||
| 4658 | } | ||
| 4659 | extent_len = min(new_extents[i].num_bytes - | ||
| 4660 | ext_offset, num_bytes); | ||
| 4661 | |||
| 4662 | ret = btrfs_insert_empty_item(trans, root, | ||
| 4663 | path, &key, | ||
| 4664 | sizeof(*fi)); | ||
| 4665 | BUG_ON(ret); | ||
| 4666 | |||
| 4667 | leaf = path->nodes[0]; | ||
| 4668 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 4669 | struct btrfs_file_extent_item); | ||
| 4670 | btrfs_set_file_extent_generation(leaf, fi, | ||
| 4671 | trans->transid); | ||
| 4672 | btrfs_set_file_extent_type(leaf, fi, | ||
| 4673 | BTRFS_FILE_EXTENT_REG); | ||
| 4674 | btrfs_set_file_extent_disk_bytenr(leaf, fi, | ||
| 4675 | new_extents[i].disk_bytenr); | ||
| 4676 | btrfs_set_file_extent_disk_num_bytes(leaf, fi, | ||
| 4677 | new_extents[i].disk_num_bytes); | ||
| 4678 | btrfs_set_file_extent_ram_bytes(leaf, fi, | ||
| 4679 | new_extents[i].ram_bytes); | ||
| 4680 | |||
| 4681 | btrfs_set_file_extent_compression(leaf, fi, | ||
| 4682 | new_extents[i].compression); | ||
| 4683 | btrfs_set_file_extent_encryption(leaf, fi, | ||
| 4684 | new_extents[i].encryption); | ||
| 4685 | btrfs_set_file_extent_other_encoding(leaf, fi, | ||
| 4686 | new_extents[i].other_encoding); | ||
| 4687 | |||
| 4688 | btrfs_set_file_extent_num_bytes(leaf, fi, | ||
| 4689 | extent_len); | ||
| 4690 | ext_offset += new_extents[i].offset; | ||
| 4691 | btrfs_set_file_extent_offset(leaf, fi, | ||
| 4692 | ext_offset); | ||
| 4693 | btrfs_mark_buffer_dirty(leaf); | ||
| 4694 | |||
| 4695 | btrfs_drop_extent_cache(inode, key.offset, | ||
| 4696 | key.offset + extent_len - 1, 0); | ||
| 4697 | |||
| 4698 | ret = btrfs_inc_extent_ref(trans, root, | ||
| 4699 | new_extents[i].disk_bytenr, | ||
| 4700 | new_extents[i].disk_num_bytes, | ||
| 4701 | leaf->start, | ||
| 4702 | root->root_key.objectid, | ||
| 4703 | trans->transid, key.objectid); | ||
| 4704 | BUG_ON(ret); | ||
| 4705 | btrfs_release_path(root, path); | ||
| 4706 | |||
| 4707 | inode_add_bytes(inode, extent_len); | ||
| 4708 | |||
| 4709 | ext_offset = 0; | ||
| 4710 | num_bytes -= extent_len; | ||
| 4711 | key.offset += extent_len; | ||
| 4712 | |||
| 4713 | if (num_bytes == 0) | ||
| 4714 | break; | ||
| 4715 | } | ||
| 4716 | BUG_ON(i >= nr_extents); | ||
| 4717 | #endif | ||
| 4718 | } | ||
| 4719 | |||
| 4720 | if (extent_locked) { | ||
| 4721 | unlock_extent(&BTRFS_I(inode)->io_tree, lock_start, | ||
| 4722 | lock_end, GFP_NOFS); | ||
| 4723 | extent_locked = 0; | ||
| 4724 | } | ||
| 4725 | skip: | ||
| 4726 | if (ref_path->owner_objectid != BTRFS_MULTIPLE_OBJECTIDS && | ||
| 4727 | key.offset >= first_pos + extent_key->offset) | ||
| 4728 | break; | ||
| 4729 | |||
| 4730 | cond_resched(); | ||
| 4731 | } | ||
| 4732 | ret = 0; | ||
| 4733 | out: | ||
| 4734 | btrfs_release_path(root, path); | ||
| 4735 | if (inode) { | ||
| 4736 | mutex_unlock(&inode->i_mutex); | ||
| 4737 | if (extent_locked) { | ||
| 4738 | unlock_extent(&BTRFS_I(inode)->io_tree, lock_start, | ||
| 4739 | lock_end, GFP_NOFS); | ||
| 4740 | } | ||
| 4741 | iput(inode); | ||
| 4742 | } | ||
| 4743 | return ret; | ||
| 4744 | } | ||
| 4745 | |||
| 4746 | int btrfs_reloc_tree_cache_ref(struct btrfs_trans_handle *trans, | ||
| 4747 | struct btrfs_root *root, | ||
| 4748 | struct extent_buffer *buf, u64 orig_start) | ||
| 4749 | { | ||
| 4750 | int level; | ||
| 4751 | int ret; | ||
| 4752 | |||
| 4753 | BUG_ON(btrfs_header_generation(buf) != trans->transid); | ||
| 4754 | BUG_ON(root->root_key.objectid != BTRFS_TREE_RELOC_OBJECTID); | ||
| 4755 | |||
| 4756 | level = btrfs_header_level(buf); | ||
| 4757 | if (level == 0) { | ||
| 4758 | struct btrfs_leaf_ref *ref; | ||
| 4759 | struct btrfs_leaf_ref *orig_ref; | ||
| 4760 | |||
| 4761 | orig_ref = btrfs_lookup_leaf_ref(root, orig_start); | ||
| 4762 | if (!orig_ref) | ||
| 4763 | return -ENOENT; | ||
| 4764 | |||
| 4765 | ref = btrfs_alloc_leaf_ref(root, orig_ref->nritems); | ||
| 4766 | if (!ref) { | ||
| 4767 | btrfs_free_leaf_ref(root, orig_ref); | ||
| 4768 | return -ENOMEM; | ||
| 4769 | } | ||
| 4770 | |||
| 4771 | ref->nritems = orig_ref->nritems; | ||
| 4772 | memcpy(ref->extents, orig_ref->extents, | ||
| 4773 | sizeof(ref->extents[0]) * ref->nritems); | ||
| 4774 | |||
| 4775 | btrfs_free_leaf_ref(root, orig_ref); | ||
| 4776 | |||
| 4777 | ref->root_gen = trans->transid; | ||
| 4778 | ref->bytenr = buf->start; | ||
| 4779 | ref->owner = btrfs_header_owner(buf); | ||
| 4780 | ref->generation = btrfs_header_generation(buf); | ||
| 4781 | ret = btrfs_add_leaf_ref(root, ref, 0); | ||
| 4782 | WARN_ON(ret); | ||
| 4783 | btrfs_free_leaf_ref(root, ref); | ||
| 4784 | } | ||
| 4785 | return 0; | ||
| 4786 | } | ||
| 4787 | |||
| 4788 | static noinline int invalidate_extent_cache(struct btrfs_root *root, | ||
| 4789 | struct extent_buffer *leaf, | ||
| 4790 | struct btrfs_block_group_cache *group, | ||
| 4791 | struct btrfs_root *target_root) | ||
| 4792 | { | ||
| 4793 | struct btrfs_key key; | ||
| 4794 | struct inode *inode = NULL; | ||
| 4795 | struct btrfs_file_extent_item *fi; | ||
| 4796 | u64 num_bytes; | ||
| 4797 | u64 skip_objectid = 0; | ||
| 4798 | u32 nritems; | ||
| 4799 | u32 i; | ||
| 4800 | |||
| 4801 | nritems = btrfs_header_nritems(leaf); | ||
| 4802 | for (i = 0; i < nritems; i++) { | ||
| 4803 | btrfs_item_key_to_cpu(leaf, &key, i); | ||
| 4804 | if (key.objectid == skip_objectid || | ||
| 4805 | key.type != BTRFS_EXTENT_DATA_KEY) | ||
| 4806 | continue; | ||
| 4807 | fi = btrfs_item_ptr(leaf, i, struct btrfs_file_extent_item); | ||
| 4808 | if (btrfs_file_extent_type(leaf, fi) == | ||
| 4809 | BTRFS_FILE_EXTENT_INLINE) | ||
| 4810 | continue; | ||
| 4811 | if (btrfs_file_extent_disk_bytenr(leaf, fi) == 0) | ||
| 4812 | continue; | ||
| 4813 | if (!inode || inode->i_ino != key.objectid) { | ||
| 4814 | iput(inode); | ||
| 4815 | inode = btrfs_ilookup(target_root->fs_info->sb, | ||
| 4816 | key.objectid, target_root, 1); | ||
| 4817 | } | ||
| 4818 | if (!inode) { | ||
| 4819 | skip_objectid = key.objectid; | ||
| 4820 | continue; | ||
| 4821 | } | ||
| 4822 | num_bytes = btrfs_file_extent_num_bytes(leaf, fi); | ||
| 4823 | |||
| 4824 | lock_extent(&BTRFS_I(inode)->io_tree, key.offset, | ||
| 4825 | key.offset + num_bytes - 1, GFP_NOFS); | ||
| 4826 | btrfs_drop_extent_cache(inode, key.offset, | ||
| 4827 | key.offset + num_bytes - 1, 1); | ||
| 4828 | unlock_extent(&BTRFS_I(inode)->io_tree, key.offset, | ||
| 4829 | key.offset + num_bytes - 1, GFP_NOFS); | ||
| 4830 | cond_resched(); | ||
| 4831 | } | ||
| 4832 | iput(inode); | ||
| 4833 | return 0; | ||
| 4834 | } | ||
| 4835 | |||
| 4836 | static noinline int replace_extents_in_leaf(struct btrfs_trans_handle *trans, | ||
| 4837 | struct btrfs_root *root, | ||
| 4838 | struct extent_buffer *leaf, | ||
| 4839 | struct btrfs_block_group_cache *group, | ||
| 4840 | struct inode *reloc_inode) | ||
| 4841 | { | ||
| 4842 | struct btrfs_key key; | ||
| 4843 | struct btrfs_key extent_key; | ||
| 4844 | struct btrfs_file_extent_item *fi; | ||
| 4845 | struct btrfs_leaf_ref *ref; | ||
| 4846 | struct disk_extent *new_extent; | ||
| 4847 | u64 bytenr; | ||
| 4848 | u64 num_bytes; | ||
| 4849 | u32 nritems; | ||
| 4850 | u32 i; | ||
| 4851 | int ext_index; | ||
| 4852 | int nr_extent; | ||
| 4853 | int ret; | ||
| 4854 | |||
| 4855 | new_extent = kmalloc(sizeof(*new_extent), GFP_NOFS); | ||
| 4856 | BUG_ON(!new_extent); | ||
| 4857 | |||
| 4858 | ref = btrfs_lookup_leaf_ref(root, leaf->start); | ||
| 4859 | BUG_ON(!ref); | ||
| 4860 | |||
| 4861 | ext_index = -1; | ||
| 4862 | nritems = btrfs_header_nritems(leaf); | ||
| 4863 | for (i = 0; i < nritems; i++) { | ||
| 4864 | btrfs_item_key_to_cpu(leaf, &key, i); | ||
| 4865 | if (btrfs_key_type(&key) != BTRFS_EXTENT_DATA_KEY) | ||
| 4866 | continue; | ||
| 4867 | fi = btrfs_item_ptr(leaf, i, struct btrfs_file_extent_item); | ||
| 4868 | if (btrfs_file_extent_type(leaf, fi) == | ||
| 4869 | BTRFS_FILE_EXTENT_INLINE) | ||
| 4870 | continue; | ||
| 4871 | bytenr = btrfs_file_extent_disk_bytenr(leaf, fi); | ||
| 4872 | num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi); | ||
| 4873 | if (bytenr == 0) | ||
| 4874 | continue; | ||
| 4875 | |||
| 4876 | ext_index++; | ||
| 4877 | if (bytenr >= group->key.objectid + group->key.offset || | ||
| 4878 | bytenr + num_bytes <= group->key.objectid) | ||
| 4879 | continue; | ||
| 4880 | |||
| 4881 | extent_key.objectid = bytenr; | ||
| 4882 | extent_key.offset = num_bytes; | ||
| 4883 | extent_key.type = BTRFS_EXTENT_ITEM_KEY; | ||
| 4884 | nr_extent = 1; | ||
| 4885 | ret = get_new_locations(reloc_inode, &extent_key, | ||
| 4886 | group->key.objectid, 1, | ||
| 4887 | &new_extent, &nr_extent); | ||
| 4888 | if (ret > 0) | ||
| 4889 | continue; | ||
| 4890 | BUG_ON(ret < 0); | ||
| 4891 | |||
| 4892 | BUG_ON(ref->extents[ext_index].bytenr != bytenr); | ||
| 4893 | BUG_ON(ref->extents[ext_index].num_bytes != num_bytes); | ||
| 4894 | ref->extents[ext_index].bytenr = new_extent->disk_bytenr; | ||
| 4895 | ref->extents[ext_index].num_bytes = new_extent->disk_num_bytes; | ||
| 4896 | |||
| 4897 | btrfs_set_file_extent_disk_bytenr(leaf, fi, | ||
| 4898 | new_extent->disk_bytenr); | ||
| 4899 | btrfs_set_file_extent_disk_num_bytes(leaf, fi, | ||
| 4900 | new_extent->disk_num_bytes); | ||
| 4901 | btrfs_mark_buffer_dirty(leaf); | ||
| 4902 | |||
| 4903 | ret = btrfs_inc_extent_ref(trans, root, | ||
| 4904 | new_extent->disk_bytenr, | ||
| 4905 | new_extent->disk_num_bytes, | ||
| 4906 | leaf->start, | ||
| 4907 | root->root_key.objectid, | ||
| 4908 | trans->transid, key.objectid); | ||
| 4909 | BUG_ON(ret); | ||
| 4910 | ret = btrfs_free_extent(trans, root, | ||
| 4911 | bytenr, num_bytes, leaf->start, | ||
| 4912 | btrfs_header_owner(leaf), | ||
| 4913 | btrfs_header_generation(leaf), | ||
| 4914 | key.objectid, 0); | ||
| 4915 | BUG_ON(ret); | ||
| 4916 | cond_resched(); | ||
| 4917 | } | ||
| 4918 | kfree(new_extent); | ||
| 4919 | BUG_ON(ext_index + 1 != ref->nritems); | ||
| 4920 | btrfs_free_leaf_ref(root, ref); | ||
| 4921 | return 0; | ||
| 4922 | } | ||
| 4923 | |||
| 4924 | int btrfs_free_reloc_root(struct btrfs_trans_handle *trans, | ||
| 4925 | struct btrfs_root *root) | ||
| 4926 | { | ||
| 4927 | struct btrfs_root *reloc_root; | ||
| 4928 | int ret; | ||
| 4929 | |||
| 4930 | if (root->reloc_root) { | ||
| 4931 | reloc_root = root->reloc_root; | ||
| 4932 | root->reloc_root = NULL; | ||
| 4933 | list_add(&reloc_root->dead_list, | ||
| 4934 | &root->fs_info->dead_reloc_roots); | ||
| 4935 | |||
| 4936 | btrfs_set_root_bytenr(&reloc_root->root_item, | ||
| 4937 | reloc_root->node->start); | ||
| 4938 | btrfs_set_root_level(&root->root_item, | ||
| 4939 | btrfs_header_level(reloc_root->node)); | ||
| 4940 | memset(&reloc_root->root_item.drop_progress, 0, | ||
| 4941 | sizeof(struct btrfs_disk_key)); | ||
| 4942 | reloc_root->root_item.drop_level = 0; | ||
| 4943 | |||
| 4944 | ret = btrfs_update_root(trans, root->fs_info->tree_root, | ||
| 4945 | &reloc_root->root_key, | ||
| 4946 | &reloc_root->root_item); | ||
| 4947 | BUG_ON(ret); | ||
| 4948 | } | ||
| 4949 | return 0; | ||
| 4950 | } | ||
| 4951 | |||
| 4952 | int btrfs_drop_dead_reloc_roots(struct btrfs_root *root) | ||
| 4953 | { | ||
| 4954 | struct btrfs_trans_handle *trans; | ||
| 4955 | struct btrfs_root *reloc_root; | ||
| 4956 | struct btrfs_root *prev_root = NULL; | ||
| 4957 | struct list_head dead_roots; | ||
| 4958 | int ret; | ||
| 4959 | unsigned long nr; | ||
| 4960 | |||
| 4961 | INIT_LIST_HEAD(&dead_roots); | ||
| 4962 | list_splice_init(&root->fs_info->dead_reloc_roots, &dead_roots); | ||
| 4963 | |||
| 4964 | while (!list_empty(&dead_roots)) { | ||
| 4965 | reloc_root = list_entry(dead_roots.prev, | ||
| 4966 | struct btrfs_root, dead_list); | ||
| 4967 | list_del_init(&reloc_root->dead_list); | ||
| 4968 | |||
| 4969 | BUG_ON(reloc_root->commit_root != NULL); | ||
| 4970 | while (1) { | ||
| 4971 | trans = btrfs_join_transaction(root, 1); | ||
| 4972 | BUG_ON(!trans); | ||
| 4973 | |||
| 4974 | mutex_lock(&root->fs_info->drop_mutex); | ||
| 4975 | ret = btrfs_drop_snapshot(trans, reloc_root); | ||
| 4976 | if (ret != -EAGAIN) | ||
| 4977 | break; | ||
| 4978 | mutex_unlock(&root->fs_info->drop_mutex); | ||
| 4979 | |||
| 4980 | nr = trans->blocks_used; | ||
| 4981 | ret = btrfs_end_transaction(trans, root); | ||
| 4982 | BUG_ON(ret); | ||
| 4983 | btrfs_btree_balance_dirty(root, nr); | ||
| 4984 | } | ||
| 4985 | |||
| 4986 | free_extent_buffer(reloc_root->node); | ||
| 4987 | |||
| 4988 | ret = btrfs_del_root(trans, root->fs_info->tree_root, | ||
| 4989 | &reloc_root->root_key); | ||
| 4990 | BUG_ON(ret); | ||
| 4991 | mutex_unlock(&root->fs_info->drop_mutex); | ||
| 4992 | |||
| 4993 | nr = trans->blocks_used; | ||
| 4994 | ret = btrfs_end_transaction(trans, root); | ||
| 4995 | BUG_ON(ret); | ||
| 4996 | btrfs_btree_balance_dirty(root, nr); | ||
| 4997 | |||
| 4998 | kfree(prev_root); | ||
| 4999 | prev_root = reloc_root; | ||
| 5000 | } | ||
| 5001 | if (prev_root) { | ||
| 5002 | btrfs_remove_leaf_refs(prev_root, (u64)-1, 0); | ||
| 5003 | kfree(prev_root); | ||
| 5004 | } | ||
| 5005 | return 0; | ||
| 5006 | } | ||
| 5007 | |||
| 5008 | int btrfs_add_dead_reloc_root(struct btrfs_root *root) | ||
| 5009 | { | ||
| 5010 | list_add(&root->dead_list, &root->fs_info->dead_reloc_roots); | ||
| 5011 | return 0; | ||
| 5012 | } | ||
| 5013 | |||
| 5014 | int btrfs_cleanup_reloc_trees(struct btrfs_root *root) | ||
| 5015 | { | ||
| 5016 | struct btrfs_root *reloc_root; | ||
| 5017 | struct btrfs_trans_handle *trans; | ||
| 5018 | struct btrfs_key location; | ||
| 5019 | int found; | ||
| 5020 | int ret; | ||
| 5021 | |||
| 5022 | mutex_lock(&root->fs_info->tree_reloc_mutex); | ||
| 5023 | ret = btrfs_find_dead_roots(root, BTRFS_TREE_RELOC_OBJECTID, NULL); | ||
| 5024 | BUG_ON(ret); | ||
| 5025 | found = !list_empty(&root->fs_info->dead_reloc_roots); | ||
| 5026 | mutex_unlock(&root->fs_info->tree_reloc_mutex); | ||
| 5027 | |||
| 5028 | if (found) { | ||
| 5029 | trans = btrfs_start_transaction(root, 1); | ||
| 5030 | BUG_ON(!trans); | ||
| 5031 | ret = btrfs_commit_transaction(trans, root); | ||
| 5032 | BUG_ON(ret); | ||
| 5033 | } | ||
| 5034 | |||
| 5035 | location.objectid = BTRFS_DATA_RELOC_TREE_OBJECTID; | ||
| 5036 | location.offset = (u64)-1; | ||
| 5037 | location.type = BTRFS_ROOT_ITEM_KEY; | ||
| 5038 | |||
| 5039 | reloc_root = btrfs_read_fs_root_no_name(root->fs_info, &location); | ||
| 5040 | BUG_ON(!reloc_root); | ||
| 5041 | btrfs_orphan_cleanup(reloc_root); | ||
| 5042 | return 0; | ||
| 5043 | } | ||
| 5044 | |||
| 5045 | static noinline int init_reloc_tree(struct btrfs_trans_handle *trans, | ||
| 5046 | struct btrfs_root *root) | ||
| 5047 | { | ||
| 5048 | struct btrfs_root *reloc_root; | ||
| 5049 | struct extent_buffer *eb; | ||
| 5050 | struct btrfs_root_item *root_item; | ||
| 5051 | struct btrfs_key root_key; | ||
| 5052 | int ret; | ||
| 5053 | |||
| 5054 | BUG_ON(!root->ref_cows); | ||
| 5055 | if (root->reloc_root) | ||
| 5056 | return 0; | ||
| 5057 | |||
| 5058 | root_item = kmalloc(sizeof(*root_item), GFP_NOFS); | ||
| 5059 | BUG_ON(!root_item); | ||
| 5060 | |||
| 5061 | ret = btrfs_copy_root(trans, root, root->commit_root, | ||
| 5062 | &eb, BTRFS_TREE_RELOC_OBJECTID); | ||
| 5063 | BUG_ON(ret); | ||
| 5064 | |||
| 5065 | root_key.objectid = BTRFS_TREE_RELOC_OBJECTID; | ||
| 5066 | root_key.offset = root->root_key.objectid; | ||
| 5067 | root_key.type = BTRFS_ROOT_ITEM_KEY; | ||
| 5068 | |||
| 5069 | memcpy(root_item, &root->root_item, sizeof(root_item)); | ||
| 5070 | btrfs_set_root_refs(root_item, 0); | ||
| 5071 | btrfs_set_root_bytenr(root_item, eb->start); | ||
| 5072 | btrfs_set_root_level(root_item, btrfs_header_level(eb)); | ||
| 5073 | btrfs_set_root_generation(root_item, trans->transid); | ||
| 5074 | |||
| 5075 | btrfs_tree_unlock(eb); | ||
| 5076 | free_extent_buffer(eb); | ||
| 5077 | |||
| 5078 | ret = btrfs_insert_root(trans, root->fs_info->tree_root, | ||
| 5079 | &root_key, root_item); | ||
| 5080 | BUG_ON(ret); | ||
| 5081 | kfree(root_item); | ||
| 5082 | |||
| 5083 | reloc_root = btrfs_read_fs_root_no_radix(root->fs_info->tree_root, | ||
| 5084 | &root_key); | ||
| 5085 | BUG_ON(!reloc_root); | ||
| 5086 | reloc_root->last_trans = trans->transid; | ||
| 5087 | reloc_root->commit_root = NULL; | ||
| 5088 | reloc_root->ref_tree = &root->fs_info->reloc_ref_tree; | ||
| 5089 | |||
| 5090 | root->reloc_root = reloc_root; | ||
| 5091 | return 0; | ||
| 5092 | } | ||
| 5093 | |||
| 5094 | /* | ||
| 5095 | * Core function of space balance. | ||
| 5096 | * | ||
| 5097 | * The idea is using reloc trees to relocate tree blocks in reference | ||
| 5098 | * counted roots. There is one reloc tree for each subvol, and all | ||
| 5099 | * reloc trees share same root key objectid. Reloc trees are snapshots | ||
| 5100 | * of the latest committed roots of subvols (root->commit_root). | ||
| 5101 | * | ||
| 5102 | * To relocate a tree block referenced by a subvol, there are two steps. | ||
| 5103 | * COW the block through subvol's reloc tree, then update block pointer | ||
| 5104 | * in the subvol to point to the new block. Since all reloc trees share | ||
| 5105 | * same root key objectid, doing special handing for tree blocks owned | ||
| 5106 | * by them is easy. Once a tree block has been COWed in one reloc tree, | ||
| 5107 | * we can use the resulting new block directly when the same block is | ||
| 5108 | * required to COW again through other reloc trees. By this way, relocated | ||
| 5109 | * tree blocks are shared between reloc trees, so they are also shared | ||
| 5110 | * between subvols. | ||
| 5111 | */ | ||
| 5112 | static noinline int relocate_one_path(struct btrfs_trans_handle *trans, | ||
| 5113 | struct btrfs_root *root, | ||
| 5114 | struct btrfs_path *path, | ||
| 5115 | struct btrfs_key *first_key, | ||
| 5116 | struct btrfs_ref_path *ref_path, | ||
| 5117 | struct btrfs_block_group_cache *group, | ||
| 5118 | struct inode *reloc_inode) | ||
| 5119 | { | ||
| 5120 | struct btrfs_root *reloc_root; | ||
| 5121 | struct extent_buffer *eb = NULL; | ||
| 5122 | struct btrfs_key *keys; | ||
| 5123 | u64 *nodes; | ||
| 5124 | int level; | ||
| 5125 | int shared_level; | ||
| 5126 | int lowest_level = 0; | ||
| 5127 | int ret; | ||
| 5128 | |||
| 5129 | if (ref_path->owner_objectid < BTRFS_FIRST_FREE_OBJECTID) | ||
| 5130 | lowest_level = ref_path->owner_objectid; | ||
| 5131 | |||
| 5132 | if (!root->ref_cows) { | ||
| 5133 | path->lowest_level = lowest_level; | ||
| 5134 | ret = btrfs_search_slot(trans, root, first_key, path, 0, 1); | ||
| 5135 | BUG_ON(ret < 0); | ||
| 5136 | path->lowest_level = 0; | ||
| 5137 | btrfs_release_path(root, path); | ||
| 5138 | return 0; | ||
| 5139 | } | ||
| 5140 | |||
| 5141 | mutex_lock(&root->fs_info->tree_reloc_mutex); | ||
| 5142 | ret = init_reloc_tree(trans, root); | ||
| 5143 | BUG_ON(ret); | ||
| 5144 | reloc_root = root->reloc_root; | ||
| 5145 | |||
| 5146 | shared_level = ref_path->shared_level; | ||
| 5147 | ref_path->shared_level = BTRFS_MAX_LEVEL - 1; | ||
| 5148 | |||
| 5149 | keys = ref_path->node_keys; | ||
| 5150 | nodes = ref_path->new_nodes; | ||
| 5151 | memset(&keys[shared_level + 1], 0, | ||
| 5152 | sizeof(*keys) * (BTRFS_MAX_LEVEL - shared_level - 1)); | ||
| 5153 | memset(&nodes[shared_level + 1], 0, | ||
| 5154 | sizeof(*nodes) * (BTRFS_MAX_LEVEL - shared_level - 1)); | ||
| 5155 | |||
| 5156 | if (nodes[lowest_level] == 0) { | ||
| 5157 | path->lowest_level = lowest_level; | ||
| 5158 | ret = btrfs_search_slot(trans, reloc_root, first_key, path, | ||
| 5159 | 0, 1); | ||
| 5160 | BUG_ON(ret); | ||
| 5161 | for (level = lowest_level; level < BTRFS_MAX_LEVEL; level++) { | ||
| 5162 | eb = path->nodes[level]; | ||
| 5163 | if (!eb || eb == reloc_root->node) | ||
| 5164 | break; | ||
| 5165 | nodes[level] = eb->start; | ||
| 5166 | if (level == 0) | ||
| 5167 | btrfs_item_key_to_cpu(eb, &keys[level], 0); | ||
| 5168 | else | ||
| 5169 | btrfs_node_key_to_cpu(eb, &keys[level], 0); | ||
| 5170 | } | ||
| 5171 | if (nodes[0] && | ||
| 5172 | ref_path->owner_objectid >= BTRFS_FIRST_FREE_OBJECTID) { | ||
| 5173 | eb = path->nodes[0]; | ||
| 5174 | ret = replace_extents_in_leaf(trans, reloc_root, eb, | ||
| 5175 | group, reloc_inode); | ||
| 5176 | BUG_ON(ret); | ||
| 5177 | } | ||
| 5178 | btrfs_release_path(reloc_root, path); | ||
| 5179 | } else { | ||
| 5180 | ret = btrfs_merge_path(trans, reloc_root, keys, nodes, | ||
| 5181 | lowest_level); | ||
| 5182 | BUG_ON(ret); | ||
| 5183 | } | ||
| 5184 | |||
| 5185 | /* | ||
| 5186 | * replace tree blocks in the fs tree with tree blocks in | ||
| 5187 | * the reloc tree. | ||
| 5188 | */ | ||
| 5189 | ret = btrfs_merge_path(trans, root, keys, nodes, lowest_level); | ||
| 5190 | BUG_ON(ret < 0); | ||
| 5191 | |||
| 5192 | if (ref_path->owner_objectid >= BTRFS_FIRST_FREE_OBJECTID) { | ||
| 5193 | ret = btrfs_search_slot(trans, reloc_root, first_key, path, | ||
| 5194 | 0, 0); | ||
| 5195 | BUG_ON(ret); | ||
| 5196 | extent_buffer_get(path->nodes[0]); | ||
| 5197 | eb = path->nodes[0]; | ||
| 5198 | btrfs_release_path(reloc_root, path); | ||
| 5199 | ret = invalidate_extent_cache(reloc_root, eb, group, root); | ||
| 5200 | BUG_ON(ret); | ||
| 5201 | free_extent_buffer(eb); | ||
| 5202 | } | ||
| 5203 | |||
| 5204 | mutex_unlock(&root->fs_info->tree_reloc_mutex); | ||
| 5205 | path->lowest_level = 0; | ||
| 5206 | return 0; | ||
| 5207 | } | ||
| 5208 | |||
| 5209 | static noinline int relocate_tree_block(struct btrfs_trans_handle *trans, | ||
| 5210 | struct btrfs_root *root, | ||
| 5211 | struct btrfs_path *path, | ||
| 5212 | struct btrfs_key *first_key, | ||
| 5213 | struct btrfs_ref_path *ref_path) | ||
| 5214 | { | ||
| 5215 | int ret; | ||
| 5216 | |||
| 5217 | ret = relocate_one_path(trans, root, path, first_key, | ||
| 5218 | ref_path, NULL, NULL); | ||
| 5219 | BUG_ON(ret); | ||
| 5220 | |||
| 5221 | if (root == root->fs_info->extent_root) | ||
| 5222 | btrfs_extent_post_op(trans, root); | ||
| 5223 | |||
| 5224 | return 0; | ||
| 5225 | } | ||
| 5226 | |||
| 5227 | static noinline int del_extent_zero(struct btrfs_trans_handle *trans, | ||
| 5228 | struct btrfs_root *extent_root, | ||
| 5229 | struct btrfs_path *path, | ||
| 5230 | struct btrfs_key *extent_key) | ||
| 5231 | { | ||
| 5232 | int ret; | ||
| 5233 | |||
| 5234 | ret = btrfs_search_slot(trans, extent_root, extent_key, path, -1, 1); | ||
| 5235 | if (ret) | ||
| 5236 | goto out; | ||
| 5237 | ret = btrfs_del_item(trans, extent_root, path); | ||
| 5238 | out: | ||
| 5239 | btrfs_release_path(extent_root, path); | ||
| 5240 | return ret; | ||
| 5241 | } | ||
| 5242 | |||
| 5243 | static noinline struct btrfs_root *read_ref_root(struct btrfs_fs_info *fs_info, | ||
| 5244 | struct btrfs_ref_path *ref_path) | ||
| 5245 | { | ||
| 5246 | struct btrfs_key root_key; | ||
| 5247 | |||
| 5248 | root_key.objectid = ref_path->root_objectid; | ||
| 5249 | root_key.type = BTRFS_ROOT_ITEM_KEY; | ||
| 5250 | if (is_cowonly_root(ref_path->root_objectid)) | ||
| 5251 | root_key.offset = 0; | ||
| 5252 | else | ||
| 5253 | root_key.offset = (u64)-1; | ||
| 5254 | |||
| 5255 | return btrfs_read_fs_root_no_name(fs_info, &root_key); | ||
| 5256 | } | ||
| 5257 | |||
| 5258 | static noinline int relocate_one_extent(struct btrfs_root *extent_root, | ||
| 5259 | struct btrfs_path *path, | ||
| 5260 | struct btrfs_key *extent_key, | ||
| 5261 | struct btrfs_block_group_cache *group, | ||
| 5262 | struct inode *reloc_inode, int pass) | ||
| 5263 | { | ||
| 5264 | struct btrfs_trans_handle *trans; | ||
| 5265 | struct btrfs_root *found_root; | ||
| 5266 | struct btrfs_ref_path *ref_path = NULL; | ||
| 5267 | struct disk_extent *new_extents = NULL; | ||
| 5268 | int nr_extents = 0; | ||
| 5269 | int loops; | ||
| 5270 | int ret; | ||
| 5271 | int level; | ||
| 5272 | struct btrfs_key first_key; | ||
| 5273 | u64 prev_block = 0; | ||
| 5274 | |||
| 5275 | |||
| 5276 | trans = btrfs_start_transaction(extent_root, 1); | ||
| 5277 | BUG_ON(!trans); | ||
| 5278 | |||
| 5279 | if (extent_key->objectid == 0) { | ||
| 5280 | ret = del_extent_zero(trans, extent_root, path, extent_key); | ||
| 5281 | goto out; | ||
| 5282 | } | ||
| 5283 | |||
| 5284 | ref_path = kmalloc(sizeof(*ref_path), GFP_NOFS); | ||
| 5285 | if (!ref_path) { | ||
| 5286 | ret = -ENOMEM; | ||
| 5287 | goto out; | ||
| 5288 | } | ||
| 5289 | |||
| 5290 | for (loops = 0; ; loops++) { | ||
| 5291 | if (loops == 0) { | ||
| 5292 | ret = btrfs_first_ref_path(trans, extent_root, ref_path, | ||
| 5293 | extent_key->objectid); | ||
| 5294 | } else { | ||
| 5295 | ret = btrfs_next_ref_path(trans, extent_root, ref_path); | ||
| 5296 | } | ||
| 5297 | if (ret < 0) | ||
| 5298 | goto out; | ||
| 5299 | if (ret > 0) | ||
| 5300 | break; | ||
| 5301 | |||
| 5302 | if (ref_path->root_objectid == BTRFS_TREE_LOG_OBJECTID || | ||
| 5303 | ref_path->root_objectid == BTRFS_TREE_RELOC_OBJECTID) | ||
| 5304 | continue; | ||
| 5305 | |||
| 5306 | found_root = read_ref_root(extent_root->fs_info, ref_path); | ||
| 5307 | BUG_ON(!found_root); | ||
| 5308 | /* | ||
| 5309 | * for reference counted tree, only process reference paths | ||
| 5310 | * rooted at the latest committed root. | ||
| 5311 | */ | ||
| 5312 | if (found_root->ref_cows && | ||
| 5313 | ref_path->root_generation != found_root->root_key.offset) | ||
| 5314 | continue; | ||
| 5315 | |||
| 5316 | if (ref_path->owner_objectid >= BTRFS_FIRST_FREE_OBJECTID) { | ||
| 5317 | if (pass == 0) { | ||
| 5318 | /* | ||
| 5319 | * copy data extents to new locations | ||
| 5320 | */ | ||
| 5321 | u64 group_start = group->key.objectid; | ||
| 5322 | ret = relocate_data_extent(reloc_inode, | ||
| 5323 | extent_key, | ||
| 5324 | group_start); | ||
| 5325 | if (ret < 0) | ||
| 5326 | goto out; | ||
| 5327 | break; | ||
| 5328 | } | ||
| 5329 | level = 0; | ||
| 5330 | } else { | ||
| 5331 | level = ref_path->owner_objectid; | ||
| 5332 | } | ||
| 5333 | |||
| 5334 | if (prev_block != ref_path->nodes[level]) { | ||
| 5335 | struct extent_buffer *eb; | ||
| 5336 | u64 block_start = ref_path->nodes[level]; | ||
| 5337 | u64 block_size = btrfs_level_size(found_root, level); | ||
| 5338 | |||
| 5339 | eb = read_tree_block(found_root, block_start, | ||
| 5340 | block_size, 0); | ||
| 5341 | btrfs_tree_lock(eb); | ||
| 5342 | BUG_ON(level != btrfs_header_level(eb)); | ||
| 5343 | |||
| 5344 | if (level == 0) | ||
| 5345 | btrfs_item_key_to_cpu(eb, &first_key, 0); | ||
| 5346 | else | ||
| 5347 | btrfs_node_key_to_cpu(eb, &first_key, 0); | ||
| 5348 | |||
| 5349 | btrfs_tree_unlock(eb); | ||
| 5350 | free_extent_buffer(eb); | ||
| 5351 | prev_block = block_start; | ||
| 5352 | } | ||
| 5353 | |||
| 5354 | btrfs_record_root_in_trans(found_root); | ||
| 5355 | if (ref_path->owner_objectid >= BTRFS_FIRST_FREE_OBJECTID) { | ||
| 5356 | /* | ||
| 5357 | * try to update data extent references while | ||
| 5358 | * keeping metadata shared between snapshots. | ||
| 5359 | */ | ||
| 5360 | if (pass == 1) { | ||
| 5361 | ret = relocate_one_path(trans, found_root, | ||
| 5362 | path, &first_key, ref_path, | ||
| 5363 | group, reloc_inode); | ||
| 5364 | if (ret < 0) | ||
| 5365 | goto out; | ||
| 5366 | continue; | ||
| 5367 | } | ||
| 5368 | /* | ||
| 5369 | * use fallback method to process the remaining | ||
| 5370 | * references. | ||
| 5371 | */ | ||
| 5372 | if (!new_extents) { | ||
| 5373 | u64 group_start = group->key.objectid; | ||
| 5374 | new_extents = kmalloc(sizeof(*new_extents), | ||
| 5375 | GFP_NOFS); | ||
| 5376 | nr_extents = 1; | ||
| 5377 | ret = get_new_locations(reloc_inode, | ||
| 5378 | extent_key, | ||
| 5379 | group_start, 1, | ||
| 5380 | &new_extents, | ||
| 5381 | &nr_extents); | ||
| 5382 | if (ret) | ||
| 5383 | goto out; | ||
| 5384 | } | ||
| 5385 | ret = replace_one_extent(trans, found_root, | ||
| 5386 | path, extent_key, | ||
| 5387 | &first_key, ref_path, | ||
| 5388 | new_extents, nr_extents); | ||
| 5389 | } else { | ||
| 5390 | ret = relocate_tree_block(trans, found_root, path, | ||
| 5391 | &first_key, ref_path); | ||
| 5392 | } | ||
| 5393 | if (ret < 0) | ||
| 5394 | goto out; | ||
| 5395 | } | ||
| 5396 | ret = 0; | ||
| 5397 | out: | ||
| 5398 | btrfs_end_transaction(trans, extent_root); | ||
| 5399 | kfree(new_extents); | ||
| 5400 | kfree(ref_path); | ||
| 5401 | return ret; | ||
| 5402 | } | ||
| 5403 | |||
| 5404 | static u64 update_block_group_flags(struct btrfs_root *root, u64 flags) | ||
| 5405 | { | ||
| 5406 | u64 num_devices; | ||
| 5407 | u64 stripped = BTRFS_BLOCK_GROUP_RAID0 | | ||
| 5408 | BTRFS_BLOCK_GROUP_RAID1 | BTRFS_BLOCK_GROUP_RAID10; | ||
| 5409 | |||
| 5410 | num_devices = root->fs_info->fs_devices->rw_devices; | ||
| 5411 | if (num_devices == 1) { | ||
| 5412 | stripped |= BTRFS_BLOCK_GROUP_DUP; | ||
| 5413 | stripped = flags & ~stripped; | ||
| 5414 | |||
| 5415 | /* turn raid0 into single device chunks */ | ||
| 5416 | if (flags & BTRFS_BLOCK_GROUP_RAID0) | ||
| 5417 | return stripped; | ||
| 5418 | |||
| 5419 | /* turn mirroring into duplication */ | ||
| 5420 | if (flags & (BTRFS_BLOCK_GROUP_RAID1 | | ||
| 5421 | BTRFS_BLOCK_GROUP_RAID10)) | ||
| 5422 | return stripped | BTRFS_BLOCK_GROUP_DUP; | ||
| 5423 | return flags; | ||
| 5424 | } else { | ||
| 5425 | /* they already had raid on here, just return */ | ||
| 5426 | if (flags & stripped) | ||
| 5427 | return flags; | ||
| 5428 | |||
| 5429 | stripped |= BTRFS_BLOCK_GROUP_DUP; | ||
| 5430 | stripped = flags & ~stripped; | ||
| 5431 | |||
| 5432 | /* switch duplicated blocks with raid1 */ | ||
| 5433 | if (flags & BTRFS_BLOCK_GROUP_DUP) | ||
| 5434 | return stripped | BTRFS_BLOCK_GROUP_RAID1; | ||
| 5435 | |||
| 5436 | /* turn single device chunks into raid0 */ | ||
| 5437 | return stripped | BTRFS_BLOCK_GROUP_RAID0; | ||
| 5438 | } | ||
| 5439 | return flags; | ||
| 5440 | } | ||
| 5441 | |||
| 5442 | static int __alloc_chunk_for_shrink(struct btrfs_root *root, | ||
| 5443 | struct btrfs_block_group_cache *shrink_block_group, | ||
| 5444 | int force) | ||
| 5445 | { | ||
| 5446 | struct btrfs_trans_handle *trans; | ||
| 5447 | u64 new_alloc_flags; | ||
| 5448 | u64 calc; | ||
| 5449 | |||
| 5450 | spin_lock(&shrink_block_group->lock); | ||
| 5451 | if (btrfs_block_group_used(&shrink_block_group->item) > 0) { | ||
| 5452 | spin_unlock(&shrink_block_group->lock); | ||
| 5453 | |||
| 5454 | trans = btrfs_start_transaction(root, 1); | ||
| 5455 | spin_lock(&shrink_block_group->lock); | ||
| 5456 | |||
| 5457 | new_alloc_flags = update_block_group_flags(root, | ||
| 5458 | shrink_block_group->flags); | ||
| 5459 | if (new_alloc_flags != shrink_block_group->flags) { | ||
| 5460 | calc = | ||
| 5461 | btrfs_block_group_used(&shrink_block_group->item); | ||
| 5462 | } else { | ||
| 5463 | calc = shrink_block_group->key.offset; | ||
| 5464 | } | ||
| 5465 | spin_unlock(&shrink_block_group->lock); | ||
| 5466 | |||
| 5467 | do_chunk_alloc(trans, root->fs_info->extent_root, | ||
| 5468 | calc + 2 * 1024 * 1024, new_alloc_flags, force); | ||
| 5469 | |||
| 5470 | btrfs_end_transaction(trans, root); | ||
| 5471 | } else | ||
| 5472 | spin_unlock(&shrink_block_group->lock); | ||
| 5473 | return 0; | ||
| 5474 | } | ||
| 5475 | |||
| 5476 | static int __insert_orphan_inode(struct btrfs_trans_handle *trans, | ||
| 5477 | struct btrfs_root *root, | ||
| 5478 | u64 objectid, u64 size) | ||
| 5479 | { | ||
| 5480 | struct btrfs_path *path; | ||
| 5481 | struct btrfs_inode_item *item; | ||
| 5482 | struct extent_buffer *leaf; | ||
| 5483 | int ret; | ||
| 5484 | |||
| 5485 | path = btrfs_alloc_path(); | ||
| 5486 | if (!path) | ||
| 5487 | return -ENOMEM; | ||
| 5488 | |||
| 5489 | ret = btrfs_insert_empty_inode(trans, root, path, objectid); | ||
| 5490 | if (ret) | ||
| 5491 | goto out; | ||
| 5492 | |||
| 5493 | leaf = path->nodes[0]; | ||
| 5494 | item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_inode_item); | ||
| 5495 | memset_extent_buffer(leaf, 0, (unsigned long)item, sizeof(*item)); | ||
| 5496 | btrfs_set_inode_generation(leaf, item, 1); | ||
| 5497 | btrfs_set_inode_size(leaf, item, size); | ||
| 5498 | btrfs_set_inode_mode(leaf, item, S_IFREG | 0600); | ||
| 5499 | btrfs_set_inode_flags(leaf, item, BTRFS_INODE_NOCOMPRESS); | ||
| 5500 | btrfs_mark_buffer_dirty(leaf); | ||
| 5501 | btrfs_release_path(root, path); | ||
| 5502 | out: | ||
| 5503 | btrfs_free_path(path); | ||
| 5504 | return ret; | ||
| 5505 | } | ||
| 5506 | |||
| 5507 | static noinline struct inode *create_reloc_inode(struct btrfs_fs_info *fs_info, | ||
| 5508 | struct btrfs_block_group_cache *group) | ||
| 5509 | { | ||
| 5510 | struct inode *inode = NULL; | ||
| 5511 | struct btrfs_trans_handle *trans; | ||
| 5512 | struct btrfs_root *root; | ||
| 5513 | struct btrfs_key root_key; | ||
| 5514 | u64 objectid = BTRFS_FIRST_FREE_OBJECTID; | ||
| 5515 | int err = 0; | ||
| 5516 | |||
| 5517 | root_key.objectid = BTRFS_DATA_RELOC_TREE_OBJECTID; | ||
| 5518 | root_key.type = BTRFS_ROOT_ITEM_KEY; | ||
| 5519 | root_key.offset = (u64)-1; | ||
| 5520 | root = btrfs_read_fs_root_no_name(fs_info, &root_key); | ||
| 5521 | if (IS_ERR(root)) | ||
| 5522 | return ERR_CAST(root); | ||
| 5523 | |||
| 5524 | trans = btrfs_start_transaction(root, 1); | ||
| 5525 | BUG_ON(!trans); | ||
| 5526 | |||
| 5527 | err = btrfs_find_free_objectid(trans, root, objectid, &objectid); | ||
| 5528 | if (err) | ||
| 5529 | goto out; | ||
| 5530 | |||
| 5531 | err = __insert_orphan_inode(trans, root, objectid, group->key.offset); | ||
| 5532 | BUG_ON(err); | ||
| 5533 | |||
| 5534 | err = btrfs_insert_file_extent(trans, root, objectid, 0, 0, 0, | ||
| 5535 | group->key.offset, 0, group->key.offset, | ||
| 5536 | 0, 0, 0); | ||
| 5537 | BUG_ON(err); | ||
| 5538 | |||
| 5539 | inode = btrfs_iget_locked(root->fs_info->sb, objectid, root); | ||
| 5540 | if (inode->i_state & I_NEW) { | ||
| 5541 | BTRFS_I(inode)->root = root; | ||
| 5542 | BTRFS_I(inode)->location.objectid = objectid; | ||
| 5543 | BTRFS_I(inode)->location.type = BTRFS_INODE_ITEM_KEY; | ||
| 5544 | BTRFS_I(inode)->location.offset = 0; | ||
| 5545 | btrfs_read_locked_inode(inode); | ||
| 5546 | unlock_new_inode(inode); | ||
| 5547 | BUG_ON(is_bad_inode(inode)); | ||
| 5548 | } else { | ||
| 5549 | BUG_ON(1); | ||
| 5550 | } | ||
| 5551 | BTRFS_I(inode)->index_cnt = group->key.objectid; | ||
| 5552 | |||
| 5553 | err = btrfs_orphan_add(trans, inode); | ||
| 5554 | out: | ||
| 5555 | btrfs_end_transaction(trans, root); | ||
| 5556 | if (err) { | ||
| 5557 | if (inode) | ||
| 5558 | iput(inode); | ||
| 5559 | inode = ERR_PTR(err); | ||
| 5560 | } | ||
| 5561 | return inode; | ||
| 5562 | } | ||
| 5563 | |||
| 5564 | int btrfs_reloc_clone_csums(struct inode *inode, u64 file_pos, u64 len) | ||
| 5565 | { | ||
| 5566 | |||
| 5567 | struct btrfs_ordered_sum *sums; | ||
| 5568 | struct btrfs_sector_sum *sector_sum; | ||
| 5569 | struct btrfs_ordered_extent *ordered; | ||
| 5570 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 5571 | struct list_head list; | ||
| 5572 | size_t offset; | ||
| 5573 | int ret; | ||
| 5574 | u64 disk_bytenr; | ||
| 5575 | |||
| 5576 | INIT_LIST_HEAD(&list); | ||
| 5577 | |||
| 5578 | ordered = btrfs_lookup_ordered_extent(inode, file_pos); | ||
| 5579 | BUG_ON(ordered->file_offset != file_pos || ordered->len != len); | ||
| 5580 | |||
| 5581 | disk_bytenr = file_pos + BTRFS_I(inode)->index_cnt; | ||
| 5582 | ret = btrfs_lookup_csums_range(root->fs_info->csum_root, disk_bytenr, | ||
| 5583 | disk_bytenr + len - 1, &list); | ||
| 5584 | |||
| 5585 | while (!list_empty(&list)) { | ||
| 5586 | sums = list_entry(list.next, struct btrfs_ordered_sum, list); | ||
| 5587 | list_del_init(&sums->list); | ||
| 5588 | |||
| 5589 | sector_sum = sums->sums; | ||
| 5590 | sums->bytenr = ordered->start; | ||
| 5591 | |||
| 5592 | offset = 0; | ||
| 5593 | while (offset < sums->len) { | ||
| 5594 | sector_sum->bytenr += ordered->start - disk_bytenr; | ||
| 5595 | sector_sum++; | ||
| 5596 | offset += root->sectorsize; | ||
| 5597 | } | ||
| 5598 | |||
| 5599 | btrfs_add_ordered_sum(inode, ordered, sums); | ||
| 5600 | } | ||
| 5601 | btrfs_put_ordered_extent(ordered); | ||
| 5602 | return 0; | ||
| 5603 | } | ||
| 5604 | |||
| 5605 | int btrfs_relocate_block_group(struct btrfs_root *root, u64 group_start) | ||
| 5606 | { | ||
| 5607 | struct btrfs_trans_handle *trans; | ||
| 5608 | struct btrfs_path *path; | ||
| 5609 | struct btrfs_fs_info *info = root->fs_info; | ||
| 5610 | struct extent_buffer *leaf; | ||
| 5611 | struct inode *reloc_inode; | ||
| 5612 | struct btrfs_block_group_cache *block_group; | ||
| 5613 | struct btrfs_key key; | ||
| 5614 | u64 skipped; | ||
| 5615 | u64 cur_byte; | ||
| 5616 | u64 total_found; | ||
| 5617 | u32 nritems; | ||
| 5618 | int ret; | ||
| 5619 | int progress; | ||
| 5620 | int pass = 0; | ||
| 5621 | |||
| 5622 | root = root->fs_info->extent_root; | ||
| 5623 | |||
| 5624 | block_group = btrfs_lookup_block_group(info, group_start); | ||
| 5625 | BUG_ON(!block_group); | ||
| 5626 | |||
| 5627 | printk(KERN_INFO "btrfs relocating block group %llu flags %llu\n", | ||
| 5628 | (unsigned long long)block_group->key.objectid, | ||
| 5629 | (unsigned long long)block_group->flags); | ||
| 5630 | |||
| 5631 | path = btrfs_alloc_path(); | ||
| 5632 | BUG_ON(!path); | ||
| 5633 | |||
| 5634 | reloc_inode = create_reloc_inode(info, block_group); | ||
| 5635 | BUG_ON(IS_ERR(reloc_inode)); | ||
| 5636 | |||
| 5637 | __alloc_chunk_for_shrink(root, block_group, 1); | ||
| 5638 | set_block_group_readonly(block_group); | ||
| 5639 | |||
| 5640 | btrfs_start_delalloc_inodes(info->tree_root); | ||
| 5641 | btrfs_wait_ordered_extents(info->tree_root, 0); | ||
| 5642 | again: | ||
| 5643 | skipped = 0; | ||
| 5644 | total_found = 0; | ||
| 5645 | progress = 0; | ||
| 5646 | key.objectid = block_group->key.objectid; | ||
| 5647 | key.offset = 0; | ||
| 5648 | key.type = 0; | ||
| 5649 | cur_byte = key.objectid; | ||
| 5650 | |||
| 5651 | trans = btrfs_start_transaction(info->tree_root, 1); | ||
| 5652 | btrfs_commit_transaction(trans, info->tree_root); | ||
| 5653 | |||
| 5654 | mutex_lock(&root->fs_info->cleaner_mutex); | ||
| 5655 | btrfs_clean_old_snapshots(info->tree_root); | ||
| 5656 | btrfs_remove_leaf_refs(info->tree_root, (u64)-1, 1); | ||
| 5657 | mutex_unlock(&root->fs_info->cleaner_mutex); | ||
| 5658 | |||
| 5659 | while (1) { | ||
| 5660 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 5661 | if (ret < 0) | ||
| 5662 | goto out; | ||
| 5663 | next: | ||
| 5664 | leaf = path->nodes[0]; | ||
| 5665 | nritems = btrfs_header_nritems(leaf); | ||
| 5666 | if (path->slots[0] >= nritems) { | ||
| 5667 | ret = btrfs_next_leaf(root, path); | ||
| 5668 | if (ret < 0) | ||
| 5669 | goto out; | ||
| 5670 | if (ret == 1) { | ||
| 5671 | ret = 0; | ||
| 5672 | break; | ||
| 5673 | } | ||
| 5674 | leaf = path->nodes[0]; | ||
| 5675 | nritems = btrfs_header_nritems(leaf); | ||
| 5676 | } | ||
| 5677 | |||
| 5678 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 5679 | |||
| 5680 | if (key.objectid >= block_group->key.objectid + | ||
| 5681 | block_group->key.offset) | ||
| 5682 | break; | ||
| 5683 | |||
| 5684 | if (progress && need_resched()) { | ||
| 5685 | btrfs_release_path(root, path); | ||
| 5686 | cond_resched(); | ||
| 5687 | progress = 0; | ||
| 5688 | continue; | ||
| 5689 | } | ||
| 5690 | progress = 1; | ||
| 5691 | |||
| 5692 | if (btrfs_key_type(&key) != BTRFS_EXTENT_ITEM_KEY || | ||
| 5693 | key.objectid + key.offset <= cur_byte) { | ||
| 5694 | path->slots[0]++; | ||
| 5695 | goto next; | ||
| 5696 | } | ||
| 5697 | |||
| 5698 | total_found++; | ||
| 5699 | cur_byte = key.objectid + key.offset; | ||
| 5700 | btrfs_release_path(root, path); | ||
| 5701 | |||
| 5702 | __alloc_chunk_for_shrink(root, block_group, 0); | ||
| 5703 | ret = relocate_one_extent(root, path, &key, block_group, | ||
| 5704 | reloc_inode, pass); | ||
| 5705 | BUG_ON(ret < 0); | ||
| 5706 | if (ret > 0) | ||
| 5707 | skipped++; | ||
| 5708 | |||
| 5709 | key.objectid = cur_byte; | ||
| 5710 | key.type = 0; | ||
| 5711 | key.offset = 0; | ||
| 5712 | } | ||
| 5713 | |||
| 5714 | btrfs_release_path(root, path); | ||
| 5715 | |||
| 5716 | if (pass == 0) { | ||
| 5717 | btrfs_wait_ordered_range(reloc_inode, 0, (u64)-1); | ||
| 5718 | invalidate_mapping_pages(reloc_inode->i_mapping, 0, -1); | ||
| 5719 | } | ||
| 5720 | |||
| 5721 | if (total_found > 0) { | ||
| 5722 | printk(KERN_INFO "btrfs found %llu extents in pass %d\n", | ||
| 5723 | (unsigned long long)total_found, pass); | ||
| 5724 | pass++; | ||
| 5725 | if (total_found == skipped && pass > 2) { | ||
| 5726 | iput(reloc_inode); | ||
| 5727 | reloc_inode = create_reloc_inode(info, block_group); | ||
| 5728 | pass = 0; | ||
| 5729 | } | ||
| 5730 | goto again; | ||
| 5731 | } | ||
| 5732 | |||
| 5733 | /* delete reloc_inode */ | ||
| 5734 | iput(reloc_inode); | ||
| 5735 | |||
| 5736 | /* unpin extents in this range */ | ||
| 5737 | trans = btrfs_start_transaction(info->tree_root, 1); | ||
| 5738 | btrfs_commit_transaction(trans, info->tree_root); | ||
| 5739 | |||
| 5740 | spin_lock(&block_group->lock); | ||
| 5741 | WARN_ON(block_group->pinned > 0); | ||
| 5742 | WARN_ON(block_group->reserved > 0); | ||
| 5743 | WARN_ON(btrfs_block_group_used(&block_group->item) > 0); | ||
| 5744 | spin_unlock(&block_group->lock); | ||
| 5745 | put_block_group(block_group); | ||
| 5746 | ret = 0; | ||
| 5747 | out: | ||
| 5748 | btrfs_free_path(path); | ||
| 5749 | return ret; | ||
| 5750 | } | ||
| 5751 | |||
| 5752 | static int find_first_block_group(struct btrfs_root *root, | ||
| 5753 | struct btrfs_path *path, struct btrfs_key *key) | ||
| 5754 | { | ||
| 5755 | int ret = 0; | ||
| 5756 | struct btrfs_key found_key; | ||
| 5757 | struct extent_buffer *leaf; | ||
| 5758 | int slot; | ||
| 5759 | |||
| 5760 | ret = btrfs_search_slot(NULL, root, key, path, 0, 0); | ||
| 5761 | if (ret < 0) | ||
| 5762 | goto out; | ||
| 5763 | |||
| 5764 | while (1) { | ||
| 5765 | slot = path->slots[0]; | ||
| 5766 | leaf = path->nodes[0]; | ||
| 5767 | if (slot >= btrfs_header_nritems(leaf)) { | ||
| 5768 | ret = btrfs_next_leaf(root, path); | ||
| 5769 | if (ret == 0) | ||
| 5770 | continue; | ||
| 5771 | if (ret < 0) | ||
| 5772 | goto out; | ||
| 5773 | break; | ||
| 5774 | } | ||
| 5775 | btrfs_item_key_to_cpu(leaf, &found_key, slot); | ||
| 5776 | |||
| 5777 | if (found_key.objectid >= key->objectid && | ||
| 5778 | found_key.type == BTRFS_BLOCK_GROUP_ITEM_KEY) { | ||
| 5779 | ret = 0; | ||
| 5780 | goto out; | ||
| 5781 | } | ||
| 5782 | path->slots[0]++; | ||
| 5783 | } | ||
| 5784 | ret = -ENOENT; | ||
| 5785 | out: | ||
| 5786 | return ret; | ||
| 5787 | } | ||
| 5788 | |||
| 5789 | int btrfs_free_block_groups(struct btrfs_fs_info *info) | ||
| 5790 | { | ||
| 5791 | struct btrfs_block_group_cache *block_group; | ||
| 5792 | struct rb_node *n; | ||
| 5793 | |||
| 5794 | spin_lock(&info->block_group_cache_lock); | ||
| 5795 | while ((n = rb_last(&info->block_group_cache_tree)) != NULL) { | ||
| 5796 | block_group = rb_entry(n, struct btrfs_block_group_cache, | ||
| 5797 | cache_node); | ||
| 5798 | rb_erase(&block_group->cache_node, | ||
| 5799 | &info->block_group_cache_tree); | ||
| 5800 | spin_unlock(&info->block_group_cache_lock); | ||
| 5801 | |||
| 5802 | btrfs_remove_free_space_cache(block_group); | ||
| 5803 | down_write(&block_group->space_info->groups_sem); | ||
| 5804 | list_del(&block_group->list); | ||
| 5805 | up_write(&block_group->space_info->groups_sem); | ||
| 5806 | |||
| 5807 | WARN_ON(atomic_read(&block_group->count) != 1); | ||
| 5808 | kfree(block_group); | ||
| 5809 | |||
| 5810 | spin_lock(&info->block_group_cache_lock); | ||
| 5811 | } | ||
| 5812 | spin_unlock(&info->block_group_cache_lock); | ||
| 5813 | return 0; | ||
| 5814 | } | ||
| 5815 | |||
| 5816 | int btrfs_read_block_groups(struct btrfs_root *root) | ||
| 5817 | { | ||
| 5818 | struct btrfs_path *path; | ||
| 5819 | int ret; | ||
| 5820 | struct btrfs_block_group_cache *cache; | ||
| 5821 | struct btrfs_fs_info *info = root->fs_info; | ||
| 5822 | struct btrfs_space_info *space_info; | ||
| 5823 | struct btrfs_key key; | ||
| 5824 | struct btrfs_key found_key; | ||
| 5825 | struct extent_buffer *leaf; | ||
| 5826 | |||
| 5827 | root = info->extent_root; | ||
| 5828 | key.objectid = 0; | ||
| 5829 | key.offset = 0; | ||
| 5830 | btrfs_set_key_type(&key, BTRFS_BLOCK_GROUP_ITEM_KEY); | ||
| 5831 | path = btrfs_alloc_path(); | ||
| 5832 | if (!path) | ||
| 5833 | return -ENOMEM; | ||
| 5834 | |||
| 5835 | while (1) { | ||
| 5836 | ret = find_first_block_group(root, path, &key); | ||
| 5837 | if (ret > 0) { | ||
| 5838 | ret = 0; | ||
| 5839 | goto error; | ||
| 5840 | } | ||
| 5841 | if (ret != 0) | ||
| 5842 | goto error; | ||
| 5843 | |||
| 5844 | leaf = path->nodes[0]; | ||
| 5845 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 5846 | cache = kzalloc(sizeof(*cache), GFP_NOFS); | ||
| 5847 | if (!cache) { | ||
| 5848 | ret = -ENOMEM; | ||
| 5849 | break; | ||
| 5850 | } | ||
| 5851 | |||
| 5852 | atomic_set(&cache->count, 1); | ||
| 5853 | spin_lock_init(&cache->lock); | ||
| 5854 | mutex_init(&cache->alloc_mutex); | ||
| 5855 | mutex_init(&cache->cache_mutex); | ||
| 5856 | INIT_LIST_HEAD(&cache->list); | ||
| 5857 | read_extent_buffer(leaf, &cache->item, | ||
| 5858 | btrfs_item_ptr_offset(leaf, path->slots[0]), | ||
| 5859 | sizeof(cache->item)); | ||
| 5860 | memcpy(&cache->key, &found_key, sizeof(found_key)); | ||
| 5861 | |||
| 5862 | key.objectid = found_key.objectid + found_key.offset; | ||
| 5863 | btrfs_release_path(root, path); | ||
| 5864 | cache->flags = btrfs_block_group_flags(&cache->item); | ||
| 5865 | |||
| 5866 | ret = update_space_info(info, cache->flags, found_key.offset, | ||
| 5867 | btrfs_block_group_used(&cache->item), | ||
| 5868 | &space_info); | ||
| 5869 | BUG_ON(ret); | ||
| 5870 | cache->space_info = space_info; | ||
| 5871 | down_write(&space_info->groups_sem); | ||
| 5872 | list_add_tail(&cache->list, &space_info->block_groups); | ||
| 5873 | up_write(&space_info->groups_sem); | ||
| 5874 | |||
| 5875 | ret = btrfs_add_block_group_cache(root->fs_info, cache); | ||
| 5876 | BUG_ON(ret); | ||
| 5877 | |||
| 5878 | set_avail_alloc_bits(root->fs_info, cache->flags); | ||
| 5879 | if (btrfs_chunk_readonly(root, cache->key.objectid)) | ||
| 5880 | set_block_group_readonly(cache); | ||
| 5881 | } | ||
| 5882 | ret = 0; | ||
| 5883 | error: | ||
| 5884 | btrfs_free_path(path); | ||
| 5885 | return ret; | ||
| 5886 | } | ||
| 5887 | |||
| 5888 | int btrfs_make_block_group(struct btrfs_trans_handle *trans, | ||
| 5889 | struct btrfs_root *root, u64 bytes_used, | ||
| 5890 | u64 type, u64 chunk_objectid, u64 chunk_offset, | ||
| 5891 | u64 size) | ||
| 5892 | { | ||
| 5893 | int ret; | ||
| 5894 | struct btrfs_root *extent_root; | ||
| 5895 | struct btrfs_block_group_cache *cache; | ||
| 5896 | |||
| 5897 | extent_root = root->fs_info->extent_root; | ||
| 5898 | |||
| 5899 | root->fs_info->last_trans_new_blockgroup = trans->transid; | ||
| 5900 | |||
| 5901 | cache = kzalloc(sizeof(*cache), GFP_NOFS); | ||
| 5902 | if (!cache) | ||
| 5903 | return -ENOMEM; | ||
| 5904 | |||
| 5905 | cache->key.objectid = chunk_offset; | ||
| 5906 | cache->key.offset = size; | ||
| 5907 | cache->key.type = BTRFS_BLOCK_GROUP_ITEM_KEY; | ||
| 5908 | atomic_set(&cache->count, 1); | ||
| 5909 | spin_lock_init(&cache->lock); | ||
| 5910 | mutex_init(&cache->alloc_mutex); | ||
| 5911 | mutex_init(&cache->cache_mutex); | ||
| 5912 | INIT_LIST_HEAD(&cache->list); | ||
| 5913 | |||
| 5914 | btrfs_set_block_group_used(&cache->item, bytes_used); | ||
| 5915 | btrfs_set_block_group_chunk_objectid(&cache->item, chunk_objectid); | ||
| 5916 | cache->flags = type; | ||
| 5917 | btrfs_set_block_group_flags(&cache->item, type); | ||
| 5918 | |||
| 5919 | ret = update_space_info(root->fs_info, cache->flags, size, bytes_used, | ||
| 5920 | &cache->space_info); | ||
| 5921 | BUG_ON(ret); | ||
| 5922 | down_write(&cache->space_info->groups_sem); | ||
| 5923 | list_add_tail(&cache->list, &cache->space_info->block_groups); | ||
| 5924 | up_write(&cache->space_info->groups_sem); | ||
| 5925 | |||
| 5926 | ret = btrfs_add_block_group_cache(root->fs_info, cache); | ||
| 5927 | BUG_ON(ret); | ||
| 5928 | |||
| 5929 | ret = btrfs_insert_item(trans, extent_root, &cache->key, &cache->item, | ||
| 5930 | sizeof(cache->item)); | ||
| 5931 | BUG_ON(ret); | ||
| 5932 | |||
| 5933 | finish_current_insert(trans, extent_root, 0); | ||
| 5934 | ret = del_pending_extents(trans, extent_root, 0); | ||
| 5935 | BUG_ON(ret); | ||
| 5936 | set_avail_alloc_bits(extent_root->fs_info, type); | ||
| 5937 | |||
| 5938 | return 0; | ||
| 5939 | } | ||
| 5940 | |||
| 5941 | int btrfs_remove_block_group(struct btrfs_trans_handle *trans, | ||
| 5942 | struct btrfs_root *root, u64 group_start) | ||
| 5943 | { | ||
| 5944 | struct btrfs_path *path; | ||
| 5945 | struct btrfs_block_group_cache *block_group; | ||
| 5946 | struct btrfs_key key; | ||
| 5947 | int ret; | ||
| 5948 | |||
| 5949 | root = root->fs_info->extent_root; | ||
| 5950 | |||
| 5951 | block_group = btrfs_lookup_block_group(root->fs_info, group_start); | ||
| 5952 | BUG_ON(!block_group); | ||
| 5953 | BUG_ON(!block_group->ro); | ||
| 5954 | |||
| 5955 | memcpy(&key, &block_group->key, sizeof(key)); | ||
| 5956 | |||
| 5957 | path = btrfs_alloc_path(); | ||
| 5958 | BUG_ON(!path); | ||
| 5959 | |||
| 5960 | btrfs_remove_free_space_cache(block_group); | ||
| 5961 | rb_erase(&block_group->cache_node, | ||
| 5962 | &root->fs_info->block_group_cache_tree); | ||
| 5963 | down_write(&block_group->space_info->groups_sem); | ||
| 5964 | list_del(&block_group->list); | ||
| 5965 | up_write(&block_group->space_info->groups_sem); | ||
| 5966 | |||
| 5967 | spin_lock(&block_group->space_info->lock); | ||
| 5968 | block_group->space_info->total_bytes -= block_group->key.offset; | ||
| 5969 | block_group->space_info->bytes_readonly -= block_group->key.offset; | ||
| 5970 | spin_unlock(&block_group->space_info->lock); | ||
| 5971 | block_group->space_info->full = 0; | ||
| 5972 | |||
| 5973 | put_block_group(block_group); | ||
| 5974 | put_block_group(block_group); | ||
| 5975 | |||
| 5976 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 5977 | if (ret > 0) | ||
| 5978 | ret = -EIO; | ||
| 5979 | if (ret < 0) | ||
| 5980 | goto out; | ||
| 5981 | |||
| 5982 | ret = btrfs_del_item(trans, root, path); | ||
| 5983 | out: | ||
| 5984 | btrfs_free_path(path); | ||
| 5985 | return ret; | ||
| 5986 | } | ||
diff --git a/fs/btrfs/extent_io.c b/fs/btrfs/extent_io.c new file mode 100644 index 000000000000..e086d407f1fa --- /dev/null +++ b/fs/btrfs/extent_io.c | |||
| @@ -0,0 +1,3717 @@ | |||
| 1 | #include <linux/bitops.h> | ||
| 2 | #include <linux/slab.h> | ||
| 3 | #include <linux/bio.h> | ||
| 4 | #include <linux/mm.h> | ||
| 5 | #include <linux/gfp.h> | ||
| 6 | #include <linux/pagemap.h> | ||
| 7 | #include <linux/page-flags.h> | ||
| 8 | #include <linux/module.h> | ||
| 9 | #include <linux/spinlock.h> | ||
| 10 | #include <linux/blkdev.h> | ||
| 11 | #include <linux/swap.h> | ||
| 12 | #include <linux/version.h> | ||
| 13 | #include <linux/writeback.h> | ||
| 14 | #include <linux/pagevec.h> | ||
| 15 | #include "extent_io.h" | ||
| 16 | #include "extent_map.h" | ||
| 17 | #include "compat.h" | ||
| 18 | #include "ctree.h" | ||
| 19 | #include "btrfs_inode.h" | ||
| 20 | |||
| 21 | /* temporary define until extent_map moves out of btrfs */ | ||
| 22 | struct kmem_cache *btrfs_cache_create(const char *name, size_t size, | ||
| 23 | unsigned long extra_flags, | ||
| 24 | void (*ctor)(void *, struct kmem_cache *, | ||
| 25 | unsigned long)); | ||
| 26 | |||
| 27 | static struct kmem_cache *extent_state_cache; | ||
| 28 | static struct kmem_cache *extent_buffer_cache; | ||
| 29 | |||
| 30 | static LIST_HEAD(buffers); | ||
| 31 | static LIST_HEAD(states); | ||
| 32 | |||
| 33 | #define LEAK_DEBUG 0 | ||
| 34 | #ifdef LEAK_DEBUG | ||
| 35 | static DEFINE_SPINLOCK(leak_lock); | ||
| 36 | #endif | ||
| 37 | |||
| 38 | #define BUFFER_LRU_MAX 64 | ||
| 39 | |||
| 40 | struct tree_entry { | ||
| 41 | u64 start; | ||
| 42 | u64 end; | ||
| 43 | struct rb_node rb_node; | ||
| 44 | }; | ||
| 45 | |||
| 46 | struct extent_page_data { | ||
| 47 | struct bio *bio; | ||
| 48 | struct extent_io_tree *tree; | ||
| 49 | get_extent_t *get_extent; | ||
| 50 | |||
| 51 | /* tells writepage not to lock the state bits for this range | ||
| 52 | * it still does the unlocking | ||
| 53 | */ | ||
| 54 | int extent_locked; | ||
| 55 | }; | ||
| 56 | |||
| 57 | int __init extent_io_init(void) | ||
| 58 | { | ||
| 59 | extent_state_cache = btrfs_cache_create("extent_state", | ||
| 60 | sizeof(struct extent_state), 0, | ||
| 61 | NULL); | ||
| 62 | if (!extent_state_cache) | ||
| 63 | return -ENOMEM; | ||
| 64 | |||
| 65 | extent_buffer_cache = btrfs_cache_create("extent_buffers", | ||
| 66 | sizeof(struct extent_buffer), 0, | ||
| 67 | NULL); | ||
| 68 | if (!extent_buffer_cache) | ||
| 69 | goto free_state_cache; | ||
| 70 | return 0; | ||
| 71 | |||
| 72 | free_state_cache: | ||
| 73 | kmem_cache_destroy(extent_state_cache); | ||
| 74 | return -ENOMEM; | ||
| 75 | } | ||
| 76 | |||
| 77 | void extent_io_exit(void) | ||
| 78 | { | ||
| 79 | struct extent_state *state; | ||
| 80 | struct extent_buffer *eb; | ||
| 81 | |||
| 82 | while (!list_empty(&states)) { | ||
| 83 | state = list_entry(states.next, struct extent_state, leak_list); | ||
| 84 | printk(KERN_ERR "btrfs state leak: start %llu end %llu " | ||
| 85 | "state %lu in tree %p refs %d\n", | ||
| 86 | (unsigned long long)state->start, | ||
| 87 | (unsigned long long)state->end, | ||
| 88 | state->state, state->tree, atomic_read(&state->refs)); | ||
| 89 | list_del(&state->leak_list); | ||
| 90 | kmem_cache_free(extent_state_cache, state); | ||
| 91 | |||
| 92 | } | ||
| 93 | |||
| 94 | while (!list_empty(&buffers)) { | ||
| 95 | eb = list_entry(buffers.next, struct extent_buffer, leak_list); | ||
| 96 | printk(KERN_ERR "btrfs buffer leak start %llu len %lu " | ||
| 97 | "refs %d\n", (unsigned long long)eb->start, | ||
| 98 | eb->len, atomic_read(&eb->refs)); | ||
| 99 | list_del(&eb->leak_list); | ||
| 100 | kmem_cache_free(extent_buffer_cache, eb); | ||
| 101 | } | ||
| 102 | if (extent_state_cache) | ||
| 103 | kmem_cache_destroy(extent_state_cache); | ||
| 104 | if (extent_buffer_cache) | ||
| 105 | kmem_cache_destroy(extent_buffer_cache); | ||
| 106 | } | ||
| 107 | |||
| 108 | void extent_io_tree_init(struct extent_io_tree *tree, | ||
| 109 | struct address_space *mapping, gfp_t mask) | ||
| 110 | { | ||
| 111 | tree->state.rb_node = NULL; | ||
| 112 | tree->buffer.rb_node = NULL; | ||
| 113 | tree->ops = NULL; | ||
| 114 | tree->dirty_bytes = 0; | ||
| 115 | spin_lock_init(&tree->lock); | ||
| 116 | spin_lock_init(&tree->buffer_lock); | ||
| 117 | tree->mapping = mapping; | ||
| 118 | } | ||
| 119 | |||
| 120 | static struct extent_state *alloc_extent_state(gfp_t mask) | ||
| 121 | { | ||
| 122 | struct extent_state *state; | ||
| 123 | #ifdef LEAK_DEBUG | ||
| 124 | unsigned long flags; | ||
| 125 | #endif | ||
| 126 | |||
| 127 | state = kmem_cache_alloc(extent_state_cache, mask); | ||
| 128 | if (!state) | ||
| 129 | return state; | ||
| 130 | state->state = 0; | ||
| 131 | state->private = 0; | ||
| 132 | state->tree = NULL; | ||
| 133 | #ifdef LEAK_DEBUG | ||
| 134 | spin_lock_irqsave(&leak_lock, flags); | ||
| 135 | list_add(&state->leak_list, &states); | ||
| 136 | spin_unlock_irqrestore(&leak_lock, flags); | ||
| 137 | #endif | ||
| 138 | atomic_set(&state->refs, 1); | ||
| 139 | init_waitqueue_head(&state->wq); | ||
| 140 | return state; | ||
| 141 | } | ||
| 142 | |||
| 143 | static void free_extent_state(struct extent_state *state) | ||
| 144 | { | ||
| 145 | if (!state) | ||
| 146 | return; | ||
| 147 | if (atomic_dec_and_test(&state->refs)) { | ||
| 148 | #ifdef LEAK_DEBUG | ||
| 149 | unsigned long flags; | ||
| 150 | #endif | ||
| 151 | WARN_ON(state->tree); | ||
| 152 | #ifdef LEAK_DEBUG | ||
| 153 | spin_lock_irqsave(&leak_lock, flags); | ||
| 154 | list_del(&state->leak_list); | ||
| 155 | spin_unlock_irqrestore(&leak_lock, flags); | ||
| 156 | #endif | ||
| 157 | kmem_cache_free(extent_state_cache, state); | ||
| 158 | } | ||
| 159 | } | ||
| 160 | |||
| 161 | static struct rb_node *tree_insert(struct rb_root *root, u64 offset, | ||
| 162 | struct rb_node *node) | ||
| 163 | { | ||
| 164 | struct rb_node **p = &root->rb_node; | ||
| 165 | struct rb_node *parent = NULL; | ||
| 166 | struct tree_entry *entry; | ||
| 167 | |||
| 168 | while (*p) { | ||
| 169 | parent = *p; | ||
| 170 | entry = rb_entry(parent, struct tree_entry, rb_node); | ||
| 171 | |||
| 172 | if (offset < entry->start) | ||
| 173 | p = &(*p)->rb_left; | ||
| 174 | else if (offset > entry->end) | ||
| 175 | p = &(*p)->rb_right; | ||
| 176 | else | ||
| 177 | return parent; | ||
| 178 | } | ||
| 179 | |||
| 180 | entry = rb_entry(node, struct tree_entry, rb_node); | ||
| 181 | rb_link_node(node, parent, p); | ||
| 182 | rb_insert_color(node, root); | ||
| 183 | return NULL; | ||
| 184 | } | ||
| 185 | |||
| 186 | static struct rb_node *__etree_search(struct extent_io_tree *tree, u64 offset, | ||
| 187 | struct rb_node **prev_ret, | ||
| 188 | struct rb_node **next_ret) | ||
| 189 | { | ||
| 190 | struct rb_root *root = &tree->state; | ||
| 191 | struct rb_node *n = root->rb_node; | ||
| 192 | struct rb_node *prev = NULL; | ||
| 193 | struct rb_node *orig_prev = NULL; | ||
| 194 | struct tree_entry *entry; | ||
| 195 | struct tree_entry *prev_entry = NULL; | ||
| 196 | |||
| 197 | while (n) { | ||
| 198 | entry = rb_entry(n, struct tree_entry, rb_node); | ||
| 199 | prev = n; | ||
| 200 | prev_entry = entry; | ||
| 201 | |||
| 202 | if (offset < entry->start) | ||
| 203 | n = n->rb_left; | ||
| 204 | else if (offset > entry->end) | ||
| 205 | n = n->rb_right; | ||
| 206 | else | ||
| 207 | return n; | ||
| 208 | } | ||
| 209 | |||
| 210 | if (prev_ret) { | ||
| 211 | orig_prev = prev; | ||
| 212 | while (prev && offset > prev_entry->end) { | ||
| 213 | prev = rb_next(prev); | ||
| 214 | prev_entry = rb_entry(prev, struct tree_entry, rb_node); | ||
| 215 | } | ||
| 216 | *prev_ret = prev; | ||
| 217 | prev = orig_prev; | ||
| 218 | } | ||
| 219 | |||
| 220 | if (next_ret) { | ||
| 221 | prev_entry = rb_entry(prev, struct tree_entry, rb_node); | ||
| 222 | while (prev && offset < prev_entry->start) { | ||
| 223 | prev = rb_prev(prev); | ||
| 224 | prev_entry = rb_entry(prev, struct tree_entry, rb_node); | ||
| 225 | } | ||
| 226 | *next_ret = prev; | ||
| 227 | } | ||
| 228 | return NULL; | ||
| 229 | } | ||
| 230 | |||
| 231 | static inline struct rb_node *tree_search(struct extent_io_tree *tree, | ||
| 232 | u64 offset) | ||
| 233 | { | ||
| 234 | struct rb_node *prev = NULL; | ||
| 235 | struct rb_node *ret; | ||
| 236 | |||
| 237 | ret = __etree_search(tree, offset, &prev, NULL); | ||
| 238 | if (!ret) | ||
| 239 | return prev; | ||
| 240 | return ret; | ||
| 241 | } | ||
| 242 | |||
| 243 | static struct extent_buffer *buffer_tree_insert(struct extent_io_tree *tree, | ||
| 244 | u64 offset, struct rb_node *node) | ||
| 245 | { | ||
| 246 | struct rb_root *root = &tree->buffer; | ||
| 247 | struct rb_node **p = &root->rb_node; | ||
| 248 | struct rb_node *parent = NULL; | ||
| 249 | struct extent_buffer *eb; | ||
| 250 | |||
| 251 | while (*p) { | ||
| 252 | parent = *p; | ||
| 253 | eb = rb_entry(parent, struct extent_buffer, rb_node); | ||
| 254 | |||
| 255 | if (offset < eb->start) | ||
| 256 | p = &(*p)->rb_left; | ||
| 257 | else if (offset > eb->start) | ||
| 258 | p = &(*p)->rb_right; | ||
| 259 | else | ||
| 260 | return eb; | ||
| 261 | } | ||
| 262 | |||
| 263 | rb_link_node(node, parent, p); | ||
| 264 | rb_insert_color(node, root); | ||
| 265 | return NULL; | ||
| 266 | } | ||
| 267 | |||
| 268 | static struct extent_buffer *buffer_search(struct extent_io_tree *tree, | ||
| 269 | u64 offset) | ||
| 270 | { | ||
| 271 | struct rb_root *root = &tree->buffer; | ||
| 272 | struct rb_node *n = root->rb_node; | ||
| 273 | struct extent_buffer *eb; | ||
| 274 | |||
| 275 | while (n) { | ||
| 276 | eb = rb_entry(n, struct extent_buffer, rb_node); | ||
| 277 | if (offset < eb->start) | ||
| 278 | n = n->rb_left; | ||
| 279 | else if (offset > eb->start) | ||
| 280 | n = n->rb_right; | ||
| 281 | else | ||
| 282 | return eb; | ||
| 283 | } | ||
| 284 | return NULL; | ||
| 285 | } | ||
| 286 | |||
| 287 | /* | ||
| 288 | * utility function to look for merge candidates inside a given range. | ||
| 289 | * Any extents with matching state are merged together into a single | ||
| 290 | * extent in the tree. Extents with EXTENT_IO in their state field | ||
| 291 | * are not merged because the end_io handlers need to be able to do | ||
| 292 | * operations on them without sleeping (or doing allocations/splits). | ||
| 293 | * | ||
| 294 | * This should be called with the tree lock held. | ||
| 295 | */ | ||
| 296 | static int merge_state(struct extent_io_tree *tree, | ||
| 297 | struct extent_state *state) | ||
| 298 | { | ||
| 299 | struct extent_state *other; | ||
| 300 | struct rb_node *other_node; | ||
| 301 | |||
| 302 | if (state->state & (EXTENT_IOBITS | EXTENT_BOUNDARY)) | ||
| 303 | return 0; | ||
| 304 | |||
| 305 | other_node = rb_prev(&state->rb_node); | ||
| 306 | if (other_node) { | ||
| 307 | other = rb_entry(other_node, struct extent_state, rb_node); | ||
| 308 | if (other->end == state->start - 1 && | ||
| 309 | other->state == state->state) { | ||
| 310 | state->start = other->start; | ||
| 311 | other->tree = NULL; | ||
| 312 | rb_erase(&other->rb_node, &tree->state); | ||
| 313 | free_extent_state(other); | ||
| 314 | } | ||
| 315 | } | ||
| 316 | other_node = rb_next(&state->rb_node); | ||
| 317 | if (other_node) { | ||
| 318 | other = rb_entry(other_node, struct extent_state, rb_node); | ||
| 319 | if (other->start == state->end + 1 && | ||
| 320 | other->state == state->state) { | ||
| 321 | other->start = state->start; | ||
| 322 | state->tree = NULL; | ||
| 323 | rb_erase(&state->rb_node, &tree->state); | ||
| 324 | free_extent_state(state); | ||
| 325 | } | ||
| 326 | } | ||
| 327 | return 0; | ||
| 328 | } | ||
| 329 | |||
| 330 | static void set_state_cb(struct extent_io_tree *tree, | ||
| 331 | struct extent_state *state, | ||
| 332 | unsigned long bits) | ||
| 333 | { | ||
| 334 | if (tree->ops && tree->ops->set_bit_hook) { | ||
| 335 | tree->ops->set_bit_hook(tree->mapping->host, state->start, | ||
| 336 | state->end, state->state, bits); | ||
| 337 | } | ||
| 338 | } | ||
| 339 | |||
| 340 | static void clear_state_cb(struct extent_io_tree *tree, | ||
| 341 | struct extent_state *state, | ||
| 342 | unsigned long bits) | ||
| 343 | { | ||
| 344 | if (tree->ops && tree->ops->clear_bit_hook) { | ||
| 345 | tree->ops->clear_bit_hook(tree->mapping->host, state->start, | ||
| 346 | state->end, state->state, bits); | ||
| 347 | } | ||
| 348 | } | ||
| 349 | |||
| 350 | /* | ||
| 351 | * insert an extent_state struct into the tree. 'bits' are set on the | ||
| 352 | * struct before it is inserted. | ||
| 353 | * | ||
| 354 | * This may return -EEXIST if the extent is already there, in which case the | ||
| 355 | * state struct is freed. | ||
| 356 | * | ||
| 357 | * The tree lock is not taken internally. This is a utility function and | ||
| 358 | * probably isn't what you want to call (see set/clear_extent_bit). | ||
| 359 | */ | ||
| 360 | static int insert_state(struct extent_io_tree *tree, | ||
| 361 | struct extent_state *state, u64 start, u64 end, | ||
| 362 | int bits) | ||
| 363 | { | ||
| 364 | struct rb_node *node; | ||
| 365 | |||
| 366 | if (end < start) { | ||
| 367 | printk(KERN_ERR "btrfs end < start %llu %llu\n", | ||
| 368 | (unsigned long long)end, | ||
| 369 | (unsigned long long)start); | ||
| 370 | WARN_ON(1); | ||
| 371 | } | ||
| 372 | if (bits & EXTENT_DIRTY) | ||
| 373 | tree->dirty_bytes += end - start + 1; | ||
| 374 | set_state_cb(tree, state, bits); | ||
| 375 | state->state |= bits; | ||
| 376 | state->start = start; | ||
| 377 | state->end = end; | ||
| 378 | node = tree_insert(&tree->state, end, &state->rb_node); | ||
| 379 | if (node) { | ||
| 380 | struct extent_state *found; | ||
| 381 | found = rb_entry(node, struct extent_state, rb_node); | ||
| 382 | printk(KERN_ERR "btrfs found node %llu %llu on insert of " | ||
| 383 | "%llu %llu\n", (unsigned long long)found->start, | ||
| 384 | (unsigned long long)found->end, | ||
| 385 | (unsigned long long)start, (unsigned long long)end); | ||
| 386 | free_extent_state(state); | ||
| 387 | return -EEXIST; | ||
| 388 | } | ||
| 389 | state->tree = tree; | ||
| 390 | merge_state(tree, state); | ||
| 391 | return 0; | ||
| 392 | } | ||
| 393 | |||
| 394 | /* | ||
| 395 | * split a given extent state struct in two, inserting the preallocated | ||
| 396 | * struct 'prealloc' as the newly created second half. 'split' indicates an | ||
| 397 | * offset inside 'orig' where it should be split. | ||
| 398 | * | ||
| 399 | * Before calling, | ||
| 400 | * the tree has 'orig' at [orig->start, orig->end]. After calling, there | ||
| 401 | * are two extent state structs in the tree: | ||
| 402 | * prealloc: [orig->start, split - 1] | ||
| 403 | * orig: [ split, orig->end ] | ||
| 404 | * | ||
| 405 | * The tree locks are not taken by this function. They need to be held | ||
| 406 | * by the caller. | ||
| 407 | */ | ||
| 408 | static int split_state(struct extent_io_tree *tree, struct extent_state *orig, | ||
| 409 | struct extent_state *prealloc, u64 split) | ||
| 410 | { | ||
| 411 | struct rb_node *node; | ||
| 412 | prealloc->start = orig->start; | ||
| 413 | prealloc->end = split - 1; | ||
| 414 | prealloc->state = orig->state; | ||
| 415 | orig->start = split; | ||
| 416 | |||
| 417 | node = tree_insert(&tree->state, prealloc->end, &prealloc->rb_node); | ||
| 418 | if (node) { | ||
| 419 | struct extent_state *found; | ||
| 420 | found = rb_entry(node, struct extent_state, rb_node); | ||
| 421 | free_extent_state(prealloc); | ||
| 422 | return -EEXIST; | ||
| 423 | } | ||
| 424 | prealloc->tree = tree; | ||
| 425 | return 0; | ||
| 426 | } | ||
| 427 | |||
| 428 | /* | ||
| 429 | * utility function to clear some bits in an extent state struct. | ||
| 430 | * it will optionally wake up any one waiting on this state (wake == 1), or | ||
| 431 | * forcibly remove the state from the tree (delete == 1). | ||
| 432 | * | ||
| 433 | * If no bits are set on the state struct after clearing things, the | ||
| 434 | * struct is freed and removed from the tree | ||
| 435 | */ | ||
| 436 | static int clear_state_bit(struct extent_io_tree *tree, | ||
| 437 | struct extent_state *state, int bits, int wake, | ||
| 438 | int delete) | ||
| 439 | { | ||
| 440 | int ret = state->state & bits; | ||
| 441 | |||
| 442 | if ((bits & EXTENT_DIRTY) && (state->state & EXTENT_DIRTY)) { | ||
| 443 | u64 range = state->end - state->start + 1; | ||
| 444 | WARN_ON(range > tree->dirty_bytes); | ||
| 445 | tree->dirty_bytes -= range; | ||
| 446 | } | ||
| 447 | clear_state_cb(tree, state, bits); | ||
| 448 | state->state &= ~bits; | ||
| 449 | if (wake) | ||
| 450 | wake_up(&state->wq); | ||
| 451 | if (delete || state->state == 0) { | ||
| 452 | if (state->tree) { | ||
| 453 | clear_state_cb(tree, state, state->state); | ||
| 454 | rb_erase(&state->rb_node, &tree->state); | ||
| 455 | state->tree = NULL; | ||
| 456 | free_extent_state(state); | ||
| 457 | } else { | ||
| 458 | WARN_ON(1); | ||
| 459 | } | ||
| 460 | } else { | ||
| 461 | merge_state(tree, state); | ||
| 462 | } | ||
| 463 | return ret; | ||
| 464 | } | ||
| 465 | |||
| 466 | /* | ||
| 467 | * clear some bits on a range in the tree. This may require splitting | ||
| 468 | * or inserting elements in the tree, so the gfp mask is used to | ||
| 469 | * indicate which allocations or sleeping are allowed. | ||
| 470 | * | ||
| 471 | * pass 'wake' == 1 to kick any sleepers, and 'delete' == 1 to remove | ||
| 472 | * the given range from the tree regardless of state (ie for truncate). | ||
| 473 | * | ||
| 474 | * the range [start, end] is inclusive. | ||
| 475 | * | ||
| 476 | * This takes the tree lock, and returns < 0 on error, > 0 if any of the | ||
| 477 | * bits were already set, or zero if none of the bits were already set. | ||
| 478 | */ | ||
| 479 | int clear_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 480 | int bits, int wake, int delete, gfp_t mask) | ||
| 481 | { | ||
| 482 | struct extent_state *state; | ||
| 483 | struct extent_state *prealloc = NULL; | ||
| 484 | struct rb_node *node; | ||
| 485 | int err; | ||
| 486 | int set = 0; | ||
| 487 | |||
| 488 | again: | ||
| 489 | if (!prealloc && (mask & __GFP_WAIT)) { | ||
| 490 | prealloc = alloc_extent_state(mask); | ||
| 491 | if (!prealloc) | ||
| 492 | return -ENOMEM; | ||
| 493 | } | ||
| 494 | |||
| 495 | spin_lock(&tree->lock); | ||
| 496 | /* | ||
| 497 | * this search will find the extents that end after | ||
| 498 | * our range starts | ||
| 499 | */ | ||
| 500 | node = tree_search(tree, start); | ||
| 501 | if (!node) | ||
| 502 | goto out; | ||
| 503 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 504 | if (state->start > end) | ||
| 505 | goto out; | ||
| 506 | WARN_ON(state->end < start); | ||
| 507 | |||
| 508 | /* | ||
| 509 | * | ---- desired range ---- | | ||
| 510 | * | state | or | ||
| 511 | * | ------------- state -------------- | | ||
| 512 | * | ||
| 513 | * We need to split the extent we found, and may flip | ||
| 514 | * bits on second half. | ||
| 515 | * | ||
| 516 | * If the extent we found extends past our range, we | ||
| 517 | * just split and search again. It'll get split again | ||
| 518 | * the next time though. | ||
| 519 | * | ||
| 520 | * If the extent we found is inside our range, we clear | ||
| 521 | * the desired bit on it. | ||
| 522 | */ | ||
| 523 | |||
| 524 | if (state->start < start) { | ||
| 525 | if (!prealloc) | ||
| 526 | prealloc = alloc_extent_state(GFP_ATOMIC); | ||
| 527 | err = split_state(tree, state, prealloc, start); | ||
| 528 | BUG_ON(err == -EEXIST); | ||
| 529 | prealloc = NULL; | ||
| 530 | if (err) | ||
| 531 | goto out; | ||
| 532 | if (state->end <= end) { | ||
| 533 | start = state->end + 1; | ||
| 534 | set |= clear_state_bit(tree, state, bits, | ||
| 535 | wake, delete); | ||
| 536 | } else { | ||
| 537 | start = state->start; | ||
| 538 | } | ||
| 539 | goto search_again; | ||
| 540 | } | ||
| 541 | /* | ||
| 542 | * | ---- desired range ---- | | ||
| 543 | * | state | | ||
| 544 | * We need to split the extent, and clear the bit | ||
| 545 | * on the first half | ||
| 546 | */ | ||
| 547 | if (state->start <= end && state->end > end) { | ||
| 548 | if (!prealloc) | ||
| 549 | prealloc = alloc_extent_state(GFP_ATOMIC); | ||
| 550 | err = split_state(tree, state, prealloc, end + 1); | ||
| 551 | BUG_ON(err == -EEXIST); | ||
| 552 | |||
| 553 | if (wake) | ||
| 554 | wake_up(&state->wq); | ||
| 555 | set |= clear_state_bit(tree, prealloc, bits, | ||
| 556 | wake, delete); | ||
| 557 | prealloc = NULL; | ||
| 558 | goto out; | ||
| 559 | } | ||
| 560 | |||
| 561 | start = state->end + 1; | ||
| 562 | set |= clear_state_bit(tree, state, bits, wake, delete); | ||
| 563 | goto search_again; | ||
| 564 | |||
| 565 | out: | ||
| 566 | spin_unlock(&tree->lock); | ||
| 567 | if (prealloc) | ||
| 568 | free_extent_state(prealloc); | ||
| 569 | |||
| 570 | return set; | ||
| 571 | |||
| 572 | search_again: | ||
| 573 | if (start > end) | ||
| 574 | goto out; | ||
| 575 | spin_unlock(&tree->lock); | ||
| 576 | if (mask & __GFP_WAIT) | ||
| 577 | cond_resched(); | ||
| 578 | goto again; | ||
| 579 | } | ||
| 580 | |||
| 581 | static int wait_on_state(struct extent_io_tree *tree, | ||
| 582 | struct extent_state *state) | ||
| 583 | __releases(tree->lock) | ||
| 584 | __acquires(tree->lock) | ||
| 585 | { | ||
| 586 | DEFINE_WAIT(wait); | ||
| 587 | prepare_to_wait(&state->wq, &wait, TASK_UNINTERRUPTIBLE); | ||
| 588 | spin_unlock(&tree->lock); | ||
| 589 | schedule(); | ||
| 590 | spin_lock(&tree->lock); | ||
| 591 | finish_wait(&state->wq, &wait); | ||
| 592 | return 0; | ||
| 593 | } | ||
| 594 | |||
| 595 | /* | ||
| 596 | * waits for one or more bits to clear on a range in the state tree. | ||
| 597 | * The range [start, end] is inclusive. | ||
| 598 | * The tree lock is taken by this function | ||
| 599 | */ | ||
| 600 | int wait_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, int bits) | ||
| 601 | { | ||
| 602 | struct extent_state *state; | ||
| 603 | struct rb_node *node; | ||
| 604 | |||
| 605 | spin_lock(&tree->lock); | ||
| 606 | again: | ||
| 607 | while (1) { | ||
| 608 | /* | ||
| 609 | * this search will find all the extents that end after | ||
| 610 | * our range starts | ||
| 611 | */ | ||
| 612 | node = tree_search(tree, start); | ||
| 613 | if (!node) | ||
| 614 | break; | ||
| 615 | |||
| 616 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 617 | |||
| 618 | if (state->start > end) | ||
| 619 | goto out; | ||
| 620 | |||
| 621 | if (state->state & bits) { | ||
| 622 | start = state->start; | ||
| 623 | atomic_inc(&state->refs); | ||
| 624 | wait_on_state(tree, state); | ||
| 625 | free_extent_state(state); | ||
| 626 | goto again; | ||
| 627 | } | ||
| 628 | start = state->end + 1; | ||
| 629 | |||
| 630 | if (start > end) | ||
| 631 | break; | ||
| 632 | |||
| 633 | if (need_resched()) { | ||
| 634 | spin_unlock(&tree->lock); | ||
| 635 | cond_resched(); | ||
| 636 | spin_lock(&tree->lock); | ||
| 637 | } | ||
| 638 | } | ||
| 639 | out: | ||
| 640 | spin_unlock(&tree->lock); | ||
| 641 | return 0; | ||
| 642 | } | ||
| 643 | |||
| 644 | static void set_state_bits(struct extent_io_tree *tree, | ||
| 645 | struct extent_state *state, | ||
| 646 | int bits) | ||
| 647 | { | ||
| 648 | if ((bits & EXTENT_DIRTY) && !(state->state & EXTENT_DIRTY)) { | ||
| 649 | u64 range = state->end - state->start + 1; | ||
| 650 | tree->dirty_bytes += range; | ||
| 651 | } | ||
| 652 | set_state_cb(tree, state, bits); | ||
| 653 | state->state |= bits; | ||
| 654 | } | ||
| 655 | |||
| 656 | /* | ||
| 657 | * set some bits on a range in the tree. This may require allocations | ||
| 658 | * or sleeping, so the gfp mask is used to indicate what is allowed. | ||
| 659 | * | ||
| 660 | * If 'exclusive' == 1, this will fail with -EEXIST if some part of the | ||
| 661 | * range already has the desired bits set. The start of the existing | ||
| 662 | * range is returned in failed_start in this case. | ||
| 663 | * | ||
| 664 | * [start, end] is inclusive | ||
| 665 | * This takes the tree lock. | ||
| 666 | */ | ||
| 667 | static int set_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 668 | int bits, int exclusive, u64 *failed_start, | ||
| 669 | gfp_t mask) | ||
| 670 | { | ||
| 671 | struct extent_state *state; | ||
| 672 | struct extent_state *prealloc = NULL; | ||
| 673 | struct rb_node *node; | ||
| 674 | int err = 0; | ||
| 675 | int set; | ||
| 676 | u64 last_start; | ||
| 677 | u64 last_end; | ||
| 678 | again: | ||
| 679 | if (!prealloc && (mask & __GFP_WAIT)) { | ||
| 680 | prealloc = alloc_extent_state(mask); | ||
| 681 | if (!prealloc) | ||
| 682 | return -ENOMEM; | ||
| 683 | } | ||
| 684 | |||
| 685 | spin_lock(&tree->lock); | ||
| 686 | /* | ||
| 687 | * this search will find all the extents that end after | ||
| 688 | * our range starts. | ||
| 689 | */ | ||
| 690 | node = tree_search(tree, start); | ||
| 691 | if (!node) { | ||
| 692 | err = insert_state(tree, prealloc, start, end, bits); | ||
| 693 | prealloc = NULL; | ||
| 694 | BUG_ON(err == -EEXIST); | ||
| 695 | goto out; | ||
| 696 | } | ||
| 697 | |||
| 698 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 699 | last_start = state->start; | ||
| 700 | last_end = state->end; | ||
| 701 | |||
| 702 | /* | ||
| 703 | * | ---- desired range ---- | | ||
| 704 | * | state | | ||
| 705 | * | ||
| 706 | * Just lock what we found and keep going | ||
| 707 | */ | ||
| 708 | if (state->start == start && state->end <= end) { | ||
| 709 | set = state->state & bits; | ||
| 710 | if (set && exclusive) { | ||
| 711 | *failed_start = state->start; | ||
| 712 | err = -EEXIST; | ||
| 713 | goto out; | ||
| 714 | } | ||
| 715 | set_state_bits(tree, state, bits); | ||
| 716 | start = state->end + 1; | ||
| 717 | merge_state(tree, state); | ||
| 718 | goto search_again; | ||
| 719 | } | ||
| 720 | |||
| 721 | /* | ||
| 722 | * | ---- desired range ---- | | ||
| 723 | * | state | | ||
| 724 | * or | ||
| 725 | * | ------------- state -------------- | | ||
| 726 | * | ||
| 727 | * We need to split the extent we found, and may flip bits on | ||
| 728 | * second half. | ||
| 729 | * | ||
| 730 | * If the extent we found extends past our | ||
| 731 | * range, we just split and search again. It'll get split | ||
| 732 | * again the next time though. | ||
| 733 | * | ||
| 734 | * If the extent we found is inside our range, we set the | ||
| 735 | * desired bit on it. | ||
| 736 | */ | ||
| 737 | if (state->start < start) { | ||
| 738 | set = state->state & bits; | ||
| 739 | if (exclusive && set) { | ||
| 740 | *failed_start = start; | ||
| 741 | err = -EEXIST; | ||
| 742 | goto out; | ||
| 743 | } | ||
| 744 | err = split_state(tree, state, prealloc, start); | ||
| 745 | BUG_ON(err == -EEXIST); | ||
| 746 | prealloc = NULL; | ||
| 747 | if (err) | ||
| 748 | goto out; | ||
| 749 | if (state->end <= end) { | ||
| 750 | set_state_bits(tree, state, bits); | ||
| 751 | start = state->end + 1; | ||
| 752 | merge_state(tree, state); | ||
| 753 | } else { | ||
| 754 | start = state->start; | ||
| 755 | } | ||
| 756 | goto search_again; | ||
| 757 | } | ||
| 758 | /* | ||
| 759 | * | ---- desired range ---- | | ||
| 760 | * | state | or | state | | ||
| 761 | * | ||
| 762 | * There's a hole, we need to insert something in it and | ||
| 763 | * ignore the extent we found. | ||
| 764 | */ | ||
| 765 | if (state->start > start) { | ||
| 766 | u64 this_end; | ||
| 767 | if (end < last_start) | ||
| 768 | this_end = end; | ||
| 769 | else | ||
| 770 | this_end = last_start - 1; | ||
| 771 | err = insert_state(tree, prealloc, start, this_end, | ||
| 772 | bits); | ||
| 773 | prealloc = NULL; | ||
| 774 | BUG_ON(err == -EEXIST); | ||
| 775 | if (err) | ||
| 776 | goto out; | ||
| 777 | start = this_end + 1; | ||
| 778 | goto search_again; | ||
| 779 | } | ||
| 780 | /* | ||
| 781 | * | ---- desired range ---- | | ||
| 782 | * | state | | ||
| 783 | * We need to split the extent, and set the bit | ||
| 784 | * on the first half | ||
| 785 | */ | ||
| 786 | if (state->start <= end && state->end > end) { | ||
| 787 | set = state->state & bits; | ||
| 788 | if (exclusive && set) { | ||
| 789 | *failed_start = start; | ||
| 790 | err = -EEXIST; | ||
| 791 | goto out; | ||
| 792 | } | ||
| 793 | err = split_state(tree, state, prealloc, end + 1); | ||
| 794 | BUG_ON(err == -EEXIST); | ||
| 795 | |||
| 796 | set_state_bits(tree, prealloc, bits); | ||
| 797 | merge_state(tree, prealloc); | ||
| 798 | prealloc = NULL; | ||
| 799 | goto out; | ||
| 800 | } | ||
| 801 | |||
| 802 | goto search_again; | ||
| 803 | |||
| 804 | out: | ||
| 805 | spin_unlock(&tree->lock); | ||
| 806 | if (prealloc) | ||
| 807 | free_extent_state(prealloc); | ||
| 808 | |||
| 809 | return err; | ||
| 810 | |||
| 811 | search_again: | ||
| 812 | if (start > end) | ||
| 813 | goto out; | ||
| 814 | spin_unlock(&tree->lock); | ||
| 815 | if (mask & __GFP_WAIT) | ||
| 816 | cond_resched(); | ||
| 817 | goto again; | ||
| 818 | } | ||
| 819 | |||
| 820 | /* wrappers around set/clear extent bit */ | ||
| 821 | int set_extent_dirty(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 822 | gfp_t mask) | ||
| 823 | { | ||
| 824 | return set_extent_bit(tree, start, end, EXTENT_DIRTY, 0, NULL, | ||
| 825 | mask); | ||
| 826 | } | ||
| 827 | |||
| 828 | int set_extent_ordered(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 829 | gfp_t mask) | ||
| 830 | { | ||
| 831 | return set_extent_bit(tree, start, end, EXTENT_ORDERED, 0, NULL, mask); | ||
| 832 | } | ||
| 833 | |||
| 834 | int set_extent_bits(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 835 | int bits, gfp_t mask) | ||
| 836 | { | ||
| 837 | return set_extent_bit(tree, start, end, bits, 0, NULL, | ||
| 838 | mask); | ||
| 839 | } | ||
| 840 | |||
| 841 | int clear_extent_bits(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 842 | int bits, gfp_t mask) | ||
| 843 | { | ||
| 844 | return clear_extent_bit(tree, start, end, bits, 0, 0, mask); | ||
| 845 | } | ||
| 846 | |||
| 847 | int set_extent_delalloc(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 848 | gfp_t mask) | ||
| 849 | { | ||
| 850 | return set_extent_bit(tree, start, end, | ||
| 851 | EXTENT_DELALLOC | EXTENT_DIRTY, | ||
| 852 | 0, NULL, mask); | ||
| 853 | } | ||
| 854 | |||
| 855 | int clear_extent_dirty(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 856 | gfp_t mask) | ||
| 857 | { | ||
| 858 | return clear_extent_bit(tree, start, end, | ||
| 859 | EXTENT_DIRTY | EXTENT_DELALLOC, 0, 0, mask); | ||
| 860 | } | ||
| 861 | |||
| 862 | int clear_extent_ordered(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 863 | gfp_t mask) | ||
| 864 | { | ||
| 865 | return clear_extent_bit(tree, start, end, EXTENT_ORDERED, 1, 0, mask); | ||
| 866 | } | ||
| 867 | |||
| 868 | int set_extent_new(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 869 | gfp_t mask) | ||
| 870 | { | ||
| 871 | return set_extent_bit(tree, start, end, EXTENT_NEW, 0, NULL, | ||
| 872 | mask); | ||
| 873 | } | ||
| 874 | |||
| 875 | static int clear_extent_new(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 876 | gfp_t mask) | ||
| 877 | { | ||
| 878 | return clear_extent_bit(tree, start, end, EXTENT_NEW, 0, 0, mask); | ||
| 879 | } | ||
| 880 | |||
| 881 | int set_extent_uptodate(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 882 | gfp_t mask) | ||
| 883 | { | ||
| 884 | return set_extent_bit(tree, start, end, EXTENT_UPTODATE, 0, NULL, | ||
| 885 | mask); | ||
| 886 | } | ||
| 887 | |||
| 888 | static int clear_extent_uptodate(struct extent_io_tree *tree, u64 start, | ||
| 889 | u64 end, gfp_t mask) | ||
| 890 | { | ||
| 891 | return clear_extent_bit(tree, start, end, EXTENT_UPTODATE, 0, 0, mask); | ||
| 892 | } | ||
| 893 | |||
| 894 | static int set_extent_writeback(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 895 | gfp_t mask) | ||
| 896 | { | ||
| 897 | return set_extent_bit(tree, start, end, EXTENT_WRITEBACK, | ||
| 898 | 0, NULL, mask); | ||
| 899 | } | ||
| 900 | |||
| 901 | static int clear_extent_writeback(struct extent_io_tree *tree, u64 start, | ||
| 902 | u64 end, gfp_t mask) | ||
| 903 | { | ||
| 904 | return clear_extent_bit(tree, start, end, EXTENT_WRITEBACK, 1, 0, mask); | ||
| 905 | } | ||
| 906 | |||
| 907 | int wait_on_extent_writeback(struct extent_io_tree *tree, u64 start, u64 end) | ||
| 908 | { | ||
| 909 | return wait_extent_bit(tree, start, end, EXTENT_WRITEBACK); | ||
| 910 | } | ||
| 911 | |||
| 912 | /* | ||
| 913 | * either insert or lock state struct between start and end use mask to tell | ||
| 914 | * us if waiting is desired. | ||
| 915 | */ | ||
| 916 | int lock_extent(struct extent_io_tree *tree, u64 start, u64 end, gfp_t mask) | ||
| 917 | { | ||
| 918 | int err; | ||
| 919 | u64 failed_start; | ||
| 920 | while (1) { | ||
| 921 | err = set_extent_bit(tree, start, end, EXTENT_LOCKED, 1, | ||
| 922 | &failed_start, mask); | ||
| 923 | if (err == -EEXIST && (mask & __GFP_WAIT)) { | ||
| 924 | wait_extent_bit(tree, failed_start, end, EXTENT_LOCKED); | ||
| 925 | start = failed_start; | ||
| 926 | } else { | ||
| 927 | break; | ||
| 928 | } | ||
| 929 | WARN_ON(start > end); | ||
| 930 | } | ||
| 931 | return err; | ||
| 932 | } | ||
| 933 | |||
| 934 | int try_lock_extent(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 935 | gfp_t mask) | ||
| 936 | { | ||
| 937 | int err; | ||
| 938 | u64 failed_start; | ||
| 939 | |||
| 940 | err = set_extent_bit(tree, start, end, EXTENT_LOCKED, 1, | ||
| 941 | &failed_start, mask); | ||
| 942 | if (err == -EEXIST) { | ||
| 943 | if (failed_start > start) | ||
| 944 | clear_extent_bit(tree, start, failed_start - 1, | ||
| 945 | EXTENT_LOCKED, 1, 0, mask); | ||
| 946 | return 0; | ||
| 947 | } | ||
| 948 | return 1; | ||
| 949 | } | ||
| 950 | |||
| 951 | int unlock_extent(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 952 | gfp_t mask) | ||
| 953 | { | ||
| 954 | return clear_extent_bit(tree, start, end, EXTENT_LOCKED, 1, 0, mask); | ||
| 955 | } | ||
| 956 | |||
| 957 | /* | ||
| 958 | * helper function to set pages and extents in the tree dirty | ||
| 959 | */ | ||
| 960 | int set_range_dirty(struct extent_io_tree *tree, u64 start, u64 end) | ||
| 961 | { | ||
| 962 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 963 | unsigned long end_index = end >> PAGE_CACHE_SHIFT; | ||
| 964 | struct page *page; | ||
| 965 | |||
| 966 | while (index <= end_index) { | ||
| 967 | page = find_get_page(tree->mapping, index); | ||
| 968 | BUG_ON(!page); | ||
| 969 | __set_page_dirty_nobuffers(page); | ||
| 970 | page_cache_release(page); | ||
| 971 | index++; | ||
| 972 | } | ||
| 973 | set_extent_dirty(tree, start, end, GFP_NOFS); | ||
| 974 | return 0; | ||
| 975 | } | ||
| 976 | |||
| 977 | /* | ||
| 978 | * helper function to set both pages and extents in the tree writeback | ||
| 979 | */ | ||
| 980 | static int set_range_writeback(struct extent_io_tree *tree, u64 start, u64 end) | ||
| 981 | { | ||
| 982 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 983 | unsigned long end_index = end >> PAGE_CACHE_SHIFT; | ||
| 984 | struct page *page; | ||
| 985 | |||
| 986 | while (index <= end_index) { | ||
| 987 | page = find_get_page(tree->mapping, index); | ||
| 988 | BUG_ON(!page); | ||
| 989 | set_page_writeback(page); | ||
| 990 | page_cache_release(page); | ||
| 991 | index++; | ||
| 992 | } | ||
| 993 | set_extent_writeback(tree, start, end, GFP_NOFS); | ||
| 994 | return 0; | ||
| 995 | } | ||
| 996 | |||
| 997 | /* | ||
| 998 | * find the first offset in the io tree with 'bits' set. zero is | ||
| 999 | * returned if we find something, and *start_ret and *end_ret are | ||
| 1000 | * set to reflect the state struct that was found. | ||
| 1001 | * | ||
| 1002 | * If nothing was found, 1 is returned, < 0 on error | ||
| 1003 | */ | ||
| 1004 | int find_first_extent_bit(struct extent_io_tree *tree, u64 start, | ||
| 1005 | u64 *start_ret, u64 *end_ret, int bits) | ||
| 1006 | { | ||
| 1007 | struct rb_node *node; | ||
| 1008 | struct extent_state *state; | ||
| 1009 | int ret = 1; | ||
| 1010 | |||
| 1011 | spin_lock(&tree->lock); | ||
| 1012 | /* | ||
| 1013 | * this search will find all the extents that end after | ||
| 1014 | * our range starts. | ||
| 1015 | */ | ||
| 1016 | node = tree_search(tree, start); | ||
| 1017 | if (!node) | ||
| 1018 | goto out; | ||
| 1019 | |||
| 1020 | while (1) { | ||
| 1021 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 1022 | if (state->end >= start && (state->state & bits)) { | ||
| 1023 | *start_ret = state->start; | ||
| 1024 | *end_ret = state->end; | ||
| 1025 | ret = 0; | ||
| 1026 | break; | ||
| 1027 | } | ||
| 1028 | node = rb_next(node); | ||
| 1029 | if (!node) | ||
| 1030 | break; | ||
| 1031 | } | ||
| 1032 | out: | ||
| 1033 | spin_unlock(&tree->lock); | ||
| 1034 | return ret; | ||
| 1035 | } | ||
| 1036 | |||
| 1037 | /* find the first state struct with 'bits' set after 'start', and | ||
| 1038 | * return it. tree->lock must be held. NULL will returned if | ||
| 1039 | * nothing was found after 'start' | ||
| 1040 | */ | ||
| 1041 | struct extent_state *find_first_extent_bit_state(struct extent_io_tree *tree, | ||
| 1042 | u64 start, int bits) | ||
| 1043 | { | ||
| 1044 | struct rb_node *node; | ||
| 1045 | struct extent_state *state; | ||
| 1046 | |||
| 1047 | /* | ||
| 1048 | * this search will find all the extents that end after | ||
| 1049 | * our range starts. | ||
| 1050 | */ | ||
| 1051 | node = tree_search(tree, start); | ||
| 1052 | if (!node) | ||
| 1053 | goto out; | ||
| 1054 | |||
| 1055 | while (1) { | ||
| 1056 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 1057 | if (state->end >= start && (state->state & bits)) | ||
| 1058 | return state; | ||
| 1059 | |||
| 1060 | node = rb_next(node); | ||
| 1061 | if (!node) | ||
| 1062 | break; | ||
| 1063 | } | ||
| 1064 | out: | ||
| 1065 | return NULL; | ||
| 1066 | } | ||
| 1067 | |||
| 1068 | /* | ||
| 1069 | * find a contiguous range of bytes in the file marked as delalloc, not | ||
| 1070 | * more than 'max_bytes'. start and end are used to return the range, | ||
| 1071 | * | ||
| 1072 | * 1 is returned if we find something, 0 if nothing was in the tree | ||
| 1073 | */ | ||
| 1074 | static noinline u64 find_delalloc_range(struct extent_io_tree *tree, | ||
| 1075 | u64 *start, u64 *end, u64 max_bytes) | ||
| 1076 | { | ||
| 1077 | struct rb_node *node; | ||
| 1078 | struct extent_state *state; | ||
| 1079 | u64 cur_start = *start; | ||
| 1080 | u64 found = 0; | ||
| 1081 | u64 total_bytes = 0; | ||
| 1082 | |||
| 1083 | spin_lock(&tree->lock); | ||
| 1084 | |||
| 1085 | /* | ||
| 1086 | * this search will find all the extents that end after | ||
| 1087 | * our range starts. | ||
| 1088 | */ | ||
| 1089 | node = tree_search(tree, cur_start); | ||
| 1090 | if (!node) { | ||
| 1091 | if (!found) | ||
| 1092 | *end = (u64)-1; | ||
| 1093 | goto out; | ||
| 1094 | } | ||
| 1095 | |||
| 1096 | while (1) { | ||
| 1097 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 1098 | if (found && (state->start != cur_start || | ||
| 1099 | (state->state & EXTENT_BOUNDARY))) { | ||
| 1100 | goto out; | ||
| 1101 | } | ||
| 1102 | if (!(state->state & EXTENT_DELALLOC)) { | ||
| 1103 | if (!found) | ||
| 1104 | *end = state->end; | ||
| 1105 | goto out; | ||
| 1106 | } | ||
| 1107 | if (!found) | ||
| 1108 | *start = state->start; | ||
| 1109 | found++; | ||
| 1110 | *end = state->end; | ||
| 1111 | cur_start = state->end + 1; | ||
| 1112 | node = rb_next(node); | ||
| 1113 | if (!node) | ||
| 1114 | break; | ||
| 1115 | total_bytes += state->end - state->start + 1; | ||
| 1116 | if (total_bytes >= max_bytes) | ||
| 1117 | break; | ||
| 1118 | } | ||
| 1119 | out: | ||
| 1120 | spin_unlock(&tree->lock); | ||
| 1121 | return found; | ||
| 1122 | } | ||
| 1123 | |||
| 1124 | static noinline int __unlock_for_delalloc(struct inode *inode, | ||
| 1125 | struct page *locked_page, | ||
| 1126 | u64 start, u64 end) | ||
| 1127 | { | ||
| 1128 | int ret; | ||
| 1129 | struct page *pages[16]; | ||
| 1130 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 1131 | unsigned long end_index = end >> PAGE_CACHE_SHIFT; | ||
| 1132 | unsigned long nr_pages = end_index - index + 1; | ||
| 1133 | int i; | ||
| 1134 | |||
| 1135 | if (index == locked_page->index && end_index == index) | ||
| 1136 | return 0; | ||
| 1137 | |||
| 1138 | while (nr_pages > 0) { | ||
| 1139 | ret = find_get_pages_contig(inode->i_mapping, index, | ||
| 1140 | min_t(unsigned long, nr_pages, | ||
| 1141 | ARRAY_SIZE(pages)), pages); | ||
| 1142 | for (i = 0; i < ret; i++) { | ||
| 1143 | if (pages[i] != locked_page) | ||
| 1144 | unlock_page(pages[i]); | ||
| 1145 | page_cache_release(pages[i]); | ||
| 1146 | } | ||
| 1147 | nr_pages -= ret; | ||
| 1148 | index += ret; | ||
| 1149 | cond_resched(); | ||
| 1150 | } | ||
| 1151 | return 0; | ||
| 1152 | } | ||
| 1153 | |||
| 1154 | static noinline int lock_delalloc_pages(struct inode *inode, | ||
| 1155 | struct page *locked_page, | ||
| 1156 | u64 delalloc_start, | ||
| 1157 | u64 delalloc_end) | ||
| 1158 | { | ||
| 1159 | unsigned long index = delalloc_start >> PAGE_CACHE_SHIFT; | ||
| 1160 | unsigned long start_index = index; | ||
| 1161 | unsigned long end_index = delalloc_end >> PAGE_CACHE_SHIFT; | ||
| 1162 | unsigned long pages_locked = 0; | ||
| 1163 | struct page *pages[16]; | ||
| 1164 | unsigned long nrpages; | ||
| 1165 | int ret; | ||
| 1166 | int i; | ||
| 1167 | |||
| 1168 | /* the caller is responsible for locking the start index */ | ||
| 1169 | if (index == locked_page->index && index == end_index) | ||
| 1170 | return 0; | ||
| 1171 | |||
| 1172 | /* skip the page at the start index */ | ||
| 1173 | nrpages = end_index - index + 1; | ||
| 1174 | while (nrpages > 0) { | ||
| 1175 | ret = find_get_pages_contig(inode->i_mapping, index, | ||
| 1176 | min_t(unsigned long, | ||
| 1177 | nrpages, ARRAY_SIZE(pages)), pages); | ||
| 1178 | if (ret == 0) { | ||
| 1179 | ret = -EAGAIN; | ||
| 1180 | goto done; | ||
| 1181 | } | ||
| 1182 | /* now we have an array of pages, lock them all */ | ||
| 1183 | for (i = 0; i < ret; i++) { | ||
| 1184 | /* | ||
| 1185 | * the caller is taking responsibility for | ||
| 1186 | * locked_page | ||
| 1187 | */ | ||
| 1188 | if (pages[i] != locked_page) { | ||
| 1189 | lock_page(pages[i]); | ||
| 1190 | if (!PageDirty(pages[i]) || | ||
| 1191 | pages[i]->mapping != inode->i_mapping) { | ||
| 1192 | ret = -EAGAIN; | ||
| 1193 | unlock_page(pages[i]); | ||
| 1194 | page_cache_release(pages[i]); | ||
| 1195 | goto done; | ||
| 1196 | } | ||
| 1197 | } | ||
| 1198 | page_cache_release(pages[i]); | ||
| 1199 | pages_locked++; | ||
| 1200 | } | ||
| 1201 | nrpages -= ret; | ||
| 1202 | index += ret; | ||
| 1203 | cond_resched(); | ||
| 1204 | } | ||
| 1205 | ret = 0; | ||
| 1206 | done: | ||
| 1207 | if (ret && pages_locked) { | ||
| 1208 | __unlock_for_delalloc(inode, locked_page, | ||
| 1209 | delalloc_start, | ||
| 1210 | ((u64)(start_index + pages_locked - 1)) << | ||
| 1211 | PAGE_CACHE_SHIFT); | ||
| 1212 | } | ||
| 1213 | return ret; | ||
| 1214 | } | ||
| 1215 | |||
| 1216 | /* | ||
| 1217 | * find a contiguous range of bytes in the file marked as delalloc, not | ||
| 1218 | * more than 'max_bytes'. start and end are used to return the range, | ||
| 1219 | * | ||
| 1220 | * 1 is returned if we find something, 0 if nothing was in the tree | ||
| 1221 | */ | ||
| 1222 | static noinline u64 find_lock_delalloc_range(struct inode *inode, | ||
| 1223 | struct extent_io_tree *tree, | ||
| 1224 | struct page *locked_page, | ||
| 1225 | u64 *start, u64 *end, | ||
| 1226 | u64 max_bytes) | ||
| 1227 | { | ||
| 1228 | u64 delalloc_start; | ||
| 1229 | u64 delalloc_end; | ||
| 1230 | u64 found; | ||
| 1231 | int ret; | ||
| 1232 | int loops = 0; | ||
| 1233 | |||
| 1234 | again: | ||
| 1235 | /* step one, find a bunch of delalloc bytes starting at start */ | ||
| 1236 | delalloc_start = *start; | ||
| 1237 | delalloc_end = 0; | ||
| 1238 | found = find_delalloc_range(tree, &delalloc_start, &delalloc_end, | ||
| 1239 | max_bytes); | ||
| 1240 | if (!found || delalloc_end <= *start) { | ||
| 1241 | *start = delalloc_start; | ||
| 1242 | *end = delalloc_end; | ||
| 1243 | return found; | ||
| 1244 | } | ||
| 1245 | |||
| 1246 | /* | ||
| 1247 | * start comes from the offset of locked_page. We have to lock | ||
| 1248 | * pages in order, so we can't process delalloc bytes before | ||
| 1249 | * locked_page | ||
| 1250 | */ | ||
| 1251 | if (delalloc_start < *start) | ||
| 1252 | delalloc_start = *start; | ||
| 1253 | |||
| 1254 | /* | ||
| 1255 | * make sure to limit the number of pages we try to lock down | ||
| 1256 | * if we're looping. | ||
| 1257 | */ | ||
| 1258 | if (delalloc_end + 1 - delalloc_start > max_bytes && loops) | ||
| 1259 | delalloc_end = delalloc_start + PAGE_CACHE_SIZE - 1; | ||
| 1260 | |||
| 1261 | /* step two, lock all the pages after the page that has start */ | ||
| 1262 | ret = lock_delalloc_pages(inode, locked_page, | ||
| 1263 | delalloc_start, delalloc_end); | ||
| 1264 | if (ret == -EAGAIN) { | ||
| 1265 | /* some of the pages are gone, lets avoid looping by | ||
| 1266 | * shortening the size of the delalloc range we're searching | ||
| 1267 | */ | ||
| 1268 | if (!loops) { | ||
| 1269 | unsigned long offset = (*start) & (PAGE_CACHE_SIZE - 1); | ||
| 1270 | max_bytes = PAGE_CACHE_SIZE - offset; | ||
| 1271 | loops = 1; | ||
| 1272 | goto again; | ||
| 1273 | } else { | ||
| 1274 | found = 0; | ||
| 1275 | goto out_failed; | ||
| 1276 | } | ||
| 1277 | } | ||
| 1278 | BUG_ON(ret); | ||
| 1279 | |||
| 1280 | /* step three, lock the state bits for the whole range */ | ||
| 1281 | lock_extent(tree, delalloc_start, delalloc_end, GFP_NOFS); | ||
| 1282 | |||
| 1283 | /* then test to make sure it is all still delalloc */ | ||
| 1284 | ret = test_range_bit(tree, delalloc_start, delalloc_end, | ||
| 1285 | EXTENT_DELALLOC, 1); | ||
| 1286 | if (!ret) { | ||
| 1287 | unlock_extent(tree, delalloc_start, delalloc_end, GFP_NOFS); | ||
| 1288 | __unlock_for_delalloc(inode, locked_page, | ||
| 1289 | delalloc_start, delalloc_end); | ||
| 1290 | cond_resched(); | ||
| 1291 | goto again; | ||
| 1292 | } | ||
| 1293 | *start = delalloc_start; | ||
| 1294 | *end = delalloc_end; | ||
| 1295 | out_failed: | ||
| 1296 | return found; | ||
| 1297 | } | ||
| 1298 | |||
| 1299 | int extent_clear_unlock_delalloc(struct inode *inode, | ||
| 1300 | struct extent_io_tree *tree, | ||
| 1301 | u64 start, u64 end, struct page *locked_page, | ||
| 1302 | int unlock_pages, | ||
| 1303 | int clear_unlock, | ||
| 1304 | int clear_delalloc, int clear_dirty, | ||
| 1305 | int set_writeback, | ||
| 1306 | int end_writeback) | ||
| 1307 | { | ||
| 1308 | int ret; | ||
| 1309 | struct page *pages[16]; | ||
| 1310 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 1311 | unsigned long end_index = end >> PAGE_CACHE_SHIFT; | ||
| 1312 | unsigned long nr_pages = end_index - index + 1; | ||
| 1313 | int i; | ||
| 1314 | int clear_bits = 0; | ||
| 1315 | |||
| 1316 | if (clear_unlock) | ||
| 1317 | clear_bits |= EXTENT_LOCKED; | ||
| 1318 | if (clear_dirty) | ||
| 1319 | clear_bits |= EXTENT_DIRTY; | ||
| 1320 | |||
| 1321 | if (clear_delalloc) | ||
| 1322 | clear_bits |= EXTENT_DELALLOC; | ||
| 1323 | |||
| 1324 | clear_extent_bit(tree, start, end, clear_bits, 1, 0, GFP_NOFS); | ||
| 1325 | if (!(unlock_pages || clear_dirty || set_writeback || end_writeback)) | ||
| 1326 | return 0; | ||
| 1327 | |||
| 1328 | while (nr_pages > 0) { | ||
| 1329 | ret = find_get_pages_contig(inode->i_mapping, index, | ||
| 1330 | min_t(unsigned long, | ||
| 1331 | nr_pages, ARRAY_SIZE(pages)), pages); | ||
| 1332 | for (i = 0; i < ret; i++) { | ||
| 1333 | if (pages[i] == locked_page) { | ||
| 1334 | page_cache_release(pages[i]); | ||
| 1335 | continue; | ||
| 1336 | } | ||
| 1337 | if (clear_dirty) | ||
| 1338 | clear_page_dirty_for_io(pages[i]); | ||
| 1339 | if (set_writeback) | ||
| 1340 | set_page_writeback(pages[i]); | ||
| 1341 | if (end_writeback) | ||
| 1342 | end_page_writeback(pages[i]); | ||
| 1343 | if (unlock_pages) | ||
| 1344 | unlock_page(pages[i]); | ||
| 1345 | page_cache_release(pages[i]); | ||
| 1346 | } | ||
| 1347 | nr_pages -= ret; | ||
| 1348 | index += ret; | ||
| 1349 | cond_resched(); | ||
| 1350 | } | ||
| 1351 | return 0; | ||
| 1352 | } | ||
| 1353 | |||
| 1354 | /* | ||
| 1355 | * count the number of bytes in the tree that have a given bit(s) | ||
| 1356 | * set. This can be fairly slow, except for EXTENT_DIRTY which is | ||
| 1357 | * cached. The total number found is returned. | ||
| 1358 | */ | ||
| 1359 | u64 count_range_bits(struct extent_io_tree *tree, | ||
| 1360 | u64 *start, u64 search_end, u64 max_bytes, | ||
| 1361 | unsigned long bits) | ||
| 1362 | { | ||
| 1363 | struct rb_node *node; | ||
| 1364 | struct extent_state *state; | ||
| 1365 | u64 cur_start = *start; | ||
| 1366 | u64 total_bytes = 0; | ||
| 1367 | int found = 0; | ||
| 1368 | |||
| 1369 | if (search_end <= cur_start) { | ||
| 1370 | WARN_ON(1); | ||
| 1371 | return 0; | ||
| 1372 | } | ||
| 1373 | |||
| 1374 | spin_lock(&tree->lock); | ||
| 1375 | if (cur_start == 0 && bits == EXTENT_DIRTY) { | ||
| 1376 | total_bytes = tree->dirty_bytes; | ||
| 1377 | goto out; | ||
| 1378 | } | ||
| 1379 | /* | ||
| 1380 | * this search will find all the extents that end after | ||
| 1381 | * our range starts. | ||
| 1382 | */ | ||
| 1383 | node = tree_search(tree, cur_start); | ||
| 1384 | if (!node) | ||
| 1385 | goto out; | ||
| 1386 | |||
| 1387 | while (1) { | ||
| 1388 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 1389 | if (state->start > search_end) | ||
| 1390 | break; | ||
| 1391 | if (state->end >= cur_start && (state->state & bits)) { | ||
| 1392 | total_bytes += min(search_end, state->end) + 1 - | ||
| 1393 | max(cur_start, state->start); | ||
| 1394 | if (total_bytes >= max_bytes) | ||
| 1395 | break; | ||
| 1396 | if (!found) { | ||
| 1397 | *start = state->start; | ||
| 1398 | found = 1; | ||
| 1399 | } | ||
| 1400 | } | ||
| 1401 | node = rb_next(node); | ||
| 1402 | if (!node) | ||
| 1403 | break; | ||
| 1404 | } | ||
| 1405 | out: | ||
| 1406 | spin_unlock(&tree->lock); | ||
| 1407 | return total_bytes; | ||
| 1408 | } | ||
| 1409 | |||
| 1410 | #if 0 | ||
| 1411 | /* | ||
| 1412 | * helper function to lock both pages and extents in the tree. | ||
| 1413 | * pages must be locked first. | ||
| 1414 | */ | ||
| 1415 | static int lock_range(struct extent_io_tree *tree, u64 start, u64 end) | ||
| 1416 | { | ||
| 1417 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 1418 | unsigned long end_index = end >> PAGE_CACHE_SHIFT; | ||
| 1419 | struct page *page; | ||
| 1420 | int err; | ||
| 1421 | |||
| 1422 | while (index <= end_index) { | ||
| 1423 | page = grab_cache_page(tree->mapping, index); | ||
| 1424 | if (!page) { | ||
| 1425 | err = -ENOMEM; | ||
| 1426 | goto failed; | ||
| 1427 | } | ||
| 1428 | if (IS_ERR(page)) { | ||
| 1429 | err = PTR_ERR(page); | ||
| 1430 | goto failed; | ||
| 1431 | } | ||
| 1432 | index++; | ||
| 1433 | } | ||
| 1434 | lock_extent(tree, start, end, GFP_NOFS); | ||
| 1435 | return 0; | ||
| 1436 | |||
| 1437 | failed: | ||
| 1438 | /* | ||
| 1439 | * we failed above in getting the page at 'index', so we undo here | ||
| 1440 | * up to but not including the page at 'index' | ||
| 1441 | */ | ||
| 1442 | end_index = index; | ||
| 1443 | index = start >> PAGE_CACHE_SHIFT; | ||
| 1444 | while (index < end_index) { | ||
| 1445 | page = find_get_page(tree->mapping, index); | ||
| 1446 | unlock_page(page); | ||
| 1447 | page_cache_release(page); | ||
| 1448 | index++; | ||
| 1449 | } | ||
| 1450 | return err; | ||
| 1451 | } | ||
| 1452 | |||
| 1453 | /* | ||
| 1454 | * helper function to unlock both pages and extents in the tree. | ||
| 1455 | */ | ||
| 1456 | static int unlock_range(struct extent_io_tree *tree, u64 start, u64 end) | ||
| 1457 | { | ||
| 1458 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 1459 | unsigned long end_index = end >> PAGE_CACHE_SHIFT; | ||
| 1460 | struct page *page; | ||
| 1461 | |||
| 1462 | while (index <= end_index) { | ||
| 1463 | page = find_get_page(tree->mapping, index); | ||
| 1464 | unlock_page(page); | ||
| 1465 | page_cache_release(page); | ||
| 1466 | index++; | ||
| 1467 | } | ||
| 1468 | unlock_extent(tree, start, end, GFP_NOFS); | ||
| 1469 | return 0; | ||
| 1470 | } | ||
| 1471 | #endif | ||
| 1472 | |||
| 1473 | /* | ||
| 1474 | * set the private field for a given byte offset in the tree. If there isn't | ||
| 1475 | * an extent_state there already, this does nothing. | ||
| 1476 | */ | ||
| 1477 | int set_state_private(struct extent_io_tree *tree, u64 start, u64 private) | ||
| 1478 | { | ||
| 1479 | struct rb_node *node; | ||
| 1480 | struct extent_state *state; | ||
| 1481 | int ret = 0; | ||
| 1482 | |||
| 1483 | spin_lock(&tree->lock); | ||
| 1484 | /* | ||
| 1485 | * this search will find all the extents that end after | ||
| 1486 | * our range starts. | ||
| 1487 | */ | ||
| 1488 | node = tree_search(tree, start); | ||
| 1489 | if (!node) { | ||
| 1490 | ret = -ENOENT; | ||
| 1491 | goto out; | ||
| 1492 | } | ||
| 1493 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 1494 | if (state->start != start) { | ||
| 1495 | ret = -ENOENT; | ||
| 1496 | goto out; | ||
| 1497 | } | ||
| 1498 | state->private = private; | ||
| 1499 | out: | ||
| 1500 | spin_unlock(&tree->lock); | ||
| 1501 | return ret; | ||
| 1502 | } | ||
| 1503 | |||
| 1504 | int get_state_private(struct extent_io_tree *tree, u64 start, u64 *private) | ||
| 1505 | { | ||
| 1506 | struct rb_node *node; | ||
| 1507 | struct extent_state *state; | ||
| 1508 | int ret = 0; | ||
| 1509 | |||
| 1510 | spin_lock(&tree->lock); | ||
| 1511 | /* | ||
| 1512 | * this search will find all the extents that end after | ||
| 1513 | * our range starts. | ||
| 1514 | */ | ||
| 1515 | node = tree_search(tree, start); | ||
| 1516 | if (!node) { | ||
| 1517 | ret = -ENOENT; | ||
| 1518 | goto out; | ||
| 1519 | } | ||
| 1520 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 1521 | if (state->start != start) { | ||
| 1522 | ret = -ENOENT; | ||
| 1523 | goto out; | ||
| 1524 | } | ||
| 1525 | *private = state->private; | ||
| 1526 | out: | ||
| 1527 | spin_unlock(&tree->lock); | ||
| 1528 | return ret; | ||
| 1529 | } | ||
| 1530 | |||
| 1531 | /* | ||
| 1532 | * searches a range in the state tree for a given mask. | ||
| 1533 | * If 'filled' == 1, this returns 1 only if every extent in the tree | ||
| 1534 | * has the bits set. Otherwise, 1 is returned if any bit in the | ||
| 1535 | * range is found set. | ||
| 1536 | */ | ||
| 1537 | int test_range_bit(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 1538 | int bits, int filled) | ||
| 1539 | { | ||
| 1540 | struct extent_state *state = NULL; | ||
| 1541 | struct rb_node *node; | ||
| 1542 | int bitset = 0; | ||
| 1543 | |||
| 1544 | spin_lock(&tree->lock); | ||
| 1545 | node = tree_search(tree, start); | ||
| 1546 | while (node && start <= end) { | ||
| 1547 | state = rb_entry(node, struct extent_state, rb_node); | ||
| 1548 | |||
| 1549 | if (filled && state->start > start) { | ||
| 1550 | bitset = 0; | ||
| 1551 | break; | ||
| 1552 | } | ||
| 1553 | |||
| 1554 | if (state->start > end) | ||
| 1555 | break; | ||
| 1556 | |||
| 1557 | if (state->state & bits) { | ||
| 1558 | bitset = 1; | ||
| 1559 | if (!filled) | ||
| 1560 | break; | ||
| 1561 | } else if (filled) { | ||
| 1562 | bitset = 0; | ||
| 1563 | break; | ||
| 1564 | } | ||
| 1565 | start = state->end + 1; | ||
| 1566 | if (start > end) | ||
| 1567 | break; | ||
| 1568 | node = rb_next(node); | ||
| 1569 | if (!node) { | ||
| 1570 | if (filled) | ||
| 1571 | bitset = 0; | ||
| 1572 | break; | ||
| 1573 | } | ||
| 1574 | } | ||
| 1575 | spin_unlock(&tree->lock); | ||
| 1576 | return bitset; | ||
| 1577 | } | ||
| 1578 | |||
| 1579 | /* | ||
| 1580 | * helper function to set a given page up to date if all the | ||
| 1581 | * extents in the tree for that page are up to date | ||
| 1582 | */ | ||
| 1583 | static int check_page_uptodate(struct extent_io_tree *tree, | ||
| 1584 | struct page *page) | ||
| 1585 | { | ||
| 1586 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 1587 | u64 end = start + PAGE_CACHE_SIZE - 1; | ||
| 1588 | if (test_range_bit(tree, start, end, EXTENT_UPTODATE, 1)) | ||
| 1589 | SetPageUptodate(page); | ||
| 1590 | return 0; | ||
| 1591 | } | ||
| 1592 | |||
| 1593 | /* | ||
| 1594 | * helper function to unlock a page if all the extents in the tree | ||
| 1595 | * for that page are unlocked | ||
| 1596 | */ | ||
| 1597 | static int check_page_locked(struct extent_io_tree *tree, | ||
| 1598 | struct page *page) | ||
| 1599 | { | ||
| 1600 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 1601 | u64 end = start + PAGE_CACHE_SIZE - 1; | ||
| 1602 | if (!test_range_bit(tree, start, end, EXTENT_LOCKED, 0)) | ||
| 1603 | unlock_page(page); | ||
| 1604 | return 0; | ||
| 1605 | } | ||
| 1606 | |||
| 1607 | /* | ||
| 1608 | * helper function to end page writeback if all the extents | ||
| 1609 | * in the tree for that page are done with writeback | ||
| 1610 | */ | ||
| 1611 | static int check_page_writeback(struct extent_io_tree *tree, | ||
| 1612 | struct page *page) | ||
| 1613 | { | ||
| 1614 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 1615 | u64 end = start + PAGE_CACHE_SIZE - 1; | ||
| 1616 | if (!test_range_bit(tree, start, end, EXTENT_WRITEBACK, 0)) | ||
| 1617 | end_page_writeback(page); | ||
| 1618 | return 0; | ||
| 1619 | } | ||
| 1620 | |||
| 1621 | /* lots and lots of room for performance fixes in the end_bio funcs */ | ||
| 1622 | |||
| 1623 | /* | ||
| 1624 | * after a writepage IO is done, we need to: | ||
| 1625 | * clear the uptodate bits on error | ||
| 1626 | * clear the writeback bits in the extent tree for this IO | ||
| 1627 | * end_page_writeback if the page has no more pending IO | ||
| 1628 | * | ||
| 1629 | * Scheduling is not allowed, so the extent state tree is expected | ||
| 1630 | * to have one and only one object corresponding to this IO. | ||
| 1631 | */ | ||
| 1632 | static void end_bio_extent_writepage(struct bio *bio, int err) | ||
| 1633 | { | ||
| 1634 | int uptodate = err == 0; | ||
| 1635 | struct bio_vec *bvec = bio->bi_io_vec + bio->bi_vcnt - 1; | ||
| 1636 | struct extent_io_tree *tree; | ||
| 1637 | u64 start; | ||
| 1638 | u64 end; | ||
| 1639 | int whole_page; | ||
| 1640 | int ret; | ||
| 1641 | |||
| 1642 | do { | ||
| 1643 | struct page *page = bvec->bv_page; | ||
| 1644 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 1645 | |||
| 1646 | start = ((u64)page->index << PAGE_CACHE_SHIFT) + | ||
| 1647 | bvec->bv_offset; | ||
| 1648 | end = start + bvec->bv_len - 1; | ||
| 1649 | |||
| 1650 | if (bvec->bv_offset == 0 && bvec->bv_len == PAGE_CACHE_SIZE) | ||
| 1651 | whole_page = 1; | ||
| 1652 | else | ||
| 1653 | whole_page = 0; | ||
| 1654 | |||
| 1655 | if (--bvec >= bio->bi_io_vec) | ||
| 1656 | prefetchw(&bvec->bv_page->flags); | ||
| 1657 | if (tree->ops && tree->ops->writepage_end_io_hook) { | ||
| 1658 | ret = tree->ops->writepage_end_io_hook(page, start, | ||
| 1659 | end, NULL, uptodate); | ||
| 1660 | if (ret) | ||
| 1661 | uptodate = 0; | ||
| 1662 | } | ||
| 1663 | |||
| 1664 | if (!uptodate && tree->ops && | ||
| 1665 | tree->ops->writepage_io_failed_hook) { | ||
| 1666 | ret = tree->ops->writepage_io_failed_hook(bio, page, | ||
| 1667 | start, end, NULL); | ||
| 1668 | if (ret == 0) { | ||
| 1669 | uptodate = (err == 0); | ||
| 1670 | continue; | ||
| 1671 | } | ||
| 1672 | } | ||
| 1673 | |||
| 1674 | if (!uptodate) { | ||
| 1675 | clear_extent_uptodate(tree, start, end, GFP_ATOMIC); | ||
| 1676 | ClearPageUptodate(page); | ||
| 1677 | SetPageError(page); | ||
| 1678 | } | ||
| 1679 | |||
| 1680 | clear_extent_writeback(tree, start, end, GFP_ATOMIC); | ||
| 1681 | |||
| 1682 | if (whole_page) | ||
| 1683 | end_page_writeback(page); | ||
| 1684 | else | ||
| 1685 | check_page_writeback(tree, page); | ||
| 1686 | } while (bvec >= bio->bi_io_vec); | ||
| 1687 | |||
| 1688 | bio_put(bio); | ||
| 1689 | } | ||
| 1690 | |||
| 1691 | /* | ||
| 1692 | * after a readpage IO is done, we need to: | ||
| 1693 | * clear the uptodate bits on error | ||
| 1694 | * set the uptodate bits if things worked | ||
| 1695 | * set the page up to date if all extents in the tree are uptodate | ||
| 1696 | * clear the lock bit in the extent tree | ||
| 1697 | * unlock the page if there are no other extents locked for it | ||
| 1698 | * | ||
| 1699 | * Scheduling is not allowed, so the extent state tree is expected | ||
| 1700 | * to have one and only one object corresponding to this IO. | ||
| 1701 | */ | ||
| 1702 | static void end_bio_extent_readpage(struct bio *bio, int err) | ||
| 1703 | { | ||
| 1704 | int uptodate = test_bit(BIO_UPTODATE, &bio->bi_flags); | ||
| 1705 | struct bio_vec *bvec = bio->bi_io_vec + bio->bi_vcnt - 1; | ||
| 1706 | struct extent_io_tree *tree; | ||
| 1707 | u64 start; | ||
| 1708 | u64 end; | ||
| 1709 | int whole_page; | ||
| 1710 | int ret; | ||
| 1711 | |||
| 1712 | if (err) | ||
| 1713 | uptodate = 0; | ||
| 1714 | |||
| 1715 | do { | ||
| 1716 | struct page *page = bvec->bv_page; | ||
| 1717 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 1718 | |||
| 1719 | start = ((u64)page->index << PAGE_CACHE_SHIFT) + | ||
| 1720 | bvec->bv_offset; | ||
| 1721 | end = start + bvec->bv_len - 1; | ||
| 1722 | |||
| 1723 | if (bvec->bv_offset == 0 && bvec->bv_len == PAGE_CACHE_SIZE) | ||
| 1724 | whole_page = 1; | ||
| 1725 | else | ||
| 1726 | whole_page = 0; | ||
| 1727 | |||
| 1728 | if (--bvec >= bio->bi_io_vec) | ||
| 1729 | prefetchw(&bvec->bv_page->flags); | ||
| 1730 | |||
| 1731 | if (uptodate && tree->ops && tree->ops->readpage_end_io_hook) { | ||
| 1732 | ret = tree->ops->readpage_end_io_hook(page, start, end, | ||
| 1733 | NULL); | ||
| 1734 | if (ret) | ||
| 1735 | uptodate = 0; | ||
| 1736 | } | ||
| 1737 | if (!uptodate && tree->ops && | ||
| 1738 | tree->ops->readpage_io_failed_hook) { | ||
| 1739 | ret = tree->ops->readpage_io_failed_hook(bio, page, | ||
| 1740 | start, end, NULL); | ||
| 1741 | if (ret == 0) { | ||
| 1742 | uptodate = | ||
| 1743 | test_bit(BIO_UPTODATE, &bio->bi_flags); | ||
| 1744 | if (err) | ||
| 1745 | uptodate = 0; | ||
| 1746 | continue; | ||
| 1747 | } | ||
| 1748 | } | ||
| 1749 | |||
| 1750 | if (uptodate) { | ||
| 1751 | set_extent_uptodate(tree, start, end, | ||
| 1752 | GFP_ATOMIC); | ||
| 1753 | } | ||
| 1754 | unlock_extent(tree, start, end, GFP_ATOMIC); | ||
| 1755 | |||
| 1756 | if (whole_page) { | ||
| 1757 | if (uptodate) { | ||
| 1758 | SetPageUptodate(page); | ||
| 1759 | } else { | ||
| 1760 | ClearPageUptodate(page); | ||
| 1761 | SetPageError(page); | ||
| 1762 | } | ||
| 1763 | unlock_page(page); | ||
| 1764 | } else { | ||
| 1765 | if (uptodate) { | ||
| 1766 | check_page_uptodate(tree, page); | ||
| 1767 | } else { | ||
| 1768 | ClearPageUptodate(page); | ||
| 1769 | SetPageError(page); | ||
| 1770 | } | ||
| 1771 | check_page_locked(tree, page); | ||
| 1772 | } | ||
| 1773 | } while (bvec >= bio->bi_io_vec); | ||
| 1774 | |||
| 1775 | bio_put(bio); | ||
| 1776 | } | ||
| 1777 | |||
| 1778 | /* | ||
| 1779 | * IO done from prepare_write is pretty simple, we just unlock | ||
| 1780 | * the structs in the extent tree when done, and set the uptodate bits | ||
| 1781 | * as appropriate. | ||
| 1782 | */ | ||
| 1783 | static void end_bio_extent_preparewrite(struct bio *bio, int err) | ||
| 1784 | { | ||
| 1785 | const int uptodate = test_bit(BIO_UPTODATE, &bio->bi_flags); | ||
| 1786 | struct bio_vec *bvec = bio->bi_io_vec + bio->bi_vcnt - 1; | ||
| 1787 | struct extent_io_tree *tree; | ||
| 1788 | u64 start; | ||
| 1789 | u64 end; | ||
| 1790 | |||
| 1791 | do { | ||
| 1792 | struct page *page = bvec->bv_page; | ||
| 1793 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 1794 | |||
| 1795 | start = ((u64)page->index << PAGE_CACHE_SHIFT) + | ||
| 1796 | bvec->bv_offset; | ||
| 1797 | end = start + bvec->bv_len - 1; | ||
| 1798 | |||
| 1799 | if (--bvec >= bio->bi_io_vec) | ||
| 1800 | prefetchw(&bvec->bv_page->flags); | ||
| 1801 | |||
| 1802 | if (uptodate) { | ||
| 1803 | set_extent_uptodate(tree, start, end, GFP_ATOMIC); | ||
| 1804 | } else { | ||
| 1805 | ClearPageUptodate(page); | ||
| 1806 | SetPageError(page); | ||
| 1807 | } | ||
| 1808 | |||
| 1809 | unlock_extent(tree, start, end, GFP_ATOMIC); | ||
| 1810 | |||
| 1811 | } while (bvec >= bio->bi_io_vec); | ||
| 1812 | |||
| 1813 | bio_put(bio); | ||
| 1814 | } | ||
| 1815 | |||
| 1816 | static struct bio * | ||
| 1817 | extent_bio_alloc(struct block_device *bdev, u64 first_sector, int nr_vecs, | ||
| 1818 | gfp_t gfp_flags) | ||
| 1819 | { | ||
| 1820 | struct bio *bio; | ||
| 1821 | |||
| 1822 | bio = bio_alloc(gfp_flags, nr_vecs); | ||
| 1823 | |||
| 1824 | if (bio == NULL && (current->flags & PF_MEMALLOC)) { | ||
| 1825 | while (!bio && (nr_vecs /= 2)) | ||
| 1826 | bio = bio_alloc(gfp_flags, nr_vecs); | ||
| 1827 | } | ||
| 1828 | |||
| 1829 | if (bio) { | ||
| 1830 | bio->bi_size = 0; | ||
| 1831 | bio->bi_bdev = bdev; | ||
| 1832 | bio->bi_sector = first_sector; | ||
| 1833 | } | ||
| 1834 | return bio; | ||
| 1835 | } | ||
| 1836 | |||
| 1837 | static int submit_one_bio(int rw, struct bio *bio, int mirror_num, | ||
| 1838 | unsigned long bio_flags) | ||
| 1839 | { | ||
| 1840 | int ret = 0; | ||
| 1841 | struct bio_vec *bvec = bio->bi_io_vec + bio->bi_vcnt - 1; | ||
| 1842 | struct page *page = bvec->bv_page; | ||
| 1843 | struct extent_io_tree *tree = bio->bi_private; | ||
| 1844 | u64 start; | ||
| 1845 | u64 end; | ||
| 1846 | |||
| 1847 | start = ((u64)page->index << PAGE_CACHE_SHIFT) + bvec->bv_offset; | ||
| 1848 | end = start + bvec->bv_len - 1; | ||
| 1849 | |||
| 1850 | bio->bi_private = NULL; | ||
| 1851 | |||
| 1852 | bio_get(bio); | ||
| 1853 | |||
| 1854 | if (tree->ops && tree->ops->submit_bio_hook) | ||
| 1855 | tree->ops->submit_bio_hook(page->mapping->host, rw, bio, | ||
| 1856 | mirror_num, bio_flags); | ||
| 1857 | else | ||
| 1858 | submit_bio(rw, bio); | ||
| 1859 | if (bio_flagged(bio, BIO_EOPNOTSUPP)) | ||
| 1860 | ret = -EOPNOTSUPP; | ||
| 1861 | bio_put(bio); | ||
| 1862 | return ret; | ||
| 1863 | } | ||
| 1864 | |||
| 1865 | static int submit_extent_page(int rw, struct extent_io_tree *tree, | ||
| 1866 | struct page *page, sector_t sector, | ||
| 1867 | size_t size, unsigned long offset, | ||
| 1868 | struct block_device *bdev, | ||
| 1869 | struct bio **bio_ret, | ||
| 1870 | unsigned long max_pages, | ||
| 1871 | bio_end_io_t end_io_func, | ||
| 1872 | int mirror_num, | ||
| 1873 | unsigned long prev_bio_flags, | ||
| 1874 | unsigned long bio_flags) | ||
| 1875 | { | ||
| 1876 | int ret = 0; | ||
| 1877 | struct bio *bio; | ||
| 1878 | int nr; | ||
| 1879 | int contig = 0; | ||
| 1880 | int this_compressed = bio_flags & EXTENT_BIO_COMPRESSED; | ||
| 1881 | int old_compressed = prev_bio_flags & EXTENT_BIO_COMPRESSED; | ||
| 1882 | size_t page_size = min_t(size_t, size, PAGE_CACHE_SIZE); | ||
| 1883 | |||
| 1884 | if (bio_ret && *bio_ret) { | ||
| 1885 | bio = *bio_ret; | ||
| 1886 | if (old_compressed) | ||
| 1887 | contig = bio->bi_sector == sector; | ||
| 1888 | else | ||
| 1889 | contig = bio->bi_sector + (bio->bi_size >> 9) == | ||
| 1890 | sector; | ||
| 1891 | |||
| 1892 | if (prev_bio_flags != bio_flags || !contig || | ||
| 1893 | (tree->ops && tree->ops->merge_bio_hook && | ||
| 1894 | tree->ops->merge_bio_hook(page, offset, page_size, bio, | ||
| 1895 | bio_flags)) || | ||
| 1896 | bio_add_page(bio, page, page_size, offset) < page_size) { | ||
| 1897 | ret = submit_one_bio(rw, bio, mirror_num, | ||
| 1898 | prev_bio_flags); | ||
| 1899 | bio = NULL; | ||
| 1900 | } else { | ||
| 1901 | return 0; | ||
| 1902 | } | ||
| 1903 | } | ||
| 1904 | if (this_compressed) | ||
| 1905 | nr = BIO_MAX_PAGES; | ||
| 1906 | else | ||
| 1907 | nr = bio_get_nr_vecs(bdev); | ||
| 1908 | |||
| 1909 | bio = extent_bio_alloc(bdev, sector, nr, GFP_NOFS | __GFP_HIGH); | ||
| 1910 | |||
| 1911 | bio_add_page(bio, page, page_size, offset); | ||
| 1912 | bio->bi_end_io = end_io_func; | ||
| 1913 | bio->bi_private = tree; | ||
| 1914 | |||
| 1915 | if (bio_ret) | ||
| 1916 | *bio_ret = bio; | ||
| 1917 | else | ||
| 1918 | ret = submit_one_bio(rw, bio, mirror_num, bio_flags); | ||
| 1919 | |||
| 1920 | return ret; | ||
| 1921 | } | ||
| 1922 | |||
| 1923 | void set_page_extent_mapped(struct page *page) | ||
| 1924 | { | ||
| 1925 | if (!PagePrivate(page)) { | ||
| 1926 | SetPagePrivate(page); | ||
| 1927 | page_cache_get(page); | ||
| 1928 | set_page_private(page, EXTENT_PAGE_PRIVATE); | ||
| 1929 | } | ||
| 1930 | } | ||
| 1931 | |||
| 1932 | static void set_page_extent_head(struct page *page, unsigned long len) | ||
| 1933 | { | ||
| 1934 | set_page_private(page, EXTENT_PAGE_PRIVATE_FIRST_PAGE | len << 2); | ||
| 1935 | } | ||
| 1936 | |||
| 1937 | /* | ||
| 1938 | * basic readpage implementation. Locked extent state structs are inserted | ||
| 1939 | * into the tree that are removed when the IO is done (by the end_io | ||
| 1940 | * handlers) | ||
| 1941 | */ | ||
| 1942 | static int __extent_read_full_page(struct extent_io_tree *tree, | ||
| 1943 | struct page *page, | ||
| 1944 | get_extent_t *get_extent, | ||
| 1945 | struct bio **bio, int mirror_num, | ||
| 1946 | unsigned long *bio_flags) | ||
| 1947 | { | ||
| 1948 | struct inode *inode = page->mapping->host; | ||
| 1949 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 1950 | u64 page_end = start + PAGE_CACHE_SIZE - 1; | ||
| 1951 | u64 end; | ||
| 1952 | u64 cur = start; | ||
| 1953 | u64 extent_offset; | ||
| 1954 | u64 last_byte = i_size_read(inode); | ||
| 1955 | u64 block_start; | ||
| 1956 | u64 cur_end; | ||
| 1957 | sector_t sector; | ||
| 1958 | struct extent_map *em; | ||
| 1959 | struct block_device *bdev; | ||
| 1960 | int ret; | ||
| 1961 | int nr = 0; | ||
| 1962 | size_t page_offset = 0; | ||
| 1963 | size_t iosize; | ||
| 1964 | size_t disk_io_size; | ||
| 1965 | size_t blocksize = inode->i_sb->s_blocksize; | ||
| 1966 | unsigned long this_bio_flag = 0; | ||
| 1967 | |||
| 1968 | set_page_extent_mapped(page); | ||
| 1969 | |||
| 1970 | end = page_end; | ||
| 1971 | lock_extent(tree, start, end, GFP_NOFS); | ||
| 1972 | |||
| 1973 | if (page->index == last_byte >> PAGE_CACHE_SHIFT) { | ||
| 1974 | char *userpage; | ||
| 1975 | size_t zero_offset = last_byte & (PAGE_CACHE_SIZE - 1); | ||
| 1976 | |||
| 1977 | if (zero_offset) { | ||
| 1978 | iosize = PAGE_CACHE_SIZE - zero_offset; | ||
| 1979 | userpage = kmap_atomic(page, KM_USER0); | ||
| 1980 | memset(userpage + zero_offset, 0, iosize); | ||
| 1981 | flush_dcache_page(page); | ||
| 1982 | kunmap_atomic(userpage, KM_USER0); | ||
| 1983 | } | ||
| 1984 | } | ||
| 1985 | while (cur <= end) { | ||
| 1986 | if (cur >= last_byte) { | ||
| 1987 | char *userpage; | ||
| 1988 | iosize = PAGE_CACHE_SIZE - page_offset; | ||
| 1989 | userpage = kmap_atomic(page, KM_USER0); | ||
| 1990 | memset(userpage + page_offset, 0, iosize); | ||
| 1991 | flush_dcache_page(page); | ||
| 1992 | kunmap_atomic(userpage, KM_USER0); | ||
| 1993 | set_extent_uptodate(tree, cur, cur + iosize - 1, | ||
| 1994 | GFP_NOFS); | ||
| 1995 | unlock_extent(tree, cur, cur + iosize - 1, GFP_NOFS); | ||
| 1996 | break; | ||
| 1997 | } | ||
| 1998 | em = get_extent(inode, page, page_offset, cur, | ||
| 1999 | end - cur + 1, 0); | ||
| 2000 | if (IS_ERR(em) || !em) { | ||
| 2001 | SetPageError(page); | ||
| 2002 | unlock_extent(tree, cur, end, GFP_NOFS); | ||
| 2003 | break; | ||
| 2004 | } | ||
| 2005 | extent_offset = cur - em->start; | ||
| 2006 | BUG_ON(extent_map_end(em) <= cur); | ||
| 2007 | BUG_ON(end < cur); | ||
| 2008 | |||
| 2009 | if (test_bit(EXTENT_FLAG_COMPRESSED, &em->flags)) | ||
| 2010 | this_bio_flag = EXTENT_BIO_COMPRESSED; | ||
| 2011 | |||
| 2012 | iosize = min(extent_map_end(em) - cur, end - cur + 1); | ||
| 2013 | cur_end = min(extent_map_end(em) - 1, end); | ||
| 2014 | iosize = (iosize + blocksize - 1) & ~((u64)blocksize - 1); | ||
| 2015 | if (this_bio_flag & EXTENT_BIO_COMPRESSED) { | ||
| 2016 | disk_io_size = em->block_len; | ||
| 2017 | sector = em->block_start >> 9; | ||
| 2018 | } else { | ||
| 2019 | sector = (em->block_start + extent_offset) >> 9; | ||
| 2020 | disk_io_size = iosize; | ||
| 2021 | } | ||
| 2022 | bdev = em->bdev; | ||
| 2023 | block_start = em->block_start; | ||
| 2024 | if (test_bit(EXTENT_FLAG_PREALLOC, &em->flags)) | ||
| 2025 | block_start = EXTENT_MAP_HOLE; | ||
| 2026 | free_extent_map(em); | ||
| 2027 | em = NULL; | ||
| 2028 | |||
| 2029 | /* we've found a hole, just zero and go on */ | ||
| 2030 | if (block_start == EXTENT_MAP_HOLE) { | ||
| 2031 | char *userpage; | ||
| 2032 | userpage = kmap_atomic(page, KM_USER0); | ||
| 2033 | memset(userpage + page_offset, 0, iosize); | ||
| 2034 | flush_dcache_page(page); | ||
| 2035 | kunmap_atomic(userpage, KM_USER0); | ||
| 2036 | |||
| 2037 | set_extent_uptodate(tree, cur, cur + iosize - 1, | ||
| 2038 | GFP_NOFS); | ||
| 2039 | unlock_extent(tree, cur, cur + iosize - 1, GFP_NOFS); | ||
| 2040 | cur = cur + iosize; | ||
| 2041 | page_offset += iosize; | ||
| 2042 | continue; | ||
| 2043 | } | ||
| 2044 | /* the get_extent function already copied into the page */ | ||
| 2045 | if (test_range_bit(tree, cur, cur_end, EXTENT_UPTODATE, 1)) { | ||
| 2046 | check_page_uptodate(tree, page); | ||
| 2047 | unlock_extent(tree, cur, cur + iosize - 1, GFP_NOFS); | ||
| 2048 | cur = cur + iosize; | ||
| 2049 | page_offset += iosize; | ||
| 2050 | continue; | ||
| 2051 | } | ||
| 2052 | /* we have an inline extent but it didn't get marked up | ||
| 2053 | * to date. Error out | ||
| 2054 | */ | ||
| 2055 | if (block_start == EXTENT_MAP_INLINE) { | ||
| 2056 | SetPageError(page); | ||
| 2057 | unlock_extent(tree, cur, cur + iosize - 1, GFP_NOFS); | ||
| 2058 | cur = cur + iosize; | ||
| 2059 | page_offset += iosize; | ||
| 2060 | continue; | ||
| 2061 | } | ||
| 2062 | |||
| 2063 | ret = 0; | ||
| 2064 | if (tree->ops && tree->ops->readpage_io_hook) { | ||
| 2065 | ret = tree->ops->readpage_io_hook(page, cur, | ||
| 2066 | cur + iosize - 1); | ||
| 2067 | } | ||
| 2068 | if (!ret) { | ||
| 2069 | unsigned long pnr = (last_byte >> PAGE_CACHE_SHIFT) + 1; | ||
| 2070 | pnr -= page->index; | ||
| 2071 | ret = submit_extent_page(READ, tree, page, | ||
| 2072 | sector, disk_io_size, page_offset, | ||
| 2073 | bdev, bio, pnr, | ||
| 2074 | end_bio_extent_readpage, mirror_num, | ||
| 2075 | *bio_flags, | ||
| 2076 | this_bio_flag); | ||
| 2077 | nr++; | ||
| 2078 | *bio_flags = this_bio_flag; | ||
| 2079 | } | ||
| 2080 | if (ret) | ||
| 2081 | SetPageError(page); | ||
| 2082 | cur = cur + iosize; | ||
| 2083 | page_offset += iosize; | ||
| 2084 | } | ||
| 2085 | if (!nr) { | ||
| 2086 | if (!PageError(page)) | ||
| 2087 | SetPageUptodate(page); | ||
| 2088 | unlock_page(page); | ||
| 2089 | } | ||
| 2090 | return 0; | ||
| 2091 | } | ||
| 2092 | |||
| 2093 | int extent_read_full_page(struct extent_io_tree *tree, struct page *page, | ||
| 2094 | get_extent_t *get_extent) | ||
| 2095 | { | ||
| 2096 | struct bio *bio = NULL; | ||
| 2097 | unsigned long bio_flags = 0; | ||
| 2098 | int ret; | ||
| 2099 | |||
| 2100 | ret = __extent_read_full_page(tree, page, get_extent, &bio, 0, | ||
| 2101 | &bio_flags); | ||
| 2102 | if (bio) | ||
| 2103 | submit_one_bio(READ, bio, 0, bio_flags); | ||
| 2104 | return ret; | ||
| 2105 | } | ||
| 2106 | |||
| 2107 | /* | ||
| 2108 | * the writepage semantics are similar to regular writepage. extent | ||
| 2109 | * records are inserted to lock ranges in the tree, and as dirty areas | ||
| 2110 | * are found, they are marked writeback. Then the lock bits are removed | ||
| 2111 | * and the end_io handler clears the writeback ranges | ||
| 2112 | */ | ||
| 2113 | static int __extent_writepage(struct page *page, struct writeback_control *wbc, | ||
| 2114 | void *data) | ||
| 2115 | { | ||
| 2116 | struct inode *inode = page->mapping->host; | ||
| 2117 | struct extent_page_data *epd = data; | ||
| 2118 | struct extent_io_tree *tree = epd->tree; | ||
| 2119 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 2120 | u64 delalloc_start; | ||
| 2121 | u64 page_end = start + PAGE_CACHE_SIZE - 1; | ||
| 2122 | u64 end; | ||
| 2123 | u64 cur = start; | ||
| 2124 | u64 extent_offset; | ||
| 2125 | u64 last_byte = i_size_read(inode); | ||
| 2126 | u64 block_start; | ||
| 2127 | u64 iosize; | ||
| 2128 | u64 unlock_start; | ||
| 2129 | sector_t sector; | ||
| 2130 | struct extent_map *em; | ||
| 2131 | struct block_device *bdev; | ||
| 2132 | int ret; | ||
| 2133 | int nr = 0; | ||
| 2134 | size_t pg_offset = 0; | ||
| 2135 | size_t blocksize; | ||
| 2136 | loff_t i_size = i_size_read(inode); | ||
| 2137 | unsigned long end_index = i_size >> PAGE_CACHE_SHIFT; | ||
| 2138 | u64 nr_delalloc; | ||
| 2139 | u64 delalloc_end; | ||
| 2140 | int page_started; | ||
| 2141 | int compressed; | ||
| 2142 | unsigned long nr_written = 0; | ||
| 2143 | |||
| 2144 | WARN_ON(!PageLocked(page)); | ||
| 2145 | pg_offset = i_size & (PAGE_CACHE_SIZE - 1); | ||
| 2146 | if (page->index > end_index || | ||
| 2147 | (page->index == end_index && !pg_offset)) { | ||
| 2148 | page->mapping->a_ops->invalidatepage(page, 0); | ||
| 2149 | unlock_page(page); | ||
| 2150 | return 0; | ||
| 2151 | } | ||
| 2152 | |||
| 2153 | if (page->index == end_index) { | ||
| 2154 | char *userpage; | ||
| 2155 | |||
| 2156 | userpage = kmap_atomic(page, KM_USER0); | ||
| 2157 | memset(userpage + pg_offset, 0, | ||
| 2158 | PAGE_CACHE_SIZE - pg_offset); | ||
| 2159 | kunmap_atomic(userpage, KM_USER0); | ||
| 2160 | flush_dcache_page(page); | ||
| 2161 | } | ||
| 2162 | pg_offset = 0; | ||
| 2163 | |||
| 2164 | set_page_extent_mapped(page); | ||
| 2165 | |||
| 2166 | delalloc_start = start; | ||
| 2167 | delalloc_end = 0; | ||
| 2168 | page_started = 0; | ||
| 2169 | if (!epd->extent_locked) { | ||
| 2170 | while (delalloc_end < page_end) { | ||
| 2171 | nr_delalloc = find_lock_delalloc_range(inode, tree, | ||
| 2172 | page, | ||
| 2173 | &delalloc_start, | ||
| 2174 | &delalloc_end, | ||
| 2175 | 128 * 1024 * 1024); | ||
| 2176 | if (nr_delalloc == 0) { | ||
| 2177 | delalloc_start = delalloc_end + 1; | ||
| 2178 | continue; | ||
| 2179 | } | ||
| 2180 | tree->ops->fill_delalloc(inode, page, delalloc_start, | ||
| 2181 | delalloc_end, &page_started, | ||
| 2182 | &nr_written); | ||
| 2183 | delalloc_start = delalloc_end + 1; | ||
| 2184 | } | ||
| 2185 | |||
| 2186 | /* did the fill delalloc function already unlock and start | ||
| 2187 | * the IO? | ||
| 2188 | */ | ||
| 2189 | if (page_started) { | ||
| 2190 | ret = 0; | ||
| 2191 | goto update_nr_written; | ||
| 2192 | } | ||
| 2193 | } | ||
| 2194 | lock_extent(tree, start, page_end, GFP_NOFS); | ||
| 2195 | |||
| 2196 | unlock_start = start; | ||
| 2197 | |||
| 2198 | if (tree->ops && tree->ops->writepage_start_hook) { | ||
| 2199 | ret = tree->ops->writepage_start_hook(page, start, | ||
| 2200 | page_end); | ||
| 2201 | if (ret == -EAGAIN) { | ||
| 2202 | unlock_extent(tree, start, page_end, GFP_NOFS); | ||
| 2203 | redirty_page_for_writepage(wbc, page); | ||
| 2204 | unlock_page(page); | ||
| 2205 | ret = 0; | ||
| 2206 | goto update_nr_written; | ||
| 2207 | } | ||
| 2208 | } | ||
| 2209 | |||
| 2210 | nr_written++; | ||
| 2211 | |||
| 2212 | end = page_end; | ||
| 2213 | if (test_range_bit(tree, start, page_end, EXTENT_DELALLOC, 0)) | ||
| 2214 | printk(KERN_ERR "btrfs delalloc bits after lock_extent\n"); | ||
| 2215 | |||
| 2216 | if (last_byte <= start) { | ||
| 2217 | clear_extent_dirty(tree, start, page_end, GFP_NOFS); | ||
| 2218 | unlock_extent(tree, start, page_end, GFP_NOFS); | ||
| 2219 | if (tree->ops && tree->ops->writepage_end_io_hook) | ||
| 2220 | tree->ops->writepage_end_io_hook(page, start, | ||
| 2221 | page_end, NULL, 1); | ||
| 2222 | unlock_start = page_end + 1; | ||
| 2223 | goto done; | ||
| 2224 | } | ||
| 2225 | |||
| 2226 | set_extent_uptodate(tree, start, page_end, GFP_NOFS); | ||
| 2227 | blocksize = inode->i_sb->s_blocksize; | ||
| 2228 | |||
| 2229 | while (cur <= end) { | ||
| 2230 | if (cur >= last_byte) { | ||
| 2231 | clear_extent_dirty(tree, cur, page_end, GFP_NOFS); | ||
| 2232 | unlock_extent(tree, unlock_start, page_end, GFP_NOFS); | ||
| 2233 | if (tree->ops && tree->ops->writepage_end_io_hook) | ||
| 2234 | tree->ops->writepage_end_io_hook(page, cur, | ||
| 2235 | page_end, NULL, 1); | ||
| 2236 | unlock_start = page_end + 1; | ||
| 2237 | break; | ||
| 2238 | } | ||
| 2239 | em = epd->get_extent(inode, page, pg_offset, cur, | ||
| 2240 | end - cur + 1, 1); | ||
| 2241 | if (IS_ERR(em) || !em) { | ||
| 2242 | SetPageError(page); | ||
| 2243 | break; | ||
| 2244 | } | ||
| 2245 | |||
| 2246 | extent_offset = cur - em->start; | ||
| 2247 | BUG_ON(extent_map_end(em) <= cur); | ||
| 2248 | BUG_ON(end < cur); | ||
| 2249 | iosize = min(extent_map_end(em) - cur, end - cur + 1); | ||
| 2250 | iosize = (iosize + blocksize - 1) & ~((u64)blocksize - 1); | ||
| 2251 | sector = (em->block_start + extent_offset) >> 9; | ||
| 2252 | bdev = em->bdev; | ||
| 2253 | block_start = em->block_start; | ||
| 2254 | compressed = test_bit(EXTENT_FLAG_COMPRESSED, &em->flags); | ||
| 2255 | free_extent_map(em); | ||
| 2256 | em = NULL; | ||
| 2257 | |||
| 2258 | /* | ||
| 2259 | * compressed and inline extents are written through other | ||
| 2260 | * paths in the FS | ||
| 2261 | */ | ||
| 2262 | if (compressed || block_start == EXTENT_MAP_HOLE || | ||
| 2263 | block_start == EXTENT_MAP_INLINE) { | ||
| 2264 | clear_extent_dirty(tree, cur, | ||
| 2265 | cur + iosize - 1, GFP_NOFS); | ||
| 2266 | |||
| 2267 | unlock_extent(tree, unlock_start, cur + iosize - 1, | ||
| 2268 | GFP_NOFS); | ||
| 2269 | |||
| 2270 | /* | ||
| 2271 | * end_io notification does not happen here for | ||
| 2272 | * compressed extents | ||
| 2273 | */ | ||
| 2274 | if (!compressed && tree->ops && | ||
| 2275 | tree->ops->writepage_end_io_hook) | ||
| 2276 | tree->ops->writepage_end_io_hook(page, cur, | ||
| 2277 | cur + iosize - 1, | ||
| 2278 | NULL, 1); | ||
| 2279 | else if (compressed) { | ||
| 2280 | /* we don't want to end_page_writeback on | ||
| 2281 | * a compressed extent. this happens | ||
| 2282 | * elsewhere | ||
| 2283 | */ | ||
| 2284 | nr++; | ||
| 2285 | } | ||
| 2286 | |||
| 2287 | cur += iosize; | ||
| 2288 | pg_offset += iosize; | ||
| 2289 | unlock_start = cur; | ||
| 2290 | continue; | ||
| 2291 | } | ||
| 2292 | /* leave this out until we have a page_mkwrite call */ | ||
| 2293 | if (0 && !test_range_bit(tree, cur, cur + iosize - 1, | ||
| 2294 | EXTENT_DIRTY, 0)) { | ||
| 2295 | cur = cur + iosize; | ||
| 2296 | pg_offset += iosize; | ||
| 2297 | continue; | ||
| 2298 | } | ||
| 2299 | |||
| 2300 | clear_extent_dirty(tree, cur, cur + iosize - 1, GFP_NOFS); | ||
| 2301 | if (tree->ops && tree->ops->writepage_io_hook) { | ||
| 2302 | ret = tree->ops->writepage_io_hook(page, cur, | ||
| 2303 | cur + iosize - 1); | ||
| 2304 | } else { | ||
| 2305 | ret = 0; | ||
| 2306 | } | ||
| 2307 | if (ret) { | ||
| 2308 | SetPageError(page); | ||
| 2309 | } else { | ||
| 2310 | unsigned long max_nr = end_index + 1; | ||
| 2311 | |||
| 2312 | set_range_writeback(tree, cur, cur + iosize - 1); | ||
| 2313 | if (!PageWriteback(page)) { | ||
| 2314 | printk(KERN_ERR "btrfs warning page %lu not " | ||
| 2315 | "writeback, cur %llu end %llu\n", | ||
| 2316 | page->index, (unsigned long long)cur, | ||
| 2317 | (unsigned long long)end); | ||
| 2318 | } | ||
| 2319 | |||
| 2320 | ret = submit_extent_page(WRITE, tree, page, sector, | ||
| 2321 | iosize, pg_offset, bdev, | ||
| 2322 | &epd->bio, max_nr, | ||
| 2323 | end_bio_extent_writepage, | ||
| 2324 | 0, 0, 0); | ||
| 2325 | if (ret) | ||
| 2326 | SetPageError(page); | ||
| 2327 | } | ||
| 2328 | cur = cur + iosize; | ||
| 2329 | pg_offset += iosize; | ||
| 2330 | nr++; | ||
| 2331 | } | ||
| 2332 | done: | ||
| 2333 | if (nr == 0) { | ||
| 2334 | /* make sure the mapping tag for page dirty gets cleared */ | ||
| 2335 | set_page_writeback(page); | ||
| 2336 | end_page_writeback(page); | ||
| 2337 | } | ||
| 2338 | if (unlock_start <= page_end) | ||
| 2339 | unlock_extent(tree, unlock_start, page_end, GFP_NOFS); | ||
| 2340 | unlock_page(page); | ||
| 2341 | |||
| 2342 | update_nr_written: | ||
| 2343 | wbc->nr_to_write -= nr_written; | ||
| 2344 | if (wbc->range_cyclic || (wbc->nr_to_write > 0 && | ||
| 2345 | wbc->range_start == 0 && wbc->range_end == LLONG_MAX)) | ||
| 2346 | page->mapping->writeback_index = page->index + nr_written; | ||
| 2347 | return 0; | ||
| 2348 | } | ||
| 2349 | |||
| 2350 | /** | ||
| 2351 | * write_cache_pages - walk the list of dirty pages of the given address space and write all of them. | ||
| 2352 | * @mapping: address space structure to write | ||
| 2353 | * @wbc: subtract the number of written pages from *@wbc->nr_to_write | ||
| 2354 | * @writepage: function called for each page | ||
| 2355 | * @data: data passed to writepage function | ||
| 2356 | * | ||
| 2357 | * If a page is already under I/O, write_cache_pages() skips it, even | ||
| 2358 | * if it's dirty. This is desirable behaviour for memory-cleaning writeback, | ||
| 2359 | * but it is INCORRECT for data-integrity system calls such as fsync(). fsync() | ||
| 2360 | * and msync() need to guarantee that all the data which was dirty at the time | ||
| 2361 | * the call was made get new I/O started against them. If wbc->sync_mode is | ||
| 2362 | * WB_SYNC_ALL then we were called for data integrity and we must wait for | ||
| 2363 | * existing IO to complete. | ||
| 2364 | */ | ||
| 2365 | static int extent_write_cache_pages(struct extent_io_tree *tree, | ||
| 2366 | struct address_space *mapping, | ||
| 2367 | struct writeback_control *wbc, | ||
| 2368 | writepage_t writepage, void *data, | ||
| 2369 | void (*flush_fn)(void *)) | ||
| 2370 | { | ||
| 2371 | struct backing_dev_info *bdi = mapping->backing_dev_info; | ||
| 2372 | int ret = 0; | ||
| 2373 | int done = 0; | ||
| 2374 | struct pagevec pvec; | ||
| 2375 | int nr_pages; | ||
| 2376 | pgoff_t index; | ||
| 2377 | pgoff_t end; /* Inclusive */ | ||
| 2378 | int scanned = 0; | ||
| 2379 | int range_whole = 0; | ||
| 2380 | |||
| 2381 | if (wbc->nonblocking && bdi_write_congested(bdi)) { | ||
| 2382 | wbc->encountered_congestion = 1; | ||
| 2383 | return 0; | ||
| 2384 | } | ||
| 2385 | |||
| 2386 | pagevec_init(&pvec, 0); | ||
| 2387 | if (wbc->range_cyclic) { | ||
| 2388 | index = mapping->writeback_index; /* Start from prev offset */ | ||
| 2389 | end = -1; | ||
| 2390 | } else { | ||
| 2391 | index = wbc->range_start >> PAGE_CACHE_SHIFT; | ||
| 2392 | end = wbc->range_end >> PAGE_CACHE_SHIFT; | ||
| 2393 | if (wbc->range_start == 0 && wbc->range_end == LLONG_MAX) | ||
| 2394 | range_whole = 1; | ||
| 2395 | scanned = 1; | ||
| 2396 | } | ||
| 2397 | retry: | ||
| 2398 | while (!done && (index <= end) && | ||
| 2399 | (nr_pages = pagevec_lookup_tag(&pvec, mapping, &index, | ||
| 2400 | PAGECACHE_TAG_DIRTY, min(end - index, | ||
| 2401 | (pgoff_t)PAGEVEC_SIZE-1) + 1))) { | ||
| 2402 | unsigned i; | ||
| 2403 | |||
| 2404 | scanned = 1; | ||
| 2405 | for (i = 0; i < nr_pages; i++) { | ||
| 2406 | struct page *page = pvec.pages[i]; | ||
| 2407 | |||
| 2408 | /* | ||
| 2409 | * At this point we hold neither mapping->tree_lock nor | ||
| 2410 | * lock on the page itself: the page may be truncated or | ||
| 2411 | * invalidated (changing page->mapping to NULL), or even | ||
| 2412 | * swizzled back from swapper_space to tmpfs file | ||
| 2413 | * mapping | ||
| 2414 | */ | ||
| 2415 | if (tree->ops && tree->ops->write_cache_pages_lock_hook) | ||
| 2416 | tree->ops->write_cache_pages_lock_hook(page); | ||
| 2417 | else | ||
| 2418 | lock_page(page); | ||
| 2419 | |||
| 2420 | if (unlikely(page->mapping != mapping)) { | ||
| 2421 | unlock_page(page); | ||
| 2422 | continue; | ||
| 2423 | } | ||
| 2424 | |||
| 2425 | if (!wbc->range_cyclic && page->index > end) { | ||
| 2426 | done = 1; | ||
| 2427 | unlock_page(page); | ||
| 2428 | continue; | ||
| 2429 | } | ||
| 2430 | |||
| 2431 | if (wbc->sync_mode != WB_SYNC_NONE) { | ||
| 2432 | if (PageWriteback(page)) | ||
| 2433 | flush_fn(data); | ||
| 2434 | wait_on_page_writeback(page); | ||
| 2435 | } | ||
| 2436 | |||
| 2437 | if (PageWriteback(page) || | ||
| 2438 | !clear_page_dirty_for_io(page)) { | ||
| 2439 | unlock_page(page); | ||
| 2440 | continue; | ||
| 2441 | } | ||
| 2442 | |||
| 2443 | ret = (*writepage)(page, wbc, data); | ||
| 2444 | |||
| 2445 | if (unlikely(ret == AOP_WRITEPAGE_ACTIVATE)) { | ||
| 2446 | unlock_page(page); | ||
| 2447 | ret = 0; | ||
| 2448 | } | ||
| 2449 | if (ret || wbc->nr_to_write <= 0) | ||
| 2450 | done = 1; | ||
| 2451 | if (wbc->nonblocking && bdi_write_congested(bdi)) { | ||
| 2452 | wbc->encountered_congestion = 1; | ||
| 2453 | done = 1; | ||
| 2454 | } | ||
| 2455 | } | ||
| 2456 | pagevec_release(&pvec); | ||
| 2457 | cond_resched(); | ||
| 2458 | } | ||
| 2459 | if (!scanned && !done) { | ||
| 2460 | /* | ||
| 2461 | * We hit the last page and there is more work to be done: wrap | ||
| 2462 | * back to the start of the file | ||
| 2463 | */ | ||
| 2464 | scanned = 1; | ||
| 2465 | index = 0; | ||
| 2466 | goto retry; | ||
| 2467 | } | ||
| 2468 | return ret; | ||
| 2469 | } | ||
| 2470 | |||
| 2471 | static noinline void flush_write_bio(void *data) | ||
| 2472 | { | ||
| 2473 | struct extent_page_data *epd = data; | ||
| 2474 | if (epd->bio) { | ||
| 2475 | submit_one_bio(WRITE, epd->bio, 0, 0); | ||
| 2476 | epd->bio = NULL; | ||
| 2477 | } | ||
| 2478 | } | ||
| 2479 | |||
| 2480 | int extent_write_full_page(struct extent_io_tree *tree, struct page *page, | ||
| 2481 | get_extent_t *get_extent, | ||
| 2482 | struct writeback_control *wbc) | ||
| 2483 | { | ||
| 2484 | int ret; | ||
| 2485 | struct address_space *mapping = page->mapping; | ||
| 2486 | struct extent_page_data epd = { | ||
| 2487 | .bio = NULL, | ||
| 2488 | .tree = tree, | ||
| 2489 | .get_extent = get_extent, | ||
| 2490 | .extent_locked = 0, | ||
| 2491 | }; | ||
| 2492 | struct writeback_control wbc_writepages = { | ||
| 2493 | .bdi = wbc->bdi, | ||
| 2494 | .sync_mode = WB_SYNC_NONE, | ||
| 2495 | .older_than_this = NULL, | ||
| 2496 | .nr_to_write = 64, | ||
| 2497 | .range_start = page_offset(page) + PAGE_CACHE_SIZE, | ||
| 2498 | .range_end = (loff_t)-1, | ||
| 2499 | }; | ||
| 2500 | |||
| 2501 | |||
| 2502 | ret = __extent_writepage(page, wbc, &epd); | ||
| 2503 | |||
| 2504 | extent_write_cache_pages(tree, mapping, &wbc_writepages, | ||
| 2505 | __extent_writepage, &epd, flush_write_bio); | ||
| 2506 | if (epd.bio) | ||
| 2507 | submit_one_bio(WRITE, epd.bio, 0, 0); | ||
| 2508 | return ret; | ||
| 2509 | } | ||
| 2510 | |||
| 2511 | int extent_write_locked_range(struct extent_io_tree *tree, struct inode *inode, | ||
| 2512 | u64 start, u64 end, get_extent_t *get_extent, | ||
| 2513 | int mode) | ||
| 2514 | { | ||
| 2515 | int ret = 0; | ||
| 2516 | struct address_space *mapping = inode->i_mapping; | ||
| 2517 | struct page *page; | ||
| 2518 | unsigned long nr_pages = (end - start + PAGE_CACHE_SIZE) >> | ||
| 2519 | PAGE_CACHE_SHIFT; | ||
| 2520 | |||
| 2521 | struct extent_page_data epd = { | ||
| 2522 | .bio = NULL, | ||
| 2523 | .tree = tree, | ||
| 2524 | .get_extent = get_extent, | ||
| 2525 | .extent_locked = 1, | ||
| 2526 | }; | ||
| 2527 | struct writeback_control wbc_writepages = { | ||
| 2528 | .bdi = inode->i_mapping->backing_dev_info, | ||
| 2529 | .sync_mode = mode, | ||
| 2530 | .older_than_this = NULL, | ||
| 2531 | .nr_to_write = nr_pages * 2, | ||
| 2532 | .range_start = start, | ||
| 2533 | .range_end = end + 1, | ||
| 2534 | }; | ||
| 2535 | |||
| 2536 | while (start <= end) { | ||
| 2537 | page = find_get_page(mapping, start >> PAGE_CACHE_SHIFT); | ||
| 2538 | if (clear_page_dirty_for_io(page)) | ||
| 2539 | ret = __extent_writepage(page, &wbc_writepages, &epd); | ||
| 2540 | else { | ||
| 2541 | if (tree->ops && tree->ops->writepage_end_io_hook) | ||
| 2542 | tree->ops->writepage_end_io_hook(page, start, | ||
| 2543 | start + PAGE_CACHE_SIZE - 1, | ||
| 2544 | NULL, 1); | ||
| 2545 | unlock_page(page); | ||
| 2546 | } | ||
| 2547 | page_cache_release(page); | ||
| 2548 | start += PAGE_CACHE_SIZE; | ||
| 2549 | } | ||
| 2550 | |||
| 2551 | if (epd.bio) | ||
| 2552 | submit_one_bio(WRITE, epd.bio, 0, 0); | ||
| 2553 | return ret; | ||
| 2554 | } | ||
| 2555 | |||
| 2556 | int extent_writepages(struct extent_io_tree *tree, | ||
| 2557 | struct address_space *mapping, | ||
| 2558 | get_extent_t *get_extent, | ||
| 2559 | struct writeback_control *wbc) | ||
| 2560 | { | ||
| 2561 | int ret = 0; | ||
| 2562 | struct extent_page_data epd = { | ||
| 2563 | .bio = NULL, | ||
| 2564 | .tree = tree, | ||
| 2565 | .get_extent = get_extent, | ||
| 2566 | .extent_locked = 0, | ||
| 2567 | }; | ||
| 2568 | |||
| 2569 | ret = extent_write_cache_pages(tree, mapping, wbc, | ||
| 2570 | __extent_writepage, &epd, | ||
| 2571 | flush_write_bio); | ||
| 2572 | if (epd.bio) | ||
| 2573 | submit_one_bio(WRITE, epd.bio, 0, 0); | ||
| 2574 | return ret; | ||
| 2575 | } | ||
| 2576 | |||
| 2577 | int extent_readpages(struct extent_io_tree *tree, | ||
| 2578 | struct address_space *mapping, | ||
| 2579 | struct list_head *pages, unsigned nr_pages, | ||
| 2580 | get_extent_t get_extent) | ||
| 2581 | { | ||
| 2582 | struct bio *bio = NULL; | ||
| 2583 | unsigned page_idx; | ||
| 2584 | struct pagevec pvec; | ||
| 2585 | unsigned long bio_flags = 0; | ||
| 2586 | |||
| 2587 | pagevec_init(&pvec, 0); | ||
| 2588 | for (page_idx = 0; page_idx < nr_pages; page_idx++) { | ||
| 2589 | struct page *page = list_entry(pages->prev, struct page, lru); | ||
| 2590 | |||
| 2591 | prefetchw(&page->flags); | ||
| 2592 | list_del(&page->lru); | ||
| 2593 | /* | ||
| 2594 | * what we want to do here is call add_to_page_cache_lru, | ||
| 2595 | * but that isn't exported, so we reproduce it here | ||
| 2596 | */ | ||
| 2597 | if (!add_to_page_cache(page, mapping, | ||
| 2598 | page->index, GFP_KERNEL)) { | ||
| 2599 | |||
| 2600 | /* open coding of lru_cache_add, also not exported */ | ||
| 2601 | page_cache_get(page); | ||
| 2602 | if (!pagevec_add(&pvec, page)) | ||
| 2603 | __pagevec_lru_add_file(&pvec); | ||
| 2604 | __extent_read_full_page(tree, page, get_extent, | ||
| 2605 | &bio, 0, &bio_flags); | ||
| 2606 | } | ||
| 2607 | page_cache_release(page); | ||
| 2608 | } | ||
| 2609 | if (pagevec_count(&pvec)) | ||
| 2610 | __pagevec_lru_add_file(&pvec); | ||
| 2611 | BUG_ON(!list_empty(pages)); | ||
| 2612 | if (bio) | ||
| 2613 | submit_one_bio(READ, bio, 0, bio_flags); | ||
| 2614 | return 0; | ||
| 2615 | } | ||
| 2616 | |||
| 2617 | /* | ||
| 2618 | * basic invalidatepage code, this waits on any locked or writeback | ||
| 2619 | * ranges corresponding to the page, and then deletes any extent state | ||
| 2620 | * records from the tree | ||
| 2621 | */ | ||
| 2622 | int extent_invalidatepage(struct extent_io_tree *tree, | ||
| 2623 | struct page *page, unsigned long offset) | ||
| 2624 | { | ||
| 2625 | u64 start = ((u64)page->index << PAGE_CACHE_SHIFT); | ||
| 2626 | u64 end = start + PAGE_CACHE_SIZE - 1; | ||
| 2627 | size_t blocksize = page->mapping->host->i_sb->s_blocksize; | ||
| 2628 | |||
| 2629 | start += (offset + blocksize - 1) & ~(blocksize - 1); | ||
| 2630 | if (start > end) | ||
| 2631 | return 0; | ||
| 2632 | |||
| 2633 | lock_extent(tree, start, end, GFP_NOFS); | ||
| 2634 | wait_on_extent_writeback(tree, start, end); | ||
| 2635 | clear_extent_bit(tree, start, end, | ||
| 2636 | EXTENT_LOCKED | EXTENT_DIRTY | EXTENT_DELALLOC, | ||
| 2637 | 1, 1, GFP_NOFS); | ||
| 2638 | return 0; | ||
| 2639 | } | ||
| 2640 | |||
| 2641 | /* | ||
| 2642 | * simple commit_write call, set_range_dirty is used to mark both | ||
| 2643 | * the pages and the extent records as dirty | ||
| 2644 | */ | ||
| 2645 | int extent_commit_write(struct extent_io_tree *tree, | ||
| 2646 | struct inode *inode, struct page *page, | ||
| 2647 | unsigned from, unsigned to) | ||
| 2648 | { | ||
| 2649 | loff_t pos = ((loff_t)page->index << PAGE_CACHE_SHIFT) + to; | ||
| 2650 | |||
| 2651 | set_page_extent_mapped(page); | ||
| 2652 | set_page_dirty(page); | ||
| 2653 | |||
| 2654 | if (pos > inode->i_size) { | ||
| 2655 | i_size_write(inode, pos); | ||
| 2656 | mark_inode_dirty(inode); | ||
| 2657 | } | ||
| 2658 | return 0; | ||
| 2659 | } | ||
| 2660 | |||
| 2661 | int extent_prepare_write(struct extent_io_tree *tree, | ||
| 2662 | struct inode *inode, struct page *page, | ||
| 2663 | unsigned from, unsigned to, get_extent_t *get_extent) | ||
| 2664 | { | ||
| 2665 | u64 page_start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 2666 | u64 page_end = page_start + PAGE_CACHE_SIZE - 1; | ||
| 2667 | u64 block_start; | ||
| 2668 | u64 orig_block_start; | ||
| 2669 | u64 block_end; | ||
| 2670 | u64 cur_end; | ||
| 2671 | struct extent_map *em; | ||
| 2672 | unsigned blocksize = 1 << inode->i_blkbits; | ||
| 2673 | size_t page_offset = 0; | ||
| 2674 | size_t block_off_start; | ||
| 2675 | size_t block_off_end; | ||
| 2676 | int err = 0; | ||
| 2677 | int iocount = 0; | ||
| 2678 | int ret = 0; | ||
| 2679 | int isnew; | ||
| 2680 | |||
| 2681 | set_page_extent_mapped(page); | ||
| 2682 | |||
| 2683 | block_start = (page_start + from) & ~((u64)blocksize - 1); | ||
| 2684 | block_end = (page_start + to - 1) | (blocksize - 1); | ||
| 2685 | orig_block_start = block_start; | ||
| 2686 | |||
| 2687 | lock_extent(tree, page_start, page_end, GFP_NOFS); | ||
| 2688 | while (block_start <= block_end) { | ||
| 2689 | em = get_extent(inode, page, page_offset, block_start, | ||
| 2690 | block_end - block_start + 1, 1); | ||
| 2691 | if (IS_ERR(em) || !em) | ||
| 2692 | goto err; | ||
| 2693 | |||
| 2694 | cur_end = min(block_end, extent_map_end(em) - 1); | ||
| 2695 | block_off_start = block_start & (PAGE_CACHE_SIZE - 1); | ||
| 2696 | block_off_end = block_off_start + blocksize; | ||
| 2697 | isnew = clear_extent_new(tree, block_start, cur_end, GFP_NOFS); | ||
| 2698 | |||
| 2699 | if (!PageUptodate(page) && isnew && | ||
| 2700 | (block_off_end > to || block_off_start < from)) { | ||
| 2701 | void *kaddr; | ||
| 2702 | |||
| 2703 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 2704 | if (block_off_end > to) | ||
| 2705 | memset(kaddr + to, 0, block_off_end - to); | ||
| 2706 | if (block_off_start < from) | ||
| 2707 | memset(kaddr + block_off_start, 0, | ||
| 2708 | from - block_off_start); | ||
| 2709 | flush_dcache_page(page); | ||
| 2710 | kunmap_atomic(kaddr, KM_USER0); | ||
| 2711 | } | ||
| 2712 | if ((em->block_start != EXTENT_MAP_HOLE && | ||
| 2713 | em->block_start != EXTENT_MAP_INLINE) && | ||
| 2714 | !isnew && !PageUptodate(page) && | ||
| 2715 | (block_off_end > to || block_off_start < from) && | ||
| 2716 | !test_range_bit(tree, block_start, cur_end, | ||
| 2717 | EXTENT_UPTODATE, 1)) { | ||
| 2718 | u64 sector; | ||
| 2719 | u64 extent_offset = block_start - em->start; | ||
| 2720 | size_t iosize; | ||
| 2721 | sector = (em->block_start + extent_offset) >> 9; | ||
| 2722 | iosize = (cur_end - block_start + blocksize) & | ||
| 2723 | ~((u64)blocksize - 1); | ||
| 2724 | /* | ||
| 2725 | * we've already got the extent locked, but we | ||
| 2726 | * need to split the state such that our end_bio | ||
| 2727 | * handler can clear the lock. | ||
| 2728 | */ | ||
| 2729 | set_extent_bit(tree, block_start, | ||
| 2730 | block_start + iosize - 1, | ||
| 2731 | EXTENT_LOCKED, 0, NULL, GFP_NOFS); | ||
| 2732 | ret = submit_extent_page(READ, tree, page, | ||
| 2733 | sector, iosize, page_offset, em->bdev, | ||
| 2734 | NULL, 1, | ||
| 2735 | end_bio_extent_preparewrite, 0, | ||
| 2736 | 0, 0); | ||
| 2737 | iocount++; | ||
| 2738 | block_start = block_start + iosize; | ||
| 2739 | } else { | ||
| 2740 | set_extent_uptodate(tree, block_start, cur_end, | ||
| 2741 | GFP_NOFS); | ||
| 2742 | unlock_extent(tree, block_start, cur_end, GFP_NOFS); | ||
| 2743 | block_start = cur_end + 1; | ||
| 2744 | } | ||
| 2745 | page_offset = block_start & (PAGE_CACHE_SIZE - 1); | ||
| 2746 | free_extent_map(em); | ||
| 2747 | } | ||
| 2748 | if (iocount) { | ||
| 2749 | wait_extent_bit(tree, orig_block_start, | ||
| 2750 | block_end, EXTENT_LOCKED); | ||
| 2751 | } | ||
| 2752 | check_page_uptodate(tree, page); | ||
| 2753 | err: | ||
| 2754 | /* FIXME, zero out newly allocated blocks on error */ | ||
| 2755 | return err; | ||
| 2756 | } | ||
| 2757 | |||
| 2758 | /* | ||
| 2759 | * a helper for releasepage, this tests for areas of the page that | ||
| 2760 | * are locked or under IO and drops the related state bits if it is safe | ||
| 2761 | * to drop the page. | ||
| 2762 | */ | ||
| 2763 | int try_release_extent_state(struct extent_map_tree *map, | ||
| 2764 | struct extent_io_tree *tree, struct page *page, | ||
| 2765 | gfp_t mask) | ||
| 2766 | { | ||
| 2767 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 2768 | u64 end = start + PAGE_CACHE_SIZE - 1; | ||
| 2769 | int ret = 1; | ||
| 2770 | |||
| 2771 | if (test_range_bit(tree, start, end, | ||
| 2772 | EXTENT_IOBITS | EXTENT_ORDERED, 0)) | ||
| 2773 | ret = 0; | ||
| 2774 | else { | ||
| 2775 | if ((mask & GFP_NOFS) == GFP_NOFS) | ||
| 2776 | mask = GFP_NOFS; | ||
| 2777 | clear_extent_bit(tree, start, end, EXTENT_UPTODATE, | ||
| 2778 | 1, 1, mask); | ||
| 2779 | } | ||
| 2780 | return ret; | ||
| 2781 | } | ||
| 2782 | |||
| 2783 | /* | ||
| 2784 | * a helper for releasepage. As long as there are no locked extents | ||
| 2785 | * in the range corresponding to the page, both state records and extent | ||
| 2786 | * map records are removed | ||
| 2787 | */ | ||
| 2788 | int try_release_extent_mapping(struct extent_map_tree *map, | ||
| 2789 | struct extent_io_tree *tree, struct page *page, | ||
| 2790 | gfp_t mask) | ||
| 2791 | { | ||
| 2792 | struct extent_map *em; | ||
| 2793 | u64 start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 2794 | u64 end = start + PAGE_CACHE_SIZE - 1; | ||
| 2795 | |||
| 2796 | if ((mask & __GFP_WAIT) && | ||
| 2797 | page->mapping->host->i_size > 16 * 1024 * 1024) { | ||
| 2798 | u64 len; | ||
| 2799 | while (start <= end) { | ||
| 2800 | len = end - start + 1; | ||
| 2801 | spin_lock(&map->lock); | ||
| 2802 | em = lookup_extent_mapping(map, start, len); | ||
| 2803 | if (!em || IS_ERR(em)) { | ||
| 2804 | spin_unlock(&map->lock); | ||
| 2805 | break; | ||
| 2806 | } | ||
| 2807 | if (test_bit(EXTENT_FLAG_PINNED, &em->flags) || | ||
| 2808 | em->start != start) { | ||
| 2809 | spin_unlock(&map->lock); | ||
| 2810 | free_extent_map(em); | ||
| 2811 | break; | ||
| 2812 | } | ||
| 2813 | if (!test_range_bit(tree, em->start, | ||
| 2814 | extent_map_end(em) - 1, | ||
| 2815 | EXTENT_LOCKED | EXTENT_WRITEBACK | | ||
| 2816 | EXTENT_ORDERED, | ||
| 2817 | 0)) { | ||
| 2818 | remove_extent_mapping(map, em); | ||
| 2819 | /* once for the rb tree */ | ||
| 2820 | free_extent_map(em); | ||
| 2821 | } | ||
| 2822 | start = extent_map_end(em); | ||
| 2823 | spin_unlock(&map->lock); | ||
| 2824 | |||
| 2825 | /* once for us */ | ||
| 2826 | free_extent_map(em); | ||
| 2827 | } | ||
| 2828 | } | ||
| 2829 | return try_release_extent_state(map, tree, page, mask); | ||
| 2830 | } | ||
| 2831 | |||
| 2832 | sector_t extent_bmap(struct address_space *mapping, sector_t iblock, | ||
| 2833 | get_extent_t *get_extent) | ||
| 2834 | { | ||
| 2835 | struct inode *inode = mapping->host; | ||
| 2836 | u64 start = iblock << inode->i_blkbits; | ||
| 2837 | sector_t sector = 0; | ||
| 2838 | size_t blksize = (1 << inode->i_blkbits); | ||
| 2839 | struct extent_map *em; | ||
| 2840 | |||
| 2841 | lock_extent(&BTRFS_I(inode)->io_tree, start, start + blksize - 1, | ||
| 2842 | GFP_NOFS); | ||
| 2843 | em = get_extent(inode, NULL, 0, start, blksize, 0); | ||
| 2844 | unlock_extent(&BTRFS_I(inode)->io_tree, start, start + blksize - 1, | ||
| 2845 | GFP_NOFS); | ||
| 2846 | if (!em || IS_ERR(em)) | ||
| 2847 | return 0; | ||
| 2848 | |||
| 2849 | if (em->block_start > EXTENT_MAP_LAST_BYTE) | ||
| 2850 | goto out; | ||
| 2851 | |||
| 2852 | sector = (em->block_start + start - em->start) >> inode->i_blkbits; | ||
| 2853 | out: | ||
| 2854 | free_extent_map(em); | ||
| 2855 | return sector; | ||
| 2856 | } | ||
| 2857 | |||
| 2858 | static inline struct page *extent_buffer_page(struct extent_buffer *eb, | ||
| 2859 | unsigned long i) | ||
| 2860 | { | ||
| 2861 | struct page *p; | ||
| 2862 | struct address_space *mapping; | ||
| 2863 | |||
| 2864 | if (i == 0) | ||
| 2865 | return eb->first_page; | ||
| 2866 | i += eb->start >> PAGE_CACHE_SHIFT; | ||
| 2867 | mapping = eb->first_page->mapping; | ||
| 2868 | if (!mapping) | ||
| 2869 | return NULL; | ||
| 2870 | |||
| 2871 | /* | ||
| 2872 | * extent_buffer_page is only called after pinning the page | ||
| 2873 | * by increasing the reference count. So we know the page must | ||
| 2874 | * be in the radix tree. | ||
| 2875 | */ | ||
| 2876 | rcu_read_lock(); | ||
| 2877 | p = radix_tree_lookup(&mapping->page_tree, i); | ||
| 2878 | rcu_read_unlock(); | ||
| 2879 | |||
| 2880 | return p; | ||
| 2881 | } | ||
| 2882 | |||
| 2883 | static inline unsigned long num_extent_pages(u64 start, u64 len) | ||
| 2884 | { | ||
| 2885 | return ((start + len + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT) - | ||
| 2886 | (start >> PAGE_CACHE_SHIFT); | ||
| 2887 | } | ||
| 2888 | |||
| 2889 | static struct extent_buffer *__alloc_extent_buffer(struct extent_io_tree *tree, | ||
| 2890 | u64 start, | ||
| 2891 | unsigned long len, | ||
| 2892 | gfp_t mask) | ||
| 2893 | { | ||
| 2894 | struct extent_buffer *eb = NULL; | ||
| 2895 | #ifdef LEAK_DEBUG | ||
| 2896 | unsigned long flags; | ||
| 2897 | #endif | ||
| 2898 | |||
| 2899 | eb = kmem_cache_zalloc(extent_buffer_cache, mask); | ||
| 2900 | eb->start = start; | ||
| 2901 | eb->len = len; | ||
| 2902 | mutex_init(&eb->mutex); | ||
| 2903 | #ifdef LEAK_DEBUG | ||
| 2904 | spin_lock_irqsave(&leak_lock, flags); | ||
| 2905 | list_add(&eb->leak_list, &buffers); | ||
| 2906 | spin_unlock_irqrestore(&leak_lock, flags); | ||
| 2907 | #endif | ||
| 2908 | atomic_set(&eb->refs, 1); | ||
| 2909 | |||
| 2910 | return eb; | ||
| 2911 | } | ||
| 2912 | |||
| 2913 | static void __free_extent_buffer(struct extent_buffer *eb) | ||
| 2914 | { | ||
| 2915 | #ifdef LEAK_DEBUG | ||
| 2916 | unsigned long flags; | ||
| 2917 | spin_lock_irqsave(&leak_lock, flags); | ||
| 2918 | list_del(&eb->leak_list); | ||
| 2919 | spin_unlock_irqrestore(&leak_lock, flags); | ||
| 2920 | #endif | ||
| 2921 | kmem_cache_free(extent_buffer_cache, eb); | ||
| 2922 | } | ||
| 2923 | |||
| 2924 | struct extent_buffer *alloc_extent_buffer(struct extent_io_tree *tree, | ||
| 2925 | u64 start, unsigned long len, | ||
| 2926 | struct page *page0, | ||
| 2927 | gfp_t mask) | ||
| 2928 | { | ||
| 2929 | unsigned long num_pages = num_extent_pages(start, len); | ||
| 2930 | unsigned long i; | ||
| 2931 | unsigned long index = start >> PAGE_CACHE_SHIFT; | ||
| 2932 | struct extent_buffer *eb; | ||
| 2933 | struct extent_buffer *exists = NULL; | ||
| 2934 | struct page *p; | ||
| 2935 | struct address_space *mapping = tree->mapping; | ||
| 2936 | int uptodate = 1; | ||
| 2937 | |||
| 2938 | spin_lock(&tree->buffer_lock); | ||
| 2939 | eb = buffer_search(tree, start); | ||
| 2940 | if (eb) { | ||
| 2941 | atomic_inc(&eb->refs); | ||
| 2942 | spin_unlock(&tree->buffer_lock); | ||
| 2943 | mark_page_accessed(eb->first_page); | ||
| 2944 | return eb; | ||
| 2945 | } | ||
| 2946 | spin_unlock(&tree->buffer_lock); | ||
| 2947 | |||
| 2948 | eb = __alloc_extent_buffer(tree, start, len, mask); | ||
| 2949 | if (!eb) | ||
| 2950 | return NULL; | ||
| 2951 | |||
| 2952 | if (page0) { | ||
| 2953 | eb->first_page = page0; | ||
| 2954 | i = 1; | ||
| 2955 | index++; | ||
| 2956 | page_cache_get(page0); | ||
| 2957 | mark_page_accessed(page0); | ||
| 2958 | set_page_extent_mapped(page0); | ||
| 2959 | set_page_extent_head(page0, len); | ||
| 2960 | uptodate = PageUptodate(page0); | ||
| 2961 | } else { | ||
| 2962 | i = 0; | ||
| 2963 | } | ||
| 2964 | for (; i < num_pages; i++, index++) { | ||
| 2965 | p = find_or_create_page(mapping, index, mask | __GFP_HIGHMEM); | ||
| 2966 | if (!p) { | ||
| 2967 | WARN_ON(1); | ||
| 2968 | goto free_eb; | ||
| 2969 | } | ||
| 2970 | set_page_extent_mapped(p); | ||
| 2971 | mark_page_accessed(p); | ||
| 2972 | if (i == 0) { | ||
| 2973 | eb->first_page = p; | ||
| 2974 | set_page_extent_head(p, len); | ||
| 2975 | } else { | ||
| 2976 | set_page_private(p, EXTENT_PAGE_PRIVATE); | ||
| 2977 | } | ||
| 2978 | if (!PageUptodate(p)) | ||
| 2979 | uptodate = 0; | ||
| 2980 | unlock_page(p); | ||
| 2981 | } | ||
| 2982 | if (uptodate) | ||
| 2983 | eb->flags |= EXTENT_UPTODATE; | ||
| 2984 | eb->flags |= EXTENT_BUFFER_FILLED; | ||
| 2985 | |||
| 2986 | spin_lock(&tree->buffer_lock); | ||
| 2987 | exists = buffer_tree_insert(tree, start, &eb->rb_node); | ||
| 2988 | if (exists) { | ||
| 2989 | /* add one reference for the caller */ | ||
| 2990 | atomic_inc(&exists->refs); | ||
| 2991 | spin_unlock(&tree->buffer_lock); | ||
| 2992 | goto free_eb; | ||
| 2993 | } | ||
| 2994 | spin_unlock(&tree->buffer_lock); | ||
| 2995 | |||
| 2996 | /* add one reference for the tree */ | ||
| 2997 | atomic_inc(&eb->refs); | ||
| 2998 | return eb; | ||
| 2999 | |||
| 3000 | free_eb: | ||
| 3001 | if (!atomic_dec_and_test(&eb->refs)) | ||
| 3002 | return exists; | ||
| 3003 | for (index = 1; index < i; index++) | ||
| 3004 | page_cache_release(extent_buffer_page(eb, index)); | ||
| 3005 | page_cache_release(extent_buffer_page(eb, 0)); | ||
| 3006 | __free_extent_buffer(eb); | ||
| 3007 | return exists; | ||
| 3008 | } | ||
| 3009 | |||
| 3010 | struct extent_buffer *find_extent_buffer(struct extent_io_tree *tree, | ||
| 3011 | u64 start, unsigned long len, | ||
| 3012 | gfp_t mask) | ||
| 3013 | { | ||
| 3014 | struct extent_buffer *eb; | ||
| 3015 | |||
| 3016 | spin_lock(&tree->buffer_lock); | ||
| 3017 | eb = buffer_search(tree, start); | ||
| 3018 | if (eb) | ||
| 3019 | atomic_inc(&eb->refs); | ||
| 3020 | spin_unlock(&tree->buffer_lock); | ||
| 3021 | |||
| 3022 | if (eb) | ||
| 3023 | mark_page_accessed(eb->first_page); | ||
| 3024 | |||
| 3025 | return eb; | ||
| 3026 | } | ||
| 3027 | |||
| 3028 | void free_extent_buffer(struct extent_buffer *eb) | ||
| 3029 | { | ||
| 3030 | if (!eb) | ||
| 3031 | return; | ||
| 3032 | |||
| 3033 | if (!atomic_dec_and_test(&eb->refs)) | ||
| 3034 | return; | ||
| 3035 | |||
| 3036 | WARN_ON(1); | ||
| 3037 | } | ||
| 3038 | |||
| 3039 | int clear_extent_buffer_dirty(struct extent_io_tree *tree, | ||
| 3040 | struct extent_buffer *eb) | ||
| 3041 | { | ||
| 3042 | int set; | ||
| 3043 | unsigned long i; | ||
| 3044 | unsigned long num_pages; | ||
| 3045 | struct page *page; | ||
| 3046 | |||
| 3047 | u64 start = eb->start; | ||
| 3048 | u64 end = start + eb->len - 1; | ||
| 3049 | |||
| 3050 | set = clear_extent_dirty(tree, start, end, GFP_NOFS); | ||
| 3051 | num_pages = num_extent_pages(eb->start, eb->len); | ||
| 3052 | |||
| 3053 | for (i = 0; i < num_pages; i++) { | ||
| 3054 | page = extent_buffer_page(eb, i); | ||
| 3055 | if (!set && !PageDirty(page)) | ||
| 3056 | continue; | ||
| 3057 | |||
| 3058 | lock_page(page); | ||
| 3059 | if (i == 0) | ||
| 3060 | set_page_extent_head(page, eb->len); | ||
| 3061 | else | ||
| 3062 | set_page_private(page, EXTENT_PAGE_PRIVATE); | ||
| 3063 | |||
| 3064 | /* | ||
| 3065 | * if we're on the last page or the first page and the | ||
| 3066 | * block isn't aligned on a page boundary, do extra checks | ||
| 3067 | * to make sure we don't clean page that is partially dirty | ||
| 3068 | */ | ||
| 3069 | if ((i == 0 && (eb->start & (PAGE_CACHE_SIZE - 1))) || | ||
| 3070 | ((i == num_pages - 1) && | ||
| 3071 | ((eb->start + eb->len) & (PAGE_CACHE_SIZE - 1)))) { | ||
| 3072 | start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 3073 | end = start + PAGE_CACHE_SIZE - 1; | ||
| 3074 | if (test_range_bit(tree, start, end, | ||
| 3075 | EXTENT_DIRTY, 0)) { | ||
| 3076 | unlock_page(page); | ||
| 3077 | continue; | ||
| 3078 | } | ||
| 3079 | } | ||
| 3080 | clear_page_dirty_for_io(page); | ||
| 3081 | spin_lock_irq(&page->mapping->tree_lock); | ||
| 3082 | if (!PageDirty(page)) { | ||
| 3083 | radix_tree_tag_clear(&page->mapping->page_tree, | ||
| 3084 | page_index(page), | ||
| 3085 | PAGECACHE_TAG_DIRTY); | ||
| 3086 | } | ||
| 3087 | spin_unlock_irq(&page->mapping->tree_lock); | ||
| 3088 | unlock_page(page); | ||
| 3089 | } | ||
| 3090 | return 0; | ||
| 3091 | } | ||
| 3092 | |||
| 3093 | int wait_on_extent_buffer_writeback(struct extent_io_tree *tree, | ||
| 3094 | struct extent_buffer *eb) | ||
| 3095 | { | ||
| 3096 | return wait_on_extent_writeback(tree, eb->start, | ||
| 3097 | eb->start + eb->len - 1); | ||
| 3098 | } | ||
| 3099 | |||
| 3100 | int set_extent_buffer_dirty(struct extent_io_tree *tree, | ||
| 3101 | struct extent_buffer *eb) | ||
| 3102 | { | ||
| 3103 | unsigned long i; | ||
| 3104 | unsigned long num_pages; | ||
| 3105 | |||
| 3106 | num_pages = num_extent_pages(eb->start, eb->len); | ||
| 3107 | for (i = 0; i < num_pages; i++) { | ||
| 3108 | struct page *page = extent_buffer_page(eb, i); | ||
| 3109 | /* writepage may need to do something special for the | ||
| 3110 | * first page, we have to make sure page->private is | ||
| 3111 | * properly set. releasepage may drop page->private | ||
| 3112 | * on us if the page isn't already dirty. | ||
| 3113 | */ | ||
| 3114 | lock_page(page); | ||
| 3115 | if (i == 0) { | ||
| 3116 | set_page_extent_head(page, eb->len); | ||
| 3117 | } else if (PagePrivate(page) && | ||
| 3118 | page->private != EXTENT_PAGE_PRIVATE) { | ||
| 3119 | set_page_extent_mapped(page); | ||
| 3120 | } | ||
| 3121 | __set_page_dirty_nobuffers(extent_buffer_page(eb, i)); | ||
| 3122 | set_extent_dirty(tree, page_offset(page), | ||
| 3123 | page_offset(page) + PAGE_CACHE_SIZE - 1, | ||
| 3124 | GFP_NOFS); | ||
| 3125 | unlock_page(page); | ||
| 3126 | } | ||
| 3127 | return 0; | ||
| 3128 | } | ||
| 3129 | |||
| 3130 | int clear_extent_buffer_uptodate(struct extent_io_tree *tree, | ||
| 3131 | struct extent_buffer *eb) | ||
| 3132 | { | ||
| 3133 | unsigned long i; | ||
| 3134 | struct page *page; | ||
| 3135 | unsigned long num_pages; | ||
| 3136 | |||
| 3137 | num_pages = num_extent_pages(eb->start, eb->len); | ||
| 3138 | eb->flags &= ~EXTENT_UPTODATE; | ||
| 3139 | |||
| 3140 | clear_extent_uptodate(tree, eb->start, eb->start + eb->len - 1, | ||
| 3141 | GFP_NOFS); | ||
| 3142 | for (i = 0; i < num_pages; i++) { | ||
| 3143 | page = extent_buffer_page(eb, i); | ||
| 3144 | if (page) | ||
| 3145 | ClearPageUptodate(page); | ||
| 3146 | } | ||
| 3147 | return 0; | ||
| 3148 | } | ||
| 3149 | |||
| 3150 | int set_extent_buffer_uptodate(struct extent_io_tree *tree, | ||
| 3151 | struct extent_buffer *eb) | ||
| 3152 | { | ||
| 3153 | unsigned long i; | ||
| 3154 | struct page *page; | ||
| 3155 | unsigned long num_pages; | ||
| 3156 | |||
| 3157 | num_pages = num_extent_pages(eb->start, eb->len); | ||
| 3158 | |||
| 3159 | set_extent_uptodate(tree, eb->start, eb->start + eb->len - 1, | ||
| 3160 | GFP_NOFS); | ||
| 3161 | for (i = 0; i < num_pages; i++) { | ||
| 3162 | page = extent_buffer_page(eb, i); | ||
| 3163 | if ((i == 0 && (eb->start & (PAGE_CACHE_SIZE - 1))) || | ||
| 3164 | ((i == num_pages - 1) && | ||
| 3165 | ((eb->start + eb->len) & (PAGE_CACHE_SIZE - 1)))) { | ||
| 3166 | check_page_uptodate(tree, page); | ||
| 3167 | continue; | ||
| 3168 | } | ||
| 3169 | SetPageUptodate(page); | ||
| 3170 | } | ||
| 3171 | return 0; | ||
| 3172 | } | ||
| 3173 | |||
| 3174 | int extent_range_uptodate(struct extent_io_tree *tree, | ||
| 3175 | u64 start, u64 end) | ||
| 3176 | { | ||
| 3177 | struct page *page; | ||
| 3178 | int ret; | ||
| 3179 | int pg_uptodate = 1; | ||
| 3180 | int uptodate; | ||
| 3181 | unsigned long index; | ||
| 3182 | |||
| 3183 | ret = test_range_bit(tree, start, end, EXTENT_UPTODATE, 1); | ||
| 3184 | if (ret) | ||
| 3185 | return 1; | ||
| 3186 | while (start <= end) { | ||
| 3187 | index = start >> PAGE_CACHE_SHIFT; | ||
| 3188 | page = find_get_page(tree->mapping, index); | ||
| 3189 | uptodate = PageUptodate(page); | ||
| 3190 | page_cache_release(page); | ||
| 3191 | if (!uptodate) { | ||
| 3192 | pg_uptodate = 0; | ||
| 3193 | break; | ||
| 3194 | } | ||
| 3195 | start += PAGE_CACHE_SIZE; | ||
| 3196 | } | ||
| 3197 | return pg_uptodate; | ||
| 3198 | } | ||
| 3199 | |||
| 3200 | int extent_buffer_uptodate(struct extent_io_tree *tree, | ||
| 3201 | struct extent_buffer *eb) | ||
| 3202 | { | ||
| 3203 | int ret = 0; | ||
| 3204 | unsigned long num_pages; | ||
| 3205 | unsigned long i; | ||
| 3206 | struct page *page; | ||
| 3207 | int pg_uptodate = 1; | ||
| 3208 | |||
| 3209 | if (eb->flags & EXTENT_UPTODATE) | ||
| 3210 | return 1; | ||
| 3211 | |||
| 3212 | ret = test_range_bit(tree, eb->start, eb->start + eb->len - 1, | ||
| 3213 | EXTENT_UPTODATE, 1); | ||
| 3214 | if (ret) | ||
| 3215 | return ret; | ||
| 3216 | |||
| 3217 | num_pages = num_extent_pages(eb->start, eb->len); | ||
| 3218 | for (i = 0; i < num_pages; i++) { | ||
| 3219 | page = extent_buffer_page(eb, i); | ||
| 3220 | if (!PageUptodate(page)) { | ||
| 3221 | pg_uptodate = 0; | ||
| 3222 | break; | ||
| 3223 | } | ||
| 3224 | } | ||
| 3225 | return pg_uptodate; | ||
| 3226 | } | ||
| 3227 | |||
| 3228 | int read_extent_buffer_pages(struct extent_io_tree *tree, | ||
| 3229 | struct extent_buffer *eb, | ||
| 3230 | u64 start, int wait, | ||
| 3231 | get_extent_t *get_extent, int mirror_num) | ||
| 3232 | { | ||
| 3233 | unsigned long i; | ||
| 3234 | unsigned long start_i; | ||
| 3235 | struct page *page; | ||
| 3236 | int err; | ||
| 3237 | int ret = 0; | ||
| 3238 | int locked_pages = 0; | ||
| 3239 | int all_uptodate = 1; | ||
| 3240 | int inc_all_pages = 0; | ||
| 3241 | unsigned long num_pages; | ||
| 3242 | struct bio *bio = NULL; | ||
| 3243 | unsigned long bio_flags = 0; | ||
| 3244 | |||
| 3245 | if (eb->flags & EXTENT_UPTODATE) | ||
| 3246 | return 0; | ||
| 3247 | |||
| 3248 | if (test_range_bit(tree, eb->start, eb->start + eb->len - 1, | ||
| 3249 | EXTENT_UPTODATE, 1)) { | ||
| 3250 | return 0; | ||
| 3251 | } | ||
| 3252 | |||
| 3253 | if (start) { | ||
| 3254 | WARN_ON(start < eb->start); | ||
| 3255 | start_i = (start >> PAGE_CACHE_SHIFT) - | ||
| 3256 | (eb->start >> PAGE_CACHE_SHIFT); | ||
| 3257 | } else { | ||
| 3258 | start_i = 0; | ||
| 3259 | } | ||
| 3260 | |||
| 3261 | num_pages = num_extent_pages(eb->start, eb->len); | ||
| 3262 | for (i = start_i; i < num_pages; i++) { | ||
| 3263 | page = extent_buffer_page(eb, i); | ||
| 3264 | if (!wait) { | ||
| 3265 | if (!trylock_page(page)) | ||
| 3266 | goto unlock_exit; | ||
| 3267 | } else { | ||
| 3268 | lock_page(page); | ||
| 3269 | } | ||
| 3270 | locked_pages++; | ||
| 3271 | if (!PageUptodate(page)) | ||
| 3272 | all_uptodate = 0; | ||
| 3273 | } | ||
| 3274 | if (all_uptodate) { | ||
| 3275 | if (start_i == 0) | ||
| 3276 | eb->flags |= EXTENT_UPTODATE; | ||
| 3277 | goto unlock_exit; | ||
| 3278 | } | ||
| 3279 | |||
| 3280 | for (i = start_i; i < num_pages; i++) { | ||
| 3281 | page = extent_buffer_page(eb, i); | ||
| 3282 | if (inc_all_pages) | ||
| 3283 | page_cache_get(page); | ||
| 3284 | if (!PageUptodate(page)) { | ||
| 3285 | if (start_i == 0) | ||
| 3286 | inc_all_pages = 1; | ||
| 3287 | ClearPageError(page); | ||
| 3288 | err = __extent_read_full_page(tree, page, | ||
| 3289 | get_extent, &bio, | ||
| 3290 | mirror_num, &bio_flags); | ||
| 3291 | if (err) | ||
| 3292 | ret = err; | ||
| 3293 | } else { | ||
| 3294 | unlock_page(page); | ||
| 3295 | } | ||
| 3296 | } | ||
| 3297 | |||
| 3298 | if (bio) | ||
| 3299 | submit_one_bio(READ, bio, mirror_num, bio_flags); | ||
| 3300 | |||
| 3301 | if (ret || !wait) | ||
| 3302 | return ret; | ||
| 3303 | |||
| 3304 | for (i = start_i; i < num_pages; i++) { | ||
| 3305 | page = extent_buffer_page(eb, i); | ||
| 3306 | wait_on_page_locked(page); | ||
| 3307 | if (!PageUptodate(page)) | ||
| 3308 | ret = -EIO; | ||
| 3309 | } | ||
| 3310 | |||
| 3311 | if (!ret) | ||
| 3312 | eb->flags |= EXTENT_UPTODATE; | ||
| 3313 | return ret; | ||
| 3314 | |||
| 3315 | unlock_exit: | ||
| 3316 | i = start_i; | ||
| 3317 | while (locked_pages > 0) { | ||
| 3318 | page = extent_buffer_page(eb, i); | ||
| 3319 | i++; | ||
| 3320 | unlock_page(page); | ||
| 3321 | locked_pages--; | ||
| 3322 | } | ||
| 3323 | return ret; | ||
| 3324 | } | ||
| 3325 | |||
| 3326 | void read_extent_buffer(struct extent_buffer *eb, void *dstv, | ||
| 3327 | unsigned long start, | ||
| 3328 | unsigned long len) | ||
| 3329 | { | ||
| 3330 | size_t cur; | ||
| 3331 | size_t offset; | ||
| 3332 | struct page *page; | ||
| 3333 | char *kaddr; | ||
| 3334 | char *dst = (char *)dstv; | ||
| 3335 | size_t start_offset = eb->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3336 | unsigned long i = (start_offset + start) >> PAGE_CACHE_SHIFT; | ||
| 3337 | |||
| 3338 | WARN_ON(start > eb->len); | ||
| 3339 | WARN_ON(start + len > eb->start + eb->len); | ||
| 3340 | |||
| 3341 | offset = (start_offset + start) & ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3342 | |||
| 3343 | while (len > 0) { | ||
| 3344 | page = extent_buffer_page(eb, i); | ||
| 3345 | |||
| 3346 | cur = min(len, (PAGE_CACHE_SIZE - offset)); | ||
| 3347 | kaddr = kmap_atomic(page, KM_USER1); | ||
| 3348 | memcpy(dst, kaddr + offset, cur); | ||
| 3349 | kunmap_atomic(kaddr, KM_USER1); | ||
| 3350 | |||
| 3351 | dst += cur; | ||
| 3352 | len -= cur; | ||
| 3353 | offset = 0; | ||
| 3354 | i++; | ||
| 3355 | } | ||
| 3356 | } | ||
| 3357 | |||
| 3358 | int map_private_extent_buffer(struct extent_buffer *eb, unsigned long start, | ||
| 3359 | unsigned long min_len, char **token, char **map, | ||
| 3360 | unsigned long *map_start, | ||
| 3361 | unsigned long *map_len, int km) | ||
| 3362 | { | ||
| 3363 | size_t offset = start & (PAGE_CACHE_SIZE - 1); | ||
| 3364 | char *kaddr; | ||
| 3365 | struct page *p; | ||
| 3366 | size_t start_offset = eb->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3367 | unsigned long i = (start_offset + start) >> PAGE_CACHE_SHIFT; | ||
| 3368 | unsigned long end_i = (start_offset + start + min_len - 1) >> | ||
| 3369 | PAGE_CACHE_SHIFT; | ||
| 3370 | |||
| 3371 | if (i != end_i) | ||
| 3372 | return -EINVAL; | ||
| 3373 | |||
| 3374 | if (i == 0) { | ||
| 3375 | offset = start_offset; | ||
| 3376 | *map_start = 0; | ||
| 3377 | } else { | ||
| 3378 | offset = 0; | ||
| 3379 | *map_start = ((u64)i << PAGE_CACHE_SHIFT) - start_offset; | ||
| 3380 | } | ||
| 3381 | |||
| 3382 | if (start + min_len > eb->len) { | ||
| 3383 | printk(KERN_ERR "btrfs bad mapping eb start %llu len %lu, " | ||
| 3384 | "wanted %lu %lu\n", (unsigned long long)eb->start, | ||
| 3385 | eb->len, start, min_len); | ||
| 3386 | WARN_ON(1); | ||
| 3387 | } | ||
| 3388 | |||
| 3389 | p = extent_buffer_page(eb, i); | ||
| 3390 | kaddr = kmap_atomic(p, km); | ||
| 3391 | *token = kaddr; | ||
| 3392 | *map = kaddr + offset; | ||
| 3393 | *map_len = PAGE_CACHE_SIZE - offset; | ||
| 3394 | return 0; | ||
| 3395 | } | ||
| 3396 | |||
| 3397 | int map_extent_buffer(struct extent_buffer *eb, unsigned long start, | ||
| 3398 | unsigned long min_len, | ||
| 3399 | char **token, char **map, | ||
| 3400 | unsigned long *map_start, | ||
| 3401 | unsigned long *map_len, int km) | ||
| 3402 | { | ||
| 3403 | int err; | ||
| 3404 | int save = 0; | ||
| 3405 | if (eb->map_token) { | ||
| 3406 | unmap_extent_buffer(eb, eb->map_token, km); | ||
| 3407 | eb->map_token = NULL; | ||
| 3408 | save = 1; | ||
| 3409 | WARN_ON(!mutex_is_locked(&eb->mutex)); | ||
| 3410 | } | ||
| 3411 | err = map_private_extent_buffer(eb, start, min_len, token, map, | ||
| 3412 | map_start, map_len, km); | ||
| 3413 | if (!err && save) { | ||
| 3414 | eb->map_token = *token; | ||
| 3415 | eb->kaddr = *map; | ||
| 3416 | eb->map_start = *map_start; | ||
| 3417 | eb->map_len = *map_len; | ||
| 3418 | } | ||
| 3419 | return err; | ||
| 3420 | } | ||
| 3421 | |||
| 3422 | void unmap_extent_buffer(struct extent_buffer *eb, char *token, int km) | ||
| 3423 | { | ||
| 3424 | kunmap_atomic(token, km); | ||
| 3425 | } | ||
| 3426 | |||
| 3427 | int memcmp_extent_buffer(struct extent_buffer *eb, const void *ptrv, | ||
| 3428 | unsigned long start, | ||
| 3429 | unsigned long len) | ||
| 3430 | { | ||
| 3431 | size_t cur; | ||
| 3432 | size_t offset; | ||
| 3433 | struct page *page; | ||
| 3434 | char *kaddr; | ||
| 3435 | char *ptr = (char *)ptrv; | ||
| 3436 | size_t start_offset = eb->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3437 | unsigned long i = (start_offset + start) >> PAGE_CACHE_SHIFT; | ||
| 3438 | int ret = 0; | ||
| 3439 | |||
| 3440 | WARN_ON(start > eb->len); | ||
| 3441 | WARN_ON(start + len > eb->start + eb->len); | ||
| 3442 | |||
| 3443 | offset = (start_offset + start) & ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3444 | |||
| 3445 | while (len > 0) { | ||
| 3446 | page = extent_buffer_page(eb, i); | ||
| 3447 | |||
| 3448 | cur = min(len, (PAGE_CACHE_SIZE - offset)); | ||
| 3449 | |||
| 3450 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 3451 | ret = memcmp(ptr, kaddr + offset, cur); | ||
| 3452 | kunmap_atomic(kaddr, KM_USER0); | ||
| 3453 | if (ret) | ||
| 3454 | break; | ||
| 3455 | |||
| 3456 | ptr += cur; | ||
| 3457 | len -= cur; | ||
| 3458 | offset = 0; | ||
| 3459 | i++; | ||
| 3460 | } | ||
| 3461 | return ret; | ||
| 3462 | } | ||
| 3463 | |||
| 3464 | void write_extent_buffer(struct extent_buffer *eb, const void *srcv, | ||
| 3465 | unsigned long start, unsigned long len) | ||
| 3466 | { | ||
| 3467 | size_t cur; | ||
| 3468 | size_t offset; | ||
| 3469 | struct page *page; | ||
| 3470 | char *kaddr; | ||
| 3471 | char *src = (char *)srcv; | ||
| 3472 | size_t start_offset = eb->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3473 | unsigned long i = (start_offset + start) >> PAGE_CACHE_SHIFT; | ||
| 3474 | |||
| 3475 | WARN_ON(start > eb->len); | ||
| 3476 | WARN_ON(start + len > eb->start + eb->len); | ||
| 3477 | |||
| 3478 | offset = (start_offset + start) & ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3479 | |||
| 3480 | while (len > 0) { | ||
| 3481 | page = extent_buffer_page(eb, i); | ||
| 3482 | WARN_ON(!PageUptodate(page)); | ||
| 3483 | |||
| 3484 | cur = min(len, PAGE_CACHE_SIZE - offset); | ||
| 3485 | kaddr = kmap_atomic(page, KM_USER1); | ||
| 3486 | memcpy(kaddr + offset, src, cur); | ||
| 3487 | kunmap_atomic(kaddr, KM_USER1); | ||
| 3488 | |||
| 3489 | src += cur; | ||
| 3490 | len -= cur; | ||
| 3491 | offset = 0; | ||
| 3492 | i++; | ||
| 3493 | } | ||
| 3494 | } | ||
| 3495 | |||
| 3496 | void memset_extent_buffer(struct extent_buffer *eb, char c, | ||
| 3497 | unsigned long start, unsigned long len) | ||
| 3498 | { | ||
| 3499 | size_t cur; | ||
| 3500 | size_t offset; | ||
| 3501 | struct page *page; | ||
| 3502 | char *kaddr; | ||
| 3503 | size_t start_offset = eb->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3504 | unsigned long i = (start_offset + start) >> PAGE_CACHE_SHIFT; | ||
| 3505 | |||
| 3506 | WARN_ON(start > eb->len); | ||
| 3507 | WARN_ON(start + len > eb->start + eb->len); | ||
| 3508 | |||
| 3509 | offset = (start_offset + start) & ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3510 | |||
| 3511 | while (len > 0) { | ||
| 3512 | page = extent_buffer_page(eb, i); | ||
| 3513 | WARN_ON(!PageUptodate(page)); | ||
| 3514 | |||
| 3515 | cur = min(len, PAGE_CACHE_SIZE - offset); | ||
| 3516 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 3517 | memset(kaddr + offset, c, cur); | ||
| 3518 | kunmap_atomic(kaddr, KM_USER0); | ||
| 3519 | |||
| 3520 | len -= cur; | ||
| 3521 | offset = 0; | ||
| 3522 | i++; | ||
| 3523 | } | ||
| 3524 | } | ||
| 3525 | |||
| 3526 | void copy_extent_buffer(struct extent_buffer *dst, struct extent_buffer *src, | ||
| 3527 | unsigned long dst_offset, unsigned long src_offset, | ||
| 3528 | unsigned long len) | ||
| 3529 | { | ||
| 3530 | u64 dst_len = dst->len; | ||
| 3531 | size_t cur; | ||
| 3532 | size_t offset; | ||
| 3533 | struct page *page; | ||
| 3534 | char *kaddr; | ||
| 3535 | size_t start_offset = dst->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3536 | unsigned long i = (start_offset + dst_offset) >> PAGE_CACHE_SHIFT; | ||
| 3537 | |||
| 3538 | WARN_ON(src->len != dst_len); | ||
| 3539 | |||
| 3540 | offset = (start_offset + dst_offset) & | ||
| 3541 | ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3542 | |||
| 3543 | while (len > 0) { | ||
| 3544 | page = extent_buffer_page(dst, i); | ||
| 3545 | WARN_ON(!PageUptodate(page)); | ||
| 3546 | |||
| 3547 | cur = min(len, (unsigned long)(PAGE_CACHE_SIZE - offset)); | ||
| 3548 | |||
| 3549 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 3550 | read_extent_buffer(src, kaddr + offset, src_offset, cur); | ||
| 3551 | kunmap_atomic(kaddr, KM_USER0); | ||
| 3552 | |||
| 3553 | src_offset += cur; | ||
| 3554 | len -= cur; | ||
| 3555 | offset = 0; | ||
| 3556 | i++; | ||
| 3557 | } | ||
| 3558 | } | ||
| 3559 | |||
| 3560 | static void move_pages(struct page *dst_page, struct page *src_page, | ||
| 3561 | unsigned long dst_off, unsigned long src_off, | ||
| 3562 | unsigned long len) | ||
| 3563 | { | ||
| 3564 | char *dst_kaddr = kmap_atomic(dst_page, KM_USER0); | ||
| 3565 | if (dst_page == src_page) { | ||
| 3566 | memmove(dst_kaddr + dst_off, dst_kaddr + src_off, len); | ||
| 3567 | } else { | ||
| 3568 | char *src_kaddr = kmap_atomic(src_page, KM_USER1); | ||
| 3569 | char *p = dst_kaddr + dst_off + len; | ||
| 3570 | char *s = src_kaddr + src_off + len; | ||
| 3571 | |||
| 3572 | while (len--) | ||
| 3573 | *--p = *--s; | ||
| 3574 | |||
| 3575 | kunmap_atomic(src_kaddr, KM_USER1); | ||
| 3576 | } | ||
| 3577 | kunmap_atomic(dst_kaddr, KM_USER0); | ||
| 3578 | } | ||
| 3579 | |||
| 3580 | static void copy_pages(struct page *dst_page, struct page *src_page, | ||
| 3581 | unsigned long dst_off, unsigned long src_off, | ||
| 3582 | unsigned long len) | ||
| 3583 | { | ||
| 3584 | char *dst_kaddr = kmap_atomic(dst_page, KM_USER0); | ||
| 3585 | char *src_kaddr; | ||
| 3586 | |||
| 3587 | if (dst_page != src_page) | ||
| 3588 | src_kaddr = kmap_atomic(src_page, KM_USER1); | ||
| 3589 | else | ||
| 3590 | src_kaddr = dst_kaddr; | ||
| 3591 | |||
| 3592 | memcpy(dst_kaddr + dst_off, src_kaddr + src_off, len); | ||
| 3593 | kunmap_atomic(dst_kaddr, KM_USER0); | ||
| 3594 | if (dst_page != src_page) | ||
| 3595 | kunmap_atomic(src_kaddr, KM_USER1); | ||
| 3596 | } | ||
| 3597 | |||
| 3598 | void memcpy_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset, | ||
| 3599 | unsigned long src_offset, unsigned long len) | ||
| 3600 | { | ||
| 3601 | size_t cur; | ||
| 3602 | size_t dst_off_in_page; | ||
| 3603 | size_t src_off_in_page; | ||
| 3604 | size_t start_offset = dst->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3605 | unsigned long dst_i; | ||
| 3606 | unsigned long src_i; | ||
| 3607 | |||
| 3608 | if (src_offset + len > dst->len) { | ||
| 3609 | printk(KERN_ERR "btrfs memmove bogus src_offset %lu move " | ||
| 3610 | "len %lu dst len %lu\n", src_offset, len, dst->len); | ||
| 3611 | BUG_ON(1); | ||
| 3612 | } | ||
| 3613 | if (dst_offset + len > dst->len) { | ||
| 3614 | printk(KERN_ERR "btrfs memmove bogus dst_offset %lu move " | ||
| 3615 | "len %lu dst len %lu\n", dst_offset, len, dst->len); | ||
| 3616 | BUG_ON(1); | ||
| 3617 | } | ||
| 3618 | |||
| 3619 | while (len > 0) { | ||
| 3620 | dst_off_in_page = (start_offset + dst_offset) & | ||
| 3621 | ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3622 | src_off_in_page = (start_offset + src_offset) & | ||
| 3623 | ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3624 | |||
| 3625 | dst_i = (start_offset + dst_offset) >> PAGE_CACHE_SHIFT; | ||
| 3626 | src_i = (start_offset + src_offset) >> PAGE_CACHE_SHIFT; | ||
| 3627 | |||
| 3628 | cur = min(len, (unsigned long)(PAGE_CACHE_SIZE - | ||
| 3629 | src_off_in_page)); | ||
| 3630 | cur = min_t(unsigned long, cur, | ||
| 3631 | (unsigned long)(PAGE_CACHE_SIZE - dst_off_in_page)); | ||
| 3632 | |||
| 3633 | copy_pages(extent_buffer_page(dst, dst_i), | ||
| 3634 | extent_buffer_page(dst, src_i), | ||
| 3635 | dst_off_in_page, src_off_in_page, cur); | ||
| 3636 | |||
| 3637 | src_offset += cur; | ||
| 3638 | dst_offset += cur; | ||
| 3639 | len -= cur; | ||
| 3640 | } | ||
| 3641 | } | ||
| 3642 | |||
| 3643 | void memmove_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset, | ||
| 3644 | unsigned long src_offset, unsigned long len) | ||
| 3645 | { | ||
| 3646 | size_t cur; | ||
| 3647 | size_t dst_off_in_page; | ||
| 3648 | size_t src_off_in_page; | ||
| 3649 | unsigned long dst_end = dst_offset + len - 1; | ||
| 3650 | unsigned long src_end = src_offset + len - 1; | ||
| 3651 | size_t start_offset = dst->start & ((u64)PAGE_CACHE_SIZE - 1); | ||
| 3652 | unsigned long dst_i; | ||
| 3653 | unsigned long src_i; | ||
| 3654 | |||
| 3655 | if (src_offset + len > dst->len) { | ||
| 3656 | printk(KERN_ERR "btrfs memmove bogus src_offset %lu move " | ||
| 3657 | "len %lu len %lu\n", src_offset, len, dst->len); | ||
| 3658 | BUG_ON(1); | ||
| 3659 | } | ||
| 3660 | if (dst_offset + len > dst->len) { | ||
| 3661 | printk(KERN_ERR "btrfs memmove bogus dst_offset %lu move " | ||
| 3662 | "len %lu len %lu\n", dst_offset, len, dst->len); | ||
| 3663 | BUG_ON(1); | ||
| 3664 | } | ||
| 3665 | if (dst_offset < src_offset) { | ||
| 3666 | memcpy_extent_buffer(dst, dst_offset, src_offset, len); | ||
| 3667 | return; | ||
| 3668 | } | ||
| 3669 | while (len > 0) { | ||
| 3670 | dst_i = (start_offset + dst_end) >> PAGE_CACHE_SHIFT; | ||
| 3671 | src_i = (start_offset + src_end) >> PAGE_CACHE_SHIFT; | ||
| 3672 | |||
| 3673 | dst_off_in_page = (start_offset + dst_end) & | ||
| 3674 | ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3675 | src_off_in_page = (start_offset + src_end) & | ||
| 3676 | ((unsigned long)PAGE_CACHE_SIZE - 1); | ||
| 3677 | |||
| 3678 | cur = min_t(unsigned long, len, src_off_in_page + 1); | ||
| 3679 | cur = min(cur, dst_off_in_page + 1); | ||
| 3680 | move_pages(extent_buffer_page(dst, dst_i), | ||
| 3681 | extent_buffer_page(dst, src_i), | ||
| 3682 | dst_off_in_page - cur + 1, | ||
| 3683 | src_off_in_page - cur + 1, cur); | ||
| 3684 | |||
| 3685 | dst_end -= cur; | ||
| 3686 | src_end -= cur; | ||
| 3687 | len -= cur; | ||
| 3688 | } | ||
| 3689 | } | ||
| 3690 | |||
| 3691 | int try_release_extent_buffer(struct extent_io_tree *tree, struct page *page) | ||
| 3692 | { | ||
| 3693 | u64 start = page_offset(page); | ||
| 3694 | struct extent_buffer *eb; | ||
| 3695 | int ret = 1; | ||
| 3696 | unsigned long i; | ||
| 3697 | unsigned long num_pages; | ||
| 3698 | |||
| 3699 | spin_lock(&tree->buffer_lock); | ||
| 3700 | eb = buffer_search(tree, start); | ||
| 3701 | if (!eb) | ||
| 3702 | goto out; | ||
| 3703 | |||
| 3704 | if (atomic_read(&eb->refs) > 1) { | ||
| 3705 | ret = 0; | ||
| 3706 | goto out; | ||
| 3707 | } | ||
| 3708 | /* at this point we can safely release the extent buffer */ | ||
| 3709 | num_pages = num_extent_pages(eb->start, eb->len); | ||
| 3710 | for (i = 0; i < num_pages; i++) | ||
| 3711 | page_cache_release(extent_buffer_page(eb, i)); | ||
| 3712 | rb_erase(&eb->rb_node, &tree->buffer); | ||
| 3713 | __free_extent_buffer(eb); | ||
| 3714 | out: | ||
| 3715 | spin_unlock(&tree->buffer_lock); | ||
| 3716 | return ret; | ||
| 3717 | } | ||
diff --git a/fs/btrfs/extent_io.h b/fs/btrfs/extent_io.h new file mode 100644 index 000000000000..c5b483a79137 --- /dev/null +++ b/fs/btrfs/extent_io.h | |||
| @@ -0,0 +1,269 @@ | |||
| 1 | #ifndef __EXTENTIO__ | ||
| 2 | #define __EXTENTIO__ | ||
| 3 | |||
| 4 | #include <linux/rbtree.h> | ||
| 5 | |||
| 6 | /* bits for the extent state */ | ||
| 7 | #define EXTENT_DIRTY 1 | ||
| 8 | #define EXTENT_WRITEBACK (1 << 1) | ||
| 9 | #define EXTENT_UPTODATE (1 << 2) | ||
| 10 | #define EXTENT_LOCKED (1 << 3) | ||
| 11 | #define EXTENT_NEW (1 << 4) | ||
| 12 | #define EXTENT_DELALLOC (1 << 5) | ||
| 13 | #define EXTENT_DEFRAG (1 << 6) | ||
| 14 | #define EXTENT_DEFRAG_DONE (1 << 7) | ||
| 15 | #define EXTENT_BUFFER_FILLED (1 << 8) | ||
| 16 | #define EXTENT_ORDERED (1 << 9) | ||
| 17 | #define EXTENT_ORDERED_METADATA (1 << 10) | ||
| 18 | #define EXTENT_BOUNDARY (1 << 11) | ||
| 19 | #define EXTENT_NODATASUM (1 << 12) | ||
| 20 | #define EXTENT_IOBITS (EXTENT_LOCKED | EXTENT_WRITEBACK) | ||
| 21 | |||
| 22 | /* flags for bio submission */ | ||
| 23 | #define EXTENT_BIO_COMPRESSED 1 | ||
| 24 | |||
| 25 | /* | ||
| 26 | * page->private values. Every page that is controlled by the extent | ||
| 27 | * map has page->private set to one. | ||
| 28 | */ | ||
| 29 | #define EXTENT_PAGE_PRIVATE 1 | ||
| 30 | #define EXTENT_PAGE_PRIVATE_FIRST_PAGE 3 | ||
| 31 | |||
| 32 | struct extent_state; | ||
| 33 | |||
| 34 | typedef int (extent_submit_bio_hook_t)(struct inode *inode, int rw, | ||
| 35 | struct bio *bio, int mirror_num, | ||
| 36 | unsigned long bio_flags); | ||
| 37 | struct extent_io_ops { | ||
| 38 | int (*fill_delalloc)(struct inode *inode, struct page *locked_page, | ||
| 39 | u64 start, u64 end, int *page_started, | ||
| 40 | unsigned long *nr_written); | ||
| 41 | int (*writepage_start_hook)(struct page *page, u64 start, u64 end); | ||
| 42 | int (*writepage_io_hook)(struct page *page, u64 start, u64 end); | ||
| 43 | extent_submit_bio_hook_t *submit_bio_hook; | ||
| 44 | int (*merge_bio_hook)(struct page *page, unsigned long offset, | ||
| 45 | size_t size, struct bio *bio, | ||
| 46 | unsigned long bio_flags); | ||
| 47 | int (*readpage_io_hook)(struct page *page, u64 start, u64 end); | ||
| 48 | int (*readpage_io_failed_hook)(struct bio *bio, struct page *page, | ||
| 49 | u64 start, u64 end, | ||
| 50 | struct extent_state *state); | ||
| 51 | int (*writepage_io_failed_hook)(struct bio *bio, struct page *page, | ||
| 52 | u64 start, u64 end, | ||
| 53 | struct extent_state *state); | ||
| 54 | int (*readpage_end_io_hook)(struct page *page, u64 start, u64 end, | ||
| 55 | struct extent_state *state); | ||
| 56 | int (*writepage_end_io_hook)(struct page *page, u64 start, u64 end, | ||
| 57 | struct extent_state *state, int uptodate); | ||
| 58 | int (*set_bit_hook)(struct inode *inode, u64 start, u64 end, | ||
| 59 | unsigned long old, unsigned long bits); | ||
| 60 | int (*clear_bit_hook)(struct inode *inode, u64 start, u64 end, | ||
| 61 | unsigned long old, unsigned long bits); | ||
| 62 | int (*write_cache_pages_lock_hook)(struct page *page); | ||
| 63 | }; | ||
| 64 | |||
| 65 | struct extent_io_tree { | ||
| 66 | struct rb_root state; | ||
| 67 | struct rb_root buffer; | ||
| 68 | struct address_space *mapping; | ||
| 69 | u64 dirty_bytes; | ||
| 70 | spinlock_t lock; | ||
| 71 | spinlock_t buffer_lock; | ||
| 72 | struct extent_io_ops *ops; | ||
| 73 | }; | ||
| 74 | |||
| 75 | struct extent_state { | ||
| 76 | u64 start; | ||
| 77 | u64 end; /* inclusive */ | ||
| 78 | struct rb_node rb_node; | ||
| 79 | struct extent_io_tree *tree; | ||
| 80 | wait_queue_head_t wq; | ||
| 81 | atomic_t refs; | ||
| 82 | unsigned long state; | ||
| 83 | |||
| 84 | /* for use by the FS */ | ||
| 85 | u64 private; | ||
| 86 | |||
| 87 | struct list_head leak_list; | ||
| 88 | }; | ||
| 89 | |||
| 90 | struct extent_buffer { | ||
| 91 | u64 start; | ||
| 92 | unsigned long len; | ||
| 93 | char *map_token; | ||
| 94 | char *kaddr; | ||
| 95 | unsigned long map_start; | ||
| 96 | unsigned long map_len; | ||
| 97 | struct page *first_page; | ||
| 98 | atomic_t refs; | ||
| 99 | int flags; | ||
| 100 | struct list_head leak_list; | ||
| 101 | struct rb_node rb_node; | ||
| 102 | struct mutex mutex; | ||
| 103 | }; | ||
| 104 | |||
| 105 | struct extent_map_tree; | ||
| 106 | |||
| 107 | static inline struct extent_state *extent_state_next(struct extent_state *state) | ||
| 108 | { | ||
| 109 | struct rb_node *node; | ||
| 110 | node = rb_next(&state->rb_node); | ||
| 111 | if (!node) | ||
| 112 | return NULL; | ||
| 113 | return rb_entry(node, struct extent_state, rb_node); | ||
| 114 | } | ||
| 115 | |||
| 116 | typedef struct extent_map *(get_extent_t)(struct inode *inode, | ||
| 117 | struct page *page, | ||
| 118 | size_t page_offset, | ||
| 119 | u64 start, u64 len, | ||
| 120 | int create); | ||
| 121 | |||
| 122 | void extent_io_tree_init(struct extent_io_tree *tree, | ||
| 123 | struct address_space *mapping, gfp_t mask); | ||
| 124 | int try_release_extent_mapping(struct extent_map_tree *map, | ||
| 125 | struct extent_io_tree *tree, struct page *page, | ||
| 126 | gfp_t mask); | ||
| 127 | int try_release_extent_buffer(struct extent_io_tree *tree, struct page *page); | ||
| 128 | int try_release_extent_state(struct extent_map_tree *map, | ||
| 129 | struct extent_io_tree *tree, struct page *page, | ||
| 130 | gfp_t mask); | ||
| 131 | int lock_extent(struct extent_io_tree *tree, u64 start, u64 end, gfp_t mask); | ||
| 132 | int unlock_extent(struct extent_io_tree *tree, u64 start, u64 end, gfp_t mask); | ||
| 133 | int try_lock_extent(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 134 | gfp_t mask); | ||
| 135 | int extent_read_full_page(struct extent_io_tree *tree, struct page *page, | ||
| 136 | get_extent_t *get_extent); | ||
| 137 | int __init extent_io_init(void); | ||
| 138 | void extent_io_exit(void); | ||
| 139 | |||
| 140 | u64 count_range_bits(struct extent_io_tree *tree, | ||
| 141 | u64 *start, u64 search_end, | ||
| 142 | u64 max_bytes, unsigned long bits); | ||
| 143 | |||
| 144 | int test_range_bit(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 145 | int bits, int filled); | ||
| 146 | int clear_extent_bits(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 147 | int bits, gfp_t mask); | ||
| 148 | int clear_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 149 | int bits, int wake, int delete, gfp_t mask); | ||
| 150 | int set_extent_bits(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 151 | int bits, gfp_t mask); | ||
| 152 | int set_extent_uptodate(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 153 | gfp_t mask); | ||
| 154 | int set_extent_new(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 155 | gfp_t mask); | ||
| 156 | int set_extent_dirty(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 157 | gfp_t mask); | ||
| 158 | int clear_extent_dirty(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 159 | gfp_t mask); | ||
| 160 | int clear_extent_ordered(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 161 | gfp_t mask); | ||
| 162 | int clear_extent_ordered_metadata(struct extent_io_tree *tree, u64 start, | ||
| 163 | u64 end, gfp_t mask); | ||
| 164 | int set_extent_delalloc(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 165 | gfp_t mask); | ||
| 166 | int set_extent_ordered(struct extent_io_tree *tree, u64 start, u64 end, | ||
| 167 | gfp_t mask); | ||
| 168 | int find_first_extent_bit(struct extent_io_tree *tree, u64 start, | ||
| 169 | u64 *start_ret, u64 *end_ret, int bits); | ||
| 170 | struct extent_state *find_first_extent_bit_state(struct extent_io_tree *tree, | ||
| 171 | u64 start, int bits); | ||
| 172 | int extent_invalidatepage(struct extent_io_tree *tree, | ||
| 173 | struct page *page, unsigned long offset); | ||
| 174 | int extent_write_full_page(struct extent_io_tree *tree, struct page *page, | ||
| 175 | get_extent_t *get_extent, | ||
| 176 | struct writeback_control *wbc); | ||
| 177 | int extent_write_locked_range(struct extent_io_tree *tree, struct inode *inode, | ||
| 178 | u64 start, u64 end, get_extent_t *get_extent, | ||
| 179 | int mode); | ||
| 180 | int extent_writepages(struct extent_io_tree *tree, | ||
| 181 | struct address_space *mapping, | ||
| 182 | get_extent_t *get_extent, | ||
| 183 | struct writeback_control *wbc); | ||
| 184 | int extent_readpages(struct extent_io_tree *tree, | ||
| 185 | struct address_space *mapping, | ||
| 186 | struct list_head *pages, unsigned nr_pages, | ||
| 187 | get_extent_t get_extent); | ||
| 188 | int extent_prepare_write(struct extent_io_tree *tree, | ||
| 189 | struct inode *inode, struct page *page, | ||
| 190 | unsigned from, unsigned to, get_extent_t *get_extent); | ||
| 191 | int extent_commit_write(struct extent_io_tree *tree, | ||
| 192 | struct inode *inode, struct page *page, | ||
| 193 | unsigned from, unsigned to); | ||
| 194 | sector_t extent_bmap(struct address_space *mapping, sector_t iblock, | ||
| 195 | get_extent_t *get_extent); | ||
| 196 | int set_range_dirty(struct extent_io_tree *tree, u64 start, u64 end); | ||
| 197 | int set_state_private(struct extent_io_tree *tree, u64 start, u64 private); | ||
| 198 | int get_state_private(struct extent_io_tree *tree, u64 start, u64 *private); | ||
| 199 | void set_page_extent_mapped(struct page *page); | ||
| 200 | |||
| 201 | struct extent_buffer *alloc_extent_buffer(struct extent_io_tree *tree, | ||
| 202 | u64 start, unsigned long len, | ||
| 203 | struct page *page0, | ||
| 204 | gfp_t mask); | ||
| 205 | struct extent_buffer *find_extent_buffer(struct extent_io_tree *tree, | ||
| 206 | u64 start, unsigned long len, | ||
| 207 | gfp_t mask); | ||
| 208 | void free_extent_buffer(struct extent_buffer *eb); | ||
| 209 | int read_extent_buffer_pages(struct extent_io_tree *tree, | ||
| 210 | struct extent_buffer *eb, u64 start, int wait, | ||
| 211 | get_extent_t *get_extent, int mirror_num); | ||
| 212 | |||
| 213 | static inline void extent_buffer_get(struct extent_buffer *eb) | ||
| 214 | { | ||
| 215 | atomic_inc(&eb->refs); | ||
| 216 | } | ||
| 217 | |||
| 218 | int memcmp_extent_buffer(struct extent_buffer *eb, const void *ptrv, | ||
| 219 | unsigned long start, | ||
| 220 | unsigned long len); | ||
| 221 | void read_extent_buffer(struct extent_buffer *eb, void *dst, | ||
| 222 | unsigned long start, | ||
| 223 | unsigned long len); | ||
| 224 | void write_extent_buffer(struct extent_buffer *eb, const void *src, | ||
| 225 | unsigned long start, unsigned long len); | ||
| 226 | void copy_extent_buffer(struct extent_buffer *dst, struct extent_buffer *src, | ||
| 227 | unsigned long dst_offset, unsigned long src_offset, | ||
| 228 | unsigned long len); | ||
| 229 | void memcpy_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset, | ||
| 230 | unsigned long src_offset, unsigned long len); | ||
| 231 | void memmove_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset, | ||
| 232 | unsigned long src_offset, unsigned long len); | ||
| 233 | void memset_extent_buffer(struct extent_buffer *eb, char c, | ||
| 234 | unsigned long start, unsigned long len); | ||
| 235 | int wait_on_extent_buffer_writeback(struct extent_io_tree *tree, | ||
| 236 | struct extent_buffer *eb); | ||
| 237 | int wait_on_extent_writeback(struct extent_io_tree *tree, u64 start, u64 end); | ||
| 238 | int wait_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, int bits); | ||
| 239 | int clear_extent_buffer_dirty(struct extent_io_tree *tree, | ||
| 240 | struct extent_buffer *eb); | ||
| 241 | int set_extent_buffer_dirty(struct extent_io_tree *tree, | ||
| 242 | struct extent_buffer *eb); | ||
| 243 | int set_extent_buffer_uptodate(struct extent_io_tree *tree, | ||
| 244 | struct extent_buffer *eb); | ||
| 245 | int clear_extent_buffer_uptodate(struct extent_io_tree *tree, | ||
| 246 | struct extent_buffer *eb); | ||
| 247 | int extent_buffer_uptodate(struct extent_io_tree *tree, | ||
| 248 | struct extent_buffer *eb); | ||
| 249 | int map_extent_buffer(struct extent_buffer *eb, unsigned long offset, | ||
| 250 | unsigned long min_len, char **token, char **map, | ||
| 251 | unsigned long *map_start, | ||
| 252 | unsigned long *map_len, int km); | ||
| 253 | int map_private_extent_buffer(struct extent_buffer *eb, unsigned long offset, | ||
| 254 | unsigned long min_len, char **token, char **map, | ||
| 255 | unsigned long *map_start, | ||
| 256 | unsigned long *map_len, int km); | ||
| 257 | void unmap_extent_buffer(struct extent_buffer *eb, char *token, int km); | ||
| 258 | int release_extent_buffer_tail_pages(struct extent_buffer *eb); | ||
| 259 | int extent_range_uptodate(struct extent_io_tree *tree, | ||
| 260 | u64 start, u64 end); | ||
| 261 | int extent_clear_unlock_delalloc(struct inode *inode, | ||
| 262 | struct extent_io_tree *tree, | ||
| 263 | u64 start, u64 end, struct page *locked_page, | ||
| 264 | int unlock_page, | ||
| 265 | int clear_unlock, | ||
| 266 | int clear_delalloc, int clear_dirty, | ||
| 267 | int set_writeback, | ||
| 268 | int end_writeback); | ||
| 269 | #endif | ||
diff --git a/fs/btrfs/extent_map.c b/fs/btrfs/extent_map.c new file mode 100644 index 000000000000..4a83e33ada32 --- /dev/null +++ b/fs/btrfs/extent_map.c | |||
| @@ -0,0 +1,351 @@ | |||
| 1 | #include <linux/err.h> | ||
| 2 | #include <linux/gfp.h> | ||
| 3 | #include <linux/slab.h> | ||
| 4 | #include <linux/module.h> | ||
| 5 | #include <linux/spinlock.h> | ||
| 6 | #include <linux/version.h> | ||
| 7 | #include <linux/hardirq.h> | ||
| 8 | #include "extent_map.h" | ||
| 9 | |||
| 10 | /* temporary define until extent_map moves out of btrfs */ | ||
| 11 | struct kmem_cache *btrfs_cache_create(const char *name, size_t size, | ||
| 12 | unsigned long extra_flags, | ||
| 13 | void (*ctor)(void *, struct kmem_cache *, | ||
| 14 | unsigned long)); | ||
| 15 | |||
| 16 | static struct kmem_cache *extent_map_cache; | ||
| 17 | |||
| 18 | int __init extent_map_init(void) | ||
| 19 | { | ||
| 20 | extent_map_cache = btrfs_cache_create("extent_map", | ||
| 21 | sizeof(struct extent_map), 0, | ||
| 22 | NULL); | ||
| 23 | if (!extent_map_cache) | ||
| 24 | return -ENOMEM; | ||
| 25 | return 0; | ||
| 26 | } | ||
| 27 | |||
| 28 | void extent_map_exit(void) | ||
| 29 | { | ||
| 30 | if (extent_map_cache) | ||
| 31 | kmem_cache_destroy(extent_map_cache); | ||
| 32 | } | ||
| 33 | |||
| 34 | /** | ||
| 35 | * extent_map_tree_init - initialize extent map tree | ||
| 36 | * @tree: tree to initialize | ||
| 37 | * @mask: flags for memory allocations during tree operations | ||
| 38 | * | ||
| 39 | * Initialize the extent tree @tree. Should be called for each new inode | ||
| 40 | * or other user of the extent_map interface. | ||
| 41 | */ | ||
| 42 | void extent_map_tree_init(struct extent_map_tree *tree, gfp_t mask) | ||
| 43 | { | ||
| 44 | tree->map.rb_node = NULL; | ||
| 45 | spin_lock_init(&tree->lock); | ||
| 46 | } | ||
| 47 | EXPORT_SYMBOL(extent_map_tree_init); | ||
| 48 | |||
| 49 | /** | ||
| 50 | * alloc_extent_map - allocate new extent map structure | ||
| 51 | * @mask: memory allocation flags | ||
| 52 | * | ||
| 53 | * Allocate a new extent_map structure. The new structure is | ||
| 54 | * returned with a reference count of one and needs to be | ||
| 55 | * freed using free_extent_map() | ||
| 56 | */ | ||
| 57 | struct extent_map *alloc_extent_map(gfp_t mask) | ||
| 58 | { | ||
| 59 | struct extent_map *em; | ||
| 60 | em = kmem_cache_alloc(extent_map_cache, mask); | ||
| 61 | if (!em || IS_ERR(em)) | ||
| 62 | return em; | ||
| 63 | em->in_tree = 0; | ||
| 64 | em->flags = 0; | ||
| 65 | atomic_set(&em->refs, 1); | ||
| 66 | return em; | ||
| 67 | } | ||
| 68 | EXPORT_SYMBOL(alloc_extent_map); | ||
| 69 | |||
| 70 | /** | ||
| 71 | * free_extent_map - drop reference count of an extent_map | ||
| 72 | * @em: extent map beeing releasead | ||
| 73 | * | ||
| 74 | * Drops the reference out on @em by one and free the structure | ||
| 75 | * if the reference count hits zero. | ||
| 76 | */ | ||
| 77 | void free_extent_map(struct extent_map *em) | ||
| 78 | { | ||
| 79 | if (!em) | ||
| 80 | return; | ||
| 81 | WARN_ON(atomic_read(&em->refs) == 0); | ||
| 82 | if (atomic_dec_and_test(&em->refs)) { | ||
| 83 | WARN_ON(em->in_tree); | ||
| 84 | kmem_cache_free(extent_map_cache, em); | ||
| 85 | } | ||
| 86 | } | ||
| 87 | EXPORT_SYMBOL(free_extent_map); | ||
| 88 | |||
| 89 | static struct rb_node *tree_insert(struct rb_root *root, u64 offset, | ||
| 90 | struct rb_node *node) | ||
| 91 | { | ||
| 92 | struct rb_node **p = &root->rb_node; | ||
| 93 | struct rb_node *parent = NULL; | ||
| 94 | struct extent_map *entry; | ||
| 95 | |||
| 96 | while (*p) { | ||
| 97 | parent = *p; | ||
| 98 | entry = rb_entry(parent, struct extent_map, rb_node); | ||
| 99 | |||
| 100 | WARN_ON(!entry->in_tree); | ||
| 101 | |||
| 102 | if (offset < entry->start) | ||
| 103 | p = &(*p)->rb_left; | ||
| 104 | else if (offset >= extent_map_end(entry)) | ||
| 105 | p = &(*p)->rb_right; | ||
| 106 | else | ||
| 107 | return parent; | ||
| 108 | } | ||
| 109 | |||
| 110 | entry = rb_entry(node, struct extent_map, rb_node); | ||
| 111 | entry->in_tree = 1; | ||
| 112 | rb_link_node(node, parent, p); | ||
| 113 | rb_insert_color(node, root); | ||
| 114 | return NULL; | ||
| 115 | } | ||
| 116 | |||
| 117 | /* | ||
| 118 | * search through the tree for an extent_map with a given offset. If | ||
| 119 | * it can't be found, try to find some neighboring extents | ||
| 120 | */ | ||
| 121 | static struct rb_node *__tree_search(struct rb_root *root, u64 offset, | ||
| 122 | struct rb_node **prev_ret, | ||
| 123 | struct rb_node **next_ret) | ||
| 124 | { | ||
| 125 | struct rb_node *n = root->rb_node; | ||
| 126 | struct rb_node *prev = NULL; | ||
| 127 | struct rb_node *orig_prev = NULL; | ||
| 128 | struct extent_map *entry; | ||
| 129 | struct extent_map *prev_entry = NULL; | ||
| 130 | |||
| 131 | while (n) { | ||
| 132 | entry = rb_entry(n, struct extent_map, rb_node); | ||
| 133 | prev = n; | ||
| 134 | prev_entry = entry; | ||
| 135 | |||
| 136 | WARN_ON(!entry->in_tree); | ||
| 137 | |||
| 138 | if (offset < entry->start) | ||
| 139 | n = n->rb_left; | ||
| 140 | else if (offset >= extent_map_end(entry)) | ||
| 141 | n = n->rb_right; | ||
| 142 | else | ||
| 143 | return n; | ||
| 144 | } | ||
| 145 | |||
| 146 | if (prev_ret) { | ||
| 147 | orig_prev = prev; | ||
| 148 | while (prev && offset >= extent_map_end(prev_entry)) { | ||
| 149 | prev = rb_next(prev); | ||
| 150 | prev_entry = rb_entry(prev, struct extent_map, rb_node); | ||
| 151 | } | ||
| 152 | *prev_ret = prev; | ||
| 153 | prev = orig_prev; | ||
| 154 | } | ||
| 155 | |||
| 156 | if (next_ret) { | ||
| 157 | prev_entry = rb_entry(prev, struct extent_map, rb_node); | ||
| 158 | while (prev && offset < prev_entry->start) { | ||
| 159 | prev = rb_prev(prev); | ||
| 160 | prev_entry = rb_entry(prev, struct extent_map, rb_node); | ||
| 161 | } | ||
| 162 | *next_ret = prev; | ||
| 163 | } | ||
| 164 | return NULL; | ||
| 165 | } | ||
| 166 | |||
| 167 | /* | ||
| 168 | * look for an offset in the tree, and if it can't be found, return | ||
| 169 | * the first offset we can find smaller than 'offset'. | ||
| 170 | */ | ||
| 171 | static inline struct rb_node *tree_search(struct rb_root *root, u64 offset) | ||
| 172 | { | ||
| 173 | struct rb_node *prev; | ||
| 174 | struct rb_node *ret; | ||
| 175 | ret = __tree_search(root, offset, &prev, NULL); | ||
| 176 | if (!ret) | ||
| 177 | return prev; | ||
| 178 | return ret; | ||
| 179 | } | ||
| 180 | |||
| 181 | /* check to see if two extent_map structs are adjacent and safe to merge */ | ||
| 182 | static int mergable_maps(struct extent_map *prev, struct extent_map *next) | ||
| 183 | { | ||
| 184 | if (test_bit(EXTENT_FLAG_PINNED, &prev->flags)) | ||
| 185 | return 0; | ||
| 186 | |||
| 187 | /* | ||
| 188 | * don't merge compressed extents, we need to know their | ||
| 189 | * actual size | ||
| 190 | */ | ||
| 191 | if (test_bit(EXTENT_FLAG_COMPRESSED, &prev->flags)) | ||
| 192 | return 0; | ||
| 193 | |||
| 194 | if (extent_map_end(prev) == next->start && | ||
| 195 | prev->flags == next->flags && | ||
| 196 | prev->bdev == next->bdev && | ||
| 197 | ((next->block_start == EXTENT_MAP_HOLE && | ||
| 198 | prev->block_start == EXTENT_MAP_HOLE) || | ||
| 199 | (next->block_start == EXTENT_MAP_INLINE && | ||
| 200 | prev->block_start == EXTENT_MAP_INLINE) || | ||
| 201 | (next->block_start == EXTENT_MAP_DELALLOC && | ||
| 202 | prev->block_start == EXTENT_MAP_DELALLOC) || | ||
| 203 | (next->block_start < EXTENT_MAP_LAST_BYTE - 1 && | ||
| 204 | next->block_start == extent_map_block_end(prev)))) { | ||
| 205 | return 1; | ||
| 206 | } | ||
| 207 | return 0; | ||
| 208 | } | ||
| 209 | |||
| 210 | /** | ||
| 211 | * add_extent_mapping - add new extent map to the extent tree | ||
| 212 | * @tree: tree to insert new map in | ||
| 213 | * @em: map to insert | ||
| 214 | * | ||
| 215 | * Insert @em into @tree or perform a simple forward/backward merge with | ||
| 216 | * existing mappings. The extent_map struct passed in will be inserted | ||
| 217 | * into the tree directly, with an additional reference taken, or a | ||
| 218 | * reference dropped if the merge attempt was sucessfull. | ||
| 219 | */ | ||
| 220 | int add_extent_mapping(struct extent_map_tree *tree, | ||
| 221 | struct extent_map *em) | ||
| 222 | { | ||
| 223 | int ret = 0; | ||
| 224 | struct extent_map *merge = NULL; | ||
| 225 | struct rb_node *rb; | ||
| 226 | struct extent_map *exist; | ||
| 227 | |||
| 228 | exist = lookup_extent_mapping(tree, em->start, em->len); | ||
| 229 | if (exist) { | ||
| 230 | free_extent_map(exist); | ||
| 231 | ret = -EEXIST; | ||
| 232 | goto out; | ||
| 233 | } | ||
| 234 | assert_spin_locked(&tree->lock); | ||
| 235 | rb = tree_insert(&tree->map, em->start, &em->rb_node); | ||
| 236 | if (rb) { | ||
| 237 | ret = -EEXIST; | ||
| 238 | free_extent_map(merge); | ||
| 239 | goto out; | ||
| 240 | } | ||
| 241 | atomic_inc(&em->refs); | ||
| 242 | if (em->start != 0) { | ||
| 243 | rb = rb_prev(&em->rb_node); | ||
| 244 | if (rb) | ||
| 245 | merge = rb_entry(rb, struct extent_map, rb_node); | ||
| 246 | if (rb && mergable_maps(merge, em)) { | ||
| 247 | em->start = merge->start; | ||
| 248 | em->len += merge->len; | ||
| 249 | em->block_len += merge->block_len; | ||
| 250 | em->block_start = merge->block_start; | ||
| 251 | merge->in_tree = 0; | ||
| 252 | rb_erase(&merge->rb_node, &tree->map); | ||
| 253 | free_extent_map(merge); | ||
| 254 | } | ||
| 255 | } | ||
| 256 | rb = rb_next(&em->rb_node); | ||
| 257 | if (rb) | ||
| 258 | merge = rb_entry(rb, struct extent_map, rb_node); | ||
| 259 | if (rb && mergable_maps(em, merge)) { | ||
| 260 | em->len += merge->len; | ||
| 261 | em->block_len += merge->len; | ||
| 262 | rb_erase(&merge->rb_node, &tree->map); | ||
| 263 | merge->in_tree = 0; | ||
| 264 | free_extent_map(merge); | ||
| 265 | } | ||
| 266 | out: | ||
| 267 | return ret; | ||
| 268 | } | ||
| 269 | EXPORT_SYMBOL(add_extent_mapping); | ||
| 270 | |||
| 271 | /* simple helper to do math around the end of an extent, handling wrap */ | ||
| 272 | static u64 range_end(u64 start, u64 len) | ||
| 273 | { | ||
| 274 | if (start + len < start) | ||
| 275 | return (u64)-1; | ||
| 276 | return start + len; | ||
| 277 | } | ||
| 278 | |||
| 279 | /** | ||
| 280 | * lookup_extent_mapping - lookup extent_map | ||
| 281 | * @tree: tree to lookup in | ||
| 282 | * @start: byte offset to start the search | ||
| 283 | * @len: length of the lookup range | ||
| 284 | * | ||
| 285 | * Find and return the first extent_map struct in @tree that intersects the | ||
| 286 | * [start, len] range. There may be additional objects in the tree that | ||
| 287 | * intersect, so check the object returned carefully to make sure that no | ||
| 288 | * additional lookups are needed. | ||
| 289 | */ | ||
| 290 | struct extent_map *lookup_extent_mapping(struct extent_map_tree *tree, | ||
| 291 | u64 start, u64 len) | ||
| 292 | { | ||
| 293 | struct extent_map *em; | ||
| 294 | struct rb_node *rb_node; | ||
| 295 | struct rb_node *prev = NULL; | ||
| 296 | struct rb_node *next = NULL; | ||
| 297 | u64 end = range_end(start, len); | ||
| 298 | |||
| 299 | assert_spin_locked(&tree->lock); | ||
| 300 | rb_node = __tree_search(&tree->map, start, &prev, &next); | ||
| 301 | if (!rb_node && prev) { | ||
| 302 | em = rb_entry(prev, struct extent_map, rb_node); | ||
| 303 | if (end > em->start && start < extent_map_end(em)) | ||
| 304 | goto found; | ||
| 305 | } | ||
| 306 | if (!rb_node && next) { | ||
| 307 | em = rb_entry(next, struct extent_map, rb_node); | ||
| 308 | if (end > em->start && start < extent_map_end(em)) | ||
| 309 | goto found; | ||
| 310 | } | ||
| 311 | if (!rb_node) { | ||
| 312 | em = NULL; | ||
| 313 | goto out; | ||
| 314 | } | ||
| 315 | if (IS_ERR(rb_node)) { | ||
| 316 | em = ERR_PTR(PTR_ERR(rb_node)); | ||
| 317 | goto out; | ||
| 318 | } | ||
| 319 | em = rb_entry(rb_node, struct extent_map, rb_node); | ||
| 320 | if (end > em->start && start < extent_map_end(em)) | ||
| 321 | goto found; | ||
| 322 | |||
| 323 | em = NULL; | ||
| 324 | goto out; | ||
| 325 | |||
| 326 | found: | ||
| 327 | atomic_inc(&em->refs); | ||
| 328 | out: | ||
| 329 | return em; | ||
| 330 | } | ||
| 331 | EXPORT_SYMBOL(lookup_extent_mapping); | ||
| 332 | |||
| 333 | /** | ||
| 334 | * remove_extent_mapping - removes an extent_map from the extent tree | ||
| 335 | * @tree: extent tree to remove from | ||
| 336 | * @em: extent map beeing removed | ||
| 337 | * | ||
| 338 | * Removes @em from @tree. No reference counts are dropped, and no checks | ||
| 339 | * are done to see if the range is in use | ||
| 340 | */ | ||
| 341 | int remove_extent_mapping(struct extent_map_tree *tree, struct extent_map *em) | ||
| 342 | { | ||
| 343 | int ret = 0; | ||
| 344 | |||
| 345 | WARN_ON(test_bit(EXTENT_FLAG_PINNED, &em->flags)); | ||
| 346 | assert_spin_locked(&tree->lock); | ||
| 347 | rb_erase(&em->rb_node, &tree->map); | ||
| 348 | em->in_tree = 0; | ||
| 349 | return ret; | ||
| 350 | } | ||
| 351 | EXPORT_SYMBOL(remove_extent_mapping); | ||
diff --git a/fs/btrfs/extent_map.h b/fs/btrfs/extent_map.h new file mode 100644 index 000000000000..fb6eeef06bb0 --- /dev/null +++ b/fs/btrfs/extent_map.h | |||
| @@ -0,0 +1,62 @@ | |||
| 1 | #ifndef __EXTENTMAP__ | ||
| 2 | #define __EXTENTMAP__ | ||
| 3 | |||
| 4 | #include <linux/rbtree.h> | ||
| 5 | |||
| 6 | #define EXTENT_MAP_LAST_BYTE (u64)-4 | ||
| 7 | #define EXTENT_MAP_HOLE (u64)-3 | ||
| 8 | #define EXTENT_MAP_INLINE (u64)-2 | ||
| 9 | #define EXTENT_MAP_DELALLOC (u64)-1 | ||
| 10 | |||
| 11 | /* bits for the flags field */ | ||
| 12 | #define EXTENT_FLAG_PINNED 0 /* this entry not yet on disk, don't free it */ | ||
| 13 | #define EXTENT_FLAG_COMPRESSED 1 | ||
| 14 | #define EXTENT_FLAG_VACANCY 2 /* no file extent item found */ | ||
| 15 | #define EXTENT_FLAG_PREALLOC 3 /* pre-allocated extent */ | ||
| 16 | |||
| 17 | struct extent_map { | ||
| 18 | struct rb_node rb_node; | ||
| 19 | |||
| 20 | /* all of these are in bytes */ | ||
| 21 | u64 start; | ||
| 22 | u64 len; | ||
| 23 | u64 orig_start; | ||
| 24 | u64 block_start; | ||
| 25 | u64 block_len; | ||
| 26 | unsigned long flags; | ||
| 27 | struct block_device *bdev; | ||
| 28 | atomic_t refs; | ||
| 29 | int in_tree; | ||
| 30 | }; | ||
| 31 | |||
| 32 | struct extent_map_tree { | ||
| 33 | struct rb_root map; | ||
| 34 | spinlock_t lock; | ||
| 35 | }; | ||
| 36 | |||
| 37 | static inline u64 extent_map_end(struct extent_map *em) | ||
| 38 | { | ||
| 39 | if (em->start + em->len < em->start) | ||
| 40 | return (u64)-1; | ||
| 41 | return em->start + em->len; | ||
| 42 | } | ||
| 43 | |||
| 44 | static inline u64 extent_map_block_end(struct extent_map *em) | ||
| 45 | { | ||
| 46 | if (em->block_start + em->block_len < em->block_start) | ||
| 47 | return (u64)-1; | ||
| 48 | return em->block_start + em->block_len; | ||
| 49 | } | ||
| 50 | |||
| 51 | void extent_map_tree_init(struct extent_map_tree *tree, gfp_t mask); | ||
| 52 | struct extent_map *lookup_extent_mapping(struct extent_map_tree *tree, | ||
| 53 | u64 start, u64 len); | ||
| 54 | int add_extent_mapping(struct extent_map_tree *tree, | ||
| 55 | struct extent_map *em); | ||
| 56 | int remove_extent_mapping(struct extent_map_tree *tree, struct extent_map *em); | ||
| 57 | |||
| 58 | struct extent_map *alloc_extent_map(gfp_t mask); | ||
| 59 | void free_extent_map(struct extent_map *em); | ||
| 60 | int __init extent_map_init(void); | ||
| 61 | void extent_map_exit(void); | ||
| 62 | #endif | ||
diff --git a/fs/btrfs/file-item.c b/fs/btrfs/file-item.c new file mode 100644 index 000000000000..964652435fd1 --- /dev/null +++ b/fs/btrfs/file-item.c | |||
| @@ -0,0 +1,831 @@ | |||
| 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 | |||
| 19 | #include <linux/bio.h> | ||
| 20 | #include <linux/pagemap.h> | ||
| 21 | #include <linux/highmem.h> | ||
| 22 | #include "ctree.h" | ||
| 23 | #include "disk-io.h" | ||
| 24 | #include "transaction.h" | ||
| 25 | #include "print-tree.h" | ||
| 26 | |||
| 27 | #define MAX_CSUM_ITEMS(r, size) ((((BTRFS_LEAF_DATA_SIZE(r) - \ | ||
| 28 | sizeof(struct btrfs_item) * 2) / \ | ||
| 29 | size) - 1)) | ||
| 30 | |||
| 31 | #define MAX_ORDERED_SUM_BYTES(r) ((PAGE_SIZE - \ | ||
| 32 | sizeof(struct btrfs_ordered_sum)) / \ | ||
| 33 | sizeof(struct btrfs_sector_sum) * \ | ||
| 34 | (r)->sectorsize - (r)->sectorsize) | ||
| 35 | |||
| 36 | int btrfs_insert_file_extent(struct btrfs_trans_handle *trans, | ||
| 37 | struct btrfs_root *root, | ||
| 38 | u64 objectid, u64 pos, | ||
| 39 | u64 disk_offset, u64 disk_num_bytes, | ||
| 40 | u64 num_bytes, u64 offset, u64 ram_bytes, | ||
| 41 | u8 compression, u8 encryption, u16 other_encoding) | ||
| 42 | { | ||
| 43 | int ret = 0; | ||
| 44 | struct btrfs_file_extent_item *item; | ||
| 45 | struct btrfs_key file_key; | ||
| 46 | struct btrfs_path *path; | ||
| 47 | struct extent_buffer *leaf; | ||
| 48 | |||
| 49 | path = btrfs_alloc_path(); | ||
| 50 | BUG_ON(!path); | ||
| 51 | file_key.objectid = objectid; | ||
| 52 | file_key.offset = pos; | ||
| 53 | btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); | ||
| 54 | |||
| 55 | ret = btrfs_insert_empty_item(trans, root, path, &file_key, | ||
| 56 | sizeof(*item)); | ||
| 57 | if (ret < 0) | ||
| 58 | goto out; | ||
| 59 | BUG_ON(ret); | ||
| 60 | leaf = path->nodes[0]; | ||
| 61 | item = btrfs_item_ptr(leaf, path->slots[0], | ||
| 62 | struct btrfs_file_extent_item); | ||
| 63 | btrfs_set_file_extent_disk_bytenr(leaf, item, disk_offset); | ||
| 64 | btrfs_set_file_extent_disk_num_bytes(leaf, item, disk_num_bytes); | ||
| 65 | btrfs_set_file_extent_offset(leaf, item, offset); | ||
| 66 | btrfs_set_file_extent_num_bytes(leaf, item, num_bytes); | ||
| 67 | btrfs_set_file_extent_ram_bytes(leaf, item, ram_bytes); | ||
| 68 | btrfs_set_file_extent_generation(leaf, item, trans->transid); | ||
| 69 | btrfs_set_file_extent_type(leaf, item, BTRFS_FILE_EXTENT_REG); | ||
| 70 | btrfs_set_file_extent_compression(leaf, item, compression); | ||
| 71 | btrfs_set_file_extent_encryption(leaf, item, encryption); | ||
| 72 | btrfs_set_file_extent_other_encoding(leaf, item, other_encoding); | ||
| 73 | |||
| 74 | btrfs_mark_buffer_dirty(leaf); | ||
| 75 | out: | ||
| 76 | btrfs_free_path(path); | ||
| 77 | return ret; | ||
| 78 | } | ||
| 79 | |||
| 80 | struct btrfs_csum_item *btrfs_lookup_csum(struct btrfs_trans_handle *trans, | ||
| 81 | struct btrfs_root *root, | ||
| 82 | struct btrfs_path *path, | ||
| 83 | u64 bytenr, int cow) | ||
| 84 | { | ||
| 85 | int ret; | ||
| 86 | struct btrfs_key file_key; | ||
| 87 | struct btrfs_key found_key; | ||
| 88 | struct btrfs_csum_item *item; | ||
| 89 | struct extent_buffer *leaf; | ||
| 90 | u64 csum_offset = 0; | ||
| 91 | u16 csum_size = | ||
| 92 | btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 93 | int csums_in_item; | ||
| 94 | |||
| 95 | file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; | ||
| 96 | file_key.offset = bytenr; | ||
| 97 | btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY); | ||
| 98 | ret = btrfs_search_slot(trans, root, &file_key, path, 0, cow); | ||
| 99 | if (ret < 0) | ||
| 100 | goto fail; | ||
| 101 | leaf = path->nodes[0]; | ||
| 102 | if (ret > 0) { | ||
| 103 | ret = 1; | ||
| 104 | if (path->slots[0] == 0) | ||
| 105 | goto fail; | ||
| 106 | path->slots[0]--; | ||
| 107 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 108 | if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY) | ||
| 109 | goto fail; | ||
| 110 | |||
| 111 | csum_offset = (bytenr - found_key.offset) >> | ||
| 112 | root->fs_info->sb->s_blocksize_bits; | ||
| 113 | csums_in_item = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 114 | csums_in_item /= csum_size; | ||
| 115 | |||
| 116 | if (csum_offset >= csums_in_item) { | ||
| 117 | ret = -EFBIG; | ||
| 118 | goto fail; | ||
| 119 | } | ||
| 120 | } | ||
| 121 | item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item); | ||
| 122 | item = (struct btrfs_csum_item *)((unsigned char *)item + | ||
| 123 | csum_offset * csum_size); | ||
| 124 | return item; | ||
| 125 | fail: | ||
| 126 | if (ret > 0) | ||
| 127 | ret = -ENOENT; | ||
| 128 | return ERR_PTR(ret); | ||
| 129 | } | ||
| 130 | |||
| 131 | |||
| 132 | int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans, | ||
| 133 | struct btrfs_root *root, | ||
| 134 | struct btrfs_path *path, u64 objectid, | ||
| 135 | u64 offset, int mod) | ||
| 136 | { | ||
| 137 | int ret; | ||
| 138 | struct btrfs_key file_key; | ||
| 139 | int ins_len = mod < 0 ? -1 : 0; | ||
| 140 | int cow = mod != 0; | ||
| 141 | |||
| 142 | file_key.objectid = objectid; | ||
| 143 | file_key.offset = offset; | ||
| 144 | btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY); | ||
| 145 | ret = btrfs_search_slot(trans, root, &file_key, path, ins_len, cow); | ||
| 146 | return ret; | ||
| 147 | } | ||
| 148 | |||
| 149 | |||
| 150 | int btrfs_lookup_bio_sums(struct btrfs_root *root, struct inode *inode, | ||
| 151 | struct bio *bio, u32 *dst) | ||
| 152 | { | ||
| 153 | u32 sum; | ||
| 154 | struct bio_vec *bvec = bio->bi_io_vec; | ||
| 155 | int bio_index = 0; | ||
| 156 | u64 offset; | ||
| 157 | u64 item_start_offset = 0; | ||
| 158 | u64 item_last_offset = 0; | ||
| 159 | u64 disk_bytenr; | ||
| 160 | u32 diff; | ||
| 161 | u16 csum_size = | ||
| 162 | btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 163 | int ret; | ||
| 164 | struct btrfs_path *path; | ||
| 165 | struct btrfs_csum_item *item = NULL; | ||
| 166 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 167 | |||
| 168 | path = btrfs_alloc_path(); | ||
| 169 | if (bio->bi_size > PAGE_CACHE_SIZE * 8) | ||
| 170 | path->reada = 2; | ||
| 171 | |||
| 172 | WARN_ON(bio->bi_vcnt <= 0); | ||
| 173 | |||
| 174 | disk_bytenr = (u64)bio->bi_sector << 9; | ||
| 175 | while (bio_index < bio->bi_vcnt) { | ||
| 176 | offset = page_offset(bvec->bv_page) + bvec->bv_offset; | ||
| 177 | ret = btrfs_find_ordered_sum(inode, offset, disk_bytenr, &sum); | ||
| 178 | if (ret == 0) | ||
| 179 | goto found; | ||
| 180 | |||
| 181 | if (!item || disk_bytenr < item_start_offset || | ||
| 182 | disk_bytenr >= item_last_offset) { | ||
| 183 | struct btrfs_key found_key; | ||
| 184 | u32 item_size; | ||
| 185 | |||
| 186 | if (item) | ||
| 187 | btrfs_release_path(root, path); | ||
| 188 | item = btrfs_lookup_csum(NULL, root->fs_info->csum_root, | ||
| 189 | path, disk_bytenr, 0); | ||
| 190 | if (IS_ERR(item)) { | ||
| 191 | ret = PTR_ERR(item); | ||
| 192 | if (ret == -ENOENT || ret == -EFBIG) | ||
| 193 | ret = 0; | ||
| 194 | sum = 0; | ||
| 195 | if (BTRFS_I(inode)->root->root_key.objectid == | ||
| 196 | BTRFS_DATA_RELOC_TREE_OBJECTID) { | ||
| 197 | set_extent_bits(io_tree, offset, | ||
| 198 | offset + bvec->bv_len - 1, | ||
| 199 | EXTENT_NODATASUM, GFP_NOFS); | ||
| 200 | } else { | ||
| 201 | printk(KERN_INFO "btrfs no csum found " | ||
| 202 | "for inode %lu start %llu\n", | ||
| 203 | inode->i_ino, | ||
| 204 | (unsigned long long)offset); | ||
| 205 | } | ||
| 206 | item = NULL; | ||
| 207 | btrfs_release_path(root, path); | ||
| 208 | goto found; | ||
| 209 | } | ||
| 210 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 211 | path->slots[0]); | ||
| 212 | |||
| 213 | item_start_offset = found_key.offset; | ||
| 214 | item_size = btrfs_item_size_nr(path->nodes[0], | ||
| 215 | path->slots[0]); | ||
| 216 | item_last_offset = item_start_offset + | ||
| 217 | (item_size / csum_size) * | ||
| 218 | root->sectorsize; | ||
| 219 | item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 220 | struct btrfs_csum_item); | ||
| 221 | } | ||
| 222 | /* | ||
| 223 | * this byte range must be able to fit inside | ||
| 224 | * a single leaf so it will also fit inside a u32 | ||
| 225 | */ | ||
| 226 | diff = disk_bytenr - item_start_offset; | ||
| 227 | diff = diff / root->sectorsize; | ||
| 228 | diff = diff * csum_size; | ||
| 229 | |||
| 230 | read_extent_buffer(path->nodes[0], &sum, | ||
| 231 | ((unsigned long)item) + diff, | ||
| 232 | csum_size); | ||
| 233 | found: | ||
| 234 | if (dst) | ||
| 235 | *dst++ = sum; | ||
| 236 | else | ||
| 237 | set_state_private(io_tree, offset, sum); | ||
| 238 | disk_bytenr += bvec->bv_len; | ||
| 239 | bio_index++; | ||
| 240 | bvec++; | ||
| 241 | } | ||
| 242 | btrfs_free_path(path); | ||
| 243 | return 0; | ||
| 244 | } | ||
| 245 | |||
| 246 | int btrfs_lookup_csums_range(struct btrfs_root *root, u64 start, u64 end, | ||
| 247 | struct list_head *list) | ||
| 248 | { | ||
| 249 | struct btrfs_key key; | ||
| 250 | struct btrfs_path *path; | ||
| 251 | struct extent_buffer *leaf; | ||
| 252 | struct btrfs_ordered_sum *sums; | ||
| 253 | struct btrfs_sector_sum *sector_sum; | ||
| 254 | struct btrfs_csum_item *item; | ||
| 255 | unsigned long offset; | ||
| 256 | int ret; | ||
| 257 | size_t size; | ||
| 258 | u64 csum_end; | ||
| 259 | u16 csum_size = btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 260 | |||
| 261 | path = btrfs_alloc_path(); | ||
| 262 | BUG_ON(!path); | ||
| 263 | |||
| 264 | key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; | ||
| 265 | key.offset = start; | ||
| 266 | key.type = BTRFS_EXTENT_CSUM_KEY; | ||
| 267 | |||
| 268 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 269 | if (ret < 0) | ||
| 270 | goto fail; | ||
| 271 | if (ret > 0 && path->slots[0] > 0) { | ||
| 272 | leaf = path->nodes[0]; | ||
| 273 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0] - 1); | ||
| 274 | if (key.objectid == BTRFS_EXTENT_CSUM_OBJECTID && | ||
| 275 | key.type == BTRFS_EXTENT_CSUM_KEY) { | ||
| 276 | offset = (start - key.offset) >> | ||
| 277 | root->fs_info->sb->s_blocksize_bits; | ||
| 278 | if (offset * csum_size < | ||
| 279 | btrfs_item_size_nr(leaf, path->slots[0] - 1)) | ||
| 280 | path->slots[0]--; | ||
| 281 | } | ||
| 282 | } | ||
| 283 | |||
| 284 | while (start <= end) { | ||
| 285 | leaf = path->nodes[0]; | ||
| 286 | if (path->slots[0] >= btrfs_header_nritems(leaf)) { | ||
| 287 | ret = btrfs_next_leaf(root, path); | ||
| 288 | if (ret < 0) | ||
| 289 | goto fail; | ||
| 290 | if (ret > 0) | ||
| 291 | break; | ||
| 292 | leaf = path->nodes[0]; | ||
| 293 | } | ||
| 294 | |||
| 295 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 296 | if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || | ||
| 297 | key.type != BTRFS_EXTENT_CSUM_KEY) | ||
| 298 | break; | ||
| 299 | |||
| 300 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 301 | if (key.offset > end) | ||
| 302 | break; | ||
| 303 | |||
| 304 | if (key.offset > start) | ||
| 305 | start = key.offset; | ||
| 306 | |||
| 307 | size = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 308 | csum_end = key.offset + (size / csum_size) * root->sectorsize; | ||
| 309 | if (csum_end <= start) { | ||
| 310 | path->slots[0]++; | ||
| 311 | continue; | ||
| 312 | } | ||
| 313 | |||
| 314 | csum_end = min(csum_end, end + 1); | ||
| 315 | item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 316 | struct btrfs_csum_item); | ||
| 317 | while (start < csum_end) { | ||
| 318 | size = min_t(size_t, csum_end - start, | ||
| 319 | MAX_ORDERED_SUM_BYTES(root)); | ||
| 320 | sums = kzalloc(btrfs_ordered_sum_size(root, size), | ||
| 321 | GFP_NOFS); | ||
| 322 | BUG_ON(!sums); | ||
| 323 | |||
| 324 | sector_sum = sums->sums; | ||
| 325 | sums->bytenr = start; | ||
| 326 | sums->len = size; | ||
| 327 | |||
| 328 | offset = (start - key.offset) >> | ||
| 329 | root->fs_info->sb->s_blocksize_bits; | ||
| 330 | offset *= csum_size; | ||
| 331 | |||
| 332 | while (size > 0) { | ||
| 333 | read_extent_buffer(path->nodes[0], | ||
| 334 | §or_sum->sum, | ||
| 335 | ((unsigned long)item) + | ||
| 336 | offset, csum_size); | ||
| 337 | sector_sum->bytenr = start; | ||
| 338 | |||
| 339 | size -= root->sectorsize; | ||
| 340 | start += root->sectorsize; | ||
| 341 | offset += csum_size; | ||
| 342 | sector_sum++; | ||
| 343 | } | ||
| 344 | list_add_tail(&sums->list, list); | ||
| 345 | } | ||
| 346 | path->slots[0]++; | ||
| 347 | } | ||
| 348 | ret = 0; | ||
| 349 | fail: | ||
| 350 | btrfs_free_path(path); | ||
| 351 | return ret; | ||
| 352 | } | ||
| 353 | |||
| 354 | int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode, | ||
| 355 | struct bio *bio, u64 file_start, int contig) | ||
| 356 | { | ||
| 357 | struct btrfs_ordered_sum *sums; | ||
| 358 | struct btrfs_sector_sum *sector_sum; | ||
| 359 | struct btrfs_ordered_extent *ordered; | ||
| 360 | char *data; | ||
| 361 | struct bio_vec *bvec = bio->bi_io_vec; | ||
| 362 | int bio_index = 0; | ||
| 363 | unsigned long total_bytes = 0; | ||
| 364 | unsigned long this_sum_bytes = 0; | ||
| 365 | u64 offset; | ||
| 366 | u64 disk_bytenr; | ||
| 367 | |||
| 368 | WARN_ON(bio->bi_vcnt <= 0); | ||
| 369 | sums = kzalloc(btrfs_ordered_sum_size(root, bio->bi_size), GFP_NOFS); | ||
| 370 | if (!sums) | ||
| 371 | return -ENOMEM; | ||
| 372 | |||
| 373 | sector_sum = sums->sums; | ||
| 374 | disk_bytenr = (u64)bio->bi_sector << 9; | ||
| 375 | sums->len = bio->bi_size; | ||
| 376 | INIT_LIST_HEAD(&sums->list); | ||
| 377 | |||
| 378 | if (contig) | ||
| 379 | offset = file_start; | ||
| 380 | else | ||
| 381 | offset = page_offset(bvec->bv_page) + bvec->bv_offset; | ||
| 382 | |||
| 383 | ordered = btrfs_lookup_ordered_extent(inode, offset); | ||
| 384 | BUG_ON(!ordered); | ||
| 385 | sums->bytenr = ordered->start; | ||
| 386 | |||
| 387 | while (bio_index < bio->bi_vcnt) { | ||
| 388 | if (!contig) | ||
| 389 | offset = page_offset(bvec->bv_page) + bvec->bv_offset; | ||
| 390 | |||
| 391 | if (!contig && (offset >= ordered->file_offset + ordered->len || | ||
| 392 | offset < ordered->file_offset)) { | ||
| 393 | unsigned long bytes_left; | ||
| 394 | sums->len = this_sum_bytes; | ||
| 395 | this_sum_bytes = 0; | ||
| 396 | btrfs_add_ordered_sum(inode, ordered, sums); | ||
| 397 | btrfs_put_ordered_extent(ordered); | ||
| 398 | |||
| 399 | bytes_left = bio->bi_size - total_bytes; | ||
| 400 | |||
| 401 | sums = kzalloc(btrfs_ordered_sum_size(root, bytes_left), | ||
| 402 | GFP_NOFS); | ||
| 403 | BUG_ON(!sums); | ||
| 404 | sector_sum = sums->sums; | ||
| 405 | sums->len = bytes_left; | ||
| 406 | ordered = btrfs_lookup_ordered_extent(inode, offset); | ||
| 407 | BUG_ON(!ordered); | ||
| 408 | sums->bytenr = ordered->start; | ||
| 409 | } | ||
| 410 | |||
| 411 | data = kmap_atomic(bvec->bv_page, KM_USER0); | ||
| 412 | sector_sum->sum = ~(u32)0; | ||
| 413 | sector_sum->sum = btrfs_csum_data(root, | ||
| 414 | data + bvec->bv_offset, | ||
| 415 | sector_sum->sum, | ||
| 416 | bvec->bv_len); | ||
| 417 | kunmap_atomic(data, KM_USER0); | ||
| 418 | btrfs_csum_final(sector_sum->sum, | ||
| 419 | (char *)§or_sum->sum); | ||
| 420 | sector_sum->bytenr = disk_bytenr; | ||
| 421 | |||
| 422 | sector_sum++; | ||
| 423 | bio_index++; | ||
| 424 | total_bytes += bvec->bv_len; | ||
| 425 | this_sum_bytes += bvec->bv_len; | ||
| 426 | disk_bytenr += bvec->bv_len; | ||
| 427 | offset += bvec->bv_len; | ||
| 428 | bvec++; | ||
| 429 | } | ||
| 430 | this_sum_bytes = 0; | ||
| 431 | btrfs_add_ordered_sum(inode, ordered, sums); | ||
| 432 | btrfs_put_ordered_extent(ordered); | ||
| 433 | return 0; | ||
| 434 | } | ||
| 435 | |||
| 436 | /* | ||
| 437 | * helper function for csum removal, this expects the | ||
| 438 | * key to describe the csum pointed to by the path, and it expects | ||
| 439 | * the csum to overlap the range [bytenr, len] | ||
| 440 | * | ||
| 441 | * The csum should not be entirely contained in the range and the | ||
| 442 | * range should not be entirely contained in the csum. | ||
| 443 | * | ||
| 444 | * This calls btrfs_truncate_item with the correct args based on the | ||
| 445 | * overlap, and fixes up the key as required. | ||
| 446 | */ | ||
| 447 | static noinline int truncate_one_csum(struct btrfs_trans_handle *trans, | ||
| 448 | struct btrfs_root *root, | ||
| 449 | struct btrfs_path *path, | ||
| 450 | struct btrfs_key *key, | ||
| 451 | u64 bytenr, u64 len) | ||
| 452 | { | ||
| 453 | struct extent_buffer *leaf; | ||
| 454 | u16 csum_size = | ||
| 455 | btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 456 | u64 csum_end; | ||
| 457 | u64 end_byte = bytenr + len; | ||
| 458 | u32 blocksize_bits = root->fs_info->sb->s_blocksize_bits; | ||
| 459 | int ret; | ||
| 460 | |||
| 461 | leaf = path->nodes[0]; | ||
| 462 | csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size; | ||
| 463 | csum_end <<= root->fs_info->sb->s_blocksize_bits; | ||
| 464 | csum_end += key->offset; | ||
| 465 | |||
| 466 | if (key->offset < bytenr && csum_end <= end_byte) { | ||
| 467 | /* | ||
| 468 | * [ bytenr - len ] | ||
| 469 | * [ ] | ||
| 470 | * [csum ] | ||
| 471 | * A simple truncate off the end of the item | ||
| 472 | */ | ||
| 473 | u32 new_size = (bytenr - key->offset) >> blocksize_bits; | ||
| 474 | new_size *= csum_size; | ||
| 475 | ret = btrfs_truncate_item(trans, root, path, new_size, 1); | ||
| 476 | BUG_ON(ret); | ||
| 477 | } else if (key->offset >= bytenr && csum_end > end_byte && | ||
| 478 | end_byte > key->offset) { | ||
| 479 | /* | ||
| 480 | * [ bytenr - len ] | ||
| 481 | * [ ] | ||
| 482 | * [csum ] | ||
| 483 | * we need to truncate from the beginning of the csum | ||
| 484 | */ | ||
| 485 | u32 new_size = (csum_end - end_byte) >> blocksize_bits; | ||
| 486 | new_size *= csum_size; | ||
| 487 | |||
| 488 | ret = btrfs_truncate_item(trans, root, path, new_size, 0); | ||
| 489 | BUG_ON(ret); | ||
| 490 | |||
| 491 | key->offset = end_byte; | ||
| 492 | ret = btrfs_set_item_key_safe(trans, root, path, key); | ||
| 493 | BUG_ON(ret); | ||
| 494 | } else { | ||
| 495 | BUG(); | ||
| 496 | } | ||
| 497 | return 0; | ||
| 498 | } | ||
| 499 | |||
| 500 | /* | ||
| 501 | * deletes the csum items from the csum tree for a given | ||
| 502 | * range of bytes. | ||
| 503 | */ | ||
| 504 | int btrfs_del_csums(struct btrfs_trans_handle *trans, | ||
| 505 | struct btrfs_root *root, u64 bytenr, u64 len) | ||
| 506 | { | ||
| 507 | struct btrfs_path *path; | ||
| 508 | struct btrfs_key key; | ||
| 509 | u64 end_byte = bytenr + len; | ||
| 510 | u64 csum_end; | ||
| 511 | struct extent_buffer *leaf; | ||
| 512 | int ret; | ||
| 513 | u16 csum_size = | ||
| 514 | btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 515 | int blocksize_bits = root->fs_info->sb->s_blocksize_bits; | ||
| 516 | |||
| 517 | root = root->fs_info->csum_root; | ||
| 518 | |||
| 519 | path = btrfs_alloc_path(); | ||
| 520 | |||
| 521 | while (1) { | ||
| 522 | key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; | ||
| 523 | key.offset = end_byte - 1; | ||
| 524 | key.type = BTRFS_EXTENT_CSUM_KEY; | ||
| 525 | |||
| 526 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 527 | if (ret > 0) { | ||
| 528 | if (path->slots[0] == 0) | ||
| 529 | goto out; | ||
| 530 | path->slots[0]--; | ||
| 531 | } | ||
| 532 | leaf = path->nodes[0]; | ||
| 533 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 534 | |||
| 535 | if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || | ||
| 536 | key.type != BTRFS_EXTENT_CSUM_KEY) { | ||
| 537 | break; | ||
| 538 | } | ||
| 539 | |||
| 540 | if (key.offset >= end_byte) | ||
| 541 | break; | ||
| 542 | |||
| 543 | csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size; | ||
| 544 | csum_end <<= blocksize_bits; | ||
| 545 | csum_end += key.offset; | ||
| 546 | |||
| 547 | /* this csum ends before we start, we're done */ | ||
| 548 | if (csum_end <= bytenr) | ||
| 549 | break; | ||
| 550 | |||
| 551 | /* delete the entire item, it is inside our range */ | ||
| 552 | if (key.offset >= bytenr && csum_end <= end_byte) { | ||
| 553 | ret = btrfs_del_item(trans, root, path); | ||
| 554 | BUG_ON(ret); | ||
| 555 | if (key.offset == bytenr) | ||
| 556 | break; | ||
| 557 | } else if (key.offset < bytenr && csum_end > end_byte) { | ||
| 558 | unsigned long offset; | ||
| 559 | unsigned long shift_len; | ||
| 560 | unsigned long item_offset; | ||
| 561 | /* | ||
| 562 | * [ bytenr - len ] | ||
| 563 | * [csum ] | ||
| 564 | * | ||
| 565 | * Our bytes are in the middle of the csum, | ||
| 566 | * we need to split this item and insert a new one. | ||
| 567 | * | ||
| 568 | * But we can't drop the path because the | ||
| 569 | * csum could change, get removed, extended etc. | ||
| 570 | * | ||
| 571 | * The trick here is the max size of a csum item leaves | ||
| 572 | * enough room in the tree block for a single | ||
| 573 | * item header. So, we split the item in place, | ||
| 574 | * adding a new header pointing to the existing | ||
| 575 | * bytes. Then we loop around again and we have | ||
| 576 | * a nicely formed csum item that we can neatly | ||
| 577 | * truncate. | ||
| 578 | */ | ||
| 579 | offset = (bytenr - key.offset) >> blocksize_bits; | ||
| 580 | offset *= csum_size; | ||
| 581 | |||
| 582 | shift_len = (len >> blocksize_bits) * csum_size; | ||
| 583 | |||
| 584 | item_offset = btrfs_item_ptr_offset(leaf, | ||
| 585 | path->slots[0]); | ||
| 586 | |||
| 587 | memset_extent_buffer(leaf, 0, item_offset + offset, | ||
| 588 | shift_len); | ||
| 589 | key.offset = bytenr; | ||
| 590 | |||
| 591 | /* | ||
| 592 | * btrfs_split_item returns -EAGAIN when the | ||
| 593 | * item changed size or key | ||
| 594 | */ | ||
| 595 | ret = btrfs_split_item(trans, root, path, &key, offset); | ||
| 596 | BUG_ON(ret && ret != -EAGAIN); | ||
| 597 | |||
| 598 | key.offset = end_byte - 1; | ||
| 599 | } else { | ||
| 600 | ret = truncate_one_csum(trans, root, path, | ||
| 601 | &key, bytenr, len); | ||
| 602 | BUG_ON(ret); | ||
| 603 | if (key.offset < bytenr) | ||
| 604 | break; | ||
| 605 | } | ||
| 606 | btrfs_release_path(root, path); | ||
| 607 | } | ||
| 608 | out: | ||
| 609 | btrfs_free_path(path); | ||
| 610 | return 0; | ||
| 611 | } | ||
| 612 | |||
| 613 | int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans, | ||
| 614 | struct btrfs_root *root, | ||
| 615 | struct btrfs_ordered_sum *sums) | ||
| 616 | { | ||
| 617 | u64 bytenr; | ||
| 618 | int ret; | ||
| 619 | struct btrfs_key file_key; | ||
| 620 | struct btrfs_key found_key; | ||
| 621 | u64 next_offset; | ||
| 622 | u64 total_bytes = 0; | ||
| 623 | int found_next; | ||
| 624 | struct btrfs_path *path; | ||
| 625 | struct btrfs_csum_item *item; | ||
| 626 | struct btrfs_csum_item *item_end; | ||
| 627 | struct extent_buffer *leaf = NULL; | ||
| 628 | u64 csum_offset; | ||
| 629 | struct btrfs_sector_sum *sector_sum; | ||
| 630 | u32 nritems; | ||
| 631 | u32 ins_size; | ||
| 632 | char *eb_map; | ||
| 633 | char *eb_token; | ||
| 634 | unsigned long map_len; | ||
| 635 | unsigned long map_start; | ||
| 636 | u16 csum_size = | ||
| 637 | btrfs_super_csum_size(&root->fs_info->super_copy); | ||
| 638 | |||
| 639 | path = btrfs_alloc_path(); | ||
| 640 | BUG_ON(!path); | ||
| 641 | sector_sum = sums->sums; | ||
| 642 | again: | ||
| 643 | next_offset = (u64)-1; | ||
| 644 | found_next = 0; | ||
| 645 | file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID; | ||
| 646 | file_key.offset = sector_sum->bytenr; | ||
| 647 | bytenr = sector_sum->bytenr; | ||
| 648 | btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY); | ||
| 649 | |||
| 650 | item = btrfs_lookup_csum(trans, root, path, sector_sum->bytenr, 1); | ||
| 651 | if (!IS_ERR(item)) { | ||
| 652 | leaf = path->nodes[0]; | ||
| 653 | ret = 0; | ||
| 654 | goto found; | ||
| 655 | } | ||
| 656 | ret = PTR_ERR(item); | ||
| 657 | if (ret == -EFBIG) { | ||
| 658 | u32 item_size; | ||
| 659 | /* we found one, but it isn't big enough yet */ | ||
| 660 | leaf = path->nodes[0]; | ||
| 661 | item_size = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 662 | if ((item_size / csum_size) >= | ||
| 663 | MAX_CSUM_ITEMS(root, csum_size)) { | ||
| 664 | /* already at max size, make a new one */ | ||
| 665 | goto insert; | ||
| 666 | } | ||
| 667 | } else { | ||
| 668 | int slot = path->slots[0] + 1; | ||
| 669 | /* we didn't find a csum item, insert one */ | ||
| 670 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 671 | if (path->slots[0] >= nritems - 1) { | ||
| 672 | ret = btrfs_next_leaf(root, path); | ||
| 673 | if (ret == 1) | ||
| 674 | found_next = 1; | ||
| 675 | if (ret != 0) | ||
| 676 | goto insert; | ||
| 677 | slot = 0; | ||
| 678 | } | ||
| 679 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, slot); | ||
| 680 | if (found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || | ||
| 681 | found_key.type != BTRFS_EXTENT_CSUM_KEY) { | ||
| 682 | found_next = 1; | ||
| 683 | goto insert; | ||
| 684 | } | ||
| 685 | next_offset = found_key.offset; | ||
| 686 | found_next = 1; | ||
| 687 | goto insert; | ||
| 688 | } | ||
| 689 | |||
| 690 | /* | ||
| 691 | * at this point, we know the tree has an item, but it isn't big | ||
| 692 | * enough yet to put our csum in. Grow it | ||
| 693 | */ | ||
| 694 | btrfs_release_path(root, path); | ||
| 695 | ret = btrfs_search_slot(trans, root, &file_key, path, | ||
| 696 | csum_size, 1); | ||
| 697 | if (ret < 0) | ||
| 698 | goto fail_unlock; | ||
| 699 | |||
| 700 | if (ret > 0) { | ||
| 701 | if (path->slots[0] == 0) | ||
| 702 | goto insert; | ||
| 703 | path->slots[0]--; | ||
| 704 | } | ||
| 705 | |||
| 706 | leaf = path->nodes[0]; | ||
| 707 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 708 | csum_offset = (bytenr - found_key.offset) >> | ||
| 709 | root->fs_info->sb->s_blocksize_bits; | ||
| 710 | |||
| 711 | if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY || | ||
| 712 | found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID || | ||
| 713 | csum_offset >= MAX_CSUM_ITEMS(root, csum_size)) { | ||
| 714 | goto insert; | ||
| 715 | } | ||
| 716 | |||
| 717 | if (csum_offset >= btrfs_item_size_nr(leaf, path->slots[0]) / | ||
| 718 | csum_size) { | ||
| 719 | u32 diff = (csum_offset + 1) * csum_size; | ||
| 720 | |||
| 721 | /* | ||
| 722 | * is the item big enough already? we dropped our lock | ||
| 723 | * before and need to recheck | ||
| 724 | */ | ||
| 725 | if (diff < btrfs_item_size_nr(leaf, path->slots[0])) | ||
| 726 | goto csum; | ||
| 727 | |||
| 728 | diff = diff - btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 729 | if (diff != csum_size) | ||
| 730 | goto insert; | ||
| 731 | |||
| 732 | ret = btrfs_extend_item(trans, root, path, diff); | ||
| 733 | BUG_ON(ret); | ||
| 734 | goto csum; | ||
| 735 | } | ||
| 736 | |||
| 737 | insert: | ||
| 738 | btrfs_release_path(root, path); | ||
| 739 | csum_offset = 0; | ||
| 740 | if (found_next) { | ||
| 741 | u64 tmp = total_bytes + root->sectorsize; | ||
| 742 | u64 next_sector = sector_sum->bytenr; | ||
| 743 | struct btrfs_sector_sum *next = sector_sum + 1; | ||
| 744 | |||
| 745 | while (tmp < sums->len) { | ||
| 746 | if (next_sector + root->sectorsize != next->bytenr) | ||
| 747 | break; | ||
| 748 | tmp += root->sectorsize; | ||
| 749 | next_sector = next->bytenr; | ||
| 750 | next++; | ||
| 751 | } | ||
| 752 | tmp = min(tmp, next_offset - file_key.offset); | ||
| 753 | tmp >>= root->fs_info->sb->s_blocksize_bits; | ||
| 754 | tmp = max((u64)1, tmp); | ||
| 755 | tmp = min(tmp, (u64)MAX_CSUM_ITEMS(root, csum_size)); | ||
| 756 | ins_size = csum_size * tmp; | ||
| 757 | } else { | ||
| 758 | ins_size = csum_size; | ||
| 759 | } | ||
| 760 | ret = btrfs_insert_empty_item(trans, root, path, &file_key, | ||
| 761 | ins_size); | ||
| 762 | if (ret < 0) | ||
| 763 | goto fail_unlock; | ||
| 764 | if (ret != 0) { | ||
| 765 | WARN_ON(1); | ||
| 766 | goto fail_unlock; | ||
| 767 | } | ||
| 768 | csum: | ||
| 769 | leaf = path->nodes[0]; | ||
| 770 | item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item); | ||
| 771 | ret = 0; | ||
| 772 | item = (struct btrfs_csum_item *)((unsigned char *)item + | ||
| 773 | csum_offset * csum_size); | ||
| 774 | found: | ||
| 775 | item_end = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item); | ||
| 776 | item_end = (struct btrfs_csum_item *)((unsigned char *)item_end + | ||
| 777 | btrfs_item_size_nr(leaf, path->slots[0])); | ||
| 778 | eb_token = NULL; | ||
| 779 | cond_resched(); | ||
| 780 | next_sector: | ||
| 781 | |||
| 782 | if (!eb_token || | ||
| 783 | (unsigned long)item + csum_size >= map_start + map_len) { | ||
| 784 | int err; | ||
| 785 | |||
| 786 | if (eb_token) | ||
| 787 | unmap_extent_buffer(leaf, eb_token, KM_USER1); | ||
| 788 | eb_token = NULL; | ||
| 789 | err = map_private_extent_buffer(leaf, (unsigned long)item, | ||
| 790 | csum_size, | ||
| 791 | &eb_token, &eb_map, | ||
| 792 | &map_start, &map_len, KM_USER1); | ||
| 793 | if (err) | ||
| 794 | eb_token = NULL; | ||
| 795 | } | ||
| 796 | if (eb_token) { | ||
| 797 | memcpy(eb_token + ((unsigned long)item & (PAGE_CACHE_SIZE - 1)), | ||
| 798 | §or_sum->sum, csum_size); | ||
| 799 | } else { | ||
| 800 | write_extent_buffer(leaf, §or_sum->sum, | ||
| 801 | (unsigned long)item, csum_size); | ||
| 802 | } | ||
| 803 | |||
| 804 | total_bytes += root->sectorsize; | ||
| 805 | sector_sum++; | ||
| 806 | if (total_bytes < sums->len) { | ||
| 807 | item = (struct btrfs_csum_item *)((char *)item + | ||
| 808 | csum_size); | ||
| 809 | if (item < item_end && bytenr + PAGE_CACHE_SIZE == | ||
| 810 | sector_sum->bytenr) { | ||
| 811 | bytenr = sector_sum->bytenr; | ||
| 812 | goto next_sector; | ||
| 813 | } | ||
| 814 | } | ||
| 815 | if (eb_token) { | ||
| 816 | unmap_extent_buffer(leaf, eb_token, KM_USER1); | ||
| 817 | eb_token = NULL; | ||
| 818 | } | ||
| 819 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 820 | cond_resched(); | ||
| 821 | if (total_bytes < sums->len) { | ||
| 822 | btrfs_release_path(root, path); | ||
| 823 | goto again; | ||
| 824 | } | ||
| 825 | out: | ||
| 826 | btrfs_free_path(path); | ||
| 827 | return ret; | ||
| 828 | |||
| 829 | fail_unlock: | ||
| 830 | goto out; | ||
| 831 | } | ||
diff --git a/fs/btrfs/file.c b/fs/btrfs/file.c new file mode 100644 index 000000000000..90268334145e --- /dev/null +++ b/fs/btrfs/file.c | |||
| @@ -0,0 +1,1288 @@ | |||
| 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 | |||
| 19 | #include <linux/fs.h> | ||
| 20 | #include <linux/pagemap.h> | ||
| 21 | #include <linux/highmem.h> | ||
| 22 | #include <linux/time.h> | ||
| 23 | #include <linux/init.h> | ||
| 24 | #include <linux/string.h> | ||
| 25 | #include <linux/smp_lock.h> | ||
| 26 | #include <linux/backing-dev.h> | ||
| 27 | #include <linux/mpage.h> | ||
| 28 | #include <linux/swap.h> | ||
| 29 | #include <linux/writeback.h> | ||
| 30 | #include <linux/statfs.h> | ||
| 31 | #include <linux/compat.h> | ||
| 32 | #include <linux/version.h> | ||
| 33 | #include "ctree.h" | ||
| 34 | #include "disk-io.h" | ||
| 35 | #include "transaction.h" | ||
| 36 | #include "btrfs_inode.h" | ||
| 37 | #include "ioctl.h" | ||
| 38 | #include "print-tree.h" | ||
| 39 | #include "tree-log.h" | ||
| 40 | #include "locking.h" | ||
| 41 | #include "compat.h" | ||
| 42 | |||
| 43 | |||
| 44 | /* simple helper to fault in pages and copy. This should go away | ||
| 45 | * and be replaced with calls into generic code. | ||
| 46 | */ | ||
| 47 | static noinline int btrfs_copy_from_user(loff_t pos, int num_pages, | ||
| 48 | int write_bytes, | ||
| 49 | struct page **prepared_pages, | ||
| 50 | const char __user *buf) | ||
| 51 | { | ||
| 52 | long page_fault = 0; | ||
| 53 | int i; | ||
| 54 | int offset = pos & (PAGE_CACHE_SIZE - 1); | ||
| 55 | |||
| 56 | for (i = 0; i < num_pages && write_bytes > 0; i++, offset = 0) { | ||
| 57 | size_t count = min_t(size_t, | ||
| 58 | PAGE_CACHE_SIZE - offset, write_bytes); | ||
| 59 | struct page *page = prepared_pages[i]; | ||
| 60 | fault_in_pages_readable(buf, count); | ||
| 61 | |||
| 62 | /* Copy data from userspace to the current page */ | ||
| 63 | kmap(page); | ||
| 64 | page_fault = __copy_from_user(page_address(page) + offset, | ||
| 65 | buf, count); | ||
| 66 | /* Flush processor's dcache for this page */ | ||
| 67 | flush_dcache_page(page); | ||
| 68 | kunmap(page); | ||
| 69 | buf += count; | ||
| 70 | write_bytes -= count; | ||
| 71 | |||
| 72 | if (page_fault) | ||
| 73 | break; | ||
| 74 | } | ||
| 75 | return page_fault ? -EFAULT : 0; | ||
| 76 | } | ||
| 77 | |||
| 78 | /* | ||
| 79 | * unlocks pages after btrfs_file_write is done with them | ||
| 80 | */ | ||
| 81 | static noinline void btrfs_drop_pages(struct page **pages, size_t num_pages) | ||
| 82 | { | ||
| 83 | size_t i; | ||
| 84 | for (i = 0; i < num_pages; i++) { | ||
| 85 | if (!pages[i]) | ||
| 86 | break; | ||
| 87 | /* page checked is some magic around finding pages that | ||
| 88 | * have been modified without going through btrfs_set_page_dirty | ||
| 89 | * clear it here | ||
| 90 | */ | ||
| 91 | ClearPageChecked(pages[i]); | ||
| 92 | unlock_page(pages[i]); | ||
| 93 | mark_page_accessed(pages[i]); | ||
| 94 | page_cache_release(pages[i]); | ||
| 95 | } | ||
| 96 | } | ||
| 97 | |||
| 98 | /* | ||
| 99 | * after copy_from_user, pages need to be dirtied and we need to make | ||
| 100 | * sure holes are created between the current EOF and the start of | ||
| 101 | * any next extents (if required). | ||
| 102 | * | ||
| 103 | * this also makes the decision about creating an inline extent vs | ||
| 104 | * doing real data extents, marking pages dirty and delalloc as required. | ||
| 105 | */ | ||
| 106 | static noinline int dirty_and_release_pages(struct btrfs_trans_handle *trans, | ||
| 107 | struct btrfs_root *root, | ||
| 108 | struct file *file, | ||
| 109 | struct page **pages, | ||
| 110 | size_t num_pages, | ||
| 111 | loff_t pos, | ||
| 112 | size_t write_bytes) | ||
| 113 | { | ||
| 114 | int err = 0; | ||
| 115 | int i; | ||
| 116 | struct inode *inode = fdentry(file)->d_inode; | ||
| 117 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 118 | u64 hint_byte; | ||
| 119 | u64 num_bytes; | ||
| 120 | u64 start_pos; | ||
| 121 | u64 end_of_last_block; | ||
| 122 | u64 end_pos = pos + write_bytes; | ||
| 123 | loff_t isize = i_size_read(inode); | ||
| 124 | |||
| 125 | start_pos = pos & ~((u64)root->sectorsize - 1); | ||
| 126 | num_bytes = (write_bytes + pos - start_pos + | ||
| 127 | root->sectorsize - 1) & ~((u64)root->sectorsize - 1); | ||
| 128 | |||
| 129 | end_of_last_block = start_pos + num_bytes - 1; | ||
| 130 | |||
| 131 | lock_extent(io_tree, start_pos, end_of_last_block, GFP_NOFS); | ||
| 132 | trans = btrfs_join_transaction(root, 1); | ||
| 133 | if (!trans) { | ||
| 134 | err = -ENOMEM; | ||
| 135 | goto out_unlock; | ||
| 136 | } | ||
| 137 | btrfs_set_trans_block_group(trans, inode); | ||
| 138 | hint_byte = 0; | ||
| 139 | |||
| 140 | set_extent_uptodate(io_tree, start_pos, end_of_last_block, GFP_NOFS); | ||
| 141 | |||
| 142 | /* check for reserved extents on each page, we don't want | ||
| 143 | * to reset the delalloc bit on things that already have | ||
| 144 | * extents reserved. | ||
| 145 | */ | ||
| 146 | btrfs_set_extent_delalloc(inode, start_pos, end_of_last_block); | ||
| 147 | for (i = 0; i < num_pages; i++) { | ||
| 148 | struct page *p = pages[i]; | ||
| 149 | SetPageUptodate(p); | ||
| 150 | ClearPageChecked(p); | ||
| 151 | set_page_dirty(p); | ||
| 152 | } | ||
| 153 | if (end_pos > isize) { | ||
| 154 | i_size_write(inode, end_pos); | ||
| 155 | btrfs_update_inode(trans, root, inode); | ||
| 156 | } | ||
| 157 | err = btrfs_end_transaction(trans, root); | ||
| 158 | out_unlock: | ||
| 159 | unlock_extent(io_tree, start_pos, end_of_last_block, GFP_NOFS); | ||
| 160 | return err; | ||
| 161 | } | ||
| 162 | |||
| 163 | /* | ||
| 164 | * this drops all the extents in the cache that intersect the range | ||
| 165 | * [start, end]. Existing extents are split as required. | ||
| 166 | */ | ||
| 167 | int btrfs_drop_extent_cache(struct inode *inode, u64 start, u64 end, | ||
| 168 | int skip_pinned) | ||
| 169 | { | ||
| 170 | struct extent_map *em; | ||
| 171 | struct extent_map *split = NULL; | ||
| 172 | struct extent_map *split2 = NULL; | ||
| 173 | struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 174 | u64 len = end - start + 1; | ||
| 175 | int ret; | ||
| 176 | int testend = 1; | ||
| 177 | unsigned long flags; | ||
| 178 | int compressed = 0; | ||
| 179 | |||
| 180 | WARN_ON(end < start); | ||
| 181 | if (end == (u64)-1) { | ||
| 182 | len = (u64)-1; | ||
| 183 | testend = 0; | ||
| 184 | } | ||
| 185 | while (1) { | ||
| 186 | if (!split) | ||
| 187 | split = alloc_extent_map(GFP_NOFS); | ||
| 188 | if (!split2) | ||
| 189 | split2 = alloc_extent_map(GFP_NOFS); | ||
| 190 | |||
| 191 | spin_lock(&em_tree->lock); | ||
| 192 | em = lookup_extent_mapping(em_tree, start, len); | ||
| 193 | if (!em) { | ||
| 194 | spin_unlock(&em_tree->lock); | ||
| 195 | break; | ||
| 196 | } | ||
| 197 | flags = em->flags; | ||
| 198 | if (skip_pinned && test_bit(EXTENT_FLAG_PINNED, &em->flags)) { | ||
| 199 | spin_unlock(&em_tree->lock); | ||
| 200 | if (em->start <= start && | ||
| 201 | (!testend || em->start + em->len >= start + len)) { | ||
| 202 | free_extent_map(em); | ||
| 203 | break; | ||
| 204 | } | ||
| 205 | if (start < em->start) { | ||
| 206 | len = em->start - start; | ||
| 207 | } else { | ||
| 208 | len = start + len - (em->start + em->len); | ||
| 209 | start = em->start + em->len; | ||
| 210 | } | ||
| 211 | free_extent_map(em); | ||
| 212 | continue; | ||
| 213 | } | ||
| 214 | compressed = test_bit(EXTENT_FLAG_COMPRESSED, &em->flags); | ||
| 215 | clear_bit(EXTENT_FLAG_PINNED, &em->flags); | ||
| 216 | remove_extent_mapping(em_tree, em); | ||
| 217 | |||
| 218 | if (em->block_start < EXTENT_MAP_LAST_BYTE && | ||
| 219 | em->start < start) { | ||
| 220 | split->start = em->start; | ||
| 221 | split->len = start - em->start; | ||
| 222 | split->orig_start = em->orig_start; | ||
| 223 | split->block_start = em->block_start; | ||
| 224 | |||
| 225 | if (compressed) | ||
| 226 | split->block_len = em->block_len; | ||
| 227 | else | ||
| 228 | split->block_len = split->len; | ||
| 229 | |||
| 230 | split->bdev = em->bdev; | ||
| 231 | split->flags = flags; | ||
| 232 | ret = add_extent_mapping(em_tree, split); | ||
| 233 | BUG_ON(ret); | ||
| 234 | free_extent_map(split); | ||
| 235 | split = split2; | ||
| 236 | split2 = NULL; | ||
| 237 | } | ||
| 238 | if (em->block_start < EXTENT_MAP_LAST_BYTE && | ||
| 239 | testend && em->start + em->len > start + len) { | ||
| 240 | u64 diff = start + len - em->start; | ||
| 241 | |||
| 242 | split->start = start + len; | ||
| 243 | split->len = em->start + em->len - (start + len); | ||
| 244 | split->bdev = em->bdev; | ||
| 245 | split->flags = flags; | ||
| 246 | |||
| 247 | if (compressed) { | ||
| 248 | split->block_len = em->block_len; | ||
| 249 | split->block_start = em->block_start; | ||
| 250 | split->orig_start = em->orig_start; | ||
| 251 | } else { | ||
| 252 | split->block_len = split->len; | ||
| 253 | split->block_start = em->block_start + diff; | ||
| 254 | split->orig_start = split->start; | ||
| 255 | } | ||
| 256 | |||
| 257 | ret = add_extent_mapping(em_tree, split); | ||
| 258 | BUG_ON(ret); | ||
| 259 | free_extent_map(split); | ||
| 260 | split = NULL; | ||
| 261 | } | ||
| 262 | spin_unlock(&em_tree->lock); | ||
| 263 | |||
| 264 | /* once for us */ | ||
| 265 | free_extent_map(em); | ||
| 266 | /* once for the tree*/ | ||
| 267 | free_extent_map(em); | ||
| 268 | } | ||
| 269 | if (split) | ||
| 270 | free_extent_map(split); | ||
| 271 | if (split2) | ||
| 272 | free_extent_map(split2); | ||
| 273 | return 0; | ||
| 274 | } | ||
| 275 | |||
| 276 | int btrfs_check_file(struct btrfs_root *root, struct inode *inode) | ||
| 277 | { | ||
| 278 | return 0; | ||
| 279 | #if 0 | ||
| 280 | struct btrfs_path *path; | ||
| 281 | struct btrfs_key found_key; | ||
| 282 | struct extent_buffer *leaf; | ||
| 283 | struct btrfs_file_extent_item *extent; | ||
| 284 | u64 last_offset = 0; | ||
| 285 | int nritems; | ||
| 286 | int slot; | ||
| 287 | int found_type; | ||
| 288 | int ret; | ||
| 289 | int err = 0; | ||
| 290 | u64 extent_end = 0; | ||
| 291 | |||
| 292 | path = btrfs_alloc_path(); | ||
| 293 | ret = btrfs_lookup_file_extent(NULL, root, path, inode->i_ino, | ||
| 294 | last_offset, 0); | ||
| 295 | while (1) { | ||
| 296 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 297 | if (path->slots[0] >= nritems) { | ||
| 298 | ret = btrfs_next_leaf(root, path); | ||
| 299 | if (ret) | ||
| 300 | goto out; | ||
| 301 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 302 | } | ||
| 303 | slot = path->slots[0]; | ||
| 304 | leaf = path->nodes[0]; | ||
| 305 | btrfs_item_key_to_cpu(leaf, &found_key, slot); | ||
| 306 | if (found_key.objectid != inode->i_ino) | ||
| 307 | break; | ||
| 308 | if (found_key.type != BTRFS_EXTENT_DATA_KEY) | ||
| 309 | goto out; | ||
| 310 | |||
| 311 | if (found_key.offset < last_offset) { | ||
| 312 | WARN_ON(1); | ||
| 313 | btrfs_print_leaf(root, leaf); | ||
| 314 | printk(KERN_ERR "inode %lu found offset %llu " | ||
| 315 | "expected %llu\n", inode->i_ino, | ||
| 316 | (unsigned long long)found_key.offset, | ||
| 317 | (unsigned long long)last_offset); | ||
| 318 | err = 1; | ||
| 319 | goto out; | ||
| 320 | } | ||
| 321 | extent = btrfs_item_ptr(leaf, slot, | ||
| 322 | struct btrfs_file_extent_item); | ||
| 323 | found_type = btrfs_file_extent_type(leaf, extent); | ||
| 324 | if (found_type == BTRFS_FILE_EXTENT_REG) { | ||
| 325 | extent_end = found_key.offset + | ||
| 326 | btrfs_file_extent_num_bytes(leaf, extent); | ||
| 327 | } else if (found_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 328 | struct btrfs_item *item; | ||
| 329 | item = btrfs_item_nr(leaf, slot); | ||
| 330 | extent_end = found_key.offset + | ||
| 331 | btrfs_file_extent_inline_len(leaf, extent); | ||
| 332 | extent_end = (extent_end + root->sectorsize - 1) & | ||
| 333 | ~((u64)root->sectorsize - 1); | ||
| 334 | } | ||
| 335 | last_offset = extent_end; | ||
| 336 | path->slots[0]++; | ||
| 337 | } | ||
| 338 | if (0 && last_offset < inode->i_size) { | ||
| 339 | WARN_ON(1); | ||
| 340 | btrfs_print_leaf(root, leaf); | ||
| 341 | printk(KERN_ERR "inode %lu found offset %llu size %llu\n", | ||
| 342 | inode->i_ino, (unsigned long long)last_offset, | ||
| 343 | (unsigned long long)inode->i_size); | ||
| 344 | err = 1; | ||
| 345 | |||
| 346 | } | ||
| 347 | out: | ||
| 348 | btrfs_free_path(path); | ||
| 349 | return err; | ||
| 350 | #endif | ||
| 351 | } | ||
| 352 | |||
| 353 | /* | ||
| 354 | * this is very complex, but the basic idea is to drop all extents | ||
| 355 | * in the range start - end. hint_block is filled in with a block number | ||
| 356 | * that would be a good hint to the block allocator for this file. | ||
| 357 | * | ||
| 358 | * If an extent intersects the range but is not entirely inside the range | ||
| 359 | * it is either truncated or split. Anything entirely inside the range | ||
| 360 | * is deleted from the tree. | ||
| 361 | * | ||
| 362 | * inline_limit is used to tell this code which offsets in the file to keep | ||
| 363 | * if they contain inline extents. | ||
| 364 | */ | ||
| 365 | noinline int btrfs_drop_extents(struct btrfs_trans_handle *trans, | ||
| 366 | struct btrfs_root *root, struct inode *inode, | ||
| 367 | u64 start, u64 end, u64 inline_limit, u64 *hint_byte) | ||
| 368 | { | ||
| 369 | u64 extent_end = 0; | ||
| 370 | u64 locked_end = end; | ||
| 371 | u64 search_start = start; | ||
| 372 | u64 leaf_start; | ||
| 373 | u64 ram_bytes = 0; | ||
| 374 | u64 orig_parent = 0; | ||
| 375 | u64 disk_bytenr = 0; | ||
| 376 | u8 compression; | ||
| 377 | u8 encryption; | ||
| 378 | u16 other_encoding = 0; | ||
| 379 | u64 root_gen; | ||
| 380 | u64 root_owner; | ||
| 381 | struct extent_buffer *leaf; | ||
| 382 | struct btrfs_file_extent_item *extent; | ||
| 383 | struct btrfs_path *path; | ||
| 384 | struct btrfs_key key; | ||
| 385 | struct btrfs_file_extent_item old; | ||
| 386 | int keep; | ||
| 387 | int slot; | ||
| 388 | int bookend; | ||
| 389 | int found_type = 0; | ||
| 390 | int found_extent; | ||
| 391 | int found_inline; | ||
| 392 | int recow; | ||
| 393 | int ret; | ||
| 394 | |||
| 395 | inline_limit = 0; | ||
| 396 | btrfs_drop_extent_cache(inode, start, end - 1, 0); | ||
| 397 | |||
| 398 | path = btrfs_alloc_path(); | ||
| 399 | if (!path) | ||
| 400 | return -ENOMEM; | ||
| 401 | while (1) { | ||
| 402 | recow = 0; | ||
| 403 | btrfs_release_path(root, path); | ||
| 404 | ret = btrfs_lookup_file_extent(trans, root, path, inode->i_ino, | ||
| 405 | search_start, -1); | ||
| 406 | if (ret < 0) | ||
| 407 | goto out; | ||
| 408 | if (ret > 0) { | ||
| 409 | if (path->slots[0] == 0) { | ||
| 410 | ret = 0; | ||
| 411 | goto out; | ||
| 412 | } | ||
| 413 | path->slots[0]--; | ||
| 414 | } | ||
| 415 | next_slot: | ||
| 416 | keep = 0; | ||
| 417 | bookend = 0; | ||
| 418 | found_extent = 0; | ||
| 419 | found_inline = 0; | ||
| 420 | leaf_start = 0; | ||
| 421 | root_gen = 0; | ||
| 422 | root_owner = 0; | ||
| 423 | compression = 0; | ||
| 424 | encryption = 0; | ||
| 425 | extent = NULL; | ||
| 426 | leaf = path->nodes[0]; | ||
| 427 | slot = path->slots[0]; | ||
| 428 | ret = 0; | ||
| 429 | btrfs_item_key_to_cpu(leaf, &key, slot); | ||
| 430 | if (btrfs_key_type(&key) == BTRFS_EXTENT_DATA_KEY && | ||
| 431 | key.offset >= end) { | ||
| 432 | goto out; | ||
| 433 | } | ||
| 434 | if (btrfs_key_type(&key) > BTRFS_EXTENT_DATA_KEY || | ||
| 435 | key.objectid != inode->i_ino) { | ||
| 436 | goto out; | ||
| 437 | } | ||
| 438 | if (recow) { | ||
| 439 | search_start = max(key.offset, start); | ||
| 440 | continue; | ||
| 441 | } | ||
| 442 | if (btrfs_key_type(&key) == BTRFS_EXTENT_DATA_KEY) { | ||
| 443 | extent = btrfs_item_ptr(leaf, slot, | ||
| 444 | struct btrfs_file_extent_item); | ||
| 445 | found_type = btrfs_file_extent_type(leaf, extent); | ||
| 446 | compression = btrfs_file_extent_compression(leaf, | ||
| 447 | extent); | ||
| 448 | encryption = btrfs_file_extent_encryption(leaf, | ||
| 449 | extent); | ||
| 450 | other_encoding = btrfs_file_extent_other_encoding(leaf, | ||
| 451 | extent); | ||
| 452 | if (found_type == BTRFS_FILE_EXTENT_REG || | ||
| 453 | found_type == BTRFS_FILE_EXTENT_PREALLOC) { | ||
| 454 | extent_end = | ||
| 455 | btrfs_file_extent_disk_bytenr(leaf, | ||
| 456 | extent); | ||
| 457 | if (extent_end) | ||
| 458 | *hint_byte = extent_end; | ||
| 459 | |||
| 460 | extent_end = key.offset + | ||
| 461 | btrfs_file_extent_num_bytes(leaf, extent); | ||
| 462 | ram_bytes = btrfs_file_extent_ram_bytes(leaf, | ||
| 463 | extent); | ||
| 464 | found_extent = 1; | ||
| 465 | } else if (found_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 466 | found_inline = 1; | ||
| 467 | extent_end = key.offset + | ||
| 468 | btrfs_file_extent_inline_len(leaf, extent); | ||
| 469 | } | ||
| 470 | } else { | ||
| 471 | extent_end = search_start; | ||
| 472 | } | ||
| 473 | |||
| 474 | /* we found nothing we can drop */ | ||
| 475 | if ((!found_extent && !found_inline) || | ||
| 476 | search_start >= extent_end) { | ||
| 477 | int nextret; | ||
| 478 | u32 nritems; | ||
| 479 | nritems = btrfs_header_nritems(leaf); | ||
| 480 | if (slot >= nritems - 1) { | ||
| 481 | nextret = btrfs_next_leaf(root, path); | ||
| 482 | if (nextret) | ||
| 483 | goto out; | ||
| 484 | recow = 1; | ||
| 485 | } else { | ||
| 486 | path->slots[0]++; | ||
| 487 | } | ||
| 488 | goto next_slot; | ||
| 489 | } | ||
| 490 | |||
| 491 | if (end <= extent_end && start >= key.offset && found_inline) | ||
| 492 | *hint_byte = EXTENT_MAP_INLINE; | ||
| 493 | |||
| 494 | if (found_extent) { | ||
| 495 | read_extent_buffer(leaf, &old, (unsigned long)extent, | ||
| 496 | sizeof(old)); | ||
| 497 | root_gen = btrfs_header_generation(leaf); | ||
| 498 | root_owner = btrfs_header_owner(leaf); | ||
| 499 | leaf_start = leaf->start; | ||
| 500 | } | ||
| 501 | |||
| 502 | if (end < extent_end && end >= key.offset) { | ||
| 503 | bookend = 1; | ||
| 504 | if (found_inline && start <= key.offset) | ||
| 505 | keep = 1; | ||
| 506 | } | ||
| 507 | |||
| 508 | if (bookend && found_extent) { | ||
| 509 | if (locked_end < extent_end) { | ||
| 510 | ret = try_lock_extent(&BTRFS_I(inode)->io_tree, | ||
| 511 | locked_end, extent_end - 1, | ||
| 512 | GFP_NOFS); | ||
| 513 | if (!ret) { | ||
| 514 | btrfs_release_path(root, path); | ||
| 515 | lock_extent(&BTRFS_I(inode)->io_tree, | ||
| 516 | locked_end, extent_end - 1, | ||
| 517 | GFP_NOFS); | ||
| 518 | locked_end = extent_end; | ||
| 519 | continue; | ||
| 520 | } | ||
| 521 | locked_end = extent_end; | ||
| 522 | } | ||
| 523 | orig_parent = path->nodes[0]->start; | ||
| 524 | disk_bytenr = le64_to_cpu(old.disk_bytenr); | ||
| 525 | if (disk_bytenr != 0) { | ||
| 526 | ret = btrfs_inc_extent_ref(trans, root, | ||
| 527 | disk_bytenr, | ||
| 528 | le64_to_cpu(old.disk_num_bytes), | ||
| 529 | orig_parent, root->root_key.objectid, | ||
| 530 | trans->transid, inode->i_ino); | ||
| 531 | BUG_ON(ret); | ||
| 532 | } | ||
| 533 | } | ||
| 534 | |||
| 535 | if (found_inline) { | ||
| 536 | u64 mask = root->sectorsize - 1; | ||
| 537 | search_start = (extent_end + mask) & ~mask; | ||
| 538 | } else | ||
| 539 | search_start = extent_end; | ||
| 540 | |||
| 541 | /* truncate existing extent */ | ||
| 542 | if (start > key.offset) { | ||
| 543 | u64 new_num; | ||
| 544 | u64 old_num; | ||
| 545 | keep = 1; | ||
| 546 | WARN_ON(start & (root->sectorsize - 1)); | ||
| 547 | if (found_extent) { | ||
| 548 | new_num = start - key.offset; | ||
| 549 | old_num = btrfs_file_extent_num_bytes(leaf, | ||
| 550 | extent); | ||
| 551 | *hint_byte = | ||
| 552 | btrfs_file_extent_disk_bytenr(leaf, | ||
| 553 | extent); | ||
| 554 | if (btrfs_file_extent_disk_bytenr(leaf, | ||
| 555 | extent)) { | ||
| 556 | inode_sub_bytes(inode, old_num - | ||
| 557 | new_num); | ||
| 558 | } | ||
| 559 | btrfs_set_file_extent_num_bytes(leaf, | ||
| 560 | extent, new_num); | ||
| 561 | btrfs_mark_buffer_dirty(leaf); | ||
| 562 | } else if (key.offset < inline_limit && | ||
| 563 | (end > extent_end) && | ||
| 564 | (inline_limit < extent_end)) { | ||
| 565 | u32 new_size; | ||
| 566 | new_size = btrfs_file_extent_calc_inline_size( | ||
| 567 | inline_limit - key.offset); | ||
| 568 | inode_sub_bytes(inode, extent_end - | ||
| 569 | inline_limit); | ||
| 570 | btrfs_set_file_extent_ram_bytes(leaf, extent, | ||
| 571 | new_size); | ||
| 572 | if (!compression && !encryption) { | ||
| 573 | btrfs_truncate_item(trans, root, path, | ||
| 574 | new_size, 1); | ||
| 575 | } | ||
| 576 | } | ||
| 577 | } | ||
| 578 | /* delete the entire extent */ | ||
| 579 | if (!keep) { | ||
| 580 | if (found_inline) | ||
| 581 | inode_sub_bytes(inode, extent_end - | ||
| 582 | key.offset); | ||
| 583 | ret = btrfs_del_item(trans, root, path); | ||
| 584 | /* TODO update progress marker and return */ | ||
| 585 | BUG_ON(ret); | ||
| 586 | extent = NULL; | ||
| 587 | btrfs_release_path(root, path); | ||
| 588 | /* the extent will be freed later */ | ||
| 589 | } | ||
| 590 | if (bookend && found_inline && start <= key.offset) { | ||
| 591 | u32 new_size; | ||
| 592 | new_size = btrfs_file_extent_calc_inline_size( | ||
| 593 | extent_end - end); | ||
| 594 | inode_sub_bytes(inode, end - key.offset); | ||
| 595 | btrfs_set_file_extent_ram_bytes(leaf, extent, | ||
| 596 | new_size); | ||
| 597 | if (!compression && !encryption) | ||
| 598 | ret = btrfs_truncate_item(trans, root, path, | ||
| 599 | new_size, 0); | ||
| 600 | BUG_ON(ret); | ||
| 601 | } | ||
| 602 | /* create bookend, splitting the extent in two */ | ||
| 603 | if (bookend && found_extent) { | ||
| 604 | struct btrfs_key ins; | ||
| 605 | ins.objectid = inode->i_ino; | ||
| 606 | ins.offset = end; | ||
| 607 | btrfs_set_key_type(&ins, BTRFS_EXTENT_DATA_KEY); | ||
| 608 | |||
| 609 | btrfs_release_path(root, path); | ||
| 610 | ret = btrfs_insert_empty_item(trans, root, path, &ins, | ||
| 611 | sizeof(*extent)); | ||
| 612 | BUG_ON(ret); | ||
| 613 | |||
| 614 | leaf = path->nodes[0]; | ||
| 615 | extent = btrfs_item_ptr(leaf, path->slots[0], | ||
| 616 | struct btrfs_file_extent_item); | ||
| 617 | write_extent_buffer(leaf, &old, | ||
| 618 | (unsigned long)extent, sizeof(old)); | ||
| 619 | |||
| 620 | btrfs_set_file_extent_compression(leaf, extent, | ||
| 621 | compression); | ||
| 622 | btrfs_set_file_extent_encryption(leaf, extent, | ||
| 623 | encryption); | ||
| 624 | btrfs_set_file_extent_other_encoding(leaf, extent, | ||
| 625 | other_encoding); | ||
| 626 | btrfs_set_file_extent_offset(leaf, extent, | ||
| 627 | le64_to_cpu(old.offset) + end - key.offset); | ||
| 628 | WARN_ON(le64_to_cpu(old.num_bytes) < | ||
| 629 | (extent_end - end)); | ||
| 630 | btrfs_set_file_extent_num_bytes(leaf, extent, | ||
| 631 | extent_end - end); | ||
| 632 | |||
| 633 | /* | ||
| 634 | * set the ram bytes to the size of the full extent | ||
| 635 | * before splitting. This is a worst case flag, | ||
| 636 | * but its the best we can do because we don't know | ||
| 637 | * how splitting affects compression | ||
| 638 | */ | ||
| 639 | btrfs_set_file_extent_ram_bytes(leaf, extent, | ||
| 640 | ram_bytes); | ||
| 641 | btrfs_set_file_extent_type(leaf, extent, found_type); | ||
| 642 | |||
| 643 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 644 | |||
| 645 | if (disk_bytenr != 0) { | ||
| 646 | ret = btrfs_update_extent_ref(trans, root, | ||
| 647 | disk_bytenr, orig_parent, | ||
| 648 | leaf->start, | ||
| 649 | root->root_key.objectid, | ||
| 650 | trans->transid, ins.objectid); | ||
| 651 | |||
| 652 | BUG_ON(ret); | ||
| 653 | } | ||
| 654 | btrfs_release_path(root, path); | ||
| 655 | if (disk_bytenr != 0) | ||
| 656 | inode_add_bytes(inode, extent_end - end); | ||
| 657 | } | ||
| 658 | |||
| 659 | if (found_extent && !keep) { | ||
| 660 | u64 old_disk_bytenr = le64_to_cpu(old.disk_bytenr); | ||
| 661 | |||
| 662 | if (old_disk_bytenr != 0) { | ||
| 663 | inode_sub_bytes(inode, | ||
| 664 | le64_to_cpu(old.num_bytes)); | ||
| 665 | ret = btrfs_free_extent(trans, root, | ||
| 666 | old_disk_bytenr, | ||
| 667 | le64_to_cpu(old.disk_num_bytes), | ||
| 668 | leaf_start, root_owner, | ||
| 669 | root_gen, key.objectid, 0); | ||
| 670 | BUG_ON(ret); | ||
| 671 | *hint_byte = old_disk_bytenr; | ||
| 672 | } | ||
| 673 | } | ||
| 674 | |||
| 675 | if (search_start >= end) { | ||
| 676 | ret = 0; | ||
| 677 | goto out; | ||
| 678 | } | ||
| 679 | } | ||
| 680 | out: | ||
| 681 | btrfs_free_path(path); | ||
| 682 | if (locked_end > end) { | ||
| 683 | unlock_extent(&BTRFS_I(inode)->io_tree, end, locked_end - 1, | ||
| 684 | GFP_NOFS); | ||
| 685 | } | ||
| 686 | btrfs_check_file(root, inode); | ||
| 687 | return ret; | ||
| 688 | } | ||
| 689 | |||
| 690 | static int extent_mergeable(struct extent_buffer *leaf, int slot, | ||
| 691 | u64 objectid, u64 bytenr, u64 *start, u64 *end) | ||
| 692 | { | ||
| 693 | struct btrfs_file_extent_item *fi; | ||
| 694 | struct btrfs_key key; | ||
| 695 | u64 extent_end; | ||
| 696 | |||
| 697 | if (slot < 0 || slot >= btrfs_header_nritems(leaf)) | ||
| 698 | return 0; | ||
| 699 | |||
| 700 | btrfs_item_key_to_cpu(leaf, &key, slot); | ||
| 701 | if (key.objectid != objectid || key.type != BTRFS_EXTENT_DATA_KEY) | ||
| 702 | return 0; | ||
| 703 | |||
| 704 | fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item); | ||
| 705 | if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG || | ||
| 706 | btrfs_file_extent_disk_bytenr(leaf, fi) != bytenr || | ||
| 707 | btrfs_file_extent_compression(leaf, fi) || | ||
| 708 | btrfs_file_extent_encryption(leaf, fi) || | ||
| 709 | btrfs_file_extent_other_encoding(leaf, fi)) | ||
| 710 | return 0; | ||
| 711 | |||
| 712 | extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi); | ||
| 713 | if ((*start && *start != key.offset) || (*end && *end != extent_end)) | ||
| 714 | return 0; | ||
| 715 | |||
| 716 | *start = key.offset; | ||
| 717 | *end = extent_end; | ||
| 718 | return 1; | ||
| 719 | } | ||
| 720 | |||
| 721 | /* | ||
| 722 | * Mark extent in the range start - end as written. | ||
| 723 | * | ||
| 724 | * This changes extent type from 'pre-allocated' to 'regular'. If only | ||
| 725 | * part of extent is marked as written, the extent will be split into | ||
| 726 | * two or three. | ||
| 727 | */ | ||
| 728 | int btrfs_mark_extent_written(struct btrfs_trans_handle *trans, | ||
| 729 | struct btrfs_root *root, | ||
| 730 | struct inode *inode, u64 start, u64 end) | ||
| 731 | { | ||
| 732 | struct extent_buffer *leaf; | ||
| 733 | struct btrfs_path *path; | ||
| 734 | struct btrfs_file_extent_item *fi; | ||
| 735 | struct btrfs_key key; | ||
| 736 | u64 bytenr; | ||
| 737 | u64 num_bytes; | ||
| 738 | u64 extent_end; | ||
| 739 | u64 extent_offset; | ||
| 740 | u64 other_start; | ||
| 741 | u64 other_end; | ||
| 742 | u64 split = start; | ||
| 743 | u64 locked_end = end; | ||
| 744 | u64 orig_parent; | ||
| 745 | int extent_type; | ||
| 746 | int split_end = 1; | ||
| 747 | int ret; | ||
| 748 | |||
| 749 | btrfs_drop_extent_cache(inode, start, end - 1, 0); | ||
| 750 | |||
| 751 | path = btrfs_alloc_path(); | ||
| 752 | BUG_ON(!path); | ||
| 753 | again: | ||
| 754 | key.objectid = inode->i_ino; | ||
| 755 | key.type = BTRFS_EXTENT_DATA_KEY; | ||
| 756 | if (split == start) | ||
| 757 | key.offset = split; | ||
| 758 | else | ||
| 759 | key.offset = split - 1; | ||
| 760 | |||
| 761 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 762 | if (ret > 0 && path->slots[0] > 0) | ||
| 763 | path->slots[0]--; | ||
| 764 | |||
| 765 | leaf = path->nodes[0]; | ||
| 766 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 767 | BUG_ON(key.objectid != inode->i_ino || | ||
| 768 | key.type != BTRFS_EXTENT_DATA_KEY); | ||
| 769 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 770 | struct btrfs_file_extent_item); | ||
| 771 | extent_type = btrfs_file_extent_type(leaf, fi); | ||
| 772 | BUG_ON(extent_type != BTRFS_FILE_EXTENT_PREALLOC); | ||
| 773 | extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi); | ||
| 774 | BUG_ON(key.offset > start || extent_end < end); | ||
| 775 | |||
| 776 | bytenr = btrfs_file_extent_disk_bytenr(leaf, fi); | ||
| 777 | num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi); | ||
| 778 | extent_offset = btrfs_file_extent_offset(leaf, fi); | ||
| 779 | |||
| 780 | if (key.offset == start) | ||
| 781 | split = end; | ||
| 782 | |||
| 783 | if (key.offset == start && extent_end == end) { | ||
| 784 | int del_nr = 0; | ||
| 785 | int del_slot = 0; | ||
| 786 | u64 leaf_owner = btrfs_header_owner(leaf); | ||
| 787 | u64 leaf_gen = btrfs_header_generation(leaf); | ||
| 788 | other_start = end; | ||
| 789 | other_end = 0; | ||
| 790 | if (extent_mergeable(leaf, path->slots[0] + 1, inode->i_ino, | ||
| 791 | bytenr, &other_start, &other_end)) { | ||
| 792 | extent_end = other_end; | ||
| 793 | del_slot = path->slots[0] + 1; | ||
| 794 | del_nr++; | ||
| 795 | ret = btrfs_free_extent(trans, root, bytenr, num_bytes, | ||
| 796 | leaf->start, leaf_owner, | ||
| 797 | leaf_gen, inode->i_ino, 0); | ||
| 798 | BUG_ON(ret); | ||
| 799 | } | ||
| 800 | other_start = 0; | ||
| 801 | other_end = start; | ||
| 802 | if (extent_mergeable(leaf, path->slots[0] - 1, inode->i_ino, | ||
| 803 | bytenr, &other_start, &other_end)) { | ||
| 804 | key.offset = other_start; | ||
| 805 | del_slot = path->slots[0]; | ||
| 806 | del_nr++; | ||
| 807 | ret = btrfs_free_extent(trans, root, bytenr, num_bytes, | ||
| 808 | leaf->start, leaf_owner, | ||
| 809 | leaf_gen, inode->i_ino, 0); | ||
| 810 | BUG_ON(ret); | ||
| 811 | } | ||
| 812 | split_end = 0; | ||
| 813 | if (del_nr == 0) { | ||
| 814 | btrfs_set_file_extent_type(leaf, fi, | ||
| 815 | BTRFS_FILE_EXTENT_REG); | ||
| 816 | goto done; | ||
| 817 | } | ||
| 818 | |||
| 819 | fi = btrfs_item_ptr(leaf, del_slot - 1, | ||
| 820 | struct btrfs_file_extent_item); | ||
| 821 | btrfs_set_file_extent_type(leaf, fi, BTRFS_FILE_EXTENT_REG); | ||
| 822 | btrfs_set_file_extent_num_bytes(leaf, fi, | ||
| 823 | extent_end - key.offset); | ||
| 824 | btrfs_mark_buffer_dirty(leaf); | ||
| 825 | |||
| 826 | ret = btrfs_del_items(trans, root, path, del_slot, del_nr); | ||
| 827 | BUG_ON(ret); | ||
| 828 | goto done; | ||
| 829 | } else if (split == start) { | ||
| 830 | if (locked_end < extent_end) { | ||
| 831 | ret = try_lock_extent(&BTRFS_I(inode)->io_tree, | ||
| 832 | locked_end, extent_end - 1, GFP_NOFS); | ||
| 833 | if (!ret) { | ||
| 834 | btrfs_release_path(root, path); | ||
| 835 | lock_extent(&BTRFS_I(inode)->io_tree, | ||
| 836 | locked_end, extent_end - 1, GFP_NOFS); | ||
| 837 | locked_end = extent_end; | ||
| 838 | goto again; | ||
| 839 | } | ||
| 840 | locked_end = extent_end; | ||
| 841 | } | ||
| 842 | btrfs_set_file_extent_num_bytes(leaf, fi, split - key.offset); | ||
| 843 | extent_offset += split - key.offset; | ||
| 844 | } else { | ||
| 845 | BUG_ON(key.offset != start); | ||
| 846 | btrfs_set_file_extent_offset(leaf, fi, extent_offset + | ||
| 847 | split - key.offset); | ||
| 848 | btrfs_set_file_extent_num_bytes(leaf, fi, extent_end - split); | ||
| 849 | key.offset = split; | ||
| 850 | btrfs_set_item_key_safe(trans, root, path, &key); | ||
| 851 | extent_end = split; | ||
| 852 | } | ||
| 853 | |||
| 854 | if (extent_end == end) { | ||
| 855 | split_end = 0; | ||
| 856 | extent_type = BTRFS_FILE_EXTENT_REG; | ||
| 857 | } | ||
| 858 | if (extent_end == end && split == start) { | ||
| 859 | other_start = end; | ||
| 860 | other_end = 0; | ||
| 861 | if (extent_mergeable(leaf, path->slots[0] + 1, inode->i_ino, | ||
| 862 | bytenr, &other_start, &other_end)) { | ||
| 863 | path->slots[0]++; | ||
| 864 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 865 | struct btrfs_file_extent_item); | ||
| 866 | key.offset = split; | ||
| 867 | btrfs_set_item_key_safe(trans, root, path, &key); | ||
| 868 | btrfs_set_file_extent_offset(leaf, fi, extent_offset); | ||
| 869 | btrfs_set_file_extent_num_bytes(leaf, fi, | ||
| 870 | other_end - split); | ||
| 871 | goto done; | ||
| 872 | } | ||
| 873 | } | ||
| 874 | if (extent_end == end && split == end) { | ||
| 875 | other_start = 0; | ||
| 876 | other_end = start; | ||
| 877 | if (extent_mergeable(leaf, path->slots[0] - 1 , inode->i_ino, | ||
| 878 | bytenr, &other_start, &other_end)) { | ||
| 879 | path->slots[0]--; | ||
| 880 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 881 | struct btrfs_file_extent_item); | ||
| 882 | btrfs_set_file_extent_num_bytes(leaf, fi, extent_end - | ||
| 883 | other_start); | ||
| 884 | goto done; | ||
| 885 | } | ||
| 886 | } | ||
| 887 | |||
| 888 | btrfs_mark_buffer_dirty(leaf); | ||
| 889 | |||
| 890 | orig_parent = leaf->start; | ||
| 891 | ret = btrfs_inc_extent_ref(trans, root, bytenr, num_bytes, | ||
| 892 | orig_parent, root->root_key.objectid, | ||
| 893 | trans->transid, inode->i_ino); | ||
| 894 | BUG_ON(ret); | ||
| 895 | btrfs_release_path(root, path); | ||
| 896 | |||
| 897 | key.offset = start; | ||
| 898 | ret = btrfs_insert_empty_item(trans, root, path, &key, sizeof(*fi)); | ||
| 899 | BUG_ON(ret); | ||
| 900 | |||
| 901 | leaf = path->nodes[0]; | ||
| 902 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 903 | struct btrfs_file_extent_item); | ||
| 904 | btrfs_set_file_extent_generation(leaf, fi, trans->transid); | ||
| 905 | btrfs_set_file_extent_type(leaf, fi, extent_type); | ||
| 906 | btrfs_set_file_extent_disk_bytenr(leaf, fi, bytenr); | ||
| 907 | btrfs_set_file_extent_disk_num_bytes(leaf, fi, num_bytes); | ||
| 908 | btrfs_set_file_extent_offset(leaf, fi, extent_offset); | ||
| 909 | btrfs_set_file_extent_num_bytes(leaf, fi, extent_end - key.offset); | ||
| 910 | btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes); | ||
| 911 | btrfs_set_file_extent_compression(leaf, fi, 0); | ||
| 912 | btrfs_set_file_extent_encryption(leaf, fi, 0); | ||
| 913 | btrfs_set_file_extent_other_encoding(leaf, fi, 0); | ||
| 914 | |||
| 915 | if (orig_parent != leaf->start) { | ||
| 916 | ret = btrfs_update_extent_ref(trans, root, bytenr, | ||
| 917 | orig_parent, leaf->start, | ||
| 918 | root->root_key.objectid, | ||
| 919 | trans->transid, inode->i_ino); | ||
| 920 | BUG_ON(ret); | ||
| 921 | } | ||
| 922 | done: | ||
| 923 | btrfs_mark_buffer_dirty(leaf); | ||
| 924 | btrfs_release_path(root, path); | ||
| 925 | if (split_end && split == start) { | ||
| 926 | split = end; | ||
| 927 | goto again; | ||
| 928 | } | ||
| 929 | if (locked_end > end) { | ||
| 930 | unlock_extent(&BTRFS_I(inode)->io_tree, end, locked_end - 1, | ||
| 931 | GFP_NOFS); | ||
| 932 | } | ||
| 933 | btrfs_free_path(path); | ||
| 934 | return 0; | ||
| 935 | } | ||
| 936 | |||
| 937 | /* | ||
| 938 | * this gets pages into the page cache and locks them down, it also properly | ||
| 939 | * waits for data=ordered extents to finish before allowing the pages to be | ||
| 940 | * modified. | ||
| 941 | */ | ||
| 942 | static noinline int prepare_pages(struct btrfs_root *root, struct file *file, | ||
| 943 | struct page **pages, size_t num_pages, | ||
| 944 | loff_t pos, unsigned long first_index, | ||
| 945 | unsigned long last_index, size_t write_bytes) | ||
| 946 | { | ||
| 947 | int i; | ||
| 948 | unsigned long index = pos >> PAGE_CACHE_SHIFT; | ||
| 949 | struct inode *inode = fdentry(file)->d_inode; | ||
| 950 | int err = 0; | ||
| 951 | u64 start_pos; | ||
| 952 | u64 last_pos; | ||
| 953 | |||
| 954 | start_pos = pos & ~((u64)root->sectorsize - 1); | ||
| 955 | last_pos = ((u64)index + num_pages) << PAGE_CACHE_SHIFT; | ||
| 956 | |||
| 957 | if (start_pos > inode->i_size) { | ||
| 958 | err = btrfs_cont_expand(inode, start_pos); | ||
| 959 | if (err) | ||
| 960 | return err; | ||
| 961 | } | ||
| 962 | |||
| 963 | memset(pages, 0, num_pages * sizeof(struct page *)); | ||
| 964 | again: | ||
| 965 | for (i = 0; i < num_pages; i++) { | ||
| 966 | pages[i] = grab_cache_page(inode->i_mapping, index + i); | ||
| 967 | if (!pages[i]) { | ||
| 968 | err = -ENOMEM; | ||
| 969 | BUG_ON(1); | ||
| 970 | } | ||
| 971 | wait_on_page_writeback(pages[i]); | ||
| 972 | } | ||
| 973 | if (start_pos < inode->i_size) { | ||
| 974 | struct btrfs_ordered_extent *ordered; | ||
| 975 | lock_extent(&BTRFS_I(inode)->io_tree, | ||
| 976 | start_pos, last_pos - 1, GFP_NOFS); | ||
| 977 | ordered = btrfs_lookup_first_ordered_extent(inode, | ||
| 978 | last_pos - 1); | ||
| 979 | if (ordered && | ||
| 980 | ordered->file_offset + ordered->len > start_pos && | ||
| 981 | ordered->file_offset < last_pos) { | ||
| 982 | btrfs_put_ordered_extent(ordered); | ||
| 983 | unlock_extent(&BTRFS_I(inode)->io_tree, | ||
| 984 | start_pos, last_pos - 1, GFP_NOFS); | ||
| 985 | for (i = 0; i < num_pages; i++) { | ||
| 986 | unlock_page(pages[i]); | ||
| 987 | page_cache_release(pages[i]); | ||
| 988 | } | ||
| 989 | btrfs_wait_ordered_range(inode, start_pos, | ||
| 990 | last_pos - start_pos); | ||
| 991 | goto again; | ||
| 992 | } | ||
| 993 | if (ordered) | ||
| 994 | btrfs_put_ordered_extent(ordered); | ||
| 995 | |||
| 996 | clear_extent_bits(&BTRFS_I(inode)->io_tree, start_pos, | ||
| 997 | last_pos - 1, EXTENT_DIRTY | EXTENT_DELALLOC, | ||
| 998 | GFP_NOFS); | ||
| 999 | unlock_extent(&BTRFS_I(inode)->io_tree, | ||
| 1000 | start_pos, last_pos - 1, GFP_NOFS); | ||
| 1001 | } | ||
| 1002 | for (i = 0; i < num_pages; i++) { | ||
| 1003 | clear_page_dirty_for_io(pages[i]); | ||
| 1004 | set_page_extent_mapped(pages[i]); | ||
| 1005 | WARN_ON(!PageLocked(pages[i])); | ||
| 1006 | } | ||
| 1007 | return 0; | ||
| 1008 | } | ||
| 1009 | |||
| 1010 | static ssize_t btrfs_file_write(struct file *file, const char __user *buf, | ||
| 1011 | size_t count, loff_t *ppos) | ||
| 1012 | { | ||
| 1013 | loff_t pos; | ||
| 1014 | loff_t start_pos; | ||
| 1015 | ssize_t num_written = 0; | ||
| 1016 | ssize_t err = 0; | ||
| 1017 | int ret = 0; | ||
| 1018 | struct inode *inode = fdentry(file)->d_inode; | ||
| 1019 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1020 | struct page **pages = NULL; | ||
| 1021 | int nrptrs; | ||
| 1022 | struct page *pinned[2]; | ||
| 1023 | unsigned long first_index; | ||
| 1024 | unsigned long last_index; | ||
| 1025 | int will_write; | ||
| 1026 | |||
| 1027 | will_write = ((file->f_flags & O_SYNC) || IS_SYNC(inode) || | ||
| 1028 | (file->f_flags & O_DIRECT)); | ||
| 1029 | |||
| 1030 | nrptrs = min((count + PAGE_CACHE_SIZE - 1) / PAGE_CACHE_SIZE, | ||
| 1031 | PAGE_CACHE_SIZE / (sizeof(struct page *))); | ||
| 1032 | pinned[0] = NULL; | ||
| 1033 | pinned[1] = NULL; | ||
| 1034 | |||
| 1035 | pos = *ppos; | ||
| 1036 | start_pos = pos; | ||
| 1037 | |||
| 1038 | vfs_check_frozen(inode->i_sb, SB_FREEZE_WRITE); | ||
| 1039 | current->backing_dev_info = inode->i_mapping->backing_dev_info; | ||
| 1040 | err = generic_write_checks(file, &pos, &count, S_ISBLK(inode->i_mode)); | ||
| 1041 | if (err) | ||
| 1042 | goto out_nolock; | ||
| 1043 | if (count == 0) | ||
| 1044 | goto out_nolock; | ||
| 1045 | |||
| 1046 | err = file_remove_suid(file); | ||
| 1047 | if (err) | ||
| 1048 | goto out_nolock; | ||
| 1049 | file_update_time(file); | ||
| 1050 | |||
| 1051 | pages = kmalloc(nrptrs * sizeof(struct page *), GFP_KERNEL); | ||
| 1052 | |||
| 1053 | mutex_lock(&inode->i_mutex); | ||
| 1054 | BTRFS_I(inode)->sequence++; | ||
| 1055 | first_index = pos >> PAGE_CACHE_SHIFT; | ||
| 1056 | last_index = (pos + count) >> PAGE_CACHE_SHIFT; | ||
| 1057 | |||
| 1058 | /* | ||
| 1059 | * there are lots of better ways to do this, but this code | ||
| 1060 | * makes sure the first and last page in the file range are | ||
| 1061 | * up to date and ready for cow | ||
| 1062 | */ | ||
| 1063 | if ((pos & (PAGE_CACHE_SIZE - 1))) { | ||
| 1064 | pinned[0] = grab_cache_page(inode->i_mapping, first_index); | ||
| 1065 | if (!PageUptodate(pinned[0])) { | ||
| 1066 | ret = btrfs_readpage(NULL, pinned[0]); | ||
| 1067 | BUG_ON(ret); | ||
| 1068 | wait_on_page_locked(pinned[0]); | ||
| 1069 | } else { | ||
| 1070 | unlock_page(pinned[0]); | ||
| 1071 | } | ||
| 1072 | } | ||
| 1073 | if ((pos + count) & (PAGE_CACHE_SIZE - 1)) { | ||
| 1074 | pinned[1] = grab_cache_page(inode->i_mapping, last_index); | ||
| 1075 | if (!PageUptodate(pinned[1])) { | ||
| 1076 | ret = btrfs_readpage(NULL, pinned[1]); | ||
| 1077 | BUG_ON(ret); | ||
| 1078 | wait_on_page_locked(pinned[1]); | ||
| 1079 | } else { | ||
| 1080 | unlock_page(pinned[1]); | ||
| 1081 | } | ||
| 1082 | } | ||
| 1083 | |||
| 1084 | while (count > 0) { | ||
| 1085 | size_t offset = pos & (PAGE_CACHE_SIZE - 1); | ||
| 1086 | size_t write_bytes = min(count, nrptrs * | ||
| 1087 | (size_t)PAGE_CACHE_SIZE - | ||
| 1088 | offset); | ||
| 1089 | size_t num_pages = (write_bytes + PAGE_CACHE_SIZE - 1) >> | ||
| 1090 | PAGE_CACHE_SHIFT; | ||
| 1091 | |||
| 1092 | WARN_ON(num_pages > nrptrs); | ||
| 1093 | memset(pages, 0, sizeof(struct page *) * nrptrs); | ||
| 1094 | |||
| 1095 | ret = btrfs_check_free_space(root, write_bytes, 0); | ||
| 1096 | if (ret) | ||
| 1097 | goto out; | ||
| 1098 | |||
| 1099 | ret = prepare_pages(root, file, pages, num_pages, | ||
| 1100 | pos, first_index, last_index, | ||
| 1101 | write_bytes); | ||
| 1102 | if (ret) | ||
| 1103 | goto out; | ||
| 1104 | |||
| 1105 | ret = btrfs_copy_from_user(pos, num_pages, | ||
| 1106 | write_bytes, pages, buf); | ||
| 1107 | if (ret) { | ||
| 1108 | btrfs_drop_pages(pages, num_pages); | ||
| 1109 | goto out; | ||
| 1110 | } | ||
| 1111 | |||
| 1112 | ret = dirty_and_release_pages(NULL, root, file, pages, | ||
| 1113 | num_pages, pos, write_bytes); | ||
| 1114 | btrfs_drop_pages(pages, num_pages); | ||
| 1115 | if (ret) | ||
| 1116 | goto out; | ||
| 1117 | |||
| 1118 | if (will_write) { | ||
| 1119 | btrfs_fdatawrite_range(inode->i_mapping, pos, | ||
| 1120 | pos + write_bytes - 1, | ||
| 1121 | WB_SYNC_NONE); | ||
| 1122 | } else { | ||
| 1123 | balance_dirty_pages_ratelimited_nr(inode->i_mapping, | ||
| 1124 | num_pages); | ||
| 1125 | if (num_pages < | ||
| 1126 | (root->leafsize >> PAGE_CACHE_SHIFT) + 1) | ||
| 1127 | btrfs_btree_balance_dirty(root, 1); | ||
| 1128 | btrfs_throttle(root); | ||
| 1129 | } | ||
| 1130 | |||
| 1131 | buf += write_bytes; | ||
| 1132 | count -= write_bytes; | ||
| 1133 | pos += write_bytes; | ||
| 1134 | num_written += write_bytes; | ||
| 1135 | |||
| 1136 | cond_resched(); | ||
| 1137 | } | ||
| 1138 | out: | ||
| 1139 | mutex_unlock(&inode->i_mutex); | ||
| 1140 | |||
| 1141 | out_nolock: | ||
| 1142 | kfree(pages); | ||
| 1143 | if (pinned[0]) | ||
| 1144 | page_cache_release(pinned[0]); | ||
| 1145 | if (pinned[1]) | ||
| 1146 | page_cache_release(pinned[1]); | ||
| 1147 | *ppos = pos; | ||
| 1148 | |||
| 1149 | if (num_written > 0 && will_write) { | ||
| 1150 | struct btrfs_trans_handle *trans; | ||
| 1151 | |||
| 1152 | err = btrfs_wait_ordered_range(inode, start_pos, num_written); | ||
| 1153 | if (err) | ||
| 1154 | num_written = err; | ||
| 1155 | |||
| 1156 | if ((file->f_flags & O_SYNC) || IS_SYNC(inode)) { | ||
| 1157 | trans = btrfs_start_transaction(root, 1); | ||
| 1158 | ret = btrfs_log_dentry_safe(trans, root, | ||
| 1159 | file->f_dentry); | ||
| 1160 | if (ret == 0) { | ||
| 1161 | btrfs_sync_log(trans, root); | ||
| 1162 | btrfs_end_transaction(trans, root); | ||
| 1163 | } else { | ||
| 1164 | btrfs_commit_transaction(trans, root); | ||
| 1165 | } | ||
| 1166 | } | ||
| 1167 | if (file->f_flags & O_DIRECT) { | ||
| 1168 | invalidate_mapping_pages(inode->i_mapping, | ||
| 1169 | start_pos >> PAGE_CACHE_SHIFT, | ||
| 1170 | (start_pos + num_written - 1) >> PAGE_CACHE_SHIFT); | ||
| 1171 | } | ||
| 1172 | } | ||
| 1173 | current->backing_dev_info = NULL; | ||
| 1174 | return num_written ? num_written : err; | ||
| 1175 | } | ||
| 1176 | |||
| 1177 | int btrfs_release_file(struct inode *inode, struct file *filp) | ||
| 1178 | { | ||
| 1179 | if (filp->private_data) | ||
| 1180 | btrfs_ioctl_trans_end(filp); | ||
| 1181 | return 0; | ||
| 1182 | } | ||
| 1183 | |||
| 1184 | /* | ||
| 1185 | * fsync call for both files and directories. This logs the inode into | ||
| 1186 | * the tree log instead of forcing full commits whenever possible. | ||
| 1187 | * | ||
| 1188 | * It needs to call filemap_fdatawait so that all ordered extent updates are | ||
| 1189 | * in the metadata btree are up to date for copying to the log. | ||
| 1190 | * | ||
| 1191 | * It drops the inode mutex before doing the tree log commit. This is an | ||
| 1192 | * important optimization for directories because holding the mutex prevents | ||
| 1193 | * new operations on the dir while we write to disk. | ||
| 1194 | */ | ||
| 1195 | int btrfs_sync_file(struct file *file, struct dentry *dentry, int datasync) | ||
| 1196 | { | ||
| 1197 | struct inode *inode = dentry->d_inode; | ||
| 1198 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1199 | int ret = 0; | ||
| 1200 | struct btrfs_trans_handle *trans; | ||
| 1201 | |||
| 1202 | /* | ||
| 1203 | * check the transaction that last modified this inode | ||
| 1204 | * and see if its already been committed | ||
| 1205 | */ | ||
| 1206 | if (!BTRFS_I(inode)->last_trans) | ||
| 1207 | goto out; | ||
| 1208 | |||
| 1209 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 1210 | if (BTRFS_I(inode)->last_trans <= | ||
| 1211 | root->fs_info->last_trans_committed) { | ||
| 1212 | BTRFS_I(inode)->last_trans = 0; | ||
| 1213 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1214 | goto out; | ||
| 1215 | } | ||
| 1216 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1217 | |||
| 1218 | root->fs_info->tree_log_batch++; | ||
| 1219 | filemap_fdatawrite(inode->i_mapping); | ||
| 1220 | btrfs_wait_ordered_range(inode, 0, (u64)-1); | ||
| 1221 | root->fs_info->tree_log_batch++; | ||
| 1222 | |||
| 1223 | /* | ||
| 1224 | * ok we haven't committed the transaction yet, lets do a commit | ||
| 1225 | */ | ||
| 1226 | if (file->private_data) | ||
| 1227 | btrfs_ioctl_trans_end(file); | ||
| 1228 | |||
| 1229 | trans = btrfs_start_transaction(root, 1); | ||
| 1230 | if (!trans) { | ||
| 1231 | ret = -ENOMEM; | ||
| 1232 | goto out; | ||
| 1233 | } | ||
| 1234 | |||
| 1235 | ret = btrfs_log_dentry_safe(trans, root, file->f_dentry); | ||
| 1236 | if (ret < 0) | ||
| 1237 | goto out; | ||
| 1238 | |||
| 1239 | /* we've logged all the items and now have a consistent | ||
| 1240 | * version of the file in the log. It is possible that | ||
| 1241 | * someone will come in and modify the file, but that's | ||
| 1242 | * fine because the log is consistent on disk, and we | ||
| 1243 | * have references to all of the file's extents | ||
| 1244 | * | ||
| 1245 | * It is possible that someone will come in and log the | ||
| 1246 | * file again, but that will end up using the synchronization | ||
| 1247 | * inside btrfs_sync_log to keep things safe. | ||
| 1248 | */ | ||
| 1249 | mutex_unlock(&file->f_dentry->d_inode->i_mutex); | ||
| 1250 | |||
| 1251 | if (ret > 0) { | ||
| 1252 | ret = btrfs_commit_transaction(trans, root); | ||
| 1253 | } else { | ||
| 1254 | btrfs_sync_log(trans, root); | ||
| 1255 | ret = btrfs_end_transaction(trans, root); | ||
| 1256 | } | ||
| 1257 | mutex_lock(&file->f_dentry->d_inode->i_mutex); | ||
| 1258 | out: | ||
| 1259 | return ret > 0 ? EIO : ret; | ||
| 1260 | } | ||
| 1261 | |||
| 1262 | static struct vm_operations_struct btrfs_file_vm_ops = { | ||
| 1263 | .fault = filemap_fault, | ||
| 1264 | .page_mkwrite = btrfs_page_mkwrite, | ||
| 1265 | }; | ||
| 1266 | |||
| 1267 | static int btrfs_file_mmap(struct file *filp, struct vm_area_struct *vma) | ||
| 1268 | { | ||
| 1269 | vma->vm_ops = &btrfs_file_vm_ops; | ||
| 1270 | file_accessed(filp); | ||
| 1271 | return 0; | ||
| 1272 | } | ||
| 1273 | |||
| 1274 | struct file_operations btrfs_file_operations = { | ||
| 1275 | .llseek = generic_file_llseek, | ||
| 1276 | .read = do_sync_read, | ||
| 1277 | .aio_read = generic_file_aio_read, | ||
| 1278 | .splice_read = generic_file_splice_read, | ||
| 1279 | .write = btrfs_file_write, | ||
| 1280 | .mmap = btrfs_file_mmap, | ||
| 1281 | .open = generic_file_open, | ||
| 1282 | .release = btrfs_release_file, | ||
| 1283 | .fsync = btrfs_sync_file, | ||
| 1284 | .unlocked_ioctl = btrfs_ioctl, | ||
| 1285 | #ifdef CONFIG_COMPAT | ||
| 1286 | .compat_ioctl = btrfs_ioctl, | ||
| 1287 | #endif | ||
| 1288 | }; | ||
diff --git a/fs/btrfs/free-space-cache.c b/fs/btrfs/free-space-cache.c new file mode 100644 index 000000000000..d1e5f0e84c58 --- /dev/null +++ b/fs/btrfs/free-space-cache.c | |||
| @@ -0,0 +1,495 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 Red Hat. 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 | |||
| 19 | #include <linux/sched.h> | ||
| 20 | #include "ctree.h" | ||
| 21 | |||
| 22 | static int tree_insert_offset(struct rb_root *root, u64 offset, | ||
| 23 | struct rb_node *node) | ||
| 24 | { | ||
| 25 | struct rb_node **p = &root->rb_node; | ||
| 26 | struct rb_node *parent = NULL; | ||
| 27 | struct btrfs_free_space *info; | ||
| 28 | |||
| 29 | while (*p) { | ||
| 30 | parent = *p; | ||
| 31 | info = rb_entry(parent, struct btrfs_free_space, offset_index); | ||
| 32 | |||
| 33 | if (offset < info->offset) | ||
| 34 | p = &(*p)->rb_left; | ||
| 35 | else if (offset > info->offset) | ||
| 36 | p = &(*p)->rb_right; | ||
| 37 | else | ||
| 38 | return -EEXIST; | ||
| 39 | } | ||
| 40 | |||
| 41 | rb_link_node(node, parent, p); | ||
| 42 | rb_insert_color(node, root); | ||
| 43 | |||
| 44 | return 0; | ||
| 45 | } | ||
| 46 | |||
| 47 | static int tree_insert_bytes(struct rb_root *root, u64 bytes, | ||
| 48 | struct rb_node *node) | ||
| 49 | { | ||
| 50 | struct rb_node **p = &root->rb_node; | ||
| 51 | struct rb_node *parent = NULL; | ||
| 52 | struct btrfs_free_space *info; | ||
| 53 | |||
| 54 | while (*p) { | ||
| 55 | parent = *p; | ||
| 56 | info = rb_entry(parent, struct btrfs_free_space, bytes_index); | ||
| 57 | |||
| 58 | if (bytes < info->bytes) | ||
| 59 | p = &(*p)->rb_left; | ||
| 60 | else | ||
| 61 | p = &(*p)->rb_right; | ||
| 62 | } | ||
| 63 | |||
| 64 | rb_link_node(node, parent, p); | ||
| 65 | rb_insert_color(node, root); | ||
| 66 | |||
| 67 | return 0; | ||
| 68 | } | ||
| 69 | |||
| 70 | /* | ||
| 71 | * searches the tree for the given offset. If contains is set we will return | ||
| 72 | * the free space that contains the given offset. If contains is not set we | ||
| 73 | * will return the free space that starts at or after the given offset and is | ||
| 74 | * at least bytes long. | ||
| 75 | */ | ||
| 76 | static struct btrfs_free_space *tree_search_offset(struct rb_root *root, | ||
| 77 | u64 offset, u64 bytes, | ||
| 78 | int contains) | ||
| 79 | { | ||
| 80 | struct rb_node *n = root->rb_node; | ||
| 81 | struct btrfs_free_space *entry, *ret = NULL; | ||
| 82 | |||
| 83 | while (n) { | ||
| 84 | entry = rb_entry(n, struct btrfs_free_space, offset_index); | ||
| 85 | |||
| 86 | if (offset < entry->offset) { | ||
| 87 | if (!contains && | ||
| 88 | (!ret || entry->offset < ret->offset) && | ||
| 89 | (bytes <= entry->bytes)) | ||
| 90 | ret = entry; | ||
| 91 | n = n->rb_left; | ||
| 92 | } else if (offset > entry->offset) { | ||
| 93 | if ((entry->offset + entry->bytes - 1) >= offset && | ||
| 94 | bytes <= entry->bytes) { | ||
| 95 | ret = entry; | ||
| 96 | break; | ||
| 97 | } | ||
| 98 | n = n->rb_right; | ||
| 99 | } else { | ||
| 100 | if (bytes > entry->bytes) { | ||
| 101 | n = n->rb_right; | ||
| 102 | continue; | ||
| 103 | } | ||
| 104 | ret = entry; | ||
| 105 | break; | ||
| 106 | } | ||
| 107 | } | ||
| 108 | |||
| 109 | return ret; | ||
| 110 | } | ||
| 111 | |||
| 112 | /* | ||
| 113 | * return a chunk at least bytes size, as close to offset that we can get. | ||
| 114 | */ | ||
| 115 | static struct btrfs_free_space *tree_search_bytes(struct rb_root *root, | ||
| 116 | u64 offset, u64 bytes) | ||
| 117 | { | ||
| 118 | struct rb_node *n = root->rb_node; | ||
| 119 | struct btrfs_free_space *entry, *ret = NULL; | ||
| 120 | |||
| 121 | while (n) { | ||
| 122 | entry = rb_entry(n, struct btrfs_free_space, bytes_index); | ||
| 123 | |||
| 124 | if (bytes < entry->bytes) { | ||
| 125 | /* | ||
| 126 | * We prefer to get a hole size as close to the size we | ||
| 127 | * are asking for so we don't take small slivers out of | ||
| 128 | * huge holes, but we also want to get as close to the | ||
| 129 | * offset as possible so we don't have a whole lot of | ||
| 130 | * fragmentation. | ||
| 131 | */ | ||
| 132 | if (offset <= entry->offset) { | ||
| 133 | if (!ret) | ||
| 134 | ret = entry; | ||
| 135 | else if (entry->bytes < ret->bytes) | ||
| 136 | ret = entry; | ||
| 137 | else if (entry->offset < ret->offset) | ||
| 138 | ret = entry; | ||
| 139 | } | ||
| 140 | n = n->rb_left; | ||
| 141 | } else if (bytes > entry->bytes) { | ||
| 142 | n = n->rb_right; | ||
| 143 | } else { | ||
| 144 | /* | ||
| 145 | * Ok we may have multiple chunks of the wanted size, | ||
| 146 | * so we don't want to take the first one we find, we | ||
| 147 | * want to take the one closest to our given offset, so | ||
| 148 | * keep searching just in case theres a better match. | ||
| 149 | */ | ||
| 150 | n = n->rb_right; | ||
| 151 | if (offset > entry->offset) | ||
| 152 | continue; | ||
| 153 | else if (!ret || entry->offset < ret->offset) | ||
| 154 | ret = entry; | ||
| 155 | } | ||
| 156 | } | ||
| 157 | |||
| 158 | return ret; | ||
| 159 | } | ||
| 160 | |||
| 161 | static void unlink_free_space(struct btrfs_block_group_cache *block_group, | ||
| 162 | struct btrfs_free_space *info) | ||
| 163 | { | ||
| 164 | rb_erase(&info->offset_index, &block_group->free_space_offset); | ||
| 165 | rb_erase(&info->bytes_index, &block_group->free_space_bytes); | ||
| 166 | } | ||
| 167 | |||
| 168 | static int link_free_space(struct btrfs_block_group_cache *block_group, | ||
| 169 | struct btrfs_free_space *info) | ||
| 170 | { | ||
| 171 | int ret = 0; | ||
| 172 | |||
| 173 | |||
| 174 | ret = tree_insert_offset(&block_group->free_space_offset, info->offset, | ||
| 175 | &info->offset_index); | ||
| 176 | if (ret) | ||
| 177 | return ret; | ||
| 178 | |||
| 179 | ret = tree_insert_bytes(&block_group->free_space_bytes, info->bytes, | ||
| 180 | &info->bytes_index); | ||
| 181 | if (ret) | ||
| 182 | return ret; | ||
| 183 | |||
| 184 | return ret; | ||
| 185 | } | ||
| 186 | |||
| 187 | static int __btrfs_add_free_space(struct btrfs_block_group_cache *block_group, | ||
| 188 | u64 offset, u64 bytes) | ||
| 189 | { | ||
| 190 | struct btrfs_free_space *right_info; | ||
| 191 | struct btrfs_free_space *left_info; | ||
| 192 | struct btrfs_free_space *info = NULL; | ||
| 193 | struct btrfs_free_space *alloc_info; | ||
| 194 | int ret = 0; | ||
| 195 | |||
| 196 | alloc_info = kzalloc(sizeof(struct btrfs_free_space), GFP_NOFS); | ||
| 197 | if (!alloc_info) | ||
| 198 | return -ENOMEM; | ||
| 199 | |||
| 200 | /* | ||
| 201 | * first we want to see if there is free space adjacent to the range we | ||
| 202 | * are adding, if there is remove that struct and add a new one to | ||
| 203 | * cover the entire range | ||
| 204 | */ | ||
| 205 | right_info = tree_search_offset(&block_group->free_space_offset, | ||
| 206 | offset+bytes, 0, 1); | ||
| 207 | left_info = tree_search_offset(&block_group->free_space_offset, | ||
| 208 | offset-1, 0, 1); | ||
| 209 | |||
| 210 | if (right_info && right_info->offset == offset+bytes) { | ||
| 211 | unlink_free_space(block_group, right_info); | ||
| 212 | info = right_info; | ||
| 213 | info->offset = offset; | ||
| 214 | info->bytes += bytes; | ||
| 215 | } else if (right_info && right_info->offset != offset+bytes) { | ||
| 216 | printk(KERN_ERR "btrfs adding space in the middle of an " | ||
| 217 | "existing free space area. existing: " | ||
| 218 | "offset=%llu, bytes=%llu. new: offset=%llu, " | ||
| 219 | "bytes=%llu\n", (unsigned long long)right_info->offset, | ||
| 220 | (unsigned long long)right_info->bytes, | ||
| 221 | (unsigned long long)offset, | ||
| 222 | (unsigned long long)bytes); | ||
| 223 | BUG(); | ||
| 224 | } | ||
| 225 | |||
| 226 | if (left_info) { | ||
| 227 | unlink_free_space(block_group, left_info); | ||
| 228 | |||
| 229 | if (unlikely((left_info->offset + left_info->bytes) != | ||
| 230 | offset)) { | ||
| 231 | printk(KERN_ERR "btrfs free space to the left " | ||
| 232 | "of new free space isn't " | ||
| 233 | "quite right. existing: offset=%llu, " | ||
| 234 | "bytes=%llu. new: offset=%llu, bytes=%llu\n", | ||
| 235 | (unsigned long long)left_info->offset, | ||
| 236 | (unsigned long long)left_info->bytes, | ||
| 237 | (unsigned long long)offset, | ||
| 238 | (unsigned long long)bytes); | ||
| 239 | BUG(); | ||
| 240 | } | ||
| 241 | |||
| 242 | if (info) { | ||
| 243 | info->offset = left_info->offset; | ||
| 244 | info->bytes += left_info->bytes; | ||
| 245 | kfree(left_info); | ||
| 246 | } else { | ||
| 247 | info = left_info; | ||
| 248 | info->bytes += bytes; | ||
| 249 | } | ||
| 250 | } | ||
| 251 | |||
| 252 | if (info) { | ||
| 253 | ret = link_free_space(block_group, info); | ||
| 254 | if (!ret) | ||
| 255 | info = NULL; | ||
| 256 | goto out; | ||
| 257 | } | ||
| 258 | |||
| 259 | info = alloc_info; | ||
| 260 | alloc_info = NULL; | ||
| 261 | info->offset = offset; | ||
| 262 | info->bytes = bytes; | ||
| 263 | |||
| 264 | ret = link_free_space(block_group, info); | ||
| 265 | if (ret) | ||
| 266 | kfree(info); | ||
| 267 | out: | ||
| 268 | if (ret) { | ||
| 269 | printk(KERN_ERR "btrfs: unable to add free space :%d\n", ret); | ||
| 270 | if (ret == -EEXIST) | ||
| 271 | BUG(); | ||
| 272 | } | ||
| 273 | |||
| 274 | kfree(alloc_info); | ||
| 275 | |||
| 276 | return ret; | ||
| 277 | } | ||
| 278 | |||
| 279 | static int | ||
| 280 | __btrfs_remove_free_space(struct btrfs_block_group_cache *block_group, | ||
| 281 | u64 offset, u64 bytes) | ||
| 282 | { | ||
| 283 | struct btrfs_free_space *info; | ||
| 284 | int ret = 0; | ||
| 285 | |||
| 286 | info = tree_search_offset(&block_group->free_space_offset, offset, 0, | ||
| 287 | 1); | ||
| 288 | |||
| 289 | if (info && info->offset == offset) { | ||
| 290 | if (info->bytes < bytes) { | ||
| 291 | printk(KERN_ERR "Found free space at %llu, size %llu," | ||
| 292 | "trying to use %llu\n", | ||
| 293 | (unsigned long long)info->offset, | ||
| 294 | (unsigned long long)info->bytes, | ||
| 295 | (unsigned long long)bytes); | ||
| 296 | WARN_ON(1); | ||
| 297 | ret = -EINVAL; | ||
| 298 | goto out; | ||
| 299 | } | ||
| 300 | unlink_free_space(block_group, info); | ||
| 301 | |||
| 302 | if (info->bytes == bytes) { | ||
| 303 | kfree(info); | ||
| 304 | goto out; | ||
| 305 | } | ||
| 306 | |||
| 307 | info->offset += bytes; | ||
| 308 | info->bytes -= bytes; | ||
| 309 | |||
| 310 | ret = link_free_space(block_group, info); | ||
| 311 | BUG_ON(ret); | ||
| 312 | } else if (info && info->offset < offset && | ||
| 313 | info->offset + info->bytes >= offset + bytes) { | ||
| 314 | u64 old_start = info->offset; | ||
| 315 | /* | ||
| 316 | * we're freeing space in the middle of the info, | ||
| 317 | * this can happen during tree log replay | ||
| 318 | * | ||
| 319 | * first unlink the old info and then | ||
| 320 | * insert it again after the hole we're creating | ||
| 321 | */ | ||
| 322 | unlink_free_space(block_group, info); | ||
| 323 | if (offset + bytes < info->offset + info->bytes) { | ||
| 324 | u64 old_end = info->offset + info->bytes; | ||
| 325 | |||
| 326 | info->offset = offset + bytes; | ||
| 327 | info->bytes = old_end - info->offset; | ||
| 328 | ret = link_free_space(block_group, info); | ||
| 329 | BUG_ON(ret); | ||
| 330 | } else { | ||
| 331 | /* the hole we're creating ends at the end | ||
| 332 | * of the info struct, just free the info | ||
| 333 | */ | ||
| 334 | kfree(info); | ||
| 335 | } | ||
| 336 | |||
| 337 | /* step two, insert a new info struct to cover anything | ||
| 338 | * before the hole | ||
| 339 | */ | ||
| 340 | ret = __btrfs_add_free_space(block_group, old_start, | ||
| 341 | offset - old_start); | ||
| 342 | BUG_ON(ret); | ||
| 343 | } else { | ||
| 344 | WARN_ON(1); | ||
| 345 | } | ||
| 346 | out: | ||
| 347 | return ret; | ||
| 348 | } | ||
| 349 | |||
| 350 | int btrfs_add_free_space(struct btrfs_block_group_cache *block_group, | ||
| 351 | u64 offset, u64 bytes) | ||
| 352 | { | ||
| 353 | int ret; | ||
| 354 | struct btrfs_free_space *sp; | ||
| 355 | |||
| 356 | mutex_lock(&block_group->alloc_mutex); | ||
| 357 | ret = __btrfs_add_free_space(block_group, offset, bytes); | ||
| 358 | sp = tree_search_offset(&block_group->free_space_offset, offset, 0, 1); | ||
| 359 | BUG_ON(!sp); | ||
| 360 | mutex_unlock(&block_group->alloc_mutex); | ||
| 361 | |||
| 362 | return ret; | ||
| 363 | } | ||
| 364 | |||
| 365 | int btrfs_add_free_space_lock(struct btrfs_block_group_cache *block_group, | ||
| 366 | u64 offset, u64 bytes) | ||
| 367 | { | ||
| 368 | int ret; | ||
| 369 | struct btrfs_free_space *sp; | ||
| 370 | |||
| 371 | ret = __btrfs_add_free_space(block_group, offset, bytes); | ||
| 372 | sp = tree_search_offset(&block_group->free_space_offset, offset, 0, 1); | ||
| 373 | BUG_ON(!sp); | ||
| 374 | |||
| 375 | return ret; | ||
| 376 | } | ||
| 377 | |||
| 378 | int btrfs_remove_free_space(struct btrfs_block_group_cache *block_group, | ||
| 379 | u64 offset, u64 bytes) | ||
| 380 | { | ||
| 381 | int ret = 0; | ||
| 382 | |||
| 383 | mutex_lock(&block_group->alloc_mutex); | ||
| 384 | ret = __btrfs_remove_free_space(block_group, offset, bytes); | ||
| 385 | mutex_unlock(&block_group->alloc_mutex); | ||
| 386 | |||
| 387 | return ret; | ||
| 388 | } | ||
| 389 | |||
| 390 | int btrfs_remove_free_space_lock(struct btrfs_block_group_cache *block_group, | ||
| 391 | u64 offset, u64 bytes) | ||
| 392 | { | ||
| 393 | int ret; | ||
| 394 | |||
| 395 | ret = __btrfs_remove_free_space(block_group, offset, bytes); | ||
| 396 | |||
| 397 | return ret; | ||
| 398 | } | ||
| 399 | |||
| 400 | void btrfs_dump_free_space(struct btrfs_block_group_cache *block_group, | ||
| 401 | u64 bytes) | ||
| 402 | { | ||
| 403 | struct btrfs_free_space *info; | ||
| 404 | struct rb_node *n; | ||
| 405 | int count = 0; | ||
| 406 | |||
| 407 | for (n = rb_first(&block_group->free_space_offset); n; n = rb_next(n)) { | ||
| 408 | info = rb_entry(n, struct btrfs_free_space, offset_index); | ||
| 409 | if (info->bytes >= bytes) | ||
| 410 | count++; | ||
| 411 | } | ||
| 412 | printk(KERN_INFO "%d blocks of free space at or bigger than bytes is" | ||
| 413 | "\n", count); | ||
| 414 | } | ||
| 415 | |||
| 416 | u64 btrfs_block_group_free_space(struct btrfs_block_group_cache *block_group) | ||
| 417 | { | ||
| 418 | struct btrfs_free_space *info; | ||
| 419 | struct rb_node *n; | ||
| 420 | u64 ret = 0; | ||
| 421 | |||
| 422 | for (n = rb_first(&block_group->free_space_offset); n; | ||
| 423 | n = rb_next(n)) { | ||
| 424 | info = rb_entry(n, struct btrfs_free_space, offset_index); | ||
| 425 | ret += info->bytes; | ||
| 426 | } | ||
| 427 | |||
| 428 | return ret; | ||
| 429 | } | ||
| 430 | |||
| 431 | void btrfs_remove_free_space_cache(struct btrfs_block_group_cache *block_group) | ||
| 432 | { | ||
| 433 | struct btrfs_free_space *info; | ||
| 434 | struct rb_node *node; | ||
| 435 | |||
| 436 | mutex_lock(&block_group->alloc_mutex); | ||
| 437 | while ((node = rb_last(&block_group->free_space_bytes)) != NULL) { | ||
| 438 | info = rb_entry(node, struct btrfs_free_space, bytes_index); | ||
| 439 | unlink_free_space(block_group, info); | ||
| 440 | kfree(info); | ||
| 441 | if (need_resched()) { | ||
| 442 | mutex_unlock(&block_group->alloc_mutex); | ||
| 443 | cond_resched(); | ||
| 444 | mutex_lock(&block_group->alloc_mutex); | ||
| 445 | } | ||
| 446 | } | ||
| 447 | mutex_unlock(&block_group->alloc_mutex); | ||
| 448 | } | ||
| 449 | |||
| 450 | #if 0 | ||
| 451 | static struct btrfs_free_space *btrfs_find_free_space_offset(struct | ||
| 452 | btrfs_block_group_cache | ||
| 453 | *block_group, u64 offset, | ||
| 454 | u64 bytes) | ||
| 455 | { | ||
| 456 | struct btrfs_free_space *ret; | ||
| 457 | |||
| 458 | mutex_lock(&block_group->alloc_mutex); | ||
| 459 | ret = tree_search_offset(&block_group->free_space_offset, offset, | ||
| 460 | bytes, 0); | ||
| 461 | mutex_unlock(&block_group->alloc_mutex); | ||
| 462 | |||
| 463 | return ret; | ||
| 464 | } | ||
| 465 | |||
| 466 | static struct btrfs_free_space *btrfs_find_free_space_bytes(struct | ||
| 467 | btrfs_block_group_cache | ||
| 468 | *block_group, u64 offset, | ||
| 469 | u64 bytes) | ||
| 470 | { | ||
| 471 | struct btrfs_free_space *ret; | ||
| 472 | |||
| 473 | mutex_lock(&block_group->alloc_mutex); | ||
| 474 | |||
| 475 | ret = tree_search_bytes(&block_group->free_space_bytes, offset, bytes); | ||
| 476 | mutex_unlock(&block_group->alloc_mutex); | ||
| 477 | |||
| 478 | return ret; | ||
| 479 | } | ||
| 480 | #endif | ||
| 481 | |||
| 482 | struct btrfs_free_space *btrfs_find_free_space(struct btrfs_block_group_cache | ||
| 483 | *block_group, u64 offset, | ||
| 484 | u64 bytes) | ||
| 485 | { | ||
| 486 | struct btrfs_free_space *ret = NULL; | ||
| 487 | |||
| 488 | ret = tree_search_offset(&block_group->free_space_offset, offset, | ||
| 489 | bytes, 0); | ||
| 490 | if (!ret) | ||
| 491 | ret = tree_search_bytes(&block_group->free_space_bytes, | ||
| 492 | offset, bytes); | ||
| 493 | |||
| 494 | return ret; | ||
| 495 | } | ||
diff --git a/fs/btrfs/hash.h b/fs/btrfs/hash.h new file mode 100644 index 000000000000..2a020b276768 --- /dev/null +++ b/fs/btrfs/hash.h | |||
| @@ -0,0 +1,27 @@ | |||
| 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 | |||
| 19 | #ifndef __HASH__ | ||
| 20 | #define __HASH__ | ||
| 21 | |||
| 22 | #include "crc32c.h" | ||
| 23 | static inline u64 btrfs_name_hash(const char *name, int len) | ||
| 24 | { | ||
| 25 | return btrfs_crc32c((u32)~1, name, len); | ||
| 26 | } | ||
| 27 | #endif | ||
diff --git a/fs/btrfs/inode-item.c b/fs/btrfs/inode-item.c new file mode 100644 index 000000000000..3d46fa1f29a4 --- /dev/null +++ b/fs/btrfs/inode-item.c | |||
| @@ -0,0 +1,206 @@ | |||
| 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 | |||
| 19 | #include "ctree.h" | ||
| 20 | #include "disk-io.h" | ||
| 21 | #include "transaction.h" | ||
| 22 | |||
| 23 | static int find_name_in_backref(struct btrfs_path *path, const char *name, | ||
| 24 | int name_len, struct btrfs_inode_ref **ref_ret) | ||
| 25 | { | ||
| 26 | struct extent_buffer *leaf; | ||
| 27 | struct btrfs_inode_ref *ref; | ||
| 28 | unsigned long ptr; | ||
| 29 | unsigned long name_ptr; | ||
| 30 | u32 item_size; | ||
| 31 | u32 cur_offset = 0; | ||
| 32 | int len; | ||
| 33 | |||
| 34 | leaf = path->nodes[0]; | ||
| 35 | item_size = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 36 | ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); | ||
| 37 | while (cur_offset < item_size) { | ||
| 38 | ref = (struct btrfs_inode_ref *)(ptr + cur_offset); | ||
| 39 | len = btrfs_inode_ref_name_len(leaf, ref); | ||
| 40 | name_ptr = (unsigned long)(ref + 1); | ||
| 41 | cur_offset += len + sizeof(*ref); | ||
| 42 | if (len != name_len) | ||
| 43 | continue; | ||
| 44 | if (memcmp_extent_buffer(leaf, name, name_ptr, name_len) == 0) { | ||
| 45 | *ref_ret = ref; | ||
| 46 | return 1; | ||
| 47 | } | ||
| 48 | } | ||
| 49 | return 0; | ||
| 50 | } | ||
| 51 | |||
| 52 | int btrfs_del_inode_ref(struct btrfs_trans_handle *trans, | ||
| 53 | struct btrfs_root *root, | ||
| 54 | const char *name, int name_len, | ||
| 55 | u64 inode_objectid, u64 ref_objectid, u64 *index) | ||
| 56 | { | ||
| 57 | struct btrfs_path *path; | ||
| 58 | struct btrfs_key key; | ||
| 59 | struct btrfs_inode_ref *ref; | ||
| 60 | struct extent_buffer *leaf; | ||
| 61 | unsigned long ptr; | ||
| 62 | unsigned long item_start; | ||
| 63 | u32 item_size; | ||
| 64 | u32 sub_item_len; | ||
| 65 | int ret; | ||
| 66 | int del_len = name_len + sizeof(*ref); | ||
| 67 | |||
| 68 | key.objectid = inode_objectid; | ||
| 69 | key.offset = ref_objectid; | ||
| 70 | btrfs_set_key_type(&key, BTRFS_INODE_REF_KEY); | ||
| 71 | |||
| 72 | path = btrfs_alloc_path(); | ||
| 73 | if (!path) | ||
| 74 | return -ENOMEM; | ||
| 75 | |||
| 76 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 77 | if (ret > 0) { | ||
| 78 | ret = -ENOENT; | ||
| 79 | goto out; | ||
| 80 | } else if (ret < 0) { | ||
| 81 | goto out; | ||
| 82 | } | ||
| 83 | if (!find_name_in_backref(path, name, name_len, &ref)) { | ||
| 84 | ret = -ENOENT; | ||
| 85 | goto out; | ||
| 86 | } | ||
| 87 | leaf = path->nodes[0]; | ||
| 88 | item_size = btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 89 | |||
| 90 | if (index) | ||
| 91 | *index = btrfs_inode_ref_index(leaf, ref); | ||
| 92 | |||
| 93 | if (del_len == item_size) { | ||
| 94 | ret = btrfs_del_item(trans, root, path); | ||
| 95 | goto out; | ||
| 96 | } | ||
| 97 | ptr = (unsigned long)ref; | ||
| 98 | sub_item_len = name_len + sizeof(*ref); | ||
| 99 | item_start = btrfs_item_ptr_offset(leaf, path->slots[0]); | ||
| 100 | memmove_extent_buffer(leaf, ptr, ptr + sub_item_len, | ||
| 101 | item_size - (ptr + sub_item_len - item_start)); | ||
| 102 | ret = btrfs_truncate_item(trans, root, path, | ||
| 103 | item_size - sub_item_len, 1); | ||
| 104 | BUG_ON(ret); | ||
| 105 | out: | ||
| 106 | btrfs_free_path(path); | ||
| 107 | return ret; | ||
| 108 | } | ||
| 109 | |||
| 110 | int btrfs_insert_inode_ref(struct btrfs_trans_handle *trans, | ||
| 111 | struct btrfs_root *root, | ||
| 112 | const char *name, int name_len, | ||
| 113 | u64 inode_objectid, u64 ref_objectid, u64 index) | ||
| 114 | { | ||
| 115 | struct btrfs_path *path; | ||
| 116 | struct btrfs_key key; | ||
| 117 | struct btrfs_inode_ref *ref; | ||
| 118 | unsigned long ptr; | ||
| 119 | int ret; | ||
| 120 | int ins_len = name_len + sizeof(*ref); | ||
| 121 | |||
| 122 | key.objectid = inode_objectid; | ||
| 123 | key.offset = ref_objectid; | ||
| 124 | btrfs_set_key_type(&key, BTRFS_INODE_REF_KEY); | ||
| 125 | |||
| 126 | path = btrfs_alloc_path(); | ||
| 127 | if (!path) | ||
| 128 | return -ENOMEM; | ||
| 129 | |||
| 130 | ret = btrfs_insert_empty_item(trans, root, path, &key, | ||
| 131 | ins_len); | ||
| 132 | if (ret == -EEXIST) { | ||
| 133 | u32 old_size; | ||
| 134 | |||
| 135 | if (find_name_in_backref(path, name, name_len, &ref)) | ||
| 136 | goto out; | ||
| 137 | |||
| 138 | old_size = btrfs_item_size_nr(path->nodes[0], path->slots[0]); | ||
| 139 | ret = btrfs_extend_item(trans, root, path, ins_len); | ||
| 140 | BUG_ON(ret); | ||
| 141 | ref = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 142 | struct btrfs_inode_ref); | ||
| 143 | ref = (struct btrfs_inode_ref *)((unsigned long)ref + old_size); | ||
| 144 | btrfs_set_inode_ref_name_len(path->nodes[0], ref, name_len); | ||
| 145 | btrfs_set_inode_ref_index(path->nodes[0], ref, index); | ||
| 146 | ptr = (unsigned long)(ref + 1); | ||
| 147 | ret = 0; | ||
| 148 | } else if (ret < 0) { | ||
| 149 | goto out; | ||
| 150 | } else { | ||
| 151 | ref = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 152 | struct btrfs_inode_ref); | ||
| 153 | btrfs_set_inode_ref_name_len(path->nodes[0], ref, name_len); | ||
| 154 | btrfs_set_inode_ref_index(path->nodes[0], ref, index); | ||
| 155 | ptr = (unsigned long)(ref + 1); | ||
| 156 | } | ||
| 157 | write_extent_buffer(path->nodes[0], name, ptr, name_len); | ||
| 158 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 159 | |||
| 160 | out: | ||
| 161 | btrfs_free_path(path); | ||
| 162 | return ret; | ||
| 163 | } | ||
| 164 | |||
| 165 | int btrfs_insert_empty_inode(struct btrfs_trans_handle *trans, | ||
| 166 | struct btrfs_root *root, | ||
| 167 | struct btrfs_path *path, u64 objectid) | ||
| 168 | { | ||
| 169 | struct btrfs_key key; | ||
| 170 | int ret; | ||
| 171 | key.objectid = objectid; | ||
| 172 | btrfs_set_key_type(&key, BTRFS_INODE_ITEM_KEY); | ||
| 173 | key.offset = 0; | ||
| 174 | |||
| 175 | ret = btrfs_insert_empty_item(trans, root, path, &key, | ||
| 176 | sizeof(struct btrfs_inode_item)); | ||
| 177 | if (ret == 0 && objectid > root->highest_inode) | ||
| 178 | root->highest_inode = objectid; | ||
| 179 | return ret; | ||
| 180 | } | ||
| 181 | |||
| 182 | int btrfs_lookup_inode(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 183 | *root, struct btrfs_path *path, | ||
| 184 | struct btrfs_key *location, int mod) | ||
| 185 | { | ||
| 186 | int ins_len = mod < 0 ? -1 : 0; | ||
| 187 | int cow = mod != 0; | ||
| 188 | int ret; | ||
| 189 | int slot; | ||
| 190 | struct extent_buffer *leaf; | ||
| 191 | struct btrfs_key found_key; | ||
| 192 | |||
| 193 | ret = btrfs_search_slot(trans, root, location, path, ins_len, cow); | ||
| 194 | if (ret > 0 && btrfs_key_type(location) == BTRFS_ROOT_ITEM_KEY && | ||
| 195 | location->offset == (u64)-1 && path->slots[0] != 0) { | ||
| 196 | slot = path->slots[0] - 1; | ||
| 197 | leaf = path->nodes[0]; | ||
| 198 | btrfs_item_key_to_cpu(leaf, &found_key, slot); | ||
| 199 | if (found_key.objectid == location->objectid && | ||
| 200 | btrfs_key_type(&found_key) == btrfs_key_type(location)) { | ||
| 201 | path->slots[0]--; | ||
| 202 | return 0; | ||
| 203 | } | ||
| 204 | } | ||
| 205 | return ret; | ||
| 206 | } | ||
diff --git a/fs/btrfs/inode-map.c b/fs/btrfs/inode-map.c new file mode 100644 index 000000000000..2aa79873eb46 --- /dev/null +++ b/fs/btrfs/inode-map.c | |||
| @@ -0,0 +1,144 @@ | |||
| 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 | |||
| 19 | #include "ctree.h" | ||
| 20 | #include "disk-io.h" | ||
| 21 | #include "transaction.h" | ||
| 22 | |||
| 23 | int btrfs_find_highest_inode(struct btrfs_root *root, u64 *objectid) | ||
| 24 | { | ||
| 25 | struct btrfs_path *path; | ||
| 26 | int ret; | ||
| 27 | struct extent_buffer *l; | ||
| 28 | struct btrfs_key search_key; | ||
| 29 | struct btrfs_key found_key; | ||
| 30 | int slot; | ||
| 31 | |||
| 32 | path = btrfs_alloc_path(); | ||
| 33 | BUG_ON(!path); | ||
| 34 | |||
| 35 | search_key.objectid = BTRFS_LAST_FREE_OBJECTID; | ||
| 36 | search_key.type = -1; | ||
| 37 | search_key.offset = (u64)-1; | ||
| 38 | ret = btrfs_search_slot(NULL, root, &search_key, path, 0, 0); | ||
| 39 | if (ret < 0) | ||
| 40 | goto error; | ||
| 41 | BUG_ON(ret == 0); | ||
| 42 | if (path->slots[0] > 0) { | ||
| 43 | slot = path->slots[0] - 1; | ||
| 44 | l = path->nodes[0]; | ||
| 45 | btrfs_item_key_to_cpu(l, &found_key, slot); | ||
| 46 | *objectid = found_key.objectid; | ||
| 47 | } else { | ||
| 48 | *objectid = BTRFS_FIRST_FREE_OBJECTID; | ||
| 49 | } | ||
| 50 | ret = 0; | ||
| 51 | error: | ||
| 52 | btrfs_free_path(path); | ||
| 53 | return ret; | ||
| 54 | } | ||
| 55 | |||
| 56 | /* | ||
| 57 | * walks the btree of allocated inodes and find a hole. | ||
| 58 | */ | ||
| 59 | int btrfs_find_free_objectid(struct btrfs_trans_handle *trans, | ||
| 60 | struct btrfs_root *root, | ||
| 61 | u64 dirid, u64 *objectid) | ||
| 62 | { | ||
| 63 | struct btrfs_path *path; | ||
| 64 | struct btrfs_key key; | ||
| 65 | int ret; | ||
| 66 | int slot = 0; | ||
| 67 | u64 last_ino = 0; | ||
| 68 | int start_found; | ||
| 69 | struct extent_buffer *l; | ||
| 70 | struct btrfs_key search_key; | ||
| 71 | u64 search_start = dirid; | ||
| 72 | |||
| 73 | mutex_lock(&root->objectid_mutex); | ||
| 74 | if (root->last_inode_alloc >= BTRFS_FIRST_FREE_OBJECTID && | ||
| 75 | root->last_inode_alloc < BTRFS_LAST_FREE_OBJECTID) { | ||
| 76 | *objectid = ++root->last_inode_alloc; | ||
| 77 | mutex_unlock(&root->objectid_mutex); | ||
| 78 | return 0; | ||
| 79 | } | ||
| 80 | path = btrfs_alloc_path(); | ||
| 81 | BUG_ON(!path); | ||
| 82 | search_start = max(search_start, BTRFS_FIRST_FREE_OBJECTID); | ||
| 83 | search_key.objectid = search_start; | ||
| 84 | search_key.type = 0; | ||
| 85 | search_key.offset = 0; | ||
| 86 | |||
| 87 | btrfs_init_path(path); | ||
| 88 | start_found = 0; | ||
| 89 | ret = btrfs_search_slot(trans, root, &search_key, path, 0, 0); | ||
| 90 | if (ret < 0) | ||
| 91 | goto error; | ||
| 92 | |||
| 93 | while (1) { | ||
| 94 | l = path->nodes[0]; | ||
| 95 | slot = path->slots[0]; | ||
| 96 | if (slot >= btrfs_header_nritems(l)) { | ||
| 97 | ret = btrfs_next_leaf(root, path); | ||
| 98 | if (ret == 0) | ||
| 99 | continue; | ||
| 100 | if (ret < 0) | ||
| 101 | goto error; | ||
| 102 | if (!start_found) { | ||
| 103 | *objectid = search_start; | ||
| 104 | start_found = 1; | ||
| 105 | goto found; | ||
| 106 | } | ||
| 107 | *objectid = last_ino > search_start ? | ||
| 108 | last_ino : search_start; | ||
| 109 | goto found; | ||
| 110 | } | ||
| 111 | btrfs_item_key_to_cpu(l, &key, slot); | ||
| 112 | if (key.objectid >= search_start) { | ||
| 113 | if (start_found) { | ||
| 114 | if (last_ino < search_start) | ||
| 115 | last_ino = search_start; | ||
| 116 | if (key.objectid > last_ino) { | ||
| 117 | *objectid = last_ino; | ||
| 118 | goto found; | ||
| 119 | } | ||
| 120 | } else if (key.objectid > search_start) { | ||
| 121 | *objectid = search_start; | ||
| 122 | goto found; | ||
| 123 | } | ||
| 124 | } | ||
| 125 | if (key.objectid >= BTRFS_LAST_FREE_OBJECTID) | ||
| 126 | break; | ||
| 127 | |||
| 128 | start_found = 1; | ||
| 129 | last_ino = key.objectid + 1; | ||
| 130 | path->slots[0]++; | ||
| 131 | } | ||
| 132 | BUG_ON(1); | ||
| 133 | found: | ||
| 134 | btrfs_release_path(root, path); | ||
| 135 | btrfs_free_path(path); | ||
| 136 | BUG_ON(*objectid < search_start); | ||
| 137 | mutex_unlock(&root->objectid_mutex); | ||
| 138 | return 0; | ||
| 139 | error: | ||
| 140 | btrfs_release_path(root, path); | ||
| 141 | btrfs_free_path(path); | ||
| 142 | mutex_unlock(&root->objectid_mutex); | ||
| 143 | return ret; | ||
| 144 | } | ||
diff --git a/fs/btrfs/inode.c b/fs/btrfs/inode.c new file mode 100644 index 000000000000..8adfe059ab41 --- /dev/null +++ b/fs/btrfs/inode.c | |||
| @@ -0,0 +1,5035 @@ | |||
| 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 | |||
| 19 | #include <linux/kernel.h> | ||
| 20 | #include <linux/bio.h> | ||
| 21 | #include <linux/buffer_head.h> | ||
| 22 | #include <linux/file.h> | ||
| 23 | #include <linux/fs.h> | ||
| 24 | #include <linux/pagemap.h> | ||
| 25 | #include <linux/highmem.h> | ||
| 26 | #include <linux/time.h> | ||
| 27 | #include <linux/init.h> | ||
| 28 | #include <linux/string.h> | ||
| 29 | #include <linux/smp_lock.h> | ||
| 30 | #include <linux/backing-dev.h> | ||
| 31 | #include <linux/mpage.h> | ||
| 32 | #include <linux/swap.h> | ||
| 33 | #include <linux/writeback.h> | ||
| 34 | #include <linux/statfs.h> | ||
| 35 | #include <linux/compat.h> | ||
| 36 | #include <linux/bit_spinlock.h> | ||
| 37 | #include <linux/version.h> | ||
| 38 | #include <linux/xattr.h> | ||
| 39 | #include <linux/posix_acl.h> | ||
| 40 | #include <linux/falloc.h> | ||
| 41 | #include "compat.h" | ||
| 42 | #include "ctree.h" | ||
| 43 | #include "disk-io.h" | ||
| 44 | #include "transaction.h" | ||
| 45 | #include "btrfs_inode.h" | ||
| 46 | #include "ioctl.h" | ||
| 47 | #include "print-tree.h" | ||
| 48 | #include "volumes.h" | ||
| 49 | #include "ordered-data.h" | ||
| 50 | #include "xattr.h" | ||
| 51 | #include "tree-log.h" | ||
| 52 | #include "ref-cache.h" | ||
| 53 | #include "compression.h" | ||
| 54 | |||
| 55 | struct btrfs_iget_args { | ||
| 56 | u64 ino; | ||
| 57 | struct btrfs_root *root; | ||
| 58 | }; | ||
| 59 | |||
| 60 | static struct inode_operations btrfs_dir_inode_operations; | ||
| 61 | static struct inode_operations btrfs_symlink_inode_operations; | ||
| 62 | static struct inode_operations btrfs_dir_ro_inode_operations; | ||
| 63 | static struct inode_operations btrfs_special_inode_operations; | ||
| 64 | static struct inode_operations btrfs_file_inode_operations; | ||
| 65 | static struct address_space_operations btrfs_aops; | ||
| 66 | static struct address_space_operations btrfs_symlink_aops; | ||
| 67 | static struct file_operations btrfs_dir_file_operations; | ||
| 68 | static struct extent_io_ops btrfs_extent_io_ops; | ||
| 69 | |||
| 70 | static struct kmem_cache *btrfs_inode_cachep; | ||
| 71 | struct kmem_cache *btrfs_trans_handle_cachep; | ||
| 72 | struct kmem_cache *btrfs_transaction_cachep; | ||
| 73 | struct kmem_cache *btrfs_bit_radix_cachep; | ||
| 74 | struct kmem_cache *btrfs_path_cachep; | ||
| 75 | |||
| 76 | #define S_SHIFT 12 | ||
| 77 | static unsigned char btrfs_type_by_mode[S_IFMT >> S_SHIFT] = { | ||
| 78 | [S_IFREG >> S_SHIFT] = BTRFS_FT_REG_FILE, | ||
| 79 | [S_IFDIR >> S_SHIFT] = BTRFS_FT_DIR, | ||
| 80 | [S_IFCHR >> S_SHIFT] = BTRFS_FT_CHRDEV, | ||
| 81 | [S_IFBLK >> S_SHIFT] = BTRFS_FT_BLKDEV, | ||
| 82 | [S_IFIFO >> S_SHIFT] = BTRFS_FT_FIFO, | ||
| 83 | [S_IFSOCK >> S_SHIFT] = BTRFS_FT_SOCK, | ||
| 84 | [S_IFLNK >> S_SHIFT] = BTRFS_FT_SYMLINK, | ||
| 85 | }; | ||
| 86 | |||
| 87 | static void btrfs_truncate(struct inode *inode); | ||
| 88 | static int btrfs_finish_ordered_io(struct inode *inode, u64 start, u64 end); | ||
| 89 | static noinline int cow_file_range(struct inode *inode, | ||
| 90 | struct page *locked_page, | ||
| 91 | u64 start, u64 end, int *page_started, | ||
| 92 | unsigned long *nr_written, int unlock); | ||
| 93 | |||
| 94 | /* | ||
| 95 | * a very lame attempt at stopping writes when the FS is 85% full. There | ||
| 96 | * are countless ways this is incorrect, but it is better than nothing. | ||
| 97 | */ | ||
| 98 | int btrfs_check_free_space(struct btrfs_root *root, u64 num_required, | ||
| 99 | int for_del) | ||
| 100 | { | ||
| 101 | u64 total; | ||
| 102 | u64 used; | ||
| 103 | u64 thresh; | ||
| 104 | int ret = 0; | ||
| 105 | |||
| 106 | spin_lock(&root->fs_info->delalloc_lock); | ||
| 107 | total = btrfs_super_total_bytes(&root->fs_info->super_copy); | ||
| 108 | used = btrfs_super_bytes_used(&root->fs_info->super_copy); | ||
| 109 | if (for_del) | ||
| 110 | thresh = total * 90; | ||
| 111 | else | ||
| 112 | thresh = total * 85; | ||
| 113 | |||
| 114 | do_div(thresh, 100); | ||
| 115 | |||
| 116 | if (used + root->fs_info->delalloc_bytes + num_required > thresh) | ||
| 117 | ret = -ENOSPC; | ||
| 118 | spin_unlock(&root->fs_info->delalloc_lock); | ||
| 119 | return ret; | ||
| 120 | } | ||
| 121 | |||
| 122 | /* | ||
| 123 | * this does all the hard work for inserting an inline extent into | ||
| 124 | * the btree. The caller should have done a btrfs_drop_extents so that | ||
| 125 | * no overlapping inline items exist in the btree | ||
| 126 | */ | ||
| 127 | static noinline int insert_inline_extent(struct btrfs_trans_handle *trans, | ||
| 128 | struct btrfs_root *root, struct inode *inode, | ||
| 129 | u64 start, size_t size, size_t compressed_size, | ||
| 130 | struct page **compressed_pages) | ||
| 131 | { | ||
| 132 | struct btrfs_key key; | ||
| 133 | struct btrfs_path *path; | ||
| 134 | struct extent_buffer *leaf; | ||
| 135 | struct page *page = NULL; | ||
| 136 | char *kaddr; | ||
| 137 | unsigned long ptr; | ||
| 138 | struct btrfs_file_extent_item *ei; | ||
| 139 | int err = 0; | ||
| 140 | int ret; | ||
| 141 | size_t cur_size = size; | ||
| 142 | size_t datasize; | ||
| 143 | unsigned long offset; | ||
| 144 | int use_compress = 0; | ||
| 145 | |||
| 146 | if (compressed_size && compressed_pages) { | ||
| 147 | use_compress = 1; | ||
| 148 | cur_size = compressed_size; | ||
| 149 | } | ||
| 150 | |||
| 151 | path = btrfs_alloc_path(); | ||
| 152 | if (!path) | ||
| 153 | return -ENOMEM; | ||
| 154 | |||
| 155 | btrfs_set_trans_block_group(trans, inode); | ||
| 156 | |||
| 157 | key.objectid = inode->i_ino; | ||
| 158 | key.offset = start; | ||
| 159 | btrfs_set_key_type(&key, BTRFS_EXTENT_DATA_KEY); | ||
| 160 | datasize = btrfs_file_extent_calc_inline_size(cur_size); | ||
| 161 | |||
| 162 | inode_add_bytes(inode, size); | ||
| 163 | ret = btrfs_insert_empty_item(trans, root, path, &key, | ||
| 164 | datasize); | ||
| 165 | BUG_ON(ret); | ||
| 166 | if (ret) { | ||
| 167 | err = ret; | ||
| 168 | goto fail; | ||
| 169 | } | ||
| 170 | leaf = path->nodes[0]; | ||
| 171 | ei = btrfs_item_ptr(leaf, path->slots[0], | ||
| 172 | struct btrfs_file_extent_item); | ||
| 173 | btrfs_set_file_extent_generation(leaf, ei, trans->transid); | ||
| 174 | btrfs_set_file_extent_type(leaf, ei, BTRFS_FILE_EXTENT_INLINE); | ||
| 175 | btrfs_set_file_extent_encryption(leaf, ei, 0); | ||
| 176 | btrfs_set_file_extent_other_encoding(leaf, ei, 0); | ||
| 177 | btrfs_set_file_extent_ram_bytes(leaf, ei, size); | ||
| 178 | ptr = btrfs_file_extent_inline_start(ei); | ||
| 179 | |||
| 180 | if (use_compress) { | ||
| 181 | struct page *cpage; | ||
| 182 | int i = 0; | ||
| 183 | while (compressed_size > 0) { | ||
| 184 | cpage = compressed_pages[i]; | ||
| 185 | cur_size = min_t(unsigned long, compressed_size, | ||
| 186 | PAGE_CACHE_SIZE); | ||
| 187 | |||
| 188 | kaddr = kmap(cpage); | ||
| 189 | write_extent_buffer(leaf, kaddr, ptr, cur_size); | ||
| 190 | kunmap(cpage); | ||
| 191 | |||
| 192 | i++; | ||
| 193 | ptr += cur_size; | ||
| 194 | compressed_size -= cur_size; | ||
| 195 | } | ||
| 196 | btrfs_set_file_extent_compression(leaf, ei, | ||
| 197 | BTRFS_COMPRESS_ZLIB); | ||
| 198 | } else { | ||
| 199 | page = find_get_page(inode->i_mapping, | ||
| 200 | start >> PAGE_CACHE_SHIFT); | ||
| 201 | btrfs_set_file_extent_compression(leaf, ei, 0); | ||
| 202 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 203 | offset = start & (PAGE_CACHE_SIZE - 1); | ||
| 204 | write_extent_buffer(leaf, kaddr + offset, ptr, size); | ||
| 205 | kunmap_atomic(kaddr, KM_USER0); | ||
| 206 | page_cache_release(page); | ||
| 207 | } | ||
| 208 | btrfs_mark_buffer_dirty(leaf); | ||
| 209 | btrfs_free_path(path); | ||
| 210 | |||
| 211 | BTRFS_I(inode)->disk_i_size = inode->i_size; | ||
| 212 | btrfs_update_inode(trans, root, inode); | ||
| 213 | return 0; | ||
| 214 | fail: | ||
| 215 | btrfs_free_path(path); | ||
| 216 | return err; | ||
| 217 | } | ||
| 218 | |||
| 219 | |||
| 220 | /* | ||
| 221 | * conditionally insert an inline extent into the file. This | ||
| 222 | * does the checks required to make sure the data is small enough | ||
| 223 | * to fit as an inline extent. | ||
| 224 | */ | ||
| 225 | static int cow_file_range_inline(struct btrfs_trans_handle *trans, | ||
| 226 | struct btrfs_root *root, | ||
| 227 | struct inode *inode, u64 start, u64 end, | ||
| 228 | size_t compressed_size, | ||
| 229 | struct page **compressed_pages) | ||
| 230 | { | ||
| 231 | u64 isize = i_size_read(inode); | ||
| 232 | u64 actual_end = min(end + 1, isize); | ||
| 233 | u64 inline_len = actual_end - start; | ||
| 234 | u64 aligned_end = (end + root->sectorsize - 1) & | ||
| 235 | ~((u64)root->sectorsize - 1); | ||
| 236 | u64 hint_byte; | ||
| 237 | u64 data_len = inline_len; | ||
| 238 | int ret; | ||
| 239 | |||
| 240 | if (compressed_size) | ||
| 241 | data_len = compressed_size; | ||
| 242 | |||
| 243 | if (start > 0 || | ||
| 244 | actual_end >= PAGE_CACHE_SIZE || | ||
| 245 | data_len >= BTRFS_MAX_INLINE_DATA_SIZE(root) || | ||
| 246 | (!compressed_size && | ||
| 247 | (actual_end & (root->sectorsize - 1)) == 0) || | ||
| 248 | end + 1 < isize || | ||
| 249 | data_len > root->fs_info->max_inline) { | ||
| 250 | return 1; | ||
| 251 | } | ||
| 252 | |||
| 253 | ret = btrfs_drop_extents(trans, root, inode, start, | ||
| 254 | aligned_end, start, &hint_byte); | ||
| 255 | BUG_ON(ret); | ||
| 256 | |||
| 257 | if (isize > actual_end) | ||
| 258 | inline_len = min_t(u64, isize, actual_end); | ||
| 259 | ret = insert_inline_extent(trans, root, inode, start, | ||
| 260 | inline_len, compressed_size, | ||
| 261 | compressed_pages); | ||
| 262 | BUG_ON(ret); | ||
| 263 | btrfs_drop_extent_cache(inode, start, aligned_end, 0); | ||
| 264 | return 0; | ||
| 265 | } | ||
| 266 | |||
| 267 | struct async_extent { | ||
| 268 | u64 start; | ||
| 269 | u64 ram_size; | ||
| 270 | u64 compressed_size; | ||
| 271 | struct page **pages; | ||
| 272 | unsigned long nr_pages; | ||
| 273 | struct list_head list; | ||
| 274 | }; | ||
| 275 | |||
| 276 | struct async_cow { | ||
| 277 | struct inode *inode; | ||
| 278 | struct btrfs_root *root; | ||
| 279 | struct page *locked_page; | ||
| 280 | u64 start; | ||
| 281 | u64 end; | ||
| 282 | struct list_head extents; | ||
| 283 | struct btrfs_work work; | ||
| 284 | }; | ||
| 285 | |||
| 286 | static noinline int add_async_extent(struct async_cow *cow, | ||
| 287 | u64 start, u64 ram_size, | ||
| 288 | u64 compressed_size, | ||
| 289 | struct page **pages, | ||
| 290 | unsigned long nr_pages) | ||
| 291 | { | ||
| 292 | struct async_extent *async_extent; | ||
| 293 | |||
| 294 | async_extent = kmalloc(sizeof(*async_extent), GFP_NOFS); | ||
| 295 | async_extent->start = start; | ||
| 296 | async_extent->ram_size = ram_size; | ||
| 297 | async_extent->compressed_size = compressed_size; | ||
| 298 | async_extent->pages = pages; | ||
| 299 | async_extent->nr_pages = nr_pages; | ||
| 300 | list_add_tail(&async_extent->list, &cow->extents); | ||
| 301 | return 0; | ||
| 302 | } | ||
| 303 | |||
| 304 | /* | ||
| 305 | * we create compressed extents in two phases. The first | ||
| 306 | * phase compresses a range of pages that have already been | ||
| 307 | * locked (both pages and state bits are locked). | ||
| 308 | * | ||
| 309 | * This is done inside an ordered work queue, and the compression | ||
| 310 | * is spread across many cpus. The actual IO submission is step | ||
| 311 | * two, and the ordered work queue takes care of making sure that | ||
| 312 | * happens in the same order things were put onto the queue by | ||
| 313 | * writepages and friends. | ||
| 314 | * | ||
| 315 | * If this code finds it can't get good compression, it puts an | ||
| 316 | * entry onto the work queue to write the uncompressed bytes. This | ||
| 317 | * makes sure that both compressed inodes and uncompressed inodes | ||
| 318 | * are written in the same order that pdflush sent them down. | ||
| 319 | */ | ||
| 320 | static noinline int compress_file_range(struct inode *inode, | ||
| 321 | struct page *locked_page, | ||
| 322 | u64 start, u64 end, | ||
| 323 | struct async_cow *async_cow, | ||
| 324 | int *num_added) | ||
| 325 | { | ||
| 326 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 327 | struct btrfs_trans_handle *trans; | ||
| 328 | u64 num_bytes; | ||
| 329 | u64 orig_start; | ||
| 330 | u64 disk_num_bytes; | ||
| 331 | u64 blocksize = root->sectorsize; | ||
| 332 | u64 actual_end; | ||
| 333 | u64 isize = i_size_read(inode); | ||
| 334 | int ret = 0; | ||
| 335 | struct page **pages = NULL; | ||
| 336 | unsigned long nr_pages; | ||
| 337 | unsigned long nr_pages_ret = 0; | ||
| 338 | unsigned long total_compressed = 0; | ||
| 339 | unsigned long total_in = 0; | ||
| 340 | unsigned long max_compressed = 128 * 1024; | ||
| 341 | unsigned long max_uncompressed = 128 * 1024; | ||
| 342 | int i; | ||
| 343 | int will_compress; | ||
| 344 | |||
| 345 | orig_start = start; | ||
| 346 | |||
| 347 | actual_end = min_t(u64, isize, end + 1); | ||
| 348 | again: | ||
| 349 | will_compress = 0; | ||
| 350 | nr_pages = (end >> PAGE_CACHE_SHIFT) - (start >> PAGE_CACHE_SHIFT) + 1; | ||
| 351 | nr_pages = min(nr_pages, (128 * 1024UL) / PAGE_CACHE_SIZE); | ||
| 352 | |||
| 353 | total_compressed = actual_end - start; | ||
| 354 | |||
| 355 | /* we want to make sure that amount of ram required to uncompress | ||
| 356 | * an extent is reasonable, so we limit the total size in ram | ||
| 357 | * of a compressed extent to 128k. This is a crucial number | ||
| 358 | * because it also controls how easily we can spread reads across | ||
| 359 | * cpus for decompression. | ||
| 360 | * | ||
| 361 | * We also want to make sure the amount of IO required to do | ||
| 362 | * a random read is reasonably small, so we limit the size of | ||
| 363 | * a compressed extent to 128k. | ||
| 364 | */ | ||
| 365 | total_compressed = min(total_compressed, max_uncompressed); | ||
| 366 | num_bytes = (end - start + blocksize) & ~(blocksize - 1); | ||
| 367 | num_bytes = max(blocksize, num_bytes); | ||
| 368 | disk_num_bytes = num_bytes; | ||
| 369 | total_in = 0; | ||
| 370 | ret = 0; | ||
| 371 | |||
| 372 | /* | ||
| 373 | * we do compression for mount -o compress and when the | ||
| 374 | * inode has not been flagged as nocompress. This flag can | ||
| 375 | * change at any time if we discover bad compression ratios. | ||
| 376 | */ | ||
| 377 | if (!btrfs_test_flag(inode, NOCOMPRESS) && | ||
| 378 | btrfs_test_opt(root, COMPRESS)) { | ||
| 379 | WARN_ON(pages); | ||
| 380 | pages = kzalloc(sizeof(struct page *) * nr_pages, GFP_NOFS); | ||
| 381 | |||
| 382 | ret = btrfs_zlib_compress_pages(inode->i_mapping, start, | ||
| 383 | total_compressed, pages, | ||
| 384 | nr_pages, &nr_pages_ret, | ||
| 385 | &total_in, | ||
| 386 | &total_compressed, | ||
| 387 | max_compressed); | ||
| 388 | |||
| 389 | if (!ret) { | ||
| 390 | unsigned long offset = total_compressed & | ||
| 391 | (PAGE_CACHE_SIZE - 1); | ||
| 392 | struct page *page = pages[nr_pages_ret - 1]; | ||
| 393 | char *kaddr; | ||
| 394 | |||
| 395 | /* zero the tail end of the last page, we might be | ||
| 396 | * sending it down to disk | ||
| 397 | */ | ||
| 398 | if (offset) { | ||
| 399 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 400 | memset(kaddr + offset, 0, | ||
| 401 | PAGE_CACHE_SIZE - offset); | ||
| 402 | kunmap_atomic(kaddr, KM_USER0); | ||
| 403 | } | ||
| 404 | will_compress = 1; | ||
| 405 | } | ||
| 406 | } | ||
| 407 | if (start == 0) { | ||
| 408 | trans = btrfs_join_transaction(root, 1); | ||
| 409 | BUG_ON(!trans); | ||
| 410 | btrfs_set_trans_block_group(trans, inode); | ||
| 411 | |||
| 412 | /* lets try to make an inline extent */ | ||
| 413 | if (ret || total_in < (actual_end - start)) { | ||
| 414 | /* we didn't compress the entire range, try | ||
| 415 | * to make an uncompressed inline extent. | ||
| 416 | */ | ||
| 417 | ret = cow_file_range_inline(trans, root, inode, | ||
| 418 | start, end, 0, NULL); | ||
| 419 | } else { | ||
| 420 | /* try making a compressed inline extent */ | ||
| 421 | ret = cow_file_range_inline(trans, root, inode, | ||
| 422 | start, end, | ||
| 423 | total_compressed, pages); | ||
| 424 | } | ||
| 425 | btrfs_end_transaction(trans, root); | ||
| 426 | if (ret == 0) { | ||
| 427 | /* | ||
| 428 | * inline extent creation worked, we don't need | ||
| 429 | * to create any more async work items. Unlock | ||
| 430 | * and free up our temp pages. | ||
| 431 | */ | ||
| 432 | extent_clear_unlock_delalloc(inode, | ||
| 433 | &BTRFS_I(inode)->io_tree, | ||
| 434 | start, end, NULL, 1, 0, | ||
| 435 | 0, 1, 1, 1); | ||
| 436 | ret = 0; | ||
| 437 | goto free_pages_out; | ||
| 438 | } | ||
| 439 | } | ||
| 440 | |||
| 441 | if (will_compress) { | ||
| 442 | /* | ||
| 443 | * we aren't doing an inline extent round the compressed size | ||
| 444 | * up to a block size boundary so the allocator does sane | ||
| 445 | * things | ||
| 446 | */ | ||
| 447 | total_compressed = (total_compressed + blocksize - 1) & | ||
| 448 | ~(blocksize - 1); | ||
| 449 | |||
| 450 | /* | ||
| 451 | * one last check to make sure the compression is really a | ||
| 452 | * win, compare the page count read with the blocks on disk | ||
| 453 | */ | ||
| 454 | total_in = (total_in + PAGE_CACHE_SIZE - 1) & | ||
| 455 | ~(PAGE_CACHE_SIZE - 1); | ||
| 456 | if (total_compressed >= total_in) { | ||
| 457 | will_compress = 0; | ||
| 458 | } else { | ||
| 459 | disk_num_bytes = total_compressed; | ||
| 460 | num_bytes = total_in; | ||
| 461 | } | ||
| 462 | } | ||
| 463 | if (!will_compress && pages) { | ||
| 464 | /* | ||
| 465 | * the compression code ran but failed to make things smaller, | ||
| 466 | * free any pages it allocated and our page pointer array | ||
| 467 | */ | ||
| 468 | for (i = 0; i < nr_pages_ret; i++) { | ||
| 469 | WARN_ON(pages[i]->mapping); | ||
| 470 | page_cache_release(pages[i]); | ||
| 471 | } | ||
| 472 | kfree(pages); | ||
| 473 | pages = NULL; | ||
| 474 | total_compressed = 0; | ||
| 475 | nr_pages_ret = 0; | ||
| 476 | |||
| 477 | /* flag the file so we don't compress in the future */ | ||
| 478 | btrfs_set_flag(inode, NOCOMPRESS); | ||
| 479 | } | ||
| 480 | if (will_compress) { | ||
| 481 | *num_added += 1; | ||
| 482 | |||
| 483 | /* the async work queues will take care of doing actual | ||
| 484 | * allocation on disk for these compressed pages, | ||
| 485 | * and will submit them to the elevator. | ||
| 486 | */ | ||
| 487 | add_async_extent(async_cow, start, num_bytes, | ||
| 488 | total_compressed, pages, nr_pages_ret); | ||
| 489 | |||
| 490 | if (start + num_bytes < end && start + num_bytes < actual_end) { | ||
| 491 | start += num_bytes; | ||
| 492 | pages = NULL; | ||
| 493 | cond_resched(); | ||
| 494 | goto again; | ||
| 495 | } | ||
| 496 | } else { | ||
| 497 | /* | ||
| 498 | * No compression, but we still need to write the pages in | ||
| 499 | * the file we've been given so far. redirty the locked | ||
| 500 | * page if it corresponds to our extent and set things up | ||
| 501 | * for the async work queue to run cow_file_range to do | ||
| 502 | * the normal delalloc dance | ||
| 503 | */ | ||
| 504 | if (page_offset(locked_page) >= start && | ||
| 505 | page_offset(locked_page) <= end) { | ||
| 506 | __set_page_dirty_nobuffers(locked_page); | ||
| 507 | /* unlocked later on in the async handlers */ | ||
| 508 | } | ||
| 509 | add_async_extent(async_cow, start, end - start + 1, 0, NULL, 0); | ||
| 510 | *num_added += 1; | ||
| 511 | } | ||
| 512 | |||
| 513 | out: | ||
| 514 | return 0; | ||
| 515 | |||
| 516 | free_pages_out: | ||
| 517 | for (i = 0; i < nr_pages_ret; i++) { | ||
| 518 | WARN_ON(pages[i]->mapping); | ||
| 519 | page_cache_release(pages[i]); | ||
| 520 | } | ||
| 521 | kfree(pages); | ||
| 522 | |||
| 523 | goto out; | ||
| 524 | } | ||
| 525 | |||
| 526 | /* | ||
| 527 | * phase two of compressed writeback. This is the ordered portion | ||
| 528 | * of the code, which only gets called in the order the work was | ||
| 529 | * queued. We walk all the async extents created by compress_file_range | ||
| 530 | * and send them down to the disk. | ||
| 531 | */ | ||
| 532 | static noinline int submit_compressed_extents(struct inode *inode, | ||
| 533 | struct async_cow *async_cow) | ||
| 534 | { | ||
| 535 | struct async_extent *async_extent; | ||
| 536 | u64 alloc_hint = 0; | ||
| 537 | struct btrfs_trans_handle *trans; | ||
| 538 | struct btrfs_key ins; | ||
| 539 | struct extent_map *em; | ||
| 540 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 541 | struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 542 | struct extent_io_tree *io_tree; | ||
| 543 | int ret; | ||
| 544 | |||
| 545 | if (list_empty(&async_cow->extents)) | ||
| 546 | return 0; | ||
| 547 | |||
| 548 | trans = btrfs_join_transaction(root, 1); | ||
| 549 | |||
| 550 | while (!list_empty(&async_cow->extents)) { | ||
| 551 | async_extent = list_entry(async_cow->extents.next, | ||
| 552 | struct async_extent, list); | ||
| 553 | list_del(&async_extent->list); | ||
| 554 | |||
| 555 | io_tree = &BTRFS_I(inode)->io_tree; | ||
| 556 | |||
| 557 | /* did the compression code fall back to uncompressed IO? */ | ||
| 558 | if (!async_extent->pages) { | ||
| 559 | int page_started = 0; | ||
| 560 | unsigned long nr_written = 0; | ||
| 561 | |||
| 562 | lock_extent(io_tree, async_extent->start, | ||
| 563 | async_extent->start + | ||
| 564 | async_extent->ram_size - 1, GFP_NOFS); | ||
| 565 | |||
| 566 | /* allocate blocks */ | ||
| 567 | cow_file_range(inode, async_cow->locked_page, | ||
| 568 | async_extent->start, | ||
| 569 | async_extent->start + | ||
| 570 | async_extent->ram_size - 1, | ||
| 571 | &page_started, &nr_written, 0); | ||
| 572 | |||
| 573 | /* | ||
| 574 | * if page_started, cow_file_range inserted an | ||
| 575 | * inline extent and took care of all the unlocking | ||
| 576 | * and IO for us. Otherwise, we need to submit | ||
| 577 | * all those pages down to the drive. | ||
| 578 | */ | ||
| 579 | if (!page_started) | ||
| 580 | extent_write_locked_range(io_tree, | ||
| 581 | inode, async_extent->start, | ||
| 582 | async_extent->start + | ||
| 583 | async_extent->ram_size - 1, | ||
| 584 | btrfs_get_extent, | ||
| 585 | WB_SYNC_ALL); | ||
| 586 | kfree(async_extent); | ||
| 587 | cond_resched(); | ||
| 588 | continue; | ||
| 589 | } | ||
| 590 | |||
| 591 | lock_extent(io_tree, async_extent->start, | ||
| 592 | async_extent->start + async_extent->ram_size - 1, | ||
| 593 | GFP_NOFS); | ||
| 594 | /* | ||
| 595 | * here we're doing allocation and writeback of the | ||
| 596 | * compressed pages | ||
| 597 | */ | ||
| 598 | btrfs_drop_extent_cache(inode, async_extent->start, | ||
| 599 | async_extent->start + | ||
| 600 | async_extent->ram_size - 1, 0); | ||
| 601 | |||
| 602 | ret = btrfs_reserve_extent(trans, root, | ||
| 603 | async_extent->compressed_size, | ||
| 604 | async_extent->compressed_size, | ||
| 605 | 0, alloc_hint, | ||
| 606 | (u64)-1, &ins, 1); | ||
| 607 | BUG_ON(ret); | ||
| 608 | em = alloc_extent_map(GFP_NOFS); | ||
| 609 | em->start = async_extent->start; | ||
| 610 | em->len = async_extent->ram_size; | ||
| 611 | em->orig_start = em->start; | ||
| 612 | |||
| 613 | em->block_start = ins.objectid; | ||
| 614 | em->block_len = ins.offset; | ||
| 615 | em->bdev = root->fs_info->fs_devices->latest_bdev; | ||
| 616 | set_bit(EXTENT_FLAG_PINNED, &em->flags); | ||
| 617 | set_bit(EXTENT_FLAG_COMPRESSED, &em->flags); | ||
| 618 | |||
| 619 | while (1) { | ||
| 620 | spin_lock(&em_tree->lock); | ||
| 621 | ret = add_extent_mapping(em_tree, em); | ||
| 622 | spin_unlock(&em_tree->lock); | ||
| 623 | if (ret != -EEXIST) { | ||
| 624 | free_extent_map(em); | ||
| 625 | break; | ||
| 626 | } | ||
| 627 | btrfs_drop_extent_cache(inode, async_extent->start, | ||
| 628 | async_extent->start + | ||
| 629 | async_extent->ram_size - 1, 0); | ||
| 630 | } | ||
| 631 | |||
| 632 | ret = btrfs_add_ordered_extent(inode, async_extent->start, | ||
| 633 | ins.objectid, | ||
| 634 | async_extent->ram_size, | ||
| 635 | ins.offset, | ||
| 636 | BTRFS_ORDERED_COMPRESSED); | ||
| 637 | BUG_ON(ret); | ||
| 638 | |||
| 639 | btrfs_end_transaction(trans, root); | ||
| 640 | |||
| 641 | /* | ||
| 642 | * clear dirty, set writeback and unlock the pages. | ||
| 643 | */ | ||
| 644 | extent_clear_unlock_delalloc(inode, | ||
| 645 | &BTRFS_I(inode)->io_tree, | ||
| 646 | async_extent->start, | ||
| 647 | async_extent->start + | ||
| 648 | async_extent->ram_size - 1, | ||
| 649 | NULL, 1, 1, 0, 1, 1, 0); | ||
| 650 | |||
| 651 | ret = btrfs_submit_compressed_write(inode, | ||
| 652 | async_extent->start, | ||
| 653 | async_extent->ram_size, | ||
| 654 | ins.objectid, | ||
| 655 | ins.offset, async_extent->pages, | ||
| 656 | async_extent->nr_pages); | ||
| 657 | |||
| 658 | BUG_ON(ret); | ||
| 659 | trans = btrfs_join_transaction(root, 1); | ||
| 660 | alloc_hint = ins.objectid + ins.offset; | ||
| 661 | kfree(async_extent); | ||
| 662 | cond_resched(); | ||
| 663 | } | ||
| 664 | |||
| 665 | btrfs_end_transaction(trans, root); | ||
| 666 | return 0; | ||
| 667 | } | ||
| 668 | |||
| 669 | /* | ||
| 670 | * when extent_io.c finds a delayed allocation range in the file, | ||
| 671 | * the call backs end up in this code. The basic idea is to | ||
| 672 | * allocate extents on disk for the range, and create ordered data structs | ||
| 673 | * in ram to track those extents. | ||
| 674 | * | ||
| 675 | * locked_page is the page that writepage had locked already. We use | ||
| 676 | * it to make sure we don't do extra locks or unlocks. | ||
| 677 | * | ||
| 678 | * *page_started is set to one if we unlock locked_page and do everything | ||
| 679 | * required to start IO on it. It may be clean and already done with | ||
| 680 | * IO when we return. | ||
| 681 | */ | ||
| 682 | static noinline int cow_file_range(struct inode *inode, | ||
| 683 | struct page *locked_page, | ||
| 684 | u64 start, u64 end, int *page_started, | ||
| 685 | unsigned long *nr_written, | ||
| 686 | int unlock) | ||
| 687 | { | ||
| 688 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 689 | struct btrfs_trans_handle *trans; | ||
| 690 | u64 alloc_hint = 0; | ||
| 691 | u64 num_bytes; | ||
| 692 | unsigned long ram_size; | ||
| 693 | u64 disk_num_bytes; | ||
| 694 | u64 cur_alloc_size; | ||
| 695 | u64 blocksize = root->sectorsize; | ||
| 696 | u64 actual_end; | ||
| 697 | u64 isize = i_size_read(inode); | ||
| 698 | struct btrfs_key ins; | ||
| 699 | struct extent_map *em; | ||
| 700 | struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 701 | int ret = 0; | ||
| 702 | |||
| 703 | trans = btrfs_join_transaction(root, 1); | ||
| 704 | BUG_ON(!trans); | ||
| 705 | btrfs_set_trans_block_group(trans, inode); | ||
| 706 | |||
| 707 | actual_end = min_t(u64, isize, end + 1); | ||
| 708 | |||
| 709 | num_bytes = (end - start + blocksize) & ~(blocksize - 1); | ||
| 710 | num_bytes = max(blocksize, num_bytes); | ||
| 711 | disk_num_bytes = num_bytes; | ||
| 712 | ret = 0; | ||
| 713 | |||
| 714 | if (start == 0) { | ||
| 715 | /* lets try to make an inline extent */ | ||
| 716 | ret = cow_file_range_inline(trans, root, inode, | ||
| 717 | start, end, 0, NULL); | ||
| 718 | if (ret == 0) { | ||
| 719 | extent_clear_unlock_delalloc(inode, | ||
| 720 | &BTRFS_I(inode)->io_tree, | ||
| 721 | start, end, NULL, 1, 1, | ||
| 722 | 1, 1, 1, 1); | ||
| 723 | *nr_written = *nr_written + | ||
| 724 | (end - start + PAGE_CACHE_SIZE) / PAGE_CACHE_SIZE; | ||
| 725 | *page_started = 1; | ||
| 726 | ret = 0; | ||
| 727 | goto out; | ||
| 728 | } | ||
| 729 | } | ||
| 730 | |||
| 731 | BUG_ON(disk_num_bytes > | ||
| 732 | btrfs_super_total_bytes(&root->fs_info->super_copy)); | ||
| 733 | |||
| 734 | btrfs_drop_extent_cache(inode, start, start + num_bytes - 1, 0); | ||
| 735 | |||
| 736 | while (disk_num_bytes > 0) { | ||
| 737 | cur_alloc_size = min(disk_num_bytes, root->fs_info->max_extent); | ||
| 738 | ret = btrfs_reserve_extent(trans, root, cur_alloc_size, | ||
| 739 | root->sectorsize, 0, alloc_hint, | ||
| 740 | (u64)-1, &ins, 1); | ||
| 741 | BUG_ON(ret); | ||
| 742 | |||
| 743 | em = alloc_extent_map(GFP_NOFS); | ||
| 744 | em->start = start; | ||
| 745 | em->orig_start = em->start; | ||
| 746 | |||
| 747 | ram_size = ins.offset; | ||
| 748 | em->len = ins.offset; | ||
| 749 | |||
| 750 | em->block_start = ins.objectid; | ||
| 751 | em->block_len = ins.offset; | ||
| 752 | em->bdev = root->fs_info->fs_devices->latest_bdev; | ||
| 753 | set_bit(EXTENT_FLAG_PINNED, &em->flags); | ||
| 754 | |||
| 755 | while (1) { | ||
| 756 | spin_lock(&em_tree->lock); | ||
| 757 | ret = add_extent_mapping(em_tree, em); | ||
| 758 | spin_unlock(&em_tree->lock); | ||
| 759 | if (ret != -EEXIST) { | ||
| 760 | free_extent_map(em); | ||
| 761 | break; | ||
| 762 | } | ||
| 763 | btrfs_drop_extent_cache(inode, start, | ||
| 764 | start + ram_size - 1, 0); | ||
| 765 | } | ||
| 766 | |||
| 767 | cur_alloc_size = ins.offset; | ||
| 768 | ret = btrfs_add_ordered_extent(inode, start, ins.objectid, | ||
| 769 | ram_size, cur_alloc_size, 0); | ||
| 770 | BUG_ON(ret); | ||
| 771 | |||
| 772 | if (root->root_key.objectid == | ||
| 773 | BTRFS_DATA_RELOC_TREE_OBJECTID) { | ||
| 774 | ret = btrfs_reloc_clone_csums(inode, start, | ||
| 775 | cur_alloc_size); | ||
| 776 | BUG_ON(ret); | ||
| 777 | } | ||
| 778 | |||
| 779 | if (disk_num_bytes < cur_alloc_size) | ||
| 780 | break; | ||
| 781 | |||
| 782 | /* we're not doing compressed IO, don't unlock the first | ||
| 783 | * page (which the caller expects to stay locked), don't | ||
| 784 | * clear any dirty bits and don't set any writeback bits | ||
| 785 | */ | ||
| 786 | extent_clear_unlock_delalloc(inode, &BTRFS_I(inode)->io_tree, | ||
| 787 | start, start + ram_size - 1, | ||
| 788 | locked_page, unlock, 1, | ||
| 789 | 1, 0, 0, 0); | ||
| 790 | disk_num_bytes -= cur_alloc_size; | ||
| 791 | num_bytes -= cur_alloc_size; | ||
| 792 | alloc_hint = ins.objectid + ins.offset; | ||
| 793 | start += cur_alloc_size; | ||
| 794 | } | ||
| 795 | out: | ||
| 796 | ret = 0; | ||
| 797 | btrfs_end_transaction(trans, root); | ||
| 798 | |||
| 799 | return ret; | ||
| 800 | } | ||
| 801 | |||
| 802 | /* | ||
| 803 | * work queue call back to started compression on a file and pages | ||
| 804 | */ | ||
| 805 | static noinline void async_cow_start(struct btrfs_work *work) | ||
| 806 | { | ||
| 807 | struct async_cow *async_cow; | ||
| 808 | int num_added = 0; | ||
| 809 | async_cow = container_of(work, struct async_cow, work); | ||
| 810 | |||
| 811 | compress_file_range(async_cow->inode, async_cow->locked_page, | ||
| 812 | async_cow->start, async_cow->end, async_cow, | ||
| 813 | &num_added); | ||
| 814 | if (num_added == 0) | ||
| 815 | async_cow->inode = NULL; | ||
| 816 | } | ||
| 817 | |||
| 818 | /* | ||
| 819 | * work queue call back to submit previously compressed pages | ||
| 820 | */ | ||
| 821 | static noinline void async_cow_submit(struct btrfs_work *work) | ||
| 822 | { | ||
| 823 | struct async_cow *async_cow; | ||
| 824 | struct btrfs_root *root; | ||
| 825 | unsigned long nr_pages; | ||
| 826 | |||
| 827 | async_cow = container_of(work, struct async_cow, work); | ||
| 828 | |||
| 829 | root = async_cow->root; | ||
| 830 | nr_pages = (async_cow->end - async_cow->start + PAGE_CACHE_SIZE) >> | ||
| 831 | PAGE_CACHE_SHIFT; | ||
| 832 | |||
| 833 | atomic_sub(nr_pages, &root->fs_info->async_delalloc_pages); | ||
| 834 | |||
| 835 | if (atomic_read(&root->fs_info->async_delalloc_pages) < | ||
| 836 | 5 * 1042 * 1024 && | ||
| 837 | waitqueue_active(&root->fs_info->async_submit_wait)) | ||
| 838 | wake_up(&root->fs_info->async_submit_wait); | ||
| 839 | |||
| 840 | if (async_cow->inode) | ||
| 841 | submit_compressed_extents(async_cow->inode, async_cow); | ||
| 842 | } | ||
| 843 | |||
| 844 | static noinline void async_cow_free(struct btrfs_work *work) | ||
| 845 | { | ||
| 846 | struct async_cow *async_cow; | ||
| 847 | async_cow = container_of(work, struct async_cow, work); | ||
| 848 | kfree(async_cow); | ||
| 849 | } | ||
| 850 | |||
| 851 | static int cow_file_range_async(struct inode *inode, struct page *locked_page, | ||
| 852 | u64 start, u64 end, int *page_started, | ||
| 853 | unsigned long *nr_written) | ||
| 854 | { | ||
| 855 | struct async_cow *async_cow; | ||
| 856 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 857 | unsigned long nr_pages; | ||
| 858 | u64 cur_end; | ||
| 859 | int limit = 10 * 1024 * 1042; | ||
| 860 | |||
| 861 | if (!btrfs_test_opt(root, COMPRESS)) { | ||
| 862 | return cow_file_range(inode, locked_page, start, end, | ||
| 863 | page_started, nr_written, 1); | ||
| 864 | } | ||
| 865 | |||
| 866 | clear_extent_bit(&BTRFS_I(inode)->io_tree, start, end, EXTENT_LOCKED | | ||
| 867 | EXTENT_DELALLOC, 1, 0, GFP_NOFS); | ||
| 868 | while (start < end) { | ||
| 869 | async_cow = kmalloc(sizeof(*async_cow), GFP_NOFS); | ||
| 870 | async_cow->inode = inode; | ||
| 871 | async_cow->root = root; | ||
| 872 | async_cow->locked_page = locked_page; | ||
| 873 | async_cow->start = start; | ||
| 874 | |||
| 875 | if (btrfs_test_flag(inode, NOCOMPRESS)) | ||
| 876 | cur_end = end; | ||
| 877 | else | ||
| 878 | cur_end = min(end, start + 512 * 1024 - 1); | ||
| 879 | |||
| 880 | async_cow->end = cur_end; | ||
| 881 | INIT_LIST_HEAD(&async_cow->extents); | ||
| 882 | |||
| 883 | async_cow->work.func = async_cow_start; | ||
| 884 | async_cow->work.ordered_func = async_cow_submit; | ||
| 885 | async_cow->work.ordered_free = async_cow_free; | ||
| 886 | async_cow->work.flags = 0; | ||
| 887 | |||
| 888 | nr_pages = (cur_end - start + PAGE_CACHE_SIZE) >> | ||
| 889 | PAGE_CACHE_SHIFT; | ||
| 890 | atomic_add(nr_pages, &root->fs_info->async_delalloc_pages); | ||
| 891 | |||
| 892 | btrfs_queue_worker(&root->fs_info->delalloc_workers, | ||
| 893 | &async_cow->work); | ||
| 894 | |||
| 895 | if (atomic_read(&root->fs_info->async_delalloc_pages) > limit) { | ||
| 896 | wait_event(root->fs_info->async_submit_wait, | ||
| 897 | (atomic_read(&root->fs_info->async_delalloc_pages) < | ||
| 898 | limit)); | ||
| 899 | } | ||
| 900 | |||
| 901 | while (atomic_read(&root->fs_info->async_submit_draining) && | ||
| 902 | atomic_read(&root->fs_info->async_delalloc_pages)) { | ||
| 903 | wait_event(root->fs_info->async_submit_wait, | ||
| 904 | (atomic_read(&root->fs_info->async_delalloc_pages) == | ||
| 905 | 0)); | ||
| 906 | } | ||
| 907 | |||
| 908 | *nr_written += nr_pages; | ||
| 909 | start = cur_end + 1; | ||
| 910 | } | ||
| 911 | *page_started = 1; | ||
| 912 | return 0; | ||
| 913 | } | ||
| 914 | |||
| 915 | static noinline int csum_exist_in_range(struct btrfs_root *root, | ||
| 916 | u64 bytenr, u64 num_bytes) | ||
| 917 | { | ||
| 918 | int ret; | ||
| 919 | struct btrfs_ordered_sum *sums; | ||
| 920 | LIST_HEAD(list); | ||
| 921 | |||
| 922 | ret = btrfs_lookup_csums_range(root->fs_info->csum_root, bytenr, | ||
| 923 | bytenr + num_bytes - 1, &list); | ||
| 924 | if (ret == 0 && list_empty(&list)) | ||
| 925 | return 0; | ||
| 926 | |||
| 927 | while (!list_empty(&list)) { | ||
| 928 | sums = list_entry(list.next, struct btrfs_ordered_sum, list); | ||
| 929 | list_del(&sums->list); | ||
| 930 | kfree(sums); | ||
| 931 | } | ||
| 932 | return 1; | ||
| 933 | } | ||
| 934 | |||
| 935 | /* | ||
| 936 | * when nowcow writeback call back. This checks for snapshots or COW copies | ||
| 937 | * of the extents that exist in the file, and COWs the file as required. | ||
| 938 | * | ||
| 939 | * If no cow copies or snapshots exist, we write directly to the existing | ||
| 940 | * blocks on disk | ||
| 941 | */ | ||
| 942 | static int run_delalloc_nocow(struct inode *inode, struct page *locked_page, | ||
| 943 | u64 start, u64 end, int *page_started, int force, | ||
| 944 | unsigned long *nr_written) | ||
| 945 | { | ||
| 946 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 947 | struct btrfs_trans_handle *trans; | ||
| 948 | struct extent_buffer *leaf; | ||
| 949 | struct btrfs_path *path; | ||
| 950 | struct btrfs_file_extent_item *fi; | ||
| 951 | struct btrfs_key found_key; | ||
| 952 | u64 cow_start; | ||
| 953 | u64 cur_offset; | ||
| 954 | u64 extent_end; | ||
| 955 | u64 disk_bytenr; | ||
| 956 | u64 num_bytes; | ||
| 957 | int extent_type; | ||
| 958 | int ret; | ||
| 959 | int type; | ||
| 960 | int nocow; | ||
| 961 | int check_prev = 1; | ||
| 962 | |||
| 963 | path = btrfs_alloc_path(); | ||
| 964 | BUG_ON(!path); | ||
| 965 | trans = btrfs_join_transaction(root, 1); | ||
| 966 | BUG_ON(!trans); | ||
| 967 | |||
| 968 | cow_start = (u64)-1; | ||
| 969 | cur_offset = start; | ||
| 970 | while (1) { | ||
| 971 | ret = btrfs_lookup_file_extent(trans, root, path, inode->i_ino, | ||
| 972 | cur_offset, 0); | ||
| 973 | BUG_ON(ret < 0); | ||
| 974 | if (ret > 0 && path->slots[0] > 0 && check_prev) { | ||
| 975 | leaf = path->nodes[0]; | ||
| 976 | btrfs_item_key_to_cpu(leaf, &found_key, | ||
| 977 | path->slots[0] - 1); | ||
| 978 | if (found_key.objectid == inode->i_ino && | ||
| 979 | found_key.type == BTRFS_EXTENT_DATA_KEY) | ||
| 980 | path->slots[0]--; | ||
| 981 | } | ||
| 982 | check_prev = 0; | ||
| 983 | next_slot: | ||
| 984 | leaf = path->nodes[0]; | ||
| 985 | if (path->slots[0] >= btrfs_header_nritems(leaf)) { | ||
| 986 | ret = btrfs_next_leaf(root, path); | ||
| 987 | if (ret < 0) | ||
| 988 | BUG_ON(1); | ||
| 989 | if (ret > 0) | ||
| 990 | break; | ||
| 991 | leaf = path->nodes[0]; | ||
| 992 | } | ||
| 993 | |||
| 994 | nocow = 0; | ||
| 995 | disk_bytenr = 0; | ||
| 996 | num_bytes = 0; | ||
| 997 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 998 | |||
| 999 | if (found_key.objectid > inode->i_ino || | ||
| 1000 | found_key.type > BTRFS_EXTENT_DATA_KEY || | ||
| 1001 | found_key.offset > end) | ||
| 1002 | break; | ||
| 1003 | |||
| 1004 | if (found_key.offset > cur_offset) { | ||
| 1005 | extent_end = found_key.offset; | ||
| 1006 | goto out_check; | ||
| 1007 | } | ||
| 1008 | |||
| 1009 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 1010 | struct btrfs_file_extent_item); | ||
| 1011 | extent_type = btrfs_file_extent_type(leaf, fi); | ||
| 1012 | |||
| 1013 | if (extent_type == BTRFS_FILE_EXTENT_REG || | ||
| 1014 | extent_type == BTRFS_FILE_EXTENT_PREALLOC) { | ||
| 1015 | disk_bytenr = btrfs_file_extent_disk_bytenr(leaf, fi); | ||
| 1016 | extent_end = found_key.offset + | ||
| 1017 | btrfs_file_extent_num_bytes(leaf, fi); | ||
| 1018 | if (extent_end <= start) { | ||
| 1019 | path->slots[0]++; | ||
| 1020 | goto next_slot; | ||
| 1021 | } | ||
| 1022 | if (disk_bytenr == 0) | ||
| 1023 | goto out_check; | ||
| 1024 | if (btrfs_file_extent_compression(leaf, fi) || | ||
| 1025 | btrfs_file_extent_encryption(leaf, fi) || | ||
| 1026 | btrfs_file_extent_other_encoding(leaf, fi)) | ||
| 1027 | goto out_check; | ||
| 1028 | if (extent_type == BTRFS_FILE_EXTENT_REG && !force) | ||
| 1029 | goto out_check; | ||
| 1030 | if (btrfs_extent_readonly(root, disk_bytenr)) | ||
| 1031 | goto out_check; | ||
| 1032 | if (btrfs_cross_ref_exist(trans, root, inode->i_ino, | ||
| 1033 | disk_bytenr)) | ||
| 1034 | goto out_check; | ||
| 1035 | disk_bytenr += btrfs_file_extent_offset(leaf, fi); | ||
| 1036 | disk_bytenr += cur_offset - found_key.offset; | ||
| 1037 | num_bytes = min(end + 1, extent_end) - cur_offset; | ||
| 1038 | /* | ||
| 1039 | * force cow if csum exists in the range. | ||
| 1040 | * this ensure that csum for a given extent are | ||
| 1041 | * either valid or do not exist. | ||
| 1042 | */ | ||
| 1043 | if (csum_exist_in_range(root, disk_bytenr, num_bytes)) | ||
| 1044 | goto out_check; | ||
| 1045 | nocow = 1; | ||
| 1046 | } else if (extent_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 1047 | extent_end = found_key.offset + | ||
| 1048 | btrfs_file_extent_inline_len(leaf, fi); | ||
| 1049 | extent_end = ALIGN(extent_end, root->sectorsize); | ||
| 1050 | } else { | ||
| 1051 | BUG_ON(1); | ||
| 1052 | } | ||
| 1053 | out_check: | ||
| 1054 | if (extent_end <= start) { | ||
| 1055 | path->slots[0]++; | ||
| 1056 | goto next_slot; | ||
| 1057 | } | ||
| 1058 | if (!nocow) { | ||
| 1059 | if (cow_start == (u64)-1) | ||
| 1060 | cow_start = cur_offset; | ||
| 1061 | cur_offset = extent_end; | ||
| 1062 | if (cur_offset > end) | ||
| 1063 | break; | ||
| 1064 | path->slots[0]++; | ||
| 1065 | goto next_slot; | ||
| 1066 | } | ||
| 1067 | |||
| 1068 | btrfs_release_path(root, path); | ||
| 1069 | if (cow_start != (u64)-1) { | ||
| 1070 | ret = cow_file_range(inode, locked_page, cow_start, | ||
| 1071 | found_key.offset - 1, page_started, | ||
| 1072 | nr_written, 1); | ||
| 1073 | BUG_ON(ret); | ||
| 1074 | cow_start = (u64)-1; | ||
| 1075 | } | ||
| 1076 | |||
| 1077 | if (extent_type == BTRFS_FILE_EXTENT_PREALLOC) { | ||
| 1078 | struct extent_map *em; | ||
| 1079 | struct extent_map_tree *em_tree; | ||
| 1080 | em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 1081 | em = alloc_extent_map(GFP_NOFS); | ||
| 1082 | em->start = cur_offset; | ||
| 1083 | em->orig_start = em->start; | ||
| 1084 | em->len = num_bytes; | ||
| 1085 | em->block_len = num_bytes; | ||
| 1086 | em->block_start = disk_bytenr; | ||
| 1087 | em->bdev = root->fs_info->fs_devices->latest_bdev; | ||
| 1088 | set_bit(EXTENT_FLAG_PINNED, &em->flags); | ||
| 1089 | while (1) { | ||
| 1090 | spin_lock(&em_tree->lock); | ||
| 1091 | ret = add_extent_mapping(em_tree, em); | ||
| 1092 | spin_unlock(&em_tree->lock); | ||
| 1093 | if (ret != -EEXIST) { | ||
| 1094 | free_extent_map(em); | ||
| 1095 | break; | ||
| 1096 | } | ||
| 1097 | btrfs_drop_extent_cache(inode, em->start, | ||
| 1098 | em->start + em->len - 1, 0); | ||
| 1099 | } | ||
| 1100 | type = BTRFS_ORDERED_PREALLOC; | ||
| 1101 | } else { | ||
| 1102 | type = BTRFS_ORDERED_NOCOW; | ||
| 1103 | } | ||
| 1104 | |||
| 1105 | ret = btrfs_add_ordered_extent(inode, cur_offset, disk_bytenr, | ||
| 1106 | num_bytes, num_bytes, type); | ||
| 1107 | BUG_ON(ret); | ||
| 1108 | |||
| 1109 | extent_clear_unlock_delalloc(inode, &BTRFS_I(inode)->io_tree, | ||
| 1110 | cur_offset, cur_offset + num_bytes - 1, | ||
| 1111 | locked_page, 1, 1, 1, 0, 0, 0); | ||
| 1112 | cur_offset = extent_end; | ||
| 1113 | if (cur_offset > end) | ||
| 1114 | break; | ||
| 1115 | } | ||
| 1116 | btrfs_release_path(root, path); | ||
| 1117 | |||
| 1118 | if (cur_offset <= end && cow_start == (u64)-1) | ||
| 1119 | cow_start = cur_offset; | ||
| 1120 | if (cow_start != (u64)-1) { | ||
| 1121 | ret = cow_file_range(inode, locked_page, cow_start, end, | ||
| 1122 | page_started, nr_written, 1); | ||
| 1123 | BUG_ON(ret); | ||
| 1124 | } | ||
| 1125 | |||
| 1126 | ret = btrfs_end_transaction(trans, root); | ||
| 1127 | BUG_ON(ret); | ||
| 1128 | btrfs_free_path(path); | ||
| 1129 | return 0; | ||
| 1130 | } | ||
| 1131 | |||
| 1132 | /* | ||
| 1133 | * extent_io.c call back to do delayed allocation processing | ||
| 1134 | */ | ||
| 1135 | static int run_delalloc_range(struct inode *inode, struct page *locked_page, | ||
| 1136 | u64 start, u64 end, int *page_started, | ||
| 1137 | unsigned long *nr_written) | ||
| 1138 | { | ||
| 1139 | int ret; | ||
| 1140 | |||
| 1141 | if (btrfs_test_flag(inode, NODATACOW)) | ||
| 1142 | ret = run_delalloc_nocow(inode, locked_page, start, end, | ||
| 1143 | page_started, 1, nr_written); | ||
| 1144 | else if (btrfs_test_flag(inode, PREALLOC)) | ||
| 1145 | ret = run_delalloc_nocow(inode, locked_page, start, end, | ||
| 1146 | page_started, 0, nr_written); | ||
| 1147 | else | ||
| 1148 | ret = cow_file_range_async(inode, locked_page, start, end, | ||
| 1149 | page_started, nr_written); | ||
| 1150 | |||
| 1151 | return ret; | ||
| 1152 | } | ||
| 1153 | |||
| 1154 | /* | ||
| 1155 | * extent_io.c set_bit_hook, used to track delayed allocation | ||
| 1156 | * bytes in this file, and to maintain the list of inodes that | ||
| 1157 | * have pending delalloc work to be done. | ||
| 1158 | */ | ||
| 1159 | static int btrfs_set_bit_hook(struct inode *inode, u64 start, u64 end, | ||
| 1160 | unsigned long old, unsigned long bits) | ||
| 1161 | { | ||
| 1162 | /* | ||
| 1163 | * set_bit and clear bit hooks normally require _irqsave/restore | ||
| 1164 | * but in this case, we are only testeing for the DELALLOC | ||
| 1165 | * bit, which is only set or cleared with irqs on | ||
| 1166 | */ | ||
| 1167 | if (!(old & EXTENT_DELALLOC) && (bits & EXTENT_DELALLOC)) { | ||
| 1168 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1169 | spin_lock(&root->fs_info->delalloc_lock); | ||
| 1170 | BTRFS_I(inode)->delalloc_bytes += end - start + 1; | ||
| 1171 | root->fs_info->delalloc_bytes += end - start + 1; | ||
| 1172 | if (list_empty(&BTRFS_I(inode)->delalloc_inodes)) { | ||
| 1173 | list_add_tail(&BTRFS_I(inode)->delalloc_inodes, | ||
| 1174 | &root->fs_info->delalloc_inodes); | ||
| 1175 | } | ||
| 1176 | spin_unlock(&root->fs_info->delalloc_lock); | ||
| 1177 | } | ||
| 1178 | return 0; | ||
| 1179 | } | ||
| 1180 | |||
| 1181 | /* | ||
| 1182 | * extent_io.c clear_bit_hook, see set_bit_hook for why | ||
| 1183 | */ | ||
| 1184 | static int btrfs_clear_bit_hook(struct inode *inode, u64 start, u64 end, | ||
| 1185 | unsigned long old, unsigned long bits) | ||
| 1186 | { | ||
| 1187 | /* | ||
| 1188 | * set_bit and clear bit hooks normally require _irqsave/restore | ||
| 1189 | * but in this case, we are only testeing for the DELALLOC | ||
| 1190 | * bit, which is only set or cleared with irqs on | ||
| 1191 | */ | ||
| 1192 | if ((old & EXTENT_DELALLOC) && (bits & EXTENT_DELALLOC)) { | ||
| 1193 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1194 | |||
| 1195 | spin_lock(&root->fs_info->delalloc_lock); | ||
| 1196 | if (end - start + 1 > root->fs_info->delalloc_bytes) { | ||
| 1197 | printk(KERN_INFO "btrfs warning: delalloc account " | ||
| 1198 | "%llu %llu\n", | ||
| 1199 | (unsigned long long)end - start + 1, | ||
| 1200 | (unsigned long long) | ||
| 1201 | root->fs_info->delalloc_bytes); | ||
| 1202 | root->fs_info->delalloc_bytes = 0; | ||
| 1203 | BTRFS_I(inode)->delalloc_bytes = 0; | ||
| 1204 | } else { | ||
| 1205 | root->fs_info->delalloc_bytes -= end - start + 1; | ||
| 1206 | BTRFS_I(inode)->delalloc_bytes -= end - start + 1; | ||
| 1207 | } | ||
| 1208 | if (BTRFS_I(inode)->delalloc_bytes == 0 && | ||
| 1209 | !list_empty(&BTRFS_I(inode)->delalloc_inodes)) { | ||
| 1210 | list_del_init(&BTRFS_I(inode)->delalloc_inodes); | ||
| 1211 | } | ||
| 1212 | spin_unlock(&root->fs_info->delalloc_lock); | ||
| 1213 | } | ||
| 1214 | return 0; | ||
| 1215 | } | ||
| 1216 | |||
| 1217 | /* | ||
| 1218 | * extent_io.c merge_bio_hook, this must check the chunk tree to make sure | ||
| 1219 | * we don't create bios that span stripes or chunks | ||
| 1220 | */ | ||
| 1221 | int btrfs_merge_bio_hook(struct page *page, unsigned long offset, | ||
| 1222 | size_t size, struct bio *bio, | ||
| 1223 | unsigned long bio_flags) | ||
| 1224 | { | ||
| 1225 | struct btrfs_root *root = BTRFS_I(page->mapping->host)->root; | ||
| 1226 | struct btrfs_mapping_tree *map_tree; | ||
| 1227 | u64 logical = (u64)bio->bi_sector << 9; | ||
| 1228 | u64 length = 0; | ||
| 1229 | u64 map_length; | ||
| 1230 | int ret; | ||
| 1231 | |||
| 1232 | if (bio_flags & EXTENT_BIO_COMPRESSED) | ||
| 1233 | return 0; | ||
| 1234 | |||
| 1235 | length = bio->bi_size; | ||
| 1236 | map_tree = &root->fs_info->mapping_tree; | ||
| 1237 | map_length = length; | ||
| 1238 | ret = btrfs_map_block(map_tree, READ, logical, | ||
| 1239 | &map_length, NULL, 0); | ||
| 1240 | |||
| 1241 | if (map_length < length + size) | ||
| 1242 | return 1; | ||
| 1243 | return 0; | ||
| 1244 | } | ||
| 1245 | |||
| 1246 | /* | ||
| 1247 | * in order to insert checksums into the metadata in large chunks, | ||
| 1248 | * we wait until bio submission time. All the pages in the bio are | ||
| 1249 | * checksummed and sums are attached onto the ordered extent record. | ||
| 1250 | * | ||
| 1251 | * At IO completion time the cums attached on the ordered extent record | ||
| 1252 | * are inserted into the btree | ||
| 1253 | */ | ||
| 1254 | static int __btrfs_submit_bio_start(struct inode *inode, int rw, | ||
| 1255 | struct bio *bio, int mirror_num, | ||
| 1256 | unsigned long bio_flags) | ||
| 1257 | { | ||
| 1258 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1259 | int ret = 0; | ||
| 1260 | |||
| 1261 | ret = btrfs_csum_one_bio(root, inode, bio, 0, 0); | ||
| 1262 | BUG_ON(ret); | ||
| 1263 | return 0; | ||
| 1264 | } | ||
| 1265 | |||
| 1266 | /* | ||
| 1267 | * in order to insert checksums into the metadata in large chunks, | ||
| 1268 | * we wait until bio submission time. All the pages in the bio are | ||
| 1269 | * checksummed and sums are attached onto the ordered extent record. | ||
| 1270 | * | ||
| 1271 | * At IO completion time the cums attached on the ordered extent record | ||
| 1272 | * are inserted into the btree | ||
| 1273 | */ | ||
| 1274 | static int __btrfs_submit_bio_done(struct inode *inode, int rw, struct bio *bio, | ||
| 1275 | int mirror_num, unsigned long bio_flags) | ||
| 1276 | { | ||
| 1277 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1278 | return btrfs_map_bio(root, rw, bio, mirror_num, 1); | ||
| 1279 | } | ||
| 1280 | |||
| 1281 | /* | ||
| 1282 | * extent_io.c submission hook. This does the right thing for csum calculation | ||
| 1283 | * on write, or reading the csums from the tree before a read | ||
| 1284 | */ | ||
| 1285 | static int btrfs_submit_bio_hook(struct inode *inode, int rw, struct bio *bio, | ||
| 1286 | int mirror_num, unsigned long bio_flags) | ||
| 1287 | { | ||
| 1288 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1289 | int ret = 0; | ||
| 1290 | int skip_sum; | ||
| 1291 | |||
| 1292 | skip_sum = btrfs_test_flag(inode, NODATASUM); | ||
| 1293 | |||
| 1294 | ret = btrfs_bio_wq_end_io(root->fs_info, bio, 0); | ||
| 1295 | BUG_ON(ret); | ||
| 1296 | |||
| 1297 | if (!(rw & (1 << BIO_RW))) { | ||
| 1298 | if (bio_flags & EXTENT_BIO_COMPRESSED) { | ||
| 1299 | return btrfs_submit_compressed_read(inode, bio, | ||
| 1300 | mirror_num, bio_flags); | ||
| 1301 | } else if (!skip_sum) | ||
| 1302 | btrfs_lookup_bio_sums(root, inode, bio, NULL); | ||
| 1303 | goto mapit; | ||
| 1304 | } else if (!skip_sum) { | ||
| 1305 | /* csum items have already been cloned */ | ||
| 1306 | if (root->root_key.objectid == BTRFS_DATA_RELOC_TREE_OBJECTID) | ||
| 1307 | goto mapit; | ||
| 1308 | /* we're doing a write, do the async checksumming */ | ||
| 1309 | return btrfs_wq_submit_bio(BTRFS_I(inode)->root->fs_info, | ||
| 1310 | inode, rw, bio, mirror_num, | ||
| 1311 | bio_flags, __btrfs_submit_bio_start, | ||
| 1312 | __btrfs_submit_bio_done); | ||
| 1313 | } | ||
| 1314 | |||
| 1315 | mapit: | ||
| 1316 | return btrfs_map_bio(root, rw, bio, mirror_num, 0); | ||
| 1317 | } | ||
| 1318 | |||
| 1319 | /* | ||
| 1320 | * given a list of ordered sums record them in the inode. This happens | ||
| 1321 | * at IO completion time based on sums calculated at bio submission time. | ||
| 1322 | */ | ||
| 1323 | static noinline int add_pending_csums(struct btrfs_trans_handle *trans, | ||
| 1324 | struct inode *inode, u64 file_offset, | ||
| 1325 | struct list_head *list) | ||
| 1326 | { | ||
| 1327 | struct list_head *cur; | ||
| 1328 | struct btrfs_ordered_sum *sum; | ||
| 1329 | |||
| 1330 | btrfs_set_trans_block_group(trans, inode); | ||
| 1331 | list_for_each(cur, list) { | ||
| 1332 | sum = list_entry(cur, struct btrfs_ordered_sum, list); | ||
| 1333 | btrfs_csum_file_blocks(trans, | ||
| 1334 | BTRFS_I(inode)->root->fs_info->csum_root, sum); | ||
| 1335 | } | ||
| 1336 | return 0; | ||
| 1337 | } | ||
| 1338 | |||
| 1339 | int btrfs_set_extent_delalloc(struct inode *inode, u64 start, u64 end) | ||
| 1340 | { | ||
| 1341 | if ((end & (PAGE_CACHE_SIZE - 1)) == 0) | ||
| 1342 | WARN_ON(1); | ||
| 1343 | return set_extent_delalloc(&BTRFS_I(inode)->io_tree, start, end, | ||
| 1344 | GFP_NOFS); | ||
| 1345 | } | ||
| 1346 | |||
| 1347 | /* see btrfs_writepage_start_hook for details on why this is required */ | ||
| 1348 | struct btrfs_writepage_fixup { | ||
| 1349 | struct page *page; | ||
| 1350 | struct btrfs_work work; | ||
| 1351 | }; | ||
| 1352 | |||
| 1353 | static void btrfs_writepage_fixup_worker(struct btrfs_work *work) | ||
| 1354 | { | ||
| 1355 | struct btrfs_writepage_fixup *fixup; | ||
| 1356 | struct btrfs_ordered_extent *ordered; | ||
| 1357 | struct page *page; | ||
| 1358 | struct inode *inode; | ||
| 1359 | u64 page_start; | ||
| 1360 | u64 page_end; | ||
| 1361 | |||
| 1362 | fixup = container_of(work, struct btrfs_writepage_fixup, work); | ||
| 1363 | page = fixup->page; | ||
| 1364 | again: | ||
| 1365 | lock_page(page); | ||
| 1366 | if (!page->mapping || !PageDirty(page) || !PageChecked(page)) { | ||
| 1367 | ClearPageChecked(page); | ||
| 1368 | goto out_page; | ||
| 1369 | } | ||
| 1370 | |||
| 1371 | inode = page->mapping->host; | ||
| 1372 | page_start = page_offset(page); | ||
| 1373 | page_end = page_offset(page) + PAGE_CACHE_SIZE - 1; | ||
| 1374 | |||
| 1375 | lock_extent(&BTRFS_I(inode)->io_tree, page_start, page_end, GFP_NOFS); | ||
| 1376 | |||
| 1377 | /* already ordered? We're done */ | ||
| 1378 | if (test_range_bit(&BTRFS_I(inode)->io_tree, page_start, page_end, | ||
| 1379 | EXTENT_ORDERED, 0)) { | ||
| 1380 | goto out; | ||
| 1381 | } | ||
| 1382 | |||
| 1383 | ordered = btrfs_lookup_ordered_extent(inode, page_start); | ||
| 1384 | if (ordered) { | ||
| 1385 | unlock_extent(&BTRFS_I(inode)->io_tree, page_start, | ||
| 1386 | page_end, GFP_NOFS); | ||
| 1387 | unlock_page(page); | ||
| 1388 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 1389 | goto again; | ||
| 1390 | } | ||
| 1391 | |||
| 1392 | btrfs_set_extent_delalloc(inode, page_start, page_end); | ||
| 1393 | ClearPageChecked(page); | ||
| 1394 | out: | ||
| 1395 | unlock_extent(&BTRFS_I(inode)->io_tree, page_start, page_end, GFP_NOFS); | ||
| 1396 | out_page: | ||
| 1397 | unlock_page(page); | ||
| 1398 | page_cache_release(page); | ||
| 1399 | } | ||
| 1400 | |||
| 1401 | /* | ||
| 1402 | * There are a few paths in the higher layers of the kernel that directly | ||
| 1403 | * set the page dirty bit without asking the filesystem if it is a | ||
| 1404 | * good idea. This causes problems because we want to make sure COW | ||
| 1405 | * properly happens and the data=ordered rules are followed. | ||
| 1406 | * | ||
| 1407 | * In our case any range that doesn't have the ORDERED bit set | ||
| 1408 | * hasn't been properly setup for IO. We kick off an async process | ||
| 1409 | * to fix it up. The async helper will wait for ordered extents, set | ||
| 1410 | * the delalloc bit and make it safe to write the page. | ||
| 1411 | */ | ||
| 1412 | static int btrfs_writepage_start_hook(struct page *page, u64 start, u64 end) | ||
| 1413 | { | ||
| 1414 | struct inode *inode = page->mapping->host; | ||
| 1415 | struct btrfs_writepage_fixup *fixup; | ||
| 1416 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1417 | int ret; | ||
| 1418 | |||
| 1419 | ret = test_range_bit(&BTRFS_I(inode)->io_tree, start, end, | ||
| 1420 | EXTENT_ORDERED, 0); | ||
| 1421 | if (ret) | ||
| 1422 | return 0; | ||
| 1423 | |||
| 1424 | if (PageChecked(page)) | ||
| 1425 | return -EAGAIN; | ||
| 1426 | |||
| 1427 | fixup = kzalloc(sizeof(*fixup), GFP_NOFS); | ||
| 1428 | if (!fixup) | ||
| 1429 | return -EAGAIN; | ||
| 1430 | |||
| 1431 | SetPageChecked(page); | ||
| 1432 | page_cache_get(page); | ||
| 1433 | fixup->work.func = btrfs_writepage_fixup_worker; | ||
| 1434 | fixup->page = page; | ||
| 1435 | btrfs_queue_worker(&root->fs_info->fixup_workers, &fixup->work); | ||
| 1436 | return -EAGAIN; | ||
| 1437 | } | ||
| 1438 | |||
| 1439 | static int insert_reserved_file_extent(struct btrfs_trans_handle *trans, | ||
| 1440 | struct inode *inode, u64 file_pos, | ||
| 1441 | u64 disk_bytenr, u64 disk_num_bytes, | ||
| 1442 | u64 num_bytes, u64 ram_bytes, | ||
| 1443 | u8 compression, u8 encryption, | ||
| 1444 | u16 other_encoding, int extent_type) | ||
| 1445 | { | ||
| 1446 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1447 | struct btrfs_file_extent_item *fi; | ||
| 1448 | struct btrfs_path *path; | ||
| 1449 | struct extent_buffer *leaf; | ||
| 1450 | struct btrfs_key ins; | ||
| 1451 | u64 hint; | ||
| 1452 | int ret; | ||
| 1453 | |||
| 1454 | path = btrfs_alloc_path(); | ||
| 1455 | BUG_ON(!path); | ||
| 1456 | |||
| 1457 | ret = btrfs_drop_extents(trans, root, inode, file_pos, | ||
| 1458 | file_pos + num_bytes, file_pos, &hint); | ||
| 1459 | BUG_ON(ret); | ||
| 1460 | |||
| 1461 | ins.objectid = inode->i_ino; | ||
| 1462 | ins.offset = file_pos; | ||
| 1463 | ins.type = BTRFS_EXTENT_DATA_KEY; | ||
| 1464 | ret = btrfs_insert_empty_item(trans, root, path, &ins, sizeof(*fi)); | ||
| 1465 | BUG_ON(ret); | ||
| 1466 | leaf = path->nodes[0]; | ||
| 1467 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 1468 | struct btrfs_file_extent_item); | ||
| 1469 | btrfs_set_file_extent_generation(leaf, fi, trans->transid); | ||
| 1470 | btrfs_set_file_extent_type(leaf, fi, extent_type); | ||
| 1471 | btrfs_set_file_extent_disk_bytenr(leaf, fi, disk_bytenr); | ||
| 1472 | btrfs_set_file_extent_disk_num_bytes(leaf, fi, disk_num_bytes); | ||
| 1473 | btrfs_set_file_extent_offset(leaf, fi, 0); | ||
| 1474 | btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes); | ||
| 1475 | btrfs_set_file_extent_ram_bytes(leaf, fi, ram_bytes); | ||
| 1476 | btrfs_set_file_extent_compression(leaf, fi, compression); | ||
| 1477 | btrfs_set_file_extent_encryption(leaf, fi, encryption); | ||
| 1478 | btrfs_set_file_extent_other_encoding(leaf, fi, other_encoding); | ||
| 1479 | btrfs_mark_buffer_dirty(leaf); | ||
| 1480 | |||
| 1481 | inode_add_bytes(inode, num_bytes); | ||
| 1482 | btrfs_drop_extent_cache(inode, file_pos, file_pos + num_bytes - 1, 0); | ||
| 1483 | |||
| 1484 | ins.objectid = disk_bytenr; | ||
| 1485 | ins.offset = disk_num_bytes; | ||
| 1486 | ins.type = BTRFS_EXTENT_ITEM_KEY; | ||
| 1487 | ret = btrfs_alloc_reserved_extent(trans, root, leaf->start, | ||
| 1488 | root->root_key.objectid, | ||
| 1489 | trans->transid, inode->i_ino, &ins); | ||
| 1490 | BUG_ON(ret); | ||
| 1491 | |||
| 1492 | btrfs_free_path(path); | ||
| 1493 | return 0; | ||
| 1494 | } | ||
| 1495 | |||
| 1496 | /* as ordered data IO finishes, this gets called so we can finish | ||
| 1497 | * an ordered extent if the range of bytes in the file it covers are | ||
| 1498 | * fully written. | ||
| 1499 | */ | ||
| 1500 | static int btrfs_finish_ordered_io(struct inode *inode, u64 start, u64 end) | ||
| 1501 | { | ||
| 1502 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1503 | struct btrfs_trans_handle *trans; | ||
| 1504 | struct btrfs_ordered_extent *ordered_extent; | ||
| 1505 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 1506 | int compressed = 0; | ||
| 1507 | int ret; | ||
| 1508 | |||
| 1509 | ret = btrfs_dec_test_ordered_pending(inode, start, end - start + 1); | ||
| 1510 | if (!ret) | ||
| 1511 | return 0; | ||
| 1512 | |||
| 1513 | trans = btrfs_join_transaction(root, 1); | ||
| 1514 | |||
| 1515 | ordered_extent = btrfs_lookup_ordered_extent(inode, start); | ||
| 1516 | BUG_ON(!ordered_extent); | ||
| 1517 | if (test_bit(BTRFS_ORDERED_NOCOW, &ordered_extent->flags)) | ||
| 1518 | goto nocow; | ||
| 1519 | |||
| 1520 | lock_extent(io_tree, ordered_extent->file_offset, | ||
| 1521 | ordered_extent->file_offset + ordered_extent->len - 1, | ||
| 1522 | GFP_NOFS); | ||
| 1523 | |||
| 1524 | if (test_bit(BTRFS_ORDERED_COMPRESSED, &ordered_extent->flags)) | ||
| 1525 | compressed = 1; | ||
| 1526 | if (test_bit(BTRFS_ORDERED_PREALLOC, &ordered_extent->flags)) { | ||
| 1527 | BUG_ON(compressed); | ||
| 1528 | ret = btrfs_mark_extent_written(trans, root, inode, | ||
| 1529 | ordered_extent->file_offset, | ||
| 1530 | ordered_extent->file_offset + | ||
| 1531 | ordered_extent->len); | ||
| 1532 | BUG_ON(ret); | ||
| 1533 | } else { | ||
| 1534 | ret = insert_reserved_file_extent(trans, inode, | ||
| 1535 | ordered_extent->file_offset, | ||
| 1536 | ordered_extent->start, | ||
| 1537 | ordered_extent->disk_len, | ||
| 1538 | ordered_extent->len, | ||
| 1539 | ordered_extent->len, | ||
| 1540 | compressed, 0, 0, | ||
| 1541 | BTRFS_FILE_EXTENT_REG); | ||
| 1542 | BUG_ON(ret); | ||
| 1543 | } | ||
| 1544 | unlock_extent(io_tree, ordered_extent->file_offset, | ||
| 1545 | ordered_extent->file_offset + ordered_extent->len - 1, | ||
| 1546 | GFP_NOFS); | ||
| 1547 | nocow: | ||
| 1548 | add_pending_csums(trans, inode, ordered_extent->file_offset, | ||
| 1549 | &ordered_extent->list); | ||
| 1550 | |||
| 1551 | mutex_lock(&BTRFS_I(inode)->extent_mutex); | ||
| 1552 | btrfs_ordered_update_i_size(inode, ordered_extent); | ||
| 1553 | btrfs_update_inode(trans, root, inode); | ||
| 1554 | btrfs_remove_ordered_extent(inode, ordered_extent); | ||
| 1555 | mutex_unlock(&BTRFS_I(inode)->extent_mutex); | ||
| 1556 | |||
| 1557 | /* once for us */ | ||
| 1558 | btrfs_put_ordered_extent(ordered_extent); | ||
| 1559 | /* once for the tree */ | ||
| 1560 | btrfs_put_ordered_extent(ordered_extent); | ||
| 1561 | |||
| 1562 | btrfs_end_transaction(trans, root); | ||
| 1563 | return 0; | ||
| 1564 | } | ||
| 1565 | |||
| 1566 | static int btrfs_writepage_end_io_hook(struct page *page, u64 start, u64 end, | ||
| 1567 | struct extent_state *state, int uptodate) | ||
| 1568 | { | ||
| 1569 | return btrfs_finish_ordered_io(page->mapping->host, start, end); | ||
| 1570 | } | ||
| 1571 | |||
| 1572 | /* | ||
| 1573 | * When IO fails, either with EIO or csum verification fails, we | ||
| 1574 | * try other mirrors that might have a good copy of the data. This | ||
| 1575 | * io_failure_record is used to record state as we go through all the | ||
| 1576 | * mirrors. If another mirror has good data, the page is set up to date | ||
| 1577 | * and things continue. If a good mirror can't be found, the original | ||
| 1578 | * bio end_io callback is called to indicate things have failed. | ||
| 1579 | */ | ||
| 1580 | struct io_failure_record { | ||
| 1581 | struct page *page; | ||
| 1582 | u64 start; | ||
| 1583 | u64 len; | ||
| 1584 | u64 logical; | ||
| 1585 | unsigned long bio_flags; | ||
| 1586 | int last_mirror; | ||
| 1587 | }; | ||
| 1588 | |||
| 1589 | static int btrfs_io_failed_hook(struct bio *failed_bio, | ||
| 1590 | struct page *page, u64 start, u64 end, | ||
| 1591 | struct extent_state *state) | ||
| 1592 | { | ||
| 1593 | struct io_failure_record *failrec = NULL; | ||
| 1594 | u64 private; | ||
| 1595 | struct extent_map *em; | ||
| 1596 | struct inode *inode = page->mapping->host; | ||
| 1597 | struct extent_io_tree *failure_tree = &BTRFS_I(inode)->io_failure_tree; | ||
| 1598 | struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 1599 | struct bio *bio; | ||
| 1600 | int num_copies; | ||
| 1601 | int ret; | ||
| 1602 | int rw; | ||
| 1603 | u64 logical; | ||
| 1604 | |||
| 1605 | ret = get_state_private(failure_tree, start, &private); | ||
| 1606 | if (ret) { | ||
| 1607 | failrec = kmalloc(sizeof(*failrec), GFP_NOFS); | ||
| 1608 | if (!failrec) | ||
| 1609 | return -ENOMEM; | ||
| 1610 | failrec->start = start; | ||
| 1611 | failrec->len = end - start + 1; | ||
| 1612 | failrec->last_mirror = 0; | ||
| 1613 | failrec->bio_flags = 0; | ||
| 1614 | |||
| 1615 | spin_lock(&em_tree->lock); | ||
| 1616 | em = lookup_extent_mapping(em_tree, start, failrec->len); | ||
| 1617 | if (em->start > start || em->start + em->len < start) { | ||
| 1618 | free_extent_map(em); | ||
| 1619 | em = NULL; | ||
| 1620 | } | ||
| 1621 | spin_unlock(&em_tree->lock); | ||
| 1622 | |||
| 1623 | if (!em || IS_ERR(em)) { | ||
| 1624 | kfree(failrec); | ||
| 1625 | return -EIO; | ||
| 1626 | } | ||
| 1627 | logical = start - em->start; | ||
| 1628 | logical = em->block_start + logical; | ||
| 1629 | if (test_bit(EXTENT_FLAG_COMPRESSED, &em->flags)) { | ||
| 1630 | logical = em->block_start; | ||
| 1631 | failrec->bio_flags = EXTENT_BIO_COMPRESSED; | ||
| 1632 | } | ||
| 1633 | failrec->logical = logical; | ||
| 1634 | free_extent_map(em); | ||
| 1635 | set_extent_bits(failure_tree, start, end, EXTENT_LOCKED | | ||
| 1636 | EXTENT_DIRTY, GFP_NOFS); | ||
| 1637 | set_state_private(failure_tree, start, | ||
| 1638 | (u64)(unsigned long)failrec); | ||
| 1639 | } else { | ||
| 1640 | failrec = (struct io_failure_record *)(unsigned long)private; | ||
| 1641 | } | ||
| 1642 | num_copies = btrfs_num_copies( | ||
| 1643 | &BTRFS_I(inode)->root->fs_info->mapping_tree, | ||
| 1644 | failrec->logical, failrec->len); | ||
| 1645 | failrec->last_mirror++; | ||
| 1646 | if (!state) { | ||
| 1647 | spin_lock(&BTRFS_I(inode)->io_tree.lock); | ||
| 1648 | state = find_first_extent_bit_state(&BTRFS_I(inode)->io_tree, | ||
| 1649 | failrec->start, | ||
| 1650 | EXTENT_LOCKED); | ||
| 1651 | if (state && state->start != failrec->start) | ||
| 1652 | state = NULL; | ||
| 1653 | spin_unlock(&BTRFS_I(inode)->io_tree.lock); | ||
| 1654 | } | ||
| 1655 | if (!state || failrec->last_mirror > num_copies) { | ||
| 1656 | set_state_private(failure_tree, failrec->start, 0); | ||
| 1657 | clear_extent_bits(failure_tree, failrec->start, | ||
| 1658 | failrec->start + failrec->len - 1, | ||
| 1659 | EXTENT_LOCKED | EXTENT_DIRTY, GFP_NOFS); | ||
| 1660 | kfree(failrec); | ||
| 1661 | return -EIO; | ||
| 1662 | } | ||
| 1663 | bio = bio_alloc(GFP_NOFS, 1); | ||
| 1664 | bio->bi_private = state; | ||
| 1665 | bio->bi_end_io = failed_bio->bi_end_io; | ||
| 1666 | bio->bi_sector = failrec->logical >> 9; | ||
| 1667 | bio->bi_bdev = failed_bio->bi_bdev; | ||
| 1668 | bio->bi_size = 0; | ||
| 1669 | |||
| 1670 | bio_add_page(bio, page, failrec->len, start - page_offset(page)); | ||
| 1671 | if (failed_bio->bi_rw & (1 << BIO_RW)) | ||
| 1672 | rw = WRITE; | ||
| 1673 | else | ||
| 1674 | rw = READ; | ||
| 1675 | |||
| 1676 | BTRFS_I(inode)->io_tree.ops->submit_bio_hook(inode, rw, bio, | ||
| 1677 | failrec->last_mirror, | ||
| 1678 | failrec->bio_flags); | ||
| 1679 | return 0; | ||
| 1680 | } | ||
| 1681 | |||
| 1682 | /* | ||
| 1683 | * each time an IO finishes, we do a fast check in the IO failure tree | ||
| 1684 | * to see if we need to process or clean up an io_failure_record | ||
| 1685 | */ | ||
| 1686 | static int btrfs_clean_io_failures(struct inode *inode, u64 start) | ||
| 1687 | { | ||
| 1688 | u64 private; | ||
| 1689 | u64 private_failure; | ||
| 1690 | struct io_failure_record *failure; | ||
| 1691 | int ret; | ||
| 1692 | |||
| 1693 | private = 0; | ||
| 1694 | if (count_range_bits(&BTRFS_I(inode)->io_failure_tree, &private, | ||
| 1695 | (u64)-1, 1, EXTENT_DIRTY)) { | ||
| 1696 | ret = get_state_private(&BTRFS_I(inode)->io_failure_tree, | ||
| 1697 | start, &private_failure); | ||
| 1698 | if (ret == 0) { | ||
| 1699 | failure = (struct io_failure_record *)(unsigned long) | ||
| 1700 | private_failure; | ||
| 1701 | set_state_private(&BTRFS_I(inode)->io_failure_tree, | ||
| 1702 | failure->start, 0); | ||
| 1703 | clear_extent_bits(&BTRFS_I(inode)->io_failure_tree, | ||
| 1704 | failure->start, | ||
| 1705 | failure->start + failure->len - 1, | ||
| 1706 | EXTENT_DIRTY | EXTENT_LOCKED, | ||
| 1707 | GFP_NOFS); | ||
| 1708 | kfree(failure); | ||
| 1709 | } | ||
| 1710 | } | ||
| 1711 | return 0; | ||
| 1712 | } | ||
| 1713 | |||
| 1714 | /* | ||
| 1715 | * when reads are done, we need to check csums to verify the data is correct | ||
| 1716 | * if there's a match, we allow the bio to finish. If not, we go through | ||
| 1717 | * the io_failure_record routines to find good copies | ||
| 1718 | */ | ||
| 1719 | static int btrfs_readpage_end_io_hook(struct page *page, u64 start, u64 end, | ||
| 1720 | struct extent_state *state) | ||
| 1721 | { | ||
| 1722 | size_t offset = start - ((u64)page->index << PAGE_CACHE_SHIFT); | ||
| 1723 | struct inode *inode = page->mapping->host; | ||
| 1724 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 1725 | char *kaddr; | ||
| 1726 | u64 private = ~(u32)0; | ||
| 1727 | int ret; | ||
| 1728 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1729 | u32 csum = ~(u32)0; | ||
| 1730 | |||
| 1731 | if (PageChecked(page)) { | ||
| 1732 | ClearPageChecked(page); | ||
| 1733 | goto good; | ||
| 1734 | } | ||
| 1735 | if (btrfs_test_flag(inode, NODATASUM)) | ||
| 1736 | return 0; | ||
| 1737 | |||
| 1738 | if (root->root_key.objectid == BTRFS_DATA_RELOC_TREE_OBJECTID && | ||
| 1739 | test_range_bit(io_tree, start, end, EXTENT_NODATASUM, 1)) { | ||
| 1740 | clear_extent_bits(io_tree, start, end, EXTENT_NODATASUM, | ||
| 1741 | GFP_NOFS); | ||
| 1742 | return 0; | ||
| 1743 | } | ||
| 1744 | |||
| 1745 | if (state && state->start == start) { | ||
| 1746 | private = state->private; | ||
| 1747 | ret = 0; | ||
| 1748 | } else { | ||
| 1749 | ret = get_state_private(io_tree, start, &private); | ||
| 1750 | } | ||
| 1751 | kaddr = kmap_atomic(page, KM_USER0); | ||
| 1752 | if (ret) | ||
| 1753 | goto zeroit; | ||
| 1754 | |||
| 1755 | csum = btrfs_csum_data(root, kaddr + offset, csum, end - start + 1); | ||
| 1756 | btrfs_csum_final(csum, (char *)&csum); | ||
| 1757 | if (csum != private) | ||
| 1758 | goto zeroit; | ||
| 1759 | |||
| 1760 | kunmap_atomic(kaddr, KM_USER0); | ||
| 1761 | good: | ||
| 1762 | /* if the io failure tree for this inode is non-empty, | ||
| 1763 | * check to see if we've recovered from a failed IO | ||
| 1764 | */ | ||
| 1765 | btrfs_clean_io_failures(inode, start); | ||
| 1766 | return 0; | ||
| 1767 | |||
| 1768 | zeroit: | ||
| 1769 | printk(KERN_INFO "btrfs csum failed ino %lu off %llu csum %u " | ||
| 1770 | "private %llu\n", page->mapping->host->i_ino, | ||
| 1771 | (unsigned long long)start, csum, | ||
| 1772 | (unsigned long long)private); | ||
| 1773 | memset(kaddr + offset, 1, end - start + 1); | ||
| 1774 | flush_dcache_page(page); | ||
| 1775 | kunmap_atomic(kaddr, KM_USER0); | ||
| 1776 | if (private == 0) | ||
| 1777 | return 0; | ||
| 1778 | return -EIO; | ||
| 1779 | } | ||
| 1780 | |||
| 1781 | /* | ||
| 1782 | * This creates an orphan entry for the given inode in case something goes | ||
| 1783 | * wrong in the middle of an unlink/truncate. | ||
| 1784 | */ | ||
| 1785 | int btrfs_orphan_add(struct btrfs_trans_handle *trans, struct inode *inode) | ||
| 1786 | { | ||
| 1787 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1788 | int ret = 0; | ||
| 1789 | |||
| 1790 | spin_lock(&root->list_lock); | ||
| 1791 | |||
| 1792 | /* already on the orphan list, we're good */ | ||
| 1793 | if (!list_empty(&BTRFS_I(inode)->i_orphan)) { | ||
| 1794 | spin_unlock(&root->list_lock); | ||
| 1795 | return 0; | ||
| 1796 | } | ||
| 1797 | |||
| 1798 | list_add(&BTRFS_I(inode)->i_orphan, &root->orphan_list); | ||
| 1799 | |||
| 1800 | spin_unlock(&root->list_lock); | ||
| 1801 | |||
| 1802 | /* | ||
| 1803 | * insert an orphan item to track this unlinked/truncated file | ||
| 1804 | */ | ||
| 1805 | ret = btrfs_insert_orphan_item(trans, root, inode->i_ino); | ||
| 1806 | |||
| 1807 | return ret; | ||
| 1808 | } | ||
| 1809 | |||
| 1810 | /* | ||
| 1811 | * We have done the truncate/delete so we can go ahead and remove the orphan | ||
| 1812 | * item for this particular inode. | ||
| 1813 | */ | ||
| 1814 | int btrfs_orphan_del(struct btrfs_trans_handle *trans, struct inode *inode) | ||
| 1815 | { | ||
| 1816 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1817 | int ret = 0; | ||
| 1818 | |||
| 1819 | spin_lock(&root->list_lock); | ||
| 1820 | |||
| 1821 | if (list_empty(&BTRFS_I(inode)->i_orphan)) { | ||
| 1822 | spin_unlock(&root->list_lock); | ||
| 1823 | return 0; | ||
| 1824 | } | ||
| 1825 | |||
| 1826 | list_del_init(&BTRFS_I(inode)->i_orphan); | ||
| 1827 | if (!trans) { | ||
| 1828 | spin_unlock(&root->list_lock); | ||
| 1829 | return 0; | ||
| 1830 | } | ||
| 1831 | |||
| 1832 | spin_unlock(&root->list_lock); | ||
| 1833 | |||
| 1834 | ret = btrfs_del_orphan_item(trans, root, inode->i_ino); | ||
| 1835 | |||
| 1836 | return ret; | ||
| 1837 | } | ||
| 1838 | |||
| 1839 | /* | ||
| 1840 | * this cleans up any orphans that may be left on the list from the last use | ||
| 1841 | * of this root. | ||
| 1842 | */ | ||
| 1843 | void btrfs_orphan_cleanup(struct btrfs_root *root) | ||
| 1844 | { | ||
| 1845 | struct btrfs_path *path; | ||
| 1846 | struct extent_buffer *leaf; | ||
| 1847 | struct btrfs_item *item; | ||
| 1848 | struct btrfs_key key, found_key; | ||
| 1849 | struct btrfs_trans_handle *trans; | ||
| 1850 | struct inode *inode; | ||
| 1851 | int ret = 0, nr_unlink = 0, nr_truncate = 0; | ||
| 1852 | |||
| 1853 | path = btrfs_alloc_path(); | ||
| 1854 | if (!path) | ||
| 1855 | return; | ||
| 1856 | path->reada = -1; | ||
| 1857 | |||
| 1858 | key.objectid = BTRFS_ORPHAN_OBJECTID; | ||
| 1859 | btrfs_set_key_type(&key, BTRFS_ORPHAN_ITEM_KEY); | ||
| 1860 | key.offset = (u64)-1; | ||
| 1861 | |||
| 1862 | |||
| 1863 | while (1) { | ||
| 1864 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 1865 | if (ret < 0) { | ||
| 1866 | printk(KERN_ERR "Error searching slot for orphan: %d" | ||
| 1867 | "\n", ret); | ||
| 1868 | break; | ||
| 1869 | } | ||
| 1870 | |||
| 1871 | /* | ||
| 1872 | * if ret == 0 means we found what we were searching for, which | ||
| 1873 | * is weird, but possible, so only screw with path if we didnt | ||
| 1874 | * find the key and see if we have stuff that matches | ||
| 1875 | */ | ||
| 1876 | if (ret > 0) { | ||
| 1877 | if (path->slots[0] == 0) | ||
| 1878 | break; | ||
| 1879 | path->slots[0]--; | ||
| 1880 | } | ||
| 1881 | |||
| 1882 | /* pull out the item */ | ||
| 1883 | leaf = path->nodes[0]; | ||
| 1884 | item = btrfs_item_nr(leaf, path->slots[0]); | ||
| 1885 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 1886 | |||
| 1887 | /* make sure the item matches what we want */ | ||
| 1888 | if (found_key.objectid != BTRFS_ORPHAN_OBJECTID) | ||
| 1889 | break; | ||
| 1890 | if (btrfs_key_type(&found_key) != BTRFS_ORPHAN_ITEM_KEY) | ||
| 1891 | break; | ||
| 1892 | |||
| 1893 | /* release the path since we're done with it */ | ||
| 1894 | btrfs_release_path(root, path); | ||
| 1895 | |||
| 1896 | /* | ||
| 1897 | * this is where we are basically btrfs_lookup, without the | ||
| 1898 | * crossing root thing. we store the inode number in the | ||
| 1899 | * offset of the orphan item. | ||
| 1900 | */ | ||
| 1901 | inode = btrfs_iget_locked(root->fs_info->sb, | ||
| 1902 | found_key.offset, root); | ||
| 1903 | if (!inode) | ||
| 1904 | break; | ||
| 1905 | |||
| 1906 | if (inode->i_state & I_NEW) { | ||
| 1907 | BTRFS_I(inode)->root = root; | ||
| 1908 | |||
| 1909 | /* have to set the location manually */ | ||
| 1910 | BTRFS_I(inode)->location.objectid = inode->i_ino; | ||
| 1911 | BTRFS_I(inode)->location.type = BTRFS_INODE_ITEM_KEY; | ||
| 1912 | BTRFS_I(inode)->location.offset = 0; | ||
| 1913 | |||
| 1914 | btrfs_read_locked_inode(inode); | ||
| 1915 | unlock_new_inode(inode); | ||
| 1916 | } | ||
| 1917 | |||
| 1918 | /* | ||
| 1919 | * add this inode to the orphan list so btrfs_orphan_del does | ||
| 1920 | * the proper thing when we hit it | ||
| 1921 | */ | ||
| 1922 | spin_lock(&root->list_lock); | ||
| 1923 | list_add(&BTRFS_I(inode)->i_orphan, &root->orphan_list); | ||
| 1924 | spin_unlock(&root->list_lock); | ||
| 1925 | |||
| 1926 | /* | ||
| 1927 | * if this is a bad inode, means we actually succeeded in | ||
| 1928 | * removing the inode, but not the orphan record, which means | ||
| 1929 | * we need to manually delete the orphan since iput will just | ||
| 1930 | * do a destroy_inode | ||
| 1931 | */ | ||
| 1932 | if (is_bad_inode(inode)) { | ||
| 1933 | trans = btrfs_start_transaction(root, 1); | ||
| 1934 | btrfs_orphan_del(trans, inode); | ||
| 1935 | btrfs_end_transaction(trans, root); | ||
| 1936 | iput(inode); | ||
| 1937 | continue; | ||
| 1938 | } | ||
| 1939 | |||
| 1940 | /* if we have links, this was a truncate, lets do that */ | ||
| 1941 | if (inode->i_nlink) { | ||
| 1942 | nr_truncate++; | ||
| 1943 | btrfs_truncate(inode); | ||
| 1944 | } else { | ||
| 1945 | nr_unlink++; | ||
| 1946 | } | ||
| 1947 | |||
| 1948 | /* this will do delete_inode and everything for us */ | ||
| 1949 | iput(inode); | ||
| 1950 | } | ||
| 1951 | |||
| 1952 | if (nr_unlink) | ||
| 1953 | printk(KERN_INFO "btrfs: unlinked %d orphans\n", nr_unlink); | ||
| 1954 | if (nr_truncate) | ||
| 1955 | printk(KERN_INFO "btrfs: truncated %d orphans\n", nr_truncate); | ||
| 1956 | |||
| 1957 | btrfs_free_path(path); | ||
| 1958 | } | ||
| 1959 | |||
| 1960 | /* | ||
| 1961 | * read an inode from the btree into the in-memory inode | ||
| 1962 | */ | ||
| 1963 | void btrfs_read_locked_inode(struct inode *inode) | ||
| 1964 | { | ||
| 1965 | struct btrfs_path *path; | ||
| 1966 | struct extent_buffer *leaf; | ||
| 1967 | struct btrfs_inode_item *inode_item; | ||
| 1968 | struct btrfs_timespec *tspec; | ||
| 1969 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1970 | struct btrfs_key location; | ||
| 1971 | u64 alloc_group_block; | ||
| 1972 | u32 rdev; | ||
| 1973 | int ret; | ||
| 1974 | |||
| 1975 | path = btrfs_alloc_path(); | ||
| 1976 | BUG_ON(!path); | ||
| 1977 | memcpy(&location, &BTRFS_I(inode)->location, sizeof(location)); | ||
| 1978 | |||
| 1979 | ret = btrfs_lookup_inode(NULL, root, path, &location, 0); | ||
| 1980 | if (ret) | ||
| 1981 | goto make_bad; | ||
| 1982 | |||
| 1983 | leaf = path->nodes[0]; | ||
| 1984 | inode_item = btrfs_item_ptr(leaf, path->slots[0], | ||
| 1985 | struct btrfs_inode_item); | ||
| 1986 | |||
| 1987 | inode->i_mode = btrfs_inode_mode(leaf, inode_item); | ||
| 1988 | inode->i_nlink = btrfs_inode_nlink(leaf, inode_item); | ||
| 1989 | inode->i_uid = btrfs_inode_uid(leaf, inode_item); | ||
| 1990 | inode->i_gid = btrfs_inode_gid(leaf, inode_item); | ||
| 1991 | btrfs_i_size_write(inode, btrfs_inode_size(leaf, inode_item)); | ||
| 1992 | |||
| 1993 | tspec = btrfs_inode_atime(inode_item); | ||
| 1994 | inode->i_atime.tv_sec = btrfs_timespec_sec(leaf, tspec); | ||
| 1995 | inode->i_atime.tv_nsec = btrfs_timespec_nsec(leaf, tspec); | ||
| 1996 | |||
| 1997 | tspec = btrfs_inode_mtime(inode_item); | ||
| 1998 | inode->i_mtime.tv_sec = btrfs_timespec_sec(leaf, tspec); | ||
| 1999 | inode->i_mtime.tv_nsec = btrfs_timespec_nsec(leaf, tspec); | ||
| 2000 | |||
| 2001 | tspec = btrfs_inode_ctime(inode_item); | ||
| 2002 | inode->i_ctime.tv_sec = btrfs_timespec_sec(leaf, tspec); | ||
| 2003 | inode->i_ctime.tv_nsec = btrfs_timespec_nsec(leaf, tspec); | ||
| 2004 | |||
| 2005 | inode_set_bytes(inode, btrfs_inode_nbytes(leaf, inode_item)); | ||
| 2006 | BTRFS_I(inode)->generation = btrfs_inode_generation(leaf, inode_item); | ||
| 2007 | BTRFS_I(inode)->sequence = btrfs_inode_sequence(leaf, inode_item); | ||
| 2008 | inode->i_generation = BTRFS_I(inode)->generation; | ||
| 2009 | inode->i_rdev = 0; | ||
| 2010 | rdev = btrfs_inode_rdev(leaf, inode_item); | ||
| 2011 | |||
| 2012 | BTRFS_I(inode)->index_cnt = (u64)-1; | ||
| 2013 | BTRFS_I(inode)->flags = btrfs_inode_flags(leaf, inode_item); | ||
| 2014 | |||
| 2015 | alloc_group_block = btrfs_inode_block_group(leaf, inode_item); | ||
| 2016 | BTRFS_I(inode)->block_group = btrfs_find_block_group(root, 0, | ||
| 2017 | alloc_group_block, 0); | ||
| 2018 | btrfs_free_path(path); | ||
| 2019 | inode_item = NULL; | ||
| 2020 | |||
| 2021 | switch (inode->i_mode & S_IFMT) { | ||
| 2022 | case S_IFREG: | ||
| 2023 | inode->i_mapping->a_ops = &btrfs_aops; | ||
| 2024 | inode->i_mapping->backing_dev_info = &root->fs_info->bdi; | ||
| 2025 | BTRFS_I(inode)->io_tree.ops = &btrfs_extent_io_ops; | ||
| 2026 | inode->i_fop = &btrfs_file_operations; | ||
| 2027 | inode->i_op = &btrfs_file_inode_operations; | ||
| 2028 | break; | ||
| 2029 | case S_IFDIR: | ||
| 2030 | inode->i_fop = &btrfs_dir_file_operations; | ||
| 2031 | if (root == root->fs_info->tree_root) | ||
| 2032 | inode->i_op = &btrfs_dir_ro_inode_operations; | ||
| 2033 | else | ||
| 2034 | inode->i_op = &btrfs_dir_inode_operations; | ||
| 2035 | break; | ||
| 2036 | case S_IFLNK: | ||
| 2037 | inode->i_op = &btrfs_symlink_inode_operations; | ||
| 2038 | inode->i_mapping->a_ops = &btrfs_symlink_aops; | ||
| 2039 | inode->i_mapping->backing_dev_info = &root->fs_info->bdi; | ||
| 2040 | break; | ||
| 2041 | default: | ||
| 2042 | init_special_inode(inode, inode->i_mode, rdev); | ||
| 2043 | break; | ||
| 2044 | } | ||
| 2045 | return; | ||
| 2046 | |||
| 2047 | make_bad: | ||
| 2048 | btrfs_free_path(path); | ||
| 2049 | make_bad_inode(inode); | ||
| 2050 | } | ||
| 2051 | |||
| 2052 | /* | ||
| 2053 | * given a leaf and an inode, copy the inode fields into the leaf | ||
| 2054 | */ | ||
| 2055 | static void fill_inode_item(struct btrfs_trans_handle *trans, | ||
| 2056 | struct extent_buffer *leaf, | ||
| 2057 | struct btrfs_inode_item *item, | ||
| 2058 | struct inode *inode) | ||
| 2059 | { | ||
| 2060 | btrfs_set_inode_uid(leaf, item, inode->i_uid); | ||
| 2061 | btrfs_set_inode_gid(leaf, item, inode->i_gid); | ||
| 2062 | btrfs_set_inode_size(leaf, item, BTRFS_I(inode)->disk_i_size); | ||
| 2063 | btrfs_set_inode_mode(leaf, item, inode->i_mode); | ||
| 2064 | btrfs_set_inode_nlink(leaf, item, inode->i_nlink); | ||
| 2065 | |||
| 2066 | btrfs_set_timespec_sec(leaf, btrfs_inode_atime(item), | ||
| 2067 | inode->i_atime.tv_sec); | ||
| 2068 | btrfs_set_timespec_nsec(leaf, btrfs_inode_atime(item), | ||
| 2069 | inode->i_atime.tv_nsec); | ||
| 2070 | |||
| 2071 | btrfs_set_timespec_sec(leaf, btrfs_inode_mtime(item), | ||
| 2072 | inode->i_mtime.tv_sec); | ||
| 2073 | btrfs_set_timespec_nsec(leaf, btrfs_inode_mtime(item), | ||
| 2074 | inode->i_mtime.tv_nsec); | ||
| 2075 | |||
| 2076 | btrfs_set_timespec_sec(leaf, btrfs_inode_ctime(item), | ||
| 2077 | inode->i_ctime.tv_sec); | ||
| 2078 | btrfs_set_timespec_nsec(leaf, btrfs_inode_ctime(item), | ||
| 2079 | inode->i_ctime.tv_nsec); | ||
| 2080 | |||
| 2081 | btrfs_set_inode_nbytes(leaf, item, inode_get_bytes(inode)); | ||
| 2082 | btrfs_set_inode_generation(leaf, item, BTRFS_I(inode)->generation); | ||
| 2083 | btrfs_set_inode_sequence(leaf, item, BTRFS_I(inode)->sequence); | ||
| 2084 | btrfs_set_inode_transid(leaf, item, trans->transid); | ||
| 2085 | btrfs_set_inode_rdev(leaf, item, inode->i_rdev); | ||
| 2086 | btrfs_set_inode_flags(leaf, item, BTRFS_I(inode)->flags); | ||
| 2087 | btrfs_set_inode_block_group(leaf, item, BTRFS_I(inode)->block_group); | ||
| 2088 | } | ||
| 2089 | |||
| 2090 | /* | ||
| 2091 | * copy everything in the in-memory inode into the btree. | ||
| 2092 | */ | ||
| 2093 | noinline int btrfs_update_inode(struct btrfs_trans_handle *trans, | ||
| 2094 | struct btrfs_root *root, struct inode *inode) | ||
| 2095 | { | ||
| 2096 | struct btrfs_inode_item *inode_item; | ||
| 2097 | struct btrfs_path *path; | ||
| 2098 | struct extent_buffer *leaf; | ||
| 2099 | int ret; | ||
| 2100 | |||
| 2101 | path = btrfs_alloc_path(); | ||
| 2102 | BUG_ON(!path); | ||
| 2103 | ret = btrfs_lookup_inode(trans, root, path, | ||
| 2104 | &BTRFS_I(inode)->location, 1); | ||
| 2105 | if (ret) { | ||
| 2106 | if (ret > 0) | ||
| 2107 | ret = -ENOENT; | ||
| 2108 | goto failed; | ||
| 2109 | } | ||
| 2110 | |||
| 2111 | leaf = path->nodes[0]; | ||
| 2112 | inode_item = btrfs_item_ptr(leaf, path->slots[0], | ||
| 2113 | struct btrfs_inode_item); | ||
| 2114 | |||
| 2115 | fill_inode_item(trans, leaf, inode_item, inode); | ||
| 2116 | btrfs_mark_buffer_dirty(leaf); | ||
| 2117 | btrfs_set_inode_last_trans(trans, inode); | ||
| 2118 | ret = 0; | ||
| 2119 | failed: | ||
| 2120 | btrfs_free_path(path); | ||
| 2121 | return ret; | ||
| 2122 | } | ||
| 2123 | |||
| 2124 | |||
| 2125 | /* | ||
| 2126 | * unlink helper that gets used here in inode.c and in the tree logging | ||
| 2127 | * recovery code. It remove a link in a directory with a given name, and | ||
| 2128 | * also drops the back refs in the inode to the directory | ||
| 2129 | */ | ||
| 2130 | int btrfs_unlink_inode(struct btrfs_trans_handle *trans, | ||
| 2131 | struct btrfs_root *root, | ||
| 2132 | struct inode *dir, struct inode *inode, | ||
| 2133 | const char *name, int name_len) | ||
| 2134 | { | ||
| 2135 | struct btrfs_path *path; | ||
| 2136 | int ret = 0; | ||
| 2137 | struct extent_buffer *leaf; | ||
| 2138 | struct btrfs_dir_item *di; | ||
| 2139 | struct btrfs_key key; | ||
| 2140 | u64 index; | ||
| 2141 | |||
| 2142 | path = btrfs_alloc_path(); | ||
| 2143 | if (!path) { | ||
| 2144 | ret = -ENOMEM; | ||
| 2145 | goto err; | ||
| 2146 | } | ||
| 2147 | |||
| 2148 | di = btrfs_lookup_dir_item(trans, root, path, dir->i_ino, | ||
| 2149 | name, name_len, -1); | ||
| 2150 | if (IS_ERR(di)) { | ||
| 2151 | ret = PTR_ERR(di); | ||
| 2152 | goto err; | ||
| 2153 | } | ||
| 2154 | if (!di) { | ||
| 2155 | ret = -ENOENT; | ||
| 2156 | goto err; | ||
| 2157 | } | ||
| 2158 | leaf = path->nodes[0]; | ||
| 2159 | btrfs_dir_item_key_to_cpu(leaf, di, &key); | ||
| 2160 | ret = btrfs_delete_one_dir_name(trans, root, path, di); | ||
| 2161 | if (ret) | ||
| 2162 | goto err; | ||
| 2163 | btrfs_release_path(root, path); | ||
| 2164 | |||
| 2165 | ret = btrfs_del_inode_ref(trans, root, name, name_len, | ||
| 2166 | inode->i_ino, | ||
| 2167 | dir->i_ino, &index); | ||
| 2168 | if (ret) { | ||
| 2169 | printk(KERN_INFO "btrfs failed to delete reference to %.*s, " | ||
| 2170 | "inode %lu parent %lu\n", name_len, name, | ||
| 2171 | inode->i_ino, dir->i_ino); | ||
| 2172 | goto err; | ||
| 2173 | } | ||
| 2174 | |||
| 2175 | di = btrfs_lookup_dir_index_item(trans, root, path, dir->i_ino, | ||
| 2176 | index, name, name_len, -1); | ||
| 2177 | if (IS_ERR(di)) { | ||
| 2178 | ret = PTR_ERR(di); | ||
| 2179 | goto err; | ||
| 2180 | } | ||
| 2181 | if (!di) { | ||
| 2182 | ret = -ENOENT; | ||
| 2183 | goto err; | ||
| 2184 | } | ||
| 2185 | ret = btrfs_delete_one_dir_name(trans, root, path, di); | ||
| 2186 | btrfs_release_path(root, path); | ||
| 2187 | |||
| 2188 | ret = btrfs_del_inode_ref_in_log(trans, root, name, name_len, | ||
| 2189 | inode, dir->i_ino); | ||
| 2190 | BUG_ON(ret != 0 && ret != -ENOENT); | ||
| 2191 | if (ret != -ENOENT) | ||
| 2192 | BTRFS_I(dir)->log_dirty_trans = trans->transid; | ||
| 2193 | |||
| 2194 | ret = btrfs_del_dir_entries_in_log(trans, root, name, name_len, | ||
| 2195 | dir, index); | ||
| 2196 | BUG_ON(ret); | ||
| 2197 | err: | ||
| 2198 | btrfs_free_path(path); | ||
| 2199 | if (ret) | ||
| 2200 | goto out; | ||
| 2201 | |||
| 2202 | btrfs_i_size_write(dir, dir->i_size - name_len * 2); | ||
| 2203 | inode->i_ctime = dir->i_mtime = dir->i_ctime = CURRENT_TIME; | ||
| 2204 | btrfs_update_inode(trans, root, dir); | ||
| 2205 | btrfs_drop_nlink(inode); | ||
| 2206 | ret = btrfs_update_inode(trans, root, inode); | ||
| 2207 | dir->i_sb->s_dirt = 1; | ||
| 2208 | out: | ||
| 2209 | return ret; | ||
| 2210 | } | ||
| 2211 | |||
| 2212 | static int btrfs_unlink(struct inode *dir, struct dentry *dentry) | ||
| 2213 | { | ||
| 2214 | struct btrfs_root *root; | ||
| 2215 | struct btrfs_trans_handle *trans; | ||
| 2216 | struct inode *inode = dentry->d_inode; | ||
| 2217 | int ret; | ||
| 2218 | unsigned long nr = 0; | ||
| 2219 | |||
| 2220 | root = BTRFS_I(dir)->root; | ||
| 2221 | |||
| 2222 | ret = btrfs_check_free_space(root, 1, 1); | ||
| 2223 | if (ret) | ||
| 2224 | goto fail; | ||
| 2225 | |||
| 2226 | trans = btrfs_start_transaction(root, 1); | ||
| 2227 | |||
| 2228 | btrfs_set_trans_block_group(trans, dir); | ||
| 2229 | ret = btrfs_unlink_inode(trans, root, dir, dentry->d_inode, | ||
| 2230 | dentry->d_name.name, dentry->d_name.len); | ||
| 2231 | |||
| 2232 | if (inode->i_nlink == 0) | ||
| 2233 | ret = btrfs_orphan_add(trans, inode); | ||
| 2234 | |||
| 2235 | nr = trans->blocks_used; | ||
| 2236 | |||
| 2237 | btrfs_end_transaction_throttle(trans, root); | ||
| 2238 | fail: | ||
| 2239 | btrfs_btree_balance_dirty(root, nr); | ||
| 2240 | return ret; | ||
| 2241 | } | ||
| 2242 | |||
| 2243 | static int btrfs_rmdir(struct inode *dir, struct dentry *dentry) | ||
| 2244 | { | ||
| 2245 | struct inode *inode = dentry->d_inode; | ||
| 2246 | int err = 0; | ||
| 2247 | int ret; | ||
| 2248 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 2249 | struct btrfs_trans_handle *trans; | ||
| 2250 | unsigned long nr = 0; | ||
| 2251 | |||
| 2252 | /* | ||
| 2253 | * the FIRST_FREE_OBJECTID check makes sure we don't try to rmdir | ||
| 2254 | * the root of a subvolume or snapshot | ||
| 2255 | */ | ||
| 2256 | if (inode->i_size > BTRFS_EMPTY_DIR_SIZE || | ||
| 2257 | inode->i_ino == BTRFS_FIRST_FREE_OBJECTID) { | ||
| 2258 | return -ENOTEMPTY; | ||
| 2259 | } | ||
| 2260 | |||
| 2261 | ret = btrfs_check_free_space(root, 1, 1); | ||
| 2262 | if (ret) | ||
| 2263 | goto fail; | ||
| 2264 | |||
| 2265 | trans = btrfs_start_transaction(root, 1); | ||
| 2266 | btrfs_set_trans_block_group(trans, dir); | ||
| 2267 | |||
| 2268 | err = btrfs_orphan_add(trans, inode); | ||
| 2269 | if (err) | ||
| 2270 | goto fail_trans; | ||
| 2271 | |||
| 2272 | /* now the directory is empty */ | ||
| 2273 | err = btrfs_unlink_inode(trans, root, dir, dentry->d_inode, | ||
| 2274 | dentry->d_name.name, dentry->d_name.len); | ||
| 2275 | if (!err) | ||
| 2276 | btrfs_i_size_write(inode, 0); | ||
| 2277 | |||
| 2278 | fail_trans: | ||
| 2279 | nr = trans->blocks_used; | ||
| 2280 | ret = btrfs_end_transaction_throttle(trans, root); | ||
| 2281 | fail: | ||
| 2282 | btrfs_btree_balance_dirty(root, nr); | ||
| 2283 | |||
| 2284 | if (ret && !err) | ||
| 2285 | err = ret; | ||
| 2286 | return err; | ||
| 2287 | } | ||
| 2288 | |||
| 2289 | #if 0 | ||
| 2290 | /* | ||
| 2291 | * when truncating bytes in a file, it is possible to avoid reading | ||
| 2292 | * the leaves that contain only checksum items. This can be the | ||
| 2293 | * majority of the IO required to delete a large file, but it must | ||
| 2294 | * be done carefully. | ||
| 2295 | * | ||
| 2296 | * The keys in the level just above the leaves are checked to make sure | ||
| 2297 | * the lowest key in a given leaf is a csum key, and starts at an offset | ||
| 2298 | * after the new size. | ||
| 2299 | * | ||
| 2300 | * Then the key for the next leaf is checked to make sure it also has | ||
| 2301 | * a checksum item for the same file. If it does, we know our target leaf | ||
| 2302 | * contains only checksum items, and it can be safely freed without reading | ||
| 2303 | * it. | ||
| 2304 | * | ||
| 2305 | * This is just an optimization targeted at large files. It may do | ||
| 2306 | * nothing. It will return 0 unless things went badly. | ||
| 2307 | */ | ||
| 2308 | static noinline int drop_csum_leaves(struct btrfs_trans_handle *trans, | ||
| 2309 | struct btrfs_root *root, | ||
| 2310 | struct btrfs_path *path, | ||
| 2311 | struct inode *inode, u64 new_size) | ||
| 2312 | { | ||
| 2313 | struct btrfs_key key; | ||
| 2314 | int ret; | ||
| 2315 | int nritems; | ||
| 2316 | struct btrfs_key found_key; | ||
| 2317 | struct btrfs_key other_key; | ||
| 2318 | struct btrfs_leaf_ref *ref; | ||
| 2319 | u64 leaf_gen; | ||
| 2320 | u64 leaf_start; | ||
| 2321 | |||
| 2322 | path->lowest_level = 1; | ||
| 2323 | key.objectid = inode->i_ino; | ||
| 2324 | key.type = BTRFS_CSUM_ITEM_KEY; | ||
| 2325 | key.offset = new_size; | ||
| 2326 | again: | ||
| 2327 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 2328 | if (ret < 0) | ||
| 2329 | goto out; | ||
| 2330 | |||
| 2331 | if (path->nodes[1] == NULL) { | ||
| 2332 | ret = 0; | ||
| 2333 | goto out; | ||
| 2334 | } | ||
| 2335 | ret = 0; | ||
| 2336 | btrfs_node_key_to_cpu(path->nodes[1], &found_key, path->slots[1]); | ||
| 2337 | nritems = btrfs_header_nritems(path->nodes[1]); | ||
| 2338 | |||
| 2339 | if (!nritems) | ||
| 2340 | goto out; | ||
| 2341 | |||
| 2342 | if (path->slots[1] >= nritems) | ||
| 2343 | goto next_node; | ||
| 2344 | |||
| 2345 | /* did we find a key greater than anything we want to delete? */ | ||
| 2346 | if (found_key.objectid > inode->i_ino || | ||
| 2347 | (found_key.objectid == inode->i_ino && found_key.type > key.type)) | ||
| 2348 | goto out; | ||
| 2349 | |||
| 2350 | /* we check the next key in the node to make sure the leave contains | ||
| 2351 | * only checksum items. This comparison doesn't work if our | ||
| 2352 | * leaf is the last one in the node | ||
| 2353 | */ | ||
| 2354 | if (path->slots[1] + 1 >= nritems) { | ||
| 2355 | next_node: | ||
| 2356 | /* search forward from the last key in the node, this | ||
| 2357 | * will bring us into the next node in the tree | ||
| 2358 | */ | ||
| 2359 | btrfs_node_key_to_cpu(path->nodes[1], &found_key, nritems - 1); | ||
| 2360 | |||
| 2361 | /* unlikely, but we inc below, so check to be safe */ | ||
| 2362 | if (found_key.offset == (u64)-1) | ||
| 2363 | goto out; | ||
| 2364 | |||
| 2365 | /* search_forward needs a path with locks held, do the | ||
| 2366 | * search again for the original key. It is possible | ||
| 2367 | * this will race with a balance and return a path that | ||
| 2368 | * we could modify, but this drop is just an optimization | ||
| 2369 | * and is allowed to miss some leaves. | ||
| 2370 | */ | ||
| 2371 | btrfs_release_path(root, path); | ||
| 2372 | found_key.offset++; | ||
| 2373 | |||
| 2374 | /* setup a max key for search_forward */ | ||
| 2375 | other_key.offset = (u64)-1; | ||
| 2376 | other_key.type = key.type; | ||
| 2377 | other_key.objectid = key.objectid; | ||
| 2378 | |||
| 2379 | path->keep_locks = 1; | ||
| 2380 | ret = btrfs_search_forward(root, &found_key, &other_key, | ||
| 2381 | path, 0, 0); | ||
| 2382 | path->keep_locks = 0; | ||
| 2383 | if (ret || found_key.objectid != key.objectid || | ||
| 2384 | found_key.type != key.type) { | ||
| 2385 | ret = 0; | ||
| 2386 | goto out; | ||
| 2387 | } | ||
| 2388 | |||
| 2389 | key.offset = found_key.offset; | ||
| 2390 | btrfs_release_path(root, path); | ||
| 2391 | cond_resched(); | ||
| 2392 | goto again; | ||
| 2393 | } | ||
| 2394 | |||
| 2395 | /* we know there's one more slot after us in the tree, | ||
| 2396 | * read that key so we can verify it is also a checksum item | ||
| 2397 | */ | ||
| 2398 | btrfs_node_key_to_cpu(path->nodes[1], &other_key, path->slots[1] + 1); | ||
| 2399 | |||
| 2400 | if (found_key.objectid < inode->i_ino) | ||
| 2401 | goto next_key; | ||
| 2402 | |||
| 2403 | if (found_key.type != key.type || found_key.offset < new_size) | ||
| 2404 | goto next_key; | ||
| 2405 | |||
| 2406 | /* | ||
| 2407 | * if the key for the next leaf isn't a csum key from this objectid, | ||
| 2408 | * we can't be sure there aren't good items inside this leaf. | ||
| 2409 | * Bail out | ||
| 2410 | */ | ||
| 2411 | if (other_key.objectid != inode->i_ino || other_key.type != key.type) | ||
| 2412 | goto out; | ||
| 2413 | |||
| 2414 | leaf_start = btrfs_node_blockptr(path->nodes[1], path->slots[1]); | ||
| 2415 | leaf_gen = btrfs_node_ptr_generation(path->nodes[1], path->slots[1]); | ||
| 2416 | /* | ||
| 2417 | * it is safe to delete this leaf, it contains only | ||
| 2418 | * csum items from this inode at an offset >= new_size | ||
| 2419 | */ | ||
| 2420 | ret = btrfs_del_leaf(trans, root, path, leaf_start); | ||
| 2421 | BUG_ON(ret); | ||
| 2422 | |||
| 2423 | if (root->ref_cows && leaf_gen < trans->transid) { | ||
| 2424 | ref = btrfs_alloc_leaf_ref(root, 0); | ||
| 2425 | if (ref) { | ||
| 2426 | ref->root_gen = root->root_key.offset; | ||
| 2427 | ref->bytenr = leaf_start; | ||
| 2428 | ref->owner = 0; | ||
| 2429 | ref->generation = leaf_gen; | ||
| 2430 | ref->nritems = 0; | ||
| 2431 | |||
| 2432 | ret = btrfs_add_leaf_ref(root, ref, 0); | ||
| 2433 | WARN_ON(ret); | ||
| 2434 | btrfs_free_leaf_ref(root, ref); | ||
| 2435 | } else { | ||
| 2436 | WARN_ON(1); | ||
| 2437 | } | ||
| 2438 | } | ||
| 2439 | next_key: | ||
| 2440 | btrfs_release_path(root, path); | ||
| 2441 | |||
| 2442 | if (other_key.objectid == inode->i_ino && | ||
| 2443 | other_key.type == key.type && other_key.offset > key.offset) { | ||
| 2444 | key.offset = other_key.offset; | ||
| 2445 | cond_resched(); | ||
| 2446 | goto again; | ||
| 2447 | } | ||
| 2448 | ret = 0; | ||
| 2449 | out: | ||
| 2450 | /* fixup any changes we've made to the path */ | ||
| 2451 | path->lowest_level = 0; | ||
| 2452 | path->keep_locks = 0; | ||
| 2453 | btrfs_release_path(root, path); | ||
| 2454 | return ret; | ||
| 2455 | } | ||
| 2456 | |||
| 2457 | #endif | ||
| 2458 | |||
| 2459 | /* | ||
| 2460 | * this can truncate away extent items, csum items and directory items. | ||
| 2461 | * It starts at a high offset and removes keys until it can't find | ||
| 2462 | * any higher than new_size | ||
| 2463 | * | ||
| 2464 | * csum items that cross the new i_size are truncated to the new size | ||
| 2465 | * as well. | ||
| 2466 | * | ||
| 2467 | * min_type is the minimum key type to truncate down to. If set to 0, this | ||
| 2468 | * will kill all the items on this inode, including the INODE_ITEM_KEY. | ||
| 2469 | */ | ||
| 2470 | noinline int btrfs_truncate_inode_items(struct btrfs_trans_handle *trans, | ||
| 2471 | struct btrfs_root *root, | ||
| 2472 | struct inode *inode, | ||
| 2473 | u64 new_size, u32 min_type) | ||
| 2474 | { | ||
| 2475 | int ret; | ||
| 2476 | struct btrfs_path *path; | ||
| 2477 | struct btrfs_key key; | ||
| 2478 | struct btrfs_key found_key; | ||
| 2479 | u32 found_type; | ||
| 2480 | struct extent_buffer *leaf; | ||
| 2481 | struct btrfs_file_extent_item *fi; | ||
| 2482 | u64 extent_start = 0; | ||
| 2483 | u64 extent_num_bytes = 0; | ||
| 2484 | u64 item_end = 0; | ||
| 2485 | u64 root_gen = 0; | ||
| 2486 | u64 root_owner = 0; | ||
| 2487 | int found_extent; | ||
| 2488 | int del_item; | ||
| 2489 | int pending_del_nr = 0; | ||
| 2490 | int pending_del_slot = 0; | ||
| 2491 | int extent_type = -1; | ||
| 2492 | int encoding; | ||
| 2493 | u64 mask = root->sectorsize - 1; | ||
| 2494 | |||
| 2495 | if (root->ref_cows) | ||
| 2496 | btrfs_drop_extent_cache(inode, new_size & (~mask), (u64)-1, 0); | ||
| 2497 | path = btrfs_alloc_path(); | ||
| 2498 | path->reada = -1; | ||
| 2499 | BUG_ON(!path); | ||
| 2500 | |||
| 2501 | /* FIXME, add redo link to tree so we don't leak on crash */ | ||
| 2502 | key.objectid = inode->i_ino; | ||
| 2503 | key.offset = (u64)-1; | ||
| 2504 | key.type = (u8)-1; | ||
| 2505 | |||
| 2506 | btrfs_init_path(path); | ||
| 2507 | |||
| 2508 | search_again: | ||
| 2509 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 2510 | if (ret < 0) | ||
| 2511 | goto error; | ||
| 2512 | |||
| 2513 | if (ret > 0) { | ||
| 2514 | /* there are no items in the tree for us to truncate, we're | ||
| 2515 | * done | ||
| 2516 | */ | ||
| 2517 | if (path->slots[0] == 0) { | ||
| 2518 | ret = 0; | ||
| 2519 | goto error; | ||
| 2520 | } | ||
| 2521 | path->slots[0]--; | ||
| 2522 | } | ||
| 2523 | |||
| 2524 | while (1) { | ||
| 2525 | fi = NULL; | ||
| 2526 | leaf = path->nodes[0]; | ||
| 2527 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 2528 | found_type = btrfs_key_type(&found_key); | ||
| 2529 | encoding = 0; | ||
| 2530 | |||
| 2531 | if (found_key.objectid != inode->i_ino) | ||
| 2532 | break; | ||
| 2533 | |||
| 2534 | if (found_type < min_type) | ||
| 2535 | break; | ||
| 2536 | |||
| 2537 | item_end = found_key.offset; | ||
| 2538 | if (found_type == BTRFS_EXTENT_DATA_KEY) { | ||
| 2539 | fi = btrfs_item_ptr(leaf, path->slots[0], | ||
| 2540 | struct btrfs_file_extent_item); | ||
| 2541 | extent_type = btrfs_file_extent_type(leaf, fi); | ||
| 2542 | encoding = btrfs_file_extent_compression(leaf, fi); | ||
| 2543 | encoding |= btrfs_file_extent_encryption(leaf, fi); | ||
| 2544 | encoding |= btrfs_file_extent_other_encoding(leaf, fi); | ||
| 2545 | |||
| 2546 | if (extent_type != BTRFS_FILE_EXTENT_INLINE) { | ||
| 2547 | item_end += | ||
| 2548 | btrfs_file_extent_num_bytes(leaf, fi); | ||
| 2549 | } else if (extent_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 2550 | item_end += btrfs_file_extent_inline_len(leaf, | ||
| 2551 | fi); | ||
| 2552 | } | ||
| 2553 | item_end--; | ||
| 2554 | } | ||
| 2555 | if (item_end < new_size) { | ||
| 2556 | if (found_type == BTRFS_DIR_ITEM_KEY) | ||
| 2557 | found_type = BTRFS_INODE_ITEM_KEY; | ||
| 2558 | else if (found_type == BTRFS_EXTENT_ITEM_KEY) | ||
| 2559 | found_type = BTRFS_EXTENT_DATA_KEY; | ||
| 2560 | else if (found_type == BTRFS_EXTENT_DATA_KEY) | ||
| 2561 | found_type = BTRFS_XATTR_ITEM_KEY; | ||
| 2562 | else if (found_type == BTRFS_XATTR_ITEM_KEY) | ||
| 2563 | found_type = BTRFS_INODE_REF_KEY; | ||
| 2564 | else if (found_type) | ||
| 2565 | found_type--; | ||
| 2566 | else | ||
| 2567 | break; | ||
| 2568 | btrfs_set_key_type(&key, found_type); | ||
| 2569 | goto next; | ||
| 2570 | } | ||
| 2571 | if (found_key.offset >= new_size) | ||
| 2572 | del_item = 1; | ||
| 2573 | else | ||
| 2574 | del_item = 0; | ||
| 2575 | found_extent = 0; | ||
| 2576 | |||
| 2577 | /* FIXME, shrink the extent if the ref count is only 1 */ | ||
| 2578 | if (found_type != BTRFS_EXTENT_DATA_KEY) | ||
| 2579 | goto delete; | ||
| 2580 | |||
| 2581 | if (extent_type != BTRFS_FILE_EXTENT_INLINE) { | ||
| 2582 | u64 num_dec; | ||
| 2583 | extent_start = btrfs_file_extent_disk_bytenr(leaf, fi); | ||
| 2584 | if (!del_item && !encoding) { | ||
| 2585 | u64 orig_num_bytes = | ||
| 2586 | btrfs_file_extent_num_bytes(leaf, fi); | ||
| 2587 | extent_num_bytes = new_size - | ||
| 2588 | found_key.offset + root->sectorsize - 1; | ||
| 2589 | extent_num_bytes = extent_num_bytes & | ||
| 2590 | ~((u64)root->sectorsize - 1); | ||
| 2591 | btrfs_set_file_extent_num_bytes(leaf, fi, | ||
| 2592 | extent_num_bytes); | ||
| 2593 | num_dec = (orig_num_bytes - | ||
| 2594 | extent_num_bytes); | ||
| 2595 | if (root->ref_cows && extent_start != 0) | ||
| 2596 | inode_sub_bytes(inode, num_dec); | ||
| 2597 | btrfs_mark_buffer_dirty(leaf); | ||
| 2598 | } else { | ||
| 2599 | extent_num_bytes = | ||
| 2600 | btrfs_file_extent_disk_num_bytes(leaf, | ||
| 2601 | fi); | ||
| 2602 | /* FIXME blocksize != 4096 */ | ||
| 2603 | num_dec = btrfs_file_extent_num_bytes(leaf, fi); | ||
| 2604 | if (extent_start != 0) { | ||
| 2605 | found_extent = 1; | ||
| 2606 | if (root->ref_cows) | ||
| 2607 | inode_sub_bytes(inode, num_dec); | ||
| 2608 | } | ||
| 2609 | root_gen = btrfs_header_generation(leaf); | ||
| 2610 | root_owner = btrfs_header_owner(leaf); | ||
| 2611 | } | ||
| 2612 | } else if (extent_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 2613 | /* | ||
| 2614 | * we can't truncate inline items that have had | ||
| 2615 | * special encodings | ||
| 2616 | */ | ||
| 2617 | if (!del_item && | ||
| 2618 | btrfs_file_extent_compression(leaf, fi) == 0 && | ||
| 2619 | btrfs_file_extent_encryption(leaf, fi) == 0 && | ||
| 2620 | btrfs_file_extent_other_encoding(leaf, fi) == 0) { | ||
| 2621 | u32 size = new_size - found_key.offset; | ||
| 2622 | |||
| 2623 | if (root->ref_cows) { | ||
| 2624 | inode_sub_bytes(inode, item_end + 1 - | ||
| 2625 | new_size); | ||
| 2626 | } | ||
| 2627 | size = | ||
| 2628 | btrfs_file_extent_calc_inline_size(size); | ||
| 2629 | ret = btrfs_truncate_item(trans, root, path, | ||
| 2630 | size, 1); | ||
| 2631 | BUG_ON(ret); | ||
| 2632 | } else if (root->ref_cows) { | ||
| 2633 | inode_sub_bytes(inode, item_end + 1 - | ||
| 2634 | found_key.offset); | ||
| 2635 | } | ||
| 2636 | } | ||
| 2637 | delete: | ||
| 2638 | if (del_item) { | ||
| 2639 | if (!pending_del_nr) { | ||
| 2640 | /* no pending yet, add ourselves */ | ||
| 2641 | pending_del_slot = path->slots[0]; | ||
| 2642 | pending_del_nr = 1; | ||
| 2643 | } else if (pending_del_nr && | ||
| 2644 | path->slots[0] + 1 == pending_del_slot) { | ||
| 2645 | /* hop on the pending chunk */ | ||
| 2646 | pending_del_nr++; | ||
| 2647 | pending_del_slot = path->slots[0]; | ||
| 2648 | } else { | ||
| 2649 | BUG(); | ||
| 2650 | } | ||
| 2651 | } else { | ||
| 2652 | break; | ||
| 2653 | } | ||
| 2654 | if (found_extent) { | ||
| 2655 | ret = btrfs_free_extent(trans, root, extent_start, | ||
| 2656 | extent_num_bytes, | ||
| 2657 | leaf->start, root_owner, | ||
| 2658 | root_gen, inode->i_ino, 0); | ||
| 2659 | BUG_ON(ret); | ||
| 2660 | } | ||
| 2661 | next: | ||
| 2662 | if (path->slots[0] == 0) { | ||
| 2663 | if (pending_del_nr) | ||
| 2664 | goto del_pending; | ||
| 2665 | btrfs_release_path(root, path); | ||
| 2666 | goto search_again; | ||
| 2667 | } | ||
| 2668 | |||
| 2669 | path->slots[0]--; | ||
| 2670 | if (pending_del_nr && | ||
| 2671 | path->slots[0] + 1 != pending_del_slot) { | ||
| 2672 | struct btrfs_key debug; | ||
| 2673 | del_pending: | ||
| 2674 | btrfs_item_key_to_cpu(path->nodes[0], &debug, | ||
| 2675 | pending_del_slot); | ||
| 2676 | ret = btrfs_del_items(trans, root, path, | ||
| 2677 | pending_del_slot, | ||
| 2678 | pending_del_nr); | ||
| 2679 | BUG_ON(ret); | ||
| 2680 | pending_del_nr = 0; | ||
| 2681 | btrfs_release_path(root, path); | ||
| 2682 | goto search_again; | ||
| 2683 | } | ||
| 2684 | } | ||
| 2685 | ret = 0; | ||
| 2686 | error: | ||
| 2687 | if (pending_del_nr) { | ||
| 2688 | ret = btrfs_del_items(trans, root, path, pending_del_slot, | ||
| 2689 | pending_del_nr); | ||
| 2690 | } | ||
| 2691 | btrfs_free_path(path); | ||
| 2692 | inode->i_sb->s_dirt = 1; | ||
| 2693 | return ret; | ||
| 2694 | } | ||
| 2695 | |||
| 2696 | /* | ||
| 2697 | * taken from block_truncate_page, but does cow as it zeros out | ||
| 2698 | * any bytes left in the last page in the file. | ||
| 2699 | */ | ||
| 2700 | static int btrfs_truncate_page(struct address_space *mapping, loff_t from) | ||
| 2701 | { | ||
| 2702 | struct inode *inode = mapping->host; | ||
| 2703 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 2704 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 2705 | struct btrfs_ordered_extent *ordered; | ||
| 2706 | char *kaddr; | ||
| 2707 | u32 blocksize = root->sectorsize; | ||
| 2708 | pgoff_t index = from >> PAGE_CACHE_SHIFT; | ||
| 2709 | unsigned offset = from & (PAGE_CACHE_SIZE-1); | ||
| 2710 | struct page *page; | ||
| 2711 | int ret = 0; | ||
| 2712 | u64 page_start; | ||
| 2713 | u64 page_end; | ||
| 2714 | |||
| 2715 | if ((offset & (blocksize - 1)) == 0) | ||
| 2716 | goto out; | ||
| 2717 | |||
| 2718 | ret = -ENOMEM; | ||
| 2719 | again: | ||
| 2720 | page = grab_cache_page(mapping, index); | ||
| 2721 | if (!page) | ||
| 2722 | goto out; | ||
| 2723 | |||
| 2724 | page_start = page_offset(page); | ||
| 2725 | page_end = page_start + PAGE_CACHE_SIZE - 1; | ||
| 2726 | |||
| 2727 | if (!PageUptodate(page)) { | ||
| 2728 | ret = btrfs_readpage(NULL, page); | ||
| 2729 | lock_page(page); | ||
| 2730 | if (page->mapping != mapping) { | ||
| 2731 | unlock_page(page); | ||
| 2732 | page_cache_release(page); | ||
| 2733 | goto again; | ||
| 2734 | } | ||
| 2735 | if (!PageUptodate(page)) { | ||
| 2736 | ret = -EIO; | ||
| 2737 | goto out_unlock; | ||
| 2738 | } | ||
| 2739 | } | ||
| 2740 | wait_on_page_writeback(page); | ||
| 2741 | |||
| 2742 | lock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 2743 | set_page_extent_mapped(page); | ||
| 2744 | |||
| 2745 | ordered = btrfs_lookup_ordered_extent(inode, page_start); | ||
| 2746 | if (ordered) { | ||
| 2747 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 2748 | unlock_page(page); | ||
| 2749 | page_cache_release(page); | ||
| 2750 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 2751 | btrfs_put_ordered_extent(ordered); | ||
| 2752 | goto again; | ||
| 2753 | } | ||
| 2754 | |||
| 2755 | btrfs_set_extent_delalloc(inode, page_start, page_end); | ||
| 2756 | ret = 0; | ||
| 2757 | if (offset != PAGE_CACHE_SIZE) { | ||
| 2758 | kaddr = kmap(page); | ||
| 2759 | memset(kaddr + offset, 0, PAGE_CACHE_SIZE - offset); | ||
| 2760 | flush_dcache_page(page); | ||
| 2761 | kunmap(page); | ||
| 2762 | } | ||
| 2763 | ClearPageChecked(page); | ||
| 2764 | set_page_dirty(page); | ||
| 2765 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 2766 | |||
| 2767 | out_unlock: | ||
| 2768 | unlock_page(page); | ||
| 2769 | page_cache_release(page); | ||
| 2770 | out: | ||
| 2771 | return ret; | ||
| 2772 | } | ||
| 2773 | |||
| 2774 | int btrfs_cont_expand(struct inode *inode, loff_t size) | ||
| 2775 | { | ||
| 2776 | struct btrfs_trans_handle *trans; | ||
| 2777 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 2778 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 2779 | struct extent_map *em; | ||
| 2780 | u64 mask = root->sectorsize - 1; | ||
| 2781 | u64 hole_start = (inode->i_size + mask) & ~mask; | ||
| 2782 | u64 block_end = (size + mask) & ~mask; | ||
| 2783 | u64 last_byte; | ||
| 2784 | u64 cur_offset; | ||
| 2785 | u64 hole_size; | ||
| 2786 | int err; | ||
| 2787 | |||
| 2788 | if (size <= hole_start) | ||
| 2789 | return 0; | ||
| 2790 | |||
| 2791 | err = btrfs_check_free_space(root, 1, 0); | ||
| 2792 | if (err) | ||
| 2793 | return err; | ||
| 2794 | |||
| 2795 | btrfs_truncate_page(inode->i_mapping, inode->i_size); | ||
| 2796 | |||
| 2797 | while (1) { | ||
| 2798 | struct btrfs_ordered_extent *ordered; | ||
| 2799 | btrfs_wait_ordered_range(inode, hole_start, | ||
| 2800 | block_end - hole_start); | ||
| 2801 | lock_extent(io_tree, hole_start, block_end - 1, GFP_NOFS); | ||
| 2802 | ordered = btrfs_lookup_ordered_extent(inode, hole_start); | ||
| 2803 | if (!ordered) | ||
| 2804 | break; | ||
| 2805 | unlock_extent(io_tree, hole_start, block_end - 1, GFP_NOFS); | ||
| 2806 | btrfs_put_ordered_extent(ordered); | ||
| 2807 | } | ||
| 2808 | |||
| 2809 | trans = btrfs_start_transaction(root, 1); | ||
| 2810 | btrfs_set_trans_block_group(trans, inode); | ||
| 2811 | |||
| 2812 | cur_offset = hole_start; | ||
| 2813 | while (1) { | ||
| 2814 | em = btrfs_get_extent(inode, NULL, 0, cur_offset, | ||
| 2815 | block_end - cur_offset, 0); | ||
| 2816 | BUG_ON(IS_ERR(em) || !em); | ||
| 2817 | last_byte = min(extent_map_end(em), block_end); | ||
| 2818 | last_byte = (last_byte + mask) & ~mask; | ||
| 2819 | if (test_bit(EXTENT_FLAG_VACANCY, &em->flags)) { | ||
| 2820 | u64 hint_byte = 0; | ||
| 2821 | hole_size = last_byte - cur_offset; | ||
| 2822 | err = btrfs_drop_extents(trans, root, inode, | ||
| 2823 | cur_offset, | ||
| 2824 | cur_offset + hole_size, | ||
| 2825 | cur_offset, &hint_byte); | ||
| 2826 | if (err) | ||
| 2827 | break; | ||
| 2828 | err = btrfs_insert_file_extent(trans, root, | ||
| 2829 | inode->i_ino, cur_offset, 0, | ||
| 2830 | 0, hole_size, 0, hole_size, | ||
| 2831 | 0, 0, 0); | ||
| 2832 | btrfs_drop_extent_cache(inode, hole_start, | ||
| 2833 | last_byte - 1, 0); | ||
| 2834 | } | ||
| 2835 | free_extent_map(em); | ||
| 2836 | cur_offset = last_byte; | ||
| 2837 | if (err || cur_offset >= block_end) | ||
| 2838 | break; | ||
| 2839 | } | ||
| 2840 | |||
| 2841 | btrfs_end_transaction(trans, root); | ||
| 2842 | unlock_extent(io_tree, hole_start, block_end - 1, GFP_NOFS); | ||
| 2843 | return err; | ||
| 2844 | } | ||
| 2845 | |||
| 2846 | static int btrfs_setattr(struct dentry *dentry, struct iattr *attr) | ||
| 2847 | { | ||
| 2848 | struct inode *inode = dentry->d_inode; | ||
| 2849 | int err; | ||
| 2850 | |||
| 2851 | err = inode_change_ok(inode, attr); | ||
| 2852 | if (err) | ||
| 2853 | return err; | ||
| 2854 | |||
| 2855 | if (S_ISREG(inode->i_mode) && | ||
| 2856 | attr->ia_valid & ATTR_SIZE && attr->ia_size > inode->i_size) { | ||
| 2857 | err = btrfs_cont_expand(inode, attr->ia_size); | ||
| 2858 | if (err) | ||
| 2859 | return err; | ||
| 2860 | } | ||
| 2861 | |||
| 2862 | err = inode_setattr(inode, attr); | ||
| 2863 | |||
| 2864 | if (!err && ((attr->ia_valid & ATTR_MODE))) | ||
| 2865 | err = btrfs_acl_chmod(inode); | ||
| 2866 | return err; | ||
| 2867 | } | ||
| 2868 | |||
| 2869 | void btrfs_delete_inode(struct inode *inode) | ||
| 2870 | { | ||
| 2871 | struct btrfs_trans_handle *trans; | ||
| 2872 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 2873 | unsigned long nr; | ||
| 2874 | int ret; | ||
| 2875 | |||
| 2876 | truncate_inode_pages(&inode->i_data, 0); | ||
| 2877 | if (is_bad_inode(inode)) { | ||
| 2878 | btrfs_orphan_del(NULL, inode); | ||
| 2879 | goto no_delete; | ||
| 2880 | } | ||
| 2881 | btrfs_wait_ordered_range(inode, 0, (u64)-1); | ||
| 2882 | |||
| 2883 | btrfs_i_size_write(inode, 0); | ||
| 2884 | trans = btrfs_join_transaction(root, 1); | ||
| 2885 | |||
| 2886 | btrfs_set_trans_block_group(trans, inode); | ||
| 2887 | ret = btrfs_truncate_inode_items(trans, root, inode, inode->i_size, 0); | ||
| 2888 | if (ret) { | ||
| 2889 | btrfs_orphan_del(NULL, inode); | ||
| 2890 | goto no_delete_lock; | ||
| 2891 | } | ||
| 2892 | |||
| 2893 | btrfs_orphan_del(trans, inode); | ||
| 2894 | |||
| 2895 | nr = trans->blocks_used; | ||
| 2896 | clear_inode(inode); | ||
| 2897 | |||
| 2898 | btrfs_end_transaction(trans, root); | ||
| 2899 | btrfs_btree_balance_dirty(root, nr); | ||
| 2900 | return; | ||
| 2901 | |||
| 2902 | no_delete_lock: | ||
| 2903 | nr = trans->blocks_used; | ||
| 2904 | btrfs_end_transaction(trans, root); | ||
| 2905 | btrfs_btree_balance_dirty(root, nr); | ||
| 2906 | no_delete: | ||
| 2907 | clear_inode(inode); | ||
| 2908 | } | ||
| 2909 | |||
| 2910 | /* | ||
| 2911 | * this returns the key found in the dir entry in the location pointer. | ||
| 2912 | * If no dir entries were found, location->objectid is 0. | ||
| 2913 | */ | ||
| 2914 | static int btrfs_inode_by_name(struct inode *dir, struct dentry *dentry, | ||
| 2915 | struct btrfs_key *location) | ||
| 2916 | { | ||
| 2917 | const char *name = dentry->d_name.name; | ||
| 2918 | int namelen = dentry->d_name.len; | ||
| 2919 | struct btrfs_dir_item *di; | ||
| 2920 | struct btrfs_path *path; | ||
| 2921 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 2922 | int ret = 0; | ||
| 2923 | |||
| 2924 | path = btrfs_alloc_path(); | ||
| 2925 | BUG_ON(!path); | ||
| 2926 | |||
| 2927 | di = btrfs_lookup_dir_item(NULL, root, path, dir->i_ino, name, | ||
| 2928 | namelen, 0); | ||
| 2929 | if (IS_ERR(di)) | ||
| 2930 | ret = PTR_ERR(di); | ||
| 2931 | |||
| 2932 | if (!di || IS_ERR(di)) | ||
| 2933 | goto out_err; | ||
| 2934 | |||
| 2935 | btrfs_dir_item_key_to_cpu(path->nodes[0], di, location); | ||
| 2936 | out: | ||
| 2937 | btrfs_free_path(path); | ||
| 2938 | return ret; | ||
| 2939 | out_err: | ||
| 2940 | location->objectid = 0; | ||
| 2941 | goto out; | ||
| 2942 | } | ||
| 2943 | |||
| 2944 | /* | ||
| 2945 | * when we hit a tree root in a directory, the btrfs part of the inode | ||
| 2946 | * needs to be changed to reflect the root directory of the tree root. This | ||
| 2947 | * is kind of like crossing a mount point. | ||
| 2948 | */ | ||
| 2949 | static int fixup_tree_root_location(struct btrfs_root *root, | ||
| 2950 | struct btrfs_key *location, | ||
| 2951 | struct btrfs_root **sub_root, | ||
| 2952 | struct dentry *dentry) | ||
| 2953 | { | ||
| 2954 | struct btrfs_root_item *ri; | ||
| 2955 | |||
| 2956 | if (btrfs_key_type(location) != BTRFS_ROOT_ITEM_KEY) | ||
| 2957 | return 0; | ||
| 2958 | if (location->objectid == BTRFS_ROOT_TREE_OBJECTID) | ||
| 2959 | return 0; | ||
| 2960 | |||
| 2961 | *sub_root = btrfs_read_fs_root(root->fs_info, location, | ||
| 2962 | dentry->d_name.name, | ||
| 2963 | dentry->d_name.len); | ||
| 2964 | if (IS_ERR(*sub_root)) | ||
| 2965 | return PTR_ERR(*sub_root); | ||
| 2966 | |||
| 2967 | ri = &(*sub_root)->root_item; | ||
| 2968 | location->objectid = btrfs_root_dirid(ri); | ||
| 2969 | btrfs_set_key_type(location, BTRFS_INODE_ITEM_KEY); | ||
| 2970 | location->offset = 0; | ||
| 2971 | |||
| 2972 | return 0; | ||
| 2973 | } | ||
| 2974 | |||
| 2975 | static noinline void init_btrfs_i(struct inode *inode) | ||
| 2976 | { | ||
| 2977 | struct btrfs_inode *bi = BTRFS_I(inode); | ||
| 2978 | |||
| 2979 | bi->i_acl = NULL; | ||
| 2980 | bi->i_default_acl = NULL; | ||
| 2981 | |||
| 2982 | bi->generation = 0; | ||
| 2983 | bi->sequence = 0; | ||
| 2984 | bi->last_trans = 0; | ||
| 2985 | bi->logged_trans = 0; | ||
| 2986 | bi->delalloc_bytes = 0; | ||
| 2987 | bi->disk_i_size = 0; | ||
| 2988 | bi->flags = 0; | ||
| 2989 | bi->index_cnt = (u64)-1; | ||
| 2990 | bi->log_dirty_trans = 0; | ||
| 2991 | extent_map_tree_init(&BTRFS_I(inode)->extent_tree, GFP_NOFS); | ||
| 2992 | extent_io_tree_init(&BTRFS_I(inode)->io_tree, | ||
| 2993 | inode->i_mapping, GFP_NOFS); | ||
| 2994 | extent_io_tree_init(&BTRFS_I(inode)->io_failure_tree, | ||
| 2995 | inode->i_mapping, GFP_NOFS); | ||
| 2996 | INIT_LIST_HEAD(&BTRFS_I(inode)->delalloc_inodes); | ||
| 2997 | btrfs_ordered_inode_tree_init(&BTRFS_I(inode)->ordered_tree); | ||
| 2998 | mutex_init(&BTRFS_I(inode)->extent_mutex); | ||
| 2999 | mutex_init(&BTRFS_I(inode)->log_mutex); | ||
| 3000 | } | ||
| 3001 | |||
| 3002 | static int btrfs_init_locked_inode(struct inode *inode, void *p) | ||
| 3003 | { | ||
| 3004 | struct btrfs_iget_args *args = p; | ||
| 3005 | inode->i_ino = args->ino; | ||
| 3006 | init_btrfs_i(inode); | ||
| 3007 | BTRFS_I(inode)->root = args->root; | ||
| 3008 | return 0; | ||
| 3009 | } | ||
| 3010 | |||
| 3011 | static int btrfs_find_actor(struct inode *inode, void *opaque) | ||
| 3012 | { | ||
| 3013 | struct btrfs_iget_args *args = opaque; | ||
| 3014 | return args->ino == inode->i_ino && | ||
| 3015 | args->root == BTRFS_I(inode)->root; | ||
| 3016 | } | ||
| 3017 | |||
| 3018 | struct inode *btrfs_ilookup(struct super_block *s, u64 objectid, | ||
| 3019 | struct btrfs_root *root, int wait) | ||
| 3020 | { | ||
| 3021 | struct inode *inode; | ||
| 3022 | struct btrfs_iget_args args; | ||
| 3023 | args.ino = objectid; | ||
| 3024 | args.root = root; | ||
| 3025 | |||
| 3026 | if (wait) { | ||
| 3027 | inode = ilookup5(s, objectid, btrfs_find_actor, | ||
| 3028 | (void *)&args); | ||
| 3029 | } else { | ||
| 3030 | inode = ilookup5_nowait(s, objectid, btrfs_find_actor, | ||
| 3031 | (void *)&args); | ||
| 3032 | } | ||
| 3033 | return inode; | ||
| 3034 | } | ||
| 3035 | |||
| 3036 | struct inode *btrfs_iget_locked(struct super_block *s, u64 objectid, | ||
| 3037 | struct btrfs_root *root) | ||
| 3038 | { | ||
| 3039 | struct inode *inode; | ||
| 3040 | struct btrfs_iget_args args; | ||
| 3041 | args.ino = objectid; | ||
| 3042 | args.root = root; | ||
| 3043 | |||
| 3044 | inode = iget5_locked(s, objectid, btrfs_find_actor, | ||
| 3045 | btrfs_init_locked_inode, | ||
| 3046 | (void *)&args); | ||
| 3047 | return inode; | ||
| 3048 | } | ||
| 3049 | |||
| 3050 | /* Get an inode object given its location and corresponding root. | ||
| 3051 | * Returns in *is_new if the inode was read from disk | ||
| 3052 | */ | ||
| 3053 | struct inode *btrfs_iget(struct super_block *s, struct btrfs_key *location, | ||
| 3054 | struct btrfs_root *root, int *is_new) | ||
| 3055 | { | ||
| 3056 | struct inode *inode; | ||
| 3057 | |||
| 3058 | inode = btrfs_iget_locked(s, location->objectid, root); | ||
| 3059 | if (!inode) | ||
| 3060 | return ERR_PTR(-EACCES); | ||
| 3061 | |||
| 3062 | if (inode->i_state & I_NEW) { | ||
| 3063 | BTRFS_I(inode)->root = root; | ||
| 3064 | memcpy(&BTRFS_I(inode)->location, location, sizeof(*location)); | ||
| 3065 | btrfs_read_locked_inode(inode); | ||
| 3066 | unlock_new_inode(inode); | ||
| 3067 | if (is_new) | ||
| 3068 | *is_new = 1; | ||
| 3069 | } else { | ||
| 3070 | if (is_new) | ||
| 3071 | *is_new = 0; | ||
| 3072 | } | ||
| 3073 | |||
| 3074 | return inode; | ||
| 3075 | } | ||
| 3076 | |||
| 3077 | struct inode *btrfs_lookup_dentry(struct inode *dir, struct dentry *dentry) | ||
| 3078 | { | ||
| 3079 | struct inode *inode; | ||
| 3080 | struct btrfs_inode *bi = BTRFS_I(dir); | ||
| 3081 | struct btrfs_root *root = bi->root; | ||
| 3082 | struct btrfs_root *sub_root = root; | ||
| 3083 | struct btrfs_key location; | ||
| 3084 | int ret, new; | ||
| 3085 | |||
| 3086 | if (dentry->d_name.len > BTRFS_NAME_LEN) | ||
| 3087 | return ERR_PTR(-ENAMETOOLONG); | ||
| 3088 | |||
| 3089 | ret = btrfs_inode_by_name(dir, dentry, &location); | ||
| 3090 | |||
| 3091 | if (ret < 0) | ||
| 3092 | return ERR_PTR(ret); | ||
| 3093 | |||
| 3094 | inode = NULL; | ||
| 3095 | if (location.objectid) { | ||
| 3096 | ret = fixup_tree_root_location(root, &location, &sub_root, | ||
| 3097 | dentry); | ||
| 3098 | if (ret < 0) | ||
| 3099 | return ERR_PTR(ret); | ||
| 3100 | if (ret > 0) | ||
| 3101 | return ERR_PTR(-ENOENT); | ||
| 3102 | inode = btrfs_iget(dir->i_sb, &location, sub_root, &new); | ||
| 3103 | if (IS_ERR(inode)) | ||
| 3104 | return ERR_CAST(inode); | ||
| 3105 | } | ||
| 3106 | return inode; | ||
| 3107 | } | ||
| 3108 | |||
| 3109 | static struct dentry *btrfs_lookup(struct inode *dir, struct dentry *dentry, | ||
| 3110 | struct nameidata *nd) | ||
| 3111 | { | ||
| 3112 | struct inode *inode; | ||
| 3113 | |||
| 3114 | if (dentry->d_name.len > BTRFS_NAME_LEN) | ||
| 3115 | return ERR_PTR(-ENAMETOOLONG); | ||
| 3116 | |||
| 3117 | inode = btrfs_lookup_dentry(dir, dentry); | ||
| 3118 | if (IS_ERR(inode)) | ||
| 3119 | return ERR_CAST(inode); | ||
| 3120 | |||
| 3121 | return d_splice_alias(inode, dentry); | ||
| 3122 | } | ||
| 3123 | |||
| 3124 | static unsigned char btrfs_filetype_table[] = { | ||
| 3125 | DT_UNKNOWN, DT_REG, DT_DIR, DT_CHR, DT_BLK, DT_FIFO, DT_SOCK, DT_LNK | ||
| 3126 | }; | ||
| 3127 | |||
| 3128 | static int btrfs_real_readdir(struct file *filp, void *dirent, | ||
| 3129 | filldir_t filldir) | ||
| 3130 | { | ||
| 3131 | struct inode *inode = filp->f_dentry->d_inode; | ||
| 3132 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 3133 | struct btrfs_item *item; | ||
| 3134 | struct btrfs_dir_item *di; | ||
| 3135 | struct btrfs_key key; | ||
| 3136 | struct btrfs_key found_key; | ||
| 3137 | struct btrfs_path *path; | ||
| 3138 | int ret; | ||
| 3139 | u32 nritems; | ||
| 3140 | struct extent_buffer *leaf; | ||
| 3141 | int slot; | ||
| 3142 | int advance; | ||
| 3143 | unsigned char d_type; | ||
| 3144 | int over = 0; | ||
| 3145 | u32 di_cur; | ||
| 3146 | u32 di_total; | ||
| 3147 | u32 di_len; | ||
| 3148 | int key_type = BTRFS_DIR_INDEX_KEY; | ||
| 3149 | char tmp_name[32]; | ||
| 3150 | char *name_ptr; | ||
| 3151 | int name_len; | ||
| 3152 | |||
| 3153 | /* FIXME, use a real flag for deciding about the key type */ | ||
| 3154 | if (root->fs_info->tree_root == root) | ||
| 3155 | key_type = BTRFS_DIR_ITEM_KEY; | ||
| 3156 | |||
| 3157 | /* special case for "." */ | ||
| 3158 | if (filp->f_pos == 0) { | ||
| 3159 | over = filldir(dirent, ".", 1, | ||
| 3160 | 1, inode->i_ino, | ||
| 3161 | DT_DIR); | ||
| 3162 | if (over) | ||
| 3163 | return 0; | ||
| 3164 | filp->f_pos = 1; | ||
| 3165 | } | ||
| 3166 | /* special case for .., just use the back ref */ | ||
| 3167 | if (filp->f_pos == 1) { | ||
| 3168 | u64 pino = parent_ino(filp->f_path.dentry); | ||
| 3169 | over = filldir(dirent, "..", 2, | ||
| 3170 | 2, pino, DT_DIR); | ||
| 3171 | if (over) | ||
| 3172 | return 0; | ||
| 3173 | filp->f_pos = 2; | ||
| 3174 | } | ||
| 3175 | path = btrfs_alloc_path(); | ||
| 3176 | path->reada = 2; | ||
| 3177 | |||
| 3178 | btrfs_set_key_type(&key, key_type); | ||
| 3179 | key.offset = filp->f_pos; | ||
| 3180 | key.objectid = inode->i_ino; | ||
| 3181 | |||
| 3182 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 3183 | if (ret < 0) | ||
| 3184 | goto err; | ||
| 3185 | advance = 0; | ||
| 3186 | |||
| 3187 | while (1) { | ||
| 3188 | leaf = path->nodes[0]; | ||
| 3189 | nritems = btrfs_header_nritems(leaf); | ||
| 3190 | slot = path->slots[0]; | ||
| 3191 | if (advance || slot >= nritems) { | ||
| 3192 | if (slot >= nritems - 1) { | ||
| 3193 | ret = btrfs_next_leaf(root, path); | ||
| 3194 | if (ret) | ||
| 3195 | break; | ||
| 3196 | leaf = path->nodes[0]; | ||
| 3197 | nritems = btrfs_header_nritems(leaf); | ||
| 3198 | slot = path->slots[0]; | ||
| 3199 | } else { | ||
| 3200 | slot++; | ||
| 3201 | path->slots[0]++; | ||
| 3202 | } | ||
| 3203 | } | ||
| 3204 | |||
| 3205 | advance = 1; | ||
| 3206 | item = btrfs_item_nr(leaf, slot); | ||
| 3207 | btrfs_item_key_to_cpu(leaf, &found_key, slot); | ||
| 3208 | |||
| 3209 | if (found_key.objectid != key.objectid) | ||
| 3210 | break; | ||
| 3211 | if (btrfs_key_type(&found_key) != key_type) | ||
| 3212 | break; | ||
| 3213 | if (found_key.offset < filp->f_pos) | ||
| 3214 | continue; | ||
| 3215 | |||
| 3216 | filp->f_pos = found_key.offset; | ||
| 3217 | |||
| 3218 | di = btrfs_item_ptr(leaf, slot, struct btrfs_dir_item); | ||
| 3219 | di_cur = 0; | ||
| 3220 | di_total = btrfs_item_size(leaf, item); | ||
| 3221 | |||
| 3222 | while (di_cur < di_total) { | ||
| 3223 | struct btrfs_key location; | ||
| 3224 | |||
| 3225 | name_len = btrfs_dir_name_len(leaf, di); | ||
| 3226 | if (name_len <= sizeof(tmp_name)) { | ||
| 3227 | name_ptr = tmp_name; | ||
| 3228 | } else { | ||
| 3229 | name_ptr = kmalloc(name_len, GFP_NOFS); | ||
| 3230 | if (!name_ptr) { | ||
| 3231 | ret = -ENOMEM; | ||
| 3232 | goto err; | ||
| 3233 | } | ||
| 3234 | } | ||
| 3235 | read_extent_buffer(leaf, name_ptr, | ||
| 3236 | (unsigned long)(di + 1), name_len); | ||
| 3237 | |||
| 3238 | d_type = btrfs_filetype_table[btrfs_dir_type(leaf, di)]; | ||
| 3239 | btrfs_dir_item_key_to_cpu(leaf, di, &location); | ||
| 3240 | |||
| 3241 | /* is this a reference to our own snapshot? If so | ||
| 3242 | * skip it | ||
| 3243 | */ | ||
| 3244 | if (location.type == BTRFS_ROOT_ITEM_KEY && | ||
| 3245 | location.objectid == root->root_key.objectid) { | ||
| 3246 | over = 0; | ||
| 3247 | goto skip; | ||
| 3248 | } | ||
| 3249 | over = filldir(dirent, name_ptr, name_len, | ||
| 3250 | found_key.offset, location.objectid, | ||
| 3251 | d_type); | ||
| 3252 | |||
| 3253 | skip: | ||
| 3254 | if (name_ptr != tmp_name) | ||
| 3255 | kfree(name_ptr); | ||
| 3256 | |||
| 3257 | if (over) | ||
| 3258 | goto nopos; | ||
| 3259 | di_len = btrfs_dir_name_len(leaf, di) + | ||
| 3260 | btrfs_dir_data_len(leaf, di) + sizeof(*di); | ||
| 3261 | di_cur += di_len; | ||
| 3262 | di = (struct btrfs_dir_item *)((char *)di + di_len); | ||
| 3263 | } | ||
| 3264 | } | ||
| 3265 | |||
| 3266 | /* Reached end of directory/root. Bump pos past the last item. */ | ||
| 3267 | if (key_type == BTRFS_DIR_INDEX_KEY) | ||
| 3268 | filp->f_pos = INT_LIMIT(typeof(filp->f_pos)); | ||
| 3269 | else | ||
| 3270 | filp->f_pos++; | ||
| 3271 | nopos: | ||
| 3272 | ret = 0; | ||
| 3273 | err: | ||
| 3274 | btrfs_free_path(path); | ||
| 3275 | return ret; | ||
| 3276 | } | ||
| 3277 | |||
| 3278 | int btrfs_write_inode(struct inode *inode, int wait) | ||
| 3279 | { | ||
| 3280 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 3281 | struct btrfs_trans_handle *trans; | ||
| 3282 | int ret = 0; | ||
| 3283 | |||
| 3284 | if (root->fs_info->btree_inode == inode) | ||
| 3285 | return 0; | ||
| 3286 | |||
| 3287 | if (wait) { | ||
| 3288 | trans = btrfs_join_transaction(root, 1); | ||
| 3289 | btrfs_set_trans_block_group(trans, inode); | ||
| 3290 | ret = btrfs_commit_transaction(trans, root); | ||
| 3291 | } | ||
| 3292 | return ret; | ||
| 3293 | } | ||
| 3294 | |||
| 3295 | /* | ||
| 3296 | * This is somewhat expensive, updating the tree every time the | ||
| 3297 | * inode changes. But, it is most likely to find the inode in cache. | ||
| 3298 | * FIXME, needs more benchmarking...there are no reasons other than performance | ||
| 3299 | * to keep or drop this code. | ||
| 3300 | */ | ||
| 3301 | void btrfs_dirty_inode(struct inode *inode) | ||
| 3302 | { | ||
| 3303 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 3304 | struct btrfs_trans_handle *trans; | ||
| 3305 | |||
| 3306 | trans = btrfs_join_transaction(root, 1); | ||
| 3307 | btrfs_set_trans_block_group(trans, inode); | ||
| 3308 | btrfs_update_inode(trans, root, inode); | ||
| 3309 | btrfs_end_transaction(trans, root); | ||
| 3310 | } | ||
| 3311 | |||
| 3312 | /* | ||
| 3313 | * find the highest existing sequence number in a directory | ||
| 3314 | * and then set the in-memory index_cnt variable to reflect | ||
| 3315 | * free sequence numbers | ||
| 3316 | */ | ||
| 3317 | static int btrfs_set_inode_index_count(struct inode *inode) | ||
| 3318 | { | ||
| 3319 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 3320 | struct btrfs_key key, found_key; | ||
| 3321 | struct btrfs_path *path; | ||
| 3322 | struct extent_buffer *leaf; | ||
| 3323 | int ret; | ||
| 3324 | |||
| 3325 | key.objectid = inode->i_ino; | ||
| 3326 | btrfs_set_key_type(&key, BTRFS_DIR_INDEX_KEY); | ||
| 3327 | key.offset = (u64)-1; | ||
| 3328 | |||
| 3329 | path = btrfs_alloc_path(); | ||
| 3330 | if (!path) | ||
| 3331 | return -ENOMEM; | ||
| 3332 | |||
| 3333 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 3334 | if (ret < 0) | ||
| 3335 | goto out; | ||
| 3336 | /* FIXME: we should be able to handle this */ | ||
| 3337 | if (ret == 0) | ||
| 3338 | goto out; | ||
| 3339 | ret = 0; | ||
| 3340 | |||
| 3341 | /* | ||
| 3342 | * MAGIC NUMBER EXPLANATION: | ||
| 3343 | * since we search a directory based on f_pos we have to start at 2 | ||
| 3344 | * since '.' and '..' have f_pos of 0 and 1 respectively, so everybody | ||
| 3345 | * else has to start at 2 | ||
| 3346 | */ | ||
| 3347 | if (path->slots[0] == 0) { | ||
| 3348 | BTRFS_I(inode)->index_cnt = 2; | ||
| 3349 | goto out; | ||
| 3350 | } | ||
| 3351 | |||
| 3352 | path->slots[0]--; | ||
| 3353 | |||
| 3354 | leaf = path->nodes[0]; | ||
| 3355 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 3356 | |||
| 3357 | if (found_key.objectid != inode->i_ino || | ||
| 3358 | btrfs_key_type(&found_key) != BTRFS_DIR_INDEX_KEY) { | ||
| 3359 | BTRFS_I(inode)->index_cnt = 2; | ||
| 3360 | goto out; | ||
| 3361 | } | ||
| 3362 | |||
| 3363 | BTRFS_I(inode)->index_cnt = found_key.offset + 1; | ||
| 3364 | out: | ||
| 3365 | btrfs_free_path(path); | ||
| 3366 | return ret; | ||
| 3367 | } | ||
| 3368 | |||
| 3369 | /* | ||
| 3370 | * helper to find a free sequence number in a given directory. This current | ||
| 3371 | * code is very simple, later versions will do smarter things in the btree | ||
| 3372 | */ | ||
| 3373 | int btrfs_set_inode_index(struct inode *dir, u64 *index) | ||
| 3374 | { | ||
| 3375 | int ret = 0; | ||
| 3376 | |||
| 3377 | if (BTRFS_I(dir)->index_cnt == (u64)-1) { | ||
| 3378 | ret = btrfs_set_inode_index_count(dir); | ||
| 3379 | if (ret) | ||
| 3380 | return ret; | ||
| 3381 | } | ||
| 3382 | |||
| 3383 | *index = BTRFS_I(dir)->index_cnt; | ||
| 3384 | BTRFS_I(dir)->index_cnt++; | ||
| 3385 | |||
| 3386 | return ret; | ||
| 3387 | } | ||
| 3388 | |||
| 3389 | static struct inode *btrfs_new_inode(struct btrfs_trans_handle *trans, | ||
| 3390 | struct btrfs_root *root, | ||
| 3391 | struct inode *dir, | ||
| 3392 | const char *name, int name_len, | ||
| 3393 | u64 ref_objectid, u64 objectid, | ||
| 3394 | u64 alloc_hint, int mode, u64 *index) | ||
| 3395 | { | ||
| 3396 | struct inode *inode; | ||
| 3397 | struct btrfs_inode_item *inode_item; | ||
| 3398 | struct btrfs_key *location; | ||
| 3399 | struct btrfs_path *path; | ||
| 3400 | struct btrfs_inode_ref *ref; | ||
| 3401 | struct btrfs_key key[2]; | ||
| 3402 | u32 sizes[2]; | ||
| 3403 | unsigned long ptr; | ||
| 3404 | int ret; | ||
| 3405 | int owner; | ||
| 3406 | |||
| 3407 | path = btrfs_alloc_path(); | ||
| 3408 | BUG_ON(!path); | ||
| 3409 | |||
| 3410 | inode = new_inode(root->fs_info->sb); | ||
| 3411 | if (!inode) | ||
| 3412 | return ERR_PTR(-ENOMEM); | ||
| 3413 | |||
| 3414 | if (dir) { | ||
| 3415 | ret = btrfs_set_inode_index(dir, index); | ||
| 3416 | if (ret) | ||
| 3417 | return ERR_PTR(ret); | ||
| 3418 | } | ||
| 3419 | /* | ||
| 3420 | * index_cnt is ignored for everything but a dir, | ||
| 3421 | * btrfs_get_inode_index_count has an explanation for the magic | ||
| 3422 | * number | ||
| 3423 | */ | ||
| 3424 | init_btrfs_i(inode); | ||
| 3425 | BTRFS_I(inode)->index_cnt = 2; | ||
| 3426 | BTRFS_I(inode)->root = root; | ||
| 3427 | BTRFS_I(inode)->generation = trans->transid; | ||
| 3428 | |||
| 3429 | if (mode & S_IFDIR) | ||
| 3430 | owner = 0; | ||
| 3431 | else | ||
| 3432 | owner = 1; | ||
| 3433 | BTRFS_I(inode)->block_group = | ||
| 3434 | btrfs_find_block_group(root, 0, alloc_hint, owner); | ||
| 3435 | if ((mode & S_IFREG)) { | ||
| 3436 | if (btrfs_test_opt(root, NODATASUM)) | ||
| 3437 | btrfs_set_flag(inode, NODATASUM); | ||
| 3438 | if (btrfs_test_opt(root, NODATACOW)) | ||
| 3439 | btrfs_set_flag(inode, NODATACOW); | ||
| 3440 | } | ||
| 3441 | |||
| 3442 | key[0].objectid = objectid; | ||
| 3443 | btrfs_set_key_type(&key[0], BTRFS_INODE_ITEM_KEY); | ||
| 3444 | key[0].offset = 0; | ||
| 3445 | |||
| 3446 | key[1].objectid = objectid; | ||
| 3447 | btrfs_set_key_type(&key[1], BTRFS_INODE_REF_KEY); | ||
| 3448 | key[1].offset = ref_objectid; | ||
| 3449 | |||
| 3450 | sizes[0] = sizeof(struct btrfs_inode_item); | ||
| 3451 | sizes[1] = name_len + sizeof(*ref); | ||
| 3452 | |||
| 3453 | ret = btrfs_insert_empty_items(trans, root, path, key, sizes, 2); | ||
| 3454 | if (ret != 0) | ||
| 3455 | goto fail; | ||
| 3456 | |||
| 3457 | if (objectid > root->highest_inode) | ||
| 3458 | root->highest_inode = objectid; | ||
| 3459 | |||
| 3460 | inode->i_uid = current_fsuid(); | ||
| 3461 | inode->i_gid = current_fsgid(); | ||
| 3462 | inode->i_mode = mode; | ||
| 3463 | inode->i_ino = objectid; | ||
| 3464 | inode_set_bytes(inode, 0); | ||
| 3465 | inode->i_mtime = inode->i_atime = inode->i_ctime = CURRENT_TIME; | ||
| 3466 | inode_item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 3467 | struct btrfs_inode_item); | ||
| 3468 | fill_inode_item(trans, path->nodes[0], inode_item, inode); | ||
| 3469 | |||
| 3470 | ref = btrfs_item_ptr(path->nodes[0], path->slots[0] + 1, | ||
| 3471 | struct btrfs_inode_ref); | ||
| 3472 | btrfs_set_inode_ref_name_len(path->nodes[0], ref, name_len); | ||
| 3473 | btrfs_set_inode_ref_index(path->nodes[0], ref, *index); | ||
| 3474 | ptr = (unsigned long)(ref + 1); | ||
| 3475 | write_extent_buffer(path->nodes[0], name, ptr, name_len); | ||
| 3476 | |||
| 3477 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 3478 | btrfs_free_path(path); | ||
| 3479 | |||
| 3480 | location = &BTRFS_I(inode)->location; | ||
| 3481 | location->objectid = objectid; | ||
| 3482 | location->offset = 0; | ||
| 3483 | btrfs_set_key_type(location, BTRFS_INODE_ITEM_KEY); | ||
| 3484 | |||
| 3485 | insert_inode_hash(inode); | ||
| 3486 | return inode; | ||
| 3487 | fail: | ||
| 3488 | if (dir) | ||
| 3489 | BTRFS_I(dir)->index_cnt--; | ||
| 3490 | btrfs_free_path(path); | ||
| 3491 | return ERR_PTR(ret); | ||
| 3492 | } | ||
| 3493 | |||
| 3494 | static inline u8 btrfs_inode_type(struct inode *inode) | ||
| 3495 | { | ||
| 3496 | return btrfs_type_by_mode[(inode->i_mode & S_IFMT) >> S_SHIFT]; | ||
| 3497 | } | ||
| 3498 | |||
| 3499 | /* | ||
| 3500 | * utility function to add 'inode' into 'parent_inode' with | ||
| 3501 | * a give name and a given sequence number. | ||
| 3502 | * if 'add_backref' is true, also insert a backref from the | ||
| 3503 | * inode to the parent directory. | ||
| 3504 | */ | ||
| 3505 | int btrfs_add_link(struct btrfs_trans_handle *trans, | ||
| 3506 | struct inode *parent_inode, struct inode *inode, | ||
| 3507 | const char *name, int name_len, int add_backref, u64 index) | ||
| 3508 | { | ||
| 3509 | int ret; | ||
| 3510 | struct btrfs_key key; | ||
| 3511 | struct btrfs_root *root = BTRFS_I(parent_inode)->root; | ||
| 3512 | |||
| 3513 | key.objectid = inode->i_ino; | ||
| 3514 | btrfs_set_key_type(&key, BTRFS_INODE_ITEM_KEY); | ||
| 3515 | key.offset = 0; | ||
| 3516 | |||
| 3517 | ret = btrfs_insert_dir_item(trans, root, name, name_len, | ||
| 3518 | parent_inode->i_ino, | ||
| 3519 | &key, btrfs_inode_type(inode), | ||
| 3520 | index); | ||
| 3521 | if (ret == 0) { | ||
| 3522 | if (add_backref) { | ||
| 3523 | ret = btrfs_insert_inode_ref(trans, root, | ||
| 3524 | name, name_len, | ||
| 3525 | inode->i_ino, | ||
| 3526 | parent_inode->i_ino, | ||
| 3527 | index); | ||
| 3528 | } | ||
| 3529 | btrfs_i_size_write(parent_inode, parent_inode->i_size + | ||
| 3530 | name_len * 2); | ||
| 3531 | parent_inode->i_mtime = parent_inode->i_ctime = CURRENT_TIME; | ||
| 3532 | ret = btrfs_update_inode(trans, root, parent_inode); | ||
| 3533 | } | ||
| 3534 | return ret; | ||
| 3535 | } | ||
| 3536 | |||
| 3537 | static int btrfs_add_nondir(struct btrfs_trans_handle *trans, | ||
| 3538 | struct dentry *dentry, struct inode *inode, | ||
| 3539 | int backref, u64 index) | ||
| 3540 | { | ||
| 3541 | int err = btrfs_add_link(trans, dentry->d_parent->d_inode, | ||
| 3542 | inode, dentry->d_name.name, | ||
| 3543 | dentry->d_name.len, backref, index); | ||
| 3544 | if (!err) { | ||
| 3545 | d_instantiate(dentry, inode); | ||
| 3546 | return 0; | ||
| 3547 | } | ||
| 3548 | if (err > 0) | ||
| 3549 | err = -EEXIST; | ||
| 3550 | return err; | ||
| 3551 | } | ||
| 3552 | |||
| 3553 | static int btrfs_mknod(struct inode *dir, struct dentry *dentry, | ||
| 3554 | int mode, dev_t rdev) | ||
| 3555 | { | ||
| 3556 | struct btrfs_trans_handle *trans; | ||
| 3557 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 3558 | struct inode *inode = NULL; | ||
| 3559 | int err; | ||
| 3560 | int drop_inode = 0; | ||
| 3561 | u64 objectid; | ||
| 3562 | unsigned long nr = 0; | ||
| 3563 | u64 index = 0; | ||
| 3564 | |||
| 3565 | if (!new_valid_dev(rdev)) | ||
| 3566 | return -EINVAL; | ||
| 3567 | |||
| 3568 | err = btrfs_check_free_space(root, 1, 0); | ||
| 3569 | if (err) | ||
| 3570 | goto fail; | ||
| 3571 | |||
| 3572 | trans = btrfs_start_transaction(root, 1); | ||
| 3573 | btrfs_set_trans_block_group(trans, dir); | ||
| 3574 | |||
| 3575 | err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid); | ||
| 3576 | if (err) { | ||
| 3577 | err = -ENOSPC; | ||
| 3578 | goto out_unlock; | ||
| 3579 | } | ||
| 3580 | |||
| 3581 | inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name, | ||
| 3582 | dentry->d_name.len, | ||
| 3583 | dentry->d_parent->d_inode->i_ino, objectid, | ||
| 3584 | BTRFS_I(dir)->block_group, mode, &index); | ||
| 3585 | err = PTR_ERR(inode); | ||
| 3586 | if (IS_ERR(inode)) | ||
| 3587 | goto out_unlock; | ||
| 3588 | |||
| 3589 | err = btrfs_init_acl(inode, dir); | ||
| 3590 | if (err) { | ||
| 3591 | drop_inode = 1; | ||
| 3592 | goto out_unlock; | ||
| 3593 | } | ||
| 3594 | |||
| 3595 | btrfs_set_trans_block_group(trans, inode); | ||
| 3596 | err = btrfs_add_nondir(trans, dentry, inode, 0, index); | ||
| 3597 | if (err) | ||
| 3598 | drop_inode = 1; | ||
| 3599 | else { | ||
| 3600 | inode->i_op = &btrfs_special_inode_operations; | ||
| 3601 | init_special_inode(inode, inode->i_mode, rdev); | ||
| 3602 | btrfs_update_inode(trans, root, inode); | ||
| 3603 | } | ||
| 3604 | dir->i_sb->s_dirt = 1; | ||
| 3605 | btrfs_update_inode_block_group(trans, inode); | ||
| 3606 | btrfs_update_inode_block_group(trans, dir); | ||
| 3607 | out_unlock: | ||
| 3608 | nr = trans->blocks_used; | ||
| 3609 | btrfs_end_transaction_throttle(trans, root); | ||
| 3610 | fail: | ||
| 3611 | if (drop_inode) { | ||
| 3612 | inode_dec_link_count(inode); | ||
| 3613 | iput(inode); | ||
| 3614 | } | ||
| 3615 | btrfs_btree_balance_dirty(root, nr); | ||
| 3616 | return err; | ||
| 3617 | } | ||
| 3618 | |||
| 3619 | static int btrfs_create(struct inode *dir, struct dentry *dentry, | ||
| 3620 | int mode, struct nameidata *nd) | ||
| 3621 | { | ||
| 3622 | struct btrfs_trans_handle *trans; | ||
| 3623 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 3624 | struct inode *inode = NULL; | ||
| 3625 | int err; | ||
| 3626 | int drop_inode = 0; | ||
| 3627 | unsigned long nr = 0; | ||
| 3628 | u64 objectid; | ||
| 3629 | u64 index = 0; | ||
| 3630 | |||
| 3631 | err = btrfs_check_free_space(root, 1, 0); | ||
| 3632 | if (err) | ||
| 3633 | goto fail; | ||
| 3634 | trans = btrfs_start_transaction(root, 1); | ||
| 3635 | btrfs_set_trans_block_group(trans, dir); | ||
| 3636 | |||
| 3637 | err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid); | ||
| 3638 | if (err) { | ||
| 3639 | err = -ENOSPC; | ||
| 3640 | goto out_unlock; | ||
| 3641 | } | ||
| 3642 | |||
| 3643 | inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name, | ||
| 3644 | dentry->d_name.len, | ||
| 3645 | dentry->d_parent->d_inode->i_ino, | ||
| 3646 | objectid, BTRFS_I(dir)->block_group, mode, | ||
| 3647 | &index); | ||
| 3648 | err = PTR_ERR(inode); | ||
| 3649 | if (IS_ERR(inode)) | ||
| 3650 | goto out_unlock; | ||
| 3651 | |||
| 3652 | err = btrfs_init_acl(inode, dir); | ||
| 3653 | if (err) { | ||
| 3654 | drop_inode = 1; | ||
| 3655 | goto out_unlock; | ||
| 3656 | } | ||
| 3657 | |||
| 3658 | btrfs_set_trans_block_group(trans, inode); | ||
| 3659 | err = btrfs_add_nondir(trans, dentry, inode, 0, index); | ||
| 3660 | if (err) | ||
| 3661 | drop_inode = 1; | ||
| 3662 | else { | ||
| 3663 | inode->i_mapping->a_ops = &btrfs_aops; | ||
| 3664 | inode->i_mapping->backing_dev_info = &root->fs_info->bdi; | ||
| 3665 | inode->i_fop = &btrfs_file_operations; | ||
| 3666 | inode->i_op = &btrfs_file_inode_operations; | ||
| 3667 | BTRFS_I(inode)->io_tree.ops = &btrfs_extent_io_ops; | ||
| 3668 | } | ||
| 3669 | dir->i_sb->s_dirt = 1; | ||
| 3670 | btrfs_update_inode_block_group(trans, inode); | ||
| 3671 | btrfs_update_inode_block_group(trans, dir); | ||
| 3672 | out_unlock: | ||
| 3673 | nr = trans->blocks_used; | ||
| 3674 | btrfs_end_transaction_throttle(trans, root); | ||
| 3675 | fail: | ||
| 3676 | if (drop_inode) { | ||
| 3677 | inode_dec_link_count(inode); | ||
| 3678 | iput(inode); | ||
| 3679 | } | ||
| 3680 | btrfs_btree_balance_dirty(root, nr); | ||
| 3681 | return err; | ||
| 3682 | } | ||
| 3683 | |||
| 3684 | static int btrfs_link(struct dentry *old_dentry, struct inode *dir, | ||
| 3685 | struct dentry *dentry) | ||
| 3686 | { | ||
| 3687 | struct btrfs_trans_handle *trans; | ||
| 3688 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 3689 | struct inode *inode = old_dentry->d_inode; | ||
| 3690 | u64 index; | ||
| 3691 | unsigned long nr = 0; | ||
| 3692 | int err; | ||
| 3693 | int drop_inode = 0; | ||
| 3694 | |||
| 3695 | if (inode->i_nlink == 0) | ||
| 3696 | return -ENOENT; | ||
| 3697 | |||
| 3698 | btrfs_inc_nlink(inode); | ||
| 3699 | err = btrfs_check_free_space(root, 1, 0); | ||
| 3700 | if (err) | ||
| 3701 | goto fail; | ||
| 3702 | err = btrfs_set_inode_index(dir, &index); | ||
| 3703 | if (err) | ||
| 3704 | goto fail; | ||
| 3705 | |||
| 3706 | trans = btrfs_start_transaction(root, 1); | ||
| 3707 | |||
| 3708 | btrfs_set_trans_block_group(trans, dir); | ||
| 3709 | atomic_inc(&inode->i_count); | ||
| 3710 | |||
| 3711 | err = btrfs_add_nondir(trans, dentry, inode, 1, index); | ||
| 3712 | |||
| 3713 | if (err) | ||
| 3714 | drop_inode = 1; | ||
| 3715 | |||
| 3716 | dir->i_sb->s_dirt = 1; | ||
| 3717 | btrfs_update_inode_block_group(trans, dir); | ||
| 3718 | err = btrfs_update_inode(trans, root, inode); | ||
| 3719 | |||
| 3720 | if (err) | ||
| 3721 | drop_inode = 1; | ||
| 3722 | |||
| 3723 | nr = trans->blocks_used; | ||
| 3724 | btrfs_end_transaction_throttle(trans, root); | ||
| 3725 | fail: | ||
| 3726 | if (drop_inode) { | ||
| 3727 | inode_dec_link_count(inode); | ||
| 3728 | iput(inode); | ||
| 3729 | } | ||
| 3730 | btrfs_btree_balance_dirty(root, nr); | ||
| 3731 | return err; | ||
| 3732 | } | ||
| 3733 | |||
| 3734 | static int btrfs_mkdir(struct inode *dir, struct dentry *dentry, int mode) | ||
| 3735 | { | ||
| 3736 | struct inode *inode = NULL; | ||
| 3737 | struct btrfs_trans_handle *trans; | ||
| 3738 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 3739 | int err = 0; | ||
| 3740 | int drop_on_err = 0; | ||
| 3741 | u64 objectid = 0; | ||
| 3742 | u64 index = 0; | ||
| 3743 | unsigned long nr = 1; | ||
| 3744 | |||
| 3745 | err = btrfs_check_free_space(root, 1, 0); | ||
| 3746 | if (err) | ||
| 3747 | goto out_unlock; | ||
| 3748 | |||
| 3749 | trans = btrfs_start_transaction(root, 1); | ||
| 3750 | btrfs_set_trans_block_group(trans, dir); | ||
| 3751 | |||
| 3752 | if (IS_ERR(trans)) { | ||
| 3753 | err = PTR_ERR(trans); | ||
| 3754 | goto out_unlock; | ||
| 3755 | } | ||
| 3756 | |||
| 3757 | err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid); | ||
| 3758 | if (err) { | ||
| 3759 | err = -ENOSPC; | ||
| 3760 | goto out_unlock; | ||
| 3761 | } | ||
| 3762 | |||
| 3763 | inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name, | ||
| 3764 | dentry->d_name.len, | ||
| 3765 | dentry->d_parent->d_inode->i_ino, objectid, | ||
| 3766 | BTRFS_I(dir)->block_group, S_IFDIR | mode, | ||
| 3767 | &index); | ||
| 3768 | if (IS_ERR(inode)) { | ||
| 3769 | err = PTR_ERR(inode); | ||
| 3770 | goto out_fail; | ||
| 3771 | } | ||
| 3772 | |||
| 3773 | drop_on_err = 1; | ||
| 3774 | |||
| 3775 | err = btrfs_init_acl(inode, dir); | ||
| 3776 | if (err) | ||
| 3777 | goto out_fail; | ||
| 3778 | |||
| 3779 | inode->i_op = &btrfs_dir_inode_operations; | ||
| 3780 | inode->i_fop = &btrfs_dir_file_operations; | ||
| 3781 | btrfs_set_trans_block_group(trans, inode); | ||
| 3782 | |||
| 3783 | btrfs_i_size_write(inode, 0); | ||
| 3784 | err = btrfs_update_inode(trans, root, inode); | ||
| 3785 | if (err) | ||
| 3786 | goto out_fail; | ||
| 3787 | |||
| 3788 | err = btrfs_add_link(trans, dentry->d_parent->d_inode, | ||
| 3789 | inode, dentry->d_name.name, | ||
| 3790 | dentry->d_name.len, 0, index); | ||
| 3791 | if (err) | ||
| 3792 | goto out_fail; | ||
| 3793 | |||
| 3794 | d_instantiate(dentry, inode); | ||
| 3795 | drop_on_err = 0; | ||
| 3796 | dir->i_sb->s_dirt = 1; | ||
| 3797 | btrfs_update_inode_block_group(trans, inode); | ||
| 3798 | btrfs_update_inode_block_group(trans, dir); | ||
| 3799 | |||
| 3800 | out_fail: | ||
| 3801 | nr = trans->blocks_used; | ||
| 3802 | btrfs_end_transaction_throttle(trans, root); | ||
| 3803 | |||
| 3804 | out_unlock: | ||
| 3805 | if (drop_on_err) | ||
| 3806 | iput(inode); | ||
| 3807 | btrfs_btree_balance_dirty(root, nr); | ||
| 3808 | return err; | ||
| 3809 | } | ||
| 3810 | |||
| 3811 | /* helper for btfs_get_extent. Given an existing extent in the tree, | ||
| 3812 | * and an extent that you want to insert, deal with overlap and insert | ||
| 3813 | * the new extent into the tree. | ||
| 3814 | */ | ||
| 3815 | static int merge_extent_mapping(struct extent_map_tree *em_tree, | ||
| 3816 | struct extent_map *existing, | ||
| 3817 | struct extent_map *em, | ||
| 3818 | u64 map_start, u64 map_len) | ||
| 3819 | { | ||
| 3820 | u64 start_diff; | ||
| 3821 | |||
| 3822 | BUG_ON(map_start < em->start || map_start >= extent_map_end(em)); | ||
| 3823 | start_diff = map_start - em->start; | ||
| 3824 | em->start = map_start; | ||
| 3825 | em->len = map_len; | ||
| 3826 | if (em->block_start < EXTENT_MAP_LAST_BYTE && | ||
| 3827 | !test_bit(EXTENT_FLAG_COMPRESSED, &em->flags)) { | ||
| 3828 | em->block_start += start_diff; | ||
| 3829 | em->block_len -= start_diff; | ||
| 3830 | } | ||
| 3831 | return add_extent_mapping(em_tree, em); | ||
| 3832 | } | ||
| 3833 | |||
| 3834 | static noinline int uncompress_inline(struct btrfs_path *path, | ||
| 3835 | struct inode *inode, struct page *page, | ||
| 3836 | size_t pg_offset, u64 extent_offset, | ||
| 3837 | struct btrfs_file_extent_item *item) | ||
| 3838 | { | ||
| 3839 | int ret; | ||
| 3840 | struct extent_buffer *leaf = path->nodes[0]; | ||
| 3841 | char *tmp; | ||
| 3842 | size_t max_size; | ||
| 3843 | unsigned long inline_size; | ||
| 3844 | unsigned long ptr; | ||
| 3845 | |||
| 3846 | WARN_ON(pg_offset != 0); | ||
| 3847 | max_size = btrfs_file_extent_ram_bytes(leaf, item); | ||
| 3848 | inline_size = btrfs_file_extent_inline_item_len(leaf, | ||
| 3849 | btrfs_item_nr(leaf, path->slots[0])); | ||
| 3850 | tmp = kmalloc(inline_size, GFP_NOFS); | ||
| 3851 | ptr = btrfs_file_extent_inline_start(item); | ||
| 3852 | |||
| 3853 | read_extent_buffer(leaf, tmp, ptr, inline_size); | ||
| 3854 | |||
| 3855 | max_size = min_t(unsigned long, PAGE_CACHE_SIZE, max_size); | ||
| 3856 | ret = btrfs_zlib_decompress(tmp, page, extent_offset, | ||
| 3857 | inline_size, max_size); | ||
| 3858 | if (ret) { | ||
| 3859 | char *kaddr = kmap_atomic(page, KM_USER0); | ||
| 3860 | unsigned long copy_size = min_t(u64, | ||
| 3861 | PAGE_CACHE_SIZE - pg_offset, | ||
| 3862 | max_size - extent_offset); | ||
| 3863 | memset(kaddr + pg_offset, 0, copy_size); | ||
| 3864 | kunmap_atomic(kaddr, KM_USER0); | ||
| 3865 | } | ||
| 3866 | kfree(tmp); | ||
| 3867 | return 0; | ||
| 3868 | } | ||
| 3869 | |||
| 3870 | /* | ||
| 3871 | * a bit scary, this does extent mapping from logical file offset to the disk. | ||
| 3872 | * the ugly parts come from merging extents from the disk with the in-ram | ||
| 3873 | * representation. This gets more complex because of the data=ordered code, | ||
| 3874 | * where the in-ram extents might be locked pending data=ordered completion. | ||
| 3875 | * | ||
| 3876 | * This also copies inline extents directly into the page. | ||
| 3877 | */ | ||
| 3878 | |||
| 3879 | struct extent_map *btrfs_get_extent(struct inode *inode, struct page *page, | ||
| 3880 | size_t pg_offset, u64 start, u64 len, | ||
| 3881 | int create) | ||
| 3882 | { | ||
| 3883 | int ret; | ||
| 3884 | int err = 0; | ||
| 3885 | u64 bytenr; | ||
| 3886 | u64 extent_start = 0; | ||
| 3887 | u64 extent_end = 0; | ||
| 3888 | u64 objectid = inode->i_ino; | ||
| 3889 | u32 found_type; | ||
| 3890 | struct btrfs_path *path = NULL; | ||
| 3891 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 3892 | struct btrfs_file_extent_item *item; | ||
| 3893 | struct extent_buffer *leaf; | ||
| 3894 | struct btrfs_key found_key; | ||
| 3895 | struct extent_map *em = NULL; | ||
| 3896 | struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree; | ||
| 3897 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 3898 | struct btrfs_trans_handle *trans = NULL; | ||
| 3899 | int compressed; | ||
| 3900 | |||
| 3901 | again: | ||
| 3902 | spin_lock(&em_tree->lock); | ||
| 3903 | em = lookup_extent_mapping(em_tree, start, len); | ||
| 3904 | if (em) | ||
| 3905 | em->bdev = root->fs_info->fs_devices->latest_bdev; | ||
| 3906 | spin_unlock(&em_tree->lock); | ||
| 3907 | |||
| 3908 | if (em) { | ||
| 3909 | if (em->start > start || em->start + em->len <= start) | ||
| 3910 | free_extent_map(em); | ||
| 3911 | else if (em->block_start == EXTENT_MAP_INLINE && page) | ||
| 3912 | free_extent_map(em); | ||
| 3913 | else | ||
| 3914 | goto out; | ||
| 3915 | } | ||
| 3916 | em = alloc_extent_map(GFP_NOFS); | ||
| 3917 | if (!em) { | ||
| 3918 | err = -ENOMEM; | ||
| 3919 | goto out; | ||
| 3920 | } | ||
| 3921 | em->bdev = root->fs_info->fs_devices->latest_bdev; | ||
| 3922 | em->start = EXTENT_MAP_HOLE; | ||
| 3923 | em->orig_start = EXTENT_MAP_HOLE; | ||
| 3924 | em->len = (u64)-1; | ||
| 3925 | em->block_len = (u64)-1; | ||
| 3926 | |||
| 3927 | if (!path) { | ||
| 3928 | path = btrfs_alloc_path(); | ||
| 3929 | BUG_ON(!path); | ||
| 3930 | } | ||
| 3931 | |||
| 3932 | ret = btrfs_lookup_file_extent(trans, root, path, | ||
| 3933 | objectid, start, trans != NULL); | ||
| 3934 | if (ret < 0) { | ||
| 3935 | err = ret; | ||
| 3936 | goto out; | ||
| 3937 | } | ||
| 3938 | |||
| 3939 | if (ret != 0) { | ||
| 3940 | if (path->slots[0] == 0) | ||
| 3941 | goto not_found; | ||
| 3942 | path->slots[0]--; | ||
| 3943 | } | ||
| 3944 | |||
| 3945 | leaf = path->nodes[0]; | ||
| 3946 | item = btrfs_item_ptr(leaf, path->slots[0], | ||
| 3947 | struct btrfs_file_extent_item); | ||
| 3948 | /* are we inside the extent that was found? */ | ||
| 3949 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 3950 | found_type = btrfs_key_type(&found_key); | ||
| 3951 | if (found_key.objectid != objectid || | ||
| 3952 | found_type != BTRFS_EXTENT_DATA_KEY) { | ||
| 3953 | goto not_found; | ||
| 3954 | } | ||
| 3955 | |||
| 3956 | found_type = btrfs_file_extent_type(leaf, item); | ||
| 3957 | extent_start = found_key.offset; | ||
| 3958 | compressed = btrfs_file_extent_compression(leaf, item); | ||
| 3959 | if (found_type == BTRFS_FILE_EXTENT_REG || | ||
| 3960 | found_type == BTRFS_FILE_EXTENT_PREALLOC) { | ||
| 3961 | extent_end = extent_start + | ||
| 3962 | btrfs_file_extent_num_bytes(leaf, item); | ||
| 3963 | } else if (found_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 3964 | size_t size; | ||
| 3965 | size = btrfs_file_extent_inline_len(leaf, item); | ||
| 3966 | extent_end = (extent_start + size + root->sectorsize - 1) & | ||
| 3967 | ~((u64)root->sectorsize - 1); | ||
| 3968 | } | ||
| 3969 | |||
| 3970 | if (start >= extent_end) { | ||
| 3971 | path->slots[0]++; | ||
| 3972 | if (path->slots[0] >= btrfs_header_nritems(leaf)) { | ||
| 3973 | ret = btrfs_next_leaf(root, path); | ||
| 3974 | if (ret < 0) { | ||
| 3975 | err = ret; | ||
| 3976 | goto out; | ||
| 3977 | } | ||
| 3978 | if (ret > 0) | ||
| 3979 | goto not_found; | ||
| 3980 | leaf = path->nodes[0]; | ||
| 3981 | } | ||
| 3982 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 3983 | if (found_key.objectid != objectid || | ||
| 3984 | found_key.type != BTRFS_EXTENT_DATA_KEY) | ||
| 3985 | goto not_found; | ||
| 3986 | if (start + len <= found_key.offset) | ||
| 3987 | goto not_found; | ||
| 3988 | em->start = start; | ||
| 3989 | em->len = found_key.offset - start; | ||
| 3990 | goto not_found_em; | ||
| 3991 | } | ||
| 3992 | |||
| 3993 | if (found_type == BTRFS_FILE_EXTENT_REG || | ||
| 3994 | found_type == BTRFS_FILE_EXTENT_PREALLOC) { | ||
| 3995 | em->start = extent_start; | ||
| 3996 | em->len = extent_end - extent_start; | ||
| 3997 | em->orig_start = extent_start - | ||
| 3998 | btrfs_file_extent_offset(leaf, item); | ||
| 3999 | bytenr = btrfs_file_extent_disk_bytenr(leaf, item); | ||
| 4000 | if (bytenr == 0) { | ||
| 4001 | em->block_start = EXTENT_MAP_HOLE; | ||
| 4002 | goto insert; | ||
| 4003 | } | ||
| 4004 | if (compressed) { | ||
| 4005 | set_bit(EXTENT_FLAG_COMPRESSED, &em->flags); | ||
| 4006 | em->block_start = bytenr; | ||
| 4007 | em->block_len = btrfs_file_extent_disk_num_bytes(leaf, | ||
| 4008 | item); | ||
| 4009 | } else { | ||
| 4010 | bytenr += btrfs_file_extent_offset(leaf, item); | ||
| 4011 | em->block_start = bytenr; | ||
| 4012 | em->block_len = em->len; | ||
| 4013 | if (found_type == BTRFS_FILE_EXTENT_PREALLOC) | ||
| 4014 | set_bit(EXTENT_FLAG_PREALLOC, &em->flags); | ||
| 4015 | } | ||
| 4016 | goto insert; | ||
| 4017 | } else if (found_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 4018 | unsigned long ptr; | ||
| 4019 | char *map; | ||
| 4020 | size_t size; | ||
| 4021 | size_t extent_offset; | ||
| 4022 | size_t copy_size; | ||
| 4023 | |||
| 4024 | em->block_start = EXTENT_MAP_INLINE; | ||
| 4025 | if (!page || create) { | ||
| 4026 | em->start = extent_start; | ||
| 4027 | em->len = extent_end - extent_start; | ||
| 4028 | goto out; | ||
| 4029 | } | ||
| 4030 | |||
| 4031 | size = btrfs_file_extent_inline_len(leaf, item); | ||
| 4032 | extent_offset = page_offset(page) + pg_offset - extent_start; | ||
| 4033 | copy_size = min_t(u64, PAGE_CACHE_SIZE - pg_offset, | ||
| 4034 | size - extent_offset); | ||
| 4035 | em->start = extent_start + extent_offset; | ||
| 4036 | em->len = (copy_size + root->sectorsize - 1) & | ||
| 4037 | ~((u64)root->sectorsize - 1); | ||
| 4038 | em->orig_start = EXTENT_MAP_INLINE; | ||
| 4039 | if (compressed) | ||
| 4040 | set_bit(EXTENT_FLAG_COMPRESSED, &em->flags); | ||
| 4041 | ptr = btrfs_file_extent_inline_start(item) + extent_offset; | ||
| 4042 | if (create == 0 && !PageUptodate(page)) { | ||
| 4043 | if (btrfs_file_extent_compression(leaf, item) == | ||
| 4044 | BTRFS_COMPRESS_ZLIB) { | ||
| 4045 | ret = uncompress_inline(path, inode, page, | ||
| 4046 | pg_offset, | ||
| 4047 | extent_offset, item); | ||
| 4048 | BUG_ON(ret); | ||
| 4049 | } else { | ||
| 4050 | map = kmap(page); | ||
| 4051 | read_extent_buffer(leaf, map + pg_offset, ptr, | ||
| 4052 | copy_size); | ||
| 4053 | kunmap(page); | ||
| 4054 | } | ||
| 4055 | flush_dcache_page(page); | ||
| 4056 | } else if (create && PageUptodate(page)) { | ||
| 4057 | if (!trans) { | ||
| 4058 | kunmap(page); | ||
| 4059 | free_extent_map(em); | ||
| 4060 | em = NULL; | ||
| 4061 | btrfs_release_path(root, path); | ||
| 4062 | trans = btrfs_join_transaction(root, 1); | ||
| 4063 | goto again; | ||
| 4064 | } | ||
| 4065 | map = kmap(page); | ||
| 4066 | write_extent_buffer(leaf, map + pg_offset, ptr, | ||
| 4067 | copy_size); | ||
| 4068 | kunmap(page); | ||
| 4069 | btrfs_mark_buffer_dirty(leaf); | ||
| 4070 | } | ||
| 4071 | set_extent_uptodate(io_tree, em->start, | ||
| 4072 | extent_map_end(em) - 1, GFP_NOFS); | ||
| 4073 | goto insert; | ||
| 4074 | } else { | ||
| 4075 | printk(KERN_ERR "btrfs unknown found_type %d\n", found_type); | ||
| 4076 | WARN_ON(1); | ||
| 4077 | } | ||
| 4078 | not_found: | ||
| 4079 | em->start = start; | ||
| 4080 | em->len = len; | ||
| 4081 | not_found_em: | ||
| 4082 | em->block_start = EXTENT_MAP_HOLE; | ||
| 4083 | set_bit(EXTENT_FLAG_VACANCY, &em->flags); | ||
| 4084 | insert: | ||
| 4085 | btrfs_release_path(root, path); | ||
| 4086 | if (em->start > start || extent_map_end(em) <= start) { | ||
| 4087 | printk(KERN_ERR "Btrfs: bad extent! em: [%llu %llu] passed " | ||
| 4088 | "[%llu %llu]\n", (unsigned long long)em->start, | ||
| 4089 | (unsigned long long)em->len, | ||
| 4090 | (unsigned long long)start, | ||
| 4091 | (unsigned long long)len); | ||
| 4092 | err = -EIO; | ||
| 4093 | goto out; | ||
| 4094 | } | ||
| 4095 | |||
| 4096 | err = 0; | ||
| 4097 | spin_lock(&em_tree->lock); | ||
| 4098 | ret = add_extent_mapping(em_tree, em); | ||
| 4099 | /* it is possible that someone inserted the extent into the tree | ||
| 4100 | * while we had the lock dropped. It is also possible that | ||
| 4101 | * an overlapping map exists in the tree | ||
| 4102 | */ | ||
| 4103 | if (ret == -EEXIST) { | ||
| 4104 | struct extent_map *existing; | ||
| 4105 | |||
| 4106 | ret = 0; | ||
| 4107 | |||
| 4108 | existing = lookup_extent_mapping(em_tree, start, len); | ||
| 4109 | if (existing && (existing->start > start || | ||
| 4110 | existing->start + existing->len <= start)) { | ||
| 4111 | free_extent_map(existing); | ||
| 4112 | existing = NULL; | ||
| 4113 | } | ||
| 4114 | if (!existing) { | ||
| 4115 | existing = lookup_extent_mapping(em_tree, em->start, | ||
| 4116 | em->len); | ||
| 4117 | if (existing) { | ||
| 4118 | err = merge_extent_mapping(em_tree, existing, | ||
| 4119 | em, start, | ||
| 4120 | root->sectorsize); | ||
| 4121 | free_extent_map(existing); | ||
| 4122 | if (err) { | ||
| 4123 | free_extent_map(em); | ||
| 4124 | em = NULL; | ||
| 4125 | } | ||
| 4126 | } else { | ||
| 4127 | err = -EIO; | ||
| 4128 | free_extent_map(em); | ||
| 4129 | em = NULL; | ||
| 4130 | } | ||
| 4131 | } else { | ||
| 4132 | free_extent_map(em); | ||
| 4133 | em = existing; | ||
| 4134 | err = 0; | ||
| 4135 | } | ||
| 4136 | } | ||
| 4137 | spin_unlock(&em_tree->lock); | ||
| 4138 | out: | ||
| 4139 | if (path) | ||
| 4140 | btrfs_free_path(path); | ||
| 4141 | if (trans) { | ||
| 4142 | ret = btrfs_end_transaction(trans, root); | ||
| 4143 | if (!err) | ||
| 4144 | err = ret; | ||
| 4145 | } | ||
| 4146 | if (err) { | ||
| 4147 | free_extent_map(em); | ||
| 4148 | WARN_ON(1); | ||
| 4149 | return ERR_PTR(err); | ||
| 4150 | } | ||
| 4151 | return em; | ||
| 4152 | } | ||
| 4153 | |||
| 4154 | static ssize_t btrfs_direct_IO(int rw, struct kiocb *iocb, | ||
| 4155 | const struct iovec *iov, loff_t offset, | ||
| 4156 | unsigned long nr_segs) | ||
| 4157 | { | ||
| 4158 | return -EINVAL; | ||
| 4159 | } | ||
| 4160 | |||
| 4161 | static sector_t btrfs_bmap(struct address_space *mapping, sector_t iblock) | ||
| 4162 | { | ||
| 4163 | return extent_bmap(mapping, iblock, btrfs_get_extent); | ||
| 4164 | } | ||
| 4165 | |||
| 4166 | int btrfs_readpage(struct file *file, struct page *page) | ||
| 4167 | { | ||
| 4168 | struct extent_io_tree *tree; | ||
| 4169 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 4170 | return extent_read_full_page(tree, page, btrfs_get_extent); | ||
| 4171 | } | ||
| 4172 | |||
| 4173 | static int btrfs_writepage(struct page *page, struct writeback_control *wbc) | ||
| 4174 | { | ||
| 4175 | struct extent_io_tree *tree; | ||
| 4176 | |||
| 4177 | |||
| 4178 | if (current->flags & PF_MEMALLOC) { | ||
| 4179 | redirty_page_for_writepage(wbc, page); | ||
| 4180 | unlock_page(page); | ||
| 4181 | return 0; | ||
| 4182 | } | ||
| 4183 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 4184 | return extent_write_full_page(tree, page, btrfs_get_extent, wbc); | ||
| 4185 | } | ||
| 4186 | |||
| 4187 | int btrfs_writepages(struct address_space *mapping, | ||
| 4188 | struct writeback_control *wbc) | ||
| 4189 | { | ||
| 4190 | struct extent_io_tree *tree; | ||
| 4191 | |||
| 4192 | tree = &BTRFS_I(mapping->host)->io_tree; | ||
| 4193 | return extent_writepages(tree, mapping, btrfs_get_extent, wbc); | ||
| 4194 | } | ||
| 4195 | |||
| 4196 | static int | ||
| 4197 | btrfs_readpages(struct file *file, struct address_space *mapping, | ||
| 4198 | struct list_head *pages, unsigned nr_pages) | ||
| 4199 | { | ||
| 4200 | struct extent_io_tree *tree; | ||
| 4201 | tree = &BTRFS_I(mapping->host)->io_tree; | ||
| 4202 | return extent_readpages(tree, mapping, pages, nr_pages, | ||
| 4203 | btrfs_get_extent); | ||
| 4204 | } | ||
| 4205 | static int __btrfs_releasepage(struct page *page, gfp_t gfp_flags) | ||
| 4206 | { | ||
| 4207 | struct extent_io_tree *tree; | ||
| 4208 | struct extent_map_tree *map; | ||
| 4209 | int ret; | ||
| 4210 | |||
| 4211 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 4212 | map = &BTRFS_I(page->mapping->host)->extent_tree; | ||
| 4213 | ret = try_release_extent_mapping(map, tree, page, gfp_flags); | ||
| 4214 | if (ret == 1) { | ||
| 4215 | ClearPagePrivate(page); | ||
| 4216 | set_page_private(page, 0); | ||
| 4217 | page_cache_release(page); | ||
| 4218 | } | ||
| 4219 | return ret; | ||
| 4220 | } | ||
| 4221 | |||
| 4222 | static int btrfs_releasepage(struct page *page, gfp_t gfp_flags) | ||
| 4223 | { | ||
| 4224 | if (PageWriteback(page) || PageDirty(page)) | ||
| 4225 | return 0; | ||
| 4226 | return __btrfs_releasepage(page, gfp_flags); | ||
| 4227 | } | ||
| 4228 | |||
| 4229 | static void btrfs_invalidatepage(struct page *page, unsigned long offset) | ||
| 4230 | { | ||
| 4231 | struct extent_io_tree *tree; | ||
| 4232 | struct btrfs_ordered_extent *ordered; | ||
| 4233 | u64 page_start = page_offset(page); | ||
| 4234 | u64 page_end = page_start + PAGE_CACHE_SIZE - 1; | ||
| 4235 | |||
| 4236 | wait_on_page_writeback(page); | ||
| 4237 | tree = &BTRFS_I(page->mapping->host)->io_tree; | ||
| 4238 | if (offset) { | ||
| 4239 | btrfs_releasepage(page, GFP_NOFS); | ||
| 4240 | return; | ||
| 4241 | } | ||
| 4242 | |||
| 4243 | lock_extent(tree, page_start, page_end, GFP_NOFS); | ||
| 4244 | ordered = btrfs_lookup_ordered_extent(page->mapping->host, | ||
| 4245 | page_offset(page)); | ||
| 4246 | if (ordered) { | ||
| 4247 | /* | ||
| 4248 | * IO on this page will never be started, so we need | ||
| 4249 | * to account for any ordered extents now | ||
| 4250 | */ | ||
| 4251 | clear_extent_bit(tree, page_start, page_end, | ||
| 4252 | EXTENT_DIRTY | EXTENT_DELALLOC | | ||
| 4253 | EXTENT_LOCKED, 1, 0, GFP_NOFS); | ||
| 4254 | btrfs_finish_ordered_io(page->mapping->host, | ||
| 4255 | page_start, page_end); | ||
| 4256 | btrfs_put_ordered_extent(ordered); | ||
| 4257 | lock_extent(tree, page_start, page_end, GFP_NOFS); | ||
| 4258 | } | ||
| 4259 | clear_extent_bit(tree, page_start, page_end, | ||
| 4260 | EXTENT_LOCKED | EXTENT_DIRTY | EXTENT_DELALLOC | | ||
| 4261 | EXTENT_ORDERED, | ||
| 4262 | 1, 1, GFP_NOFS); | ||
| 4263 | __btrfs_releasepage(page, GFP_NOFS); | ||
| 4264 | |||
| 4265 | ClearPageChecked(page); | ||
| 4266 | if (PagePrivate(page)) { | ||
| 4267 | ClearPagePrivate(page); | ||
| 4268 | set_page_private(page, 0); | ||
| 4269 | page_cache_release(page); | ||
| 4270 | } | ||
| 4271 | } | ||
| 4272 | |||
| 4273 | /* | ||
| 4274 | * btrfs_page_mkwrite() is not allowed to change the file size as it gets | ||
| 4275 | * called from a page fault handler when a page is first dirtied. Hence we must | ||
| 4276 | * be careful to check for EOF conditions here. We set the page up correctly | ||
| 4277 | * for a written page which means we get ENOSPC checking when writing into | ||
| 4278 | * holes and correct delalloc and unwritten extent mapping on filesystems that | ||
| 4279 | * support these features. | ||
| 4280 | * | ||
| 4281 | * We are not allowed to take the i_mutex here so we have to play games to | ||
| 4282 | * protect against truncate races as the page could now be beyond EOF. Because | ||
| 4283 | * vmtruncate() writes the inode size before removing pages, once we have the | ||
| 4284 | * page lock we can determine safely if the page is beyond EOF. If it is not | ||
| 4285 | * beyond EOF, then the page is guaranteed safe against truncation until we | ||
| 4286 | * unlock the page. | ||
| 4287 | */ | ||
| 4288 | int btrfs_page_mkwrite(struct vm_area_struct *vma, struct page *page) | ||
| 4289 | { | ||
| 4290 | struct inode *inode = fdentry(vma->vm_file)->d_inode; | ||
| 4291 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 4292 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 4293 | struct btrfs_ordered_extent *ordered; | ||
| 4294 | char *kaddr; | ||
| 4295 | unsigned long zero_start; | ||
| 4296 | loff_t size; | ||
| 4297 | int ret; | ||
| 4298 | u64 page_start; | ||
| 4299 | u64 page_end; | ||
| 4300 | |||
| 4301 | ret = btrfs_check_free_space(root, PAGE_CACHE_SIZE, 0); | ||
| 4302 | if (ret) | ||
| 4303 | goto out; | ||
| 4304 | |||
| 4305 | ret = -EINVAL; | ||
| 4306 | again: | ||
| 4307 | lock_page(page); | ||
| 4308 | size = i_size_read(inode); | ||
| 4309 | page_start = page_offset(page); | ||
| 4310 | page_end = page_start + PAGE_CACHE_SIZE - 1; | ||
| 4311 | |||
| 4312 | if ((page->mapping != inode->i_mapping) || | ||
| 4313 | (page_start >= size)) { | ||
| 4314 | /* page got truncated out from underneath us */ | ||
| 4315 | goto out_unlock; | ||
| 4316 | } | ||
| 4317 | wait_on_page_writeback(page); | ||
| 4318 | |||
| 4319 | lock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 4320 | set_page_extent_mapped(page); | ||
| 4321 | |||
| 4322 | /* | ||
| 4323 | * we can't set the delalloc bits if there are pending ordered | ||
| 4324 | * extents. Drop our locks and wait for them to finish | ||
| 4325 | */ | ||
| 4326 | ordered = btrfs_lookup_ordered_extent(inode, page_start); | ||
| 4327 | if (ordered) { | ||
| 4328 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 4329 | unlock_page(page); | ||
| 4330 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 4331 | btrfs_put_ordered_extent(ordered); | ||
| 4332 | goto again; | ||
| 4333 | } | ||
| 4334 | |||
| 4335 | btrfs_set_extent_delalloc(inode, page_start, page_end); | ||
| 4336 | ret = 0; | ||
| 4337 | |||
| 4338 | /* page is wholly or partially inside EOF */ | ||
| 4339 | if (page_start + PAGE_CACHE_SIZE > size) | ||
| 4340 | zero_start = size & ~PAGE_CACHE_MASK; | ||
| 4341 | else | ||
| 4342 | zero_start = PAGE_CACHE_SIZE; | ||
| 4343 | |||
| 4344 | if (zero_start != PAGE_CACHE_SIZE) { | ||
| 4345 | kaddr = kmap(page); | ||
| 4346 | memset(kaddr + zero_start, 0, PAGE_CACHE_SIZE - zero_start); | ||
| 4347 | flush_dcache_page(page); | ||
| 4348 | kunmap(page); | ||
| 4349 | } | ||
| 4350 | ClearPageChecked(page); | ||
| 4351 | set_page_dirty(page); | ||
| 4352 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 4353 | |||
| 4354 | out_unlock: | ||
| 4355 | unlock_page(page); | ||
| 4356 | out: | ||
| 4357 | return ret; | ||
| 4358 | } | ||
| 4359 | |||
| 4360 | static void btrfs_truncate(struct inode *inode) | ||
| 4361 | { | ||
| 4362 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 4363 | int ret; | ||
| 4364 | struct btrfs_trans_handle *trans; | ||
| 4365 | unsigned long nr; | ||
| 4366 | u64 mask = root->sectorsize - 1; | ||
| 4367 | |||
| 4368 | if (!S_ISREG(inode->i_mode)) | ||
| 4369 | return; | ||
| 4370 | if (IS_APPEND(inode) || IS_IMMUTABLE(inode)) | ||
| 4371 | return; | ||
| 4372 | |||
| 4373 | btrfs_truncate_page(inode->i_mapping, inode->i_size); | ||
| 4374 | btrfs_wait_ordered_range(inode, inode->i_size & (~mask), (u64)-1); | ||
| 4375 | |||
| 4376 | trans = btrfs_start_transaction(root, 1); | ||
| 4377 | btrfs_set_trans_block_group(trans, inode); | ||
| 4378 | btrfs_i_size_write(inode, inode->i_size); | ||
| 4379 | |||
| 4380 | ret = btrfs_orphan_add(trans, inode); | ||
| 4381 | if (ret) | ||
| 4382 | goto out; | ||
| 4383 | /* FIXME, add redo link to tree so we don't leak on crash */ | ||
| 4384 | ret = btrfs_truncate_inode_items(trans, root, inode, inode->i_size, | ||
| 4385 | BTRFS_EXTENT_DATA_KEY); | ||
| 4386 | btrfs_update_inode(trans, root, inode); | ||
| 4387 | |||
| 4388 | ret = btrfs_orphan_del(trans, inode); | ||
| 4389 | BUG_ON(ret); | ||
| 4390 | |||
| 4391 | out: | ||
| 4392 | nr = trans->blocks_used; | ||
| 4393 | ret = btrfs_end_transaction_throttle(trans, root); | ||
| 4394 | BUG_ON(ret); | ||
| 4395 | btrfs_btree_balance_dirty(root, nr); | ||
| 4396 | } | ||
| 4397 | |||
| 4398 | /* | ||
| 4399 | * create a new subvolume directory/inode (helper for the ioctl). | ||
| 4400 | */ | ||
| 4401 | int btrfs_create_subvol_root(struct btrfs_trans_handle *trans, | ||
| 4402 | struct btrfs_root *new_root, struct dentry *dentry, | ||
| 4403 | u64 new_dirid, u64 alloc_hint) | ||
| 4404 | { | ||
| 4405 | struct inode *inode; | ||
| 4406 | int error; | ||
| 4407 | u64 index = 0; | ||
| 4408 | |||
| 4409 | inode = btrfs_new_inode(trans, new_root, NULL, "..", 2, new_dirid, | ||
| 4410 | new_dirid, alloc_hint, S_IFDIR | 0700, &index); | ||
| 4411 | if (IS_ERR(inode)) | ||
| 4412 | return PTR_ERR(inode); | ||
| 4413 | inode->i_op = &btrfs_dir_inode_operations; | ||
| 4414 | inode->i_fop = &btrfs_dir_file_operations; | ||
| 4415 | |||
| 4416 | inode->i_nlink = 1; | ||
| 4417 | btrfs_i_size_write(inode, 0); | ||
| 4418 | |||
| 4419 | error = btrfs_update_inode(trans, new_root, inode); | ||
| 4420 | if (error) | ||
| 4421 | return error; | ||
| 4422 | |||
| 4423 | d_instantiate(dentry, inode); | ||
| 4424 | return 0; | ||
| 4425 | } | ||
| 4426 | |||
| 4427 | /* helper function for file defrag and space balancing. This | ||
| 4428 | * forces readahead on a given range of bytes in an inode | ||
| 4429 | */ | ||
| 4430 | unsigned long btrfs_force_ra(struct address_space *mapping, | ||
| 4431 | struct file_ra_state *ra, struct file *file, | ||
| 4432 | pgoff_t offset, pgoff_t last_index) | ||
| 4433 | { | ||
| 4434 | pgoff_t req_size = last_index - offset + 1; | ||
| 4435 | |||
| 4436 | page_cache_sync_readahead(mapping, ra, file, offset, req_size); | ||
| 4437 | return offset + req_size; | ||
| 4438 | } | ||
| 4439 | |||
| 4440 | struct inode *btrfs_alloc_inode(struct super_block *sb) | ||
| 4441 | { | ||
| 4442 | struct btrfs_inode *ei; | ||
| 4443 | |||
| 4444 | ei = kmem_cache_alloc(btrfs_inode_cachep, GFP_NOFS); | ||
| 4445 | if (!ei) | ||
| 4446 | return NULL; | ||
| 4447 | ei->last_trans = 0; | ||
| 4448 | ei->logged_trans = 0; | ||
| 4449 | btrfs_ordered_inode_tree_init(&ei->ordered_tree); | ||
| 4450 | ei->i_acl = BTRFS_ACL_NOT_CACHED; | ||
| 4451 | ei->i_default_acl = BTRFS_ACL_NOT_CACHED; | ||
| 4452 | INIT_LIST_HEAD(&ei->i_orphan); | ||
| 4453 | return &ei->vfs_inode; | ||
| 4454 | } | ||
| 4455 | |||
| 4456 | void btrfs_destroy_inode(struct inode *inode) | ||
| 4457 | { | ||
| 4458 | struct btrfs_ordered_extent *ordered; | ||
| 4459 | WARN_ON(!list_empty(&inode->i_dentry)); | ||
| 4460 | WARN_ON(inode->i_data.nrpages); | ||
| 4461 | |||
| 4462 | if (BTRFS_I(inode)->i_acl && | ||
| 4463 | BTRFS_I(inode)->i_acl != BTRFS_ACL_NOT_CACHED) | ||
| 4464 | posix_acl_release(BTRFS_I(inode)->i_acl); | ||
| 4465 | if (BTRFS_I(inode)->i_default_acl && | ||
| 4466 | BTRFS_I(inode)->i_default_acl != BTRFS_ACL_NOT_CACHED) | ||
| 4467 | posix_acl_release(BTRFS_I(inode)->i_default_acl); | ||
| 4468 | |||
| 4469 | spin_lock(&BTRFS_I(inode)->root->list_lock); | ||
| 4470 | if (!list_empty(&BTRFS_I(inode)->i_orphan)) { | ||
| 4471 | printk(KERN_ERR "BTRFS: inode %lu: inode still on the orphan" | ||
| 4472 | " list\n", inode->i_ino); | ||
| 4473 | dump_stack(); | ||
| 4474 | } | ||
| 4475 | spin_unlock(&BTRFS_I(inode)->root->list_lock); | ||
| 4476 | |||
| 4477 | while (1) { | ||
| 4478 | ordered = btrfs_lookup_first_ordered_extent(inode, (u64)-1); | ||
| 4479 | if (!ordered) | ||
| 4480 | break; | ||
| 4481 | else { | ||
| 4482 | printk(KERN_ERR "btrfs found ordered " | ||
| 4483 | "extent %llu %llu on inode cleanup\n", | ||
| 4484 | (unsigned long long)ordered->file_offset, | ||
| 4485 | (unsigned long long)ordered->len); | ||
| 4486 | btrfs_remove_ordered_extent(inode, ordered); | ||
| 4487 | btrfs_put_ordered_extent(ordered); | ||
| 4488 | btrfs_put_ordered_extent(ordered); | ||
| 4489 | } | ||
| 4490 | } | ||
| 4491 | btrfs_drop_extent_cache(inode, 0, (u64)-1, 0); | ||
| 4492 | kmem_cache_free(btrfs_inode_cachep, BTRFS_I(inode)); | ||
| 4493 | } | ||
| 4494 | |||
| 4495 | static void init_once(void *foo) | ||
| 4496 | { | ||
| 4497 | struct btrfs_inode *ei = (struct btrfs_inode *) foo; | ||
| 4498 | |||
| 4499 | inode_init_once(&ei->vfs_inode); | ||
| 4500 | } | ||
| 4501 | |||
| 4502 | void btrfs_destroy_cachep(void) | ||
| 4503 | { | ||
| 4504 | if (btrfs_inode_cachep) | ||
| 4505 | kmem_cache_destroy(btrfs_inode_cachep); | ||
| 4506 | if (btrfs_trans_handle_cachep) | ||
| 4507 | kmem_cache_destroy(btrfs_trans_handle_cachep); | ||
| 4508 | if (btrfs_transaction_cachep) | ||
| 4509 | kmem_cache_destroy(btrfs_transaction_cachep); | ||
| 4510 | if (btrfs_bit_radix_cachep) | ||
| 4511 | kmem_cache_destroy(btrfs_bit_radix_cachep); | ||
| 4512 | if (btrfs_path_cachep) | ||
| 4513 | kmem_cache_destroy(btrfs_path_cachep); | ||
| 4514 | } | ||
| 4515 | |||
| 4516 | struct kmem_cache *btrfs_cache_create(const char *name, size_t size, | ||
| 4517 | unsigned long extra_flags, | ||
| 4518 | void (*ctor)(void *)) | ||
| 4519 | { | ||
| 4520 | return kmem_cache_create(name, size, 0, (SLAB_RECLAIM_ACCOUNT | | ||
| 4521 | SLAB_MEM_SPREAD | extra_flags), ctor); | ||
| 4522 | } | ||
| 4523 | |||
| 4524 | int btrfs_init_cachep(void) | ||
| 4525 | { | ||
| 4526 | btrfs_inode_cachep = btrfs_cache_create("btrfs_inode_cache", | ||
| 4527 | sizeof(struct btrfs_inode), | ||
| 4528 | 0, init_once); | ||
| 4529 | if (!btrfs_inode_cachep) | ||
| 4530 | goto fail; | ||
| 4531 | btrfs_trans_handle_cachep = | ||
| 4532 | btrfs_cache_create("btrfs_trans_handle_cache", | ||
| 4533 | sizeof(struct btrfs_trans_handle), | ||
| 4534 | 0, NULL); | ||
| 4535 | if (!btrfs_trans_handle_cachep) | ||
| 4536 | goto fail; | ||
| 4537 | btrfs_transaction_cachep = btrfs_cache_create("btrfs_transaction_cache", | ||
| 4538 | sizeof(struct btrfs_transaction), | ||
| 4539 | 0, NULL); | ||
| 4540 | if (!btrfs_transaction_cachep) | ||
| 4541 | goto fail; | ||
| 4542 | btrfs_path_cachep = btrfs_cache_create("btrfs_path_cache", | ||
| 4543 | sizeof(struct btrfs_path), | ||
| 4544 | 0, NULL); | ||
| 4545 | if (!btrfs_path_cachep) | ||
| 4546 | goto fail; | ||
| 4547 | btrfs_bit_radix_cachep = btrfs_cache_create("btrfs_radix", 256, | ||
| 4548 | SLAB_DESTROY_BY_RCU, NULL); | ||
| 4549 | if (!btrfs_bit_radix_cachep) | ||
| 4550 | goto fail; | ||
| 4551 | return 0; | ||
| 4552 | fail: | ||
| 4553 | btrfs_destroy_cachep(); | ||
| 4554 | return -ENOMEM; | ||
| 4555 | } | ||
| 4556 | |||
| 4557 | static int btrfs_getattr(struct vfsmount *mnt, | ||
| 4558 | struct dentry *dentry, struct kstat *stat) | ||
| 4559 | { | ||
| 4560 | struct inode *inode = dentry->d_inode; | ||
| 4561 | generic_fillattr(inode, stat); | ||
| 4562 | stat->dev = BTRFS_I(inode)->root->anon_super.s_dev; | ||
| 4563 | stat->blksize = PAGE_CACHE_SIZE; | ||
| 4564 | stat->blocks = (inode_get_bytes(inode) + | ||
| 4565 | BTRFS_I(inode)->delalloc_bytes) >> 9; | ||
| 4566 | return 0; | ||
| 4567 | } | ||
| 4568 | |||
| 4569 | static int btrfs_rename(struct inode *old_dir, struct dentry *old_dentry, | ||
| 4570 | struct inode *new_dir, struct dentry *new_dentry) | ||
| 4571 | { | ||
| 4572 | struct btrfs_trans_handle *trans; | ||
| 4573 | struct btrfs_root *root = BTRFS_I(old_dir)->root; | ||
| 4574 | struct inode *new_inode = new_dentry->d_inode; | ||
| 4575 | struct inode *old_inode = old_dentry->d_inode; | ||
| 4576 | struct timespec ctime = CURRENT_TIME; | ||
| 4577 | u64 index = 0; | ||
| 4578 | int ret; | ||
| 4579 | |||
| 4580 | /* we're not allowed to rename between subvolumes */ | ||
| 4581 | if (BTRFS_I(old_inode)->root->root_key.objectid != | ||
| 4582 | BTRFS_I(new_dir)->root->root_key.objectid) | ||
| 4583 | return -EXDEV; | ||
| 4584 | |||
| 4585 | if (S_ISDIR(old_inode->i_mode) && new_inode && | ||
| 4586 | new_inode->i_size > BTRFS_EMPTY_DIR_SIZE) { | ||
| 4587 | return -ENOTEMPTY; | ||
| 4588 | } | ||
| 4589 | |||
| 4590 | /* to rename a snapshot or subvolume, we need to juggle the | ||
| 4591 | * backrefs. This isn't coded yet | ||
| 4592 | */ | ||
| 4593 | if (old_inode->i_ino == BTRFS_FIRST_FREE_OBJECTID) | ||
| 4594 | return -EXDEV; | ||
| 4595 | |||
| 4596 | ret = btrfs_check_free_space(root, 1, 0); | ||
| 4597 | if (ret) | ||
| 4598 | goto out_unlock; | ||
| 4599 | |||
| 4600 | trans = btrfs_start_transaction(root, 1); | ||
| 4601 | |||
| 4602 | btrfs_set_trans_block_group(trans, new_dir); | ||
| 4603 | |||
| 4604 | btrfs_inc_nlink(old_dentry->d_inode); | ||
| 4605 | old_dir->i_ctime = old_dir->i_mtime = ctime; | ||
| 4606 | new_dir->i_ctime = new_dir->i_mtime = ctime; | ||
| 4607 | old_inode->i_ctime = ctime; | ||
| 4608 | |||
| 4609 | ret = btrfs_unlink_inode(trans, root, old_dir, old_dentry->d_inode, | ||
| 4610 | old_dentry->d_name.name, | ||
| 4611 | old_dentry->d_name.len); | ||
| 4612 | if (ret) | ||
| 4613 | goto out_fail; | ||
| 4614 | |||
| 4615 | if (new_inode) { | ||
| 4616 | new_inode->i_ctime = CURRENT_TIME; | ||
| 4617 | ret = btrfs_unlink_inode(trans, root, new_dir, | ||
| 4618 | new_dentry->d_inode, | ||
| 4619 | new_dentry->d_name.name, | ||
| 4620 | new_dentry->d_name.len); | ||
| 4621 | if (ret) | ||
| 4622 | goto out_fail; | ||
| 4623 | if (new_inode->i_nlink == 0) { | ||
| 4624 | ret = btrfs_orphan_add(trans, new_dentry->d_inode); | ||
| 4625 | if (ret) | ||
| 4626 | goto out_fail; | ||
| 4627 | } | ||
| 4628 | |||
| 4629 | } | ||
| 4630 | ret = btrfs_set_inode_index(new_dir, &index); | ||
| 4631 | if (ret) | ||
| 4632 | goto out_fail; | ||
| 4633 | |||
| 4634 | ret = btrfs_add_link(trans, new_dentry->d_parent->d_inode, | ||
| 4635 | old_inode, new_dentry->d_name.name, | ||
| 4636 | new_dentry->d_name.len, 1, index); | ||
| 4637 | if (ret) | ||
| 4638 | goto out_fail; | ||
| 4639 | |||
| 4640 | out_fail: | ||
| 4641 | btrfs_end_transaction_throttle(trans, root); | ||
| 4642 | out_unlock: | ||
| 4643 | return ret; | ||
| 4644 | } | ||
| 4645 | |||
| 4646 | /* | ||
| 4647 | * some fairly slow code that needs optimization. This walks the list | ||
| 4648 | * of all the inodes with pending delalloc and forces them to disk. | ||
| 4649 | */ | ||
| 4650 | int btrfs_start_delalloc_inodes(struct btrfs_root *root) | ||
| 4651 | { | ||
| 4652 | struct list_head *head = &root->fs_info->delalloc_inodes; | ||
| 4653 | struct btrfs_inode *binode; | ||
| 4654 | struct inode *inode; | ||
| 4655 | |||
| 4656 | if (root->fs_info->sb->s_flags & MS_RDONLY) | ||
| 4657 | return -EROFS; | ||
| 4658 | |||
| 4659 | spin_lock(&root->fs_info->delalloc_lock); | ||
| 4660 | while (!list_empty(head)) { | ||
| 4661 | binode = list_entry(head->next, struct btrfs_inode, | ||
| 4662 | delalloc_inodes); | ||
| 4663 | inode = igrab(&binode->vfs_inode); | ||
| 4664 | if (!inode) | ||
| 4665 | list_del_init(&binode->delalloc_inodes); | ||
| 4666 | spin_unlock(&root->fs_info->delalloc_lock); | ||
| 4667 | if (inode) { | ||
| 4668 | filemap_flush(inode->i_mapping); | ||
| 4669 | iput(inode); | ||
| 4670 | } | ||
| 4671 | cond_resched(); | ||
| 4672 | spin_lock(&root->fs_info->delalloc_lock); | ||
| 4673 | } | ||
| 4674 | spin_unlock(&root->fs_info->delalloc_lock); | ||
| 4675 | |||
| 4676 | /* the filemap_flush will queue IO into the worker threads, but | ||
| 4677 | * we have to make sure the IO is actually started and that | ||
| 4678 | * ordered extents get created before we return | ||
| 4679 | */ | ||
| 4680 | atomic_inc(&root->fs_info->async_submit_draining); | ||
| 4681 | while (atomic_read(&root->fs_info->nr_async_submits) || | ||
| 4682 | atomic_read(&root->fs_info->async_delalloc_pages)) { | ||
| 4683 | wait_event(root->fs_info->async_submit_wait, | ||
| 4684 | (atomic_read(&root->fs_info->nr_async_submits) == 0 && | ||
| 4685 | atomic_read(&root->fs_info->async_delalloc_pages) == 0)); | ||
| 4686 | } | ||
| 4687 | atomic_dec(&root->fs_info->async_submit_draining); | ||
| 4688 | return 0; | ||
| 4689 | } | ||
| 4690 | |||
| 4691 | static int btrfs_symlink(struct inode *dir, struct dentry *dentry, | ||
| 4692 | const char *symname) | ||
| 4693 | { | ||
| 4694 | struct btrfs_trans_handle *trans; | ||
| 4695 | struct btrfs_root *root = BTRFS_I(dir)->root; | ||
| 4696 | struct btrfs_path *path; | ||
| 4697 | struct btrfs_key key; | ||
| 4698 | struct inode *inode = NULL; | ||
| 4699 | int err; | ||
| 4700 | int drop_inode = 0; | ||
| 4701 | u64 objectid; | ||
| 4702 | u64 index = 0 ; | ||
| 4703 | int name_len; | ||
| 4704 | int datasize; | ||
| 4705 | unsigned long ptr; | ||
| 4706 | struct btrfs_file_extent_item *ei; | ||
| 4707 | struct extent_buffer *leaf; | ||
| 4708 | unsigned long nr = 0; | ||
| 4709 | |||
| 4710 | name_len = strlen(symname) + 1; | ||
| 4711 | if (name_len > BTRFS_MAX_INLINE_DATA_SIZE(root)) | ||
| 4712 | return -ENAMETOOLONG; | ||
| 4713 | |||
| 4714 | err = btrfs_check_free_space(root, 1, 0); | ||
| 4715 | if (err) | ||
| 4716 | goto out_fail; | ||
| 4717 | |||
| 4718 | trans = btrfs_start_transaction(root, 1); | ||
| 4719 | btrfs_set_trans_block_group(trans, dir); | ||
| 4720 | |||
| 4721 | err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid); | ||
| 4722 | if (err) { | ||
| 4723 | err = -ENOSPC; | ||
| 4724 | goto out_unlock; | ||
| 4725 | } | ||
| 4726 | |||
| 4727 | inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name, | ||
| 4728 | dentry->d_name.len, | ||
| 4729 | dentry->d_parent->d_inode->i_ino, objectid, | ||
| 4730 | BTRFS_I(dir)->block_group, S_IFLNK|S_IRWXUGO, | ||
| 4731 | &index); | ||
| 4732 | err = PTR_ERR(inode); | ||
| 4733 | if (IS_ERR(inode)) | ||
| 4734 | goto out_unlock; | ||
| 4735 | |||
| 4736 | err = btrfs_init_acl(inode, dir); | ||
| 4737 | if (err) { | ||
| 4738 | drop_inode = 1; | ||
| 4739 | goto out_unlock; | ||
| 4740 | } | ||
| 4741 | |||
| 4742 | btrfs_set_trans_block_group(trans, inode); | ||
| 4743 | err = btrfs_add_nondir(trans, dentry, inode, 0, index); | ||
| 4744 | if (err) | ||
| 4745 | drop_inode = 1; | ||
| 4746 | else { | ||
| 4747 | inode->i_mapping->a_ops = &btrfs_aops; | ||
| 4748 | inode->i_mapping->backing_dev_info = &root->fs_info->bdi; | ||
| 4749 | inode->i_fop = &btrfs_file_operations; | ||
| 4750 | inode->i_op = &btrfs_file_inode_operations; | ||
| 4751 | BTRFS_I(inode)->io_tree.ops = &btrfs_extent_io_ops; | ||
| 4752 | } | ||
| 4753 | dir->i_sb->s_dirt = 1; | ||
| 4754 | btrfs_update_inode_block_group(trans, inode); | ||
| 4755 | btrfs_update_inode_block_group(trans, dir); | ||
| 4756 | if (drop_inode) | ||
| 4757 | goto out_unlock; | ||
| 4758 | |||
| 4759 | path = btrfs_alloc_path(); | ||
| 4760 | BUG_ON(!path); | ||
| 4761 | key.objectid = inode->i_ino; | ||
| 4762 | key.offset = 0; | ||
| 4763 | btrfs_set_key_type(&key, BTRFS_EXTENT_DATA_KEY); | ||
| 4764 | datasize = btrfs_file_extent_calc_inline_size(name_len); | ||
| 4765 | err = btrfs_insert_empty_item(trans, root, path, &key, | ||
| 4766 | datasize); | ||
| 4767 | if (err) { | ||
| 4768 | drop_inode = 1; | ||
| 4769 | goto out_unlock; | ||
| 4770 | } | ||
| 4771 | leaf = path->nodes[0]; | ||
| 4772 | ei = btrfs_item_ptr(leaf, path->slots[0], | ||
| 4773 | struct btrfs_file_extent_item); | ||
| 4774 | btrfs_set_file_extent_generation(leaf, ei, trans->transid); | ||
| 4775 | btrfs_set_file_extent_type(leaf, ei, | ||
| 4776 | BTRFS_FILE_EXTENT_INLINE); | ||
| 4777 | btrfs_set_file_extent_encryption(leaf, ei, 0); | ||
| 4778 | btrfs_set_file_extent_compression(leaf, ei, 0); | ||
| 4779 | btrfs_set_file_extent_other_encoding(leaf, ei, 0); | ||
| 4780 | btrfs_set_file_extent_ram_bytes(leaf, ei, name_len); | ||
| 4781 | |||
| 4782 | ptr = btrfs_file_extent_inline_start(ei); | ||
| 4783 | write_extent_buffer(leaf, symname, ptr, name_len); | ||
| 4784 | btrfs_mark_buffer_dirty(leaf); | ||
| 4785 | btrfs_free_path(path); | ||
| 4786 | |||
| 4787 | inode->i_op = &btrfs_symlink_inode_operations; | ||
| 4788 | inode->i_mapping->a_ops = &btrfs_symlink_aops; | ||
| 4789 | inode->i_mapping->backing_dev_info = &root->fs_info->bdi; | ||
| 4790 | inode_set_bytes(inode, name_len); | ||
| 4791 | btrfs_i_size_write(inode, name_len - 1); | ||
| 4792 | err = btrfs_update_inode(trans, root, inode); | ||
| 4793 | if (err) | ||
| 4794 | drop_inode = 1; | ||
| 4795 | |||
| 4796 | out_unlock: | ||
| 4797 | nr = trans->blocks_used; | ||
| 4798 | btrfs_end_transaction_throttle(trans, root); | ||
| 4799 | out_fail: | ||
| 4800 | if (drop_inode) { | ||
| 4801 | inode_dec_link_count(inode); | ||
| 4802 | iput(inode); | ||
| 4803 | } | ||
| 4804 | btrfs_btree_balance_dirty(root, nr); | ||
| 4805 | return err; | ||
| 4806 | } | ||
| 4807 | |||
| 4808 | static int prealloc_file_range(struct inode *inode, u64 start, u64 end, | ||
| 4809 | u64 alloc_hint, int mode) | ||
| 4810 | { | ||
| 4811 | struct btrfs_trans_handle *trans; | ||
| 4812 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 4813 | struct btrfs_key ins; | ||
| 4814 | u64 alloc_size; | ||
| 4815 | u64 cur_offset = start; | ||
| 4816 | u64 num_bytes = end - start; | ||
| 4817 | int ret = 0; | ||
| 4818 | |||
| 4819 | trans = btrfs_join_transaction(root, 1); | ||
| 4820 | BUG_ON(!trans); | ||
| 4821 | btrfs_set_trans_block_group(trans, inode); | ||
| 4822 | |||
| 4823 | while (num_bytes > 0) { | ||
| 4824 | alloc_size = min(num_bytes, root->fs_info->max_extent); | ||
| 4825 | ret = btrfs_reserve_extent(trans, root, alloc_size, | ||
| 4826 | root->sectorsize, 0, alloc_hint, | ||
| 4827 | (u64)-1, &ins, 1); | ||
| 4828 | if (ret) { | ||
| 4829 | WARN_ON(1); | ||
| 4830 | goto out; | ||
| 4831 | } | ||
| 4832 | ret = insert_reserved_file_extent(trans, inode, | ||
| 4833 | cur_offset, ins.objectid, | ||
| 4834 | ins.offset, ins.offset, | ||
| 4835 | ins.offset, 0, 0, 0, | ||
| 4836 | BTRFS_FILE_EXTENT_PREALLOC); | ||
| 4837 | BUG_ON(ret); | ||
| 4838 | num_bytes -= ins.offset; | ||
| 4839 | cur_offset += ins.offset; | ||
| 4840 | alloc_hint = ins.objectid + ins.offset; | ||
| 4841 | } | ||
| 4842 | out: | ||
| 4843 | if (cur_offset > start) { | ||
| 4844 | inode->i_ctime = CURRENT_TIME; | ||
| 4845 | btrfs_set_flag(inode, PREALLOC); | ||
| 4846 | if (!(mode & FALLOC_FL_KEEP_SIZE) && | ||
| 4847 | cur_offset > i_size_read(inode)) | ||
| 4848 | btrfs_i_size_write(inode, cur_offset); | ||
| 4849 | ret = btrfs_update_inode(trans, root, inode); | ||
| 4850 | BUG_ON(ret); | ||
| 4851 | } | ||
| 4852 | |||
| 4853 | btrfs_end_transaction(trans, root); | ||
| 4854 | return ret; | ||
| 4855 | } | ||
| 4856 | |||
| 4857 | static long btrfs_fallocate(struct inode *inode, int mode, | ||
| 4858 | loff_t offset, loff_t len) | ||
| 4859 | { | ||
| 4860 | u64 cur_offset; | ||
| 4861 | u64 last_byte; | ||
| 4862 | u64 alloc_start; | ||
| 4863 | u64 alloc_end; | ||
| 4864 | u64 alloc_hint = 0; | ||
| 4865 | u64 mask = BTRFS_I(inode)->root->sectorsize - 1; | ||
| 4866 | struct extent_map *em; | ||
| 4867 | int ret; | ||
| 4868 | |||
| 4869 | alloc_start = offset & ~mask; | ||
| 4870 | alloc_end = (offset + len + mask) & ~mask; | ||
| 4871 | |||
| 4872 | mutex_lock(&inode->i_mutex); | ||
| 4873 | if (alloc_start > inode->i_size) { | ||
| 4874 | ret = btrfs_cont_expand(inode, alloc_start); | ||
| 4875 | if (ret) | ||
| 4876 | goto out; | ||
| 4877 | } | ||
| 4878 | |||
| 4879 | while (1) { | ||
| 4880 | struct btrfs_ordered_extent *ordered; | ||
| 4881 | lock_extent(&BTRFS_I(inode)->io_tree, alloc_start, | ||
| 4882 | alloc_end - 1, GFP_NOFS); | ||
| 4883 | ordered = btrfs_lookup_first_ordered_extent(inode, | ||
| 4884 | alloc_end - 1); | ||
| 4885 | if (ordered && | ||
| 4886 | ordered->file_offset + ordered->len > alloc_start && | ||
| 4887 | ordered->file_offset < alloc_end) { | ||
| 4888 | btrfs_put_ordered_extent(ordered); | ||
| 4889 | unlock_extent(&BTRFS_I(inode)->io_tree, | ||
| 4890 | alloc_start, alloc_end - 1, GFP_NOFS); | ||
| 4891 | btrfs_wait_ordered_range(inode, alloc_start, | ||
| 4892 | alloc_end - alloc_start); | ||
| 4893 | } else { | ||
| 4894 | if (ordered) | ||
| 4895 | btrfs_put_ordered_extent(ordered); | ||
| 4896 | break; | ||
| 4897 | } | ||
| 4898 | } | ||
| 4899 | |||
| 4900 | cur_offset = alloc_start; | ||
| 4901 | while (1) { | ||
| 4902 | em = btrfs_get_extent(inode, NULL, 0, cur_offset, | ||
| 4903 | alloc_end - cur_offset, 0); | ||
| 4904 | BUG_ON(IS_ERR(em) || !em); | ||
| 4905 | last_byte = min(extent_map_end(em), alloc_end); | ||
| 4906 | last_byte = (last_byte + mask) & ~mask; | ||
| 4907 | if (em->block_start == EXTENT_MAP_HOLE) { | ||
| 4908 | ret = prealloc_file_range(inode, cur_offset, | ||
| 4909 | last_byte, alloc_hint, mode); | ||
| 4910 | if (ret < 0) { | ||
| 4911 | free_extent_map(em); | ||
| 4912 | break; | ||
| 4913 | } | ||
| 4914 | } | ||
| 4915 | if (em->block_start <= EXTENT_MAP_LAST_BYTE) | ||
| 4916 | alloc_hint = em->block_start; | ||
| 4917 | free_extent_map(em); | ||
| 4918 | |||
| 4919 | cur_offset = last_byte; | ||
| 4920 | if (cur_offset >= alloc_end) { | ||
| 4921 | ret = 0; | ||
| 4922 | break; | ||
| 4923 | } | ||
| 4924 | } | ||
| 4925 | unlock_extent(&BTRFS_I(inode)->io_tree, alloc_start, alloc_end - 1, | ||
| 4926 | GFP_NOFS); | ||
| 4927 | out: | ||
| 4928 | mutex_unlock(&inode->i_mutex); | ||
| 4929 | return ret; | ||
| 4930 | } | ||
| 4931 | |||
| 4932 | static int btrfs_set_page_dirty(struct page *page) | ||
| 4933 | { | ||
| 4934 | return __set_page_dirty_nobuffers(page); | ||
| 4935 | } | ||
| 4936 | |||
| 4937 | static int btrfs_permission(struct inode *inode, int mask) | ||
| 4938 | { | ||
| 4939 | if (btrfs_test_flag(inode, READONLY) && (mask & MAY_WRITE)) | ||
| 4940 | return -EACCES; | ||
| 4941 | return generic_permission(inode, mask, btrfs_check_acl); | ||
| 4942 | } | ||
| 4943 | |||
| 4944 | static struct inode_operations btrfs_dir_inode_operations = { | ||
| 4945 | .getattr = btrfs_getattr, | ||
| 4946 | .lookup = btrfs_lookup, | ||
| 4947 | .create = btrfs_create, | ||
| 4948 | .unlink = btrfs_unlink, | ||
| 4949 | .link = btrfs_link, | ||
| 4950 | .mkdir = btrfs_mkdir, | ||
| 4951 | .rmdir = btrfs_rmdir, | ||
| 4952 | .rename = btrfs_rename, | ||
| 4953 | .symlink = btrfs_symlink, | ||
| 4954 | .setattr = btrfs_setattr, | ||
| 4955 | .mknod = btrfs_mknod, | ||
| 4956 | .setxattr = btrfs_setxattr, | ||
| 4957 | .getxattr = btrfs_getxattr, | ||
| 4958 | .listxattr = btrfs_listxattr, | ||
| 4959 | .removexattr = btrfs_removexattr, | ||
| 4960 | .permission = btrfs_permission, | ||
| 4961 | }; | ||
| 4962 | static struct inode_operations btrfs_dir_ro_inode_operations = { | ||
| 4963 | .lookup = btrfs_lookup, | ||
| 4964 | .permission = btrfs_permission, | ||
| 4965 | }; | ||
| 4966 | static struct file_operations btrfs_dir_file_operations = { | ||
| 4967 | .llseek = generic_file_llseek, | ||
| 4968 | .read = generic_read_dir, | ||
| 4969 | .readdir = btrfs_real_readdir, | ||
| 4970 | .unlocked_ioctl = btrfs_ioctl, | ||
| 4971 | #ifdef CONFIG_COMPAT | ||
| 4972 | .compat_ioctl = btrfs_ioctl, | ||
| 4973 | #endif | ||
| 4974 | .release = btrfs_release_file, | ||
| 4975 | .fsync = btrfs_sync_file, | ||
| 4976 | }; | ||
| 4977 | |||
| 4978 | static struct extent_io_ops btrfs_extent_io_ops = { | ||
| 4979 | .fill_delalloc = run_delalloc_range, | ||
| 4980 | .submit_bio_hook = btrfs_submit_bio_hook, | ||
| 4981 | .merge_bio_hook = btrfs_merge_bio_hook, | ||
| 4982 | .readpage_end_io_hook = btrfs_readpage_end_io_hook, | ||
| 4983 | .writepage_end_io_hook = btrfs_writepage_end_io_hook, | ||
| 4984 | .writepage_start_hook = btrfs_writepage_start_hook, | ||
| 4985 | .readpage_io_failed_hook = btrfs_io_failed_hook, | ||
| 4986 | .set_bit_hook = btrfs_set_bit_hook, | ||
| 4987 | .clear_bit_hook = btrfs_clear_bit_hook, | ||
| 4988 | }; | ||
| 4989 | |||
| 4990 | static struct address_space_operations btrfs_aops = { | ||
| 4991 | .readpage = btrfs_readpage, | ||
| 4992 | .writepage = btrfs_writepage, | ||
| 4993 | .writepages = btrfs_writepages, | ||
| 4994 | .readpages = btrfs_readpages, | ||
| 4995 | .sync_page = block_sync_page, | ||
| 4996 | .bmap = btrfs_bmap, | ||
| 4997 | .direct_IO = btrfs_direct_IO, | ||
| 4998 | .invalidatepage = btrfs_invalidatepage, | ||
| 4999 | .releasepage = btrfs_releasepage, | ||
| 5000 | .set_page_dirty = btrfs_set_page_dirty, | ||
| 5001 | }; | ||
| 5002 | |||
| 5003 | static struct address_space_operations btrfs_symlink_aops = { | ||
| 5004 | .readpage = btrfs_readpage, | ||
| 5005 | .writepage = btrfs_writepage, | ||
| 5006 | .invalidatepage = btrfs_invalidatepage, | ||
| 5007 | .releasepage = btrfs_releasepage, | ||
| 5008 | }; | ||
| 5009 | |||
| 5010 | static struct inode_operations btrfs_file_inode_operations = { | ||
| 5011 | .truncate = btrfs_truncate, | ||
| 5012 | .getattr = btrfs_getattr, | ||
| 5013 | .setattr = btrfs_setattr, | ||
| 5014 | .setxattr = btrfs_setxattr, | ||
| 5015 | .getxattr = btrfs_getxattr, | ||
| 5016 | .listxattr = btrfs_listxattr, | ||
| 5017 | .removexattr = btrfs_removexattr, | ||
| 5018 | .permission = btrfs_permission, | ||
| 5019 | .fallocate = btrfs_fallocate, | ||
| 5020 | }; | ||
| 5021 | static struct inode_operations btrfs_special_inode_operations = { | ||
| 5022 | .getattr = btrfs_getattr, | ||
| 5023 | .setattr = btrfs_setattr, | ||
| 5024 | .permission = btrfs_permission, | ||
| 5025 | .setxattr = btrfs_setxattr, | ||
| 5026 | .getxattr = btrfs_getxattr, | ||
| 5027 | .listxattr = btrfs_listxattr, | ||
| 5028 | .removexattr = btrfs_removexattr, | ||
| 5029 | }; | ||
| 5030 | static struct inode_operations btrfs_symlink_inode_operations = { | ||
| 5031 | .readlink = generic_readlink, | ||
| 5032 | .follow_link = page_follow_link_light, | ||
| 5033 | .put_link = page_put_link, | ||
| 5034 | .permission = btrfs_permission, | ||
| 5035 | }; | ||
diff --git a/fs/btrfs/ioctl.c b/fs/btrfs/ioctl.c new file mode 100644 index 000000000000..c2aa33e3feb5 --- /dev/null +++ b/fs/btrfs/ioctl.c | |||
| @@ -0,0 +1,1132 @@ | |||
| 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 | |||
| 19 | #include <linux/kernel.h> | ||
| 20 | #include <linux/bio.h> | ||
| 21 | #include <linux/buffer_head.h> | ||
| 22 | #include <linux/file.h> | ||
| 23 | #include <linux/fs.h> | ||
| 24 | #include <linux/fsnotify.h> | ||
| 25 | #include <linux/pagemap.h> | ||
| 26 | #include <linux/highmem.h> | ||
| 27 | #include <linux/time.h> | ||
| 28 | #include <linux/init.h> | ||
| 29 | #include <linux/string.h> | ||
| 30 | #include <linux/smp_lock.h> | ||
| 31 | #include <linux/backing-dev.h> | ||
| 32 | #include <linux/mount.h> | ||
| 33 | #include <linux/mpage.h> | ||
| 34 | #include <linux/namei.h> | ||
| 35 | #include <linux/swap.h> | ||
| 36 | #include <linux/writeback.h> | ||
| 37 | #include <linux/statfs.h> | ||
| 38 | #include <linux/compat.h> | ||
| 39 | #include <linux/bit_spinlock.h> | ||
| 40 | #include <linux/security.h> | ||
| 41 | #include <linux/version.h> | ||
| 42 | #include <linux/xattr.h> | ||
| 43 | #include <linux/vmalloc.h> | ||
| 44 | #include "compat.h" | ||
| 45 | #include "ctree.h" | ||
| 46 | #include "disk-io.h" | ||
| 47 | #include "transaction.h" | ||
| 48 | #include "btrfs_inode.h" | ||
| 49 | #include "ioctl.h" | ||
| 50 | #include "print-tree.h" | ||
| 51 | #include "volumes.h" | ||
| 52 | #include "locking.h" | ||
| 53 | |||
| 54 | |||
| 55 | |||
| 56 | static noinline int create_subvol(struct btrfs_root *root, | ||
| 57 | struct dentry *dentry, | ||
| 58 | char *name, int namelen) | ||
| 59 | { | ||
| 60 | struct btrfs_trans_handle *trans; | ||
| 61 | struct btrfs_key key; | ||
| 62 | struct btrfs_root_item root_item; | ||
| 63 | struct btrfs_inode_item *inode_item; | ||
| 64 | struct extent_buffer *leaf; | ||
| 65 | struct btrfs_root *new_root = root; | ||
| 66 | struct inode *dir; | ||
| 67 | int ret; | ||
| 68 | int err; | ||
| 69 | u64 objectid; | ||
| 70 | u64 new_dirid = BTRFS_FIRST_FREE_OBJECTID; | ||
| 71 | u64 index = 0; | ||
| 72 | unsigned long nr = 1; | ||
| 73 | |||
| 74 | ret = btrfs_check_free_space(root, 1, 0); | ||
| 75 | if (ret) | ||
| 76 | goto fail_commit; | ||
| 77 | |||
| 78 | trans = btrfs_start_transaction(root, 1); | ||
| 79 | BUG_ON(!trans); | ||
| 80 | |||
| 81 | ret = btrfs_find_free_objectid(trans, root->fs_info->tree_root, | ||
| 82 | 0, &objectid); | ||
| 83 | if (ret) | ||
| 84 | goto fail; | ||
| 85 | |||
| 86 | leaf = btrfs_alloc_free_block(trans, root, root->leafsize, 0, | ||
| 87 | objectid, trans->transid, 0, 0, 0); | ||
| 88 | if (IS_ERR(leaf)) { | ||
| 89 | ret = PTR_ERR(leaf); | ||
| 90 | goto fail; | ||
| 91 | } | ||
| 92 | |||
| 93 | btrfs_set_header_nritems(leaf, 0); | ||
| 94 | btrfs_set_header_level(leaf, 0); | ||
| 95 | btrfs_set_header_bytenr(leaf, leaf->start); | ||
| 96 | btrfs_set_header_generation(leaf, trans->transid); | ||
| 97 | btrfs_set_header_owner(leaf, objectid); | ||
| 98 | |||
| 99 | write_extent_buffer(leaf, root->fs_info->fsid, | ||
| 100 | (unsigned long)btrfs_header_fsid(leaf), | ||
| 101 | BTRFS_FSID_SIZE); | ||
| 102 | btrfs_mark_buffer_dirty(leaf); | ||
| 103 | |||
| 104 | inode_item = &root_item.inode; | ||
| 105 | memset(inode_item, 0, sizeof(*inode_item)); | ||
| 106 | inode_item->generation = cpu_to_le64(1); | ||
| 107 | inode_item->size = cpu_to_le64(3); | ||
| 108 | inode_item->nlink = cpu_to_le32(1); | ||
| 109 | inode_item->nbytes = cpu_to_le64(root->leafsize); | ||
| 110 | inode_item->mode = cpu_to_le32(S_IFDIR | 0755); | ||
| 111 | |||
| 112 | btrfs_set_root_bytenr(&root_item, leaf->start); | ||
| 113 | btrfs_set_root_generation(&root_item, trans->transid); | ||
| 114 | btrfs_set_root_level(&root_item, 0); | ||
| 115 | btrfs_set_root_refs(&root_item, 1); | ||
| 116 | btrfs_set_root_used(&root_item, 0); | ||
| 117 | btrfs_set_root_last_snapshot(&root_item, 0); | ||
| 118 | |||
| 119 | memset(&root_item.drop_progress, 0, sizeof(root_item.drop_progress)); | ||
| 120 | root_item.drop_level = 0; | ||
| 121 | |||
| 122 | btrfs_tree_unlock(leaf); | ||
| 123 | free_extent_buffer(leaf); | ||
| 124 | leaf = NULL; | ||
| 125 | |||
| 126 | btrfs_set_root_dirid(&root_item, new_dirid); | ||
| 127 | |||
| 128 | key.objectid = objectid; | ||
| 129 | key.offset = 1; | ||
| 130 | btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY); | ||
| 131 | ret = btrfs_insert_root(trans, root->fs_info->tree_root, &key, | ||
| 132 | &root_item); | ||
| 133 | if (ret) | ||
| 134 | goto fail; | ||
| 135 | |||
| 136 | /* | ||
| 137 | * insert the directory item | ||
| 138 | */ | ||
| 139 | key.offset = (u64)-1; | ||
| 140 | dir = dentry->d_parent->d_inode; | ||
| 141 | ret = btrfs_set_inode_index(dir, &index); | ||
| 142 | BUG_ON(ret); | ||
| 143 | |||
| 144 | ret = btrfs_insert_dir_item(trans, root, | ||
| 145 | name, namelen, dir->i_ino, &key, | ||
| 146 | BTRFS_FT_DIR, index); | ||
| 147 | if (ret) | ||
| 148 | goto fail; | ||
| 149 | |||
| 150 | btrfs_i_size_write(dir, dir->i_size + namelen * 2); | ||
| 151 | ret = btrfs_update_inode(trans, root, dir); | ||
| 152 | BUG_ON(ret); | ||
| 153 | |||
| 154 | /* add the backref first */ | ||
| 155 | ret = btrfs_add_root_ref(trans, root->fs_info->tree_root, | ||
| 156 | objectid, BTRFS_ROOT_BACKREF_KEY, | ||
| 157 | root->root_key.objectid, | ||
| 158 | dir->i_ino, index, name, namelen); | ||
| 159 | |||
| 160 | BUG_ON(ret); | ||
| 161 | |||
| 162 | /* now add the forward ref */ | ||
| 163 | ret = btrfs_add_root_ref(trans, root->fs_info->tree_root, | ||
| 164 | root->root_key.objectid, BTRFS_ROOT_REF_KEY, | ||
| 165 | objectid, | ||
| 166 | dir->i_ino, index, name, namelen); | ||
| 167 | |||
| 168 | BUG_ON(ret); | ||
| 169 | |||
| 170 | ret = btrfs_commit_transaction(trans, root); | ||
| 171 | if (ret) | ||
| 172 | goto fail_commit; | ||
| 173 | |||
| 174 | new_root = btrfs_read_fs_root_no_name(root->fs_info, &key); | ||
| 175 | BUG_ON(!new_root); | ||
| 176 | |||
| 177 | trans = btrfs_start_transaction(new_root, 1); | ||
| 178 | BUG_ON(!trans); | ||
| 179 | |||
| 180 | ret = btrfs_create_subvol_root(trans, new_root, dentry, new_dirid, | ||
| 181 | BTRFS_I(dir)->block_group); | ||
| 182 | if (ret) | ||
| 183 | goto fail; | ||
| 184 | |||
| 185 | fail: | ||
| 186 | nr = trans->blocks_used; | ||
| 187 | err = btrfs_commit_transaction(trans, new_root); | ||
| 188 | if (err && !ret) | ||
| 189 | ret = err; | ||
| 190 | fail_commit: | ||
| 191 | btrfs_btree_balance_dirty(root, nr); | ||
| 192 | return ret; | ||
| 193 | } | ||
| 194 | |||
| 195 | static int create_snapshot(struct btrfs_root *root, struct dentry *dentry, | ||
| 196 | char *name, int namelen) | ||
| 197 | { | ||
| 198 | struct btrfs_pending_snapshot *pending_snapshot; | ||
| 199 | struct btrfs_trans_handle *trans; | ||
| 200 | int ret = 0; | ||
| 201 | int err; | ||
| 202 | unsigned long nr = 0; | ||
| 203 | |||
| 204 | if (!root->ref_cows) | ||
| 205 | return -EINVAL; | ||
| 206 | |||
| 207 | ret = btrfs_check_free_space(root, 1, 0); | ||
| 208 | if (ret) | ||
| 209 | goto fail_unlock; | ||
| 210 | |||
| 211 | pending_snapshot = kzalloc(sizeof(*pending_snapshot), GFP_NOFS); | ||
| 212 | if (!pending_snapshot) { | ||
| 213 | ret = -ENOMEM; | ||
| 214 | goto fail_unlock; | ||
| 215 | } | ||
| 216 | pending_snapshot->name = kmalloc(namelen + 1, GFP_NOFS); | ||
| 217 | if (!pending_snapshot->name) { | ||
| 218 | ret = -ENOMEM; | ||
| 219 | kfree(pending_snapshot); | ||
| 220 | goto fail_unlock; | ||
| 221 | } | ||
| 222 | memcpy(pending_snapshot->name, name, namelen); | ||
| 223 | pending_snapshot->name[namelen] = '\0'; | ||
| 224 | pending_snapshot->dentry = dentry; | ||
| 225 | trans = btrfs_start_transaction(root, 1); | ||
| 226 | BUG_ON(!trans); | ||
| 227 | pending_snapshot->root = root; | ||
| 228 | list_add(&pending_snapshot->list, | ||
| 229 | &trans->transaction->pending_snapshots); | ||
| 230 | err = btrfs_commit_transaction(trans, root); | ||
| 231 | |||
| 232 | fail_unlock: | ||
| 233 | btrfs_btree_balance_dirty(root, nr); | ||
| 234 | return ret; | ||
| 235 | } | ||
| 236 | |||
| 237 | /* copy of may_create in fs/namei.c() */ | ||
| 238 | static inline int btrfs_may_create(struct inode *dir, struct dentry *child) | ||
| 239 | { | ||
| 240 | if (child->d_inode) | ||
| 241 | return -EEXIST; | ||
| 242 | if (IS_DEADDIR(dir)) | ||
| 243 | return -ENOENT; | ||
| 244 | return inode_permission(dir, MAY_WRITE | MAY_EXEC); | ||
| 245 | } | ||
| 246 | |||
| 247 | /* | ||
| 248 | * Create a new subvolume below @parent. This is largely modeled after | ||
| 249 | * sys_mkdirat and vfs_mkdir, but we only do a single component lookup | ||
| 250 | * inside this filesystem so it's quite a bit simpler. | ||
| 251 | */ | ||
| 252 | static noinline int btrfs_mksubvol(struct path *parent, char *name, | ||
| 253 | int mode, int namelen, | ||
| 254 | struct btrfs_root *snap_src) | ||
| 255 | { | ||
| 256 | struct dentry *dentry; | ||
| 257 | int error; | ||
| 258 | |||
| 259 | mutex_lock_nested(&parent->dentry->d_inode->i_mutex, I_MUTEX_PARENT); | ||
| 260 | |||
| 261 | dentry = lookup_one_len(name, parent->dentry, namelen); | ||
| 262 | error = PTR_ERR(dentry); | ||
| 263 | if (IS_ERR(dentry)) | ||
| 264 | goto out_unlock; | ||
| 265 | |||
| 266 | error = -EEXIST; | ||
| 267 | if (dentry->d_inode) | ||
| 268 | goto out_dput; | ||
| 269 | |||
| 270 | if (!IS_POSIXACL(parent->dentry->d_inode)) | ||
| 271 | mode &= ~current->fs->umask; | ||
| 272 | |||
| 273 | error = mnt_want_write(parent->mnt); | ||
| 274 | if (error) | ||
| 275 | goto out_dput; | ||
| 276 | |||
| 277 | error = btrfs_may_create(parent->dentry->d_inode, dentry); | ||
| 278 | if (error) | ||
| 279 | goto out_drop_write; | ||
| 280 | |||
| 281 | /* | ||
| 282 | * Actually perform the low-level subvolume creation after all | ||
| 283 | * this VFS fuzz. | ||
| 284 | * | ||
| 285 | * Eventually we want to pass in an inode under which we create this | ||
| 286 | * subvolume, but for now all are under the filesystem root. | ||
| 287 | * | ||
| 288 | * Also we should pass on the mode eventually to allow creating new | ||
| 289 | * subvolume with specific mode bits. | ||
| 290 | */ | ||
| 291 | if (snap_src) { | ||
| 292 | struct dentry *dir = dentry->d_parent; | ||
| 293 | struct dentry *test = dir->d_parent; | ||
| 294 | struct btrfs_path *path = btrfs_alloc_path(); | ||
| 295 | int ret; | ||
| 296 | u64 test_oid; | ||
| 297 | u64 parent_oid = BTRFS_I(dir->d_inode)->root->root_key.objectid; | ||
| 298 | |||
| 299 | test_oid = snap_src->root_key.objectid; | ||
| 300 | |||
| 301 | ret = btrfs_find_root_ref(snap_src->fs_info->tree_root, | ||
| 302 | path, parent_oid, test_oid); | ||
| 303 | if (ret == 0) | ||
| 304 | goto create; | ||
| 305 | btrfs_release_path(snap_src->fs_info->tree_root, path); | ||
| 306 | |||
| 307 | /* we need to make sure we aren't creating a directory loop | ||
| 308 | * by taking a snapshot of something that has our current | ||
| 309 | * subvol in its directory tree. So, this loops through | ||
| 310 | * the dentries and checks the forward refs for each subvolume | ||
| 311 | * to see if is references the subvolume where we are | ||
| 312 | * placing this new snapshot. | ||
| 313 | */ | ||
| 314 | while (1) { | ||
| 315 | if (!test || | ||
| 316 | dir == snap_src->fs_info->sb->s_root || | ||
| 317 | test == snap_src->fs_info->sb->s_root || | ||
| 318 | test->d_inode->i_sb != snap_src->fs_info->sb) { | ||
| 319 | break; | ||
| 320 | } | ||
| 321 | if (S_ISLNK(test->d_inode->i_mode)) { | ||
| 322 | printk(KERN_INFO "Btrfs symlink in snapshot " | ||
| 323 | "path, failed\n"); | ||
| 324 | error = -EMLINK; | ||
| 325 | btrfs_free_path(path); | ||
| 326 | goto out_drop_write; | ||
| 327 | } | ||
| 328 | test_oid = | ||
| 329 | BTRFS_I(test->d_inode)->root->root_key.objectid; | ||
| 330 | ret = btrfs_find_root_ref(snap_src->fs_info->tree_root, | ||
| 331 | path, test_oid, parent_oid); | ||
| 332 | if (ret == 0) { | ||
| 333 | printk(KERN_INFO "Btrfs snapshot creation " | ||
| 334 | "failed, looping\n"); | ||
| 335 | error = -EMLINK; | ||
| 336 | btrfs_free_path(path); | ||
| 337 | goto out_drop_write; | ||
| 338 | } | ||
| 339 | btrfs_release_path(snap_src->fs_info->tree_root, path); | ||
| 340 | test = test->d_parent; | ||
| 341 | } | ||
| 342 | create: | ||
| 343 | btrfs_free_path(path); | ||
| 344 | error = create_snapshot(snap_src, dentry, name, namelen); | ||
| 345 | } else { | ||
| 346 | error = create_subvol(BTRFS_I(parent->dentry->d_inode)->root, | ||
| 347 | dentry, name, namelen); | ||
| 348 | } | ||
| 349 | if (error) | ||
| 350 | goto out_drop_write; | ||
| 351 | |||
| 352 | fsnotify_mkdir(parent->dentry->d_inode, dentry); | ||
| 353 | out_drop_write: | ||
| 354 | mnt_drop_write(parent->mnt); | ||
| 355 | out_dput: | ||
| 356 | dput(dentry); | ||
| 357 | out_unlock: | ||
| 358 | mutex_unlock(&parent->dentry->d_inode->i_mutex); | ||
| 359 | return error; | ||
| 360 | } | ||
| 361 | |||
| 362 | |||
| 363 | static int btrfs_defrag_file(struct file *file) | ||
| 364 | { | ||
| 365 | struct inode *inode = fdentry(file)->d_inode; | ||
| 366 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 367 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 368 | struct btrfs_ordered_extent *ordered; | ||
| 369 | struct page *page; | ||
| 370 | unsigned long last_index; | ||
| 371 | unsigned long ra_pages = root->fs_info->bdi.ra_pages; | ||
| 372 | unsigned long total_read = 0; | ||
| 373 | u64 page_start; | ||
| 374 | u64 page_end; | ||
| 375 | unsigned long i; | ||
| 376 | int ret; | ||
| 377 | |||
| 378 | ret = btrfs_check_free_space(root, inode->i_size, 0); | ||
| 379 | if (ret) | ||
| 380 | return -ENOSPC; | ||
| 381 | |||
| 382 | mutex_lock(&inode->i_mutex); | ||
| 383 | last_index = inode->i_size >> PAGE_CACHE_SHIFT; | ||
| 384 | for (i = 0; i <= last_index; i++) { | ||
| 385 | if (total_read % ra_pages == 0) { | ||
| 386 | btrfs_force_ra(inode->i_mapping, &file->f_ra, file, i, | ||
| 387 | min(last_index, i + ra_pages - 1)); | ||
| 388 | } | ||
| 389 | total_read++; | ||
| 390 | again: | ||
| 391 | page = grab_cache_page(inode->i_mapping, i); | ||
| 392 | if (!page) | ||
| 393 | goto out_unlock; | ||
| 394 | if (!PageUptodate(page)) { | ||
| 395 | btrfs_readpage(NULL, page); | ||
| 396 | lock_page(page); | ||
| 397 | if (!PageUptodate(page)) { | ||
| 398 | unlock_page(page); | ||
| 399 | page_cache_release(page); | ||
| 400 | goto out_unlock; | ||
| 401 | } | ||
| 402 | } | ||
| 403 | |||
| 404 | wait_on_page_writeback(page); | ||
| 405 | |||
| 406 | page_start = (u64)page->index << PAGE_CACHE_SHIFT; | ||
| 407 | page_end = page_start + PAGE_CACHE_SIZE - 1; | ||
| 408 | lock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 409 | |||
| 410 | ordered = btrfs_lookup_ordered_extent(inode, page_start); | ||
| 411 | if (ordered) { | ||
| 412 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 413 | unlock_page(page); | ||
| 414 | page_cache_release(page); | ||
| 415 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 416 | btrfs_put_ordered_extent(ordered); | ||
| 417 | goto again; | ||
| 418 | } | ||
| 419 | set_page_extent_mapped(page); | ||
| 420 | |||
| 421 | /* | ||
| 422 | * this makes sure page_mkwrite is called on the | ||
| 423 | * page if it is dirtied again later | ||
| 424 | */ | ||
| 425 | clear_page_dirty_for_io(page); | ||
| 426 | |||
| 427 | btrfs_set_extent_delalloc(inode, page_start, page_end); | ||
| 428 | |||
| 429 | unlock_extent(io_tree, page_start, page_end, GFP_NOFS); | ||
| 430 | set_page_dirty(page); | ||
| 431 | unlock_page(page); | ||
| 432 | page_cache_release(page); | ||
| 433 | balance_dirty_pages_ratelimited_nr(inode->i_mapping, 1); | ||
| 434 | } | ||
| 435 | |||
| 436 | out_unlock: | ||
| 437 | mutex_unlock(&inode->i_mutex); | ||
| 438 | return 0; | ||
| 439 | } | ||
| 440 | |||
| 441 | /* | ||
| 442 | * Called inside transaction, so use GFP_NOFS | ||
| 443 | */ | ||
| 444 | |||
| 445 | static int btrfs_ioctl_resize(struct btrfs_root *root, void __user *arg) | ||
| 446 | { | ||
| 447 | u64 new_size; | ||
| 448 | u64 old_size; | ||
| 449 | u64 devid = 1; | ||
| 450 | struct btrfs_ioctl_vol_args *vol_args; | ||
| 451 | struct btrfs_trans_handle *trans; | ||
| 452 | struct btrfs_device *device = NULL; | ||
| 453 | char *sizestr; | ||
| 454 | char *devstr = NULL; | ||
| 455 | int ret = 0; | ||
| 456 | int namelen; | ||
| 457 | int mod = 0; | ||
| 458 | |||
| 459 | if (root->fs_info->sb->s_flags & MS_RDONLY) | ||
| 460 | return -EROFS; | ||
| 461 | |||
| 462 | if (!capable(CAP_SYS_ADMIN)) | ||
| 463 | return -EPERM; | ||
| 464 | |||
| 465 | vol_args = kmalloc(sizeof(*vol_args), GFP_NOFS); | ||
| 466 | |||
| 467 | if (!vol_args) | ||
| 468 | return -ENOMEM; | ||
| 469 | |||
| 470 | if (copy_from_user(vol_args, arg, sizeof(*vol_args))) { | ||
| 471 | ret = -EFAULT; | ||
| 472 | goto out; | ||
| 473 | } | ||
| 474 | |||
| 475 | vol_args->name[BTRFS_PATH_NAME_MAX] = '\0'; | ||
| 476 | namelen = strlen(vol_args->name); | ||
| 477 | |||
| 478 | mutex_lock(&root->fs_info->volume_mutex); | ||
| 479 | sizestr = vol_args->name; | ||
| 480 | devstr = strchr(sizestr, ':'); | ||
| 481 | if (devstr) { | ||
| 482 | char *end; | ||
| 483 | sizestr = devstr + 1; | ||
| 484 | *devstr = '\0'; | ||
| 485 | devstr = vol_args->name; | ||
| 486 | devid = simple_strtoull(devstr, &end, 10); | ||
| 487 | printk(KERN_INFO "resizing devid %llu\n", devid); | ||
| 488 | } | ||
| 489 | device = btrfs_find_device(root, devid, NULL, NULL); | ||
| 490 | if (!device) { | ||
| 491 | printk(KERN_INFO "resizer unable to find device %llu\n", devid); | ||
| 492 | ret = -EINVAL; | ||
| 493 | goto out_unlock; | ||
| 494 | } | ||
| 495 | if (!strcmp(sizestr, "max")) | ||
| 496 | new_size = device->bdev->bd_inode->i_size; | ||
| 497 | else { | ||
| 498 | if (sizestr[0] == '-') { | ||
| 499 | mod = -1; | ||
| 500 | sizestr++; | ||
| 501 | } else if (sizestr[0] == '+') { | ||
| 502 | mod = 1; | ||
| 503 | sizestr++; | ||
| 504 | } | ||
| 505 | new_size = btrfs_parse_size(sizestr); | ||
| 506 | if (new_size == 0) { | ||
| 507 | ret = -EINVAL; | ||
| 508 | goto out_unlock; | ||
| 509 | } | ||
| 510 | } | ||
| 511 | |||
| 512 | old_size = device->total_bytes; | ||
| 513 | |||
| 514 | if (mod < 0) { | ||
| 515 | if (new_size > old_size) { | ||
| 516 | ret = -EINVAL; | ||
| 517 | goto out_unlock; | ||
| 518 | } | ||
| 519 | new_size = old_size - new_size; | ||
| 520 | } else if (mod > 0) { | ||
| 521 | new_size = old_size + new_size; | ||
| 522 | } | ||
| 523 | |||
| 524 | if (new_size < 256 * 1024 * 1024) { | ||
| 525 | ret = -EINVAL; | ||
| 526 | goto out_unlock; | ||
| 527 | } | ||
| 528 | if (new_size > device->bdev->bd_inode->i_size) { | ||
| 529 | ret = -EFBIG; | ||
| 530 | goto out_unlock; | ||
| 531 | } | ||
| 532 | |||
| 533 | do_div(new_size, root->sectorsize); | ||
| 534 | new_size *= root->sectorsize; | ||
| 535 | |||
| 536 | printk(KERN_INFO "new size for %s is %llu\n", | ||
| 537 | device->name, (unsigned long long)new_size); | ||
| 538 | |||
| 539 | if (new_size > old_size) { | ||
| 540 | trans = btrfs_start_transaction(root, 1); | ||
| 541 | ret = btrfs_grow_device(trans, device, new_size); | ||
| 542 | btrfs_commit_transaction(trans, root); | ||
| 543 | } else { | ||
| 544 | ret = btrfs_shrink_device(device, new_size); | ||
| 545 | } | ||
| 546 | |||
| 547 | out_unlock: | ||
| 548 | mutex_unlock(&root->fs_info->volume_mutex); | ||
| 549 | out: | ||
| 550 | kfree(vol_args); | ||
| 551 | return ret; | ||
| 552 | } | ||
| 553 | |||
| 554 | static noinline int btrfs_ioctl_snap_create(struct file *file, | ||
| 555 | void __user *arg, int subvol) | ||
| 556 | { | ||
| 557 | struct btrfs_root *root = BTRFS_I(fdentry(file)->d_inode)->root; | ||
| 558 | struct btrfs_ioctl_vol_args *vol_args; | ||
| 559 | struct btrfs_dir_item *di; | ||
| 560 | struct btrfs_path *path; | ||
| 561 | struct file *src_file; | ||
| 562 | u64 root_dirid; | ||
| 563 | int namelen; | ||
| 564 | int ret = 0; | ||
| 565 | |||
| 566 | if (root->fs_info->sb->s_flags & MS_RDONLY) | ||
| 567 | return -EROFS; | ||
| 568 | |||
| 569 | vol_args = kmalloc(sizeof(*vol_args), GFP_NOFS); | ||
| 570 | |||
| 571 | if (!vol_args) | ||
| 572 | return -ENOMEM; | ||
| 573 | |||
| 574 | if (copy_from_user(vol_args, arg, sizeof(*vol_args))) { | ||
| 575 | ret = -EFAULT; | ||
| 576 | goto out; | ||
| 577 | } | ||
| 578 | |||
| 579 | vol_args->name[BTRFS_PATH_NAME_MAX] = '\0'; | ||
| 580 | namelen = strlen(vol_args->name); | ||
| 581 | if (strchr(vol_args->name, '/')) { | ||
| 582 | ret = -EINVAL; | ||
| 583 | goto out; | ||
| 584 | } | ||
| 585 | |||
| 586 | path = btrfs_alloc_path(); | ||
| 587 | if (!path) { | ||
| 588 | ret = -ENOMEM; | ||
| 589 | goto out; | ||
| 590 | } | ||
| 591 | |||
| 592 | root_dirid = root->fs_info->sb->s_root->d_inode->i_ino, | ||
| 593 | di = btrfs_lookup_dir_item(NULL, root->fs_info->tree_root, | ||
| 594 | path, root_dirid, | ||
| 595 | vol_args->name, namelen, 0); | ||
| 596 | btrfs_free_path(path); | ||
| 597 | |||
| 598 | if (di && !IS_ERR(di)) { | ||
| 599 | ret = -EEXIST; | ||
| 600 | goto out; | ||
| 601 | } | ||
| 602 | |||
| 603 | if (IS_ERR(di)) { | ||
| 604 | ret = PTR_ERR(di); | ||
| 605 | goto out; | ||
| 606 | } | ||
| 607 | |||
| 608 | if (subvol) { | ||
| 609 | ret = btrfs_mksubvol(&file->f_path, vol_args->name, | ||
| 610 | file->f_path.dentry->d_inode->i_mode, | ||
| 611 | namelen, NULL); | ||
| 612 | } else { | ||
| 613 | struct inode *src_inode; | ||
| 614 | src_file = fget(vol_args->fd); | ||
| 615 | if (!src_file) { | ||
| 616 | ret = -EINVAL; | ||
| 617 | goto out; | ||
| 618 | } | ||
| 619 | |||
| 620 | src_inode = src_file->f_path.dentry->d_inode; | ||
| 621 | if (src_inode->i_sb != file->f_path.dentry->d_inode->i_sb) { | ||
| 622 | printk(KERN_INFO "btrfs: Snapshot src from " | ||
| 623 | "another FS\n"); | ||
| 624 | ret = -EINVAL; | ||
| 625 | fput(src_file); | ||
| 626 | goto out; | ||
| 627 | } | ||
| 628 | ret = btrfs_mksubvol(&file->f_path, vol_args->name, | ||
| 629 | file->f_path.dentry->d_inode->i_mode, | ||
| 630 | namelen, BTRFS_I(src_inode)->root); | ||
| 631 | fput(src_file); | ||
| 632 | } | ||
| 633 | |||
| 634 | out: | ||
| 635 | kfree(vol_args); | ||
| 636 | return ret; | ||
| 637 | } | ||
| 638 | |||
| 639 | static int btrfs_ioctl_defrag(struct file *file) | ||
| 640 | { | ||
| 641 | struct inode *inode = fdentry(file)->d_inode; | ||
| 642 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 643 | int ret; | ||
| 644 | |||
| 645 | ret = mnt_want_write(file->f_path.mnt); | ||
| 646 | if (ret) | ||
| 647 | return ret; | ||
| 648 | |||
| 649 | switch (inode->i_mode & S_IFMT) { | ||
| 650 | case S_IFDIR: | ||
| 651 | if (!capable(CAP_SYS_ADMIN)) { | ||
| 652 | ret = -EPERM; | ||
| 653 | goto out; | ||
| 654 | } | ||
| 655 | btrfs_defrag_root(root, 0); | ||
| 656 | btrfs_defrag_root(root->fs_info->extent_root, 0); | ||
| 657 | break; | ||
| 658 | case S_IFREG: | ||
| 659 | if (!(file->f_mode & FMODE_WRITE)) { | ||
| 660 | ret = -EINVAL; | ||
| 661 | goto out; | ||
| 662 | } | ||
| 663 | btrfs_defrag_file(file); | ||
| 664 | break; | ||
| 665 | } | ||
| 666 | out: | ||
| 667 | mnt_drop_write(file->f_path.mnt); | ||
| 668 | return ret; | ||
| 669 | } | ||
| 670 | |||
| 671 | static long btrfs_ioctl_add_dev(struct btrfs_root *root, void __user *arg) | ||
| 672 | { | ||
| 673 | struct btrfs_ioctl_vol_args *vol_args; | ||
| 674 | int ret; | ||
| 675 | |||
| 676 | if (!capable(CAP_SYS_ADMIN)) | ||
| 677 | return -EPERM; | ||
| 678 | |||
| 679 | vol_args = kmalloc(sizeof(*vol_args), GFP_NOFS); | ||
| 680 | |||
| 681 | if (!vol_args) | ||
| 682 | return -ENOMEM; | ||
| 683 | |||
| 684 | if (copy_from_user(vol_args, arg, sizeof(*vol_args))) { | ||
| 685 | ret = -EFAULT; | ||
| 686 | goto out; | ||
| 687 | } | ||
| 688 | vol_args->name[BTRFS_PATH_NAME_MAX] = '\0'; | ||
| 689 | ret = btrfs_init_new_device(root, vol_args->name); | ||
| 690 | |||
| 691 | out: | ||
| 692 | kfree(vol_args); | ||
| 693 | return ret; | ||
| 694 | } | ||
| 695 | |||
| 696 | static long btrfs_ioctl_rm_dev(struct btrfs_root *root, void __user *arg) | ||
| 697 | { | ||
| 698 | struct btrfs_ioctl_vol_args *vol_args; | ||
| 699 | int ret; | ||
| 700 | |||
| 701 | if (!capable(CAP_SYS_ADMIN)) | ||
| 702 | return -EPERM; | ||
| 703 | |||
| 704 | if (root->fs_info->sb->s_flags & MS_RDONLY) | ||
| 705 | return -EROFS; | ||
| 706 | |||
| 707 | vol_args = kmalloc(sizeof(*vol_args), GFP_NOFS); | ||
| 708 | |||
| 709 | if (!vol_args) | ||
| 710 | return -ENOMEM; | ||
| 711 | |||
| 712 | if (copy_from_user(vol_args, arg, sizeof(*vol_args))) { | ||
| 713 | ret = -EFAULT; | ||
| 714 | goto out; | ||
| 715 | } | ||
| 716 | vol_args->name[BTRFS_PATH_NAME_MAX] = '\0'; | ||
| 717 | ret = btrfs_rm_device(root, vol_args->name); | ||
| 718 | |||
| 719 | out: | ||
| 720 | kfree(vol_args); | ||
| 721 | return ret; | ||
| 722 | } | ||
| 723 | |||
| 724 | static long btrfs_ioctl_clone(struct file *file, unsigned long srcfd, | ||
| 725 | u64 off, u64 olen, u64 destoff) | ||
| 726 | { | ||
| 727 | struct inode *inode = fdentry(file)->d_inode; | ||
| 728 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 729 | struct file *src_file; | ||
| 730 | struct inode *src; | ||
| 731 | struct btrfs_trans_handle *trans; | ||
| 732 | struct btrfs_path *path; | ||
| 733 | struct extent_buffer *leaf; | ||
| 734 | char *buf; | ||
| 735 | struct btrfs_key key; | ||
| 736 | u32 nritems; | ||
| 737 | int slot; | ||
| 738 | int ret; | ||
| 739 | u64 len = olen; | ||
| 740 | u64 bs = root->fs_info->sb->s_blocksize; | ||
| 741 | u64 hint_byte; | ||
| 742 | |||
| 743 | /* | ||
| 744 | * TODO: | ||
| 745 | * - split compressed inline extents. annoying: we need to | ||
| 746 | * decompress into destination's address_space (the file offset | ||
| 747 | * may change, so source mapping won't do), then recompress (or | ||
| 748 | * otherwise reinsert) a subrange. | ||
| 749 | * - allow ranges within the same file to be cloned (provided | ||
| 750 | * they don't overlap)? | ||
| 751 | */ | ||
| 752 | |||
| 753 | /* the destination must be opened for writing */ | ||
| 754 | if (!(file->f_mode & FMODE_WRITE)) | ||
| 755 | return -EINVAL; | ||
| 756 | |||
| 757 | ret = mnt_want_write(file->f_path.mnt); | ||
| 758 | if (ret) | ||
| 759 | return ret; | ||
| 760 | |||
| 761 | src_file = fget(srcfd); | ||
| 762 | if (!src_file) { | ||
| 763 | ret = -EBADF; | ||
| 764 | goto out_drop_write; | ||
| 765 | } | ||
| 766 | src = src_file->f_dentry->d_inode; | ||
| 767 | |||
| 768 | ret = -EINVAL; | ||
| 769 | if (src == inode) | ||
| 770 | goto out_fput; | ||
| 771 | |||
| 772 | ret = -EISDIR; | ||
| 773 | if (S_ISDIR(src->i_mode) || S_ISDIR(inode->i_mode)) | ||
| 774 | goto out_fput; | ||
| 775 | |||
| 776 | ret = -EXDEV; | ||
| 777 | if (src->i_sb != inode->i_sb || BTRFS_I(src)->root != root) | ||
| 778 | goto out_fput; | ||
| 779 | |||
| 780 | ret = -ENOMEM; | ||
| 781 | buf = vmalloc(btrfs_level_size(root, 0)); | ||
| 782 | if (!buf) | ||
| 783 | goto out_fput; | ||
| 784 | |||
| 785 | path = btrfs_alloc_path(); | ||
| 786 | if (!path) { | ||
| 787 | vfree(buf); | ||
| 788 | goto out_fput; | ||
| 789 | } | ||
| 790 | path->reada = 2; | ||
| 791 | |||
| 792 | if (inode < src) { | ||
| 793 | mutex_lock(&inode->i_mutex); | ||
| 794 | mutex_lock(&src->i_mutex); | ||
| 795 | } else { | ||
| 796 | mutex_lock(&src->i_mutex); | ||
| 797 | mutex_lock(&inode->i_mutex); | ||
| 798 | } | ||
| 799 | |||
| 800 | /* determine range to clone */ | ||
| 801 | ret = -EINVAL; | ||
| 802 | if (off >= src->i_size || off + len > src->i_size) | ||
| 803 | goto out_unlock; | ||
| 804 | if (len == 0) | ||
| 805 | olen = len = src->i_size - off; | ||
| 806 | /* if we extend to eof, continue to block boundary */ | ||
| 807 | if (off + len == src->i_size) | ||
| 808 | len = ((src->i_size + bs-1) & ~(bs-1)) | ||
| 809 | - off; | ||
| 810 | |||
| 811 | /* verify the end result is block aligned */ | ||
| 812 | if ((off & (bs-1)) || | ||
| 813 | ((off + len) & (bs-1))) | ||
| 814 | goto out_unlock; | ||
| 815 | |||
| 816 | /* do any pending delalloc/csum calc on src, one way or | ||
| 817 | another, and lock file content */ | ||
| 818 | while (1) { | ||
| 819 | struct btrfs_ordered_extent *ordered; | ||
| 820 | lock_extent(&BTRFS_I(src)->io_tree, off, off+len, GFP_NOFS); | ||
| 821 | ordered = btrfs_lookup_first_ordered_extent(inode, off+len); | ||
| 822 | if (BTRFS_I(src)->delalloc_bytes == 0 && !ordered) | ||
| 823 | break; | ||
| 824 | unlock_extent(&BTRFS_I(src)->io_tree, off, off+len, GFP_NOFS); | ||
| 825 | if (ordered) | ||
| 826 | btrfs_put_ordered_extent(ordered); | ||
| 827 | btrfs_wait_ordered_range(src, off, off+len); | ||
| 828 | } | ||
| 829 | |||
| 830 | trans = btrfs_start_transaction(root, 1); | ||
| 831 | BUG_ON(!trans); | ||
| 832 | |||
| 833 | /* punch hole in destination first */ | ||
| 834 | btrfs_drop_extents(trans, root, inode, off, off+len, 0, &hint_byte); | ||
| 835 | |||
| 836 | /* clone data */ | ||
| 837 | key.objectid = src->i_ino; | ||
| 838 | key.type = BTRFS_EXTENT_DATA_KEY; | ||
| 839 | key.offset = 0; | ||
| 840 | |||
| 841 | while (1) { | ||
| 842 | /* | ||
| 843 | * note the key will change type as we walk through the | ||
| 844 | * tree. | ||
| 845 | */ | ||
| 846 | ret = btrfs_search_slot(trans, root, &key, path, 0, 0); | ||
| 847 | if (ret < 0) | ||
| 848 | goto out; | ||
| 849 | |||
| 850 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 851 | if (path->slots[0] >= nritems) { | ||
| 852 | ret = btrfs_next_leaf(root, path); | ||
| 853 | if (ret < 0) | ||
| 854 | goto out; | ||
| 855 | if (ret > 0) | ||
| 856 | break; | ||
| 857 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 858 | } | ||
| 859 | leaf = path->nodes[0]; | ||
| 860 | slot = path->slots[0]; | ||
| 861 | |||
| 862 | btrfs_item_key_to_cpu(leaf, &key, slot); | ||
| 863 | if (btrfs_key_type(&key) > BTRFS_EXTENT_DATA_KEY || | ||
| 864 | key.objectid != src->i_ino) | ||
| 865 | break; | ||
| 866 | |||
| 867 | if (btrfs_key_type(&key) == BTRFS_EXTENT_DATA_KEY) { | ||
| 868 | struct btrfs_file_extent_item *extent; | ||
| 869 | int type; | ||
| 870 | u32 size; | ||
| 871 | struct btrfs_key new_key; | ||
| 872 | u64 disko = 0, diskl = 0; | ||
| 873 | u64 datao = 0, datal = 0; | ||
| 874 | u8 comp; | ||
| 875 | |||
| 876 | size = btrfs_item_size_nr(leaf, slot); | ||
| 877 | read_extent_buffer(leaf, buf, | ||
| 878 | btrfs_item_ptr_offset(leaf, slot), | ||
| 879 | size); | ||
| 880 | |||
| 881 | extent = btrfs_item_ptr(leaf, slot, | ||
| 882 | struct btrfs_file_extent_item); | ||
| 883 | comp = btrfs_file_extent_compression(leaf, extent); | ||
| 884 | type = btrfs_file_extent_type(leaf, extent); | ||
| 885 | if (type == BTRFS_FILE_EXTENT_REG) { | ||
| 886 | disko = btrfs_file_extent_disk_bytenr(leaf, | ||
| 887 | extent); | ||
| 888 | diskl = btrfs_file_extent_disk_num_bytes(leaf, | ||
| 889 | extent); | ||
| 890 | datao = btrfs_file_extent_offset(leaf, extent); | ||
| 891 | datal = btrfs_file_extent_num_bytes(leaf, | ||
| 892 | extent); | ||
| 893 | } else if (type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 894 | /* take upper bound, may be compressed */ | ||
| 895 | datal = btrfs_file_extent_ram_bytes(leaf, | ||
| 896 | extent); | ||
| 897 | } | ||
| 898 | btrfs_release_path(root, path); | ||
| 899 | |||
| 900 | if (key.offset + datal < off || | ||
| 901 | key.offset >= off+len) | ||
| 902 | goto next; | ||
| 903 | |||
| 904 | memcpy(&new_key, &key, sizeof(new_key)); | ||
| 905 | new_key.objectid = inode->i_ino; | ||
| 906 | new_key.offset = key.offset + destoff - off; | ||
| 907 | |||
| 908 | if (type == BTRFS_FILE_EXTENT_REG) { | ||
| 909 | ret = btrfs_insert_empty_item(trans, root, path, | ||
| 910 | &new_key, size); | ||
| 911 | if (ret) | ||
| 912 | goto out; | ||
| 913 | |||
| 914 | leaf = path->nodes[0]; | ||
| 915 | slot = path->slots[0]; | ||
| 916 | write_extent_buffer(leaf, buf, | ||
| 917 | btrfs_item_ptr_offset(leaf, slot), | ||
| 918 | size); | ||
| 919 | |||
| 920 | extent = btrfs_item_ptr(leaf, slot, | ||
| 921 | struct btrfs_file_extent_item); | ||
| 922 | |||
| 923 | if (off > key.offset) { | ||
| 924 | datao += off - key.offset; | ||
| 925 | datal -= off - key.offset; | ||
| 926 | } | ||
| 927 | if (key.offset + datao + datal + key.offset > | ||
| 928 | off + len) | ||
| 929 | datal = off + len - key.offset - datao; | ||
| 930 | /* disko == 0 means it's a hole */ | ||
| 931 | if (!disko) | ||
| 932 | datao = 0; | ||
| 933 | |||
| 934 | btrfs_set_file_extent_offset(leaf, extent, | ||
| 935 | datao); | ||
| 936 | btrfs_set_file_extent_num_bytes(leaf, extent, | ||
| 937 | datal); | ||
| 938 | if (disko) { | ||
| 939 | inode_add_bytes(inode, datal); | ||
| 940 | ret = btrfs_inc_extent_ref(trans, root, | ||
| 941 | disko, diskl, leaf->start, | ||
| 942 | root->root_key.objectid, | ||
| 943 | trans->transid, | ||
| 944 | inode->i_ino); | ||
| 945 | BUG_ON(ret); | ||
| 946 | } | ||
| 947 | } else if (type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 948 | u64 skip = 0; | ||
| 949 | u64 trim = 0; | ||
| 950 | if (off > key.offset) { | ||
| 951 | skip = off - key.offset; | ||
| 952 | new_key.offset += skip; | ||
| 953 | } | ||
| 954 | |||
| 955 | if (key.offset + datal > off+len) | ||
| 956 | trim = key.offset + datal - (off+len); | ||
| 957 | |||
| 958 | if (comp && (skip || trim)) { | ||
| 959 | ret = -EINVAL; | ||
| 960 | goto out; | ||
| 961 | } | ||
| 962 | size -= skip + trim; | ||
| 963 | datal -= skip + trim; | ||
| 964 | ret = btrfs_insert_empty_item(trans, root, path, | ||
| 965 | &new_key, size); | ||
| 966 | if (ret) | ||
| 967 | goto out; | ||
| 968 | |||
| 969 | if (skip) { | ||
| 970 | u32 start = | ||
| 971 | btrfs_file_extent_calc_inline_size(0); | ||
| 972 | memmove(buf+start, buf+start+skip, | ||
| 973 | datal); | ||
| 974 | } | ||
| 975 | |||
| 976 | leaf = path->nodes[0]; | ||
| 977 | slot = path->slots[0]; | ||
| 978 | write_extent_buffer(leaf, buf, | ||
| 979 | btrfs_item_ptr_offset(leaf, slot), | ||
| 980 | size); | ||
| 981 | inode_add_bytes(inode, datal); | ||
| 982 | } | ||
| 983 | |||
| 984 | btrfs_mark_buffer_dirty(leaf); | ||
| 985 | } | ||
| 986 | |||
| 987 | next: | ||
| 988 | btrfs_release_path(root, path); | ||
| 989 | key.offset++; | ||
| 990 | } | ||
| 991 | ret = 0; | ||
| 992 | out: | ||
| 993 | btrfs_release_path(root, path); | ||
| 994 | if (ret == 0) { | ||
| 995 | inode->i_mtime = inode->i_ctime = CURRENT_TIME; | ||
| 996 | if (destoff + olen > inode->i_size) | ||
| 997 | btrfs_i_size_write(inode, destoff + olen); | ||
| 998 | BTRFS_I(inode)->flags = BTRFS_I(src)->flags; | ||
| 999 | ret = btrfs_update_inode(trans, root, inode); | ||
| 1000 | } | ||
| 1001 | btrfs_end_transaction(trans, root); | ||
| 1002 | unlock_extent(&BTRFS_I(src)->io_tree, off, off+len, GFP_NOFS); | ||
| 1003 | if (ret) | ||
| 1004 | vmtruncate(inode, 0); | ||
| 1005 | out_unlock: | ||
| 1006 | mutex_unlock(&src->i_mutex); | ||
| 1007 | mutex_unlock(&inode->i_mutex); | ||
| 1008 | vfree(buf); | ||
| 1009 | btrfs_free_path(path); | ||
| 1010 | out_fput: | ||
| 1011 | fput(src_file); | ||
| 1012 | out_drop_write: | ||
| 1013 | mnt_drop_write(file->f_path.mnt); | ||
| 1014 | return ret; | ||
| 1015 | } | ||
| 1016 | |||
| 1017 | static long btrfs_ioctl_clone_range(struct file *file, void __user *argp) | ||
| 1018 | { | ||
| 1019 | struct btrfs_ioctl_clone_range_args args; | ||
| 1020 | |||
| 1021 | if (copy_from_user(&args, argp, sizeof(args))) | ||
| 1022 | return -EFAULT; | ||
| 1023 | return btrfs_ioctl_clone(file, args.src_fd, args.src_offset, | ||
| 1024 | args.src_length, args.dest_offset); | ||
| 1025 | } | ||
| 1026 | |||
| 1027 | /* | ||
| 1028 | * there are many ways the trans_start and trans_end ioctls can lead | ||
| 1029 | * to deadlocks. They should only be used by applications that | ||
| 1030 | * basically own the machine, and have a very in depth understanding | ||
| 1031 | * of all the possible deadlocks and enospc problems. | ||
| 1032 | */ | ||
| 1033 | static long btrfs_ioctl_trans_start(struct file *file) | ||
| 1034 | { | ||
| 1035 | struct inode *inode = fdentry(file)->d_inode; | ||
| 1036 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1037 | struct btrfs_trans_handle *trans; | ||
| 1038 | int ret = 0; | ||
| 1039 | |||
| 1040 | if (!capable(CAP_SYS_ADMIN)) | ||
| 1041 | return -EPERM; | ||
| 1042 | |||
| 1043 | if (file->private_data) { | ||
| 1044 | ret = -EINPROGRESS; | ||
| 1045 | goto out; | ||
| 1046 | } | ||
| 1047 | |||
| 1048 | ret = mnt_want_write(file->f_path.mnt); | ||
| 1049 | if (ret) | ||
| 1050 | goto out; | ||
| 1051 | |||
| 1052 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 1053 | root->fs_info->open_ioctl_trans++; | ||
| 1054 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1055 | |||
| 1056 | trans = btrfs_start_ioctl_transaction(root, 0); | ||
| 1057 | if (trans) | ||
| 1058 | file->private_data = trans; | ||
| 1059 | else | ||
| 1060 | ret = -ENOMEM; | ||
| 1061 | /*printk(KERN_INFO "btrfs_ioctl_trans_start on %p\n", file);*/ | ||
| 1062 | out: | ||
| 1063 | return ret; | ||
| 1064 | } | ||
| 1065 | |||
| 1066 | /* | ||
| 1067 | * there are many ways the trans_start and trans_end ioctls can lead | ||
| 1068 | * to deadlocks. They should only be used by applications that | ||
| 1069 | * basically own the machine, and have a very in depth understanding | ||
| 1070 | * of all the possible deadlocks and enospc problems. | ||
| 1071 | */ | ||
| 1072 | long btrfs_ioctl_trans_end(struct file *file) | ||
| 1073 | { | ||
| 1074 | struct inode *inode = fdentry(file)->d_inode; | ||
| 1075 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 1076 | struct btrfs_trans_handle *trans; | ||
| 1077 | int ret = 0; | ||
| 1078 | |||
| 1079 | trans = file->private_data; | ||
| 1080 | if (!trans) { | ||
| 1081 | ret = -EINVAL; | ||
| 1082 | goto out; | ||
| 1083 | } | ||
| 1084 | btrfs_end_transaction(trans, root); | ||
| 1085 | file->private_data = NULL; | ||
| 1086 | |||
| 1087 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 1088 | root->fs_info->open_ioctl_trans--; | ||
| 1089 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1090 | |||
| 1091 | mnt_drop_write(file->f_path.mnt); | ||
| 1092 | |||
| 1093 | out: | ||
| 1094 | return ret; | ||
| 1095 | } | ||
| 1096 | |||
| 1097 | long btrfs_ioctl(struct file *file, unsigned int | ||
| 1098 | cmd, unsigned long arg) | ||
| 1099 | { | ||
| 1100 | struct btrfs_root *root = BTRFS_I(fdentry(file)->d_inode)->root; | ||
| 1101 | void __user *argp = (void __user *)arg; | ||
| 1102 | |||
| 1103 | switch (cmd) { | ||
| 1104 | case BTRFS_IOC_SNAP_CREATE: | ||
| 1105 | return btrfs_ioctl_snap_create(file, argp, 0); | ||
| 1106 | case BTRFS_IOC_SUBVOL_CREATE: | ||
| 1107 | return btrfs_ioctl_snap_create(file, argp, 1); | ||
| 1108 | case BTRFS_IOC_DEFRAG: | ||
| 1109 | return btrfs_ioctl_defrag(file); | ||
| 1110 | case BTRFS_IOC_RESIZE: | ||
| 1111 | return btrfs_ioctl_resize(root, argp); | ||
| 1112 | case BTRFS_IOC_ADD_DEV: | ||
| 1113 | return btrfs_ioctl_add_dev(root, argp); | ||
| 1114 | case BTRFS_IOC_RM_DEV: | ||
| 1115 | return btrfs_ioctl_rm_dev(root, argp); | ||
| 1116 | case BTRFS_IOC_BALANCE: | ||
| 1117 | return btrfs_balance(root->fs_info->dev_root); | ||
| 1118 | case BTRFS_IOC_CLONE: | ||
| 1119 | return btrfs_ioctl_clone(file, arg, 0, 0, 0); | ||
| 1120 | case BTRFS_IOC_CLONE_RANGE: | ||
| 1121 | return btrfs_ioctl_clone_range(file, argp); | ||
| 1122 | case BTRFS_IOC_TRANS_START: | ||
| 1123 | return btrfs_ioctl_trans_start(file); | ||
| 1124 | case BTRFS_IOC_TRANS_END: | ||
| 1125 | return btrfs_ioctl_trans_end(file); | ||
| 1126 | case BTRFS_IOC_SYNC: | ||
| 1127 | btrfs_sync_fs(file->f_dentry->d_sb, 1); | ||
| 1128 | return 0; | ||
| 1129 | } | ||
| 1130 | |||
| 1131 | return -ENOTTY; | ||
| 1132 | } | ||
diff --git a/fs/btrfs/ioctl.h b/fs/btrfs/ioctl.h new file mode 100644 index 000000000000..78049ea208db --- /dev/null +++ b/fs/btrfs/ioctl.h | |||
| @@ -0,0 +1,67 @@ | |||
| 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 | |||
| 19 | #ifndef __IOCTL_ | ||
| 20 | #define __IOCTL_ | ||
| 21 | #include <linux/ioctl.h> | ||
| 22 | |||
| 23 | #define BTRFS_IOCTL_MAGIC 0x94 | ||
| 24 | #define BTRFS_VOL_NAME_MAX 255 | ||
| 25 | #define BTRFS_PATH_NAME_MAX 3072 | ||
| 26 | |||
| 27 | struct btrfs_ioctl_vol_args { | ||
| 28 | __s64 fd; | ||
| 29 | char name[BTRFS_PATH_NAME_MAX + 1]; | ||
| 30 | }; | ||
| 31 | |||
| 32 | #define BTRFS_IOC_SNAP_CREATE _IOW(BTRFS_IOCTL_MAGIC, 1, \ | ||
| 33 | struct btrfs_ioctl_vol_args) | ||
| 34 | #define BTRFS_IOC_DEFRAG _IOW(BTRFS_IOCTL_MAGIC, 2, \ | ||
| 35 | struct btrfs_ioctl_vol_args) | ||
| 36 | #define BTRFS_IOC_RESIZE _IOW(BTRFS_IOCTL_MAGIC, 3, \ | ||
| 37 | struct btrfs_ioctl_vol_args) | ||
| 38 | #define BTRFS_IOC_SCAN_DEV _IOW(BTRFS_IOCTL_MAGIC, 4, \ | ||
| 39 | struct btrfs_ioctl_vol_args) | ||
| 40 | /* trans start and trans end are dangerous, and only for | ||
| 41 | * use by applications that know how to avoid the | ||
| 42 | * resulting deadlocks | ||
| 43 | */ | ||
| 44 | #define BTRFS_IOC_TRANS_START _IO(BTRFS_IOCTL_MAGIC, 6) | ||
| 45 | #define BTRFS_IOC_TRANS_END _IO(BTRFS_IOCTL_MAGIC, 7) | ||
| 46 | #define BTRFS_IOC_SYNC _IO(BTRFS_IOCTL_MAGIC, 8) | ||
| 47 | |||
| 48 | #define BTRFS_IOC_CLONE _IOW(BTRFS_IOCTL_MAGIC, 9, int) | ||
| 49 | #define BTRFS_IOC_ADD_DEV _IOW(BTRFS_IOCTL_MAGIC, 10, \ | ||
| 50 | struct btrfs_ioctl_vol_args) | ||
| 51 | #define BTRFS_IOC_RM_DEV _IOW(BTRFS_IOCTL_MAGIC, 11, \ | ||
| 52 | struct btrfs_ioctl_vol_args) | ||
| 53 | #define BTRFS_IOC_BALANCE _IOW(BTRFS_IOCTL_MAGIC, 12, \ | ||
| 54 | struct btrfs_ioctl_vol_args) | ||
| 55 | struct btrfs_ioctl_clone_range_args { | ||
| 56 | __s64 src_fd; | ||
| 57 | __u64 src_offset, src_length; | ||
| 58 | __u64 dest_offset; | ||
| 59 | }; | ||
| 60 | |||
| 61 | #define BTRFS_IOC_CLONE_RANGE _IOW(BTRFS_IOCTL_MAGIC, 13, \ | ||
| 62 | struct btrfs_ioctl_clone_range_args) | ||
| 63 | |||
| 64 | #define BTRFS_IOC_SUBVOL_CREATE _IOW(BTRFS_IOCTL_MAGIC, 14, \ | ||
| 65 | struct btrfs_ioctl_vol_args) | ||
| 66 | |||
| 67 | #endif | ||
diff --git a/fs/btrfs/locking.c b/fs/btrfs/locking.c new file mode 100644 index 000000000000..39bae7761db6 --- /dev/null +++ b/fs/btrfs/locking.c | |||
| @@ -0,0 +1,88 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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/sched.h> | ||
| 19 | #include <linux/gfp.h> | ||
| 20 | #include <linux/pagemap.h> | ||
| 21 | #include <linux/spinlock.h> | ||
| 22 | #include <linux/page-flags.h> | ||
| 23 | #include <asm/bug.h> | ||
| 24 | #include "ctree.h" | ||
| 25 | #include "extent_io.h" | ||
| 26 | #include "locking.h" | ||
| 27 | |||
| 28 | /* | ||
| 29 | * locks the per buffer mutex in an extent buffer. This uses adaptive locks | ||
| 30 | * and the spin is not tuned very extensively. The spinning does make a big | ||
| 31 | * difference in almost every workload, but spinning for the right amount of | ||
| 32 | * time needs some help. | ||
| 33 | * | ||
| 34 | * In general, we want to spin as long as the lock holder is doing btree | ||
| 35 | * searches, and we should give up if they are in more expensive code. | ||
| 36 | */ | ||
| 37 | |||
| 38 | int btrfs_tree_lock(struct extent_buffer *eb) | ||
| 39 | { | ||
| 40 | int i; | ||
| 41 | |||
| 42 | if (mutex_trylock(&eb->mutex)) | ||
| 43 | return 0; | ||
| 44 | for (i = 0; i < 512; i++) { | ||
| 45 | cpu_relax(); | ||
| 46 | if (mutex_trylock(&eb->mutex)) | ||
| 47 | return 0; | ||
| 48 | } | ||
| 49 | cpu_relax(); | ||
| 50 | mutex_lock_nested(&eb->mutex, BTRFS_MAX_LEVEL - btrfs_header_level(eb)); | ||
| 51 | return 0; | ||
| 52 | } | ||
| 53 | |||
| 54 | int btrfs_try_tree_lock(struct extent_buffer *eb) | ||
| 55 | { | ||
| 56 | return mutex_trylock(&eb->mutex); | ||
| 57 | } | ||
| 58 | |||
| 59 | int btrfs_tree_unlock(struct extent_buffer *eb) | ||
| 60 | { | ||
| 61 | mutex_unlock(&eb->mutex); | ||
| 62 | return 0; | ||
| 63 | } | ||
| 64 | |||
| 65 | int btrfs_tree_locked(struct extent_buffer *eb) | ||
| 66 | { | ||
| 67 | return mutex_is_locked(&eb->mutex); | ||
| 68 | } | ||
| 69 | |||
| 70 | /* | ||
| 71 | * btrfs_search_slot uses this to decide if it should drop its locks | ||
| 72 | * before doing something expensive like allocating free blocks for cow. | ||
| 73 | */ | ||
| 74 | int btrfs_path_lock_waiting(struct btrfs_path *path, int level) | ||
| 75 | { | ||
| 76 | int i; | ||
| 77 | struct extent_buffer *eb; | ||
| 78 | for (i = level; i <= level + 1 && i < BTRFS_MAX_LEVEL; i++) { | ||
| 79 | eb = path->nodes[i]; | ||
| 80 | if (!eb) | ||
| 81 | break; | ||
| 82 | smp_mb(); | ||
| 83 | if (!list_empty(&eb->mutex.wait_list)) | ||
| 84 | return 1; | ||
| 85 | } | ||
| 86 | return 0; | ||
| 87 | } | ||
| 88 | |||
diff --git a/fs/btrfs/locking.h b/fs/btrfs/locking.h new file mode 100644 index 000000000000..bc1faef12519 --- /dev/null +++ b/fs/btrfs/locking.h | |||
| @@ -0,0 +1,27 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | |||
| 19 | #ifndef __BTRFS_LOCKING_ | ||
| 20 | #define __BTRFS_LOCKING_ | ||
| 21 | |||
| 22 | int btrfs_tree_lock(struct extent_buffer *eb); | ||
| 23 | int btrfs_tree_unlock(struct extent_buffer *eb); | ||
| 24 | int btrfs_tree_locked(struct extent_buffer *eb); | ||
| 25 | int btrfs_try_tree_lock(struct extent_buffer *eb); | ||
| 26 | int btrfs_path_lock_waiting(struct btrfs_path *path, int level); | ||
| 27 | #endif | ||
diff --git a/fs/btrfs/ordered-data.c b/fs/btrfs/ordered-data.c new file mode 100644 index 000000000000..a20940170274 --- /dev/null +++ b/fs/btrfs/ordered-data.c | |||
| @@ -0,0 +1,730 @@ | |||
| 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 | |||
| 19 | #include <linux/gfp.h> | ||
| 20 | #include <linux/slab.h> | ||
| 21 | #include <linux/blkdev.h> | ||
| 22 | #include <linux/writeback.h> | ||
| 23 | #include <linux/pagevec.h> | ||
| 24 | #include "ctree.h" | ||
| 25 | #include "transaction.h" | ||
| 26 | #include "btrfs_inode.h" | ||
| 27 | #include "extent_io.h" | ||
| 28 | |||
| 29 | static u64 entry_end(struct btrfs_ordered_extent *entry) | ||
| 30 | { | ||
| 31 | if (entry->file_offset + entry->len < entry->file_offset) | ||
| 32 | return (u64)-1; | ||
| 33 | return entry->file_offset + entry->len; | ||
| 34 | } | ||
| 35 | |||
| 36 | /* returns NULL if the insertion worked, or it returns the node it did find | ||
| 37 | * in the tree | ||
| 38 | */ | ||
| 39 | static struct rb_node *tree_insert(struct rb_root *root, u64 file_offset, | ||
| 40 | struct rb_node *node) | ||
| 41 | { | ||
| 42 | struct rb_node **p = &root->rb_node; | ||
| 43 | struct rb_node *parent = NULL; | ||
| 44 | struct btrfs_ordered_extent *entry; | ||
| 45 | |||
| 46 | while (*p) { | ||
| 47 | parent = *p; | ||
| 48 | entry = rb_entry(parent, struct btrfs_ordered_extent, rb_node); | ||
| 49 | |||
| 50 | if (file_offset < entry->file_offset) | ||
| 51 | p = &(*p)->rb_left; | ||
| 52 | else if (file_offset >= entry_end(entry)) | ||
| 53 | p = &(*p)->rb_right; | ||
| 54 | else | ||
| 55 | return parent; | ||
| 56 | } | ||
| 57 | |||
| 58 | rb_link_node(node, parent, p); | ||
| 59 | rb_insert_color(node, root); | ||
| 60 | return NULL; | ||
| 61 | } | ||
| 62 | |||
| 63 | /* | ||
| 64 | * look for a given offset in the tree, and if it can't be found return the | ||
| 65 | * first lesser offset | ||
| 66 | */ | ||
| 67 | static struct rb_node *__tree_search(struct rb_root *root, u64 file_offset, | ||
| 68 | struct rb_node **prev_ret) | ||
| 69 | { | ||
| 70 | struct rb_node *n = root->rb_node; | ||
| 71 | struct rb_node *prev = NULL; | ||
| 72 | struct rb_node *test; | ||
| 73 | struct btrfs_ordered_extent *entry; | ||
| 74 | struct btrfs_ordered_extent *prev_entry = NULL; | ||
| 75 | |||
| 76 | while (n) { | ||
| 77 | entry = rb_entry(n, struct btrfs_ordered_extent, rb_node); | ||
| 78 | prev = n; | ||
| 79 | prev_entry = entry; | ||
| 80 | |||
| 81 | if (file_offset < entry->file_offset) | ||
| 82 | n = n->rb_left; | ||
| 83 | else if (file_offset >= entry_end(entry)) | ||
| 84 | n = n->rb_right; | ||
| 85 | else | ||
| 86 | return n; | ||
| 87 | } | ||
| 88 | if (!prev_ret) | ||
| 89 | return NULL; | ||
| 90 | |||
| 91 | while (prev && file_offset >= entry_end(prev_entry)) { | ||
| 92 | test = rb_next(prev); | ||
| 93 | if (!test) | ||
| 94 | break; | ||
| 95 | prev_entry = rb_entry(test, struct btrfs_ordered_extent, | ||
| 96 | rb_node); | ||
| 97 | if (file_offset < entry_end(prev_entry)) | ||
| 98 | break; | ||
| 99 | |||
| 100 | prev = test; | ||
| 101 | } | ||
| 102 | if (prev) | ||
| 103 | prev_entry = rb_entry(prev, struct btrfs_ordered_extent, | ||
| 104 | rb_node); | ||
| 105 | while (prev && file_offset < entry_end(prev_entry)) { | ||
| 106 | test = rb_prev(prev); | ||
| 107 | if (!test) | ||
| 108 | break; | ||
| 109 | prev_entry = rb_entry(test, struct btrfs_ordered_extent, | ||
| 110 | rb_node); | ||
| 111 | prev = test; | ||
| 112 | } | ||
| 113 | *prev_ret = prev; | ||
| 114 | return NULL; | ||
| 115 | } | ||
| 116 | |||
| 117 | /* | ||
| 118 | * helper to check if a given offset is inside a given entry | ||
| 119 | */ | ||
| 120 | static int offset_in_entry(struct btrfs_ordered_extent *entry, u64 file_offset) | ||
| 121 | { | ||
| 122 | if (file_offset < entry->file_offset || | ||
| 123 | entry->file_offset + entry->len <= file_offset) | ||
| 124 | return 0; | ||
| 125 | return 1; | ||
| 126 | } | ||
| 127 | |||
| 128 | /* | ||
| 129 | * look find the first ordered struct that has this offset, otherwise | ||
| 130 | * the first one less than this offset | ||
| 131 | */ | ||
| 132 | static inline struct rb_node *tree_search(struct btrfs_ordered_inode_tree *tree, | ||
| 133 | u64 file_offset) | ||
| 134 | { | ||
| 135 | struct rb_root *root = &tree->tree; | ||
| 136 | struct rb_node *prev; | ||
| 137 | struct rb_node *ret; | ||
| 138 | struct btrfs_ordered_extent *entry; | ||
| 139 | |||
| 140 | if (tree->last) { | ||
| 141 | entry = rb_entry(tree->last, struct btrfs_ordered_extent, | ||
| 142 | rb_node); | ||
| 143 | if (offset_in_entry(entry, file_offset)) | ||
| 144 | return tree->last; | ||
| 145 | } | ||
| 146 | ret = __tree_search(root, file_offset, &prev); | ||
| 147 | if (!ret) | ||
| 148 | ret = prev; | ||
| 149 | if (ret) | ||
| 150 | tree->last = ret; | ||
| 151 | return ret; | ||
| 152 | } | ||
| 153 | |||
| 154 | /* allocate and add a new ordered_extent into the per-inode tree. | ||
| 155 | * file_offset is the logical offset in the file | ||
| 156 | * | ||
| 157 | * start is the disk block number of an extent already reserved in the | ||
| 158 | * extent allocation tree | ||
| 159 | * | ||
| 160 | * len is the length of the extent | ||
| 161 | * | ||
| 162 | * This also sets the EXTENT_ORDERED bit on the range in the inode. | ||
| 163 | * | ||
| 164 | * The tree is given a single reference on the ordered extent that was | ||
| 165 | * inserted. | ||
| 166 | */ | ||
| 167 | int btrfs_add_ordered_extent(struct inode *inode, u64 file_offset, | ||
| 168 | u64 start, u64 len, u64 disk_len, int type) | ||
| 169 | { | ||
| 170 | struct btrfs_ordered_inode_tree *tree; | ||
| 171 | struct rb_node *node; | ||
| 172 | struct btrfs_ordered_extent *entry; | ||
| 173 | |||
| 174 | tree = &BTRFS_I(inode)->ordered_tree; | ||
| 175 | entry = kzalloc(sizeof(*entry), GFP_NOFS); | ||
| 176 | if (!entry) | ||
| 177 | return -ENOMEM; | ||
| 178 | |||
| 179 | mutex_lock(&tree->mutex); | ||
| 180 | entry->file_offset = file_offset; | ||
| 181 | entry->start = start; | ||
| 182 | entry->len = len; | ||
| 183 | entry->disk_len = disk_len; | ||
| 184 | entry->inode = inode; | ||
| 185 | if (type != BTRFS_ORDERED_IO_DONE && type != BTRFS_ORDERED_COMPLETE) | ||
| 186 | set_bit(type, &entry->flags); | ||
| 187 | |||
| 188 | /* one ref for the tree */ | ||
| 189 | atomic_set(&entry->refs, 1); | ||
| 190 | init_waitqueue_head(&entry->wait); | ||
| 191 | INIT_LIST_HEAD(&entry->list); | ||
| 192 | INIT_LIST_HEAD(&entry->root_extent_list); | ||
| 193 | |||
| 194 | node = tree_insert(&tree->tree, file_offset, | ||
| 195 | &entry->rb_node); | ||
| 196 | BUG_ON(node); | ||
| 197 | |||
| 198 | set_extent_ordered(&BTRFS_I(inode)->io_tree, file_offset, | ||
| 199 | entry_end(entry) - 1, GFP_NOFS); | ||
| 200 | |||
| 201 | spin_lock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock); | ||
| 202 | list_add_tail(&entry->root_extent_list, | ||
| 203 | &BTRFS_I(inode)->root->fs_info->ordered_extents); | ||
| 204 | spin_unlock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock); | ||
| 205 | |||
| 206 | mutex_unlock(&tree->mutex); | ||
| 207 | BUG_ON(node); | ||
| 208 | return 0; | ||
| 209 | } | ||
| 210 | |||
| 211 | /* | ||
| 212 | * Add a struct btrfs_ordered_sum into the list of checksums to be inserted | ||
| 213 | * when an ordered extent is finished. If the list covers more than one | ||
| 214 | * ordered extent, it is split across multiples. | ||
| 215 | */ | ||
| 216 | int btrfs_add_ordered_sum(struct inode *inode, | ||
| 217 | struct btrfs_ordered_extent *entry, | ||
| 218 | struct btrfs_ordered_sum *sum) | ||
| 219 | { | ||
| 220 | struct btrfs_ordered_inode_tree *tree; | ||
| 221 | |||
| 222 | tree = &BTRFS_I(inode)->ordered_tree; | ||
| 223 | mutex_lock(&tree->mutex); | ||
| 224 | list_add_tail(&sum->list, &entry->list); | ||
| 225 | mutex_unlock(&tree->mutex); | ||
| 226 | return 0; | ||
| 227 | } | ||
| 228 | |||
| 229 | /* | ||
| 230 | * this is used to account for finished IO across a given range | ||
| 231 | * of the file. The IO should not span ordered extents. If | ||
| 232 | * a given ordered_extent is completely done, 1 is returned, otherwise | ||
| 233 | * 0. | ||
| 234 | * | ||
| 235 | * test_and_set_bit on a flag in the struct btrfs_ordered_extent is used | ||
| 236 | * to make sure this function only returns 1 once for a given ordered extent. | ||
| 237 | */ | ||
| 238 | int btrfs_dec_test_ordered_pending(struct inode *inode, | ||
| 239 | u64 file_offset, u64 io_size) | ||
| 240 | { | ||
| 241 | struct btrfs_ordered_inode_tree *tree; | ||
| 242 | struct rb_node *node; | ||
| 243 | struct btrfs_ordered_extent *entry; | ||
| 244 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 245 | int ret; | ||
| 246 | |||
| 247 | tree = &BTRFS_I(inode)->ordered_tree; | ||
| 248 | mutex_lock(&tree->mutex); | ||
| 249 | clear_extent_ordered(io_tree, file_offset, file_offset + io_size - 1, | ||
| 250 | GFP_NOFS); | ||
| 251 | node = tree_search(tree, file_offset); | ||
| 252 | if (!node) { | ||
| 253 | ret = 1; | ||
| 254 | goto out; | ||
| 255 | } | ||
| 256 | |||
| 257 | entry = rb_entry(node, struct btrfs_ordered_extent, rb_node); | ||
| 258 | if (!offset_in_entry(entry, file_offset)) { | ||
| 259 | ret = 1; | ||
| 260 | goto out; | ||
| 261 | } | ||
| 262 | |||
| 263 | ret = test_range_bit(io_tree, entry->file_offset, | ||
| 264 | entry->file_offset + entry->len - 1, | ||
| 265 | EXTENT_ORDERED, 0); | ||
| 266 | if (ret == 0) | ||
| 267 | ret = test_and_set_bit(BTRFS_ORDERED_IO_DONE, &entry->flags); | ||
| 268 | out: | ||
| 269 | mutex_unlock(&tree->mutex); | ||
| 270 | return ret == 0; | ||
| 271 | } | ||
| 272 | |||
| 273 | /* | ||
| 274 | * used to drop a reference on an ordered extent. This will free | ||
| 275 | * the extent if the last reference is dropped | ||
| 276 | */ | ||
| 277 | int btrfs_put_ordered_extent(struct btrfs_ordered_extent *entry) | ||
| 278 | { | ||
| 279 | struct list_head *cur; | ||
| 280 | struct btrfs_ordered_sum *sum; | ||
| 281 | |||
| 282 | if (atomic_dec_and_test(&entry->refs)) { | ||
| 283 | while (!list_empty(&entry->list)) { | ||
| 284 | cur = entry->list.next; | ||
| 285 | sum = list_entry(cur, struct btrfs_ordered_sum, list); | ||
| 286 | list_del(&sum->list); | ||
| 287 | kfree(sum); | ||
| 288 | } | ||
| 289 | kfree(entry); | ||
| 290 | } | ||
| 291 | return 0; | ||
| 292 | } | ||
| 293 | |||
| 294 | /* | ||
| 295 | * remove an ordered extent from the tree. No references are dropped | ||
| 296 | * but, anyone waiting on this extent is woken up. | ||
| 297 | */ | ||
| 298 | int btrfs_remove_ordered_extent(struct inode *inode, | ||
| 299 | struct btrfs_ordered_extent *entry) | ||
| 300 | { | ||
| 301 | struct btrfs_ordered_inode_tree *tree; | ||
| 302 | struct rb_node *node; | ||
| 303 | |||
| 304 | tree = &BTRFS_I(inode)->ordered_tree; | ||
| 305 | mutex_lock(&tree->mutex); | ||
| 306 | node = &entry->rb_node; | ||
| 307 | rb_erase(node, &tree->tree); | ||
| 308 | tree->last = NULL; | ||
| 309 | set_bit(BTRFS_ORDERED_COMPLETE, &entry->flags); | ||
| 310 | |||
| 311 | spin_lock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock); | ||
| 312 | list_del_init(&entry->root_extent_list); | ||
| 313 | spin_unlock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock); | ||
| 314 | |||
| 315 | mutex_unlock(&tree->mutex); | ||
| 316 | wake_up(&entry->wait); | ||
| 317 | return 0; | ||
| 318 | } | ||
| 319 | |||
| 320 | /* | ||
| 321 | * wait for all the ordered extents in a root. This is done when balancing | ||
| 322 | * space between drives. | ||
| 323 | */ | ||
| 324 | int btrfs_wait_ordered_extents(struct btrfs_root *root, int nocow_only) | ||
| 325 | { | ||
| 326 | struct list_head splice; | ||
| 327 | struct list_head *cur; | ||
| 328 | struct btrfs_ordered_extent *ordered; | ||
| 329 | struct inode *inode; | ||
| 330 | |||
| 331 | INIT_LIST_HEAD(&splice); | ||
| 332 | |||
| 333 | spin_lock(&root->fs_info->ordered_extent_lock); | ||
| 334 | list_splice_init(&root->fs_info->ordered_extents, &splice); | ||
| 335 | while (!list_empty(&splice)) { | ||
| 336 | cur = splice.next; | ||
| 337 | ordered = list_entry(cur, struct btrfs_ordered_extent, | ||
| 338 | root_extent_list); | ||
| 339 | if (nocow_only && | ||
| 340 | !test_bit(BTRFS_ORDERED_NOCOW, &ordered->flags) && | ||
| 341 | !test_bit(BTRFS_ORDERED_PREALLOC, &ordered->flags)) { | ||
| 342 | list_move(&ordered->root_extent_list, | ||
| 343 | &root->fs_info->ordered_extents); | ||
| 344 | cond_resched_lock(&root->fs_info->ordered_extent_lock); | ||
| 345 | continue; | ||
| 346 | } | ||
| 347 | |||
| 348 | list_del_init(&ordered->root_extent_list); | ||
| 349 | atomic_inc(&ordered->refs); | ||
| 350 | |||
| 351 | /* | ||
| 352 | * the inode may be getting freed (in sys_unlink path). | ||
| 353 | */ | ||
| 354 | inode = igrab(ordered->inode); | ||
| 355 | |||
| 356 | spin_unlock(&root->fs_info->ordered_extent_lock); | ||
| 357 | |||
| 358 | if (inode) { | ||
| 359 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 360 | btrfs_put_ordered_extent(ordered); | ||
| 361 | iput(inode); | ||
| 362 | } else { | ||
| 363 | btrfs_put_ordered_extent(ordered); | ||
| 364 | } | ||
| 365 | |||
| 366 | spin_lock(&root->fs_info->ordered_extent_lock); | ||
| 367 | } | ||
| 368 | spin_unlock(&root->fs_info->ordered_extent_lock); | ||
| 369 | return 0; | ||
| 370 | } | ||
| 371 | |||
| 372 | /* | ||
| 373 | * Used to start IO or wait for a given ordered extent to finish. | ||
| 374 | * | ||
| 375 | * If wait is one, this effectively waits on page writeback for all the pages | ||
| 376 | * in the extent, and it waits on the io completion code to insert | ||
| 377 | * metadata into the btree corresponding to the extent | ||
| 378 | */ | ||
| 379 | void btrfs_start_ordered_extent(struct inode *inode, | ||
| 380 | struct btrfs_ordered_extent *entry, | ||
| 381 | int wait) | ||
| 382 | { | ||
| 383 | u64 start = entry->file_offset; | ||
| 384 | u64 end = start + entry->len - 1; | ||
| 385 | |||
| 386 | /* | ||
| 387 | * pages in the range can be dirty, clean or writeback. We | ||
| 388 | * start IO on any dirty ones so the wait doesn't stall waiting | ||
| 389 | * for pdflush to find them | ||
| 390 | */ | ||
| 391 | btrfs_fdatawrite_range(inode->i_mapping, start, end, WB_SYNC_ALL); | ||
| 392 | if (wait) { | ||
| 393 | wait_event(entry->wait, test_bit(BTRFS_ORDERED_COMPLETE, | ||
| 394 | &entry->flags)); | ||
| 395 | } | ||
| 396 | } | ||
| 397 | |||
| 398 | /* | ||
| 399 | * Used to wait on ordered extents across a large range of bytes. | ||
| 400 | */ | ||
| 401 | int btrfs_wait_ordered_range(struct inode *inode, u64 start, u64 len) | ||
| 402 | { | ||
| 403 | u64 end; | ||
| 404 | u64 orig_end; | ||
| 405 | u64 wait_end; | ||
| 406 | struct btrfs_ordered_extent *ordered; | ||
| 407 | |||
| 408 | if (start + len < start) { | ||
| 409 | orig_end = INT_LIMIT(loff_t); | ||
| 410 | } else { | ||
| 411 | orig_end = start + len - 1; | ||
| 412 | if (orig_end > INT_LIMIT(loff_t)) | ||
| 413 | orig_end = INT_LIMIT(loff_t); | ||
| 414 | } | ||
| 415 | wait_end = orig_end; | ||
| 416 | again: | ||
| 417 | /* start IO across the range first to instantiate any delalloc | ||
| 418 | * extents | ||
| 419 | */ | ||
| 420 | btrfs_fdatawrite_range(inode->i_mapping, start, orig_end, WB_SYNC_NONE); | ||
| 421 | |||
| 422 | /* The compression code will leave pages locked but return from | ||
| 423 | * writepage without setting the page writeback. Starting again | ||
| 424 | * with WB_SYNC_ALL will end up waiting for the IO to actually start. | ||
| 425 | */ | ||
| 426 | btrfs_fdatawrite_range(inode->i_mapping, start, orig_end, WB_SYNC_ALL); | ||
| 427 | |||
| 428 | btrfs_wait_on_page_writeback_range(inode->i_mapping, | ||
| 429 | start >> PAGE_CACHE_SHIFT, | ||
| 430 | orig_end >> PAGE_CACHE_SHIFT); | ||
| 431 | |||
| 432 | end = orig_end; | ||
| 433 | while (1) { | ||
| 434 | ordered = btrfs_lookup_first_ordered_extent(inode, end); | ||
| 435 | if (!ordered) | ||
| 436 | break; | ||
| 437 | if (ordered->file_offset > orig_end) { | ||
| 438 | btrfs_put_ordered_extent(ordered); | ||
| 439 | break; | ||
| 440 | } | ||
| 441 | if (ordered->file_offset + ordered->len < start) { | ||
| 442 | btrfs_put_ordered_extent(ordered); | ||
| 443 | break; | ||
| 444 | } | ||
| 445 | btrfs_start_ordered_extent(inode, ordered, 1); | ||
| 446 | end = ordered->file_offset; | ||
| 447 | btrfs_put_ordered_extent(ordered); | ||
| 448 | if (end == 0 || end == start) | ||
| 449 | break; | ||
| 450 | end--; | ||
| 451 | } | ||
| 452 | if (test_range_bit(&BTRFS_I(inode)->io_tree, start, orig_end, | ||
| 453 | EXTENT_ORDERED | EXTENT_DELALLOC, 0)) { | ||
| 454 | schedule_timeout(1); | ||
| 455 | goto again; | ||
| 456 | } | ||
| 457 | return 0; | ||
| 458 | } | ||
| 459 | |||
| 460 | /* | ||
| 461 | * find an ordered extent corresponding to file_offset. return NULL if | ||
| 462 | * nothing is found, otherwise take a reference on the extent and return it | ||
| 463 | */ | ||
| 464 | struct btrfs_ordered_extent *btrfs_lookup_ordered_extent(struct inode *inode, | ||
| 465 | u64 file_offset) | ||
| 466 | { | ||
| 467 | struct btrfs_ordered_inode_tree *tree; | ||
| 468 | struct rb_node *node; | ||
| 469 | struct btrfs_ordered_extent *entry = NULL; | ||
| 470 | |||
| 471 | tree = &BTRFS_I(inode)->ordered_tree; | ||
| 472 | mutex_lock(&tree->mutex); | ||
| 473 | node = tree_search(tree, file_offset); | ||
| 474 | if (!node) | ||
| 475 | goto out; | ||
| 476 | |||
| 477 | entry = rb_entry(node, struct btrfs_ordered_extent, rb_node); | ||
| 478 | if (!offset_in_entry(entry, file_offset)) | ||
| 479 | entry = NULL; | ||
| 480 | if (entry) | ||
| 481 | atomic_inc(&entry->refs); | ||
| 482 | out: | ||
| 483 | mutex_unlock(&tree->mutex); | ||
| 484 | return entry; | ||
| 485 | } | ||
| 486 | |||
| 487 | /* | ||
| 488 | * lookup and return any extent before 'file_offset'. NULL is returned | ||
| 489 | * if none is found | ||
| 490 | */ | ||
| 491 | struct btrfs_ordered_extent * | ||
| 492 | btrfs_lookup_first_ordered_extent(struct inode *inode, u64 file_offset) | ||
| 493 | { | ||
| 494 | struct btrfs_ordered_inode_tree *tree; | ||
| 495 | struct rb_node *node; | ||
| 496 | struct btrfs_ordered_extent *entry = NULL; | ||
| 497 | |||
| 498 | tree = &BTRFS_I(inode)->ordered_tree; | ||
| 499 | mutex_lock(&tree->mutex); | ||
| 500 | node = tree_search(tree, file_offset); | ||
| 501 | if (!node) | ||
| 502 | goto out; | ||
| 503 | |||
| 504 | entry = rb_entry(node, struct btrfs_ordered_extent, rb_node); | ||
| 505 | atomic_inc(&entry->refs); | ||
| 506 | out: | ||
| 507 | mutex_unlock(&tree->mutex); | ||
| 508 | return entry; | ||
| 509 | } | ||
| 510 | |||
| 511 | /* | ||
| 512 | * After an extent is done, call this to conditionally update the on disk | ||
| 513 | * i_size. i_size is updated to cover any fully written part of the file. | ||
| 514 | */ | ||
| 515 | int btrfs_ordered_update_i_size(struct inode *inode, | ||
| 516 | struct btrfs_ordered_extent *ordered) | ||
| 517 | { | ||
| 518 | struct btrfs_ordered_inode_tree *tree = &BTRFS_I(inode)->ordered_tree; | ||
| 519 | struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree; | ||
| 520 | u64 disk_i_size; | ||
| 521 | u64 new_i_size; | ||
| 522 | u64 i_size_test; | ||
| 523 | struct rb_node *node; | ||
| 524 | struct btrfs_ordered_extent *test; | ||
| 525 | |||
| 526 | mutex_lock(&tree->mutex); | ||
| 527 | disk_i_size = BTRFS_I(inode)->disk_i_size; | ||
| 528 | |||
| 529 | /* | ||
| 530 | * if the disk i_size is already at the inode->i_size, or | ||
| 531 | * this ordered extent is inside the disk i_size, we're done | ||
| 532 | */ | ||
| 533 | if (disk_i_size >= inode->i_size || | ||
| 534 | ordered->file_offset + ordered->len <= disk_i_size) { | ||
| 535 | goto out; | ||
| 536 | } | ||
| 537 | |||
| 538 | /* | ||
| 539 | * we can't update the disk_isize if there are delalloc bytes | ||
| 540 | * between disk_i_size and this ordered extent | ||
| 541 | */ | ||
| 542 | if (test_range_bit(io_tree, disk_i_size, | ||
| 543 | ordered->file_offset + ordered->len - 1, | ||
| 544 | EXTENT_DELALLOC, 0)) { | ||
| 545 | goto out; | ||
| 546 | } | ||
| 547 | /* | ||
| 548 | * walk backward from this ordered extent to disk_i_size. | ||
| 549 | * if we find an ordered extent then we can't update disk i_size | ||
| 550 | * yet | ||
| 551 | */ | ||
| 552 | node = &ordered->rb_node; | ||
| 553 | while (1) { | ||
| 554 | node = rb_prev(node); | ||
| 555 | if (!node) | ||
| 556 | break; | ||
| 557 | test = rb_entry(node, struct btrfs_ordered_extent, rb_node); | ||
| 558 | if (test->file_offset + test->len <= disk_i_size) | ||
| 559 | break; | ||
| 560 | if (test->file_offset >= inode->i_size) | ||
| 561 | break; | ||
| 562 | if (test->file_offset >= disk_i_size) | ||
| 563 | goto out; | ||
| 564 | } | ||
| 565 | new_i_size = min_t(u64, entry_end(ordered), i_size_read(inode)); | ||
| 566 | |||
| 567 | /* | ||
| 568 | * at this point, we know we can safely update i_size to at least | ||
| 569 | * the offset from this ordered extent. But, we need to | ||
| 570 | * walk forward and see if ios from higher up in the file have | ||
| 571 | * finished. | ||
| 572 | */ | ||
| 573 | node = rb_next(&ordered->rb_node); | ||
| 574 | i_size_test = 0; | ||
| 575 | if (node) { | ||
| 576 | /* | ||
| 577 | * do we have an area where IO might have finished | ||
| 578 | * between our ordered extent and the next one. | ||
| 579 | */ | ||
| 580 | test = rb_entry(node, struct btrfs_ordered_extent, rb_node); | ||
| 581 | if (test->file_offset > entry_end(ordered)) | ||
| 582 | i_size_test = test->file_offset; | ||
| 583 | } else { | ||
| 584 | i_size_test = i_size_read(inode); | ||
| 585 | } | ||
| 586 | |||
| 587 | /* | ||
| 588 | * i_size_test is the end of a region after this ordered | ||
| 589 | * extent where there are no ordered extents. As long as there | ||
| 590 | * are no delalloc bytes in this area, it is safe to update | ||
| 591 | * disk_i_size to the end of the region. | ||
| 592 | */ | ||
| 593 | if (i_size_test > entry_end(ordered) && | ||
| 594 | !test_range_bit(io_tree, entry_end(ordered), i_size_test - 1, | ||
| 595 | EXTENT_DELALLOC, 0)) { | ||
| 596 | new_i_size = min_t(u64, i_size_test, i_size_read(inode)); | ||
| 597 | } | ||
| 598 | BTRFS_I(inode)->disk_i_size = new_i_size; | ||
| 599 | out: | ||
| 600 | mutex_unlock(&tree->mutex); | ||
| 601 | return 0; | ||
| 602 | } | ||
| 603 | |||
| 604 | /* | ||
| 605 | * search the ordered extents for one corresponding to 'offset' and | ||
| 606 | * try to find a checksum. This is used because we allow pages to | ||
| 607 | * be reclaimed before their checksum is actually put into the btree | ||
| 608 | */ | ||
| 609 | int btrfs_find_ordered_sum(struct inode *inode, u64 offset, u64 disk_bytenr, | ||
| 610 | u32 *sum) | ||
| 611 | { | ||
| 612 | struct btrfs_ordered_sum *ordered_sum; | ||
| 613 | struct btrfs_sector_sum *sector_sums; | ||
| 614 | struct btrfs_ordered_extent *ordered; | ||
| 615 | struct btrfs_ordered_inode_tree *tree = &BTRFS_I(inode)->ordered_tree; | ||
| 616 | struct list_head *cur; | ||
| 617 | unsigned long num_sectors; | ||
| 618 | unsigned long i; | ||
| 619 | u32 sectorsize = BTRFS_I(inode)->root->sectorsize; | ||
| 620 | int ret = 1; | ||
| 621 | |||
| 622 | ordered = btrfs_lookup_ordered_extent(inode, offset); | ||
| 623 | if (!ordered) | ||
| 624 | return 1; | ||
| 625 | |||
| 626 | mutex_lock(&tree->mutex); | ||
| 627 | list_for_each_prev(cur, &ordered->list) { | ||
| 628 | ordered_sum = list_entry(cur, struct btrfs_ordered_sum, list); | ||
| 629 | if (disk_bytenr >= ordered_sum->bytenr) { | ||
| 630 | num_sectors = ordered_sum->len / sectorsize; | ||
| 631 | sector_sums = ordered_sum->sums; | ||
| 632 | for (i = 0; i < num_sectors; i++) { | ||
| 633 | if (sector_sums[i].bytenr == disk_bytenr) { | ||
| 634 | *sum = sector_sums[i].sum; | ||
| 635 | ret = 0; | ||
| 636 | goto out; | ||
| 637 | } | ||
| 638 | } | ||
| 639 | } | ||
| 640 | } | ||
| 641 | out: | ||
| 642 | mutex_unlock(&tree->mutex); | ||
| 643 | btrfs_put_ordered_extent(ordered); | ||
| 644 | return ret; | ||
| 645 | } | ||
| 646 | |||
| 647 | |||
| 648 | /** | ||
| 649 | * taken from mm/filemap.c because it isn't exported | ||
| 650 | * | ||
| 651 | * __filemap_fdatawrite_range - start writeback on mapping dirty pages in range | ||
| 652 | * @mapping: address space structure to write | ||
| 653 | * @start: offset in bytes where the range starts | ||
| 654 | * @end: offset in bytes where the range ends (inclusive) | ||
| 655 | * @sync_mode: enable synchronous operation | ||
| 656 | * | ||
| 657 | * Start writeback against all of a mapping's dirty pages that lie | ||
| 658 | * within the byte offsets <start, end> inclusive. | ||
| 659 | * | ||
| 660 | * If sync_mode is WB_SYNC_ALL then this is a "data integrity" operation, as | ||
| 661 | * opposed to a regular memory cleansing writeback. The difference between | ||
| 662 | * these two operations is that if a dirty page/buffer is encountered, it must | ||
| 663 | * be waited upon, and not just skipped over. | ||
| 664 | */ | ||
| 665 | int btrfs_fdatawrite_range(struct address_space *mapping, loff_t start, | ||
| 666 | loff_t end, int sync_mode) | ||
| 667 | { | ||
| 668 | struct writeback_control wbc = { | ||
| 669 | .sync_mode = sync_mode, | ||
| 670 | .nr_to_write = mapping->nrpages * 2, | ||
| 671 | .range_start = start, | ||
| 672 | .range_end = end, | ||
| 673 | .for_writepages = 1, | ||
| 674 | }; | ||
| 675 | return btrfs_writepages(mapping, &wbc); | ||
| 676 | } | ||
| 677 | |||
| 678 | /** | ||
| 679 | * taken from mm/filemap.c because it isn't exported | ||
| 680 | * | ||
| 681 | * wait_on_page_writeback_range - wait for writeback to complete | ||
| 682 | * @mapping: target address_space | ||
| 683 | * @start: beginning page index | ||
| 684 | * @end: ending page index | ||
| 685 | * | ||
| 686 | * Wait for writeback to complete against pages indexed by start->end | ||
| 687 | * inclusive | ||
| 688 | */ | ||
| 689 | int btrfs_wait_on_page_writeback_range(struct address_space *mapping, | ||
| 690 | pgoff_t start, pgoff_t end) | ||
| 691 | { | ||
| 692 | struct pagevec pvec; | ||
| 693 | int nr_pages; | ||
| 694 | int ret = 0; | ||
| 695 | pgoff_t index; | ||
| 696 | |||
| 697 | if (end < start) | ||
| 698 | return 0; | ||
| 699 | |||
| 700 | pagevec_init(&pvec, 0); | ||
| 701 | index = start; | ||
| 702 | while ((index <= end) && | ||
| 703 | (nr_pages = pagevec_lookup_tag(&pvec, mapping, &index, | ||
| 704 | PAGECACHE_TAG_WRITEBACK, | ||
| 705 | min(end - index, (pgoff_t)PAGEVEC_SIZE-1) + 1)) != 0) { | ||
| 706 | unsigned i; | ||
| 707 | |||
| 708 | for (i = 0; i < nr_pages; i++) { | ||
| 709 | struct page *page = pvec.pages[i]; | ||
| 710 | |||
| 711 | /* until radix tree lookup accepts end_index */ | ||
| 712 | if (page->index > end) | ||
| 713 | continue; | ||
| 714 | |||
| 715 | wait_on_page_writeback(page); | ||
| 716 | if (PageError(page)) | ||
| 717 | ret = -EIO; | ||
| 718 | } | ||
| 719 | pagevec_release(&pvec); | ||
| 720 | cond_resched(); | ||
| 721 | } | ||
| 722 | |||
| 723 | /* Check for outstanding write errors */ | ||
| 724 | if (test_and_clear_bit(AS_ENOSPC, &mapping->flags)) | ||
| 725 | ret = -ENOSPC; | ||
| 726 | if (test_and_clear_bit(AS_EIO, &mapping->flags)) | ||
| 727 | ret = -EIO; | ||
| 728 | |||
| 729 | return ret; | ||
| 730 | } | ||
diff --git a/fs/btrfs/ordered-data.h b/fs/btrfs/ordered-data.h new file mode 100644 index 000000000000..ab66d5e8d6d6 --- /dev/null +++ b/fs/btrfs/ordered-data.h | |||
| @@ -0,0 +1,158 @@ | |||
| 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 | |||
| 19 | #ifndef __BTRFS_ORDERED_DATA__ | ||
| 20 | #define __BTRFS_ORDERED_DATA__ | ||
| 21 | |||
| 22 | /* one of these per inode */ | ||
| 23 | struct btrfs_ordered_inode_tree { | ||
| 24 | struct mutex mutex; | ||
| 25 | struct rb_root tree; | ||
| 26 | struct rb_node *last; | ||
| 27 | }; | ||
| 28 | |||
| 29 | /* | ||
| 30 | * these are used to collect checksums done just before bios submission. | ||
| 31 | * They are attached via a list into the ordered extent, and | ||
| 32 | * checksum items are inserted into the tree after all the blocks in | ||
| 33 | * the ordered extent are on disk | ||
| 34 | */ | ||
| 35 | struct btrfs_sector_sum { | ||
| 36 | /* bytenr on disk */ | ||
| 37 | u64 bytenr; | ||
| 38 | u32 sum; | ||
| 39 | }; | ||
| 40 | |||
| 41 | struct btrfs_ordered_sum { | ||
| 42 | /* bytenr is the start of this extent on disk */ | ||
| 43 | u64 bytenr; | ||
| 44 | |||
| 45 | /* | ||
| 46 | * this is the length in bytes covered by the sums array below. | ||
| 47 | */ | ||
| 48 | unsigned long len; | ||
| 49 | struct list_head list; | ||
| 50 | /* last field is a variable length array of btrfs_sector_sums */ | ||
| 51 | struct btrfs_sector_sum sums[]; | ||
| 52 | }; | ||
| 53 | |||
| 54 | /* | ||
| 55 | * bits for the flags field: | ||
| 56 | * | ||
| 57 | * BTRFS_ORDERED_IO_DONE is set when all of the blocks are written. | ||
| 58 | * It is used to make sure metadata is inserted into the tree only once | ||
| 59 | * per extent. | ||
| 60 | * | ||
| 61 | * BTRFS_ORDERED_COMPLETE is set when the extent is removed from the | ||
| 62 | * rbtree, just before waking any waiters. It is used to indicate the | ||
| 63 | * IO is done and any metadata is inserted into the tree. | ||
| 64 | */ | ||
| 65 | #define BTRFS_ORDERED_IO_DONE 0 /* set when all the pages are written */ | ||
| 66 | |||
| 67 | #define BTRFS_ORDERED_COMPLETE 1 /* set when removed from the tree */ | ||
| 68 | |||
| 69 | #define BTRFS_ORDERED_NOCOW 2 /* set when we want to write in place */ | ||
| 70 | |||
| 71 | #define BTRFS_ORDERED_COMPRESSED 3 /* writing a compressed extent */ | ||
| 72 | |||
| 73 | #define BTRFS_ORDERED_PREALLOC 4 /* set when writing to prealloced extent */ | ||
| 74 | |||
| 75 | struct btrfs_ordered_extent { | ||
| 76 | /* logical offset in the file */ | ||
| 77 | u64 file_offset; | ||
| 78 | |||
| 79 | /* disk byte number */ | ||
| 80 | u64 start; | ||
| 81 | |||
| 82 | /* ram length of the extent in bytes */ | ||
| 83 | u64 len; | ||
| 84 | |||
| 85 | /* extent length on disk */ | ||
| 86 | u64 disk_len; | ||
| 87 | |||
| 88 | /* flags (described above) */ | ||
| 89 | unsigned long flags; | ||
| 90 | |||
| 91 | /* reference count */ | ||
| 92 | atomic_t refs; | ||
| 93 | |||
| 94 | /* the inode we belong to */ | ||
| 95 | struct inode *inode; | ||
| 96 | |||
| 97 | /* list of checksums for insertion when the extent io is done */ | ||
| 98 | struct list_head list; | ||
| 99 | |||
| 100 | /* used to wait for the BTRFS_ORDERED_COMPLETE bit */ | ||
| 101 | wait_queue_head_t wait; | ||
| 102 | |||
| 103 | /* our friendly rbtree entry */ | ||
| 104 | struct rb_node rb_node; | ||
| 105 | |||
| 106 | /* a per root list of all the pending ordered extents */ | ||
| 107 | struct list_head root_extent_list; | ||
| 108 | }; | ||
| 109 | |||
| 110 | |||
| 111 | /* | ||
| 112 | * calculates the total size you need to allocate for an ordered sum | ||
| 113 | * structure spanning 'bytes' in the file | ||
| 114 | */ | ||
| 115 | static inline int btrfs_ordered_sum_size(struct btrfs_root *root, | ||
| 116 | unsigned long bytes) | ||
| 117 | { | ||
| 118 | unsigned long num_sectors = (bytes + root->sectorsize - 1) / | ||
| 119 | root->sectorsize; | ||
| 120 | num_sectors++; | ||
| 121 | return sizeof(struct btrfs_ordered_sum) + | ||
| 122 | num_sectors * sizeof(struct btrfs_sector_sum); | ||
| 123 | } | ||
| 124 | |||
| 125 | static inline void | ||
| 126 | btrfs_ordered_inode_tree_init(struct btrfs_ordered_inode_tree *t) | ||
| 127 | { | ||
| 128 | mutex_init(&t->mutex); | ||
| 129 | t->tree.rb_node = NULL; | ||
| 130 | t->last = NULL; | ||
| 131 | } | ||
| 132 | |||
| 133 | int btrfs_put_ordered_extent(struct btrfs_ordered_extent *entry); | ||
| 134 | int btrfs_remove_ordered_extent(struct inode *inode, | ||
| 135 | struct btrfs_ordered_extent *entry); | ||
| 136 | int btrfs_dec_test_ordered_pending(struct inode *inode, | ||
| 137 | u64 file_offset, u64 io_size); | ||
| 138 | int btrfs_add_ordered_extent(struct inode *inode, u64 file_offset, | ||
| 139 | u64 start, u64 len, u64 disk_len, int tyep); | ||
| 140 | int btrfs_add_ordered_sum(struct inode *inode, | ||
| 141 | struct btrfs_ordered_extent *entry, | ||
| 142 | struct btrfs_ordered_sum *sum); | ||
| 143 | struct btrfs_ordered_extent *btrfs_lookup_ordered_extent(struct inode *inode, | ||
| 144 | u64 file_offset); | ||
| 145 | void btrfs_start_ordered_extent(struct inode *inode, | ||
| 146 | struct btrfs_ordered_extent *entry, int wait); | ||
| 147 | int btrfs_wait_ordered_range(struct inode *inode, u64 start, u64 len); | ||
| 148 | struct btrfs_ordered_extent * | ||
| 149 | btrfs_lookup_first_ordered_extent(struct inode * inode, u64 file_offset); | ||
| 150 | int btrfs_ordered_update_i_size(struct inode *inode, | ||
| 151 | struct btrfs_ordered_extent *ordered); | ||
| 152 | int btrfs_find_ordered_sum(struct inode *inode, u64 offset, u64 disk_bytenr, u32 *sum); | ||
| 153 | int btrfs_wait_on_page_writeback_range(struct address_space *mapping, | ||
| 154 | pgoff_t start, pgoff_t end); | ||
| 155 | int btrfs_fdatawrite_range(struct address_space *mapping, loff_t start, | ||
| 156 | loff_t end, int sync_mode); | ||
| 157 | int btrfs_wait_ordered_extents(struct btrfs_root *root, int nocow_only); | ||
| 158 | #endif | ||
diff --git a/fs/btrfs/orphan.c b/fs/btrfs/orphan.c new file mode 100644 index 000000000000..3c0d52af4f80 --- /dev/null +++ b/fs/btrfs/orphan.c | |||
| @@ -0,0 +1,67 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 Red Hat. 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 | |||
| 19 | #include "ctree.h" | ||
| 20 | #include "disk-io.h" | ||
| 21 | |||
| 22 | int btrfs_insert_orphan_item(struct btrfs_trans_handle *trans, | ||
| 23 | struct btrfs_root *root, u64 offset) | ||
| 24 | { | ||
| 25 | struct btrfs_path *path; | ||
| 26 | struct btrfs_key key; | ||
| 27 | int ret = 0; | ||
| 28 | |||
| 29 | key.objectid = BTRFS_ORPHAN_OBJECTID; | ||
| 30 | btrfs_set_key_type(&key, BTRFS_ORPHAN_ITEM_KEY); | ||
| 31 | key.offset = offset; | ||
| 32 | |||
| 33 | path = btrfs_alloc_path(); | ||
| 34 | if (!path) | ||
| 35 | return -ENOMEM; | ||
| 36 | |||
| 37 | ret = btrfs_insert_empty_item(trans, root, path, &key, 0); | ||
| 38 | |||
| 39 | btrfs_free_path(path); | ||
| 40 | return ret; | ||
| 41 | } | ||
| 42 | |||
| 43 | int btrfs_del_orphan_item(struct btrfs_trans_handle *trans, | ||
| 44 | struct btrfs_root *root, u64 offset) | ||
| 45 | { | ||
| 46 | struct btrfs_path *path; | ||
| 47 | struct btrfs_key key; | ||
| 48 | int ret = 0; | ||
| 49 | |||
| 50 | key.objectid = BTRFS_ORPHAN_OBJECTID; | ||
| 51 | btrfs_set_key_type(&key, BTRFS_ORPHAN_ITEM_KEY); | ||
| 52 | key.offset = offset; | ||
| 53 | |||
| 54 | path = btrfs_alloc_path(); | ||
| 55 | if (!path) | ||
| 56 | return -ENOMEM; | ||
| 57 | |||
| 58 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 59 | if (ret) | ||
| 60 | goto out; | ||
| 61 | |||
| 62 | ret = btrfs_del_item(trans, root, path); | ||
| 63 | |||
| 64 | out: | ||
| 65 | btrfs_free_path(path); | ||
| 66 | return ret; | ||
| 67 | } | ||
diff --git a/fs/btrfs/print-tree.c b/fs/btrfs/print-tree.c new file mode 100644 index 000000000000..5f8f218c1005 --- /dev/null +++ b/fs/btrfs/print-tree.c | |||
| @@ -0,0 +1,216 @@ | |||
| 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 | |||
| 19 | #include "ctree.h" | ||
| 20 | #include "disk-io.h" | ||
| 21 | #include "print-tree.h" | ||
| 22 | |||
| 23 | static void print_chunk(struct extent_buffer *eb, struct btrfs_chunk *chunk) | ||
| 24 | { | ||
| 25 | int num_stripes = btrfs_chunk_num_stripes(eb, chunk); | ||
| 26 | int i; | ||
| 27 | printk(KERN_INFO "\t\tchunk length %llu owner %llu type %llu " | ||
| 28 | "num_stripes %d\n", | ||
| 29 | (unsigned long long)btrfs_chunk_length(eb, chunk), | ||
| 30 | (unsigned long long)btrfs_chunk_owner(eb, chunk), | ||
| 31 | (unsigned long long)btrfs_chunk_type(eb, chunk), | ||
| 32 | num_stripes); | ||
| 33 | for (i = 0 ; i < num_stripes ; i++) { | ||
| 34 | printk(KERN_INFO "\t\t\tstripe %d devid %llu offset %llu\n", i, | ||
| 35 | (unsigned long long)btrfs_stripe_devid_nr(eb, chunk, i), | ||
| 36 | (unsigned long long)btrfs_stripe_offset_nr(eb, chunk, i)); | ||
| 37 | } | ||
| 38 | } | ||
| 39 | static void print_dev_item(struct extent_buffer *eb, | ||
| 40 | struct btrfs_dev_item *dev_item) | ||
| 41 | { | ||
| 42 | printk(KERN_INFO "\t\tdev item devid %llu " | ||
| 43 | "total_bytes %llu bytes used %llu\n", | ||
| 44 | (unsigned long long)btrfs_device_id(eb, dev_item), | ||
| 45 | (unsigned long long)btrfs_device_total_bytes(eb, dev_item), | ||
| 46 | (unsigned long long)btrfs_device_bytes_used(eb, dev_item)); | ||
| 47 | } | ||
| 48 | void btrfs_print_leaf(struct btrfs_root *root, struct extent_buffer *l) | ||
| 49 | { | ||
| 50 | int i; | ||
| 51 | u32 nr = btrfs_header_nritems(l); | ||
| 52 | struct btrfs_item *item; | ||
| 53 | struct btrfs_extent_item *ei; | ||
| 54 | struct btrfs_root_item *ri; | ||
| 55 | struct btrfs_dir_item *di; | ||
| 56 | struct btrfs_inode_item *ii; | ||
| 57 | struct btrfs_block_group_item *bi; | ||
| 58 | struct btrfs_file_extent_item *fi; | ||
| 59 | struct btrfs_key key; | ||
| 60 | struct btrfs_key found_key; | ||
| 61 | struct btrfs_extent_ref *ref; | ||
| 62 | struct btrfs_dev_extent *dev_extent; | ||
| 63 | u32 type; | ||
| 64 | |||
| 65 | printk(KERN_INFO "leaf %llu total ptrs %d free space %d\n", | ||
| 66 | (unsigned long long)btrfs_header_bytenr(l), nr, | ||
| 67 | btrfs_leaf_free_space(root, l)); | ||
| 68 | for (i = 0 ; i < nr ; i++) { | ||
| 69 | item = btrfs_item_nr(l, i); | ||
| 70 | btrfs_item_key_to_cpu(l, &key, i); | ||
| 71 | type = btrfs_key_type(&key); | ||
| 72 | printk(KERN_INFO "\titem %d key (%llu %x %llu) itemoff %d " | ||
| 73 | "itemsize %d\n", | ||
| 74 | i, | ||
| 75 | (unsigned long long)key.objectid, type, | ||
| 76 | (unsigned long long)key.offset, | ||
| 77 | btrfs_item_offset(l, item), btrfs_item_size(l, item)); | ||
| 78 | switch (type) { | ||
| 79 | case BTRFS_INODE_ITEM_KEY: | ||
| 80 | ii = btrfs_item_ptr(l, i, struct btrfs_inode_item); | ||
| 81 | printk(KERN_INFO "\t\tinode generation %llu size %llu " | ||
| 82 | "mode %o\n", | ||
| 83 | (unsigned long long) | ||
| 84 | btrfs_inode_generation(l, ii), | ||
| 85 | (unsigned long long)btrfs_inode_size(l, ii), | ||
| 86 | btrfs_inode_mode(l, ii)); | ||
| 87 | break; | ||
| 88 | case BTRFS_DIR_ITEM_KEY: | ||
| 89 | di = btrfs_item_ptr(l, i, struct btrfs_dir_item); | ||
| 90 | btrfs_dir_item_key_to_cpu(l, di, &found_key); | ||
| 91 | printk(KERN_INFO "\t\tdir oid %llu type %u\n", | ||
| 92 | (unsigned long long)found_key.objectid, | ||
| 93 | btrfs_dir_type(l, di)); | ||
| 94 | break; | ||
| 95 | case BTRFS_ROOT_ITEM_KEY: | ||
| 96 | ri = btrfs_item_ptr(l, i, struct btrfs_root_item); | ||
| 97 | printk(KERN_INFO "\t\troot data bytenr %llu refs %u\n", | ||
| 98 | (unsigned long long) | ||
| 99 | btrfs_disk_root_bytenr(l, ri), | ||
| 100 | btrfs_disk_root_refs(l, ri)); | ||
| 101 | break; | ||
| 102 | case BTRFS_EXTENT_ITEM_KEY: | ||
| 103 | ei = btrfs_item_ptr(l, i, struct btrfs_extent_item); | ||
| 104 | printk(KERN_INFO "\t\textent data refs %u\n", | ||
| 105 | btrfs_extent_refs(l, ei)); | ||
| 106 | break; | ||
| 107 | case BTRFS_EXTENT_REF_KEY: | ||
| 108 | ref = btrfs_item_ptr(l, i, struct btrfs_extent_ref); | ||
| 109 | printk(KERN_INFO "\t\textent back ref root %llu " | ||
| 110 | "gen %llu owner %llu num_refs %lu\n", | ||
| 111 | (unsigned long long)btrfs_ref_root(l, ref), | ||
| 112 | (unsigned long long)btrfs_ref_generation(l, ref), | ||
| 113 | (unsigned long long)btrfs_ref_objectid(l, ref), | ||
| 114 | (unsigned long)btrfs_ref_num_refs(l, ref)); | ||
| 115 | break; | ||
| 116 | |||
| 117 | case BTRFS_EXTENT_DATA_KEY: | ||
| 118 | fi = btrfs_item_ptr(l, i, | ||
| 119 | struct btrfs_file_extent_item); | ||
| 120 | if (btrfs_file_extent_type(l, fi) == | ||
| 121 | BTRFS_FILE_EXTENT_INLINE) { | ||
| 122 | printk(KERN_INFO "\t\tinline extent data " | ||
| 123 | "size %u\n", | ||
| 124 | btrfs_file_extent_inline_len(l, fi)); | ||
| 125 | break; | ||
| 126 | } | ||
| 127 | printk(KERN_INFO "\t\textent data disk bytenr %llu " | ||
| 128 | "nr %llu\n", | ||
| 129 | (unsigned long long) | ||
| 130 | btrfs_file_extent_disk_bytenr(l, fi), | ||
| 131 | (unsigned long long) | ||
| 132 | btrfs_file_extent_disk_num_bytes(l, fi)); | ||
| 133 | printk(KERN_INFO "\t\textent data offset %llu " | ||
| 134 | "nr %llu ram %llu\n", | ||
| 135 | (unsigned long long) | ||
| 136 | btrfs_file_extent_offset(l, fi), | ||
| 137 | (unsigned long long) | ||
| 138 | btrfs_file_extent_num_bytes(l, fi), | ||
| 139 | (unsigned long long) | ||
| 140 | btrfs_file_extent_ram_bytes(l, fi)); | ||
| 141 | break; | ||
| 142 | case BTRFS_BLOCK_GROUP_ITEM_KEY: | ||
| 143 | bi = btrfs_item_ptr(l, i, | ||
| 144 | struct btrfs_block_group_item); | ||
| 145 | printk(KERN_INFO "\t\tblock group used %llu\n", | ||
| 146 | (unsigned long long) | ||
| 147 | btrfs_disk_block_group_used(l, bi)); | ||
| 148 | break; | ||
| 149 | case BTRFS_CHUNK_ITEM_KEY: | ||
| 150 | print_chunk(l, btrfs_item_ptr(l, i, | ||
| 151 | struct btrfs_chunk)); | ||
| 152 | break; | ||
| 153 | case BTRFS_DEV_ITEM_KEY: | ||
| 154 | print_dev_item(l, btrfs_item_ptr(l, i, | ||
| 155 | struct btrfs_dev_item)); | ||
| 156 | break; | ||
| 157 | case BTRFS_DEV_EXTENT_KEY: | ||
| 158 | dev_extent = btrfs_item_ptr(l, i, | ||
| 159 | struct btrfs_dev_extent); | ||
| 160 | printk(KERN_INFO "\t\tdev extent chunk_tree %llu\n" | ||
| 161 | "\t\tchunk objectid %llu chunk offset %llu " | ||
| 162 | "length %llu\n", | ||
| 163 | (unsigned long long) | ||
| 164 | btrfs_dev_extent_chunk_tree(l, dev_extent), | ||
| 165 | (unsigned long long) | ||
| 166 | btrfs_dev_extent_chunk_objectid(l, dev_extent), | ||
| 167 | (unsigned long long) | ||
| 168 | btrfs_dev_extent_chunk_offset(l, dev_extent), | ||
| 169 | (unsigned long long) | ||
| 170 | btrfs_dev_extent_length(l, dev_extent)); | ||
| 171 | }; | ||
| 172 | } | ||
| 173 | } | ||
| 174 | |||
| 175 | void btrfs_print_tree(struct btrfs_root *root, struct extent_buffer *c) | ||
| 176 | { | ||
| 177 | int i; u32 nr; | ||
| 178 | struct btrfs_key key; | ||
| 179 | int level; | ||
| 180 | |||
| 181 | if (!c) | ||
| 182 | return; | ||
| 183 | nr = btrfs_header_nritems(c); | ||
| 184 | level = btrfs_header_level(c); | ||
| 185 | if (level == 0) { | ||
| 186 | btrfs_print_leaf(root, c); | ||
| 187 | return; | ||
| 188 | } | ||
| 189 | printk(KERN_INFO "node %llu level %d total ptrs %d free spc %u\n", | ||
| 190 | (unsigned long long)btrfs_header_bytenr(c), | ||
| 191 | btrfs_header_level(c), nr, | ||
| 192 | (u32)BTRFS_NODEPTRS_PER_BLOCK(root) - nr); | ||
| 193 | for (i = 0; i < nr; i++) { | ||
| 194 | btrfs_node_key_to_cpu(c, &key, i); | ||
| 195 | printk(KERN_INFO "\tkey %d (%llu %u %llu) block %llu\n", | ||
| 196 | i, | ||
| 197 | (unsigned long long)key.objectid, | ||
| 198 | key.type, | ||
| 199 | (unsigned long long)key.offset, | ||
| 200 | (unsigned long long)btrfs_node_blockptr(c, i)); | ||
| 201 | } | ||
| 202 | for (i = 0; i < nr; i++) { | ||
| 203 | struct extent_buffer *next = read_tree_block(root, | ||
| 204 | btrfs_node_blockptr(c, i), | ||
| 205 | btrfs_level_size(root, level - 1), | ||
| 206 | btrfs_node_ptr_generation(c, i)); | ||
| 207 | if (btrfs_is_leaf(next) && | ||
| 208 | btrfs_header_level(c) != 1) | ||
| 209 | BUG(); | ||
| 210 | if (btrfs_header_level(next) != | ||
| 211 | btrfs_header_level(c) - 1) | ||
| 212 | BUG(); | ||
| 213 | btrfs_print_tree(root, next); | ||
| 214 | free_extent_buffer(next); | ||
| 215 | } | ||
| 216 | } | ||
diff --git a/fs/btrfs/print-tree.h b/fs/btrfs/print-tree.h new file mode 100644 index 000000000000..da75efe534d5 --- /dev/null +++ b/fs/btrfs/print-tree.h | |||
| @@ -0,0 +1,23 @@ | |||
| 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 | |||
| 19 | #ifndef __PRINT_TREE_ | ||
| 20 | #define __PRINT_TREE_ | ||
| 21 | void btrfs_print_leaf(struct btrfs_root *root, struct extent_buffer *l); | ||
| 22 | void btrfs_print_tree(struct btrfs_root *root, struct extent_buffer *t); | ||
| 23 | #endif | ||
diff --git a/fs/btrfs/ref-cache.c b/fs/btrfs/ref-cache.c new file mode 100644 index 000000000000..6f0acc4c9eab --- /dev/null +++ b/fs/btrfs/ref-cache.c | |||
| @@ -0,0 +1,230 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | |||
| 19 | #include <linux/sched.h> | ||
| 20 | #include "ctree.h" | ||
| 21 | #include "ref-cache.h" | ||
| 22 | #include "transaction.h" | ||
| 23 | |||
| 24 | /* | ||
| 25 | * leaf refs are used to cache the information about which extents | ||
| 26 | * a given leaf has references on. This allows us to process that leaf | ||
| 27 | * in btrfs_drop_snapshot without needing to read it back from disk. | ||
| 28 | */ | ||
| 29 | |||
| 30 | /* | ||
| 31 | * kmalloc a leaf reference struct and update the counters for the | ||
| 32 | * total ref cache size | ||
| 33 | */ | ||
| 34 | struct btrfs_leaf_ref *btrfs_alloc_leaf_ref(struct btrfs_root *root, | ||
| 35 | int nr_extents) | ||
| 36 | { | ||
| 37 | struct btrfs_leaf_ref *ref; | ||
| 38 | size_t size = btrfs_leaf_ref_size(nr_extents); | ||
| 39 | |||
| 40 | ref = kmalloc(size, GFP_NOFS); | ||
| 41 | if (ref) { | ||
| 42 | spin_lock(&root->fs_info->ref_cache_lock); | ||
| 43 | root->fs_info->total_ref_cache_size += size; | ||
| 44 | spin_unlock(&root->fs_info->ref_cache_lock); | ||
| 45 | |||
| 46 | memset(ref, 0, sizeof(*ref)); | ||
| 47 | atomic_set(&ref->usage, 1); | ||
| 48 | INIT_LIST_HEAD(&ref->list); | ||
| 49 | } | ||
| 50 | return ref; | ||
| 51 | } | ||
| 52 | |||
| 53 | /* | ||
| 54 | * free a leaf reference struct and update the counters for the | ||
| 55 | * total ref cache size | ||
| 56 | */ | ||
| 57 | void btrfs_free_leaf_ref(struct btrfs_root *root, struct btrfs_leaf_ref *ref) | ||
| 58 | { | ||
| 59 | if (!ref) | ||
| 60 | return; | ||
| 61 | WARN_ON(atomic_read(&ref->usage) == 0); | ||
| 62 | if (atomic_dec_and_test(&ref->usage)) { | ||
| 63 | size_t size = btrfs_leaf_ref_size(ref->nritems); | ||
| 64 | |||
| 65 | BUG_ON(ref->in_tree); | ||
| 66 | kfree(ref); | ||
| 67 | |||
| 68 | spin_lock(&root->fs_info->ref_cache_lock); | ||
| 69 | root->fs_info->total_ref_cache_size -= size; | ||
| 70 | spin_unlock(&root->fs_info->ref_cache_lock); | ||
| 71 | } | ||
| 72 | } | ||
| 73 | |||
| 74 | static struct rb_node *tree_insert(struct rb_root *root, u64 bytenr, | ||
| 75 | struct rb_node *node) | ||
| 76 | { | ||
| 77 | struct rb_node **p = &root->rb_node; | ||
| 78 | struct rb_node *parent = NULL; | ||
| 79 | struct btrfs_leaf_ref *entry; | ||
| 80 | |||
| 81 | while (*p) { | ||
| 82 | parent = *p; | ||
| 83 | entry = rb_entry(parent, struct btrfs_leaf_ref, rb_node); | ||
| 84 | |||
| 85 | if (bytenr < entry->bytenr) | ||
| 86 | p = &(*p)->rb_left; | ||
| 87 | else if (bytenr > entry->bytenr) | ||
| 88 | p = &(*p)->rb_right; | ||
| 89 | else | ||
| 90 | return parent; | ||
| 91 | } | ||
| 92 | |||
| 93 | entry = rb_entry(node, struct btrfs_leaf_ref, rb_node); | ||
| 94 | rb_link_node(node, parent, p); | ||
| 95 | rb_insert_color(node, root); | ||
| 96 | return NULL; | ||
| 97 | } | ||
| 98 | |||
| 99 | static struct rb_node *tree_search(struct rb_root *root, u64 bytenr) | ||
| 100 | { | ||
| 101 | struct rb_node *n = root->rb_node; | ||
| 102 | struct btrfs_leaf_ref *entry; | ||
| 103 | |||
| 104 | while (n) { | ||
| 105 | entry = rb_entry(n, struct btrfs_leaf_ref, rb_node); | ||
| 106 | WARN_ON(!entry->in_tree); | ||
| 107 | |||
| 108 | if (bytenr < entry->bytenr) | ||
| 109 | n = n->rb_left; | ||
| 110 | else if (bytenr > entry->bytenr) | ||
| 111 | n = n->rb_right; | ||
| 112 | else | ||
| 113 | return n; | ||
| 114 | } | ||
| 115 | return NULL; | ||
| 116 | } | ||
| 117 | |||
| 118 | int btrfs_remove_leaf_refs(struct btrfs_root *root, u64 max_root_gen, | ||
| 119 | int shared) | ||
| 120 | { | ||
| 121 | struct btrfs_leaf_ref *ref = NULL; | ||
| 122 | struct btrfs_leaf_ref_tree *tree = root->ref_tree; | ||
| 123 | |||
| 124 | if (shared) | ||
| 125 | tree = &root->fs_info->shared_ref_tree; | ||
| 126 | if (!tree) | ||
| 127 | return 0; | ||
| 128 | |||
| 129 | spin_lock(&tree->lock); | ||
| 130 | while (!list_empty(&tree->list)) { | ||
| 131 | ref = list_entry(tree->list.next, struct btrfs_leaf_ref, list); | ||
| 132 | BUG_ON(ref->tree != tree); | ||
| 133 | if (ref->root_gen > max_root_gen) | ||
| 134 | break; | ||
| 135 | if (!xchg(&ref->in_tree, 0)) { | ||
| 136 | cond_resched_lock(&tree->lock); | ||
| 137 | continue; | ||
| 138 | } | ||
| 139 | |||
| 140 | rb_erase(&ref->rb_node, &tree->root); | ||
| 141 | list_del_init(&ref->list); | ||
| 142 | |||
| 143 | spin_unlock(&tree->lock); | ||
| 144 | btrfs_free_leaf_ref(root, ref); | ||
| 145 | cond_resched(); | ||
| 146 | spin_lock(&tree->lock); | ||
| 147 | } | ||
| 148 | spin_unlock(&tree->lock); | ||
| 149 | return 0; | ||
| 150 | } | ||
| 151 | |||
| 152 | /* | ||
| 153 | * find the leaf ref for a given extent. This returns the ref struct with | ||
| 154 | * a usage reference incremented | ||
| 155 | */ | ||
| 156 | struct btrfs_leaf_ref *btrfs_lookup_leaf_ref(struct btrfs_root *root, | ||
| 157 | u64 bytenr) | ||
| 158 | { | ||
| 159 | struct rb_node *rb; | ||
| 160 | struct btrfs_leaf_ref *ref = NULL; | ||
| 161 | struct btrfs_leaf_ref_tree *tree = root->ref_tree; | ||
| 162 | again: | ||
| 163 | if (tree) { | ||
| 164 | spin_lock(&tree->lock); | ||
| 165 | rb = tree_search(&tree->root, bytenr); | ||
| 166 | if (rb) | ||
| 167 | ref = rb_entry(rb, struct btrfs_leaf_ref, rb_node); | ||
| 168 | if (ref) | ||
| 169 | atomic_inc(&ref->usage); | ||
| 170 | spin_unlock(&tree->lock); | ||
| 171 | if (ref) | ||
| 172 | return ref; | ||
| 173 | } | ||
| 174 | if (tree != &root->fs_info->shared_ref_tree) { | ||
| 175 | tree = &root->fs_info->shared_ref_tree; | ||
| 176 | goto again; | ||
| 177 | } | ||
| 178 | return NULL; | ||
| 179 | } | ||
| 180 | |||
| 181 | /* | ||
| 182 | * add a fully filled in leaf ref struct | ||
| 183 | * remove all the refs older than a given root generation | ||
| 184 | */ | ||
| 185 | int btrfs_add_leaf_ref(struct btrfs_root *root, struct btrfs_leaf_ref *ref, | ||
| 186 | int shared) | ||
| 187 | { | ||
| 188 | int ret = 0; | ||
| 189 | struct rb_node *rb; | ||
| 190 | struct btrfs_leaf_ref_tree *tree = root->ref_tree; | ||
| 191 | |||
| 192 | if (shared) | ||
| 193 | tree = &root->fs_info->shared_ref_tree; | ||
| 194 | |||
| 195 | spin_lock(&tree->lock); | ||
| 196 | rb = tree_insert(&tree->root, ref->bytenr, &ref->rb_node); | ||
| 197 | if (rb) { | ||
| 198 | ret = -EEXIST; | ||
| 199 | } else { | ||
| 200 | atomic_inc(&ref->usage); | ||
| 201 | ref->tree = tree; | ||
| 202 | ref->in_tree = 1; | ||
| 203 | list_add_tail(&ref->list, &tree->list); | ||
| 204 | } | ||
| 205 | spin_unlock(&tree->lock); | ||
| 206 | return ret; | ||
| 207 | } | ||
| 208 | |||
| 209 | /* | ||
| 210 | * remove a single leaf ref from the tree. This drops the ref held by the tree | ||
| 211 | * only | ||
| 212 | */ | ||
| 213 | int btrfs_remove_leaf_ref(struct btrfs_root *root, struct btrfs_leaf_ref *ref) | ||
| 214 | { | ||
| 215 | struct btrfs_leaf_ref_tree *tree; | ||
| 216 | |||
| 217 | if (!xchg(&ref->in_tree, 0)) | ||
| 218 | return 0; | ||
| 219 | |||
| 220 | tree = ref->tree; | ||
| 221 | spin_lock(&tree->lock); | ||
| 222 | |||
| 223 | rb_erase(&ref->rb_node, &tree->root); | ||
| 224 | list_del_init(&ref->list); | ||
| 225 | |||
| 226 | spin_unlock(&tree->lock); | ||
| 227 | |||
| 228 | btrfs_free_leaf_ref(root, ref); | ||
| 229 | return 0; | ||
| 230 | } | ||
diff --git a/fs/btrfs/ref-cache.h b/fs/btrfs/ref-cache.h new file mode 100644 index 000000000000..16f3183d7c59 --- /dev/null +++ b/fs/btrfs/ref-cache.h | |||
| @@ -0,0 +1,77 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | #ifndef __REFCACHE__ | ||
| 19 | #define __REFCACHE__ | ||
| 20 | |||
| 21 | struct btrfs_extent_info { | ||
| 22 | /* bytenr and num_bytes find the extent in the extent allocation tree */ | ||
| 23 | u64 bytenr; | ||
| 24 | u64 num_bytes; | ||
| 25 | |||
| 26 | /* objectid and offset find the back reference for the file */ | ||
| 27 | u64 objectid; | ||
| 28 | u64 offset; | ||
| 29 | }; | ||
| 30 | |||
| 31 | struct btrfs_leaf_ref { | ||
| 32 | struct rb_node rb_node; | ||
| 33 | struct btrfs_leaf_ref_tree *tree; | ||
| 34 | int in_tree; | ||
| 35 | atomic_t usage; | ||
| 36 | |||
| 37 | u64 root_gen; | ||
| 38 | u64 bytenr; | ||
| 39 | u64 owner; | ||
| 40 | u64 generation; | ||
| 41 | int nritems; | ||
| 42 | |||
| 43 | struct list_head list; | ||
| 44 | struct btrfs_extent_info extents[]; | ||
| 45 | }; | ||
| 46 | |||
| 47 | static inline size_t btrfs_leaf_ref_size(int nr_extents) | ||
| 48 | { | ||
| 49 | return sizeof(struct btrfs_leaf_ref) + | ||
| 50 | sizeof(struct btrfs_extent_info) * nr_extents; | ||
| 51 | } | ||
| 52 | |||
| 53 | static inline void btrfs_leaf_ref_tree_init(struct btrfs_leaf_ref_tree *tree) | ||
| 54 | { | ||
| 55 | tree->root.rb_node = NULL; | ||
| 56 | INIT_LIST_HEAD(&tree->list); | ||
| 57 | spin_lock_init(&tree->lock); | ||
| 58 | } | ||
| 59 | |||
| 60 | static inline int btrfs_leaf_ref_tree_empty(struct btrfs_leaf_ref_tree *tree) | ||
| 61 | { | ||
| 62 | return RB_EMPTY_ROOT(&tree->root); | ||
| 63 | } | ||
| 64 | |||
| 65 | void btrfs_leaf_ref_tree_init(struct btrfs_leaf_ref_tree *tree); | ||
| 66 | struct btrfs_leaf_ref *btrfs_alloc_leaf_ref(struct btrfs_root *root, | ||
| 67 | int nr_extents); | ||
| 68 | void btrfs_free_leaf_ref(struct btrfs_root *root, struct btrfs_leaf_ref *ref); | ||
| 69 | struct btrfs_leaf_ref *btrfs_lookup_leaf_ref(struct btrfs_root *root, | ||
| 70 | u64 bytenr); | ||
| 71 | int btrfs_add_leaf_ref(struct btrfs_root *root, struct btrfs_leaf_ref *ref, | ||
| 72 | int shared); | ||
| 73 | int btrfs_remove_leaf_refs(struct btrfs_root *root, u64 max_root_gen, | ||
| 74 | int shared); | ||
| 75 | int btrfs_remove_leaf_ref(struct btrfs_root *root, struct btrfs_leaf_ref *ref); | ||
| 76 | |||
| 77 | #endif | ||
diff --git a/fs/btrfs/root-tree.c b/fs/btrfs/root-tree.c new file mode 100644 index 000000000000..b48650de4472 --- /dev/null +++ b/fs/btrfs/root-tree.c | |||
| @@ -0,0 +1,366 @@ | |||
| 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 | |||
| 19 | #include "ctree.h" | ||
| 20 | #include "transaction.h" | ||
| 21 | #include "disk-io.h" | ||
| 22 | #include "print-tree.h" | ||
| 23 | |||
| 24 | /* | ||
| 25 | * search forward for a root, starting with objectid 'search_start' | ||
| 26 | * if a root key is found, the objectid we find is filled into 'found_objectid' | ||
| 27 | * and 0 is returned. < 0 is returned on error, 1 if there is nothing | ||
| 28 | * left in the tree. | ||
| 29 | */ | ||
| 30 | int btrfs_search_root(struct btrfs_root *root, u64 search_start, | ||
| 31 | u64 *found_objectid) | ||
| 32 | { | ||
| 33 | struct btrfs_path *path; | ||
| 34 | struct btrfs_key search_key; | ||
| 35 | int ret; | ||
| 36 | |||
| 37 | root = root->fs_info->tree_root; | ||
| 38 | search_key.objectid = search_start; | ||
| 39 | search_key.type = (u8)-1; | ||
| 40 | search_key.offset = (u64)-1; | ||
| 41 | |||
| 42 | path = btrfs_alloc_path(); | ||
| 43 | BUG_ON(!path); | ||
| 44 | again: | ||
| 45 | ret = btrfs_search_slot(NULL, root, &search_key, path, 0, 0); | ||
| 46 | if (ret < 0) | ||
| 47 | goto out; | ||
| 48 | if (ret == 0) { | ||
| 49 | ret = 1; | ||
| 50 | goto out; | ||
| 51 | } | ||
| 52 | if (path->slots[0] >= btrfs_header_nritems(path->nodes[0])) { | ||
| 53 | ret = btrfs_next_leaf(root, path); | ||
| 54 | if (ret) | ||
| 55 | goto out; | ||
| 56 | } | ||
| 57 | btrfs_item_key_to_cpu(path->nodes[0], &search_key, path->slots[0]); | ||
| 58 | if (search_key.type != BTRFS_ROOT_ITEM_KEY) { | ||
| 59 | search_key.offset++; | ||
| 60 | btrfs_release_path(root, path); | ||
| 61 | goto again; | ||
| 62 | } | ||
| 63 | ret = 0; | ||
| 64 | *found_objectid = search_key.objectid; | ||
| 65 | |||
| 66 | out: | ||
| 67 | btrfs_free_path(path); | ||
| 68 | return ret; | ||
| 69 | } | ||
| 70 | |||
| 71 | /* | ||
| 72 | * lookup the root with the highest offset for a given objectid. The key we do | ||
| 73 | * find is copied into 'key'. If we find something return 0, otherwise 1, < 0 | ||
| 74 | * on error. | ||
| 75 | */ | ||
| 76 | int btrfs_find_last_root(struct btrfs_root *root, u64 objectid, | ||
| 77 | struct btrfs_root_item *item, struct btrfs_key *key) | ||
| 78 | { | ||
| 79 | struct btrfs_path *path; | ||
| 80 | struct btrfs_key search_key; | ||
| 81 | struct btrfs_key found_key; | ||
| 82 | struct extent_buffer *l; | ||
| 83 | int ret; | ||
| 84 | int slot; | ||
| 85 | |||
| 86 | search_key.objectid = objectid; | ||
| 87 | search_key.type = BTRFS_ROOT_ITEM_KEY; | ||
| 88 | search_key.offset = (u64)-1; | ||
| 89 | |||
| 90 | path = btrfs_alloc_path(); | ||
| 91 | BUG_ON(!path); | ||
| 92 | ret = btrfs_search_slot(NULL, root, &search_key, path, 0, 0); | ||
| 93 | if (ret < 0) | ||
| 94 | goto out; | ||
| 95 | |||
| 96 | BUG_ON(ret == 0); | ||
| 97 | l = path->nodes[0]; | ||
| 98 | BUG_ON(path->slots[0] == 0); | ||
| 99 | slot = path->slots[0] - 1; | ||
| 100 | btrfs_item_key_to_cpu(l, &found_key, slot); | ||
| 101 | if (found_key.objectid != objectid) { | ||
| 102 | ret = 1; | ||
| 103 | goto out; | ||
| 104 | } | ||
| 105 | read_extent_buffer(l, item, btrfs_item_ptr_offset(l, slot), | ||
| 106 | sizeof(*item)); | ||
| 107 | memcpy(key, &found_key, sizeof(found_key)); | ||
| 108 | ret = 0; | ||
| 109 | out: | ||
| 110 | btrfs_free_path(path); | ||
| 111 | return ret; | ||
| 112 | } | ||
| 113 | |||
| 114 | /* | ||
| 115 | * copy the data in 'item' into the btree | ||
| 116 | */ | ||
| 117 | int btrfs_update_root(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 118 | *root, struct btrfs_key *key, struct btrfs_root_item | ||
| 119 | *item) | ||
| 120 | { | ||
| 121 | struct btrfs_path *path; | ||
| 122 | struct extent_buffer *l; | ||
| 123 | int ret; | ||
| 124 | int slot; | ||
| 125 | unsigned long ptr; | ||
| 126 | |||
| 127 | path = btrfs_alloc_path(); | ||
| 128 | BUG_ON(!path); | ||
| 129 | ret = btrfs_search_slot(trans, root, key, path, 0, 1); | ||
| 130 | if (ret < 0) | ||
| 131 | goto out; | ||
| 132 | |||
| 133 | if (ret != 0) { | ||
| 134 | btrfs_print_leaf(root, path->nodes[0]); | ||
| 135 | printk(KERN_CRIT "unable to update root key %llu %u %llu\n", | ||
| 136 | (unsigned long long)key->objectid, key->type, | ||
| 137 | (unsigned long long)key->offset); | ||
| 138 | BUG_ON(1); | ||
| 139 | } | ||
| 140 | |||
| 141 | l = path->nodes[0]; | ||
| 142 | slot = path->slots[0]; | ||
| 143 | ptr = btrfs_item_ptr_offset(l, slot); | ||
| 144 | write_extent_buffer(l, item, ptr, sizeof(*item)); | ||
| 145 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 146 | out: | ||
| 147 | btrfs_release_path(root, path); | ||
| 148 | btrfs_free_path(path); | ||
| 149 | return ret; | ||
| 150 | } | ||
| 151 | |||
| 152 | int btrfs_insert_root(struct btrfs_trans_handle *trans, struct btrfs_root | ||
| 153 | *root, struct btrfs_key *key, struct btrfs_root_item | ||
| 154 | *item) | ||
| 155 | { | ||
| 156 | int ret; | ||
| 157 | ret = btrfs_insert_item(trans, root, key, item, sizeof(*item)); | ||
| 158 | return ret; | ||
| 159 | } | ||
| 160 | |||
| 161 | /* | ||
| 162 | * at mount time we want to find all the old transaction snapshots that were in | ||
| 163 | * the process of being deleted if we crashed. This is any root item with an | ||
| 164 | * offset lower than the latest root. They need to be queued for deletion to | ||
| 165 | * finish what was happening when we crashed. | ||
| 166 | */ | ||
| 167 | int btrfs_find_dead_roots(struct btrfs_root *root, u64 objectid, | ||
| 168 | struct btrfs_root *latest) | ||
| 169 | { | ||
| 170 | struct btrfs_root *dead_root; | ||
| 171 | struct btrfs_item *item; | ||
| 172 | struct btrfs_root_item *ri; | ||
| 173 | struct btrfs_key key; | ||
| 174 | struct btrfs_key found_key; | ||
| 175 | struct btrfs_path *path; | ||
| 176 | int ret; | ||
| 177 | u32 nritems; | ||
| 178 | struct extent_buffer *leaf; | ||
| 179 | int slot; | ||
| 180 | |||
| 181 | key.objectid = objectid; | ||
| 182 | btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY); | ||
| 183 | key.offset = 0; | ||
| 184 | path = btrfs_alloc_path(); | ||
| 185 | if (!path) | ||
| 186 | return -ENOMEM; | ||
| 187 | |||
| 188 | again: | ||
| 189 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 190 | if (ret < 0) | ||
| 191 | goto err; | ||
| 192 | while (1) { | ||
| 193 | leaf = path->nodes[0]; | ||
| 194 | nritems = btrfs_header_nritems(leaf); | ||
| 195 | slot = path->slots[0]; | ||
| 196 | if (slot >= nritems) { | ||
| 197 | ret = btrfs_next_leaf(root, path); | ||
| 198 | if (ret) | ||
| 199 | break; | ||
| 200 | leaf = path->nodes[0]; | ||
| 201 | nritems = btrfs_header_nritems(leaf); | ||
| 202 | slot = path->slots[0]; | ||
| 203 | } | ||
| 204 | item = btrfs_item_nr(leaf, slot); | ||
| 205 | btrfs_item_key_to_cpu(leaf, &key, slot); | ||
| 206 | if (btrfs_key_type(&key) != BTRFS_ROOT_ITEM_KEY) | ||
| 207 | goto next; | ||
| 208 | |||
| 209 | if (key.objectid < objectid) | ||
| 210 | goto next; | ||
| 211 | |||
| 212 | if (key.objectid > objectid) | ||
| 213 | break; | ||
| 214 | |||
| 215 | ri = btrfs_item_ptr(leaf, slot, struct btrfs_root_item); | ||
| 216 | if (btrfs_disk_root_refs(leaf, ri) != 0) | ||
| 217 | goto next; | ||
| 218 | |||
| 219 | memcpy(&found_key, &key, sizeof(key)); | ||
| 220 | key.offset++; | ||
| 221 | btrfs_release_path(root, path); | ||
| 222 | dead_root = | ||
| 223 | btrfs_read_fs_root_no_radix(root->fs_info->tree_root, | ||
| 224 | &found_key); | ||
| 225 | if (IS_ERR(dead_root)) { | ||
| 226 | ret = PTR_ERR(dead_root); | ||
| 227 | goto err; | ||
| 228 | } | ||
| 229 | |||
| 230 | if (objectid == BTRFS_TREE_RELOC_OBJECTID) | ||
| 231 | ret = btrfs_add_dead_reloc_root(dead_root); | ||
| 232 | else | ||
| 233 | ret = btrfs_add_dead_root(dead_root, latest); | ||
| 234 | if (ret) | ||
| 235 | goto err; | ||
| 236 | goto again; | ||
| 237 | next: | ||
| 238 | slot++; | ||
| 239 | path->slots[0]++; | ||
| 240 | } | ||
| 241 | ret = 0; | ||
| 242 | err: | ||
| 243 | btrfs_free_path(path); | ||
| 244 | return ret; | ||
| 245 | } | ||
| 246 | |||
| 247 | /* drop the root item for 'key' from 'root' */ | ||
| 248 | int btrfs_del_root(struct btrfs_trans_handle *trans, struct btrfs_root *root, | ||
| 249 | struct btrfs_key *key) | ||
| 250 | { | ||
| 251 | struct btrfs_path *path; | ||
| 252 | int ret; | ||
| 253 | u32 refs; | ||
| 254 | struct btrfs_root_item *ri; | ||
| 255 | struct extent_buffer *leaf; | ||
| 256 | |||
| 257 | path = btrfs_alloc_path(); | ||
| 258 | BUG_ON(!path); | ||
| 259 | ret = btrfs_search_slot(trans, root, key, path, -1, 1); | ||
| 260 | if (ret < 0) | ||
| 261 | goto out; | ||
| 262 | |||
| 263 | BUG_ON(ret != 0); | ||
| 264 | leaf = path->nodes[0]; | ||
| 265 | ri = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_root_item); | ||
| 266 | |||
| 267 | refs = btrfs_disk_root_refs(leaf, ri); | ||
| 268 | BUG_ON(refs != 0); | ||
| 269 | ret = btrfs_del_item(trans, root, path); | ||
| 270 | out: | ||
| 271 | btrfs_release_path(root, path); | ||
| 272 | btrfs_free_path(path); | ||
| 273 | return ret; | ||
| 274 | } | ||
| 275 | |||
| 276 | #if 0 /* this will get used when snapshot deletion is implemented */ | ||
| 277 | int btrfs_del_root_ref(struct btrfs_trans_handle *trans, | ||
| 278 | struct btrfs_root *tree_root, | ||
| 279 | u64 root_id, u8 type, u64 ref_id) | ||
| 280 | { | ||
| 281 | struct btrfs_key key; | ||
| 282 | int ret; | ||
| 283 | struct btrfs_path *path; | ||
| 284 | |||
| 285 | path = btrfs_alloc_path(); | ||
| 286 | |||
| 287 | key.objectid = root_id; | ||
| 288 | key.type = type; | ||
| 289 | key.offset = ref_id; | ||
| 290 | |||
| 291 | ret = btrfs_search_slot(trans, tree_root, &key, path, -1, 1); | ||
| 292 | BUG_ON(ret); | ||
| 293 | |||
| 294 | ret = btrfs_del_item(trans, tree_root, path); | ||
| 295 | BUG_ON(ret); | ||
| 296 | |||
| 297 | btrfs_free_path(path); | ||
| 298 | return ret; | ||
| 299 | } | ||
| 300 | #endif | ||
| 301 | |||
| 302 | int btrfs_find_root_ref(struct btrfs_root *tree_root, | ||
| 303 | struct btrfs_path *path, | ||
| 304 | u64 root_id, u64 ref_id) | ||
| 305 | { | ||
| 306 | struct btrfs_key key; | ||
| 307 | int ret; | ||
| 308 | |||
| 309 | key.objectid = root_id; | ||
| 310 | key.type = BTRFS_ROOT_REF_KEY; | ||
| 311 | key.offset = ref_id; | ||
| 312 | |||
| 313 | ret = btrfs_search_slot(NULL, tree_root, &key, path, 0, 0); | ||
| 314 | return ret; | ||
| 315 | } | ||
| 316 | |||
| 317 | |||
| 318 | /* | ||
| 319 | * add a btrfs_root_ref item. type is either BTRFS_ROOT_REF_KEY | ||
| 320 | * or BTRFS_ROOT_BACKREF_KEY. | ||
| 321 | * | ||
| 322 | * The dirid, sequence, name and name_len refer to the directory entry | ||
| 323 | * that is referencing the root. | ||
| 324 | * | ||
| 325 | * For a forward ref, the root_id is the id of the tree referencing | ||
| 326 | * the root and ref_id is the id of the subvol or snapshot. | ||
| 327 | * | ||
| 328 | * For a back ref the root_id is the id of the subvol or snapshot and | ||
| 329 | * ref_id is the id of the tree referencing it. | ||
| 330 | */ | ||
| 331 | int btrfs_add_root_ref(struct btrfs_trans_handle *trans, | ||
| 332 | struct btrfs_root *tree_root, | ||
| 333 | u64 root_id, u8 type, u64 ref_id, | ||
| 334 | u64 dirid, u64 sequence, | ||
| 335 | const char *name, int name_len) | ||
| 336 | { | ||
| 337 | struct btrfs_key key; | ||
| 338 | int ret; | ||
| 339 | struct btrfs_path *path; | ||
| 340 | struct btrfs_root_ref *ref; | ||
| 341 | struct extent_buffer *leaf; | ||
| 342 | unsigned long ptr; | ||
| 343 | |||
| 344 | |||
| 345 | path = btrfs_alloc_path(); | ||
| 346 | |||
| 347 | key.objectid = root_id; | ||
| 348 | key.type = type; | ||
| 349 | key.offset = ref_id; | ||
| 350 | |||
| 351 | ret = btrfs_insert_empty_item(trans, tree_root, path, &key, | ||
| 352 | sizeof(*ref) + name_len); | ||
| 353 | BUG_ON(ret); | ||
| 354 | |||
| 355 | leaf = path->nodes[0]; | ||
| 356 | ref = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_root_ref); | ||
| 357 | btrfs_set_root_ref_dirid(leaf, ref, dirid); | ||
| 358 | btrfs_set_root_ref_sequence(leaf, ref, sequence); | ||
| 359 | btrfs_set_root_ref_name_len(leaf, ref, name_len); | ||
| 360 | ptr = (unsigned long)(ref + 1); | ||
| 361 | write_extent_buffer(leaf, name, ptr, name_len); | ||
| 362 | btrfs_mark_buffer_dirty(leaf); | ||
| 363 | |||
| 364 | btrfs_free_path(path); | ||
| 365 | return ret; | ||
| 366 | } | ||
diff --git a/fs/btrfs/struct-funcs.c b/fs/btrfs/struct-funcs.c new file mode 100644 index 000000000000..c0f7ecaf1e79 --- /dev/null +++ b/fs/btrfs/struct-funcs.c | |||
| @@ -0,0 +1,139 @@ | |||
| 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 | |||
| 19 | #include <linux/highmem.h> | ||
| 20 | |||
| 21 | /* this is some deeply nasty code. ctree.h has a different | ||
| 22 | * definition for this BTRFS_SETGET_FUNCS macro, behind a #ifndef | ||
| 23 | * | ||
| 24 | * The end result is that anyone who #includes ctree.h gets a | ||
| 25 | * declaration for the btrfs_set_foo functions and btrfs_foo functions | ||
| 26 | * | ||
| 27 | * This file declares the macros and then #includes ctree.h, which results | ||
| 28 | * in cpp creating the function here based on the template below. | ||
| 29 | * | ||
| 30 | * These setget functions do all the extent_buffer related mapping | ||
| 31 | * required to efficiently read and write specific fields in the extent | ||
| 32 | * buffers. Every pointer to metadata items in btrfs is really just | ||
| 33 | * an unsigned long offset into the extent buffer which has been | ||
| 34 | * cast to a specific type. This gives us all the gcc type checking. | ||
| 35 | * | ||
| 36 | * The extent buffer api is used to do all the kmapping and page | ||
| 37 | * spanning work required to get extent buffers in highmem and have | ||
| 38 | * a metadata blocksize different from the page size. | ||
| 39 | * | ||
| 40 | * The macro starts with a simple function prototype declaration so that | ||
| 41 | * sparse won't complain about it being static. | ||
| 42 | */ | ||
| 43 | |||
| 44 | #define BTRFS_SETGET_FUNCS(name, type, member, bits) \ | ||
| 45 | u##bits btrfs_##name(struct extent_buffer *eb, type *s); \ | ||
| 46 | void btrfs_set_##name(struct extent_buffer *eb, type *s, u##bits val); \ | ||
| 47 | u##bits btrfs_##name(struct extent_buffer *eb, \ | ||
| 48 | type *s) \ | ||
| 49 | { \ | ||
| 50 | unsigned long part_offset = (unsigned long)s; \ | ||
| 51 | unsigned long offset = part_offset + offsetof(type, member); \ | ||
| 52 | type *p; \ | ||
| 53 | /* ugly, but we want the fast path here */ \ | ||
| 54 | if (eb->map_token && offset >= eb->map_start && \ | ||
| 55 | offset + sizeof(((type *)0)->member) <= eb->map_start + \ | ||
| 56 | eb->map_len) { \ | ||
| 57 | p = (type *)(eb->kaddr + part_offset - eb->map_start); \ | ||
| 58 | return le##bits##_to_cpu(p->member); \ | ||
| 59 | } \ | ||
| 60 | { \ | ||
| 61 | int err; \ | ||
| 62 | char *map_token; \ | ||
| 63 | char *kaddr; \ | ||
| 64 | int unmap_on_exit = (eb->map_token == NULL); \ | ||
| 65 | unsigned long map_start; \ | ||
| 66 | unsigned long map_len; \ | ||
| 67 | u##bits res; \ | ||
| 68 | err = map_extent_buffer(eb, offset, \ | ||
| 69 | sizeof(((type *)0)->member), \ | ||
| 70 | &map_token, &kaddr, \ | ||
| 71 | &map_start, &map_len, KM_USER1); \ | ||
| 72 | if (err) { \ | ||
| 73 | __le##bits leres; \ | ||
| 74 | read_eb_member(eb, s, type, member, &leres); \ | ||
| 75 | return le##bits##_to_cpu(leres); \ | ||
| 76 | } \ | ||
| 77 | p = (type *)(kaddr + part_offset - map_start); \ | ||
| 78 | res = le##bits##_to_cpu(p->member); \ | ||
| 79 | if (unmap_on_exit) \ | ||
| 80 | unmap_extent_buffer(eb, map_token, KM_USER1); \ | ||
| 81 | return res; \ | ||
| 82 | } \ | ||
| 83 | } \ | ||
| 84 | void btrfs_set_##name(struct extent_buffer *eb, \ | ||
| 85 | type *s, u##bits val) \ | ||
| 86 | { \ | ||
| 87 | unsigned long part_offset = (unsigned long)s; \ | ||
| 88 | unsigned long offset = part_offset + offsetof(type, member); \ | ||
| 89 | type *p; \ | ||
| 90 | /* ugly, but we want the fast path here */ \ | ||
| 91 | if (eb->map_token && offset >= eb->map_start && \ | ||
| 92 | offset + sizeof(((type *)0)->member) <= eb->map_start + \ | ||
| 93 | eb->map_len) { \ | ||
| 94 | p = (type *)(eb->kaddr + part_offset - eb->map_start); \ | ||
| 95 | p->member = cpu_to_le##bits(val); \ | ||
| 96 | return; \ | ||
| 97 | } \ | ||
| 98 | { \ | ||
| 99 | int err; \ | ||
| 100 | char *map_token; \ | ||
| 101 | char *kaddr; \ | ||
| 102 | int unmap_on_exit = (eb->map_token == NULL); \ | ||
| 103 | unsigned long map_start; \ | ||
| 104 | unsigned long map_len; \ | ||
| 105 | err = map_extent_buffer(eb, offset, \ | ||
| 106 | sizeof(((type *)0)->member), \ | ||
| 107 | &map_token, &kaddr, \ | ||
| 108 | &map_start, &map_len, KM_USER1); \ | ||
| 109 | if (err) { \ | ||
| 110 | __le##bits val2; \ | ||
| 111 | val2 = cpu_to_le##bits(val); \ | ||
| 112 | write_eb_member(eb, s, type, member, &val2); \ | ||
| 113 | return; \ | ||
| 114 | } \ | ||
| 115 | p = (type *)(kaddr + part_offset - map_start); \ | ||
| 116 | p->member = cpu_to_le##bits(val); \ | ||
| 117 | if (unmap_on_exit) \ | ||
| 118 | unmap_extent_buffer(eb, map_token, KM_USER1); \ | ||
| 119 | } \ | ||
| 120 | } | ||
| 121 | |||
| 122 | #include "ctree.h" | ||
| 123 | |||
| 124 | void btrfs_node_key(struct extent_buffer *eb, | ||
| 125 | struct btrfs_disk_key *disk_key, int nr) | ||
| 126 | { | ||
| 127 | unsigned long ptr = btrfs_node_key_ptr_offset(nr); | ||
| 128 | if (eb->map_token && ptr >= eb->map_start && | ||
| 129 | ptr + sizeof(*disk_key) <= eb->map_start + eb->map_len) { | ||
| 130 | memcpy(disk_key, eb->kaddr + ptr - eb->map_start, | ||
| 131 | sizeof(*disk_key)); | ||
| 132 | return; | ||
| 133 | } else if (eb->map_token) { | ||
| 134 | unmap_extent_buffer(eb, eb->map_token, KM_USER1); | ||
| 135 | eb->map_token = NULL; | ||
| 136 | } | ||
| 137 | read_eb_member(eb, (struct btrfs_key_ptr *)ptr, | ||
| 138 | struct btrfs_key_ptr, key, disk_key); | ||
| 139 | } | ||
diff --git a/fs/btrfs/super.c b/fs/btrfs/super.c new file mode 100644 index 000000000000..0a14b495532f --- /dev/null +++ b/fs/btrfs/super.c | |||
| @@ -0,0 +1,722 @@ | |||
| 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 | |||
| 19 | #include <linux/blkdev.h> | ||
| 20 | #include <linux/module.h> | ||
| 21 | #include <linux/buffer_head.h> | ||
| 22 | #include <linux/fs.h> | ||
| 23 | #include <linux/pagemap.h> | ||
| 24 | #include <linux/highmem.h> | ||
| 25 | #include <linux/time.h> | ||
| 26 | #include <linux/init.h> | ||
| 27 | #include <linux/string.h> | ||
| 28 | #include <linux/smp_lock.h> | ||
| 29 | #include <linux/backing-dev.h> | ||
| 30 | #include <linux/mount.h> | ||
| 31 | #include <linux/mpage.h> | ||
| 32 | #include <linux/swap.h> | ||
| 33 | #include <linux/writeback.h> | ||
| 34 | #include <linux/statfs.h> | ||
| 35 | #include <linux/compat.h> | ||
| 36 | #include <linux/parser.h> | ||
| 37 | #include <linux/ctype.h> | ||
| 38 | #include <linux/namei.h> | ||
| 39 | #include <linux/miscdevice.h> | ||
| 40 | #include <linux/version.h> | ||
| 41 | #include "compat.h" | ||
| 42 | #include "ctree.h" | ||
| 43 | #include "disk-io.h" | ||
| 44 | #include "transaction.h" | ||
| 45 | #include "btrfs_inode.h" | ||
| 46 | #include "ioctl.h" | ||
| 47 | #include "print-tree.h" | ||
| 48 | #include "xattr.h" | ||
| 49 | #include "volumes.h" | ||
| 50 | #include "version.h" | ||
| 51 | #include "export.h" | ||
| 52 | #include "compression.h" | ||
| 53 | |||
| 54 | #define BTRFS_SUPER_MAGIC 0x9123683E | ||
| 55 | |||
| 56 | static struct super_operations btrfs_super_ops; | ||
| 57 | |||
| 58 | static void btrfs_put_super(struct super_block *sb) | ||
| 59 | { | ||
| 60 | struct btrfs_root *root = btrfs_sb(sb); | ||
| 61 | int ret; | ||
| 62 | |||
| 63 | ret = close_ctree(root); | ||
| 64 | sb->s_fs_info = NULL; | ||
| 65 | } | ||
| 66 | |||
| 67 | enum { | ||
| 68 | Opt_degraded, Opt_subvol, Opt_device, Opt_nodatasum, Opt_nodatacow, | ||
| 69 | Opt_max_extent, Opt_max_inline, Opt_alloc_start, Opt_nobarrier, | ||
| 70 | Opt_ssd, Opt_thread_pool, Opt_noacl, Opt_compress, Opt_err, | ||
| 71 | }; | ||
| 72 | |||
| 73 | static match_table_t tokens = { | ||
| 74 | {Opt_degraded, "degraded"}, | ||
| 75 | {Opt_subvol, "subvol=%s"}, | ||
| 76 | {Opt_device, "device=%s"}, | ||
| 77 | {Opt_nodatasum, "nodatasum"}, | ||
| 78 | {Opt_nodatacow, "nodatacow"}, | ||
| 79 | {Opt_nobarrier, "nobarrier"}, | ||
| 80 | {Opt_max_extent, "max_extent=%s"}, | ||
| 81 | {Opt_max_inline, "max_inline=%s"}, | ||
| 82 | {Opt_alloc_start, "alloc_start=%s"}, | ||
| 83 | {Opt_thread_pool, "thread_pool=%d"}, | ||
| 84 | {Opt_compress, "compress"}, | ||
| 85 | {Opt_ssd, "ssd"}, | ||
| 86 | {Opt_noacl, "noacl"}, | ||
| 87 | {Opt_err, NULL}, | ||
| 88 | }; | ||
| 89 | |||
| 90 | u64 btrfs_parse_size(char *str) | ||
| 91 | { | ||
| 92 | u64 res; | ||
| 93 | int mult = 1; | ||
| 94 | char *end; | ||
| 95 | char last; | ||
| 96 | |||
| 97 | res = simple_strtoul(str, &end, 10); | ||
| 98 | |||
| 99 | last = end[0]; | ||
| 100 | if (isalpha(last)) { | ||
| 101 | last = tolower(last); | ||
| 102 | switch (last) { | ||
| 103 | case 'g': | ||
| 104 | mult *= 1024; | ||
| 105 | case 'm': | ||
| 106 | mult *= 1024; | ||
| 107 | case 'k': | ||
| 108 | mult *= 1024; | ||
| 109 | } | ||
| 110 | res = res * mult; | ||
| 111 | } | ||
| 112 | return res; | ||
| 113 | } | ||
| 114 | |||
| 115 | /* | ||
| 116 | * Regular mount options parser. Everything that is needed only when | ||
| 117 | * reading in a new superblock is parsed here. | ||
| 118 | */ | ||
| 119 | int btrfs_parse_options(struct btrfs_root *root, char *options) | ||
| 120 | { | ||
| 121 | struct btrfs_fs_info *info = root->fs_info; | ||
| 122 | substring_t args[MAX_OPT_ARGS]; | ||
| 123 | char *p, *num; | ||
| 124 | int intarg; | ||
| 125 | |||
| 126 | if (!options) | ||
| 127 | return 0; | ||
| 128 | |||
| 129 | /* | ||
| 130 | * strsep changes the string, duplicate it because parse_options | ||
| 131 | * gets called twice | ||
| 132 | */ | ||
| 133 | options = kstrdup(options, GFP_NOFS); | ||
| 134 | if (!options) | ||
| 135 | return -ENOMEM; | ||
| 136 | |||
| 137 | |||
| 138 | while ((p = strsep(&options, ",")) != NULL) { | ||
| 139 | int token; | ||
| 140 | if (!*p) | ||
| 141 | continue; | ||
| 142 | |||
| 143 | token = match_token(p, tokens, args); | ||
| 144 | switch (token) { | ||
| 145 | case Opt_degraded: | ||
| 146 | printk(KERN_INFO "btrfs: allowing degraded mounts\n"); | ||
| 147 | btrfs_set_opt(info->mount_opt, DEGRADED); | ||
| 148 | break; | ||
| 149 | case Opt_subvol: | ||
| 150 | case Opt_device: | ||
| 151 | /* | ||
| 152 | * These are parsed by btrfs_parse_early_options | ||
| 153 | * and can be happily ignored here. | ||
| 154 | */ | ||
| 155 | break; | ||
| 156 | case Opt_nodatasum: | ||
| 157 | printk(KERN_INFO "btrfs: setting nodatacsum\n"); | ||
| 158 | btrfs_set_opt(info->mount_opt, NODATASUM); | ||
| 159 | break; | ||
| 160 | case Opt_nodatacow: | ||
| 161 | printk(KERN_INFO "btrfs: setting nodatacow\n"); | ||
| 162 | btrfs_set_opt(info->mount_opt, NODATACOW); | ||
| 163 | btrfs_set_opt(info->mount_opt, NODATASUM); | ||
| 164 | break; | ||
| 165 | case Opt_compress: | ||
| 166 | printk(KERN_INFO "btrfs: use compression\n"); | ||
| 167 | btrfs_set_opt(info->mount_opt, COMPRESS); | ||
| 168 | break; | ||
| 169 | case Opt_ssd: | ||
| 170 | printk(KERN_INFO "btrfs: use ssd allocation scheme\n"); | ||
| 171 | btrfs_set_opt(info->mount_opt, SSD); | ||
| 172 | break; | ||
| 173 | case Opt_nobarrier: | ||
| 174 | printk(KERN_INFO "btrfs: turning off barriers\n"); | ||
| 175 | btrfs_set_opt(info->mount_opt, NOBARRIER); | ||
| 176 | break; | ||
| 177 | case Opt_thread_pool: | ||
| 178 | intarg = 0; | ||
| 179 | match_int(&args[0], &intarg); | ||
| 180 | if (intarg) { | ||
| 181 | info->thread_pool_size = intarg; | ||
| 182 | printk(KERN_INFO "btrfs: thread pool %d\n", | ||
| 183 | info->thread_pool_size); | ||
| 184 | } | ||
| 185 | break; | ||
| 186 | case Opt_max_extent: | ||
| 187 | num = match_strdup(&args[0]); | ||
| 188 | if (num) { | ||
| 189 | info->max_extent = btrfs_parse_size(num); | ||
| 190 | kfree(num); | ||
| 191 | |||
| 192 | info->max_extent = max_t(u64, | ||
| 193 | info->max_extent, root->sectorsize); | ||
| 194 | printk(KERN_INFO "btrfs: max_extent at %llu\n", | ||
| 195 | info->max_extent); | ||
| 196 | } | ||
| 197 | break; | ||
| 198 | case Opt_max_inline: | ||
| 199 | num = match_strdup(&args[0]); | ||
| 200 | if (num) { | ||
| 201 | info->max_inline = btrfs_parse_size(num); | ||
| 202 | kfree(num); | ||
| 203 | |||
| 204 | if (info->max_inline) { | ||
| 205 | info->max_inline = max_t(u64, | ||
| 206 | info->max_inline, | ||
| 207 | root->sectorsize); | ||
| 208 | } | ||
| 209 | printk(KERN_INFO "btrfs: max_inline at %llu\n", | ||
| 210 | info->max_inline); | ||
| 211 | } | ||
| 212 | break; | ||
| 213 | case Opt_alloc_start: | ||
| 214 | num = match_strdup(&args[0]); | ||
| 215 | if (num) { | ||
| 216 | info->alloc_start = btrfs_parse_size(num); | ||
| 217 | kfree(num); | ||
| 218 | printk(KERN_INFO | ||
| 219 | "btrfs: allocations start at %llu\n", | ||
| 220 | info->alloc_start); | ||
| 221 | } | ||
| 222 | break; | ||
| 223 | case Opt_noacl: | ||
| 224 | root->fs_info->sb->s_flags &= ~MS_POSIXACL; | ||
| 225 | break; | ||
| 226 | default: | ||
| 227 | break; | ||
| 228 | } | ||
| 229 | } | ||
| 230 | kfree(options); | ||
| 231 | return 0; | ||
| 232 | } | ||
| 233 | |||
| 234 | /* | ||
| 235 | * Parse mount options that are required early in the mount process. | ||
| 236 | * | ||
| 237 | * All other options will be parsed on much later in the mount process and | ||
| 238 | * only when we need to allocate a new super block. | ||
| 239 | */ | ||
| 240 | static int btrfs_parse_early_options(const char *options, fmode_t flags, | ||
| 241 | void *holder, char **subvol_name, | ||
| 242 | struct btrfs_fs_devices **fs_devices) | ||
| 243 | { | ||
| 244 | substring_t args[MAX_OPT_ARGS]; | ||
| 245 | char *opts, *p; | ||
| 246 | int error = 0; | ||
| 247 | |||
| 248 | if (!options) | ||
| 249 | goto out; | ||
| 250 | |||
| 251 | /* | ||
| 252 | * strsep changes the string, duplicate it because parse_options | ||
| 253 | * gets called twice | ||
| 254 | */ | ||
| 255 | opts = kstrdup(options, GFP_KERNEL); | ||
| 256 | if (!opts) | ||
| 257 | return -ENOMEM; | ||
| 258 | |||
| 259 | while ((p = strsep(&opts, ",")) != NULL) { | ||
| 260 | int token; | ||
| 261 | if (!*p) | ||
| 262 | continue; | ||
| 263 | |||
| 264 | token = match_token(p, tokens, args); | ||
| 265 | switch (token) { | ||
| 266 | case Opt_subvol: | ||
| 267 | *subvol_name = match_strdup(&args[0]); | ||
| 268 | break; | ||
| 269 | case Opt_device: | ||
| 270 | error = btrfs_scan_one_device(match_strdup(&args[0]), | ||
| 271 | flags, holder, fs_devices); | ||
| 272 | if (error) | ||
| 273 | goto out_free_opts; | ||
| 274 | break; | ||
| 275 | default: | ||
| 276 | break; | ||
| 277 | } | ||
| 278 | } | ||
| 279 | |||
| 280 | out_free_opts: | ||
| 281 | kfree(opts); | ||
| 282 | out: | ||
| 283 | /* | ||
| 284 | * If no subvolume name is specified we use the default one. Allocate | ||
| 285 | * a copy of the string "." here so that code later in the | ||
| 286 | * mount path doesn't care if it's the default volume or another one. | ||
| 287 | */ | ||
| 288 | if (!*subvol_name) { | ||
| 289 | *subvol_name = kstrdup(".", GFP_KERNEL); | ||
| 290 | if (!*subvol_name) | ||
| 291 | return -ENOMEM; | ||
| 292 | } | ||
| 293 | return error; | ||
| 294 | } | ||
| 295 | |||
| 296 | static int btrfs_fill_super(struct super_block *sb, | ||
| 297 | struct btrfs_fs_devices *fs_devices, | ||
| 298 | void *data, int silent) | ||
| 299 | { | ||
| 300 | struct inode *inode; | ||
| 301 | struct dentry *root_dentry; | ||
| 302 | struct btrfs_super_block *disk_super; | ||
| 303 | struct btrfs_root *tree_root; | ||
| 304 | struct btrfs_inode *bi; | ||
| 305 | int err; | ||
| 306 | |||
| 307 | sb->s_maxbytes = MAX_LFS_FILESIZE; | ||
| 308 | sb->s_magic = BTRFS_SUPER_MAGIC; | ||
| 309 | sb->s_op = &btrfs_super_ops; | ||
| 310 | sb->s_export_op = &btrfs_export_ops; | ||
| 311 | sb->s_xattr = btrfs_xattr_handlers; | ||
| 312 | sb->s_time_gran = 1; | ||
| 313 | sb->s_flags |= MS_POSIXACL; | ||
| 314 | |||
| 315 | tree_root = open_ctree(sb, fs_devices, (char *)data); | ||
| 316 | |||
| 317 | if (IS_ERR(tree_root)) { | ||
| 318 | printk("btrfs: open_ctree failed\n"); | ||
| 319 | return PTR_ERR(tree_root); | ||
| 320 | } | ||
| 321 | sb->s_fs_info = tree_root; | ||
| 322 | disk_super = &tree_root->fs_info->super_copy; | ||
| 323 | inode = btrfs_iget_locked(sb, BTRFS_FIRST_FREE_OBJECTID, | ||
| 324 | tree_root->fs_info->fs_root); | ||
| 325 | bi = BTRFS_I(inode); | ||
| 326 | bi->location.objectid = inode->i_ino; | ||
| 327 | bi->location.offset = 0; | ||
| 328 | bi->root = tree_root->fs_info->fs_root; | ||
| 329 | |||
| 330 | btrfs_set_key_type(&bi->location, BTRFS_INODE_ITEM_KEY); | ||
| 331 | |||
| 332 | if (!inode) { | ||
| 333 | err = -ENOMEM; | ||
| 334 | goto fail_close; | ||
| 335 | } | ||
| 336 | if (inode->i_state & I_NEW) { | ||
| 337 | btrfs_read_locked_inode(inode); | ||
| 338 | unlock_new_inode(inode); | ||
| 339 | } | ||
| 340 | |||
| 341 | root_dentry = d_alloc_root(inode); | ||
| 342 | if (!root_dentry) { | ||
| 343 | iput(inode); | ||
| 344 | err = -ENOMEM; | ||
| 345 | goto fail_close; | ||
| 346 | } | ||
| 347 | #if 0 | ||
| 348 | /* this does the super kobj at the same time */ | ||
| 349 | err = btrfs_sysfs_add_super(tree_root->fs_info); | ||
| 350 | if (err) | ||
| 351 | goto fail_close; | ||
| 352 | #endif | ||
| 353 | |||
| 354 | sb->s_root = root_dentry; | ||
| 355 | |||
| 356 | save_mount_options(sb, data); | ||
| 357 | return 0; | ||
| 358 | |||
| 359 | fail_close: | ||
| 360 | close_ctree(tree_root); | ||
| 361 | return err; | ||
| 362 | } | ||
| 363 | |||
| 364 | int btrfs_sync_fs(struct super_block *sb, int wait) | ||
| 365 | { | ||
| 366 | struct btrfs_trans_handle *trans; | ||
| 367 | struct btrfs_root *root; | ||
| 368 | int ret; | ||
| 369 | root = btrfs_sb(sb); | ||
| 370 | |||
| 371 | if (sb->s_flags & MS_RDONLY) | ||
| 372 | return 0; | ||
| 373 | |||
| 374 | sb->s_dirt = 0; | ||
| 375 | if (!wait) { | ||
| 376 | filemap_flush(root->fs_info->btree_inode->i_mapping); | ||
| 377 | return 0; | ||
| 378 | } | ||
| 379 | |||
| 380 | btrfs_start_delalloc_inodes(root); | ||
| 381 | btrfs_wait_ordered_extents(root, 0); | ||
| 382 | |||
| 383 | btrfs_clean_old_snapshots(root); | ||
| 384 | trans = btrfs_start_transaction(root, 1); | ||
| 385 | ret = btrfs_commit_transaction(trans, root); | ||
| 386 | sb->s_dirt = 0; | ||
| 387 | return ret; | ||
| 388 | } | ||
| 389 | |||
| 390 | static void btrfs_write_super(struct super_block *sb) | ||
| 391 | { | ||
| 392 | sb->s_dirt = 0; | ||
| 393 | } | ||
| 394 | |||
| 395 | static int btrfs_test_super(struct super_block *s, void *data) | ||
| 396 | { | ||
| 397 | struct btrfs_fs_devices *test_fs_devices = data; | ||
| 398 | struct btrfs_root *root = btrfs_sb(s); | ||
| 399 | |||
| 400 | return root->fs_info->fs_devices == test_fs_devices; | ||
| 401 | } | ||
| 402 | |||
| 403 | /* | ||
| 404 | * Find a superblock for the given device / mount point. | ||
| 405 | * | ||
| 406 | * Note: This is based on get_sb_bdev from fs/super.c with a few additions | ||
| 407 | * for multiple device setup. Make sure to keep it in sync. | ||
| 408 | */ | ||
| 409 | static int btrfs_get_sb(struct file_system_type *fs_type, int flags, | ||
| 410 | const char *dev_name, void *data, struct vfsmount *mnt) | ||
| 411 | { | ||
| 412 | char *subvol_name = NULL; | ||
| 413 | struct block_device *bdev = NULL; | ||
| 414 | struct super_block *s; | ||
| 415 | struct dentry *root; | ||
| 416 | struct btrfs_fs_devices *fs_devices = NULL; | ||
| 417 | fmode_t mode = FMODE_READ; | ||
| 418 | int error = 0; | ||
| 419 | |||
| 420 | if (!(flags & MS_RDONLY)) | ||
| 421 | mode |= FMODE_WRITE; | ||
| 422 | |||
| 423 | error = btrfs_parse_early_options(data, mode, fs_type, | ||
| 424 | &subvol_name, &fs_devices); | ||
| 425 | if (error) | ||
| 426 | return error; | ||
| 427 | |||
| 428 | error = btrfs_scan_one_device(dev_name, mode, fs_type, &fs_devices); | ||
| 429 | if (error) | ||
| 430 | goto error_free_subvol_name; | ||
| 431 | |||
| 432 | error = btrfs_open_devices(fs_devices, mode, fs_type); | ||
| 433 | if (error) | ||
| 434 | goto error_free_subvol_name; | ||
| 435 | |||
| 436 | if (!(flags & MS_RDONLY) && fs_devices->rw_devices == 0) { | ||
| 437 | error = -EACCES; | ||
| 438 | goto error_close_devices; | ||
| 439 | } | ||
| 440 | |||
| 441 | bdev = fs_devices->latest_bdev; | ||
| 442 | s = sget(fs_type, btrfs_test_super, set_anon_super, fs_devices); | ||
| 443 | if (IS_ERR(s)) | ||
| 444 | goto error_s; | ||
| 445 | |||
| 446 | if (s->s_root) { | ||
| 447 | if ((flags ^ s->s_flags) & MS_RDONLY) { | ||
| 448 | up_write(&s->s_umount); | ||
| 449 | deactivate_super(s); | ||
| 450 | error = -EBUSY; | ||
| 451 | goto error_close_devices; | ||
| 452 | } | ||
| 453 | |||
| 454 | btrfs_close_devices(fs_devices); | ||
| 455 | } else { | ||
| 456 | char b[BDEVNAME_SIZE]; | ||
| 457 | |||
| 458 | s->s_flags = flags; | ||
| 459 | strlcpy(s->s_id, bdevname(bdev, b), sizeof(s->s_id)); | ||
| 460 | error = btrfs_fill_super(s, fs_devices, data, | ||
| 461 | flags & MS_SILENT ? 1 : 0); | ||
| 462 | if (error) { | ||
| 463 | up_write(&s->s_umount); | ||
| 464 | deactivate_super(s); | ||
| 465 | goto error_free_subvol_name; | ||
| 466 | } | ||
| 467 | |||
| 468 | btrfs_sb(s)->fs_info->bdev_holder = fs_type; | ||
| 469 | s->s_flags |= MS_ACTIVE; | ||
| 470 | } | ||
| 471 | |||
| 472 | if (!strcmp(subvol_name, ".")) | ||
| 473 | root = dget(s->s_root); | ||
| 474 | else { | ||
| 475 | mutex_lock(&s->s_root->d_inode->i_mutex); | ||
| 476 | root = lookup_one_len(subvol_name, s->s_root, | ||
| 477 | strlen(subvol_name)); | ||
| 478 | mutex_unlock(&s->s_root->d_inode->i_mutex); | ||
| 479 | |||
| 480 | if (IS_ERR(root)) { | ||
| 481 | up_write(&s->s_umount); | ||
| 482 | deactivate_super(s); | ||
| 483 | error = PTR_ERR(root); | ||
| 484 | goto error_free_subvol_name; | ||
| 485 | } | ||
| 486 | if (!root->d_inode) { | ||
| 487 | dput(root); | ||
| 488 | up_write(&s->s_umount); | ||
| 489 | deactivate_super(s); | ||
| 490 | error = -ENXIO; | ||
| 491 | goto error_free_subvol_name; | ||
| 492 | } | ||
| 493 | } | ||
| 494 | |||
| 495 | mnt->mnt_sb = s; | ||
| 496 | mnt->mnt_root = root; | ||
| 497 | |||
| 498 | kfree(subvol_name); | ||
| 499 | return 0; | ||
| 500 | |||
| 501 | error_s: | ||
| 502 | error = PTR_ERR(s); | ||
| 503 | error_close_devices: | ||
| 504 | btrfs_close_devices(fs_devices); | ||
| 505 | error_free_subvol_name: | ||
| 506 | kfree(subvol_name); | ||
| 507 | return error; | ||
| 508 | } | ||
| 509 | |||
| 510 | static int btrfs_remount(struct super_block *sb, int *flags, char *data) | ||
| 511 | { | ||
| 512 | struct btrfs_root *root = btrfs_sb(sb); | ||
| 513 | int ret; | ||
| 514 | |||
| 515 | if ((*flags & MS_RDONLY) == (sb->s_flags & MS_RDONLY)) | ||
| 516 | return 0; | ||
| 517 | |||
| 518 | if (*flags & MS_RDONLY) { | ||
| 519 | sb->s_flags |= MS_RDONLY; | ||
| 520 | |||
| 521 | ret = btrfs_commit_super(root); | ||
| 522 | WARN_ON(ret); | ||
| 523 | } else { | ||
| 524 | if (root->fs_info->fs_devices->rw_devices == 0) | ||
| 525 | return -EACCES; | ||
| 526 | |||
| 527 | if (btrfs_super_log_root(&root->fs_info->super_copy) != 0) | ||
| 528 | return -EINVAL; | ||
| 529 | |||
| 530 | ret = btrfs_cleanup_reloc_trees(root); | ||
| 531 | WARN_ON(ret); | ||
| 532 | |||
| 533 | ret = btrfs_cleanup_fs_roots(root->fs_info); | ||
| 534 | WARN_ON(ret); | ||
| 535 | |||
| 536 | sb->s_flags &= ~MS_RDONLY; | ||
| 537 | } | ||
| 538 | |||
| 539 | return 0; | ||
| 540 | } | ||
| 541 | |||
| 542 | static int btrfs_statfs(struct dentry *dentry, struct kstatfs *buf) | ||
| 543 | { | ||
| 544 | struct btrfs_root *root = btrfs_sb(dentry->d_sb); | ||
| 545 | struct btrfs_super_block *disk_super = &root->fs_info->super_copy; | ||
| 546 | int bits = dentry->d_sb->s_blocksize_bits; | ||
| 547 | __be32 *fsid = (__be32 *)root->fs_info->fsid; | ||
| 548 | |||
| 549 | buf->f_namelen = BTRFS_NAME_LEN; | ||
| 550 | buf->f_blocks = btrfs_super_total_bytes(disk_super) >> bits; | ||
| 551 | buf->f_bfree = buf->f_blocks - | ||
| 552 | (btrfs_super_bytes_used(disk_super) >> bits); | ||
| 553 | buf->f_bavail = buf->f_bfree; | ||
| 554 | buf->f_bsize = dentry->d_sb->s_blocksize; | ||
| 555 | buf->f_type = BTRFS_SUPER_MAGIC; | ||
| 556 | |||
| 557 | /* We treat it as constant endianness (it doesn't matter _which_) | ||
| 558 | because we want the fsid to come out the same whether mounted | ||
| 559 | on a big-endian or little-endian host */ | ||
| 560 | buf->f_fsid.val[0] = be32_to_cpu(fsid[0]) ^ be32_to_cpu(fsid[2]); | ||
| 561 | buf->f_fsid.val[1] = be32_to_cpu(fsid[1]) ^ be32_to_cpu(fsid[3]); | ||
| 562 | /* Mask in the root object ID too, to disambiguate subvols */ | ||
| 563 | buf->f_fsid.val[0] ^= BTRFS_I(dentry->d_inode)->root->objectid >> 32; | ||
| 564 | buf->f_fsid.val[1] ^= BTRFS_I(dentry->d_inode)->root->objectid; | ||
| 565 | |||
| 566 | return 0; | ||
| 567 | } | ||
| 568 | |||
| 569 | static struct file_system_type btrfs_fs_type = { | ||
| 570 | .owner = THIS_MODULE, | ||
| 571 | .name = "btrfs", | ||
| 572 | .get_sb = btrfs_get_sb, | ||
| 573 | .kill_sb = kill_anon_super, | ||
| 574 | .fs_flags = FS_REQUIRES_DEV, | ||
| 575 | }; | ||
| 576 | |||
| 577 | /* | ||
| 578 | * used by btrfsctl to scan devices when no FS is mounted | ||
| 579 | */ | ||
| 580 | static long btrfs_control_ioctl(struct file *file, unsigned int cmd, | ||
| 581 | unsigned long arg) | ||
| 582 | { | ||
| 583 | struct btrfs_ioctl_vol_args *vol; | ||
| 584 | struct btrfs_fs_devices *fs_devices; | ||
| 585 | int ret = 0; | ||
| 586 | int len; | ||
| 587 | |||
| 588 | if (!capable(CAP_SYS_ADMIN)) | ||
| 589 | return -EPERM; | ||
| 590 | |||
| 591 | vol = kmalloc(sizeof(*vol), GFP_KERNEL); | ||
| 592 | if (copy_from_user(vol, (void __user *)arg, sizeof(*vol))) { | ||
| 593 | ret = -EFAULT; | ||
| 594 | goto out; | ||
| 595 | } | ||
| 596 | len = strnlen(vol->name, BTRFS_PATH_NAME_MAX); | ||
| 597 | switch (cmd) { | ||
| 598 | case BTRFS_IOC_SCAN_DEV: | ||
| 599 | ret = btrfs_scan_one_device(vol->name, FMODE_READ, | ||
| 600 | &btrfs_fs_type, &fs_devices); | ||
| 601 | break; | ||
| 602 | } | ||
| 603 | out: | ||
| 604 | kfree(vol); | ||
| 605 | return ret; | ||
| 606 | } | ||
| 607 | |||
| 608 | static int btrfs_freeze(struct super_block *sb) | ||
| 609 | { | ||
| 610 | struct btrfs_root *root = btrfs_sb(sb); | ||
| 611 | mutex_lock(&root->fs_info->transaction_kthread_mutex); | ||
| 612 | mutex_lock(&root->fs_info->cleaner_mutex); | ||
| 613 | return 0; | ||
| 614 | } | ||
| 615 | |||
| 616 | static int btrfs_unfreeze(struct super_block *sb) | ||
| 617 | { | ||
| 618 | struct btrfs_root *root = btrfs_sb(sb); | ||
| 619 | mutex_unlock(&root->fs_info->cleaner_mutex); | ||
| 620 | mutex_unlock(&root->fs_info->transaction_kthread_mutex); | ||
| 621 | return 0; | ||
| 622 | } | ||
| 623 | |||
| 624 | static struct super_operations btrfs_super_ops = { | ||
| 625 | .delete_inode = btrfs_delete_inode, | ||
| 626 | .put_super = btrfs_put_super, | ||
| 627 | .write_super = btrfs_write_super, | ||
| 628 | .sync_fs = btrfs_sync_fs, | ||
| 629 | .show_options = generic_show_options, | ||
| 630 | .write_inode = btrfs_write_inode, | ||
| 631 | .dirty_inode = btrfs_dirty_inode, | ||
| 632 | .alloc_inode = btrfs_alloc_inode, | ||
| 633 | .destroy_inode = btrfs_destroy_inode, | ||
| 634 | .statfs = btrfs_statfs, | ||
| 635 | .remount_fs = btrfs_remount, | ||
| 636 | .freeze_fs = btrfs_freeze, | ||
| 637 | .unfreeze_fs = btrfs_unfreeze, | ||
| 638 | }; | ||
| 639 | |||
| 640 | static const struct file_operations btrfs_ctl_fops = { | ||
| 641 | .unlocked_ioctl = btrfs_control_ioctl, | ||
| 642 | .compat_ioctl = btrfs_control_ioctl, | ||
| 643 | .owner = THIS_MODULE, | ||
| 644 | }; | ||
| 645 | |||
| 646 | static struct miscdevice btrfs_misc = { | ||
| 647 | .minor = MISC_DYNAMIC_MINOR, | ||
| 648 | .name = "btrfs-control", | ||
| 649 | .fops = &btrfs_ctl_fops | ||
| 650 | }; | ||
| 651 | |||
| 652 | static int btrfs_interface_init(void) | ||
| 653 | { | ||
| 654 | return misc_register(&btrfs_misc); | ||
| 655 | } | ||
| 656 | |||
| 657 | static void btrfs_interface_exit(void) | ||
| 658 | { | ||
| 659 | if (misc_deregister(&btrfs_misc) < 0) | ||
| 660 | printk(KERN_INFO "misc_deregister failed for control device"); | ||
| 661 | } | ||
| 662 | |||
| 663 | static int __init init_btrfs_fs(void) | ||
| 664 | { | ||
| 665 | int err; | ||
| 666 | |||
| 667 | err = btrfs_init_sysfs(); | ||
| 668 | if (err) | ||
| 669 | return err; | ||
| 670 | |||
| 671 | err = btrfs_init_cachep(); | ||
| 672 | if (err) | ||
| 673 | goto free_sysfs; | ||
| 674 | |||
| 675 | err = extent_io_init(); | ||
| 676 | if (err) | ||
| 677 | goto free_cachep; | ||
| 678 | |||
| 679 | err = extent_map_init(); | ||
| 680 | if (err) | ||
| 681 | goto free_extent_io; | ||
| 682 | |||
| 683 | err = btrfs_interface_init(); | ||
| 684 | if (err) | ||
| 685 | goto free_extent_map; | ||
| 686 | |||
| 687 | err = register_filesystem(&btrfs_fs_type); | ||
| 688 | if (err) | ||
| 689 | goto unregister_ioctl; | ||
| 690 | |||
| 691 | printk(KERN_INFO "%s loaded\n", BTRFS_BUILD_VERSION); | ||
| 692 | return 0; | ||
| 693 | |||
| 694 | unregister_ioctl: | ||
| 695 | btrfs_interface_exit(); | ||
| 696 | free_extent_map: | ||
| 697 | extent_map_exit(); | ||
| 698 | free_extent_io: | ||
| 699 | extent_io_exit(); | ||
| 700 | free_cachep: | ||
| 701 | btrfs_destroy_cachep(); | ||
| 702 | free_sysfs: | ||
| 703 | btrfs_exit_sysfs(); | ||
| 704 | return err; | ||
| 705 | } | ||
| 706 | |||
| 707 | static void __exit exit_btrfs_fs(void) | ||
| 708 | { | ||
| 709 | btrfs_destroy_cachep(); | ||
| 710 | extent_map_exit(); | ||
| 711 | extent_io_exit(); | ||
| 712 | btrfs_interface_exit(); | ||
| 713 | unregister_filesystem(&btrfs_fs_type); | ||
| 714 | btrfs_exit_sysfs(); | ||
| 715 | btrfs_cleanup_fs_uuids(); | ||
| 716 | btrfs_zlib_exit(); | ||
| 717 | } | ||
| 718 | |||
| 719 | module_init(init_btrfs_fs) | ||
| 720 | module_exit(exit_btrfs_fs) | ||
| 721 | |||
| 722 | MODULE_LICENSE("GPL"); | ||
diff --git a/fs/btrfs/sysfs.c b/fs/btrfs/sysfs.c new file mode 100644 index 000000000000..a240b6fa81df --- /dev/null +++ b/fs/btrfs/sysfs.c | |||
| @@ -0,0 +1,269 @@ | |||
| 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 | |||
| 19 | #include <linux/sched.h> | ||
| 20 | #include <linux/slab.h> | ||
| 21 | #include <linux/spinlock.h> | ||
| 22 | #include <linux/completion.h> | ||
| 23 | #include <linux/buffer_head.h> | ||
| 24 | #include <linux/module.h> | ||
| 25 | #include <linux/kobject.h> | ||
| 26 | |||
| 27 | #include "ctree.h" | ||
| 28 | #include "disk-io.h" | ||
| 29 | #include "transaction.h" | ||
| 30 | |||
| 31 | static ssize_t root_blocks_used_show(struct btrfs_root *root, char *buf) | ||
| 32 | { | ||
| 33 | return snprintf(buf, PAGE_SIZE, "%llu\n", | ||
| 34 | (unsigned long long)btrfs_root_used(&root->root_item)); | ||
| 35 | } | ||
| 36 | |||
| 37 | static ssize_t root_block_limit_show(struct btrfs_root *root, char *buf) | ||
| 38 | { | ||
| 39 | return snprintf(buf, PAGE_SIZE, "%llu\n", | ||
| 40 | (unsigned long long)btrfs_root_limit(&root->root_item)); | ||
| 41 | } | ||
| 42 | |||
| 43 | static ssize_t super_blocks_used_show(struct btrfs_fs_info *fs, char *buf) | ||
| 44 | { | ||
| 45 | |||
| 46 | return snprintf(buf, PAGE_SIZE, "%llu\n", | ||
| 47 | (unsigned long long)btrfs_super_bytes_used(&fs->super_copy)); | ||
| 48 | } | ||
| 49 | |||
| 50 | static ssize_t super_total_blocks_show(struct btrfs_fs_info *fs, char *buf) | ||
| 51 | { | ||
| 52 | return snprintf(buf, PAGE_SIZE, "%llu\n", | ||
| 53 | (unsigned long long)btrfs_super_total_bytes(&fs->super_copy)); | ||
| 54 | } | ||
| 55 | |||
| 56 | static ssize_t super_blocksize_show(struct btrfs_fs_info *fs, char *buf) | ||
| 57 | { | ||
| 58 | return snprintf(buf, PAGE_SIZE, "%llu\n", | ||
| 59 | (unsigned long long)btrfs_super_sectorsize(&fs->super_copy)); | ||
| 60 | } | ||
| 61 | |||
| 62 | /* this is for root attrs (subvols/snapshots) */ | ||
| 63 | struct btrfs_root_attr { | ||
| 64 | struct attribute attr; | ||
| 65 | ssize_t (*show)(struct btrfs_root *, char *); | ||
| 66 | ssize_t (*store)(struct btrfs_root *, const char *, size_t); | ||
| 67 | }; | ||
| 68 | |||
| 69 | #define ROOT_ATTR(name, mode, show, store) \ | ||
| 70 | static struct btrfs_root_attr btrfs_root_attr_##name = __ATTR(name, mode, \ | ||
| 71 | show, store) | ||
| 72 | |||
| 73 | ROOT_ATTR(blocks_used, 0444, root_blocks_used_show, NULL); | ||
| 74 | ROOT_ATTR(block_limit, 0644, root_block_limit_show, NULL); | ||
| 75 | |||
| 76 | static struct attribute *btrfs_root_attrs[] = { | ||
| 77 | &btrfs_root_attr_blocks_used.attr, | ||
| 78 | &btrfs_root_attr_block_limit.attr, | ||
| 79 | NULL, | ||
| 80 | }; | ||
| 81 | |||
| 82 | /* this is for super attrs (actual full fs) */ | ||
| 83 | struct btrfs_super_attr { | ||
| 84 | struct attribute attr; | ||
| 85 | ssize_t (*show)(struct btrfs_fs_info *, char *); | ||
| 86 | ssize_t (*store)(struct btrfs_fs_info *, const char *, size_t); | ||
| 87 | }; | ||
| 88 | |||
| 89 | #define SUPER_ATTR(name, mode, show, store) \ | ||
| 90 | static struct btrfs_super_attr btrfs_super_attr_##name = __ATTR(name, mode, \ | ||
| 91 | show, store) | ||
| 92 | |||
| 93 | SUPER_ATTR(blocks_used, 0444, super_blocks_used_show, NULL); | ||
| 94 | SUPER_ATTR(total_blocks, 0444, super_total_blocks_show, NULL); | ||
| 95 | SUPER_ATTR(blocksize, 0444, super_blocksize_show, NULL); | ||
| 96 | |||
| 97 | static struct attribute *btrfs_super_attrs[] = { | ||
| 98 | &btrfs_super_attr_blocks_used.attr, | ||
| 99 | &btrfs_super_attr_total_blocks.attr, | ||
| 100 | &btrfs_super_attr_blocksize.attr, | ||
| 101 | NULL, | ||
| 102 | }; | ||
| 103 | |||
| 104 | static ssize_t btrfs_super_attr_show(struct kobject *kobj, | ||
| 105 | struct attribute *attr, char *buf) | ||
| 106 | { | ||
| 107 | struct btrfs_fs_info *fs = container_of(kobj, struct btrfs_fs_info, | ||
| 108 | super_kobj); | ||
| 109 | struct btrfs_super_attr *a = container_of(attr, | ||
| 110 | struct btrfs_super_attr, | ||
| 111 | attr); | ||
| 112 | |||
| 113 | return a->show ? a->show(fs, buf) : 0; | ||
| 114 | } | ||
| 115 | |||
| 116 | static ssize_t btrfs_super_attr_store(struct kobject *kobj, | ||
| 117 | struct attribute *attr, | ||
| 118 | const char *buf, size_t len) | ||
| 119 | { | ||
| 120 | struct btrfs_fs_info *fs = container_of(kobj, struct btrfs_fs_info, | ||
| 121 | super_kobj); | ||
| 122 | struct btrfs_super_attr *a = container_of(attr, | ||
| 123 | struct btrfs_super_attr, | ||
| 124 | attr); | ||
| 125 | |||
| 126 | return a->store ? a->store(fs, buf, len) : 0; | ||
| 127 | } | ||
| 128 | |||
| 129 | static ssize_t btrfs_root_attr_show(struct kobject *kobj, | ||
| 130 | struct attribute *attr, char *buf) | ||
| 131 | { | ||
| 132 | struct btrfs_root *root = container_of(kobj, struct btrfs_root, | ||
| 133 | root_kobj); | ||
| 134 | struct btrfs_root_attr *a = container_of(attr, | ||
| 135 | struct btrfs_root_attr, | ||
| 136 | attr); | ||
| 137 | |||
| 138 | return a->show ? a->show(root, buf) : 0; | ||
| 139 | } | ||
| 140 | |||
| 141 | static ssize_t btrfs_root_attr_store(struct kobject *kobj, | ||
| 142 | struct attribute *attr, | ||
| 143 | const char *buf, size_t len) | ||
| 144 | { | ||
| 145 | struct btrfs_root *root = container_of(kobj, struct btrfs_root, | ||
| 146 | root_kobj); | ||
| 147 | struct btrfs_root_attr *a = container_of(attr, | ||
| 148 | struct btrfs_root_attr, | ||
| 149 | attr); | ||
| 150 | return a->store ? a->store(root, buf, len) : 0; | ||
| 151 | } | ||
| 152 | |||
| 153 | static void btrfs_super_release(struct kobject *kobj) | ||
| 154 | { | ||
| 155 | struct btrfs_fs_info *fs = container_of(kobj, struct btrfs_fs_info, | ||
| 156 | super_kobj); | ||
| 157 | complete(&fs->kobj_unregister); | ||
| 158 | } | ||
| 159 | |||
| 160 | static void btrfs_root_release(struct kobject *kobj) | ||
| 161 | { | ||
| 162 | struct btrfs_root *root = container_of(kobj, struct btrfs_root, | ||
| 163 | root_kobj); | ||
| 164 | complete(&root->kobj_unregister); | ||
| 165 | } | ||
| 166 | |||
| 167 | static struct sysfs_ops btrfs_super_attr_ops = { | ||
| 168 | .show = btrfs_super_attr_show, | ||
| 169 | .store = btrfs_super_attr_store, | ||
| 170 | }; | ||
| 171 | |||
| 172 | static struct sysfs_ops btrfs_root_attr_ops = { | ||
| 173 | .show = btrfs_root_attr_show, | ||
| 174 | .store = btrfs_root_attr_store, | ||
| 175 | }; | ||
| 176 | |||
| 177 | static struct kobj_type btrfs_root_ktype = { | ||
| 178 | .default_attrs = btrfs_root_attrs, | ||
| 179 | .sysfs_ops = &btrfs_root_attr_ops, | ||
| 180 | .release = btrfs_root_release, | ||
| 181 | }; | ||
| 182 | |||
| 183 | static struct kobj_type btrfs_super_ktype = { | ||
| 184 | .default_attrs = btrfs_super_attrs, | ||
| 185 | .sysfs_ops = &btrfs_super_attr_ops, | ||
| 186 | .release = btrfs_super_release, | ||
| 187 | }; | ||
| 188 | |||
| 189 | /* /sys/fs/btrfs/ entry */ | ||
| 190 | static struct kset *btrfs_kset; | ||
| 191 | |||
| 192 | int btrfs_sysfs_add_super(struct btrfs_fs_info *fs) | ||
| 193 | { | ||
| 194 | int error; | ||
| 195 | char *name; | ||
| 196 | char c; | ||
| 197 | int len = strlen(fs->sb->s_id) + 1; | ||
| 198 | int i; | ||
| 199 | |||
| 200 | name = kmalloc(len, GFP_NOFS); | ||
| 201 | if (!name) { | ||
| 202 | error = -ENOMEM; | ||
| 203 | goto fail; | ||
| 204 | } | ||
| 205 | |||
| 206 | for (i = 0; i < len; i++) { | ||
| 207 | c = fs->sb->s_id[i]; | ||
| 208 | if (c == '/' || c == '\\') | ||
| 209 | c = '!'; | ||
| 210 | name[i] = c; | ||
| 211 | } | ||
| 212 | name[len] = '\0'; | ||
| 213 | |||
| 214 | fs->super_kobj.kset = btrfs_kset; | ||
| 215 | error = kobject_init_and_add(&fs->super_kobj, &btrfs_super_ktype, | ||
| 216 | NULL, "%s", name); | ||
| 217 | kfree(name); | ||
| 218 | if (error) | ||
| 219 | goto fail; | ||
| 220 | |||
| 221 | return 0; | ||
| 222 | |||
| 223 | fail: | ||
| 224 | printk(KERN_ERR "btrfs: sysfs creation for super failed\n"); | ||
| 225 | return error; | ||
| 226 | } | ||
| 227 | |||
| 228 | int btrfs_sysfs_add_root(struct btrfs_root *root) | ||
| 229 | { | ||
| 230 | int error; | ||
| 231 | |||
| 232 | error = kobject_init_and_add(&root->root_kobj, &btrfs_root_ktype, | ||
| 233 | &root->fs_info->super_kobj, | ||
| 234 | "%s", root->name); | ||
| 235 | if (error) | ||
| 236 | goto fail; | ||
| 237 | |||
| 238 | return 0; | ||
| 239 | |||
| 240 | fail: | ||
| 241 | printk(KERN_ERR "btrfs: sysfs creation for root failed\n"); | ||
| 242 | return error; | ||
| 243 | } | ||
| 244 | |||
| 245 | void btrfs_sysfs_del_root(struct btrfs_root *root) | ||
| 246 | { | ||
| 247 | kobject_put(&root->root_kobj); | ||
| 248 | wait_for_completion(&root->kobj_unregister); | ||
| 249 | } | ||
| 250 | |||
| 251 | void btrfs_sysfs_del_super(struct btrfs_fs_info *fs) | ||
| 252 | { | ||
| 253 | kobject_put(&fs->super_kobj); | ||
| 254 | wait_for_completion(&fs->kobj_unregister); | ||
| 255 | } | ||
| 256 | |||
| 257 | int btrfs_init_sysfs(void) | ||
| 258 | { | ||
| 259 | btrfs_kset = kset_create_and_add("btrfs", NULL, fs_kobj); | ||
| 260 | if (!btrfs_kset) | ||
| 261 | return -ENOMEM; | ||
| 262 | return 0; | ||
| 263 | } | ||
| 264 | |||
| 265 | void btrfs_exit_sysfs(void) | ||
| 266 | { | ||
| 267 | kset_unregister(btrfs_kset); | ||
| 268 | } | ||
| 269 | |||
diff --git a/fs/btrfs/transaction.c b/fs/btrfs/transaction.c new file mode 100644 index 000000000000..8a08f9443340 --- /dev/null +++ b/fs/btrfs/transaction.c | |||
| @@ -0,0 +1,1097 @@ | |||
| 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 | |||
| 19 | #include <linux/fs.h> | ||
| 20 | #include <linux/sched.h> | ||
| 21 | #include <linux/writeback.h> | ||
| 22 | #include <linux/pagemap.h> | ||
| 23 | #include <linux/blkdev.h> | ||
| 24 | #include "ctree.h" | ||
| 25 | #include "disk-io.h" | ||
| 26 | #include "transaction.h" | ||
| 27 | #include "locking.h" | ||
| 28 | #include "ref-cache.h" | ||
| 29 | #include "tree-log.h" | ||
| 30 | |||
| 31 | #define BTRFS_ROOT_TRANS_TAG 0 | ||
| 32 | |||
| 33 | static noinline void put_transaction(struct btrfs_transaction *transaction) | ||
| 34 | { | ||
| 35 | WARN_ON(transaction->use_count == 0); | ||
| 36 | transaction->use_count--; | ||
| 37 | if (transaction->use_count == 0) { | ||
| 38 | list_del_init(&transaction->list); | ||
| 39 | memset(transaction, 0, sizeof(*transaction)); | ||
| 40 | kmem_cache_free(btrfs_transaction_cachep, transaction); | ||
| 41 | } | ||
| 42 | } | ||
| 43 | |||
| 44 | /* | ||
| 45 | * either allocate a new transaction or hop into the existing one | ||
| 46 | */ | ||
| 47 | static noinline int join_transaction(struct btrfs_root *root) | ||
| 48 | { | ||
| 49 | struct btrfs_transaction *cur_trans; | ||
| 50 | cur_trans = root->fs_info->running_transaction; | ||
| 51 | if (!cur_trans) { | ||
| 52 | cur_trans = kmem_cache_alloc(btrfs_transaction_cachep, | ||
| 53 | GFP_NOFS); | ||
| 54 | BUG_ON(!cur_trans); | ||
| 55 | root->fs_info->generation++; | ||
| 56 | root->fs_info->last_alloc = 0; | ||
| 57 | root->fs_info->last_data_alloc = 0; | ||
| 58 | cur_trans->num_writers = 1; | ||
| 59 | cur_trans->num_joined = 0; | ||
| 60 | cur_trans->transid = root->fs_info->generation; | ||
| 61 | init_waitqueue_head(&cur_trans->writer_wait); | ||
| 62 | init_waitqueue_head(&cur_trans->commit_wait); | ||
| 63 | cur_trans->in_commit = 0; | ||
| 64 | cur_trans->blocked = 0; | ||
| 65 | cur_trans->use_count = 1; | ||
| 66 | cur_trans->commit_done = 0; | ||
| 67 | cur_trans->start_time = get_seconds(); | ||
| 68 | INIT_LIST_HEAD(&cur_trans->pending_snapshots); | ||
| 69 | list_add_tail(&cur_trans->list, &root->fs_info->trans_list); | ||
| 70 | extent_io_tree_init(&cur_trans->dirty_pages, | ||
| 71 | root->fs_info->btree_inode->i_mapping, | ||
| 72 | GFP_NOFS); | ||
| 73 | spin_lock(&root->fs_info->new_trans_lock); | ||
| 74 | root->fs_info->running_transaction = cur_trans; | ||
| 75 | spin_unlock(&root->fs_info->new_trans_lock); | ||
| 76 | } else { | ||
| 77 | cur_trans->num_writers++; | ||
| 78 | cur_trans->num_joined++; | ||
| 79 | } | ||
| 80 | |||
| 81 | return 0; | ||
| 82 | } | ||
| 83 | |||
| 84 | /* | ||
| 85 | * this does all the record keeping required to make sure that a reference | ||
| 86 | * counted root is properly recorded in a given transaction. This is required | ||
| 87 | * to make sure the old root from before we joined the transaction is deleted | ||
| 88 | * when the transaction commits | ||
| 89 | */ | ||
| 90 | noinline int btrfs_record_root_in_trans(struct btrfs_root *root) | ||
| 91 | { | ||
| 92 | struct btrfs_dirty_root *dirty; | ||
| 93 | u64 running_trans_id = root->fs_info->running_transaction->transid; | ||
| 94 | if (root->ref_cows && root->last_trans < running_trans_id) { | ||
| 95 | WARN_ON(root == root->fs_info->extent_root); | ||
| 96 | if (root->root_item.refs != 0) { | ||
| 97 | radix_tree_tag_set(&root->fs_info->fs_roots_radix, | ||
| 98 | (unsigned long)root->root_key.objectid, | ||
| 99 | BTRFS_ROOT_TRANS_TAG); | ||
| 100 | |||
| 101 | dirty = kmalloc(sizeof(*dirty), GFP_NOFS); | ||
| 102 | BUG_ON(!dirty); | ||
| 103 | dirty->root = kmalloc(sizeof(*dirty->root), GFP_NOFS); | ||
| 104 | BUG_ON(!dirty->root); | ||
| 105 | dirty->latest_root = root; | ||
| 106 | INIT_LIST_HEAD(&dirty->list); | ||
| 107 | |||
| 108 | root->commit_root = btrfs_root_node(root); | ||
| 109 | |||
| 110 | memcpy(dirty->root, root, sizeof(*root)); | ||
| 111 | spin_lock_init(&dirty->root->node_lock); | ||
| 112 | spin_lock_init(&dirty->root->list_lock); | ||
| 113 | mutex_init(&dirty->root->objectid_mutex); | ||
| 114 | mutex_init(&dirty->root->log_mutex); | ||
| 115 | INIT_LIST_HEAD(&dirty->root->dead_list); | ||
| 116 | dirty->root->node = root->commit_root; | ||
| 117 | dirty->root->commit_root = NULL; | ||
| 118 | |||
| 119 | spin_lock(&root->list_lock); | ||
| 120 | list_add(&dirty->root->dead_list, &root->dead_list); | ||
| 121 | spin_unlock(&root->list_lock); | ||
| 122 | |||
| 123 | root->dirty_root = dirty; | ||
| 124 | } else { | ||
| 125 | WARN_ON(1); | ||
| 126 | } | ||
| 127 | root->last_trans = running_trans_id; | ||
| 128 | } | ||
| 129 | return 0; | ||
| 130 | } | ||
| 131 | |||
| 132 | /* wait for commit against the current transaction to become unblocked | ||
| 133 | * when this is done, it is safe to start a new transaction, but the current | ||
| 134 | * transaction might not be fully on disk. | ||
| 135 | */ | ||
| 136 | static void wait_current_trans(struct btrfs_root *root) | ||
| 137 | { | ||
| 138 | struct btrfs_transaction *cur_trans; | ||
| 139 | |||
| 140 | cur_trans = root->fs_info->running_transaction; | ||
| 141 | if (cur_trans && cur_trans->blocked) { | ||
| 142 | DEFINE_WAIT(wait); | ||
| 143 | cur_trans->use_count++; | ||
| 144 | while (1) { | ||
| 145 | prepare_to_wait(&root->fs_info->transaction_wait, &wait, | ||
| 146 | TASK_UNINTERRUPTIBLE); | ||
| 147 | if (cur_trans->blocked) { | ||
| 148 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 149 | schedule(); | ||
| 150 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 151 | finish_wait(&root->fs_info->transaction_wait, | ||
| 152 | &wait); | ||
| 153 | } else { | ||
| 154 | finish_wait(&root->fs_info->transaction_wait, | ||
| 155 | &wait); | ||
| 156 | break; | ||
| 157 | } | ||
| 158 | } | ||
| 159 | put_transaction(cur_trans); | ||
| 160 | } | ||
| 161 | } | ||
| 162 | |||
| 163 | static struct btrfs_trans_handle *start_transaction(struct btrfs_root *root, | ||
| 164 | int num_blocks, int wait) | ||
| 165 | { | ||
| 166 | struct btrfs_trans_handle *h = | ||
| 167 | kmem_cache_alloc(btrfs_trans_handle_cachep, GFP_NOFS); | ||
| 168 | int ret; | ||
| 169 | |||
| 170 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 171 | if (!root->fs_info->log_root_recovering && | ||
| 172 | ((wait == 1 && !root->fs_info->open_ioctl_trans) || wait == 2)) | ||
| 173 | wait_current_trans(root); | ||
| 174 | ret = join_transaction(root); | ||
| 175 | BUG_ON(ret); | ||
| 176 | |||
| 177 | btrfs_record_root_in_trans(root); | ||
| 178 | h->transid = root->fs_info->running_transaction->transid; | ||
| 179 | h->transaction = root->fs_info->running_transaction; | ||
| 180 | h->blocks_reserved = num_blocks; | ||
| 181 | h->blocks_used = 0; | ||
| 182 | h->block_group = 0; | ||
| 183 | h->alloc_exclude_nr = 0; | ||
| 184 | h->alloc_exclude_start = 0; | ||
| 185 | root->fs_info->running_transaction->use_count++; | ||
| 186 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 187 | return h; | ||
| 188 | } | ||
| 189 | |||
| 190 | struct btrfs_trans_handle *btrfs_start_transaction(struct btrfs_root *root, | ||
| 191 | int num_blocks) | ||
| 192 | { | ||
| 193 | return start_transaction(root, num_blocks, 1); | ||
| 194 | } | ||
| 195 | struct btrfs_trans_handle *btrfs_join_transaction(struct btrfs_root *root, | ||
| 196 | int num_blocks) | ||
| 197 | { | ||
| 198 | return start_transaction(root, num_blocks, 0); | ||
| 199 | } | ||
| 200 | |||
| 201 | struct btrfs_trans_handle *btrfs_start_ioctl_transaction(struct btrfs_root *r, | ||
| 202 | int num_blocks) | ||
| 203 | { | ||
| 204 | return start_transaction(r, num_blocks, 2); | ||
| 205 | } | ||
| 206 | |||
| 207 | /* wait for a transaction commit to be fully complete */ | ||
| 208 | static noinline int wait_for_commit(struct btrfs_root *root, | ||
| 209 | struct btrfs_transaction *commit) | ||
| 210 | { | ||
| 211 | DEFINE_WAIT(wait); | ||
| 212 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 213 | while (!commit->commit_done) { | ||
| 214 | prepare_to_wait(&commit->commit_wait, &wait, | ||
| 215 | TASK_UNINTERRUPTIBLE); | ||
| 216 | if (commit->commit_done) | ||
| 217 | break; | ||
| 218 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 219 | schedule(); | ||
| 220 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 221 | } | ||
| 222 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 223 | finish_wait(&commit->commit_wait, &wait); | ||
| 224 | return 0; | ||
| 225 | } | ||
| 226 | |||
| 227 | /* | ||
| 228 | * rate limit against the drop_snapshot code. This helps to slow down new | ||
| 229 | * operations if the drop_snapshot code isn't able to keep up. | ||
| 230 | */ | ||
| 231 | static void throttle_on_drops(struct btrfs_root *root) | ||
| 232 | { | ||
| 233 | struct btrfs_fs_info *info = root->fs_info; | ||
| 234 | int harder_count = 0; | ||
| 235 | |||
| 236 | harder: | ||
| 237 | if (atomic_read(&info->throttles)) { | ||
| 238 | DEFINE_WAIT(wait); | ||
| 239 | int thr; | ||
| 240 | thr = atomic_read(&info->throttle_gen); | ||
| 241 | |||
| 242 | do { | ||
| 243 | prepare_to_wait(&info->transaction_throttle, | ||
| 244 | &wait, TASK_UNINTERRUPTIBLE); | ||
| 245 | if (!atomic_read(&info->throttles)) { | ||
| 246 | finish_wait(&info->transaction_throttle, &wait); | ||
| 247 | break; | ||
| 248 | } | ||
| 249 | schedule(); | ||
| 250 | finish_wait(&info->transaction_throttle, &wait); | ||
| 251 | } while (thr == atomic_read(&info->throttle_gen)); | ||
| 252 | harder_count++; | ||
| 253 | |||
| 254 | if (root->fs_info->total_ref_cache_size > 1 * 1024 * 1024 && | ||
| 255 | harder_count < 2) | ||
| 256 | goto harder; | ||
| 257 | |||
| 258 | if (root->fs_info->total_ref_cache_size > 5 * 1024 * 1024 && | ||
| 259 | harder_count < 10) | ||
| 260 | goto harder; | ||
| 261 | |||
| 262 | if (root->fs_info->total_ref_cache_size > 10 * 1024 * 1024 && | ||
| 263 | harder_count < 20) | ||
| 264 | goto harder; | ||
| 265 | } | ||
| 266 | } | ||
| 267 | |||
| 268 | void btrfs_throttle(struct btrfs_root *root) | ||
| 269 | { | ||
| 270 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 271 | if (!root->fs_info->open_ioctl_trans) | ||
| 272 | wait_current_trans(root); | ||
| 273 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 274 | |||
| 275 | throttle_on_drops(root); | ||
| 276 | } | ||
| 277 | |||
| 278 | static int __btrfs_end_transaction(struct btrfs_trans_handle *trans, | ||
| 279 | struct btrfs_root *root, int throttle) | ||
| 280 | { | ||
| 281 | struct btrfs_transaction *cur_trans; | ||
| 282 | struct btrfs_fs_info *info = root->fs_info; | ||
| 283 | |||
| 284 | mutex_lock(&info->trans_mutex); | ||
| 285 | cur_trans = info->running_transaction; | ||
| 286 | WARN_ON(cur_trans != trans->transaction); | ||
| 287 | WARN_ON(cur_trans->num_writers < 1); | ||
| 288 | cur_trans->num_writers--; | ||
| 289 | |||
| 290 | if (waitqueue_active(&cur_trans->writer_wait)) | ||
| 291 | wake_up(&cur_trans->writer_wait); | ||
| 292 | put_transaction(cur_trans); | ||
| 293 | mutex_unlock(&info->trans_mutex); | ||
| 294 | memset(trans, 0, sizeof(*trans)); | ||
| 295 | kmem_cache_free(btrfs_trans_handle_cachep, trans); | ||
| 296 | |||
| 297 | if (throttle) | ||
| 298 | throttle_on_drops(root); | ||
| 299 | |||
| 300 | return 0; | ||
| 301 | } | ||
| 302 | |||
| 303 | int btrfs_end_transaction(struct btrfs_trans_handle *trans, | ||
| 304 | struct btrfs_root *root) | ||
| 305 | { | ||
| 306 | return __btrfs_end_transaction(trans, root, 0); | ||
| 307 | } | ||
| 308 | |||
| 309 | int btrfs_end_transaction_throttle(struct btrfs_trans_handle *trans, | ||
| 310 | struct btrfs_root *root) | ||
| 311 | { | ||
| 312 | return __btrfs_end_transaction(trans, root, 1); | ||
| 313 | } | ||
| 314 | |||
| 315 | /* | ||
| 316 | * when btree blocks are allocated, they have some corresponding bits set for | ||
| 317 | * them in one of two extent_io trees. This is used to make sure all of | ||
| 318 | * those extents are on disk for transaction or log commit | ||
| 319 | */ | ||
| 320 | int btrfs_write_and_wait_marked_extents(struct btrfs_root *root, | ||
| 321 | struct extent_io_tree *dirty_pages) | ||
| 322 | { | ||
| 323 | int ret; | ||
| 324 | int err = 0; | ||
| 325 | int werr = 0; | ||
| 326 | struct page *page; | ||
| 327 | struct inode *btree_inode = root->fs_info->btree_inode; | ||
| 328 | u64 start = 0; | ||
| 329 | u64 end; | ||
| 330 | unsigned long index; | ||
| 331 | |||
| 332 | while (1) { | ||
| 333 | ret = find_first_extent_bit(dirty_pages, start, &start, &end, | ||
| 334 | EXTENT_DIRTY); | ||
| 335 | if (ret) | ||
| 336 | break; | ||
| 337 | while (start <= end) { | ||
| 338 | cond_resched(); | ||
| 339 | |||
| 340 | index = start >> PAGE_CACHE_SHIFT; | ||
| 341 | start = (u64)(index + 1) << PAGE_CACHE_SHIFT; | ||
| 342 | page = find_get_page(btree_inode->i_mapping, index); | ||
| 343 | if (!page) | ||
| 344 | continue; | ||
| 345 | |||
| 346 | btree_lock_page_hook(page); | ||
| 347 | if (!page->mapping) { | ||
| 348 | unlock_page(page); | ||
| 349 | page_cache_release(page); | ||
| 350 | continue; | ||
| 351 | } | ||
| 352 | |||
| 353 | if (PageWriteback(page)) { | ||
| 354 | if (PageDirty(page)) | ||
| 355 | wait_on_page_writeback(page); | ||
| 356 | else { | ||
| 357 | unlock_page(page); | ||
| 358 | page_cache_release(page); | ||
| 359 | continue; | ||
| 360 | } | ||
| 361 | } | ||
| 362 | err = write_one_page(page, 0); | ||
| 363 | if (err) | ||
| 364 | werr = err; | ||
| 365 | page_cache_release(page); | ||
| 366 | } | ||
| 367 | } | ||
| 368 | while (1) { | ||
| 369 | ret = find_first_extent_bit(dirty_pages, 0, &start, &end, | ||
| 370 | EXTENT_DIRTY); | ||
| 371 | if (ret) | ||
| 372 | break; | ||
| 373 | |||
| 374 | clear_extent_dirty(dirty_pages, start, end, GFP_NOFS); | ||
| 375 | while (start <= end) { | ||
| 376 | index = start >> PAGE_CACHE_SHIFT; | ||
| 377 | start = (u64)(index + 1) << PAGE_CACHE_SHIFT; | ||
| 378 | page = find_get_page(btree_inode->i_mapping, index); | ||
| 379 | if (!page) | ||
| 380 | continue; | ||
| 381 | if (PageDirty(page)) { | ||
| 382 | btree_lock_page_hook(page); | ||
| 383 | wait_on_page_writeback(page); | ||
| 384 | err = write_one_page(page, 0); | ||
| 385 | if (err) | ||
| 386 | werr = err; | ||
| 387 | } | ||
| 388 | wait_on_page_writeback(page); | ||
| 389 | page_cache_release(page); | ||
| 390 | cond_resched(); | ||
| 391 | } | ||
| 392 | } | ||
| 393 | if (err) | ||
| 394 | werr = err; | ||
| 395 | return werr; | ||
| 396 | } | ||
| 397 | |||
| 398 | int btrfs_write_and_wait_transaction(struct btrfs_trans_handle *trans, | ||
| 399 | struct btrfs_root *root) | ||
| 400 | { | ||
| 401 | if (!trans || !trans->transaction) { | ||
| 402 | struct inode *btree_inode; | ||
| 403 | btree_inode = root->fs_info->btree_inode; | ||
| 404 | return filemap_write_and_wait(btree_inode->i_mapping); | ||
| 405 | } | ||
| 406 | return btrfs_write_and_wait_marked_extents(root, | ||
| 407 | &trans->transaction->dirty_pages); | ||
| 408 | } | ||
| 409 | |||
| 410 | /* | ||
| 411 | * this is used to update the root pointer in the tree of tree roots. | ||
| 412 | * | ||
| 413 | * But, in the case of the extent allocation tree, updating the root | ||
| 414 | * pointer may allocate blocks which may change the root of the extent | ||
| 415 | * allocation tree. | ||
| 416 | * | ||
| 417 | * So, this loops and repeats and makes sure the cowonly root didn't | ||
| 418 | * change while the root pointer was being updated in the metadata. | ||
| 419 | */ | ||
| 420 | static int update_cowonly_root(struct btrfs_trans_handle *trans, | ||
| 421 | struct btrfs_root *root) | ||
| 422 | { | ||
| 423 | int ret; | ||
| 424 | u64 old_root_bytenr; | ||
| 425 | struct btrfs_root *tree_root = root->fs_info->tree_root; | ||
| 426 | |||
| 427 | btrfs_extent_post_op(trans, root); | ||
| 428 | btrfs_write_dirty_block_groups(trans, root); | ||
| 429 | btrfs_extent_post_op(trans, root); | ||
| 430 | |||
| 431 | while (1) { | ||
| 432 | old_root_bytenr = btrfs_root_bytenr(&root->root_item); | ||
| 433 | if (old_root_bytenr == root->node->start) | ||
| 434 | break; | ||
| 435 | btrfs_set_root_bytenr(&root->root_item, | ||
| 436 | root->node->start); | ||
| 437 | btrfs_set_root_level(&root->root_item, | ||
| 438 | btrfs_header_level(root->node)); | ||
| 439 | btrfs_set_root_generation(&root->root_item, trans->transid); | ||
| 440 | |||
| 441 | btrfs_extent_post_op(trans, root); | ||
| 442 | |||
| 443 | ret = btrfs_update_root(trans, tree_root, | ||
| 444 | &root->root_key, | ||
| 445 | &root->root_item); | ||
| 446 | BUG_ON(ret); | ||
| 447 | btrfs_write_dirty_block_groups(trans, root); | ||
| 448 | btrfs_extent_post_op(trans, root); | ||
| 449 | } | ||
| 450 | return 0; | ||
| 451 | } | ||
| 452 | |||
| 453 | /* | ||
| 454 | * update all the cowonly tree roots on disk | ||
| 455 | */ | ||
| 456 | int btrfs_commit_tree_roots(struct btrfs_trans_handle *trans, | ||
| 457 | struct btrfs_root *root) | ||
| 458 | { | ||
| 459 | struct btrfs_fs_info *fs_info = root->fs_info; | ||
| 460 | struct list_head *next; | ||
| 461 | struct extent_buffer *eb; | ||
| 462 | |||
| 463 | btrfs_extent_post_op(trans, fs_info->tree_root); | ||
| 464 | |||
| 465 | eb = btrfs_lock_root_node(fs_info->tree_root); | ||
| 466 | btrfs_cow_block(trans, fs_info->tree_root, eb, NULL, 0, &eb, 0); | ||
| 467 | btrfs_tree_unlock(eb); | ||
| 468 | free_extent_buffer(eb); | ||
| 469 | |||
| 470 | btrfs_extent_post_op(trans, fs_info->tree_root); | ||
| 471 | |||
| 472 | while (!list_empty(&fs_info->dirty_cowonly_roots)) { | ||
| 473 | next = fs_info->dirty_cowonly_roots.next; | ||
| 474 | list_del_init(next); | ||
| 475 | root = list_entry(next, struct btrfs_root, dirty_list); | ||
| 476 | |||
| 477 | update_cowonly_root(trans, root); | ||
| 478 | } | ||
| 479 | return 0; | ||
| 480 | } | ||
| 481 | |||
| 482 | /* | ||
| 483 | * dead roots are old snapshots that need to be deleted. This allocates | ||
| 484 | * a dirty root struct and adds it into the list of dead roots that need to | ||
| 485 | * be deleted | ||
| 486 | */ | ||
| 487 | int btrfs_add_dead_root(struct btrfs_root *root, struct btrfs_root *latest) | ||
| 488 | { | ||
| 489 | struct btrfs_dirty_root *dirty; | ||
| 490 | |||
| 491 | dirty = kmalloc(sizeof(*dirty), GFP_NOFS); | ||
| 492 | if (!dirty) | ||
| 493 | return -ENOMEM; | ||
| 494 | dirty->root = root; | ||
| 495 | dirty->latest_root = latest; | ||
| 496 | |||
| 497 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 498 | list_add(&dirty->list, &latest->fs_info->dead_roots); | ||
| 499 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 500 | return 0; | ||
| 501 | } | ||
| 502 | |||
| 503 | /* | ||
| 504 | * at transaction commit time we need to schedule the old roots for | ||
| 505 | * deletion via btrfs_drop_snapshot. This runs through all the | ||
| 506 | * reference counted roots that were modified in the current | ||
| 507 | * transaction and puts them into the drop list | ||
| 508 | */ | ||
| 509 | static noinline int add_dirty_roots(struct btrfs_trans_handle *trans, | ||
| 510 | struct radix_tree_root *radix, | ||
| 511 | struct list_head *list) | ||
| 512 | { | ||
| 513 | struct btrfs_dirty_root *dirty; | ||
| 514 | struct btrfs_root *gang[8]; | ||
| 515 | struct btrfs_root *root; | ||
| 516 | int i; | ||
| 517 | int ret; | ||
| 518 | int err = 0; | ||
| 519 | u32 refs; | ||
| 520 | |||
| 521 | while (1) { | ||
| 522 | ret = radix_tree_gang_lookup_tag(radix, (void **)gang, 0, | ||
| 523 | ARRAY_SIZE(gang), | ||
| 524 | BTRFS_ROOT_TRANS_TAG); | ||
| 525 | if (ret == 0) | ||
| 526 | break; | ||
| 527 | for (i = 0; i < ret; i++) { | ||
| 528 | root = gang[i]; | ||
| 529 | radix_tree_tag_clear(radix, | ||
| 530 | (unsigned long)root->root_key.objectid, | ||
| 531 | BTRFS_ROOT_TRANS_TAG); | ||
| 532 | |||
| 533 | BUG_ON(!root->ref_tree); | ||
| 534 | dirty = root->dirty_root; | ||
| 535 | |||
| 536 | btrfs_free_log(trans, root); | ||
| 537 | btrfs_free_reloc_root(trans, root); | ||
| 538 | |||
| 539 | if (root->commit_root == root->node) { | ||
| 540 | WARN_ON(root->node->start != | ||
| 541 | btrfs_root_bytenr(&root->root_item)); | ||
| 542 | |||
| 543 | free_extent_buffer(root->commit_root); | ||
| 544 | root->commit_root = NULL; | ||
| 545 | root->dirty_root = NULL; | ||
| 546 | |||
| 547 | spin_lock(&root->list_lock); | ||
| 548 | list_del_init(&dirty->root->dead_list); | ||
| 549 | spin_unlock(&root->list_lock); | ||
| 550 | |||
| 551 | kfree(dirty->root); | ||
| 552 | kfree(dirty); | ||
| 553 | |||
| 554 | /* make sure to update the root on disk | ||
| 555 | * so we get any updates to the block used | ||
| 556 | * counts | ||
| 557 | */ | ||
| 558 | err = btrfs_update_root(trans, | ||
| 559 | root->fs_info->tree_root, | ||
| 560 | &root->root_key, | ||
| 561 | &root->root_item); | ||
| 562 | continue; | ||
| 563 | } | ||
| 564 | |||
| 565 | memset(&root->root_item.drop_progress, 0, | ||
| 566 | sizeof(struct btrfs_disk_key)); | ||
| 567 | root->root_item.drop_level = 0; | ||
| 568 | root->commit_root = NULL; | ||
| 569 | root->dirty_root = NULL; | ||
| 570 | root->root_key.offset = root->fs_info->generation; | ||
| 571 | btrfs_set_root_bytenr(&root->root_item, | ||
| 572 | root->node->start); | ||
| 573 | btrfs_set_root_level(&root->root_item, | ||
| 574 | btrfs_header_level(root->node)); | ||
| 575 | btrfs_set_root_generation(&root->root_item, | ||
| 576 | root->root_key.offset); | ||
| 577 | |||
| 578 | err = btrfs_insert_root(trans, root->fs_info->tree_root, | ||
| 579 | &root->root_key, | ||
| 580 | &root->root_item); | ||
| 581 | if (err) | ||
| 582 | break; | ||
| 583 | |||
| 584 | refs = btrfs_root_refs(&dirty->root->root_item); | ||
| 585 | btrfs_set_root_refs(&dirty->root->root_item, refs - 1); | ||
| 586 | err = btrfs_update_root(trans, root->fs_info->tree_root, | ||
| 587 | &dirty->root->root_key, | ||
| 588 | &dirty->root->root_item); | ||
| 589 | |||
| 590 | BUG_ON(err); | ||
| 591 | if (refs == 1) { | ||
| 592 | list_add(&dirty->list, list); | ||
| 593 | } else { | ||
| 594 | WARN_ON(1); | ||
| 595 | free_extent_buffer(dirty->root->node); | ||
| 596 | kfree(dirty->root); | ||
| 597 | kfree(dirty); | ||
| 598 | } | ||
| 599 | } | ||
| 600 | } | ||
| 601 | return err; | ||
| 602 | } | ||
| 603 | |||
| 604 | /* | ||
| 605 | * defrag a given btree. If cacheonly == 1, this won't read from the disk, | ||
| 606 | * otherwise every leaf in the btree is read and defragged. | ||
| 607 | */ | ||
| 608 | int btrfs_defrag_root(struct btrfs_root *root, int cacheonly) | ||
| 609 | { | ||
| 610 | struct btrfs_fs_info *info = root->fs_info; | ||
| 611 | int ret; | ||
| 612 | struct btrfs_trans_handle *trans; | ||
| 613 | unsigned long nr; | ||
| 614 | |||
| 615 | smp_mb(); | ||
| 616 | if (root->defrag_running) | ||
| 617 | return 0; | ||
| 618 | trans = btrfs_start_transaction(root, 1); | ||
| 619 | while (1) { | ||
| 620 | root->defrag_running = 1; | ||
| 621 | ret = btrfs_defrag_leaves(trans, root, cacheonly); | ||
| 622 | nr = trans->blocks_used; | ||
| 623 | btrfs_end_transaction(trans, root); | ||
| 624 | btrfs_btree_balance_dirty(info->tree_root, nr); | ||
| 625 | cond_resched(); | ||
| 626 | |||
| 627 | trans = btrfs_start_transaction(root, 1); | ||
| 628 | if (root->fs_info->closing || ret != -EAGAIN) | ||
| 629 | break; | ||
| 630 | } | ||
| 631 | root->defrag_running = 0; | ||
| 632 | smp_mb(); | ||
| 633 | btrfs_end_transaction(trans, root); | ||
| 634 | return 0; | ||
| 635 | } | ||
| 636 | |||
| 637 | /* | ||
| 638 | * Given a list of roots that need to be deleted, call btrfs_drop_snapshot on | ||
| 639 | * all of them | ||
| 640 | */ | ||
| 641 | static noinline int drop_dirty_roots(struct btrfs_root *tree_root, | ||
| 642 | struct list_head *list) | ||
| 643 | { | ||
| 644 | struct btrfs_dirty_root *dirty; | ||
| 645 | struct btrfs_trans_handle *trans; | ||
| 646 | unsigned long nr; | ||
| 647 | u64 num_bytes; | ||
| 648 | u64 bytes_used; | ||
| 649 | u64 max_useless; | ||
| 650 | int ret = 0; | ||
| 651 | int err; | ||
| 652 | |||
| 653 | while (!list_empty(list)) { | ||
| 654 | struct btrfs_root *root; | ||
| 655 | |||
| 656 | dirty = list_entry(list->prev, struct btrfs_dirty_root, list); | ||
| 657 | list_del_init(&dirty->list); | ||
| 658 | |||
| 659 | num_bytes = btrfs_root_used(&dirty->root->root_item); | ||
| 660 | root = dirty->latest_root; | ||
| 661 | atomic_inc(&root->fs_info->throttles); | ||
| 662 | |||
| 663 | while (1) { | ||
| 664 | trans = btrfs_start_transaction(tree_root, 1); | ||
| 665 | mutex_lock(&root->fs_info->drop_mutex); | ||
| 666 | ret = btrfs_drop_snapshot(trans, dirty->root); | ||
| 667 | if (ret != -EAGAIN) | ||
| 668 | break; | ||
| 669 | mutex_unlock(&root->fs_info->drop_mutex); | ||
| 670 | |||
| 671 | err = btrfs_update_root(trans, | ||
| 672 | tree_root, | ||
| 673 | &dirty->root->root_key, | ||
| 674 | &dirty->root->root_item); | ||
| 675 | if (err) | ||
| 676 | ret = err; | ||
| 677 | nr = trans->blocks_used; | ||
| 678 | ret = btrfs_end_transaction(trans, tree_root); | ||
| 679 | BUG_ON(ret); | ||
| 680 | |||
| 681 | btrfs_btree_balance_dirty(tree_root, nr); | ||
| 682 | cond_resched(); | ||
| 683 | } | ||
| 684 | BUG_ON(ret); | ||
| 685 | atomic_dec(&root->fs_info->throttles); | ||
| 686 | wake_up(&root->fs_info->transaction_throttle); | ||
| 687 | |||
| 688 | num_bytes -= btrfs_root_used(&dirty->root->root_item); | ||
| 689 | bytes_used = btrfs_root_used(&root->root_item); | ||
| 690 | if (num_bytes) { | ||
| 691 | btrfs_record_root_in_trans(root); | ||
| 692 | btrfs_set_root_used(&root->root_item, | ||
| 693 | bytes_used - num_bytes); | ||
| 694 | } | ||
| 695 | |||
| 696 | ret = btrfs_del_root(trans, tree_root, &dirty->root->root_key); | ||
| 697 | if (ret) { | ||
| 698 | BUG(); | ||
| 699 | break; | ||
| 700 | } | ||
| 701 | mutex_unlock(&root->fs_info->drop_mutex); | ||
| 702 | |||
| 703 | spin_lock(&root->list_lock); | ||
| 704 | list_del_init(&dirty->root->dead_list); | ||
| 705 | if (!list_empty(&root->dead_list)) { | ||
| 706 | struct btrfs_root *oldest; | ||
| 707 | oldest = list_entry(root->dead_list.prev, | ||
| 708 | struct btrfs_root, dead_list); | ||
| 709 | max_useless = oldest->root_key.offset - 1; | ||
| 710 | } else { | ||
| 711 | max_useless = root->root_key.offset - 1; | ||
| 712 | } | ||
| 713 | spin_unlock(&root->list_lock); | ||
| 714 | |||
| 715 | nr = trans->blocks_used; | ||
| 716 | ret = btrfs_end_transaction(trans, tree_root); | ||
| 717 | BUG_ON(ret); | ||
| 718 | |||
| 719 | ret = btrfs_remove_leaf_refs(root, max_useless, 0); | ||
| 720 | BUG_ON(ret); | ||
| 721 | |||
| 722 | free_extent_buffer(dirty->root->node); | ||
| 723 | kfree(dirty->root); | ||
| 724 | kfree(dirty); | ||
| 725 | |||
| 726 | btrfs_btree_balance_dirty(tree_root, nr); | ||
| 727 | cond_resched(); | ||
| 728 | } | ||
| 729 | return ret; | ||
| 730 | } | ||
| 731 | |||
| 732 | /* | ||
| 733 | * new snapshots need to be created at a very specific time in the | ||
| 734 | * transaction commit. This does the actual creation | ||
| 735 | */ | ||
| 736 | static noinline int create_pending_snapshot(struct btrfs_trans_handle *trans, | ||
| 737 | struct btrfs_fs_info *fs_info, | ||
| 738 | struct btrfs_pending_snapshot *pending) | ||
| 739 | { | ||
| 740 | struct btrfs_key key; | ||
| 741 | struct btrfs_root_item *new_root_item; | ||
| 742 | struct btrfs_root *tree_root = fs_info->tree_root; | ||
| 743 | struct btrfs_root *root = pending->root; | ||
| 744 | struct extent_buffer *tmp; | ||
| 745 | struct extent_buffer *old; | ||
| 746 | int ret; | ||
| 747 | u64 objectid; | ||
| 748 | |||
| 749 | new_root_item = kmalloc(sizeof(*new_root_item), GFP_NOFS); | ||
| 750 | if (!new_root_item) { | ||
| 751 | ret = -ENOMEM; | ||
| 752 | goto fail; | ||
| 753 | } | ||
| 754 | ret = btrfs_find_free_objectid(trans, tree_root, 0, &objectid); | ||
| 755 | if (ret) | ||
| 756 | goto fail; | ||
| 757 | |||
| 758 | btrfs_record_root_in_trans(root); | ||
| 759 | btrfs_set_root_last_snapshot(&root->root_item, trans->transid); | ||
| 760 | memcpy(new_root_item, &root->root_item, sizeof(*new_root_item)); | ||
| 761 | |||
| 762 | key.objectid = objectid; | ||
| 763 | key.offset = trans->transid; | ||
| 764 | btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY); | ||
| 765 | |||
| 766 | old = btrfs_lock_root_node(root); | ||
| 767 | btrfs_cow_block(trans, root, old, NULL, 0, &old, 0); | ||
| 768 | |||
| 769 | btrfs_copy_root(trans, root, old, &tmp, objectid); | ||
| 770 | btrfs_tree_unlock(old); | ||
| 771 | free_extent_buffer(old); | ||
| 772 | |||
| 773 | btrfs_set_root_bytenr(new_root_item, tmp->start); | ||
| 774 | btrfs_set_root_level(new_root_item, btrfs_header_level(tmp)); | ||
| 775 | btrfs_set_root_generation(new_root_item, trans->transid); | ||
| 776 | ret = btrfs_insert_root(trans, root->fs_info->tree_root, &key, | ||
| 777 | new_root_item); | ||
| 778 | btrfs_tree_unlock(tmp); | ||
| 779 | free_extent_buffer(tmp); | ||
| 780 | if (ret) | ||
| 781 | goto fail; | ||
| 782 | |||
| 783 | key.offset = (u64)-1; | ||
| 784 | memcpy(&pending->root_key, &key, sizeof(key)); | ||
| 785 | fail: | ||
| 786 | kfree(new_root_item); | ||
| 787 | return ret; | ||
| 788 | } | ||
| 789 | |||
| 790 | static noinline int finish_pending_snapshot(struct btrfs_fs_info *fs_info, | ||
| 791 | struct btrfs_pending_snapshot *pending) | ||
| 792 | { | ||
| 793 | int ret; | ||
| 794 | int namelen; | ||
| 795 | u64 index = 0; | ||
| 796 | struct btrfs_trans_handle *trans; | ||
| 797 | struct inode *parent_inode; | ||
| 798 | struct inode *inode; | ||
| 799 | struct btrfs_root *parent_root; | ||
| 800 | |||
| 801 | parent_inode = pending->dentry->d_parent->d_inode; | ||
| 802 | parent_root = BTRFS_I(parent_inode)->root; | ||
| 803 | trans = btrfs_join_transaction(parent_root, 1); | ||
| 804 | |||
| 805 | /* | ||
| 806 | * insert the directory item | ||
| 807 | */ | ||
| 808 | namelen = strlen(pending->name); | ||
| 809 | ret = btrfs_set_inode_index(parent_inode, &index); | ||
| 810 | ret = btrfs_insert_dir_item(trans, parent_root, | ||
| 811 | pending->name, namelen, | ||
| 812 | parent_inode->i_ino, | ||
| 813 | &pending->root_key, BTRFS_FT_DIR, index); | ||
| 814 | |||
| 815 | if (ret) | ||
| 816 | goto fail; | ||
| 817 | |||
| 818 | btrfs_i_size_write(parent_inode, parent_inode->i_size + namelen * 2); | ||
| 819 | ret = btrfs_update_inode(trans, parent_root, parent_inode); | ||
| 820 | BUG_ON(ret); | ||
| 821 | |||
| 822 | /* add the backref first */ | ||
| 823 | ret = btrfs_add_root_ref(trans, parent_root->fs_info->tree_root, | ||
| 824 | pending->root_key.objectid, | ||
| 825 | BTRFS_ROOT_BACKREF_KEY, | ||
| 826 | parent_root->root_key.objectid, | ||
| 827 | parent_inode->i_ino, index, pending->name, | ||
| 828 | namelen); | ||
| 829 | |||
| 830 | BUG_ON(ret); | ||
| 831 | |||
| 832 | /* now add the forward ref */ | ||
| 833 | ret = btrfs_add_root_ref(trans, parent_root->fs_info->tree_root, | ||
| 834 | parent_root->root_key.objectid, | ||
| 835 | BTRFS_ROOT_REF_KEY, | ||
| 836 | pending->root_key.objectid, | ||
| 837 | parent_inode->i_ino, index, pending->name, | ||
| 838 | namelen); | ||
| 839 | |||
| 840 | inode = btrfs_lookup_dentry(parent_inode, pending->dentry); | ||
| 841 | d_instantiate(pending->dentry, inode); | ||
| 842 | fail: | ||
| 843 | btrfs_end_transaction(trans, fs_info->fs_root); | ||
| 844 | return ret; | ||
| 845 | } | ||
| 846 | |||
| 847 | /* | ||
| 848 | * create all the snapshots we've scheduled for creation | ||
| 849 | */ | ||
| 850 | static noinline int create_pending_snapshots(struct btrfs_trans_handle *trans, | ||
| 851 | struct btrfs_fs_info *fs_info) | ||
| 852 | { | ||
| 853 | struct btrfs_pending_snapshot *pending; | ||
| 854 | struct list_head *head = &trans->transaction->pending_snapshots; | ||
| 855 | struct list_head *cur; | ||
| 856 | int ret; | ||
| 857 | |||
| 858 | list_for_each(cur, head) { | ||
| 859 | pending = list_entry(cur, struct btrfs_pending_snapshot, list); | ||
| 860 | ret = create_pending_snapshot(trans, fs_info, pending); | ||
| 861 | BUG_ON(ret); | ||
| 862 | } | ||
| 863 | return 0; | ||
| 864 | } | ||
| 865 | |||
| 866 | static noinline int finish_pending_snapshots(struct btrfs_trans_handle *trans, | ||
| 867 | struct btrfs_fs_info *fs_info) | ||
| 868 | { | ||
| 869 | struct btrfs_pending_snapshot *pending; | ||
| 870 | struct list_head *head = &trans->transaction->pending_snapshots; | ||
| 871 | int ret; | ||
| 872 | |||
| 873 | while (!list_empty(head)) { | ||
| 874 | pending = list_entry(head->next, | ||
| 875 | struct btrfs_pending_snapshot, list); | ||
| 876 | ret = finish_pending_snapshot(fs_info, pending); | ||
| 877 | BUG_ON(ret); | ||
| 878 | list_del(&pending->list); | ||
| 879 | kfree(pending->name); | ||
| 880 | kfree(pending); | ||
| 881 | } | ||
| 882 | return 0; | ||
| 883 | } | ||
| 884 | |||
| 885 | int btrfs_commit_transaction(struct btrfs_trans_handle *trans, | ||
| 886 | struct btrfs_root *root) | ||
| 887 | { | ||
| 888 | unsigned long joined = 0; | ||
| 889 | unsigned long timeout = 1; | ||
| 890 | struct btrfs_transaction *cur_trans; | ||
| 891 | struct btrfs_transaction *prev_trans = NULL; | ||
| 892 | struct btrfs_root *chunk_root = root->fs_info->chunk_root; | ||
| 893 | struct list_head dirty_fs_roots; | ||
| 894 | struct extent_io_tree *pinned_copy; | ||
| 895 | DEFINE_WAIT(wait); | ||
| 896 | int ret; | ||
| 897 | |||
| 898 | INIT_LIST_HEAD(&dirty_fs_roots); | ||
| 899 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 900 | if (trans->transaction->in_commit) { | ||
| 901 | cur_trans = trans->transaction; | ||
| 902 | trans->transaction->use_count++; | ||
| 903 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 904 | btrfs_end_transaction(trans, root); | ||
| 905 | |||
| 906 | ret = wait_for_commit(root, cur_trans); | ||
| 907 | BUG_ON(ret); | ||
| 908 | |||
| 909 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 910 | put_transaction(cur_trans); | ||
| 911 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 912 | |||
| 913 | return 0; | ||
| 914 | } | ||
| 915 | |||
| 916 | pinned_copy = kmalloc(sizeof(*pinned_copy), GFP_NOFS); | ||
| 917 | if (!pinned_copy) | ||
| 918 | return -ENOMEM; | ||
| 919 | |||
| 920 | extent_io_tree_init(pinned_copy, | ||
| 921 | root->fs_info->btree_inode->i_mapping, GFP_NOFS); | ||
| 922 | |||
| 923 | trans->transaction->in_commit = 1; | ||
| 924 | trans->transaction->blocked = 1; | ||
| 925 | cur_trans = trans->transaction; | ||
| 926 | if (cur_trans->list.prev != &root->fs_info->trans_list) { | ||
| 927 | prev_trans = list_entry(cur_trans->list.prev, | ||
| 928 | struct btrfs_transaction, list); | ||
| 929 | if (!prev_trans->commit_done) { | ||
| 930 | prev_trans->use_count++; | ||
| 931 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 932 | |||
| 933 | wait_for_commit(root, prev_trans); | ||
| 934 | |||
| 935 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 936 | put_transaction(prev_trans); | ||
| 937 | } | ||
| 938 | } | ||
| 939 | |||
| 940 | do { | ||
| 941 | int snap_pending = 0; | ||
| 942 | joined = cur_trans->num_joined; | ||
| 943 | if (!list_empty(&trans->transaction->pending_snapshots)) | ||
| 944 | snap_pending = 1; | ||
| 945 | |||
| 946 | WARN_ON(cur_trans != trans->transaction); | ||
| 947 | prepare_to_wait(&cur_trans->writer_wait, &wait, | ||
| 948 | TASK_UNINTERRUPTIBLE); | ||
| 949 | |||
| 950 | if (cur_trans->num_writers > 1) | ||
| 951 | timeout = MAX_SCHEDULE_TIMEOUT; | ||
| 952 | else | ||
| 953 | timeout = 1; | ||
| 954 | |||
| 955 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 956 | |||
| 957 | if (snap_pending) { | ||
| 958 | ret = btrfs_wait_ordered_extents(root, 1); | ||
| 959 | BUG_ON(ret); | ||
| 960 | } | ||
| 961 | |||
| 962 | schedule_timeout(timeout); | ||
| 963 | |||
| 964 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 965 | finish_wait(&cur_trans->writer_wait, &wait); | ||
| 966 | } while (cur_trans->num_writers > 1 || | ||
| 967 | (cur_trans->num_joined != joined)); | ||
| 968 | |||
| 969 | ret = create_pending_snapshots(trans, root->fs_info); | ||
| 970 | BUG_ON(ret); | ||
| 971 | |||
| 972 | WARN_ON(cur_trans != trans->transaction); | ||
| 973 | |||
| 974 | /* btrfs_commit_tree_roots is responsible for getting the | ||
| 975 | * various roots consistent with each other. Every pointer | ||
| 976 | * in the tree of tree roots has to point to the most up to date | ||
| 977 | * root for every subvolume and other tree. So, we have to keep | ||
| 978 | * the tree logging code from jumping in and changing any | ||
| 979 | * of the trees. | ||
| 980 | * | ||
| 981 | * At this point in the commit, there can't be any tree-log | ||
| 982 | * writers, but a little lower down we drop the trans mutex | ||
| 983 | * and let new people in. By holding the tree_log_mutex | ||
| 984 | * from now until after the super is written, we avoid races | ||
| 985 | * with the tree-log code. | ||
| 986 | */ | ||
| 987 | mutex_lock(&root->fs_info->tree_log_mutex); | ||
| 988 | /* | ||
| 989 | * keep tree reloc code from adding new reloc trees | ||
| 990 | */ | ||
| 991 | mutex_lock(&root->fs_info->tree_reloc_mutex); | ||
| 992 | |||
| 993 | |||
| 994 | ret = add_dirty_roots(trans, &root->fs_info->fs_roots_radix, | ||
| 995 | &dirty_fs_roots); | ||
| 996 | BUG_ON(ret); | ||
| 997 | |||
| 998 | /* add_dirty_roots gets rid of all the tree log roots, it is now | ||
| 999 | * safe to free the root of tree log roots | ||
| 1000 | */ | ||
| 1001 | btrfs_free_log_root_tree(trans, root->fs_info); | ||
| 1002 | |||
| 1003 | ret = btrfs_commit_tree_roots(trans, root); | ||
| 1004 | BUG_ON(ret); | ||
| 1005 | |||
| 1006 | cur_trans = root->fs_info->running_transaction; | ||
| 1007 | spin_lock(&root->fs_info->new_trans_lock); | ||
| 1008 | root->fs_info->running_transaction = NULL; | ||
| 1009 | spin_unlock(&root->fs_info->new_trans_lock); | ||
| 1010 | btrfs_set_super_generation(&root->fs_info->super_copy, | ||
| 1011 | cur_trans->transid); | ||
| 1012 | btrfs_set_super_root(&root->fs_info->super_copy, | ||
| 1013 | root->fs_info->tree_root->node->start); | ||
| 1014 | btrfs_set_super_root_level(&root->fs_info->super_copy, | ||
| 1015 | btrfs_header_level(root->fs_info->tree_root->node)); | ||
| 1016 | |||
| 1017 | btrfs_set_super_chunk_root(&root->fs_info->super_copy, | ||
| 1018 | chunk_root->node->start); | ||
| 1019 | btrfs_set_super_chunk_root_level(&root->fs_info->super_copy, | ||
| 1020 | btrfs_header_level(chunk_root->node)); | ||
| 1021 | btrfs_set_super_chunk_root_generation(&root->fs_info->super_copy, | ||
| 1022 | btrfs_header_generation(chunk_root->node)); | ||
| 1023 | |||
| 1024 | if (!root->fs_info->log_root_recovering) { | ||
| 1025 | btrfs_set_super_log_root(&root->fs_info->super_copy, 0); | ||
| 1026 | btrfs_set_super_log_root_level(&root->fs_info->super_copy, 0); | ||
| 1027 | } | ||
| 1028 | |||
| 1029 | memcpy(&root->fs_info->super_for_commit, &root->fs_info->super_copy, | ||
| 1030 | sizeof(root->fs_info->super_copy)); | ||
| 1031 | |||
| 1032 | btrfs_copy_pinned(root, pinned_copy); | ||
| 1033 | |||
| 1034 | trans->transaction->blocked = 0; | ||
| 1035 | wake_up(&root->fs_info->transaction_throttle); | ||
| 1036 | wake_up(&root->fs_info->transaction_wait); | ||
| 1037 | |||
| 1038 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1039 | ret = btrfs_write_and_wait_transaction(trans, root); | ||
| 1040 | BUG_ON(ret); | ||
| 1041 | write_ctree_super(trans, root, 0); | ||
| 1042 | |||
| 1043 | /* | ||
| 1044 | * the super is written, we can safely allow the tree-loggers | ||
| 1045 | * to go about their business | ||
| 1046 | */ | ||
| 1047 | mutex_unlock(&root->fs_info->tree_log_mutex); | ||
| 1048 | |||
| 1049 | btrfs_finish_extent_commit(trans, root, pinned_copy); | ||
| 1050 | kfree(pinned_copy); | ||
| 1051 | |||
| 1052 | btrfs_drop_dead_reloc_roots(root); | ||
| 1053 | mutex_unlock(&root->fs_info->tree_reloc_mutex); | ||
| 1054 | |||
| 1055 | /* do the directory inserts of any pending snapshot creations */ | ||
| 1056 | finish_pending_snapshots(trans, root->fs_info); | ||
| 1057 | |||
| 1058 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 1059 | |||
| 1060 | cur_trans->commit_done = 1; | ||
| 1061 | root->fs_info->last_trans_committed = cur_trans->transid; | ||
| 1062 | wake_up(&cur_trans->commit_wait); | ||
| 1063 | |||
| 1064 | put_transaction(cur_trans); | ||
| 1065 | put_transaction(cur_trans); | ||
| 1066 | |||
| 1067 | list_splice_init(&dirty_fs_roots, &root->fs_info->dead_roots); | ||
| 1068 | if (root->fs_info->closing) | ||
| 1069 | list_splice_init(&root->fs_info->dead_roots, &dirty_fs_roots); | ||
| 1070 | |||
| 1071 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1072 | |||
| 1073 | kmem_cache_free(btrfs_trans_handle_cachep, trans); | ||
| 1074 | |||
| 1075 | if (root->fs_info->closing) | ||
| 1076 | drop_dirty_roots(root->fs_info->tree_root, &dirty_fs_roots); | ||
| 1077 | return ret; | ||
| 1078 | } | ||
| 1079 | |||
| 1080 | /* | ||
| 1081 | * interface function to delete all the snapshots we have scheduled for deletion | ||
| 1082 | */ | ||
| 1083 | int btrfs_clean_old_snapshots(struct btrfs_root *root) | ||
| 1084 | { | ||
| 1085 | struct list_head dirty_roots; | ||
| 1086 | INIT_LIST_HEAD(&dirty_roots); | ||
| 1087 | again: | ||
| 1088 | mutex_lock(&root->fs_info->trans_mutex); | ||
| 1089 | list_splice_init(&root->fs_info->dead_roots, &dirty_roots); | ||
| 1090 | mutex_unlock(&root->fs_info->trans_mutex); | ||
| 1091 | |||
| 1092 | if (!list_empty(&dirty_roots)) { | ||
| 1093 | drop_dirty_roots(root, &dirty_roots); | ||
| 1094 | goto again; | ||
| 1095 | } | ||
| 1096 | return 0; | ||
| 1097 | } | ||
diff --git a/fs/btrfs/transaction.h b/fs/btrfs/transaction.h new file mode 100644 index 000000000000..ea292117f882 --- /dev/null +++ b/fs/btrfs/transaction.h | |||
| @@ -0,0 +1,106 @@ | |||
| 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 | |||
| 19 | #ifndef __BTRFS_TRANSACTION__ | ||
| 20 | #define __BTRFS_TRANSACTION__ | ||
| 21 | #include "btrfs_inode.h" | ||
| 22 | |||
| 23 | struct btrfs_transaction { | ||
| 24 | u64 transid; | ||
| 25 | unsigned long num_writers; | ||
| 26 | unsigned long num_joined; | ||
| 27 | int in_commit; | ||
| 28 | int use_count; | ||
| 29 | int commit_done; | ||
| 30 | int blocked; | ||
| 31 | struct list_head list; | ||
| 32 | struct extent_io_tree dirty_pages; | ||
| 33 | unsigned long start_time; | ||
| 34 | wait_queue_head_t writer_wait; | ||
| 35 | wait_queue_head_t commit_wait; | ||
| 36 | struct list_head pending_snapshots; | ||
| 37 | }; | ||
| 38 | |||
| 39 | struct btrfs_trans_handle { | ||
| 40 | u64 transid; | ||
| 41 | unsigned long blocks_reserved; | ||
| 42 | unsigned long blocks_used; | ||
| 43 | struct btrfs_transaction *transaction; | ||
| 44 | u64 block_group; | ||
| 45 | u64 alloc_exclude_start; | ||
| 46 | u64 alloc_exclude_nr; | ||
| 47 | }; | ||
| 48 | |||
| 49 | struct btrfs_pending_snapshot { | ||
| 50 | struct dentry *dentry; | ||
| 51 | struct btrfs_root *root; | ||
| 52 | char *name; | ||
| 53 | struct btrfs_key root_key; | ||
| 54 | struct list_head list; | ||
| 55 | }; | ||
| 56 | |||
| 57 | struct btrfs_dirty_root { | ||
| 58 | struct list_head list; | ||
| 59 | struct btrfs_root *root; | ||
| 60 | struct btrfs_root *latest_root; | ||
| 61 | }; | ||
| 62 | |||
| 63 | static inline void btrfs_set_trans_block_group(struct btrfs_trans_handle *trans, | ||
| 64 | struct inode *inode) | ||
| 65 | { | ||
| 66 | trans->block_group = BTRFS_I(inode)->block_group; | ||
| 67 | } | ||
| 68 | |||
| 69 | static inline void btrfs_update_inode_block_group( | ||
| 70 | struct btrfs_trans_handle *trans, | ||
| 71 | struct inode *inode) | ||
| 72 | { | ||
| 73 | BTRFS_I(inode)->block_group = trans->block_group; | ||
| 74 | } | ||
| 75 | |||
| 76 | static inline void btrfs_set_inode_last_trans(struct btrfs_trans_handle *trans, | ||
| 77 | struct inode *inode) | ||
| 78 | { | ||
| 79 | BTRFS_I(inode)->last_trans = trans->transaction->transid; | ||
| 80 | } | ||
| 81 | |||
| 82 | int btrfs_end_transaction(struct btrfs_trans_handle *trans, | ||
| 83 | struct btrfs_root *root); | ||
| 84 | struct btrfs_trans_handle *btrfs_start_transaction(struct btrfs_root *root, | ||
| 85 | int num_blocks); | ||
| 86 | struct btrfs_trans_handle *btrfs_join_transaction(struct btrfs_root *root, | ||
| 87 | int num_blocks); | ||
| 88 | struct btrfs_trans_handle *btrfs_start_ioctl_transaction(struct btrfs_root *r, | ||
| 89 | int num_blocks); | ||
| 90 | int btrfs_write_and_wait_transaction(struct btrfs_trans_handle *trans, | ||
| 91 | struct btrfs_root *root); | ||
| 92 | int btrfs_commit_tree_roots(struct btrfs_trans_handle *trans, | ||
| 93 | struct btrfs_root *root); | ||
| 94 | |||
| 95 | int btrfs_add_dead_root(struct btrfs_root *root, struct btrfs_root *latest); | ||
| 96 | int btrfs_defrag_root(struct btrfs_root *root, int cacheonly); | ||
| 97 | int btrfs_clean_old_snapshots(struct btrfs_root *root); | ||
| 98 | int btrfs_commit_transaction(struct btrfs_trans_handle *trans, | ||
| 99 | struct btrfs_root *root); | ||
| 100 | int btrfs_end_transaction_throttle(struct btrfs_trans_handle *trans, | ||
| 101 | struct btrfs_root *root); | ||
| 102 | void btrfs_throttle(struct btrfs_root *root); | ||
| 103 | int btrfs_record_root_in_trans(struct btrfs_root *root); | ||
| 104 | int btrfs_write_and_wait_marked_extents(struct btrfs_root *root, | ||
| 105 | struct extent_io_tree *dirty_pages); | ||
| 106 | #endif | ||
diff --git a/fs/btrfs/tree-defrag.c b/fs/btrfs/tree-defrag.c new file mode 100644 index 000000000000..3e8358c36165 --- /dev/null +++ b/fs/btrfs/tree-defrag.c | |||
| @@ -0,0 +1,147 @@ | |||
| 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 | |||
| 19 | #include <linux/sched.h> | ||
| 20 | #include "ctree.h" | ||
| 21 | #include "disk-io.h" | ||
| 22 | #include "print-tree.h" | ||
| 23 | #include "transaction.h" | ||
| 24 | #include "locking.h" | ||
| 25 | |||
| 26 | /* defrag all the leaves in a given btree. If cache_only == 1, don't read | ||
| 27 | * things from disk, otherwise read all the leaves and try to get key order to | ||
| 28 | * better reflect disk order | ||
| 29 | */ | ||
| 30 | |||
| 31 | int btrfs_defrag_leaves(struct btrfs_trans_handle *trans, | ||
| 32 | struct btrfs_root *root, int cache_only) | ||
| 33 | { | ||
| 34 | struct btrfs_path *path = NULL; | ||
| 35 | struct btrfs_key key; | ||
| 36 | int ret = 0; | ||
| 37 | int wret; | ||
| 38 | int level; | ||
| 39 | int orig_level; | ||
| 40 | int is_extent = 0; | ||
| 41 | int next_key_ret = 0; | ||
| 42 | u64 last_ret = 0; | ||
| 43 | u64 min_trans = 0; | ||
| 44 | |||
| 45 | if (cache_only) | ||
| 46 | goto out; | ||
| 47 | |||
| 48 | if (root->fs_info->extent_root == root) { | ||
| 49 | /* | ||
| 50 | * there's recursion here right now in the tree locking, | ||
| 51 | * we can't defrag the extent root without deadlock | ||
| 52 | */ | ||
| 53 | goto out; | ||
| 54 | } | ||
| 55 | |||
| 56 | if (root->ref_cows == 0 && !is_extent) | ||
| 57 | goto out; | ||
| 58 | |||
| 59 | if (btrfs_test_opt(root, SSD)) | ||
| 60 | goto out; | ||
| 61 | |||
| 62 | path = btrfs_alloc_path(); | ||
| 63 | if (!path) | ||
| 64 | return -ENOMEM; | ||
| 65 | |||
| 66 | level = btrfs_header_level(root->node); | ||
| 67 | orig_level = level; | ||
| 68 | |||
| 69 | if (level == 0) | ||
| 70 | goto out; | ||
| 71 | |||
| 72 | if (root->defrag_progress.objectid == 0) { | ||
| 73 | struct extent_buffer *root_node; | ||
| 74 | u32 nritems; | ||
| 75 | |||
| 76 | root_node = btrfs_lock_root_node(root); | ||
| 77 | nritems = btrfs_header_nritems(root_node); | ||
| 78 | root->defrag_max.objectid = 0; | ||
| 79 | /* from above we know this is not a leaf */ | ||
| 80 | btrfs_node_key_to_cpu(root_node, &root->defrag_max, | ||
| 81 | nritems - 1); | ||
| 82 | btrfs_tree_unlock(root_node); | ||
| 83 | free_extent_buffer(root_node); | ||
| 84 | memset(&key, 0, sizeof(key)); | ||
| 85 | } else { | ||
| 86 | memcpy(&key, &root->defrag_progress, sizeof(key)); | ||
| 87 | } | ||
| 88 | |||
| 89 | path->keep_locks = 1; | ||
| 90 | if (cache_only) | ||
| 91 | min_trans = root->defrag_trans_start; | ||
| 92 | |||
| 93 | ret = btrfs_search_forward(root, &key, NULL, path, | ||
| 94 | cache_only, min_trans); | ||
| 95 | if (ret < 0) | ||
| 96 | goto out; | ||
| 97 | if (ret > 0) { | ||
| 98 | ret = 0; | ||
| 99 | goto out; | ||
| 100 | } | ||
| 101 | btrfs_release_path(root, path); | ||
| 102 | wret = btrfs_search_slot(trans, root, &key, path, 0, 1); | ||
| 103 | |||
| 104 | if (wret < 0) { | ||
| 105 | ret = wret; | ||
| 106 | goto out; | ||
| 107 | } | ||
| 108 | if (!path->nodes[1]) { | ||
| 109 | ret = 0; | ||
| 110 | goto out; | ||
| 111 | } | ||
| 112 | path->slots[1] = btrfs_header_nritems(path->nodes[1]); | ||
| 113 | next_key_ret = btrfs_find_next_key(root, path, &key, 1, cache_only, | ||
| 114 | min_trans); | ||
| 115 | ret = btrfs_realloc_node(trans, root, | ||
| 116 | path->nodes[1], 0, | ||
| 117 | cache_only, &last_ret, | ||
| 118 | &root->defrag_progress); | ||
| 119 | WARN_ON(ret && ret != -EAGAIN); | ||
| 120 | if (next_key_ret == 0) { | ||
| 121 | memcpy(&root->defrag_progress, &key, sizeof(key)); | ||
| 122 | ret = -EAGAIN; | ||
| 123 | } | ||
| 124 | |||
| 125 | btrfs_release_path(root, path); | ||
| 126 | if (is_extent) | ||
| 127 | btrfs_extent_post_op(trans, root); | ||
| 128 | out: | ||
| 129 | if (path) | ||
| 130 | btrfs_free_path(path); | ||
| 131 | if (ret == -EAGAIN) { | ||
| 132 | if (root->defrag_max.objectid > root->defrag_progress.objectid) | ||
| 133 | goto done; | ||
| 134 | if (root->defrag_max.type > root->defrag_progress.type) | ||
| 135 | goto done; | ||
| 136 | if (root->defrag_max.offset > root->defrag_progress.offset) | ||
| 137 | goto done; | ||
| 138 | ret = 0; | ||
| 139 | } | ||
| 140 | done: | ||
| 141 | if (ret != -EAGAIN) { | ||
| 142 | memset(&root->defrag_progress, 0, | ||
| 143 | sizeof(root->defrag_progress)); | ||
| 144 | root->defrag_trans_start = trans->transid; | ||
| 145 | } | ||
| 146 | return ret; | ||
| 147 | } | ||
diff --git a/fs/btrfs/tree-log.c b/fs/btrfs/tree-log.c new file mode 100644 index 000000000000..d81cda2e077c --- /dev/null +++ b/fs/btrfs/tree-log.c | |||
| @@ -0,0 +1,2898 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | |||
| 19 | #include <linux/sched.h> | ||
| 20 | #include "ctree.h" | ||
| 21 | #include "transaction.h" | ||
| 22 | #include "disk-io.h" | ||
| 23 | #include "locking.h" | ||
| 24 | #include "print-tree.h" | ||
| 25 | #include "compat.h" | ||
| 26 | #include "tree-log.h" | ||
| 27 | |||
| 28 | /* magic values for the inode_only field in btrfs_log_inode: | ||
| 29 | * | ||
| 30 | * LOG_INODE_ALL means to log everything | ||
| 31 | * LOG_INODE_EXISTS means to log just enough to recreate the inode | ||
| 32 | * during log replay | ||
| 33 | */ | ||
| 34 | #define LOG_INODE_ALL 0 | ||
| 35 | #define LOG_INODE_EXISTS 1 | ||
| 36 | |||
| 37 | /* | ||
| 38 | * stages for the tree walking. The first | ||
| 39 | * stage (0) is to only pin down the blocks we find | ||
| 40 | * the second stage (1) is to make sure that all the inodes | ||
| 41 | * we find in the log are created in the subvolume. | ||
| 42 | * | ||
| 43 | * The last stage is to deal with directories and links and extents | ||
| 44 | * and all the other fun semantics | ||
| 45 | */ | ||
| 46 | #define LOG_WALK_PIN_ONLY 0 | ||
| 47 | #define LOG_WALK_REPLAY_INODES 1 | ||
| 48 | #define LOG_WALK_REPLAY_ALL 2 | ||
| 49 | |||
| 50 | static int __btrfs_log_inode(struct btrfs_trans_handle *trans, | ||
| 51 | struct btrfs_root *root, struct inode *inode, | ||
| 52 | int inode_only); | ||
| 53 | static int link_to_fixup_dir(struct btrfs_trans_handle *trans, | ||
| 54 | struct btrfs_root *root, | ||
| 55 | struct btrfs_path *path, u64 objectid); | ||
| 56 | |||
| 57 | /* | ||
| 58 | * tree logging is a special write ahead log used to make sure that | ||
| 59 | * fsyncs and O_SYNCs can happen without doing full tree commits. | ||
| 60 | * | ||
| 61 | * Full tree commits are expensive because they require commonly | ||
| 62 | * modified blocks to be recowed, creating many dirty pages in the | ||
| 63 | * extent tree an 4x-6x higher write load than ext3. | ||
| 64 | * | ||
| 65 | * Instead of doing a tree commit on every fsync, we use the | ||
| 66 | * key ranges and transaction ids to find items for a given file or directory | ||
| 67 | * that have changed in this transaction. Those items are copied into | ||
| 68 | * a special tree (one per subvolume root), that tree is written to disk | ||
| 69 | * and then the fsync is considered complete. | ||
| 70 | * | ||
| 71 | * After a crash, items are copied out of the log-tree back into the | ||
| 72 | * subvolume tree. Any file data extents found are recorded in the extent | ||
| 73 | * allocation tree, and the log-tree freed. | ||
| 74 | * | ||
| 75 | * The log tree is read three times, once to pin down all the extents it is | ||
| 76 | * using in ram and once, once to create all the inodes logged in the tree | ||
| 77 | * and once to do all the other items. | ||
| 78 | */ | ||
| 79 | |||
| 80 | /* | ||
| 81 | * btrfs_add_log_tree adds a new per-subvolume log tree into the | ||
| 82 | * tree of log tree roots. This must be called with a tree log transaction | ||
| 83 | * running (see start_log_trans). | ||
| 84 | */ | ||
| 85 | static int btrfs_add_log_tree(struct btrfs_trans_handle *trans, | ||
| 86 | struct btrfs_root *root) | ||
| 87 | { | ||
| 88 | struct btrfs_key key; | ||
| 89 | struct btrfs_root_item root_item; | ||
| 90 | struct btrfs_inode_item *inode_item; | ||
| 91 | struct extent_buffer *leaf; | ||
| 92 | struct btrfs_root *new_root = root; | ||
| 93 | int ret; | ||
| 94 | u64 objectid = root->root_key.objectid; | ||
| 95 | |||
| 96 | leaf = btrfs_alloc_free_block(trans, root, root->leafsize, 0, | ||
| 97 | BTRFS_TREE_LOG_OBJECTID, | ||
| 98 | trans->transid, 0, 0, 0); | ||
| 99 | if (IS_ERR(leaf)) { | ||
| 100 | ret = PTR_ERR(leaf); | ||
| 101 | return ret; | ||
| 102 | } | ||
| 103 | |||
| 104 | btrfs_set_header_nritems(leaf, 0); | ||
| 105 | btrfs_set_header_level(leaf, 0); | ||
| 106 | btrfs_set_header_bytenr(leaf, leaf->start); | ||
| 107 | btrfs_set_header_generation(leaf, trans->transid); | ||
| 108 | btrfs_set_header_owner(leaf, BTRFS_TREE_LOG_OBJECTID); | ||
| 109 | |||
| 110 | write_extent_buffer(leaf, root->fs_info->fsid, | ||
| 111 | (unsigned long)btrfs_header_fsid(leaf), | ||
| 112 | BTRFS_FSID_SIZE); | ||
| 113 | btrfs_mark_buffer_dirty(leaf); | ||
| 114 | |||
| 115 | inode_item = &root_item.inode; | ||
| 116 | memset(inode_item, 0, sizeof(*inode_item)); | ||
| 117 | inode_item->generation = cpu_to_le64(1); | ||
| 118 | inode_item->size = cpu_to_le64(3); | ||
| 119 | inode_item->nlink = cpu_to_le32(1); | ||
| 120 | inode_item->nbytes = cpu_to_le64(root->leafsize); | ||
| 121 | inode_item->mode = cpu_to_le32(S_IFDIR | 0755); | ||
| 122 | |||
| 123 | btrfs_set_root_bytenr(&root_item, leaf->start); | ||
| 124 | btrfs_set_root_generation(&root_item, trans->transid); | ||
| 125 | btrfs_set_root_level(&root_item, 0); | ||
| 126 | btrfs_set_root_refs(&root_item, 0); | ||
| 127 | btrfs_set_root_used(&root_item, 0); | ||
| 128 | |||
| 129 | memset(&root_item.drop_progress, 0, sizeof(root_item.drop_progress)); | ||
| 130 | root_item.drop_level = 0; | ||
| 131 | |||
| 132 | btrfs_tree_unlock(leaf); | ||
| 133 | free_extent_buffer(leaf); | ||
| 134 | leaf = NULL; | ||
| 135 | |||
| 136 | btrfs_set_root_dirid(&root_item, 0); | ||
| 137 | |||
| 138 | key.objectid = BTRFS_TREE_LOG_OBJECTID; | ||
| 139 | key.offset = objectid; | ||
| 140 | btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY); | ||
| 141 | ret = btrfs_insert_root(trans, root->fs_info->log_root_tree, &key, | ||
| 142 | &root_item); | ||
| 143 | if (ret) | ||
| 144 | goto fail; | ||
| 145 | |||
| 146 | new_root = btrfs_read_fs_root_no_radix(root->fs_info->log_root_tree, | ||
| 147 | &key); | ||
| 148 | BUG_ON(!new_root); | ||
| 149 | |||
| 150 | WARN_ON(root->log_root); | ||
| 151 | root->log_root = new_root; | ||
| 152 | |||
| 153 | /* | ||
| 154 | * log trees do not get reference counted because they go away | ||
| 155 | * before a real commit is actually done. They do store pointers | ||
| 156 | * to file data extents, and those reference counts still get | ||
| 157 | * updated (along with back refs to the log tree). | ||
| 158 | */ | ||
| 159 | new_root->ref_cows = 0; | ||
| 160 | new_root->last_trans = trans->transid; | ||
| 161 | |||
| 162 | /* | ||
| 163 | * we need to make sure the root block for this new tree | ||
| 164 | * is marked as dirty in the dirty_log_pages tree. This | ||
| 165 | * is how it gets flushed down to disk at tree log commit time. | ||
| 166 | * | ||
| 167 | * the tree logging mutex keeps others from coming in and changing | ||
| 168 | * the new_root->node, so we can safely access it here | ||
| 169 | */ | ||
| 170 | set_extent_dirty(&new_root->dirty_log_pages, new_root->node->start, | ||
| 171 | new_root->node->start + new_root->node->len - 1, | ||
| 172 | GFP_NOFS); | ||
| 173 | |||
| 174 | fail: | ||
| 175 | return ret; | ||
| 176 | } | ||
| 177 | |||
| 178 | /* | ||
| 179 | * start a sub transaction and setup the log tree | ||
| 180 | * this increments the log tree writer count to make the people | ||
| 181 | * syncing the tree wait for us to finish | ||
| 182 | */ | ||
| 183 | static int start_log_trans(struct btrfs_trans_handle *trans, | ||
| 184 | struct btrfs_root *root) | ||
| 185 | { | ||
| 186 | int ret; | ||
| 187 | mutex_lock(&root->fs_info->tree_log_mutex); | ||
| 188 | if (!root->fs_info->log_root_tree) { | ||
| 189 | ret = btrfs_init_log_root_tree(trans, root->fs_info); | ||
| 190 | BUG_ON(ret); | ||
| 191 | } | ||
| 192 | if (!root->log_root) { | ||
| 193 | ret = btrfs_add_log_tree(trans, root); | ||
| 194 | BUG_ON(ret); | ||
| 195 | } | ||
| 196 | atomic_inc(&root->fs_info->tree_log_writers); | ||
| 197 | root->fs_info->tree_log_batch++; | ||
| 198 | mutex_unlock(&root->fs_info->tree_log_mutex); | ||
| 199 | return 0; | ||
| 200 | } | ||
| 201 | |||
| 202 | /* | ||
| 203 | * returns 0 if there was a log transaction running and we were able | ||
| 204 | * to join, or returns -ENOENT if there were not transactions | ||
| 205 | * in progress | ||
| 206 | */ | ||
| 207 | static int join_running_log_trans(struct btrfs_root *root) | ||
| 208 | { | ||
| 209 | int ret = -ENOENT; | ||
| 210 | |||
| 211 | smp_mb(); | ||
| 212 | if (!root->log_root) | ||
| 213 | return -ENOENT; | ||
| 214 | |||
| 215 | mutex_lock(&root->fs_info->tree_log_mutex); | ||
| 216 | if (root->log_root) { | ||
| 217 | ret = 0; | ||
| 218 | atomic_inc(&root->fs_info->tree_log_writers); | ||
| 219 | root->fs_info->tree_log_batch++; | ||
| 220 | } | ||
| 221 | mutex_unlock(&root->fs_info->tree_log_mutex); | ||
| 222 | return ret; | ||
| 223 | } | ||
| 224 | |||
| 225 | /* | ||
| 226 | * indicate we're done making changes to the log tree | ||
| 227 | * and wake up anyone waiting to do a sync | ||
| 228 | */ | ||
| 229 | static int end_log_trans(struct btrfs_root *root) | ||
| 230 | { | ||
| 231 | atomic_dec(&root->fs_info->tree_log_writers); | ||
| 232 | smp_mb(); | ||
| 233 | if (waitqueue_active(&root->fs_info->tree_log_wait)) | ||
| 234 | wake_up(&root->fs_info->tree_log_wait); | ||
| 235 | return 0; | ||
| 236 | } | ||
| 237 | |||
| 238 | |||
| 239 | /* | ||
| 240 | * the walk control struct is used to pass state down the chain when | ||
| 241 | * processing the log tree. The stage field tells us which part | ||
| 242 | * of the log tree processing we are currently doing. The others | ||
| 243 | * are state fields used for that specific part | ||
| 244 | */ | ||
| 245 | struct walk_control { | ||
| 246 | /* should we free the extent on disk when done? This is used | ||
| 247 | * at transaction commit time while freeing a log tree | ||
| 248 | */ | ||
| 249 | int free; | ||
| 250 | |||
| 251 | /* should we write out the extent buffer? This is used | ||
| 252 | * while flushing the log tree to disk during a sync | ||
| 253 | */ | ||
| 254 | int write; | ||
| 255 | |||
| 256 | /* should we wait for the extent buffer io to finish? Also used | ||
| 257 | * while flushing the log tree to disk for a sync | ||
| 258 | */ | ||
| 259 | int wait; | ||
| 260 | |||
| 261 | /* pin only walk, we record which extents on disk belong to the | ||
| 262 | * log trees | ||
| 263 | */ | ||
| 264 | int pin; | ||
| 265 | |||
| 266 | /* what stage of the replay code we're currently in */ | ||
| 267 | int stage; | ||
| 268 | |||
| 269 | /* the root we are currently replaying */ | ||
| 270 | struct btrfs_root *replay_dest; | ||
| 271 | |||
| 272 | /* the trans handle for the current replay */ | ||
| 273 | struct btrfs_trans_handle *trans; | ||
| 274 | |||
| 275 | /* the function that gets used to process blocks we find in the | ||
| 276 | * tree. Note the extent_buffer might not be up to date when it is | ||
| 277 | * passed in, and it must be checked or read if you need the data | ||
| 278 | * inside it | ||
| 279 | */ | ||
| 280 | int (*process_func)(struct btrfs_root *log, struct extent_buffer *eb, | ||
| 281 | struct walk_control *wc, u64 gen); | ||
| 282 | }; | ||
| 283 | |||
| 284 | /* | ||
| 285 | * process_func used to pin down extents, write them or wait on them | ||
| 286 | */ | ||
| 287 | static int process_one_buffer(struct btrfs_root *log, | ||
| 288 | struct extent_buffer *eb, | ||
| 289 | struct walk_control *wc, u64 gen) | ||
| 290 | { | ||
| 291 | if (wc->pin) { | ||
| 292 | mutex_lock(&log->fs_info->pinned_mutex); | ||
| 293 | btrfs_update_pinned_extents(log->fs_info->extent_root, | ||
| 294 | eb->start, eb->len, 1); | ||
| 295 | mutex_unlock(&log->fs_info->pinned_mutex); | ||
| 296 | } | ||
| 297 | |||
| 298 | if (btrfs_buffer_uptodate(eb, gen)) { | ||
| 299 | if (wc->write) | ||
| 300 | btrfs_write_tree_block(eb); | ||
| 301 | if (wc->wait) | ||
| 302 | btrfs_wait_tree_block_writeback(eb); | ||
| 303 | } | ||
| 304 | return 0; | ||
| 305 | } | ||
| 306 | |||
| 307 | /* | ||
| 308 | * Item overwrite used by replay and tree logging. eb, slot and key all refer | ||
| 309 | * to the src data we are copying out. | ||
| 310 | * | ||
| 311 | * root is the tree we are copying into, and path is a scratch | ||
| 312 | * path for use in this function (it should be released on entry and | ||
| 313 | * will be released on exit). | ||
| 314 | * | ||
| 315 | * If the key is already in the destination tree the existing item is | ||
| 316 | * overwritten. If the existing item isn't big enough, it is extended. | ||
| 317 | * If it is too large, it is truncated. | ||
| 318 | * | ||
| 319 | * If the key isn't in the destination yet, a new item is inserted. | ||
| 320 | */ | ||
| 321 | static noinline int overwrite_item(struct btrfs_trans_handle *trans, | ||
| 322 | struct btrfs_root *root, | ||
| 323 | struct btrfs_path *path, | ||
| 324 | struct extent_buffer *eb, int slot, | ||
| 325 | struct btrfs_key *key) | ||
| 326 | { | ||
| 327 | int ret; | ||
| 328 | u32 item_size; | ||
| 329 | u64 saved_i_size = 0; | ||
| 330 | int save_old_i_size = 0; | ||
| 331 | unsigned long src_ptr; | ||
| 332 | unsigned long dst_ptr; | ||
| 333 | int overwrite_root = 0; | ||
| 334 | |||
| 335 | if (root->root_key.objectid != BTRFS_TREE_LOG_OBJECTID) | ||
| 336 | overwrite_root = 1; | ||
| 337 | |||
| 338 | item_size = btrfs_item_size_nr(eb, slot); | ||
| 339 | src_ptr = btrfs_item_ptr_offset(eb, slot); | ||
| 340 | |||
| 341 | /* look for the key in the destination tree */ | ||
| 342 | ret = btrfs_search_slot(NULL, root, key, path, 0, 0); | ||
| 343 | if (ret == 0) { | ||
| 344 | char *src_copy; | ||
| 345 | char *dst_copy; | ||
| 346 | u32 dst_size = btrfs_item_size_nr(path->nodes[0], | ||
| 347 | path->slots[0]); | ||
| 348 | if (dst_size != item_size) | ||
| 349 | goto insert; | ||
| 350 | |||
| 351 | if (item_size == 0) { | ||
| 352 | btrfs_release_path(root, path); | ||
| 353 | return 0; | ||
| 354 | } | ||
| 355 | dst_copy = kmalloc(item_size, GFP_NOFS); | ||
| 356 | src_copy = kmalloc(item_size, GFP_NOFS); | ||
| 357 | |||
| 358 | read_extent_buffer(eb, src_copy, src_ptr, item_size); | ||
| 359 | |||
| 360 | dst_ptr = btrfs_item_ptr_offset(path->nodes[0], path->slots[0]); | ||
| 361 | read_extent_buffer(path->nodes[0], dst_copy, dst_ptr, | ||
| 362 | item_size); | ||
| 363 | ret = memcmp(dst_copy, src_copy, item_size); | ||
| 364 | |||
| 365 | kfree(dst_copy); | ||
| 366 | kfree(src_copy); | ||
| 367 | /* | ||
| 368 | * they have the same contents, just return, this saves | ||
| 369 | * us from cowing blocks in the destination tree and doing | ||
| 370 | * extra writes that may not have been done by a previous | ||
| 371 | * sync | ||
| 372 | */ | ||
| 373 | if (ret == 0) { | ||
| 374 | btrfs_release_path(root, path); | ||
| 375 | return 0; | ||
| 376 | } | ||
| 377 | |||
| 378 | } | ||
| 379 | insert: | ||
| 380 | btrfs_release_path(root, path); | ||
| 381 | /* try to insert the key into the destination tree */ | ||
| 382 | ret = btrfs_insert_empty_item(trans, root, path, | ||
| 383 | key, item_size); | ||
| 384 | |||
| 385 | /* make sure any existing item is the correct size */ | ||
| 386 | if (ret == -EEXIST) { | ||
| 387 | u32 found_size; | ||
| 388 | found_size = btrfs_item_size_nr(path->nodes[0], | ||
| 389 | path->slots[0]); | ||
| 390 | if (found_size > item_size) { | ||
| 391 | btrfs_truncate_item(trans, root, path, item_size, 1); | ||
| 392 | } else if (found_size < item_size) { | ||
| 393 | ret = btrfs_extend_item(trans, root, path, | ||
| 394 | item_size - found_size); | ||
| 395 | BUG_ON(ret); | ||
| 396 | } | ||
| 397 | } else if (ret) { | ||
| 398 | BUG(); | ||
| 399 | } | ||
| 400 | dst_ptr = btrfs_item_ptr_offset(path->nodes[0], | ||
| 401 | path->slots[0]); | ||
| 402 | |||
| 403 | /* don't overwrite an existing inode if the generation number | ||
| 404 | * was logged as zero. This is done when the tree logging code | ||
| 405 | * is just logging an inode to make sure it exists after recovery. | ||
| 406 | * | ||
| 407 | * Also, don't overwrite i_size on directories during replay. | ||
| 408 | * log replay inserts and removes directory items based on the | ||
| 409 | * state of the tree found in the subvolume, and i_size is modified | ||
| 410 | * as it goes | ||
| 411 | */ | ||
| 412 | if (key->type == BTRFS_INODE_ITEM_KEY && ret == -EEXIST) { | ||
| 413 | struct btrfs_inode_item *src_item; | ||
| 414 | struct btrfs_inode_item *dst_item; | ||
| 415 | |||
| 416 | src_item = (struct btrfs_inode_item *)src_ptr; | ||
| 417 | dst_item = (struct btrfs_inode_item *)dst_ptr; | ||
| 418 | |||
| 419 | if (btrfs_inode_generation(eb, src_item) == 0) | ||
| 420 | goto no_copy; | ||
| 421 | |||
| 422 | if (overwrite_root && | ||
| 423 | S_ISDIR(btrfs_inode_mode(eb, src_item)) && | ||
| 424 | S_ISDIR(btrfs_inode_mode(path->nodes[0], dst_item))) { | ||
| 425 | save_old_i_size = 1; | ||
| 426 | saved_i_size = btrfs_inode_size(path->nodes[0], | ||
| 427 | dst_item); | ||
| 428 | } | ||
| 429 | } | ||
| 430 | |||
| 431 | copy_extent_buffer(path->nodes[0], eb, dst_ptr, | ||
| 432 | src_ptr, item_size); | ||
| 433 | |||
| 434 | if (save_old_i_size) { | ||
| 435 | struct btrfs_inode_item *dst_item; | ||
| 436 | dst_item = (struct btrfs_inode_item *)dst_ptr; | ||
| 437 | btrfs_set_inode_size(path->nodes[0], dst_item, saved_i_size); | ||
| 438 | } | ||
| 439 | |||
| 440 | /* make sure the generation is filled in */ | ||
| 441 | if (key->type == BTRFS_INODE_ITEM_KEY) { | ||
| 442 | struct btrfs_inode_item *dst_item; | ||
| 443 | dst_item = (struct btrfs_inode_item *)dst_ptr; | ||
| 444 | if (btrfs_inode_generation(path->nodes[0], dst_item) == 0) { | ||
| 445 | btrfs_set_inode_generation(path->nodes[0], dst_item, | ||
| 446 | trans->transid); | ||
| 447 | } | ||
| 448 | } | ||
| 449 | no_copy: | ||
| 450 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 451 | btrfs_release_path(root, path); | ||
| 452 | return 0; | ||
| 453 | } | ||
| 454 | |||
| 455 | /* | ||
| 456 | * simple helper to read an inode off the disk from a given root | ||
| 457 | * This can only be called for subvolume roots and not for the log | ||
| 458 | */ | ||
| 459 | static noinline struct inode *read_one_inode(struct btrfs_root *root, | ||
| 460 | u64 objectid) | ||
| 461 | { | ||
| 462 | struct inode *inode; | ||
| 463 | inode = btrfs_iget_locked(root->fs_info->sb, objectid, root); | ||
| 464 | if (inode->i_state & I_NEW) { | ||
| 465 | BTRFS_I(inode)->root = root; | ||
| 466 | BTRFS_I(inode)->location.objectid = objectid; | ||
| 467 | BTRFS_I(inode)->location.type = BTRFS_INODE_ITEM_KEY; | ||
| 468 | BTRFS_I(inode)->location.offset = 0; | ||
| 469 | btrfs_read_locked_inode(inode); | ||
| 470 | unlock_new_inode(inode); | ||
| 471 | |||
| 472 | } | ||
| 473 | if (is_bad_inode(inode)) { | ||
| 474 | iput(inode); | ||
| 475 | inode = NULL; | ||
| 476 | } | ||
| 477 | return inode; | ||
| 478 | } | ||
| 479 | |||
| 480 | /* replays a single extent in 'eb' at 'slot' with 'key' into the | ||
| 481 | * subvolume 'root'. path is released on entry and should be released | ||
| 482 | * on exit. | ||
| 483 | * | ||
| 484 | * extents in the log tree have not been allocated out of the extent | ||
| 485 | * tree yet. So, this completes the allocation, taking a reference | ||
| 486 | * as required if the extent already exists or creating a new extent | ||
| 487 | * if it isn't in the extent allocation tree yet. | ||
| 488 | * | ||
| 489 | * The extent is inserted into the file, dropping any existing extents | ||
| 490 | * from the file that overlap the new one. | ||
| 491 | */ | ||
| 492 | static noinline int replay_one_extent(struct btrfs_trans_handle *trans, | ||
| 493 | struct btrfs_root *root, | ||
| 494 | struct btrfs_path *path, | ||
| 495 | struct extent_buffer *eb, int slot, | ||
| 496 | struct btrfs_key *key) | ||
| 497 | { | ||
| 498 | int found_type; | ||
| 499 | u64 mask = root->sectorsize - 1; | ||
| 500 | u64 extent_end; | ||
| 501 | u64 alloc_hint; | ||
| 502 | u64 start = key->offset; | ||
| 503 | u64 saved_nbytes; | ||
| 504 | struct btrfs_file_extent_item *item; | ||
| 505 | struct inode *inode = NULL; | ||
| 506 | unsigned long size; | ||
| 507 | int ret = 0; | ||
| 508 | |||
| 509 | item = btrfs_item_ptr(eb, slot, struct btrfs_file_extent_item); | ||
| 510 | found_type = btrfs_file_extent_type(eb, item); | ||
| 511 | |||
| 512 | if (found_type == BTRFS_FILE_EXTENT_REG || | ||
| 513 | found_type == BTRFS_FILE_EXTENT_PREALLOC) | ||
| 514 | extent_end = start + btrfs_file_extent_num_bytes(eb, item); | ||
| 515 | else if (found_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 516 | size = btrfs_file_extent_inline_len(eb, item); | ||
| 517 | extent_end = (start + size + mask) & ~mask; | ||
| 518 | } else { | ||
| 519 | ret = 0; | ||
| 520 | goto out; | ||
| 521 | } | ||
| 522 | |||
| 523 | inode = read_one_inode(root, key->objectid); | ||
| 524 | if (!inode) { | ||
| 525 | ret = -EIO; | ||
| 526 | goto out; | ||
| 527 | } | ||
| 528 | |||
| 529 | /* | ||
| 530 | * first check to see if we already have this extent in the | ||
| 531 | * file. This must be done before the btrfs_drop_extents run | ||
| 532 | * so we don't try to drop this extent. | ||
| 533 | */ | ||
| 534 | ret = btrfs_lookup_file_extent(trans, root, path, inode->i_ino, | ||
| 535 | start, 0); | ||
| 536 | |||
| 537 | if (ret == 0 && | ||
| 538 | (found_type == BTRFS_FILE_EXTENT_REG || | ||
| 539 | found_type == BTRFS_FILE_EXTENT_PREALLOC)) { | ||
| 540 | struct btrfs_file_extent_item cmp1; | ||
| 541 | struct btrfs_file_extent_item cmp2; | ||
| 542 | struct btrfs_file_extent_item *existing; | ||
| 543 | struct extent_buffer *leaf; | ||
| 544 | |||
| 545 | leaf = path->nodes[0]; | ||
| 546 | existing = btrfs_item_ptr(leaf, path->slots[0], | ||
| 547 | struct btrfs_file_extent_item); | ||
| 548 | |||
| 549 | read_extent_buffer(eb, &cmp1, (unsigned long)item, | ||
| 550 | sizeof(cmp1)); | ||
| 551 | read_extent_buffer(leaf, &cmp2, (unsigned long)existing, | ||
| 552 | sizeof(cmp2)); | ||
| 553 | |||
| 554 | /* | ||
| 555 | * we already have a pointer to this exact extent, | ||
| 556 | * we don't have to do anything | ||
| 557 | */ | ||
| 558 | if (memcmp(&cmp1, &cmp2, sizeof(cmp1)) == 0) { | ||
| 559 | btrfs_release_path(root, path); | ||
| 560 | goto out; | ||
| 561 | } | ||
| 562 | } | ||
| 563 | btrfs_release_path(root, path); | ||
| 564 | |||
| 565 | saved_nbytes = inode_get_bytes(inode); | ||
| 566 | /* drop any overlapping extents */ | ||
| 567 | ret = btrfs_drop_extents(trans, root, inode, | ||
| 568 | start, extent_end, start, &alloc_hint); | ||
| 569 | BUG_ON(ret); | ||
| 570 | |||
| 571 | if (found_type == BTRFS_FILE_EXTENT_REG || | ||
| 572 | found_type == BTRFS_FILE_EXTENT_PREALLOC) { | ||
| 573 | unsigned long dest_offset; | ||
| 574 | struct btrfs_key ins; | ||
| 575 | |||
| 576 | ret = btrfs_insert_empty_item(trans, root, path, key, | ||
| 577 | sizeof(*item)); | ||
| 578 | BUG_ON(ret); | ||
| 579 | dest_offset = btrfs_item_ptr_offset(path->nodes[0], | ||
| 580 | path->slots[0]); | ||
| 581 | copy_extent_buffer(path->nodes[0], eb, dest_offset, | ||
| 582 | (unsigned long)item, sizeof(*item)); | ||
| 583 | |||
| 584 | ins.objectid = btrfs_file_extent_disk_bytenr(eb, item); | ||
| 585 | ins.offset = btrfs_file_extent_disk_num_bytes(eb, item); | ||
| 586 | ins.type = BTRFS_EXTENT_ITEM_KEY; | ||
| 587 | |||
| 588 | if (ins.objectid > 0) { | ||
| 589 | u64 csum_start; | ||
| 590 | u64 csum_end; | ||
| 591 | LIST_HEAD(ordered_sums); | ||
| 592 | /* | ||
| 593 | * is this extent already allocated in the extent | ||
| 594 | * allocation tree? If so, just add a reference | ||
| 595 | */ | ||
| 596 | ret = btrfs_lookup_extent(root, ins.objectid, | ||
| 597 | ins.offset); | ||
| 598 | if (ret == 0) { | ||
| 599 | ret = btrfs_inc_extent_ref(trans, root, | ||
| 600 | ins.objectid, ins.offset, | ||
| 601 | path->nodes[0]->start, | ||
| 602 | root->root_key.objectid, | ||
| 603 | trans->transid, key->objectid); | ||
| 604 | } else { | ||
| 605 | /* | ||
| 606 | * insert the extent pointer in the extent | ||
| 607 | * allocation tree | ||
| 608 | */ | ||
| 609 | ret = btrfs_alloc_logged_extent(trans, root, | ||
| 610 | path->nodes[0]->start, | ||
| 611 | root->root_key.objectid, | ||
| 612 | trans->transid, key->objectid, | ||
| 613 | &ins); | ||
| 614 | BUG_ON(ret); | ||
| 615 | } | ||
| 616 | btrfs_release_path(root, path); | ||
| 617 | |||
| 618 | if (btrfs_file_extent_compression(eb, item)) { | ||
| 619 | csum_start = ins.objectid; | ||
| 620 | csum_end = csum_start + ins.offset; | ||
| 621 | } else { | ||
| 622 | csum_start = ins.objectid + | ||
| 623 | btrfs_file_extent_offset(eb, item); | ||
| 624 | csum_end = csum_start + | ||
| 625 | btrfs_file_extent_num_bytes(eb, item); | ||
| 626 | } | ||
| 627 | |||
| 628 | ret = btrfs_lookup_csums_range(root->log_root, | ||
| 629 | csum_start, csum_end - 1, | ||
| 630 | &ordered_sums); | ||
| 631 | BUG_ON(ret); | ||
| 632 | while (!list_empty(&ordered_sums)) { | ||
| 633 | struct btrfs_ordered_sum *sums; | ||
| 634 | sums = list_entry(ordered_sums.next, | ||
| 635 | struct btrfs_ordered_sum, | ||
| 636 | list); | ||
| 637 | ret = btrfs_csum_file_blocks(trans, | ||
| 638 | root->fs_info->csum_root, | ||
| 639 | sums); | ||
| 640 | BUG_ON(ret); | ||
| 641 | list_del(&sums->list); | ||
| 642 | kfree(sums); | ||
| 643 | } | ||
| 644 | } else { | ||
| 645 | btrfs_release_path(root, path); | ||
| 646 | } | ||
| 647 | } else if (found_type == BTRFS_FILE_EXTENT_INLINE) { | ||
| 648 | /* inline extents are easy, we just overwrite them */ | ||
| 649 | ret = overwrite_item(trans, root, path, eb, slot, key); | ||
| 650 | BUG_ON(ret); | ||
| 651 | } | ||
| 652 | |||
| 653 | inode_set_bytes(inode, saved_nbytes); | ||
| 654 | btrfs_update_inode(trans, root, inode); | ||
| 655 | out: | ||
| 656 | if (inode) | ||
| 657 | iput(inode); | ||
| 658 | return ret; | ||
| 659 | } | ||
| 660 | |||
| 661 | /* | ||
| 662 | * when cleaning up conflicts between the directory names in the | ||
| 663 | * subvolume, directory names in the log and directory names in the | ||
| 664 | * inode back references, we may have to unlink inodes from directories. | ||
| 665 | * | ||
| 666 | * This is a helper function to do the unlink of a specific directory | ||
| 667 | * item | ||
| 668 | */ | ||
| 669 | static noinline int drop_one_dir_item(struct btrfs_trans_handle *trans, | ||
| 670 | struct btrfs_root *root, | ||
| 671 | struct btrfs_path *path, | ||
| 672 | struct inode *dir, | ||
| 673 | struct btrfs_dir_item *di) | ||
| 674 | { | ||
| 675 | struct inode *inode; | ||
| 676 | char *name; | ||
| 677 | int name_len; | ||
| 678 | struct extent_buffer *leaf; | ||
| 679 | struct btrfs_key location; | ||
| 680 | int ret; | ||
| 681 | |||
| 682 | leaf = path->nodes[0]; | ||
| 683 | |||
| 684 | btrfs_dir_item_key_to_cpu(leaf, di, &location); | ||
| 685 | name_len = btrfs_dir_name_len(leaf, di); | ||
| 686 | name = kmalloc(name_len, GFP_NOFS); | ||
| 687 | read_extent_buffer(leaf, name, (unsigned long)(di + 1), name_len); | ||
| 688 | btrfs_release_path(root, path); | ||
| 689 | |||
| 690 | inode = read_one_inode(root, location.objectid); | ||
| 691 | BUG_ON(!inode); | ||
| 692 | |||
| 693 | ret = link_to_fixup_dir(trans, root, path, location.objectid); | ||
| 694 | BUG_ON(ret); | ||
| 695 | ret = btrfs_unlink_inode(trans, root, dir, inode, name, name_len); | ||
| 696 | BUG_ON(ret); | ||
| 697 | kfree(name); | ||
| 698 | |||
| 699 | iput(inode); | ||
| 700 | return ret; | ||
| 701 | } | ||
| 702 | |||
| 703 | /* | ||
| 704 | * helper function to see if a given name and sequence number found | ||
| 705 | * in an inode back reference are already in a directory and correctly | ||
| 706 | * point to this inode | ||
| 707 | */ | ||
| 708 | static noinline int inode_in_dir(struct btrfs_root *root, | ||
| 709 | struct btrfs_path *path, | ||
| 710 | u64 dirid, u64 objectid, u64 index, | ||
| 711 | const char *name, int name_len) | ||
| 712 | { | ||
| 713 | struct btrfs_dir_item *di; | ||
| 714 | struct btrfs_key location; | ||
| 715 | int match = 0; | ||
| 716 | |||
| 717 | di = btrfs_lookup_dir_index_item(NULL, root, path, dirid, | ||
| 718 | index, name, name_len, 0); | ||
| 719 | if (di && !IS_ERR(di)) { | ||
| 720 | btrfs_dir_item_key_to_cpu(path->nodes[0], di, &location); | ||
| 721 | if (location.objectid != objectid) | ||
| 722 | goto out; | ||
| 723 | } else | ||
| 724 | goto out; | ||
| 725 | btrfs_release_path(root, path); | ||
| 726 | |||
| 727 | di = btrfs_lookup_dir_item(NULL, root, path, dirid, name, name_len, 0); | ||
| 728 | if (di && !IS_ERR(di)) { | ||
| 729 | btrfs_dir_item_key_to_cpu(path->nodes[0], di, &location); | ||
| 730 | if (location.objectid != objectid) | ||
| 731 | goto out; | ||
| 732 | } else | ||
| 733 | goto out; | ||
| 734 | match = 1; | ||
| 735 | out: | ||
| 736 | btrfs_release_path(root, path); | ||
| 737 | return match; | ||
| 738 | } | ||
| 739 | |||
| 740 | /* | ||
| 741 | * helper function to check a log tree for a named back reference in | ||
| 742 | * an inode. This is used to decide if a back reference that is | ||
| 743 | * found in the subvolume conflicts with what we find in the log. | ||
| 744 | * | ||
| 745 | * inode backreferences may have multiple refs in a single item, | ||
| 746 | * during replay we process one reference at a time, and we don't | ||
| 747 | * want to delete valid links to a file from the subvolume if that | ||
| 748 | * link is also in the log. | ||
| 749 | */ | ||
| 750 | static noinline int backref_in_log(struct btrfs_root *log, | ||
| 751 | struct btrfs_key *key, | ||
| 752 | char *name, int namelen) | ||
| 753 | { | ||
| 754 | struct btrfs_path *path; | ||
| 755 | struct btrfs_inode_ref *ref; | ||
| 756 | unsigned long ptr; | ||
| 757 | unsigned long ptr_end; | ||
| 758 | unsigned long name_ptr; | ||
| 759 | int found_name_len; | ||
| 760 | int item_size; | ||
| 761 | int ret; | ||
| 762 | int match = 0; | ||
| 763 | |||
| 764 | path = btrfs_alloc_path(); | ||
| 765 | ret = btrfs_search_slot(NULL, log, key, path, 0, 0); | ||
| 766 | if (ret != 0) | ||
| 767 | goto out; | ||
| 768 | |||
| 769 | item_size = btrfs_item_size_nr(path->nodes[0], path->slots[0]); | ||
| 770 | ptr = btrfs_item_ptr_offset(path->nodes[0], path->slots[0]); | ||
| 771 | ptr_end = ptr + item_size; | ||
| 772 | while (ptr < ptr_end) { | ||
| 773 | ref = (struct btrfs_inode_ref *)ptr; | ||
| 774 | found_name_len = btrfs_inode_ref_name_len(path->nodes[0], ref); | ||
| 775 | if (found_name_len == namelen) { | ||
| 776 | name_ptr = (unsigned long)(ref + 1); | ||
| 777 | ret = memcmp_extent_buffer(path->nodes[0], name, | ||
| 778 | name_ptr, namelen); | ||
| 779 | if (ret == 0) { | ||
| 780 | match = 1; | ||
| 781 | goto out; | ||
| 782 | } | ||
| 783 | } | ||
| 784 | ptr = (unsigned long)(ref + 1) + found_name_len; | ||
| 785 | } | ||
| 786 | out: | ||
| 787 | btrfs_free_path(path); | ||
| 788 | return match; | ||
| 789 | } | ||
| 790 | |||
| 791 | |||
| 792 | /* | ||
| 793 | * replay one inode back reference item found in the log tree. | ||
| 794 | * eb, slot and key refer to the buffer and key found in the log tree. | ||
| 795 | * root is the destination we are replaying into, and path is for temp | ||
| 796 | * use by this function. (it should be released on return). | ||
| 797 | */ | ||
| 798 | static noinline int add_inode_ref(struct btrfs_trans_handle *trans, | ||
| 799 | struct btrfs_root *root, | ||
| 800 | struct btrfs_root *log, | ||
| 801 | struct btrfs_path *path, | ||
| 802 | struct extent_buffer *eb, int slot, | ||
| 803 | struct btrfs_key *key) | ||
| 804 | { | ||
| 805 | struct inode *dir; | ||
| 806 | int ret; | ||
| 807 | struct btrfs_key location; | ||
| 808 | struct btrfs_inode_ref *ref; | ||
| 809 | struct btrfs_dir_item *di; | ||
| 810 | struct inode *inode; | ||
| 811 | char *name; | ||
| 812 | int namelen; | ||
| 813 | unsigned long ref_ptr; | ||
| 814 | unsigned long ref_end; | ||
| 815 | |||
| 816 | location.objectid = key->objectid; | ||
| 817 | location.type = BTRFS_INODE_ITEM_KEY; | ||
| 818 | location.offset = 0; | ||
| 819 | |||
| 820 | /* | ||
| 821 | * it is possible that we didn't log all the parent directories | ||
| 822 | * for a given inode. If we don't find the dir, just don't | ||
| 823 | * copy the back ref in. The link count fixup code will take | ||
| 824 | * care of the rest | ||
| 825 | */ | ||
| 826 | dir = read_one_inode(root, key->offset); | ||
| 827 | if (!dir) | ||
| 828 | return -ENOENT; | ||
| 829 | |||
| 830 | inode = read_one_inode(root, key->objectid); | ||
| 831 | BUG_ON(!dir); | ||
| 832 | |||
| 833 | ref_ptr = btrfs_item_ptr_offset(eb, slot); | ||
| 834 | ref_end = ref_ptr + btrfs_item_size_nr(eb, slot); | ||
| 835 | |||
| 836 | again: | ||
| 837 | ref = (struct btrfs_inode_ref *)ref_ptr; | ||
| 838 | |||
| 839 | namelen = btrfs_inode_ref_name_len(eb, ref); | ||
| 840 | name = kmalloc(namelen, GFP_NOFS); | ||
| 841 | BUG_ON(!name); | ||
| 842 | |||
| 843 | read_extent_buffer(eb, name, (unsigned long)(ref + 1), namelen); | ||
| 844 | |||
| 845 | /* if we already have a perfect match, we're done */ | ||
| 846 | if (inode_in_dir(root, path, dir->i_ino, inode->i_ino, | ||
| 847 | btrfs_inode_ref_index(eb, ref), | ||
| 848 | name, namelen)) { | ||
| 849 | goto out; | ||
| 850 | } | ||
| 851 | |||
| 852 | /* | ||
| 853 | * look for a conflicting back reference in the metadata. | ||
| 854 | * if we find one we have to unlink that name of the file | ||
| 855 | * before we add our new link. Later on, we overwrite any | ||
| 856 | * existing back reference, and we don't want to create | ||
| 857 | * dangling pointers in the directory. | ||
| 858 | */ | ||
| 859 | conflict_again: | ||
| 860 | ret = btrfs_search_slot(NULL, root, key, path, 0, 0); | ||
| 861 | if (ret == 0) { | ||
| 862 | char *victim_name; | ||
| 863 | int victim_name_len; | ||
| 864 | struct btrfs_inode_ref *victim_ref; | ||
| 865 | unsigned long ptr; | ||
| 866 | unsigned long ptr_end; | ||
| 867 | struct extent_buffer *leaf = path->nodes[0]; | ||
| 868 | |||
| 869 | /* are we trying to overwrite a back ref for the root directory | ||
| 870 | * if so, just jump out, we're done | ||
| 871 | */ | ||
| 872 | if (key->objectid == key->offset) | ||
| 873 | goto out_nowrite; | ||
| 874 | |||
| 875 | /* check all the names in this back reference to see | ||
| 876 | * if they are in the log. if so, we allow them to stay | ||
| 877 | * otherwise they must be unlinked as a conflict | ||
| 878 | */ | ||
| 879 | ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); | ||
| 880 | ptr_end = ptr + btrfs_item_size_nr(leaf, path->slots[0]); | ||
| 881 | while (ptr < ptr_end) { | ||
| 882 | victim_ref = (struct btrfs_inode_ref *)ptr; | ||
| 883 | victim_name_len = btrfs_inode_ref_name_len(leaf, | ||
| 884 | victim_ref); | ||
| 885 | victim_name = kmalloc(victim_name_len, GFP_NOFS); | ||
| 886 | BUG_ON(!victim_name); | ||
| 887 | |||
| 888 | read_extent_buffer(leaf, victim_name, | ||
| 889 | (unsigned long)(victim_ref + 1), | ||
| 890 | victim_name_len); | ||
| 891 | |||
| 892 | if (!backref_in_log(log, key, victim_name, | ||
| 893 | victim_name_len)) { | ||
| 894 | btrfs_inc_nlink(inode); | ||
| 895 | btrfs_release_path(root, path); | ||
| 896 | ret = btrfs_unlink_inode(trans, root, dir, | ||
| 897 | inode, victim_name, | ||
| 898 | victim_name_len); | ||
| 899 | kfree(victim_name); | ||
| 900 | btrfs_release_path(root, path); | ||
| 901 | goto conflict_again; | ||
| 902 | } | ||
| 903 | kfree(victim_name); | ||
| 904 | ptr = (unsigned long)(victim_ref + 1) + victim_name_len; | ||
| 905 | } | ||
| 906 | BUG_ON(ret); | ||
| 907 | } | ||
| 908 | btrfs_release_path(root, path); | ||
| 909 | |||
| 910 | /* look for a conflicting sequence number */ | ||
| 911 | di = btrfs_lookup_dir_index_item(trans, root, path, dir->i_ino, | ||
| 912 | btrfs_inode_ref_index(eb, ref), | ||
| 913 | name, namelen, 0); | ||
| 914 | if (di && !IS_ERR(di)) { | ||
| 915 | ret = drop_one_dir_item(trans, root, path, dir, di); | ||
| 916 | BUG_ON(ret); | ||
| 917 | } | ||
| 918 | btrfs_release_path(root, path); | ||
| 919 | |||
| 920 | |||
| 921 | /* look for a conflicting name */ | ||
| 922 | di = btrfs_lookup_dir_item(trans, root, path, dir->i_ino, | ||
| 923 | name, namelen, 0); | ||
| 924 | if (di && !IS_ERR(di)) { | ||
| 925 | ret = drop_one_dir_item(trans, root, path, dir, di); | ||
| 926 | BUG_ON(ret); | ||
| 927 | } | ||
| 928 | btrfs_release_path(root, path); | ||
| 929 | |||
| 930 | /* insert our name */ | ||
| 931 | ret = btrfs_add_link(trans, dir, inode, name, namelen, 0, | ||
| 932 | btrfs_inode_ref_index(eb, ref)); | ||
| 933 | BUG_ON(ret); | ||
| 934 | |||
| 935 | btrfs_update_inode(trans, root, inode); | ||
| 936 | |||
| 937 | out: | ||
| 938 | ref_ptr = (unsigned long)(ref + 1) + namelen; | ||
| 939 | kfree(name); | ||
| 940 | if (ref_ptr < ref_end) | ||
| 941 | goto again; | ||
| 942 | |||
| 943 | /* finally write the back reference in the inode */ | ||
| 944 | ret = overwrite_item(trans, root, path, eb, slot, key); | ||
| 945 | BUG_ON(ret); | ||
| 946 | |||
| 947 | out_nowrite: | ||
| 948 | btrfs_release_path(root, path); | ||
| 949 | iput(dir); | ||
| 950 | iput(inode); | ||
| 951 | return 0; | ||
| 952 | } | ||
| 953 | |||
| 954 | /* | ||
| 955 | * There are a few corners where the link count of the file can't | ||
| 956 | * be properly maintained during replay. So, instead of adding | ||
| 957 | * lots of complexity to the log code, we just scan the backrefs | ||
| 958 | * for any file that has been through replay. | ||
| 959 | * | ||
| 960 | * The scan will update the link count on the inode to reflect the | ||
| 961 | * number of back refs found. If it goes down to zero, the iput | ||
| 962 | * will free the inode. | ||
| 963 | */ | ||
| 964 | static noinline int fixup_inode_link_count(struct btrfs_trans_handle *trans, | ||
| 965 | struct btrfs_root *root, | ||
| 966 | struct inode *inode) | ||
| 967 | { | ||
| 968 | struct btrfs_path *path; | ||
| 969 | int ret; | ||
| 970 | struct btrfs_key key; | ||
| 971 | u64 nlink = 0; | ||
| 972 | unsigned long ptr; | ||
| 973 | unsigned long ptr_end; | ||
| 974 | int name_len; | ||
| 975 | |||
| 976 | key.objectid = inode->i_ino; | ||
| 977 | key.type = BTRFS_INODE_REF_KEY; | ||
| 978 | key.offset = (u64)-1; | ||
| 979 | |||
| 980 | path = btrfs_alloc_path(); | ||
| 981 | |||
| 982 | while (1) { | ||
| 983 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 984 | if (ret < 0) | ||
| 985 | break; | ||
| 986 | if (ret > 0) { | ||
| 987 | if (path->slots[0] == 0) | ||
| 988 | break; | ||
| 989 | path->slots[0]--; | ||
| 990 | } | ||
| 991 | btrfs_item_key_to_cpu(path->nodes[0], &key, | ||
| 992 | path->slots[0]); | ||
| 993 | if (key.objectid != inode->i_ino || | ||
| 994 | key.type != BTRFS_INODE_REF_KEY) | ||
| 995 | break; | ||
| 996 | ptr = btrfs_item_ptr_offset(path->nodes[0], path->slots[0]); | ||
| 997 | ptr_end = ptr + btrfs_item_size_nr(path->nodes[0], | ||
| 998 | path->slots[0]); | ||
| 999 | while (ptr < ptr_end) { | ||
| 1000 | struct btrfs_inode_ref *ref; | ||
| 1001 | |||
| 1002 | ref = (struct btrfs_inode_ref *)ptr; | ||
| 1003 | name_len = btrfs_inode_ref_name_len(path->nodes[0], | ||
| 1004 | ref); | ||
| 1005 | ptr = (unsigned long)(ref + 1) + name_len; | ||
| 1006 | nlink++; | ||
| 1007 | } | ||
| 1008 | |||
| 1009 | if (key.offset == 0) | ||
| 1010 | break; | ||
| 1011 | key.offset--; | ||
| 1012 | btrfs_release_path(root, path); | ||
| 1013 | } | ||
| 1014 | btrfs_free_path(path); | ||
| 1015 | if (nlink != inode->i_nlink) { | ||
| 1016 | inode->i_nlink = nlink; | ||
| 1017 | btrfs_update_inode(trans, root, inode); | ||
| 1018 | } | ||
| 1019 | BTRFS_I(inode)->index_cnt = (u64)-1; | ||
| 1020 | |||
| 1021 | return 0; | ||
| 1022 | } | ||
| 1023 | |||
| 1024 | static noinline int fixup_inode_link_counts(struct btrfs_trans_handle *trans, | ||
| 1025 | struct btrfs_root *root, | ||
| 1026 | struct btrfs_path *path) | ||
| 1027 | { | ||
| 1028 | int ret; | ||
| 1029 | struct btrfs_key key; | ||
| 1030 | struct inode *inode; | ||
| 1031 | |||
| 1032 | key.objectid = BTRFS_TREE_LOG_FIXUP_OBJECTID; | ||
| 1033 | key.type = BTRFS_ORPHAN_ITEM_KEY; | ||
| 1034 | key.offset = (u64)-1; | ||
| 1035 | while (1) { | ||
| 1036 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 1037 | if (ret < 0) | ||
| 1038 | break; | ||
| 1039 | |||
| 1040 | if (ret == 1) { | ||
| 1041 | if (path->slots[0] == 0) | ||
| 1042 | break; | ||
| 1043 | path->slots[0]--; | ||
| 1044 | } | ||
| 1045 | |||
| 1046 | btrfs_item_key_to_cpu(path->nodes[0], &key, path->slots[0]); | ||
| 1047 | if (key.objectid != BTRFS_TREE_LOG_FIXUP_OBJECTID || | ||
| 1048 | key.type != BTRFS_ORPHAN_ITEM_KEY) | ||
| 1049 | break; | ||
| 1050 | |||
| 1051 | ret = btrfs_del_item(trans, root, path); | ||
| 1052 | BUG_ON(ret); | ||
| 1053 | |||
| 1054 | btrfs_release_path(root, path); | ||
| 1055 | inode = read_one_inode(root, key.offset); | ||
| 1056 | BUG_ON(!inode); | ||
| 1057 | |||
| 1058 | ret = fixup_inode_link_count(trans, root, inode); | ||
| 1059 | BUG_ON(ret); | ||
| 1060 | |||
| 1061 | iput(inode); | ||
| 1062 | |||
| 1063 | if (key.offset == 0) | ||
| 1064 | break; | ||
| 1065 | key.offset--; | ||
| 1066 | } | ||
| 1067 | btrfs_release_path(root, path); | ||
| 1068 | return 0; | ||
| 1069 | } | ||
| 1070 | |||
| 1071 | |||
| 1072 | /* | ||
| 1073 | * record a given inode in the fixup dir so we can check its link | ||
| 1074 | * count when replay is done. The link count is incremented here | ||
| 1075 | * so the inode won't go away until we check it | ||
| 1076 | */ | ||
| 1077 | static noinline int link_to_fixup_dir(struct btrfs_trans_handle *trans, | ||
| 1078 | struct btrfs_root *root, | ||
| 1079 | struct btrfs_path *path, | ||
| 1080 | u64 objectid) | ||
| 1081 | { | ||
| 1082 | struct btrfs_key key; | ||
| 1083 | int ret = 0; | ||
| 1084 | struct inode *inode; | ||
| 1085 | |||
| 1086 | inode = read_one_inode(root, objectid); | ||
| 1087 | BUG_ON(!inode); | ||
| 1088 | |||
| 1089 | key.objectid = BTRFS_TREE_LOG_FIXUP_OBJECTID; | ||
| 1090 | btrfs_set_key_type(&key, BTRFS_ORPHAN_ITEM_KEY); | ||
| 1091 | key.offset = objectid; | ||
| 1092 | |||
| 1093 | ret = btrfs_insert_empty_item(trans, root, path, &key, 0); | ||
| 1094 | |||
| 1095 | btrfs_release_path(root, path); | ||
| 1096 | if (ret == 0) { | ||
| 1097 | btrfs_inc_nlink(inode); | ||
| 1098 | btrfs_update_inode(trans, root, inode); | ||
| 1099 | } else if (ret == -EEXIST) { | ||
| 1100 | ret = 0; | ||
| 1101 | } else { | ||
| 1102 | BUG(); | ||
| 1103 | } | ||
| 1104 | iput(inode); | ||
| 1105 | |||
| 1106 | return ret; | ||
| 1107 | } | ||
| 1108 | |||
| 1109 | /* | ||
| 1110 | * when replaying the log for a directory, we only insert names | ||
| 1111 | * for inodes that actually exist. This means an fsync on a directory | ||
| 1112 | * does not implicitly fsync all the new files in it | ||
| 1113 | */ | ||
| 1114 | static noinline int insert_one_name(struct btrfs_trans_handle *trans, | ||
| 1115 | struct btrfs_root *root, | ||
| 1116 | struct btrfs_path *path, | ||
| 1117 | u64 dirid, u64 index, | ||
| 1118 | char *name, int name_len, u8 type, | ||
| 1119 | struct btrfs_key *location) | ||
| 1120 | { | ||
| 1121 | struct inode *inode; | ||
| 1122 | struct inode *dir; | ||
| 1123 | int ret; | ||
| 1124 | |||
| 1125 | inode = read_one_inode(root, location->objectid); | ||
| 1126 | if (!inode) | ||
| 1127 | return -ENOENT; | ||
| 1128 | |||
| 1129 | dir = read_one_inode(root, dirid); | ||
| 1130 | if (!dir) { | ||
| 1131 | iput(inode); | ||
| 1132 | return -EIO; | ||
| 1133 | } | ||
| 1134 | ret = btrfs_add_link(trans, dir, inode, name, name_len, 1, index); | ||
| 1135 | |||
| 1136 | /* FIXME, put inode into FIXUP list */ | ||
| 1137 | |||
| 1138 | iput(inode); | ||
| 1139 | iput(dir); | ||
| 1140 | return ret; | ||
| 1141 | } | ||
| 1142 | |||
| 1143 | /* | ||
| 1144 | * take a single entry in a log directory item and replay it into | ||
| 1145 | * the subvolume. | ||
| 1146 | * | ||
| 1147 | * if a conflicting item exists in the subdirectory already, | ||
| 1148 | * the inode it points to is unlinked and put into the link count | ||
| 1149 | * fix up tree. | ||
| 1150 | * | ||
| 1151 | * If a name from the log points to a file or directory that does | ||
| 1152 | * not exist in the FS, it is skipped. fsyncs on directories | ||
| 1153 | * do not force down inodes inside that directory, just changes to the | ||
| 1154 | * names or unlinks in a directory. | ||
| 1155 | */ | ||
| 1156 | static noinline int replay_one_name(struct btrfs_trans_handle *trans, | ||
| 1157 | struct btrfs_root *root, | ||
| 1158 | struct btrfs_path *path, | ||
| 1159 | struct extent_buffer *eb, | ||
| 1160 | struct btrfs_dir_item *di, | ||
| 1161 | struct btrfs_key *key) | ||
| 1162 | { | ||
| 1163 | char *name; | ||
| 1164 | int name_len; | ||
| 1165 | struct btrfs_dir_item *dst_di; | ||
| 1166 | struct btrfs_key found_key; | ||
| 1167 | struct btrfs_key log_key; | ||
| 1168 | struct inode *dir; | ||
| 1169 | u8 log_type; | ||
| 1170 | int exists; | ||
| 1171 | int ret; | ||
| 1172 | |||
| 1173 | dir = read_one_inode(root, key->objectid); | ||
| 1174 | BUG_ON(!dir); | ||
| 1175 | |||
| 1176 | name_len = btrfs_dir_name_len(eb, di); | ||
| 1177 | name = kmalloc(name_len, GFP_NOFS); | ||
| 1178 | log_type = btrfs_dir_type(eb, di); | ||
| 1179 | read_extent_buffer(eb, name, (unsigned long)(di + 1), | ||
| 1180 | name_len); | ||
| 1181 | |||
| 1182 | btrfs_dir_item_key_to_cpu(eb, di, &log_key); | ||
| 1183 | exists = btrfs_lookup_inode(trans, root, path, &log_key, 0); | ||
| 1184 | if (exists == 0) | ||
| 1185 | exists = 1; | ||
| 1186 | else | ||
| 1187 | exists = 0; | ||
| 1188 | btrfs_release_path(root, path); | ||
| 1189 | |||
| 1190 | if (key->type == BTRFS_DIR_ITEM_KEY) { | ||
| 1191 | dst_di = btrfs_lookup_dir_item(trans, root, path, key->objectid, | ||
| 1192 | name, name_len, 1); | ||
| 1193 | } else if (key->type == BTRFS_DIR_INDEX_KEY) { | ||
| 1194 | dst_di = btrfs_lookup_dir_index_item(trans, root, path, | ||
| 1195 | key->objectid, | ||
| 1196 | key->offset, name, | ||
| 1197 | name_len, 1); | ||
| 1198 | } else { | ||
| 1199 | BUG(); | ||
| 1200 | } | ||
| 1201 | if (!dst_di || IS_ERR(dst_di)) { | ||
| 1202 | /* we need a sequence number to insert, so we only | ||
| 1203 | * do inserts for the BTRFS_DIR_INDEX_KEY types | ||
| 1204 | */ | ||
| 1205 | if (key->type != BTRFS_DIR_INDEX_KEY) | ||
| 1206 | goto out; | ||
| 1207 | goto insert; | ||
| 1208 | } | ||
| 1209 | |||
| 1210 | btrfs_dir_item_key_to_cpu(path->nodes[0], dst_di, &found_key); | ||
| 1211 | /* the existing item matches the logged item */ | ||
| 1212 | if (found_key.objectid == log_key.objectid && | ||
| 1213 | found_key.type == log_key.type && | ||
| 1214 | found_key.offset == log_key.offset && | ||
| 1215 | btrfs_dir_type(path->nodes[0], dst_di) == log_type) { | ||
| 1216 | goto out; | ||
| 1217 | } | ||
| 1218 | |||
| 1219 | /* | ||
| 1220 | * don't drop the conflicting directory entry if the inode | ||
| 1221 | * for the new entry doesn't exist | ||
| 1222 | */ | ||
| 1223 | if (!exists) | ||
| 1224 | goto out; | ||
| 1225 | |||
| 1226 | ret = drop_one_dir_item(trans, root, path, dir, dst_di); | ||
| 1227 | BUG_ON(ret); | ||
| 1228 | |||
| 1229 | if (key->type == BTRFS_DIR_INDEX_KEY) | ||
| 1230 | goto insert; | ||
| 1231 | out: | ||
| 1232 | btrfs_release_path(root, path); | ||
| 1233 | kfree(name); | ||
| 1234 | iput(dir); | ||
| 1235 | return 0; | ||
| 1236 | |||
| 1237 | insert: | ||
| 1238 | btrfs_release_path(root, path); | ||
| 1239 | ret = insert_one_name(trans, root, path, key->objectid, key->offset, | ||
| 1240 | name, name_len, log_type, &log_key); | ||
| 1241 | |||
| 1242 | if (ret && ret != -ENOENT) | ||
| 1243 | BUG(); | ||
| 1244 | goto out; | ||
| 1245 | } | ||
| 1246 | |||
| 1247 | /* | ||
| 1248 | * find all the names in a directory item and reconcile them into | ||
| 1249 | * the subvolume. Only BTRFS_DIR_ITEM_KEY types will have more than | ||
| 1250 | * one name in a directory item, but the same code gets used for | ||
| 1251 | * both directory index types | ||
| 1252 | */ | ||
| 1253 | static noinline int replay_one_dir_item(struct btrfs_trans_handle *trans, | ||
| 1254 | struct btrfs_root *root, | ||
| 1255 | struct btrfs_path *path, | ||
| 1256 | struct extent_buffer *eb, int slot, | ||
| 1257 | struct btrfs_key *key) | ||
| 1258 | { | ||
| 1259 | int ret; | ||
| 1260 | u32 item_size = btrfs_item_size_nr(eb, slot); | ||
| 1261 | struct btrfs_dir_item *di; | ||
| 1262 | int name_len; | ||
| 1263 | unsigned long ptr; | ||
| 1264 | unsigned long ptr_end; | ||
| 1265 | |||
| 1266 | ptr = btrfs_item_ptr_offset(eb, slot); | ||
| 1267 | ptr_end = ptr + item_size; | ||
| 1268 | while (ptr < ptr_end) { | ||
| 1269 | di = (struct btrfs_dir_item *)ptr; | ||
| 1270 | name_len = btrfs_dir_name_len(eb, di); | ||
| 1271 | ret = replay_one_name(trans, root, path, eb, di, key); | ||
| 1272 | BUG_ON(ret); | ||
| 1273 | ptr = (unsigned long)(di + 1); | ||
| 1274 | ptr += name_len; | ||
| 1275 | } | ||
| 1276 | return 0; | ||
| 1277 | } | ||
| 1278 | |||
| 1279 | /* | ||
| 1280 | * directory replay has two parts. There are the standard directory | ||
| 1281 | * items in the log copied from the subvolume, and range items | ||
| 1282 | * created in the log while the subvolume was logged. | ||
| 1283 | * | ||
| 1284 | * The range items tell us which parts of the key space the log | ||
| 1285 | * is authoritative for. During replay, if a key in the subvolume | ||
| 1286 | * directory is in a logged range item, but not actually in the log | ||
| 1287 | * that means it was deleted from the directory before the fsync | ||
| 1288 | * and should be removed. | ||
| 1289 | */ | ||
| 1290 | static noinline int find_dir_range(struct btrfs_root *root, | ||
| 1291 | struct btrfs_path *path, | ||
| 1292 | u64 dirid, int key_type, | ||
| 1293 | u64 *start_ret, u64 *end_ret) | ||
| 1294 | { | ||
| 1295 | struct btrfs_key key; | ||
| 1296 | u64 found_end; | ||
| 1297 | struct btrfs_dir_log_item *item; | ||
| 1298 | int ret; | ||
| 1299 | int nritems; | ||
| 1300 | |||
| 1301 | if (*start_ret == (u64)-1) | ||
| 1302 | return 1; | ||
| 1303 | |||
| 1304 | key.objectid = dirid; | ||
| 1305 | key.type = key_type; | ||
| 1306 | key.offset = *start_ret; | ||
| 1307 | |||
| 1308 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 1309 | if (ret < 0) | ||
| 1310 | goto out; | ||
| 1311 | if (ret > 0) { | ||
| 1312 | if (path->slots[0] == 0) | ||
| 1313 | goto out; | ||
| 1314 | path->slots[0]--; | ||
| 1315 | } | ||
| 1316 | if (ret != 0) | ||
| 1317 | btrfs_item_key_to_cpu(path->nodes[0], &key, path->slots[0]); | ||
| 1318 | |||
| 1319 | if (key.type != key_type || key.objectid != dirid) { | ||
| 1320 | ret = 1; | ||
| 1321 | goto next; | ||
| 1322 | } | ||
| 1323 | item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 1324 | struct btrfs_dir_log_item); | ||
| 1325 | found_end = btrfs_dir_log_end(path->nodes[0], item); | ||
| 1326 | |||
| 1327 | if (*start_ret >= key.offset && *start_ret <= found_end) { | ||
| 1328 | ret = 0; | ||
| 1329 | *start_ret = key.offset; | ||
| 1330 | *end_ret = found_end; | ||
| 1331 | goto out; | ||
| 1332 | } | ||
| 1333 | ret = 1; | ||
| 1334 | next: | ||
| 1335 | /* check the next slot in the tree to see if it is a valid item */ | ||
| 1336 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 1337 | if (path->slots[0] >= nritems) { | ||
| 1338 | ret = btrfs_next_leaf(root, path); | ||
| 1339 | if (ret) | ||
| 1340 | goto out; | ||
| 1341 | } else { | ||
| 1342 | path->slots[0]++; | ||
| 1343 | } | ||
| 1344 | |||
| 1345 | btrfs_item_key_to_cpu(path->nodes[0], &key, path->slots[0]); | ||
| 1346 | |||
| 1347 | if (key.type != key_type || key.objectid != dirid) { | ||
| 1348 | ret = 1; | ||
| 1349 | goto out; | ||
| 1350 | } | ||
| 1351 | item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 1352 | struct btrfs_dir_log_item); | ||
| 1353 | found_end = btrfs_dir_log_end(path->nodes[0], item); | ||
| 1354 | *start_ret = key.offset; | ||
| 1355 | *end_ret = found_end; | ||
| 1356 | ret = 0; | ||
| 1357 | out: | ||
| 1358 | btrfs_release_path(root, path); | ||
| 1359 | return ret; | ||
| 1360 | } | ||
| 1361 | |||
| 1362 | /* | ||
| 1363 | * this looks for a given directory item in the log. If the directory | ||
| 1364 | * item is not in the log, the item is removed and the inode it points | ||
| 1365 | * to is unlinked | ||
| 1366 | */ | ||
| 1367 | static noinline int check_item_in_log(struct btrfs_trans_handle *trans, | ||
| 1368 | struct btrfs_root *root, | ||
| 1369 | struct btrfs_root *log, | ||
| 1370 | struct btrfs_path *path, | ||
| 1371 | struct btrfs_path *log_path, | ||
| 1372 | struct inode *dir, | ||
| 1373 | struct btrfs_key *dir_key) | ||
| 1374 | { | ||
| 1375 | int ret; | ||
| 1376 | struct extent_buffer *eb; | ||
| 1377 | int slot; | ||
| 1378 | u32 item_size; | ||
| 1379 | struct btrfs_dir_item *di; | ||
| 1380 | struct btrfs_dir_item *log_di; | ||
| 1381 | int name_len; | ||
| 1382 | unsigned long ptr; | ||
| 1383 | unsigned long ptr_end; | ||
| 1384 | char *name; | ||
| 1385 | struct inode *inode; | ||
| 1386 | struct btrfs_key location; | ||
| 1387 | |||
| 1388 | again: | ||
| 1389 | eb = path->nodes[0]; | ||
| 1390 | slot = path->slots[0]; | ||
| 1391 | item_size = btrfs_item_size_nr(eb, slot); | ||
| 1392 | ptr = btrfs_item_ptr_offset(eb, slot); | ||
| 1393 | ptr_end = ptr + item_size; | ||
| 1394 | while (ptr < ptr_end) { | ||
| 1395 | di = (struct btrfs_dir_item *)ptr; | ||
| 1396 | name_len = btrfs_dir_name_len(eb, di); | ||
| 1397 | name = kmalloc(name_len, GFP_NOFS); | ||
| 1398 | if (!name) { | ||
| 1399 | ret = -ENOMEM; | ||
| 1400 | goto out; | ||
| 1401 | } | ||
| 1402 | read_extent_buffer(eb, name, (unsigned long)(di + 1), | ||
| 1403 | name_len); | ||
| 1404 | log_di = NULL; | ||
| 1405 | if (dir_key->type == BTRFS_DIR_ITEM_KEY) { | ||
| 1406 | log_di = btrfs_lookup_dir_item(trans, log, log_path, | ||
| 1407 | dir_key->objectid, | ||
| 1408 | name, name_len, 0); | ||
| 1409 | } else if (dir_key->type == BTRFS_DIR_INDEX_KEY) { | ||
| 1410 | log_di = btrfs_lookup_dir_index_item(trans, log, | ||
| 1411 | log_path, | ||
| 1412 | dir_key->objectid, | ||
| 1413 | dir_key->offset, | ||
| 1414 | name, name_len, 0); | ||
| 1415 | } | ||
| 1416 | if (!log_di || IS_ERR(log_di)) { | ||
| 1417 | btrfs_dir_item_key_to_cpu(eb, di, &location); | ||
| 1418 | btrfs_release_path(root, path); | ||
| 1419 | btrfs_release_path(log, log_path); | ||
| 1420 | inode = read_one_inode(root, location.objectid); | ||
| 1421 | BUG_ON(!inode); | ||
| 1422 | |||
| 1423 | ret = link_to_fixup_dir(trans, root, | ||
| 1424 | path, location.objectid); | ||
| 1425 | BUG_ON(ret); | ||
| 1426 | btrfs_inc_nlink(inode); | ||
| 1427 | ret = btrfs_unlink_inode(trans, root, dir, inode, | ||
| 1428 | name, name_len); | ||
| 1429 | BUG_ON(ret); | ||
| 1430 | kfree(name); | ||
| 1431 | iput(inode); | ||
| 1432 | |||
| 1433 | /* there might still be more names under this key | ||
| 1434 | * check and repeat if required | ||
| 1435 | */ | ||
| 1436 | ret = btrfs_search_slot(NULL, root, dir_key, path, | ||
| 1437 | 0, 0); | ||
| 1438 | if (ret == 0) | ||
| 1439 | goto again; | ||
| 1440 | ret = 0; | ||
| 1441 | goto out; | ||
| 1442 | } | ||
| 1443 | btrfs_release_path(log, log_path); | ||
| 1444 | kfree(name); | ||
| 1445 | |||
| 1446 | ptr = (unsigned long)(di + 1); | ||
| 1447 | ptr += name_len; | ||
| 1448 | } | ||
| 1449 | ret = 0; | ||
| 1450 | out: | ||
| 1451 | btrfs_release_path(root, path); | ||
| 1452 | btrfs_release_path(log, log_path); | ||
| 1453 | return ret; | ||
| 1454 | } | ||
| 1455 | |||
| 1456 | /* | ||
| 1457 | * deletion replay happens before we copy any new directory items | ||
| 1458 | * out of the log or out of backreferences from inodes. It | ||
| 1459 | * scans the log to find ranges of keys that log is authoritative for, | ||
| 1460 | * and then scans the directory to find items in those ranges that are | ||
| 1461 | * not present in the log. | ||
| 1462 | * | ||
| 1463 | * Anything we don't find in the log is unlinked and removed from the | ||
| 1464 | * directory. | ||
| 1465 | */ | ||
| 1466 | static noinline int replay_dir_deletes(struct btrfs_trans_handle *trans, | ||
| 1467 | struct btrfs_root *root, | ||
| 1468 | struct btrfs_root *log, | ||
| 1469 | struct btrfs_path *path, | ||
| 1470 | u64 dirid) | ||
| 1471 | { | ||
| 1472 | u64 range_start; | ||
| 1473 | u64 range_end; | ||
| 1474 | int key_type = BTRFS_DIR_LOG_ITEM_KEY; | ||
| 1475 | int ret = 0; | ||
| 1476 | struct btrfs_key dir_key; | ||
| 1477 | struct btrfs_key found_key; | ||
| 1478 | struct btrfs_path *log_path; | ||
| 1479 | struct inode *dir; | ||
| 1480 | |||
| 1481 | dir_key.objectid = dirid; | ||
| 1482 | dir_key.type = BTRFS_DIR_ITEM_KEY; | ||
| 1483 | log_path = btrfs_alloc_path(); | ||
| 1484 | if (!log_path) | ||
| 1485 | return -ENOMEM; | ||
| 1486 | |||
| 1487 | dir = read_one_inode(root, dirid); | ||
| 1488 | /* it isn't an error if the inode isn't there, that can happen | ||
| 1489 | * because we replay the deletes before we copy in the inode item | ||
| 1490 | * from the log | ||
| 1491 | */ | ||
| 1492 | if (!dir) { | ||
| 1493 | btrfs_free_path(log_path); | ||
| 1494 | return 0; | ||
| 1495 | } | ||
| 1496 | again: | ||
| 1497 | range_start = 0; | ||
| 1498 | range_end = 0; | ||
| 1499 | while (1) { | ||
| 1500 | ret = find_dir_range(log, path, dirid, key_type, | ||
| 1501 | &range_start, &range_end); | ||
| 1502 | if (ret != 0) | ||
| 1503 | break; | ||
| 1504 | |||
| 1505 | dir_key.offset = range_start; | ||
| 1506 | while (1) { | ||
| 1507 | int nritems; | ||
| 1508 | ret = btrfs_search_slot(NULL, root, &dir_key, path, | ||
| 1509 | 0, 0); | ||
| 1510 | if (ret < 0) | ||
| 1511 | goto out; | ||
| 1512 | |||
| 1513 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 1514 | if (path->slots[0] >= nritems) { | ||
| 1515 | ret = btrfs_next_leaf(root, path); | ||
| 1516 | if (ret) | ||
| 1517 | break; | ||
| 1518 | } | ||
| 1519 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 1520 | path->slots[0]); | ||
| 1521 | if (found_key.objectid != dirid || | ||
| 1522 | found_key.type != dir_key.type) | ||
| 1523 | goto next_type; | ||
| 1524 | |||
| 1525 | if (found_key.offset > range_end) | ||
| 1526 | break; | ||
| 1527 | |||
| 1528 | ret = check_item_in_log(trans, root, log, path, | ||
| 1529 | log_path, dir, &found_key); | ||
| 1530 | BUG_ON(ret); | ||
| 1531 | if (found_key.offset == (u64)-1) | ||
| 1532 | break; | ||
| 1533 | dir_key.offset = found_key.offset + 1; | ||
| 1534 | } | ||
| 1535 | btrfs_release_path(root, path); | ||
| 1536 | if (range_end == (u64)-1) | ||
| 1537 | break; | ||
| 1538 | range_start = range_end + 1; | ||
| 1539 | } | ||
| 1540 | |||
| 1541 | next_type: | ||
| 1542 | ret = 0; | ||
| 1543 | if (key_type == BTRFS_DIR_LOG_ITEM_KEY) { | ||
| 1544 | key_type = BTRFS_DIR_LOG_INDEX_KEY; | ||
| 1545 | dir_key.type = BTRFS_DIR_INDEX_KEY; | ||
| 1546 | btrfs_release_path(root, path); | ||
| 1547 | goto again; | ||
| 1548 | } | ||
| 1549 | out: | ||
| 1550 | btrfs_release_path(root, path); | ||
| 1551 | btrfs_free_path(log_path); | ||
| 1552 | iput(dir); | ||
| 1553 | return ret; | ||
| 1554 | } | ||
| 1555 | |||
| 1556 | /* | ||
| 1557 | * the process_func used to replay items from the log tree. This | ||
| 1558 | * gets called in two different stages. The first stage just looks | ||
| 1559 | * for inodes and makes sure they are all copied into the subvolume. | ||
| 1560 | * | ||
| 1561 | * The second stage copies all the other item types from the log into | ||
| 1562 | * the subvolume. The two stage approach is slower, but gets rid of | ||
| 1563 | * lots of complexity around inodes referencing other inodes that exist | ||
| 1564 | * only in the log (references come from either directory items or inode | ||
| 1565 | * back refs). | ||
| 1566 | */ | ||
| 1567 | static int replay_one_buffer(struct btrfs_root *log, struct extent_buffer *eb, | ||
| 1568 | struct walk_control *wc, u64 gen) | ||
| 1569 | { | ||
| 1570 | int nritems; | ||
| 1571 | struct btrfs_path *path; | ||
| 1572 | struct btrfs_root *root = wc->replay_dest; | ||
| 1573 | struct btrfs_key key; | ||
| 1574 | u32 item_size; | ||
| 1575 | int level; | ||
| 1576 | int i; | ||
| 1577 | int ret; | ||
| 1578 | |||
| 1579 | btrfs_read_buffer(eb, gen); | ||
| 1580 | |||
| 1581 | level = btrfs_header_level(eb); | ||
| 1582 | |||
| 1583 | if (level != 0) | ||
| 1584 | return 0; | ||
| 1585 | |||
| 1586 | path = btrfs_alloc_path(); | ||
| 1587 | BUG_ON(!path); | ||
| 1588 | |||
| 1589 | nritems = btrfs_header_nritems(eb); | ||
| 1590 | for (i = 0; i < nritems; i++) { | ||
| 1591 | btrfs_item_key_to_cpu(eb, &key, i); | ||
| 1592 | item_size = btrfs_item_size_nr(eb, i); | ||
| 1593 | |||
| 1594 | /* inode keys are done during the first stage */ | ||
| 1595 | if (key.type == BTRFS_INODE_ITEM_KEY && | ||
| 1596 | wc->stage == LOG_WALK_REPLAY_INODES) { | ||
| 1597 | struct inode *inode; | ||
| 1598 | struct btrfs_inode_item *inode_item; | ||
| 1599 | u32 mode; | ||
| 1600 | |||
| 1601 | inode_item = btrfs_item_ptr(eb, i, | ||
| 1602 | struct btrfs_inode_item); | ||
| 1603 | mode = btrfs_inode_mode(eb, inode_item); | ||
| 1604 | if (S_ISDIR(mode)) { | ||
| 1605 | ret = replay_dir_deletes(wc->trans, | ||
| 1606 | root, log, path, key.objectid); | ||
| 1607 | BUG_ON(ret); | ||
| 1608 | } | ||
| 1609 | ret = overwrite_item(wc->trans, root, path, | ||
| 1610 | eb, i, &key); | ||
| 1611 | BUG_ON(ret); | ||
| 1612 | |||
| 1613 | /* for regular files, truncate away | ||
| 1614 | * extents past the new EOF | ||
| 1615 | */ | ||
| 1616 | if (S_ISREG(mode)) { | ||
| 1617 | inode = read_one_inode(root, | ||
| 1618 | key.objectid); | ||
| 1619 | BUG_ON(!inode); | ||
| 1620 | |||
| 1621 | ret = btrfs_truncate_inode_items(wc->trans, | ||
| 1622 | root, inode, inode->i_size, | ||
| 1623 | BTRFS_EXTENT_DATA_KEY); | ||
| 1624 | BUG_ON(ret); | ||
| 1625 | iput(inode); | ||
| 1626 | } | ||
| 1627 | ret = link_to_fixup_dir(wc->trans, root, | ||
| 1628 | path, key.objectid); | ||
| 1629 | BUG_ON(ret); | ||
| 1630 | } | ||
| 1631 | if (wc->stage < LOG_WALK_REPLAY_ALL) | ||
| 1632 | continue; | ||
| 1633 | |||
| 1634 | /* these keys are simply copied */ | ||
| 1635 | if (key.type == BTRFS_XATTR_ITEM_KEY) { | ||
| 1636 | ret = overwrite_item(wc->trans, root, path, | ||
| 1637 | eb, i, &key); | ||
| 1638 | BUG_ON(ret); | ||
| 1639 | } else if (key.type == BTRFS_INODE_REF_KEY) { | ||
| 1640 | ret = add_inode_ref(wc->trans, root, log, path, | ||
| 1641 | eb, i, &key); | ||
| 1642 | BUG_ON(ret && ret != -ENOENT); | ||
| 1643 | } else if (key.type == BTRFS_EXTENT_DATA_KEY) { | ||
| 1644 | ret = replay_one_extent(wc->trans, root, path, | ||
| 1645 | eb, i, &key); | ||
| 1646 | BUG_ON(ret); | ||
| 1647 | } else if (key.type == BTRFS_DIR_ITEM_KEY || | ||
| 1648 | key.type == BTRFS_DIR_INDEX_KEY) { | ||
| 1649 | ret = replay_one_dir_item(wc->trans, root, path, | ||
| 1650 | eb, i, &key); | ||
| 1651 | BUG_ON(ret); | ||
| 1652 | } | ||
| 1653 | } | ||
| 1654 | btrfs_free_path(path); | ||
| 1655 | return 0; | ||
| 1656 | } | ||
| 1657 | |||
| 1658 | static noinline int walk_down_log_tree(struct btrfs_trans_handle *trans, | ||
| 1659 | struct btrfs_root *root, | ||
| 1660 | struct btrfs_path *path, int *level, | ||
| 1661 | struct walk_control *wc) | ||
| 1662 | { | ||
| 1663 | u64 root_owner; | ||
| 1664 | u64 root_gen; | ||
| 1665 | u64 bytenr; | ||
| 1666 | u64 ptr_gen; | ||
| 1667 | struct extent_buffer *next; | ||
| 1668 | struct extent_buffer *cur; | ||
| 1669 | struct extent_buffer *parent; | ||
| 1670 | u32 blocksize; | ||
| 1671 | int ret = 0; | ||
| 1672 | |||
| 1673 | WARN_ON(*level < 0); | ||
| 1674 | WARN_ON(*level >= BTRFS_MAX_LEVEL); | ||
| 1675 | |||
| 1676 | while (*level > 0) { | ||
| 1677 | WARN_ON(*level < 0); | ||
| 1678 | WARN_ON(*level >= BTRFS_MAX_LEVEL); | ||
| 1679 | cur = path->nodes[*level]; | ||
| 1680 | |||
| 1681 | if (btrfs_header_level(cur) != *level) | ||
| 1682 | WARN_ON(1); | ||
| 1683 | |||
| 1684 | if (path->slots[*level] >= | ||
| 1685 | btrfs_header_nritems(cur)) | ||
| 1686 | break; | ||
| 1687 | |||
| 1688 | bytenr = btrfs_node_blockptr(cur, path->slots[*level]); | ||
| 1689 | ptr_gen = btrfs_node_ptr_generation(cur, path->slots[*level]); | ||
| 1690 | blocksize = btrfs_level_size(root, *level - 1); | ||
| 1691 | |||
| 1692 | parent = path->nodes[*level]; | ||
| 1693 | root_owner = btrfs_header_owner(parent); | ||
| 1694 | root_gen = btrfs_header_generation(parent); | ||
| 1695 | |||
| 1696 | next = btrfs_find_create_tree_block(root, bytenr, blocksize); | ||
| 1697 | |||
| 1698 | wc->process_func(root, next, wc, ptr_gen); | ||
| 1699 | |||
| 1700 | if (*level == 1) { | ||
| 1701 | path->slots[*level]++; | ||
| 1702 | if (wc->free) { | ||
| 1703 | btrfs_read_buffer(next, ptr_gen); | ||
| 1704 | |||
| 1705 | btrfs_tree_lock(next); | ||
| 1706 | clean_tree_block(trans, root, next); | ||
| 1707 | btrfs_wait_tree_block_writeback(next); | ||
| 1708 | btrfs_tree_unlock(next); | ||
| 1709 | |||
| 1710 | ret = btrfs_drop_leaf_ref(trans, root, next); | ||
| 1711 | BUG_ON(ret); | ||
| 1712 | |||
| 1713 | WARN_ON(root_owner != | ||
| 1714 | BTRFS_TREE_LOG_OBJECTID); | ||
| 1715 | ret = btrfs_free_reserved_extent(root, | ||
| 1716 | bytenr, blocksize); | ||
| 1717 | BUG_ON(ret); | ||
| 1718 | } | ||
| 1719 | free_extent_buffer(next); | ||
| 1720 | continue; | ||
| 1721 | } | ||
| 1722 | btrfs_read_buffer(next, ptr_gen); | ||
| 1723 | |||
| 1724 | WARN_ON(*level <= 0); | ||
| 1725 | if (path->nodes[*level-1]) | ||
| 1726 | free_extent_buffer(path->nodes[*level-1]); | ||
| 1727 | path->nodes[*level-1] = next; | ||
| 1728 | *level = btrfs_header_level(next); | ||
| 1729 | path->slots[*level] = 0; | ||
| 1730 | cond_resched(); | ||
| 1731 | } | ||
| 1732 | WARN_ON(*level < 0); | ||
| 1733 | WARN_ON(*level >= BTRFS_MAX_LEVEL); | ||
| 1734 | |||
| 1735 | if (path->nodes[*level] == root->node) | ||
| 1736 | parent = path->nodes[*level]; | ||
| 1737 | else | ||
| 1738 | parent = path->nodes[*level + 1]; | ||
| 1739 | |||
| 1740 | bytenr = path->nodes[*level]->start; | ||
| 1741 | |||
| 1742 | blocksize = btrfs_level_size(root, *level); | ||
| 1743 | root_owner = btrfs_header_owner(parent); | ||
| 1744 | root_gen = btrfs_header_generation(parent); | ||
| 1745 | |||
| 1746 | wc->process_func(root, path->nodes[*level], wc, | ||
| 1747 | btrfs_header_generation(path->nodes[*level])); | ||
| 1748 | |||
| 1749 | if (wc->free) { | ||
| 1750 | next = path->nodes[*level]; | ||
| 1751 | btrfs_tree_lock(next); | ||
| 1752 | clean_tree_block(trans, root, next); | ||
| 1753 | btrfs_wait_tree_block_writeback(next); | ||
| 1754 | btrfs_tree_unlock(next); | ||
| 1755 | |||
| 1756 | if (*level == 0) { | ||
| 1757 | ret = btrfs_drop_leaf_ref(trans, root, next); | ||
| 1758 | BUG_ON(ret); | ||
| 1759 | } | ||
| 1760 | WARN_ON(root_owner != BTRFS_TREE_LOG_OBJECTID); | ||
| 1761 | ret = btrfs_free_reserved_extent(root, bytenr, blocksize); | ||
| 1762 | BUG_ON(ret); | ||
| 1763 | } | ||
| 1764 | free_extent_buffer(path->nodes[*level]); | ||
| 1765 | path->nodes[*level] = NULL; | ||
| 1766 | *level += 1; | ||
| 1767 | |||
| 1768 | cond_resched(); | ||
| 1769 | return 0; | ||
| 1770 | } | ||
| 1771 | |||
| 1772 | static noinline int walk_up_log_tree(struct btrfs_trans_handle *trans, | ||
| 1773 | struct btrfs_root *root, | ||
| 1774 | struct btrfs_path *path, int *level, | ||
| 1775 | struct walk_control *wc) | ||
| 1776 | { | ||
| 1777 | u64 root_owner; | ||
| 1778 | u64 root_gen; | ||
| 1779 | int i; | ||
| 1780 | int slot; | ||
| 1781 | int ret; | ||
| 1782 | |||
| 1783 | for (i = *level; i < BTRFS_MAX_LEVEL - 1 && path->nodes[i]; i++) { | ||
| 1784 | slot = path->slots[i]; | ||
| 1785 | if (slot < btrfs_header_nritems(path->nodes[i]) - 1) { | ||
| 1786 | struct extent_buffer *node; | ||
| 1787 | node = path->nodes[i]; | ||
| 1788 | path->slots[i]++; | ||
| 1789 | *level = i; | ||
| 1790 | WARN_ON(*level == 0); | ||
| 1791 | return 0; | ||
| 1792 | } else { | ||
| 1793 | struct extent_buffer *parent; | ||
| 1794 | if (path->nodes[*level] == root->node) | ||
| 1795 | parent = path->nodes[*level]; | ||
| 1796 | else | ||
| 1797 | parent = path->nodes[*level + 1]; | ||
| 1798 | |||
| 1799 | root_owner = btrfs_header_owner(parent); | ||
| 1800 | root_gen = btrfs_header_generation(parent); | ||
| 1801 | wc->process_func(root, path->nodes[*level], wc, | ||
| 1802 | btrfs_header_generation(path->nodes[*level])); | ||
| 1803 | if (wc->free) { | ||
| 1804 | struct extent_buffer *next; | ||
| 1805 | |||
| 1806 | next = path->nodes[*level]; | ||
| 1807 | |||
| 1808 | btrfs_tree_lock(next); | ||
| 1809 | clean_tree_block(trans, root, next); | ||
| 1810 | btrfs_wait_tree_block_writeback(next); | ||
| 1811 | btrfs_tree_unlock(next); | ||
| 1812 | |||
| 1813 | if (*level == 0) { | ||
| 1814 | ret = btrfs_drop_leaf_ref(trans, root, | ||
| 1815 | next); | ||
| 1816 | BUG_ON(ret); | ||
| 1817 | } | ||
| 1818 | |||
| 1819 | WARN_ON(root_owner != BTRFS_TREE_LOG_OBJECTID); | ||
| 1820 | ret = btrfs_free_reserved_extent(root, | ||
| 1821 | path->nodes[*level]->start, | ||
| 1822 | path->nodes[*level]->len); | ||
| 1823 | BUG_ON(ret); | ||
| 1824 | } | ||
| 1825 | free_extent_buffer(path->nodes[*level]); | ||
| 1826 | path->nodes[*level] = NULL; | ||
| 1827 | *level = i + 1; | ||
| 1828 | } | ||
| 1829 | } | ||
| 1830 | return 1; | ||
| 1831 | } | ||
| 1832 | |||
| 1833 | /* | ||
| 1834 | * drop the reference count on the tree rooted at 'snap'. This traverses | ||
| 1835 | * the tree freeing any blocks that have a ref count of zero after being | ||
| 1836 | * decremented. | ||
| 1837 | */ | ||
| 1838 | static int walk_log_tree(struct btrfs_trans_handle *trans, | ||
| 1839 | struct btrfs_root *log, struct walk_control *wc) | ||
| 1840 | { | ||
| 1841 | int ret = 0; | ||
| 1842 | int wret; | ||
| 1843 | int level; | ||
| 1844 | struct btrfs_path *path; | ||
| 1845 | int i; | ||
| 1846 | int orig_level; | ||
| 1847 | |||
| 1848 | path = btrfs_alloc_path(); | ||
| 1849 | BUG_ON(!path); | ||
| 1850 | |||
| 1851 | level = btrfs_header_level(log->node); | ||
| 1852 | orig_level = level; | ||
| 1853 | path->nodes[level] = log->node; | ||
| 1854 | extent_buffer_get(log->node); | ||
| 1855 | path->slots[level] = 0; | ||
| 1856 | |||
| 1857 | while (1) { | ||
| 1858 | wret = walk_down_log_tree(trans, log, path, &level, wc); | ||
| 1859 | if (wret > 0) | ||
| 1860 | break; | ||
| 1861 | if (wret < 0) | ||
| 1862 | ret = wret; | ||
| 1863 | |||
| 1864 | wret = walk_up_log_tree(trans, log, path, &level, wc); | ||
| 1865 | if (wret > 0) | ||
| 1866 | break; | ||
| 1867 | if (wret < 0) | ||
| 1868 | ret = wret; | ||
| 1869 | } | ||
| 1870 | |||
| 1871 | /* was the root node processed? if not, catch it here */ | ||
| 1872 | if (path->nodes[orig_level]) { | ||
| 1873 | wc->process_func(log, path->nodes[orig_level], wc, | ||
| 1874 | btrfs_header_generation(path->nodes[orig_level])); | ||
| 1875 | if (wc->free) { | ||
| 1876 | struct extent_buffer *next; | ||
| 1877 | |||
| 1878 | next = path->nodes[orig_level]; | ||
| 1879 | |||
| 1880 | btrfs_tree_lock(next); | ||
| 1881 | clean_tree_block(trans, log, next); | ||
| 1882 | btrfs_wait_tree_block_writeback(next); | ||
| 1883 | btrfs_tree_unlock(next); | ||
| 1884 | |||
| 1885 | if (orig_level == 0) { | ||
| 1886 | ret = btrfs_drop_leaf_ref(trans, log, | ||
| 1887 | next); | ||
| 1888 | BUG_ON(ret); | ||
| 1889 | } | ||
| 1890 | WARN_ON(log->root_key.objectid != | ||
| 1891 | BTRFS_TREE_LOG_OBJECTID); | ||
| 1892 | ret = btrfs_free_reserved_extent(log, next->start, | ||
| 1893 | next->len); | ||
| 1894 | BUG_ON(ret); | ||
| 1895 | } | ||
| 1896 | } | ||
| 1897 | |||
| 1898 | for (i = 0; i <= orig_level; i++) { | ||
| 1899 | if (path->nodes[i]) { | ||
| 1900 | free_extent_buffer(path->nodes[i]); | ||
| 1901 | path->nodes[i] = NULL; | ||
| 1902 | } | ||
| 1903 | } | ||
| 1904 | btrfs_free_path(path); | ||
| 1905 | if (wc->free) | ||
| 1906 | free_extent_buffer(log->node); | ||
| 1907 | return ret; | ||
| 1908 | } | ||
| 1909 | |||
| 1910 | static int wait_log_commit(struct btrfs_root *log) | ||
| 1911 | { | ||
| 1912 | DEFINE_WAIT(wait); | ||
| 1913 | u64 transid = log->fs_info->tree_log_transid; | ||
| 1914 | |||
| 1915 | do { | ||
| 1916 | prepare_to_wait(&log->fs_info->tree_log_wait, &wait, | ||
| 1917 | TASK_UNINTERRUPTIBLE); | ||
| 1918 | mutex_unlock(&log->fs_info->tree_log_mutex); | ||
| 1919 | if (atomic_read(&log->fs_info->tree_log_commit)) | ||
| 1920 | schedule(); | ||
| 1921 | finish_wait(&log->fs_info->tree_log_wait, &wait); | ||
| 1922 | mutex_lock(&log->fs_info->tree_log_mutex); | ||
| 1923 | } while (transid == log->fs_info->tree_log_transid && | ||
| 1924 | atomic_read(&log->fs_info->tree_log_commit)); | ||
| 1925 | return 0; | ||
| 1926 | } | ||
| 1927 | |||
| 1928 | /* | ||
| 1929 | * btrfs_sync_log does sends a given tree log down to the disk and | ||
| 1930 | * updates the super blocks to record it. When this call is done, | ||
| 1931 | * you know that any inodes previously logged are safely on disk | ||
| 1932 | */ | ||
| 1933 | int btrfs_sync_log(struct btrfs_trans_handle *trans, | ||
| 1934 | struct btrfs_root *root) | ||
| 1935 | { | ||
| 1936 | int ret; | ||
| 1937 | unsigned long batch; | ||
| 1938 | struct btrfs_root *log = root->log_root; | ||
| 1939 | |||
| 1940 | mutex_lock(&log->fs_info->tree_log_mutex); | ||
| 1941 | if (atomic_read(&log->fs_info->tree_log_commit)) { | ||
| 1942 | wait_log_commit(log); | ||
| 1943 | goto out; | ||
| 1944 | } | ||
| 1945 | atomic_set(&log->fs_info->tree_log_commit, 1); | ||
| 1946 | |||
| 1947 | while (1) { | ||
| 1948 | batch = log->fs_info->tree_log_batch; | ||
| 1949 | mutex_unlock(&log->fs_info->tree_log_mutex); | ||
| 1950 | schedule_timeout_uninterruptible(1); | ||
| 1951 | mutex_lock(&log->fs_info->tree_log_mutex); | ||
| 1952 | |||
| 1953 | while (atomic_read(&log->fs_info->tree_log_writers)) { | ||
| 1954 | DEFINE_WAIT(wait); | ||
| 1955 | prepare_to_wait(&log->fs_info->tree_log_wait, &wait, | ||
| 1956 | TASK_UNINTERRUPTIBLE); | ||
| 1957 | mutex_unlock(&log->fs_info->tree_log_mutex); | ||
| 1958 | if (atomic_read(&log->fs_info->tree_log_writers)) | ||
| 1959 | schedule(); | ||
| 1960 | mutex_lock(&log->fs_info->tree_log_mutex); | ||
| 1961 | finish_wait(&log->fs_info->tree_log_wait, &wait); | ||
| 1962 | } | ||
| 1963 | if (batch == log->fs_info->tree_log_batch) | ||
| 1964 | break; | ||
| 1965 | } | ||
| 1966 | |||
| 1967 | ret = btrfs_write_and_wait_marked_extents(log, &log->dirty_log_pages); | ||
| 1968 | BUG_ON(ret); | ||
| 1969 | ret = btrfs_write_and_wait_marked_extents(root->fs_info->log_root_tree, | ||
| 1970 | &root->fs_info->log_root_tree->dirty_log_pages); | ||
| 1971 | BUG_ON(ret); | ||
| 1972 | |||
| 1973 | btrfs_set_super_log_root(&root->fs_info->super_for_commit, | ||
| 1974 | log->fs_info->log_root_tree->node->start); | ||
| 1975 | btrfs_set_super_log_root_level(&root->fs_info->super_for_commit, | ||
| 1976 | btrfs_header_level(log->fs_info->log_root_tree->node)); | ||
| 1977 | |||
| 1978 | write_ctree_super(trans, log->fs_info->tree_root, 2); | ||
| 1979 | log->fs_info->tree_log_transid++; | ||
| 1980 | log->fs_info->tree_log_batch = 0; | ||
| 1981 | atomic_set(&log->fs_info->tree_log_commit, 0); | ||
| 1982 | smp_mb(); | ||
| 1983 | if (waitqueue_active(&log->fs_info->tree_log_wait)) | ||
| 1984 | wake_up(&log->fs_info->tree_log_wait); | ||
| 1985 | out: | ||
| 1986 | mutex_unlock(&log->fs_info->tree_log_mutex); | ||
| 1987 | return 0; | ||
| 1988 | } | ||
| 1989 | |||
| 1990 | /* * free all the extents used by the tree log. This should be called | ||
| 1991 | * at commit time of the full transaction | ||
| 1992 | */ | ||
| 1993 | int btrfs_free_log(struct btrfs_trans_handle *trans, struct btrfs_root *root) | ||
| 1994 | { | ||
| 1995 | int ret; | ||
| 1996 | struct btrfs_root *log; | ||
| 1997 | struct key; | ||
| 1998 | u64 start; | ||
| 1999 | u64 end; | ||
| 2000 | struct walk_control wc = { | ||
| 2001 | .free = 1, | ||
| 2002 | .process_func = process_one_buffer | ||
| 2003 | }; | ||
| 2004 | |||
| 2005 | if (!root->log_root || root->fs_info->log_root_recovering) | ||
| 2006 | return 0; | ||
| 2007 | |||
| 2008 | log = root->log_root; | ||
| 2009 | ret = walk_log_tree(trans, log, &wc); | ||
| 2010 | BUG_ON(ret); | ||
| 2011 | |||
| 2012 | while (1) { | ||
| 2013 | ret = find_first_extent_bit(&log->dirty_log_pages, | ||
| 2014 | 0, &start, &end, EXTENT_DIRTY); | ||
| 2015 | if (ret) | ||
| 2016 | break; | ||
| 2017 | |||
| 2018 | clear_extent_dirty(&log->dirty_log_pages, | ||
| 2019 | start, end, GFP_NOFS); | ||
| 2020 | } | ||
| 2021 | |||
| 2022 | log = root->log_root; | ||
| 2023 | ret = btrfs_del_root(trans, root->fs_info->log_root_tree, | ||
| 2024 | &log->root_key); | ||
| 2025 | BUG_ON(ret); | ||
| 2026 | root->log_root = NULL; | ||
| 2027 | kfree(root->log_root); | ||
| 2028 | return 0; | ||
| 2029 | } | ||
| 2030 | |||
| 2031 | /* | ||
| 2032 | * helper function to update the item for a given subvolumes log root | ||
| 2033 | * in the tree of log roots | ||
| 2034 | */ | ||
| 2035 | static int update_log_root(struct btrfs_trans_handle *trans, | ||
| 2036 | struct btrfs_root *log) | ||
| 2037 | { | ||
| 2038 | u64 bytenr = btrfs_root_bytenr(&log->root_item); | ||
| 2039 | int ret; | ||
| 2040 | |||
| 2041 | if (log->node->start == bytenr) | ||
| 2042 | return 0; | ||
| 2043 | |||
| 2044 | btrfs_set_root_bytenr(&log->root_item, log->node->start); | ||
| 2045 | btrfs_set_root_generation(&log->root_item, trans->transid); | ||
| 2046 | btrfs_set_root_level(&log->root_item, btrfs_header_level(log->node)); | ||
| 2047 | ret = btrfs_update_root(trans, log->fs_info->log_root_tree, | ||
| 2048 | &log->root_key, &log->root_item); | ||
| 2049 | BUG_ON(ret); | ||
| 2050 | return ret; | ||
| 2051 | } | ||
| 2052 | |||
| 2053 | /* | ||
| 2054 | * If both a file and directory are logged, and unlinks or renames are | ||
| 2055 | * mixed in, we have a few interesting corners: | ||
| 2056 | * | ||
| 2057 | * create file X in dir Y | ||
| 2058 | * link file X to X.link in dir Y | ||
| 2059 | * fsync file X | ||
| 2060 | * unlink file X but leave X.link | ||
| 2061 | * fsync dir Y | ||
| 2062 | * | ||
| 2063 | * After a crash we would expect only X.link to exist. But file X | ||
| 2064 | * didn't get fsync'd again so the log has back refs for X and X.link. | ||
| 2065 | * | ||
| 2066 | * We solve this by removing directory entries and inode backrefs from the | ||
| 2067 | * log when a file that was logged in the current transaction is | ||
| 2068 | * unlinked. Any later fsync will include the updated log entries, and | ||
| 2069 | * we'll be able to reconstruct the proper directory items from backrefs. | ||
| 2070 | * | ||
| 2071 | * This optimizations allows us to avoid relogging the entire inode | ||
| 2072 | * or the entire directory. | ||
| 2073 | */ | ||
| 2074 | int btrfs_del_dir_entries_in_log(struct btrfs_trans_handle *trans, | ||
| 2075 | struct btrfs_root *root, | ||
| 2076 | const char *name, int name_len, | ||
| 2077 | struct inode *dir, u64 index) | ||
| 2078 | { | ||
| 2079 | struct btrfs_root *log; | ||
| 2080 | struct btrfs_dir_item *di; | ||
| 2081 | struct btrfs_path *path; | ||
| 2082 | int ret; | ||
| 2083 | int bytes_del = 0; | ||
| 2084 | |||
| 2085 | if (BTRFS_I(dir)->logged_trans < trans->transid) | ||
| 2086 | return 0; | ||
| 2087 | |||
| 2088 | ret = join_running_log_trans(root); | ||
| 2089 | if (ret) | ||
| 2090 | return 0; | ||
| 2091 | |||
| 2092 | mutex_lock(&BTRFS_I(dir)->log_mutex); | ||
| 2093 | |||
| 2094 | log = root->log_root; | ||
| 2095 | path = btrfs_alloc_path(); | ||
| 2096 | di = btrfs_lookup_dir_item(trans, log, path, dir->i_ino, | ||
| 2097 | name, name_len, -1); | ||
| 2098 | if (di && !IS_ERR(di)) { | ||
| 2099 | ret = btrfs_delete_one_dir_name(trans, log, path, di); | ||
| 2100 | bytes_del += name_len; | ||
| 2101 | BUG_ON(ret); | ||
| 2102 | } | ||
| 2103 | btrfs_release_path(log, path); | ||
| 2104 | di = btrfs_lookup_dir_index_item(trans, log, path, dir->i_ino, | ||
| 2105 | index, name, name_len, -1); | ||
| 2106 | if (di && !IS_ERR(di)) { | ||
| 2107 | ret = btrfs_delete_one_dir_name(trans, log, path, di); | ||
| 2108 | bytes_del += name_len; | ||
| 2109 | BUG_ON(ret); | ||
| 2110 | } | ||
| 2111 | |||
| 2112 | /* update the directory size in the log to reflect the names | ||
| 2113 | * we have removed | ||
| 2114 | */ | ||
| 2115 | if (bytes_del) { | ||
| 2116 | struct btrfs_key key; | ||
| 2117 | |||
| 2118 | key.objectid = dir->i_ino; | ||
| 2119 | key.offset = 0; | ||
| 2120 | key.type = BTRFS_INODE_ITEM_KEY; | ||
| 2121 | btrfs_release_path(log, path); | ||
| 2122 | |||
| 2123 | ret = btrfs_search_slot(trans, log, &key, path, 0, 1); | ||
| 2124 | if (ret == 0) { | ||
| 2125 | struct btrfs_inode_item *item; | ||
| 2126 | u64 i_size; | ||
| 2127 | |||
| 2128 | item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 2129 | struct btrfs_inode_item); | ||
| 2130 | i_size = btrfs_inode_size(path->nodes[0], item); | ||
| 2131 | if (i_size > bytes_del) | ||
| 2132 | i_size -= bytes_del; | ||
| 2133 | else | ||
| 2134 | i_size = 0; | ||
| 2135 | btrfs_set_inode_size(path->nodes[0], item, i_size); | ||
| 2136 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 2137 | } else | ||
| 2138 | ret = 0; | ||
| 2139 | btrfs_release_path(log, path); | ||
| 2140 | } | ||
| 2141 | |||
| 2142 | btrfs_free_path(path); | ||
| 2143 | mutex_unlock(&BTRFS_I(dir)->log_mutex); | ||
| 2144 | end_log_trans(root); | ||
| 2145 | |||
| 2146 | return 0; | ||
| 2147 | } | ||
| 2148 | |||
| 2149 | /* see comments for btrfs_del_dir_entries_in_log */ | ||
| 2150 | int btrfs_del_inode_ref_in_log(struct btrfs_trans_handle *trans, | ||
| 2151 | struct btrfs_root *root, | ||
| 2152 | const char *name, int name_len, | ||
| 2153 | struct inode *inode, u64 dirid) | ||
| 2154 | { | ||
| 2155 | struct btrfs_root *log; | ||
| 2156 | u64 index; | ||
| 2157 | int ret; | ||
| 2158 | |||
| 2159 | if (BTRFS_I(inode)->logged_trans < trans->transid) | ||
| 2160 | return 0; | ||
| 2161 | |||
| 2162 | ret = join_running_log_trans(root); | ||
| 2163 | if (ret) | ||
| 2164 | return 0; | ||
| 2165 | log = root->log_root; | ||
| 2166 | mutex_lock(&BTRFS_I(inode)->log_mutex); | ||
| 2167 | |||
| 2168 | ret = btrfs_del_inode_ref(trans, log, name, name_len, inode->i_ino, | ||
| 2169 | dirid, &index); | ||
| 2170 | mutex_unlock(&BTRFS_I(inode)->log_mutex); | ||
| 2171 | end_log_trans(root); | ||
| 2172 | |||
| 2173 | return ret; | ||
| 2174 | } | ||
| 2175 | |||
| 2176 | /* | ||
| 2177 | * creates a range item in the log for 'dirid'. first_offset and | ||
| 2178 | * last_offset tell us which parts of the key space the log should | ||
| 2179 | * be considered authoritative for. | ||
| 2180 | */ | ||
| 2181 | static noinline int insert_dir_log_key(struct btrfs_trans_handle *trans, | ||
| 2182 | struct btrfs_root *log, | ||
| 2183 | struct btrfs_path *path, | ||
| 2184 | int key_type, u64 dirid, | ||
| 2185 | u64 first_offset, u64 last_offset) | ||
| 2186 | { | ||
| 2187 | int ret; | ||
| 2188 | struct btrfs_key key; | ||
| 2189 | struct btrfs_dir_log_item *item; | ||
| 2190 | |||
| 2191 | key.objectid = dirid; | ||
| 2192 | key.offset = first_offset; | ||
| 2193 | if (key_type == BTRFS_DIR_ITEM_KEY) | ||
| 2194 | key.type = BTRFS_DIR_LOG_ITEM_KEY; | ||
| 2195 | else | ||
| 2196 | key.type = BTRFS_DIR_LOG_INDEX_KEY; | ||
| 2197 | ret = btrfs_insert_empty_item(trans, log, path, &key, sizeof(*item)); | ||
| 2198 | BUG_ON(ret); | ||
| 2199 | |||
| 2200 | item = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 2201 | struct btrfs_dir_log_item); | ||
| 2202 | btrfs_set_dir_log_end(path->nodes[0], item, last_offset); | ||
| 2203 | btrfs_mark_buffer_dirty(path->nodes[0]); | ||
| 2204 | btrfs_release_path(log, path); | ||
| 2205 | return 0; | ||
| 2206 | } | ||
| 2207 | |||
| 2208 | /* | ||
| 2209 | * log all the items included in the current transaction for a given | ||
| 2210 | * directory. This also creates the range items in the log tree required | ||
| 2211 | * to replay anything deleted before the fsync | ||
| 2212 | */ | ||
| 2213 | static noinline int log_dir_items(struct btrfs_trans_handle *trans, | ||
| 2214 | struct btrfs_root *root, struct inode *inode, | ||
| 2215 | struct btrfs_path *path, | ||
| 2216 | struct btrfs_path *dst_path, int key_type, | ||
| 2217 | u64 min_offset, u64 *last_offset_ret) | ||
| 2218 | { | ||
| 2219 | struct btrfs_key min_key; | ||
| 2220 | struct btrfs_key max_key; | ||
| 2221 | struct btrfs_root *log = root->log_root; | ||
| 2222 | struct extent_buffer *src; | ||
| 2223 | int ret; | ||
| 2224 | int i; | ||
| 2225 | int nritems; | ||
| 2226 | u64 first_offset = min_offset; | ||
| 2227 | u64 last_offset = (u64)-1; | ||
| 2228 | |||
| 2229 | log = root->log_root; | ||
| 2230 | max_key.objectid = inode->i_ino; | ||
| 2231 | max_key.offset = (u64)-1; | ||
| 2232 | max_key.type = key_type; | ||
| 2233 | |||
| 2234 | min_key.objectid = inode->i_ino; | ||
| 2235 | min_key.type = key_type; | ||
| 2236 | min_key.offset = min_offset; | ||
| 2237 | |||
| 2238 | path->keep_locks = 1; | ||
| 2239 | |||
| 2240 | ret = btrfs_search_forward(root, &min_key, &max_key, | ||
| 2241 | path, 0, trans->transid); | ||
| 2242 | |||
| 2243 | /* | ||
| 2244 | * we didn't find anything from this transaction, see if there | ||
| 2245 | * is anything at all | ||
| 2246 | */ | ||
| 2247 | if (ret != 0 || min_key.objectid != inode->i_ino || | ||
| 2248 | min_key.type != key_type) { | ||
| 2249 | min_key.objectid = inode->i_ino; | ||
| 2250 | min_key.type = key_type; | ||
| 2251 | min_key.offset = (u64)-1; | ||
| 2252 | btrfs_release_path(root, path); | ||
| 2253 | ret = btrfs_search_slot(NULL, root, &min_key, path, 0, 0); | ||
| 2254 | if (ret < 0) { | ||
| 2255 | btrfs_release_path(root, path); | ||
| 2256 | return ret; | ||
| 2257 | } | ||
| 2258 | ret = btrfs_previous_item(root, path, inode->i_ino, key_type); | ||
| 2259 | |||
| 2260 | /* if ret == 0 there are items for this type, | ||
| 2261 | * create a range to tell us the last key of this type. | ||
| 2262 | * otherwise, there are no items in this directory after | ||
| 2263 | * *min_offset, and we create a range to indicate that. | ||
| 2264 | */ | ||
| 2265 | if (ret == 0) { | ||
| 2266 | struct btrfs_key tmp; | ||
| 2267 | btrfs_item_key_to_cpu(path->nodes[0], &tmp, | ||
| 2268 | path->slots[0]); | ||
| 2269 | if (key_type == tmp.type) | ||
| 2270 | first_offset = max(min_offset, tmp.offset) + 1; | ||
| 2271 | } | ||
| 2272 | goto done; | ||
| 2273 | } | ||
| 2274 | |||
| 2275 | /* go backward to find any previous key */ | ||
| 2276 | ret = btrfs_previous_item(root, path, inode->i_ino, key_type); | ||
| 2277 | if (ret == 0) { | ||
| 2278 | struct btrfs_key tmp; | ||
| 2279 | btrfs_item_key_to_cpu(path->nodes[0], &tmp, path->slots[0]); | ||
| 2280 | if (key_type == tmp.type) { | ||
| 2281 | first_offset = tmp.offset; | ||
| 2282 | ret = overwrite_item(trans, log, dst_path, | ||
| 2283 | path->nodes[0], path->slots[0], | ||
| 2284 | &tmp); | ||
| 2285 | } | ||
| 2286 | } | ||
| 2287 | btrfs_release_path(root, path); | ||
| 2288 | |||
| 2289 | /* find the first key from this transaction again */ | ||
| 2290 | ret = btrfs_search_slot(NULL, root, &min_key, path, 0, 0); | ||
| 2291 | if (ret != 0) { | ||
| 2292 | WARN_ON(1); | ||
| 2293 | goto done; | ||
| 2294 | } | ||
| 2295 | |||
| 2296 | /* | ||
| 2297 | * we have a block from this transaction, log every item in it | ||
| 2298 | * from our directory | ||
| 2299 | */ | ||
| 2300 | while (1) { | ||
| 2301 | struct btrfs_key tmp; | ||
| 2302 | src = path->nodes[0]; | ||
| 2303 | nritems = btrfs_header_nritems(src); | ||
| 2304 | for (i = path->slots[0]; i < nritems; i++) { | ||
| 2305 | btrfs_item_key_to_cpu(src, &min_key, i); | ||
| 2306 | |||
| 2307 | if (min_key.objectid != inode->i_ino || | ||
| 2308 | min_key.type != key_type) | ||
| 2309 | goto done; | ||
| 2310 | ret = overwrite_item(trans, log, dst_path, src, i, | ||
| 2311 | &min_key); | ||
| 2312 | BUG_ON(ret); | ||
| 2313 | } | ||
| 2314 | path->slots[0] = nritems; | ||
| 2315 | |||
| 2316 | /* | ||
| 2317 | * look ahead to the next item and see if it is also | ||
| 2318 | * from this directory and from this transaction | ||
| 2319 | */ | ||
| 2320 | ret = btrfs_next_leaf(root, path); | ||
| 2321 | if (ret == 1) { | ||
| 2322 | last_offset = (u64)-1; | ||
| 2323 | goto done; | ||
| 2324 | } | ||
| 2325 | btrfs_item_key_to_cpu(path->nodes[0], &tmp, path->slots[0]); | ||
| 2326 | if (tmp.objectid != inode->i_ino || tmp.type != key_type) { | ||
| 2327 | last_offset = (u64)-1; | ||
| 2328 | goto done; | ||
| 2329 | } | ||
| 2330 | if (btrfs_header_generation(path->nodes[0]) != trans->transid) { | ||
| 2331 | ret = overwrite_item(trans, log, dst_path, | ||
| 2332 | path->nodes[0], path->slots[0], | ||
| 2333 | &tmp); | ||
| 2334 | |||
| 2335 | BUG_ON(ret); | ||
| 2336 | last_offset = tmp.offset; | ||
| 2337 | goto done; | ||
| 2338 | } | ||
| 2339 | } | ||
| 2340 | done: | ||
| 2341 | *last_offset_ret = last_offset; | ||
| 2342 | btrfs_release_path(root, path); | ||
| 2343 | btrfs_release_path(log, dst_path); | ||
| 2344 | |||
| 2345 | /* insert the log range keys to indicate where the log is valid */ | ||
| 2346 | ret = insert_dir_log_key(trans, log, path, key_type, inode->i_ino, | ||
| 2347 | first_offset, last_offset); | ||
| 2348 | BUG_ON(ret); | ||
| 2349 | return 0; | ||
| 2350 | } | ||
| 2351 | |||
| 2352 | /* | ||
| 2353 | * logging directories is very similar to logging inodes, We find all the items | ||
| 2354 | * from the current transaction and write them to the log. | ||
| 2355 | * | ||
| 2356 | * The recovery code scans the directory in the subvolume, and if it finds a | ||
| 2357 | * key in the range logged that is not present in the log tree, then it means | ||
| 2358 | * that dir entry was unlinked during the transaction. | ||
| 2359 | * | ||
| 2360 | * In order for that scan to work, we must include one key smaller than | ||
| 2361 | * the smallest logged by this transaction and one key larger than the largest | ||
| 2362 | * key logged by this transaction. | ||
| 2363 | */ | ||
| 2364 | static noinline int log_directory_changes(struct btrfs_trans_handle *trans, | ||
| 2365 | struct btrfs_root *root, struct inode *inode, | ||
| 2366 | struct btrfs_path *path, | ||
| 2367 | struct btrfs_path *dst_path) | ||
| 2368 | { | ||
| 2369 | u64 min_key; | ||
| 2370 | u64 max_key; | ||
| 2371 | int ret; | ||
| 2372 | int key_type = BTRFS_DIR_ITEM_KEY; | ||
| 2373 | |||
| 2374 | again: | ||
| 2375 | min_key = 0; | ||
| 2376 | max_key = 0; | ||
| 2377 | while (1) { | ||
| 2378 | ret = log_dir_items(trans, root, inode, path, | ||
| 2379 | dst_path, key_type, min_key, | ||
| 2380 | &max_key); | ||
| 2381 | BUG_ON(ret); | ||
| 2382 | if (max_key == (u64)-1) | ||
| 2383 | break; | ||
| 2384 | min_key = max_key + 1; | ||
| 2385 | } | ||
| 2386 | |||
| 2387 | if (key_type == BTRFS_DIR_ITEM_KEY) { | ||
| 2388 | key_type = BTRFS_DIR_INDEX_KEY; | ||
| 2389 | goto again; | ||
| 2390 | } | ||
| 2391 | return 0; | ||
| 2392 | } | ||
| 2393 | |||
| 2394 | /* | ||
| 2395 | * a helper function to drop items from the log before we relog an | ||
| 2396 | * inode. max_key_type indicates the highest item type to remove. | ||
| 2397 | * This cannot be run for file data extents because it does not | ||
| 2398 | * free the extents they point to. | ||
| 2399 | */ | ||
| 2400 | static int drop_objectid_items(struct btrfs_trans_handle *trans, | ||
| 2401 | struct btrfs_root *log, | ||
| 2402 | struct btrfs_path *path, | ||
| 2403 | u64 objectid, int max_key_type) | ||
| 2404 | { | ||
| 2405 | int ret; | ||
| 2406 | struct btrfs_key key; | ||
| 2407 | struct btrfs_key found_key; | ||
| 2408 | |||
| 2409 | key.objectid = objectid; | ||
| 2410 | key.type = max_key_type; | ||
| 2411 | key.offset = (u64)-1; | ||
| 2412 | |||
| 2413 | while (1) { | ||
| 2414 | ret = btrfs_search_slot(trans, log, &key, path, -1, 1); | ||
| 2415 | |||
| 2416 | if (ret != 1) | ||
| 2417 | break; | ||
| 2418 | |||
| 2419 | if (path->slots[0] == 0) | ||
| 2420 | break; | ||
| 2421 | |||
| 2422 | path->slots[0]--; | ||
| 2423 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 2424 | path->slots[0]); | ||
| 2425 | |||
| 2426 | if (found_key.objectid != objectid) | ||
| 2427 | break; | ||
| 2428 | |||
| 2429 | ret = btrfs_del_item(trans, log, path); | ||
| 2430 | BUG_ON(ret); | ||
| 2431 | btrfs_release_path(log, path); | ||
| 2432 | } | ||
| 2433 | btrfs_release_path(log, path); | ||
| 2434 | return 0; | ||
| 2435 | } | ||
| 2436 | |||
| 2437 | static noinline int copy_items(struct btrfs_trans_handle *trans, | ||
| 2438 | struct btrfs_root *log, | ||
| 2439 | struct btrfs_path *dst_path, | ||
| 2440 | struct extent_buffer *src, | ||
| 2441 | int start_slot, int nr, int inode_only) | ||
| 2442 | { | ||
| 2443 | unsigned long src_offset; | ||
| 2444 | unsigned long dst_offset; | ||
| 2445 | struct btrfs_file_extent_item *extent; | ||
| 2446 | struct btrfs_inode_item *inode_item; | ||
| 2447 | int ret; | ||
| 2448 | struct btrfs_key *ins_keys; | ||
| 2449 | u32 *ins_sizes; | ||
| 2450 | char *ins_data; | ||
| 2451 | int i; | ||
| 2452 | struct list_head ordered_sums; | ||
| 2453 | |||
| 2454 | INIT_LIST_HEAD(&ordered_sums); | ||
| 2455 | |||
| 2456 | ins_data = kmalloc(nr * sizeof(struct btrfs_key) + | ||
| 2457 | nr * sizeof(u32), GFP_NOFS); | ||
| 2458 | ins_sizes = (u32 *)ins_data; | ||
| 2459 | ins_keys = (struct btrfs_key *)(ins_data + nr * sizeof(u32)); | ||
| 2460 | |||
| 2461 | for (i = 0; i < nr; i++) { | ||
| 2462 | ins_sizes[i] = btrfs_item_size_nr(src, i + start_slot); | ||
| 2463 | btrfs_item_key_to_cpu(src, ins_keys + i, i + start_slot); | ||
| 2464 | } | ||
| 2465 | ret = btrfs_insert_empty_items(trans, log, dst_path, | ||
| 2466 | ins_keys, ins_sizes, nr); | ||
| 2467 | BUG_ON(ret); | ||
| 2468 | |||
| 2469 | for (i = 0; i < nr; i++) { | ||
| 2470 | dst_offset = btrfs_item_ptr_offset(dst_path->nodes[0], | ||
| 2471 | dst_path->slots[0]); | ||
| 2472 | |||
| 2473 | src_offset = btrfs_item_ptr_offset(src, start_slot + i); | ||
| 2474 | |||
| 2475 | copy_extent_buffer(dst_path->nodes[0], src, dst_offset, | ||
| 2476 | src_offset, ins_sizes[i]); | ||
| 2477 | |||
| 2478 | if (inode_only == LOG_INODE_EXISTS && | ||
| 2479 | ins_keys[i].type == BTRFS_INODE_ITEM_KEY) { | ||
| 2480 | inode_item = btrfs_item_ptr(dst_path->nodes[0], | ||
| 2481 | dst_path->slots[0], | ||
| 2482 | struct btrfs_inode_item); | ||
| 2483 | btrfs_set_inode_size(dst_path->nodes[0], inode_item, 0); | ||
| 2484 | |||
| 2485 | /* set the generation to zero so the recover code | ||
| 2486 | * can tell the difference between an logging | ||
| 2487 | * just to say 'this inode exists' and a logging | ||
| 2488 | * to say 'update this inode with these values' | ||
| 2489 | */ | ||
| 2490 | btrfs_set_inode_generation(dst_path->nodes[0], | ||
| 2491 | inode_item, 0); | ||
| 2492 | } | ||
| 2493 | /* take a reference on file data extents so that truncates | ||
| 2494 | * or deletes of this inode don't have to relog the inode | ||
| 2495 | * again | ||
| 2496 | */ | ||
| 2497 | if (btrfs_key_type(ins_keys + i) == BTRFS_EXTENT_DATA_KEY) { | ||
| 2498 | int found_type; | ||
| 2499 | extent = btrfs_item_ptr(src, start_slot + i, | ||
| 2500 | struct btrfs_file_extent_item); | ||
| 2501 | |||
| 2502 | found_type = btrfs_file_extent_type(src, extent); | ||
| 2503 | if (found_type == BTRFS_FILE_EXTENT_REG || | ||
| 2504 | found_type == BTRFS_FILE_EXTENT_PREALLOC) { | ||
| 2505 | u64 ds = btrfs_file_extent_disk_bytenr(src, | ||
| 2506 | extent); | ||
| 2507 | u64 dl = btrfs_file_extent_disk_num_bytes(src, | ||
| 2508 | extent); | ||
| 2509 | u64 cs = btrfs_file_extent_offset(src, extent); | ||
| 2510 | u64 cl = btrfs_file_extent_num_bytes(src, | ||
| 2511 | extent);; | ||
| 2512 | if (btrfs_file_extent_compression(src, | ||
| 2513 | extent)) { | ||
| 2514 | cs = 0; | ||
| 2515 | cl = dl; | ||
| 2516 | } | ||
| 2517 | /* ds == 0 is a hole */ | ||
| 2518 | if (ds != 0) { | ||
| 2519 | ret = btrfs_inc_extent_ref(trans, log, | ||
| 2520 | ds, dl, | ||
| 2521 | dst_path->nodes[0]->start, | ||
| 2522 | BTRFS_TREE_LOG_OBJECTID, | ||
| 2523 | trans->transid, | ||
| 2524 | ins_keys[i].objectid); | ||
| 2525 | BUG_ON(ret); | ||
| 2526 | ret = btrfs_lookup_csums_range( | ||
| 2527 | log->fs_info->csum_root, | ||
| 2528 | ds + cs, ds + cs + cl - 1, | ||
| 2529 | &ordered_sums); | ||
| 2530 | BUG_ON(ret); | ||
| 2531 | } | ||
| 2532 | } | ||
| 2533 | } | ||
| 2534 | dst_path->slots[0]++; | ||
| 2535 | } | ||
| 2536 | |||
| 2537 | btrfs_mark_buffer_dirty(dst_path->nodes[0]); | ||
| 2538 | btrfs_release_path(log, dst_path); | ||
| 2539 | kfree(ins_data); | ||
| 2540 | |||
| 2541 | /* | ||
| 2542 | * we have to do this after the loop above to avoid changing the | ||
| 2543 | * log tree while trying to change the log tree. | ||
| 2544 | */ | ||
| 2545 | while (!list_empty(&ordered_sums)) { | ||
| 2546 | struct btrfs_ordered_sum *sums = list_entry(ordered_sums.next, | ||
| 2547 | struct btrfs_ordered_sum, | ||
| 2548 | list); | ||
| 2549 | ret = btrfs_csum_file_blocks(trans, log, sums); | ||
| 2550 | BUG_ON(ret); | ||
| 2551 | list_del(&sums->list); | ||
| 2552 | kfree(sums); | ||
| 2553 | } | ||
| 2554 | return 0; | ||
| 2555 | } | ||
| 2556 | |||
| 2557 | /* log a single inode in the tree log. | ||
| 2558 | * At least one parent directory for this inode must exist in the tree | ||
| 2559 | * or be logged already. | ||
| 2560 | * | ||
| 2561 | * Any items from this inode changed by the current transaction are copied | ||
| 2562 | * to the log tree. An extra reference is taken on any extents in this | ||
| 2563 | * file, allowing us to avoid a whole pile of corner cases around logging | ||
| 2564 | * blocks that have been removed from the tree. | ||
| 2565 | * | ||
| 2566 | * See LOG_INODE_ALL and related defines for a description of what inode_only | ||
| 2567 | * does. | ||
| 2568 | * | ||
| 2569 | * This handles both files and directories. | ||
| 2570 | */ | ||
| 2571 | static int __btrfs_log_inode(struct btrfs_trans_handle *trans, | ||
| 2572 | struct btrfs_root *root, struct inode *inode, | ||
| 2573 | int inode_only) | ||
| 2574 | { | ||
| 2575 | struct btrfs_path *path; | ||
| 2576 | struct btrfs_path *dst_path; | ||
| 2577 | struct btrfs_key min_key; | ||
| 2578 | struct btrfs_key max_key; | ||
| 2579 | struct btrfs_root *log = root->log_root; | ||
| 2580 | struct extent_buffer *src = NULL; | ||
| 2581 | u32 size; | ||
| 2582 | int ret; | ||
| 2583 | int nritems; | ||
| 2584 | int ins_start_slot = 0; | ||
| 2585 | int ins_nr; | ||
| 2586 | |||
| 2587 | log = root->log_root; | ||
| 2588 | |||
| 2589 | path = btrfs_alloc_path(); | ||
| 2590 | dst_path = btrfs_alloc_path(); | ||
| 2591 | |||
| 2592 | min_key.objectid = inode->i_ino; | ||
| 2593 | min_key.type = BTRFS_INODE_ITEM_KEY; | ||
| 2594 | min_key.offset = 0; | ||
| 2595 | |||
| 2596 | max_key.objectid = inode->i_ino; | ||
| 2597 | if (inode_only == LOG_INODE_EXISTS || S_ISDIR(inode->i_mode)) | ||
| 2598 | max_key.type = BTRFS_XATTR_ITEM_KEY; | ||
| 2599 | else | ||
| 2600 | max_key.type = (u8)-1; | ||
| 2601 | max_key.offset = (u64)-1; | ||
| 2602 | |||
| 2603 | /* | ||
| 2604 | * if this inode has already been logged and we're in inode_only | ||
| 2605 | * mode, we don't want to delete the things that have already | ||
| 2606 | * been written to the log. | ||
| 2607 | * | ||
| 2608 | * But, if the inode has been through an inode_only log, | ||
| 2609 | * the logged_trans field is not set. This allows us to catch | ||
| 2610 | * any new names for this inode in the backrefs by logging it | ||
| 2611 | * again | ||
| 2612 | */ | ||
| 2613 | if (inode_only == LOG_INODE_EXISTS && | ||
| 2614 | BTRFS_I(inode)->logged_trans == trans->transid) { | ||
| 2615 | btrfs_free_path(path); | ||
| 2616 | btrfs_free_path(dst_path); | ||
| 2617 | goto out; | ||
| 2618 | } | ||
| 2619 | mutex_lock(&BTRFS_I(inode)->log_mutex); | ||
| 2620 | |||
| 2621 | /* | ||
| 2622 | * a brute force approach to making sure we get the most uptodate | ||
| 2623 | * copies of everything. | ||
| 2624 | */ | ||
| 2625 | if (S_ISDIR(inode->i_mode)) { | ||
| 2626 | int max_key_type = BTRFS_DIR_LOG_INDEX_KEY; | ||
| 2627 | |||
| 2628 | if (inode_only == LOG_INODE_EXISTS) | ||
| 2629 | max_key_type = BTRFS_XATTR_ITEM_KEY; | ||
| 2630 | ret = drop_objectid_items(trans, log, path, | ||
| 2631 | inode->i_ino, max_key_type); | ||
| 2632 | } else { | ||
| 2633 | ret = btrfs_truncate_inode_items(trans, log, inode, 0, 0); | ||
| 2634 | } | ||
| 2635 | BUG_ON(ret); | ||
| 2636 | path->keep_locks = 1; | ||
| 2637 | |||
| 2638 | while (1) { | ||
| 2639 | ins_nr = 0; | ||
| 2640 | ret = btrfs_search_forward(root, &min_key, &max_key, | ||
| 2641 | path, 0, trans->transid); | ||
| 2642 | if (ret != 0) | ||
| 2643 | break; | ||
| 2644 | again: | ||
| 2645 | /* note, ins_nr might be > 0 here, cleanup outside the loop */ | ||
| 2646 | if (min_key.objectid != inode->i_ino) | ||
| 2647 | break; | ||
| 2648 | if (min_key.type > max_key.type) | ||
| 2649 | break; | ||
| 2650 | |||
| 2651 | src = path->nodes[0]; | ||
| 2652 | size = btrfs_item_size_nr(src, path->slots[0]); | ||
| 2653 | if (ins_nr && ins_start_slot + ins_nr == path->slots[0]) { | ||
| 2654 | ins_nr++; | ||
| 2655 | goto next_slot; | ||
| 2656 | } else if (!ins_nr) { | ||
| 2657 | ins_start_slot = path->slots[0]; | ||
| 2658 | ins_nr = 1; | ||
| 2659 | goto next_slot; | ||
| 2660 | } | ||
| 2661 | |||
| 2662 | ret = copy_items(trans, log, dst_path, src, ins_start_slot, | ||
| 2663 | ins_nr, inode_only); | ||
| 2664 | BUG_ON(ret); | ||
| 2665 | ins_nr = 1; | ||
| 2666 | ins_start_slot = path->slots[0]; | ||
| 2667 | next_slot: | ||
| 2668 | |||
| 2669 | nritems = btrfs_header_nritems(path->nodes[0]); | ||
| 2670 | path->slots[0]++; | ||
| 2671 | if (path->slots[0] < nritems) { | ||
| 2672 | btrfs_item_key_to_cpu(path->nodes[0], &min_key, | ||
| 2673 | path->slots[0]); | ||
| 2674 | goto again; | ||
| 2675 | } | ||
| 2676 | if (ins_nr) { | ||
| 2677 | ret = copy_items(trans, log, dst_path, src, | ||
| 2678 | ins_start_slot, | ||
| 2679 | ins_nr, inode_only); | ||
| 2680 | BUG_ON(ret); | ||
| 2681 | ins_nr = 0; | ||
| 2682 | } | ||
| 2683 | btrfs_release_path(root, path); | ||
| 2684 | |||
| 2685 | if (min_key.offset < (u64)-1) | ||
| 2686 | min_key.offset++; | ||
| 2687 | else if (min_key.type < (u8)-1) | ||
| 2688 | min_key.type++; | ||
| 2689 | else if (min_key.objectid < (u64)-1) | ||
| 2690 | min_key.objectid++; | ||
| 2691 | else | ||
| 2692 | break; | ||
| 2693 | } | ||
| 2694 | if (ins_nr) { | ||
| 2695 | ret = copy_items(trans, log, dst_path, src, | ||
| 2696 | ins_start_slot, | ||
| 2697 | ins_nr, inode_only); | ||
| 2698 | BUG_ON(ret); | ||
| 2699 | ins_nr = 0; | ||
| 2700 | } | ||
| 2701 | WARN_ON(ins_nr); | ||
| 2702 | if (inode_only == LOG_INODE_ALL && S_ISDIR(inode->i_mode)) { | ||
| 2703 | btrfs_release_path(root, path); | ||
| 2704 | btrfs_release_path(log, dst_path); | ||
| 2705 | BTRFS_I(inode)->log_dirty_trans = 0; | ||
| 2706 | ret = log_directory_changes(trans, root, inode, path, dst_path); | ||
| 2707 | BUG_ON(ret); | ||
| 2708 | } | ||
| 2709 | BTRFS_I(inode)->logged_trans = trans->transid; | ||
| 2710 | mutex_unlock(&BTRFS_I(inode)->log_mutex); | ||
| 2711 | |||
| 2712 | btrfs_free_path(path); | ||
| 2713 | btrfs_free_path(dst_path); | ||
| 2714 | |||
| 2715 | mutex_lock(&root->fs_info->tree_log_mutex); | ||
| 2716 | ret = update_log_root(trans, log); | ||
| 2717 | BUG_ON(ret); | ||
| 2718 | mutex_unlock(&root->fs_info->tree_log_mutex); | ||
| 2719 | out: | ||
| 2720 | return 0; | ||
| 2721 | } | ||
| 2722 | |||
| 2723 | int btrfs_log_inode(struct btrfs_trans_handle *trans, | ||
| 2724 | struct btrfs_root *root, struct inode *inode, | ||
| 2725 | int inode_only) | ||
| 2726 | { | ||
| 2727 | int ret; | ||
| 2728 | |||
| 2729 | start_log_trans(trans, root); | ||
| 2730 | ret = __btrfs_log_inode(trans, root, inode, inode_only); | ||
| 2731 | end_log_trans(root); | ||
| 2732 | return ret; | ||
| 2733 | } | ||
| 2734 | |||
| 2735 | /* | ||
| 2736 | * helper function around btrfs_log_inode to make sure newly created | ||
| 2737 | * parent directories also end up in the log. A minimal inode and backref | ||
| 2738 | * only logging is done of any parent directories that are older than | ||
| 2739 | * the last committed transaction | ||
| 2740 | */ | ||
| 2741 | int btrfs_log_dentry(struct btrfs_trans_handle *trans, | ||
| 2742 | struct btrfs_root *root, struct dentry *dentry) | ||
| 2743 | { | ||
| 2744 | int inode_only = LOG_INODE_ALL; | ||
| 2745 | struct super_block *sb; | ||
| 2746 | int ret; | ||
| 2747 | |||
| 2748 | start_log_trans(trans, root); | ||
| 2749 | sb = dentry->d_inode->i_sb; | ||
| 2750 | while (1) { | ||
| 2751 | ret = __btrfs_log_inode(trans, root, dentry->d_inode, | ||
| 2752 | inode_only); | ||
| 2753 | BUG_ON(ret); | ||
| 2754 | inode_only = LOG_INODE_EXISTS; | ||
| 2755 | |||
| 2756 | dentry = dentry->d_parent; | ||
| 2757 | if (!dentry || !dentry->d_inode || sb != dentry->d_inode->i_sb) | ||
| 2758 | break; | ||
| 2759 | |||
| 2760 | if (BTRFS_I(dentry->d_inode)->generation <= | ||
| 2761 | root->fs_info->last_trans_committed) | ||
| 2762 | break; | ||
| 2763 | } | ||
| 2764 | end_log_trans(root); | ||
| 2765 | return 0; | ||
| 2766 | } | ||
| 2767 | |||
| 2768 | /* | ||
| 2769 | * it is not safe to log dentry if the chunk root has added new | ||
| 2770 | * chunks. This returns 0 if the dentry was logged, and 1 otherwise. | ||
| 2771 | * If this returns 1, you must commit the transaction to safely get your | ||
| 2772 | * data on disk. | ||
| 2773 | */ | ||
| 2774 | int btrfs_log_dentry_safe(struct btrfs_trans_handle *trans, | ||
| 2775 | struct btrfs_root *root, struct dentry *dentry) | ||
| 2776 | { | ||
| 2777 | u64 gen; | ||
| 2778 | gen = root->fs_info->last_trans_new_blockgroup; | ||
| 2779 | if (gen > root->fs_info->last_trans_committed) | ||
| 2780 | return 1; | ||
| 2781 | else | ||
| 2782 | return btrfs_log_dentry(trans, root, dentry); | ||
| 2783 | } | ||
| 2784 | |||
| 2785 | /* | ||
| 2786 | * should be called during mount to recover any replay any log trees | ||
| 2787 | * from the FS | ||
| 2788 | */ | ||
| 2789 | int btrfs_recover_log_trees(struct btrfs_root *log_root_tree) | ||
| 2790 | { | ||
| 2791 | int ret; | ||
| 2792 | struct btrfs_path *path; | ||
| 2793 | struct btrfs_trans_handle *trans; | ||
| 2794 | struct btrfs_key key; | ||
| 2795 | struct btrfs_key found_key; | ||
| 2796 | struct btrfs_key tmp_key; | ||
| 2797 | struct btrfs_root *log; | ||
| 2798 | struct btrfs_fs_info *fs_info = log_root_tree->fs_info; | ||
| 2799 | u64 highest_inode; | ||
| 2800 | struct walk_control wc = { | ||
| 2801 | .process_func = process_one_buffer, | ||
| 2802 | .stage = 0, | ||
| 2803 | }; | ||
| 2804 | |||
| 2805 | fs_info->log_root_recovering = 1; | ||
| 2806 | path = btrfs_alloc_path(); | ||
| 2807 | BUG_ON(!path); | ||
| 2808 | |||
| 2809 | trans = btrfs_start_transaction(fs_info->tree_root, 1); | ||
| 2810 | |||
| 2811 | wc.trans = trans; | ||
| 2812 | wc.pin = 1; | ||
| 2813 | |||
| 2814 | walk_log_tree(trans, log_root_tree, &wc); | ||
| 2815 | |||
| 2816 | again: | ||
| 2817 | key.objectid = BTRFS_TREE_LOG_OBJECTID; | ||
| 2818 | key.offset = (u64)-1; | ||
| 2819 | btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY); | ||
| 2820 | |||
| 2821 | while (1) { | ||
| 2822 | ret = btrfs_search_slot(NULL, log_root_tree, &key, path, 0, 0); | ||
| 2823 | if (ret < 0) | ||
| 2824 | break; | ||
| 2825 | if (ret > 0) { | ||
| 2826 | if (path->slots[0] == 0) | ||
| 2827 | break; | ||
| 2828 | path->slots[0]--; | ||
| 2829 | } | ||
| 2830 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 2831 | path->slots[0]); | ||
| 2832 | btrfs_release_path(log_root_tree, path); | ||
| 2833 | if (found_key.objectid != BTRFS_TREE_LOG_OBJECTID) | ||
| 2834 | break; | ||
| 2835 | |||
| 2836 | log = btrfs_read_fs_root_no_radix(log_root_tree, | ||
| 2837 | &found_key); | ||
| 2838 | BUG_ON(!log); | ||
| 2839 | |||
| 2840 | |||
| 2841 | tmp_key.objectid = found_key.offset; | ||
| 2842 | tmp_key.type = BTRFS_ROOT_ITEM_KEY; | ||
| 2843 | tmp_key.offset = (u64)-1; | ||
| 2844 | |||
| 2845 | wc.replay_dest = btrfs_read_fs_root_no_name(fs_info, &tmp_key); | ||
| 2846 | BUG_ON(!wc.replay_dest); | ||
| 2847 | |||
| 2848 | wc.replay_dest->log_root = log; | ||
| 2849 | btrfs_record_root_in_trans(wc.replay_dest); | ||
| 2850 | ret = walk_log_tree(trans, log, &wc); | ||
| 2851 | BUG_ON(ret); | ||
| 2852 | |||
| 2853 | if (wc.stage == LOG_WALK_REPLAY_ALL) { | ||
| 2854 | ret = fixup_inode_link_counts(trans, wc.replay_dest, | ||
| 2855 | path); | ||
| 2856 | BUG_ON(ret); | ||
| 2857 | } | ||
| 2858 | ret = btrfs_find_highest_inode(wc.replay_dest, &highest_inode); | ||
| 2859 | if (ret == 0) { | ||
| 2860 | wc.replay_dest->highest_inode = highest_inode; | ||
| 2861 | wc.replay_dest->last_inode_alloc = highest_inode; | ||
| 2862 | } | ||
| 2863 | |||
| 2864 | key.offset = found_key.offset - 1; | ||
| 2865 | wc.replay_dest->log_root = NULL; | ||
| 2866 | free_extent_buffer(log->node); | ||
| 2867 | kfree(log); | ||
| 2868 | |||
| 2869 | if (found_key.offset == 0) | ||
| 2870 | break; | ||
| 2871 | } | ||
| 2872 | btrfs_release_path(log_root_tree, path); | ||
| 2873 | |||
| 2874 | /* step one is to pin it all, step two is to replay just inodes */ | ||
| 2875 | if (wc.pin) { | ||
| 2876 | wc.pin = 0; | ||
| 2877 | wc.process_func = replay_one_buffer; | ||
| 2878 | wc.stage = LOG_WALK_REPLAY_INODES; | ||
| 2879 | goto again; | ||
| 2880 | } | ||
| 2881 | /* step three is to replay everything */ | ||
| 2882 | if (wc.stage < LOG_WALK_REPLAY_ALL) { | ||
| 2883 | wc.stage++; | ||
| 2884 | goto again; | ||
| 2885 | } | ||
| 2886 | |||
| 2887 | btrfs_free_path(path); | ||
| 2888 | |||
| 2889 | free_extent_buffer(log_root_tree->node); | ||
| 2890 | log_root_tree->log_root = NULL; | ||
| 2891 | fs_info->log_root_recovering = 0; | ||
| 2892 | |||
| 2893 | /* step 4: commit the transaction, which also unpins the blocks */ | ||
| 2894 | btrfs_commit_transaction(trans, fs_info->tree_root); | ||
| 2895 | |||
| 2896 | kfree(log_root_tree); | ||
| 2897 | return 0; | ||
| 2898 | } | ||
diff --git a/fs/btrfs/tree-log.h b/fs/btrfs/tree-log.h new file mode 100644 index 000000000000..b9409b32ed02 --- /dev/null +++ b/fs/btrfs/tree-log.h | |||
| @@ -0,0 +1,41 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | |||
| 19 | #ifndef __TREE_LOG_ | ||
| 20 | #define __TREE_LOG_ | ||
| 21 | |||
| 22 | int btrfs_sync_log(struct btrfs_trans_handle *trans, | ||
| 23 | struct btrfs_root *root); | ||
| 24 | int btrfs_free_log(struct btrfs_trans_handle *trans, struct btrfs_root *root); | ||
| 25 | int btrfs_log_dentry(struct btrfs_trans_handle *trans, | ||
| 26 | struct btrfs_root *root, struct dentry *dentry); | ||
| 27 | int btrfs_recover_log_trees(struct btrfs_root *tree_root); | ||
| 28 | int btrfs_log_dentry_safe(struct btrfs_trans_handle *trans, | ||
| 29 | struct btrfs_root *root, struct dentry *dentry); | ||
| 30 | int btrfs_log_inode(struct btrfs_trans_handle *trans, | ||
| 31 | struct btrfs_root *root, struct inode *inode, | ||
| 32 | int inode_only); | ||
| 33 | int btrfs_del_dir_entries_in_log(struct btrfs_trans_handle *trans, | ||
| 34 | struct btrfs_root *root, | ||
| 35 | const char *name, int name_len, | ||
| 36 | struct inode *dir, u64 index); | ||
| 37 | int btrfs_del_inode_ref_in_log(struct btrfs_trans_handle *trans, | ||
| 38 | struct btrfs_root *root, | ||
| 39 | const char *name, int name_len, | ||
| 40 | struct inode *inode, u64 dirid); | ||
| 41 | #endif | ||
diff --git a/fs/btrfs/version.h b/fs/btrfs/version.h new file mode 100644 index 000000000000..9bf3946d5ef2 --- /dev/null +++ b/fs/btrfs/version.h | |||
| @@ -0,0 +1,4 @@ | |||
| 1 | #ifndef __BTRFS_VERSION_H | ||
| 2 | #define __BTRFS_VERSION_H | ||
| 3 | #define BTRFS_BUILD_VERSION "Btrfs" | ||
| 4 | #endif | ||
diff --git a/fs/btrfs/version.sh b/fs/btrfs/version.sh new file mode 100644 index 000000000000..1ca1952fd917 --- /dev/null +++ b/fs/btrfs/version.sh | |||
| @@ -0,0 +1,43 @@ | |||
| 1 | #!/bin/bash | ||
| 2 | # | ||
| 3 | # determine-version -- report a useful version for releases | ||
| 4 | # | ||
| 5 | # Copyright 2008, Aron Griffis <agriffis@n01se.net> | ||
| 6 | # Copyright 2008, Oracle | ||
| 7 | # Released under the GNU GPLv2 | ||
| 8 | |||
| 9 | v="v0.16" | ||
| 10 | |||
| 11 | which git &> /dev/null | ||
| 12 | if [ $? == 0 ]; then | ||
| 13 | git branch >& /dev/null | ||
| 14 | if [ $? == 0 ]; then | ||
| 15 | if head=`git rev-parse --verify HEAD 2>/dev/null`; then | ||
| 16 | if tag=`git describe --tags 2>/dev/null`; then | ||
| 17 | v="$tag" | ||
| 18 | fi | ||
| 19 | |||
| 20 | # Are there uncommitted changes? | ||
| 21 | git update-index --refresh --unmerged > /dev/null | ||
| 22 | if git diff-index --name-only HEAD | \ | ||
| 23 | grep -v "^scripts/package" \ | ||
| 24 | | read dummy; then | ||
| 25 | v="$v"-dirty | ||
| 26 | fi | ||
| 27 | fi | ||
| 28 | fi | ||
| 29 | fi | ||
| 30 | |||
| 31 | echo "#ifndef __BUILD_VERSION" > .build-version.h | ||
| 32 | echo "#define __BUILD_VERSION" >> .build-version.h | ||
| 33 | echo "#define BTRFS_BUILD_VERSION \"Btrfs $v\"" >> .build-version.h | ||
| 34 | echo "#endif" >> .build-version.h | ||
| 35 | |||
| 36 | diff -q version.h .build-version.h >& /dev/null | ||
| 37 | |||
| 38 | if [ $? == 0 ]; then | ||
| 39 | rm .build-version.h | ||
| 40 | exit 0 | ||
| 41 | fi | ||
| 42 | |||
| 43 | mv .build-version.h version.h | ||
diff --git a/fs/btrfs/volumes.c b/fs/btrfs/volumes.c new file mode 100644 index 000000000000..b187b537888e --- /dev/null +++ b/fs/btrfs/volumes.c | |||
| @@ -0,0 +1,3218 @@ | |||
| 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/sched.h> | ||
| 19 | #include <linux/bio.h> | ||
| 20 | #include <linux/buffer_head.h> | ||
| 21 | #include <linux/blkdev.h> | ||
| 22 | #include <linux/random.h> | ||
| 23 | #include <linux/version.h> | ||
| 24 | #include <asm/div64.h> | ||
| 25 | #include "compat.h" | ||
| 26 | #include "ctree.h" | ||
| 27 | #include "extent_map.h" | ||
| 28 | #include "disk-io.h" | ||
| 29 | #include "transaction.h" | ||
| 30 | #include "print-tree.h" | ||
| 31 | #include "volumes.h" | ||
| 32 | #include "async-thread.h" | ||
| 33 | |||
| 34 | struct map_lookup { | ||
| 35 | u64 type; | ||
| 36 | int io_align; | ||
| 37 | int io_width; | ||
| 38 | int stripe_len; | ||
| 39 | int sector_size; | ||
| 40 | int num_stripes; | ||
| 41 | int sub_stripes; | ||
| 42 | struct btrfs_bio_stripe stripes[]; | ||
| 43 | }; | ||
| 44 | |||
| 45 | static int init_first_rw_device(struct btrfs_trans_handle *trans, | ||
| 46 | struct btrfs_root *root, | ||
| 47 | struct btrfs_device *device); | ||
| 48 | static int btrfs_relocate_sys_chunks(struct btrfs_root *root); | ||
| 49 | |||
| 50 | #define map_lookup_size(n) (sizeof(struct map_lookup) + \ | ||
| 51 | (sizeof(struct btrfs_bio_stripe) * (n))) | ||
| 52 | |||
| 53 | static DEFINE_MUTEX(uuid_mutex); | ||
| 54 | static LIST_HEAD(fs_uuids); | ||
| 55 | |||
| 56 | void btrfs_lock_volumes(void) | ||
| 57 | { | ||
| 58 | mutex_lock(&uuid_mutex); | ||
| 59 | } | ||
| 60 | |||
| 61 | void btrfs_unlock_volumes(void) | ||
| 62 | { | ||
| 63 | mutex_unlock(&uuid_mutex); | ||
| 64 | } | ||
| 65 | |||
| 66 | static void lock_chunks(struct btrfs_root *root) | ||
| 67 | { | ||
| 68 | mutex_lock(&root->fs_info->chunk_mutex); | ||
| 69 | } | ||
| 70 | |||
| 71 | static void unlock_chunks(struct btrfs_root *root) | ||
| 72 | { | ||
| 73 | mutex_unlock(&root->fs_info->chunk_mutex); | ||
| 74 | } | ||
| 75 | |||
| 76 | static void free_fs_devices(struct btrfs_fs_devices *fs_devices) | ||
| 77 | { | ||
| 78 | struct btrfs_device *device; | ||
| 79 | WARN_ON(fs_devices->opened); | ||
| 80 | while (!list_empty(&fs_devices->devices)) { | ||
| 81 | device = list_entry(fs_devices->devices.next, | ||
| 82 | struct btrfs_device, dev_list); | ||
| 83 | list_del(&device->dev_list); | ||
| 84 | kfree(device->name); | ||
| 85 | kfree(device); | ||
| 86 | } | ||
| 87 | kfree(fs_devices); | ||
| 88 | } | ||
| 89 | |||
| 90 | int btrfs_cleanup_fs_uuids(void) | ||
| 91 | { | ||
| 92 | struct btrfs_fs_devices *fs_devices; | ||
| 93 | |||
| 94 | while (!list_empty(&fs_uuids)) { | ||
| 95 | fs_devices = list_entry(fs_uuids.next, | ||
| 96 | struct btrfs_fs_devices, list); | ||
| 97 | list_del(&fs_devices->list); | ||
| 98 | free_fs_devices(fs_devices); | ||
| 99 | } | ||
| 100 | return 0; | ||
| 101 | } | ||
| 102 | |||
| 103 | static noinline struct btrfs_device *__find_device(struct list_head *head, | ||
| 104 | u64 devid, u8 *uuid) | ||
| 105 | { | ||
| 106 | struct btrfs_device *dev; | ||
| 107 | struct list_head *cur; | ||
| 108 | |||
| 109 | list_for_each(cur, head) { | ||
| 110 | dev = list_entry(cur, struct btrfs_device, dev_list); | ||
| 111 | if (dev->devid == devid && | ||
| 112 | (!uuid || !memcmp(dev->uuid, uuid, BTRFS_UUID_SIZE))) { | ||
| 113 | return dev; | ||
| 114 | } | ||
| 115 | } | ||
| 116 | return NULL; | ||
| 117 | } | ||
| 118 | |||
| 119 | static noinline struct btrfs_fs_devices *find_fsid(u8 *fsid) | ||
| 120 | { | ||
| 121 | struct list_head *cur; | ||
| 122 | struct btrfs_fs_devices *fs_devices; | ||
| 123 | |||
| 124 | list_for_each(cur, &fs_uuids) { | ||
| 125 | fs_devices = list_entry(cur, struct btrfs_fs_devices, list); | ||
| 126 | if (memcmp(fsid, fs_devices->fsid, BTRFS_FSID_SIZE) == 0) | ||
| 127 | return fs_devices; | ||
| 128 | } | ||
| 129 | return NULL; | ||
| 130 | } | ||
| 131 | |||
| 132 | /* | ||
| 133 | * we try to collect pending bios for a device so we don't get a large | ||
| 134 | * number of procs sending bios down to the same device. This greatly | ||
| 135 | * improves the schedulers ability to collect and merge the bios. | ||
| 136 | * | ||
| 137 | * But, it also turns into a long list of bios to process and that is sure | ||
| 138 | * to eventually make the worker thread block. The solution here is to | ||
| 139 | * make some progress and then put this work struct back at the end of | ||
| 140 | * the list if the block device is congested. This way, multiple devices | ||
| 141 | * can make progress from a single worker thread. | ||
| 142 | */ | ||
| 143 | static noinline int run_scheduled_bios(struct btrfs_device *device) | ||
| 144 | { | ||
| 145 | struct bio *pending; | ||
| 146 | struct backing_dev_info *bdi; | ||
| 147 | struct btrfs_fs_info *fs_info; | ||
| 148 | struct bio *tail; | ||
| 149 | struct bio *cur; | ||
| 150 | int again = 0; | ||
| 151 | unsigned long num_run = 0; | ||
| 152 | unsigned long limit; | ||
| 153 | |||
| 154 | bdi = device->bdev->bd_inode->i_mapping->backing_dev_info; | ||
| 155 | fs_info = device->dev_root->fs_info; | ||
| 156 | limit = btrfs_async_submit_limit(fs_info); | ||
| 157 | limit = limit * 2 / 3; | ||
| 158 | |||
| 159 | loop: | ||
| 160 | spin_lock(&device->io_lock); | ||
| 161 | |||
| 162 | /* take all the bios off the list at once and process them | ||
| 163 | * later on (without the lock held). But, remember the | ||
| 164 | * tail and other pointers so the bios can be properly reinserted | ||
| 165 | * into the list if we hit congestion | ||
| 166 | */ | ||
| 167 | pending = device->pending_bios; | ||
| 168 | tail = device->pending_bio_tail; | ||
| 169 | WARN_ON(pending && !tail); | ||
| 170 | device->pending_bios = NULL; | ||
| 171 | device->pending_bio_tail = NULL; | ||
| 172 | |||
| 173 | /* | ||
| 174 | * if pending was null this time around, no bios need processing | ||
| 175 | * at all and we can stop. Otherwise it'll loop back up again | ||
| 176 | * and do an additional check so no bios are missed. | ||
| 177 | * | ||
| 178 | * device->running_pending is used to synchronize with the | ||
| 179 | * schedule_bio code. | ||
| 180 | */ | ||
| 181 | if (pending) { | ||
| 182 | again = 1; | ||
| 183 | device->running_pending = 1; | ||
| 184 | } else { | ||
| 185 | again = 0; | ||
| 186 | device->running_pending = 0; | ||
| 187 | } | ||
| 188 | spin_unlock(&device->io_lock); | ||
| 189 | |||
| 190 | while (pending) { | ||
| 191 | cur = pending; | ||
| 192 | pending = pending->bi_next; | ||
| 193 | cur->bi_next = NULL; | ||
| 194 | atomic_dec(&fs_info->nr_async_bios); | ||
| 195 | |||
| 196 | if (atomic_read(&fs_info->nr_async_bios) < limit && | ||
| 197 | waitqueue_active(&fs_info->async_submit_wait)) | ||
| 198 | wake_up(&fs_info->async_submit_wait); | ||
| 199 | |||
| 200 | BUG_ON(atomic_read(&cur->bi_cnt) == 0); | ||
| 201 | bio_get(cur); | ||
| 202 | submit_bio(cur->bi_rw, cur); | ||
| 203 | bio_put(cur); | ||
| 204 | num_run++; | ||
| 205 | |||
| 206 | /* | ||
| 207 | * we made progress, there is more work to do and the bdi | ||
| 208 | * is now congested. Back off and let other work structs | ||
| 209 | * run instead | ||
| 210 | */ | ||
| 211 | if (pending && bdi_write_congested(bdi) && | ||
| 212 | fs_info->fs_devices->open_devices > 1) { | ||
| 213 | struct bio *old_head; | ||
| 214 | |||
| 215 | spin_lock(&device->io_lock); | ||
| 216 | |||
| 217 | old_head = device->pending_bios; | ||
| 218 | device->pending_bios = pending; | ||
| 219 | if (device->pending_bio_tail) | ||
| 220 | tail->bi_next = old_head; | ||
| 221 | else | ||
| 222 | device->pending_bio_tail = tail; | ||
| 223 | |||
| 224 | spin_unlock(&device->io_lock); | ||
| 225 | btrfs_requeue_work(&device->work); | ||
| 226 | goto done; | ||
| 227 | } | ||
| 228 | } | ||
| 229 | if (again) | ||
| 230 | goto loop; | ||
| 231 | done: | ||
| 232 | return 0; | ||
| 233 | } | ||
| 234 | |||
| 235 | static void pending_bios_fn(struct btrfs_work *work) | ||
| 236 | { | ||
| 237 | struct btrfs_device *device; | ||
| 238 | |||
| 239 | device = container_of(work, struct btrfs_device, work); | ||
| 240 | run_scheduled_bios(device); | ||
| 241 | } | ||
| 242 | |||
| 243 | static noinline int device_list_add(const char *path, | ||
| 244 | struct btrfs_super_block *disk_super, | ||
| 245 | u64 devid, struct btrfs_fs_devices **fs_devices_ret) | ||
| 246 | { | ||
| 247 | struct btrfs_device *device; | ||
| 248 | struct btrfs_fs_devices *fs_devices; | ||
| 249 | u64 found_transid = btrfs_super_generation(disk_super); | ||
| 250 | |||
| 251 | fs_devices = find_fsid(disk_super->fsid); | ||
| 252 | if (!fs_devices) { | ||
| 253 | fs_devices = kzalloc(sizeof(*fs_devices), GFP_NOFS); | ||
| 254 | if (!fs_devices) | ||
| 255 | return -ENOMEM; | ||
| 256 | INIT_LIST_HEAD(&fs_devices->devices); | ||
| 257 | INIT_LIST_HEAD(&fs_devices->alloc_list); | ||
| 258 | list_add(&fs_devices->list, &fs_uuids); | ||
| 259 | memcpy(fs_devices->fsid, disk_super->fsid, BTRFS_FSID_SIZE); | ||
| 260 | fs_devices->latest_devid = devid; | ||
| 261 | fs_devices->latest_trans = found_transid; | ||
| 262 | device = NULL; | ||
| 263 | } else { | ||
| 264 | device = __find_device(&fs_devices->devices, devid, | ||
| 265 | disk_super->dev_item.uuid); | ||
| 266 | } | ||
| 267 | if (!device) { | ||
| 268 | if (fs_devices->opened) | ||
| 269 | return -EBUSY; | ||
| 270 | |||
| 271 | device = kzalloc(sizeof(*device), GFP_NOFS); | ||
| 272 | if (!device) { | ||
| 273 | /* we can safely leave the fs_devices entry around */ | ||
| 274 | return -ENOMEM; | ||
| 275 | } | ||
| 276 | device->devid = devid; | ||
| 277 | device->work.func = pending_bios_fn; | ||
| 278 | memcpy(device->uuid, disk_super->dev_item.uuid, | ||
| 279 | BTRFS_UUID_SIZE); | ||
| 280 | device->barriers = 1; | ||
| 281 | spin_lock_init(&device->io_lock); | ||
| 282 | device->name = kstrdup(path, GFP_NOFS); | ||
| 283 | if (!device->name) { | ||
| 284 | kfree(device); | ||
| 285 | return -ENOMEM; | ||
| 286 | } | ||
| 287 | INIT_LIST_HEAD(&device->dev_alloc_list); | ||
| 288 | list_add(&device->dev_list, &fs_devices->devices); | ||
| 289 | device->fs_devices = fs_devices; | ||
| 290 | fs_devices->num_devices++; | ||
| 291 | } | ||
| 292 | |||
| 293 | if (found_transid > fs_devices->latest_trans) { | ||
| 294 | fs_devices->latest_devid = devid; | ||
| 295 | fs_devices->latest_trans = found_transid; | ||
| 296 | } | ||
| 297 | *fs_devices_ret = fs_devices; | ||
| 298 | return 0; | ||
| 299 | } | ||
| 300 | |||
| 301 | static struct btrfs_fs_devices *clone_fs_devices(struct btrfs_fs_devices *orig) | ||
| 302 | { | ||
| 303 | struct btrfs_fs_devices *fs_devices; | ||
| 304 | struct btrfs_device *device; | ||
| 305 | struct btrfs_device *orig_dev; | ||
| 306 | |||
| 307 | fs_devices = kzalloc(sizeof(*fs_devices), GFP_NOFS); | ||
| 308 | if (!fs_devices) | ||
| 309 | return ERR_PTR(-ENOMEM); | ||
| 310 | |||
| 311 | INIT_LIST_HEAD(&fs_devices->devices); | ||
| 312 | INIT_LIST_HEAD(&fs_devices->alloc_list); | ||
| 313 | INIT_LIST_HEAD(&fs_devices->list); | ||
| 314 | fs_devices->latest_devid = orig->latest_devid; | ||
| 315 | fs_devices->latest_trans = orig->latest_trans; | ||
| 316 | memcpy(fs_devices->fsid, orig->fsid, sizeof(fs_devices->fsid)); | ||
| 317 | |||
| 318 | list_for_each_entry(orig_dev, &orig->devices, dev_list) { | ||
| 319 | device = kzalloc(sizeof(*device), GFP_NOFS); | ||
| 320 | if (!device) | ||
| 321 | goto error; | ||
| 322 | |||
| 323 | device->name = kstrdup(orig_dev->name, GFP_NOFS); | ||
| 324 | if (!device->name) | ||
| 325 | goto error; | ||
| 326 | |||
| 327 | device->devid = orig_dev->devid; | ||
| 328 | device->work.func = pending_bios_fn; | ||
| 329 | memcpy(device->uuid, orig_dev->uuid, sizeof(device->uuid)); | ||
| 330 | device->barriers = 1; | ||
| 331 | spin_lock_init(&device->io_lock); | ||
| 332 | INIT_LIST_HEAD(&device->dev_list); | ||
| 333 | INIT_LIST_HEAD(&device->dev_alloc_list); | ||
| 334 | |||
| 335 | list_add(&device->dev_list, &fs_devices->devices); | ||
| 336 | device->fs_devices = fs_devices; | ||
| 337 | fs_devices->num_devices++; | ||
| 338 | } | ||
| 339 | return fs_devices; | ||
| 340 | error: | ||
| 341 | free_fs_devices(fs_devices); | ||
| 342 | return ERR_PTR(-ENOMEM); | ||
| 343 | } | ||
| 344 | |||
| 345 | int btrfs_close_extra_devices(struct btrfs_fs_devices *fs_devices) | ||
| 346 | { | ||
| 347 | struct list_head *tmp; | ||
| 348 | struct list_head *cur; | ||
| 349 | struct btrfs_device *device; | ||
| 350 | |||
| 351 | mutex_lock(&uuid_mutex); | ||
| 352 | again: | ||
| 353 | list_for_each_safe(cur, tmp, &fs_devices->devices) { | ||
| 354 | device = list_entry(cur, struct btrfs_device, dev_list); | ||
| 355 | if (device->in_fs_metadata) | ||
| 356 | continue; | ||
| 357 | |||
| 358 | if (device->bdev) { | ||
| 359 | close_bdev_exclusive(device->bdev, device->mode); | ||
| 360 | device->bdev = NULL; | ||
| 361 | fs_devices->open_devices--; | ||
| 362 | } | ||
| 363 | if (device->writeable) { | ||
| 364 | list_del_init(&device->dev_alloc_list); | ||
| 365 | device->writeable = 0; | ||
| 366 | fs_devices->rw_devices--; | ||
| 367 | } | ||
| 368 | list_del_init(&device->dev_list); | ||
| 369 | fs_devices->num_devices--; | ||
| 370 | kfree(device->name); | ||
| 371 | kfree(device); | ||
| 372 | } | ||
| 373 | |||
| 374 | if (fs_devices->seed) { | ||
| 375 | fs_devices = fs_devices->seed; | ||
| 376 | goto again; | ||
| 377 | } | ||
| 378 | |||
| 379 | mutex_unlock(&uuid_mutex); | ||
| 380 | return 0; | ||
| 381 | } | ||
| 382 | |||
| 383 | static int __btrfs_close_devices(struct btrfs_fs_devices *fs_devices) | ||
| 384 | { | ||
| 385 | struct list_head *cur; | ||
| 386 | struct btrfs_device *device; | ||
| 387 | |||
| 388 | if (--fs_devices->opened > 0) | ||
| 389 | return 0; | ||
| 390 | |||
| 391 | list_for_each(cur, &fs_devices->devices) { | ||
| 392 | device = list_entry(cur, struct btrfs_device, dev_list); | ||
| 393 | if (device->bdev) { | ||
| 394 | close_bdev_exclusive(device->bdev, device->mode); | ||
| 395 | fs_devices->open_devices--; | ||
| 396 | } | ||
| 397 | if (device->writeable) { | ||
| 398 | list_del_init(&device->dev_alloc_list); | ||
| 399 | fs_devices->rw_devices--; | ||
| 400 | } | ||
| 401 | |||
| 402 | device->bdev = NULL; | ||
| 403 | device->writeable = 0; | ||
| 404 | device->in_fs_metadata = 0; | ||
| 405 | } | ||
| 406 | WARN_ON(fs_devices->open_devices); | ||
| 407 | WARN_ON(fs_devices->rw_devices); | ||
| 408 | fs_devices->opened = 0; | ||
| 409 | fs_devices->seeding = 0; | ||
| 410 | |||
| 411 | return 0; | ||
| 412 | } | ||
| 413 | |||
| 414 | int btrfs_close_devices(struct btrfs_fs_devices *fs_devices) | ||
| 415 | { | ||
| 416 | struct btrfs_fs_devices *seed_devices = NULL; | ||
| 417 | int ret; | ||
| 418 | |||
| 419 | mutex_lock(&uuid_mutex); | ||
| 420 | ret = __btrfs_close_devices(fs_devices); | ||
| 421 | if (!fs_devices->opened) { | ||
| 422 | seed_devices = fs_devices->seed; | ||
| 423 | fs_devices->seed = NULL; | ||
| 424 | } | ||
| 425 | mutex_unlock(&uuid_mutex); | ||
| 426 | |||
| 427 | while (seed_devices) { | ||
| 428 | fs_devices = seed_devices; | ||
| 429 | seed_devices = fs_devices->seed; | ||
| 430 | __btrfs_close_devices(fs_devices); | ||
| 431 | free_fs_devices(fs_devices); | ||
| 432 | } | ||
| 433 | return ret; | ||
| 434 | } | ||
| 435 | |||
| 436 | static int __btrfs_open_devices(struct btrfs_fs_devices *fs_devices, | ||
| 437 | fmode_t flags, void *holder) | ||
| 438 | { | ||
| 439 | struct block_device *bdev; | ||
| 440 | struct list_head *head = &fs_devices->devices; | ||
| 441 | struct list_head *cur; | ||
| 442 | struct btrfs_device *device; | ||
| 443 | struct block_device *latest_bdev = NULL; | ||
| 444 | struct buffer_head *bh; | ||
| 445 | struct btrfs_super_block *disk_super; | ||
| 446 | u64 latest_devid = 0; | ||
| 447 | u64 latest_transid = 0; | ||
| 448 | u64 devid; | ||
| 449 | int seeding = 1; | ||
| 450 | int ret = 0; | ||
| 451 | |||
| 452 | list_for_each(cur, head) { | ||
| 453 | device = list_entry(cur, struct btrfs_device, dev_list); | ||
| 454 | if (device->bdev) | ||
| 455 | continue; | ||
| 456 | if (!device->name) | ||
| 457 | continue; | ||
| 458 | |||
| 459 | bdev = open_bdev_exclusive(device->name, flags, holder); | ||
| 460 | if (IS_ERR(bdev)) { | ||
| 461 | printk(KERN_INFO "open %s failed\n", device->name); | ||
| 462 | goto error; | ||
| 463 | } | ||
| 464 | set_blocksize(bdev, 4096); | ||
| 465 | |||
| 466 | bh = btrfs_read_dev_super(bdev); | ||
| 467 | if (!bh) | ||
| 468 | goto error_close; | ||
| 469 | |||
| 470 | disk_super = (struct btrfs_super_block *)bh->b_data; | ||
| 471 | devid = le64_to_cpu(disk_super->dev_item.devid); | ||
| 472 | if (devid != device->devid) | ||
| 473 | goto error_brelse; | ||
| 474 | |||
| 475 | if (memcmp(device->uuid, disk_super->dev_item.uuid, | ||
| 476 | BTRFS_UUID_SIZE)) | ||
| 477 | goto error_brelse; | ||
| 478 | |||
| 479 | device->generation = btrfs_super_generation(disk_super); | ||
| 480 | if (!latest_transid || device->generation > latest_transid) { | ||
| 481 | latest_devid = devid; | ||
| 482 | latest_transid = device->generation; | ||
| 483 | latest_bdev = bdev; | ||
| 484 | } | ||
| 485 | |||
| 486 | if (btrfs_super_flags(disk_super) & BTRFS_SUPER_FLAG_SEEDING) { | ||
| 487 | device->writeable = 0; | ||
| 488 | } else { | ||
| 489 | device->writeable = !bdev_read_only(bdev); | ||
| 490 | seeding = 0; | ||
| 491 | } | ||
| 492 | |||
| 493 | device->bdev = bdev; | ||
| 494 | device->in_fs_metadata = 0; | ||
| 495 | device->mode = flags; | ||
| 496 | |||
| 497 | fs_devices->open_devices++; | ||
| 498 | if (device->writeable) { | ||
| 499 | fs_devices->rw_devices++; | ||
| 500 | list_add(&device->dev_alloc_list, | ||
| 501 | &fs_devices->alloc_list); | ||
| 502 | } | ||
| 503 | continue; | ||
| 504 | |||
| 505 | error_brelse: | ||
| 506 | brelse(bh); | ||
| 507 | error_close: | ||
| 508 | close_bdev_exclusive(bdev, FMODE_READ); | ||
| 509 | error: | ||
| 510 | continue; | ||
| 511 | } | ||
| 512 | if (fs_devices->open_devices == 0) { | ||
| 513 | ret = -EIO; | ||
| 514 | goto out; | ||
| 515 | } | ||
| 516 | fs_devices->seeding = seeding; | ||
| 517 | fs_devices->opened = 1; | ||
| 518 | fs_devices->latest_bdev = latest_bdev; | ||
| 519 | fs_devices->latest_devid = latest_devid; | ||
| 520 | fs_devices->latest_trans = latest_transid; | ||
| 521 | fs_devices->total_rw_bytes = 0; | ||
| 522 | out: | ||
| 523 | return ret; | ||
| 524 | } | ||
| 525 | |||
| 526 | int btrfs_open_devices(struct btrfs_fs_devices *fs_devices, | ||
| 527 | fmode_t flags, void *holder) | ||
| 528 | { | ||
| 529 | int ret; | ||
| 530 | |||
| 531 | mutex_lock(&uuid_mutex); | ||
| 532 | if (fs_devices->opened) { | ||
| 533 | fs_devices->opened++; | ||
| 534 | ret = 0; | ||
| 535 | } else { | ||
| 536 | ret = __btrfs_open_devices(fs_devices, flags, holder); | ||
| 537 | } | ||
| 538 | mutex_unlock(&uuid_mutex); | ||
| 539 | return ret; | ||
| 540 | } | ||
| 541 | |||
| 542 | int btrfs_scan_one_device(const char *path, fmode_t flags, void *holder, | ||
| 543 | struct btrfs_fs_devices **fs_devices_ret) | ||
| 544 | { | ||
| 545 | struct btrfs_super_block *disk_super; | ||
| 546 | struct block_device *bdev; | ||
| 547 | struct buffer_head *bh; | ||
| 548 | int ret; | ||
| 549 | u64 devid; | ||
| 550 | u64 transid; | ||
| 551 | |||
| 552 | mutex_lock(&uuid_mutex); | ||
| 553 | |||
| 554 | bdev = open_bdev_exclusive(path, flags, holder); | ||
| 555 | |||
| 556 | if (IS_ERR(bdev)) { | ||
| 557 | ret = PTR_ERR(bdev); | ||
| 558 | goto error; | ||
| 559 | } | ||
| 560 | |||
| 561 | ret = set_blocksize(bdev, 4096); | ||
| 562 | if (ret) | ||
| 563 | goto error_close; | ||
| 564 | bh = btrfs_read_dev_super(bdev); | ||
| 565 | if (!bh) { | ||
| 566 | ret = -EIO; | ||
| 567 | goto error_close; | ||
| 568 | } | ||
| 569 | disk_super = (struct btrfs_super_block *)bh->b_data; | ||
| 570 | devid = le64_to_cpu(disk_super->dev_item.devid); | ||
| 571 | transid = btrfs_super_generation(disk_super); | ||
| 572 | if (disk_super->label[0]) | ||
| 573 | printk(KERN_INFO "device label %s ", disk_super->label); | ||
| 574 | else { | ||
| 575 | /* FIXME, make a readl uuid parser */ | ||
| 576 | printk(KERN_INFO "device fsid %llx-%llx ", | ||
| 577 | *(unsigned long long *)disk_super->fsid, | ||
| 578 | *(unsigned long long *)(disk_super->fsid + 8)); | ||
| 579 | } | ||
| 580 | printk(KERN_INFO "devid %llu transid %llu %s\n", | ||
| 581 | (unsigned long long)devid, (unsigned long long)transid, path); | ||
| 582 | ret = device_list_add(path, disk_super, devid, fs_devices_ret); | ||
| 583 | |||
| 584 | brelse(bh); | ||
| 585 | error_close: | ||
| 586 | close_bdev_exclusive(bdev, flags); | ||
| 587 | error: | ||
| 588 | mutex_unlock(&uuid_mutex); | ||
| 589 | return ret; | ||
| 590 | } | ||
| 591 | |||
| 592 | /* | ||
| 593 | * this uses a pretty simple search, the expectation is that it is | ||
| 594 | * called very infrequently and that a given device has a small number | ||
| 595 | * of extents | ||
| 596 | */ | ||
| 597 | static noinline int find_free_dev_extent(struct btrfs_trans_handle *trans, | ||
| 598 | struct btrfs_device *device, | ||
| 599 | u64 num_bytes, u64 *start) | ||
| 600 | { | ||
| 601 | struct btrfs_key key; | ||
| 602 | struct btrfs_root *root = device->dev_root; | ||
| 603 | struct btrfs_dev_extent *dev_extent = NULL; | ||
| 604 | struct btrfs_path *path; | ||
| 605 | u64 hole_size = 0; | ||
| 606 | u64 last_byte = 0; | ||
| 607 | u64 search_start = 0; | ||
| 608 | u64 search_end = device->total_bytes; | ||
| 609 | int ret; | ||
| 610 | int slot = 0; | ||
| 611 | int start_found; | ||
| 612 | struct extent_buffer *l; | ||
| 613 | |||
| 614 | path = btrfs_alloc_path(); | ||
| 615 | if (!path) | ||
| 616 | return -ENOMEM; | ||
| 617 | path->reada = 2; | ||
| 618 | start_found = 0; | ||
| 619 | |||
| 620 | /* FIXME use last free of some kind */ | ||
| 621 | |||
| 622 | /* we don't want to overwrite the superblock on the drive, | ||
| 623 | * so we make sure to start at an offset of at least 1MB | ||
| 624 | */ | ||
| 625 | search_start = max((u64)1024 * 1024, search_start); | ||
| 626 | |||
| 627 | if (root->fs_info->alloc_start + num_bytes <= device->total_bytes) | ||
| 628 | search_start = max(root->fs_info->alloc_start, search_start); | ||
| 629 | |||
| 630 | key.objectid = device->devid; | ||
| 631 | key.offset = search_start; | ||
| 632 | key.type = BTRFS_DEV_EXTENT_KEY; | ||
| 633 | ret = btrfs_search_slot(trans, root, &key, path, 0, 0); | ||
| 634 | if (ret < 0) | ||
| 635 | goto error; | ||
| 636 | ret = btrfs_previous_item(root, path, 0, key.type); | ||
| 637 | if (ret < 0) | ||
| 638 | goto error; | ||
| 639 | l = path->nodes[0]; | ||
| 640 | btrfs_item_key_to_cpu(l, &key, path->slots[0]); | ||
| 641 | while (1) { | ||
| 642 | l = path->nodes[0]; | ||
| 643 | slot = path->slots[0]; | ||
| 644 | if (slot >= btrfs_header_nritems(l)) { | ||
| 645 | ret = btrfs_next_leaf(root, path); | ||
| 646 | if (ret == 0) | ||
| 647 | continue; | ||
| 648 | if (ret < 0) | ||
| 649 | goto error; | ||
| 650 | no_more_items: | ||
| 651 | if (!start_found) { | ||
| 652 | if (search_start >= search_end) { | ||
| 653 | ret = -ENOSPC; | ||
| 654 | goto error; | ||
| 655 | } | ||
| 656 | *start = search_start; | ||
| 657 | start_found = 1; | ||
| 658 | goto check_pending; | ||
| 659 | } | ||
| 660 | *start = last_byte > search_start ? | ||
| 661 | last_byte : search_start; | ||
| 662 | if (search_end <= *start) { | ||
| 663 | ret = -ENOSPC; | ||
| 664 | goto error; | ||
| 665 | } | ||
| 666 | goto check_pending; | ||
| 667 | } | ||
| 668 | btrfs_item_key_to_cpu(l, &key, slot); | ||
| 669 | |||
| 670 | if (key.objectid < device->devid) | ||
| 671 | goto next; | ||
| 672 | |||
| 673 | if (key.objectid > device->devid) | ||
| 674 | goto no_more_items; | ||
| 675 | |||
| 676 | if (key.offset >= search_start && key.offset > last_byte && | ||
| 677 | start_found) { | ||
| 678 | if (last_byte < search_start) | ||
| 679 | last_byte = search_start; | ||
| 680 | hole_size = key.offset - last_byte; | ||
| 681 | if (key.offset > last_byte && | ||
| 682 | hole_size >= num_bytes) { | ||
| 683 | *start = last_byte; | ||
| 684 | goto check_pending; | ||
| 685 | } | ||
| 686 | } | ||
| 687 | if (btrfs_key_type(&key) != BTRFS_DEV_EXTENT_KEY) | ||
| 688 | goto next; | ||
| 689 | |||
| 690 | start_found = 1; | ||
| 691 | dev_extent = btrfs_item_ptr(l, slot, struct btrfs_dev_extent); | ||
| 692 | last_byte = key.offset + btrfs_dev_extent_length(l, dev_extent); | ||
| 693 | next: | ||
| 694 | path->slots[0]++; | ||
| 695 | cond_resched(); | ||
| 696 | } | ||
| 697 | check_pending: | ||
| 698 | /* we have to make sure we didn't find an extent that has already | ||
| 699 | * been allocated by the map tree or the original allocation | ||
| 700 | */ | ||
| 701 | BUG_ON(*start < search_start); | ||
| 702 | |||
| 703 | if (*start + num_bytes > search_end) { | ||
| 704 | ret = -ENOSPC; | ||
| 705 | goto error; | ||
| 706 | } | ||
| 707 | /* check for pending inserts here */ | ||
| 708 | ret = 0; | ||
| 709 | |||
| 710 | error: | ||
| 711 | btrfs_free_path(path); | ||
| 712 | return ret; | ||
| 713 | } | ||
| 714 | |||
| 715 | static int btrfs_free_dev_extent(struct btrfs_trans_handle *trans, | ||
| 716 | struct btrfs_device *device, | ||
| 717 | u64 start) | ||
| 718 | { | ||
| 719 | int ret; | ||
| 720 | struct btrfs_path *path; | ||
| 721 | struct btrfs_root *root = device->dev_root; | ||
| 722 | struct btrfs_key key; | ||
| 723 | struct btrfs_key found_key; | ||
| 724 | struct extent_buffer *leaf = NULL; | ||
| 725 | struct btrfs_dev_extent *extent = NULL; | ||
| 726 | |||
| 727 | path = btrfs_alloc_path(); | ||
| 728 | if (!path) | ||
| 729 | return -ENOMEM; | ||
| 730 | |||
| 731 | key.objectid = device->devid; | ||
| 732 | key.offset = start; | ||
| 733 | key.type = BTRFS_DEV_EXTENT_KEY; | ||
| 734 | |||
| 735 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 736 | if (ret > 0) { | ||
| 737 | ret = btrfs_previous_item(root, path, key.objectid, | ||
| 738 | BTRFS_DEV_EXTENT_KEY); | ||
| 739 | BUG_ON(ret); | ||
| 740 | leaf = path->nodes[0]; | ||
| 741 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 742 | extent = btrfs_item_ptr(leaf, path->slots[0], | ||
| 743 | struct btrfs_dev_extent); | ||
| 744 | BUG_ON(found_key.offset > start || found_key.offset + | ||
| 745 | btrfs_dev_extent_length(leaf, extent) < start); | ||
| 746 | ret = 0; | ||
| 747 | } else if (ret == 0) { | ||
| 748 | leaf = path->nodes[0]; | ||
| 749 | extent = btrfs_item_ptr(leaf, path->slots[0], | ||
| 750 | struct btrfs_dev_extent); | ||
| 751 | } | ||
| 752 | BUG_ON(ret); | ||
| 753 | |||
| 754 | if (device->bytes_used > 0) | ||
| 755 | device->bytes_used -= btrfs_dev_extent_length(leaf, extent); | ||
| 756 | ret = btrfs_del_item(trans, root, path); | ||
| 757 | BUG_ON(ret); | ||
| 758 | |||
| 759 | btrfs_free_path(path); | ||
| 760 | return ret; | ||
| 761 | } | ||
| 762 | |||
| 763 | int btrfs_alloc_dev_extent(struct btrfs_trans_handle *trans, | ||
| 764 | struct btrfs_device *device, | ||
| 765 | u64 chunk_tree, u64 chunk_objectid, | ||
| 766 | u64 chunk_offset, u64 start, u64 num_bytes) | ||
| 767 | { | ||
| 768 | int ret; | ||
| 769 | struct btrfs_path *path; | ||
| 770 | struct btrfs_root *root = device->dev_root; | ||
| 771 | struct btrfs_dev_extent *extent; | ||
| 772 | struct extent_buffer *leaf; | ||
| 773 | struct btrfs_key key; | ||
| 774 | |||
| 775 | WARN_ON(!device->in_fs_metadata); | ||
| 776 | path = btrfs_alloc_path(); | ||
| 777 | if (!path) | ||
| 778 | return -ENOMEM; | ||
| 779 | |||
| 780 | key.objectid = device->devid; | ||
| 781 | key.offset = start; | ||
| 782 | key.type = BTRFS_DEV_EXTENT_KEY; | ||
| 783 | ret = btrfs_insert_empty_item(trans, root, path, &key, | ||
| 784 | sizeof(*extent)); | ||
| 785 | BUG_ON(ret); | ||
| 786 | |||
| 787 | leaf = path->nodes[0]; | ||
| 788 | extent = btrfs_item_ptr(leaf, path->slots[0], | ||
| 789 | struct btrfs_dev_extent); | ||
| 790 | btrfs_set_dev_extent_chunk_tree(leaf, extent, chunk_tree); | ||
| 791 | btrfs_set_dev_extent_chunk_objectid(leaf, extent, chunk_objectid); | ||
| 792 | btrfs_set_dev_extent_chunk_offset(leaf, extent, chunk_offset); | ||
| 793 | |||
| 794 | write_extent_buffer(leaf, root->fs_info->chunk_tree_uuid, | ||
| 795 | (unsigned long)btrfs_dev_extent_chunk_tree_uuid(extent), | ||
| 796 | BTRFS_UUID_SIZE); | ||
| 797 | |||
| 798 | btrfs_set_dev_extent_length(leaf, extent, num_bytes); | ||
| 799 | btrfs_mark_buffer_dirty(leaf); | ||
| 800 | btrfs_free_path(path); | ||
| 801 | return ret; | ||
| 802 | } | ||
| 803 | |||
| 804 | static noinline int find_next_chunk(struct btrfs_root *root, | ||
| 805 | u64 objectid, u64 *offset) | ||
| 806 | { | ||
| 807 | struct btrfs_path *path; | ||
| 808 | int ret; | ||
| 809 | struct btrfs_key key; | ||
| 810 | struct btrfs_chunk *chunk; | ||
| 811 | struct btrfs_key found_key; | ||
| 812 | |||
| 813 | path = btrfs_alloc_path(); | ||
| 814 | BUG_ON(!path); | ||
| 815 | |||
| 816 | key.objectid = objectid; | ||
| 817 | key.offset = (u64)-1; | ||
| 818 | key.type = BTRFS_CHUNK_ITEM_KEY; | ||
| 819 | |||
| 820 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 821 | if (ret < 0) | ||
| 822 | goto error; | ||
| 823 | |||
| 824 | BUG_ON(ret == 0); | ||
| 825 | |||
| 826 | ret = btrfs_previous_item(root, path, 0, BTRFS_CHUNK_ITEM_KEY); | ||
| 827 | if (ret) { | ||
| 828 | *offset = 0; | ||
| 829 | } else { | ||
| 830 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 831 | path->slots[0]); | ||
| 832 | if (found_key.objectid != objectid) | ||
| 833 | *offset = 0; | ||
| 834 | else { | ||
| 835 | chunk = btrfs_item_ptr(path->nodes[0], path->slots[0], | ||
| 836 | struct btrfs_chunk); | ||
| 837 | *offset = found_key.offset + | ||
| 838 | btrfs_chunk_length(path->nodes[0], chunk); | ||
| 839 | } | ||
| 840 | } | ||
| 841 | ret = 0; | ||
| 842 | error: | ||
| 843 | btrfs_free_path(path); | ||
| 844 | return ret; | ||
| 845 | } | ||
| 846 | |||
| 847 | static noinline int find_next_devid(struct btrfs_root *root, u64 *objectid) | ||
| 848 | { | ||
| 849 | int ret; | ||
| 850 | struct btrfs_key key; | ||
| 851 | struct btrfs_key found_key; | ||
| 852 | struct btrfs_path *path; | ||
| 853 | |||
| 854 | root = root->fs_info->chunk_root; | ||
| 855 | |||
| 856 | path = btrfs_alloc_path(); | ||
| 857 | if (!path) | ||
| 858 | return -ENOMEM; | ||
| 859 | |||
| 860 | key.objectid = BTRFS_DEV_ITEMS_OBJECTID; | ||
| 861 | key.type = BTRFS_DEV_ITEM_KEY; | ||
| 862 | key.offset = (u64)-1; | ||
| 863 | |||
| 864 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 865 | if (ret < 0) | ||
| 866 | goto error; | ||
| 867 | |||
| 868 | BUG_ON(ret == 0); | ||
| 869 | |||
| 870 | ret = btrfs_previous_item(root, path, BTRFS_DEV_ITEMS_OBJECTID, | ||
| 871 | BTRFS_DEV_ITEM_KEY); | ||
| 872 | if (ret) { | ||
| 873 | *objectid = 1; | ||
| 874 | } else { | ||
| 875 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 876 | path->slots[0]); | ||
| 877 | *objectid = found_key.offset + 1; | ||
| 878 | } | ||
| 879 | ret = 0; | ||
| 880 | error: | ||
| 881 | btrfs_free_path(path); | ||
| 882 | return ret; | ||
| 883 | } | ||
| 884 | |||
| 885 | /* | ||
| 886 | * the device information is stored in the chunk root | ||
| 887 | * the btrfs_device struct should be fully filled in | ||
| 888 | */ | ||
| 889 | int btrfs_add_device(struct btrfs_trans_handle *trans, | ||
| 890 | struct btrfs_root *root, | ||
| 891 | struct btrfs_device *device) | ||
| 892 | { | ||
| 893 | int ret; | ||
| 894 | struct btrfs_path *path; | ||
| 895 | struct btrfs_dev_item *dev_item; | ||
| 896 | struct extent_buffer *leaf; | ||
| 897 | struct btrfs_key key; | ||
| 898 | unsigned long ptr; | ||
| 899 | |||
| 900 | root = root->fs_info->chunk_root; | ||
| 901 | |||
| 902 | path = btrfs_alloc_path(); | ||
| 903 | if (!path) | ||
| 904 | return -ENOMEM; | ||
| 905 | |||
| 906 | key.objectid = BTRFS_DEV_ITEMS_OBJECTID; | ||
| 907 | key.type = BTRFS_DEV_ITEM_KEY; | ||
| 908 | key.offset = device->devid; | ||
| 909 | |||
| 910 | ret = btrfs_insert_empty_item(trans, root, path, &key, | ||
| 911 | sizeof(*dev_item)); | ||
| 912 | if (ret) | ||
| 913 | goto out; | ||
| 914 | |||
| 915 | leaf = path->nodes[0]; | ||
| 916 | dev_item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_dev_item); | ||
| 917 | |||
| 918 | btrfs_set_device_id(leaf, dev_item, device->devid); | ||
| 919 | btrfs_set_device_generation(leaf, dev_item, 0); | ||
| 920 | btrfs_set_device_type(leaf, dev_item, device->type); | ||
| 921 | btrfs_set_device_io_align(leaf, dev_item, device->io_align); | ||
| 922 | btrfs_set_device_io_width(leaf, dev_item, device->io_width); | ||
| 923 | btrfs_set_device_sector_size(leaf, dev_item, device->sector_size); | ||
| 924 | btrfs_set_device_total_bytes(leaf, dev_item, device->total_bytes); | ||
| 925 | btrfs_set_device_bytes_used(leaf, dev_item, device->bytes_used); | ||
| 926 | btrfs_set_device_group(leaf, dev_item, 0); | ||
| 927 | btrfs_set_device_seek_speed(leaf, dev_item, 0); | ||
| 928 | btrfs_set_device_bandwidth(leaf, dev_item, 0); | ||
| 929 | btrfs_set_device_start_offset(leaf, dev_item, 0); | ||
| 930 | |||
| 931 | ptr = (unsigned long)btrfs_device_uuid(dev_item); | ||
| 932 | write_extent_buffer(leaf, device->uuid, ptr, BTRFS_UUID_SIZE); | ||
| 933 | ptr = (unsigned long)btrfs_device_fsid(dev_item); | ||
| 934 | write_extent_buffer(leaf, root->fs_info->fsid, ptr, BTRFS_UUID_SIZE); | ||
| 935 | btrfs_mark_buffer_dirty(leaf); | ||
| 936 | |||
| 937 | ret = 0; | ||
| 938 | out: | ||
| 939 | btrfs_free_path(path); | ||
| 940 | return ret; | ||
| 941 | } | ||
| 942 | |||
| 943 | static int btrfs_rm_dev_item(struct btrfs_root *root, | ||
| 944 | struct btrfs_device *device) | ||
| 945 | { | ||
| 946 | int ret; | ||
| 947 | struct btrfs_path *path; | ||
| 948 | struct btrfs_key key; | ||
| 949 | struct btrfs_trans_handle *trans; | ||
| 950 | |||
| 951 | root = root->fs_info->chunk_root; | ||
| 952 | |||
| 953 | path = btrfs_alloc_path(); | ||
| 954 | if (!path) | ||
| 955 | return -ENOMEM; | ||
| 956 | |||
| 957 | trans = btrfs_start_transaction(root, 1); | ||
| 958 | key.objectid = BTRFS_DEV_ITEMS_OBJECTID; | ||
| 959 | key.type = BTRFS_DEV_ITEM_KEY; | ||
| 960 | key.offset = device->devid; | ||
| 961 | lock_chunks(root); | ||
| 962 | |||
| 963 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 964 | if (ret < 0) | ||
| 965 | goto out; | ||
| 966 | |||
| 967 | if (ret > 0) { | ||
| 968 | ret = -ENOENT; | ||
| 969 | goto out; | ||
| 970 | } | ||
| 971 | |||
| 972 | ret = btrfs_del_item(trans, root, path); | ||
| 973 | if (ret) | ||
| 974 | goto out; | ||
| 975 | out: | ||
| 976 | btrfs_free_path(path); | ||
| 977 | unlock_chunks(root); | ||
| 978 | btrfs_commit_transaction(trans, root); | ||
| 979 | return ret; | ||
| 980 | } | ||
| 981 | |||
| 982 | int btrfs_rm_device(struct btrfs_root *root, char *device_path) | ||
| 983 | { | ||
| 984 | struct btrfs_device *device; | ||
| 985 | struct btrfs_device *next_device; | ||
| 986 | struct block_device *bdev; | ||
| 987 | struct buffer_head *bh = NULL; | ||
| 988 | struct btrfs_super_block *disk_super; | ||
| 989 | u64 all_avail; | ||
| 990 | u64 devid; | ||
| 991 | u64 num_devices; | ||
| 992 | u8 *dev_uuid; | ||
| 993 | int ret = 0; | ||
| 994 | |||
| 995 | mutex_lock(&uuid_mutex); | ||
| 996 | mutex_lock(&root->fs_info->volume_mutex); | ||
| 997 | |||
| 998 | all_avail = root->fs_info->avail_data_alloc_bits | | ||
| 999 | root->fs_info->avail_system_alloc_bits | | ||
| 1000 | root->fs_info->avail_metadata_alloc_bits; | ||
| 1001 | |||
| 1002 | if ((all_avail & BTRFS_BLOCK_GROUP_RAID10) && | ||
| 1003 | root->fs_info->fs_devices->rw_devices <= 4) { | ||
| 1004 | printk(KERN_ERR "btrfs: unable to go below four devices " | ||
| 1005 | "on raid10\n"); | ||
| 1006 | ret = -EINVAL; | ||
| 1007 | goto out; | ||
| 1008 | } | ||
| 1009 | |||
| 1010 | if ((all_avail & BTRFS_BLOCK_GROUP_RAID1) && | ||
| 1011 | root->fs_info->fs_devices->rw_devices <= 2) { | ||
| 1012 | printk(KERN_ERR "btrfs: unable to go below two " | ||
| 1013 | "devices on raid1\n"); | ||
| 1014 | ret = -EINVAL; | ||
| 1015 | goto out; | ||
| 1016 | } | ||
| 1017 | |||
| 1018 | if (strcmp(device_path, "missing") == 0) { | ||
| 1019 | struct list_head *cur; | ||
| 1020 | struct list_head *devices; | ||
| 1021 | struct btrfs_device *tmp; | ||
| 1022 | |||
| 1023 | device = NULL; | ||
| 1024 | devices = &root->fs_info->fs_devices->devices; | ||
| 1025 | list_for_each(cur, devices) { | ||
| 1026 | tmp = list_entry(cur, struct btrfs_device, dev_list); | ||
| 1027 | if (tmp->in_fs_metadata && !tmp->bdev) { | ||
| 1028 | device = tmp; | ||
| 1029 | break; | ||
| 1030 | } | ||
| 1031 | } | ||
| 1032 | bdev = NULL; | ||
| 1033 | bh = NULL; | ||
| 1034 | disk_super = NULL; | ||
| 1035 | if (!device) { | ||
| 1036 | printk(KERN_ERR "btrfs: no missing devices found to " | ||
| 1037 | "remove\n"); | ||
| 1038 | goto out; | ||
| 1039 | } | ||
| 1040 | } else { | ||
| 1041 | bdev = open_bdev_exclusive(device_path, FMODE_READ, | ||
| 1042 | root->fs_info->bdev_holder); | ||
| 1043 | if (IS_ERR(bdev)) { | ||
| 1044 | ret = PTR_ERR(bdev); | ||
| 1045 | goto out; | ||
| 1046 | } | ||
| 1047 | |||
| 1048 | set_blocksize(bdev, 4096); | ||
| 1049 | bh = btrfs_read_dev_super(bdev); | ||
| 1050 | if (!bh) { | ||
| 1051 | ret = -EIO; | ||
| 1052 | goto error_close; | ||
| 1053 | } | ||
| 1054 | disk_super = (struct btrfs_super_block *)bh->b_data; | ||
| 1055 | devid = le64_to_cpu(disk_super->dev_item.devid); | ||
| 1056 | dev_uuid = disk_super->dev_item.uuid; | ||
| 1057 | device = btrfs_find_device(root, devid, dev_uuid, | ||
| 1058 | disk_super->fsid); | ||
| 1059 | if (!device) { | ||
| 1060 | ret = -ENOENT; | ||
| 1061 | goto error_brelse; | ||
| 1062 | } | ||
| 1063 | } | ||
| 1064 | |||
| 1065 | if (device->writeable && root->fs_info->fs_devices->rw_devices == 1) { | ||
| 1066 | printk(KERN_ERR "btrfs: unable to remove the only writeable " | ||
| 1067 | "device\n"); | ||
| 1068 | ret = -EINVAL; | ||
| 1069 | goto error_brelse; | ||
| 1070 | } | ||
| 1071 | |||
| 1072 | if (device->writeable) { | ||
| 1073 | list_del_init(&device->dev_alloc_list); | ||
| 1074 | root->fs_info->fs_devices->rw_devices--; | ||
| 1075 | } | ||
| 1076 | |||
| 1077 | ret = btrfs_shrink_device(device, 0); | ||
| 1078 | if (ret) | ||
| 1079 | goto error_brelse; | ||
| 1080 | |||
| 1081 | ret = btrfs_rm_dev_item(root->fs_info->chunk_root, device); | ||
| 1082 | if (ret) | ||
| 1083 | goto error_brelse; | ||
| 1084 | |||
| 1085 | device->in_fs_metadata = 0; | ||
| 1086 | list_del_init(&device->dev_list); | ||
| 1087 | device->fs_devices->num_devices--; | ||
| 1088 | |||
| 1089 | next_device = list_entry(root->fs_info->fs_devices->devices.next, | ||
| 1090 | struct btrfs_device, dev_list); | ||
| 1091 | if (device->bdev == root->fs_info->sb->s_bdev) | ||
| 1092 | root->fs_info->sb->s_bdev = next_device->bdev; | ||
| 1093 | if (device->bdev == root->fs_info->fs_devices->latest_bdev) | ||
| 1094 | root->fs_info->fs_devices->latest_bdev = next_device->bdev; | ||
| 1095 | |||
| 1096 | if (device->bdev) { | ||
| 1097 | close_bdev_exclusive(device->bdev, device->mode); | ||
| 1098 | device->bdev = NULL; | ||
| 1099 | device->fs_devices->open_devices--; | ||
| 1100 | } | ||
| 1101 | |||
| 1102 | num_devices = btrfs_super_num_devices(&root->fs_info->super_copy) - 1; | ||
| 1103 | btrfs_set_super_num_devices(&root->fs_info->super_copy, num_devices); | ||
| 1104 | |||
| 1105 | if (device->fs_devices->open_devices == 0) { | ||
| 1106 | struct btrfs_fs_devices *fs_devices; | ||
| 1107 | fs_devices = root->fs_info->fs_devices; | ||
| 1108 | while (fs_devices) { | ||
| 1109 | if (fs_devices->seed == device->fs_devices) | ||
| 1110 | break; | ||
| 1111 | fs_devices = fs_devices->seed; | ||
| 1112 | } | ||
| 1113 | fs_devices->seed = device->fs_devices->seed; | ||
| 1114 | device->fs_devices->seed = NULL; | ||
| 1115 | __btrfs_close_devices(device->fs_devices); | ||
| 1116 | free_fs_devices(device->fs_devices); | ||
| 1117 | } | ||
| 1118 | |||
| 1119 | /* | ||
| 1120 | * at this point, the device is zero sized. We want to | ||
| 1121 | * remove it from the devices list and zero out the old super | ||
| 1122 | */ | ||
| 1123 | if (device->writeable) { | ||
| 1124 | /* make sure this device isn't detected as part of | ||
| 1125 | * the FS anymore | ||
| 1126 | */ | ||
| 1127 | memset(&disk_super->magic, 0, sizeof(disk_super->magic)); | ||
| 1128 | set_buffer_dirty(bh); | ||
| 1129 | sync_dirty_buffer(bh); | ||
| 1130 | } | ||
| 1131 | |||
| 1132 | kfree(device->name); | ||
| 1133 | kfree(device); | ||
| 1134 | ret = 0; | ||
| 1135 | |||
| 1136 | error_brelse: | ||
| 1137 | brelse(bh); | ||
| 1138 | error_close: | ||
| 1139 | if (bdev) | ||
| 1140 | close_bdev_exclusive(bdev, FMODE_READ); | ||
| 1141 | out: | ||
| 1142 | mutex_unlock(&root->fs_info->volume_mutex); | ||
| 1143 | mutex_unlock(&uuid_mutex); | ||
| 1144 | return ret; | ||
| 1145 | } | ||
| 1146 | |||
| 1147 | /* | ||
| 1148 | * does all the dirty work required for changing file system's UUID. | ||
| 1149 | */ | ||
| 1150 | static int btrfs_prepare_sprout(struct btrfs_trans_handle *trans, | ||
| 1151 | struct btrfs_root *root) | ||
| 1152 | { | ||
| 1153 | struct btrfs_fs_devices *fs_devices = root->fs_info->fs_devices; | ||
| 1154 | struct btrfs_fs_devices *old_devices; | ||
| 1155 | struct btrfs_fs_devices *seed_devices; | ||
| 1156 | struct btrfs_super_block *disk_super = &root->fs_info->super_copy; | ||
| 1157 | struct btrfs_device *device; | ||
| 1158 | u64 super_flags; | ||
| 1159 | |||
| 1160 | BUG_ON(!mutex_is_locked(&uuid_mutex)); | ||
| 1161 | if (!fs_devices->seeding) | ||
| 1162 | return -EINVAL; | ||
| 1163 | |||
| 1164 | seed_devices = kzalloc(sizeof(*fs_devices), GFP_NOFS); | ||
| 1165 | if (!seed_devices) | ||
| 1166 | return -ENOMEM; | ||
| 1167 | |||
| 1168 | old_devices = clone_fs_devices(fs_devices); | ||
| 1169 | if (IS_ERR(old_devices)) { | ||
| 1170 | kfree(seed_devices); | ||
| 1171 | return PTR_ERR(old_devices); | ||
| 1172 | } | ||
| 1173 | |||
| 1174 | list_add(&old_devices->list, &fs_uuids); | ||
| 1175 | |||
| 1176 | memcpy(seed_devices, fs_devices, sizeof(*seed_devices)); | ||
| 1177 | seed_devices->opened = 1; | ||
| 1178 | INIT_LIST_HEAD(&seed_devices->devices); | ||
| 1179 | INIT_LIST_HEAD(&seed_devices->alloc_list); | ||
| 1180 | list_splice_init(&fs_devices->devices, &seed_devices->devices); | ||
| 1181 | list_splice_init(&fs_devices->alloc_list, &seed_devices->alloc_list); | ||
| 1182 | list_for_each_entry(device, &seed_devices->devices, dev_list) { | ||
| 1183 | device->fs_devices = seed_devices; | ||
| 1184 | } | ||
| 1185 | |||
| 1186 | fs_devices->seeding = 0; | ||
| 1187 | fs_devices->num_devices = 0; | ||
| 1188 | fs_devices->open_devices = 0; | ||
| 1189 | fs_devices->seed = seed_devices; | ||
| 1190 | |||
| 1191 | generate_random_uuid(fs_devices->fsid); | ||
| 1192 | memcpy(root->fs_info->fsid, fs_devices->fsid, BTRFS_FSID_SIZE); | ||
| 1193 | memcpy(disk_super->fsid, fs_devices->fsid, BTRFS_FSID_SIZE); | ||
| 1194 | super_flags = btrfs_super_flags(disk_super) & | ||
| 1195 | ~BTRFS_SUPER_FLAG_SEEDING; | ||
| 1196 | btrfs_set_super_flags(disk_super, super_flags); | ||
| 1197 | |||
| 1198 | return 0; | ||
| 1199 | } | ||
| 1200 | |||
| 1201 | /* | ||
| 1202 | * strore the expected generation for seed devices in device items. | ||
| 1203 | */ | ||
| 1204 | static int btrfs_finish_sprout(struct btrfs_trans_handle *trans, | ||
| 1205 | struct btrfs_root *root) | ||
| 1206 | { | ||
| 1207 | struct btrfs_path *path; | ||
| 1208 | struct extent_buffer *leaf; | ||
| 1209 | struct btrfs_dev_item *dev_item; | ||
| 1210 | struct btrfs_device *device; | ||
| 1211 | struct btrfs_key key; | ||
| 1212 | u8 fs_uuid[BTRFS_UUID_SIZE]; | ||
| 1213 | u8 dev_uuid[BTRFS_UUID_SIZE]; | ||
| 1214 | u64 devid; | ||
| 1215 | int ret; | ||
| 1216 | |||
| 1217 | path = btrfs_alloc_path(); | ||
| 1218 | if (!path) | ||
| 1219 | return -ENOMEM; | ||
| 1220 | |||
| 1221 | root = root->fs_info->chunk_root; | ||
| 1222 | key.objectid = BTRFS_DEV_ITEMS_OBJECTID; | ||
| 1223 | key.offset = 0; | ||
| 1224 | key.type = BTRFS_DEV_ITEM_KEY; | ||
| 1225 | |||
| 1226 | while (1) { | ||
| 1227 | ret = btrfs_search_slot(trans, root, &key, path, 0, 1); | ||
| 1228 | if (ret < 0) | ||
| 1229 | goto error; | ||
| 1230 | |||
| 1231 | leaf = path->nodes[0]; | ||
| 1232 | next_slot: | ||
| 1233 | if (path->slots[0] >= btrfs_header_nritems(leaf)) { | ||
| 1234 | ret = btrfs_next_leaf(root, path); | ||
| 1235 | if (ret > 0) | ||
| 1236 | break; | ||
| 1237 | if (ret < 0) | ||
| 1238 | goto error; | ||
| 1239 | leaf = path->nodes[0]; | ||
| 1240 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 1241 | btrfs_release_path(root, path); | ||
| 1242 | continue; | ||
| 1243 | } | ||
| 1244 | |||
| 1245 | btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); | ||
| 1246 | if (key.objectid != BTRFS_DEV_ITEMS_OBJECTID || | ||
| 1247 | key.type != BTRFS_DEV_ITEM_KEY) | ||
| 1248 | break; | ||
| 1249 | |||
| 1250 | dev_item = btrfs_item_ptr(leaf, path->slots[0], | ||
| 1251 | struct btrfs_dev_item); | ||
| 1252 | devid = btrfs_device_id(leaf, dev_item); | ||
| 1253 | read_extent_buffer(leaf, dev_uuid, | ||
| 1254 | (unsigned long)btrfs_device_uuid(dev_item), | ||
| 1255 | BTRFS_UUID_SIZE); | ||
| 1256 | read_extent_buffer(leaf, fs_uuid, | ||
| 1257 | (unsigned long)btrfs_device_fsid(dev_item), | ||
| 1258 | BTRFS_UUID_SIZE); | ||
| 1259 | device = btrfs_find_device(root, devid, dev_uuid, fs_uuid); | ||
| 1260 | BUG_ON(!device); | ||
| 1261 | |||
| 1262 | if (device->fs_devices->seeding) { | ||
| 1263 | btrfs_set_device_generation(leaf, dev_item, | ||
| 1264 | device->generation); | ||
| 1265 | btrfs_mark_buffer_dirty(leaf); | ||
| 1266 | } | ||
| 1267 | |||
| 1268 | path->slots[0]++; | ||
| 1269 | goto next_slot; | ||
| 1270 | } | ||
| 1271 | ret = 0; | ||
| 1272 | error: | ||
| 1273 | btrfs_free_path(path); | ||
| 1274 | return ret; | ||
| 1275 | } | ||
| 1276 | |||
| 1277 | int btrfs_init_new_device(struct btrfs_root *root, char *device_path) | ||
| 1278 | { | ||
| 1279 | struct btrfs_trans_handle *trans; | ||
| 1280 | struct btrfs_device *device; | ||
| 1281 | struct block_device *bdev; | ||
| 1282 | struct list_head *cur; | ||
| 1283 | struct list_head *devices; | ||
| 1284 | struct super_block *sb = root->fs_info->sb; | ||
| 1285 | u64 total_bytes; | ||
| 1286 | int seeding_dev = 0; | ||
| 1287 | int ret = 0; | ||
| 1288 | |||
| 1289 | if ((sb->s_flags & MS_RDONLY) && !root->fs_info->fs_devices->seeding) | ||
| 1290 | return -EINVAL; | ||
| 1291 | |||
| 1292 | bdev = open_bdev_exclusive(device_path, 0, root->fs_info->bdev_holder); | ||
| 1293 | if (!bdev) | ||
| 1294 | return -EIO; | ||
| 1295 | |||
| 1296 | if (root->fs_info->fs_devices->seeding) { | ||
| 1297 | seeding_dev = 1; | ||
| 1298 | down_write(&sb->s_umount); | ||
| 1299 | mutex_lock(&uuid_mutex); | ||
| 1300 | } | ||
| 1301 | |||
| 1302 | filemap_write_and_wait(bdev->bd_inode->i_mapping); | ||
| 1303 | mutex_lock(&root->fs_info->volume_mutex); | ||
| 1304 | |||
| 1305 | devices = &root->fs_info->fs_devices->devices; | ||
| 1306 | list_for_each(cur, devices) { | ||
| 1307 | device = list_entry(cur, struct btrfs_device, dev_list); | ||
| 1308 | if (device->bdev == bdev) { | ||
| 1309 | ret = -EEXIST; | ||
| 1310 | goto error; | ||
| 1311 | } | ||
| 1312 | } | ||
| 1313 | |||
| 1314 | device = kzalloc(sizeof(*device), GFP_NOFS); | ||
| 1315 | if (!device) { | ||
| 1316 | /* we can safely leave the fs_devices entry around */ | ||
| 1317 | ret = -ENOMEM; | ||
| 1318 | goto error; | ||
| 1319 | } | ||
| 1320 | |||
| 1321 | device->name = kstrdup(device_path, GFP_NOFS); | ||
| 1322 | if (!device->name) { | ||
| 1323 | kfree(device); | ||
| 1324 | ret = -ENOMEM; | ||
| 1325 | goto error; | ||
| 1326 | } | ||
| 1327 | |||
| 1328 | ret = find_next_devid(root, &device->devid); | ||
| 1329 | if (ret) { | ||
| 1330 | kfree(device); | ||
| 1331 | goto error; | ||
| 1332 | } | ||
| 1333 | |||
| 1334 | trans = btrfs_start_transaction(root, 1); | ||
| 1335 | lock_chunks(root); | ||
| 1336 | |||
| 1337 | device->barriers = 1; | ||
| 1338 | device->writeable = 1; | ||
| 1339 | device->work.func = pending_bios_fn; | ||
| 1340 | generate_random_uuid(device->uuid); | ||
| 1341 | spin_lock_init(&device->io_lock); | ||
| 1342 | device->generation = trans->transid; | ||
| 1343 | device->io_width = root->sectorsize; | ||
| 1344 | device->io_align = root->sectorsize; | ||
| 1345 | device->sector_size = root->sectorsize; | ||
| 1346 | device->total_bytes = i_size_read(bdev->bd_inode); | ||
| 1347 | device->dev_root = root->fs_info->dev_root; | ||
| 1348 | device->bdev = bdev; | ||
| 1349 | device->in_fs_metadata = 1; | ||
| 1350 | device->mode = 0; | ||
| 1351 | set_blocksize(device->bdev, 4096); | ||
| 1352 | |||
| 1353 | if (seeding_dev) { | ||
| 1354 | sb->s_flags &= ~MS_RDONLY; | ||
| 1355 | ret = btrfs_prepare_sprout(trans, root); | ||
| 1356 | BUG_ON(ret); | ||
| 1357 | } | ||
| 1358 | |||
| 1359 | device->fs_devices = root->fs_info->fs_devices; | ||
| 1360 | list_add(&device->dev_list, &root->fs_info->fs_devices->devices); | ||
| 1361 | list_add(&device->dev_alloc_list, | ||
| 1362 | &root->fs_info->fs_devices->alloc_list); | ||
| 1363 | root->fs_info->fs_devices->num_devices++; | ||
| 1364 | root->fs_info->fs_devices->open_devices++; | ||
| 1365 | root->fs_info->fs_devices->rw_devices++; | ||
| 1366 | root->fs_info->fs_devices->total_rw_bytes += device->total_bytes; | ||
| 1367 | |||
| 1368 | total_bytes = btrfs_super_total_bytes(&root->fs_info->super_copy); | ||
| 1369 | btrfs_set_super_total_bytes(&root->fs_info->super_copy, | ||
| 1370 | total_bytes + device->total_bytes); | ||
| 1371 | |||
| 1372 | total_bytes = btrfs_super_num_devices(&root->fs_info->super_copy); | ||
| 1373 | btrfs_set_super_num_devices(&root->fs_info->super_copy, | ||
| 1374 | total_bytes + 1); | ||
| 1375 | |||
| 1376 | if (seeding_dev) { | ||
| 1377 | ret = init_first_rw_device(trans, root, device); | ||
| 1378 | BUG_ON(ret); | ||
| 1379 | ret = btrfs_finish_sprout(trans, root); | ||
| 1380 | BUG_ON(ret); | ||
| 1381 | } else { | ||
| 1382 | ret = btrfs_add_device(trans, root, device); | ||
| 1383 | } | ||
| 1384 | |||
| 1385 | unlock_chunks(root); | ||
| 1386 | btrfs_commit_transaction(trans, root); | ||
| 1387 | |||
| 1388 | if (seeding_dev) { | ||
| 1389 | mutex_unlock(&uuid_mutex); | ||
| 1390 | up_write(&sb->s_umount); | ||
| 1391 | |||
| 1392 | ret = btrfs_relocate_sys_chunks(root); | ||
| 1393 | BUG_ON(ret); | ||
| 1394 | } | ||
| 1395 | out: | ||
| 1396 | mutex_unlock(&root->fs_info->volume_mutex); | ||
| 1397 | return ret; | ||
| 1398 | error: | ||
| 1399 | close_bdev_exclusive(bdev, 0); | ||
| 1400 | if (seeding_dev) { | ||
| 1401 | mutex_unlock(&uuid_mutex); | ||
| 1402 | up_write(&sb->s_umount); | ||
| 1403 | } | ||
| 1404 | goto out; | ||
| 1405 | } | ||
| 1406 | |||
| 1407 | static noinline int btrfs_update_device(struct btrfs_trans_handle *trans, | ||
| 1408 | struct btrfs_device *device) | ||
| 1409 | { | ||
| 1410 | int ret; | ||
| 1411 | struct btrfs_path *path; | ||
| 1412 | struct btrfs_root *root; | ||
| 1413 | struct btrfs_dev_item *dev_item; | ||
| 1414 | struct extent_buffer *leaf; | ||
| 1415 | struct btrfs_key key; | ||
| 1416 | |||
| 1417 | root = device->dev_root->fs_info->chunk_root; | ||
| 1418 | |||
| 1419 | path = btrfs_alloc_path(); | ||
| 1420 | if (!path) | ||
| 1421 | return -ENOMEM; | ||
| 1422 | |||
| 1423 | key.objectid = BTRFS_DEV_ITEMS_OBJECTID; | ||
| 1424 | key.type = BTRFS_DEV_ITEM_KEY; | ||
| 1425 | key.offset = device->devid; | ||
| 1426 | |||
| 1427 | ret = btrfs_search_slot(trans, root, &key, path, 0, 1); | ||
| 1428 | if (ret < 0) | ||
| 1429 | goto out; | ||
| 1430 | |||
| 1431 | if (ret > 0) { | ||
| 1432 | ret = -ENOENT; | ||
| 1433 | goto out; | ||
| 1434 | } | ||
| 1435 | |||
| 1436 | leaf = path->nodes[0]; | ||
| 1437 | dev_item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_dev_item); | ||
| 1438 | |||
| 1439 | btrfs_set_device_id(leaf, dev_item, device->devid); | ||
| 1440 | btrfs_set_device_type(leaf, dev_item, device->type); | ||
| 1441 | btrfs_set_device_io_align(leaf, dev_item, device->io_align); | ||
| 1442 | btrfs_set_device_io_width(leaf, dev_item, device->io_width); | ||
| 1443 | btrfs_set_device_sector_size(leaf, dev_item, device->sector_size); | ||
| 1444 | btrfs_set_device_total_bytes(leaf, dev_item, device->total_bytes); | ||
| 1445 | btrfs_set_device_bytes_used(leaf, dev_item, device->bytes_used); | ||
| 1446 | btrfs_mark_buffer_dirty(leaf); | ||
| 1447 | |||
| 1448 | out: | ||
| 1449 | btrfs_free_path(path); | ||
| 1450 | return ret; | ||
| 1451 | } | ||
| 1452 | |||
| 1453 | static int __btrfs_grow_device(struct btrfs_trans_handle *trans, | ||
| 1454 | struct btrfs_device *device, u64 new_size) | ||
| 1455 | { | ||
| 1456 | struct btrfs_super_block *super_copy = | ||
| 1457 | &device->dev_root->fs_info->super_copy; | ||
| 1458 | u64 old_total = btrfs_super_total_bytes(super_copy); | ||
| 1459 | u64 diff = new_size - device->total_bytes; | ||
| 1460 | |||
| 1461 | if (!device->writeable) | ||
| 1462 | return -EACCES; | ||
| 1463 | if (new_size <= device->total_bytes) | ||
| 1464 | return -EINVAL; | ||
| 1465 | |||
| 1466 | btrfs_set_super_total_bytes(super_copy, old_total + diff); | ||
| 1467 | device->fs_devices->total_rw_bytes += diff; | ||
| 1468 | |||
| 1469 | device->total_bytes = new_size; | ||
| 1470 | return btrfs_update_device(trans, device); | ||
| 1471 | } | ||
| 1472 | |||
| 1473 | int btrfs_grow_device(struct btrfs_trans_handle *trans, | ||
| 1474 | struct btrfs_device *device, u64 new_size) | ||
| 1475 | { | ||
| 1476 | int ret; | ||
| 1477 | lock_chunks(device->dev_root); | ||
| 1478 | ret = __btrfs_grow_device(trans, device, new_size); | ||
| 1479 | unlock_chunks(device->dev_root); | ||
| 1480 | return ret; | ||
| 1481 | } | ||
| 1482 | |||
| 1483 | static int btrfs_free_chunk(struct btrfs_trans_handle *trans, | ||
| 1484 | struct btrfs_root *root, | ||
| 1485 | u64 chunk_tree, u64 chunk_objectid, | ||
| 1486 | u64 chunk_offset) | ||
| 1487 | { | ||
| 1488 | int ret; | ||
| 1489 | struct btrfs_path *path; | ||
| 1490 | struct btrfs_key key; | ||
| 1491 | |||
| 1492 | root = root->fs_info->chunk_root; | ||
| 1493 | path = btrfs_alloc_path(); | ||
| 1494 | if (!path) | ||
| 1495 | return -ENOMEM; | ||
| 1496 | |||
| 1497 | key.objectid = chunk_objectid; | ||
| 1498 | key.offset = chunk_offset; | ||
| 1499 | key.type = BTRFS_CHUNK_ITEM_KEY; | ||
| 1500 | |||
| 1501 | ret = btrfs_search_slot(trans, root, &key, path, -1, 1); | ||
| 1502 | BUG_ON(ret); | ||
| 1503 | |||
| 1504 | ret = btrfs_del_item(trans, root, path); | ||
| 1505 | BUG_ON(ret); | ||
| 1506 | |||
| 1507 | btrfs_free_path(path); | ||
| 1508 | return 0; | ||
| 1509 | } | ||
| 1510 | |||
| 1511 | static int btrfs_del_sys_chunk(struct btrfs_root *root, u64 chunk_objectid, u64 | ||
| 1512 | chunk_offset) | ||
| 1513 | { | ||
| 1514 | struct btrfs_super_block *super_copy = &root->fs_info->super_copy; | ||
| 1515 | struct btrfs_disk_key *disk_key; | ||
| 1516 | struct btrfs_chunk *chunk; | ||
| 1517 | u8 *ptr; | ||
| 1518 | int ret = 0; | ||
| 1519 | u32 num_stripes; | ||
| 1520 | u32 array_size; | ||
| 1521 | u32 len = 0; | ||
| 1522 | u32 cur; | ||
| 1523 | struct btrfs_key key; | ||
| 1524 | |||
| 1525 | array_size = btrfs_super_sys_array_size(super_copy); | ||
| 1526 | |||
| 1527 | ptr = super_copy->sys_chunk_array; | ||
| 1528 | cur = 0; | ||
| 1529 | |||
| 1530 | while (cur < array_size) { | ||
| 1531 | disk_key = (struct btrfs_disk_key *)ptr; | ||
| 1532 | btrfs_disk_key_to_cpu(&key, disk_key); | ||
| 1533 | |||
| 1534 | len = sizeof(*disk_key); | ||
| 1535 | |||
| 1536 | if (key.type == BTRFS_CHUNK_ITEM_KEY) { | ||
| 1537 | chunk = (struct btrfs_chunk *)(ptr + len); | ||
| 1538 | num_stripes = btrfs_stack_chunk_num_stripes(chunk); | ||
| 1539 | len += btrfs_chunk_item_size(num_stripes); | ||
| 1540 | } else { | ||
| 1541 | ret = -EIO; | ||
| 1542 | break; | ||
| 1543 | } | ||
| 1544 | if (key.objectid == chunk_objectid && | ||
| 1545 | key.offset == chunk_offset) { | ||
| 1546 | memmove(ptr, ptr + len, array_size - (cur + len)); | ||
| 1547 | array_size -= len; | ||
| 1548 | btrfs_set_super_sys_array_size(super_copy, array_size); | ||
| 1549 | } else { | ||
| 1550 | ptr += len; | ||
| 1551 | cur += len; | ||
| 1552 | } | ||
| 1553 | } | ||
| 1554 | return ret; | ||
| 1555 | } | ||
| 1556 | |||
| 1557 | static int btrfs_relocate_chunk(struct btrfs_root *root, | ||
| 1558 | u64 chunk_tree, u64 chunk_objectid, | ||
| 1559 | u64 chunk_offset) | ||
| 1560 | { | ||
| 1561 | struct extent_map_tree *em_tree; | ||
| 1562 | struct btrfs_root *extent_root; | ||
| 1563 | struct btrfs_trans_handle *trans; | ||
| 1564 | struct extent_map *em; | ||
| 1565 | struct map_lookup *map; | ||
| 1566 | int ret; | ||
| 1567 | int i; | ||
| 1568 | |||
| 1569 | printk(KERN_INFO "btrfs relocating chunk %llu\n", | ||
| 1570 | (unsigned long long)chunk_offset); | ||
| 1571 | root = root->fs_info->chunk_root; | ||
| 1572 | extent_root = root->fs_info->extent_root; | ||
| 1573 | em_tree = &root->fs_info->mapping_tree.map_tree; | ||
| 1574 | |||
| 1575 | /* step one, relocate all the extents inside this chunk */ | ||
| 1576 | ret = btrfs_relocate_block_group(extent_root, chunk_offset); | ||
| 1577 | BUG_ON(ret); | ||
| 1578 | |||
| 1579 | trans = btrfs_start_transaction(root, 1); | ||
| 1580 | BUG_ON(!trans); | ||
| 1581 | |||
| 1582 | lock_chunks(root); | ||
| 1583 | |||
| 1584 | /* | ||
| 1585 | * step two, delete the device extents and the | ||
| 1586 | * chunk tree entries | ||
| 1587 | */ | ||
| 1588 | spin_lock(&em_tree->lock); | ||
| 1589 | em = lookup_extent_mapping(em_tree, chunk_offset, 1); | ||
| 1590 | spin_unlock(&em_tree->lock); | ||
| 1591 | |||
| 1592 | BUG_ON(em->start > chunk_offset || | ||
| 1593 | em->start + em->len < chunk_offset); | ||
| 1594 | map = (struct map_lookup *)em->bdev; | ||
| 1595 | |||
| 1596 | for (i = 0; i < map->num_stripes; i++) { | ||
| 1597 | ret = btrfs_free_dev_extent(trans, map->stripes[i].dev, | ||
| 1598 | map->stripes[i].physical); | ||
| 1599 | BUG_ON(ret); | ||
| 1600 | |||
| 1601 | if (map->stripes[i].dev) { | ||
| 1602 | ret = btrfs_update_device(trans, map->stripes[i].dev); | ||
| 1603 | BUG_ON(ret); | ||
| 1604 | } | ||
| 1605 | } | ||
| 1606 | ret = btrfs_free_chunk(trans, root, chunk_tree, chunk_objectid, | ||
| 1607 | chunk_offset); | ||
| 1608 | |||
| 1609 | BUG_ON(ret); | ||
| 1610 | |||
| 1611 | if (map->type & BTRFS_BLOCK_GROUP_SYSTEM) { | ||
| 1612 | ret = btrfs_del_sys_chunk(root, chunk_objectid, chunk_offset); | ||
| 1613 | BUG_ON(ret); | ||
| 1614 | } | ||
| 1615 | |||
| 1616 | ret = btrfs_remove_block_group(trans, extent_root, chunk_offset); | ||
| 1617 | BUG_ON(ret); | ||
| 1618 | |||
| 1619 | spin_lock(&em_tree->lock); | ||
| 1620 | remove_extent_mapping(em_tree, em); | ||
| 1621 | spin_unlock(&em_tree->lock); | ||
| 1622 | |||
| 1623 | kfree(map); | ||
| 1624 | em->bdev = NULL; | ||
| 1625 | |||
| 1626 | /* once for the tree */ | ||
| 1627 | free_extent_map(em); | ||
| 1628 | /* once for us */ | ||
| 1629 | free_extent_map(em); | ||
| 1630 | |||
| 1631 | unlock_chunks(root); | ||
| 1632 | btrfs_end_transaction(trans, root); | ||
| 1633 | return 0; | ||
| 1634 | } | ||
| 1635 | |||
| 1636 | static int btrfs_relocate_sys_chunks(struct btrfs_root *root) | ||
| 1637 | { | ||
| 1638 | struct btrfs_root *chunk_root = root->fs_info->chunk_root; | ||
| 1639 | struct btrfs_path *path; | ||
| 1640 | struct extent_buffer *leaf; | ||
| 1641 | struct btrfs_chunk *chunk; | ||
| 1642 | struct btrfs_key key; | ||
| 1643 | struct btrfs_key found_key; | ||
| 1644 | u64 chunk_tree = chunk_root->root_key.objectid; | ||
| 1645 | u64 chunk_type; | ||
| 1646 | int ret; | ||
| 1647 | |||
| 1648 | path = btrfs_alloc_path(); | ||
| 1649 | if (!path) | ||
| 1650 | return -ENOMEM; | ||
| 1651 | |||
| 1652 | key.objectid = BTRFS_FIRST_CHUNK_TREE_OBJECTID; | ||
| 1653 | key.offset = (u64)-1; | ||
| 1654 | key.type = BTRFS_CHUNK_ITEM_KEY; | ||
| 1655 | |||
| 1656 | while (1) { | ||
| 1657 | ret = btrfs_search_slot(NULL, chunk_root, &key, path, 0, 0); | ||
| 1658 | if (ret < 0) | ||
| 1659 | goto error; | ||
| 1660 | BUG_ON(ret == 0); | ||
| 1661 | |||
| 1662 | ret = btrfs_previous_item(chunk_root, path, key.objectid, | ||
| 1663 | key.type); | ||
| 1664 | if (ret < 0) | ||
| 1665 | goto error; | ||
| 1666 | if (ret > 0) | ||
| 1667 | break; | ||
| 1668 | |||
| 1669 | leaf = path->nodes[0]; | ||
| 1670 | btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); | ||
| 1671 | |||
| 1672 | chunk = btrfs_item_ptr(leaf, path->slots[0], | ||
| 1673 | struct btrfs_chunk); | ||
| 1674 | chunk_type = btrfs_chunk_type(leaf, chunk); | ||
| 1675 | btrfs_release_path(chunk_root, path); | ||
| 1676 | |||
| 1677 | if (chunk_type & BTRFS_BLOCK_GROUP_SYSTEM) { | ||
| 1678 | ret = btrfs_relocate_chunk(chunk_root, chunk_tree, | ||
| 1679 | found_key.objectid, | ||
| 1680 | found_key.offset); | ||
| 1681 | BUG_ON(ret); | ||
| 1682 | } | ||
| 1683 | |||
| 1684 | if (found_key.offset == 0) | ||
| 1685 | break; | ||
| 1686 | key.offset = found_key.offset - 1; | ||
| 1687 | } | ||
| 1688 | ret = 0; | ||
| 1689 | error: | ||
| 1690 | btrfs_free_path(path); | ||
| 1691 | return ret; | ||
| 1692 | } | ||
| 1693 | |||
| 1694 | static u64 div_factor(u64 num, int factor) | ||
| 1695 | { | ||
| 1696 | if (factor == 10) | ||
| 1697 | return num; | ||
| 1698 | num *= factor; | ||
| 1699 | do_div(num, 10); | ||
| 1700 | return num; | ||
| 1701 | } | ||
| 1702 | |||
| 1703 | int btrfs_balance(struct btrfs_root *dev_root) | ||
| 1704 | { | ||
| 1705 | int ret; | ||
| 1706 | struct list_head *cur; | ||
| 1707 | struct list_head *devices = &dev_root->fs_info->fs_devices->devices; | ||
| 1708 | struct btrfs_device *device; | ||
| 1709 | u64 old_size; | ||
| 1710 | u64 size_to_free; | ||
| 1711 | struct btrfs_path *path; | ||
| 1712 | struct btrfs_key key; | ||
| 1713 | struct btrfs_chunk *chunk; | ||
| 1714 | struct btrfs_root *chunk_root = dev_root->fs_info->chunk_root; | ||
| 1715 | struct btrfs_trans_handle *trans; | ||
| 1716 | struct btrfs_key found_key; | ||
| 1717 | |||
| 1718 | if (dev_root->fs_info->sb->s_flags & MS_RDONLY) | ||
| 1719 | return -EROFS; | ||
| 1720 | |||
| 1721 | mutex_lock(&dev_root->fs_info->volume_mutex); | ||
| 1722 | dev_root = dev_root->fs_info->dev_root; | ||
| 1723 | |||
| 1724 | /* step one make some room on all the devices */ | ||
| 1725 | list_for_each(cur, devices) { | ||
| 1726 | device = list_entry(cur, struct btrfs_device, dev_list); | ||
| 1727 | old_size = device->total_bytes; | ||
| 1728 | size_to_free = div_factor(old_size, 1); | ||
| 1729 | size_to_free = min(size_to_free, (u64)1 * 1024 * 1024); | ||
| 1730 | if (!device->writeable || | ||
| 1731 | device->total_bytes - device->bytes_used > size_to_free) | ||
| 1732 | continue; | ||
| 1733 | |||
| 1734 | ret = btrfs_shrink_device(device, old_size - size_to_free); | ||
| 1735 | BUG_ON(ret); | ||
| 1736 | |||
| 1737 | trans = btrfs_start_transaction(dev_root, 1); | ||
| 1738 | BUG_ON(!trans); | ||
| 1739 | |||
| 1740 | ret = btrfs_grow_device(trans, device, old_size); | ||
| 1741 | BUG_ON(ret); | ||
| 1742 | |||
| 1743 | btrfs_end_transaction(trans, dev_root); | ||
| 1744 | } | ||
| 1745 | |||
| 1746 | /* step two, relocate all the chunks */ | ||
| 1747 | path = btrfs_alloc_path(); | ||
| 1748 | BUG_ON(!path); | ||
| 1749 | |||
| 1750 | key.objectid = BTRFS_FIRST_CHUNK_TREE_OBJECTID; | ||
| 1751 | key.offset = (u64)-1; | ||
| 1752 | key.type = BTRFS_CHUNK_ITEM_KEY; | ||
| 1753 | |||
| 1754 | while (1) { | ||
| 1755 | ret = btrfs_search_slot(NULL, chunk_root, &key, path, 0, 0); | ||
| 1756 | if (ret < 0) | ||
| 1757 | goto error; | ||
| 1758 | |||
| 1759 | /* | ||
| 1760 | * this shouldn't happen, it means the last relocate | ||
| 1761 | * failed | ||
| 1762 | */ | ||
| 1763 | if (ret == 0) | ||
| 1764 | break; | ||
| 1765 | |||
| 1766 | ret = btrfs_previous_item(chunk_root, path, 0, | ||
| 1767 | BTRFS_CHUNK_ITEM_KEY); | ||
| 1768 | if (ret) | ||
| 1769 | break; | ||
| 1770 | |||
| 1771 | btrfs_item_key_to_cpu(path->nodes[0], &found_key, | ||
| 1772 | path->slots[0]); | ||
| 1773 | if (found_key.objectid != key.objectid) | ||
| 1774 | break; | ||
| 1775 | |||
| 1776 | chunk = btrfs_item_ptr(path->nodes[0], | ||
| 1777 | path->slots[0], | ||
| 1778 | struct btrfs_chunk); | ||
| 1779 | key.offset = found_key.offset; | ||
| 1780 | /* chunk zero is special */ | ||
| 1781 | if (key.offset == 0) | ||
| 1782 | break; | ||
| 1783 | |||
| 1784 | btrfs_release_path(chunk_root, path); | ||
| 1785 | ret = btrfs_relocate_chunk(chunk_root, | ||
| 1786 | chunk_root->root_key.objectid, | ||
| 1787 | found_key.objectid, | ||
| 1788 | found_key.offset); | ||
| 1789 | BUG_ON(ret); | ||
| 1790 | } | ||
| 1791 | ret = 0; | ||
| 1792 | error: | ||
| 1793 | btrfs_free_path(path); | ||
| 1794 | mutex_unlock(&dev_root->fs_info->volume_mutex); | ||
| 1795 | return ret; | ||
| 1796 | } | ||
| 1797 | |||
| 1798 | /* | ||
| 1799 | * shrinking a device means finding all of the device extents past | ||
| 1800 | * the new size, and then following the back refs to the chunks. | ||
| 1801 | * The chunk relocation code actually frees the device extent | ||
| 1802 | */ | ||
| 1803 | int btrfs_shrink_device(struct btrfs_device *device, u64 new_size) | ||
| 1804 | { | ||
| 1805 | struct btrfs_trans_handle *trans; | ||
| 1806 | struct btrfs_root *root = device->dev_root; | ||
| 1807 | struct btrfs_dev_extent *dev_extent = NULL; | ||
| 1808 | struct btrfs_path *path; | ||
| 1809 | u64 length; | ||
| 1810 | u64 chunk_tree; | ||
| 1811 | u64 chunk_objectid; | ||
| 1812 | u64 chunk_offset; | ||
| 1813 | int ret; | ||
| 1814 | int slot; | ||
| 1815 | struct extent_buffer *l; | ||
| 1816 | struct btrfs_key key; | ||
| 1817 | struct btrfs_super_block *super_copy = &root->fs_info->super_copy; | ||
| 1818 | u64 old_total = btrfs_super_total_bytes(super_copy); | ||
| 1819 | u64 diff = device->total_bytes - new_size; | ||
| 1820 | |||
| 1821 | if (new_size >= device->total_bytes) | ||
| 1822 | return -EINVAL; | ||
| 1823 | |||
| 1824 | path = btrfs_alloc_path(); | ||
| 1825 | if (!path) | ||
| 1826 | return -ENOMEM; | ||
| 1827 | |||
| 1828 | trans = btrfs_start_transaction(root, 1); | ||
| 1829 | if (!trans) { | ||
| 1830 | ret = -ENOMEM; | ||
| 1831 | goto done; | ||
| 1832 | } | ||
| 1833 | |||
| 1834 | path->reada = 2; | ||
| 1835 | |||
| 1836 | lock_chunks(root); | ||
| 1837 | |||
| 1838 | device->total_bytes = new_size; | ||
| 1839 | if (device->writeable) | ||
| 1840 | device->fs_devices->total_rw_bytes -= diff; | ||
| 1841 | ret = btrfs_update_device(trans, device); | ||
| 1842 | if (ret) { | ||
| 1843 | unlock_chunks(root); | ||
| 1844 | btrfs_end_transaction(trans, root); | ||
| 1845 | goto done; | ||
| 1846 | } | ||
| 1847 | WARN_ON(diff > old_total); | ||
| 1848 | btrfs_set_super_total_bytes(super_copy, old_total - diff); | ||
| 1849 | unlock_chunks(root); | ||
| 1850 | btrfs_end_transaction(trans, root); | ||
| 1851 | |||
| 1852 | key.objectid = device->devid; | ||
| 1853 | key.offset = (u64)-1; | ||
| 1854 | key.type = BTRFS_DEV_EXTENT_KEY; | ||
| 1855 | |||
| 1856 | while (1) { | ||
| 1857 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 1858 | if (ret < 0) | ||
| 1859 | goto done; | ||
| 1860 | |||
| 1861 | ret = btrfs_previous_item(root, path, 0, key.type); | ||
| 1862 | if (ret < 0) | ||
| 1863 | goto done; | ||
| 1864 | if (ret) { | ||
| 1865 | ret = 0; | ||
| 1866 | goto done; | ||
| 1867 | } | ||
| 1868 | |||
| 1869 | l = path->nodes[0]; | ||
| 1870 | slot = path->slots[0]; | ||
| 1871 | btrfs_item_key_to_cpu(l, &key, path->slots[0]); | ||
| 1872 | |||
| 1873 | if (key.objectid != device->devid) | ||
| 1874 | goto done; | ||
| 1875 | |||
| 1876 | dev_extent = btrfs_item_ptr(l, slot, struct btrfs_dev_extent); | ||
| 1877 | length = btrfs_dev_extent_length(l, dev_extent); | ||
| 1878 | |||
| 1879 | if (key.offset + length <= new_size) | ||
| 1880 | goto done; | ||
| 1881 | |||
| 1882 | chunk_tree = btrfs_dev_extent_chunk_tree(l, dev_extent); | ||
| 1883 | chunk_objectid = btrfs_dev_extent_chunk_objectid(l, dev_extent); | ||
| 1884 | chunk_offset = btrfs_dev_extent_chunk_offset(l, dev_extent); | ||
| 1885 | btrfs_release_path(root, path); | ||
| 1886 | |||
| 1887 | ret = btrfs_relocate_chunk(root, chunk_tree, chunk_objectid, | ||
| 1888 | chunk_offset); | ||
| 1889 | if (ret) | ||
| 1890 | goto done; | ||
| 1891 | } | ||
| 1892 | |||
| 1893 | done: | ||
| 1894 | btrfs_free_path(path); | ||
| 1895 | return ret; | ||
| 1896 | } | ||
| 1897 | |||
| 1898 | static int btrfs_add_system_chunk(struct btrfs_trans_handle *trans, | ||
| 1899 | struct btrfs_root *root, | ||
| 1900 | struct btrfs_key *key, | ||
| 1901 | struct btrfs_chunk *chunk, int item_size) | ||
| 1902 | { | ||
| 1903 | struct btrfs_super_block *super_copy = &root->fs_info->super_copy; | ||
| 1904 | struct btrfs_disk_key disk_key; | ||
| 1905 | u32 array_size; | ||
| 1906 | u8 *ptr; | ||
| 1907 | |||
| 1908 | array_size = btrfs_super_sys_array_size(super_copy); | ||
| 1909 | if (array_size + item_size > BTRFS_SYSTEM_CHUNK_ARRAY_SIZE) | ||
| 1910 | return -EFBIG; | ||
| 1911 | |||
| 1912 | ptr = super_copy->sys_chunk_array + array_size; | ||
| 1913 | btrfs_cpu_key_to_disk(&disk_key, key); | ||
| 1914 | memcpy(ptr, &disk_key, sizeof(disk_key)); | ||
| 1915 | ptr += sizeof(disk_key); | ||
| 1916 | memcpy(ptr, chunk, item_size); | ||
| 1917 | item_size += sizeof(disk_key); | ||
| 1918 | btrfs_set_super_sys_array_size(super_copy, array_size + item_size); | ||
| 1919 | return 0; | ||
| 1920 | } | ||
| 1921 | |||
| 1922 | static noinline u64 chunk_bytes_by_type(u64 type, u64 calc_size, | ||
| 1923 | int num_stripes, int sub_stripes) | ||
| 1924 | { | ||
| 1925 | if (type & (BTRFS_BLOCK_GROUP_RAID1 | BTRFS_BLOCK_GROUP_DUP)) | ||
| 1926 | return calc_size; | ||
| 1927 | else if (type & BTRFS_BLOCK_GROUP_RAID10) | ||
| 1928 | return calc_size * (num_stripes / sub_stripes); | ||
| 1929 | else | ||
| 1930 | return calc_size * num_stripes; | ||
| 1931 | } | ||
| 1932 | |||
| 1933 | static int __btrfs_alloc_chunk(struct btrfs_trans_handle *trans, | ||
| 1934 | struct btrfs_root *extent_root, | ||
| 1935 | struct map_lookup **map_ret, | ||
| 1936 | u64 *num_bytes, u64 *stripe_size, | ||
| 1937 | u64 start, u64 type) | ||
| 1938 | { | ||
| 1939 | struct btrfs_fs_info *info = extent_root->fs_info; | ||
| 1940 | struct btrfs_device *device = NULL; | ||
| 1941 | struct btrfs_fs_devices *fs_devices = info->fs_devices; | ||
| 1942 | struct list_head *cur; | ||
| 1943 | struct map_lookup *map = NULL; | ||
| 1944 | struct extent_map_tree *em_tree; | ||
| 1945 | struct extent_map *em; | ||
| 1946 | struct list_head private_devs; | ||
| 1947 | int min_stripe_size = 1 * 1024 * 1024; | ||
| 1948 | u64 calc_size = 1024 * 1024 * 1024; | ||
| 1949 | u64 max_chunk_size = calc_size; | ||
| 1950 | u64 min_free; | ||
| 1951 | u64 avail; | ||
| 1952 | u64 max_avail = 0; | ||
| 1953 | u64 dev_offset; | ||
| 1954 | int num_stripes = 1; | ||
| 1955 | int min_stripes = 1; | ||
| 1956 | int sub_stripes = 0; | ||
| 1957 | int looped = 0; | ||
| 1958 | int ret; | ||
| 1959 | int index; | ||
| 1960 | int stripe_len = 64 * 1024; | ||
| 1961 | |||
| 1962 | if ((type & BTRFS_BLOCK_GROUP_RAID1) && | ||
| 1963 | (type & BTRFS_BLOCK_GROUP_DUP)) { | ||
| 1964 | WARN_ON(1); | ||
| 1965 | type &= ~BTRFS_BLOCK_GROUP_DUP; | ||
| 1966 | } | ||
| 1967 | if (list_empty(&fs_devices->alloc_list)) | ||
| 1968 | return -ENOSPC; | ||
| 1969 | |||
| 1970 | if (type & (BTRFS_BLOCK_GROUP_RAID0)) { | ||
| 1971 | num_stripes = fs_devices->rw_devices; | ||
| 1972 | min_stripes = 2; | ||
| 1973 | } | ||
| 1974 | if (type & (BTRFS_BLOCK_GROUP_DUP)) { | ||
| 1975 | num_stripes = 2; | ||
| 1976 | min_stripes = 2; | ||
| 1977 | } | ||
| 1978 | if (type & (BTRFS_BLOCK_GROUP_RAID1)) { | ||
| 1979 | num_stripes = min_t(u64, 2, fs_devices->rw_devices); | ||
| 1980 | if (num_stripes < 2) | ||
| 1981 | return -ENOSPC; | ||
| 1982 | min_stripes = 2; | ||
| 1983 | } | ||
| 1984 | if (type & (BTRFS_BLOCK_GROUP_RAID10)) { | ||
| 1985 | num_stripes = fs_devices->rw_devices; | ||
| 1986 | if (num_stripes < 4) | ||
| 1987 | return -ENOSPC; | ||
| 1988 | num_stripes &= ~(u32)1; | ||
| 1989 | sub_stripes = 2; | ||
| 1990 | min_stripes = 4; | ||
| 1991 | } | ||
| 1992 | |||
| 1993 | if (type & BTRFS_BLOCK_GROUP_DATA) { | ||
| 1994 | max_chunk_size = 10 * calc_size; | ||
| 1995 | min_stripe_size = 64 * 1024 * 1024; | ||
| 1996 | } else if (type & BTRFS_BLOCK_GROUP_METADATA) { | ||
| 1997 | max_chunk_size = 4 * calc_size; | ||
| 1998 | min_stripe_size = 32 * 1024 * 1024; | ||
| 1999 | } else if (type & BTRFS_BLOCK_GROUP_SYSTEM) { | ||
| 2000 | calc_size = 8 * 1024 * 1024; | ||
| 2001 | max_chunk_size = calc_size * 2; | ||
| 2002 | min_stripe_size = 1 * 1024 * 1024; | ||
| 2003 | } | ||
| 2004 | |||
| 2005 | /* we don't want a chunk larger than 10% of writeable space */ | ||
| 2006 | max_chunk_size = min(div_factor(fs_devices->total_rw_bytes, 1), | ||
| 2007 | max_chunk_size); | ||
| 2008 | |||
| 2009 | again: | ||
| 2010 | if (!map || map->num_stripes != num_stripes) { | ||
| 2011 | kfree(map); | ||
| 2012 | map = kmalloc(map_lookup_size(num_stripes), GFP_NOFS); | ||
| 2013 | if (!map) | ||
| 2014 | return -ENOMEM; | ||
| 2015 | map->num_stripes = num_stripes; | ||
| 2016 | } | ||
| 2017 | |||
| 2018 | if (calc_size * num_stripes > max_chunk_size) { | ||
| 2019 | calc_size = max_chunk_size; | ||
| 2020 | do_div(calc_size, num_stripes); | ||
| 2021 | do_div(calc_size, stripe_len); | ||
| 2022 | calc_size *= stripe_len; | ||
| 2023 | } | ||
| 2024 | /* we don't want tiny stripes */ | ||
| 2025 | calc_size = max_t(u64, min_stripe_size, calc_size); | ||
| 2026 | |||
| 2027 | do_div(calc_size, stripe_len); | ||
| 2028 | calc_size *= stripe_len; | ||
| 2029 | |||
| 2030 | cur = fs_devices->alloc_list.next; | ||
| 2031 | index = 0; | ||
| 2032 | |||
| 2033 | if (type & BTRFS_BLOCK_GROUP_DUP) | ||
| 2034 | min_free = calc_size * 2; | ||
| 2035 | else | ||
| 2036 | min_free = calc_size; | ||
| 2037 | |||
| 2038 | /* | ||
| 2039 | * we add 1MB because we never use the first 1MB of the device, unless | ||
| 2040 | * we've looped, then we are likely allocating the maximum amount of | ||
| 2041 | * space left already | ||
| 2042 | */ | ||
| 2043 | if (!looped) | ||
| 2044 | min_free += 1024 * 1024; | ||
| 2045 | |||
| 2046 | INIT_LIST_HEAD(&private_devs); | ||
| 2047 | while (index < num_stripes) { | ||
| 2048 | device = list_entry(cur, struct btrfs_device, dev_alloc_list); | ||
| 2049 | BUG_ON(!device->writeable); | ||
| 2050 | if (device->total_bytes > device->bytes_used) | ||
| 2051 | avail = device->total_bytes - device->bytes_used; | ||
| 2052 | else | ||
| 2053 | avail = 0; | ||
| 2054 | cur = cur->next; | ||
| 2055 | |||
| 2056 | if (device->in_fs_metadata && avail >= min_free) { | ||
| 2057 | ret = find_free_dev_extent(trans, device, | ||
| 2058 | min_free, &dev_offset); | ||
| 2059 | if (ret == 0) { | ||
| 2060 | list_move_tail(&device->dev_alloc_list, | ||
| 2061 | &private_devs); | ||
| 2062 | map->stripes[index].dev = device; | ||
| 2063 | map->stripes[index].physical = dev_offset; | ||
| 2064 | index++; | ||
| 2065 | if (type & BTRFS_BLOCK_GROUP_DUP) { | ||
| 2066 | map->stripes[index].dev = device; | ||
| 2067 | map->stripes[index].physical = | ||
| 2068 | dev_offset + calc_size; | ||
| 2069 | index++; | ||
| 2070 | } | ||
| 2071 | } | ||
| 2072 | } else if (device->in_fs_metadata && avail > max_avail) | ||
| 2073 | max_avail = avail; | ||
| 2074 | if (cur == &fs_devices->alloc_list) | ||
| 2075 | break; | ||
| 2076 | } | ||
| 2077 | list_splice(&private_devs, &fs_devices->alloc_list); | ||
| 2078 | if (index < num_stripes) { | ||
| 2079 | if (index >= min_stripes) { | ||
| 2080 | num_stripes = index; | ||
| 2081 | if (type & (BTRFS_BLOCK_GROUP_RAID10)) { | ||
| 2082 | num_stripes /= sub_stripes; | ||
| 2083 | num_stripes *= sub_stripes; | ||
| 2084 | } | ||
| 2085 | looped = 1; | ||
| 2086 | goto again; | ||
| 2087 | } | ||
| 2088 | if (!looped && max_avail > 0) { | ||
| 2089 | looped = 1; | ||
| 2090 | calc_size = max_avail; | ||
| 2091 | goto again; | ||
| 2092 | } | ||
| 2093 | kfree(map); | ||
| 2094 | return -ENOSPC; | ||
| 2095 | } | ||
| 2096 | map->sector_size = extent_root->sectorsize; | ||
| 2097 | map->stripe_len = stripe_len; | ||
| 2098 | map->io_align = stripe_len; | ||
| 2099 | map->io_width = stripe_len; | ||
| 2100 | map->type = type; | ||
| 2101 | map->num_stripes = num_stripes; | ||
| 2102 | map->sub_stripes = sub_stripes; | ||
| 2103 | |||
| 2104 | *map_ret = map; | ||
| 2105 | *stripe_size = calc_size; | ||
| 2106 | *num_bytes = chunk_bytes_by_type(type, calc_size, | ||
| 2107 | num_stripes, sub_stripes); | ||
| 2108 | |||
| 2109 | em = alloc_extent_map(GFP_NOFS); | ||
| 2110 | if (!em) { | ||
| 2111 | kfree(map); | ||
| 2112 | return -ENOMEM; | ||
| 2113 | } | ||
| 2114 | em->bdev = (struct block_device *)map; | ||
| 2115 | em->start = start; | ||
| 2116 | em->len = *num_bytes; | ||
| 2117 | em->block_start = 0; | ||
| 2118 | em->block_len = em->len; | ||
| 2119 | |||
| 2120 | em_tree = &extent_root->fs_info->mapping_tree.map_tree; | ||
| 2121 | spin_lock(&em_tree->lock); | ||
| 2122 | ret = add_extent_mapping(em_tree, em); | ||
| 2123 | spin_unlock(&em_tree->lock); | ||
| 2124 | BUG_ON(ret); | ||
| 2125 | free_extent_map(em); | ||
| 2126 | |||
| 2127 | ret = btrfs_make_block_group(trans, extent_root, 0, type, | ||
| 2128 | BTRFS_FIRST_CHUNK_TREE_OBJECTID, | ||
| 2129 | start, *num_bytes); | ||
| 2130 | BUG_ON(ret); | ||
| 2131 | |||
| 2132 | index = 0; | ||
| 2133 | while (index < map->num_stripes) { | ||
| 2134 | device = map->stripes[index].dev; | ||
| 2135 | dev_offset = map->stripes[index].physical; | ||
| 2136 | |||
| 2137 | ret = btrfs_alloc_dev_extent(trans, device, | ||
| 2138 | info->chunk_root->root_key.objectid, | ||
| 2139 | BTRFS_FIRST_CHUNK_TREE_OBJECTID, | ||
| 2140 | start, dev_offset, calc_size); | ||
| 2141 | BUG_ON(ret); | ||
| 2142 | index++; | ||
| 2143 | } | ||
| 2144 | |||
| 2145 | return 0; | ||
| 2146 | } | ||
| 2147 | |||
| 2148 | static int __finish_chunk_alloc(struct btrfs_trans_handle *trans, | ||
| 2149 | struct btrfs_root *extent_root, | ||
| 2150 | struct map_lookup *map, u64 chunk_offset, | ||
| 2151 | u64 chunk_size, u64 stripe_size) | ||
| 2152 | { | ||
| 2153 | u64 dev_offset; | ||
| 2154 | struct btrfs_key key; | ||
| 2155 | struct btrfs_root *chunk_root = extent_root->fs_info->chunk_root; | ||
| 2156 | struct btrfs_device *device; | ||
| 2157 | struct btrfs_chunk *chunk; | ||
| 2158 | struct btrfs_stripe *stripe; | ||
| 2159 | size_t item_size = btrfs_chunk_item_size(map->num_stripes); | ||
| 2160 | int index = 0; | ||
| 2161 | int ret; | ||
| 2162 | |||
| 2163 | chunk = kzalloc(item_size, GFP_NOFS); | ||
| 2164 | if (!chunk) | ||
| 2165 | return -ENOMEM; | ||
| 2166 | |||
| 2167 | index = 0; | ||
| 2168 | while (index < map->num_stripes) { | ||
| 2169 | device = map->stripes[index].dev; | ||
| 2170 | device->bytes_used += stripe_size; | ||
| 2171 | ret = btrfs_update_device(trans, device); | ||
| 2172 | BUG_ON(ret); | ||
| 2173 | index++; | ||
| 2174 | } | ||
| 2175 | |||
| 2176 | index = 0; | ||
| 2177 | stripe = &chunk->stripe; | ||
| 2178 | while (index < map->num_stripes) { | ||
| 2179 | device = map->stripes[index].dev; | ||
| 2180 | dev_offset = map->stripes[index].physical; | ||
| 2181 | |||
| 2182 | btrfs_set_stack_stripe_devid(stripe, device->devid); | ||
| 2183 | btrfs_set_stack_stripe_offset(stripe, dev_offset); | ||
| 2184 | memcpy(stripe->dev_uuid, device->uuid, BTRFS_UUID_SIZE); | ||
| 2185 | stripe++; | ||
| 2186 | index++; | ||
| 2187 | } | ||
| 2188 | |||
| 2189 | btrfs_set_stack_chunk_length(chunk, chunk_size); | ||
| 2190 | btrfs_set_stack_chunk_owner(chunk, extent_root->root_key.objectid); | ||
| 2191 | btrfs_set_stack_chunk_stripe_len(chunk, map->stripe_len); | ||
| 2192 | btrfs_set_stack_chunk_type(chunk, map->type); | ||
| 2193 | btrfs_set_stack_chunk_num_stripes(chunk, map->num_stripes); | ||
| 2194 | btrfs_set_stack_chunk_io_align(chunk, map->stripe_len); | ||
| 2195 | btrfs_set_stack_chunk_io_width(chunk, map->stripe_len); | ||
| 2196 | btrfs_set_stack_chunk_sector_size(chunk, extent_root->sectorsize); | ||
| 2197 | btrfs_set_stack_chunk_sub_stripes(chunk, map->sub_stripes); | ||
| 2198 | |||
| 2199 | key.objectid = BTRFS_FIRST_CHUNK_TREE_OBJECTID; | ||
| 2200 | key.type = BTRFS_CHUNK_ITEM_KEY; | ||
| 2201 | key.offset = chunk_offset; | ||
| 2202 | |||
| 2203 | ret = btrfs_insert_item(trans, chunk_root, &key, chunk, item_size); | ||
| 2204 | BUG_ON(ret); | ||
| 2205 | |||
| 2206 | if (map->type & BTRFS_BLOCK_GROUP_SYSTEM) { | ||
| 2207 | ret = btrfs_add_system_chunk(trans, chunk_root, &key, chunk, | ||
| 2208 | item_size); | ||
| 2209 | BUG_ON(ret); | ||
| 2210 | } | ||
| 2211 | kfree(chunk); | ||
| 2212 | return 0; | ||
| 2213 | } | ||
| 2214 | |||
| 2215 | /* | ||
| 2216 | * Chunk allocation falls into two parts. The first part does works | ||
| 2217 | * that make the new allocated chunk useable, but not do any operation | ||
| 2218 | * that modifies the chunk tree. The second part does the works that | ||
| 2219 | * require modifying the chunk tree. This division is important for the | ||
| 2220 | * bootstrap process of adding storage to a seed btrfs. | ||
| 2221 | */ | ||
| 2222 | int btrfs_alloc_chunk(struct btrfs_trans_handle *trans, | ||
| 2223 | struct btrfs_root *extent_root, u64 type) | ||
| 2224 | { | ||
| 2225 | u64 chunk_offset; | ||
| 2226 | u64 chunk_size; | ||
| 2227 | u64 stripe_size; | ||
| 2228 | struct map_lookup *map; | ||
| 2229 | struct btrfs_root *chunk_root = extent_root->fs_info->chunk_root; | ||
| 2230 | int ret; | ||
| 2231 | |||
| 2232 | ret = find_next_chunk(chunk_root, BTRFS_FIRST_CHUNK_TREE_OBJECTID, | ||
| 2233 | &chunk_offset); | ||
| 2234 | if (ret) | ||
| 2235 | return ret; | ||
| 2236 | |||
| 2237 | ret = __btrfs_alloc_chunk(trans, extent_root, &map, &chunk_size, | ||
| 2238 | &stripe_size, chunk_offset, type); | ||
| 2239 | if (ret) | ||
| 2240 | return ret; | ||
| 2241 | |||
| 2242 | ret = __finish_chunk_alloc(trans, extent_root, map, chunk_offset, | ||
| 2243 | chunk_size, stripe_size); | ||
| 2244 | BUG_ON(ret); | ||
| 2245 | return 0; | ||
| 2246 | } | ||
| 2247 | |||
| 2248 | static noinline int init_first_rw_device(struct btrfs_trans_handle *trans, | ||
| 2249 | struct btrfs_root *root, | ||
| 2250 | struct btrfs_device *device) | ||
| 2251 | { | ||
| 2252 | u64 chunk_offset; | ||
| 2253 | u64 sys_chunk_offset; | ||
| 2254 | u64 chunk_size; | ||
| 2255 | u64 sys_chunk_size; | ||
| 2256 | u64 stripe_size; | ||
| 2257 | u64 sys_stripe_size; | ||
| 2258 | u64 alloc_profile; | ||
| 2259 | struct map_lookup *map; | ||
| 2260 | struct map_lookup *sys_map; | ||
| 2261 | struct btrfs_fs_info *fs_info = root->fs_info; | ||
| 2262 | struct btrfs_root *extent_root = fs_info->extent_root; | ||
| 2263 | int ret; | ||
| 2264 | |||
| 2265 | ret = find_next_chunk(fs_info->chunk_root, | ||
| 2266 | BTRFS_FIRST_CHUNK_TREE_OBJECTID, &chunk_offset); | ||
| 2267 | BUG_ON(ret); | ||
| 2268 | |||
| 2269 | alloc_profile = BTRFS_BLOCK_GROUP_METADATA | | ||
| 2270 | (fs_info->metadata_alloc_profile & | ||
| 2271 | fs_info->avail_metadata_alloc_bits); | ||
| 2272 | alloc_profile = btrfs_reduce_alloc_profile(root, alloc_profile); | ||
| 2273 | |||
| 2274 | ret = __btrfs_alloc_chunk(trans, extent_root, &map, &chunk_size, | ||
| 2275 | &stripe_size, chunk_offset, alloc_profile); | ||
| 2276 | BUG_ON(ret); | ||
| 2277 | |||
| 2278 | sys_chunk_offset = chunk_offset + chunk_size; | ||
| 2279 | |||
| 2280 | alloc_profile = BTRFS_BLOCK_GROUP_SYSTEM | | ||
| 2281 | (fs_info->system_alloc_profile & | ||
| 2282 | fs_info->avail_system_alloc_bits); | ||
| 2283 | alloc_profile = btrfs_reduce_alloc_profile(root, alloc_profile); | ||
| 2284 | |||
| 2285 | ret = __btrfs_alloc_chunk(trans, extent_root, &sys_map, | ||
| 2286 | &sys_chunk_size, &sys_stripe_size, | ||
| 2287 | sys_chunk_offset, alloc_profile); | ||
| 2288 | BUG_ON(ret); | ||
| 2289 | |||
| 2290 | ret = btrfs_add_device(trans, fs_info->chunk_root, device); | ||
| 2291 | BUG_ON(ret); | ||
| 2292 | |||
| 2293 | /* | ||
| 2294 | * Modifying chunk tree needs allocating new blocks from both | ||
| 2295 | * system block group and metadata block group. So we only can | ||
| 2296 | * do operations require modifying the chunk tree after both | ||
| 2297 | * block groups were created. | ||
| 2298 | */ | ||
| 2299 | ret = __finish_chunk_alloc(trans, extent_root, map, chunk_offset, | ||
| 2300 | chunk_size, stripe_size); | ||
| 2301 | BUG_ON(ret); | ||
| 2302 | |||
| 2303 | ret = __finish_chunk_alloc(trans, extent_root, sys_map, | ||
| 2304 | sys_chunk_offset, sys_chunk_size, | ||
| 2305 | sys_stripe_size); | ||
| 2306 | BUG_ON(ret); | ||
| 2307 | return 0; | ||
| 2308 | } | ||
| 2309 | |||
| 2310 | int btrfs_chunk_readonly(struct btrfs_root *root, u64 chunk_offset) | ||
| 2311 | { | ||
| 2312 | struct extent_map *em; | ||
| 2313 | struct map_lookup *map; | ||
| 2314 | struct btrfs_mapping_tree *map_tree = &root->fs_info->mapping_tree; | ||
| 2315 | int readonly = 0; | ||
| 2316 | int i; | ||
| 2317 | |||
| 2318 | spin_lock(&map_tree->map_tree.lock); | ||
| 2319 | em = lookup_extent_mapping(&map_tree->map_tree, chunk_offset, 1); | ||
| 2320 | spin_unlock(&map_tree->map_tree.lock); | ||
| 2321 | if (!em) | ||
| 2322 | return 1; | ||
| 2323 | |||
| 2324 | map = (struct map_lookup *)em->bdev; | ||
| 2325 | for (i = 0; i < map->num_stripes; i++) { | ||
| 2326 | if (!map->stripes[i].dev->writeable) { | ||
| 2327 | readonly = 1; | ||
| 2328 | break; | ||
| 2329 | } | ||
| 2330 | } | ||
| 2331 | free_extent_map(em); | ||
| 2332 | return readonly; | ||
| 2333 | } | ||
| 2334 | |||
| 2335 | void btrfs_mapping_init(struct btrfs_mapping_tree *tree) | ||
| 2336 | { | ||
| 2337 | extent_map_tree_init(&tree->map_tree, GFP_NOFS); | ||
| 2338 | } | ||
| 2339 | |||
| 2340 | void btrfs_mapping_tree_free(struct btrfs_mapping_tree *tree) | ||
| 2341 | { | ||
| 2342 | struct extent_map *em; | ||
| 2343 | |||
| 2344 | while (1) { | ||
| 2345 | spin_lock(&tree->map_tree.lock); | ||
| 2346 | em = lookup_extent_mapping(&tree->map_tree, 0, (u64)-1); | ||
| 2347 | if (em) | ||
| 2348 | remove_extent_mapping(&tree->map_tree, em); | ||
| 2349 | spin_unlock(&tree->map_tree.lock); | ||
| 2350 | if (!em) | ||
| 2351 | break; | ||
| 2352 | kfree(em->bdev); | ||
| 2353 | /* once for us */ | ||
| 2354 | free_extent_map(em); | ||
| 2355 | /* once for the tree */ | ||
| 2356 | free_extent_map(em); | ||
| 2357 | } | ||
| 2358 | } | ||
| 2359 | |||
| 2360 | int btrfs_num_copies(struct btrfs_mapping_tree *map_tree, u64 logical, u64 len) | ||
| 2361 | { | ||
| 2362 | struct extent_map *em; | ||
| 2363 | struct map_lookup *map; | ||
| 2364 | struct extent_map_tree *em_tree = &map_tree->map_tree; | ||
| 2365 | int ret; | ||
| 2366 | |||
| 2367 | spin_lock(&em_tree->lock); | ||
| 2368 | em = lookup_extent_mapping(em_tree, logical, len); | ||
| 2369 | spin_unlock(&em_tree->lock); | ||
| 2370 | BUG_ON(!em); | ||
| 2371 | |||
| 2372 | BUG_ON(em->start > logical || em->start + em->len < logical); | ||
| 2373 | map = (struct map_lookup *)em->bdev; | ||
| 2374 | if (map->type & (BTRFS_BLOCK_GROUP_DUP | BTRFS_BLOCK_GROUP_RAID1)) | ||
| 2375 | ret = map->num_stripes; | ||
| 2376 | else if (map->type & BTRFS_BLOCK_GROUP_RAID10) | ||
| 2377 | ret = map->sub_stripes; | ||
| 2378 | else | ||
| 2379 | ret = 1; | ||
| 2380 | free_extent_map(em); | ||
| 2381 | return ret; | ||
| 2382 | } | ||
| 2383 | |||
| 2384 | static int find_live_mirror(struct map_lookup *map, int first, int num, | ||
| 2385 | int optimal) | ||
| 2386 | { | ||
| 2387 | int i; | ||
| 2388 | if (map->stripes[optimal].dev->bdev) | ||
| 2389 | return optimal; | ||
| 2390 | for (i = first; i < first + num; i++) { | ||
| 2391 | if (map->stripes[i].dev->bdev) | ||
| 2392 | return i; | ||
| 2393 | } | ||
| 2394 | /* we couldn't find one that doesn't fail. Just return something | ||
| 2395 | * and the io error handling code will clean up eventually | ||
| 2396 | */ | ||
| 2397 | return optimal; | ||
| 2398 | } | ||
| 2399 | |||
| 2400 | static int __btrfs_map_block(struct btrfs_mapping_tree *map_tree, int rw, | ||
| 2401 | u64 logical, u64 *length, | ||
| 2402 | struct btrfs_multi_bio **multi_ret, | ||
| 2403 | int mirror_num, struct page *unplug_page) | ||
| 2404 | { | ||
| 2405 | struct extent_map *em; | ||
| 2406 | struct map_lookup *map; | ||
| 2407 | struct extent_map_tree *em_tree = &map_tree->map_tree; | ||
| 2408 | u64 offset; | ||
| 2409 | u64 stripe_offset; | ||
| 2410 | u64 stripe_nr; | ||
| 2411 | int stripes_allocated = 8; | ||
| 2412 | int stripes_required = 1; | ||
| 2413 | int stripe_index; | ||
| 2414 | int i; | ||
| 2415 | int num_stripes; | ||
| 2416 | int max_errors = 0; | ||
| 2417 | struct btrfs_multi_bio *multi = NULL; | ||
| 2418 | |||
| 2419 | if (multi_ret && !(rw & (1 << BIO_RW))) | ||
| 2420 | stripes_allocated = 1; | ||
| 2421 | again: | ||
| 2422 | if (multi_ret) { | ||
| 2423 | multi = kzalloc(btrfs_multi_bio_size(stripes_allocated), | ||
| 2424 | GFP_NOFS); | ||
| 2425 | if (!multi) | ||
| 2426 | return -ENOMEM; | ||
| 2427 | |||
| 2428 | atomic_set(&multi->error, 0); | ||
| 2429 | } | ||
| 2430 | |||
| 2431 | spin_lock(&em_tree->lock); | ||
| 2432 | em = lookup_extent_mapping(em_tree, logical, *length); | ||
| 2433 | spin_unlock(&em_tree->lock); | ||
| 2434 | |||
| 2435 | if (!em && unplug_page) | ||
| 2436 | return 0; | ||
| 2437 | |||
| 2438 | if (!em) { | ||
| 2439 | printk(KERN_CRIT "unable to find logical %llu len %llu\n", | ||
| 2440 | (unsigned long long)logical, | ||
| 2441 | (unsigned long long)*length); | ||
| 2442 | BUG(); | ||
| 2443 | } | ||
| 2444 | |||
| 2445 | BUG_ON(em->start > logical || em->start + em->len < logical); | ||
| 2446 | map = (struct map_lookup *)em->bdev; | ||
| 2447 | offset = logical - em->start; | ||
| 2448 | |||
| 2449 | if (mirror_num > map->num_stripes) | ||
| 2450 | mirror_num = 0; | ||
| 2451 | |||
| 2452 | /* if our multi bio struct is too small, back off and try again */ | ||
| 2453 | if (rw & (1 << BIO_RW)) { | ||
| 2454 | if (map->type & (BTRFS_BLOCK_GROUP_RAID1 | | ||
| 2455 | BTRFS_BLOCK_GROUP_DUP)) { | ||
| 2456 | stripes_required = map->num_stripes; | ||
| 2457 | max_errors = 1; | ||
| 2458 | } else if (map->type & BTRFS_BLOCK_GROUP_RAID10) { | ||
| 2459 | stripes_required = map->sub_stripes; | ||
| 2460 | max_errors = 1; | ||
| 2461 | } | ||
| 2462 | } | ||
| 2463 | if (multi_ret && rw == WRITE && | ||
| 2464 | stripes_allocated < stripes_required) { | ||
| 2465 | stripes_allocated = map->num_stripes; | ||
| 2466 | free_extent_map(em); | ||
| 2467 | kfree(multi); | ||
| 2468 | goto again; | ||
| 2469 | } | ||
| 2470 | stripe_nr = offset; | ||
| 2471 | /* | ||
| 2472 | * stripe_nr counts the total number of stripes we have to stride | ||
| 2473 | * to get to this block | ||
| 2474 | */ | ||
| 2475 | do_div(stripe_nr, map->stripe_len); | ||
| 2476 | |||
| 2477 | stripe_offset = stripe_nr * map->stripe_len; | ||
| 2478 | BUG_ON(offset < stripe_offset); | ||
| 2479 | |||
| 2480 | /* stripe_offset is the offset of this block in its stripe*/ | ||
| 2481 | stripe_offset = offset - stripe_offset; | ||
| 2482 | |||
| 2483 | if (map->type & (BTRFS_BLOCK_GROUP_RAID0 | BTRFS_BLOCK_GROUP_RAID1 | | ||
| 2484 | BTRFS_BLOCK_GROUP_RAID10 | | ||
| 2485 | BTRFS_BLOCK_GROUP_DUP)) { | ||
| 2486 | /* we limit the length of each bio to what fits in a stripe */ | ||
| 2487 | *length = min_t(u64, em->len - offset, | ||
| 2488 | map->stripe_len - stripe_offset); | ||
| 2489 | } else { | ||
| 2490 | *length = em->len - offset; | ||
| 2491 | } | ||
| 2492 | |||
| 2493 | if (!multi_ret && !unplug_page) | ||
| 2494 | goto out; | ||
| 2495 | |||
| 2496 | num_stripes = 1; | ||
| 2497 | stripe_index = 0; | ||
| 2498 | if (map->type & BTRFS_BLOCK_GROUP_RAID1) { | ||
| 2499 | if (unplug_page || (rw & (1 << BIO_RW))) | ||
| 2500 | num_stripes = map->num_stripes; | ||
| 2501 | else if (mirror_num) | ||
| 2502 | stripe_index = mirror_num - 1; | ||
| 2503 | else { | ||
| 2504 | stripe_index = find_live_mirror(map, 0, | ||
| 2505 | map->num_stripes, | ||
| 2506 | current->pid % map->num_stripes); | ||
| 2507 | } | ||
| 2508 | |||
| 2509 | } else if (map->type & BTRFS_BLOCK_GROUP_DUP) { | ||
| 2510 | if (rw & (1 << BIO_RW)) | ||
| 2511 | num_stripes = map->num_stripes; | ||
| 2512 | else if (mirror_num) | ||
| 2513 | stripe_index = mirror_num - 1; | ||
| 2514 | |||
| 2515 | } else if (map->type & BTRFS_BLOCK_GROUP_RAID10) { | ||
| 2516 | int factor = map->num_stripes / map->sub_stripes; | ||
| 2517 | |||
| 2518 | stripe_index = do_div(stripe_nr, factor); | ||
| 2519 | stripe_index *= map->sub_stripes; | ||
| 2520 | |||
| 2521 | if (unplug_page || (rw & (1 << BIO_RW))) | ||
| 2522 | num_stripes = map->sub_stripes; | ||
| 2523 | else if (mirror_num) | ||
| 2524 | stripe_index += mirror_num - 1; | ||
| 2525 | else { | ||
| 2526 | stripe_index = find_live_mirror(map, stripe_index, | ||
| 2527 | map->sub_stripes, stripe_index + | ||
| 2528 | current->pid % map->sub_stripes); | ||
| 2529 | } | ||
| 2530 | } else { | ||
| 2531 | /* | ||
| 2532 | * after this do_div call, stripe_nr is the number of stripes | ||
| 2533 | * on this device we have to walk to find the data, and | ||
| 2534 | * stripe_index is the number of our device in the stripe array | ||
| 2535 | */ | ||
| 2536 | stripe_index = do_div(stripe_nr, map->num_stripes); | ||
| 2537 | } | ||
| 2538 | BUG_ON(stripe_index >= map->num_stripes); | ||
| 2539 | |||
| 2540 | for (i = 0; i < num_stripes; i++) { | ||
| 2541 | if (unplug_page) { | ||
| 2542 | struct btrfs_device *device; | ||
| 2543 | struct backing_dev_info *bdi; | ||
| 2544 | |||
| 2545 | device = map->stripes[stripe_index].dev; | ||
| 2546 | if (device->bdev) { | ||
| 2547 | bdi = blk_get_backing_dev_info(device->bdev); | ||
| 2548 | if (bdi->unplug_io_fn) | ||
| 2549 | bdi->unplug_io_fn(bdi, unplug_page); | ||
| 2550 | } | ||
| 2551 | } else { | ||
| 2552 | multi->stripes[i].physical = | ||
| 2553 | map->stripes[stripe_index].physical + | ||
| 2554 | stripe_offset + stripe_nr * map->stripe_len; | ||
| 2555 | multi->stripes[i].dev = map->stripes[stripe_index].dev; | ||
| 2556 | } | ||
| 2557 | stripe_index++; | ||
| 2558 | } | ||
| 2559 | if (multi_ret) { | ||
| 2560 | *multi_ret = multi; | ||
| 2561 | multi->num_stripes = num_stripes; | ||
| 2562 | multi->max_errors = max_errors; | ||
| 2563 | } | ||
| 2564 | out: | ||
| 2565 | free_extent_map(em); | ||
| 2566 | return 0; | ||
| 2567 | } | ||
| 2568 | |||
| 2569 | int btrfs_map_block(struct btrfs_mapping_tree *map_tree, int rw, | ||
| 2570 | u64 logical, u64 *length, | ||
| 2571 | struct btrfs_multi_bio **multi_ret, int mirror_num) | ||
| 2572 | { | ||
| 2573 | return __btrfs_map_block(map_tree, rw, logical, length, multi_ret, | ||
| 2574 | mirror_num, NULL); | ||
| 2575 | } | ||
| 2576 | |||
| 2577 | int btrfs_rmap_block(struct btrfs_mapping_tree *map_tree, | ||
| 2578 | u64 chunk_start, u64 physical, u64 devid, | ||
| 2579 | u64 **logical, int *naddrs, int *stripe_len) | ||
| 2580 | { | ||
| 2581 | struct extent_map_tree *em_tree = &map_tree->map_tree; | ||
| 2582 | struct extent_map *em; | ||
| 2583 | struct map_lookup *map; | ||
| 2584 | u64 *buf; | ||
| 2585 | u64 bytenr; | ||
| 2586 | u64 length; | ||
| 2587 | u64 stripe_nr; | ||
| 2588 | int i, j, nr = 0; | ||
| 2589 | |||
| 2590 | spin_lock(&em_tree->lock); | ||
| 2591 | em = lookup_extent_mapping(em_tree, chunk_start, 1); | ||
| 2592 | spin_unlock(&em_tree->lock); | ||
| 2593 | |||
| 2594 | BUG_ON(!em || em->start != chunk_start); | ||
| 2595 | map = (struct map_lookup *)em->bdev; | ||
| 2596 | |||
| 2597 | length = em->len; | ||
| 2598 | if (map->type & BTRFS_BLOCK_GROUP_RAID10) | ||
| 2599 | do_div(length, map->num_stripes / map->sub_stripes); | ||
| 2600 | else if (map->type & BTRFS_BLOCK_GROUP_RAID0) | ||
| 2601 | do_div(length, map->num_stripes); | ||
| 2602 | |||
| 2603 | buf = kzalloc(sizeof(u64) * map->num_stripes, GFP_NOFS); | ||
| 2604 | BUG_ON(!buf); | ||
| 2605 | |||
| 2606 | for (i = 0; i < map->num_stripes; i++) { | ||
| 2607 | if (devid && map->stripes[i].dev->devid != devid) | ||
| 2608 | continue; | ||
| 2609 | if (map->stripes[i].physical > physical || | ||
| 2610 | map->stripes[i].physical + length <= physical) | ||
| 2611 | continue; | ||
| 2612 | |||
| 2613 | stripe_nr = physical - map->stripes[i].physical; | ||
| 2614 | do_div(stripe_nr, map->stripe_len); | ||
| 2615 | |||
| 2616 | if (map->type & BTRFS_BLOCK_GROUP_RAID10) { | ||
| 2617 | stripe_nr = stripe_nr * map->num_stripes + i; | ||
| 2618 | do_div(stripe_nr, map->sub_stripes); | ||
| 2619 | } else if (map->type & BTRFS_BLOCK_GROUP_RAID0) { | ||
| 2620 | stripe_nr = stripe_nr * map->num_stripes + i; | ||
| 2621 | } | ||
| 2622 | bytenr = chunk_start + stripe_nr * map->stripe_len; | ||
| 2623 | WARN_ON(nr >= map->num_stripes); | ||
| 2624 | for (j = 0; j < nr; j++) { | ||
| 2625 | if (buf[j] == bytenr) | ||
| 2626 | break; | ||
| 2627 | } | ||
| 2628 | if (j == nr) { | ||
| 2629 | WARN_ON(nr >= map->num_stripes); | ||
| 2630 | buf[nr++] = bytenr; | ||
| 2631 | } | ||
| 2632 | } | ||
| 2633 | |||
| 2634 | for (i = 0; i > nr; i++) { | ||
| 2635 | struct btrfs_multi_bio *multi; | ||
| 2636 | struct btrfs_bio_stripe *stripe; | ||
| 2637 | int ret; | ||
| 2638 | |||
| 2639 | length = 1; | ||
| 2640 | ret = btrfs_map_block(map_tree, WRITE, buf[i], | ||
| 2641 | &length, &multi, 0); | ||
| 2642 | BUG_ON(ret); | ||
| 2643 | |||
| 2644 | stripe = multi->stripes; | ||
| 2645 | for (j = 0; j < multi->num_stripes; j++) { | ||
| 2646 | if (stripe->physical >= physical && | ||
| 2647 | physical < stripe->physical + length) | ||
| 2648 | break; | ||
| 2649 | } | ||
| 2650 | BUG_ON(j >= multi->num_stripes); | ||
| 2651 | kfree(multi); | ||
| 2652 | } | ||
| 2653 | |||
| 2654 | *logical = buf; | ||
| 2655 | *naddrs = nr; | ||
| 2656 | *stripe_len = map->stripe_len; | ||
| 2657 | |||
| 2658 | free_extent_map(em); | ||
| 2659 | return 0; | ||
| 2660 | } | ||
| 2661 | |||
| 2662 | int btrfs_unplug_page(struct btrfs_mapping_tree *map_tree, | ||
| 2663 | u64 logical, struct page *page) | ||
| 2664 | { | ||
| 2665 | u64 length = PAGE_CACHE_SIZE; | ||
| 2666 | return __btrfs_map_block(map_tree, READ, logical, &length, | ||
| 2667 | NULL, 0, page); | ||
| 2668 | } | ||
| 2669 | |||
| 2670 | static void end_bio_multi_stripe(struct bio *bio, int err) | ||
| 2671 | { | ||
| 2672 | struct btrfs_multi_bio *multi = bio->bi_private; | ||
| 2673 | int is_orig_bio = 0; | ||
| 2674 | |||
| 2675 | if (err) | ||
| 2676 | atomic_inc(&multi->error); | ||
| 2677 | |||
| 2678 | if (bio == multi->orig_bio) | ||
| 2679 | is_orig_bio = 1; | ||
| 2680 | |||
| 2681 | if (atomic_dec_and_test(&multi->stripes_pending)) { | ||
| 2682 | if (!is_orig_bio) { | ||
| 2683 | bio_put(bio); | ||
| 2684 | bio = multi->orig_bio; | ||
| 2685 | } | ||
| 2686 | bio->bi_private = multi->private; | ||
| 2687 | bio->bi_end_io = multi->end_io; | ||
| 2688 | /* only send an error to the higher layers if it is | ||
| 2689 | * beyond the tolerance of the multi-bio | ||
| 2690 | */ | ||
| 2691 | if (atomic_read(&multi->error) > multi->max_errors) { | ||
| 2692 | err = -EIO; | ||
| 2693 | } else if (err) { | ||
| 2694 | /* | ||
| 2695 | * this bio is actually up to date, we didn't | ||
| 2696 | * go over the max number of errors | ||
| 2697 | */ | ||
| 2698 | set_bit(BIO_UPTODATE, &bio->bi_flags); | ||
| 2699 | err = 0; | ||
| 2700 | } | ||
| 2701 | kfree(multi); | ||
| 2702 | |||
| 2703 | bio_endio(bio, err); | ||
| 2704 | } else if (!is_orig_bio) { | ||
| 2705 | bio_put(bio); | ||
| 2706 | } | ||
| 2707 | } | ||
| 2708 | |||
| 2709 | struct async_sched { | ||
| 2710 | struct bio *bio; | ||
| 2711 | int rw; | ||
| 2712 | struct btrfs_fs_info *info; | ||
| 2713 | struct btrfs_work work; | ||
| 2714 | }; | ||
| 2715 | |||
| 2716 | /* | ||
| 2717 | * see run_scheduled_bios for a description of why bios are collected for | ||
| 2718 | * async submit. | ||
| 2719 | * | ||
| 2720 | * This will add one bio to the pending list for a device and make sure | ||
| 2721 | * the work struct is scheduled. | ||
| 2722 | */ | ||
| 2723 | static noinline int schedule_bio(struct btrfs_root *root, | ||
| 2724 | struct btrfs_device *device, | ||
| 2725 | int rw, struct bio *bio) | ||
| 2726 | { | ||
| 2727 | int should_queue = 1; | ||
| 2728 | |||
| 2729 | /* don't bother with additional async steps for reads, right now */ | ||
| 2730 | if (!(rw & (1 << BIO_RW))) { | ||
| 2731 | bio_get(bio); | ||
| 2732 | submit_bio(rw, bio); | ||
| 2733 | bio_put(bio); | ||
| 2734 | return 0; | ||
| 2735 | } | ||
| 2736 | |||
| 2737 | /* | ||
| 2738 | * nr_async_bios allows us to reliably return congestion to the | ||
| 2739 | * higher layers. Otherwise, the async bio makes it appear we have | ||
| 2740 | * made progress against dirty pages when we've really just put it | ||
| 2741 | * on a queue for later | ||
| 2742 | */ | ||
| 2743 | atomic_inc(&root->fs_info->nr_async_bios); | ||
| 2744 | WARN_ON(bio->bi_next); | ||
| 2745 | bio->bi_next = NULL; | ||
| 2746 | bio->bi_rw |= rw; | ||
| 2747 | |||
| 2748 | spin_lock(&device->io_lock); | ||
| 2749 | |||
| 2750 | if (device->pending_bio_tail) | ||
| 2751 | device->pending_bio_tail->bi_next = bio; | ||
| 2752 | |||
| 2753 | device->pending_bio_tail = bio; | ||
| 2754 | if (!device->pending_bios) | ||
| 2755 | device->pending_bios = bio; | ||
| 2756 | if (device->running_pending) | ||
| 2757 | should_queue = 0; | ||
| 2758 | |||
| 2759 | spin_unlock(&device->io_lock); | ||
| 2760 | |||
| 2761 | if (should_queue) | ||
| 2762 | btrfs_queue_worker(&root->fs_info->submit_workers, | ||
| 2763 | &device->work); | ||
| 2764 | return 0; | ||
| 2765 | } | ||
| 2766 | |||
| 2767 | int btrfs_map_bio(struct btrfs_root *root, int rw, struct bio *bio, | ||
| 2768 | int mirror_num, int async_submit) | ||
| 2769 | { | ||
| 2770 | struct btrfs_mapping_tree *map_tree; | ||
| 2771 | struct btrfs_device *dev; | ||
| 2772 | struct bio *first_bio = bio; | ||
| 2773 | u64 logical = (u64)bio->bi_sector << 9; | ||
| 2774 | u64 length = 0; | ||
| 2775 | u64 map_length; | ||
| 2776 | struct btrfs_multi_bio *multi = NULL; | ||
| 2777 | int ret; | ||
| 2778 | int dev_nr = 0; | ||
| 2779 | int total_devs = 1; | ||
| 2780 | |||
| 2781 | length = bio->bi_size; | ||
| 2782 | map_tree = &root->fs_info->mapping_tree; | ||
| 2783 | map_length = length; | ||
| 2784 | |||
| 2785 | ret = btrfs_map_block(map_tree, rw, logical, &map_length, &multi, | ||
| 2786 | mirror_num); | ||
| 2787 | BUG_ON(ret); | ||
| 2788 | |||
| 2789 | total_devs = multi->num_stripes; | ||
| 2790 | if (map_length < length) { | ||
| 2791 | printk(KERN_CRIT "mapping failed logical %llu bio len %llu " | ||
| 2792 | "len %llu\n", (unsigned long long)logical, | ||
| 2793 | (unsigned long long)length, | ||
| 2794 | (unsigned long long)map_length); | ||
| 2795 | BUG(); | ||
| 2796 | } | ||
| 2797 | multi->end_io = first_bio->bi_end_io; | ||
| 2798 | multi->private = first_bio->bi_private; | ||
| 2799 | multi->orig_bio = first_bio; | ||
| 2800 | atomic_set(&multi->stripes_pending, multi->num_stripes); | ||
| 2801 | |||
| 2802 | while (dev_nr < total_devs) { | ||
| 2803 | if (total_devs > 1) { | ||
| 2804 | if (dev_nr < total_devs - 1) { | ||
| 2805 | bio = bio_clone(first_bio, GFP_NOFS); | ||
| 2806 | BUG_ON(!bio); | ||
| 2807 | } else { | ||
| 2808 | bio = first_bio; | ||
| 2809 | } | ||
| 2810 | bio->bi_private = multi; | ||
| 2811 | bio->bi_end_io = end_bio_multi_stripe; | ||
| 2812 | } | ||
| 2813 | bio->bi_sector = multi->stripes[dev_nr].physical >> 9; | ||
| 2814 | dev = multi->stripes[dev_nr].dev; | ||
| 2815 | BUG_ON(rw == WRITE && !dev->writeable); | ||
| 2816 | if (dev && dev->bdev) { | ||
| 2817 | bio->bi_bdev = dev->bdev; | ||
| 2818 | if (async_submit) | ||
| 2819 | schedule_bio(root, dev, rw, bio); | ||
| 2820 | else | ||
| 2821 | submit_bio(rw, bio); | ||
| 2822 | } else { | ||
| 2823 | bio->bi_bdev = root->fs_info->fs_devices->latest_bdev; | ||
| 2824 | bio->bi_sector = logical >> 9; | ||
| 2825 | bio_endio(bio, -EIO); | ||
| 2826 | } | ||
| 2827 | dev_nr++; | ||
| 2828 | } | ||
| 2829 | if (total_devs == 1) | ||
| 2830 | kfree(multi); | ||
| 2831 | return 0; | ||
| 2832 | } | ||
| 2833 | |||
| 2834 | struct btrfs_device *btrfs_find_device(struct btrfs_root *root, u64 devid, | ||
| 2835 | u8 *uuid, u8 *fsid) | ||
| 2836 | { | ||
| 2837 | struct btrfs_device *device; | ||
| 2838 | struct btrfs_fs_devices *cur_devices; | ||
| 2839 | |||
| 2840 | cur_devices = root->fs_info->fs_devices; | ||
| 2841 | while (cur_devices) { | ||
| 2842 | if (!fsid || | ||
| 2843 | !memcmp(cur_devices->fsid, fsid, BTRFS_UUID_SIZE)) { | ||
| 2844 | device = __find_device(&cur_devices->devices, | ||
| 2845 | devid, uuid); | ||
| 2846 | if (device) | ||
| 2847 | return device; | ||
| 2848 | } | ||
| 2849 | cur_devices = cur_devices->seed; | ||
| 2850 | } | ||
| 2851 | return NULL; | ||
| 2852 | } | ||
| 2853 | |||
| 2854 | static struct btrfs_device *add_missing_dev(struct btrfs_root *root, | ||
| 2855 | u64 devid, u8 *dev_uuid) | ||
| 2856 | { | ||
| 2857 | struct btrfs_device *device; | ||
| 2858 | struct btrfs_fs_devices *fs_devices = root->fs_info->fs_devices; | ||
| 2859 | |||
| 2860 | device = kzalloc(sizeof(*device), GFP_NOFS); | ||
| 2861 | if (!device) | ||
| 2862 | return NULL; | ||
| 2863 | list_add(&device->dev_list, | ||
| 2864 | &fs_devices->devices); | ||
| 2865 | device->barriers = 1; | ||
| 2866 | device->dev_root = root->fs_info->dev_root; | ||
| 2867 | device->devid = devid; | ||
| 2868 | device->work.func = pending_bios_fn; | ||
| 2869 | device->fs_devices = fs_devices; | ||
| 2870 | fs_devices->num_devices++; | ||
| 2871 | spin_lock_init(&device->io_lock); | ||
| 2872 | INIT_LIST_HEAD(&device->dev_alloc_list); | ||
| 2873 | memcpy(device->uuid, dev_uuid, BTRFS_UUID_SIZE); | ||
| 2874 | return device; | ||
| 2875 | } | ||
| 2876 | |||
| 2877 | static int read_one_chunk(struct btrfs_root *root, struct btrfs_key *key, | ||
| 2878 | struct extent_buffer *leaf, | ||
| 2879 | struct btrfs_chunk *chunk) | ||
| 2880 | { | ||
| 2881 | struct btrfs_mapping_tree *map_tree = &root->fs_info->mapping_tree; | ||
| 2882 | struct map_lookup *map; | ||
| 2883 | struct extent_map *em; | ||
| 2884 | u64 logical; | ||
| 2885 | u64 length; | ||
| 2886 | u64 devid; | ||
| 2887 | u8 uuid[BTRFS_UUID_SIZE]; | ||
| 2888 | int num_stripes; | ||
| 2889 | int ret; | ||
| 2890 | int i; | ||
| 2891 | |||
| 2892 | logical = key->offset; | ||
| 2893 | length = btrfs_chunk_length(leaf, chunk); | ||
| 2894 | |||
| 2895 | spin_lock(&map_tree->map_tree.lock); | ||
| 2896 | em = lookup_extent_mapping(&map_tree->map_tree, logical, 1); | ||
| 2897 | spin_unlock(&map_tree->map_tree.lock); | ||
| 2898 | |||
| 2899 | /* already mapped? */ | ||
| 2900 | if (em && em->start <= logical && em->start + em->len > logical) { | ||
| 2901 | free_extent_map(em); | ||
| 2902 | return 0; | ||
| 2903 | } else if (em) { | ||
| 2904 | free_extent_map(em); | ||
| 2905 | } | ||
| 2906 | |||
| 2907 | map = kzalloc(sizeof(*map), GFP_NOFS); | ||
| 2908 | if (!map) | ||
| 2909 | return -ENOMEM; | ||
| 2910 | |||
| 2911 | em = alloc_extent_map(GFP_NOFS); | ||
| 2912 | if (!em) | ||
| 2913 | return -ENOMEM; | ||
| 2914 | num_stripes = btrfs_chunk_num_stripes(leaf, chunk); | ||
| 2915 | map = kmalloc(map_lookup_size(num_stripes), GFP_NOFS); | ||
| 2916 | if (!map) { | ||
| 2917 | free_extent_map(em); | ||
| 2918 | return -ENOMEM; | ||
| 2919 | } | ||
| 2920 | |||
| 2921 | em->bdev = (struct block_device *)map; | ||
| 2922 | em->start = logical; | ||
| 2923 | em->len = length; | ||
| 2924 | em->block_start = 0; | ||
| 2925 | em->block_len = em->len; | ||
| 2926 | |||
| 2927 | map->num_stripes = num_stripes; | ||
| 2928 | map->io_width = btrfs_chunk_io_width(leaf, chunk); | ||
| 2929 | map->io_align = btrfs_chunk_io_align(leaf, chunk); | ||
| 2930 | map->sector_size = btrfs_chunk_sector_size(leaf, chunk); | ||
| 2931 | map->stripe_len = btrfs_chunk_stripe_len(leaf, chunk); | ||
| 2932 | map->type = btrfs_chunk_type(leaf, chunk); | ||
| 2933 | map->sub_stripes = btrfs_chunk_sub_stripes(leaf, chunk); | ||
| 2934 | for (i = 0; i < num_stripes; i++) { | ||
| 2935 | map->stripes[i].physical = | ||
| 2936 | btrfs_stripe_offset_nr(leaf, chunk, i); | ||
| 2937 | devid = btrfs_stripe_devid_nr(leaf, chunk, i); | ||
| 2938 | read_extent_buffer(leaf, uuid, (unsigned long) | ||
| 2939 | btrfs_stripe_dev_uuid_nr(chunk, i), | ||
| 2940 | BTRFS_UUID_SIZE); | ||
| 2941 | map->stripes[i].dev = btrfs_find_device(root, devid, uuid, | ||
| 2942 | NULL); | ||
| 2943 | if (!map->stripes[i].dev && !btrfs_test_opt(root, DEGRADED)) { | ||
| 2944 | kfree(map); | ||
| 2945 | free_extent_map(em); | ||
| 2946 | return -EIO; | ||
| 2947 | } | ||
| 2948 | if (!map->stripes[i].dev) { | ||
| 2949 | map->stripes[i].dev = | ||
| 2950 | add_missing_dev(root, devid, uuid); | ||
| 2951 | if (!map->stripes[i].dev) { | ||
| 2952 | kfree(map); | ||
| 2953 | free_extent_map(em); | ||
| 2954 | return -EIO; | ||
| 2955 | } | ||
| 2956 | } | ||
| 2957 | map->stripes[i].dev->in_fs_metadata = 1; | ||
| 2958 | } | ||
| 2959 | |||
| 2960 | spin_lock(&map_tree->map_tree.lock); | ||
| 2961 | ret = add_extent_mapping(&map_tree->map_tree, em); | ||
| 2962 | spin_unlock(&map_tree->map_tree.lock); | ||
| 2963 | BUG_ON(ret); | ||
| 2964 | free_extent_map(em); | ||
| 2965 | |||
| 2966 | return 0; | ||
| 2967 | } | ||
| 2968 | |||
| 2969 | static int fill_device_from_item(struct extent_buffer *leaf, | ||
| 2970 | struct btrfs_dev_item *dev_item, | ||
| 2971 | struct btrfs_device *device) | ||
| 2972 | { | ||
| 2973 | unsigned long ptr; | ||
| 2974 | |||
| 2975 | device->devid = btrfs_device_id(leaf, dev_item); | ||
| 2976 | device->total_bytes = btrfs_device_total_bytes(leaf, dev_item); | ||
| 2977 | device->bytes_used = btrfs_device_bytes_used(leaf, dev_item); | ||
| 2978 | device->type = btrfs_device_type(leaf, dev_item); | ||
| 2979 | device->io_align = btrfs_device_io_align(leaf, dev_item); | ||
| 2980 | device->io_width = btrfs_device_io_width(leaf, dev_item); | ||
| 2981 | device->sector_size = btrfs_device_sector_size(leaf, dev_item); | ||
| 2982 | |||
| 2983 | ptr = (unsigned long)btrfs_device_uuid(dev_item); | ||
| 2984 | read_extent_buffer(leaf, device->uuid, ptr, BTRFS_UUID_SIZE); | ||
| 2985 | |||
| 2986 | return 0; | ||
| 2987 | } | ||
| 2988 | |||
| 2989 | static int open_seed_devices(struct btrfs_root *root, u8 *fsid) | ||
| 2990 | { | ||
| 2991 | struct btrfs_fs_devices *fs_devices; | ||
| 2992 | int ret; | ||
| 2993 | |||
| 2994 | mutex_lock(&uuid_mutex); | ||
| 2995 | |||
| 2996 | fs_devices = root->fs_info->fs_devices->seed; | ||
| 2997 | while (fs_devices) { | ||
| 2998 | if (!memcmp(fs_devices->fsid, fsid, BTRFS_UUID_SIZE)) { | ||
| 2999 | ret = 0; | ||
| 3000 | goto out; | ||
| 3001 | } | ||
| 3002 | fs_devices = fs_devices->seed; | ||
| 3003 | } | ||
| 3004 | |||
| 3005 | fs_devices = find_fsid(fsid); | ||
| 3006 | if (!fs_devices) { | ||
| 3007 | ret = -ENOENT; | ||
| 3008 | goto out; | ||
| 3009 | } | ||
| 3010 | |||
| 3011 | fs_devices = clone_fs_devices(fs_devices); | ||
| 3012 | if (IS_ERR(fs_devices)) { | ||
| 3013 | ret = PTR_ERR(fs_devices); | ||
| 3014 | goto out; | ||
| 3015 | } | ||
| 3016 | |||
| 3017 | ret = __btrfs_open_devices(fs_devices, FMODE_READ, | ||
| 3018 | root->fs_info->bdev_holder); | ||
| 3019 | if (ret) | ||
| 3020 | goto out; | ||
| 3021 | |||
| 3022 | if (!fs_devices->seeding) { | ||
| 3023 | __btrfs_close_devices(fs_devices); | ||
| 3024 | free_fs_devices(fs_devices); | ||
| 3025 | ret = -EINVAL; | ||
| 3026 | goto out; | ||
| 3027 | } | ||
| 3028 | |||
| 3029 | fs_devices->seed = root->fs_info->fs_devices->seed; | ||
| 3030 | root->fs_info->fs_devices->seed = fs_devices; | ||
| 3031 | out: | ||
| 3032 | mutex_unlock(&uuid_mutex); | ||
| 3033 | return ret; | ||
| 3034 | } | ||
| 3035 | |||
| 3036 | static int read_one_dev(struct btrfs_root *root, | ||
| 3037 | struct extent_buffer *leaf, | ||
| 3038 | struct btrfs_dev_item *dev_item) | ||
| 3039 | { | ||
| 3040 | struct btrfs_device *device; | ||
| 3041 | u64 devid; | ||
| 3042 | int ret; | ||
| 3043 | u8 fs_uuid[BTRFS_UUID_SIZE]; | ||
| 3044 | u8 dev_uuid[BTRFS_UUID_SIZE]; | ||
| 3045 | |||
| 3046 | devid = btrfs_device_id(leaf, dev_item); | ||
| 3047 | read_extent_buffer(leaf, dev_uuid, | ||
| 3048 | (unsigned long)btrfs_device_uuid(dev_item), | ||
| 3049 | BTRFS_UUID_SIZE); | ||
| 3050 | read_extent_buffer(leaf, fs_uuid, | ||
| 3051 | (unsigned long)btrfs_device_fsid(dev_item), | ||
| 3052 | BTRFS_UUID_SIZE); | ||
| 3053 | |||
| 3054 | if (memcmp(fs_uuid, root->fs_info->fsid, BTRFS_UUID_SIZE)) { | ||
| 3055 | ret = open_seed_devices(root, fs_uuid); | ||
| 3056 | if (ret && !btrfs_test_opt(root, DEGRADED)) | ||
| 3057 | return ret; | ||
| 3058 | } | ||
| 3059 | |||
| 3060 | device = btrfs_find_device(root, devid, dev_uuid, fs_uuid); | ||
| 3061 | if (!device || !device->bdev) { | ||
| 3062 | if (!btrfs_test_opt(root, DEGRADED)) | ||
| 3063 | return -EIO; | ||
| 3064 | |||
| 3065 | if (!device) { | ||
| 3066 | printk(KERN_WARNING "warning devid %llu missing\n", | ||
| 3067 | (unsigned long long)devid); | ||
| 3068 | device = add_missing_dev(root, devid, dev_uuid); | ||
| 3069 | if (!device) | ||
| 3070 | return -ENOMEM; | ||
| 3071 | } | ||
| 3072 | } | ||
| 3073 | |||
| 3074 | if (device->fs_devices != root->fs_info->fs_devices) { | ||
| 3075 | BUG_ON(device->writeable); | ||
| 3076 | if (device->generation != | ||
| 3077 | btrfs_device_generation(leaf, dev_item)) | ||
| 3078 | return -EINVAL; | ||
| 3079 | } | ||
| 3080 | |||
| 3081 | fill_device_from_item(leaf, dev_item, device); | ||
| 3082 | device->dev_root = root->fs_info->dev_root; | ||
| 3083 | device->in_fs_metadata = 1; | ||
| 3084 | if (device->writeable) | ||
| 3085 | device->fs_devices->total_rw_bytes += device->total_bytes; | ||
| 3086 | ret = 0; | ||
| 3087 | return ret; | ||
| 3088 | } | ||
| 3089 | |||
| 3090 | int btrfs_read_super_device(struct btrfs_root *root, struct extent_buffer *buf) | ||
| 3091 | { | ||
| 3092 | struct btrfs_dev_item *dev_item; | ||
| 3093 | |||
| 3094 | dev_item = (struct btrfs_dev_item *)offsetof(struct btrfs_super_block, | ||
| 3095 | dev_item); | ||
| 3096 | return read_one_dev(root, buf, dev_item); | ||
| 3097 | } | ||
| 3098 | |||
| 3099 | int btrfs_read_sys_array(struct btrfs_root *root) | ||
| 3100 | { | ||
| 3101 | struct btrfs_super_block *super_copy = &root->fs_info->super_copy; | ||
| 3102 | struct extent_buffer *sb; | ||
| 3103 | struct btrfs_disk_key *disk_key; | ||
| 3104 | struct btrfs_chunk *chunk; | ||
| 3105 | u8 *ptr; | ||
| 3106 | unsigned long sb_ptr; | ||
| 3107 | int ret = 0; | ||
| 3108 | u32 num_stripes; | ||
| 3109 | u32 array_size; | ||
| 3110 | u32 len = 0; | ||
| 3111 | u32 cur; | ||
| 3112 | struct btrfs_key key; | ||
| 3113 | |||
| 3114 | sb = btrfs_find_create_tree_block(root, BTRFS_SUPER_INFO_OFFSET, | ||
| 3115 | BTRFS_SUPER_INFO_SIZE); | ||
| 3116 | if (!sb) | ||
| 3117 | return -ENOMEM; | ||
| 3118 | btrfs_set_buffer_uptodate(sb); | ||
| 3119 | write_extent_buffer(sb, super_copy, 0, BTRFS_SUPER_INFO_SIZE); | ||
| 3120 | array_size = btrfs_super_sys_array_size(super_copy); | ||
| 3121 | |||
| 3122 | ptr = super_copy->sys_chunk_array; | ||
| 3123 | sb_ptr = offsetof(struct btrfs_super_block, sys_chunk_array); | ||
| 3124 | cur = 0; | ||
| 3125 | |||
| 3126 | while (cur < array_size) { | ||
| 3127 | disk_key = (struct btrfs_disk_key *)ptr; | ||
| 3128 | btrfs_disk_key_to_cpu(&key, disk_key); | ||
| 3129 | |||
| 3130 | len = sizeof(*disk_key); ptr += len; | ||
| 3131 | sb_ptr += len; | ||
| 3132 | cur += len; | ||
| 3133 | |||
| 3134 | if (key.type == BTRFS_CHUNK_ITEM_KEY) { | ||
| 3135 | chunk = (struct btrfs_chunk *)sb_ptr; | ||
| 3136 | ret = read_one_chunk(root, &key, sb, chunk); | ||
| 3137 | if (ret) | ||
| 3138 | break; | ||
| 3139 | num_stripes = btrfs_chunk_num_stripes(sb, chunk); | ||
| 3140 | len = btrfs_chunk_item_size(num_stripes); | ||
| 3141 | } else { | ||
| 3142 | ret = -EIO; | ||
| 3143 | break; | ||
| 3144 | } | ||
| 3145 | ptr += len; | ||
| 3146 | sb_ptr += len; | ||
| 3147 | cur += len; | ||
| 3148 | } | ||
| 3149 | free_extent_buffer(sb); | ||
| 3150 | return ret; | ||
| 3151 | } | ||
| 3152 | |||
| 3153 | int btrfs_read_chunk_tree(struct btrfs_root *root) | ||
| 3154 | { | ||
| 3155 | struct btrfs_path *path; | ||
| 3156 | struct extent_buffer *leaf; | ||
| 3157 | struct btrfs_key key; | ||
| 3158 | struct btrfs_key found_key; | ||
| 3159 | int ret; | ||
| 3160 | int slot; | ||
| 3161 | |||
| 3162 | root = root->fs_info->chunk_root; | ||
| 3163 | |||
| 3164 | path = btrfs_alloc_path(); | ||
| 3165 | if (!path) | ||
| 3166 | return -ENOMEM; | ||
| 3167 | |||
| 3168 | /* first we search for all of the device items, and then we | ||
| 3169 | * read in all of the chunk items. This way we can create chunk | ||
| 3170 | * mappings that reference all of the devices that are afound | ||
| 3171 | */ | ||
| 3172 | key.objectid = BTRFS_DEV_ITEMS_OBJECTID; | ||
| 3173 | key.offset = 0; | ||
| 3174 | key.type = 0; | ||
| 3175 | again: | ||
| 3176 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 3177 | while (1) { | ||
| 3178 | leaf = path->nodes[0]; | ||
| 3179 | slot = path->slots[0]; | ||
| 3180 | if (slot >= btrfs_header_nritems(leaf)) { | ||
| 3181 | ret = btrfs_next_leaf(root, path); | ||
| 3182 | if (ret == 0) | ||
| 3183 | continue; | ||
| 3184 | if (ret < 0) | ||
| 3185 | goto error; | ||
| 3186 | break; | ||
| 3187 | } | ||
| 3188 | btrfs_item_key_to_cpu(leaf, &found_key, slot); | ||
| 3189 | if (key.objectid == BTRFS_DEV_ITEMS_OBJECTID) { | ||
| 3190 | if (found_key.objectid != BTRFS_DEV_ITEMS_OBJECTID) | ||
| 3191 | break; | ||
| 3192 | if (found_key.type == BTRFS_DEV_ITEM_KEY) { | ||
| 3193 | struct btrfs_dev_item *dev_item; | ||
| 3194 | dev_item = btrfs_item_ptr(leaf, slot, | ||
| 3195 | struct btrfs_dev_item); | ||
| 3196 | ret = read_one_dev(root, leaf, dev_item); | ||
| 3197 | if (ret) | ||
| 3198 | goto error; | ||
| 3199 | } | ||
| 3200 | } else if (found_key.type == BTRFS_CHUNK_ITEM_KEY) { | ||
| 3201 | struct btrfs_chunk *chunk; | ||
| 3202 | chunk = btrfs_item_ptr(leaf, slot, struct btrfs_chunk); | ||
| 3203 | ret = read_one_chunk(root, &found_key, leaf, chunk); | ||
| 3204 | if (ret) | ||
| 3205 | goto error; | ||
| 3206 | } | ||
| 3207 | path->slots[0]++; | ||
| 3208 | } | ||
| 3209 | if (key.objectid == BTRFS_DEV_ITEMS_OBJECTID) { | ||
| 3210 | key.objectid = 0; | ||
| 3211 | btrfs_release_path(root, path); | ||
| 3212 | goto again; | ||
| 3213 | } | ||
| 3214 | ret = 0; | ||
| 3215 | error: | ||
| 3216 | btrfs_free_path(path); | ||
| 3217 | return ret; | ||
| 3218 | } | ||
diff --git a/fs/btrfs/volumes.h b/fs/btrfs/volumes.h new file mode 100644 index 000000000000..86c44e9ae110 --- /dev/null +++ b/fs/btrfs/volumes.h | |||
| @@ -0,0 +1,162 @@ | |||
| 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 | |||
| 19 | #ifndef __BTRFS_VOLUMES_ | ||
| 20 | #define __BTRFS_VOLUMES_ | ||
| 21 | |||
| 22 | #include <linux/bio.h> | ||
| 23 | #include "async-thread.h" | ||
| 24 | |||
| 25 | struct buffer_head; | ||
| 26 | struct btrfs_device { | ||
| 27 | struct list_head dev_list; | ||
| 28 | struct list_head dev_alloc_list; | ||
| 29 | struct btrfs_fs_devices *fs_devices; | ||
| 30 | struct btrfs_root *dev_root; | ||
| 31 | struct bio *pending_bios; | ||
| 32 | struct bio *pending_bio_tail; | ||
| 33 | int running_pending; | ||
| 34 | u64 generation; | ||
| 35 | |||
| 36 | int barriers; | ||
| 37 | int writeable; | ||
| 38 | int in_fs_metadata; | ||
| 39 | |||
| 40 | spinlock_t io_lock; | ||
| 41 | |||
| 42 | struct block_device *bdev; | ||
| 43 | |||
| 44 | /* the mode sent to open_bdev_exclusive */ | ||
| 45 | fmode_t mode; | ||
| 46 | |||
| 47 | char *name; | ||
| 48 | |||
| 49 | /* the internal btrfs device id */ | ||
| 50 | u64 devid; | ||
| 51 | |||
| 52 | /* size of the device */ | ||
| 53 | u64 total_bytes; | ||
| 54 | |||
| 55 | /* bytes used */ | ||
| 56 | u64 bytes_used; | ||
| 57 | |||
| 58 | /* optimal io alignment for this device */ | ||
| 59 | u32 io_align; | ||
| 60 | |||
| 61 | /* optimal io width for this device */ | ||
| 62 | u32 io_width; | ||
| 63 | |||
| 64 | /* minimal io size for this device */ | ||
| 65 | u32 sector_size; | ||
| 66 | |||
| 67 | /* type and info about this device */ | ||
| 68 | u64 type; | ||
| 69 | |||
| 70 | /* physical drive uuid (or lvm uuid) */ | ||
| 71 | u8 uuid[BTRFS_UUID_SIZE]; | ||
| 72 | |||
| 73 | struct btrfs_work work; | ||
| 74 | }; | ||
| 75 | |||
| 76 | struct btrfs_fs_devices { | ||
| 77 | u8 fsid[BTRFS_FSID_SIZE]; /* FS specific uuid */ | ||
| 78 | |||
| 79 | /* the device with this id has the most recent coyp of the super */ | ||
| 80 | u64 latest_devid; | ||
| 81 | u64 latest_trans; | ||
| 82 | u64 num_devices; | ||
| 83 | u64 open_devices; | ||
| 84 | u64 rw_devices; | ||
| 85 | u64 total_rw_bytes; | ||
| 86 | struct block_device *latest_bdev; | ||
| 87 | /* all of the devices in the FS */ | ||
| 88 | struct list_head devices; | ||
| 89 | |||
| 90 | /* devices not currently being allocated */ | ||
| 91 | struct list_head alloc_list; | ||
| 92 | struct list_head list; | ||
| 93 | |||
| 94 | struct btrfs_fs_devices *seed; | ||
| 95 | int seeding; | ||
| 96 | |||
| 97 | int opened; | ||
| 98 | }; | ||
| 99 | |||
| 100 | struct btrfs_bio_stripe { | ||
| 101 | struct btrfs_device *dev; | ||
| 102 | u64 physical; | ||
| 103 | }; | ||
| 104 | |||
| 105 | struct btrfs_multi_bio { | ||
| 106 | atomic_t stripes_pending; | ||
| 107 | bio_end_io_t *end_io; | ||
| 108 | struct bio *orig_bio; | ||
| 109 | void *private; | ||
| 110 | atomic_t error; | ||
| 111 | int max_errors; | ||
| 112 | int num_stripes; | ||
| 113 | struct btrfs_bio_stripe stripes[]; | ||
| 114 | }; | ||
| 115 | |||
| 116 | #define btrfs_multi_bio_size(n) (sizeof(struct btrfs_multi_bio) + \ | ||
| 117 | (sizeof(struct btrfs_bio_stripe) * (n))) | ||
| 118 | |||
| 119 | int btrfs_alloc_dev_extent(struct btrfs_trans_handle *trans, | ||
| 120 | struct btrfs_device *device, | ||
| 121 | u64 chunk_tree, u64 chunk_objectid, | ||
| 122 | u64 chunk_offset, u64 start, u64 num_bytes); | ||
| 123 | int btrfs_map_block(struct btrfs_mapping_tree *map_tree, int rw, | ||
| 124 | u64 logical, u64 *length, | ||
| 125 | struct btrfs_multi_bio **multi_ret, int mirror_num); | ||
| 126 | int btrfs_rmap_block(struct btrfs_mapping_tree *map_tree, | ||
| 127 | u64 chunk_start, u64 physical, u64 devid, | ||
| 128 | u64 **logical, int *naddrs, int *stripe_len); | ||
| 129 | int btrfs_read_sys_array(struct btrfs_root *root); | ||
| 130 | int btrfs_read_chunk_tree(struct btrfs_root *root); | ||
| 131 | int btrfs_alloc_chunk(struct btrfs_trans_handle *trans, | ||
| 132 | struct btrfs_root *extent_root, u64 type); | ||
| 133 | void btrfs_mapping_init(struct btrfs_mapping_tree *tree); | ||
| 134 | void btrfs_mapping_tree_free(struct btrfs_mapping_tree *tree); | ||
| 135 | int btrfs_map_bio(struct btrfs_root *root, int rw, struct bio *bio, | ||
| 136 | int mirror_num, int async_submit); | ||
| 137 | int btrfs_read_super_device(struct btrfs_root *root, struct extent_buffer *buf); | ||
| 138 | int btrfs_open_devices(struct btrfs_fs_devices *fs_devices, | ||
| 139 | fmode_t flags, void *holder); | ||
| 140 | int btrfs_scan_one_device(const char *path, fmode_t flags, void *holder, | ||
| 141 | struct btrfs_fs_devices **fs_devices_ret); | ||
| 142 | int btrfs_close_devices(struct btrfs_fs_devices *fs_devices); | ||
| 143 | int btrfs_close_extra_devices(struct btrfs_fs_devices *fs_devices); | ||
| 144 | int btrfs_add_device(struct btrfs_trans_handle *trans, | ||
| 145 | struct btrfs_root *root, | ||
| 146 | struct btrfs_device *device); | ||
| 147 | int btrfs_rm_device(struct btrfs_root *root, char *device_path); | ||
| 148 | int btrfs_cleanup_fs_uuids(void); | ||
| 149 | int btrfs_num_copies(struct btrfs_mapping_tree *map_tree, u64 logical, u64 len); | ||
| 150 | int btrfs_unplug_page(struct btrfs_mapping_tree *map_tree, | ||
| 151 | u64 logical, struct page *page); | ||
| 152 | int btrfs_grow_device(struct btrfs_trans_handle *trans, | ||
| 153 | struct btrfs_device *device, u64 new_size); | ||
| 154 | struct btrfs_device *btrfs_find_device(struct btrfs_root *root, u64 devid, | ||
| 155 | u8 *uuid, u8 *fsid); | ||
| 156 | int btrfs_shrink_device(struct btrfs_device *device, u64 new_size); | ||
| 157 | int btrfs_init_new_device(struct btrfs_root *root, char *path); | ||
| 158 | int btrfs_balance(struct btrfs_root *dev_root); | ||
| 159 | void btrfs_unlock_volumes(void); | ||
| 160 | void btrfs_lock_volumes(void); | ||
| 161 | int btrfs_chunk_readonly(struct btrfs_root *root, u64 chunk_offset); | ||
| 162 | #endif | ||
diff --git a/fs/btrfs/xattr.c b/fs/btrfs/xattr.c new file mode 100644 index 000000000000..7f332e270894 --- /dev/null +++ b/fs/btrfs/xattr.c | |||
| @@ -0,0 +1,322 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2007 Red Hat. 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 | |||
| 19 | #include <linux/init.h> | ||
| 20 | #include <linux/fs.h> | ||
| 21 | #include <linux/slab.h> | ||
| 22 | #include <linux/rwsem.h> | ||
| 23 | #include <linux/xattr.h> | ||
| 24 | #include "ctree.h" | ||
| 25 | #include "btrfs_inode.h" | ||
| 26 | #include "transaction.h" | ||
| 27 | #include "xattr.h" | ||
| 28 | #include "disk-io.h" | ||
| 29 | |||
| 30 | |||
| 31 | ssize_t __btrfs_getxattr(struct inode *inode, const char *name, | ||
| 32 | void *buffer, size_t size) | ||
| 33 | { | ||
| 34 | struct btrfs_dir_item *di; | ||
| 35 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 36 | struct btrfs_path *path; | ||
| 37 | struct extent_buffer *leaf; | ||
| 38 | int ret = 0; | ||
| 39 | unsigned long data_ptr; | ||
| 40 | |||
| 41 | path = btrfs_alloc_path(); | ||
| 42 | if (!path) | ||
| 43 | return -ENOMEM; | ||
| 44 | |||
| 45 | /* lookup the xattr by name */ | ||
| 46 | di = btrfs_lookup_xattr(NULL, root, path, inode->i_ino, name, | ||
| 47 | strlen(name), 0); | ||
| 48 | if (!di || IS_ERR(di)) { | ||
| 49 | ret = -ENODATA; | ||
| 50 | goto out; | ||
| 51 | } | ||
| 52 | |||
| 53 | leaf = path->nodes[0]; | ||
| 54 | /* if size is 0, that means we want the size of the attr */ | ||
| 55 | if (!size) { | ||
| 56 | ret = btrfs_dir_data_len(leaf, di); | ||
| 57 | goto out; | ||
| 58 | } | ||
| 59 | |||
| 60 | /* now get the data out of our dir_item */ | ||
| 61 | if (btrfs_dir_data_len(leaf, di) > size) { | ||
| 62 | ret = -ERANGE; | ||
| 63 | goto out; | ||
| 64 | } | ||
| 65 | data_ptr = (unsigned long)((char *)(di + 1) + | ||
| 66 | btrfs_dir_name_len(leaf, di)); | ||
| 67 | read_extent_buffer(leaf, buffer, data_ptr, | ||
| 68 | btrfs_dir_data_len(leaf, di)); | ||
| 69 | ret = btrfs_dir_data_len(leaf, di); | ||
| 70 | |||
| 71 | out: | ||
| 72 | btrfs_free_path(path); | ||
| 73 | return ret; | ||
| 74 | } | ||
| 75 | |||
| 76 | int __btrfs_setxattr(struct inode *inode, const char *name, | ||
| 77 | const void *value, size_t size, int flags) | ||
| 78 | { | ||
| 79 | struct btrfs_dir_item *di; | ||
| 80 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 81 | struct btrfs_trans_handle *trans; | ||
| 82 | struct btrfs_path *path; | ||
| 83 | int ret = 0, mod = 0; | ||
| 84 | |||
| 85 | path = btrfs_alloc_path(); | ||
| 86 | if (!path) | ||
| 87 | return -ENOMEM; | ||
| 88 | |||
| 89 | trans = btrfs_start_transaction(root, 1); | ||
| 90 | btrfs_set_trans_block_group(trans, inode); | ||
| 91 | |||
| 92 | /* first lets see if we already have this xattr */ | ||
| 93 | di = btrfs_lookup_xattr(trans, root, path, inode->i_ino, name, | ||
| 94 | strlen(name), -1); | ||
| 95 | if (IS_ERR(di)) { | ||
| 96 | ret = PTR_ERR(di); | ||
| 97 | goto out; | ||
| 98 | } | ||
| 99 | |||
| 100 | /* ok we already have this xattr, lets remove it */ | ||
| 101 | if (di) { | ||
| 102 | /* if we want create only exit */ | ||
| 103 | if (flags & XATTR_CREATE) { | ||
| 104 | ret = -EEXIST; | ||
| 105 | goto out; | ||
| 106 | } | ||
| 107 | |||
| 108 | ret = btrfs_delete_one_dir_name(trans, root, path, di); | ||
| 109 | if (ret) | ||
| 110 | goto out; | ||
| 111 | btrfs_release_path(root, path); | ||
| 112 | |||
| 113 | /* if we don't have a value then we are removing the xattr */ | ||
| 114 | if (!value) { | ||
| 115 | mod = 1; | ||
| 116 | goto out; | ||
| 117 | } | ||
| 118 | } else { | ||
| 119 | btrfs_release_path(root, path); | ||
| 120 | |||
| 121 | if (flags & XATTR_REPLACE) { | ||
| 122 | /* we couldn't find the attr to replace */ | ||
| 123 | ret = -ENODATA; | ||
| 124 | goto out; | ||
| 125 | } | ||
| 126 | } | ||
| 127 | |||
| 128 | /* ok we have to create a completely new xattr */ | ||
| 129 | ret = btrfs_insert_xattr_item(trans, root, name, strlen(name), | ||
| 130 | value, size, inode->i_ino); | ||
| 131 | if (ret) | ||
| 132 | goto out; | ||
| 133 | mod = 1; | ||
| 134 | |||
| 135 | out: | ||
| 136 | if (mod) { | ||
| 137 | inode->i_ctime = CURRENT_TIME; | ||
| 138 | ret = btrfs_update_inode(trans, root, inode); | ||
| 139 | } | ||
| 140 | |||
| 141 | btrfs_end_transaction(trans, root); | ||
| 142 | btrfs_free_path(path); | ||
| 143 | return ret; | ||
| 144 | } | ||
| 145 | |||
| 146 | ssize_t btrfs_listxattr(struct dentry *dentry, char *buffer, size_t size) | ||
| 147 | { | ||
| 148 | struct btrfs_key key, found_key; | ||
| 149 | struct inode *inode = dentry->d_inode; | ||
| 150 | struct btrfs_root *root = BTRFS_I(inode)->root; | ||
| 151 | struct btrfs_path *path; | ||
| 152 | struct btrfs_item *item; | ||
| 153 | struct extent_buffer *leaf; | ||
| 154 | struct btrfs_dir_item *di; | ||
| 155 | int ret = 0, slot, advance; | ||
| 156 | size_t total_size = 0, size_left = size; | ||
| 157 | unsigned long name_ptr; | ||
| 158 | size_t name_len; | ||
| 159 | u32 nritems; | ||
| 160 | |||
| 161 | /* | ||
| 162 | * ok we want all objects associated with this id. | ||
| 163 | * NOTE: we set key.offset = 0; because we want to start with the | ||
| 164 | * first xattr that we find and walk forward | ||
| 165 | */ | ||
| 166 | key.objectid = inode->i_ino; | ||
| 167 | btrfs_set_key_type(&key, BTRFS_XATTR_ITEM_KEY); | ||
| 168 | key.offset = 0; | ||
| 169 | |||
| 170 | path = btrfs_alloc_path(); | ||
| 171 | if (!path) | ||
| 172 | return -ENOMEM; | ||
| 173 | path->reada = 2; | ||
| 174 | |||
| 175 | /* search for our xattrs */ | ||
| 176 | ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); | ||
| 177 | if (ret < 0) | ||
| 178 | goto err; | ||
| 179 | ret = 0; | ||
| 180 | advance = 0; | ||
| 181 | while (1) { | ||
| 182 | leaf = path->nodes[0]; | ||
| 183 | nritems = btrfs_header_nritems(leaf); | ||
| 184 | slot = path->slots[0]; | ||
| 185 | |||
| 186 | /* this is where we start walking through the path */ | ||
| 187 | if (advance || slot >= nritems) { | ||
| 188 | /* | ||
| 189 | * if we've reached the last slot in this leaf we need | ||
| 190 | * to go to the next leaf and reset everything | ||
| 191 | */ | ||
| 192 | if (slot >= nritems-1) { | ||
| 193 | ret = btrfs_next_leaf(root, path); | ||
| 194 | if (ret) | ||
| 195 | break; | ||
| 196 | leaf = path->nodes[0]; | ||
| 197 | nritems = btrfs_header_nritems(leaf); | ||
| 198 | slot = path->slots[0]; | ||
| 199 | } else { | ||
| 200 | /* | ||
| 201 | * just walking through the slots on this leaf | ||
| 202 | */ | ||
| 203 | slot++; | ||
| 204 | path->slots[0]++; | ||
| 205 | } | ||
| 206 | } | ||
| 207 | advance = 1; | ||
| 208 | |||
| 209 | item = btrfs_item_nr(leaf, slot); | ||
| 210 | btrfs_item_key_to_cpu(leaf, &found_key, slot); | ||
| 211 | |||
| 212 | /* check to make sure this item is what we want */ | ||
| 213 | if (found_key.objectid != key.objectid) | ||
| 214 | break; | ||
| 215 | if (btrfs_key_type(&found_key) != BTRFS_XATTR_ITEM_KEY) | ||
| 216 | break; | ||
| 217 | |||
| 218 | di = btrfs_item_ptr(leaf, slot, struct btrfs_dir_item); | ||
| 219 | |||
| 220 | name_len = btrfs_dir_name_len(leaf, di); | ||
| 221 | total_size += name_len + 1; | ||
| 222 | |||
| 223 | /* we are just looking for how big our buffer needs to be */ | ||
| 224 | if (!size) | ||
| 225 | continue; | ||
| 226 | |||
| 227 | if (!buffer || (name_len + 1) > size_left) { | ||
| 228 | ret = -ERANGE; | ||
| 229 | goto err; | ||
| 230 | } | ||
| 231 | |||
| 232 | name_ptr = (unsigned long)(di + 1); | ||
| 233 | read_extent_buffer(leaf, buffer, name_ptr, name_len); | ||
| 234 | buffer[name_len] = '\0'; | ||
| 235 | |||
| 236 | size_left -= name_len + 1; | ||
| 237 | buffer += name_len + 1; | ||
| 238 | } | ||
| 239 | ret = total_size; | ||
| 240 | |||
| 241 | err: | ||
| 242 | btrfs_free_path(path); | ||
| 243 | |||
| 244 | return ret; | ||
| 245 | } | ||
| 246 | |||
| 247 | /* | ||
| 248 | * List of handlers for synthetic system.* attributes. All real ondisk | ||
| 249 | * attributes are handled directly. | ||
| 250 | */ | ||
| 251 | struct xattr_handler *btrfs_xattr_handlers[] = { | ||
| 252 | #ifdef CONFIG_FS_POSIX_ACL | ||
| 253 | &btrfs_xattr_acl_access_handler, | ||
| 254 | &btrfs_xattr_acl_default_handler, | ||
| 255 | #endif | ||
| 256 | NULL, | ||
| 257 | }; | ||
| 258 | |||
| 259 | /* | ||
| 260 | * Check if the attribute is in a supported namespace. | ||
| 261 | * | ||
| 262 | * This applied after the check for the synthetic attributes in the system | ||
| 263 | * namespace. | ||
| 264 | */ | ||
| 265 | static bool btrfs_is_valid_xattr(const char *name) | ||
| 266 | { | ||
| 267 | return !strncmp(name, XATTR_SECURITY_PREFIX, | ||
| 268 | XATTR_SECURITY_PREFIX_LEN) || | ||
| 269 | !strncmp(name, XATTR_SYSTEM_PREFIX, XATTR_SYSTEM_PREFIX_LEN) || | ||
| 270 | !strncmp(name, XATTR_TRUSTED_PREFIX, XATTR_TRUSTED_PREFIX_LEN) || | ||
| 271 | !strncmp(name, XATTR_USER_PREFIX, XATTR_USER_PREFIX_LEN); | ||
| 272 | } | ||
| 273 | |||
| 274 | ssize_t btrfs_getxattr(struct dentry *dentry, const char *name, | ||
| 275 | void *buffer, size_t size) | ||
| 276 | { | ||
| 277 | /* | ||
| 278 | * If this is a request for a synthetic attribute in the system.* | ||
| 279 | * namespace use the generic infrastructure to resolve a handler | ||
| 280 | * for it via sb->s_xattr. | ||
| 281 | */ | ||
| 282 | if (!strncmp(name, XATTR_SYSTEM_PREFIX, XATTR_SYSTEM_PREFIX_LEN)) | ||
| 283 | return generic_getxattr(dentry, name, buffer, size); | ||
| 284 | |||
| 285 | if (!btrfs_is_valid_xattr(name)) | ||
| 286 | return -EOPNOTSUPP; | ||
| 287 | return __btrfs_getxattr(dentry->d_inode, name, buffer, size); | ||
| 288 | } | ||
| 289 | |||
| 290 | int btrfs_setxattr(struct dentry *dentry, const char *name, const void *value, | ||
| 291 | size_t size, int flags) | ||
| 292 | { | ||
| 293 | /* | ||
| 294 | * If this is a request for a synthetic attribute in the system.* | ||
| 295 | * namespace use the generic infrastructure to resolve a handler | ||
| 296 | * for it via sb->s_xattr. | ||
| 297 | */ | ||
| 298 | if (!strncmp(name, XATTR_SYSTEM_PREFIX, XATTR_SYSTEM_PREFIX_LEN)) | ||
| 299 | return generic_setxattr(dentry, name, value, size, flags); | ||
| 300 | |||
| 301 | if (!btrfs_is_valid_xattr(name)) | ||
| 302 | return -EOPNOTSUPP; | ||
| 303 | |||
| 304 | if (size == 0) | ||
| 305 | value = ""; /* empty EA, do not remove */ | ||
| 306 | return __btrfs_setxattr(dentry->d_inode, name, value, size, flags); | ||
| 307 | } | ||
| 308 | |||
| 309 | int btrfs_removexattr(struct dentry *dentry, const char *name) | ||
| 310 | { | ||
| 311 | /* | ||
| 312 | * If this is a request for a synthetic attribute in the system.* | ||
| 313 | * namespace use the generic infrastructure to resolve a handler | ||
| 314 | * for it via sb->s_xattr. | ||
| 315 | */ | ||
| 316 | if (!strncmp(name, XATTR_SYSTEM_PREFIX, XATTR_SYSTEM_PREFIX_LEN)) | ||
| 317 | return generic_removexattr(dentry, name); | ||
| 318 | |||
| 319 | if (!btrfs_is_valid_xattr(name)) | ||
| 320 | return -EOPNOTSUPP; | ||
| 321 | return __btrfs_setxattr(dentry->d_inode, name, NULL, 0, XATTR_REPLACE); | ||
| 322 | } | ||
diff --git a/fs/btrfs/xattr.h b/fs/btrfs/xattr.h new file mode 100644 index 000000000000..5b1d08f8e68d --- /dev/null +++ b/fs/btrfs/xattr.h | |||
| @@ -0,0 +1,39 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2007 Red Hat. 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 | |||
| 19 | #ifndef __XATTR__ | ||
| 20 | #define __XATTR__ | ||
| 21 | |||
| 22 | #include <linux/xattr.h> | ||
| 23 | |||
| 24 | extern struct xattr_handler btrfs_xattr_acl_access_handler; | ||
| 25 | extern struct xattr_handler btrfs_xattr_acl_default_handler; | ||
| 26 | extern struct xattr_handler *btrfs_xattr_handlers[]; | ||
| 27 | |||
| 28 | extern ssize_t __btrfs_getxattr(struct inode *inode, const char *name, | ||
| 29 | void *buffer, size_t size); | ||
| 30 | extern int __btrfs_setxattr(struct inode *inode, const char *name, | ||
| 31 | const void *value, size_t size, int flags); | ||
| 32 | |||
| 33 | extern ssize_t btrfs_getxattr(struct dentry *dentry, const char *name, | ||
| 34 | void *buffer, size_t size); | ||
| 35 | extern int btrfs_setxattr(struct dentry *dentry, const char *name, | ||
| 36 | const void *value, size_t size, int flags); | ||
| 37 | extern int btrfs_removexattr(struct dentry *dentry, const char *name); | ||
| 38 | |||
| 39 | #endif /* __XATTR__ */ | ||
diff --git a/fs/btrfs/zlib.c b/fs/btrfs/zlib.c new file mode 100644 index 000000000000..ecfbce836d32 --- /dev/null +++ b/fs/btrfs/zlib.c | |||
| @@ -0,0 +1,632 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2008 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 | * Based on jffs2 zlib code: | ||
| 19 | * Copyright © 2001-2007 Red Hat, Inc. | ||
| 20 | * Created by David Woodhouse <dwmw2@infradead.org> | ||
| 21 | */ | ||
| 22 | |||
| 23 | #include <linux/kernel.h> | ||
| 24 | #include <linux/slab.h> | ||
| 25 | #include <linux/zlib.h> | ||
| 26 | #include <linux/zutil.h> | ||
| 27 | #include <linux/vmalloc.h> | ||
| 28 | #include <linux/init.h> | ||
| 29 | #include <linux/err.h> | ||
| 30 | #include <linux/sched.h> | ||
| 31 | #include <linux/pagemap.h> | ||
| 32 | #include <linux/bio.h> | ||
| 33 | #include "compression.h" | ||
| 34 | |||
| 35 | /* Plan: call deflate() with avail_in == *sourcelen, | ||
| 36 | avail_out = *dstlen - 12 and flush == Z_FINISH. | ||
| 37 | If it doesn't manage to finish, call it again with | ||
| 38 | avail_in == 0 and avail_out set to the remaining 12 | ||
| 39 | bytes for it to clean up. | ||
| 40 | Q: Is 12 bytes sufficient? | ||
| 41 | */ | ||
| 42 | #define STREAM_END_SPACE 12 | ||
| 43 | |||
| 44 | struct workspace { | ||
| 45 | z_stream inf_strm; | ||
| 46 | z_stream def_strm; | ||
| 47 | char *buf; | ||
| 48 | struct list_head list; | ||
| 49 | }; | ||
| 50 | |||
| 51 | static LIST_HEAD(idle_workspace); | ||
| 52 | static DEFINE_SPINLOCK(workspace_lock); | ||
| 53 | static unsigned long num_workspace; | ||
| 54 | static atomic_t alloc_workspace = ATOMIC_INIT(0); | ||
| 55 | static DECLARE_WAIT_QUEUE_HEAD(workspace_wait); | ||
| 56 | |||
| 57 | /* | ||
| 58 | * this finds an available zlib workspace or allocates a new one | ||
| 59 | * NULL or an ERR_PTR is returned if things go bad. | ||
| 60 | */ | ||
| 61 | static struct workspace *find_zlib_workspace(void) | ||
| 62 | { | ||
| 63 | struct workspace *workspace; | ||
| 64 | int ret; | ||
| 65 | int cpus = num_online_cpus(); | ||
| 66 | |||
| 67 | again: | ||
| 68 | spin_lock(&workspace_lock); | ||
| 69 | if (!list_empty(&idle_workspace)) { | ||
| 70 | workspace = list_entry(idle_workspace.next, struct workspace, | ||
| 71 | list); | ||
| 72 | list_del(&workspace->list); | ||
| 73 | num_workspace--; | ||
| 74 | spin_unlock(&workspace_lock); | ||
| 75 | return workspace; | ||
| 76 | |||
| 77 | } | ||
| 78 | spin_unlock(&workspace_lock); | ||
| 79 | if (atomic_read(&alloc_workspace) > cpus) { | ||
| 80 | DEFINE_WAIT(wait); | ||
| 81 | prepare_to_wait(&workspace_wait, &wait, TASK_UNINTERRUPTIBLE); | ||
| 82 | if (atomic_read(&alloc_workspace) > cpus) | ||
| 83 | schedule(); | ||
| 84 | finish_wait(&workspace_wait, &wait); | ||
| 85 | goto again; | ||
| 86 | } | ||
| 87 | atomic_inc(&alloc_workspace); | ||
| 88 | workspace = kzalloc(sizeof(*workspace), GFP_NOFS); | ||
| 89 | if (!workspace) { | ||
| 90 | ret = -ENOMEM; | ||
| 91 | goto fail; | ||
| 92 | } | ||
| 93 | |||
| 94 | workspace->def_strm.workspace = vmalloc(zlib_deflate_workspacesize()); | ||
| 95 | if (!workspace->def_strm.workspace) { | ||
| 96 | ret = -ENOMEM; | ||
| 97 | goto fail; | ||
| 98 | } | ||
| 99 | workspace->inf_strm.workspace = vmalloc(zlib_inflate_workspacesize()); | ||
| 100 | if (!workspace->inf_strm.workspace) { | ||
| 101 | ret = -ENOMEM; | ||
| 102 | goto fail_inflate; | ||
| 103 | } | ||
| 104 | workspace->buf = kmalloc(PAGE_CACHE_SIZE, GFP_NOFS); | ||
| 105 | if (!workspace->buf) { | ||
| 106 | ret = -ENOMEM; | ||
| 107 | goto fail_kmalloc; | ||
| 108 | } | ||
| 109 | return workspace; | ||
| 110 | |||
| 111 | fail_kmalloc: | ||
| 112 | vfree(workspace->inf_strm.workspace); | ||
| 113 | fail_inflate: | ||
| 114 | vfree(workspace->def_strm.workspace); | ||
| 115 | fail: | ||
| 116 | kfree(workspace); | ||
| 117 | atomic_dec(&alloc_workspace); | ||
| 118 | wake_up(&workspace_wait); | ||
| 119 | return ERR_PTR(ret); | ||
| 120 | } | ||
| 121 | |||
| 122 | /* | ||
| 123 | * put a workspace struct back on the list or free it if we have enough | ||
| 124 | * idle ones sitting around | ||
| 125 | */ | ||
| 126 | static int free_workspace(struct workspace *workspace) | ||
| 127 | { | ||
| 128 | spin_lock(&workspace_lock); | ||
| 129 | if (num_workspace < num_online_cpus()) { | ||
| 130 | list_add_tail(&workspace->list, &idle_workspace); | ||
| 131 | num_workspace++; | ||
| 132 | spin_unlock(&workspace_lock); | ||
| 133 | if (waitqueue_active(&workspace_wait)) | ||
| 134 | wake_up(&workspace_wait); | ||
| 135 | return 0; | ||
| 136 | } | ||
| 137 | spin_unlock(&workspace_lock); | ||
| 138 | vfree(workspace->def_strm.workspace); | ||
| 139 | vfree(workspace->inf_strm.workspace); | ||
| 140 | kfree(workspace->buf); | ||
| 141 | kfree(workspace); | ||
| 142 | |||
| 143 | atomic_dec(&alloc_workspace); | ||
| 144 | if (waitqueue_active(&workspace_wait)) | ||
| 145 | wake_up(&workspace_wait); | ||
| 146 | return 0; | ||
| 147 | } | ||
| 148 | |||
| 149 | /* | ||
| 150 | * cleanup function for module exit | ||
| 151 | */ | ||
| 152 | static void free_workspaces(void) | ||
| 153 | { | ||
| 154 | struct workspace *workspace; | ||
| 155 | while (!list_empty(&idle_workspace)) { | ||
| 156 | workspace = list_entry(idle_workspace.next, struct workspace, | ||
| 157 | list); | ||
| 158 | list_del(&workspace->list); | ||
| 159 | vfree(workspace->def_strm.workspace); | ||
| 160 | vfree(workspace->inf_strm.workspace); | ||
| 161 | kfree(workspace->buf); | ||
| 162 | kfree(workspace); | ||
| 163 | atomic_dec(&alloc_workspace); | ||
| 164 | } | ||
| 165 | } | ||
| 166 | |||
| 167 | /* | ||
| 168 | * given an address space and start/len, compress the bytes. | ||
| 169 | * | ||
| 170 | * pages are allocated to hold the compressed result and stored | ||
| 171 | * in 'pages' | ||
| 172 | * | ||
| 173 | * out_pages is used to return the number of pages allocated. There | ||
| 174 | * may be pages allocated even if we return an error | ||
| 175 | * | ||
| 176 | * total_in is used to return the number of bytes actually read. It | ||
| 177 | * may be smaller then len if we had to exit early because we | ||
| 178 | * ran out of room in the pages array or because we cross the | ||
| 179 | * max_out threshold. | ||
| 180 | * | ||
| 181 | * total_out is used to return the total number of compressed bytes | ||
| 182 | * | ||
| 183 | * max_out tells us the max number of bytes that we're allowed to | ||
| 184 | * stuff into pages | ||
| 185 | */ | ||
| 186 | int btrfs_zlib_compress_pages(struct address_space *mapping, | ||
| 187 | u64 start, unsigned long len, | ||
| 188 | struct page **pages, | ||
| 189 | unsigned long nr_dest_pages, | ||
| 190 | unsigned long *out_pages, | ||
| 191 | unsigned long *total_in, | ||
| 192 | unsigned long *total_out, | ||
| 193 | unsigned long max_out) | ||
| 194 | { | ||
| 195 | int ret; | ||
| 196 | struct workspace *workspace; | ||
| 197 | char *data_in; | ||
| 198 | char *cpage_out; | ||
| 199 | int nr_pages = 0; | ||
| 200 | struct page *in_page = NULL; | ||
| 201 | struct page *out_page = NULL; | ||
| 202 | int out_written = 0; | ||
| 203 | int in_read = 0; | ||
| 204 | unsigned long bytes_left; | ||
| 205 | |||
| 206 | *out_pages = 0; | ||
| 207 | *total_out = 0; | ||
| 208 | *total_in = 0; | ||
| 209 | |||
| 210 | workspace = find_zlib_workspace(); | ||
| 211 | if (!workspace) | ||
| 212 | return -1; | ||
| 213 | |||
| 214 | if (Z_OK != zlib_deflateInit(&workspace->def_strm, 3)) { | ||
| 215 | printk(KERN_WARNING "deflateInit failed\n"); | ||
| 216 | ret = -1; | ||
| 217 | goto out; | ||
| 218 | } | ||
| 219 | |||
| 220 | workspace->def_strm.total_in = 0; | ||
| 221 | workspace->def_strm.total_out = 0; | ||
| 222 | |||
| 223 | in_page = find_get_page(mapping, start >> PAGE_CACHE_SHIFT); | ||
| 224 | data_in = kmap(in_page); | ||
| 225 | |||
| 226 | out_page = alloc_page(GFP_NOFS | __GFP_HIGHMEM); | ||
| 227 | cpage_out = kmap(out_page); | ||
| 228 | pages[0] = out_page; | ||
| 229 | nr_pages = 1; | ||
| 230 | |||
| 231 | workspace->def_strm.next_in = data_in; | ||
| 232 | workspace->def_strm.next_out = cpage_out; | ||
| 233 | workspace->def_strm.avail_out = PAGE_CACHE_SIZE; | ||
| 234 | workspace->def_strm.avail_in = min(len, PAGE_CACHE_SIZE); | ||
| 235 | |||
| 236 | out_written = 0; | ||
| 237 | in_read = 0; | ||
| 238 | |||
| 239 | while (workspace->def_strm.total_in < len) { | ||
| 240 | ret = zlib_deflate(&workspace->def_strm, Z_SYNC_FLUSH); | ||
| 241 | if (ret != Z_OK) { | ||
| 242 | printk(KERN_DEBUG "btrfs deflate in loop returned %d\n", | ||
| 243 | ret); | ||
| 244 | zlib_deflateEnd(&workspace->def_strm); | ||
| 245 | ret = -1; | ||
| 246 | goto out; | ||
| 247 | } | ||
| 248 | |||
| 249 | /* we're making it bigger, give up */ | ||
| 250 | if (workspace->def_strm.total_in > 8192 && | ||
| 251 | workspace->def_strm.total_in < | ||
| 252 | workspace->def_strm.total_out) { | ||
| 253 | ret = -1; | ||
| 254 | goto out; | ||
| 255 | } | ||
| 256 | /* we need another page for writing out. Test this | ||
| 257 | * before the total_in so we will pull in a new page for | ||
| 258 | * the stream end if required | ||
| 259 | */ | ||
| 260 | if (workspace->def_strm.avail_out == 0) { | ||
| 261 | kunmap(out_page); | ||
| 262 | if (nr_pages == nr_dest_pages) { | ||
| 263 | out_page = NULL; | ||
| 264 | ret = -1; | ||
| 265 | goto out; | ||
| 266 | } | ||
| 267 | out_page = alloc_page(GFP_NOFS | __GFP_HIGHMEM); | ||
| 268 | cpage_out = kmap(out_page); | ||
| 269 | pages[nr_pages] = out_page; | ||
| 270 | nr_pages++; | ||
| 271 | workspace->def_strm.avail_out = PAGE_CACHE_SIZE; | ||
| 272 | workspace->def_strm.next_out = cpage_out; | ||
| 273 | } | ||
| 274 | /* we're all done */ | ||
| 275 | if (workspace->def_strm.total_in >= len) | ||
| 276 | break; | ||
| 277 | |||
| 278 | /* we've read in a full page, get a new one */ | ||
| 279 | if (workspace->def_strm.avail_in == 0) { | ||
| 280 | if (workspace->def_strm.total_out > max_out) | ||
| 281 | break; | ||
| 282 | |||
| 283 | bytes_left = len - workspace->def_strm.total_in; | ||
| 284 | kunmap(in_page); | ||
| 285 | page_cache_release(in_page); | ||
| 286 | |||
| 287 | start += PAGE_CACHE_SIZE; | ||
| 288 | in_page = find_get_page(mapping, | ||
| 289 | start >> PAGE_CACHE_SHIFT); | ||
| 290 | data_in = kmap(in_page); | ||
| 291 | workspace->def_strm.avail_in = min(bytes_left, | ||
| 292 | PAGE_CACHE_SIZE); | ||
| 293 | workspace->def_strm.next_in = data_in; | ||
| 294 | } | ||
| 295 | } | ||
| 296 | workspace->def_strm.avail_in = 0; | ||
| 297 | ret = zlib_deflate(&workspace->def_strm, Z_FINISH); | ||
| 298 | zlib_deflateEnd(&workspace->def_strm); | ||
| 299 | |||
| 300 | if (ret != Z_STREAM_END) { | ||
| 301 | ret = -1; | ||
| 302 | goto out; | ||
| 303 | } | ||
| 304 | |||
| 305 | if (workspace->def_strm.total_out >= workspace->def_strm.total_in) { | ||
| 306 | ret = -1; | ||
| 307 | goto out; | ||
| 308 | } | ||
| 309 | |||
| 310 | ret = 0; | ||
| 311 | *total_out = workspace->def_strm.total_out; | ||
| 312 | *total_in = workspace->def_strm.total_in; | ||
| 313 | out: | ||
| 314 | *out_pages = nr_pages; | ||
| 315 | if (out_page) | ||
| 316 | kunmap(out_page); | ||
| 317 | |||
| 318 | if (in_page) { | ||
| 319 | kunmap(in_page); | ||
| 320 | page_cache_release(in_page); | ||
| 321 | } | ||
| 322 | free_workspace(workspace); | ||
| 323 | return ret; | ||
| 324 | } | ||
| 325 | |||
| 326 | /* | ||
| 327 | * pages_in is an array of pages with compressed data. | ||
| 328 | * | ||
| 329 | * disk_start is the starting logical offset of this array in the file | ||
| 330 | * | ||
| 331 | * bvec is a bio_vec of pages from the file that we want to decompress into | ||
| 332 | * | ||
| 333 | * vcnt is the count of pages in the biovec | ||
| 334 | * | ||
| 335 | * srclen is the number of bytes in pages_in | ||
| 336 | * | ||
| 337 | * The basic idea is that we have a bio that was created by readpages. | ||
| 338 | * The pages in the bio are for the uncompressed data, and they may not | ||
| 339 | * be contiguous. They all correspond to the range of bytes covered by | ||
| 340 | * the compressed extent. | ||
| 341 | */ | ||
| 342 | int btrfs_zlib_decompress_biovec(struct page **pages_in, | ||
| 343 | u64 disk_start, | ||
| 344 | struct bio_vec *bvec, | ||
| 345 | int vcnt, | ||
| 346 | size_t srclen) | ||
| 347 | { | ||
| 348 | int ret = 0; | ||
| 349 | int wbits = MAX_WBITS; | ||
| 350 | struct workspace *workspace; | ||
| 351 | char *data_in; | ||
| 352 | size_t total_out = 0; | ||
| 353 | unsigned long page_bytes_left; | ||
| 354 | unsigned long page_in_index = 0; | ||
| 355 | unsigned long page_out_index = 0; | ||
| 356 | struct page *page_out; | ||
| 357 | unsigned long total_pages_in = (srclen + PAGE_CACHE_SIZE - 1) / | ||
| 358 | PAGE_CACHE_SIZE; | ||
| 359 | unsigned long buf_start; | ||
| 360 | unsigned long buf_offset; | ||
| 361 | unsigned long bytes; | ||
| 362 | unsigned long working_bytes; | ||
| 363 | unsigned long pg_offset; | ||
| 364 | unsigned long start_byte; | ||
| 365 | unsigned long current_buf_start; | ||
| 366 | char *kaddr; | ||
| 367 | |||
| 368 | workspace = find_zlib_workspace(); | ||
| 369 | if (!workspace) | ||
| 370 | return -ENOMEM; | ||
| 371 | |||
| 372 | data_in = kmap(pages_in[page_in_index]); | ||
| 373 | workspace->inf_strm.next_in = data_in; | ||
| 374 | workspace->inf_strm.avail_in = min_t(size_t, srclen, PAGE_CACHE_SIZE); | ||
| 375 | workspace->inf_strm.total_in = 0; | ||
| 376 | |||
| 377 | workspace->inf_strm.total_out = 0; | ||
| 378 | workspace->inf_strm.next_out = workspace->buf; | ||
| 379 | workspace->inf_strm.avail_out = PAGE_CACHE_SIZE; | ||
| 380 | page_out = bvec[page_out_index].bv_page; | ||
| 381 | page_bytes_left = PAGE_CACHE_SIZE; | ||
| 382 | pg_offset = 0; | ||
| 383 | |||
| 384 | /* If it's deflate, and it's got no preset dictionary, then | ||
| 385 | we can tell zlib to skip the adler32 check. */ | ||
| 386 | if (srclen > 2 && !(data_in[1] & PRESET_DICT) && | ||
| 387 | ((data_in[0] & 0x0f) == Z_DEFLATED) && | ||
| 388 | !(((data_in[0]<<8) + data_in[1]) % 31)) { | ||
| 389 | |||
| 390 | wbits = -((data_in[0] >> 4) + 8); | ||
| 391 | workspace->inf_strm.next_in += 2; | ||
| 392 | workspace->inf_strm.avail_in -= 2; | ||
| 393 | } | ||
| 394 | |||
| 395 | if (Z_OK != zlib_inflateInit2(&workspace->inf_strm, wbits)) { | ||
| 396 | printk(KERN_WARNING "inflateInit failed\n"); | ||
| 397 | ret = -1; | ||
| 398 | goto out; | ||
| 399 | } | ||
| 400 | while (workspace->inf_strm.total_in < srclen) { | ||
| 401 | ret = zlib_inflate(&workspace->inf_strm, Z_NO_FLUSH); | ||
| 402 | if (ret != Z_OK && ret != Z_STREAM_END) | ||
| 403 | break; | ||
| 404 | /* | ||
| 405 | * buf start is the byte offset we're of the start of | ||
| 406 | * our workspace buffer | ||
| 407 | */ | ||
| 408 | buf_start = total_out; | ||
| 409 | |||
| 410 | /* total_out is the last byte of the workspace buffer */ | ||
| 411 | total_out = workspace->inf_strm.total_out; | ||
| 412 | |||
| 413 | working_bytes = total_out - buf_start; | ||
| 414 | |||
| 415 | /* | ||
| 416 | * start byte is the first byte of the page we're currently | ||
| 417 | * copying into relative to the start of the compressed data. | ||
| 418 | */ | ||
| 419 | start_byte = page_offset(page_out) - disk_start; | ||
| 420 | |||
| 421 | if (working_bytes == 0) { | ||
| 422 | /* we didn't make progress in this inflate | ||
| 423 | * call, we're done | ||
| 424 | */ | ||
| 425 | if (ret != Z_STREAM_END) | ||
| 426 | ret = -1; | ||
| 427 | break; | ||
| 428 | } | ||
| 429 | |||
| 430 | /* we haven't yet hit data corresponding to this page */ | ||
| 431 | if (total_out <= start_byte) | ||
| 432 | goto next; | ||
| 433 | |||
| 434 | /* | ||
| 435 | * the start of the data we care about is offset into | ||
| 436 | * the middle of our working buffer | ||
| 437 | */ | ||
| 438 | if (total_out > start_byte && buf_start < start_byte) { | ||
| 439 | buf_offset = start_byte - buf_start; | ||
| 440 | working_bytes -= buf_offset; | ||
| 441 | } else { | ||
| 442 | buf_offset = 0; | ||
| 443 | } | ||
| 444 | current_buf_start = buf_start; | ||
| 445 | |||
| 446 | /* copy bytes from the working buffer into the pages */ | ||
| 447 | while (working_bytes > 0) { | ||
| 448 | bytes = min(PAGE_CACHE_SIZE - pg_offset, | ||
| 449 | PAGE_CACHE_SIZE - buf_offset); | ||
| 450 | bytes = min(bytes, working_bytes); | ||
| 451 | kaddr = kmap_atomic(page_out, KM_USER0); | ||
| 452 | memcpy(kaddr + pg_offset, workspace->buf + buf_offset, | ||
| 453 | bytes); | ||
| 454 | kunmap_atomic(kaddr, KM_USER0); | ||
| 455 | flush_dcache_page(page_out); | ||
| 456 | |||
| 457 | pg_offset += bytes; | ||
| 458 | page_bytes_left -= bytes; | ||
| 459 | buf_offset += bytes; | ||
| 460 | working_bytes -= bytes; | ||
| 461 | current_buf_start += bytes; | ||
| 462 | |||
| 463 | /* check if we need to pick another page */ | ||
| 464 | if (page_bytes_left == 0) { | ||
| 465 | page_out_index++; | ||
| 466 | if (page_out_index >= vcnt) { | ||
| 467 | ret = 0; | ||
| 468 | goto done; | ||
| 469 | } | ||
| 470 | |||
| 471 | page_out = bvec[page_out_index].bv_page; | ||
| 472 | pg_offset = 0; | ||
| 473 | page_bytes_left = PAGE_CACHE_SIZE; | ||
| 474 | start_byte = page_offset(page_out) - disk_start; | ||
| 475 | |||
| 476 | /* | ||
| 477 | * make sure our new page is covered by this | ||
| 478 | * working buffer | ||
| 479 | */ | ||
| 480 | if (total_out <= start_byte) | ||
| 481 | goto next; | ||
| 482 | |||
| 483 | /* the next page in the biovec might not | ||
| 484 | * be adjacent to the last page, but it | ||
| 485 | * might still be found inside this working | ||
| 486 | * buffer. bump our offset pointer | ||
| 487 | */ | ||
| 488 | if (total_out > start_byte && | ||
| 489 | current_buf_start < start_byte) { | ||
| 490 | buf_offset = start_byte - buf_start; | ||
| 491 | working_bytes = total_out - start_byte; | ||
| 492 | current_buf_start = buf_start + | ||
| 493 | buf_offset; | ||
| 494 | } | ||
| 495 | } | ||
| 496 | } | ||
| 497 | next: | ||
| 498 | workspace->inf_strm.next_out = workspace->buf; | ||
| 499 | workspace->inf_strm.avail_out = PAGE_CACHE_SIZE; | ||
| 500 | |||
| 501 | if (workspace->inf_strm.avail_in == 0) { | ||
| 502 | unsigned long tmp; | ||
| 503 | kunmap(pages_in[page_in_index]); | ||
| 504 | page_in_index++; | ||
| 505 | if (page_in_index >= total_pages_in) { | ||
| 506 | data_in = NULL; | ||
| 507 | break; | ||
| 508 | } | ||
| 509 | data_in = kmap(pages_in[page_in_index]); | ||
| 510 | workspace->inf_strm.next_in = data_in; | ||
| 511 | tmp = srclen - workspace->inf_strm.total_in; | ||
| 512 | workspace->inf_strm.avail_in = min(tmp, | ||
| 513 | PAGE_CACHE_SIZE); | ||
| 514 | } | ||
| 515 | } | ||
| 516 | if (ret != Z_STREAM_END) | ||
| 517 | ret = -1; | ||
| 518 | else | ||
| 519 | ret = 0; | ||
| 520 | done: | ||
| 521 | zlib_inflateEnd(&workspace->inf_strm); | ||
| 522 | if (data_in) | ||
| 523 | kunmap(pages_in[page_in_index]); | ||
| 524 | out: | ||
| 525 | free_workspace(workspace); | ||
| 526 | return ret; | ||
| 527 | } | ||
| 528 | |||
| 529 | /* | ||
| 530 | * a less complex decompression routine. Our compressed data fits in a | ||
| 531 | * single page, and we want to read a single page out of it. | ||
| 532 | * start_byte tells us the offset into the compressed data we're interested in | ||
| 533 | */ | ||
| 534 | int btrfs_zlib_decompress(unsigned char *data_in, | ||
| 535 | struct page *dest_page, | ||
| 536 | unsigned long start_byte, | ||
| 537 | size_t srclen, size_t destlen) | ||
| 538 | { | ||
| 539 | int ret = 0; | ||
| 540 | int wbits = MAX_WBITS; | ||
| 541 | struct workspace *workspace; | ||
| 542 | unsigned long bytes_left = destlen; | ||
| 543 | unsigned long total_out = 0; | ||
| 544 | char *kaddr; | ||
| 545 | |||
| 546 | if (destlen > PAGE_CACHE_SIZE) | ||
| 547 | return -ENOMEM; | ||
| 548 | |||
| 549 | workspace = find_zlib_workspace(); | ||
| 550 | if (!workspace) | ||
| 551 | return -ENOMEM; | ||
| 552 | |||
| 553 | workspace->inf_strm.next_in = data_in; | ||
| 554 | workspace->inf_strm.avail_in = srclen; | ||
| 555 | workspace->inf_strm.total_in = 0; | ||
| 556 | |||
| 557 | workspace->inf_strm.next_out = workspace->buf; | ||
| 558 | workspace->inf_strm.avail_out = PAGE_CACHE_SIZE; | ||
| 559 | workspace->inf_strm.total_out = 0; | ||
| 560 | /* If it's deflate, and it's got no preset dictionary, then | ||
| 561 | we can tell zlib to skip the adler32 check. */ | ||
| 562 | if (srclen > 2 && !(data_in[1] & PRESET_DICT) && | ||
| 563 | ((data_in[0] & 0x0f) == Z_DEFLATED) && | ||
| 564 | !(((data_in[0]<<8) + data_in[1]) % 31)) { | ||
| 565 | |||
| 566 | wbits = -((data_in[0] >> 4) + 8); | ||
| 567 | workspace->inf_strm.next_in += 2; | ||
| 568 | workspace->inf_strm.avail_in -= 2; | ||
| 569 | } | ||
| 570 | |||
| 571 | if (Z_OK != zlib_inflateInit2(&workspace->inf_strm, wbits)) { | ||
| 572 | printk(KERN_WARNING "inflateInit failed\n"); | ||
| 573 | ret = -1; | ||
| 574 | goto out; | ||
| 575 | } | ||
| 576 | |||
| 577 | while (bytes_left > 0) { | ||
| 578 | unsigned long buf_start; | ||
| 579 | unsigned long buf_offset; | ||
| 580 | unsigned long bytes; | ||
| 581 | unsigned long pg_offset = 0; | ||
| 582 | |||
| 583 | ret = zlib_inflate(&workspace->inf_strm, Z_NO_FLUSH); | ||
| 584 | if (ret != Z_OK && ret != Z_STREAM_END) | ||
| 585 | break; | ||
| 586 | |||
| 587 | buf_start = total_out; | ||
| 588 | total_out = workspace->inf_strm.total_out; | ||
| 589 | |||
| 590 | if (total_out == buf_start) { | ||
| 591 | ret = -1; | ||
| 592 | break; | ||
| 593 | } | ||
| 594 | |||
| 595 | if (total_out <= start_byte) | ||
| 596 | goto next; | ||
| 597 | |||
| 598 | if (total_out > start_byte && buf_start < start_byte) | ||
| 599 | buf_offset = start_byte - buf_start; | ||
| 600 | else | ||
| 601 | buf_offset = 0; | ||
| 602 | |||
| 603 | bytes = min(PAGE_CACHE_SIZE - pg_offset, | ||
| 604 | PAGE_CACHE_SIZE - buf_offset); | ||
| 605 | bytes = min(bytes, bytes_left); | ||
| 606 | |||
| 607 | kaddr = kmap_atomic(dest_page, KM_USER0); | ||
| 608 | memcpy(kaddr + pg_offset, workspace->buf + buf_offset, bytes); | ||
| 609 | kunmap_atomic(kaddr, KM_USER0); | ||
| 610 | |||
| 611 | pg_offset += bytes; | ||
| 612 | bytes_left -= bytes; | ||
| 613 | next: | ||
| 614 | workspace->inf_strm.next_out = workspace->buf; | ||
| 615 | workspace->inf_strm.avail_out = PAGE_CACHE_SIZE; | ||
| 616 | } | ||
| 617 | |||
| 618 | if (ret != Z_STREAM_END && bytes_left != 0) | ||
| 619 | ret = -1; | ||
| 620 | else | ||
| 621 | ret = 0; | ||
| 622 | |||
| 623 | zlib_inflateEnd(&workspace->inf_strm); | ||
| 624 | out: | ||
| 625 | free_workspace(workspace); | ||
| 626 | return ret; | ||
| 627 | } | ||
| 628 | |||
| 629 | void btrfs_zlib_exit(void) | ||
| 630 | { | ||
| 631 | free_workspaces(); | ||
| 632 | } | ||
