16cbd5570SChris Mason /* 2d352ac68SChris Mason * Copyright (C) 2007,2008 Oracle. All rights reserved. 36cbd5570SChris Mason * 46cbd5570SChris Mason * This program is free software; you can redistribute it and/or 56cbd5570SChris Mason * modify it under the terms of the GNU General Public 66cbd5570SChris Mason * License v2 as published by the Free Software Foundation. 76cbd5570SChris Mason * 86cbd5570SChris Mason * This program is distributed in the hope that it will be useful, 96cbd5570SChris Mason * but WITHOUT ANY WARRANTY; without even the implied warranty of 106cbd5570SChris Mason * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 116cbd5570SChris Mason * General Public License for more details. 126cbd5570SChris Mason * 136cbd5570SChris Mason * You should have received a copy of the GNU General Public 146cbd5570SChris Mason * License along with this program; if not, write to the 156cbd5570SChris Mason * Free Software Foundation, Inc., 59 Temple Place - Suite 330, 166cbd5570SChris Mason * Boston, MA 021110-1307, USA. 176cbd5570SChris Mason */ 186cbd5570SChris Mason 19a6b6e75eSChris Mason #include <linux/sched.h> 205a0e3ad6STejun Heo #include <linux/slab.h> 21bd989ba3SJan Schmidt #include <linux/rbtree.h> 22eb60ceacSChris Mason #include "ctree.h" 23eb60ceacSChris Mason #include "disk-io.h" 247f5c1516SChris Mason #include "transaction.h" 255f39d397SChris Mason #include "print-tree.h" 26925baeddSChris Mason #include "locking.h" 279a8dd150SChris Mason 28e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 29e089f05cSChris Mason *root, struct btrfs_path *path, int level); 30e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 31d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 32cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend); 335f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 345f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 35971a1f66SChris Mason struct extent_buffer *src, int empty); 365f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 375f39d397SChris Mason struct btrfs_root *root, 385f39d397SChris Mason struct extent_buffer *dst_buf, 395f39d397SChris Mason struct extent_buffer *src_buf); 40143bede5SJeff Mahoney static void del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 41f3ea38daSJan Schmidt struct btrfs_path *path, int level, int slot, 42f3ea38daSJan Schmidt int tree_mod_log); 43f230475eSJan Schmidt static void tree_mod_log_free_eb(struct btrfs_fs_info *fs_info, 44f230475eSJan Schmidt struct extent_buffer *eb); 45f230475eSJan Schmidt struct extent_buffer *read_old_tree_block(struct btrfs_root *root, u64 bytenr, 46f230475eSJan Schmidt u32 blocksize, u64 parent_transid, 47f230475eSJan Schmidt u64 time_seq); 48f230475eSJan Schmidt struct extent_buffer *btrfs_find_old_tree_block(struct btrfs_root *root, 49f230475eSJan Schmidt u64 bytenr, u32 blocksize, 50f230475eSJan Schmidt u64 time_seq); 51d97e63b6SChris Mason 522c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void) 532c90e5d6SChris Mason { 54df24a2b9SChris Mason struct btrfs_path *path; 55e00f7308SJeff Mahoney path = kmem_cache_zalloc(btrfs_path_cachep, GFP_NOFS); 56df24a2b9SChris Mason return path; 572c90e5d6SChris Mason } 582c90e5d6SChris Mason 59b4ce94deSChris Mason /* 60b4ce94deSChris Mason * set all locked nodes in the path to blocking locks. This should 61b4ce94deSChris Mason * be done before scheduling 62b4ce94deSChris Mason */ 63b4ce94deSChris Mason noinline void btrfs_set_path_blocking(struct btrfs_path *p) 64b4ce94deSChris Mason { 65b4ce94deSChris Mason int i; 66b4ce94deSChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 67bd681513SChris Mason if (!p->nodes[i] || !p->locks[i]) 68bd681513SChris Mason continue; 69bd681513SChris Mason btrfs_set_lock_blocking_rw(p->nodes[i], p->locks[i]); 70bd681513SChris Mason if (p->locks[i] == BTRFS_READ_LOCK) 71bd681513SChris Mason p->locks[i] = BTRFS_READ_LOCK_BLOCKING; 72bd681513SChris Mason else if (p->locks[i] == BTRFS_WRITE_LOCK) 73bd681513SChris Mason p->locks[i] = BTRFS_WRITE_LOCK_BLOCKING; 74b4ce94deSChris Mason } 75b4ce94deSChris Mason } 76b4ce94deSChris Mason 77b4ce94deSChris Mason /* 78b4ce94deSChris Mason * reset all the locked nodes in the patch to spinning locks. 794008c04aSChris Mason * 804008c04aSChris Mason * held is used to keep lockdep happy, when lockdep is enabled 814008c04aSChris Mason * we set held to a blocking lock before we go around and 824008c04aSChris Mason * retake all the spinlocks in the path. You can safely use NULL 834008c04aSChris Mason * for held 84b4ce94deSChris Mason */ 854008c04aSChris Mason noinline void btrfs_clear_path_blocking(struct btrfs_path *p, 86bd681513SChris Mason struct extent_buffer *held, int held_rw) 87b4ce94deSChris Mason { 88b4ce94deSChris Mason int i; 894008c04aSChris Mason 904008c04aSChris Mason #ifdef CONFIG_DEBUG_LOCK_ALLOC 914008c04aSChris Mason /* lockdep really cares that we take all of these spinlocks 924008c04aSChris Mason * in the right order. If any of the locks in the path are not 934008c04aSChris Mason * currently blocking, it is going to complain. So, make really 944008c04aSChris Mason * really sure by forcing the path to blocking before we clear 954008c04aSChris Mason * the path blocking. 964008c04aSChris Mason */ 97bd681513SChris Mason if (held) { 98bd681513SChris Mason btrfs_set_lock_blocking_rw(held, held_rw); 99bd681513SChris Mason if (held_rw == BTRFS_WRITE_LOCK) 100bd681513SChris Mason held_rw = BTRFS_WRITE_LOCK_BLOCKING; 101bd681513SChris Mason else if (held_rw == BTRFS_READ_LOCK) 102bd681513SChris Mason held_rw = BTRFS_READ_LOCK_BLOCKING; 103bd681513SChris Mason } 1044008c04aSChris Mason btrfs_set_path_blocking(p); 1054008c04aSChris Mason #endif 1064008c04aSChris Mason 1074008c04aSChris Mason for (i = BTRFS_MAX_LEVEL - 1; i >= 0; i--) { 108bd681513SChris Mason if (p->nodes[i] && p->locks[i]) { 109bd681513SChris Mason btrfs_clear_lock_blocking_rw(p->nodes[i], p->locks[i]); 110bd681513SChris Mason if (p->locks[i] == BTRFS_WRITE_LOCK_BLOCKING) 111bd681513SChris Mason p->locks[i] = BTRFS_WRITE_LOCK; 112bd681513SChris Mason else if (p->locks[i] == BTRFS_READ_LOCK_BLOCKING) 113bd681513SChris Mason p->locks[i] = BTRFS_READ_LOCK; 114bd681513SChris Mason } 115b4ce94deSChris Mason } 1164008c04aSChris Mason 1174008c04aSChris Mason #ifdef CONFIG_DEBUG_LOCK_ALLOC 1184008c04aSChris Mason if (held) 119bd681513SChris Mason btrfs_clear_lock_blocking_rw(held, held_rw); 1204008c04aSChris Mason #endif 121b4ce94deSChris Mason } 122b4ce94deSChris Mason 123d352ac68SChris Mason /* this also releases the path */ 1242c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p) 1252c90e5d6SChris Mason { 126ff175d57SJesper Juhl if (!p) 127ff175d57SJesper Juhl return; 128b3b4aa74SDavid Sterba btrfs_release_path(p); 1292c90e5d6SChris Mason kmem_cache_free(btrfs_path_cachep, p); 1302c90e5d6SChris Mason } 1312c90e5d6SChris Mason 132d352ac68SChris Mason /* 133d352ac68SChris Mason * path release drops references on the extent buffers in the path 134d352ac68SChris Mason * and it drops any locks held by this path 135d352ac68SChris Mason * 136d352ac68SChris Mason * It is safe to call this on paths that no locks or extent buffers held. 137d352ac68SChris Mason */ 138b3b4aa74SDavid Sterba noinline void btrfs_release_path(struct btrfs_path *p) 139eb60ceacSChris Mason { 140eb60ceacSChris Mason int i; 141a2135011SChris Mason 142234b63a0SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 1433f157a2fSChris Mason p->slots[i] = 0; 144eb60ceacSChris Mason if (!p->nodes[i]) 145925baeddSChris Mason continue; 146925baeddSChris Mason if (p->locks[i]) { 147bd681513SChris Mason btrfs_tree_unlock_rw(p->nodes[i], p->locks[i]); 148925baeddSChris Mason p->locks[i] = 0; 149925baeddSChris Mason } 1505f39d397SChris Mason free_extent_buffer(p->nodes[i]); 1513f157a2fSChris Mason p->nodes[i] = NULL; 152eb60ceacSChris Mason } 153eb60ceacSChris Mason } 154eb60ceacSChris Mason 155d352ac68SChris Mason /* 156d352ac68SChris Mason * safely gets a reference on the root node of a tree. A lock 157d352ac68SChris Mason * is not taken, so a concurrent writer may put a different node 158d352ac68SChris Mason * at the root of the tree. See btrfs_lock_root_node for the 159d352ac68SChris Mason * looping required. 160d352ac68SChris Mason * 161d352ac68SChris Mason * The extent buffer returned by this has a reference taken, so 162d352ac68SChris Mason * it won't disappear. It may stop being the root of the tree 163d352ac68SChris Mason * at any time because there are no locks held. 164d352ac68SChris Mason */ 165925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root) 166925baeddSChris Mason { 167925baeddSChris Mason struct extent_buffer *eb; 168240f62c8SChris Mason 1693083ee2eSJosef Bacik while (1) { 170240f62c8SChris Mason rcu_read_lock(); 171240f62c8SChris Mason eb = rcu_dereference(root->node); 1723083ee2eSJosef Bacik 1733083ee2eSJosef Bacik /* 1743083ee2eSJosef Bacik * RCU really hurts here, we could free up the root node because 1753083ee2eSJosef Bacik * it was cow'ed but we may not get the new root node yet so do 1763083ee2eSJosef Bacik * the inc_not_zero dance and if it doesn't work then 1773083ee2eSJosef Bacik * synchronize_rcu and try again. 1783083ee2eSJosef Bacik */ 1793083ee2eSJosef Bacik if (atomic_inc_not_zero(&eb->refs)) { 180240f62c8SChris Mason rcu_read_unlock(); 1813083ee2eSJosef Bacik break; 1823083ee2eSJosef Bacik } 1833083ee2eSJosef Bacik rcu_read_unlock(); 1843083ee2eSJosef Bacik synchronize_rcu(); 1853083ee2eSJosef Bacik } 186925baeddSChris Mason return eb; 187925baeddSChris Mason } 188925baeddSChris Mason 189d352ac68SChris Mason /* loop around taking references on and locking the root node of the 190d352ac68SChris Mason * tree until you end up with a lock on the root. A locked buffer 191d352ac68SChris Mason * is returned, with a reference held. 192d352ac68SChris Mason */ 193925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root) 194925baeddSChris Mason { 195925baeddSChris Mason struct extent_buffer *eb; 196925baeddSChris Mason 197925baeddSChris Mason while (1) { 198925baeddSChris Mason eb = btrfs_root_node(root); 199925baeddSChris Mason btrfs_tree_lock(eb); 200240f62c8SChris Mason if (eb == root->node) 201925baeddSChris Mason break; 202925baeddSChris Mason btrfs_tree_unlock(eb); 203925baeddSChris Mason free_extent_buffer(eb); 204925baeddSChris Mason } 205925baeddSChris Mason return eb; 206925baeddSChris Mason } 207925baeddSChris Mason 208bd681513SChris Mason /* loop around taking references on and locking the root node of the 209bd681513SChris Mason * tree until you end up with a lock on the root. A locked buffer 210bd681513SChris Mason * is returned, with a reference held. 211bd681513SChris Mason */ 212bd681513SChris Mason struct extent_buffer *btrfs_read_lock_root_node(struct btrfs_root *root) 213bd681513SChris Mason { 214bd681513SChris Mason struct extent_buffer *eb; 215bd681513SChris Mason 216bd681513SChris Mason while (1) { 217bd681513SChris Mason eb = btrfs_root_node(root); 218bd681513SChris Mason btrfs_tree_read_lock(eb); 219bd681513SChris Mason if (eb == root->node) 220bd681513SChris Mason break; 221bd681513SChris Mason btrfs_tree_read_unlock(eb); 222bd681513SChris Mason free_extent_buffer(eb); 223bd681513SChris Mason } 224bd681513SChris Mason return eb; 225bd681513SChris Mason } 226bd681513SChris Mason 227d352ac68SChris Mason /* cowonly root (everything not a reference counted cow subvolume), just get 228d352ac68SChris Mason * put onto a simple dirty list. transaction.c walks this to make sure they 229d352ac68SChris Mason * get properly updated on disk. 230d352ac68SChris Mason */ 2310b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root) 2320b86a832SChris Mason { 233e5846fc6SChris Mason spin_lock(&root->fs_info->trans_lock); 2340b86a832SChris Mason if (root->track_dirty && list_empty(&root->dirty_list)) { 2350b86a832SChris Mason list_add(&root->dirty_list, 2360b86a832SChris Mason &root->fs_info->dirty_cowonly_roots); 2370b86a832SChris Mason } 238e5846fc6SChris Mason spin_unlock(&root->fs_info->trans_lock); 2390b86a832SChris Mason } 2400b86a832SChris Mason 241d352ac68SChris Mason /* 242d352ac68SChris Mason * used by snapshot creation to make a copy of a root for a tree with 243d352ac68SChris Mason * a given objectid. The buffer with the new root node is returned in 244d352ac68SChris Mason * cow_ret, and this func returns zero on success or a negative error code. 245d352ac68SChris Mason */ 246be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans, 247be20aa9dSChris Mason struct btrfs_root *root, 248be20aa9dSChris Mason struct extent_buffer *buf, 249be20aa9dSChris Mason struct extent_buffer **cow_ret, u64 new_root_objectid) 250be20aa9dSChris Mason { 251be20aa9dSChris Mason struct extent_buffer *cow; 252be20aa9dSChris Mason int ret = 0; 253be20aa9dSChris Mason int level; 2545d4f98a2SYan Zheng struct btrfs_disk_key disk_key; 255be20aa9dSChris Mason 256be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != 257be20aa9dSChris Mason root->fs_info->running_transaction->transid); 258be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 259be20aa9dSChris Mason 260be20aa9dSChris Mason level = btrfs_header_level(buf); 2615d4f98a2SYan Zheng if (level == 0) 2625d4f98a2SYan Zheng btrfs_item_key(buf, &disk_key, 0); 2635d4f98a2SYan Zheng else 2645d4f98a2SYan Zheng btrfs_node_key(buf, &disk_key, 0); 26531840ae1SZheng Yan 2665d4f98a2SYan Zheng cow = btrfs_alloc_free_block(trans, root, buf->len, 0, 2675d4f98a2SYan Zheng new_root_objectid, &disk_key, level, 2685581a51aSJan Schmidt buf->start, 0); 2695d4f98a2SYan Zheng if (IS_ERR(cow)) 270be20aa9dSChris Mason return PTR_ERR(cow); 271be20aa9dSChris Mason 272be20aa9dSChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 273be20aa9dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 274be20aa9dSChris Mason btrfs_set_header_generation(cow, trans->transid); 2755d4f98a2SYan Zheng btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV); 2765d4f98a2SYan Zheng btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN | 2775d4f98a2SYan Zheng BTRFS_HEADER_FLAG_RELOC); 2785d4f98a2SYan Zheng if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID) 2795d4f98a2SYan Zheng btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC); 2805d4f98a2SYan Zheng else 281be20aa9dSChris Mason btrfs_set_header_owner(cow, new_root_objectid); 282be20aa9dSChris Mason 2832b82032cSYan Zheng write_extent_buffer(cow, root->fs_info->fsid, 2842b82032cSYan Zheng (unsigned long)btrfs_header_fsid(cow), 2852b82032cSYan Zheng BTRFS_FSID_SIZE); 2862b82032cSYan Zheng 287be20aa9dSChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 2885d4f98a2SYan Zheng if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID) 28966d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, cow, 1, 1); 2905d4f98a2SYan Zheng else 29166d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, cow, 0, 1); 2924aec2b52SChris Mason 293be20aa9dSChris Mason if (ret) 294be20aa9dSChris Mason return ret; 295be20aa9dSChris Mason 296be20aa9dSChris Mason btrfs_mark_buffer_dirty(cow); 297be20aa9dSChris Mason *cow_ret = cow; 298be20aa9dSChris Mason return 0; 299be20aa9dSChris Mason } 300be20aa9dSChris Mason 301bd989ba3SJan Schmidt enum mod_log_op { 302bd989ba3SJan Schmidt MOD_LOG_KEY_REPLACE, 303bd989ba3SJan Schmidt MOD_LOG_KEY_ADD, 304bd989ba3SJan Schmidt MOD_LOG_KEY_REMOVE, 305bd989ba3SJan Schmidt MOD_LOG_KEY_REMOVE_WHILE_FREEING, 306bd989ba3SJan Schmidt MOD_LOG_KEY_REMOVE_WHILE_MOVING, 307bd989ba3SJan Schmidt MOD_LOG_MOVE_KEYS, 308bd989ba3SJan Schmidt MOD_LOG_ROOT_REPLACE, 309bd989ba3SJan Schmidt }; 310bd989ba3SJan Schmidt 311bd989ba3SJan Schmidt struct tree_mod_move { 312bd989ba3SJan Schmidt int dst_slot; 313bd989ba3SJan Schmidt int nr_items; 314bd989ba3SJan Schmidt }; 315bd989ba3SJan Schmidt 316bd989ba3SJan Schmidt struct tree_mod_root { 317bd989ba3SJan Schmidt u64 logical; 318bd989ba3SJan Schmidt u8 level; 319bd989ba3SJan Schmidt }; 320bd989ba3SJan Schmidt 321bd989ba3SJan Schmidt struct tree_mod_elem { 322bd989ba3SJan Schmidt struct rb_node node; 323bd989ba3SJan Schmidt u64 index; /* shifted logical */ 324097b8a7cSJan Schmidt u64 seq; 325bd989ba3SJan Schmidt enum mod_log_op op; 326bd989ba3SJan Schmidt 327bd989ba3SJan Schmidt /* this is used for MOD_LOG_KEY_* and MOD_LOG_MOVE_KEYS operations */ 328bd989ba3SJan Schmidt int slot; 329bd989ba3SJan Schmidt 330bd989ba3SJan Schmidt /* this is used for MOD_LOG_KEY* and MOD_LOG_ROOT_REPLACE */ 331bd989ba3SJan Schmidt u64 generation; 332bd989ba3SJan Schmidt 333bd989ba3SJan Schmidt /* those are used for op == MOD_LOG_KEY_{REPLACE,REMOVE} */ 334bd989ba3SJan Schmidt struct btrfs_disk_key key; 335bd989ba3SJan Schmidt u64 blockptr; 336bd989ba3SJan Schmidt 337bd989ba3SJan Schmidt /* this is used for op == MOD_LOG_MOVE_KEYS */ 338bd989ba3SJan Schmidt struct tree_mod_move move; 339bd989ba3SJan Schmidt 340bd989ba3SJan Schmidt /* this is used for op == MOD_LOG_ROOT_REPLACE */ 341bd989ba3SJan Schmidt struct tree_mod_root old_root; 342bd989ba3SJan Schmidt }; 343bd989ba3SJan Schmidt 344097b8a7cSJan Schmidt static inline void tree_mod_log_read_lock(struct btrfs_fs_info *fs_info) 345bd989ba3SJan Schmidt { 346097b8a7cSJan Schmidt read_lock(&fs_info->tree_mod_log_lock); 347bd989ba3SJan Schmidt } 348bd989ba3SJan Schmidt 349097b8a7cSJan Schmidt static inline void tree_mod_log_read_unlock(struct btrfs_fs_info *fs_info) 350097b8a7cSJan Schmidt { 351097b8a7cSJan Schmidt read_unlock(&fs_info->tree_mod_log_lock); 352097b8a7cSJan Schmidt } 353097b8a7cSJan Schmidt 354097b8a7cSJan Schmidt static inline void tree_mod_log_write_lock(struct btrfs_fs_info *fs_info) 355097b8a7cSJan Schmidt { 356097b8a7cSJan Schmidt write_lock(&fs_info->tree_mod_log_lock); 357097b8a7cSJan Schmidt } 358097b8a7cSJan Schmidt 359097b8a7cSJan Schmidt static inline void tree_mod_log_write_unlock(struct btrfs_fs_info *fs_info) 360097b8a7cSJan Schmidt { 361097b8a7cSJan Schmidt write_unlock(&fs_info->tree_mod_log_lock); 362097b8a7cSJan Schmidt } 363097b8a7cSJan Schmidt 364097b8a7cSJan Schmidt /* 365097b8a7cSJan Schmidt * This adds a new blocker to the tree mod log's blocker list if the @elem 366097b8a7cSJan Schmidt * passed does not already have a sequence number set. So when a caller expects 367097b8a7cSJan Schmidt * to record tree modifications, it should ensure to set elem->seq to zero 368097b8a7cSJan Schmidt * before calling btrfs_get_tree_mod_seq. 369097b8a7cSJan Schmidt * Returns a fresh, unused tree log modification sequence number, even if no new 370097b8a7cSJan Schmidt * blocker was added. 371097b8a7cSJan Schmidt */ 372097b8a7cSJan Schmidt u64 btrfs_get_tree_mod_seq(struct btrfs_fs_info *fs_info, 373bd989ba3SJan Schmidt struct seq_list *elem) 374bd989ba3SJan Schmidt { 375097b8a7cSJan Schmidt u64 seq; 376097b8a7cSJan Schmidt 377097b8a7cSJan Schmidt tree_mod_log_write_lock(fs_info); 378bd989ba3SJan Schmidt spin_lock(&fs_info->tree_mod_seq_lock); 379097b8a7cSJan Schmidt if (!elem->seq) { 380097b8a7cSJan Schmidt elem->seq = btrfs_inc_tree_mod_seq(fs_info); 381097b8a7cSJan Schmidt list_add_tail(&elem->list, &fs_info->tree_mod_seq_list); 382097b8a7cSJan Schmidt } 383097b8a7cSJan Schmidt seq = btrfs_inc_tree_mod_seq(fs_info); 384bd989ba3SJan Schmidt spin_unlock(&fs_info->tree_mod_seq_lock); 385097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 386097b8a7cSJan Schmidt 387097b8a7cSJan Schmidt return seq; 388bd989ba3SJan Schmidt } 389bd989ba3SJan Schmidt 390bd989ba3SJan Schmidt void btrfs_put_tree_mod_seq(struct btrfs_fs_info *fs_info, 391bd989ba3SJan Schmidt struct seq_list *elem) 392bd989ba3SJan Schmidt { 393bd989ba3SJan Schmidt struct rb_root *tm_root; 394bd989ba3SJan Schmidt struct rb_node *node; 395bd989ba3SJan Schmidt struct rb_node *next; 396bd989ba3SJan Schmidt struct seq_list *cur_elem; 397bd989ba3SJan Schmidt struct tree_mod_elem *tm; 398bd989ba3SJan Schmidt u64 min_seq = (u64)-1; 399bd989ba3SJan Schmidt u64 seq_putting = elem->seq; 400bd989ba3SJan Schmidt 401bd989ba3SJan Schmidt if (!seq_putting) 402bd989ba3SJan Schmidt return; 403bd989ba3SJan Schmidt 404bd989ba3SJan Schmidt spin_lock(&fs_info->tree_mod_seq_lock); 405bd989ba3SJan Schmidt list_del(&elem->list); 406097b8a7cSJan Schmidt elem->seq = 0; 407bd989ba3SJan Schmidt 408bd989ba3SJan Schmidt list_for_each_entry(cur_elem, &fs_info->tree_mod_seq_list, list) { 409097b8a7cSJan Schmidt if (cur_elem->seq < min_seq) { 410bd989ba3SJan Schmidt if (seq_putting > cur_elem->seq) { 411bd989ba3SJan Schmidt /* 412bd989ba3SJan Schmidt * blocker with lower sequence number exists, we 413bd989ba3SJan Schmidt * cannot remove anything from the log 414bd989ba3SJan Schmidt */ 415097b8a7cSJan Schmidt spin_unlock(&fs_info->tree_mod_seq_lock); 416097b8a7cSJan Schmidt return; 417bd989ba3SJan Schmidt } 418bd989ba3SJan Schmidt min_seq = cur_elem->seq; 419bd989ba3SJan Schmidt } 420bd989ba3SJan Schmidt } 421097b8a7cSJan Schmidt spin_unlock(&fs_info->tree_mod_seq_lock); 422097b8a7cSJan Schmidt 423097b8a7cSJan Schmidt /* 424bd989ba3SJan Schmidt * anything that's lower than the lowest existing (read: blocked) 425bd989ba3SJan Schmidt * sequence number can be removed from the tree. 426bd989ba3SJan Schmidt */ 427097b8a7cSJan Schmidt tree_mod_log_write_lock(fs_info); 428bd989ba3SJan Schmidt tm_root = &fs_info->tree_mod_log; 429bd989ba3SJan Schmidt for (node = rb_first(tm_root); node; node = next) { 430bd989ba3SJan Schmidt next = rb_next(node); 431bd989ba3SJan Schmidt tm = container_of(node, struct tree_mod_elem, node); 432097b8a7cSJan Schmidt if (tm->seq > min_seq) 433bd989ba3SJan Schmidt continue; 434bd989ba3SJan Schmidt rb_erase(node, tm_root); 435bd989ba3SJan Schmidt kfree(tm); 436bd989ba3SJan Schmidt } 437097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 438bd989ba3SJan Schmidt } 439bd989ba3SJan Schmidt 440bd989ba3SJan Schmidt /* 441bd989ba3SJan Schmidt * key order of the log: 442bd989ba3SJan Schmidt * index -> sequence 443bd989ba3SJan Schmidt * 444bd989ba3SJan Schmidt * the index is the shifted logical of the *new* root node for root replace 445bd989ba3SJan Schmidt * operations, or the shifted logical of the affected block for all other 446bd989ba3SJan Schmidt * operations. 447bd989ba3SJan Schmidt */ 448bd989ba3SJan Schmidt static noinline int 449bd989ba3SJan Schmidt __tree_mod_log_insert(struct btrfs_fs_info *fs_info, struct tree_mod_elem *tm) 450bd989ba3SJan Schmidt { 451bd989ba3SJan Schmidt struct rb_root *tm_root; 452bd989ba3SJan Schmidt struct rb_node **new; 453bd989ba3SJan Schmidt struct rb_node *parent = NULL; 454bd989ba3SJan Schmidt struct tree_mod_elem *cur; 455bd989ba3SJan Schmidt 456097b8a7cSJan Schmidt BUG_ON(!tm || !tm->seq); 457bd989ba3SJan Schmidt 458bd989ba3SJan Schmidt tm_root = &fs_info->tree_mod_log; 459bd989ba3SJan Schmidt new = &tm_root->rb_node; 460bd989ba3SJan Schmidt while (*new) { 461bd989ba3SJan Schmidt cur = container_of(*new, struct tree_mod_elem, node); 462bd989ba3SJan Schmidt parent = *new; 463bd989ba3SJan Schmidt if (cur->index < tm->index) 464bd989ba3SJan Schmidt new = &((*new)->rb_left); 465bd989ba3SJan Schmidt else if (cur->index > tm->index) 466bd989ba3SJan Schmidt new = &((*new)->rb_right); 467097b8a7cSJan Schmidt else if (cur->seq < tm->seq) 468bd989ba3SJan Schmidt new = &((*new)->rb_left); 469097b8a7cSJan Schmidt else if (cur->seq > tm->seq) 470bd989ba3SJan Schmidt new = &((*new)->rb_right); 471bd989ba3SJan Schmidt else { 472bd989ba3SJan Schmidt kfree(tm); 473097b8a7cSJan Schmidt return -EEXIST; 474bd989ba3SJan Schmidt } 475bd989ba3SJan Schmidt } 476bd989ba3SJan Schmidt 477bd989ba3SJan Schmidt rb_link_node(&tm->node, parent, new); 478bd989ba3SJan Schmidt rb_insert_color(&tm->node, tm_root); 479097b8a7cSJan Schmidt return 0; 480bd989ba3SJan Schmidt } 481bd989ba3SJan Schmidt 482097b8a7cSJan Schmidt /* 483097b8a7cSJan Schmidt * Determines if logging can be omitted. Returns 1 if it can. Otherwise, it 484097b8a7cSJan Schmidt * returns zero with the tree_mod_log_lock acquired. The caller must hold 485097b8a7cSJan Schmidt * this until all tree mod log insertions are recorded in the rb tree and then 486097b8a7cSJan Schmidt * call tree_mod_log_write_unlock() to release. 487097b8a7cSJan Schmidt */ 488e9b7fd4dSJan Schmidt static inline int tree_mod_dont_log(struct btrfs_fs_info *fs_info, 489e9b7fd4dSJan Schmidt struct extent_buffer *eb) { 490e9b7fd4dSJan Schmidt smp_mb(); 491e9b7fd4dSJan Schmidt if (list_empty(&(fs_info)->tree_mod_seq_list)) 492e9b7fd4dSJan Schmidt return 1; 493097b8a7cSJan Schmidt if (eb && btrfs_header_level(eb) == 0) 494e9b7fd4dSJan Schmidt return 1; 495097b8a7cSJan Schmidt 496097b8a7cSJan Schmidt tree_mod_log_write_lock(fs_info); 497097b8a7cSJan Schmidt if (list_empty(&fs_info->tree_mod_seq_list)) { 498097b8a7cSJan Schmidt /* 499097b8a7cSJan Schmidt * someone emptied the list while we were waiting for the lock. 500097b8a7cSJan Schmidt * we must not add to the list when no blocker exists. 501097b8a7cSJan Schmidt */ 502097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 503097b8a7cSJan Schmidt return 1; 504097b8a7cSJan Schmidt } 505097b8a7cSJan Schmidt 506e9b7fd4dSJan Schmidt return 0; 507e9b7fd4dSJan Schmidt } 508e9b7fd4dSJan Schmidt 5093310c36eSJan Schmidt /* 510097b8a7cSJan Schmidt * This allocates memory and gets a tree modification sequence number. 5113310c36eSJan Schmidt * 512097b8a7cSJan Schmidt * Returns <0 on error. 513097b8a7cSJan Schmidt * Returns >0 (the added sequence number) on success. 5143310c36eSJan Schmidt */ 515926dd8a6SJan Schmidt static inline int tree_mod_alloc(struct btrfs_fs_info *fs_info, gfp_t flags, 516bd989ba3SJan Schmidt struct tree_mod_elem **tm_ret) 517bd989ba3SJan Schmidt { 518bd989ba3SJan Schmidt struct tree_mod_elem *tm; 519bd989ba3SJan Schmidt 520097b8a7cSJan Schmidt /* 521097b8a7cSJan Schmidt * once we switch from spin locks to something different, we should 522097b8a7cSJan Schmidt * honor the flags parameter here. 523097b8a7cSJan Schmidt */ 524097b8a7cSJan Schmidt tm = *tm_ret = kzalloc(sizeof(*tm), GFP_ATOMIC); 525bd989ba3SJan Schmidt if (!tm) 526bd989ba3SJan Schmidt return -ENOMEM; 527bd989ba3SJan Schmidt 528097b8a7cSJan Schmidt tm->seq = btrfs_inc_tree_mod_seq(fs_info); 529097b8a7cSJan Schmidt return tm->seq; 530926dd8a6SJan Schmidt } 531bd989ba3SJan Schmidt 532097b8a7cSJan Schmidt static inline int 533097b8a7cSJan Schmidt __tree_mod_log_insert_key(struct btrfs_fs_info *fs_info, 534bd989ba3SJan Schmidt struct extent_buffer *eb, int slot, 535bd989ba3SJan Schmidt enum mod_log_op op, gfp_t flags) 536bd989ba3SJan Schmidt { 537bd989ba3SJan Schmidt int ret; 538097b8a7cSJan Schmidt struct tree_mod_elem *tm; 539bd989ba3SJan Schmidt 540bd989ba3SJan Schmidt ret = tree_mod_alloc(fs_info, flags, &tm); 541097b8a7cSJan Schmidt if (ret < 0) 542bd989ba3SJan Schmidt return ret; 543bd989ba3SJan Schmidt 544bd989ba3SJan Schmidt tm->index = eb->start >> PAGE_CACHE_SHIFT; 545bd989ba3SJan Schmidt if (op != MOD_LOG_KEY_ADD) { 546bd989ba3SJan Schmidt btrfs_node_key(eb, &tm->key, slot); 547bd989ba3SJan Schmidt tm->blockptr = btrfs_node_blockptr(eb, slot); 548bd989ba3SJan Schmidt } 549bd989ba3SJan Schmidt tm->op = op; 550bd989ba3SJan Schmidt tm->slot = slot; 551bd989ba3SJan Schmidt tm->generation = btrfs_node_ptr_generation(eb, slot); 552bd989ba3SJan Schmidt 553097b8a7cSJan Schmidt return __tree_mod_log_insert(fs_info, tm); 554097b8a7cSJan Schmidt } 555097b8a7cSJan Schmidt 556097b8a7cSJan Schmidt static noinline int 557097b8a7cSJan Schmidt tree_mod_log_insert_key_mask(struct btrfs_fs_info *fs_info, 558097b8a7cSJan Schmidt struct extent_buffer *eb, int slot, 559097b8a7cSJan Schmidt enum mod_log_op op, gfp_t flags) 560097b8a7cSJan Schmidt { 561097b8a7cSJan Schmidt int ret; 562097b8a7cSJan Schmidt 563097b8a7cSJan Schmidt if (tree_mod_dont_log(fs_info, eb)) 564097b8a7cSJan Schmidt return 0; 565097b8a7cSJan Schmidt 566097b8a7cSJan Schmidt ret = __tree_mod_log_insert_key(fs_info, eb, slot, op, flags); 567097b8a7cSJan Schmidt 568097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 5693310c36eSJan Schmidt return ret; 570bd989ba3SJan Schmidt } 571bd989ba3SJan Schmidt 572bd989ba3SJan Schmidt static noinline int 573bd989ba3SJan Schmidt tree_mod_log_insert_key(struct btrfs_fs_info *fs_info, struct extent_buffer *eb, 574bd989ba3SJan Schmidt int slot, enum mod_log_op op) 575bd989ba3SJan Schmidt { 576bd989ba3SJan Schmidt return tree_mod_log_insert_key_mask(fs_info, eb, slot, op, GFP_NOFS); 577bd989ba3SJan Schmidt } 578bd989ba3SJan Schmidt 579bd989ba3SJan Schmidt static noinline int 580097b8a7cSJan Schmidt tree_mod_log_insert_key_locked(struct btrfs_fs_info *fs_info, 581097b8a7cSJan Schmidt struct extent_buffer *eb, int slot, 582097b8a7cSJan Schmidt enum mod_log_op op) 583097b8a7cSJan Schmidt { 584097b8a7cSJan Schmidt return __tree_mod_log_insert_key(fs_info, eb, slot, op, GFP_NOFS); 585097b8a7cSJan Schmidt } 586097b8a7cSJan Schmidt 587097b8a7cSJan Schmidt static noinline int 588bd989ba3SJan Schmidt tree_mod_log_insert_move(struct btrfs_fs_info *fs_info, 589bd989ba3SJan Schmidt struct extent_buffer *eb, int dst_slot, int src_slot, 590bd989ba3SJan Schmidt int nr_items, gfp_t flags) 591bd989ba3SJan Schmidt { 592bd989ba3SJan Schmidt struct tree_mod_elem *tm; 593bd989ba3SJan Schmidt int ret; 594bd989ba3SJan Schmidt int i; 595bd989ba3SJan Schmidt 596f395694cSJan Schmidt if (tree_mod_dont_log(fs_info, eb)) 597f395694cSJan Schmidt return 0; 598bd989ba3SJan Schmidt 59901763a2eSJan Schmidt /* 60001763a2eSJan Schmidt * When we override something during the move, we log these removals. 60101763a2eSJan Schmidt * This can only happen when we move towards the beginning of the 60201763a2eSJan Schmidt * buffer, i.e. dst_slot < src_slot. 60301763a2eSJan Schmidt */ 604bd989ba3SJan Schmidt for (i = 0; i + dst_slot < src_slot && i < nr_items; i++) { 605097b8a7cSJan Schmidt ret = tree_mod_log_insert_key_locked(fs_info, eb, i + dst_slot, 606bd989ba3SJan Schmidt MOD_LOG_KEY_REMOVE_WHILE_MOVING); 607bd989ba3SJan Schmidt BUG_ON(ret < 0); 608bd989ba3SJan Schmidt } 609bd989ba3SJan Schmidt 610f395694cSJan Schmidt ret = tree_mod_alloc(fs_info, flags, &tm); 611097b8a7cSJan Schmidt if (ret < 0) 612097b8a7cSJan Schmidt goto out; 613f395694cSJan Schmidt 614bd989ba3SJan Schmidt tm->index = eb->start >> PAGE_CACHE_SHIFT; 615bd989ba3SJan Schmidt tm->slot = src_slot; 616bd989ba3SJan Schmidt tm->move.dst_slot = dst_slot; 617bd989ba3SJan Schmidt tm->move.nr_items = nr_items; 618bd989ba3SJan Schmidt tm->op = MOD_LOG_MOVE_KEYS; 619bd989ba3SJan Schmidt 6203310c36eSJan Schmidt ret = __tree_mod_log_insert(fs_info, tm); 621097b8a7cSJan Schmidt out: 622097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 6233310c36eSJan Schmidt return ret; 624bd989ba3SJan Schmidt } 625bd989ba3SJan Schmidt 626097b8a7cSJan Schmidt static inline void 627097b8a7cSJan Schmidt __tree_mod_log_free_eb(struct btrfs_fs_info *fs_info, struct extent_buffer *eb) 628097b8a7cSJan Schmidt { 629097b8a7cSJan Schmidt int i; 630097b8a7cSJan Schmidt u32 nritems; 631097b8a7cSJan Schmidt int ret; 632097b8a7cSJan Schmidt 633b12a3b1eSChris Mason if (btrfs_header_level(eb) == 0) 634b12a3b1eSChris Mason return; 635b12a3b1eSChris Mason 636097b8a7cSJan Schmidt nritems = btrfs_header_nritems(eb); 637097b8a7cSJan Schmidt for (i = nritems - 1; i >= 0; i--) { 638097b8a7cSJan Schmidt ret = tree_mod_log_insert_key_locked(fs_info, eb, i, 639097b8a7cSJan Schmidt MOD_LOG_KEY_REMOVE_WHILE_FREEING); 640097b8a7cSJan Schmidt BUG_ON(ret < 0); 641097b8a7cSJan Schmidt } 642097b8a7cSJan Schmidt } 643097b8a7cSJan Schmidt 644bd989ba3SJan Schmidt static noinline int 645bd989ba3SJan Schmidt tree_mod_log_insert_root(struct btrfs_fs_info *fs_info, 646bd989ba3SJan Schmidt struct extent_buffer *old_root, 647bd989ba3SJan Schmidt struct extent_buffer *new_root, gfp_t flags) 648bd989ba3SJan Schmidt { 649bd989ba3SJan Schmidt struct tree_mod_elem *tm; 650bd989ba3SJan Schmidt int ret; 651bd989ba3SJan Schmidt 652097b8a7cSJan Schmidt if (tree_mod_dont_log(fs_info, NULL)) 653097b8a7cSJan Schmidt return 0; 654097b8a7cSJan Schmidt 655bd989ba3SJan Schmidt ret = tree_mod_alloc(fs_info, flags, &tm); 656097b8a7cSJan Schmidt if (ret < 0) 657097b8a7cSJan Schmidt goto out; 658bd989ba3SJan Schmidt 659bd989ba3SJan Schmidt tm->index = new_root->start >> PAGE_CACHE_SHIFT; 660bd989ba3SJan Schmidt tm->old_root.logical = old_root->start; 661bd989ba3SJan Schmidt tm->old_root.level = btrfs_header_level(old_root); 662bd989ba3SJan Schmidt tm->generation = btrfs_header_generation(old_root); 663bd989ba3SJan Schmidt tm->op = MOD_LOG_ROOT_REPLACE; 664bd989ba3SJan Schmidt 6653310c36eSJan Schmidt ret = __tree_mod_log_insert(fs_info, tm); 666097b8a7cSJan Schmidt out: 667097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 6683310c36eSJan Schmidt return ret; 669bd989ba3SJan Schmidt } 670bd989ba3SJan Schmidt 671bd989ba3SJan Schmidt static struct tree_mod_elem * 672bd989ba3SJan Schmidt __tree_mod_log_search(struct btrfs_fs_info *fs_info, u64 start, u64 min_seq, 673bd989ba3SJan Schmidt int smallest) 674bd989ba3SJan Schmidt { 675bd989ba3SJan Schmidt struct rb_root *tm_root; 676bd989ba3SJan Schmidt struct rb_node *node; 677bd989ba3SJan Schmidt struct tree_mod_elem *cur = NULL; 678bd989ba3SJan Schmidt struct tree_mod_elem *found = NULL; 679bd989ba3SJan Schmidt u64 index = start >> PAGE_CACHE_SHIFT; 680bd989ba3SJan Schmidt 681097b8a7cSJan Schmidt tree_mod_log_read_lock(fs_info); 682bd989ba3SJan Schmidt tm_root = &fs_info->tree_mod_log; 683bd989ba3SJan Schmidt node = tm_root->rb_node; 684bd989ba3SJan Schmidt while (node) { 685bd989ba3SJan Schmidt cur = container_of(node, struct tree_mod_elem, node); 686bd989ba3SJan Schmidt if (cur->index < index) { 687bd989ba3SJan Schmidt node = node->rb_left; 688bd989ba3SJan Schmidt } else if (cur->index > index) { 689bd989ba3SJan Schmidt node = node->rb_right; 690097b8a7cSJan Schmidt } else if (cur->seq < min_seq) { 691bd989ba3SJan Schmidt node = node->rb_left; 692bd989ba3SJan Schmidt } else if (!smallest) { 693bd989ba3SJan Schmidt /* we want the node with the highest seq */ 694bd989ba3SJan Schmidt if (found) 695097b8a7cSJan Schmidt BUG_ON(found->seq > cur->seq); 696bd989ba3SJan Schmidt found = cur; 697bd989ba3SJan Schmidt node = node->rb_left; 698097b8a7cSJan Schmidt } else if (cur->seq > min_seq) { 699bd989ba3SJan Schmidt /* we want the node with the smallest seq */ 700bd989ba3SJan Schmidt if (found) 701097b8a7cSJan Schmidt BUG_ON(found->seq < cur->seq); 702bd989ba3SJan Schmidt found = cur; 703bd989ba3SJan Schmidt node = node->rb_right; 704bd989ba3SJan Schmidt } else { 705bd989ba3SJan Schmidt found = cur; 706bd989ba3SJan Schmidt break; 707bd989ba3SJan Schmidt } 708bd989ba3SJan Schmidt } 709097b8a7cSJan Schmidt tree_mod_log_read_unlock(fs_info); 710bd989ba3SJan Schmidt 711bd989ba3SJan Schmidt return found; 712bd989ba3SJan Schmidt } 713bd989ba3SJan Schmidt 714bd989ba3SJan Schmidt /* 715bd989ba3SJan Schmidt * this returns the element from the log with the smallest time sequence 716bd989ba3SJan Schmidt * value that's in the log (the oldest log item). any element with a time 717bd989ba3SJan Schmidt * sequence lower than min_seq will be ignored. 718bd989ba3SJan Schmidt */ 719bd989ba3SJan Schmidt static struct tree_mod_elem * 720bd989ba3SJan Schmidt tree_mod_log_search_oldest(struct btrfs_fs_info *fs_info, u64 start, 721bd989ba3SJan Schmidt u64 min_seq) 722bd989ba3SJan Schmidt { 723bd989ba3SJan Schmidt return __tree_mod_log_search(fs_info, start, min_seq, 1); 724bd989ba3SJan Schmidt } 725bd989ba3SJan Schmidt 726bd989ba3SJan Schmidt /* 727bd989ba3SJan Schmidt * this returns the element from the log with the largest time sequence 728bd989ba3SJan Schmidt * value that's in the log (the most recent log item). any element with 729bd989ba3SJan Schmidt * a time sequence lower than min_seq will be ignored. 730bd989ba3SJan Schmidt */ 731bd989ba3SJan Schmidt static struct tree_mod_elem * 732bd989ba3SJan Schmidt tree_mod_log_search(struct btrfs_fs_info *fs_info, u64 start, u64 min_seq) 733bd989ba3SJan Schmidt { 734bd989ba3SJan Schmidt return __tree_mod_log_search(fs_info, start, min_seq, 0); 735bd989ba3SJan Schmidt } 736bd989ba3SJan Schmidt 737097b8a7cSJan Schmidt static noinline void 738bd989ba3SJan Schmidt tree_mod_log_eb_copy(struct btrfs_fs_info *fs_info, struct extent_buffer *dst, 739bd989ba3SJan Schmidt struct extent_buffer *src, unsigned long dst_offset, 740bd989ba3SJan Schmidt unsigned long src_offset, int nr_items) 741bd989ba3SJan Schmidt { 742bd989ba3SJan Schmidt int ret; 743bd989ba3SJan Schmidt int i; 744bd989ba3SJan Schmidt 745e9b7fd4dSJan Schmidt if (tree_mod_dont_log(fs_info, NULL)) 746bd989ba3SJan Schmidt return; 747bd989ba3SJan Schmidt 748097b8a7cSJan Schmidt if (btrfs_header_level(dst) == 0 && btrfs_header_level(src) == 0) { 749097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 750bd989ba3SJan Schmidt return; 751097b8a7cSJan Schmidt } 752bd989ba3SJan Schmidt 753bd989ba3SJan Schmidt for (i = 0; i < nr_items; i++) { 754097b8a7cSJan Schmidt ret = tree_mod_log_insert_key_locked(fs_info, src, 755097b8a7cSJan Schmidt i + src_offset, 756bd989ba3SJan Schmidt MOD_LOG_KEY_REMOVE); 757bd989ba3SJan Schmidt BUG_ON(ret < 0); 758097b8a7cSJan Schmidt ret = tree_mod_log_insert_key_locked(fs_info, dst, 759097b8a7cSJan Schmidt i + dst_offset, 760bd989ba3SJan Schmidt MOD_LOG_KEY_ADD); 761bd989ba3SJan Schmidt BUG_ON(ret < 0); 762bd989ba3SJan Schmidt } 763097b8a7cSJan Schmidt 764097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 765bd989ba3SJan Schmidt } 766bd989ba3SJan Schmidt 767bd989ba3SJan Schmidt static inline void 768bd989ba3SJan Schmidt tree_mod_log_eb_move(struct btrfs_fs_info *fs_info, struct extent_buffer *dst, 769bd989ba3SJan Schmidt int dst_offset, int src_offset, int nr_items) 770bd989ba3SJan Schmidt { 771bd989ba3SJan Schmidt int ret; 772bd989ba3SJan Schmidt ret = tree_mod_log_insert_move(fs_info, dst, dst_offset, src_offset, 773bd989ba3SJan Schmidt nr_items, GFP_NOFS); 774bd989ba3SJan Schmidt BUG_ON(ret < 0); 775bd989ba3SJan Schmidt } 776bd989ba3SJan Schmidt 777097b8a7cSJan Schmidt static noinline void 778bd989ba3SJan Schmidt tree_mod_log_set_node_key(struct btrfs_fs_info *fs_info, 779bd989ba3SJan Schmidt struct extent_buffer *eb, 780bd989ba3SJan Schmidt struct btrfs_disk_key *disk_key, int slot, int atomic) 781bd989ba3SJan Schmidt { 782bd989ba3SJan Schmidt int ret; 783bd989ba3SJan Schmidt 784bd989ba3SJan Schmidt ret = tree_mod_log_insert_key_mask(fs_info, eb, slot, 785bd989ba3SJan Schmidt MOD_LOG_KEY_REPLACE, 786bd989ba3SJan Schmidt atomic ? GFP_ATOMIC : GFP_NOFS); 787bd989ba3SJan Schmidt BUG_ON(ret < 0); 788bd989ba3SJan Schmidt } 789bd989ba3SJan Schmidt 790097b8a7cSJan Schmidt static noinline void 791097b8a7cSJan Schmidt tree_mod_log_free_eb(struct btrfs_fs_info *fs_info, struct extent_buffer *eb) 792bd989ba3SJan Schmidt { 793e9b7fd4dSJan Schmidt if (tree_mod_dont_log(fs_info, eb)) 794bd989ba3SJan Schmidt return; 795bd989ba3SJan Schmidt 796097b8a7cSJan Schmidt __tree_mod_log_free_eb(fs_info, eb); 797097b8a7cSJan Schmidt 798097b8a7cSJan Schmidt tree_mod_log_write_unlock(fs_info); 799bd989ba3SJan Schmidt } 800bd989ba3SJan Schmidt 801097b8a7cSJan Schmidt static noinline void 802bd989ba3SJan Schmidt tree_mod_log_set_root_pointer(struct btrfs_root *root, 803bd989ba3SJan Schmidt struct extent_buffer *new_root_node) 804bd989ba3SJan Schmidt { 805bd989ba3SJan Schmidt int ret; 806bd989ba3SJan Schmidt ret = tree_mod_log_insert_root(root->fs_info, root->node, 807bd989ba3SJan Schmidt new_root_node, GFP_NOFS); 808bd989ba3SJan Schmidt BUG_ON(ret < 0); 809bd989ba3SJan Schmidt } 810bd989ba3SJan Schmidt 811d352ac68SChris Mason /* 8125d4f98a2SYan Zheng * check if the tree block can be shared by multiple trees 8135d4f98a2SYan Zheng */ 8145d4f98a2SYan Zheng int btrfs_block_can_be_shared(struct btrfs_root *root, 8155d4f98a2SYan Zheng struct extent_buffer *buf) 8165d4f98a2SYan Zheng { 8175d4f98a2SYan Zheng /* 8185d4f98a2SYan Zheng * Tree blocks not in refernece counted trees and tree roots 8195d4f98a2SYan Zheng * are never shared. If a block was allocated after the last 8205d4f98a2SYan Zheng * snapshot and the block was not allocated by tree relocation, 8215d4f98a2SYan Zheng * we know the block is not shared. 8225d4f98a2SYan Zheng */ 8235d4f98a2SYan Zheng if (root->ref_cows && 8245d4f98a2SYan Zheng buf != root->node && buf != root->commit_root && 8255d4f98a2SYan Zheng (btrfs_header_generation(buf) <= 8265d4f98a2SYan Zheng btrfs_root_last_snapshot(&root->root_item) || 8275d4f98a2SYan Zheng btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC))) 8285d4f98a2SYan Zheng return 1; 8295d4f98a2SYan Zheng #ifdef BTRFS_COMPAT_EXTENT_TREE_V0 8305d4f98a2SYan Zheng if (root->ref_cows && 8315d4f98a2SYan Zheng btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV) 8325d4f98a2SYan Zheng return 1; 8335d4f98a2SYan Zheng #endif 8345d4f98a2SYan Zheng return 0; 8355d4f98a2SYan Zheng } 8365d4f98a2SYan Zheng 8375d4f98a2SYan Zheng static noinline int update_ref_for_cow(struct btrfs_trans_handle *trans, 8385d4f98a2SYan Zheng struct btrfs_root *root, 8395d4f98a2SYan Zheng struct extent_buffer *buf, 840f0486c68SYan, Zheng struct extent_buffer *cow, 841f0486c68SYan, Zheng int *last_ref) 8425d4f98a2SYan Zheng { 8435d4f98a2SYan Zheng u64 refs; 8445d4f98a2SYan Zheng u64 owner; 8455d4f98a2SYan Zheng u64 flags; 8465d4f98a2SYan Zheng u64 new_flags = 0; 8475d4f98a2SYan Zheng int ret; 8485d4f98a2SYan Zheng 8495d4f98a2SYan Zheng /* 8505d4f98a2SYan Zheng * Backrefs update rules: 8515d4f98a2SYan Zheng * 8525d4f98a2SYan Zheng * Always use full backrefs for extent pointers in tree block 8535d4f98a2SYan Zheng * allocated by tree relocation. 8545d4f98a2SYan Zheng * 8555d4f98a2SYan Zheng * If a shared tree block is no longer referenced by its owner 8565d4f98a2SYan Zheng * tree (btrfs_header_owner(buf) == root->root_key.objectid), 8575d4f98a2SYan Zheng * use full backrefs for extent pointers in tree block. 8585d4f98a2SYan Zheng * 8595d4f98a2SYan Zheng * If a tree block is been relocating 8605d4f98a2SYan Zheng * (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID), 8615d4f98a2SYan Zheng * use full backrefs for extent pointers in tree block. 8625d4f98a2SYan Zheng * The reason for this is some operations (such as drop tree) 8635d4f98a2SYan Zheng * are only allowed for blocks use full backrefs. 8645d4f98a2SYan Zheng */ 8655d4f98a2SYan Zheng 8665d4f98a2SYan Zheng if (btrfs_block_can_be_shared(root, buf)) { 8675d4f98a2SYan Zheng ret = btrfs_lookup_extent_info(trans, root, buf->start, 8685d4f98a2SYan Zheng buf->len, &refs, &flags); 869be1a5564SMark Fasheh if (ret) 870be1a5564SMark Fasheh return ret; 871e5df9573SMark Fasheh if (refs == 0) { 872e5df9573SMark Fasheh ret = -EROFS; 873e5df9573SMark Fasheh btrfs_std_error(root->fs_info, ret); 874e5df9573SMark Fasheh return ret; 875e5df9573SMark Fasheh } 8765d4f98a2SYan Zheng } else { 8775d4f98a2SYan Zheng refs = 1; 8785d4f98a2SYan Zheng if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID || 8795d4f98a2SYan Zheng btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV) 8805d4f98a2SYan Zheng flags = BTRFS_BLOCK_FLAG_FULL_BACKREF; 8815d4f98a2SYan Zheng else 8825d4f98a2SYan Zheng flags = 0; 8835d4f98a2SYan Zheng } 8845d4f98a2SYan Zheng 8855d4f98a2SYan Zheng owner = btrfs_header_owner(buf); 8865d4f98a2SYan Zheng BUG_ON(owner == BTRFS_TREE_RELOC_OBJECTID && 8875d4f98a2SYan Zheng !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF)); 8885d4f98a2SYan Zheng 8895d4f98a2SYan Zheng if (refs > 1) { 8905d4f98a2SYan Zheng if ((owner == root->root_key.objectid || 8915d4f98a2SYan Zheng root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) && 8925d4f98a2SYan Zheng !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF)) { 89366d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, buf, 1, 1); 89479787eaaSJeff Mahoney BUG_ON(ret); /* -ENOMEM */ 8955d4f98a2SYan Zheng 8965d4f98a2SYan Zheng if (root->root_key.objectid == 8975d4f98a2SYan Zheng BTRFS_TREE_RELOC_OBJECTID) { 89866d7e7f0SArne Jansen ret = btrfs_dec_ref(trans, root, buf, 0, 1); 89979787eaaSJeff Mahoney BUG_ON(ret); /* -ENOMEM */ 90066d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, cow, 1, 1); 90179787eaaSJeff Mahoney BUG_ON(ret); /* -ENOMEM */ 9025d4f98a2SYan Zheng } 9035d4f98a2SYan Zheng new_flags |= BTRFS_BLOCK_FLAG_FULL_BACKREF; 9045d4f98a2SYan Zheng } else { 9055d4f98a2SYan Zheng 9065d4f98a2SYan Zheng if (root->root_key.objectid == 9075d4f98a2SYan Zheng BTRFS_TREE_RELOC_OBJECTID) 90866d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, cow, 1, 1); 9095d4f98a2SYan Zheng else 91066d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, cow, 0, 1); 91179787eaaSJeff Mahoney BUG_ON(ret); /* -ENOMEM */ 9125d4f98a2SYan Zheng } 9135d4f98a2SYan Zheng if (new_flags != 0) { 9145d4f98a2SYan Zheng ret = btrfs_set_disk_extent_flags(trans, root, 9155d4f98a2SYan Zheng buf->start, 9165d4f98a2SYan Zheng buf->len, 9175d4f98a2SYan Zheng new_flags, 0); 918be1a5564SMark Fasheh if (ret) 919be1a5564SMark Fasheh return ret; 9205d4f98a2SYan Zheng } 9215d4f98a2SYan Zheng } else { 9225d4f98a2SYan Zheng if (flags & BTRFS_BLOCK_FLAG_FULL_BACKREF) { 9235d4f98a2SYan Zheng if (root->root_key.objectid == 9245d4f98a2SYan Zheng BTRFS_TREE_RELOC_OBJECTID) 92566d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, cow, 1, 1); 9265d4f98a2SYan Zheng else 92766d7e7f0SArne Jansen ret = btrfs_inc_ref(trans, root, cow, 0, 1); 92879787eaaSJeff Mahoney BUG_ON(ret); /* -ENOMEM */ 92966d7e7f0SArne Jansen ret = btrfs_dec_ref(trans, root, buf, 1, 1); 93079787eaaSJeff Mahoney BUG_ON(ret); /* -ENOMEM */ 9315d4f98a2SYan Zheng } 932f230475eSJan Schmidt tree_mod_log_free_eb(root->fs_info, buf); 9335d4f98a2SYan Zheng clean_tree_block(trans, root, buf); 934f0486c68SYan, Zheng *last_ref = 1; 9355d4f98a2SYan Zheng } 9365d4f98a2SYan Zheng return 0; 9375d4f98a2SYan Zheng } 9385d4f98a2SYan Zheng 9395d4f98a2SYan Zheng /* 940d397712bSChris Mason * does the dirty work in cow of a single block. The parent block (if 941d397712bSChris Mason * supplied) is updated to point to the new cow copy. The new buffer is marked 942d397712bSChris Mason * dirty and returned locked. If you modify the block it needs to be marked 943d397712bSChris Mason * dirty again. 944d352ac68SChris Mason * 945d352ac68SChris Mason * search_start -- an allocation hint for the new block 946d352ac68SChris Mason * 947d397712bSChris Mason * empty_size -- a hint that you plan on doing more cow. This is the size in 948d397712bSChris Mason * bytes the allocator should try to find free next to the block it returns. 949d397712bSChris Mason * This is just a hint and may be ignored by the allocator. 950d352ac68SChris Mason */ 951d397712bSChris Mason static noinline int __btrfs_cow_block(struct btrfs_trans_handle *trans, 9525f39d397SChris Mason struct btrfs_root *root, 9535f39d397SChris Mason struct extent_buffer *buf, 9545f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 9555f39d397SChris Mason struct extent_buffer **cow_ret, 9569fa8cfe7SChris Mason u64 search_start, u64 empty_size) 9576702ed49SChris Mason { 9585d4f98a2SYan Zheng struct btrfs_disk_key disk_key; 9595f39d397SChris Mason struct extent_buffer *cow; 960be1a5564SMark Fasheh int level, ret; 961f0486c68SYan, Zheng int last_ref = 0; 962925baeddSChris Mason int unlock_orig = 0; 9635d4f98a2SYan Zheng u64 parent_start; 9646702ed49SChris Mason 965925baeddSChris Mason if (*cow_ret == buf) 966925baeddSChris Mason unlock_orig = 1; 967925baeddSChris Mason 968b9447ef8SChris Mason btrfs_assert_tree_locked(buf); 969925baeddSChris Mason 9707bb86316SChris Mason WARN_ON(root->ref_cows && trans->transid != 9717bb86316SChris Mason root->fs_info->running_transaction->transid); 9726702ed49SChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 9735f39d397SChris Mason 9747bb86316SChris Mason level = btrfs_header_level(buf); 97531840ae1SZheng Yan 9765d4f98a2SYan Zheng if (level == 0) 9775d4f98a2SYan Zheng btrfs_item_key(buf, &disk_key, 0); 9785d4f98a2SYan Zheng else 9795d4f98a2SYan Zheng btrfs_node_key(buf, &disk_key, 0); 9805d4f98a2SYan Zheng 9815d4f98a2SYan Zheng if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) { 9825d4f98a2SYan Zheng if (parent) 9835d4f98a2SYan Zheng parent_start = parent->start; 9845d4f98a2SYan Zheng else 9855d4f98a2SYan Zheng parent_start = 0; 9865d4f98a2SYan Zheng } else 9875d4f98a2SYan Zheng parent_start = 0; 9885d4f98a2SYan Zheng 9895d4f98a2SYan Zheng cow = btrfs_alloc_free_block(trans, root, buf->len, parent_start, 9905d4f98a2SYan Zheng root->root_key.objectid, &disk_key, 9915581a51aSJan Schmidt level, search_start, empty_size); 9926702ed49SChris Mason if (IS_ERR(cow)) 9936702ed49SChris Mason return PTR_ERR(cow); 9946702ed49SChris Mason 995b4ce94deSChris Mason /* cow is set to blocking by btrfs_init_new_buffer */ 996b4ce94deSChris Mason 9975f39d397SChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 998db94535dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 9995f39d397SChris Mason btrfs_set_header_generation(cow, trans->transid); 10005d4f98a2SYan Zheng btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV); 10015d4f98a2SYan Zheng btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN | 10025d4f98a2SYan Zheng BTRFS_HEADER_FLAG_RELOC); 10035d4f98a2SYan Zheng if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) 10045d4f98a2SYan Zheng btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC); 10055d4f98a2SYan Zheng else 10065f39d397SChris Mason btrfs_set_header_owner(cow, root->root_key.objectid); 10076702ed49SChris Mason 10082b82032cSYan Zheng write_extent_buffer(cow, root->fs_info->fsid, 10092b82032cSYan Zheng (unsigned long)btrfs_header_fsid(cow), 10102b82032cSYan Zheng BTRFS_FSID_SIZE); 10112b82032cSYan Zheng 1012be1a5564SMark Fasheh ret = update_ref_for_cow(trans, root, buf, cow, &last_ref); 1013b68dc2a9SMark Fasheh if (ret) { 101479787eaaSJeff Mahoney btrfs_abort_transaction(trans, root, ret); 1015b68dc2a9SMark Fasheh return ret; 1016b68dc2a9SMark Fasheh } 10171a40e23bSZheng Yan 10183fd0a558SYan, Zheng if (root->ref_cows) 10193fd0a558SYan, Zheng btrfs_reloc_cow_block(trans, root, buf, cow); 10203fd0a558SYan, Zheng 10216702ed49SChris Mason if (buf == root->node) { 1022925baeddSChris Mason WARN_ON(parent && parent != buf); 10235d4f98a2SYan Zheng if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID || 10245d4f98a2SYan Zheng btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV) 10255d4f98a2SYan Zheng parent_start = buf->start; 10265d4f98a2SYan Zheng else 10275d4f98a2SYan Zheng parent_start = 0; 1028925baeddSChris Mason 10295f39d397SChris Mason extent_buffer_get(cow); 1030f230475eSJan Schmidt tree_mod_log_set_root_pointer(root, cow); 1031240f62c8SChris Mason rcu_assign_pointer(root->node, cow); 1032925baeddSChris Mason 1033f0486c68SYan, Zheng btrfs_free_tree_block(trans, root, buf, parent_start, 10345581a51aSJan Schmidt last_ref); 10355f39d397SChris Mason free_extent_buffer(buf); 10360b86a832SChris Mason add_root_to_dirty_list(root); 10376702ed49SChris Mason } else { 10385d4f98a2SYan Zheng if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) 10395d4f98a2SYan Zheng parent_start = parent->start; 10405d4f98a2SYan Zheng else 10415d4f98a2SYan Zheng parent_start = 0; 10425d4f98a2SYan Zheng 10435d4f98a2SYan Zheng WARN_ON(trans->transid != btrfs_header_generation(parent)); 1044f230475eSJan Schmidt tree_mod_log_insert_key(root->fs_info, parent, parent_slot, 1045f230475eSJan Schmidt MOD_LOG_KEY_REPLACE); 10465f39d397SChris Mason btrfs_set_node_blockptr(parent, parent_slot, 1047db94535dSChris Mason cow->start); 104874493f7aSChris Mason btrfs_set_node_ptr_generation(parent, parent_slot, 104974493f7aSChris Mason trans->transid); 10506702ed49SChris Mason btrfs_mark_buffer_dirty(parent); 1051f0486c68SYan, Zheng btrfs_free_tree_block(trans, root, buf, parent_start, 10525581a51aSJan Schmidt last_ref); 10536702ed49SChris Mason } 1054925baeddSChris Mason if (unlock_orig) 1055925baeddSChris Mason btrfs_tree_unlock(buf); 10563083ee2eSJosef Bacik free_extent_buffer_stale(buf); 10576702ed49SChris Mason btrfs_mark_buffer_dirty(cow); 10586702ed49SChris Mason *cow_ret = cow; 10596702ed49SChris Mason return 0; 10606702ed49SChris Mason } 10616702ed49SChris Mason 10625d9e75c4SJan Schmidt /* 10635d9e75c4SJan Schmidt * returns the logical address of the oldest predecessor of the given root. 10645d9e75c4SJan Schmidt * entries older than time_seq are ignored. 10655d9e75c4SJan Schmidt */ 10665d9e75c4SJan Schmidt static struct tree_mod_elem * 10675d9e75c4SJan Schmidt __tree_mod_log_oldest_root(struct btrfs_fs_info *fs_info, 10685d9e75c4SJan Schmidt struct btrfs_root *root, u64 time_seq) 10695d9e75c4SJan Schmidt { 10705d9e75c4SJan Schmidt struct tree_mod_elem *tm; 10715d9e75c4SJan Schmidt struct tree_mod_elem *found = NULL; 10725d9e75c4SJan Schmidt u64 root_logical = root->node->start; 10735d9e75c4SJan Schmidt int looped = 0; 10745d9e75c4SJan Schmidt 10755d9e75c4SJan Schmidt if (!time_seq) 10765d9e75c4SJan Schmidt return 0; 10775d9e75c4SJan Schmidt 10785d9e75c4SJan Schmidt /* 10795d9e75c4SJan Schmidt * the very last operation that's logged for a root is the replacement 10805d9e75c4SJan Schmidt * operation (if it is replaced at all). this has the index of the *new* 10815d9e75c4SJan Schmidt * root, making it the very first operation that's logged for this root. 10825d9e75c4SJan Schmidt */ 10835d9e75c4SJan Schmidt while (1) { 10845d9e75c4SJan Schmidt tm = tree_mod_log_search_oldest(fs_info, root_logical, 10855d9e75c4SJan Schmidt time_seq); 10865d9e75c4SJan Schmidt if (!looped && !tm) 10875d9e75c4SJan Schmidt return 0; 10885d9e75c4SJan Schmidt /* 108928da9fb4SJan Schmidt * if there are no tree operation for the oldest root, we simply 109028da9fb4SJan Schmidt * return it. this should only happen if that (old) root is at 109128da9fb4SJan Schmidt * level 0. 10925d9e75c4SJan Schmidt */ 109328da9fb4SJan Schmidt if (!tm) 109428da9fb4SJan Schmidt break; 10955d9e75c4SJan Schmidt 109628da9fb4SJan Schmidt /* 109728da9fb4SJan Schmidt * if there's an operation that's not a root replacement, we 109828da9fb4SJan Schmidt * found the oldest version of our root. normally, we'll find a 109928da9fb4SJan Schmidt * MOD_LOG_KEY_REMOVE_WHILE_FREEING operation here. 110028da9fb4SJan Schmidt */ 11015d9e75c4SJan Schmidt if (tm->op != MOD_LOG_ROOT_REPLACE) 11025d9e75c4SJan Schmidt break; 11035d9e75c4SJan Schmidt 11045d9e75c4SJan Schmidt found = tm; 11055d9e75c4SJan Schmidt root_logical = tm->old_root.logical; 11065d9e75c4SJan Schmidt BUG_ON(root_logical == root->node->start); 11075d9e75c4SJan Schmidt looped = 1; 11085d9e75c4SJan Schmidt } 11095d9e75c4SJan Schmidt 1110a95236d9SJan Schmidt /* if there's no old root to return, return what we found instead */ 1111a95236d9SJan Schmidt if (!found) 1112a95236d9SJan Schmidt found = tm; 1113a95236d9SJan Schmidt 11145d9e75c4SJan Schmidt return found; 11155d9e75c4SJan Schmidt } 11165d9e75c4SJan Schmidt 11175d9e75c4SJan Schmidt /* 11185d9e75c4SJan Schmidt * tm is a pointer to the first operation to rewind within eb. then, all 11195d9e75c4SJan Schmidt * previous operations will be rewinded (until we reach something older than 11205d9e75c4SJan Schmidt * time_seq). 11215d9e75c4SJan Schmidt */ 11225d9e75c4SJan Schmidt static void 11235d9e75c4SJan Schmidt __tree_mod_log_rewind(struct extent_buffer *eb, u64 time_seq, 11245d9e75c4SJan Schmidt struct tree_mod_elem *first_tm) 11255d9e75c4SJan Schmidt { 11265d9e75c4SJan Schmidt u32 n; 11275d9e75c4SJan Schmidt struct rb_node *next; 11285d9e75c4SJan Schmidt struct tree_mod_elem *tm = first_tm; 11295d9e75c4SJan Schmidt unsigned long o_dst; 11305d9e75c4SJan Schmidt unsigned long o_src; 11315d9e75c4SJan Schmidt unsigned long p_size = sizeof(struct btrfs_key_ptr); 11325d9e75c4SJan Schmidt 11335d9e75c4SJan Schmidt n = btrfs_header_nritems(eb); 1134097b8a7cSJan Schmidt while (tm && tm->seq >= time_seq) { 11355d9e75c4SJan Schmidt /* 11365d9e75c4SJan Schmidt * all the operations are recorded with the operator used for 11375d9e75c4SJan Schmidt * the modification. as we're going backwards, we do the 11385d9e75c4SJan Schmidt * opposite of each operation here. 11395d9e75c4SJan Schmidt */ 11405d9e75c4SJan Schmidt switch (tm->op) { 11415d9e75c4SJan Schmidt case MOD_LOG_KEY_REMOVE_WHILE_FREEING: 11425d9e75c4SJan Schmidt BUG_ON(tm->slot < n); 11435d9e75c4SJan Schmidt case MOD_LOG_KEY_REMOVE: 114495c80bb1SLiu Bo n++; 114595c80bb1SLiu Bo case MOD_LOG_KEY_REMOVE_WHILE_MOVING: 11465d9e75c4SJan Schmidt btrfs_set_node_key(eb, &tm->key, tm->slot); 11475d9e75c4SJan Schmidt btrfs_set_node_blockptr(eb, tm->slot, tm->blockptr); 11485d9e75c4SJan Schmidt btrfs_set_node_ptr_generation(eb, tm->slot, 11495d9e75c4SJan Schmidt tm->generation); 11505d9e75c4SJan Schmidt break; 11515d9e75c4SJan Schmidt case MOD_LOG_KEY_REPLACE: 11525d9e75c4SJan Schmidt BUG_ON(tm->slot >= n); 11535d9e75c4SJan Schmidt btrfs_set_node_key(eb, &tm->key, tm->slot); 11545d9e75c4SJan Schmidt btrfs_set_node_blockptr(eb, tm->slot, tm->blockptr); 11555d9e75c4SJan Schmidt btrfs_set_node_ptr_generation(eb, tm->slot, 11565d9e75c4SJan Schmidt tm->generation); 11575d9e75c4SJan Schmidt break; 11585d9e75c4SJan Schmidt case MOD_LOG_KEY_ADD: 115919956c7eSJan Schmidt /* if a move operation is needed it's in the log */ 11605d9e75c4SJan Schmidt n--; 11615d9e75c4SJan Schmidt break; 11625d9e75c4SJan Schmidt case MOD_LOG_MOVE_KEYS: 1163c3193108SJan Schmidt o_dst = btrfs_node_key_ptr_offset(tm->slot); 1164c3193108SJan Schmidt o_src = btrfs_node_key_ptr_offset(tm->move.dst_slot); 1165c3193108SJan Schmidt memmove_extent_buffer(eb, o_dst, o_src, 11665d9e75c4SJan Schmidt tm->move.nr_items * p_size); 11675d9e75c4SJan Schmidt break; 11685d9e75c4SJan Schmidt case MOD_LOG_ROOT_REPLACE: 11695d9e75c4SJan Schmidt /* 11705d9e75c4SJan Schmidt * this operation is special. for roots, this must be 11715d9e75c4SJan Schmidt * handled explicitly before rewinding. 11725d9e75c4SJan Schmidt * for non-roots, this operation may exist if the node 11735d9e75c4SJan Schmidt * was a root: root A -> child B; then A gets empty and 11745d9e75c4SJan Schmidt * B is promoted to the new root. in the mod log, we'll 11755d9e75c4SJan Schmidt * have a root-replace operation for B, a tree block 11765d9e75c4SJan Schmidt * that is no root. we simply ignore that operation. 11775d9e75c4SJan Schmidt */ 11785d9e75c4SJan Schmidt break; 11795d9e75c4SJan Schmidt } 11805d9e75c4SJan Schmidt next = rb_next(&tm->node); 11815d9e75c4SJan Schmidt if (!next) 11825d9e75c4SJan Schmidt break; 11835d9e75c4SJan Schmidt tm = container_of(next, struct tree_mod_elem, node); 11845d9e75c4SJan Schmidt if (tm->index != first_tm->index) 11855d9e75c4SJan Schmidt break; 11865d9e75c4SJan Schmidt } 11875d9e75c4SJan Schmidt btrfs_set_header_nritems(eb, n); 11885d9e75c4SJan Schmidt } 11895d9e75c4SJan Schmidt 11905d9e75c4SJan Schmidt static struct extent_buffer * 11915d9e75c4SJan Schmidt tree_mod_log_rewind(struct btrfs_fs_info *fs_info, struct extent_buffer *eb, 11925d9e75c4SJan Schmidt u64 time_seq) 11935d9e75c4SJan Schmidt { 11945d9e75c4SJan Schmidt struct extent_buffer *eb_rewin; 11955d9e75c4SJan Schmidt struct tree_mod_elem *tm; 11965d9e75c4SJan Schmidt 11975d9e75c4SJan Schmidt if (!time_seq) 11985d9e75c4SJan Schmidt return eb; 11995d9e75c4SJan Schmidt 12005d9e75c4SJan Schmidt if (btrfs_header_level(eb) == 0) 12015d9e75c4SJan Schmidt return eb; 12025d9e75c4SJan Schmidt 12035d9e75c4SJan Schmidt tm = tree_mod_log_search(fs_info, eb->start, time_seq); 12045d9e75c4SJan Schmidt if (!tm) 12055d9e75c4SJan Schmidt return eb; 12065d9e75c4SJan Schmidt 12075d9e75c4SJan Schmidt if (tm->op == MOD_LOG_KEY_REMOVE_WHILE_FREEING) { 12085d9e75c4SJan Schmidt BUG_ON(tm->slot != 0); 12095d9e75c4SJan Schmidt eb_rewin = alloc_dummy_extent_buffer(eb->start, 12105d9e75c4SJan Schmidt fs_info->tree_root->nodesize); 12115d9e75c4SJan Schmidt BUG_ON(!eb_rewin); 12125d9e75c4SJan Schmidt btrfs_set_header_bytenr(eb_rewin, eb->start); 12135d9e75c4SJan Schmidt btrfs_set_header_backref_rev(eb_rewin, 12145d9e75c4SJan Schmidt btrfs_header_backref_rev(eb)); 12155d9e75c4SJan Schmidt btrfs_set_header_owner(eb_rewin, btrfs_header_owner(eb)); 1216c3193108SJan Schmidt btrfs_set_header_level(eb_rewin, btrfs_header_level(eb)); 12175d9e75c4SJan Schmidt } else { 12185d9e75c4SJan Schmidt eb_rewin = btrfs_clone_extent_buffer(eb); 12195d9e75c4SJan Schmidt BUG_ON(!eb_rewin); 12205d9e75c4SJan Schmidt } 12215d9e75c4SJan Schmidt 12225d9e75c4SJan Schmidt extent_buffer_get(eb_rewin); 12235d9e75c4SJan Schmidt free_extent_buffer(eb); 12245d9e75c4SJan Schmidt 12255d9e75c4SJan Schmidt __tree_mod_log_rewind(eb_rewin, time_seq, tm); 122657911b8bSJan Schmidt WARN_ON(btrfs_header_nritems(eb_rewin) > 122757911b8bSJan Schmidt BTRFS_NODEPTRS_PER_BLOCK(fs_info->fs_root)); 12285d9e75c4SJan Schmidt 12295d9e75c4SJan Schmidt return eb_rewin; 12305d9e75c4SJan Schmidt } 12315d9e75c4SJan Schmidt 12328ba97a15SJan Schmidt /* 12338ba97a15SJan Schmidt * get_old_root() rewinds the state of @root's root node to the given @time_seq 12348ba97a15SJan Schmidt * value. If there are no changes, the current root->root_node is returned. If 12358ba97a15SJan Schmidt * anything changed in between, there's a fresh buffer allocated on which the 12368ba97a15SJan Schmidt * rewind operations are done. In any case, the returned buffer is read locked. 12378ba97a15SJan Schmidt * Returns NULL on error (with no locks held). 12388ba97a15SJan Schmidt */ 12395d9e75c4SJan Schmidt static inline struct extent_buffer * 12405d9e75c4SJan Schmidt get_old_root(struct btrfs_root *root, u64 time_seq) 12415d9e75c4SJan Schmidt { 12425d9e75c4SJan Schmidt struct tree_mod_elem *tm; 12435d9e75c4SJan Schmidt struct extent_buffer *eb; 12447bfdcf7fSLiu Bo struct extent_buffer *old; 1245a95236d9SJan Schmidt struct tree_mod_root *old_root = NULL; 12464325edd0SChris Mason u64 old_generation = 0; 1247a95236d9SJan Schmidt u64 logical; 1248834328a8SJan Schmidt u32 blocksize; 12495d9e75c4SJan Schmidt 12508ba97a15SJan Schmidt eb = btrfs_read_lock_root_node(root); 12515d9e75c4SJan Schmidt tm = __tree_mod_log_oldest_root(root->fs_info, root, time_seq); 12525d9e75c4SJan Schmidt if (!tm) 12535d9e75c4SJan Schmidt return root->node; 12545d9e75c4SJan Schmidt 1255a95236d9SJan Schmidt if (tm->op == MOD_LOG_ROOT_REPLACE) { 12565d9e75c4SJan Schmidt old_root = &tm->old_root; 12575d9e75c4SJan Schmidt old_generation = tm->generation; 1258a95236d9SJan Schmidt logical = old_root->logical; 1259a95236d9SJan Schmidt } else { 1260a95236d9SJan Schmidt logical = root->node->start; 1261a95236d9SJan Schmidt } 12625d9e75c4SJan Schmidt 1263a95236d9SJan Schmidt tm = tree_mod_log_search(root->fs_info, logical, time_seq); 1264834328a8SJan Schmidt if (old_root && tm && tm->op != MOD_LOG_KEY_REMOVE_WHILE_FREEING) { 1265834328a8SJan Schmidt btrfs_tree_read_unlock(root->node); 1266834328a8SJan Schmidt free_extent_buffer(root->node); 1267834328a8SJan Schmidt blocksize = btrfs_level_size(root, old_root->level); 12687bfdcf7fSLiu Bo old = read_tree_block(root, logical, blocksize, 0); 12697bfdcf7fSLiu Bo if (!old) { 1270834328a8SJan Schmidt pr_warn("btrfs: failed to read tree block %llu from get_old_root\n", 1271834328a8SJan Schmidt logical); 1272834328a8SJan Schmidt WARN_ON(1); 1273834328a8SJan Schmidt } else { 12747bfdcf7fSLiu Bo eb = btrfs_clone_extent_buffer(old); 12757bfdcf7fSLiu Bo free_extent_buffer(old); 1276834328a8SJan Schmidt } 1277834328a8SJan Schmidt } else if (old_root) { 1278834328a8SJan Schmidt btrfs_tree_read_unlock(root->node); 1279834328a8SJan Schmidt free_extent_buffer(root->node); 128028da9fb4SJan Schmidt eb = alloc_dummy_extent_buffer(logical, root->nodesize); 1281834328a8SJan Schmidt } else { 1282a95236d9SJan Schmidt eb = btrfs_clone_extent_buffer(root->node); 12838ba97a15SJan Schmidt btrfs_tree_read_unlock(root->node); 12848ba97a15SJan Schmidt free_extent_buffer(root->node); 1285834328a8SJan Schmidt } 1286834328a8SJan Schmidt 12878ba97a15SJan Schmidt if (!eb) 12888ba97a15SJan Schmidt return NULL; 1289d6381084SJan Schmidt extent_buffer_get(eb); 12908ba97a15SJan Schmidt btrfs_tree_read_lock(eb); 1291a95236d9SJan Schmidt if (old_root) { 12925d9e75c4SJan Schmidt btrfs_set_header_bytenr(eb, eb->start); 12935d9e75c4SJan Schmidt btrfs_set_header_backref_rev(eb, BTRFS_MIXED_BACKREF_REV); 12945d9e75c4SJan Schmidt btrfs_set_header_owner(eb, root->root_key.objectid); 12955d9e75c4SJan Schmidt btrfs_set_header_level(eb, old_root->level); 12965d9e75c4SJan Schmidt btrfs_set_header_generation(eb, old_generation); 1297a95236d9SJan Schmidt } 129828da9fb4SJan Schmidt if (tm) 12995d9e75c4SJan Schmidt __tree_mod_log_rewind(eb, time_seq, tm); 130028da9fb4SJan Schmidt else 130128da9fb4SJan Schmidt WARN_ON(btrfs_header_level(eb) != 0); 130257911b8bSJan Schmidt WARN_ON(btrfs_header_nritems(eb) > BTRFS_NODEPTRS_PER_BLOCK(root)); 13035d9e75c4SJan Schmidt 13045d9e75c4SJan Schmidt return eb; 13055d9e75c4SJan Schmidt } 13065d9e75c4SJan Schmidt 13075b6602e7SJan Schmidt int btrfs_old_root_level(struct btrfs_root *root, u64 time_seq) 13085b6602e7SJan Schmidt { 13095b6602e7SJan Schmidt struct tree_mod_elem *tm; 13105b6602e7SJan Schmidt int level; 13115b6602e7SJan Schmidt 13125b6602e7SJan Schmidt tm = __tree_mod_log_oldest_root(root->fs_info, root, time_seq); 13135b6602e7SJan Schmidt if (tm && tm->op == MOD_LOG_ROOT_REPLACE) { 13145b6602e7SJan Schmidt level = tm->old_root.level; 13155b6602e7SJan Schmidt } else { 13165b6602e7SJan Schmidt rcu_read_lock(); 13175b6602e7SJan Schmidt level = btrfs_header_level(root->node); 13185b6602e7SJan Schmidt rcu_read_unlock(); 13195b6602e7SJan Schmidt } 13205b6602e7SJan Schmidt 13215b6602e7SJan Schmidt return level; 13225b6602e7SJan Schmidt } 13235b6602e7SJan Schmidt 13245d4f98a2SYan Zheng static inline int should_cow_block(struct btrfs_trans_handle *trans, 13255d4f98a2SYan Zheng struct btrfs_root *root, 13265d4f98a2SYan Zheng struct extent_buffer *buf) 13275d4f98a2SYan Zheng { 1328f1ebcc74SLiu Bo /* ensure we can see the force_cow */ 1329f1ebcc74SLiu Bo smp_rmb(); 1330f1ebcc74SLiu Bo 1331f1ebcc74SLiu Bo /* 1332f1ebcc74SLiu Bo * We do not need to cow a block if 1333f1ebcc74SLiu Bo * 1) this block is not created or changed in this transaction; 1334f1ebcc74SLiu Bo * 2) this block does not belong to TREE_RELOC tree; 1335f1ebcc74SLiu Bo * 3) the root is not forced COW. 1336f1ebcc74SLiu Bo * 1337f1ebcc74SLiu Bo * What is forced COW: 1338f1ebcc74SLiu Bo * when we create snapshot during commiting the transaction, 1339f1ebcc74SLiu Bo * after we've finished coping src root, we must COW the shared 1340f1ebcc74SLiu Bo * block to ensure the metadata consistency. 1341f1ebcc74SLiu Bo */ 13425d4f98a2SYan Zheng if (btrfs_header_generation(buf) == trans->transid && 13435d4f98a2SYan Zheng !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN) && 13445d4f98a2SYan Zheng !(root->root_key.objectid != BTRFS_TREE_RELOC_OBJECTID && 1345f1ebcc74SLiu Bo btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC)) && 1346f1ebcc74SLiu Bo !root->force_cow) 13475d4f98a2SYan Zheng return 0; 13485d4f98a2SYan Zheng return 1; 13495d4f98a2SYan Zheng } 13505d4f98a2SYan Zheng 1351d352ac68SChris Mason /* 1352d352ac68SChris Mason * cows a single block, see __btrfs_cow_block for the real work. 1353d352ac68SChris Mason * This version of it has extra checks so that a block isn't cow'd more than 1354d352ac68SChris Mason * once per transaction, as long as it hasn't been written yet 1355d352ac68SChris Mason */ 1356d397712bSChris Mason noinline int btrfs_cow_block(struct btrfs_trans_handle *trans, 13575f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *buf, 13585f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 13599fa8cfe7SChris Mason struct extent_buffer **cow_ret) 136002217ed2SChris Mason { 13616702ed49SChris Mason u64 search_start; 1362f510cfecSChris Mason int ret; 1363dc17ff8fSChris Mason 1364ccd467d6SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 1365d397712bSChris Mason printk(KERN_CRIT "trans %llu running %llu\n", 1366d397712bSChris Mason (unsigned long long)trans->transid, 1367d397712bSChris Mason (unsigned long long) 1368ccd467d6SChris Mason root->fs_info->running_transaction->transid); 1369ccd467d6SChris Mason WARN_ON(1); 1370ccd467d6SChris Mason } 1371ccd467d6SChris Mason if (trans->transid != root->fs_info->generation) { 1372d397712bSChris Mason printk(KERN_CRIT "trans %llu running %llu\n", 1373d397712bSChris Mason (unsigned long long)trans->transid, 1374d397712bSChris Mason (unsigned long long)root->fs_info->generation); 1375ccd467d6SChris Mason WARN_ON(1); 1376ccd467d6SChris Mason } 1377dc17ff8fSChris Mason 13785d4f98a2SYan Zheng if (!should_cow_block(trans, root, buf)) { 137902217ed2SChris Mason *cow_ret = buf; 138002217ed2SChris Mason return 0; 138102217ed2SChris Mason } 1382c487685dSChris Mason 13830b86a832SChris Mason search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1); 1384b4ce94deSChris Mason 1385b4ce94deSChris Mason if (parent) 1386b4ce94deSChris Mason btrfs_set_lock_blocking(parent); 1387b4ce94deSChris Mason btrfs_set_lock_blocking(buf); 1388b4ce94deSChris Mason 1389f510cfecSChris Mason ret = __btrfs_cow_block(trans, root, buf, parent, 13909fa8cfe7SChris Mason parent_slot, cow_ret, search_start, 0); 13911abe9b8aSliubo 13921abe9b8aSliubo trace_btrfs_cow_block(root, buf, *cow_ret); 13931abe9b8aSliubo 1394f510cfecSChris Mason return ret; 13952c90e5d6SChris Mason } 13966702ed49SChris Mason 1397d352ac68SChris Mason /* 1398d352ac68SChris Mason * helper function for defrag to decide if two blocks pointed to by a 1399d352ac68SChris Mason * node are actually close by 1400d352ac68SChris Mason */ 14016b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize) 14026702ed49SChris Mason { 14036b80053dSChris Mason if (blocknr < other && other - (blocknr + blocksize) < 32768) 14046702ed49SChris Mason return 1; 14056b80053dSChris Mason if (blocknr > other && blocknr - (other + blocksize) < 32768) 14066702ed49SChris Mason return 1; 140702217ed2SChris Mason return 0; 140802217ed2SChris Mason } 140902217ed2SChris Mason 1410081e9573SChris Mason /* 1411081e9573SChris Mason * compare two keys in a memcmp fashion 1412081e9573SChris Mason */ 1413081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 1414081e9573SChris Mason { 1415081e9573SChris Mason struct btrfs_key k1; 1416081e9573SChris Mason 1417081e9573SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 1418081e9573SChris Mason 141920736abaSDiego Calleja return btrfs_comp_cpu_keys(&k1, k2); 1420081e9573SChris Mason } 1421081e9573SChris Mason 1422f3465ca4SJosef Bacik /* 1423f3465ca4SJosef Bacik * same as comp_keys only with two btrfs_key's 1424f3465ca4SJosef Bacik */ 14255d4f98a2SYan Zheng int btrfs_comp_cpu_keys(struct btrfs_key *k1, struct btrfs_key *k2) 1426f3465ca4SJosef Bacik { 1427f3465ca4SJosef Bacik if (k1->objectid > k2->objectid) 1428f3465ca4SJosef Bacik return 1; 1429f3465ca4SJosef Bacik if (k1->objectid < k2->objectid) 1430f3465ca4SJosef Bacik return -1; 1431f3465ca4SJosef Bacik if (k1->type > k2->type) 1432f3465ca4SJosef Bacik return 1; 1433f3465ca4SJosef Bacik if (k1->type < k2->type) 1434f3465ca4SJosef Bacik return -1; 1435f3465ca4SJosef Bacik if (k1->offset > k2->offset) 1436f3465ca4SJosef Bacik return 1; 1437f3465ca4SJosef Bacik if (k1->offset < k2->offset) 1438f3465ca4SJosef Bacik return -1; 1439f3465ca4SJosef Bacik return 0; 1440f3465ca4SJosef Bacik } 1441081e9573SChris Mason 1442d352ac68SChris Mason /* 1443d352ac68SChris Mason * this is used by the defrag code to go through all the 1444d352ac68SChris Mason * leaves pointed to by a node and reallocate them so that 1445d352ac68SChris Mason * disk order is close to key order 1446d352ac68SChris Mason */ 14476702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans, 14485f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *parent, 1449a6b6e75eSChris Mason int start_slot, int cache_only, u64 *last_ret, 1450a6b6e75eSChris Mason struct btrfs_key *progress) 14516702ed49SChris Mason { 14526b80053dSChris Mason struct extent_buffer *cur; 14536702ed49SChris Mason u64 blocknr; 1454ca7a79adSChris Mason u64 gen; 1455e9d0b13bSChris Mason u64 search_start = *last_ret; 1456e9d0b13bSChris Mason u64 last_block = 0; 14576702ed49SChris Mason u64 other; 14586702ed49SChris Mason u32 parent_nritems; 14596702ed49SChris Mason int end_slot; 14606702ed49SChris Mason int i; 14616702ed49SChris Mason int err = 0; 1462f2183bdeSChris Mason int parent_level; 14636b80053dSChris Mason int uptodate; 14646b80053dSChris Mason u32 blocksize; 1465081e9573SChris Mason int progress_passed = 0; 1466081e9573SChris Mason struct btrfs_disk_key disk_key; 14676702ed49SChris Mason 14685708b959SChris Mason parent_level = btrfs_header_level(parent); 14695708b959SChris Mason if (cache_only && parent_level != 1) 14705708b959SChris Mason return 0; 14715708b959SChris Mason 1472d397712bSChris Mason if (trans->transaction != root->fs_info->running_transaction) 14736702ed49SChris Mason WARN_ON(1); 1474d397712bSChris Mason if (trans->transid != root->fs_info->generation) 14756702ed49SChris Mason WARN_ON(1); 147686479a04SChris Mason 14776b80053dSChris Mason parent_nritems = btrfs_header_nritems(parent); 14786b80053dSChris Mason blocksize = btrfs_level_size(root, parent_level - 1); 14796702ed49SChris Mason end_slot = parent_nritems; 14806702ed49SChris Mason 14816702ed49SChris Mason if (parent_nritems == 1) 14826702ed49SChris Mason return 0; 14836702ed49SChris Mason 1484b4ce94deSChris Mason btrfs_set_lock_blocking(parent); 1485b4ce94deSChris Mason 14866702ed49SChris Mason for (i = start_slot; i < end_slot; i++) { 14876702ed49SChris Mason int close = 1; 1488a6b6e75eSChris Mason 1489081e9573SChris Mason btrfs_node_key(parent, &disk_key, i); 1490081e9573SChris Mason if (!progress_passed && comp_keys(&disk_key, progress) < 0) 1491081e9573SChris Mason continue; 1492081e9573SChris Mason 1493081e9573SChris Mason progress_passed = 1; 14946b80053dSChris Mason blocknr = btrfs_node_blockptr(parent, i); 1495ca7a79adSChris Mason gen = btrfs_node_ptr_generation(parent, i); 1496e9d0b13bSChris Mason if (last_block == 0) 1497e9d0b13bSChris Mason last_block = blocknr; 14985708b959SChris Mason 14996702ed49SChris Mason if (i > 0) { 15006b80053dSChris Mason other = btrfs_node_blockptr(parent, i - 1); 15016b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 15026702ed49SChris Mason } 15030ef3e66bSChris Mason if (!close && i < end_slot - 2) { 15046b80053dSChris Mason other = btrfs_node_blockptr(parent, i + 1); 15056b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 15066702ed49SChris Mason } 1507e9d0b13bSChris Mason if (close) { 1508e9d0b13bSChris Mason last_block = blocknr; 15096702ed49SChris Mason continue; 1510e9d0b13bSChris Mason } 15116702ed49SChris Mason 15126b80053dSChris Mason cur = btrfs_find_tree_block(root, blocknr, blocksize); 15136b80053dSChris Mason if (cur) 1514b9fab919SChris Mason uptodate = btrfs_buffer_uptodate(cur, gen, 0); 15156b80053dSChris Mason else 15166b80053dSChris Mason uptodate = 0; 15175708b959SChris Mason if (!cur || !uptodate) { 15186702ed49SChris Mason if (cache_only) { 15196b80053dSChris Mason free_extent_buffer(cur); 15206702ed49SChris Mason continue; 15216702ed49SChris Mason } 15226b80053dSChris Mason if (!cur) { 15236b80053dSChris Mason cur = read_tree_block(root, blocknr, 1524ca7a79adSChris Mason blocksize, gen); 152597d9a8a4STsutomu Itoh if (!cur) 152697d9a8a4STsutomu Itoh return -EIO; 15276b80053dSChris Mason } else if (!uptodate) { 1528018642a1STsutomu Itoh err = btrfs_read_buffer(cur, gen); 1529018642a1STsutomu Itoh if (err) { 1530018642a1STsutomu Itoh free_extent_buffer(cur); 1531018642a1STsutomu Itoh return err; 1532018642a1STsutomu Itoh } 15336702ed49SChris Mason } 1534f2183bdeSChris Mason } 1535e9d0b13bSChris Mason if (search_start == 0) 15366b80053dSChris Mason search_start = last_block; 1537e9d0b13bSChris Mason 1538e7a84565SChris Mason btrfs_tree_lock(cur); 1539b4ce94deSChris Mason btrfs_set_lock_blocking(cur); 15406b80053dSChris Mason err = __btrfs_cow_block(trans, root, cur, parent, i, 1541e7a84565SChris Mason &cur, search_start, 15426b80053dSChris Mason min(16 * blocksize, 15439fa8cfe7SChris Mason (end_slot - i) * blocksize)); 1544252c38f0SYan if (err) { 1545e7a84565SChris Mason btrfs_tree_unlock(cur); 15466b80053dSChris Mason free_extent_buffer(cur); 15476702ed49SChris Mason break; 1548252c38f0SYan } 1549e7a84565SChris Mason search_start = cur->start; 1550e7a84565SChris Mason last_block = cur->start; 1551f2183bdeSChris Mason *last_ret = search_start; 1552e7a84565SChris Mason btrfs_tree_unlock(cur); 1553e7a84565SChris Mason free_extent_buffer(cur); 15546702ed49SChris Mason } 15556702ed49SChris Mason return err; 15566702ed49SChris Mason } 15576702ed49SChris Mason 155874123bd7SChris Mason /* 155974123bd7SChris Mason * The leaf data grows from end-to-front in the node. 156074123bd7SChris Mason * this returns the address of the start of the last item, 156174123bd7SChris Mason * which is the stop of the leaf data stack 156274123bd7SChris Mason */ 1563123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 15645f39d397SChris Mason struct extent_buffer *leaf) 1565be0e5c09SChris Mason { 15665f39d397SChris Mason u32 nr = btrfs_header_nritems(leaf); 1567be0e5c09SChris Mason if (nr == 0) 1568123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 15695f39d397SChris Mason return btrfs_item_offset_nr(leaf, nr - 1); 1570be0e5c09SChris Mason } 1571be0e5c09SChris Mason 1572aa5d6bedSChris Mason 157374123bd7SChris Mason /* 15745f39d397SChris Mason * search for key in the extent_buffer. The items start at offset p, 15755f39d397SChris Mason * and they are item_size apart. There are 'max' items in p. 15765f39d397SChris Mason * 157774123bd7SChris Mason * the slot in the array is returned via slot, and it points to 157874123bd7SChris Mason * the place where you would insert key if it is not found in 157974123bd7SChris Mason * the array. 158074123bd7SChris Mason * 158174123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 158274123bd7SChris Mason */ 1583e02119d5SChris Mason static noinline int generic_bin_search(struct extent_buffer *eb, 1584e02119d5SChris Mason unsigned long p, 15855f39d397SChris Mason int item_size, struct btrfs_key *key, 1586be0e5c09SChris Mason int max, int *slot) 1587be0e5c09SChris Mason { 1588be0e5c09SChris Mason int low = 0; 1589be0e5c09SChris Mason int high = max; 1590be0e5c09SChris Mason int mid; 1591be0e5c09SChris Mason int ret; 1592479965d6SChris Mason struct btrfs_disk_key *tmp = NULL; 15935f39d397SChris Mason struct btrfs_disk_key unaligned; 15945f39d397SChris Mason unsigned long offset; 15955f39d397SChris Mason char *kaddr = NULL; 15965f39d397SChris Mason unsigned long map_start = 0; 15975f39d397SChris Mason unsigned long map_len = 0; 1598479965d6SChris Mason int err; 1599be0e5c09SChris Mason 1600be0e5c09SChris Mason while (low < high) { 1601be0e5c09SChris Mason mid = (low + high) / 2; 16025f39d397SChris Mason offset = p + mid * item_size; 16035f39d397SChris Mason 1604a6591715SChris Mason if (!kaddr || offset < map_start || 16055f39d397SChris Mason (offset + sizeof(struct btrfs_disk_key)) > 16065f39d397SChris Mason map_start + map_len) { 1607934d375bSChris Mason 1608934d375bSChris Mason err = map_private_extent_buffer(eb, offset, 1609479965d6SChris Mason sizeof(struct btrfs_disk_key), 1610a6591715SChris Mason &kaddr, &map_start, &map_len); 16115f39d397SChris Mason 1612479965d6SChris Mason if (!err) { 1613479965d6SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 1614479965d6SChris Mason map_start); 1615479965d6SChris Mason } else { 16165f39d397SChris Mason read_extent_buffer(eb, &unaligned, 16175f39d397SChris Mason offset, sizeof(unaligned)); 16185f39d397SChris Mason tmp = &unaligned; 1619479965d6SChris Mason } 1620479965d6SChris Mason 16215f39d397SChris Mason } else { 16225f39d397SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 16235f39d397SChris Mason map_start); 16245f39d397SChris Mason } 1625be0e5c09SChris Mason ret = comp_keys(tmp, key); 1626be0e5c09SChris Mason 1627be0e5c09SChris Mason if (ret < 0) 1628be0e5c09SChris Mason low = mid + 1; 1629be0e5c09SChris Mason else if (ret > 0) 1630be0e5c09SChris Mason high = mid; 1631be0e5c09SChris Mason else { 1632be0e5c09SChris Mason *slot = mid; 1633be0e5c09SChris Mason return 0; 1634be0e5c09SChris Mason } 1635be0e5c09SChris Mason } 1636be0e5c09SChris Mason *slot = low; 1637be0e5c09SChris Mason return 1; 1638be0e5c09SChris Mason } 1639be0e5c09SChris Mason 164097571fd0SChris Mason /* 164197571fd0SChris Mason * simple bin_search frontend that does the right thing for 164297571fd0SChris Mason * leaves vs nodes 164397571fd0SChris Mason */ 16445f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, 16455f39d397SChris Mason int level, int *slot) 1646be0e5c09SChris Mason { 1647f775738fSWang Sheng-Hui if (level == 0) 16485f39d397SChris Mason return generic_bin_search(eb, 16495f39d397SChris Mason offsetof(struct btrfs_leaf, items), 16500783fcfcSChris Mason sizeof(struct btrfs_item), 16515f39d397SChris Mason key, btrfs_header_nritems(eb), 16527518a238SChris Mason slot); 1653f775738fSWang Sheng-Hui else 16545f39d397SChris Mason return generic_bin_search(eb, 16555f39d397SChris Mason offsetof(struct btrfs_node, ptrs), 1656123abc88SChris Mason sizeof(struct btrfs_key_ptr), 16575f39d397SChris Mason key, btrfs_header_nritems(eb), 16587518a238SChris Mason slot); 1659be0e5c09SChris Mason } 1660be0e5c09SChris Mason 16615d4f98a2SYan Zheng int btrfs_bin_search(struct extent_buffer *eb, struct btrfs_key *key, 16625d4f98a2SYan Zheng int level, int *slot) 16635d4f98a2SYan Zheng { 16645d4f98a2SYan Zheng return bin_search(eb, key, level, slot); 16655d4f98a2SYan Zheng } 16665d4f98a2SYan Zheng 1667f0486c68SYan, Zheng static void root_add_used(struct btrfs_root *root, u32 size) 1668f0486c68SYan, Zheng { 1669f0486c68SYan, Zheng spin_lock(&root->accounting_lock); 1670f0486c68SYan, Zheng btrfs_set_root_used(&root->root_item, 1671f0486c68SYan, Zheng btrfs_root_used(&root->root_item) + size); 1672f0486c68SYan, Zheng spin_unlock(&root->accounting_lock); 1673f0486c68SYan, Zheng } 1674f0486c68SYan, Zheng 1675f0486c68SYan, Zheng static void root_sub_used(struct btrfs_root *root, u32 size) 1676f0486c68SYan, Zheng { 1677f0486c68SYan, Zheng spin_lock(&root->accounting_lock); 1678f0486c68SYan, Zheng btrfs_set_root_used(&root->root_item, 1679f0486c68SYan, Zheng btrfs_root_used(&root->root_item) - size); 1680f0486c68SYan, Zheng spin_unlock(&root->accounting_lock); 1681f0486c68SYan, Zheng } 1682f0486c68SYan, Zheng 1683d352ac68SChris Mason /* given a node and slot number, this reads the blocks it points to. The 1684d352ac68SChris Mason * extent buffer is returned with a reference taken (but unlocked). 1685d352ac68SChris Mason * NULL is returned on error. 1686d352ac68SChris Mason */ 1687e02119d5SChris Mason static noinline struct extent_buffer *read_node_slot(struct btrfs_root *root, 16885f39d397SChris Mason struct extent_buffer *parent, int slot) 1689bb803951SChris Mason { 1690ca7a79adSChris Mason int level = btrfs_header_level(parent); 1691bb803951SChris Mason if (slot < 0) 1692bb803951SChris Mason return NULL; 16935f39d397SChris Mason if (slot >= btrfs_header_nritems(parent)) 1694bb803951SChris Mason return NULL; 1695ca7a79adSChris Mason 1696ca7a79adSChris Mason BUG_ON(level == 0); 1697ca7a79adSChris Mason 1698db94535dSChris Mason return read_tree_block(root, btrfs_node_blockptr(parent, slot), 1699ca7a79adSChris Mason btrfs_level_size(root, level - 1), 1700ca7a79adSChris Mason btrfs_node_ptr_generation(parent, slot)); 1701bb803951SChris Mason } 1702bb803951SChris Mason 1703d352ac68SChris Mason /* 1704d352ac68SChris Mason * node level balancing, used to make sure nodes are in proper order for 1705d352ac68SChris Mason * item deletion. We balance from the top down, so we have to make sure 1706d352ac68SChris Mason * that a deletion won't leave an node completely empty later on. 1707d352ac68SChris Mason */ 1708e02119d5SChris Mason static noinline int balance_level(struct btrfs_trans_handle *trans, 170998ed5174SChris Mason struct btrfs_root *root, 171098ed5174SChris Mason struct btrfs_path *path, int level) 1711bb803951SChris Mason { 17125f39d397SChris Mason struct extent_buffer *right = NULL; 17135f39d397SChris Mason struct extent_buffer *mid; 17145f39d397SChris Mason struct extent_buffer *left = NULL; 17155f39d397SChris Mason struct extent_buffer *parent = NULL; 1716bb803951SChris Mason int ret = 0; 1717bb803951SChris Mason int wret; 1718bb803951SChris Mason int pslot; 1719bb803951SChris Mason int orig_slot = path->slots[level]; 172079f95c82SChris Mason u64 orig_ptr; 1721bb803951SChris Mason 1722bb803951SChris Mason if (level == 0) 1723bb803951SChris Mason return 0; 1724bb803951SChris Mason 17255f39d397SChris Mason mid = path->nodes[level]; 1726b4ce94deSChris Mason 1727bd681513SChris Mason WARN_ON(path->locks[level] != BTRFS_WRITE_LOCK && 1728bd681513SChris Mason path->locks[level] != BTRFS_WRITE_LOCK_BLOCKING); 17297bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 17307bb86316SChris Mason 17311d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 173279f95c82SChris Mason 1733a05a9bb1SLi Zefan if (level < BTRFS_MAX_LEVEL - 1) { 17345f39d397SChris Mason parent = path->nodes[level + 1]; 1735bb803951SChris Mason pslot = path->slots[level + 1]; 1736a05a9bb1SLi Zefan } 1737bb803951SChris Mason 173840689478SChris Mason /* 173940689478SChris Mason * deal with the case where there is only one pointer in the root 174040689478SChris Mason * by promoting the node below to a root 174140689478SChris Mason */ 17425f39d397SChris Mason if (!parent) { 17435f39d397SChris Mason struct extent_buffer *child; 1744bb803951SChris Mason 17455f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 1746bb803951SChris Mason return 0; 1747bb803951SChris Mason 1748bb803951SChris Mason /* promote the child to a root */ 17495f39d397SChris Mason child = read_node_slot(root, mid, 0); 1750305a26afSMark Fasheh if (!child) { 1751305a26afSMark Fasheh ret = -EROFS; 1752305a26afSMark Fasheh btrfs_std_error(root->fs_info, ret); 1753305a26afSMark Fasheh goto enospc; 1754305a26afSMark Fasheh } 1755305a26afSMark Fasheh 1756925baeddSChris Mason btrfs_tree_lock(child); 1757b4ce94deSChris Mason btrfs_set_lock_blocking(child); 17589fa8cfe7SChris Mason ret = btrfs_cow_block(trans, root, child, mid, 0, &child); 1759f0486c68SYan, Zheng if (ret) { 1760f0486c68SYan, Zheng btrfs_tree_unlock(child); 1761f0486c68SYan, Zheng free_extent_buffer(child); 1762f0486c68SYan, Zheng goto enospc; 1763f0486c68SYan, Zheng } 17642f375ab9SYan 1765ba1bfbd5SJan Schmidt tree_mod_log_free_eb(root->fs_info, root->node); 1766f230475eSJan Schmidt tree_mod_log_set_root_pointer(root, child); 1767240f62c8SChris Mason rcu_assign_pointer(root->node, child); 1768925baeddSChris Mason 17690b86a832SChris Mason add_root_to_dirty_list(root); 1770925baeddSChris Mason btrfs_tree_unlock(child); 1771b4ce94deSChris Mason 1772925baeddSChris Mason path->locks[level] = 0; 1773bb803951SChris Mason path->nodes[level] = NULL; 17745f39d397SChris Mason clean_tree_block(trans, root, mid); 1775925baeddSChris Mason btrfs_tree_unlock(mid); 1776bb803951SChris Mason /* once for the path */ 17775f39d397SChris Mason free_extent_buffer(mid); 1778f0486c68SYan, Zheng 1779f0486c68SYan, Zheng root_sub_used(root, mid->len); 17805581a51aSJan Schmidt btrfs_free_tree_block(trans, root, mid, 0, 1); 1781bb803951SChris Mason /* once for the root ptr */ 17823083ee2eSJosef Bacik free_extent_buffer_stale(mid); 1783f0486c68SYan, Zheng return 0; 1784bb803951SChris Mason } 17855f39d397SChris Mason if (btrfs_header_nritems(mid) > 1786123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 1787bb803951SChris Mason return 0; 1788bb803951SChris Mason 17895f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 17905f39d397SChris Mason if (left) { 1791925baeddSChris Mason btrfs_tree_lock(left); 1792b4ce94deSChris Mason btrfs_set_lock_blocking(left); 17935f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 17949fa8cfe7SChris Mason parent, pslot - 1, &left); 179554aa1f4dSChris Mason if (wret) { 179654aa1f4dSChris Mason ret = wret; 179754aa1f4dSChris Mason goto enospc; 179854aa1f4dSChris Mason } 17992cc58cf2SChris Mason } 18005f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 18015f39d397SChris Mason if (right) { 1802925baeddSChris Mason btrfs_tree_lock(right); 1803b4ce94deSChris Mason btrfs_set_lock_blocking(right); 18045f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 18059fa8cfe7SChris Mason parent, pslot + 1, &right); 18062cc58cf2SChris Mason if (wret) { 18072cc58cf2SChris Mason ret = wret; 18082cc58cf2SChris Mason goto enospc; 18092cc58cf2SChris Mason } 18102cc58cf2SChris Mason } 18112cc58cf2SChris Mason 18122cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 18135f39d397SChris Mason if (left) { 18145f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 1815bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 181679f95c82SChris Mason if (wret < 0) 181779f95c82SChris Mason ret = wret; 1818bb803951SChris Mason } 181979f95c82SChris Mason 182079f95c82SChris Mason /* 182179f95c82SChris Mason * then try to empty the right most buffer into the middle 182279f95c82SChris Mason */ 18235f39d397SChris Mason if (right) { 1824971a1f66SChris Mason wret = push_node_left(trans, root, mid, right, 1); 182554aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 182679f95c82SChris Mason ret = wret; 18275f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 18285f39d397SChris Mason clean_tree_block(trans, root, right); 1829925baeddSChris Mason btrfs_tree_unlock(right); 1830f3ea38daSJan Schmidt del_ptr(trans, root, path, level + 1, pslot + 1, 1); 1831f0486c68SYan, Zheng root_sub_used(root, right->len); 18325581a51aSJan Schmidt btrfs_free_tree_block(trans, root, right, 0, 1); 18333083ee2eSJosef Bacik free_extent_buffer_stale(right); 1834f0486c68SYan, Zheng right = NULL; 1835bb803951SChris Mason } else { 18365f39d397SChris Mason struct btrfs_disk_key right_key; 18375f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 1838f230475eSJan Schmidt tree_mod_log_set_node_key(root->fs_info, parent, 1839f230475eSJan Schmidt &right_key, pslot + 1, 0); 18405f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 18415f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 1842bb803951SChris Mason } 1843bb803951SChris Mason } 18445f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 184579f95c82SChris Mason /* 184679f95c82SChris Mason * we're not allowed to leave a node with one item in the 184779f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 184879f95c82SChris Mason * could try to delete the only pointer in this node. 184979f95c82SChris Mason * So, pull some keys from the left. 185079f95c82SChris Mason * There has to be a left pointer at this point because 185179f95c82SChris Mason * otherwise we would have pulled some pointers from the 185279f95c82SChris Mason * right 185379f95c82SChris Mason */ 1854305a26afSMark Fasheh if (!left) { 1855305a26afSMark Fasheh ret = -EROFS; 1856305a26afSMark Fasheh btrfs_std_error(root->fs_info, ret); 1857305a26afSMark Fasheh goto enospc; 1858305a26afSMark Fasheh } 18595f39d397SChris Mason wret = balance_node_right(trans, root, mid, left); 186054aa1f4dSChris Mason if (wret < 0) { 186179f95c82SChris Mason ret = wret; 186254aa1f4dSChris Mason goto enospc; 186354aa1f4dSChris Mason } 1864bce4eae9SChris Mason if (wret == 1) { 1865bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 1866bce4eae9SChris Mason if (wret < 0) 1867bce4eae9SChris Mason ret = wret; 1868bce4eae9SChris Mason } 186979f95c82SChris Mason BUG_ON(wret == 1); 187079f95c82SChris Mason } 18715f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 18725f39d397SChris Mason clean_tree_block(trans, root, mid); 1873925baeddSChris Mason btrfs_tree_unlock(mid); 1874f3ea38daSJan Schmidt del_ptr(trans, root, path, level + 1, pslot, 1); 1875f0486c68SYan, Zheng root_sub_used(root, mid->len); 18765581a51aSJan Schmidt btrfs_free_tree_block(trans, root, mid, 0, 1); 18773083ee2eSJosef Bacik free_extent_buffer_stale(mid); 1878f0486c68SYan, Zheng mid = NULL; 187979f95c82SChris Mason } else { 188079f95c82SChris Mason /* update the parent key to reflect our changes */ 18815f39d397SChris Mason struct btrfs_disk_key mid_key; 18825f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 1883f230475eSJan Schmidt tree_mod_log_set_node_key(root->fs_info, parent, &mid_key, 1884f230475eSJan Schmidt pslot, 0); 18855f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 18865f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 188779f95c82SChris Mason } 1888bb803951SChris Mason 188979f95c82SChris Mason /* update the path */ 18905f39d397SChris Mason if (left) { 18915f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 18925f39d397SChris Mason extent_buffer_get(left); 1893925baeddSChris Mason /* left was locked after cow */ 18945f39d397SChris Mason path->nodes[level] = left; 1895bb803951SChris Mason path->slots[level + 1] -= 1; 1896bb803951SChris Mason path->slots[level] = orig_slot; 1897925baeddSChris Mason if (mid) { 1898925baeddSChris Mason btrfs_tree_unlock(mid); 18995f39d397SChris Mason free_extent_buffer(mid); 1900925baeddSChris Mason } 1901bb803951SChris Mason } else { 19025f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 1903bb803951SChris Mason path->slots[level] = orig_slot; 1904bb803951SChris Mason } 1905bb803951SChris Mason } 190679f95c82SChris Mason /* double check we haven't messed things up */ 1907e20d96d6SChris Mason if (orig_ptr != 19085f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 190979f95c82SChris Mason BUG(); 191054aa1f4dSChris Mason enospc: 1911925baeddSChris Mason if (right) { 1912925baeddSChris Mason btrfs_tree_unlock(right); 19135f39d397SChris Mason free_extent_buffer(right); 1914925baeddSChris Mason } 1915925baeddSChris Mason if (left) { 1916925baeddSChris Mason if (path->nodes[level] != left) 1917925baeddSChris Mason btrfs_tree_unlock(left); 19185f39d397SChris Mason free_extent_buffer(left); 1919925baeddSChris Mason } 1920bb803951SChris Mason return ret; 1921bb803951SChris Mason } 1922bb803951SChris Mason 1923d352ac68SChris Mason /* Node balancing for insertion. Here we only split or push nodes around 1924d352ac68SChris Mason * when they are completely full. This is also done top down, so we 1925d352ac68SChris Mason * have to be pessimistic. 1926d352ac68SChris Mason */ 1927d397712bSChris Mason static noinline int push_nodes_for_insert(struct btrfs_trans_handle *trans, 1928e66f709bSChris Mason struct btrfs_root *root, 1929e66f709bSChris Mason struct btrfs_path *path, int level) 1930e66f709bSChris Mason { 19315f39d397SChris Mason struct extent_buffer *right = NULL; 19325f39d397SChris Mason struct extent_buffer *mid; 19335f39d397SChris Mason struct extent_buffer *left = NULL; 19345f39d397SChris Mason struct extent_buffer *parent = NULL; 1935e66f709bSChris Mason int ret = 0; 1936e66f709bSChris Mason int wret; 1937e66f709bSChris Mason int pslot; 1938e66f709bSChris Mason int orig_slot = path->slots[level]; 1939e66f709bSChris Mason 1940e66f709bSChris Mason if (level == 0) 1941e66f709bSChris Mason return 1; 1942e66f709bSChris Mason 19435f39d397SChris Mason mid = path->nodes[level]; 19447bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 1945e66f709bSChris Mason 1946a05a9bb1SLi Zefan if (level < BTRFS_MAX_LEVEL - 1) { 19475f39d397SChris Mason parent = path->nodes[level + 1]; 1948e66f709bSChris Mason pslot = path->slots[level + 1]; 1949a05a9bb1SLi Zefan } 1950e66f709bSChris Mason 19515f39d397SChris Mason if (!parent) 1952e66f709bSChris Mason return 1; 1953e66f709bSChris Mason 19545f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 1955e66f709bSChris Mason 1956e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 19575f39d397SChris Mason if (left) { 1958e66f709bSChris Mason u32 left_nr; 1959925baeddSChris Mason 1960925baeddSChris Mason btrfs_tree_lock(left); 1961b4ce94deSChris Mason btrfs_set_lock_blocking(left); 1962b4ce94deSChris Mason 19635f39d397SChris Mason left_nr = btrfs_header_nritems(left); 196433ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 196533ade1f8SChris Mason wret = 1; 196633ade1f8SChris Mason } else { 19675f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 19689fa8cfe7SChris Mason pslot - 1, &left); 196954aa1f4dSChris Mason if (ret) 197054aa1f4dSChris Mason wret = 1; 197154aa1f4dSChris Mason else { 197254aa1f4dSChris Mason wret = push_node_left(trans, root, 1973971a1f66SChris Mason left, mid, 0); 197454aa1f4dSChris Mason } 197533ade1f8SChris Mason } 1976e66f709bSChris Mason if (wret < 0) 1977e66f709bSChris Mason ret = wret; 1978e66f709bSChris Mason if (wret == 0) { 19795f39d397SChris Mason struct btrfs_disk_key disk_key; 1980e66f709bSChris Mason orig_slot += left_nr; 19815f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 1982f230475eSJan Schmidt tree_mod_log_set_node_key(root->fs_info, parent, 1983f230475eSJan Schmidt &disk_key, pslot, 0); 19845f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 19855f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 19865f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 19875f39d397SChris Mason path->nodes[level] = left; 1988e66f709bSChris Mason path->slots[level + 1] -= 1; 1989e66f709bSChris Mason path->slots[level] = orig_slot; 1990925baeddSChris Mason btrfs_tree_unlock(mid); 19915f39d397SChris Mason free_extent_buffer(mid); 1992e66f709bSChris Mason } else { 1993e66f709bSChris Mason orig_slot -= 19945f39d397SChris Mason btrfs_header_nritems(left); 1995e66f709bSChris Mason path->slots[level] = orig_slot; 1996925baeddSChris Mason btrfs_tree_unlock(left); 19975f39d397SChris Mason free_extent_buffer(left); 1998e66f709bSChris Mason } 1999e66f709bSChris Mason return 0; 2000e66f709bSChris Mason } 2001925baeddSChris Mason btrfs_tree_unlock(left); 20025f39d397SChris Mason free_extent_buffer(left); 2003e66f709bSChris Mason } 20045f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 2005e66f709bSChris Mason 2006e66f709bSChris Mason /* 2007e66f709bSChris Mason * then try to empty the right most buffer into the middle 2008e66f709bSChris Mason */ 20095f39d397SChris Mason if (right) { 201033ade1f8SChris Mason u32 right_nr; 2011b4ce94deSChris Mason 2012925baeddSChris Mason btrfs_tree_lock(right); 2013b4ce94deSChris Mason btrfs_set_lock_blocking(right); 2014b4ce94deSChris Mason 20155f39d397SChris Mason right_nr = btrfs_header_nritems(right); 201633ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 201733ade1f8SChris Mason wret = 1; 201833ade1f8SChris Mason } else { 20195f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 20205f39d397SChris Mason parent, pslot + 1, 20219fa8cfe7SChris Mason &right); 202254aa1f4dSChris Mason if (ret) 202354aa1f4dSChris Mason wret = 1; 202454aa1f4dSChris Mason else { 202533ade1f8SChris Mason wret = balance_node_right(trans, root, 20265f39d397SChris Mason right, mid); 202733ade1f8SChris Mason } 202854aa1f4dSChris Mason } 2029e66f709bSChris Mason if (wret < 0) 2030e66f709bSChris Mason ret = wret; 2031e66f709bSChris Mason if (wret == 0) { 20325f39d397SChris Mason struct btrfs_disk_key disk_key; 20335f39d397SChris Mason 20345f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 2035f230475eSJan Schmidt tree_mod_log_set_node_key(root->fs_info, parent, 2036f230475eSJan Schmidt &disk_key, pslot + 1, 0); 20375f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 20385f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 20395f39d397SChris Mason 20405f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 20415f39d397SChris Mason path->nodes[level] = right; 2042e66f709bSChris Mason path->slots[level + 1] += 1; 2043e66f709bSChris Mason path->slots[level] = orig_slot - 20445f39d397SChris Mason btrfs_header_nritems(mid); 2045925baeddSChris Mason btrfs_tree_unlock(mid); 20465f39d397SChris Mason free_extent_buffer(mid); 2047e66f709bSChris Mason } else { 2048925baeddSChris Mason btrfs_tree_unlock(right); 20495f39d397SChris Mason free_extent_buffer(right); 2050e66f709bSChris Mason } 2051e66f709bSChris Mason return 0; 2052e66f709bSChris Mason } 2053925baeddSChris Mason btrfs_tree_unlock(right); 20545f39d397SChris Mason free_extent_buffer(right); 2055e66f709bSChris Mason } 2056e66f709bSChris Mason return 1; 2057e66f709bSChris Mason } 2058e66f709bSChris Mason 205974123bd7SChris Mason /* 2060d352ac68SChris Mason * readahead one full node of leaves, finding things that are close 2061d352ac68SChris Mason * to the block in 'slot', and triggering ra on them. 20623c69faecSChris Mason */ 2063c8c42864SChris Mason static void reada_for_search(struct btrfs_root *root, 2064e02119d5SChris Mason struct btrfs_path *path, 206501f46658SChris Mason int level, int slot, u64 objectid) 20663c69faecSChris Mason { 20675f39d397SChris Mason struct extent_buffer *node; 206801f46658SChris Mason struct btrfs_disk_key disk_key; 20693c69faecSChris Mason u32 nritems; 20703c69faecSChris Mason u64 search; 2071a7175319SChris Mason u64 target; 20726b80053dSChris Mason u64 nread = 0; 2073cb25c2eaSJosef Bacik u64 gen; 20743c69faecSChris Mason int direction = path->reada; 20755f39d397SChris Mason struct extent_buffer *eb; 20766b80053dSChris Mason u32 nr; 20776b80053dSChris Mason u32 blocksize; 20786b80053dSChris Mason u32 nscan = 0; 2079db94535dSChris Mason 2080a6b6e75eSChris Mason if (level != 1) 20813c69faecSChris Mason return; 20823c69faecSChris Mason 20836702ed49SChris Mason if (!path->nodes[level]) 20846702ed49SChris Mason return; 20856702ed49SChris Mason 20865f39d397SChris Mason node = path->nodes[level]; 2087925baeddSChris Mason 20883c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 20896b80053dSChris Mason blocksize = btrfs_level_size(root, level - 1); 20906b80053dSChris Mason eb = btrfs_find_tree_block(root, search, blocksize); 20915f39d397SChris Mason if (eb) { 20925f39d397SChris Mason free_extent_buffer(eb); 20933c69faecSChris Mason return; 20943c69faecSChris Mason } 20953c69faecSChris Mason 2096a7175319SChris Mason target = search; 20976b80053dSChris Mason 20985f39d397SChris Mason nritems = btrfs_header_nritems(node); 20996b80053dSChris Mason nr = slot; 210025b8b936SJosef Bacik 21013c69faecSChris Mason while (1) { 21026b80053dSChris Mason if (direction < 0) { 21036b80053dSChris Mason if (nr == 0) 21043c69faecSChris Mason break; 21056b80053dSChris Mason nr--; 21066b80053dSChris Mason } else if (direction > 0) { 21076b80053dSChris Mason nr++; 21086b80053dSChris Mason if (nr >= nritems) 21096b80053dSChris Mason break; 21103c69faecSChris Mason } 211101f46658SChris Mason if (path->reada < 0 && objectid) { 211201f46658SChris Mason btrfs_node_key(node, &disk_key, nr); 211301f46658SChris Mason if (btrfs_disk_key_objectid(&disk_key) != objectid) 211401f46658SChris Mason break; 211501f46658SChris Mason } 21166b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 2117a7175319SChris Mason if ((search <= target && target - search <= 65536) || 2118a7175319SChris Mason (search > target && search - target <= 65536)) { 2119cb25c2eaSJosef Bacik gen = btrfs_node_ptr_generation(node, nr); 2120cb25c2eaSJosef Bacik readahead_tree_block(root, search, blocksize, gen); 21216b80053dSChris Mason nread += blocksize; 21223c69faecSChris Mason } 21236b80053dSChris Mason nscan++; 2124a7175319SChris Mason if ((nread > 65536 || nscan > 32)) 21256b80053dSChris Mason break; 21263c69faecSChris Mason } 21273c69faecSChris Mason } 2128925baeddSChris Mason 2129d352ac68SChris Mason /* 2130b4ce94deSChris Mason * returns -EAGAIN if it had to drop the path, or zero if everything was in 2131b4ce94deSChris Mason * cache 2132b4ce94deSChris Mason */ 2133b4ce94deSChris Mason static noinline int reada_for_balance(struct btrfs_root *root, 2134b4ce94deSChris Mason struct btrfs_path *path, int level) 2135b4ce94deSChris Mason { 2136b4ce94deSChris Mason int slot; 2137b4ce94deSChris Mason int nritems; 2138b4ce94deSChris Mason struct extent_buffer *parent; 2139b4ce94deSChris Mason struct extent_buffer *eb; 2140b4ce94deSChris Mason u64 gen; 2141b4ce94deSChris Mason u64 block1 = 0; 2142b4ce94deSChris Mason u64 block2 = 0; 2143b4ce94deSChris Mason int ret = 0; 2144b4ce94deSChris Mason int blocksize; 2145b4ce94deSChris Mason 21468c594ea8SChris Mason parent = path->nodes[level + 1]; 2147b4ce94deSChris Mason if (!parent) 2148b4ce94deSChris Mason return 0; 2149b4ce94deSChris Mason 2150b4ce94deSChris Mason nritems = btrfs_header_nritems(parent); 21518c594ea8SChris Mason slot = path->slots[level + 1]; 2152b4ce94deSChris Mason blocksize = btrfs_level_size(root, level); 2153b4ce94deSChris Mason 2154b4ce94deSChris Mason if (slot > 0) { 2155b4ce94deSChris Mason block1 = btrfs_node_blockptr(parent, slot - 1); 2156b4ce94deSChris Mason gen = btrfs_node_ptr_generation(parent, slot - 1); 2157b4ce94deSChris Mason eb = btrfs_find_tree_block(root, block1, blocksize); 2158b9fab919SChris Mason /* 2159b9fab919SChris Mason * if we get -eagain from btrfs_buffer_uptodate, we 2160b9fab919SChris Mason * don't want to return eagain here. That will loop 2161b9fab919SChris Mason * forever 2162b9fab919SChris Mason */ 2163b9fab919SChris Mason if (eb && btrfs_buffer_uptodate(eb, gen, 1) != 0) 2164b4ce94deSChris Mason block1 = 0; 2165b4ce94deSChris Mason free_extent_buffer(eb); 2166b4ce94deSChris Mason } 21678c594ea8SChris Mason if (slot + 1 < nritems) { 2168b4ce94deSChris Mason block2 = btrfs_node_blockptr(parent, slot + 1); 2169b4ce94deSChris Mason gen = btrfs_node_ptr_generation(parent, slot + 1); 2170b4ce94deSChris Mason eb = btrfs_find_tree_block(root, block2, blocksize); 2171b9fab919SChris Mason if (eb && btrfs_buffer_uptodate(eb, gen, 1) != 0) 2172b4ce94deSChris Mason block2 = 0; 2173b4ce94deSChris Mason free_extent_buffer(eb); 2174b4ce94deSChris Mason } 2175b4ce94deSChris Mason if (block1 || block2) { 2176b4ce94deSChris Mason ret = -EAGAIN; 21778c594ea8SChris Mason 21788c594ea8SChris Mason /* release the whole path */ 2179b3b4aa74SDavid Sterba btrfs_release_path(path); 21808c594ea8SChris Mason 21818c594ea8SChris Mason /* read the blocks */ 2182b4ce94deSChris Mason if (block1) 2183b4ce94deSChris Mason readahead_tree_block(root, block1, blocksize, 0); 2184b4ce94deSChris Mason if (block2) 2185b4ce94deSChris Mason readahead_tree_block(root, block2, blocksize, 0); 2186b4ce94deSChris Mason 2187b4ce94deSChris Mason if (block1) { 2188b4ce94deSChris Mason eb = read_tree_block(root, block1, blocksize, 0); 2189b4ce94deSChris Mason free_extent_buffer(eb); 2190b4ce94deSChris Mason } 21918c594ea8SChris Mason if (block2) { 2192b4ce94deSChris Mason eb = read_tree_block(root, block2, blocksize, 0); 2193b4ce94deSChris Mason free_extent_buffer(eb); 2194b4ce94deSChris Mason } 2195b4ce94deSChris Mason } 2196b4ce94deSChris Mason return ret; 2197b4ce94deSChris Mason } 2198b4ce94deSChris Mason 2199b4ce94deSChris Mason 2200b4ce94deSChris Mason /* 2201d397712bSChris Mason * when we walk down the tree, it is usually safe to unlock the higher layers 2202d397712bSChris Mason * in the tree. The exceptions are when our path goes through slot 0, because 2203d397712bSChris Mason * operations on the tree might require changing key pointers higher up in the 2204d397712bSChris Mason * tree. 2205d352ac68SChris Mason * 2206d397712bSChris Mason * callers might also have set path->keep_locks, which tells this code to keep 2207d397712bSChris Mason * the lock if the path points to the last slot in the block. This is part of 2208d397712bSChris Mason * walking through the tree, and selecting the next slot in the higher block. 2209d352ac68SChris Mason * 2210d397712bSChris Mason * lowest_unlock sets the lowest level in the tree we're allowed to unlock. so 2211d397712bSChris Mason * if lowest_unlock is 1, level 0 won't be unlocked 2212d352ac68SChris Mason */ 2213e02119d5SChris Mason static noinline void unlock_up(struct btrfs_path *path, int level, 2214f7c79f30SChris Mason int lowest_unlock, int min_write_lock_level, 2215f7c79f30SChris Mason int *write_lock_level) 2216925baeddSChris Mason { 2217925baeddSChris Mason int i; 2218925baeddSChris Mason int skip_level = level; 2219051e1b9fSChris Mason int no_skips = 0; 2220925baeddSChris Mason struct extent_buffer *t; 2221925baeddSChris Mason 2222925baeddSChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 2223925baeddSChris Mason if (!path->nodes[i]) 2224925baeddSChris Mason break; 2225925baeddSChris Mason if (!path->locks[i]) 2226925baeddSChris Mason break; 2227051e1b9fSChris Mason if (!no_skips && path->slots[i] == 0) { 2228925baeddSChris Mason skip_level = i + 1; 2229925baeddSChris Mason continue; 2230925baeddSChris Mason } 2231051e1b9fSChris Mason if (!no_skips && path->keep_locks) { 2232925baeddSChris Mason u32 nritems; 2233925baeddSChris Mason t = path->nodes[i]; 2234925baeddSChris Mason nritems = btrfs_header_nritems(t); 2235051e1b9fSChris Mason if (nritems < 1 || path->slots[i] >= nritems - 1) { 2236925baeddSChris Mason skip_level = i + 1; 2237925baeddSChris Mason continue; 2238925baeddSChris Mason } 2239925baeddSChris Mason } 2240051e1b9fSChris Mason if (skip_level < i && i >= lowest_unlock) 2241051e1b9fSChris Mason no_skips = 1; 2242051e1b9fSChris Mason 2243925baeddSChris Mason t = path->nodes[i]; 2244925baeddSChris Mason if (i >= lowest_unlock && i > skip_level && path->locks[i]) { 2245bd681513SChris Mason btrfs_tree_unlock_rw(t, path->locks[i]); 2246925baeddSChris Mason path->locks[i] = 0; 2247f7c79f30SChris Mason if (write_lock_level && 2248f7c79f30SChris Mason i > min_write_lock_level && 2249f7c79f30SChris Mason i <= *write_lock_level) { 2250f7c79f30SChris Mason *write_lock_level = i - 1; 2251f7c79f30SChris Mason } 2252925baeddSChris Mason } 2253925baeddSChris Mason } 2254925baeddSChris Mason } 2255925baeddSChris Mason 22563c69faecSChris Mason /* 2257b4ce94deSChris Mason * This releases any locks held in the path starting at level and 2258b4ce94deSChris Mason * going all the way up to the root. 2259b4ce94deSChris Mason * 2260b4ce94deSChris Mason * btrfs_search_slot will keep the lock held on higher nodes in a few 2261b4ce94deSChris Mason * corner cases, such as COW of the block at slot zero in the node. This 2262b4ce94deSChris Mason * ignores those rules, and it should only be called when there are no 2263b4ce94deSChris Mason * more updates to be done higher up in the tree. 2264b4ce94deSChris Mason */ 2265b4ce94deSChris Mason noinline void btrfs_unlock_up_safe(struct btrfs_path *path, int level) 2266b4ce94deSChris Mason { 2267b4ce94deSChris Mason int i; 2268b4ce94deSChris Mason 22695d4f98a2SYan Zheng if (path->keep_locks) 2270b4ce94deSChris Mason return; 2271b4ce94deSChris Mason 2272b4ce94deSChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 2273b4ce94deSChris Mason if (!path->nodes[i]) 227412f4daccSChris Mason continue; 2275b4ce94deSChris Mason if (!path->locks[i]) 227612f4daccSChris Mason continue; 2277bd681513SChris Mason btrfs_tree_unlock_rw(path->nodes[i], path->locks[i]); 2278b4ce94deSChris Mason path->locks[i] = 0; 2279b4ce94deSChris Mason } 2280b4ce94deSChris Mason } 2281b4ce94deSChris Mason 2282b4ce94deSChris Mason /* 2283c8c42864SChris Mason * helper function for btrfs_search_slot. The goal is to find a block 2284c8c42864SChris Mason * in cache without setting the path to blocking. If we find the block 2285c8c42864SChris Mason * we return zero and the path is unchanged. 2286c8c42864SChris Mason * 2287c8c42864SChris Mason * If we can't find the block, we set the path blocking and do some 2288c8c42864SChris Mason * reada. -EAGAIN is returned and the search must be repeated. 2289c8c42864SChris Mason */ 2290c8c42864SChris Mason static int 2291c8c42864SChris Mason read_block_for_search(struct btrfs_trans_handle *trans, 2292c8c42864SChris Mason struct btrfs_root *root, struct btrfs_path *p, 2293c8c42864SChris Mason struct extent_buffer **eb_ret, int level, int slot, 22945d9e75c4SJan Schmidt struct btrfs_key *key, u64 time_seq) 2295c8c42864SChris Mason { 2296c8c42864SChris Mason u64 blocknr; 2297c8c42864SChris Mason u64 gen; 2298c8c42864SChris Mason u32 blocksize; 2299c8c42864SChris Mason struct extent_buffer *b = *eb_ret; 2300c8c42864SChris Mason struct extent_buffer *tmp; 230176a05b35SChris Mason int ret; 2302c8c42864SChris Mason 2303c8c42864SChris Mason blocknr = btrfs_node_blockptr(b, slot); 2304c8c42864SChris Mason gen = btrfs_node_ptr_generation(b, slot); 2305c8c42864SChris Mason blocksize = btrfs_level_size(root, level - 1); 2306c8c42864SChris Mason 2307c8c42864SChris Mason tmp = btrfs_find_tree_block(root, blocknr, blocksize); 2308cb44921aSChris Mason if (tmp) { 2309b9fab919SChris Mason /* first we do an atomic uptodate check */ 2310b9fab919SChris Mason if (btrfs_buffer_uptodate(tmp, 0, 1) > 0) { 2311b9fab919SChris Mason if (btrfs_buffer_uptodate(tmp, gen, 1) > 0) { 231276a05b35SChris Mason /* 2313cb44921aSChris Mason * we found an up to date block without 2314cb44921aSChris Mason * sleeping, return 231576a05b35SChris Mason * right away 231676a05b35SChris Mason */ 2317c8c42864SChris Mason *eb_ret = tmp; 2318c8c42864SChris Mason return 0; 2319c8c42864SChris Mason } 2320cb44921aSChris Mason /* the pages were up to date, but we failed 2321cb44921aSChris Mason * the generation number check. Do a full 2322cb44921aSChris Mason * read for the generation number that is correct. 2323cb44921aSChris Mason * We must do this without dropping locks so 2324cb44921aSChris Mason * we can trust our generation number 2325cb44921aSChris Mason */ 2326cb44921aSChris Mason free_extent_buffer(tmp); 2327bd681513SChris Mason btrfs_set_path_blocking(p); 2328bd681513SChris Mason 2329b9fab919SChris Mason /* now we're allowed to do a blocking uptodate check */ 2330cb44921aSChris Mason tmp = read_tree_block(root, blocknr, blocksize, gen); 2331b9fab919SChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen, 0) > 0) { 2332cb44921aSChris Mason *eb_ret = tmp; 2333cb44921aSChris Mason return 0; 2334cb44921aSChris Mason } 2335cb44921aSChris Mason free_extent_buffer(tmp); 2336b3b4aa74SDavid Sterba btrfs_release_path(p); 2337cb44921aSChris Mason return -EIO; 2338cb44921aSChris Mason } 2339cb44921aSChris Mason } 2340c8c42864SChris Mason 2341c8c42864SChris Mason /* 2342c8c42864SChris Mason * reduce lock contention at high levels 2343c8c42864SChris Mason * of the btree by dropping locks before 234476a05b35SChris Mason * we read. Don't release the lock on the current 234576a05b35SChris Mason * level because we need to walk this node to figure 234676a05b35SChris Mason * out which blocks to read. 2347c8c42864SChris Mason */ 23488c594ea8SChris Mason btrfs_unlock_up_safe(p, level + 1); 23498c594ea8SChris Mason btrfs_set_path_blocking(p); 23508c594ea8SChris Mason 2351c8c42864SChris Mason free_extent_buffer(tmp); 2352c8c42864SChris Mason if (p->reada) 2353c8c42864SChris Mason reada_for_search(root, p, level, slot, key->objectid); 2354c8c42864SChris Mason 2355b3b4aa74SDavid Sterba btrfs_release_path(p); 235676a05b35SChris Mason 235776a05b35SChris Mason ret = -EAGAIN; 23585bdd3536SYan, Zheng tmp = read_tree_block(root, blocknr, blocksize, 0); 235976a05b35SChris Mason if (tmp) { 236076a05b35SChris Mason /* 236176a05b35SChris Mason * If the read above didn't mark this buffer up to date, 236276a05b35SChris Mason * it will never end up being up to date. Set ret to EIO now 236376a05b35SChris Mason * and give up so that our caller doesn't loop forever 236476a05b35SChris Mason * on our EAGAINs. 236576a05b35SChris Mason */ 2366b9fab919SChris Mason if (!btrfs_buffer_uptodate(tmp, 0, 0)) 236776a05b35SChris Mason ret = -EIO; 2368c8c42864SChris Mason free_extent_buffer(tmp); 236976a05b35SChris Mason } 237076a05b35SChris Mason return ret; 2371c8c42864SChris Mason } 2372c8c42864SChris Mason 2373c8c42864SChris Mason /* 2374c8c42864SChris Mason * helper function for btrfs_search_slot. This does all of the checks 2375c8c42864SChris Mason * for node-level blocks and does any balancing required based on 2376c8c42864SChris Mason * the ins_len. 2377c8c42864SChris Mason * 2378c8c42864SChris Mason * If no extra work was required, zero is returned. If we had to 2379c8c42864SChris Mason * drop the path, -EAGAIN is returned and btrfs_search_slot must 2380c8c42864SChris Mason * start over 2381c8c42864SChris Mason */ 2382c8c42864SChris Mason static int 2383c8c42864SChris Mason setup_nodes_for_search(struct btrfs_trans_handle *trans, 2384c8c42864SChris Mason struct btrfs_root *root, struct btrfs_path *p, 2385bd681513SChris Mason struct extent_buffer *b, int level, int ins_len, 2386bd681513SChris Mason int *write_lock_level) 2387c8c42864SChris Mason { 2388c8c42864SChris Mason int ret; 2389c8c42864SChris Mason if ((p->search_for_split || ins_len > 0) && btrfs_header_nritems(b) >= 2390c8c42864SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { 2391c8c42864SChris Mason int sret; 2392c8c42864SChris Mason 2393bd681513SChris Mason if (*write_lock_level < level + 1) { 2394bd681513SChris Mason *write_lock_level = level + 1; 2395bd681513SChris Mason btrfs_release_path(p); 2396bd681513SChris Mason goto again; 2397bd681513SChris Mason } 2398bd681513SChris Mason 2399c8c42864SChris Mason sret = reada_for_balance(root, p, level); 2400c8c42864SChris Mason if (sret) 2401c8c42864SChris Mason goto again; 2402c8c42864SChris Mason 2403c8c42864SChris Mason btrfs_set_path_blocking(p); 2404c8c42864SChris Mason sret = split_node(trans, root, p, level); 2405bd681513SChris Mason btrfs_clear_path_blocking(p, NULL, 0); 2406c8c42864SChris Mason 2407c8c42864SChris Mason BUG_ON(sret > 0); 2408c8c42864SChris Mason if (sret) { 2409c8c42864SChris Mason ret = sret; 2410c8c42864SChris Mason goto done; 2411c8c42864SChris Mason } 2412c8c42864SChris Mason b = p->nodes[level]; 2413c8c42864SChris Mason } else if (ins_len < 0 && btrfs_header_nritems(b) < 2414cfbb9308SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 2) { 2415c8c42864SChris Mason int sret; 2416c8c42864SChris Mason 2417bd681513SChris Mason if (*write_lock_level < level + 1) { 2418bd681513SChris Mason *write_lock_level = level + 1; 2419bd681513SChris Mason btrfs_release_path(p); 2420bd681513SChris Mason goto again; 2421bd681513SChris Mason } 2422bd681513SChris Mason 2423c8c42864SChris Mason sret = reada_for_balance(root, p, level); 2424c8c42864SChris Mason if (sret) 2425c8c42864SChris Mason goto again; 2426c8c42864SChris Mason 2427c8c42864SChris Mason btrfs_set_path_blocking(p); 2428c8c42864SChris Mason sret = balance_level(trans, root, p, level); 2429bd681513SChris Mason btrfs_clear_path_blocking(p, NULL, 0); 2430c8c42864SChris Mason 2431c8c42864SChris Mason if (sret) { 2432c8c42864SChris Mason ret = sret; 2433c8c42864SChris Mason goto done; 2434c8c42864SChris Mason } 2435c8c42864SChris Mason b = p->nodes[level]; 2436c8c42864SChris Mason if (!b) { 2437b3b4aa74SDavid Sterba btrfs_release_path(p); 2438c8c42864SChris Mason goto again; 2439c8c42864SChris Mason } 2440c8c42864SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 2441c8c42864SChris Mason } 2442c8c42864SChris Mason return 0; 2443c8c42864SChris Mason 2444c8c42864SChris Mason again: 2445c8c42864SChris Mason ret = -EAGAIN; 2446c8c42864SChris Mason done: 2447c8c42864SChris Mason return ret; 2448c8c42864SChris Mason } 2449c8c42864SChris Mason 2450c8c42864SChris Mason /* 245174123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 245274123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 245374123bd7SChris Mason * level of the path (level 0) 245474123bd7SChris Mason * 245574123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 2456aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 2457aa5d6bedSChris Mason * search a negative error number is returned. 245897571fd0SChris Mason * 245997571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 246097571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 246197571fd0SChris Mason * possible) 246274123bd7SChris Mason */ 2463e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 2464e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 2465e089f05cSChris Mason ins_len, int cow) 2466be0e5c09SChris Mason { 24675f39d397SChris Mason struct extent_buffer *b; 2468be0e5c09SChris Mason int slot; 2469be0e5c09SChris Mason int ret; 247033c66f43SYan Zheng int err; 2471be0e5c09SChris Mason int level; 2472925baeddSChris Mason int lowest_unlock = 1; 2473bd681513SChris Mason int root_lock; 2474bd681513SChris Mason /* everything at write_lock_level or lower must be write locked */ 2475bd681513SChris Mason int write_lock_level = 0; 24769f3a7427SChris Mason u8 lowest_level = 0; 2477f7c79f30SChris Mason int min_write_lock_level; 24789f3a7427SChris Mason 24796702ed49SChris Mason lowest_level = p->lowest_level; 2480323ac95bSChris Mason WARN_ON(lowest_level && ins_len > 0); 248122b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 248225179201SJosef Bacik 2483bd681513SChris Mason if (ins_len < 0) { 2484925baeddSChris Mason lowest_unlock = 2; 248565b51a00SChris Mason 2486bd681513SChris Mason /* when we are removing items, we might have to go up to level 2487bd681513SChris Mason * two as we update tree pointers Make sure we keep write 2488bd681513SChris Mason * for those levels as well 2489bd681513SChris Mason */ 2490bd681513SChris Mason write_lock_level = 2; 2491bd681513SChris Mason } else if (ins_len > 0) { 2492bd681513SChris Mason /* 2493bd681513SChris Mason * for inserting items, make sure we have a write lock on 2494bd681513SChris Mason * level 1 so we can update keys 2495bd681513SChris Mason */ 2496bd681513SChris Mason write_lock_level = 1; 2497bd681513SChris Mason } 2498bd681513SChris Mason 2499bd681513SChris Mason if (!cow) 2500bd681513SChris Mason write_lock_level = -1; 2501bd681513SChris Mason 2502bd681513SChris Mason if (cow && (p->keep_locks || p->lowest_level)) 2503bd681513SChris Mason write_lock_level = BTRFS_MAX_LEVEL; 2504bd681513SChris Mason 2505f7c79f30SChris Mason min_write_lock_level = write_lock_level; 2506f7c79f30SChris Mason 2507bb803951SChris Mason again: 2508bd681513SChris Mason /* 2509bd681513SChris Mason * we try very hard to do read locks on the root 2510bd681513SChris Mason */ 2511bd681513SChris Mason root_lock = BTRFS_READ_LOCK; 2512bd681513SChris Mason level = 0; 25135d4f98a2SYan Zheng if (p->search_commit_root) { 2514bd681513SChris Mason /* 2515bd681513SChris Mason * the commit roots are read only 2516bd681513SChris Mason * so we always do read locks 2517bd681513SChris Mason */ 25185d4f98a2SYan Zheng b = root->commit_root; 25195d4f98a2SYan Zheng extent_buffer_get(b); 2520bd681513SChris Mason level = btrfs_header_level(b); 25215d4f98a2SYan Zheng if (!p->skip_locking) 2522bd681513SChris Mason btrfs_tree_read_lock(b); 25235d4f98a2SYan Zheng } else { 2524bd681513SChris Mason if (p->skip_locking) { 25255cd57b2cSChris Mason b = btrfs_root_node(root); 2526bd681513SChris Mason level = btrfs_header_level(b); 2527bd681513SChris Mason } else { 2528bd681513SChris Mason /* we don't know the level of the root node 2529bd681513SChris Mason * until we actually have it read locked 2530bd681513SChris Mason */ 2531bd681513SChris Mason b = btrfs_read_lock_root_node(root); 2532bd681513SChris Mason level = btrfs_header_level(b); 2533bd681513SChris Mason if (level <= write_lock_level) { 2534bd681513SChris Mason /* whoops, must trade for write lock */ 2535bd681513SChris Mason btrfs_tree_read_unlock(b); 2536bd681513SChris Mason free_extent_buffer(b); 2537925baeddSChris Mason b = btrfs_lock_root_node(root); 2538bd681513SChris Mason root_lock = BTRFS_WRITE_LOCK; 2539bd681513SChris Mason 2540bd681513SChris Mason /* the level might have changed, check again */ 2541bd681513SChris Mason level = btrfs_header_level(b); 25425d4f98a2SYan Zheng } 2543bd681513SChris Mason } 2544bd681513SChris Mason } 2545bd681513SChris Mason p->nodes[level] = b; 2546bd681513SChris Mason if (!p->skip_locking) 2547bd681513SChris Mason p->locks[level] = root_lock; 2548925baeddSChris Mason 2549eb60ceacSChris Mason while (b) { 25505f39d397SChris Mason level = btrfs_header_level(b); 255165b51a00SChris Mason 255265b51a00SChris Mason /* 255365b51a00SChris Mason * setup the path here so we can release it under lock 255465b51a00SChris Mason * contention with the cow code 255565b51a00SChris Mason */ 255602217ed2SChris Mason if (cow) { 2557c8c42864SChris Mason /* 2558c8c42864SChris Mason * if we don't really need to cow this block 2559c8c42864SChris Mason * then we don't want to set the path blocking, 2560c8c42864SChris Mason * so we test it here 2561c8c42864SChris Mason */ 25625d4f98a2SYan Zheng if (!should_cow_block(trans, root, b)) 256365b51a00SChris Mason goto cow_done; 25645d4f98a2SYan Zheng 2565b4ce94deSChris Mason btrfs_set_path_blocking(p); 2566b4ce94deSChris Mason 2567bd681513SChris Mason /* 2568bd681513SChris Mason * must have write locks on this node and the 2569bd681513SChris Mason * parent 2570bd681513SChris Mason */ 2571bd681513SChris Mason if (level + 1 > write_lock_level) { 2572bd681513SChris Mason write_lock_level = level + 1; 2573bd681513SChris Mason btrfs_release_path(p); 2574bd681513SChris Mason goto again; 2575bd681513SChris Mason } 2576bd681513SChris Mason 257733c66f43SYan Zheng err = btrfs_cow_block(trans, root, b, 2578e20d96d6SChris Mason p->nodes[level + 1], 25799fa8cfe7SChris Mason p->slots[level + 1], &b); 258033c66f43SYan Zheng if (err) { 258133c66f43SYan Zheng ret = err; 258265b51a00SChris Mason goto done; 258354aa1f4dSChris Mason } 258402217ed2SChris Mason } 258565b51a00SChris Mason cow_done: 258602217ed2SChris Mason BUG_ON(!cow && ins_len); 258765b51a00SChris Mason 2588eb60ceacSChris Mason p->nodes[level] = b; 2589bd681513SChris Mason btrfs_clear_path_blocking(p, NULL, 0); 2590b4ce94deSChris Mason 2591b4ce94deSChris Mason /* 2592b4ce94deSChris Mason * we have a lock on b and as long as we aren't changing 2593b4ce94deSChris Mason * the tree, there is no way to for the items in b to change. 2594b4ce94deSChris Mason * It is safe to drop the lock on our parent before we 2595b4ce94deSChris Mason * go through the expensive btree search on b. 2596b4ce94deSChris Mason * 2597b4ce94deSChris Mason * If cow is true, then we might be changing slot zero, 2598b4ce94deSChris Mason * which may require changing the parent. So, we can't 2599b4ce94deSChris Mason * drop the lock until after we know which slot we're 2600b4ce94deSChris Mason * operating on. 2601b4ce94deSChris Mason */ 2602b4ce94deSChris Mason if (!cow) 2603b4ce94deSChris Mason btrfs_unlock_up_safe(p, level + 1); 2604b4ce94deSChris Mason 26055f39d397SChris Mason ret = bin_search(b, key, level, &slot); 2606b4ce94deSChris Mason 26075f39d397SChris Mason if (level != 0) { 260833c66f43SYan Zheng int dec = 0; 260933c66f43SYan Zheng if (ret && slot > 0) { 261033c66f43SYan Zheng dec = 1; 2611be0e5c09SChris Mason slot -= 1; 261233c66f43SYan Zheng } 2613be0e5c09SChris Mason p->slots[level] = slot; 261433c66f43SYan Zheng err = setup_nodes_for_search(trans, root, p, b, level, 2615bd681513SChris Mason ins_len, &write_lock_level); 261633c66f43SYan Zheng if (err == -EAGAIN) 2617b4ce94deSChris Mason goto again; 261833c66f43SYan Zheng if (err) { 261933c66f43SYan Zheng ret = err; 262065b51a00SChris Mason goto done; 262133c66f43SYan Zheng } 26225c680ed6SChris Mason b = p->nodes[level]; 26235c680ed6SChris Mason slot = p->slots[level]; 2624b4ce94deSChris Mason 2625bd681513SChris Mason /* 2626bd681513SChris Mason * slot 0 is special, if we change the key 2627bd681513SChris Mason * we have to update the parent pointer 2628bd681513SChris Mason * which means we must have a write lock 2629bd681513SChris Mason * on the parent 2630bd681513SChris Mason */ 2631bd681513SChris Mason if (slot == 0 && cow && 2632bd681513SChris Mason write_lock_level < level + 1) { 2633bd681513SChris Mason write_lock_level = level + 1; 2634bd681513SChris Mason btrfs_release_path(p); 2635bd681513SChris Mason goto again; 2636bd681513SChris Mason } 2637bd681513SChris Mason 2638f7c79f30SChris Mason unlock_up(p, level, lowest_unlock, 2639f7c79f30SChris Mason min_write_lock_level, &write_lock_level); 2640f9efa9c7SChris Mason 2641925baeddSChris Mason if (level == lowest_level) { 264233c66f43SYan Zheng if (dec) 264333c66f43SYan Zheng p->slots[level]++; 26445b21f2edSZheng Yan goto done; 2645925baeddSChris Mason } 2646ca7a79adSChris Mason 264733c66f43SYan Zheng err = read_block_for_search(trans, root, p, 26485d9e75c4SJan Schmidt &b, level, slot, key, 0); 264933c66f43SYan Zheng if (err == -EAGAIN) 2650051e1b9fSChris Mason goto again; 265133c66f43SYan Zheng if (err) { 265233c66f43SYan Zheng ret = err; 265376a05b35SChris Mason goto done; 265433c66f43SYan Zheng } 265576a05b35SChris Mason 2656b4ce94deSChris Mason if (!p->skip_locking) { 2657bd681513SChris Mason level = btrfs_header_level(b); 2658bd681513SChris Mason if (level <= write_lock_level) { 2659bd681513SChris Mason err = btrfs_try_tree_write_lock(b); 266033c66f43SYan Zheng if (!err) { 2661b4ce94deSChris Mason btrfs_set_path_blocking(p); 2662925baeddSChris Mason btrfs_tree_lock(b); 2663bd681513SChris Mason btrfs_clear_path_blocking(p, b, 2664bd681513SChris Mason BTRFS_WRITE_LOCK); 2665b4ce94deSChris Mason } 2666bd681513SChris Mason p->locks[level] = BTRFS_WRITE_LOCK; 2667bd681513SChris Mason } else { 2668bd681513SChris Mason err = btrfs_try_tree_read_lock(b); 2669bd681513SChris Mason if (!err) { 2670bd681513SChris Mason btrfs_set_path_blocking(p); 2671bd681513SChris Mason btrfs_tree_read_lock(b); 2672bd681513SChris Mason btrfs_clear_path_blocking(p, b, 2673bd681513SChris Mason BTRFS_READ_LOCK); 2674bd681513SChris Mason } 2675bd681513SChris Mason p->locks[level] = BTRFS_READ_LOCK; 2676bd681513SChris Mason } 2677bd681513SChris Mason p->nodes[level] = b; 2678b4ce94deSChris Mason } 2679be0e5c09SChris Mason } else { 2680be0e5c09SChris Mason p->slots[level] = slot; 268187b29b20SYan Zheng if (ins_len > 0 && 268287b29b20SYan Zheng btrfs_leaf_free_space(root, b) < ins_len) { 2683bd681513SChris Mason if (write_lock_level < 1) { 2684bd681513SChris Mason write_lock_level = 1; 2685bd681513SChris Mason btrfs_release_path(p); 2686bd681513SChris Mason goto again; 2687bd681513SChris Mason } 2688bd681513SChris Mason 2689b4ce94deSChris Mason btrfs_set_path_blocking(p); 269033c66f43SYan Zheng err = split_leaf(trans, root, key, 2691cc0c5538SChris Mason p, ins_len, ret == 0); 2692bd681513SChris Mason btrfs_clear_path_blocking(p, NULL, 0); 2693b4ce94deSChris Mason 269433c66f43SYan Zheng BUG_ON(err > 0); 269533c66f43SYan Zheng if (err) { 269633c66f43SYan Zheng ret = err; 269765b51a00SChris Mason goto done; 269865b51a00SChris Mason } 26995c680ed6SChris Mason } 2700459931ecSChris Mason if (!p->search_for_split) 2701f7c79f30SChris Mason unlock_up(p, level, lowest_unlock, 2702f7c79f30SChris Mason min_write_lock_level, &write_lock_level); 270365b51a00SChris Mason goto done; 270465b51a00SChris Mason } 270565b51a00SChris Mason } 270665b51a00SChris Mason ret = 1; 270765b51a00SChris Mason done: 2708b4ce94deSChris Mason /* 2709b4ce94deSChris Mason * we don't really know what they plan on doing with the path 2710b4ce94deSChris Mason * from here on, so for now just mark it as blocking 2711b4ce94deSChris Mason */ 2712b9473439SChris Mason if (!p->leave_spinning) 2713b4ce94deSChris Mason btrfs_set_path_blocking(p); 271476a05b35SChris Mason if (ret < 0) 2715b3b4aa74SDavid Sterba btrfs_release_path(p); 2716be0e5c09SChris Mason return ret; 2717be0e5c09SChris Mason } 2718be0e5c09SChris Mason 271974123bd7SChris Mason /* 27205d9e75c4SJan Schmidt * Like btrfs_search_slot, this looks for a key in the given tree. It uses the 27215d9e75c4SJan Schmidt * current state of the tree together with the operations recorded in the tree 27225d9e75c4SJan Schmidt * modification log to search for the key in a previous version of this tree, as 27235d9e75c4SJan Schmidt * denoted by the time_seq parameter. 27245d9e75c4SJan Schmidt * 27255d9e75c4SJan Schmidt * Naturally, there is no support for insert, delete or cow operations. 27265d9e75c4SJan Schmidt * 27275d9e75c4SJan Schmidt * The resulting path and return value will be set up as if we called 27285d9e75c4SJan Schmidt * btrfs_search_slot at that point in time with ins_len and cow both set to 0. 27295d9e75c4SJan Schmidt */ 27305d9e75c4SJan Schmidt int btrfs_search_old_slot(struct btrfs_root *root, struct btrfs_key *key, 27315d9e75c4SJan Schmidt struct btrfs_path *p, u64 time_seq) 27325d9e75c4SJan Schmidt { 27335d9e75c4SJan Schmidt struct extent_buffer *b; 27345d9e75c4SJan Schmidt int slot; 27355d9e75c4SJan Schmidt int ret; 27365d9e75c4SJan Schmidt int err; 27375d9e75c4SJan Schmidt int level; 27385d9e75c4SJan Schmidt int lowest_unlock = 1; 27395d9e75c4SJan Schmidt u8 lowest_level = 0; 27405d9e75c4SJan Schmidt 27415d9e75c4SJan Schmidt lowest_level = p->lowest_level; 27425d9e75c4SJan Schmidt WARN_ON(p->nodes[0] != NULL); 27435d9e75c4SJan Schmidt 27445d9e75c4SJan Schmidt if (p->search_commit_root) { 27455d9e75c4SJan Schmidt BUG_ON(time_seq); 27465d9e75c4SJan Schmidt return btrfs_search_slot(NULL, root, key, p, 0, 0); 27475d9e75c4SJan Schmidt } 27485d9e75c4SJan Schmidt 27495d9e75c4SJan Schmidt again: 27505d9e75c4SJan Schmidt b = get_old_root(root, time_seq); 27515d9e75c4SJan Schmidt level = btrfs_header_level(b); 27525d9e75c4SJan Schmidt p->locks[level] = BTRFS_READ_LOCK; 27535d9e75c4SJan Schmidt 27545d9e75c4SJan Schmidt while (b) { 27555d9e75c4SJan Schmidt level = btrfs_header_level(b); 27565d9e75c4SJan Schmidt p->nodes[level] = b; 27575d9e75c4SJan Schmidt btrfs_clear_path_blocking(p, NULL, 0); 27585d9e75c4SJan Schmidt 27595d9e75c4SJan Schmidt /* 27605d9e75c4SJan Schmidt * we have a lock on b and as long as we aren't changing 27615d9e75c4SJan Schmidt * the tree, there is no way to for the items in b to change. 27625d9e75c4SJan Schmidt * It is safe to drop the lock on our parent before we 27635d9e75c4SJan Schmidt * go through the expensive btree search on b. 27645d9e75c4SJan Schmidt */ 27655d9e75c4SJan Schmidt btrfs_unlock_up_safe(p, level + 1); 27665d9e75c4SJan Schmidt 27675d9e75c4SJan Schmidt ret = bin_search(b, key, level, &slot); 27685d9e75c4SJan Schmidt 27695d9e75c4SJan Schmidt if (level != 0) { 27705d9e75c4SJan Schmidt int dec = 0; 27715d9e75c4SJan Schmidt if (ret && slot > 0) { 27725d9e75c4SJan Schmidt dec = 1; 27735d9e75c4SJan Schmidt slot -= 1; 27745d9e75c4SJan Schmidt } 27755d9e75c4SJan Schmidt p->slots[level] = slot; 27765d9e75c4SJan Schmidt unlock_up(p, level, lowest_unlock, 0, NULL); 27775d9e75c4SJan Schmidt 27785d9e75c4SJan Schmidt if (level == lowest_level) { 27795d9e75c4SJan Schmidt if (dec) 27805d9e75c4SJan Schmidt p->slots[level]++; 27815d9e75c4SJan Schmidt goto done; 27825d9e75c4SJan Schmidt } 27835d9e75c4SJan Schmidt 27845d9e75c4SJan Schmidt err = read_block_for_search(NULL, root, p, &b, level, 27855d9e75c4SJan Schmidt slot, key, time_seq); 27865d9e75c4SJan Schmidt if (err == -EAGAIN) 27875d9e75c4SJan Schmidt goto again; 27885d9e75c4SJan Schmidt if (err) { 27895d9e75c4SJan Schmidt ret = err; 27905d9e75c4SJan Schmidt goto done; 27915d9e75c4SJan Schmidt } 27925d9e75c4SJan Schmidt 27935d9e75c4SJan Schmidt level = btrfs_header_level(b); 27945d9e75c4SJan Schmidt err = btrfs_try_tree_read_lock(b); 27955d9e75c4SJan Schmidt if (!err) { 27965d9e75c4SJan Schmidt btrfs_set_path_blocking(p); 27975d9e75c4SJan Schmidt btrfs_tree_read_lock(b); 27985d9e75c4SJan Schmidt btrfs_clear_path_blocking(p, b, 27995d9e75c4SJan Schmidt BTRFS_READ_LOCK); 28005d9e75c4SJan Schmidt } 28015d9e75c4SJan Schmidt p->locks[level] = BTRFS_READ_LOCK; 28025d9e75c4SJan Schmidt p->nodes[level] = b; 28035d9e75c4SJan Schmidt b = tree_mod_log_rewind(root->fs_info, b, time_seq); 28045d9e75c4SJan Schmidt if (b != p->nodes[level]) { 28055d9e75c4SJan Schmidt btrfs_tree_unlock_rw(p->nodes[level], 28065d9e75c4SJan Schmidt p->locks[level]); 28075d9e75c4SJan Schmidt p->locks[level] = 0; 28085d9e75c4SJan Schmidt p->nodes[level] = b; 28095d9e75c4SJan Schmidt } 28105d9e75c4SJan Schmidt } else { 28115d9e75c4SJan Schmidt p->slots[level] = slot; 28125d9e75c4SJan Schmidt unlock_up(p, level, lowest_unlock, 0, NULL); 28135d9e75c4SJan Schmidt goto done; 28145d9e75c4SJan Schmidt } 28155d9e75c4SJan Schmidt } 28165d9e75c4SJan Schmidt ret = 1; 28175d9e75c4SJan Schmidt done: 28185d9e75c4SJan Schmidt if (!p->leave_spinning) 28195d9e75c4SJan Schmidt btrfs_set_path_blocking(p); 28205d9e75c4SJan Schmidt if (ret < 0) 28215d9e75c4SJan Schmidt btrfs_release_path(p); 28225d9e75c4SJan Schmidt 28235d9e75c4SJan Schmidt return ret; 28245d9e75c4SJan Schmidt } 28255d9e75c4SJan Schmidt 28265d9e75c4SJan Schmidt /* 28272f38b3e1SArne Jansen * helper to use instead of search slot if no exact match is needed but 28282f38b3e1SArne Jansen * instead the next or previous item should be returned. 28292f38b3e1SArne Jansen * When find_higher is true, the next higher item is returned, the next lower 28302f38b3e1SArne Jansen * otherwise. 28312f38b3e1SArne Jansen * When return_any and find_higher are both true, and no higher item is found, 28322f38b3e1SArne Jansen * return the next lower instead. 28332f38b3e1SArne Jansen * When return_any is true and find_higher is false, and no lower item is found, 28342f38b3e1SArne Jansen * return the next higher instead. 28352f38b3e1SArne Jansen * It returns 0 if any item is found, 1 if none is found (tree empty), and 28362f38b3e1SArne Jansen * < 0 on error 28372f38b3e1SArne Jansen */ 28382f38b3e1SArne Jansen int btrfs_search_slot_for_read(struct btrfs_root *root, 28392f38b3e1SArne Jansen struct btrfs_key *key, struct btrfs_path *p, 28402f38b3e1SArne Jansen int find_higher, int return_any) 28412f38b3e1SArne Jansen { 28422f38b3e1SArne Jansen int ret; 28432f38b3e1SArne Jansen struct extent_buffer *leaf; 28442f38b3e1SArne Jansen 28452f38b3e1SArne Jansen again: 28462f38b3e1SArne Jansen ret = btrfs_search_slot(NULL, root, key, p, 0, 0); 28472f38b3e1SArne Jansen if (ret <= 0) 28482f38b3e1SArne Jansen return ret; 28492f38b3e1SArne Jansen /* 28502f38b3e1SArne Jansen * a return value of 1 means the path is at the position where the 28512f38b3e1SArne Jansen * item should be inserted. Normally this is the next bigger item, 28522f38b3e1SArne Jansen * but in case the previous item is the last in a leaf, path points 28532f38b3e1SArne Jansen * to the first free slot in the previous leaf, i.e. at an invalid 28542f38b3e1SArne Jansen * item. 28552f38b3e1SArne Jansen */ 28562f38b3e1SArne Jansen leaf = p->nodes[0]; 28572f38b3e1SArne Jansen 28582f38b3e1SArne Jansen if (find_higher) { 28592f38b3e1SArne Jansen if (p->slots[0] >= btrfs_header_nritems(leaf)) { 28602f38b3e1SArne Jansen ret = btrfs_next_leaf(root, p); 28612f38b3e1SArne Jansen if (ret <= 0) 28622f38b3e1SArne Jansen return ret; 28632f38b3e1SArne Jansen if (!return_any) 28642f38b3e1SArne Jansen return 1; 28652f38b3e1SArne Jansen /* 28662f38b3e1SArne Jansen * no higher item found, return the next 28672f38b3e1SArne Jansen * lower instead 28682f38b3e1SArne Jansen */ 28692f38b3e1SArne Jansen return_any = 0; 28702f38b3e1SArne Jansen find_higher = 0; 28712f38b3e1SArne Jansen btrfs_release_path(p); 28722f38b3e1SArne Jansen goto again; 28732f38b3e1SArne Jansen } 28742f38b3e1SArne Jansen } else { 28752f38b3e1SArne Jansen if (p->slots[0] == 0) { 28762f38b3e1SArne Jansen ret = btrfs_prev_leaf(root, p); 2877e6793769SArne Jansen if (ret < 0) 28782f38b3e1SArne Jansen return ret; 2879e6793769SArne Jansen if (!ret) { 2880e6793769SArne Jansen p->slots[0] = btrfs_header_nritems(leaf) - 1; 2881e6793769SArne Jansen return 0; 2882e6793769SArne Jansen } 28832f38b3e1SArne Jansen if (!return_any) 28842f38b3e1SArne Jansen return 1; 28852f38b3e1SArne Jansen /* 28862f38b3e1SArne Jansen * no lower item found, return the next 28872f38b3e1SArne Jansen * higher instead 28882f38b3e1SArne Jansen */ 28892f38b3e1SArne Jansen return_any = 0; 28902f38b3e1SArne Jansen find_higher = 1; 28912f38b3e1SArne Jansen btrfs_release_path(p); 28922f38b3e1SArne Jansen goto again; 2893e6793769SArne Jansen } else { 28942f38b3e1SArne Jansen --p->slots[0]; 28952f38b3e1SArne Jansen } 28962f38b3e1SArne Jansen } 28972f38b3e1SArne Jansen return 0; 28982f38b3e1SArne Jansen } 28992f38b3e1SArne Jansen 29002f38b3e1SArne Jansen /* 290174123bd7SChris Mason * adjust the pointers going up the tree, starting at level 290274123bd7SChris Mason * making sure the right key of each node is points to 'key'. 290374123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 290474123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 290574123bd7SChris Mason * higher levels 2906aa5d6bedSChris Mason * 290774123bd7SChris Mason */ 2908143bede5SJeff Mahoney static void fixup_low_keys(struct btrfs_trans_handle *trans, 29095f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 29105f39d397SChris Mason struct btrfs_disk_key *key, int level) 2911be0e5c09SChris Mason { 2912be0e5c09SChris Mason int i; 29135f39d397SChris Mason struct extent_buffer *t; 29145f39d397SChris Mason 2915234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 2916be0e5c09SChris Mason int tslot = path->slots[i]; 2917eb60ceacSChris Mason if (!path->nodes[i]) 2918be0e5c09SChris Mason break; 29195f39d397SChris Mason t = path->nodes[i]; 2920f230475eSJan Schmidt tree_mod_log_set_node_key(root->fs_info, t, key, tslot, 1); 29215f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 2922d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 2923be0e5c09SChris Mason if (tslot != 0) 2924be0e5c09SChris Mason break; 2925be0e5c09SChris Mason } 2926be0e5c09SChris Mason } 2927be0e5c09SChris Mason 292874123bd7SChris Mason /* 292931840ae1SZheng Yan * update item key. 293031840ae1SZheng Yan * 293131840ae1SZheng Yan * This function isn't completely safe. It's the caller's responsibility 293231840ae1SZheng Yan * that the new key won't break the order 293331840ae1SZheng Yan */ 2934143bede5SJeff Mahoney void btrfs_set_item_key_safe(struct btrfs_trans_handle *trans, 293531840ae1SZheng Yan struct btrfs_root *root, struct btrfs_path *path, 293631840ae1SZheng Yan struct btrfs_key *new_key) 293731840ae1SZheng Yan { 293831840ae1SZheng Yan struct btrfs_disk_key disk_key; 293931840ae1SZheng Yan struct extent_buffer *eb; 294031840ae1SZheng Yan int slot; 294131840ae1SZheng Yan 294231840ae1SZheng Yan eb = path->nodes[0]; 294331840ae1SZheng Yan slot = path->slots[0]; 294431840ae1SZheng Yan if (slot > 0) { 294531840ae1SZheng Yan btrfs_item_key(eb, &disk_key, slot - 1); 2946143bede5SJeff Mahoney BUG_ON(comp_keys(&disk_key, new_key) >= 0); 294731840ae1SZheng Yan } 294831840ae1SZheng Yan if (slot < btrfs_header_nritems(eb) - 1) { 294931840ae1SZheng Yan btrfs_item_key(eb, &disk_key, slot + 1); 2950143bede5SJeff Mahoney BUG_ON(comp_keys(&disk_key, new_key) <= 0); 295131840ae1SZheng Yan } 295231840ae1SZheng Yan 295331840ae1SZheng Yan btrfs_cpu_key_to_disk(&disk_key, new_key); 295431840ae1SZheng Yan btrfs_set_item_key(eb, &disk_key, slot); 295531840ae1SZheng Yan btrfs_mark_buffer_dirty(eb); 295631840ae1SZheng Yan if (slot == 0) 295731840ae1SZheng Yan fixup_low_keys(trans, root, path, &disk_key, 1); 295831840ae1SZheng Yan } 295931840ae1SZheng Yan 296031840ae1SZheng Yan /* 296174123bd7SChris Mason * try to push data from one node into the next node left in the 296279f95c82SChris Mason * tree. 2963aa5d6bedSChris Mason * 2964aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 2965aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 296674123bd7SChris Mason */ 296798ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 296898ed5174SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 2969971a1f66SChris Mason struct extent_buffer *src, int empty) 2970be0e5c09SChris Mason { 2971be0e5c09SChris Mason int push_items = 0; 2972bb803951SChris Mason int src_nritems; 2973bb803951SChris Mason int dst_nritems; 2974aa5d6bedSChris Mason int ret = 0; 2975be0e5c09SChris Mason 29765f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 29775f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 2978123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 29797bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 29807bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 298154aa1f4dSChris Mason 2982bce4eae9SChris Mason if (!empty && src_nritems <= 8) 2983971a1f66SChris Mason return 1; 2984971a1f66SChris Mason 2985d397712bSChris Mason if (push_items <= 0) 2986be0e5c09SChris Mason return 1; 2987be0e5c09SChris Mason 2988bce4eae9SChris Mason if (empty) { 2989971a1f66SChris Mason push_items = min(src_nritems, push_items); 2990bce4eae9SChris Mason if (push_items < src_nritems) { 2991bce4eae9SChris Mason /* leave at least 8 pointers in the node if 2992bce4eae9SChris Mason * we aren't going to empty it 2993bce4eae9SChris Mason */ 2994bce4eae9SChris Mason if (src_nritems - push_items < 8) { 2995bce4eae9SChris Mason if (push_items <= 8) 2996bce4eae9SChris Mason return 1; 2997bce4eae9SChris Mason push_items -= 8; 2998bce4eae9SChris Mason } 2999bce4eae9SChris Mason } 3000bce4eae9SChris Mason } else 3001bce4eae9SChris Mason push_items = min(src_nritems - 8, push_items); 300279f95c82SChris Mason 3003f230475eSJan Schmidt tree_mod_log_eb_copy(root->fs_info, dst, src, dst_nritems, 0, 3004f230475eSJan Schmidt push_items); 30055f39d397SChris Mason copy_extent_buffer(dst, src, 30065f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 30075f39d397SChris Mason btrfs_node_key_ptr_offset(0), 3008123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 30095f39d397SChris Mason 3010bb803951SChris Mason if (push_items < src_nritems) { 301157911b8bSJan Schmidt /* 301257911b8bSJan Schmidt * don't call tree_mod_log_eb_move here, key removal was already 301357911b8bSJan Schmidt * fully logged by tree_mod_log_eb_copy above. 301457911b8bSJan Schmidt */ 30155f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 30165f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 3017e2fa7227SChris Mason (src_nritems - push_items) * 3018123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 3019bb803951SChris Mason } 30205f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 30215f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 30225f39d397SChris Mason btrfs_mark_buffer_dirty(src); 30235f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 302431840ae1SZheng Yan 3025bb803951SChris Mason return ret; 3026be0e5c09SChris Mason } 3027be0e5c09SChris Mason 302897571fd0SChris Mason /* 302979f95c82SChris Mason * try to push data from one node into the next node right in the 303079f95c82SChris Mason * tree. 303179f95c82SChris Mason * 303279f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 303379f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 303479f95c82SChris Mason * 303579f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 303679f95c82SChris Mason */ 30375f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 30385f39d397SChris Mason struct btrfs_root *root, 30395f39d397SChris Mason struct extent_buffer *dst, 30405f39d397SChris Mason struct extent_buffer *src) 304179f95c82SChris Mason { 304279f95c82SChris Mason int push_items = 0; 304379f95c82SChris Mason int max_push; 304479f95c82SChris Mason int src_nritems; 304579f95c82SChris Mason int dst_nritems; 304679f95c82SChris Mason int ret = 0; 304779f95c82SChris Mason 30487bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 30497bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 30507bb86316SChris Mason 30515f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 30525f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 3053123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 3054d397712bSChris Mason if (push_items <= 0) 305579f95c82SChris Mason return 1; 3056bce4eae9SChris Mason 3057d397712bSChris Mason if (src_nritems < 4) 3058bce4eae9SChris Mason return 1; 305979f95c82SChris Mason 306079f95c82SChris Mason max_push = src_nritems / 2 + 1; 306179f95c82SChris Mason /* don't try to empty the node */ 3062d397712bSChris Mason if (max_push >= src_nritems) 306379f95c82SChris Mason return 1; 3064252c38f0SYan 306579f95c82SChris Mason if (max_push < push_items) 306679f95c82SChris Mason push_items = max_push; 306779f95c82SChris Mason 3068f230475eSJan Schmidt tree_mod_log_eb_move(root->fs_info, dst, push_items, 0, dst_nritems); 30695f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 30705f39d397SChris Mason btrfs_node_key_ptr_offset(0), 30715f39d397SChris Mason (dst_nritems) * 30725f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 3073d6025579SChris Mason 3074f230475eSJan Schmidt tree_mod_log_eb_copy(root->fs_info, dst, src, 0, 3075f230475eSJan Schmidt src_nritems - push_items, push_items); 30765f39d397SChris Mason copy_extent_buffer(dst, src, 30775f39d397SChris Mason btrfs_node_key_ptr_offset(0), 30785f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 3079123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 308079f95c82SChris Mason 30815f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 30825f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 308379f95c82SChris Mason 30845f39d397SChris Mason btrfs_mark_buffer_dirty(src); 30855f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 308631840ae1SZheng Yan 308779f95c82SChris Mason return ret; 308879f95c82SChris Mason } 308979f95c82SChris Mason 309079f95c82SChris Mason /* 309197571fd0SChris Mason * helper function to insert a new root level in the tree. 309297571fd0SChris Mason * A new node is allocated, and a single item is inserted to 309397571fd0SChris Mason * point to the existing root 3094aa5d6bedSChris Mason * 3095aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 309697571fd0SChris Mason */ 3097d397712bSChris Mason static noinline int insert_new_root(struct btrfs_trans_handle *trans, 30985f39d397SChris Mason struct btrfs_root *root, 30995f39d397SChris Mason struct btrfs_path *path, int level) 310074123bd7SChris Mason { 31017bb86316SChris Mason u64 lower_gen; 31025f39d397SChris Mason struct extent_buffer *lower; 31035f39d397SChris Mason struct extent_buffer *c; 3104925baeddSChris Mason struct extent_buffer *old; 31055f39d397SChris Mason struct btrfs_disk_key lower_key; 31065c680ed6SChris Mason 31075c680ed6SChris Mason BUG_ON(path->nodes[level]); 31085c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 31095c680ed6SChris Mason 31107bb86316SChris Mason lower = path->nodes[level-1]; 31117bb86316SChris Mason if (level == 1) 31127bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 31137bb86316SChris Mason else 31147bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 31157bb86316SChris Mason 311631840ae1SZheng Yan c = btrfs_alloc_free_block(trans, root, root->nodesize, 0, 31175d4f98a2SYan Zheng root->root_key.objectid, &lower_key, 31185581a51aSJan Schmidt level, root->node->start, 0); 31195f39d397SChris Mason if (IS_ERR(c)) 31205f39d397SChris Mason return PTR_ERR(c); 3121925baeddSChris Mason 3122f0486c68SYan, Zheng root_add_used(root, root->nodesize); 3123f0486c68SYan, Zheng 31245d4f98a2SYan Zheng memset_extent_buffer(c, 0, 0, sizeof(struct btrfs_header)); 31255f39d397SChris Mason btrfs_set_header_nritems(c, 1); 31265f39d397SChris Mason btrfs_set_header_level(c, level); 3127db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 31285f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 31295d4f98a2SYan Zheng btrfs_set_header_backref_rev(c, BTRFS_MIXED_BACKREF_REV); 31305f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 3131d5719762SChris Mason 31325f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 31335f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 31345f39d397SChris Mason BTRFS_FSID_SIZE); 3135e17cade2SChris Mason 3136e17cade2SChris Mason write_extent_buffer(c, root->fs_info->chunk_tree_uuid, 3137e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(c), 3138e17cade2SChris Mason BTRFS_UUID_SIZE); 3139e17cade2SChris Mason 31405f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 3141db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 31427bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 314331840ae1SZheng Yan WARN_ON(lower_gen != trans->transid); 31447bb86316SChris Mason 31457bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 31465f39d397SChris Mason 31475f39d397SChris Mason btrfs_mark_buffer_dirty(c); 3148d5719762SChris Mason 3149925baeddSChris Mason old = root->node; 3150f230475eSJan Schmidt tree_mod_log_set_root_pointer(root, c); 3151240f62c8SChris Mason rcu_assign_pointer(root->node, c); 3152925baeddSChris Mason 3153925baeddSChris Mason /* the super has an extra ref to root->node */ 3154925baeddSChris Mason free_extent_buffer(old); 3155925baeddSChris Mason 31560b86a832SChris Mason add_root_to_dirty_list(root); 31575f39d397SChris Mason extent_buffer_get(c); 31585f39d397SChris Mason path->nodes[level] = c; 3159bd681513SChris Mason path->locks[level] = BTRFS_WRITE_LOCK; 316074123bd7SChris Mason path->slots[level] = 0; 316174123bd7SChris Mason return 0; 316274123bd7SChris Mason } 31635c680ed6SChris Mason 31645c680ed6SChris Mason /* 31655c680ed6SChris Mason * worker function to insert a single pointer in a node. 31665c680ed6SChris Mason * the node should have enough room for the pointer already 316797571fd0SChris Mason * 31685c680ed6SChris Mason * slot and level indicate where you want the key to go, and 31695c680ed6SChris Mason * blocknr is the block the key points to. 31705c680ed6SChris Mason */ 3171143bede5SJeff Mahoney static void insert_ptr(struct btrfs_trans_handle *trans, 3172143bede5SJeff Mahoney struct btrfs_root *root, struct btrfs_path *path, 3173143bede5SJeff Mahoney struct btrfs_disk_key *key, u64 bytenr, 3174c3e06965SJan Schmidt int slot, int level) 31755c680ed6SChris Mason { 31765f39d397SChris Mason struct extent_buffer *lower; 31775c680ed6SChris Mason int nritems; 3178f3ea38daSJan Schmidt int ret; 31795c680ed6SChris Mason 31805c680ed6SChris Mason BUG_ON(!path->nodes[level]); 3181f0486c68SYan, Zheng btrfs_assert_tree_locked(path->nodes[level]); 31825f39d397SChris Mason lower = path->nodes[level]; 31835f39d397SChris Mason nritems = btrfs_header_nritems(lower); 3184c293498bSStoyan Gaydarov BUG_ON(slot > nritems); 3185143bede5SJeff Mahoney BUG_ON(nritems == BTRFS_NODEPTRS_PER_BLOCK(root)); 318674123bd7SChris Mason if (slot != nritems) { 3187c3e06965SJan Schmidt if (level) 3188f3ea38daSJan Schmidt tree_mod_log_eb_move(root->fs_info, lower, slot + 1, 3189f3ea38daSJan Schmidt slot, nritems - slot); 31905f39d397SChris Mason memmove_extent_buffer(lower, 31915f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 31925f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 3193123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 319474123bd7SChris Mason } 3195c3e06965SJan Schmidt if (level) { 3196f3ea38daSJan Schmidt ret = tree_mod_log_insert_key(root->fs_info, lower, slot, 3197f3ea38daSJan Schmidt MOD_LOG_KEY_ADD); 3198f3ea38daSJan Schmidt BUG_ON(ret < 0); 3199f3ea38daSJan Schmidt } 32005f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 3201db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 320274493f7aSChris Mason WARN_ON(trans->transid == 0); 320374493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 32045f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 32055f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 320674123bd7SChris Mason } 320774123bd7SChris Mason 320897571fd0SChris Mason /* 320997571fd0SChris Mason * split the node at the specified level in path in two. 321097571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 321197571fd0SChris Mason * 321297571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 321397571fd0SChris Mason * left and right, if either one works, it returns right away. 3214aa5d6bedSChris Mason * 3215aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 321697571fd0SChris Mason */ 3217e02119d5SChris Mason static noinline int split_node(struct btrfs_trans_handle *trans, 3218e02119d5SChris Mason struct btrfs_root *root, 3219e02119d5SChris Mason struct btrfs_path *path, int level) 3220be0e5c09SChris Mason { 32215f39d397SChris Mason struct extent_buffer *c; 32225f39d397SChris Mason struct extent_buffer *split; 32235f39d397SChris Mason struct btrfs_disk_key disk_key; 3224be0e5c09SChris Mason int mid; 32255c680ed6SChris Mason int ret; 32267518a238SChris Mason u32 c_nritems; 3227be0e5c09SChris Mason 32285f39d397SChris Mason c = path->nodes[level]; 32297bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 32305f39d397SChris Mason if (c == root->node) { 32315c680ed6SChris Mason /* trying to split the root, lets make a new one */ 3232e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 32335c680ed6SChris Mason if (ret) 32345c680ed6SChris Mason return ret; 3235b3612421SChris Mason } else { 3236e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 32375f39d397SChris Mason c = path->nodes[level]; 32385f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 3239c448acf0SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) 3240e66f709bSChris Mason return 0; 324154aa1f4dSChris Mason if (ret < 0) 324254aa1f4dSChris Mason return ret; 32435c680ed6SChris Mason } 3244e66f709bSChris Mason 32455f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 32465d4f98a2SYan Zheng mid = (c_nritems + 1) / 2; 32475d4f98a2SYan Zheng btrfs_node_key(c, &disk_key, mid); 32487bb86316SChris Mason 32495d4f98a2SYan Zheng split = btrfs_alloc_free_block(trans, root, root->nodesize, 0, 32507bb86316SChris Mason root->root_key.objectid, 32515581a51aSJan Schmidt &disk_key, level, c->start, 0); 32525f39d397SChris Mason if (IS_ERR(split)) 32535f39d397SChris Mason return PTR_ERR(split); 325454aa1f4dSChris Mason 3255f0486c68SYan, Zheng root_add_used(root, root->nodesize); 3256f0486c68SYan, Zheng 32575d4f98a2SYan Zheng memset_extent_buffer(split, 0, 0, sizeof(struct btrfs_header)); 32585f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 3259db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 32605f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 32615d4f98a2SYan Zheng btrfs_set_header_backref_rev(split, BTRFS_MIXED_BACKREF_REV); 32625f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 32635f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 32645f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 32655f39d397SChris Mason BTRFS_FSID_SIZE); 3266e17cade2SChris Mason write_extent_buffer(split, root->fs_info->chunk_tree_uuid, 3267e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(split), 3268e17cade2SChris Mason BTRFS_UUID_SIZE); 32695f39d397SChris Mason 3270f230475eSJan Schmidt tree_mod_log_eb_copy(root->fs_info, split, c, 0, mid, c_nritems - mid); 32715f39d397SChris Mason copy_extent_buffer(split, c, 32725f39d397SChris Mason btrfs_node_key_ptr_offset(0), 32735f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 3274123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 32755f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 32765f39d397SChris Mason btrfs_set_header_nritems(c, mid); 3277aa5d6bedSChris Mason ret = 0; 3278aa5d6bedSChris Mason 32795f39d397SChris Mason btrfs_mark_buffer_dirty(c); 32805f39d397SChris Mason btrfs_mark_buffer_dirty(split); 32815f39d397SChris Mason 3282143bede5SJeff Mahoney insert_ptr(trans, root, path, &disk_key, split->start, 3283c3e06965SJan Schmidt path->slots[level + 1] + 1, level + 1); 3284aa5d6bedSChris Mason 32855de08d7dSChris Mason if (path->slots[level] >= mid) { 32865c680ed6SChris Mason path->slots[level] -= mid; 3287925baeddSChris Mason btrfs_tree_unlock(c); 32885f39d397SChris Mason free_extent_buffer(c); 32895f39d397SChris Mason path->nodes[level] = split; 32905c680ed6SChris Mason path->slots[level + 1] += 1; 3291eb60ceacSChris Mason } else { 3292925baeddSChris Mason btrfs_tree_unlock(split); 32935f39d397SChris Mason free_extent_buffer(split); 3294be0e5c09SChris Mason } 3295aa5d6bedSChris Mason return ret; 3296be0e5c09SChris Mason } 3297be0e5c09SChris Mason 329874123bd7SChris Mason /* 329974123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 330074123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 330174123bd7SChris Mason * space used both by the item structs and the item data 330274123bd7SChris Mason */ 33035f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 3304be0e5c09SChris Mason { 3305be0e5c09SChris Mason int data_len; 33065f39d397SChris Mason int nritems = btrfs_header_nritems(l); 3307d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 3308be0e5c09SChris Mason 3309be0e5c09SChris Mason if (!nr) 3310be0e5c09SChris Mason return 0; 33115f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 33125f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 33130783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 3314d4dbff95SChris Mason WARN_ON(data_len < 0); 3315be0e5c09SChris Mason return data_len; 3316be0e5c09SChris Mason } 3317be0e5c09SChris Mason 331874123bd7SChris Mason /* 3319d4dbff95SChris Mason * The space between the end of the leaf items and 3320d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 3321d4dbff95SChris Mason * the leaf has left for both items and data 3322d4dbff95SChris Mason */ 3323d397712bSChris Mason noinline int btrfs_leaf_free_space(struct btrfs_root *root, 3324e02119d5SChris Mason struct extent_buffer *leaf) 3325d4dbff95SChris Mason { 33265f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 33275f39d397SChris Mason int ret; 33285f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 33295f39d397SChris Mason if (ret < 0) { 3330d397712bSChris Mason printk(KERN_CRIT "leaf free space ret %d, leaf data size %lu, " 3331d397712bSChris Mason "used %d nritems %d\n", 3332ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 33335f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 33345f39d397SChris Mason } 33355f39d397SChris Mason return ret; 3336d4dbff95SChris Mason } 3337d4dbff95SChris Mason 333899d8f83cSChris Mason /* 333999d8f83cSChris Mason * min slot controls the lowest index we're willing to push to the 334099d8f83cSChris Mason * right. We'll push up to and including min_slot, but no lower 334199d8f83cSChris Mason */ 334244871b1bSChris Mason static noinline int __push_leaf_right(struct btrfs_trans_handle *trans, 334344871b1bSChris Mason struct btrfs_root *root, 334444871b1bSChris Mason struct btrfs_path *path, 334544871b1bSChris Mason int data_size, int empty, 334644871b1bSChris Mason struct extent_buffer *right, 334799d8f83cSChris Mason int free_space, u32 left_nritems, 334899d8f83cSChris Mason u32 min_slot) 334900ec4c51SChris Mason { 33505f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 335144871b1bSChris Mason struct extent_buffer *upper = path->nodes[1]; 3352cfed81a0SChris Mason struct btrfs_map_token token; 33535f39d397SChris Mason struct btrfs_disk_key disk_key; 335400ec4c51SChris Mason int slot; 335534a38218SChris Mason u32 i; 335600ec4c51SChris Mason int push_space = 0; 335700ec4c51SChris Mason int push_items = 0; 33580783fcfcSChris Mason struct btrfs_item *item; 335934a38218SChris Mason u32 nr; 33607518a238SChris Mason u32 right_nritems; 33615f39d397SChris Mason u32 data_end; 3362db94535dSChris Mason u32 this_item_size; 336300ec4c51SChris Mason 3364cfed81a0SChris Mason btrfs_init_map_token(&token); 3365cfed81a0SChris Mason 336634a38218SChris Mason if (empty) 336734a38218SChris Mason nr = 0; 336834a38218SChris Mason else 336999d8f83cSChris Mason nr = max_t(u32, 1, min_slot); 337034a38218SChris Mason 337131840ae1SZheng Yan if (path->slots[0] >= left_nritems) 337287b29b20SYan Zheng push_space += data_size; 337331840ae1SZheng Yan 337444871b1bSChris Mason slot = path->slots[1]; 337534a38218SChris Mason i = left_nritems - 1; 337634a38218SChris Mason while (i >= nr) { 33775f39d397SChris Mason item = btrfs_item_nr(left, i); 3378db94535dSChris Mason 337931840ae1SZheng Yan if (!empty && push_items > 0) { 338031840ae1SZheng Yan if (path->slots[0] > i) 338131840ae1SZheng Yan break; 338231840ae1SZheng Yan if (path->slots[0] == i) { 338331840ae1SZheng Yan int space = btrfs_leaf_free_space(root, left); 338431840ae1SZheng Yan if (space + push_space * 2 > free_space) 338531840ae1SZheng Yan break; 338631840ae1SZheng Yan } 338731840ae1SZheng Yan } 338831840ae1SZheng Yan 338900ec4c51SChris Mason if (path->slots[0] == i) 339087b29b20SYan Zheng push_space += data_size; 3391db94535dSChris Mason 3392db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 3393db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 339400ec4c51SChris Mason break; 339531840ae1SZheng Yan 339600ec4c51SChris Mason push_items++; 3397db94535dSChris Mason push_space += this_item_size + sizeof(*item); 339834a38218SChris Mason if (i == 0) 339934a38218SChris Mason break; 340034a38218SChris Mason i--; 3401db94535dSChris Mason } 34025f39d397SChris Mason 3403925baeddSChris Mason if (push_items == 0) 3404925baeddSChris Mason goto out_unlock; 34055f39d397SChris Mason 340634a38218SChris Mason if (!empty && push_items == left_nritems) 3407a429e513SChris Mason WARN_ON(1); 34085f39d397SChris Mason 340900ec4c51SChris Mason /* push left to right */ 34105f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 341134a38218SChris Mason 34125f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 3413123abc88SChris Mason push_space -= leaf_data_end(root, left); 34145f39d397SChris Mason 341500ec4c51SChris Mason /* make room in the right data area */ 34165f39d397SChris Mason data_end = leaf_data_end(root, right); 34175f39d397SChris Mason memmove_extent_buffer(right, 34185f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 34195f39d397SChris Mason btrfs_leaf_data(right) + data_end, 34205f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 34215f39d397SChris Mason 342200ec4c51SChris Mason /* copy from the left data area */ 34235f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 3424d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 3425d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 3426d6025579SChris Mason push_space); 34275f39d397SChris Mason 34285f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 34295f39d397SChris Mason btrfs_item_nr_offset(0), 34300783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 34315f39d397SChris Mason 343200ec4c51SChris Mason /* copy the items from left to right */ 34335f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 34345f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 34350783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 343600ec4c51SChris Mason 343700ec4c51SChris Mason /* update the item pointers */ 34387518a238SChris Mason right_nritems += push_items; 34395f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 3440123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 34417518a238SChris Mason for (i = 0; i < right_nritems; i++) { 34425f39d397SChris Mason item = btrfs_item_nr(right, i); 3443cfed81a0SChris Mason push_space -= btrfs_token_item_size(right, item, &token); 3444cfed81a0SChris Mason btrfs_set_token_item_offset(right, item, push_space, &token); 3445db94535dSChris Mason } 3446db94535dSChris Mason 34477518a238SChris Mason left_nritems -= push_items; 34485f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 344900ec4c51SChris Mason 345034a38218SChris Mason if (left_nritems) 34515f39d397SChris Mason btrfs_mark_buffer_dirty(left); 3452f0486c68SYan, Zheng else 3453f0486c68SYan, Zheng clean_tree_block(trans, root, left); 3454f0486c68SYan, Zheng 34555f39d397SChris Mason btrfs_mark_buffer_dirty(right); 3456a429e513SChris Mason 34575f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 34585f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 3459d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 346002217ed2SChris Mason 346100ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 34627518a238SChris Mason if (path->slots[0] >= left_nritems) { 34637518a238SChris Mason path->slots[0] -= left_nritems; 3464925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 3465925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 3466925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 34675f39d397SChris Mason free_extent_buffer(path->nodes[0]); 34685f39d397SChris Mason path->nodes[0] = right; 346900ec4c51SChris Mason path->slots[1] += 1; 347000ec4c51SChris Mason } else { 3471925baeddSChris Mason btrfs_tree_unlock(right); 34725f39d397SChris Mason free_extent_buffer(right); 347300ec4c51SChris Mason } 347400ec4c51SChris Mason return 0; 3475925baeddSChris Mason 3476925baeddSChris Mason out_unlock: 3477925baeddSChris Mason btrfs_tree_unlock(right); 3478925baeddSChris Mason free_extent_buffer(right); 3479925baeddSChris Mason return 1; 348000ec4c51SChris Mason } 3481925baeddSChris Mason 348200ec4c51SChris Mason /* 348344871b1bSChris Mason * push some data in the path leaf to the right, trying to free up at 348474123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 348544871b1bSChris Mason * 348644871b1bSChris Mason * returns 1 if the push failed because the other node didn't have enough 348744871b1bSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 348899d8f83cSChris Mason * 348999d8f83cSChris Mason * this will push starting from min_slot to the end of the leaf. It won't 349099d8f83cSChris Mason * push any slot lower than min_slot 349174123bd7SChris Mason */ 349244871b1bSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 349399d8f83cSChris Mason *root, struct btrfs_path *path, 349499d8f83cSChris Mason int min_data_size, int data_size, 349599d8f83cSChris Mason int empty, u32 min_slot) 3496be0e5c09SChris Mason { 349744871b1bSChris Mason struct extent_buffer *left = path->nodes[0]; 349844871b1bSChris Mason struct extent_buffer *right; 349944871b1bSChris Mason struct extent_buffer *upper; 350044871b1bSChris Mason int slot; 350144871b1bSChris Mason int free_space; 350244871b1bSChris Mason u32 left_nritems; 350344871b1bSChris Mason int ret; 350444871b1bSChris Mason 350544871b1bSChris Mason if (!path->nodes[1]) 350644871b1bSChris Mason return 1; 350744871b1bSChris Mason 350844871b1bSChris Mason slot = path->slots[1]; 350944871b1bSChris Mason upper = path->nodes[1]; 351044871b1bSChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 351144871b1bSChris Mason return 1; 351244871b1bSChris Mason 351344871b1bSChris Mason btrfs_assert_tree_locked(path->nodes[1]); 351444871b1bSChris Mason 351544871b1bSChris Mason right = read_node_slot(root, upper, slot + 1); 351691ca338dSTsutomu Itoh if (right == NULL) 351791ca338dSTsutomu Itoh return 1; 351891ca338dSTsutomu Itoh 351944871b1bSChris Mason btrfs_tree_lock(right); 352044871b1bSChris Mason btrfs_set_lock_blocking(right); 352144871b1bSChris Mason 352244871b1bSChris Mason free_space = btrfs_leaf_free_space(root, right); 352344871b1bSChris Mason if (free_space < data_size) 352444871b1bSChris Mason goto out_unlock; 352544871b1bSChris Mason 352644871b1bSChris Mason /* cow and double check */ 352744871b1bSChris Mason ret = btrfs_cow_block(trans, root, right, upper, 352844871b1bSChris Mason slot + 1, &right); 352944871b1bSChris Mason if (ret) 353044871b1bSChris Mason goto out_unlock; 353144871b1bSChris Mason 353244871b1bSChris Mason free_space = btrfs_leaf_free_space(root, right); 353344871b1bSChris Mason if (free_space < data_size) 353444871b1bSChris Mason goto out_unlock; 353544871b1bSChris Mason 353644871b1bSChris Mason left_nritems = btrfs_header_nritems(left); 353744871b1bSChris Mason if (left_nritems == 0) 353844871b1bSChris Mason goto out_unlock; 353944871b1bSChris Mason 354099d8f83cSChris Mason return __push_leaf_right(trans, root, path, min_data_size, empty, 354199d8f83cSChris Mason right, free_space, left_nritems, min_slot); 354244871b1bSChris Mason out_unlock: 354344871b1bSChris Mason btrfs_tree_unlock(right); 354444871b1bSChris Mason free_extent_buffer(right); 354544871b1bSChris Mason return 1; 354644871b1bSChris Mason } 354744871b1bSChris Mason 354844871b1bSChris Mason /* 354944871b1bSChris Mason * push some data in the path leaf to the left, trying to free up at 355044871b1bSChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 355199d8f83cSChris Mason * 355299d8f83cSChris Mason * max_slot can put a limit on how far into the leaf we'll push items. The 355399d8f83cSChris Mason * item at 'max_slot' won't be touched. Use (u32)-1 to make us do all the 355499d8f83cSChris Mason * items 355544871b1bSChris Mason */ 355644871b1bSChris Mason static noinline int __push_leaf_left(struct btrfs_trans_handle *trans, 355744871b1bSChris Mason struct btrfs_root *root, 355844871b1bSChris Mason struct btrfs_path *path, int data_size, 355944871b1bSChris Mason int empty, struct extent_buffer *left, 356099d8f83cSChris Mason int free_space, u32 right_nritems, 356199d8f83cSChris Mason u32 max_slot) 356244871b1bSChris Mason { 35635f39d397SChris Mason struct btrfs_disk_key disk_key; 35645f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 3565be0e5c09SChris Mason int i; 3566be0e5c09SChris Mason int push_space = 0; 3567be0e5c09SChris Mason int push_items = 0; 35680783fcfcSChris Mason struct btrfs_item *item; 35697518a238SChris Mason u32 old_left_nritems; 357034a38218SChris Mason u32 nr; 3571aa5d6bedSChris Mason int ret = 0; 3572db94535dSChris Mason u32 this_item_size; 3573db94535dSChris Mason u32 old_left_item_size; 3574cfed81a0SChris Mason struct btrfs_map_token token; 3575cfed81a0SChris Mason 3576cfed81a0SChris Mason btrfs_init_map_token(&token); 3577be0e5c09SChris Mason 357834a38218SChris Mason if (empty) 357999d8f83cSChris Mason nr = min(right_nritems, max_slot); 358034a38218SChris Mason else 358199d8f83cSChris Mason nr = min(right_nritems - 1, max_slot); 358234a38218SChris Mason 358334a38218SChris Mason for (i = 0; i < nr; i++) { 35845f39d397SChris Mason item = btrfs_item_nr(right, i); 3585db94535dSChris Mason 358631840ae1SZheng Yan if (!empty && push_items > 0) { 358731840ae1SZheng Yan if (path->slots[0] < i) 358831840ae1SZheng Yan break; 358931840ae1SZheng Yan if (path->slots[0] == i) { 359031840ae1SZheng Yan int space = btrfs_leaf_free_space(root, right); 359131840ae1SZheng Yan if (space + push_space * 2 > free_space) 359231840ae1SZheng Yan break; 359331840ae1SZheng Yan } 359431840ae1SZheng Yan } 359531840ae1SZheng Yan 3596be0e5c09SChris Mason if (path->slots[0] == i) 359787b29b20SYan Zheng push_space += data_size; 3598db94535dSChris Mason 3599db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 3600db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 3601be0e5c09SChris Mason break; 3602db94535dSChris Mason 3603be0e5c09SChris Mason push_items++; 3604db94535dSChris Mason push_space += this_item_size + sizeof(*item); 3605be0e5c09SChris Mason } 3606db94535dSChris Mason 3607be0e5c09SChris Mason if (push_items == 0) { 3608925baeddSChris Mason ret = 1; 3609925baeddSChris Mason goto out; 3610be0e5c09SChris Mason } 361134a38218SChris Mason if (!empty && push_items == btrfs_header_nritems(right)) 3612a429e513SChris Mason WARN_ON(1); 36135f39d397SChris Mason 3614be0e5c09SChris Mason /* push data from right to left */ 36155f39d397SChris Mason copy_extent_buffer(left, right, 36165f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 36175f39d397SChris Mason btrfs_item_nr_offset(0), 36185f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 36195f39d397SChris Mason 3620123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 36215f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1); 36225f39d397SChris Mason 36235f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 3624d6025579SChris Mason leaf_data_end(root, left) - push_space, 3625123abc88SChris Mason btrfs_leaf_data(right) + 36265f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 3627be0e5c09SChris Mason push_space); 36285f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 362987b29b20SYan Zheng BUG_ON(old_left_nritems <= 0); 3630eb60ceacSChris Mason 3631db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 3632be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 36335f39d397SChris Mason u32 ioff; 3634db94535dSChris Mason 36355f39d397SChris Mason item = btrfs_item_nr(left, i); 3636db94535dSChris Mason 3637cfed81a0SChris Mason ioff = btrfs_token_item_offset(left, item, &token); 3638cfed81a0SChris Mason btrfs_set_token_item_offset(left, item, 3639cfed81a0SChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size), 3640cfed81a0SChris Mason &token); 3641be0e5c09SChris Mason } 36425f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 3643be0e5c09SChris Mason 3644be0e5c09SChris Mason /* fixup right node */ 364534a38218SChris Mason if (push_items > right_nritems) { 3646d397712bSChris Mason printk(KERN_CRIT "push items %d nr %u\n", push_items, 3647d397712bSChris Mason right_nritems); 364834a38218SChris Mason WARN_ON(1); 364934a38218SChris Mason } 365034a38218SChris Mason 365134a38218SChris Mason if (push_items < right_nritems) { 36525f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 3653123abc88SChris Mason leaf_data_end(root, right); 36545f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 3655d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 3656d6025579SChris Mason btrfs_leaf_data(right) + 3657123abc88SChris Mason leaf_data_end(root, right), push_space); 36585f39d397SChris Mason 36595f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 36605f39d397SChris Mason btrfs_item_nr_offset(push_items), 36615f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 36620783fcfcSChris Mason sizeof(struct btrfs_item)); 366334a38218SChris Mason } 3664eef1c494SYan right_nritems -= push_items; 3665eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 3666123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 36675f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 36685f39d397SChris Mason item = btrfs_item_nr(right, i); 3669db94535dSChris Mason 3670cfed81a0SChris Mason push_space = push_space - btrfs_token_item_size(right, 3671cfed81a0SChris Mason item, &token); 3672cfed81a0SChris Mason btrfs_set_token_item_offset(right, item, push_space, &token); 3673db94535dSChris Mason } 3674eb60ceacSChris Mason 36755f39d397SChris Mason btrfs_mark_buffer_dirty(left); 367634a38218SChris Mason if (right_nritems) 36775f39d397SChris Mason btrfs_mark_buffer_dirty(right); 3678f0486c68SYan, Zheng else 3679f0486c68SYan, Zheng clean_tree_block(trans, root, right); 3680098f59c2SChris Mason 36815f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 3682143bede5SJeff Mahoney fixup_low_keys(trans, root, path, &disk_key, 1); 3683be0e5c09SChris Mason 3684be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 3685be0e5c09SChris Mason if (path->slots[0] < push_items) { 3686be0e5c09SChris Mason path->slots[0] += old_left_nritems; 3687925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 36885f39d397SChris Mason free_extent_buffer(path->nodes[0]); 36895f39d397SChris Mason path->nodes[0] = left; 3690be0e5c09SChris Mason path->slots[1] -= 1; 3691be0e5c09SChris Mason } else { 3692925baeddSChris Mason btrfs_tree_unlock(left); 36935f39d397SChris Mason free_extent_buffer(left); 3694be0e5c09SChris Mason path->slots[0] -= push_items; 3695be0e5c09SChris Mason } 3696eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 3697aa5d6bedSChris Mason return ret; 3698925baeddSChris Mason out: 3699925baeddSChris Mason btrfs_tree_unlock(left); 3700925baeddSChris Mason free_extent_buffer(left); 3701925baeddSChris Mason return ret; 3702be0e5c09SChris Mason } 3703be0e5c09SChris Mason 370474123bd7SChris Mason /* 370544871b1bSChris Mason * push some data in the path leaf to the left, trying to free up at 370644871b1bSChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 370799d8f83cSChris Mason * 370899d8f83cSChris Mason * max_slot can put a limit on how far into the leaf we'll push items. The 370999d8f83cSChris Mason * item at 'max_slot' won't be touched. Use (u32)-1 to make us push all the 371099d8f83cSChris Mason * items 371144871b1bSChris Mason */ 371244871b1bSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 371399d8f83cSChris Mason *root, struct btrfs_path *path, int min_data_size, 371499d8f83cSChris Mason int data_size, int empty, u32 max_slot) 371544871b1bSChris Mason { 371644871b1bSChris Mason struct extent_buffer *right = path->nodes[0]; 371744871b1bSChris Mason struct extent_buffer *left; 371844871b1bSChris Mason int slot; 371944871b1bSChris Mason int free_space; 372044871b1bSChris Mason u32 right_nritems; 372144871b1bSChris Mason int ret = 0; 372244871b1bSChris Mason 372344871b1bSChris Mason slot = path->slots[1]; 372444871b1bSChris Mason if (slot == 0) 372544871b1bSChris Mason return 1; 372644871b1bSChris Mason if (!path->nodes[1]) 372744871b1bSChris Mason return 1; 372844871b1bSChris Mason 372944871b1bSChris Mason right_nritems = btrfs_header_nritems(right); 373044871b1bSChris Mason if (right_nritems == 0) 373144871b1bSChris Mason return 1; 373244871b1bSChris Mason 373344871b1bSChris Mason btrfs_assert_tree_locked(path->nodes[1]); 373444871b1bSChris Mason 373544871b1bSChris Mason left = read_node_slot(root, path->nodes[1], slot - 1); 373691ca338dSTsutomu Itoh if (left == NULL) 373791ca338dSTsutomu Itoh return 1; 373891ca338dSTsutomu Itoh 373944871b1bSChris Mason btrfs_tree_lock(left); 374044871b1bSChris Mason btrfs_set_lock_blocking(left); 374144871b1bSChris Mason 374244871b1bSChris Mason free_space = btrfs_leaf_free_space(root, left); 374344871b1bSChris Mason if (free_space < data_size) { 374444871b1bSChris Mason ret = 1; 374544871b1bSChris Mason goto out; 374644871b1bSChris Mason } 374744871b1bSChris Mason 374844871b1bSChris Mason /* cow and double check */ 374944871b1bSChris Mason ret = btrfs_cow_block(trans, root, left, 375044871b1bSChris Mason path->nodes[1], slot - 1, &left); 375144871b1bSChris Mason if (ret) { 375244871b1bSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 375379787eaaSJeff Mahoney if (ret == -ENOSPC) 375444871b1bSChris Mason ret = 1; 375544871b1bSChris Mason goto out; 375644871b1bSChris Mason } 375744871b1bSChris Mason 375844871b1bSChris Mason free_space = btrfs_leaf_free_space(root, left); 375944871b1bSChris Mason if (free_space < data_size) { 376044871b1bSChris Mason ret = 1; 376144871b1bSChris Mason goto out; 376244871b1bSChris Mason } 376344871b1bSChris Mason 376499d8f83cSChris Mason return __push_leaf_left(trans, root, path, min_data_size, 376599d8f83cSChris Mason empty, left, free_space, right_nritems, 376699d8f83cSChris Mason max_slot); 376744871b1bSChris Mason out: 376844871b1bSChris Mason btrfs_tree_unlock(left); 376944871b1bSChris Mason free_extent_buffer(left); 377044871b1bSChris Mason return ret; 377144871b1bSChris Mason } 377244871b1bSChris Mason 377344871b1bSChris Mason /* 377474123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 377574123bd7SChris Mason * available for the resulting leaf level of the path. 377674123bd7SChris Mason */ 3777143bede5SJeff Mahoney static noinline void copy_for_split(struct btrfs_trans_handle *trans, 3778e02119d5SChris Mason struct btrfs_root *root, 377944871b1bSChris Mason struct btrfs_path *path, 378044871b1bSChris Mason struct extent_buffer *l, 378144871b1bSChris Mason struct extent_buffer *right, 378244871b1bSChris Mason int slot, int mid, int nritems) 3783be0e5c09SChris Mason { 3784be0e5c09SChris Mason int data_copy_size; 3785be0e5c09SChris Mason int rt_data_off; 3786be0e5c09SChris Mason int i; 3787d4dbff95SChris Mason struct btrfs_disk_key disk_key; 3788cfed81a0SChris Mason struct btrfs_map_token token; 3789cfed81a0SChris Mason 3790cfed81a0SChris Mason btrfs_init_map_token(&token); 3791be0e5c09SChris Mason 37925f39d397SChris Mason nritems = nritems - mid; 37935f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 37945f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 37955f39d397SChris Mason 37965f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 37975f39d397SChris Mason btrfs_item_nr_offset(mid), 37985f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 37995f39d397SChris Mason 38005f39d397SChris Mason copy_extent_buffer(right, l, 3801d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 3802123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 3803123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 380474123bd7SChris Mason 38055f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 38065f39d397SChris Mason btrfs_item_end_nr(l, mid); 38075f39d397SChris Mason 38085f39d397SChris Mason for (i = 0; i < nritems; i++) { 38095f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 3810db94535dSChris Mason u32 ioff; 3811db94535dSChris Mason 3812cfed81a0SChris Mason ioff = btrfs_token_item_offset(right, item, &token); 3813cfed81a0SChris Mason btrfs_set_token_item_offset(right, item, 3814cfed81a0SChris Mason ioff + rt_data_off, &token); 38150783fcfcSChris Mason } 381674123bd7SChris Mason 38175f39d397SChris Mason btrfs_set_header_nritems(l, mid); 38185f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 3819143bede5SJeff Mahoney insert_ptr(trans, root, path, &disk_key, right->start, 3820c3e06965SJan Schmidt path->slots[1] + 1, 1); 38215f39d397SChris Mason 38225f39d397SChris Mason btrfs_mark_buffer_dirty(right); 38235f39d397SChris Mason btrfs_mark_buffer_dirty(l); 3824eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 38255f39d397SChris Mason 3826be0e5c09SChris Mason if (mid <= slot) { 3827925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 38285f39d397SChris Mason free_extent_buffer(path->nodes[0]); 38295f39d397SChris Mason path->nodes[0] = right; 3830be0e5c09SChris Mason path->slots[0] -= mid; 3831be0e5c09SChris Mason path->slots[1] += 1; 3832925baeddSChris Mason } else { 3833925baeddSChris Mason btrfs_tree_unlock(right); 38345f39d397SChris Mason free_extent_buffer(right); 3835925baeddSChris Mason } 38365f39d397SChris Mason 3837eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 383844871b1bSChris Mason } 383944871b1bSChris Mason 384044871b1bSChris Mason /* 384199d8f83cSChris Mason * double splits happen when we need to insert a big item in the middle 384299d8f83cSChris Mason * of a leaf. A double split can leave us with 3 mostly empty leaves: 384399d8f83cSChris Mason * leaf: [ slots 0 - N] [ our target ] [ N + 1 - total in leaf ] 384499d8f83cSChris Mason * A B C 384599d8f83cSChris Mason * 384699d8f83cSChris Mason * We avoid this by trying to push the items on either side of our target 384799d8f83cSChris Mason * into the adjacent leaves. If all goes well we can avoid the double split 384899d8f83cSChris Mason * completely. 384999d8f83cSChris Mason */ 385099d8f83cSChris Mason static noinline int push_for_double_split(struct btrfs_trans_handle *trans, 385199d8f83cSChris Mason struct btrfs_root *root, 385299d8f83cSChris Mason struct btrfs_path *path, 385399d8f83cSChris Mason int data_size) 385499d8f83cSChris Mason { 385599d8f83cSChris Mason int ret; 385699d8f83cSChris Mason int progress = 0; 385799d8f83cSChris Mason int slot; 385899d8f83cSChris Mason u32 nritems; 385999d8f83cSChris Mason 386099d8f83cSChris Mason slot = path->slots[0]; 386199d8f83cSChris Mason 386299d8f83cSChris Mason /* 386399d8f83cSChris Mason * try to push all the items after our slot into the 386499d8f83cSChris Mason * right leaf 386599d8f83cSChris Mason */ 386699d8f83cSChris Mason ret = push_leaf_right(trans, root, path, 1, data_size, 0, slot); 386799d8f83cSChris Mason if (ret < 0) 386899d8f83cSChris Mason return ret; 386999d8f83cSChris Mason 387099d8f83cSChris Mason if (ret == 0) 387199d8f83cSChris Mason progress++; 387299d8f83cSChris Mason 387399d8f83cSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 387499d8f83cSChris Mason /* 387599d8f83cSChris Mason * our goal is to get our slot at the start or end of a leaf. If 387699d8f83cSChris Mason * we've done so we're done 387799d8f83cSChris Mason */ 387899d8f83cSChris Mason if (path->slots[0] == 0 || path->slots[0] == nritems) 387999d8f83cSChris Mason return 0; 388099d8f83cSChris Mason 388199d8f83cSChris Mason if (btrfs_leaf_free_space(root, path->nodes[0]) >= data_size) 388299d8f83cSChris Mason return 0; 388399d8f83cSChris Mason 388499d8f83cSChris Mason /* try to push all the items before our slot into the next leaf */ 388599d8f83cSChris Mason slot = path->slots[0]; 388699d8f83cSChris Mason ret = push_leaf_left(trans, root, path, 1, data_size, 0, slot); 388799d8f83cSChris Mason if (ret < 0) 388899d8f83cSChris Mason return ret; 388999d8f83cSChris Mason 389099d8f83cSChris Mason if (ret == 0) 389199d8f83cSChris Mason progress++; 389299d8f83cSChris Mason 389399d8f83cSChris Mason if (progress) 389499d8f83cSChris Mason return 0; 389599d8f83cSChris Mason return 1; 389699d8f83cSChris Mason } 389799d8f83cSChris Mason 389899d8f83cSChris Mason /* 389944871b1bSChris Mason * split the path's leaf in two, making sure there is at least data_size 390044871b1bSChris Mason * available for the resulting leaf level of the path. 390144871b1bSChris Mason * 390244871b1bSChris Mason * returns 0 if all went well and < 0 on failure. 390344871b1bSChris Mason */ 390444871b1bSChris Mason static noinline int split_leaf(struct btrfs_trans_handle *trans, 390544871b1bSChris Mason struct btrfs_root *root, 390644871b1bSChris Mason struct btrfs_key *ins_key, 390744871b1bSChris Mason struct btrfs_path *path, int data_size, 390844871b1bSChris Mason int extend) 390944871b1bSChris Mason { 39105d4f98a2SYan Zheng struct btrfs_disk_key disk_key; 391144871b1bSChris Mason struct extent_buffer *l; 391244871b1bSChris Mason u32 nritems; 391344871b1bSChris Mason int mid; 391444871b1bSChris Mason int slot; 391544871b1bSChris Mason struct extent_buffer *right; 391644871b1bSChris Mason int ret = 0; 391744871b1bSChris Mason int wret; 39185d4f98a2SYan Zheng int split; 391944871b1bSChris Mason int num_doubles = 0; 392099d8f83cSChris Mason int tried_avoid_double = 0; 392144871b1bSChris Mason 3922a5719521SYan, Zheng l = path->nodes[0]; 3923a5719521SYan, Zheng slot = path->slots[0]; 3924a5719521SYan, Zheng if (extend && data_size + btrfs_item_size_nr(l, slot) + 3925a5719521SYan, Zheng sizeof(struct btrfs_item) > BTRFS_LEAF_DATA_SIZE(root)) 3926a5719521SYan, Zheng return -EOVERFLOW; 3927a5719521SYan, Zheng 392844871b1bSChris Mason /* first try to make some room by pushing left and right */ 392999d8f83cSChris Mason if (data_size) { 393099d8f83cSChris Mason wret = push_leaf_right(trans, root, path, data_size, 393199d8f83cSChris Mason data_size, 0, 0); 393244871b1bSChris Mason if (wret < 0) 393344871b1bSChris Mason return wret; 393444871b1bSChris Mason if (wret) { 393599d8f83cSChris Mason wret = push_leaf_left(trans, root, path, data_size, 393699d8f83cSChris Mason data_size, 0, (u32)-1); 393744871b1bSChris Mason if (wret < 0) 393844871b1bSChris Mason return wret; 393944871b1bSChris Mason } 394044871b1bSChris Mason l = path->nodes[0]; 394144871b1bSChris Mason 394244871b1bSChris Mason /* did the pushes work? */ 394344871b1bSChris Mason if (btrfs_leaf_free_space(root, l) >= data_size) 394444871b1bSChris Mason return 0; 394544871b1bSChris Mason } 394644871b1bSChris Mason 394744871b1bSChris Mason if (!path->nodes[1]) { 394844871b1bSChris Mason ret = insert_new_root(trans, root, path, 1); 394944871b1bSChris Mason if (ret) 395044871b1bSChris Mason return ret; 395144871b1bSChris Mason } 395244871b1bSChris Mason again: 39535d4f98a2SYan Zheng split = 1; 395444871b1bSChris Mason l = path->nodes[0]; 395544871b1bSChris Mason slot = path->slots[0]; 395644871b1bSChris Mason nritems = btrfs_header_nritems(l); 395744871b1bSChris Mason mid = (nritems + 1) / 2; 395844871b1bSChris Mason 39595d4f98a2SYan Zheng if (mid <= slot) { 39605d4f98a2SYan Zheng if (nritems == 1 || 39615d4f98a2SYan Zheng leaf_space_used(l, mid, nritems - mid) + data_size > 39625d4f98a2SYan Zheng BTRFS_LEAF_DATA_SIZE(root)) { 39635d4f98a2SYan Zheng if (slot >= nritems) { 39645d4f98a2SYan Zheng split = 0; 39655d4f98a2SYan Zheng } else { 39665d4f98a2SYan Zheng mid = slot; 39675d4f98a2SYan Zheng if (mid != nritems && 39685d4f98a2SYan Zheng leaf_space_used(l, mid, nritems - mid) + 39695d4f98a2SYan Zheng data_size > BTRFS_LEAF_DATA_SIZE(root)) { 397099d8f83cSChris Mason if (data_size && !tried_avoid_double) 397199d8f83cSChris Mason goto push_for_double; 39725d4f98a2SYan Zheng split = 2; 39735d4f98a2SYan Zheng } 39745d4f98a2SYan Zheng } 39755d4f98a2SYan Zheng } 39765d4f98a2SYan Zheng } else { 39775d4f98a2SYan Zheng if (leaf_space_used(l, 0, mid) + data_size > 39785d4f98a2SYan Zheng BTRFS_LEAF_DATA_SIZE(root)) { 39795d4f98a2SYan Zheng if (!extend && data_size && slot == 0) { 39805d4f98a2SYan Zheng split = 0; 39815d4f98a2SYan Zheng } else if ((extend || !data_size) && slot == 0) { 39825d4f98a2SYan Zheng mid = 1; 39835d4f98a2SYan Zheng } else { 39845d4f98a2SYan Zheng mid = slot; 39855d4f98a2SYan Zheng if (mid != nritems && 39865d4f98a2SYan Zheng leaf_space_used(l, mid, nritems - mid) + 39875d4f98a2SYan Zheng data_size > BTRFS_LEAF_DATA_SIZE(root)) { 398899d8f83cSChris Mason if (data_size && !tried_avoid_double) 398999d8f83cSChris Mason goto push_for_double; 39905d4f98a2SYan Zheng split = 2 ; 39915d4f98a2SYan Zheng } 39925d4f98a2SYan Zheng } 39935d4f98a2SYan Zheng } 39945d4f98a2SYan Zheng } 39955d4f98a2SYan Zheng 39965d4f98a2SYan Zheng if (split == 0) 39975d4f98a2SYan Zheng btrfs_cpu_key_to_disk(&disk_key, ins_key); 39985d4f98a2SYan Zheng else 39995d4f98a2SYan Zheng btrfs_item_key(l, &disk_key, mid); 40005d4f98a2SYan Zheng 40015d4f98a2SYan Zheng right = btrfs_alloc_free_block(trans, root, root->leafsize, 0, 400244871b1bSChris Mason root->root_key.objectid, 40035581a51aSJan Schmidt &disk_key, 0, l->start, 0); 4004f0486c68SYan, Zheng if (IS_ERR(right)) 400544871b1bSChris Mason return PTR_ERR(right); 4006f0486c68SYan, Zheng 4007f0486c68SYan, Zheng root_add_used(root, root->leafsize); 400844871b1bSChris Mason 400944871b1bSChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 401044871b1bSChris Mason btrfs_set_header_bytenr(right, right->start); 401144871b1bSChris Mason btrfs_set_header_generation(right, trans->transid); 40125d4f98a2SYan Zheng btrfs_set_header_backref_rev(right, BTRFS_MIXED_BACKREF_REV); 401344871b1bSChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 401444871b1bSChris Mason btrfs_set_header_level(right, 0); 401544871b1bSChris Mason write_extent_buffer(right, root->fs_info->fsid, 401644871b1bSChris Mason (unsigned long)btrfs_header_fsid(right), 401744871b1bSChris Mason BTRFS_FSID_SIZE); 401844871b1bSChris Mason 401944871b1bSChris Mason write_extent_buffer(right, root->fs_info->chunk_tree_uuid, 402044871b1bSChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(right), 402144871b1bSChris Mason BTRFS_UUID_SIZE); 402244871b1bSChris Mason 40235d4f98a2SYan Zheng if (split == 0) { 402444871b1bSChris Mason if (mid <= slot) { 402544871b1bSChris Mason btrfs_set_header_nritems(right, 0); 4026143bede5SJeff Mahoney insert_ptr(trans, root, path, &disk_key, right->start, 4027c3e06965SJan Schmidt path->slots[1] + 1, 1); 402844871b1bSChris Mason btrfs_tree_unlock(path->nodes[0]); 402944871b1bSChris Mason free_extent_buffer(path->nodes[0]); 403044871b1bSChris Mason path->nodes[0] = right; 403144871b1bSChris Mason path->slots[0] = 0; 403244871b1bSChris Mason path->slots[1] += 1; 403344871b1bSChris Mason } else { 403444871b1bSChris Mason btrfs_set_header_nritems(right, 0); 4035143bede5SJeff Mahoney insert_ptr(trans, root, path, &disk_key, right->start, 4036c3e06965SJan Schmidt path->slots[1], 1); 403744871b1bSChris Mason btrfs_tree_unlock(path->nodes[0]); 403844871b1bSChris Mason free_extent_buffer(path->nodes[0]); 403944871b1bSChris Mason path->nodes[0] = right; 404044871b1bSChris Mason path->slots[0] = 0; 4041143bede5SJeff Mahoney if (path->slots[1] == 0) 4042143bede5SJeff Mahoney fixup_low_keys(trans, root, path, 4043143bede5SJeff Mahoney &disk_key, 1); 40445d4f98a2SYan Zheng } 404544871b1bSChris Mason btrfs_mark_buffer_dirty(right); 404644871b1bSChris Mason return ret; 404744871b1bSChris Mason } 404844871b1bSChris Mason 4049143bede5SJeff Mahoney copy_for_split(trans, root, path, l, right, slot, mid, nritems); 405044871b1bSChris Mason 40515d4f98a2SYan Zheng if (split == 2) { 4052cc0c5538SChris Mason BUG_ON(num_doubles != 0); 4053cc0c5538SChris Mason num_doubles++; 4054cc0c5538SChris Mason goto again; 40553326d1b0SChris Mason } 405644871b1bSChris Mason 4057143bede5SJeff Mahoney return 0; 405899d8f83cSChris Mason 405999d8f83cSChris Mason push_for_double: 406099d8f83cSChris Mason push_for_double_split(trans, root, path, data_size); 406199d8f83cSChris Mason tried_avoid_double = 1; 406299d8f83cSChris Mason if (btrfs_leaf_free_space(root, path->nodes[0]) >= data_size) 406399d8f83cSChris Mason return 0; 406499d8f83cSChris Mason goto again; 4065be0e5c09SChris Mason } 4066be0e5c09SChris Mason 4067ad48fd75SYan, Zheng static noinline int setup_leaf_for_split(struct btrfs_trans_handle *trans, 4068ad48fd75SYan, Zheng struct btrfs_root *root, 4069ad48fd75SYan, Zheng struct btrfs_path *path, int ins_len) 4070ad48fd75SYan, Zheng { 4071ad48fd75SYan, Zheng struct btrfs_key key; 4072ad48fd75SYan, Zheng struct extent_buffer *leaf; 4073ad48fd75SYan, Zheng struct btrfs_file_extent_item *fi; 4074ad48fd75SYan, Zheng u64 extent_len = 0; 4075ad48fd75SYan, Zheng u32 item_size; 4076ad48fd75SYan, Zheng int ret; 4077ad48fd75SYan, Zheng 4078ad48fd75SYan, Zheng leaf = path->nodes[0]; 4079ad48fd75SYan, Zheng btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); 4080ad48fd75SYan, Zheng 4081ad48fd75SYan, Zheng BUG_ON(key.type != BTRFS_EXTENT_DATA_KEY && 4082ad48fd75SYan, Zheng key.type != BTRFS_EXTENT_CSUM_KEY); 4083ad48fd75SYan, Zheng 4084ad48fd75SYan, Zheng if (btrfs_leaf_free_space(root, leaf) >= ins_len) 4085ad48fd75SYan, Zheng return 0; 4086ad48fd75SYan, Zheng 4087ad48fd75SYan, Zheng item_size = btrfs_item_size_nr(leaf, path->slots[0]); 4088ad48fd75SYan, Zheng if (key.type == BTRFS_EXTENT_DATA_KEY) { 4089ad48fd75SYan, Zheng fi = btrfs_item_ptr(leaf, path->slots[0], 4090ad48fd75SYan, Zheng struct btrfs_file_extent_item); 4091ad48fd75SYan, Zheng extent_len = btrfs_file_extent_num_bytes(leaf, fi); 4092ad48fd75SYan, Zheng } 4093b3b4aa74SDavid Sterba btrfs_release_path(path); 4094ad48fd75SYan, Zheng 4095ad48fd75SYan, Zheng path->keep_locks = 1; 4096ad48fd75SYan, Zheng path->search_for_split = 1; 4097ad48fd75SYan, Zheng ret = btrfs_search_slot(trans, root, &key, path, 0, 1); 4098ad48fd75SYan, Zheng path->search_for_split = 0; 4099ad48fd75SYan, Zheng if (ret < 0) 4100ad48fd75SYan, Zheng goto err; 4101ad48fd75SYan, Zheng 4102ad48fd75SYan, Zheng ret = -EAGAIN; 4103ad48fd75SYan, Zheng leaf = path->nodes[0]; 4104ad48fd75SYan, Zheng /* if our item isn't there or got smaller, return now */ 4105ad48fd75SYan, Zheng if (ret > 0 || item_size != btrfs_item_size_nr(leaf, path->slots[0])) 4106ad48fd75SYan, Zheng goto err; 4107ad48fd75SYan, Zheng 4108109f6aefSChris Mason /* the leaf has changed, it now has room. return now */ 4109109f6aefSChris Mason if (btrfs_leaf_free_space(root, path->nodes[0]) >= ins_len) 4110109f6aefSChris Mason goto err; 4111109f6aefSChris Mason 4112ad48fd75SYan, Zheng if (key.type == BTRFS_EXTENT_DATA_KEY) { 4113ad48fd75SYan, Zheng fi = btrfs_item_ptr(leaf, path->slots[0], 4114ad48fd75SYan, Zheng struct btrfs_file_extent_item); 4115ad48fd75SYan, Zheng if (extent_len != btrfs_file_extent_num_bytes(leaf, fi)) 4116ad48fd75SYan, Zheng goto err; 4117ad48fd75SYan, Zheng } 4118ad48fd75SYan, Zheng 4119ad48fd75SYan, Zheng btrfs_set_path_blocking(path); 4120ad48fd75SYan, Zheng ret = split_leaf(trans, root, &key, path, ins_len, 1); 4121f0486c68SYan, Zheng if (ret) 4122f0486c68SYan, Zheng goto err; 4123ad48fd75SYan, Zheng 4124ad48fd75SYan, Zheng path->keep_locks = 0; 4125ad48fd75SYan, Zheng btrfs_unlock_up_safe(path, 1); 4126ad48fd75SYan, Zheng return 0; 4127ad48fd75SYan, Zheng err: 4128ad48fd75SYan, Zheng path->keep_locks = 0; 4129ad48fd75SYan, Zheng return ret; 4130ad48fd75SYan, Zheng } 4131ad48fd75SYan, Zheng 4132ad48fd75SYan, Zheng static noinline int split_item(struct btrfs_trans_handle *trans, 4133459931ecSChris Mason struct btrfs_root *root, 4134459931ecSChris Mason struct btrfs_path *path, 4135459931ecSChris Mason struct btrfs_key *new_key, 4136459931ecSChris Mason unsigned long split_offset) 4137459931ecSChris Mason { 4138459931ecSChris Mason struct extent_buffer *leaf; 4139459931ecSChris Mason struct btrfs_item *item; 4140459931ecSChris Mason struct btrfs_item *new_item; 4141459931ecSChris Mason int slot; 4142ad48fd75SYan, Zheng char *buf; 4143459931ecSChris Mason u32 nritems; 4144ad48fd75SYan, Zheng u32 item_size; 4145459931ecSChris Mason u32 orig_offset; 4146459931ecSChris Mason struct btrfs_disk_key disk_key; 4147459931ecSChris Mason 4148459931ecSChris Mason leaf = path->nodes[0]; 4149b9473439SChris Mason BUG_ON(btrfs_leaf_free_space(root, leaf) < sizeof(struct btrfs_item)); 4150b9473439SChris Mason 4151b4ce94deSChris Mason btrfs_set_path_blocking(path); 4152b4ce94deSChris Mason 4153459931ecSChris Mason item = btrfs_item_nr(leaf, path->slots[0]); 4154459931ecSChris Mason orig_offset = btrfs_item_offset(leaf, item); 4155459931ecSChris Mason item_size = btrfs_item_size(leaf, item); 4156459931ecSChris Mason 4157459931ecSChris Mason buf = kmalloc(item_size, GFP_NOFS); 4158ad48fd75SYan, Zheng if (!buf) 4159ad48fd75SYan, Zheng return -ENOMEM; 4160ad48fd75SYan, Zheng 4161459931ecSChris Mason read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf, 4162459931ecSChris Mason path->slots[0]), item_size); 4163ad48fd75SYan, Zheng 4164459931ecSChris Mason slot = path->slots[0] + 1; 4165459931ecSChris Mason nritems = btrfs_header_nritems(leaf); 4166459931ecSChris Mason if (slot != nritems) { 4167459931ecSChris Mason /* shift the items */ 4168459931ecSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1), 4169459931ecSChris Mason btrfs_item_nr_offset(slot), 4170459931ecSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 4171459931ecSChris Mason } 4172459931ecSChris Mason 4173459931ecSChris Mason btrfs_cpu_key_to_disk(&disk_key, new_key); 4174459931ecSChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 4175459931ecSChris Mason 4176459931ecSChris Mason new_item = btrfs_item_nr(leaf, slot); 4177459931ecSChris Mason 4178459931ecSChris Mason btrfs_set_item_offset(leaf, new_item, orig_offset); 4179459931ecSChris Mason btrfs_set_item_size(leaf, new_item, item_size - split_offset); 4180459931ecSChris Mason 4181459931ecSChris Mason btrfs_set_item_offset(leaf, item, 4182459931ecSChris Mason orig_offset + item_size - split_offset); 4183459931ecSChris Mason btrfs_set_item_size(leaf, item, split_offset); 4184459931ecSChris Mason 4185459931ecSChris Mason btrfs_set_header_nritems(leaf, nritems + 1); 4186459931ecSChris Mason 4187459931ecSChris Mason /* write the data for the start of the original item */ 4188459931ecSChris Mason write_extent_buffer(leaf, buf, 4189459931ecSChris Mason btrfs_item_ptr_offset(leaf, path->slots[0]), 4190459931ecSChris Mason split_offset); 4191459931ecSChris Mason 4192459931ecSChris Mason /* write the data for the new item */ 4193459931ecSChris Mason write_extent_buffer(leaf, buf + split_offset, 4194459931ecSChris Mason btrfs_item_ptr_offset(leaf, slot), 4195459931ecSChris Mason item_size - split_offset); 4196459931ecSChris Mason btrfs_mark_buffer_dirty(leaf); 4197459931ecSChris Mason 4198ad48fd75SYan, Zheng BUG_ON(btrfs_leaf_free_space(root, leaf) < 0); 4199459931ecSChris Mason kfree(buf); 4200ad48fd75SYan, Zheng return 0; 4201ad48fd75SYan, Zheng } 4202ad48fd75SYan, Zheng 4203ad48fd75SYan, Zheng /* 4204ad48fd75SYan, Zheng * This function splits a single item into two items, 4205ad48fd75SYan, Zheng * giving 'new_key' to the new item and splitting the 4206ad48fd75SYan, Zheng * old one at split_offset (from the start of the item). 4207ad48fd75SYan, Zheng * 4208ad48fd75SYan, Zheng * The path may be released by this operation. After 4209ad48fd75SYan, Zheng * the split, the path is pointing to the old item. The 4210ad48fd75SYan, Zheng * new item is going to be in the same node as the old one. 4211ad48fd75SYan, Zheng * 4212ad48fd75SYan, Zheng * Note, the item being split must be smaller enough to live alone on 4213ad48fd75SYan, Zheng * a tree block with room for one extra struct btrfs_item 4214ad48fd75SYan, Zheng * 4215ad48fd75SYan, Zheng * This allows us to split the item in place, keeping a lock on the 4216ad48fd75SYan, Zheng * leaf the entire time. 4217ad48fd75SYan, Zheng */ 4218ad48fd75SYan, Zheng int btrfs_split_item(struct btrfs_trans_handle *trans, 4219ad48fd75SYan, Zheng struct btrfs_root *root, 4220ad48fd75SYan, Zheng struct btrfs_path *path, 4221ad48fd75SYan, Zheng struct btrfs_key *new_key, 4222ad48fd75SYan, Zheng unsigned long split_offset) 4223ad48fd75SYan, Zheng { 4224ad48fd75SYan, Zheng int ret; 4225ad48fd75SYan, Zheng ret = setup_leaf_for_split(trans, root, path, 4226ad48fd75SYan, Zheng sizeof(struct btrfs_item)); 4227ad48fd75SYan, Zheng if (ret) 4228459931ecSChris Mason return ret; 4229ad48fd75SYan, Zheng 4230ad48fd75SYan, Zheng ret = split_item(trans, root, path, new_key, split_offset); 4231ad48fd75SYan, Zheng return ret; 4232ad48fd75SYan, Zheng } 4233ad48fd75SYan, Zheng 4234ad48fd75SYan, Zheng /* 4235ad48fd75SYan, Zheng * This function duplicate a item, giving 'new_key' to the new item. 4236ad48fd75SYan, Zheng * It guarantees both items live in the same tree leaf and the new item 4237ad48fd75SYan, Zheng * is contiguous with the original item. 4238ad48fd75SYan, Zheng * 4239ad48fd75SYan, Zheng * This allows us to split file extent in place, keeping a lock on the 4240ad48fd75SYan, Zheng * leaf the entire time. 4241ad48fd75SYan, Zheng */ 4242ad48fd75SYan, Zheng int btrfs_duplicate_item(struct btrfs_trans_handle *trans, 4243ad48fd75SYan, Zheng struct btrfs_root *root, 4244ad48fd75SYan, Zheng struct btrfs_path *path, 4245ad48fd75SYan, Zheng struct btrfs_key *new_key) 4246ad48fd75SYan, Zheng { 4247ad48fd75SYan, Zheng struct extent_buffer *leaf; 4248ad48fd75SYan, Zheng int ret; 4249ad48fd75SYan, Zheng u32 item_size; 4250ad48fd75SYan, Zheng 4251ad48fd75SYan, Zheng leaf = path->nodes[0]; 4252ad48fd75SYan, Zheng item_size = btrfs_item_size_nr(leaf, path->slots[0]); 4253ad48fd75SYan, Zheng ret = setup_leaf_for_split(trans, root, path, 4254ad48fd75SYan, Zheng item_size + sizeof(struct btrfs_item)); 4255ad48fd75SYan, Zheng if (ret) 4256ad48fd75SYan, Zheng return ret; 4257ad48fd75SYan, Zheng 4258ad48fd75SYan, Zheng path->slots[0]++; 4259143bede5SJeff Mahoney setup_items_for_insert(trans, root, path, new_key, &item_size, 4260ad48fd75SYan, Zheng item_size, item_size + 4261ad48fd75SYan, Zheng sizeof(struct btrfs_item), 1); 4262ad48fd75SYan, Zheng leaf = path->nodes[0]; 4263ad48fd75SYan, Zheng memcpy_extent_buffer(leaf, 4264ad48fd75SYan, Zheng btrfs_item_ptr_offset(leaf, path->slots[0]), 4265ad48fd75SYan, Zheng btrfs_item_ptr_offset(leaf, path->slots[0] - 1), 4266ad48fd75SYan, Zheng item_size); 4267ad48fd75SYan, Zheng return 0; 4268459931ecSChris Mason } 4269459931ecSChris Mason 4270459931ecSChris Mason /* 4271d352ac68SChris Mason * make the item pointed to by the path smaller. new_size indicates 4272d352ac68SChris Mason * how small to make it, and from_end tells us if we just chop bytes 4273d352ac68SChris Mason * off the end of the item or if we shift the item to chop bytes off 4274d352ac68SChris Mason * the front. 4275d352ac68SChris Mason */ 4276143bede5SJeff Mahoney void btrfs_truncate_item(struct btrfs_trans_handle *trans, 4277b18c6685SChris Mason struct btrfs_root *root, 4278b18c6685SChris Mason struct btrfs_path *path, 4279179e29e4SChris Mason u32 new_size, int from_end) 4280b18c6685SChris Mason { 4281b18c6685SChris Mason int slot; 42825f39d397SChris Mason struct extent_buffer *leaf; 42835f39d397SChris Mason struct btrfs_item *item; 4284b18c6685SChris Mason u32 nritems; 4285b18c6685SChris Mason unsigned int data_end; 4286b18c6685SChris Mason unsigned int old_data_start; 4287b18c6685SChris Mason unsigned int old_size; 4288b18c6685SChris Mason unsigned int size_diff; 4289b18c6685SChris Mason int i; 4290cfed81a0SChris Mason struct btrfs_map_token token; 4291cfed81a0SChris Mason 4292cfed81a0SChris Mason btrfs_init_map_token(&token); 4293b18c6685SChris Mason 42945f39d397SChris Mason leaf = path->nodes[0]; 4295179e29e4SChris Mason slot = path->slots[0]; 4296179e29e4SChris Mason 4297179e29e4SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 4298179e29e4SChris Mason if (old_size == new_size) 4299143bede5SJeff Mahoney return; 4300b18c6685SChris Mason 43015f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 4302b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 4303b18c6685SChris Mason 43045f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 4305179e29e4SChris Mason 4306b18c6685SChris Mason size_diff = old_size - new_size; 4307b18c6685SChris Mason 4308b18c6685SChris Mason BUG_ON(slot < 0); 4309b18c6685SChris Mason BUG_ON(slot >= nritems); 4310b18c6685SChris Mason 4311b18c6685SChris Mason /* 4312b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 4313b18c6685SChris Mason */ 4314b18c6685SChris Mason /* first correct the data pointers */ 4315b18c6685SChris Mason for (i = slot; i < nritems; i++) { 43165f39d397SChris Mason u32 ioff; 43175f39d397SChris Mason item = btrfs_item_nr(leaf, i); 4318db94535dSChris Mason 4319cfed81a0SChris Mason ioff = btrfs_token_item_offset(leaf, item, &token); 4320cfed81a0SChris Mason btrfs_set_token_item_offset(leaf, item, 4321cfed81a0SChris Mason ioff + size_diff, &token); 4322b18c6685SChris Mason } 4323db94535dSChris Mason 4324b18c6685SChris Mason /* shift the data */ 4325179e29e4SChris Mason if (from_end) { 43265f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 4327b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 4328b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 4329179e29e4SChris Mason } else { 4330179e29e4SChris Mason struct btrfs_disk_key disk_key; 4331179e29e4SChris Mason u64 offset; 4332179e29e4SChris Mason 4333179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 4334179e29e4SChris Mason 4335179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 4336179e29e4SChris Mason unsigned long ptr; 4337179e29e4SChris Mason struct btrfs_file_extent_item *fi; 4338179e29e4SChris Mason 4339179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 4340179e29e4SChris Mason struct btrfs_file_extent_item); 4341179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 4342179e29e4SChris Mason (unsigned long)fi - size_diff); 4343179e29e4SChris Mason 4344179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 4345179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 4346179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 4347179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 4348179e29e4SChris Mason (unsigned long)fi, 4349179e29e4SChris Mason offsetof(struct btrfs_file_extent_item, 4350179e29e4SChris Mason disk_bytenr)); 4351179e29e4SChris Mason } 4352179e29e4SChris Mason } 4353179e29e4SChris Mason 4354179e29e4SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 4355179e29e4SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 4356179e29e4SChris Mason data_end, old_data_start - data_end); 4357179e29e4SChris Mason 4358179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 4359179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 4360179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 4361179e29e4SChris Mason if (slot == 0) 4362179e29e4SChris Mason fixup_low_keys(trans, root, path, &disk_key, 1); 4363179e29e4SChris Mason } 43645f39d397SChris Mason 43655f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 43665f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 43675f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 4368b18c6685SChris Mason 43695f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 43705f39d397SChris Mason btrfs_print_leaf(root, leaf); 4371b18c6685SChris Mason BUG(); 43725f39d397SChris Mason } 4373b18c6685SChris Mason } 4374b18c6685SChris Mason 4375d352ac68SChris Mason /* 4376d352ac68SChris Mason * make the item pointed to by the path bigger, data_size is the new size. 4377d352ac68SChris Mason */ 4378143bede5SJeff Mahoney void btrfs_extend_item(struct btrfs_trans_handle *trans, 43795f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 43805f39d397SChris Mason u32 data_size) 43816567e837SChris Mason { 43826567e837SChris Mason int slot; 43835f39d397SChris Mason struct extent_buffer *leaf; 43845f39d397SChris Mason struct btrfs_item *item; 43856567e837SChris Mason u32 nritems; 43866567e837SChris Mason unsigned int data_end; 43876567e837SChris Mason unsigned int old_data; 43886567e837SChris Mason unsigned int old_size; 43896567e837SChris Mason int i; 4390cfed81a0SChris Mason struct btrfs_map_token token; 4391cfed81a0SChris Mason 4392cfed81a0SChris Mason btrfs_init_map_token(&token); 43936567e837SChris Mason 43945f39d397SChris Mason leaf = path->nodes[0]; 43956567e837SChris Mason 43965f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 43976567e837SChris Mason data_end = leaf_data_end(root, leaf); 43986567e837SChris Mason 43995f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 44005f39d397SChris Mason btrfs_print_leaf(root, leaf); 44016567e837SChris Mason BUG(); 44025f39d397SChris Mason } 44036567e837SChris Mason slot = path->slots[0]; 44045f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 44056567e837SChris Mason 44066567e837SChris Mason BUG_ON(slot < 0); 44073326d1b0SChris Mason if (slot >= nritems) { 44083326d1b0SChris Mason btrfs_print_leaf(root, leaf); 4409d397712bSChris Mason printk(KERN_CRIT "slot %d too large, nritems %d\n", 4410d397712bSChris Mason slot, nritems); 44113326d1b0SChris Mason BUG_ON(1); 44123326d1b0SChris Mason } 44136567e837SChris Mason 44146567e837SChris Mason /* 44156567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 44166567e837SChris Mason */ 44176567e837SChris Mason /* first correct the data pointers */ 44186567e837SChris Mason for (i = slot; i < nritems; i++) { 44195f39d397SChris Mason u32 ioff; 44205f39d397SChris Mason item = btrfs_item_nr(leaf, i); 4421db94535dSChris Mason 4422cfed81a0SChris Mason ioff = btrfs_token_item_offset(leaf, item, &token); 4423cfed81a0SChris Mason btrfs_set_token_item_offset(leaf, item, 4424cfed81a0SChris Mason ioff - data_size, &token); 44256567e837SChris Mason } 44265f39d397SChris Mason 44276567e837SChris Mason /* shift the data */ 44285f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 44296567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 44306567e837SChris Mason data_end, old_data - data_end); 44315f39d397SChris Mason 44326567e837SChris Mason data_end = old_data; 44335f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 44345f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 44355f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 44365f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 44376567e837SChris Mason 44385f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 44395f39d397SChris Mason btrfs_print_leaf(root, leaf); 44406567e837SChris Mason BUG(); 44415f39d397SChris Mason } 44426567e837SChris Mason } 44436567e837SChris Mason 444474123bd7SChris Mason /* 444544871b1bSChris Mason * this is a helper for btrfs_insert_empty_items, the main goal here is 444644871b1bSChris Mason * to save stack depth by doing the bulk of the work in a function 444744871b1bSChris Mason * that doesn't call btrfs_search_slot 444874123bd7SChris Mason */ 4449143bede5SJeff Mahoney void setup_items_for_insert(struct btrfs_trans_handle *trans, 445044871b1bSChris Mason struct btrfs_root *root, struct btrfs_path *path, 44519c58309dSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 445244871b1bSChris Mason u32 total_data, u32 total_size, int nr) 4453be0e5c09SChris Mason { 44545f39d397SChris Mason struct btrfs_item *item; 44559c58309dSChris Mason int i; 44567518a238SChris Mason u32 nritems; 4457be0e5c09SChris Mason unsigned int data_end; 4458e2fa7227SChris Mason struct btrfs_disk_key disk_key; 445944871b1bSChris Mason struct extent_buffer *leaf; 446044871b1bSChris Mason int slot; 4461cfed81a0SChris Mason struct btrfs_map_token token; 4462cfed81a0SChris Mason 4463cfed81a0SChris Mason btrfs_init_map_token(&token); 4464e2fa7227SChris Mason 44655f39d397SChris Mason leaf = path->nodes[0]; 446644871b1bSChris Mason slot = path->slots[0]; 446774123bd7SChris Mason 44685f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 4469123abc88SChris Mason data_end = leaf_data_end(root, leaf); 4470eb60ceacSChris Mason 4471f25956ccSChris Mason if (btrfs_leaf_free_space(root, leaf) < total_size) { 44723326d1b0SChris Mason btrfs_print_leaf(root, leaf); 4473d397712bSChris Mason printk(KERN_CRIT "not enough freespace need %u have %d\n", 44749c58309dSChris Mason total_size, btrfs_leaf_free_space(root, leaf)); 4475be0e5c09SChris Mason BUG(); 4476d4dbff95SChris Mason } 44775f39d397SChris Mason 4478be0e5c09SChris Mason if (slot != nritems) { 44795f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 4480be0e5c09SChris Mason 44815f39d397SChris Mason if (old_data < data_end) { 44825f39d397SChris Mason btrfs_print_leaf(root, leaf); 4483d397712bSChris Mason printk(KERN_CRIT "slot %d old_data %d data_end %d\n", 44845f39d397SChris Mason slot, old_data, data_end); 44855f39d397SChris Mason BUG_ON(1); 44865f39d397SChris Mason } 4487be0e5c09SChris Mason /* 4488be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 4489be0e5c09SChris Mason */ 4490be0e5c09SChris Mason /* first correct the data pointers */ 44910783fcfcSChris Mason for (i = slot; i < nritems; i++) { 44925f39d397SChris Mason u32 ioff; 4493db94535dSChris Mason 44945f39d397SChris Mason item = btrfs_item_nr(leaf, i); 4495cfed81a0SChris Mason ioff = btrfs_token_item_offset(leaf, item, &token); 4496cfed81a0SChris Mason btrfs_set_token_item_offset(leaf, item, 4497cfed81a0SChris Mason ioff - total_data, &token); 44980783fcfcSChris Mason } 4499be0e5c09SChris Mason /* shift the items */ 45009c58309dSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 45015f39d397SChris Mason btrfs_item_nr_offset(slot), 45020783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 4503be0e5c09SChris Mason 4504be0e5c09SChris Mason /* shift the data */ 45055f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 45069c58309dSChris Mason data_end - total_data, btrfs_leaf_data(leaf) + 4507be0e5c09SChris Mason data_end, old_data - data_end); 4508be0e5c09SChris Mason data_end = old_data; 4509be0e5c09SChris Mason } 45105f39d397SChris Mason 451162e2749eSChris Mason /* setup the item for the new data */ 45129c58309dSChris Mason for (i = 0; i < nr; i++) { 45139c58309dSChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 45149c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 45159c58309dSChris Mason item = btrfs_item_nr(leaf, slot + i); 4516cfed81a0SChris Mason btrfs_set_token_item_offset(leaf, item, 4517cfed81a0SChris Mason data_end - data_size[i], &token); 45189c58309dSChris Mason data_end -= data_size[i]; 4519cfed81a0SChris Mason btrfs_set_token_item_size(leaf, item, data_size[i], &token); 45209c58309dSChris Mason } 452144871b1bSChris Mason 45229c58309dSChris Mason btrfs_set_header_nritems(leaf, nritems + nr); 4523aa5d6bedSChris Mason 45245a01a2e3SChris Mason if (slot == 0) { 45255a01a2e3SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 4526143bede5SJeff Mahoney fixup_low_keys(trans, root, path, &disk_key, 1); 45275a01a2e3SChris Mason } 4528b9473439SChris Mason btrfs_unlock_up_safe(path, 1); 4529b9473439SChris Mason btrfs_mark_buffer_dirty(leaf); 4530aa5d6bedSChris Mason 45315f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 45325f39d397SChris Mason btrfs_print_leaf(root, leaf); 4533be0e5c09SChris Mason BUG(); 45345f39d397SChris Mason } 453544871b1bSChris Mason } 453644871b1bSChris Mason 453744871b1bSChris Mason /* 453844871b1bSChris Mason * Given a key and some data, insert items into the tree. 453944871b1bSChris Mason * This does all the path init required, making room in the tree if needed. 454044871b1bSChris Mason */ 454144871b1bSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 454244871b1bSChris Mason struct btrfs_root *root, 454344871b1bSChris Mason struct btrfs_path *path, 454444871b1bSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 454544871b1bSChris Mason int nr) 454644871b1bSChris Mason { 454744871b1bSChris Mason int ret = 0; 454844871b1bSChris Mason int slot; 454944871b1bSChris Mason int i; 455044871b1bSChris Mason u32 total_size = 0; 455144871b1bSChris Mason u32 total_data = 0; 455244871b1bSChris Mason 455344871b1bSChris Mason for (i = 0; i < nr; i++) 455444871b1bSChris Mason total_data += data_size[i]; 455544871b1bSChris Mason 455644871b1bSChris Mason total_size = total_data + (nr * sizeof(struct btrfs_item)); 455744871b1bSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 455844871b1bSChris Mason if (ret == 0) 455944871b1bSChris Mason return -EEXIST; 456044871b1bSChris Mason if (ret < 0) 4561143bede5SJeff Mahoney return ret; 456244871b1bSChris Mason 456344871b1bSChris Mason slot = path->slots[0]; 456444871b1bSChris Mason BUG_ON(slot < 0); 456544871b1bSChris Mason 4566143bede5SJeff Mahoney setup_items_for_insert(trans, root, path, cpu_key, data_size, 456744871b1bSChris Mason total_data, total_size, nr); 4568143bede5SJeff Mahoney return 0; 456962e2749eSChris Mason } 457062e2749eSChris Mason 457162e2749eSChris Mason /* 457262e2749eSChris Mason * Given a key and some data, insert an item into the tree. 457362e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 457462e2749eSChris Mason */ 4575e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 4576e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 4577e089f05cSChris Mason data_size) 457862e2749eSChris Mason { 457962e2749eSChris Mason int ret = 0; 45802c90e5d6SChris Mason struct btrfs_path *path; 45815f39d397SChris Mason struct extent_buffer *leaf; 45825f39d397SChris Mason unsigned long ptr; 458362e2749eSChris Mason 45842c90e5d6SChris Mason path = btrfs_alloc_path(); 4585db5b493aSTsutomu Itoh if (!path) 4586db5b493aSTsutomu Itoh return -ENOMEM; 45872c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 458862e2749eSChris Mason if (!ret) { 45895f39d397SChris Mason leaf = path->nodes[0]; 45905f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 45915f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 45925f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 459362e2749eSChris Mason } 45942c90e5d6SChris Mason btrfs_free_path(path); 4595aa5d6bedSChris Mason return ret; 4596be0e5c09SChris Mason } 4597be0e5c09SChris Mason 459874123bd7SChris Mason /* 45995de08d7dSChris Mason * delete the pointer from a given node. 460074123bd7SChris Mason * 4601d352ac68SChris Mason * the tree should have been previously balanced so the deletion does not 4602d352ac68SChris Mason * empty a node. 460374123bd7SChris Mason */ 4604143bede5SJeff Mahoney static void del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 4605f3ea38daSJan Schmidt struct btrfs_path *path, int level, int slot, 4606f3ea38daSJan Schmidt int tree_mod_log) 4607be0e5c09SChris Mason { 46085f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 46097518a238SChris Mason u32 nritems; 4610f3ea38daSJan Schmidt int ret; 4611be0e5c09SChris Mason 4612*6a7a665dSLiu Bo if (tree_mod_log && level) { 4613*6a7a665dSLiu Bo ret = tree_mod_log_insert_key(root->fs_info, parent, slot, 4614*6a7a665dSLiu Bo MOD_LOG_KEY_REMOVE); 4615*6a7a665dSLiu Bo BUG_ON(ret < 0); 4616*6a7a665dSLiu Bo } 4617*6a7a665dSLiu Bo 46185f39d397SChris Mason nritems = btrfs_header_nritems(parent); 4619be0e5c09SChris Mason if (slot != nritems - 1) { 4620f3ea38daSJan Schmidt if (tree_mod_log && level) 4621f3ea38daSJan Schmidt tree_mod_log_eb_move(root->fs_info, parent, slot, 4622f3ea38daSJan Schmidt slot + 1, nritems - slot - 1); 46235f39d397SChris Mason memmove_extent_buffer(parent, 46245f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 46255f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 4626d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 4627d6025579SChris Mason (nritems - slot - 1)); 4628be0e5c09SChris Mason } 4629f3ea38daSJan Schmidt 46307518a238SChris Mason nritems--; 46315f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 46327518a238SChris Mason if (nritems == 0 && parent == root->node) { 46335f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 4634eb60ceacSChris Mason /* just turn the root into a leaf and break */ 46355f39d397SChris Mason btrfs_set_header_level(root->node, 0); 4636bb803951SChris Mason } else if (slot == 0) { 46375f39d397SChris Mason struct btrfs_disk_key disk_key; 46385f39d397SChris Mason 46395f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 4640143bede5SJeff Mahoney fixup_low_keys(trans, root, path, &disk_key, level + 1); 4641be0e5c09SChris Mason } 4642d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 4643be0e5c09SChris Mason } 4644be0e5c09SChris Mason 464574123bd7SChris Mason /* 4646323ac95bSChris Mason * a helper function to delete the leaf pointed to by path->slots[1] and 46475d4f98a2SYan Zheng * path->nodes[1]. 4648323ac95bSChris Mason * 4649323ac95bSChris Mason * This deletes the pointer in path->nodes[1] and frees the leaf 4650323ac95bSChris Mason * block extent. zero is returned if it all worked out, < 0 otherwise. 4651323ac95bSChris Mason * 4652323ac95bSChris Mason * The path must have already been setup for deleting the leaf, including 4653323ac95bSChris Mason * all the proper balancing. path->nodes[1] must be locked. 4654323ac95bSChris Mason */ 4655143bede5SJeff Mahoney static noinline void btrfs_del_leaf(struct btrfs_trans_handle *trans, 4656323ac95bSChris Mason struct btrfs_root *root, 46575d4f98a2SYan Zheng struct btrfs_path *path, 46585d4f98a2SYan Zheng struct extent_buffer *leaf) 4659323ac95bSChris Mason { 46605d4f98a2SYan Zheng WARN_ON(btrfs_header_generation(leaf) != trans->transid); 4661f3ea38daSJan Schmidt del_ptr(trans, root, path, 1, path->slots[1], 1); 4662323ac95bSChris Mason 46634d081c41SChris Mason /* 46644d081c41SChris Mason * btrfs_free_extent is expensive, we want to make sure we 46654d081c41SChris Mason * aren't holding any locks when we call it 46664d081c41SChris Mason */ 46674d081c41SChris Mason btrfs_unlock_up_safe(path, 0); 46684d081c41SChris Mason 4669f0486c68SYan, Zheng root_sub_used(root, leaf->len); 4670f0486c68SYan, Zheng 46713083ee2eSJosef Bacik extent_buffer_get(leaf); 46725581a51aSJan Schmidt btrfs_free_tree_block(trans, root, leaf, 0, 1); 46733083ee2eSJosef Bacik free_extent_buffer_stale(leaf); 4674323ac95bSChris Mason } 4675323ac95bSChris Mason /* 467674123bd7SChris Mason * delete the item at the leaf level in path. If that empties 467774123bd7SChris Mason * the leaf, remove it from the tree 467874123bd7SChris Mason */ 467985e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 468085e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 4681be0e5c09SChris Mason { 46825f39d397SChris Mason struct extent_buffer *leaf; 46835f39d397SChris Mason struct btrfs_item *item; 468485e21bacSChris Mason int last_off; 468585e21bacSChris Mason int dsize = 0; 4686aa5d6bedSChris Mason int ret = 0; 4687aa5d6bedSChris Mason int wret; 468885e21bacSChris Mason int i; 46897518a238SChris Mason u32 nritems; 4690cfed81a0SChris Mason struct btrfs_map_token token; 4691cfed81a0SChris Mason 4692cfed81a0SChris Mason btrfs_init_map_token(&token); 4693be0e5c09SChris Mason 46945f39d397SChris Mason leaf = path->nodes[0]; 469585e21bacSChris Mason last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); 469685e21bacSChris Mason 469785e21bacSChris Mason for (i = 0; i < nr; i++) 469885e21bacSChris Mason dsize += btrfs_item_size_nr(leaf, slot + i); 469985e21bacSChris Mason 47005f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 4701be0e5c09SChris Mason 470285e21bacSChris Mason if (slot + nr != nritems) { 4703123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 47045f39d397SChris Mason 47055f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 4706d6025579SChris Mason data_end + dsize, 4707123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 470885e21bacSChris Mason last_off - data_end); 47095f39d397SChris Mason 471085e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 47115f39d397SChris Mason u32 ioff; 4712db94535dSChris Mason 47135f39d397SChris Mason item = btrfs_item_nr(leaf, i); 4714cfed81a0SChris Mason ioff = btrfs_token_item_offset(leaf, item, &token); 4715cfed81a0SChris Mason btrfs_set_token_item_offset(leaf, item, 4716cfed81a0SChris Mason ioff + dsize, &token); 47170783fcfcSChris Mason } 4718db94535dSChris Mason 47195f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 472085e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 47210783fcfcSChris Mason sizeof(struct btrfs_item) * 472285e21bacSChris Mason (nritems - slot - nr)); 4723be0e5c09SChris Mason } 472485e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 472585e21bacSChris Mason nritems -= nr; 47265f39d397SChris Mason 472774123bd7SChris Mason /* delete the leaf if we've emptied it */ 47287518a238SChris Mason if (nritems == 0) { 47295f39d397SChris Mason if (leaf == root->node) { 47305f39d397SChris Mason btrfs_set_header_level(leaf, 0); 47319a8dd150SChris Mason } else { 4732f0486c68SYan, Zheng btrfs_set_path_blocking(path); 4733f0486c68SYan, Zheng clean_tree_block(trans, root, leaf); 4734143bede5SJeff Mahoney btrfs_del_leaf(trans, root, path, leaf); 47359a8dd150SChris Mason } 4736be0e5c09SChris Mason } else { 47377518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 4738aa5d6bedSChris Mason if (slot == 0) { 47395f39d397SChris Mason struct btrfs_disk_key disk_key; 47405f39d397SChris Mason 47415f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 4742143bede5SJeff Mahoney fixup_low_keys(trans, root, path, &disk_key, 1); 4743aa5d6bedSChris Mason } 4744aa5d6bedSChris Mason 474574123bd7SChris Mason /* delete the leaf if it is mostly empty */ 4746d717aa1dSYan Zheng if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) { 4747be0e5c09SChris Mason /* push_leaf_left fixes the path. 4748be0e5c09SChris Mason * make sure the path still points to our leaf 4749be0e5c09SChris Mason * for possible call to del_ptr below 4750be0e5c09SChris Mason */ 47514920c9acSChris Mason slot = path->slots[1]; 47525f39d397SChris Mason extent_buffer_get(leaf); 47535f39d397SChris Mason 4754b9473439SChris Mason btrfs_set_path_blocking(path); 475599d8f83cSChris Mason wret = push_leaf_left(trans, root, path, 1, 1, 475699d8f83cSChris Mason 1, (u32)-1); 475754aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 4758aa5d6bedSChris Mason ret = wret; 47595f39d397SChris Mason 47605f39d397SChris Mason if (path->nodes[0] == leaf && 47615f39d397SChris Mason btrfs_header_nritems(leaf)) { 476299d8f83cSChris Mason wret = push_leaf_right(trans, root, path, 1, 476399d8f83cSChris Mason 1, 1, 0); 476454aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 4765aa5d6bedSChris Mason ret = wret; 4766aa5d6bedSChris Mason } 47675f39d397SChris Mason 47685f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 4769323ac95bSChris Mason path->slots[1] = slot; 4770143bede5SJeff Mahoney btrfs_del_leaf(trans, root, path, leaf); 47715f39d397SChris Mason free_extent_buffer(leaf); 4772143bede5SJeff Mahoney ret = 0; 47735de08d7dSChris Mason } else { 4774925baeddSChris Mason /* if we're still in the path, make sure 4775925baeddSChris Mason * we're dirty. Otherwise, one of the 4776925baeddSChris Mason * push_leaf functions must have already 4777925baeddSChris Mason * dirtied this buffer 4778925baeddSChris Mason */ 4779925baeddSChris Mason if (path->nodes[0] == leaf) 47805f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 47815f39d397SChris Mason free_extent_buffer(leaf); 4782be0e5c09SChris Mason } 4783d5719762SChris Mason } else { 47845f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 4785be0e5c09SChris Mason } 47869a8dd150SChris Mason } 4787aa5d6bedSChris Mason return ret; 47889a8dd150SChris Mason } 47899a8dd150SChris Mason 479097571fd0SChris Mason /* 4791925baeddSChris Mason * search the tree again to find a leaf with lesser keys 47927bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 47937bb86316SChris Mason * returns < 0 on io errors. 4794d352ac68SChris Mason * 4795d352ac68SChris Mason * This may release the path, and so you may lose any locks held at the 4796d352ac68SChris Mason * time you call it. 47977bb86316SChris Mason */ 47987bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 47997bb86316SChris Mason { 4800925baeddSChris Mason struct btrfs_key key; 4801925baeddSChris Mason struct btrfs_disk_key found_key; 4802925baeddSChris Mason int ret; 48037bb86316SChris Mason 4804925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, 0); 4805925baeddSChris Mason 4806925baeddSChris Mason if (key.offset > 0) 4807925baeddSChris Mason key.offset--; 4808925baeddSChris Mason else if (key.type > 0) 4809925baeddSChris Mason key.type--; 4810925baeddSChris Mason else if (key.objectid > 0) 4811925baeddSChris Mason key.objectid--; 4812925baeddSChris Mason else 48137bb86316SChris Mason return 1; 48147bb86316SChris Mason 4815b3b4aa74SDavid Sterba btrfs_release_path(path); 4816925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 4817925baeddSChris Mason if (ret < 0) 4818925baeddSChris Mason return ret; 4819925baeddSChris Mason btrfs_item_key(path->nodes[0], &found_key, 0); 4820925baeddSChris Mason ret = comp_keys(&found_key, &key); 4821925baeddSChris Mason if (ret < 0) 48227bb86316SChris Mason return 0; 4823925baeddSChris Mason return 1; 48247bb86316SChris Mason } 48257bb86316SChris Mason 48263f157a2fSChris Mason /* 48273f157a2fSChris Mason * A helper function to walk down the tree starting at min_key, and looking 48283f157a2fSChris Mason * for nodes or leaves that are either in cache or have a minimum 4829d352ac68SChris Mason * transaction id. This is used by the btree defrag code, and tree logging 48303f157a2fSChris Mason * 48313f157a2fSChris Mason * This does not cow, but it does stuff the starting key it finds back 48323f157a2fSChris Mason * into min_key, so you can call btrfs_search_slot with cow=1 on the 48333f157a2fSChris Mason * key and get a writable path. 48343f157a2fSChris Mason * 48353f157a2fSChris Mason * This does lock as it descends, and path->keep_locks should be set 48363f157a2fSChris Mason * to 1 by the caller. 48373f157a2fSChris Mason * 48383f157a2fSChris Mason * This honors path->lowest_level to prevent descent past a given level 48393f157a2fSChris Mason * of the tree. 48403f157a2fSChris Mason * 4841d352ac68SChris Mason * min_trans indicates the oldest transaction that you are interested 4842d352ac68SChris Mason * in walking through. Any nodes or leaves older than min_trans are 4843d352ac68SChris Mason * skipped over (without reading them). 4844d352ac68SChris Mason * 48453f157a2fSChris Mason * returns zero if something useful was found, < 0 on error and 1 if there 48463f157a2fSChris Mason * was nothing in the tree that matched the search criteria. 48473f157a2fSChris Mason */ 48483f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key, 4849e02119d5SChris Mason struct btrfs_key *max_key, 48503f157a2fSChris Mason struct btrfs_path *path, int cache_only, 48513f157a2fSChris Mason u64 min_trans) 48523f157a2fSChris Mason { 48533f157a2fSChris Mason struct extent_buffer *cur; 48543f157a2fSChris Mason struct btrfs_key found_key; 48553f157a2fSChris Mason int slot; 48569652480bSYan int sret; 48573f157a2fSChris Mason u32 nritems; 48583f157a2fSChris Mason int level; 48593f157a2fSChris Mason int ret = 1; 48603f157a2fSChris Mason 4861934d375bSChris Mason WARN_ON(!path->keep_locks); 48623f157a2fSChris Mason again: 4863bd681513SChris Mason cur = btrfs_read_lock_root_node(root); 48643f157a2fSChris Mason level = btrfs_header_level(cur); 4865e02119d5SChris Mason WARN_ON(path->nodes[level]); 48663f157a2fSChris Mason path->nodes[level] = cur; 4867bd681513SChris Mason path->locks[level] = BTRFS_READ_LOCK; 48683f157a2fSChris Mason 48693f157a2fSChris Mason if (btrfs_header_generation(cur) < min_trans) { 48703f157a2fSChris Mason ret = 1; 48713f157a2fSChris Mason goto out; 48723f157a2fSChris Mason } 48733f157a2fSChris Mason while (1) { 48743f157a2fSChris Mason nritems = btrfs_header_nritems(cur); 48753f157a2fSChris Mason level = btrfs_header_level(cur); 48769652480bSYan sret = bin_search(cur, min_key, level, &slot); 48773f157a2fSChris Mason 4878323ac95bSChris Mason /* at the lowest level, we're done, setup the path and exit */ 4879323ac95bSChris Mason if (level == path->lowest_level) { 4880e02119d5SChris Mason if (slot >= nritems) 4881e02119d5SChris Mason goto find_next_key; 48823f157a2fSChris Mason ret = 0; 48833f157a2fSChris Mason path->slots[level] = slot; 48843f157a2fSChris Mason btrfs_item_key_to_cpu(cur, &found_key, slot); 48853f157a2fSChris Mason goto out; 48863f157a2fSChris Mason } 48879652480bSYan if (sret && slot > 0) 48889652480bSYan slot--; 48893f157a2fSChris Mason /* 48903f157a2fSChris Mason * check this node pointer against the cache_only and 48913f157a2fSChris Mason * min_trans parameters. If it isn't in cache or is too 48923f157a2fSChris Mason * old, skip to the next one. 48933f157a2fSChris Mason */ 48943f157a2fSChris Mason while (slot < nritems) { 48953f157a2fSChris Mason u64 blockptr; 48963f157a2fSChris Mason u64 gen; 48973f157a2fSChris Mason struct extent_buffer *tmp; 4898e02119d5SChris Mason struct btrfs_disk_key disk_key; 4899e02119d5SChris Mason 49003f157a2fSChris Mason blockptr = btrfs_node_blockptr(cur, slot); 49013f157a2fSChris Mason gen = btrfs_node_ptr_generation(cur, slot); 49023f157a2fSChris Mason if (gen < min_trans) { 49033f157a2fSChris Mason slot++; 49043f157a2fSChris Mason continue; 49053f157a2fSChris Mason } 49063f157a2fSChris Mason if (!cache_only) 49073f157a2fSChris Mason break; 49083f157a2fSChris Mason 4909e02119d5SChris Mason if (max_key) { 4910e02119d5SChris Mason btrfs_node_key(cur, &disk_key, slot); 4911e02119d5SChris Mason if (comp_keys(&disk_key, max_key) >= 0) { 4912e02119d5SChris Mason ret = 1; 4913e02119d5SChris Mason goto out; 4914e02119d5SChris Mason } 4915e02119d5SChris Mason } 4916e02119d5SChris Mason 49173f157a2fSChris Mason tmp = btrfs_find_tree_block(root, blockptr, 49183f157a2fSChris Mason btrfs_level_size(root, level - 1)); 49193f157a2fSChris Mason 4920b9fab919SChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen, 1) > 0) { 49213f157a2fSChris Mason free_extent_buffer(tmp); 49223f157a2fSChris Mason break; 49233f157a2fSChris Mason } 49243f157a2fSChris Mason if (tmp) 49253f157a2fSChris Mason free_extent_buffer(tmp); 49263f157a2fSChris Mason slot++; 49273f157a2fSChris Mason } 4928e02119d5SChris Mason find_next_key: 49293f157a2fSChris Mason /* 49303f157a2fSChris Mason * we didn't find a candidate key in this node, walk forward 49313f157a2fSChris Mason * and find another one 49323f157a2fSChris Mason */ 49333f157a2fSChris Mason if (slot >= nritems) { 4934e02119d5SChris Mason path->slots[level] = slot; 4935b4ce94deSChris Mason btrfs_set_path_blocking(path); 4936e02119d5SChris Mason sret = btrfs_find_next_key(root, path, min_key, level, 49373f157a2fSChris Mason cache_only, min_trans); 4938e02119d5SChris Mason if (sret == 0) { 4939b3b4aa74SDavid Sterba btrfs_release_path(path); 49403f157a2fSChris Mason goto again; 49413f157a2fSChris Mason } else { 49423f157a2fSChris Mason goto out; 49433f157a2fSChris Mason } 49443f157a2fSChris Mason } 49453f157a2fSChris Mason /* save our key for returning back */ 49463f157a2fSChris Mason btrfs_node_key_to_cpu(cur, &found_key, slot); 49473f157a2fSChris Mason path->slots[level] = slot; 49483f157a2fSChris Mason if (level == path->lowest_level) { 49493f157a2fSChris Mason ret = 0; 4950f7c79f30SChris Mason unlock_up(path, level, 1, 0, NULL); 49513f157a2fSChris Mason goto out; 49523f157a2fSChris Mason } 4953b4ce94deSChris Mason btrfs_set_path_blocking(path); 49543f157a2fSChris Mason cur = read_node_slot(root, cur, slot); 495579787eaaSJeff Mahoney BUG_ON(!cur); /* -ENOMEM */ 49563f157a2fSChris Mason 4957bd681513SChris Mason btrfs_tree_read_lock(cur); 4958b4ce94deSChris Mason 4959bd681513SChris Mason path->locks[level - 1] = BTRFS_READ_LOCK; 49603f157a2fSChris Mason path->nodes[level - 1] = cur; 4961f7c79f30SChris Mason unlock_up(path, level, 1, 0, NULL); 4962bd681513SChris Mason btrfs_clear_path_blocking(path, NULL, 0); 49633f157a2fSChris Mason } 49643f157a2fSChris Mason out: 49653f157a2fSChris Mason if (ret == 0) 49663f157a2fSChris Mason memcpy(min_key, &found_key, sizeof(found_key)); 4967b4ce94deSChris Mason btrfs_set_path_blocking(path); 49683f157a2fSChris Mason return ret; 49693f157a2fSChris Mason } 49703f157a2fSChris Mason 49717069830aSAlexander Block static void tree_move_down(struct btrfs_root *root, 49727069830aSAlexander Block struct btrfs_path *path, 49737069830aSAlexander Block int *level, int root_level) 49747069830aSAlexander Block { 497574dd17fbSChris Mason BUG_ON(*level == 0); 49767069830aSAlexander Block path->nodes[*level - 1] = read_node_slot(root, path->nodes[*level], 49777069830aSAlexander Block path->slots[*level]); 49787069830aSAlexander Block path->slots[*level - 1] = 0; 49797069830aSAlexander Block (*level)--; 49807069830aSAlexander Block } 49817069830aSAlexander Block 49827069830aSAlexander Block static int tree_move_next_or_upnext(struct btrfs_root *root, 49837069830aSAlexander Block struct btrfs_path *path, 49847069830aSAlexander Block int *level, int root_level) 49857069830aSAlexander Block { 49867069830aSAlexander Block int ret = 0; 49877069830aSAlexander Block int nritems; 49887069830aSAlexander Block nritems = btrfs_header_nritems(path->nodes[*level]); 49897069830aSAlexander Block 49907069830aSAlexander Block path->slots[*level]++; 49917069830aSAlexander Block 499274dd17fbSChris Mason while (path->slots[*level] >= nritems) { 49937069830aSAlexander Block if (*level == root_level) 49947069830aSAlexander Block return -1; 49957069830aSAlexander Block 49967069830aSAlexander Block /* move upnext */ 49977069830aSAlexander Block path->slots[*level] = 0; 49987069830aSAlexander Block free_extent_buffer(path->nodes[*level]); 49997069830aSAlexander Block path->nodes[*level] = NULL; 50007069830aSAlexander Block (*level)++; 50017069830aSAlexander Block path->slots[*level]++; 50027069830aSAlexander Block 50037069830aSAlexander Block nritems = btrfs_header_nritems(path->nodes[*level]); 50047069830aSAlexander Block ret = 1; 50057069830aSAlexander Block } 50067069830aSAlexander Block return ret; 50077069830aSAlexander Block } 50087069830aSAlexander Block 50097069830aSAlexander Block /* 50107069830aSAlexander Block * Returns 1 if it had to move up and next. 0 is returned if it moved only next 50117069830aSAlexander Block * or down. 50127069830aSAlexander Block */ 50137069830aSAlexander Block static int tree_advance(struct btrfs_root *root, 50147069830aSAlexander Block struct btrfs_path *path, 50157069830aSAlexander Block int *level, int root_level, 50167069830aSAlexander Block int allow_down, 50177069830aSAlexander Block struct btrfs_key *key) 50187069830aSAlexander Block { 50197069830aSAlexander Block int ret; 50207069830aSAlexander Block 50217069830aSAlexander Block if (*level == 0 || !allow_down) { 50227069830aSAlexander Block ret = tree_move_next_or_upnext(root, path, level, root_level); 50237069830aSAlexander Block } else { 50247069830aSAlexander Block tree_move_down(root, path, level, root_level); 50257069830aSAlexander Block ret = 0; 50267069830aSAlexander Block } 50277069830aSAlexander Block if (ret >= 0) { 50287069830aSAlexander Block if (*level == 0) 50297069830aSAlexander Block btrfs_item_key_to_cpu(path->nodes[*level], key, 50307069830aSAlexander Block path->slots[*level]); 50317069830aSAlexander Block else 50327069830aSAlexander Block btrfs_node_key_to_cpu(path->nodes[*level], key, 50337069830aSAlexander Block path->slots[*level]); 50347069830aSAlexander Block } 50357069830aSAlexander Block return ret; 50367069830aSAlexander Block } 50377069830aSAlexander Block 50387069830aSAlexander Block static int tree_compare_item(struct btrfs_root *left_root, 50397069830aSAlexander Block struct btrfs_path *left_path, 50407069830aSAlexander Block struct btrfs_path *right_path, 50417069830aSAlexander Block char *tmp_buf) 50427069830aSAlexander Block { 50437069830aSAlexander Block int cmp; 50447069830aSAlexander Block int len1, len2; 50457069830aSAlexander Block unsigned long off1, off2; 50467069830aSAlexander Block 50477069830aSAlexander Block len1 = btrfs_item_size_nr(left_path->nodes[0], left_path->slots[0]); 50487069830aSAlexander Block len2 = btrfs_item_size_nr(right_path->nodes[0], right_path->slots[0]); 50497069830aSAlexander Block if (len1 != len2) 50507069830aSAlexander Block return 1; 50517069830aSAlexander Block 50527069830aSAlexander Block off1 = btrfs_item_ptr_offset(left_path->nodes[0], left_path->slots[0]); 50537069830aSAlexander Block off2 = btrfs_item_ptr_offset(right_path->nodes[0], 50547069830aSAlexander Block right_path->slots[0]); 50557069830aSAlexander Block 50567069830aSAlexander Block read_extent_buffer(left_path->nodes[0], tmp_buf, off1, len1); 50577069830aSAlexander Block 50587069830aSAlexander Block cmp = memcmp_extent_buffer(right_path->nodes[0], tmp_buf, off2, len1); 50597069830aSAlexander Block if (cmp) 50607069830aSAlexander Block return 1; 50617069830aSAlexander Block return 0; 50627069830aSAlexander Block } 50637069830aSAlexander Block 50647069830aSAlexander Block #define ADVANCE 1 50657069830aSAlexander Block #define ADVANCE_ONLY_NEXT -1 50667069830aSAlexander Block 50677069830aSAlexander Block /* 50687069830aSAlexander Block * This function compares two trees and calls the provided callback for 50697069830aSAlexander Block * every changed/new/deleted item it finds. 50707069830aSAlexander Block * If shared tree blocks are encountered, whole subtrees are skipped, making 50717069830aSAlexander Block * the compare pretty fast on snapshotted subvolumes. 50727069830aSAlexander Block * 50737069830aSAlexander Block * This currently works on commit roots only. As commit roots are read only, 50747069830aSAlexander Block * we don't do any locking. The commit roots are protected with transactions. 50757069830aSAlexander Block * Transactions are ended and rejoined when a commit is tried in between. 50767069830aSAlexander Block * 50777069830aSAlexander Block * This function checks for modifications done to the trees while comparing. 50787069830aSAlexander Block * If it detects a change, it aborts immediately. 50797069830aSAlexander Block */ 50807069830aSAlexander Block int btrfs_compare_trees(struct btrfs_root *left_root, 50817069830aSAlexander Block struct btrfs_root *right_root, 50827069830aSAlexander Block btrfs_changed_cb_t changed_cb, void *ctx) 50837069830aSAlexander Block { 50847069830aSAlexander Block int ret; 50857069830aSAlexander Block int cmp; 50867069830aSAlexander Block struct btrfs_trans_handle *trans = NULL; 50877069830aSAlexander Block struct btrfs_path *left_path = NULL; 50887069830aSAlexander Block struct btrfs_path *right_path = NULL; 50897069830aSAlexander Block struct btrfs_key left_key; 50907069830aSAlexander Block struct btrfs_key right_key; 50917069830aSAlexander Block char *tmp_buf = NULL; 50927069830aSAlexander Block int left_root_level; 50937069830aSAlexander Block int right_root_level; 50947069830aSAlexander Block int left_level; 50957069830aSAlexander Block int right_level; 50967069830aSAlexander Block int left_end_reached; 50977069830aSAlexander Block int right_end_reached; 50987069830aSAlexander Block int advance_left; 50997069830aSAlexander Block int advance_right; 51007069830aSAlexander Block u64 left_blockptr; 51017069830aSAlexander Block u64 right_blockptr; 51027069830aSAlexander Block u64 left_start_ctransid; 51037069830aSAlexander Block u64 right_start_ctransid; 51047069830aSAlexander Block u64 ctransid; 51057069830aSAlexander Block 51067069830aSAlexander Block left_path = btrfs_alloc_path(); 51077069830aSAlexander Block if (!left_path) { 51087069830aSAlexander Block ret = -ENOMEM; 51097069830aSAlexander Block goto out; 51107069830aSAlexander Block } 51117069830aSAlexander Block right_path = btrfs_alloc_path(); 51127069830aSAlexander Block if (!right_path) { 51137069830aSAlexander Block ret = -ENOMEM; 51147069830aSAlexander Block goto out; 51157069830aSAlexander Block } 51167069830aSAlexander Block 51177069830aSAlexander Block tmp_buf = kmalloc(left_root->leafsize, GFP_NOFS); 51187069830aSAlexander Block if (!tmp_buf) { 51197069830aSAlexander Block ret = -ENOMEM; 51207069830aSAlexander Block goto out; 51217069830aSAlexander Block } 51227069830aSAlexander Block 51237069830aSAlexander Block left_path->search_commit_root = 1; 51247069830aSAlexander Block left_path->skip_locking = 1; 51257069830aSAlexander Block right_path->search_commit_root = 1; 51267069830aSAlexander Block right_path->skip_locking = 1; 51277069830aSAlexander Block 51287069830aSAlexander Block spin_lock(&left_root->root_times_lock); 51297069830aSAlexander Block left_start_ctransid = btrfs_root_ctransid(&left_root->root_item); 51307069830aSAlexander Block spin_unlock(&left_root->root_times_lock); 51317069830aSAlexander Block 51327069830aSAlexander Block spin_lock(&right_root->root_times_lock); 51337069830aSAlexander Block right_start_ctransid = btrfs_root_ctransid(&right_root->root_item); 51347069830aSAlexander Block spin_unlock(&right_root->root_times_lock); 51357069830aSAlexander Block 51367069830aSAlexander Block trans = btrfs_join_transaction(left_root); 51377069830aSAlexander Block if (IS_ERR(trans)) { 51387069830aSAlexander Block ret = PTR_ERR(trans); 51397069830aSAlexander Block trans = NULL; 51407069830aSAlexander Block goto out; 51417069830aSAlexander Block } 51427069830aSAlexander Block 51437069830aSAlexander Block /* 51447069830aSAlexander Block * Strategy: Go to the first items of both trees. Then do 51457069830aSAlexander Block * 51467069830aSAlexander Block * If both trees are at level 0 51477069830aSAlexander Block * Compare keys of current items 51487069830aSAlexander Block * If left < right treat left item as new, advance left tree 51497069830aSAlexander Block * and repeat 51507069830aSAlexander Block * If left > right treat right item as deleted, advance right tree 51517069830aSAlexander Block * and repeat 51527069830aSAlexander Block * If left == right do deep compare of items, treat as changed if 51537069830aSAlexander Block * needed, advance both trees and repeat 51547069830aSAlexander Block * If both trees are at the same level but not at level 0 51557069830aSAlexander Block * Compare keys of current nodes/leafs 51567069830aSAlexander Block * If left < right advance left tree and repeat 51577069830aSAlexander Block * If left > right advance right tree and repeat 51587069830aSAlexander Block * If left == right compare blockptrs of the next nodes/leafs 51597069830aSAlexander Block * If they match advance both trees but stay at the same level 51607069830aSAlexander Block * and repeat 51617069830aSAlexander Block * If they don't match advance both trees while allowing to go 51627069830aSAlexander Block * deeper and repeat 51637069830aSAlexander Block * If tree levels are different 51647069830aSAlexander Block * Advance the tree that needs it and repeat 51657069830aSAlexander Block * 51667069830aSAlexander Block * Advancing a tree means: 51677069830aSAlexander Block * If we are at level 0, try to go to the next slot. If that's not 51687069830aSAlexander Block * possible, go one level up and repeat. Stop when we found a level 51697069830aSAlexander Block * where we could go to the next slot. We may at this point be on a 51707069830aSAlexander Block * node or a leaf. 51717069830aSAlexander Block * 51727069830aSAlexander Block * If we are not at level 0 and not on shared tree blocks, go one 51737069830aSAlexander Block * level deeper. 51747069830aSAlexander Block * 51757069830aSAlexander Block * If we are not at level 0 and on shared tree blocks, go one slot to 51767069830aSAlexander Block * the right if possible or go up and right. 51777069830aSAlexander Block */ 51787069830aSAlexander Block 51797069830aSAlexander Block left_level = btrfs_header_level(left_root->commit_root); 51807069830aSAlexander Block left_root_level = left_level; 51817069830aSAlexander Block left_path->nodes[left_level] = left_root->commit_root; 51827069830aSAlexander Block extent_buffer_get(left_path->nodes[left_level]); 51837069830aSAlexander Block 51847069830aSAlexander Block right_level = btrfs_header_level(right_root->commit_root); 51857069830aSAlexander Block right_root_level = right_level; 51867069830aSAlexander Block right_path->nodes[right_level] = right_root->commit_root; 51877069830aSAlexander Block extent_buffer_get(right_path->nodes[right_level]); 51887069830aSAlexander Block 51897069830aSAlexander Block if (left_level == 0) 51907069830aSAlexander Block btrfs_item_key_to_cpu(left_path->nodes[left_level], 51917069830aSAlexander Block &left_key, left_path->slots[left_level]); 51927069830aSAlexander Block else 51937069830aSAlexander Block btrfs_node_key_to_cpu(left_path->nodes[left_level], 51947069830aSAlexander Block &left_key, left_path->slots[left_level]); 51957069830aSAlexander Block if (right_level == 0) 51967069830aSAlexander Block btrfs_item_key_to_cpu(right_path->nodes[right_level], 51977069830aSAlexander Block &right_key, right_path->slots[right_level]); 51987069830aSAlexander Block else 51997069830aSAlexander Block btrfs_node_key_to_cpu(right_path->nodes[right_level], 52007069830aSAlexander Block &right_key, right_path->slots[right_level]); 52017069830aSAlexander Block 52027069830aSAlexander Block left_end_reached = right_end_reached = 0; 52037069830aSAlexander Block advance_left = advance_right = 0; 52047069830aSAlexander Block 52057069830aSAlexander Block while (1) { 52067069830aSAlexander Block /* 52077069830aSAlexander Block * We need to make sure the transaction does not get committed 52087069830aSAlexander Block * while we do anything on commit roots. This means, we need to 52097069830aSAlexander Block * join and leave transactions for every item that we process. 52107069830aSAlexander Block */ 52117069830aSAlexander Block if (trans && btrfs_should_end_transaction(trans, left_root)) { 52127069830aSAlexander Block btrfs_release_path(left_path); 52137069830aSAlexander Block btrfs_release_path(right_path); 52147069830aSAlexander Block 52157069830aSAlexander Block ret = btrfs_end_transaction(trans, left_root); 52167069830aSAlexander Block trans = NULL; 52177069830aSAlexander Block if (ret < 0) 52187069830aSAlexander Block goto out; 52197069830aSAlexander Block } 52207069830aSAlexander Block /* now rejoin the transaction */ 52217069830aSAlexander Block if (!trans) { 52227069830aSAlexander Block trans = btrfs_join_transaction(left_root); 52237069830aSAlexander Block if (IS_ERR(trans)) { 52247069830aSAlexander Block ret = PTR_ERR(trans); 52257069830aSAlexander Block trans = NULL; 52267069830aSAlexander Block goto out; 52277069830aSAlexander Block } 52287069830aSAlexander Block 52297069830aSAlexander Block spin_lock(&left_root->root_times_lock); 52307069830aSAlexander Block ctransid = btrfs_root_ctransid(&left_root->root_item); 52317069830aSAlexander Block spin_unlock(&left_root->root_times_lock); 52327069830aSAlexander Block if (ctransid != left_start_ctransid) 52337069830aSAlexander Block left_start_ctransid = 0; 52347069830aSAlexander Block 52357069830aSAlexander Block spin_lock(&right_root->root_times_lock); 52367069830aSAlexander Block ctransid = btrfs_root_ctransid(&right_root->root_item); 52377069830aSAlexander Block spin_unlock(&right_root->root_times_lock); 52387069830aSAlexander Block if (ctransid != right_start_ctransid) 52397069830aSAlexander Block right_start_ctransid = 0; 52407069830aSAlexander Block 52417069830aSAlexander Block if (!left_start_ctransid || !right_start_ctransid) { 52427069830aSAlexander Block WARN(1, KERN_WARNING 52437069830aSAlexander Block "btrfs: btrfs_compare_tree detected " 52447069830aSAlexander Block "a change in one of the trees while " 52457069830aSAlexander Block "iterating. This is probably a " 52467069830aSAlexander Block "bug.\n"); 52477069830aSAlexander Block ret = -EIO; 52487069830aSAlexander Block goto out; 52497069830aSAlexander Block } 52507069830aSAlexander Block 52517069830aSAlexander Block /* 52527069830aSAlexander Block * the commit root may have changed, so start again 52537069830aSAlexander Block * where we stopped 52547069830aSAlexander Block */ 52557069830aSAlexander Block left_path->lowest_level = left_level; 52567069830aSAlexander Block right_path->lowest_level = right_level; 52577069830aSAlexander Block ret = btrfs_search_slot(NULL, left_root, 52587069830aSAlexander Block &left_key, left_path, 0, 0); 52597069830aSAlexander Block if (ret < 0) 52607069830aSAlexander Block goto out; 52617069830aSAlexander Block ret = btrfs_search_slot(NULL, right_root, 52627069830aSAlexander Block &right_key, right_path, 0, 0); 52637069830aSAlexander Block if (ret < 0) 52647069830aSAlexander Block goto out; 52657069830aSAlexander Block } 52667069830aSAlexander Block 52677069830aSAlexander Block if (advance_left && !left_end_reached) { 52687069830aSAlexander Block ret = tree_advance(left_root, left_path, &left_level, 52697069830aSAlexander Block left_root_level, 52707069830aSAlexander Block advance_left != ADVANCE_ONLY_NEXT, 52717069830aSAlexander Block &left_key); 52727069830aSAlexander Block if (ret < 0) 52737069830aSAlexander Block left_end_reached = ADVANCE; 52747069830aSAlexander Block advance_left = 0; 52757069830aSAlexander Block } 52767069830aSAlexander Block if (advance_right && !right_end_reached) { 52777069830aSAlexander Block ret = tree_advance(right_root, right_path, &right_level, 52787069830aSAlexander Block right_root_level, 52797069830aSAlexander Block advance_right != ADVANCE_ONLY_NEXT, 52807069830aSAlexander Block &right_key); 52817069830aSAlexander Block if (ret < 0) 52827069830aSAlexander Block right_end_reached = ADVANCE; 52837069830aSAlexander Block advance_right = 0; 52847069830aSAlexander Block } 52857069830aSAlexander Block 52867069830aSAlexander Block if (left_end_reached && right_end_reached) { 52877069830aSAlexander Block ret = 0; 52887069830aSAlexander Block goto out; 52897069830aSAlexander Block } else if (left_end_reached) { 52907069830aSAlexander Block if (right_level == 0) { 52917069830aSAlexander Block ret = changed_cb(left_root, right_root, 52927069830aSAlexander Block left_path, right_path, 52937069830aSAlexander Block &right_key, 52947069830aSAlexander Block BTRFS_COMPARE_TREE_DELETED, 52957069830aSAlexander Block ctx); 52967069830aSAlexander Block if (ret < 0) 52977069830aSAlexander Block goto out; 52987069830aSAlexander Block } 52997069830aSAlexander Block advance_right = ADVANCE; 53007069830aSAlexander Block continue; 53017069830aSAlexander Block } else if (right_end_reached) { 53027069830aSAlexander Block if (left_level == 0) { 53037069830aSAlexander Block ret = changed_cb(left_root, right_root, 53047069830aSAlexander Block left_path, right_path, 53057069830aSAlexander Block &left_key, 53067069830aSAlexander Block BTRFS_COMPARE_TREE_NEW, 53077069830aSAlexander Block ctx); 53087069830aSAlexander Block if (ret < 0) 53097069830aSAlexander Block goto out; 53107069830aSAlexander Block } 53117069830aSAlexander Block advance_left = ADVANCE; 53127069830aSAlexander Block continue; 53137069830aSAlexander Block } 53147069830aSAlexander Block 53157069830aSAlexander Block if (left_level == 0 && right_level == 0) { 53167069830aSAlexander Block cmp = btrfs_comp_cpu_keys(&left_key, &right_key); 53177069830aSAlexander Block if (cmp < 0) { 53187069830aSAlexander Block ret = changed_cb(left_root, right_root, 53197069830aSAlexander Block left_path, right_path, 53207069830aSAlexander Block &left_key, 53217069830aSAlexander Block BTRFS_COMPARE_TREE_NEW, 53227069830aSAlexander Block ctx); 53237069830aSAlexander Block if (ret < 0) 53247069830aSAlexander Block goto out; 53257069830aSAlexander Block advance_left = ADVANCE; 53267069830aSAlexander Block } else if (cmp > 0) { 53277069830aSAlexander Block ret = changed_cb(left_root, right_root, 53287069830aSAlexander Block left_path, right_path, 53297069830aSAlexander Block &right_key, 53307069830aSAlexander Block BTRFS_COMPARE_TREE_DELETED, 53317069830aSAlexander Block ctx); 53327069830aSAlexander Block if (ret < 0) 53337069830aSAlexander Block goto out; 53347069830aSAlexander Block advance_right = ADVANCE; 53357069830aSAlexander Block } else { 533674dd17fbSChris Mason WARN_ON(!extent_buffer_uptodate(left_path->nodes[0])); 53377069830aSAlexander Block ret = tree_compare_item(left_root, left_path, 53387069830aSAlexander Block right_path, tmp_buf); 53397069830aSAlexander Block if (ret) { 534074dd17fbSChris Mason WARN_ON(!extent_buffer_uptodate(left_path->nodes[0])); 53417069830aSAlexander Block ret = changed_cb(left_root, right_root, 53427069830aSAlexander Block left_path, right_path, 53437069830aSAlexander Block &left_key, 53447069830aSAlexander Block BTRFS_COMPARE_TREE_CHANGED, 53457069830aSAlexander Block ctx); 53467069830aSAlexander Block if (ret < 0) 53477069830aSAlexander Block goto out; 53487069830aSAlexander Block } 53497069830aSAlexander Block advance_left = ADVANCE; 53507069830aSAlexander Block advance_right = ADVANCE; 53517069830aSAlexander Block } 53527069830aSAlexander Block } else if (left_level == right_level) { 53537069830aSAlexander Block cmp = btrfs_comp_cpu_keys(&left_key, &right_key); 53547069830aSAlexander Block if (cmp < 0) { 53557069830aSAlexander Block advance_left = ADVANCE; 53567069830aSAlexander Block } else if (cmp > 0) { 53577069830aSAlexander Block advance_right = ADVANCE; 53587069830aSAlexander Block } else { 53597069830aSAlexander Block left_blockptr = btrfs_node_blockptr( 53607069830aSAlexander Block left_path->nodes[left_level], 53617069830aSAlexander Block left_path->slots[left_level]); 53627069830aSAlexander Block right_blockptr = btrfs_node_blockptr( 53637069830aSAlexander Block right_path->nodes[right_level], 53647069830aSAlexander Block right_path->slots[right_level]); 53657069830aSAlexander Block if (left_blockptr == right_blockptr) { 53667069830aSAlexander Block /* 53677069830aSAlexander Block * As we're on a shared block, don't 53687069830aSAlexander Block * allow to go deeper. 53697069830aSAlexander Block */ 53707069830aSAlexander Block advance_left = ADVANCE_ONLY_NEXT; 53717069830aSAlexander Block advance_right = ADVANCE_ONLY_NEXT; 53727069830aSAlexander Block } else { 53737069830aSAlexander Block advance_left = ADVANCE; 53747069830aSAlexander Block advance_right = ADVANCE; 53757069830aSAlexander Block } 53767069830aSAlexander Block } 53777069830aSAlexander Block } else if (left_level < right_level) { 53787069830aSAlexander Block advance_right = ADVANCE; 53797069830aSAlexander Block } else { 53807069830aSAlexander Block advance_left = ADVANCE; 53817069830aSAlexander Block } 53827069830aSAlexander Block } 53837069830aSAlexander Block 53847069830aSAlexander Block out: 53857069830aSAlexander Block btrfs_free_path(left_path); 53867069830aSAlexander Block btrfs_free_path(right_path); 53877069830aSAlexander Block kfree(tmp_buf); 53887069830aSAlexander Block 53897069830aSAlexander Block if (trans) { 53907069830aSAlexander Block if (!ret) 53917069830aSAlexander Block ret = btrfs_end_transaction(trans, left_root); 53927069830aSAlexander Block else 53937069830aSAlexander Block btrfs_end_transaction(trans, left_root); 53947069830aSAlexander Block } 53957069830aSAlexander Block 53967069830aSAlexander Block return ret; 53977069830aSAlexander Block } 53987069830aSAlexander Block 53993f157a2fSChris Mason /* 54003f157a2fSChris Mason * this is similar to btrfs_next_leaf, but does not try to preserve 54013f157a2fSChris Mason * and fixup the path. It looks for and returns the next key in the 54023f157a2fSChris Mason * tree based on the current path and the cache_only and min_trans 54033f157a2fSChris Mason * parameters. 54043f157a2fSChris Mason * 54053f157a2fSChris Mason * 0 is returned if another key is found, < 0 if there are any errors 54063f157a2fSChris Mason * and 1 is returned if there are no higher keys in the tree 54073f157a2fSChris Mason * 54083f157a2fSChris Mason * path->keep_locks should be set to 1 on the search made before 54093f157a2fSChris Mason * calling this function. 54103f157a2fSChris Mason */ 5411e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path, 541233c66f43SYan Zheng struct btrfs_key *key, int level, 54133f157a2fSChris Mason int cache_only, u64 min_trans) 5414e7a84565SChris Mason { 5415e7a84565SChris Mason int slot; 5416e7a84565SChris Mason struct extent_buffer *c; 5417e7a84565SChris Mason 5418934d375bSChris Mason WARN_ON(!path->keep_locks); 5419e7a84565SChris Mason while (level < BTRFS_MAX_LEVEL) { 5420e7a84565SChris Mason if (!path->nodes[level]) 5421e7a84565SChris Mason return 1; 5422e7a84565SChris Mason 5423e7a84565SChris Mason slot = path->slots[level] + 1; 5424e7a84565SChris Mason c = path->nodes[level]; 54253f157a2fSChris Mason next: 5426e7a84565SChris Mason if (slot >= btrfs_header_nritems(c)) { 542733c66f43SYan Zheng int ret; 542833c66f43SYan Zheng int orig_lowest; 542933c66f43SYan Zheng struct btrfs_key cur_key; 543033c66f43SYan Zheng if (level + 1 >= BTRFS_MAX_LEVEL || 543133c66f43SYan Zheng !path->nodes[level + 1]) 5432e7a84565SChris Mason return 1; 543333c66f43SYan Zheng 543433c66f43SYan Zheng if (path->locks[level + 1]) { 543533c66f43SYan Zheng level++; 5436e7a84565SChris Mason continue; 5437e7a84565SChris Mason } 543833c66f43SYan Zheng 543933c66f43SYan Zheng slot = btrfs_header_nritems(c) - 1; 544033c66f43SYan Zheng if (level == 0) 544133c66f43SYan Zheng btrfs_item_key_to_cpu(c, &cur_key, slot); 544233c66f43SYan Zheng else 544333c66f43SYan Zheng btrfs_node_key_to_cpu(c, &cur_key, slot); 544433c66f43SYan Zheng 544533c66f43SYan Zheng orig_lowest = path->lowest_level; 5446b3b4aa74SDavid Sterba btrfs_release_path(path); 544733c66f43SYan Zheng path->lowest_level = level; 544833c66f43SYan Zheng ret = btrfs_search_slot(NULL, root, &cur_key, path, 544933c66f43SYan Zheng 0, 0); 545033c66f43SYan Zheng path->lowest_level = orig_lowest; 545133c66f43SYan Zheng if (ret < 0) 545233c66f43SYan Zheng return ret; 545333c66f43SYan Zheng 545433c66f43SYan Zheng c = path->nodes[level]; 545533c66f43SYan Zheng slot = path->slots[level]; 545633c66f43SYan Zheng if (ret == 0) 545733c66f43SYan Zheng slot++; 545833c66f43SYan Zheng goto next; 545933c66f43SYan Zheng } 546033c66f43SYan Zheng 5461e7a84565SChris Mason if (level == 0) 5462e7a84565SChris Mason btrfs_item_key_to_cpu(c, key, slot); 54633f157a2fSChris Mason else { 54643f157a2fSChris Mason u64 blockptr = btrfs_node_blockptr(c, slot); 54653f157a2fSChris Mason u64 gen = btrfs_node_ptr_generation(c, slot); 54663f157a2fSChris Mason 54673f157a2fSChris Mason if (cache_only) { 54683f157a2fSChris Mason struct extent_buffer *cur; 54693f157a2fSChris Mason cur = btrfs_find_tree_block(root, blockptr, 54703f157a2fSChris Mason btrfs_level_size(root, level - 1)); 5471b9fab919SChris Mason if (!cur || 5472b9fab919SChris Mason btrfs_buffer_uptodate(cur, gen, 1) <= 0) { 54733f157a2fSChris Mason slot++; 54743f157a2fSChris Mason if (cur) 54753f157a2fSChris Mason free_extent_buffer(cur); 54763f157a2fSChris Mason goto next; 54773f157a2fSChris Mason } 54783f157a2fSChris Mason free_extent_buffer(cur); 54793f157a2fSChris Mason } 54803f157a2fSChris Mason if (gen < min_trans) { 54813f157a2fSChris Mason slot++; 54823f157a2fSChris Mason goto next; 54833f157a2fSChris Mason } 5484e7a84565SChris Mason btrfs_node_key_to_cpu(c, key, slot); 54853f157a2fSChris Mason } 5486e7a84565SChris Mason return 0; 5487e7a84565SChris Mason } 5488e7a84565SChris Mason return 1; 5489e7a84565SChris Mason } 5490e7a84565SChris Mason 54917bb86316SChris Mason /* 5492925baeddSChris Mason * search the tree again to find a leaf with greater keys 54930f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 54940f70abe2SChris Mason * returns < 0 on io errors. 549597571fd0SChris Mason */ 5496234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 5497d97e63b6SChris Mason { 54983d7806ecSJan Schmidt return btrfs_next_old_leaf(root, path, 0); 54993d7806ecSJan Schmidt } 55003d7806ecSJan Schmidt 55013d7806ecSJan Schmidt int btrfs_next_old_leaf(struct btrfs_root *root, struct btrfs_path *path, 55023d7806ecSJan Schmidt u64 time_seq) 55033d7806ecSJan Schmidt { 5504d97e63b6SChris Mason int slot; 55058e73f275SChris Mason int level; 55065f39d397SChris Mason struct extent_buffer *c; 55078e73f275SChris Mason struct extent_buffer *next; 5508925baeddSChris Mason struct btrfs_key key; 5509925baeddSChris Mason u32 nritems; 5510925baeddSChris Mason int ret; 55118e73f275SChris Mason int old_spinning = path->leave_spinning; 5512bd681513SChris Mason int next_rw_lock = 0; 5513925baeddSChris Mason 5514925baeddSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 5515d397712bSChris Mason if (nritems == 0) 5516925baeddSChris Mason return 1; 5517925baeddSChris Mason 55188e73f275SChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1); 55198e73f275SChris Mason again: 55208e73f275SChris Mason level = 1; 55218e73f275SChris Mason next = NULL; 5522bd681513SChris Mason next_rw_lock = 0; 5523b3b4aa74SDavid Sterba btrfs_release_path(path); 55248e73f275SChris Mason 5525a2135011SChris Mason path->keep_locks = 1; 55268e73f275SChris Mason path->leave_spinning = 1; 55278e73f275SChris Mason 55283d7806ecSJan Schmidt if (time_seq) 55293d7806ecSJan Schmidt ret = btrfs_search_old_slot(root, &key, path, time_seq); 55303d7806ecSJan Schmidt else 5531925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 5532925baeddSChris Mason path->keep_locks = 0; 5533925baeddSChris Mason 5534925baeddSChris Mason if (ret < 0) 5535925baeddSChris Mason return ret; 5536925baeddSChris Mason 5537a2135011SChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 5538168fd7d2SChris Mason /* 5539168fd7d2SChris Mason * by releasing the path above we dropped all our locks. A balance 5540168fd7d2SChris Mason * could have added more items next to the key that used to be 5541168fd7d2SChris Mason * at the very end of the block. So, check again here and 5542168fd7d2SChris Mason * advance the path if there are now more items available. 5543168fd7d2SChris Mason */ 5544a2135011SChris Mason if (nritems > 0 && path->slots[0] < nritems - 1) { 5545e457afecSYan Zheng if (ret == 0) 5546168fd7d2SChris Mason path->slots[0]++; 55478e73f275SChris Mason ret = 0; 5548925baeddSChris Mason goto done; 5549925baeddSChris Mason } 5550d97e63b6SChris Mason 5551234b63a0SChris Mason while (level < BTRFS_MAX_LEVEL) { 55528e73f275SChris Mason if (!path->nodes[level]) { 55538e73f275SChris Mason ret = 1; 55548e73f275SChris Mason goto done; 55558e73f275SChris Mason } 55565f39d397SChris Mason 5557d97e63b6SChris Mason slot = path->slots[level] + 1; 5558d97e63b6SChris Mason c = path->nodes[level]; 55595f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 5560d97e63b6SChris Mason level++; 55618e73f275SChris Mason if (level == BTRFS_MAX_LEVEL) { 55628e73f275SChris Mason ret = 1; 55638e73f275SChris Mason goto done; 55648e73f275SChris Mason } 5565d97e63b6SChris Mason continue; 5566d97e63b6SChris Mason } 55675f39d397SChris Mason 5568925baeddSChris Mason if (next) { 5569bd681513SChris Mason btrfs_tree_unlock_rw(next, next_rw_lock); 55705f39d397SChris Mason free_extent_buffer(next); 5571925baeddSChris Mason } 55725f39d397SChris Mason 55738e73f275SChris Mason next = c; 5574bd681513SChris Mason next_rw_lock = path->locks[level]; 55758e73f275SChris Mason ret = read_block_for_search(NULL, root, path, &next, level, 55765d9e75c4SJan Schmidt slot, &key, 0); 55778e73f275SChris Mason if (ret == -EAGAIN) 55788e73f275SChris Mason goto again; 55795f39d397SChris Mason 558076a05b35SChris Mason if (ret < 0) { 5581b3b4aa74SDavid Sterba btrfs_release_path(path); 558276a05b35SChris Mason goto done; 558376a05b35SChris Mason } 558476a05b35SChris Mason 55855cd57b2cSChris Mason if (!path->skip_locking) { 5586bd681513SChris Mason ret = btrfs_try_tree_read_lock(next); 5587d42244a0SJan Schmidt if (!ret && time_seq) { 5588d42244a0SJan Schmidt /* 5589d42244a0SJan Schmidt * If we don't get the lock, we may be racing 5590d42244a0SJan Schmidt * with push_leaf_left, holding that lock while 5591d42244a0SJan Schmidt * itself waiting for the leaf we've currently 5592d42244a0SJan Schmidt * locked. To solve this situation, we give up 5593d42244a0SJan Schmidt * on our lock and cycle. 5594d42244a0SJan Schmidt */ 5595cf538830SJan Schmidt free_extent_buffer(next); 5596d42244a0SJan Schmidt btrfs_release_path(path); 5597d42244a0SJan Schmidt cond_resched(); 5598d42244a0SJan Schmidt goto again; 5599d42244a0SJan Schmidt } 56008e73f275SChris Mason if (!ret) { 56018e73f275SChris Mason btrfs_set_path_blocking(path); 5602bd681513SChris Mason btrfs_tree_read_lock(next); 5603bd681513SChris Mason btrfs_clear_path_blocking(path, next, 5604bd681513SChris Mason BTRFS_READ_LOCK); 56058e73f275SChris Mason } 5606bd681513SChris Mason next_rw_lock = BTRFS_READ_LOCK; 5607bd681513SChris Mason } 5608d97e63b6SChris Mason break; 5609d97e63b6SChris Mason } 5610d97e63b6SChris Mason path->slots[level] = slot; 5611d97e63b6SChris Mason while (1) { 5612d97e63b6SChris Mason level--; 5613d97e63b6SChris Mason c = path->nodes[level]; 5614925baeddSChris Mason if (path->locks[level]) 5615bd681513SChris Mason btrfs_tree_unlock_rw(c, path->locks[level]); 56168e73f275SChris Mason 56175f39d397SChris Mason free_extent_buffer(c); 5618d97e63b6SChris Mason path->nodes[level] = next; 5619d97e63b6SChris Mason path->slots[level] = 0; 5620a74a4b97SChris Mason if (!path->skip_locking) 5621bd681513SChris Mason path->locks[level] = next_rw_lock; 5622d97e63b6SChris Mason if (!level) 5623d97e63b6SChris Mason break; 5624b4ce94deSChris Mason 56258e73f275SChris Mason ret = read_block_for_search(NULL, root, path, &next, level, 56265d9e75c4SJan Schmidt 0, &key, 0); 56278e73f275SChris Mason if (ret == -EAGAIN) 56288e73f275SChris Mason goto again; 56298e73f275SChris Mason 563076a05b35SChris Mason if (ret < 0) { 5631b3b4aa74SDavid Sterba btrfs_release_path(path); 563276a05b35SChris Mason goto done; 563376a05b35SChris Mason } 563476a05b35SChris Mason 56355cd57b2cSChris Mason if (!path->skip_locking) { 5636bd681513SChris Mason ret = btrfs_try_tree_read_lock(next); 56378e73f275SChris Mason if (!ret) { 56388e73f275SChris Mason btrfs_set_path_blocking(path); 5639bd681513SChris Mason btrfs_tree_read_lock(next); 5640bd681513SChris Mason btrfs_clear_path_blocking(path, next, 5641bd681513SChris Mason BTRFS_READ_LOCK); 56428e73f275SChris Mason } 5643bd681513SChris Mason next_rw_lock = BTRFS_READ_LOCK; 5644bd681513SChris Mason } 5645d97e63b6SChris Mason } 56468e73f275SChris Mason ret = 0; 5647925baeddSChris Mason done: 5648f7c79f30SChris Mason unlock_up(path, 0, 1, 0, NULL); 56498e73f275SChris Mason path->leave_spinning = old_spinning; 56508e73f275SChris Mason if (!old_spinning) 56518e73f275SChris Mason btrfs_set_path_blocking(path); 56528e73f275SChris Mason 56538e73f275SChris Mason return ret; 5654d97e63b6SChris Mason } 56550b86a832SChris Mason 56563f157a2fSChris Mason /* 56573f157a2fSChris Mason * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps 56583f157a2fSChris Mason * searching until it gets past min_objectid or finds an item of 'type' 56593f157a2fSChris Mason * 56603f157a2fSChris Mason * returns 0 if something is found, 1 if nothing was found and < 0 on error 56613f157a2fSChris Mason */ 56620b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 56630b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 56640b86a832SChris Mason int type) 56650b86a832SChris Mason { 56660b86a832SChris Mason struct btrfs_key found_key; 56670b86a832SChris Mason struct extent_buffer *leaf; 5668e02119d5SChris Mason u32 nritems; 56690b86a832SChris Mason int ret; 56700b86a832SChris Mason 56710b86a832SChris Mason while (1) { 56720b86a832SChris Mason if (path->slots[0] == 0) { 5673b4ce94deSChris Mason btrfs_set_path_blocking(path); 56740b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 56750b86a832SChris Mason if (ret != 0) 56760b86a832SChris Mason return ret; 56770b86a832SChris Mason } else { 56780b86a832SChris Mason path->slots[0]--; 56790b86a832SChris Mason } 56800b86a832SChris Mason leaf = path->nodes[0]; 5681e02119d5SChris Mason nritems = btrfs_header_nritems(leaf); 5682e02119d5SChris Mason if (nritems == 0) 5683e02119d5SChris Mason return 1; 5684e02119d5SChris Mason if (path->slots[0] == nritems) 5685e02119d5SChris Mason path->slots[0]--; 5686e02119d5SChris Mason 56870b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 5688e02119d5SChris Mason if (found_key.objectid < min_objectid) 5689e02119d5SChris Mason break; 56900a4eefbbSYan Zheng if (found_key.type == type) 56910a4eefbbSYan Zheng return 0; 5692e02119d5SChris Mason if (found_key.objectid == min_objectid && 5693e02119d5SChris Mason found_key.type < type) 5694e02119d5SChris Mason break; 56950b86a832SChris Mason } 56960b86a832SChris Mason return 1; 56970b86a832SChris Mason } 5698