blob: 08ddf1873ab1f0c55da018e116a5fae011253597 [file] [log] [blame]
Chris Mason6cbd5572007-06-12 09:07:21 -04001/*
2 * Copyright (C) 2007 Oracle. All rights reserved.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public
6 * License v2 as published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public
14 * License along with this program; if not, write to the
15 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16 * Boston, MA 021110-1307, USA.
17 */
18
Chris Mason5f39d392007-10-15 16:14:19 -040019#include <linux/highmem.h>
Chris Masoneb60cea2007-02-02 09:18:22 -050020#include "ctree.h"
21#include "disk-io.h"
Chris Mason7f5c1512007-03-23 15:56:19 -040022#include "transaction.h"
Chris Mason5f39d392007-10-15 16:14:19 -040023#include "print-tree.h"
Chris Mason9a8dd152007-02-23 08:38:36 -050024
Chris Masone089f052007-03-16 16:20:31 -040025static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
26 *root, struct btrfs_path *path, int level);
27static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
Chris Masond4dbff92007-04-04 14:08:15 -040028 *root, struct btrfs_key *ins_key,
29 struct btrfs_path *path, int data_size);
Chris Mason5f39d392007-10-15 16:14:19 -040030static int push_node_left(struct btrfs_trans_handle *trans,
31 struct btrfs_root *root, struct extent_buffer *dst,
32 struct extent_buffer *src);
33static int balance_node_right(struct btrfs_trans_handle *trans,
34 struct btrfs_root *root,
35 struct extent_buffer *dst_buf,
36 struct extent_buffer *src_buf);
Chris Masone089f052007-03-16 16:20:31 -040037static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
38 struct btrfs_path *path, int level, int slot);
Chris Masond97e63b2007-02-20 16:40:44 -050039
Chris Masondf24a2b2007-04-04 09:36:31 -040040inline void btrfs_init_path(struct btrfs_path *p)
41{
42 memset(p, 0, sizeof(*p));
43}
44
Chris Mason2c90e5d2007-04-02 10:50:19 -040045struct btrfs_path *btrfs_alloc_path(void)
46{
Chris Masondf24a2b2007-04-04 09:36:31 -040047 struct btrfs_path *path;
48 path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS);
Chris Mason2cc58cf2007-08-27 16:49:44 -040049 if (path) {
Chris Masondf24a2b2007-04-04 09:36:31 -040050 btrfs_init_path(path);
Chris Mason2cc58cf2007-08-27 16:49:44 -040051 path->reada = 1;
52 }
Chris Masondf24a2b2007-04-04 09:36:31 -040053 return path;
Chris Mason2c90e5d2007-04-02 10:50:19 -040054}
55
56void btrfs_free_path(struct btrfs_path *p)
57{
Chris Masondf24a2b2007-04-04 09:36:31 -040058 btrfs_release_path(NULL, p);
Chris Mason2c90e5d2007-04-02 10:50:19 -040059 kmem_cache_free(btrfs_path_cachep, p);
60}
61
Chris Mason234b63a2007-03-13 10:46:10 -040062void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
Chris Masoneb60cea2007-02-02 09:18:22 -050063{
64 int i;
Chris Mason234b63a2007-03-13 10:46:10 -040065 for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
Chris Masoneb60cea2007-02-02 09:18:22 -050066 if (!p->nodes[i])
67 break;
Chris Mason5f39d392007-10-15 16:14:19 -040068 free_extent_buffer(p->nodes[i]);
Chris Masoneb60cea2007-02-02 09:18:22 -050069 }
Chris Masonaa5d6be2007-02-28 16:35:06 -050070 memset(p, 0, sizeof(*p));
Chris Masoneb60cea2007-02-02 09:18:22 -050071}
72
Chris Mason5f39d392007-10-15 16:14:19 -040073static int __btrfs_cow_block(struct btrfs_trans_handle *trans,
74 struct btrfs_root *root,
75 struct extent_buffer *buf,
76 struct extent_buffer *parent, int parent_slot,
77 struct extent_buffer **cow_ret,
78 u64 search_start, u64 empty_size)
Chris Mason6702ed42007-08-07 16:15:09 -040079{
Chris Mason5f39d392007-10-15 16:14:19 -040080 struct extent_buffer *cow;
Chris Mason6702ed42007-08-07 16:15:09 -040081 int ret = 0;
82 int different_trans = 0;
83
84 WARN_ON(root->ref_cows && trans->transid != root->last_trans);
Chris Mason5f39d392007-10-15 16:14:19 -040085
Chris Masondb945352007-10-15 16:15:53 -040086 cow = btrfs_alloc_free_block(trans, root, buf->len,
87 search_start, empty_size);
Chris Mason6702ed42007-08-07 16:15:09 -040088 if (IS_ERR(cow))
89 return PTR_ERR(cow);
90
Chris Masonf510cfe2007-10-15 16:14:48 -040091 cow->alloc_addr = (unsigned long)__builtin_return_address(0);
Chris Mason6702ed42007-08-07 16:15:09 -040092
Chris Mason5f39d392007-10-15 16:14:19 -040093 copy_extent_buffer(cow, buf, 0, 0, cow->len);
Chris Masondb945352007-10-15 16:15:53 -040094 btrfs_set_header_bytenr(cow, cow->start);
Chris Mason5f39d392007-10-15 16:14:19 -040095 btrfs_set_header_generation(cow, trans->transid);
96 btrfs_set_header_owner(cow, root->root_key.objectid);
Chris Mason6702ed42007-08-07 16:15:09 -040097
Chris Mason5f39d392007-10-15 16:14:19 -040098 WARN_ON(btrfs_header_generation(buf) > trans->transid);
99 if (btrfs_header_generation(buf) != trans->transid) {
Chris Mason6702ed42007-08-07 16:15:09 -0400100 different_trans = 1;
101 ret = btrfs_inc_ref(trans, root, buf);
102 if (ret)
103 return ret;
104 } else {
Chris Mason6702ed42007-08-07 16:15:09 -0400105 clean_tree_block(trans, root, buf);
106 }
107
108 if (buf == root->node) {
109 root->node = cow;
Chris Mason5f39d392007-10-15 16:14:19 -0400110 extent_buffer_get(cow);
Chris Mason6702ed42007-08-07 16:15:09 -0400111 if (buf != root->commit_root) {
Chris Masondb945352007-10-15 16:15:53 -0400112 btrfs_free_extent(trans, root, buf->start,
113 buf->len, 1);
Chris Mason6702ed42007-08-07 16:15:09 -0400114 }
Chris Mason5f39d392007-10-15 16:14:19 -0400115 free_extent_buffer(buf);
Chris Mason6702ed42007-08-07 16:15:09 -0400116 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400117 btrfs_set_node_blockptr(parent, parent_slot,
Chris Masondb945352007-10-15 16:15:53 -0400118 cow->start);
Chris Mason6702ed42007-08-07 16:15:09 -0400119 btrfs_mark_buffer_dirty(parent);
Chris Mason5f39d392007-10-15 16:14:19 -0400120 WARN_ON(btrfs_header_generation(parent) != trans->transid);
Chris Masondb945352007-10-15 16:15:53 -0400121 btrfs_free_extent(trans, root, buf->start, buf->len, 1);
Chris Mason6702ed42007-08-07 16:15:09 -0400122 }
Chris Mason5f39d392007-10-15 16:14:19 -0400123 free_extent_buffer(buf);
Chris Mason6702ed42007-08-07 16:15:09 -0400124 btrfs_mark_buffer_dirty(cow);
125 *cow_ret = cow;
126 return 0;
127}
128
Chris Mason5f39d392007-10-15 16:14:19 -0400129int btrfs_cow_block(struct btrfs_trans_handle *trans,
130 struct btrfs_root *root, struct extent_buffer *buf,
131 struct extent_buffer *parent, int parent_slot,
132 struct extent_buffer **cow_ret)
Chris Mason02217ed2007-03-02 16:08:05 -0500133{
Chris Mason6702ed42007-08-07 16:15:09 -0400134 u64 search_start;
Chris Masonf510cfe2007-10-15 16:14:48 -0400135 int ret;
Chris Masonccd467d2007-06-28 15:57:36 -0400136 if (trans->transaction != root->fs_info->running_transaction) {
137 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
138 root->fs_info->running_transaction->transid);
139 WARN_ON(1);
140 }
141 if (trans->transid != root->fs_info->generation) {
142 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
143 root->fs_info->generation);
144 WARN_ON(1);
145 }
Chris Mason5f39d392007-10-15 16:14:19 -0400146 if (btrfs_header_generation(buf) == trans->transid) {
Chris Mason02217ed2007-03-02 16:08:05 -0500147 *cow_ret = buf;
148 return 0;
149 }
Chris Mason6702ed42007-08-07 16:15:09 -0400150
Chris Masondb945352007-10-15 16:15:53 -0400151 search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1);
Chris Masonf510cfe2007-10-15 16:14:48 -0400152 ret = __btrfs_cow_block(trans, root, buf, parent,
Chris Mason6702ed42007-08-07 16:15:09 -0400153 parent_slot, cow_ret, search_start, 0);
Chris Masonf510cfe2007-10-15 16:14:48 -0400154 (*cow_ret)->alloc_addr = (unsigned long)__builtin_return_address(0);
155 return ret;
Chris Mason6702ed42007-08-07 16:15:09 -0400156}
157
Chris Mason6b800532007-10-15 16:17:34 -0400158static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
Chris Mason6702ed42007-08-07 16:15:09 -0400159{
Chris Mason6b800532007-10-15 16:17:34 -0400160 if (blocknr < other && other - (blocknr + blocksize) < 32768)
Chris Mason6702ed42007-08-07 16:15:09 -0400161 return 1;
Chris Mason6b800532007-10-15 16:17:34 -0400162 if (blocknr > other && blocknr - (other + blocksize) < 32768)
Chris Mason6702ed42007-08-07 16:15:09 -0400163 return 1;
Chris Mason02217ed2007-03-02 16:08:05 -0500164 return 0;
165}
166
Chris Mason6b800532007-10-15 16:17:34 -0400167static int should_defrag_leaf(struct extent_buffer *leaf)
Chris Mason2cc58cf2007-08-27 16:49:44 -0400168{
Chris Mason6b800532007-10-15 16:17:34 -0400169 struct btrfs_key key;
Chris Mason2cc58cf2007-08-27 16:49:44 -0400170 u32 nritems;
171
Chris Mason6b800532007-10-15 16:17:34 -0400172 if (btrfs_buffer_defrag(leaf))
Chris Mason2cc58cf2007-08-27 16:49:44 -0400173 return 1;
174
Chris Mason6b800532007-10-15 16:17:34 -0400175 nritems = btrfs_header_nritems(leaf);
Chris Mason2cc58cf2007-08-27 16:49:44 -0400176 if (nritems == 0)
177 return 0;
178
Chris Mason6b800532007-10-15 16:17:34 -0400179 btrfs_item_key_to_cpu(leaf, &key, 0);
180 if (key.type == BTRFS_DIR_ITEM_KEY)
Chris Mason2cc58cf2007-08-27 16:49:44 -0400181 return 1;
182
Chris Mason6b800532007-10-15 16:17:34 -0400183
184 btrfs_item_key_to_cpu(leaf, &key, nritems - 1);
185 if (key.type == BTRFS_DIR_ITEM_KEY)
Chris Mason2cc58cf2007-08-27 16:49:44 -0400186 return 1;
187 if (nritems > 4) {
Chris Mason6b800532007-10-15 16:17:34 -0400188 btrfs_item_key_to_cpu(leaf, &key, nritems / 2);
189 if (key.type == BTRFS_DIR_ITEM_KEY)
Chris Mason2cc58cf2007-08-27 16:49:44 -0400190 return 1;
191 }
192 return 0;
193}
194
Chris Mason6702ed42007-08-07 16:15:09 -0400195int btrfs_realloc_node(struct btrfs_trans_handle *trans,
Chris Mason5f39d392007-10-15 16:14:19 -0400196 struct btrfs_root *root, struct extent_buffer *parent,
Chris Masone9d0b132007-08-10 14:06:19 -0400197 int cache_only, u64 *last_ret)
Chris Mason6702ed42007-08-07 16:15:09 -0400198{
Chris Mason6b800532007-10-15 16:17:34 -0400199 struct extent_buffer *cur;
200 struct extent_buffer *tmp;
Chris Mason6702ed42007-08-07 16:15:09 -0400201 u64 blocknr;
Chris Masone9d0b132007-08-10 14:06:19 -0400202 u64 search_start = *last_ret;
203 u64 last_block = 0;
Chris Mason6702ed42007-08-07 16:15:09 -0400204 u64 other;
205 u32 parent_nritems;
206 int start_slot;
207 int end_slot;
208 int i;
209 int err = 0;
Chris Masonf2183bd2007-08-10 14:42:37 -0400210 int parent_level;
Chris Mason6b800532007-10-15 16:17:34 -0400211 int uptodate;
212 u32 blocksize;
Chris Mason6702ed42007-08-07 16:15:09 -0400213
214 if (trans->transaction != root->fs_info->running_transaction) {
215 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
216 root->fs_info->running_transaction->transid);
217 WARN_ON(1);
218 }
219 if (trans->transid != root->fs_info->generation) {
220 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
221 root->fs_info->generation);
222 WARN_ON(1);
223 }
Chris Mason6b800532007-10-15 16:17:34 -0400224 if (btrfs_buffer_defrag_done(parent))
Chris Mason86479a02007-09-10 19:58:16 -0400225 return 0;
226
Chris Mason6b800532007-10-15 16:17:34 -0400227 parent_nritems = btrfs_header_nritems(parent);
228 parent_level = btrfs_header_level(parent);
229 blocksize = btrfs_level_size(root, parent_level - 1);
Chris Mason6702ed42007-08-07 16:15:09 -0400230
231 start_slot = 0;
232 end_slot = parent_nritems;
233
234 if (parent_nritems == 1)
235 return 0;
236
237 for (i = start_slot; i < end_slot; i++) {
238 int close = 1;
Chris Mason6b800532007-10-15 16:17:34 -0400239 blocknr = btrfs_node_blockptr(parent, i);
Chris Masone9d0b132007-08-10 14:06:19 -0400240 if (last_block == 0)
241 last_block = blocknr;
Chris Mason6702ed42007-08-07 16:15:09 -0400242 if (i > 0) {
Chris Mason6b800532007-10-15 16:17:34 -0400243 other = btrfs_node_blockptr(parent, i - 1);
244 close = close_blocks(blocknr, other, blocksize);
Chris Mason6702ed42007-08-07 16:15:09 -0400245 }
246 if (close && i < end_slot - 1) {
Chris Mason6b800532007-10-15 16:17:34 -0400247 other = btrfs_node_blockptr(parent, i + 1);
248 close = close_blocks(blocknr, other, blocksize);
Chris Mason6702ed42007-08-07 16:15:09 -0400249 }
Chris Masone9d0b132007-08-10 14:06:19 -0400250 if (close) {
251 last_block = blocknr;
Chris Mason6702ed42007-08-07 16:15:09 -0400252 continue;
Chris Masone9d0b132007-08-10 14:06:19 -0400253 }
Chris Mason6702ed42007-08-07 16:15:09 -0400254
Chris Mason6b800532007-10-15 16:17:34 -0400255 cur = btrfs_find_tree_block(root, blocknr, blocksize);
256 if (cur)
257 uptodate = btrfs_buffer_uptodate(cur);
258 else
259 uptodate = 0;
260 if (!cur || !uptodate ||
261 (parent_level != 1 && !btrfs_buffer_defrag(cur)) ||
262 (parent_level == 1 && !should_defrag_leaf(cur))) {
Chris Mason6702ed42007-08-07 16:15:09 -0400263 if (cache_only) {
Chris Mason6b800532007-10-15 16:17:34 -0400264 free_extent_buffer(cur);
Chris Mason6702ed42007-08-07 16:15:09 -0400265 continue;
266 }
Chris Mason6b800532007-10-15 16:17:34 -0400267 if (!cur) {
268 cur = read_tree_block(root, blocknr,
269 blocksize);
270 } else if (!uptodate) {
271 btrfs_read_buffer(cur);
Chris Masonf2183bd2007-08-10 14:42:37 -0400272 }
Chris Mason6702ed42007-08-07 16:15:09 -0400273 }
Chris Masone9d0b132007-08-10 14:06:19 -0400274 if (search_start == 0)
Chris Mason6b800532007-10-15 16:17:34 -0400275 search_start = last_block;
Chris Masone9d0b132007-08-10 14:06:19 -0400276
Chris Mason6b800532007-10-15 16:17:34 -0400277 err = __btrfs_cow_block(trans, root, cur, parent, i,
278 &tmp, search_start,
279 min(16 * blocksize,
280 (end_slot - i) * blocksize));
Yan252c38f2007-08-29 09:11:44 -0400281 if (err) {
Chris Mason6b800532007-10-15 16:17:34 -0400282 free_extent_buffer(cur);
Chris Mason6702ed42007-08-07 16:15:09 -0400283 break;
Yan252c38f2007-08-29 09:11:44 -0400284 }
Chris Mason6b800532007-10-15 16:17:34 -0400285 search_start = tmp->start;
Chris Masonf2183bd2007-08-10 14:42:37 -0400286 *last_ret = search_start;
287 if (parent_level == 1)
Chris Mason6b800532007-10-15 16:17:34 -0400288 btrfs_clear_buffer_defrag(tmp);
289 btrfs_set_buffer_defrag_done(tmp);
290 free_extent_buffer(tmp);
Chris Mason6702ed42007-08-07 16:15:09 -0400291 }
292 return err;
293}
294
Chris Mason74123bd2007-02-02 11:05:29 -0500295/*
296 * The leaf data grows from end-to-front in the node.
297 * this returns the address of the start of the last item,
298 * which is the stop of the leaf data stack
299 */
Chris Mason123abc82007-03-14 14:14:43 -0400300static inline unsigned int leaf_data_end(struct btrfs_root *root,
Chris Mason5f39d392007-10-15 16:14:19 -0400301 struct extent_buffer *leaf)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500302{
Chris Mason5f39d392007-10-15 16:14:19 -0400303 u32 nr = btrfs_header_nritems(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500304 if (nr == 0)
Chris Mason123abc82007-03-14 14:14:43 -0400305 return BTRFS_LEAF_DATA_SIZE(root);
Chris Mason5f39d392007-10-15 16:14:19 -0400306 return btrfs_item_offset_nr(leaf, nr - 1);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500307}
308
Chris Mason74123bd2007-02-02 11:05:29 -0500309/*
Chris Mason74123bd2007-02-02 11:05:29 -0500310 * compare two keys in a memcmp fashion
311 */
Chris Mason9aca1d52007-03-13 11:09:37 -0400312static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500313{
Chris Masone2fa7222007-03-12 16:22:34 -0400314 struct btrfs_key k1;
315
316 btrfs_disk_key_to_cpu(&k1, disk);
317
318 if (k1.objectid > k2->objectid)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500319 return 1;
Chris Masone2fa7222007-03-12 16:22:34 -0400320 if (k1.objectid < k2->objectid)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500321 return -1;
Chris Mason5f39d392007-10-15 16:14:19 -0400322 if (k1.type > k2->type)
Chris Masonf254e522007-03-29 15:15:27 -0400323 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -0400324 if (k1.type < k2->type)
Chris Masonf254e522007-03-29 15:15:27 -0400325 return -1;
Chris Mason70b2bef2007-04-17 15:39:32 -0400326 if (k1.offset > k2->offset)
327 return 1;
328 if (k1.offset < k2->offset)
329 return -1;
Chris Masonbe0e5c02007-01-26 15:51:26 -0500330 return 0;
331}
Chris Mason74123bd2007-02-02 11:05:29 -0500332
Chris Mason123abc82007-03-14 14:14:43 -0400333static int check_node(struct btrfs_root *root, struct btrfs_path *path,
334 int level)
Chris Masonaa5d6be2007-02-28 16:35:06 -0500335{
Chris Mason5f39d392007-10-15 16:14:19 -0400336 struct extent_buffer *parent = NULL;
337 struct extent_buffer *node = path->nodes[level];
338 struct btrfs_disk_key parent_key;
339 struct btrfs_disk_key node_key;
Chris Masonaa5d6be2007-02-28 16:35:06 -0500340 int parent_slot;
Chris Mason8d7be552007-05-10 11:24:42 -0400341 int slot;
342 struct btrfs_key cpukey;
Chris Mason5f39d392007-10-15 16:14:19 -0400343 u32 nritems = btrfs_header_nritems(node);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500344
345 if (path->nodes[level + 1])
Chris Mason5f39d392007-10-15 16:14:19 -0400346 parent = path->nodes[level + 1];
Aneesha1f39632007-07-11 10:03:27 -0400347
Chris Mason8d7be552007-05-10 11:24:42 -0400348 slot = path->slots[level];
Chris Mason7518a232007-03-12 12:01:18 -0400349 BUG_ON(nritems == 0);
350 if (parent) {
Aneesha1f39632007-07-11 10:03:27 -0400351 parent_slot = path->slots[level + 1];
Chris Mason5f39d392007-10-15 16:14:19 -0400352 btrfs_node_key(parent, &parent_key, parent_slot);
353 btrfs_node_key(node, &node_key, 0);
354 BUG_ON(memcmp(&parent_key, &node_key,
Chris Masone2fa7222007-03-12 16:22:34 -0400355 sizeof(struct btrfs_disk_key)));
Chris Mason1d4f8a02007-03-13 09:28:32 -0400356 BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
Chris Masondb945352007-10-15 16:15:53 -0400357 btrfs_header_bytenr(node));
Chris Masonaa5d6be2007-02-28 16:35:06 -0500358 }
Chris Mason123abc82007-03-14 14:14:43 -0400359 BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
Chris Mason8d7be552007-05-10 11:24:42 -0400360 if (slot != 0) {
Chris Mason5f39d392007-10-15 16:14:19 -0400361 btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
362 btrfs_node_key(node, &node_key, slot);
363 BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
Chris Mason8d7be552007-05-10 11:24:42 -0400364 }
365 if (slot < nritems - 1) {
Chris Mason5f39d392007-10-15 16:14:19 -0400366 btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
367 btrfs_node_key(node, &node_key, slot);
368 BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500369 }
370 return 0;
371}
372
Chris Mason123abc82007-03-14 14:14:43 -0400373static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
374 int level)
Chris Masonaa5d6be2007-02-28 16:35:06 -0500375{
Chris Mason5f39d392007-10-15 16:14:19 -0400376 struct extent_buffer *leaf = path->nodes[level];
377 struct extent_buffer *parent = NULL;
Chris Masonaa5d6be2007-02-28 16:35:06 -0500378 int parent_slot;
Chris Mason8d7be552007-05-10 11:24:42 -0400379 struct btrfs_key cpukey;
Chris Mason5f39d392007-10-15 16:14:19 -0400380 struct btrfs_disk_key parent_key;
381 struct btrfs_disk_key leaf_key;
382 int slot = path->slots[0];
Chris Mason8d7be552007-05-10 11:24:42 -0400383
Chris Mason5f39d392007-10-15 16:14:19 -0400384 u32 nritems = btrfs_header_nritems(leaf);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500385
386 if (path->nodes[level + 1])
Chris Mason5f39d392007-10-15 16:14:19 -0400387 parent = path->nodes[level + 1];
Chris Mason7518a232007-03-12 12:01:18 -0400388
389 if (nritems == 0)
390 return 0;
391
392 if (parent) {
Aneesha1f39632007-07-11 10:03:27 -0400393 parent_slot = path->slots[level + 1];
Chris Mason5f39d392007-10-15 16:14:19 -0400394 btrfs_node_key(parent, &parent_key, parent_slot);
395 btrfs_item_key(leaf, &leaf_key, 0);
Chris Mason6702ed42007-08-07 16:15:09 -0400396
Chris Mason5f39d392007-10-15 16:14:19 -0400397 BUG_ON(memcmp(&parent_key, &leaf_key,
Chris Masone2fa7222007-03-12 16:22:34 -0400398 sizeof(struct btrfs_disk_key)));
Chris Mason1d4f8a02007-03-13 09:28:32 -0400399 BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
Chris Masondb945352007-10-15 16:15:53 -0400400 btrfs_header_bytenr(leaf));
Chris Masonaa5d6be2007-02-28 16:35:06 -0500401 }
Chris Mason5f39d392007-10-15 16:14:19 -0400402#if 0
403 for (i = 0; nritems > 1 && i < nritems - 2; i++) {
404 btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
405 btrfs_item_key(leaf, &leaf_key, i);
406 if (comp_keys(&leaf_key, &cpukey) >= 0) {
407 btrfs_print_leaf(root, leaf);
408 printk("slot %d offset bad key\n", i);
409 BUG_ON(1);
410 }
411 if (btrfs_item_offset_nr(leaf, i) !=
412 btrfs_item_end_nr(leaf, i + 1)) {
413 btrfs_print_leaf(root, leaf);
414 printk("slot %d offset bad\n", i);
415 BUG_ON(1);
416 }
417 if (i == 0) {
418 if (btrfs_item_offset_nr(leaf, i) +
419 btrfs_item_size_nr(leaf, i) !=
420 BTRFS_LEAF_DATA_SIZE(root)) {
421 btrfs_print_leaf(root, leaf);
422 printk("slot %d first offset bad\n", i);
423 BUG_ON(1);
424 }
425 }
426 }
427 if (nritems > 0) {
428 if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
429 btrfs_print_leaf(root, leaf);
430 printk("slot %d bad size \n", nritems - 1);
431 BUG_ON(1);
432 }
433 }
434#endif
435 if (slot != 0 && slot < nritems - 1) {
436 btrfs_item_key(leaf, &leaf_key, slot);
437 btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
438 if (comp_keys(&leaf_key, &cpukey) <= 0) {
439 btrfs_print_leaf(root, leaf);
440 printk("slot %d offset bad key\n", slot);
441 BUG_ON(1);
442 }
443 if (btrfs_item_offset_nr(leaf, slot - 1) !=
444 btrfs_item_end_nr(leaf, slot)) {
445 btrfs_print_leaf(root, leaf);
446 printk("slot %d offset bad\n", slot);
447 BUG_ON(1);
448 }
Chris Masonaa5d6be2007-02-28 16:35:06 -0500449 }
Chris Mason8d7be552007-05-10 11:24:42 -0400450 if (slot < nritems - 1) {
Chris Mason5f39d392007-10-15 16:14:19 -0400451 btrfs_item_key(leaf, &leaf_key, slot);
452 btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
453 BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
454 if (btrfs_item_offset_nr(leaf, slot) !=
455 btrfs_item_end_nr(leaf, slot + 1)) {
456 btrfs_print_leaf(root, leaf);
457 printk("slot %d offset bad\n", slot);
458 BUG_ON(1);
459 }
Chris Mason8d7be552007-05-10 11:24:42 -0400460 }
Chris Mason5f39d392007-10-15 16:14:19 -0400461 BUG_ON(btrfs_item_offset_nr(leaf, 0) +
462 btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
Chris Masonaa5d6be2007-02-28 16:35:06 -0500463 return 0;
464}
465
Chris Mason123abc82007-03-14 14:14:43 -0400466static int check_block(struct btrfs_root *root, struct btrfs_path *path,
467 int level)
Chris Masonaa5d6be2007-02-28 16:35:06 -0500468{
Chris Masondb945352007-10-15 16:15:53 -0400469#if 0
Chris Mason5f39d392007-10-15 16:14:19 -0400470 struct extent_buffer *buf = path->nodes[level];
Chris Mason5f39d392007-10-15 16:14:19 -0400471
Chris Mason479965d2007-10-15 16:14:27 -0400472 if (memcmp_extent_buffer(buf, root->fs_info->fsid,
473 (unsigned long)btrfs_header_fsid(buf),
474 BTRFS_FSID_SIZE)) {
Chris Mason5f39d392007-10-15 16:14:19 -0400475 printk("warning bad block %Lu\n", buf->start);
Chris Masondb945352007-10-15 16:15:53 -0400476 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -0400477 }
Chris Masondb945352007-10-15 16:15:53 -0400478#endif
Chris Masonaa5d6be2007-02-28 16:35:06 -0500479 if (level == 0)
Chris Mason123abc82007-03-14 14:14:43 -0400480 return check_leaf(root, path, level);
481 return check_node(root, path, level);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500482}
483
Chris Mason74123bd2007-02-02 11:05:29 -0500484/*
Chris Mason5f39d392007-10-15 16:14:19 -0400485 * search for key in the extent_buffer. The items start at offset p,
486 * and they are item_size apart. There are 'max' items in p.
487 *
Chris Mason74123bd2007-02-02 11:05:29 -0500488 * the slot in the array is returned via slot, and it points to
489 * the place where you would insert key if it is not found in
490 * the array.
491 *
492 * slot may point to max if the key is bigger than all of the keys
493 */
Chris Mason5f39d392007-10-15 16:14:19 -0400494static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
495 int item_size, struct btrfs_key *key,
496 int max, int *slot)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500497{
498 int low = 0;
499 int high = max;
500 int mid;
501 int ret;
Chris Mason479965d2007-10-15 16:14:27 -0400502 struct btrfs_disk_key *tmp = NULL;
Chris Mason5f39d392007-10-15 16:14:19 -0400503 struct btrfs_disk_key unaligned;
504 unsigned long offset;
505 char *map_token = NULL;
506 char *kaddr = NULL;
507 unsigned long map_start = 0;
508 unsigned long map_len = 0;
Chris Mason479965d2007-10-15 16:14:27 -0400509 int err;
Chris Masonbe0e5c02007-01-26 15:51:26 -0500510
511 while(low < high) {
512 mid = (low + high) / 2;
Chris Mason5f39d392007-10-15 16:14:19 -0400513 offset = p + mid * item_size;
514
515 if (!map_token || offset < map_start ||
516 (offset + sizeof(struct btrfs_disk_key)) >
517 map_start + map_len) {
Chris Mason479965d2007-10-15 16:14:27 -0400518 if (map_token) {
Chris Mason5f39d392007-10-15 16:14:19 -0400519 unmap_extent_buffer(eb, map_token, KM_USER0);
Chris Mason479965d2007-10-15 16:14:27 -0400520 map_token = NULL;
521 }
522 err = map_extent_buffer(eb, offset,
523 sizeof(struct btrfs_disk_key),
524 &map_token, &kaddr,
525 &map_start, &map_len, KM_USER0);
Chris Mason5f39d392007-10-15 16:14:19 -0400526
Chris Mason479965d2007-10-15 16:14:27 -0400527 if (!err) {
528 tmp = (struct btrfs_disk_key *)(kaddr + offset -
529 map_start);
530 } else {
531 read_extent_buffer(eb, &unaligned,
532 offset, sizeof(unaligned));
533 tmp = &unaligned;
534 }
535
Chris Mason5f39d392007-10-15 16:14:19 -0400536 } else {
537 tmp = (struct btrfs_disk_key *)(kaddr + offset -
538 map_start);
539 }
Chris Masonbe0e5c02007-01-26 15:51:26 -0500540 ret = comp_keys(tmp, key);
541
542 if (ret < 0)
543 low = mid + 1;
544 else if (ret > 0)
545 high = mid;
546 else {
547 *slot = mid;
Chris Mason479965d2007-10-15 16:14:27 -0400548 if (map_token)
549 unmap_extent_buffer(eb, map_token, KM_USER0);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500550 return 0;
551 }
552 }
553 *slot = low;
Chris Mason5f39d392007-10-15 16:14:19 -0400554 if (map_token)
555 unmap_extent_buffer(eb, map_token, KM_USER0);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500556 return 1;
557}
558
Chris Mason97571fd2007-02-24 13:39:08 -0500559/*
560 * simple bin_search frontend that does the right thing for
561 * leaves vs nodes
562 */
Chris Mason5f39d392007-10-15 16:14:19 -0400563static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
564 int level, int *slot)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500565{
Chris Mason5f39d392007-10-15 16:14:19 -0400566 if (level == 0) {
567 return generic_bin_search(eb,
568 offsetof(struct btrfs_leaf, items),
Chris Mason0783fcf2007-03-12 20:12:07 -0400569 sizeof(struct btrfs_item),
Chris Mason5f39d392007-10-15 16:14:19 -0400570 key, btrfs_header_nritems(eb),
Chris Mason7518a232007-03-12 12:01:18 -0400571 slot);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500572 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400573 return generic_bin_search(eb,
574 offsetof(struct btrfs_node, ptrs),
Chris Mason123abc82007-03-14 14:14:43 -0400575 sizeof(struct btrfs_key_ptr),
Chris Mason5f39d392007-10-15 16:14:19 -0400576 key, btrfs_header_nritems(eb),
Chris Mason7518a232007-03-12 12:01:18 -0400577 slot);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500578 }
579 return -1;
580}
581
Chris Mason5f39d392007-10-15 16:14:19 -0400582static struct extent_buffer *read_node_slot(struct btrfs_root *root,
583 struct extent_buffer *parent, int slot)
Chris Masonbb803952007-03-01 12:04:21 -0500584{
Chris Masonbb803952007-03-01 12:04:21 -0500585 if (slot < 0)
586 return NULL;
Chris Mason5f39d392007-10-15 16:14:19 -0400587 if (slot >= btrfs_header_nritems(parent))
Chris Masonbb803952007-03-01 12:04:21 -0500588 return NULL;
Chris Masondb945352007-10-15 16:15:53 -0400589 return read_tree_block(root, btrfs_node_blockptr(parent, slot),
590 btrfs_level_size(root, btrfs_header_level(parent) - 1));
Chris Masonbb803952007-03-01 12:04:21 -0500591}
592
Chris Masone089f052007-03-16 16:20:31 -0400593static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root
594 *root, struct btrfs_path *path, int level)
Chris Masonbb803952007-03-01 12:04:21 -0500595{
Chris Mason5f39d392007-10-15 16:14:19 -0400596 struct extent_buffer *right = NULL;
597 struct extent_buffer *mid;
598 struct extent_buffer *left = NULL;
599 struct extent_buffer *parent = NULL;
Chris Masonbb803952007-03-01 12:04:21 -0500600 int ret = 0;
601 int wret;
602 int pslot;
Chris Masonbb803952007-03-01 12:04:21 -0500603 int orig_slot = path->slots[level];
Chris Mason54aa1f42007-06-22 14:16:25 -0400604 int err_on_enospc = 0;
Chris Mason79f95c82007-03-01 15:16:26 -0500605 u64 orig_ptr;
Chris Masonbb803952007-03-01 12:04:21 -0500606
607 if (level == 0)
608 return 0;
609
Chris Mason5f39d392007-10-15 16:14:19 -0400610 mid = path->nodes[level];
Chris Mason1d4f8a02007-03-13 09:28:32 -0400611 orig_ptr = btrfs_node_blockptr(mid, orig_slot);
Chris Mason79f95c82007-03-01 15:16:26 -0500612
Chris Mason234b63a2007-03-13 10:46:10 -0400613 if (level < BTRFS_MAX_LEVEL - 1)
Chris Mason5f39d392007-10-15 16:14:19 -0400614 parent = path->nodes[level + 1];
Chris Masonbb803952007-03-01 12:04:21 -0500615 pslot = path->slots[level + 1];
616
Chris Mason40689472007-03-17 14:29:23 -0400617 /*
618 * deal with the case where there is only one pointer in the root
619 * by promoting the node below to a root
620 */
Chris Mason5f39d392007-10-15 16:14:19 -0400621 if (!parent) {
622 struct extent_buffer *child;
Chris Masonbb803952007-03-01 12:04:21 -0500623
Chris Mason5f39d392007-10-15 16:14:19 -0400624 if (btrfs_header_nritems(mid) != 1)
Chris Masonbb803952007-03-01 12:04:21 -0500625 return 0;
626
627 /* promote the child to a root */
Chris Mason5f39d392007-10-15 16:14:19 -0400628 child = read_node_slot(root, mid, 0);
Chris Masonbb803952007-03-01 12:04:21 -0500629 BUG_ON(!child);
630 root->node = child;
631 path->nodes[level] = NULL;
Chris Mason5f39d392007-10-15 16:14:19 -0400632 clean_tree_block(trans, root, mid);
633 wait_on_tree_block_writeback(root, mid);
Chris Masonbb803952007-03-01 12:04:21 -0500634 /* once for the path */
Chris Mason5f39d392007-10-15 16:14:19 -0400635 free_extent_buffer(mid);
Chris Masondb945352007-10-15 16:15:53 -0400636 ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1);
Chris Masonbb803952007-03-01 12:04:21 -0500637 /* once for the root ptr */
Chris Mason5f39d392007-10-15 16:14:19 -0400638 free_extent_buffer(mid);
Chris Masondb945352007-10-15 16:15:53 -0400639 return ret;
Chris Masonbb803952007-03-01 12:04:21 -0500640 }
Chris Mason5f39d392007-10-15 16:14:19 -0400641 if (btrfs_header_nritems(mid) >
Chris Mason123abc82007-03-14 14:14:43 -0400642 BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
Chris Masonbb803952007-03-01 12:04:21 -0500643 return 0;
644
Chris Mason5f39d392007-10-15 16:14:19 -0400645 if (btrfs_header_nritems(mid) < 2)
Chris Mason54aa1f42007-06-22 14:16:25 -0400646 err_on_enospc = 1;
647
Chris Mason5f39d392007-10-15 16:14:19 -0400648 left = read_node_slot(root, parent, pslot - 1);
649 if (left) {
650 wret = btrfs_cow_block(trans, root, left,
651 parent, pslot - 1, &left);
Chris Mason54aa1f42007-06-22 14:16:25 -0400652 if (wret) {
653 ret = wret;
654 goto enospc;
655 }
Chris Mason2cc58cf2007-08-27 16:49:44 -0400656 }
Chris Mason5f39d392007-10-15 16:14:19 -0400657 right = read_node_slot(root, parent, pslot + 1);
658 if (right) {
659 wret = btrfs_cow_block(trans, root, right,
660 parent, pslot + 1, &right);
Chris Mason2cc58cf2007-08-27 16:49:44 -0400661 if (wret) {
662 ret = wret;
663 goto enospc;
664 }
665 }
666
667 /* first, try to make some room in the middle buffer */
Chris Mason5f39d392007-10-15 16:14:19 -0400668 if (left) {
669 orig_slot += btrfs_header_nritems(left);
670 wret = push_node_left(trans, root, left, mid);
Chris Mason79f95c82007-03-01 15:16:26 -0500671 if (wret < 0)
672 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -0400673 if (btrfs_header_nritems(mid) < 2)
Chris Mason54aa1f42007-06-22 14:16:25 -0400674 err_on_enospc = 1;
Chris Masonbb803952007-03-01 12:04:21 -0500675 }
Chris Mason79f95c82007-03-01 15:16:26 -0500676
677 /*
678 * then try to empty the right most buffer into the middle
679 */
Chris Mason5f39d392007-10-15 16:14:19 -0400680 if (right) {
681 wret = push_node_left(trans, root, mid, right);
Chris Mason54aa1f42007-06-22 14:16:25 -0400682 if (wret < 0 && wret != -ENOSPC)
Chris Mason79f95c82007-03-01 15:16:26 -0500683 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -0400684 if (btrfs_header_nritems(right) == 0) {
Chris Masondb945352007-10-15 16:15:53 -0400685 u64 bytenr = right->start;
686 u32 blocksize = right->len;
687
Chris Mason5f39d392007-10-15 16:14:19 -0400688 clean_tree_block(trans, root, right);
689 wait_on_tree_block_writeback(root, right);
690 free_extent_buffer(right);
Chris Masonbb803952007-03-01 12:04:21 -0500691 right = NULL;
Chris Masone089f052007-03-16 16:20:31 -0400692 wret = del_ptr(trans, root, path, level + 1, pslot +
693 1);
Chris Masonbb803952007-03-01 12:04:21 -0500694 if (wret)
695 ret = wret;
Chris Masondb945352007-10-15 16:15:53 -0400696 wret = btrfs_free_extent(trans, root, bytenr,
697 blocksize, 1);
Chris Masonbb803952007-03-01 12:04:21 -0500698 if (wret)
699 ret = wret;
700 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400701 struct btrfs_disk_key right_key;
702 btrfs_node_key(right, &right_key, 0);
703 btrfs_set_node_key(parent, &right_key, pslot + 1);
704 btrfs_mark_buffer_dirty(parent);
Chris Masonbb803952007-03-01 12:04:21 -0500705 }
706 }
Chris Mason5f39d392007-10-15 16:14:19 -0400707 if (btrfs_header_nritems(mid) == 1) {
Chris Mason79f95c82007-03-01 15:16:26 -0500708 /*
709 * we're not allowed to leave a node with one item in the
710 * tree during a delete. A deletion from lower in the tree
711 * could try to delete the only pointer in this node.
712 * So, pull some keys from the left.
713 * There has to be a left pointer at this point because
714 * otherwise we would have pulled some pointers from the
715 * right
716 */
Chris Mason5f39d392007-10-15 16:14:19 -0400717 BUG_ON(!left);
718 wret = balance_node_right(trans, root, mid, left);
Chris Mason54aa1f42007-06-22 14:16:25 -0400719 if (wret < 0) {
Chris Mason79f95c82007-03-01 15:16:26 -0500720 ret = wret;
Chris Mason54aa1f42007-06-22 14:16:25 -0400721 goto enospc;
722 }
Chris Mason79f95c82007-03-01 15:16:26 -0500723 BUG_ON(wret == 1);
724 }
Chris Mason5f39d392007-10-15 16:14:19 -0400725 if (btrfs_header_nritems(mid) == 0) {
Chris Mason79f95c82007-03-01 15:16:26 -0500726 /* we've managed to empty the middle node, drop it */
Chris Masondb945352007-10-15 16:15:53 -0400727 u64 bytenr = mid->start;
728 u32 blocksize = mid->len;
Chris Mason5f39d392007-10-15 16:14:19 -0400729 clean_tree_block(trans, root, mid);
730 wait_on_tree_block_writeback(root, mid);
731 free_extent_buffer(mid);
Chris Masonbb803952007-03-01 12:04:21 -0500732 mid = NULL;
Chris Masone089f052007-03-16 16:20:31 -0400733 wret = del_ptr(trans, root, path, level + 1, pslot);
Chris Masonbb803952007-03-01 12:04:21 -0500734 if (wret)
735 ret = wret;
Chris Masondb945352007-10-15 16:15:53 -0400736 wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1);
Chris Masonbb803952007-03-01 12:04:21 -0500737 if (wret)
738 ret = wret;
Chris Mason79f95c82007-03-01 15:16:26 -0500739 } else {
740 /* update the parent key to reflect our changes */
Chris Mason5f39d392007-10-15 16:14:19 -0400741 struct btrfs_disk_key mid_key;
742 btrfs_node_key(mid, &mid_key, 0);
743 btrfs_set_node_key(parent, &mid_key, pslot);
744 btrfs_mark_buffer_dirty(parent);
Chris Mason79f95c82007-03-01 15:16:26 -0500745 }
Chris Masonbb803952007-03-01 12:04:21 -0500746
Chris Mason79f95c82007-03-01 15:16:26 -0500747 /* update the path */
Chris Mason5f39d392007-10-15 16:14:19 -0400748 if (left) {
749 if (btrfs_header_nritems(left) > orig_slot) {
750 extent_buffer_get(left);
751 path->nodes[level] = left;
Chris Masonbb803952007-03-01 12:04:21 -0500752 path->slots[level + 1] -= 1;
753 path->slots[level] = orig_slot;
Chris Mason5f39d392007-10-15 16:14:19 -0400754 if (mid)
755 free_extent_buffer(mid);
Chris Masonbb803952007-03-01 12:04:21 -0500756 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400757 orig_slot -= btrfs_header_nritems(left);
Chris Masonbb803952007-03-01 12:04:21 -0500758 path->slots[level] = orig_slot;
759 }
760 }
Chris Mason79f95c82007-03-01 15:16:26 -0500761 /* double check we haven't messed things up */
Chris Mason123abc82007-03-14 14:14:43 -0400762 check_block(root, path, level);
Chris Masone20d96d2007-03-22 12:13:20 -0400763 if (orig_ptr !=
Chris Mason5f39d392007-10-15 16:14:19 -0400764 btrfs_node_blockptr(path->nodes[level], path->slots[level]))
Chris Mason79f95c82007-03-01 15:16:26 -0500765 BUG();
Chris Mason54aa1f42007-06-22 14:16:25 -0400766enospc:
Chris Mason5f39d392007-10-15 16:14:19 -0400767 if (right)
768 free_extent_buffer(right);
769 if (left)
770 free_extent_buffer(left);
Chris Masonbb803952007-03-01 12:04:21 -0500771 return ret;
772}
773
Chris Masone66f7092007-04-20 13:16:02 -0400774/* returns zero if the push worked, non-zero otherwise */
775static int push_nodes_for_insert(struct btrfs_trans_handle *trans,
776 struct btrfs_root *root,
777 struct btrfs_path *path, int level)
778{
Chris Mason5f39d392007-10-15 16:14:19 -0400779 struct extent_buffer *right = NULL;
780 struct extent_buffer *mid;
781 struct extent_buffer *left = NULL;
782 struct extent_buffer *parent = NULL;
Chris Masone66f7092007-04-20 13:16:02 -0400783 int ret = 0;
784 int wret;
785 int pslot;
786 int orig_slot = path->slots[level];
787 u64 orig_ptr;
788
789 if (level == 0)
790 return 1;
791
Chris Mason5f39d392007-10-15 16:14:19 -0400792 mid = path->nodes[level];
Chris Masone66f7092007-04-20 13:16:02 -0400793 orig_ptr = btrfs_node_blockptr(mid, orig_slot);
794
795 if (level < BTRFS_MAX_LEVEL - 1)
Chris Mason5f39d392007-10-15 16:14:19 -0400796 parent = path->nodes[level + 1];
Chris Masone66f7092007-04-20 13:16:02 -0400797 pslot = path->slots[level + 1];
798
Chris Mason5f39d392007-10-15 16:14:19 -0400799 if (!parent)
Chris Masone66f7092007-04-20 13:16:02 -0400800 return 1;
Chris Masone66f7092007-04-20 13:16:02 -0400801
Chris Mason5f39d392007-10-15 16:14:19 -0400802 left = read_node_slot(root, parent, pslot - 1);
Chris Masone66f7092007-04-20 13:16:02 -0400803
804 /* first, try to make some room in the middle buffer */
Chris Mason5f39d392007-10-15 16:14:19 -0400805 if (left) {
Chris Masone66f7092007-04-20 13:16:02 -0400806 u32 left_nr;
Chris Mason5f39d392007-10-15 16:14:19 -0400807 left_nr = btrfs_header_nritems(left);
Chris Mason33ade1f2007-04-20 13:48:57 -0400808 if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
809 wret = 1;
810 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400811 ret = btrfs_cow_block(trans, root, left, parent,
812 pslot - 1, &left);
Chris Mason54aa1f42007-06-22 14:16:25 -0400813 if (ret)
814 wret = 1;
815 else {
Chris Mason54aa1f42007-06-22 14:16:25 -0400816 wret = push_node_left(trans, root,
Chris Mason5f39d392007-10-15 16:14:19 -0400817 left, mid);
Chris Mason54aa1f42007-06-22 14:16:25 -0400818 }
Chris Mason33ade1f2007-04-20 13:48:57 -0400819 }
Chris Masone66f7092007-04-20 13:16:02 -0400820 if (wret < 0)
821 ret = wret;
822 if (wret == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -0400823 struct btrfs_disk_key disk_key;
Chris Masone66f7092007-04-20 13:16:02 -0400824 orig_slot += left_nr;
Chris Mason5f39d392007-10-15 16:14:19 -0400825 btrfs_node_key(mid, &disk_key, 0);
826 btrfs_set_node_key(parent, &disk_key, pslot);
827 btrfs_mark_buffer_dirty(parent);
828 if (btrfs_header_nritems(left) > orig_slot) {
829 path->nodes[level] = left;
Chris Masone66f7092007-04-20 13:16:02 -0400830 path->slots[level + 1] -= 1;
831 path->slots[level] = orig_slot;
Chris Mason5f39d392007-10-15 16:14:19 -0400832 free_extent_buffer(mid);
Chris Masone66f7092007-04-20 13:16:02 -0400833 } else {
834 orig_slot -=
Chris Mason5f39d392007-10-15 16:14:19 -0400835 btrfs_header_nritems(left);
Chris Masone66f7092007-04-20 13:16:02 -0400836 path->slots[level] = orig_slot;
Chris Mason5f39d392007-10-15 16:14:19 -0400837 free_extent_buffer(left);
Chris Masone66f7092007-04-20 13:16:02 -0400838 }
Chris Masone66f7092007-04-20 13:16:02 -0400839 return 0;
840 }
Chris Mason5f39d392007-10-15 16:14:19 -0400841 free_extent_buffer(left);
Chris Masone66f7092007-04-20 13:16:02 -0400842 }
Chris Mason5f39d392007-10-15 16:14:19 -0400843 right= read_node_slot(root, parent, pslot + 1);
Chris Masone66f7092007-04-20 13:16:02 -0400844
845 /*
846 * then try to empty the right most buffer into the middle
847 */
Chris Mason5f39d392007-10-15 16:14:19 -0400848 if (right) {
Chris Mason33ade1f2007-04-20 13:48:57 -0400849 u32 right_nr;
Chris Mason5f39d392007-10-15 16:14:19 -0400850 right_nr = btrfs_header_nritems(right);
Chris Mason33ade1f2007-04-20 13:48:57 -0400851 if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
852 wret = 1;
853 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400854 ret = btrfs_cow_block(trans, root, right,
855 parent, pslot + 1,
856 &right);
Chris Mason54aa1f42007-06-22 14:16:25 -0400857 if (ret)
858 wret = 1;
859 else {
Chris Mason54aa1f42007-06-22 14:16:25 -0400860 wret = balance_node_right(trans, root,
Chris Mason5f39d392007-10-15 16:14:19 -0400861 right, mid);
Chris Mason54aa1f42007-06-22 14:16:25 -0400862 }
Chris Mason33ade1f2007-04-20 13:48:57 -0400863 }
Chris Masone66f7092007-04-20 13:16:02 -0400864 if (wret < 0)
865 ret = wret;
866 if (wret == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -0400867 struct btrfs_disk_key disk_key;
868
869 btrfs_node_key(right, &disk_key, 0);
870 btrfs_set_node_key(parent, &disk_key, pslot + 1);
871 btrfs_mark_buffer_dirty(parent);
872
873 if (btrfs_header_nritems(mid) <= orig_slot) {
874 path->nodes[level] = right;
Chris Masone66f7092007-04-20 13:16:02 -0400875 path->slots[level + 1] += 1;
876 path->slots[level] = orig_slot -
Chris Mason5f39d392007-10-15 16:14:19 -0400877 btrfs_header_nritems(mid);
878 free_extent_buffer(mid);
Chris Masone66f7092007-04-20 13:16:02 -0400879 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400880 free_extent_buffer(right);
Chris Masone66f7092007-04-20 13:16:02 -0400881 }
Chris Masone66f7092007-04-20 13:16:02 -0400882 return 0;
883 }
Chris Mason5f39d392007-10-15 16:14:19 -0400884 free_extent_buffer(right);
Chris Masone66f7092007-04-20 13:16:02 -0400885 }
Chris Masone66f7092007-04-20 13:16:02 -0400886 return 1;
887}
888
Chris Mason74123bd2007-02-02 11:05:29 -0500889/*
Chris Mason3c69fae2007-08-07 15:52:22 -0400890 * readahead one full node of leaves
891 */
892static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
Chris Mason6702ed42007-08-07 16:15:09 -0400893 int level, int slot)
Chris Mason3c69fae2007-08-07 15:52:22 -0400894{
Chris Mason5f39d392007-10-15 16:14:19 -0400895 struct extent_buffer *node;
Chris Mason3c69fae2007-08-07 15:52:22 -0400896 u32 nritems;
Chris Mason3c69fae2007-08-07 15:52:22 -0400897 u64 search;
Chris Mason6b800532007-10-15 16:17:34 -0400898 u64 lowest_read;
899 u64 highest_read;
900 u64 nread = 0;
Chris Mason3c69fae2007-08-07 15:52:22 -0400901 int direction = path->reada;
Chris Mason5f39d392007-10-15 16:14:19 -0400902 struct extent_buffer *eb;
Chris Mason6b800532007-10-15 16:17:34 -0400903 u32 nr;
904 u32 blocksize;
905 u32 nscan = 0;
Chris Masondb945352007-10-15 16:15:53 -0400906
Chris Mason6702ed42007-08-07 16:15:09 -0400907 if (level == 0)
Chris Mason3c69fae2007-08-07 15:52:22 -0400908 return;
909
Chris Mason6702ed42007-08-07 16:15:09 -0400910 if (!path->nodes[level])
911 return;
912
Chris Mason5f39d392007-10-15 16:14:19 -0400913 node = path->nodes[level];
Chris Mason3c69fae2007-08-07 15:52:22 -0400914 search = btrfs_node_blockptr(node, slot);
Chris Mason6b800532007-10-15 16:17:34 -0400915 blocksize = btrfs_level_size(root, level - 1);
916 eb = btrfs_find_tree_block(root, search, blocksize);
Chris Mason5f39d392007-10-15 16:14:19 -0400917 if (eb) {
918 free_extent_buffer(eb);
Chris Mason3c69fae2007-08-07 15:52:22 -0400919 return;
920 }
921
Chris Mason6b800532007-10-15 16:17:34 -0400922 highest_read = search;
923 lowest_read = search;
924
Chris Mason5f39d392007-10-15 16:14:19 -0400925 nritems = btrfs_header_nritems(node);
Chris Mason6b800532007-10-15 16:17:34 -0400926 nr = slot;
Chris Mason3c69fae2007-08-07 15:52:22 -0400927 while(1) {
Chris Mason6b800532007-10-15 16:17:34 -0400928 if (direction < 0) {
929 if (nr == 0)
930 break;
931 nr--;
932 } else if (direction > 0) {
933 nr++;
934 if (nr >= nritems)
935 break;
Chris Mason3c69fae2007-08-07 15:52:22 -0400936 }
Chris Mason6b800532007-10-15 16:17:34 -0400937 search = btrfs_node_blockptr(node, nr);
938 if ((search >= lowest_read && search <= highest_read) ||
939 (search < lowest_read && lowest_read - search <= 32768) ||
940 (search > highest_read && search - highest_read <= 32768)) {
941 readahead_tree_block(root, search, blocksize);
942 nread += blocksize;
943 }
944 nscan++;
945 if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
946 break;
947 if(nread > (1024 * 1024) || nscan > 128)
948 break;
949
950 if (search < lowest_read)
951 lowest_read = search;
952 if (search > highest_read)
953 highest_read = search;
Chris Mason3c69fae2007-08-07 15:52:22 -0400954 }
955}
956/*
Chris Mason74123bd2007-02-02 11:05:29 -0500957 * look for key in the tree. path is filled in with nodes along the way
958 * if key is found, we return zero and you can find the item in the leaf
959 * level of the path (level 0)
960 *
961 * If the key isn't found, the path points to the slot where it should
Chris Masonaa5d6be2007-02-28 16:35:06 -0500962 * be inserted, and 1 is returned. If there are other errors during the
963 * search a negative error number is returned.
Chris Mason97571fd2007-02-24 13:39:08 -0500964 *
965 * if ins_len > 0, nodes and leaves will be split as we walk down the
966 * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if
967 * possible)
Chris Mason74123bd2007-02-02 11:05:29 -0500968 */
Chris Masone089f052007-03-16 16:20:31 -0400969int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
970 *root, struct btrfs_key *key, struct btrfs_path *p, int
971 ins_len, int cow)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500972{
Chris Mason5f39d392007-10-15 16:14:19 -0400973 struct extent_buffer *b;
Chris Masondb945352007-10-15 16:15:53 -0400974 u64 bytenr;
Chris Masonbe0e5c02007-01-26 15:51:26 -0500975 int slot;
976 int ret;
977 int level;
Chris Mason3c69fae2007-08-07 15:52:22 -0400978 int should_reada = p->reada;
Chris Mason9f3a7422007-08-07 15:52:19 -0400979 u8 lowest_level = 0;
980
Chris Mason6702ed42007-08-07 16:15:09 -0400981 lowest_level = p->lowest_level;
982 WARN_ON(lowest_level && ins_len);
Chris Mason22b0ebd2007-03-30 08:47:31 -0400983 WARN_ON(p->nodes[0] != NULL);
984 WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
Chris Masonbb803952007-03-01 12:04:21 -0500985again:
986 b = root->node;
Chris Mason5f39d392007-10-15 16:14:19 -0400987 extent_buffer_get(b);
Chris Masoneb60cea2007-02-02 09:18:22 -0500988 while (b) {
Chris Mason5f39d392007-10-15 16:14:19 -0400989 level = btrfs_header_level(b);
Chris Mason02217ed2007-03-02 16:08:05 -0500990 if (cow) {
991 int wret;
Chris Masone20d96d2007-03-22 12:13:20 -0400992 wret = btrfs_cow_block(trans, root, b,
993 p->nodes[level + 1],
994 p->slots[level + 1],
Yan252c38f2007-08-29 09:11:44 -0400995 &b);
Chris Mason54aa1f42007-06-22 14:16:25 -0400996 if (wret) {
Chris Mason5f39d392007-10-15 16:14:19 -0400997 free_extent_buffer(b);
Chris Mason54aa1f42007-06-22 14:16:25 -0400998 return wret;
999 }
Chris Mason02217ed2007-03-02 16:08:05 -05001000 }
1001 BUG_ON(!cow && ins_len);
Chris Mason5f39d392007-10-15 16:14:19 -04001002 if (level != btrfs_header_level(b))
Chris Mason2c90e5d2007-04-02 10:50:19 -04001003 WARN_ON(1);
Chris Mason5f39d392007-10-15 16:14:19 -04001004 level = btrfs_header_level(b);
Chris Masoneb60cea2007-02-02 09:18:22 -05001005 p->nodes[level] = b;
Chris Mason123abc82007-03-14 14:14:43 -04001006 ret = check_block(root, p, level);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001007 if (ret)
1008 return -1;
Chris Mason5f39d392007-10-15 16:14:19 -04001009 ret = bin_search(b, key, level, &slot);
1010 if (level != 0) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05001011 if (ret && slot > 0)
1012 slot -= 1;
1013 p->slots[level] = slot;
Chris Mason5f39d392007-10-15 16:14:19 -04001014 if (ins_len > 0 && btrfs_header_nritems(b) >=
Chris Masond4dbff92007-04-04 14:08:15 -04001015 BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
Chris Masone089f052007-03-16 16:20:31 -04001016 int sret = split_node(trans, root, p, level);
Chris Mason5c680ed2007-02-22 11:39:13 -05001017 BUG_ON(sret > 0);
1018 if (sret)
1019 return sret;
1020 b = p->nodes[level];
Chris Mason5c680ed2007-02-22 11:39:13 -05001021 slot = p->slots[level];
Chris Masonbb803952007-03-01 12:04:21 -05001022 } else if (ins_len < 0) {
Chris Masone089f052007-03-16 16:20:31 -04001023 int sret = balance_level(trans, root, p,
1024 level);
Chris Masonbb803952007-03-01 12:04:21 -05001025 if (sret)
1026 return sret;
1027 b = p->nodes[level];
Chris Masonf510cfe2007-10-15 16:14:48 -04001028 if (!b) {
1029 btrfs_release_path(NULL, p);
Chris Masonbb803952007-03-01 12:04:21 -05001030 goto again;
Chris Masonf510cfe2007-10-15 16:14:48 -04001031 }
Chris Masonbb803952007-03-01 12:04:21 -05001032 slot = p->slots[level];
Chris Mason5f39d392007-10-15 16:14:19 -04001033 BUG_ON(btrfs_header_nritems(b) == 1);
Chris Mason5c680ed2007-02-22 11:39:13 -05001034 }
Chris Mason9f3a7422007-08-07 15:52:19 -04001035 /* this is only true while dropping a snapshot */
1036 if (level == lowest_level)
1037 break;
Chris Masondb945352007-10-15 16:15:53 -04001038 bytenr = btrfs_node_blockptr(b, slot);
Chris Mason6702ed42007-08-07 16:15:09 -04001039 if (should_reada)
1040 reada_for_search(root, p, level, slot);
Chris Masondb945352007-10-15 16:15:53 -04001041 b = read_tree_block(root, bytenr,
1042 btrfs_level_size(root, level - 1));
Chris Masonbe0e5c02007-01-26 15:51:26 -05001043 } else {
1044 p->slots[level] = slot;
Chris Mason5f39d392007-10-15 16:14:19 -04001045 if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
Chris Mason0783fcf2007-03-12 20:12:07 -04001046 sizeof(struct btrfs_item) + ins_len) {
Chris Masond4dbff92007-04-04 14:08:15 -04001047 int sret = split_leaf(trans, root, key,
1048 p, ins_len);
Chris Mason5c680ed2007-02-22 11:39:13 -05001049 BUG_ON(sret > 0);
1050 if (sret)
1051 return sret;
1052 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05001053 return ret;
1054 }
1055 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05001056 return 1;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001057}
1058
Chris Mason74123bd2007-02-02 11:05:29 -05001059/*
1060 * adjust the pointers going up the tree, starting at level
1061 * making sure the right key of each node is points to 'key'.
1062 * This is used after shifting pointers to the left, so it stops
1063 * fixing up pointers when a given leaf/node is not in slot 0 of the
1064 * higher levels
Chris Masonaa5d6be2007-02-28 16:35:06 -05001065 *
1066 * If this fails to write a tree block, it returns -1, but continues
1067 * fixing up the blocks in ram so the tree is consistent.
Chris Mason74123bd2007-02-02 11:05:29 -05001068 */
Chris Mason5f39d392007-10-15 16:14:19 -04001069static int fixup_low_keys(struct btrfs_trans_handle *trans,
1070 struct btrfs_root *root, struct btrfs_path *path,
1071 struct btrfs_disk_key *key, int level)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001072{
1073 int i;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001074 int ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04001075 struct extent_buffer *t;
1076
Chris Mason234b63a2007-03-13 10:46:10 -04001077 for (i = level; i < BTRFS_MAX_LEVEL; i++) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05001078 int tslot = path->slots[i];
Chris Masoneb60cea2007-02-02 09:18:22 -05001079 if (!path->nodes[i])
Chris Masonbe0e5c02007-01-26 15:51:26 -05001080 break;
Chris Mason5f39d392007-10-15 16:14:19 -04001081 t = path->nodes[i];
1082 btrfs_set_node_key(t, key, tslot);
Chris Masond6025572007-03-30 14:27:56 -04001083 btrfs_mark_buffer_dirty(path->nodes[i]);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001084 if (tslot != 0)
1085 break;
1086 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05001087 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001088}
1089
Chris Mason74123bd2007-02-02 11:05:29 -05001090/*
1091 * try to push data from one node into the next node left in the
Chris Mason79f95c82007-03-01 15:16:26 -05001092 * tree.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001093 *
1094 * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1095 * error, and > 0 if there was no room in the left hand block.
Chris Mason74123bd2007-02-02 11:05:29 -05001096 */
Chris Masone089f052007-03-16 16:20:31 -04001097static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root
Chris Mason5f39d392007-10-15 16:14:19 -04001098 *root, struct extent_buffer *dst,
1099 struct extent_buffer *src)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001100{
Chris Masonbe0e5c02007-01-26 15:51:26 -05001101 int push_items = 0;
Chris Masonbb803952007-03-01 12:04:21 -05001102 int src_nritems;
1103 int dst_nritems;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001104 int ret = 0;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001105
Chris Mason5f39d392007-10-15 16:14:19 -04001106 src_nritems = btrfs_header_nritems(src);
1107 dst_nritems = btrfs_header_nritems(dst);
Chris Mason123abc82007-03-14 14:14:43 -04001108 push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
Chris Mason54aa1f42007-06-22 14:16:25 -04001109
Chris Masoneb60cea2007-02-02 09:18:22 -05001110 if (push_items <= 0) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05001111 return 1;
Chris Masoneb60cea2007-02-02 09:18:22 -05001112 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05001113
1114 if (src_nritems < push_items)
Chris Mason79f95c82007-03-01 15:16:26 -05001115 push_items = src_nritems;
1116
Chris Mason5f39d392007-10-15 16:14:19 -04001117 copy_extent_buffer(dst, src,
1118 btrfs_node_key_ptr_offset(dst_nritems),
1119 btrfs_node_key_ptr_offset(0),
1120 push_items * sizeof(struct btrfs_key_ptr));
1121
Chris Masonbb803952007-03-01 12:04:21 -05001122 if (push_items < src_nritems) {
Chris Mason5f39d392007-10-15 16:14:19 -04001123 memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
1124 btrfs_node_key_ptr_offset(push_items),
1125 (src_nritems - push_items) *
1126 sizeof(struct btrfs_key_ptr));
Chris Masonbb803952007-03-01 12:04:21 -05001127 }
Chris Mason5f39d392007-10-15 16:14:19 -04001128 btrfs_set_header_nritems(src, src_nritems - push_items);
1129 btrfs_set_header_nritems(dst, dst_nritems + push_items);
1130 btrfs_mark_buffer_dirty(src);
1131 btrfs_mark_buffer_dirty(dst);
Chris Masonbb803952007-03-01 12:04:21 -05001132 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001133}
1134
Chris Mason97571fd2007-02-24 13:39:08 -05001135/*
Chris Mason79f95c82007-03-01 15:16:26 -05001136 * try to push data from one node into the next node right in the
1137 * tree.
1138 *
1139 * returns 0 if some ptrs were pushed, < 0 if there was some horrible
1140 * error, and > 0 if there was no room in the right hand block.
1141 *
1142 * this will only push up to 1/2 the contents of the left node over
1143 */
Chris Mason5f39d392007-10-15 16:14:19 -04001144static int balance_node_right(struct btrfs_trans_handle *trans,
1145 struct btrfs_root *root,
1146 struct extent_buffer *dst,
1147 struct extent_buffer *src)
Chris Mason79f95c82007-03-01 15:16:26 -05001148{
Chris Mason79f95c82007-03-01 15:16:26 -05001149 int push_items = 0;
1150 int max_push;
1151 int src_nritems;
1152 int dst_nritems;
1153 int ret = 0;
Chris Mason79f95c82007-03-01 15:16:26 -05001154
Chris Mason5f39d392007-10-15 16:14:19 -04001155 src_nritems = btrfs_header_nritems(src);
1156 dst_nritems = btrfs_header_nritems(dst);
Chris Mason123abc82007-03-14 14:14:43 -04001157 push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001158 if (push_items <= 0)
Chris Mason79f95c82007-03-01 15:16:26 -05001159 return 1;
Chris Mason79f95c82007-03-01 15:16:26 -05001160
1161 max_push = src_nritems / 2 + 1;
1162 /* don't try to empty the node */
Yan252c38f2007-08-29 09:11:44 -04001163 if (max_push >= src_nritems)
Chris Mason79f95c82007-03-01 15:16:26 -05001164 return 1;
Yan252c38f2007-08-29 09:11:44 -04001165
Chris Mason79f95c82007-03-01 15:16:26 -05001166 if (max_push < push_items)
1167 push_items = max_push;
1168
Chris Mason5f39d392007-10-15 16:14:19 -04001169 memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
1170 btrfs_node_key_ptr_offset(0),
1171 (dst_nritems) *
1172 sizeof(struct btrfs_key_ptr));
Chris Masond6025572007-03-30 14:27:56 -04001173
Chris Mason5f39d392007-10-15 16:14:19 -04001174 copy_extent_buffer(dst, src,
1175 btrfs_node_key_ptr_offset(0),
1176 btrfs_node_key_ptr_offset(src_nritems - push_items),
1177 push_items * sizeof(struct btrfs_key_ptr));
Chris Mason79f95c82007-03-01 15:16:26 -05001178
Chris Mason5f39d392007-10-15 16:14:19 -04001179 btrfs_set_header_nritems(src, src_nritems - push_items);
1180 btrfs_set_header_nritems(dst, dst_nritems + push_items);
Chris Mason79f95c82007-03-01 15:16:26 -05001181
Chris Mason5f39d392007-10-15 16:14:19 -04001182 btrfs_mark_buffer_dirty(src);
1183 btrfs_mark_buffer_dirty(dst);
Chris Mason79f95c82007-03-01 15:16:26 -05001184 return ret;
1185}
1186
1187/*
Chris Mason97571fd2007-02-24 13:39:08 -05001188 * helper function to insert a new root level in the tree.
1189 * A new node is allocated, and a single item is inserted to
1190 * point to the existing root
Chris Masonaa5d6be2007-02-28 16:35:06 -05001191 *
1192 * returns zero on success or < 0 on failure.
Chris Mason97571fd2007-02-24 13:39:08 -05001193 */
Chris Mason5f39d392007-10-15 16:14:19 -04001194static int insert_new_root(struct btrfs_trans_handle *trans,
1195 struct btrfs_root *root,
1196 struct btrfs_path *path, int level)
Chris Mason5c680ed2007-02-22 11:39:13 -05001197{
Chris Mason5f39d392007-10-15 16:14:19 -04001198 struct extent_buffer *lower;
1199 struct extent_buffer *c;
1200 struct btrfs_disk_key lower_key;
Chris Mason5c680ed2007-02-22 11:39:13 -05001201
1202 BUG_ON(path->nodes[level]);
1203 BUG_ON(path->nodes[level-1] != root->node);
1204
Chris Masondb945352007-10-15 16:15:53 -04001205 c = btrfs_alloc_free_block(trans, root, root->nodesize,
1206 root->node->start, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04001207 if (IS_ERR(c))
1208 return PTR_ERR(c);
1209 memset_extent_buffer(c, 0, 0, root->nodesize);
1210 btrfs_set_header_nritems(c, 1);
1211 btrfs_set_header_level(c, level);
Chris Masondb945352007-10-15 16:15:53 -04001212 btrfs_set_header_bytenr(c, c->start);
Chris Mason5f39d392007-10-15 16:14:19 -04001213 btrfs_set_header_generation(c, trans->transid);
1214 btrfs_set_header_owner(c, root->root_key.objectid);
1215 lower = path->nodes[level-1];
Chris Masond5719762007-03-23 10:01:08 -04001216
Chris Mason5f39d392007-10-15 16:14:19 -04001217 write_extent_buffer(c, root->fs_info->fsid,
1218 (unsigned long)btrfs_header_fsid(c),
1219 BTRFS_FSID_SIZE);
1220 if (level == 1)
1221 btrfs_item_key(lower, &lower_key, 0);
1222 else
1223 btrfs_node_key(lower, &lower_key, 0);
1224 btrfs_set_node_key(c, &lower_key, 0);
Chris Masondb945352007-10-15 16:15:53 -04001225 btrfs_set_node_blockptr(c, 0, lower->start);
Chris Mason5f39d392007-10-15 16:14:19 -04001226
1227 btrfs_mark_buffer_dirty(c);
Chris Masond5719762007-03-23 10:01:08 -04001228
Chris Mason5c680ed2007-02-22 11:39:13 -05001229 /* the super has an extra ref to root->node */
Chris Mason5f39d392007-10-15 16:14:19 -04001230 free_extent_buffer(root->node);
1231 root->node = c;
1232 extent_buffer_get(c);
1233 path->nodes[level] = c;
Chris Mason5c680ed2007-02-22 11:39:13 -05001234 path->slots[level] = 0;
1235 return 0;
1236}
1237
Chris Mason74123bd2007-02-02 11:05:29 -05001238/*
1239 * worker function to insert a single pointer in a node.
1240 * the node should have enough room for the pointer already
Chris Mason97571fd2007-02-24 13:39:08 -05001241 *
Chris Mason74123bd2007-02-02 11:05:29 -05001242 * slot and level indicate where you want the key to go, and
1243 * blocknr is the block the key points to.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001244 *
1245 * returns zero on success and < 0 on any error
Chris Mason74123bd2007-02-02 11:05:29 -05001246 */
Chris Masone089f052007-03-16 16:20:31 -04001247static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1248 *root, struct btrfs_path *path, struct btrfs_disk_key
Chris Masondb945352007-10-15 16:15:53 -04001249 *key, u64 bytenr, int slot, int level)
Chris Mason74123bd2007-02-02 11:05:29 -05001250{
Chris Mason5f39d392007-10-15 16:14:19 -04001251 struct extent_buffer *lower;
Chris Mason74123bd2007-02-02 11:05:29 -05001252 int nritems;
Chris Mason5c680ed2007-02-22 11:39:13 -05001253
1254 BUG_ON(!path->nodes[level]);
Chris Mason5f39d392007-10-15 16:14:19 -04001255 lower = path->nodes[level];
1256 nritems = btrfs_header_nritems(lower);
Chris Mason74123bd2007-02-02 11:05:29 -05001257 if (slot > nritems)
1258 BUG();
Chris Mason123abc82007-03-14 14:14:43 -04001259 if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
Chris Mason74123bd2007-02-02 11:05:29 -05001260 BUG();
1261 if (slot != nritems) {
Chris Mason5f39d392007-10-15 16:14:19 -04001262 memmove_extent_buffer(lower,
1263 btrfs_node_key_ptr_offset(slot + 1),
1264 btrfs_node_key_ptr_offset(slot),
Chris Masond6025572007-03-30 14:27:56 -04001265 (nritems - slot) * sizeof(struct btrfs_key_ptr));
Chris Mason74123bd2007-02-02 11:05:29 -05001266 }
Chris Mason5f39d392007-10-15 16:14:19 -04001267 btrfs_set_node_key(lower, key, slot);
Chris Masondb945352007-10-15 16:15:53 -04001268 btrfs_set_node_blockptr(lower, slot, bytenr);
Chris Mason5f39d392007-10-15 16:14:19 -04001269 btrfs_set_header_nritems(lower, nritems + 1);
1270 btrfs_mark_buffer_dirty(lower);
Chris Mason74123bd2007-02-02 11:05:29 -05001271 return 0;
1272}
1273
Chris Mason97571fd2007-02-24 13:39:08 -05001274/*
1275 * split the node at the specified level in path in two.
1276 * The path is corrected to point to the appropriate node after the split
1277 *
1278 * Before splitting this tries to make some room in the node by pushing
1279 * left and right, if either one works, it returns right away.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001280 *
1281 * returns 0 on success and < 0 on failure
Chris Mason97571fd2007-02-24 13:39:08 -05001282 */
Chris Masone089f052007-03-16 16:20:31 -04001283static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1284 *root, struct btrfs_path *path, int level)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001285{
Chris Mason5f39d392007-10-15 16:14:19 -04001286 struct extent_buffer *c;
1287 struct extent_buffer *split;
1288 struct btrfs_disk_key disk_key;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001289 int mid;
Chris Mason5c680ed2007-02-22 11:39:13 -05001290 int ret;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001291 int wret;
Chris Mason7518a232007-03-12 12:01:18 -04001292 u32 c_nritems;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001293
Chris Mason5f39d392007-10-15 16:14:19 -04001294 c = path->nodes[level];
1295 if (c == root->node) {
Chris Mason5c680ed2007-02-22 11:39:13 -05001296 /* trying to split the root, lets make a new one */
Chris Masone089f052007-03-16 16:20:31 -04001297 ret = insert_new_root(trans, root, path, level + 1);
Chris Mason5c680ed2007-02-22 11:39:13 -05001298 if (ret)
1299 return ret;
Chris Masone66f7092007-04-20 13:16:02 -04001300 } else {
1301 ret = push_nodes_for_insert(trans, root, path, level);
Chris Mason5f39d392007-10-15 16:14:19 -04001302 c = path->nodes[level];
1303 if (!ret && btrfs_header_nritems(c) <
Chris Masone66f7092007-04-20 13:16:02 -04001304 BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
1305 return 0;
Chris Mason54aa1f42007-06-22 14:16:25 -04001306 if (ret < 0)
1307 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001308 }
Chris Masone66f7092007-04-20 13:16:02 -04001309
Chris Mason5f39d392007-10-15 16:14:19 -04001310 c_nritems = btrfs_header_nritems(c);
Chris Masondb945352007-10-15 16:15:53 -04001311 split = btrfs_alloc_free_block(trans, root, root->nodesize,
1312 c->start, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04001313 if (IS_ERR(split))
1314 return PTR_ERR(split);
Chris Mason54aa1f42007-06-22 14:16:25 -04001315
Chris Mason5f39d392007-10-15 16:14:19 -04001316 btrfs_set_header_flags(split, btrfs_header_flags(c));
1317 btrfs_set_header_level(split, btrfs_header_level(c));
Chris Masondb945352007-10-15 16:15:53 -04001318 btrfs_set_header_bytenr(split, split->start);
Chris Mason5f39d392007-10-15 16:14:19 -04001319 btrfs_set_header_generation(split, trans->transid);
1320 btrfs_set_header_owner(split, root->root_key.objectid);
1321 write_extent_buffer(split, root->fs_info->fsid,
1322 (unsigned long)btrfs_header_fsid(split),
1323 BTRFS_FSID_SIZE);
1324
Chris Mason7518a232007-03-12 12:01:18 -04001325 mid = (c_nritems + 1) / 2;
Chris Mason5f39d392007-10-15 16:14:19 -04001326
1327 copy_extent_buffer(split, c,
1328 btrfs_node_key_ptr_offset(0),
1329 btrfs_node_key_ptr_offset(mid),
1330 (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
1331 btrfs_set_header_nritems(split, c_nritems - mid);
1332 btrfs_set_header_nritems(c, mid);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001333 ret = 0;
1334
Chris Mason5f39d392007-10-15 16:14:19 -04001335 btrfs_mark_buffer_dirty(c);
1336 btrfs_mark_buffer_dirty(split);
1337
1338 btrfs_node_key(split, &disk_key, 0);
Chris Masondb945352007-10-15 16:15:53 -04001339 wret = insert_ptr(trans, root, path, &disk_key, split->start,
Chris Mason5f39d392007-10-15 16:14:19 -04001340 path->slots[level + 1] + 1,
Chris Mason123abc82007-03-14 14:14:43 -04001341 level + 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001342 if (wret)
1343 ret = wret;
1344
Chris Mason5de08d72007-02-24 06:24:44 -05001345 if (path->slots[level] >= mid) {
Chris Mason5c680ed2007-02-22 11:39:13 -05001346 path->slots[level] -= mid;
Chris Mason5f39d392007-10-15 16:14:19 -04001347 free_extent_buffer(c);
1348 path->nodes[level] = split;
Chris Mason5c680ed2007-02-22 11:39:13 -05001349 path->slots[level + 1] += 1;
1350 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04001351 free_extent_buffer(split);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001352 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05001353 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001354}
1355
Chris Mason74123bd2007-02-02 11:05:29 -05001356/*
1357 * how many bytes are required to store the items in a leaf. start
1358 * and nr indicate which items in the leaf to check. This totals up the
1359 * space used both by the item structs and the item data
1360 */
Chris Mason5f39d392007-10-15 16:14:19 -04001361static int leaf_space_used(struct extent_buffer *l, int start, int nr)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001362{
1363 int data_len;
Chris Mason5f39d392007-10-15 16:14:19 -04001364 int nritems = btrfs_header_nritems(l);
Chris Masond4dbff92007-04-04 14:08:15 -04001365 int end = min(nritems, start + nr) - 1;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001366
1367 if (!nr)
1368 return 0;
Chris Mason5f39d392007-10-15 16:14:19 -04001369 data_len = btrfs_item_end_nr(l, start);
1370 data_len = data_len - btrfs_item_offset_nr(l, end);
Chris Mason0783fcf2007-03-12 20:12:07 -04001371 data_len += sizeof(struct btrfs_item) * nr;
Chris Masond4dbff92007-04-04 14:08:15 -04001372 WARN_ON(data_len < 0);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001373 return data_len;
1374}
1375
Chris Mason74123bd2007-02-02 11:05:29 -05001376/*
Chris Masond4dbff92007-04-04 14:08:15 -04001377 * The space between the end of the leaf items and
1378 * the start of the leaf data. IOW, how much room
1379 * the leaf has left for both items and data
1380 */
Chris Mason5f39d392007-10-15 16:14:19 -04001381int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
Chris Masond4dbff92007-04-04 14:08:15 -04001382{
Chris Mason5f39d392007-10-15 16:14:19 -04001383 int nritems = btrfs_header_nritems(leaf);
1384 int ret;
1385 ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
1386 if (ret < 0) {
1387 printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1388 ret, BTRFS_LEAF_DATA_SIZE(root),
1389 leaf_space_used(leaf, 0, nritems), nritems);
1390 }
1391 return ret;
Chris Masond4dbff92007-04-04 14:08:15 -04001392}
1393
1394/*
Chris Mason00ec4c52007-02-24 12:47:20 -05001395 * push some data in the path leaf to the right, trying to free up at
1396 * least data_size bytes. returns zero if the push worked, nonzero otherwise
Chris Masonaa5d6be2007-02-28 16:35:06 -05001397 *
1398 * returns 1 if the push failed because the other node didn't have enough
1399 * room, 0 if everything worked out and < 0 if there were major errors.
Chris Mason00ec4c52007-02-24 12:47:20 -05001400 */
Chris Masone089f052007-03-16 16:20:31 -04001401static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1402 *root, struct btrfs_path *path, int data_size)
Chris Mason00ec4c52007-02-24 12:47:20 -05001403{
Chris Mason5f39d392007-10-15 16:14:19 -04001404 struct extent_buffer *left = path->nodes[0];
1405 struct extent_buffer *right;
1406 struct extent_buffer *upper;
1407 struct btrfs_disk_key disk_key;
Chris Mason00ec4c52007-02-24 12:47:20 -05001408 int slot;
1409 int i;
1410 int free_space;
1411 int push_space = 0;
1412 int push_items = 0;
Chris Mason0783fcf2007-03-12 20:12:07 -04001413 struct btrfs_item *item;
Chris Mason7518a232007-03-12 12:01:18 -04001414 u32 left_nritems;
1415 u32 right_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001416 u32 data_end;
Chris Masondb945352007-10-15 16:15:53 -04001417 u32 this_item_size;
Chris Mason54aa1f42007-06-22 14:16:25 -04001418 int ret;
Chris Mason00ec4c52007-02-24 12:47:20 -05001419
1420 slot = path->slots[1];
1421 if (!path->nodes[1]) {
1422 return 1;
1423 }
1424 upper = path->nodes[1];
Chris Mason5f39d392007-10-15 16:14:19 -04001425 if (slot >= btrfs_header_nritems(upper) - 1)
Chris Mason00ec4c52007-02-24 12:47:20 -05001426 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04001427
Chris Masondb945352007-10-15 16:15:53 -04001428 right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
1429 root->leafsize);
Chris Mason123abc82007-03-14 14:14:43 -04001430 free_space = btrfs_leaf_free_space(root, right);
Chris Mason0783fcf2007-03-12 20:12:07 -04001431 if (free_space < data_size + sizeof(struct btrfs_item)) {
Chris Mason5f39d392007-10-15 16:14:19 -04001432 free_extent_buffer(right);
Chris Mason02217ed2007-03-02 16:08:05 -05001433 return 1;
1434 }
1435
Chris Mason5f39d392007-10-15 16:14:19 -04001436 /* cow and double check */
1437 ret = btrfs_cow_block(trans, root, right, upper,
1438 slot + 1, &right);
1439 if (ret) {
1440 free_extent_buffer(right);
Chris Masona429e512007-04-18 16:15:28 -04001441 return 1;
1442 }
Chris Mason5f39d392007-10-15 16:14:19 -04001443 free_space = btrfs_leaf_free_space(root, right);
1444 if (free_space < data_size + sizeof(struct btrfs_item)) {
1445 free_extent_buffer(right);
1446 return 1;
1447 }
1448
1449 left_nritems = btrfs_header_nritems(left);
1450 if (left_nritems == 0) {
1451 free_extent_buffer(right);
1452 return 1;
1453 }
1454
Chris Masona429e512007-04-18 16:15:28 -04001455 for (i = left_nritems - 1; i >= 1; i--) {
Chris Mason5f39d392007-10-15 16:14:19 -04001456 item = btrfs_item_nr(left, i);
Chris Masondb945352007-10-15 16:15:53 -04001457
Chris Mason00ec4c52007-02-24 12:47:20 -05001458 if (path->slots[0] == i)
1459 push_space += data_size + sizeof(*item);
Chris Masondb945352007-10-15 16:15:53 -04001460
1461 if (!left->map_token) {
1462 map_extent_buffer(left, (unsigned long)item,
1463 sizeof(struct btrfs_item),
1464 &left->map_token, &left->kaddr,
1465 &left->map_start, &left->map_len,
1466 KM_USER1);
1467 }
1468
1469 this_item_size = btrfs_item_size(left, item);
1470 if (this_item_size + sizeof(*item) + push_space > free_space)
Chris Mason00ec4c52007-02-24 12:47:20 -05001471 break;
1472 push_items++;
Chris Masondb945352007-10-15 16:15:53 -04001473 push_space += this_item_size + sizeof(*item);
1474 }
1475 if (left->map_token) {
1476 unmap_extent_buffer(left, left->map_token, KM_USER1);
1477 left->map_token = NULL;
Chris Mason00ec4c52007-02-24 12:47:20 -05001478 }
Chris Mason5f39d392007-10-15 16:14:19 -04001479
Chris Mason00ec4c52007-02-24 12:47:20 -05001480 if (push_items == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04001481 free_extent_buffer(right);
Chris Mason00ec4c52007-02-24 12:47:20 -05001482 return 1;
1483 }
Chris Mason5f39d392007-10-15 16:14:19 -04001484
Chris Masona429e512007-04-18 16:15:28 -04001485 if (push_items == left_nritems)
1486 WARN_ON(1);
Chris Mason5f39d392007-10-15 16:14:19 -04001487
Chris Mason00ec4c52007-02-24 12:47:20 -05001488 /* push left to right */
Chris Mason5f39d392007-10-15 16:14:19 -04001489 right_nritems = btrfs_header_nritems(right);
1490 push_space = btrfs_item_end_nr(left, left_nritems - push_items);
Chris Mason123abc82007-03-14 14:14:43 -04001491 push_space -= leaf_data_end(root, left);
Chris Mason5f39d392007-10-15 16:14:19 -04001492
Chris Mason00ec4c52007-02-24 12:47:20 -05001493 /* make room in the right data area */
Chris Mason5f39d392007-10-15 16:14:19 -04001494 data_end = leaf_data_end(root, right);
1495 memmove_extent_buffer(right,
1496 btrfs_leaf_data(right) + data_end - push_space,
1497 btrfs_leaf_data(right) + data_end,
1498 BTRFS_LEAF_DATA_SIZE(root) - data_end);
1499
Chris Mason00ec4c52007-02-24 12:47:20 -05001500 /* copy from the left data area */
Chris Mason5f39d392007-10-15 16:14:19 -04001501 copy_extent_buffer(right, left, btrfs_leaf_data(right) +
Chris Masond6025572007-03-30 14:27:56 -04001502 BTRFS_LEAF_DATA_SIZE(root) - push_space,
1503 btrfs_leaf_data(left) + leaf_data_end(root, left),
1504 push_space);
Chris Mason5f39d392007-10-15 16:14:19 -04001505
1506 memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
1507 btrfs_item_nr_offset(0),
1508 right_nritems * sizeof(struct btrfs_item));
1509
Chris Mason00ec4c52007-02-24 12:47:20 -05001510 /* copy the items from left to right */
Chris Mason5f39d392007-10-15 16:14:19 -04001511 copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
1512 btrfs_item_nr_offset(left_nritems - push_items),
1513 push_items * sizeof(struct btrfs_item));
Chris Mason00ec4c52007-02-24 12:47:20 -05001514
1515 /* update the item pointers */
Chris Mason7518a232007-03-12 12:01:18 -04001516 right_nritems += push_items;
Chris Mason5f39d392007-10-15 16:14:19 -04001517 btrfs_set_header_nritems(right, right_nritems);
Chris Mason123abc82007-03-14 14:14:43 -04001518 push_space = BTRFS_LEAF_DATA_SIZE(root);
Chris Masondb945352007-10-15 16:15:53 -04001519
Chris Mason7518a232007-03-12 12:01:18 -04001520 for (i = 0; i < right_nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04001521 item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04001522 if (!right->map_token) {
1523 map_extent_buffer(right, (unsigned long)item,
1524 sizeof(struct btrfs_item),
1525 &right->map_token, &right->kaddr,
1526 &right->map_start, &right->map_len,
1527 KM_USER1);
1528 }
1529 push_space -= btrfs_item_size(right, item);
1530 btrfs_set_item_offset(right, item, push_space);
1531 }
1532
1533 if (right->map_token) {
1534 unmap_extent_buffer(right, right->map_token, KM_USER1);
1535 right->map_token = NULL;
Chris Mason00ec4c52007-02-24 12:47:20 -05001536 }
Chris Mason7518a232007-03-12 12:01:18 -04001537 left_nritems -= push_items;
Chris Mason5f39d392007-10-15 16:14:19 -04001538 btrfs_set_header_nritems(left, left_nritems);
Chris Mason00ec4c52007-02-24 12:47:20 -05001539
Chris Mason5f39d392007-10-15 16:14:19 -04001540 btrfs_mark_buffer_dirty(left);
1541 btrfs_mark_buffer_dirty(right);
Chris Masona429e512007-04-18 16:15:28 -04001542
Chris Mason5f39d392007-10-15 16:14:19 -04001543 btrfs_item_key(right, &disk_key, 0);
1544 btrfs_set_node_key(upper, &disk_key, slot + 1);
Chris Masond6025572007-03-30 14:27:56 -04001545 btrfs_mark_buffer_dirty(upper);
Chris Mason02217ed2007-03-02 16:08:05 -05001546
Chris Mason00ec4c52007-02-24 12:47:20 -05001547 /* then fixup the leaf pointer in the path */
Chris Mason7518a232007-03-12 12:01:18 -04001548 if (path->slots[0] >= left_nritems) {
1549 path->slots[0] -= left_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001550 free_extent_buffer(path->nodes[0]);
1551 path->nodes[0] = right;
Chris Mason00ec4c52007-02-24 12:47:20 -05001552 path->slots[1] += 1;
1553 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04001554 free_extent_buffer(right);
Chris Mason00ec4c52007-02-24 12:47:20 -05001555 }
1556 return 0;
1557}
1558/*
Chris Mason74123bd2007-02-02 11:05:29 -05001559 * push some data in the path leaf to the left, trying to free up at
1560 * least data_size bytes. returns zero if the push worked, nonzero otherwise
1561 */
Chris Masone089f052007-03-16 16:20:31 -04001562static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
1563 *root, struct btrfs_path *path, int data_size)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001564{
Chris Mason5f39d392007-10-15 16:14:19 -04001565 struct btrfs_disk_key disk_key;
1566 struct extent_buffer *right = path->nodes[0];
1567 struct extent_buffer *left;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001568 int slot;
1569 int i;
1570 int free_space;
1571 int push_space = 0;
1572 int push_items = 0;
Chris Mason0783fcf2007-03-12 20:12:07 -04001573 struct btrfs_item *item;
Chris Mason7518a232007-03-12 12:01:18 -04001574 u32 old_left_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001575 u32 right_nritems;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001576 int ret = 0;
1577 int wret;
Chris Masondb945352007-10-15 16:15:53 -04001578 u32 this_item_size;
1579 u32 old_left_item_size;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001580
1581 slot = path->slots[1];
Chris Mason5f39d392007-10-15 16:14:19 -04001582 if (slot == 0)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001583 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04001584 if (!path->nodes[1])
Chris Masonbe0e5c02007-01-26 15:51:26 -05001585 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04001586
1587 left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
Chris Masondb945352007-10-15 16:15:53 -04001588 slot - 1), root->leafsize);
Chris Mason123abc82007-03-14 14:14:43 -04001589 free_space = btrfs_leaf_free_space(root, left);
Chris Mason0783fcf2007-03-12 20:12:07 -04001590 if (free_space < data_size + sizeof(struct btrfs_item)) {
Chris Mason5f39d392007-10-15 16:14:19 -04001591 free_extent_buffer(left);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001592 return 1;
1593 }
Chris Mason02217ed2007-03-02 16:08:05 -05001594
1595 /* cow and double check */
Chris Mason5f39d392007-10-15 16:14:19 -04001596 ret = btrfs_cow_block(trans, root, left,
1597 path->nodes[1], slot - 1, &left);
Chris Mason54aa1f42007-06-22 14:16:25 -04001598 if (ret) {
1599 /* we hit -ENOSPC, but it isn't fatal here */
Chris Mason5f39d392007-10-15 16:14:19 -04001600 free_extent_buffer(left);
Chris Mason54aa1f42007-06-22 14:16:25 -04001601 return 1;
1602 }
Chris Mason123abc82007-03-14 14:14:43 -04001603 free_space = btrfs_leaf_free_space(root, left);
Chris Mason0783fcf2007-03-12 20:12:07 -04001604 if (free_space < data_size + sizeof(struct btrfs_item)) {
Chris Mason5f39d392007-10-15 16:14:19 -04001605 free_extent_buffer(left);
Chris Mason02217ed2007-03-02 16:08:05 -05001606 return 1;
1607 }
1608
Chris Mason5f39d392007-10-15 16:14:19 -04001609 right_nritems = btrfs_header_nritems(right);
1610 if (right_nritems == 0) {
1611 free_extent_buffer(left);
Chris Masona429e512007-04-18 16:15:28 -04001612 return 1;
1613 }
1614
Chris Mason5f39d392007-10-15 16:14:19 -04001615 for (i = 0; i < right_nritems - 1; i++) {
1616 item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04001617 if (!right->map_token) {
1618 map_extent_buffer(right, (unsigned long)item,
1619 sizeof(struct btrfs_item),
1620 &right->map_token, &right->kaddr,
1621 &right->map_start, &right->map_len,
1622 KM_USER1);
1623 }
1624
Chris Masonbe0e5c02007-01-26 15:51:26 -05001625 if (path->slots[0] == i)
1626 push_space += data_size + sizeof(*item);
Chris Masondb945352007-10-15 16:15:53 -04001627
1628 this_item_size = btrfs_item_size(right, item);
1629 if (this_item_size + sizeof(*item) + push_space > free_space)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001630 break;
Chris Masondb945352007-10-15 16:15:53 -04001631
Chris Masonbe0e5c02007-01-26 15:51:26 -05001632 push_items++;
Chris Masondb945352007-10-15 16:15:53 -04001633 push_space += this_item_size + sizeof(*item);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001634 }
Chris Masondb945352007-10-15 16:15:53 -04001635
1636 if (right->map_token) {
1637 unmap_extent_buffer(right, right->map_token, KM_USER1);
1638 right->map_token = NULL;
1639 }
1640
Chris Masonbe0e5c02007-01-26 15:51:26 -05001641 if (push_items == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04001642 free_extent_buffer(left);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001643 return 1;
1644 }
Chris Mason5f39d392007-10-15 16:14:19 -04001645 if (push_items == btrfs_header_nritems(right))
Chris Masona429e512007-04-18 16:15:28 -04001646 WARN_ON(1);
Chris Mason5f39d392007-10-15 16:14:19 -04001647
Chris Masonbe0e5c02007-01-26 15:51:26 -05001648 /* push data from right to left */
Chris Mason5f39d392007-10-15 16:14:19 -04001649 copy_extent_buffer(left, right,
1650 btrfs_item_nr_offset(btrfs_header_nritems(left)),
1651 btrfs_item_nr_offset(0),
1652 push_items * sizeof(struct btrfs_item));
1653
Chris Mason123abc82007-03-14 14:14:43 -04001654 push_space = BTRFS_LEAF_DATA_SIZE(root) -
Chris Mason5f39d392007-10-15 16:14:19 -04001655 btrfs_item_offset_nr(right, push_items -1);
1656
1657 copy_extent_buffer(left, right, btrfs_leaf_data(left) +
Chris Masond6025572007-03-30 14:27:56 -04001658 leaf_data_end(root, left) - push_space,
1659 btrfs_leaf_data(right) +
Chris Mason5f39d392007-10-15 16:14:19 -04001660 btrfs_item_offset_nr(right, push_items - 1),
Chris Masond6025572007-03-30 14:27:56 -04001661 push_space);
Chris Mason5f39d392007-10-15 16:14:19 -04001662 old_left_nritems = btrfs_header_nritems(left);
Chris Masoneb60cea2007-02-02 09:18:22 -05001663 BUG_ON(old_left_nritems < 0);
1664
Chris Masondb945352007-10-15 16:15:53 -04001665 old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
Chris Mason0783fcf2007-03-12 20:12:07 -04001666 for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04001667 u32 ioff;
Chris Masondb945352007-10-15 16:15:53 -04001668
Chris Mason5f39d392007-10-15 16:14:19 -04001669 item = btrfs_item_nr(left, i);
Chris Masondb945352007-10-15 16:15:53 -04001670 if (!left->map_token) {
1671 map_extent_buffer(left, (unsigned long)item,
1672 sizeof(struct btrfs_item),
1673 &left->map_token, &left->kaddr,
1674 &left->map_start, &left->map_len,
1675 KM_USER1);
1676 }
1677
Chris Mason5f39d392007-10-15 16:14:19 -04001678 ioff = btrfs_item_offset(left, item);
1679 btrfs_set_item_offset(left, item,
Chris Masondb945352007-10-15 16:15:53 -04001680 ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
Chris Masonbe0e5c02007-01-26 15:51:26 -05001681 }
Chris Mason5f39d392007-10-15 16:14:19 -04001682 btrfs_set_header_nritems(left, old_left_nritems + push_items);
Chris Masondb945352007-10-15 16:15:53 -04001683 if (left->map_token) {
1684 unmap_extent_buffer(left, left->map_token, KM_USER1);
1685 left->map_token = NULL;
1686 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05001687
1688 /* fixup right node */
Chris Mason5f39d392007-10-15 16:14:19 -04001689 push_space = btrfs_item_offset_nr(right, push_items - 1) -
1690 leaf_data_end(root, right);
1691 memmove_extent_buffer(right, btrfs_leaf_data(right) +
1692 BTRFS_LEAF_DATA_SIZE(root) - push_space,
1693 btrfs_leaf_data(right) +
1694 leaf_data_end(root, right), push_space);
1695
1696 memmove_extent_buffer(right, btrfs_item_nr_offset(0),
1697 btrfs_item_nr_offset(push_items),
1698 (btrfs_header_nritems(right) - push_items) *
1699 sizeof(struct btrfs_item));
1700
1701 right_nritems = btrfs_header_nritems(right) - push_items;
1702 btrfs_set_header_nritems(right, right_nritems);
Chris Mason123abc82007-03-14 14:14:43 -04001703 push_space = BTRFS_LEAF_DATA_SIZE(root);
Chris Masoneb60cea2007-02-02 09:18:22 -05001704
Chris Mason5f39d392007-10-15 16:14:19 -04001705 for (i = 0; i < right_nritems; i++) {
1706 item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04001707
1708 if (!right->map_token) {
1709 map_extent_buffer(right, (unsigned long)item,
1710 sizeof(struct btrfs_item),
1711 &right->map_token, &right->kaddr,
1712 &right->map_start, &right->map_len,
1713 KM_USER1);
1714 }
1715
1716 push_space = push_space - btrfs_item_size(right, item);
1717 btrfs_set_item_offset(right, item, push_space);
1718 }
1719 if (right->map_token) {
1720 unmap_extent_buffer(right, right->map_token, KM_USER1);
1721 right->map_token = NULL;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001722 }
Chris Masoneb60cea2007-02-02 09:18:22 -05001723
Chris Mason5f39d392007-10-15 16:14:19 -04001724 btrfs_mark_buffer_dirty(left);
1725 btrfs_mark_buffer_dirty(right);
Chris Mason098f59c2007-05-11 11:33:21 -04001726
Chris Mason5f39d392007-10-15 16:14:19 -04001727 btrfs_item_key(right, &disk_key, 0);
1728 wret = fixup_low_keys(trans, root, path, &disk_key, 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001729 if (wret)
1730 ret = wret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001731
1732 /* then fixup the leaf pointer in the path */
1733 if (path->slots[0] < push_items) {
1734 path->slots[0] += old_left_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001735 free_extent_buffer(path->nodes[0]);
1736 path->nodes[0] = left;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001737 path->slots[1] -= 1;
1738 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04001739 free_extent_buffer(left);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001740 path->slots[0] -= push_items;
1741 }
Chris Masoneb60cea2007-02-02 09:18:22 -05001742 BUG_ON(path->slots[0] < 0);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001743 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001744}
1745
Chris Mason74123bd2007-02-02 11:05:29 -05001746/*
1747 * split the path's leaf in two, making sure there is at least data_size
1748 * available for the resulting leaf level of the path.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001749 *
1750 * returns 0 if all went well and < 0 on failure.
Chris Mason74123bd2007-02-02 11:05:29 -05001751 */
Chris Masone089f052007-03-16 16:20:31 -04001752static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
Chris Masond4dbff92007-04-04 14:08:15 -04001753 *root, struct btrfs_key *ins_key,
1754 struct btrfs_path *path, int data_size)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001755{
Chris Mason5f39d392007-10-15 16:14:19 -04001756 struct extent_buffer *l;
Chris Mason7518a232007-03-12 12:01:18 -04001757 u32 nritems;
Chris Masoneb60cea2007-02-02 09:18:22 -05001758 int mid;
1759 int slot;
Chris Mason5f39d392007-10-15 16:14:19 -04001760 struct extent_buffer *right;
Chris Mason0783fcf2007-03-12 20:12:07 -04001761 int space_needed = data_size + sizeof(struct btrfs_item);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001762 int data_copy_size;
1763 int rt_data_off;
1764 int i;
Chris Masond4dbff92007-04-04 14:08:15 -04001765 int ret = 0;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001766 int wret;
Chris Masond4dbff92007-04-04 14:08:15 -04001767 int double_split = 0;
1768 struct btrfs_disk_key disk_key;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001769
Chris Mason40689472007-03-17 14:29:23 -04001770 /* first try to make some room by pushing left and right */
Chris Masone089f052007-03-16 16:20:31 -04001771 wret = push_leaf_left(trans, root, path, data_size);
Chris Masoneaee50e2007-03-13 11:17:52 -04001772 if (wret < 0)
1773 return wret;
1774 if (wret) {
Chris Masone089f052007-03-16 16:20:31 -04001775 wret = push_leaf_right(trans, root, path, data_size);
Chris Masoneaee50e2007-03-13 11:17:52 -04001776 if (wret < 0)
1777 return wret;
1778 }
Chris Mason5f39d392007-10-15 16:14:19 -04001779 l = path->nodes[0];
Chris Masonaa5d6be2007-02-28 16:35:06 -05001780
1781 /* did the pushes work? */
Chris Mason123abc82007-03-14 14:14:43 -04001782 if (btrfs_leaf_free_space(root, l) >=
1783 sizeof(struct btrfs_item) + data_size)
Chris Masonaa5d6be2007-02-28 16:35:06 -05001784 return 0;
1785
Chris Mason5c680ed2007-02-22 11:39:13 -05001786 if (!path->nodes[1]) {
Chris Masone089f052007-03-16 16:20:31 -04001787 ret = insert_new_root(trans, root, path, 1);
Chris Mason5c680ed2007-02-22 11:39:13 -05001788 if (ret)
1789 return ret;
1790 }
Chris Masoneb60cea2007-02-02 09:18:22 -05001791 slot = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04001792 nritems = btrfs_header_nritems(l);
Chris Masoneb60cea2007-02-02 09:18:22 -05001793 mid = (nritems + 1)/ 2;
Chris Mason54aa1f42007-06-22 14:16:25 -04001794
Chris Masondb945352007-10-15 16:15:53 -04001795 right = btrfs_alloc_free_block(trans, root, root->leafsize,
1796 l->start, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04001797 if (IS_ERR(right))
1798 return PTR_ERR(right);
Chris Mason54aa1f42007-06-22 14:16:25 -04001799
Chris Mason5f39d392007-10-15 16:14:19 -04001800 memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
Chris Masondb945352007-10-15 16:15:53 -04001801 btrfs_set_header_bytenr(right, right->start);
Chris Mason5f39d392007-10-15 16:14:19 -04001802 btrfs_set_header_generation(right, trans->transid);
1803 btrfs_set_header_owner(right, root->root_key.objectid);
1804 btrfs_set_header_level(right, 0);
1805 write_extent_buffer(right, root->fs_info->fsid,
1806 (unsigned long)btrfs_header_fsid(right),
1807 BTRFS_FSID_SIZE);
1808
Chris Masond4dbff92007-04-04 14:08:15 -04001809 if (mid <= slot) {
1810 if (nritems == 1 ||
1811 leaf_space_used(l, mid, nritems - mid) + space_needed >
1812 BTRFS_LEAF_DATA_SIZE(root)) {
1813 if (slot >= nritems) {
1814 btrfs_cpu_key_to_disk(&disk_key, ins_key);
Chris Mason5f39d392007-10-15 16:14:19 -04001815 btrfs_set_header_nritems(right, 0);
Chris Masond4dbff92007-04-04 14:08:15 -04001816 wret = insert_ptr(trans, root, path,
Chris Masondb945352007-10-15 16:15:53 -04001817 &disk_key, right->start,
Chris Masond4dbff92007-04-04 14:08:15 -04001818 path->slots[1] + 1, 1);
1819 if (wret)
1820 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04001821 free_extent_buffer(path->nodes[0]);
1822 path->nodes[0] = right;
Chris Masond4dbff92007-04-04 14:08:15 -04001823 path->slots[0] = 0;
1824 path->slots[1] += 1;
1825 return ret;
1826 }
1827 mid = slot;
1828 double_split = 1;
1829 }
1830 } else {
1831 if (leaf_space_used(l, 0, mid + 1) + space_needed >
1832 BTRFS_LEAF_DATA_SIZE(root)) {
1833 if (slot == 0) {
1834 btrfs_cpu_key_to_disk(&disk_key, ins_key);
Chris Mason5f39d392007-10-15 16:14:19 -04001835 btrfs_set_header_nritems(right, 0);
Chris Masond4dbff92007-04-04 14:08:15 -04001836 wret = insert_ptr(trans, root, path,
1837 &disk_key,
Chris Masondb945352007-10-15 16:15:53 -04001838 right->start,
Chris Mason098f59c2007-05-11 11:33:21 -04001839 path->slots[1], 1);
Chris Masond4dbff92007-04-04 14:08:15 -04001840 if (wret)
1841 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04001842 free_extent_buffer(path->nodes[0]);
1843 path->nodes[0] = right;
Chris Masond4dbff92007-04-04 14:08:15 -04001844 path->slots[0] = 0;
Chris Masona429e512007-04-18 16:15:28 -04001845 if (path->slots[1] == 0) {
1846 wret = fixup_low_keys(trans, root,
1847 path, &disk_key, 1);
1848 if (wret)
1849 ret = wret;
1850 }
Chris Masond4dbff92007-04-04 14:08:15 -04001851 return ret;
1852 }
1853 mid = slot;
1854 double_split = 1;
1855 }
1856 }
Chris Mason5f39d392007-10-15 16:14:19 -04001857 nritems = nritems - mid;
1858 btrfs_set_header_nritems(right, nritems);
1859 data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
1860
1861 copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
1862 btrfs_item_nr_offset(mid),
1863 nritems * sizeof(struct btrfs_item));
1864
1865 copy_extent_buffer(right, l,
Chris Masond6025572007-03-30 14:27:56 -04001866 btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
1867 data_copy_size, btrfs_leaf_data(l) +
1868 leaf_data_end(root, l), data_copy_size);
Chris Mason74123bd2007-02-02 11:05:29 -05001869
Chris Mason5f39d392007-10-15 16:14:19 -04001870 rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
1871 btrfs_item_end_nr(l, mid);
1872
1873 for (i = 0; i < nritems; i++) {
1874 struct btrfs_item *item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04001875 u32 ioff;
1876
1877 if (!right->map_token) {
1878 map_extent_buffer(right, (unsigned long)item,
1879 sizeof(struct btrfs_item),
1880 &right->map_token, &right->kaddr,
1881 &right->map_start, &right->map_len,
1882 KM_USER1);
1883 }
1884
1885 ioff = btrfs_item_offset(right, item);
Chris Mason5f39d392007-10-15 16:14:19 -04001886 btrfs_set_item_offset(right, item, ioff + rt_data_off);
Chris Mason0783fcf2007-03-12 20:12:07 -04001887 }
Chris Mason74123bd2007-02-02 11:05:29 -05001888
Chris Masondb945352007-10-15 16:15:53 -04001889 if (right->map_token) {
1890 unmap_extent_buffer(right, right->map_token, KM_USER1);
1891 right->map_token = NULL;
1892 }
1893
Chris Mason5f39d392007-10-15 16:14:19 -04001894 btrfs_set_header_nritems(l, mid);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001895 ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04001896 btrfs_item_key(right, &disk_key, 0);
Chris Masondb945352007-10-15 16:15:53 -04001897 wret = insert_ptr(trans, root, path, &disk_key, right->start,
1898 path->slots[1] + 1, 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001899 if (wret)
1900 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04001901
1902 btrfs_mark_buffer_dirty(right);
1903 btrfs_mark_buffer_dirty(l);
Chris Masoneb60cea2007-02-02 09:18:22 -05001904 BUG_ON(path->slots[0] != slot);
Chris Mason5f39d392007-10-15 16:14:19 -04001905
Chris Masonbe0e5c02007-01-26 15:51:26 -05001906 if (mid <= slot) {
Chris Mason5f39d392007-10-15 16:14:19 -04001907 free_extent_buffer(path->nodes[0]);
1908 path->nodes[0] = right;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001909 path->slots[0] -= mid;
1910 path->slots[1] += 1;
Chris Masoneb60cea2007-02-02 09:18:22 -05001911 } else
Chris Mason5f39d392007-10-15 16:14:19 -04001912 free_extent_buffer(right);
1913
Chris Masoneb60cea2007-02-02 09:18:22 -05001914 BUG_ON(path->slots[0] < 0);
Chris Masond4dbff92007-04-04 14:08:15 -04001915
1916 if (!double_split)
1917 return ret;
Chris Mason54aa1f42007-06-22 14:16:25 -04001918
Chris Masondb945352007-10-15 16:15:53 -04001919 right = btrfs_alloc_free_block(trans, root, root->leafsize,
1920 l->start, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04001921 if (IS_ERR(right))
1922 return PTR_ERR(right);
1923
1924 memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
Chris Masondb945352007-10-15 16:15:53 -04001925 btrfs_set_header_bytenr(right, right->start);
Chris Mason5f39d392007-10-15 16:14:19 -04001926 btrfs_set_header_generation(right, trans->transid);
1927 btrfs_set_header_owner(right, root->root_key.objectid);
1928 btrfs_set_header_level(right, 0);
1929 write_extent_buffer(right, root->fs_info->fsid,
1930 (unsigned long)btrfs_header_fsid(right),
1931 BTRFS_FSID_SIZE);
1932
Chris Masond4dbff92007-04-04 14:08:15 -04001933 btrfs_cpu_key_to_disk(&disk_key, ins_key);
Chris Mason5f39d392007-10-15 16:14:19 -04001934 btrfs_set_header_nritems(right, 0);
Chris Masond4dbff92007-04-04 14:08:15 -04001935 wret = insert_ptr(trans, root, path,
Chris Masondb945352007-10-15 16:15:53 -04001936 &disk_key, right->start,
Chris Masond4dbff92007-04-04 14:08:15 -04001937 path->slots[1], 1);
1938 if (wret)
1939 ret = wret;
Chris Masona429e512007-04-18 16:15:28 -04001940 if (path->slots[1] == 0) {
1941 wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1942 if (wret)
1943 ret = wret;
1944 }
Chris Mason5f39d392007-10-15 16:14:19 -04001945 free_extent_buffer(path->nodes[0]);
1946 path->nodes[0] = right;
Chris Masond4dbff92007-04-04 14:08:15 -04001947 path->slots[0] = 0;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001948 return ret;
1949}
1950
Chris Masonb18c6682007-04-17 13:26:50 -04001951int btrfs_truncate_item(struct btrfs_trans_handle *trans,
1952 struct btrfs_root *root,
1953 struct btrfs_path *path,
1954 u32 new_size)
1955{
1956 int ret = 0;
1957 int slot;
1958 int slot_orig;
Chris Mason5f39d392007-10-15 16:14:19 -04001959 struct extent_buffer *leaf;
1960 struct btrfs_item *item;
Chris Masonb18c6682007-04-17 13:26:50 -04001961 u32 nritems;
1962 unsigned int data_end;
1963 unsigned int old_data_start;
1964 unsigned int old_size;
1965 unsigned int size_diff;
1966 int i;
1967
1968 slot_orig = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04001969 leaf = path->nodes[0];
Chris Masonb18c6682007-04-17 13:26:50 -04001970
Chris Mason5f39d392007-10-15 16:14:19 -04001971 nritems = btrfs_header_nritems(leaf);
Chris Masonb18c6682007-04-17 13:26:50 -04001972 data_end = leaf_data_end(root, leaf);
1973
1974 slot = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04001975 old_data_start = btrfs_item_offset_nr(leaf, slot);
1976 old_size = btrfs_item_size_nr(leaf, slot);
Chris Masonb18c6682007-04-17 13:26:50 -04001977 BUG_ON(old_size <= new_size);
1978 size_diff = old_size - new_size;
1979
1980 BUG_ON(slot < 0);
1981 BUG_ON(slot >= nritems);
1982
1983 /*
1984 * item0..itemN ... dataN.offset..dataN.size .. data0.size
1985 */
1986 /* first correct the data pointers */
1987 for (i = slot; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04001988 u32 ioff;
1989 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04001990
1991 if (!leaf->map_token) {
1992 map_extent_buffer(leaf, (unsigned long)item,
1993 sizeof(struct btrfs_item),
1994 &leaf->map_token, &leaf->kaddr,
1995 &leaf->map_start, &leaf->map_len,
1996 KM_USER1);
1997 }
1998
Chris Mason5f39d392007-10-15 16:14:19 -04001999 ioff = btrfs_item_offset(leaf, item);
2000 btrfs_set_item_offset(leaf, item, ioff + size_diff);
Chris Masonb18c6682007-04-17 13:26:50 -04002001 }
Chris Masondb945352007-10-15 16:15:53 -04002002
2003 if (leaf->map_token) {
2004 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2005 leaf->map_token = NULL;
2006 }
2007
Chris Masonb18c6682007-04-17 13:26:50 -04002008 /* shift the data */
Chris Mason5f39d392007-10-15 16:14:19 -04002009 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
Chris Masonb18c6682007-04-17 13:26:50 -04002010 data_end + size_diff, btrfs_leaf_data(leaf) +
2011 data_end, old_data_start + new_size - data_end);
Chris Mason5f39d392007-10-15 16:14:19 -04002012
2013 item = btrfs_item_nr(leaf, slot);
2014 btrfs_set_item_size(leaf, item, new_size);
2015 btrfs_mark_buffer_dirty(leaf);
Chris Masonb18c6682007-04-17 13:26:50 -04002016
2017 ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04002018 if (btrfs_leaf_free_space(root, leaf) < 0) {
2019 btrfs_print_leaf(root, leaf);
Chris Masonb18c6682007-04-17 13:26:50 -04002020 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002021 }
Chris Masonb18c6682007-04-17 13:26:50 -04002022 return ret;
2023}
2024
Chris Mason5f39d392007-10-15 16:14:19 -04002025int btrfs_extend_item(struct btrfs_trans_handle *trans,
2026 struct btrfs_root *root, struct btrfs_path *path,
2027 u32 data_size)
Chris Mason6567e832007-04-16 09:22:45 -04002028{
2029 int ret = 0;
2030 int slot;
2031 int slot_orig;
Chris Mason5f39d392007-10-15 16:14:19 -04002032 struct extent_buffer *leaf;
2033 struct btrfs_item *item;
Chris Mason6567e832007-04-16 09:22:45 -04002034 u32 nritems;
2035 unsigned int data_end;
2036 unsigned int old_data;
2037 unsigned int old_size;
2038 int i;
2039
2040 slot_orig = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002041 leaf = path->nodes[0];
Chris Mason6567e832007-04-16 09:22:45 -04002042
Chris Mason5f39d392007-10-15 16:14:19 -04002043 nritems = btrfs_header_nritems(leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002044 data_end = leaf_data_end(root, leaf);
2045
Chris Mason5f39d392007-10-15 16:14:19 -04002046 if (btrfs_leaf_free_space(root, leaf) < data_size) {
2047 btrfs_print_leaf(root, leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002048 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002049 }
Chris Mason6567e832007-04-16 09:22:45 -04002050 slot = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002051 old_data = btrfs_item_end_nr(leaf, slot);
Chris Mason6567e832007-04-16 09:22:45 -04002052
2053 BUG_ON(slot < 0);
2054 BUG_ON(slot >= nritems);
2055
2056 /*
2057 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2058 */
2059 /* first correct the data pointers */
2060 for (i = slot; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04002061 u32 ioff;
2062 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04002063
2064 if (!leaf->map_token) {
2065 map_extent_buffer(leaf, (unsigned long)item,
2066 sizeof(struct btrfs_item),
2067 &leaf->map_token, &leaf->kaddr,
2068 &leaf->map_start, &leaf->map_len,
2069 KM_USER1);
2070 }
Chris Mason5f39d392007-10-15 16:14:19 -04002071 ioff = btrfs_item_offset(leaf, item);
2072 btrfs_set_item_offset(leaf, item, ioff - data_size);
Chris Mason6567e832007-04-16 09:22:45 -04002073 }
Chris Mason5f39d392007-10-15 16:14:19 -04002074
Chris Masondb945352007-10-15 16:15:53 -04002075 if (leaf->map_token) {
2076 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2077 leaf->map_token = NULL;
2078 }
2079
Chris Mason6567e832007-04-16 09:22:45 -04002080 /* shift the data */
Chris Mason5f39d392007-10-15 16:14:19 -04002081 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
Chris Mason6567e832007-04-16 09:22:45 -04002082 data_end - data_size, btrfs_leaf_data(leaf) +
2083 data_end, old_data - data_end);
Chris Mason5f39d392007-10-15 16:14:19 -04002084
Chris Mason6567e832007-04-16 09:22:45 -04002085 data_end = old_data;
Chris Mason5f39d392007-10-15 16:14:19 -04002086 old_size = btrfs_item_size_nr(leaf, slot);
2087 item = btrfs_item_nr(leaf, slot);
2088 btrfs_set_item_size(leaf, item, old_size + data_size);
2089 btrfs_mark_buffer_dirty(leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002090
2091 ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04002092 if (btrfs_leaf_free_space(root, leaf) < 0) {
2093 btrfs_print_leaf(root, leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002094 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002095 }
Chris Mason6567e832007-04-16 09:22:45 -04002096 return ret;
2097}
2098
Chris Mason74123bd2007-02-02 11:05:29 -05002099/*
2100 * Given a key and some data, insert an item into the tree.
2101 * This does all the path init required, making room in the tree if needed.
2102 */
Chris Mason5f39d392007-10-15 16:14:19 -04002103int btrfs_insert_empty_item(struct btrfs_trans_handle *trans,
2104 struct btrfs_root *root,
2105 struct btrfs_path *path,
2106 struct btrfs_key *cpu_key, u32 data_size)
Chris Masonbe0e5c02007-01-26 15:51:26 -05002107{
Chris Mason5f39d392007-10-15 16:14:19 -04002108 struct extent_buffer *leaf;
2109 struct btrfs_item *item;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002110 int ret = 0;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002111 int slot;
Chris Masoneb60cea2007-02-02 09:18:22 -05002112 int slot_orig;
Chris Mason7518a232007-03-12 12:01:18 -04002113 u32 nritems;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002114 unsigned int data_end;
Chris Masone2fa7222007-03-12 16:22:34 -04002115 struct btrfs_disk_key disk_key;
2116
2117 btrfs_cpu_key_to_disk(&disk_key, cpu_key);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002118
Chris Mason74123bd2007-02-02 11:05:29 -05002119 /* create a root if there isn't one */
Chris Mason5c680ed2007-02-22 11:39:13 -05002120 if (!root->node)
Chris Masoncfaa7292007-02-21 17:04:57 -05002121 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002122
Chris Masone089f052007-03-16 16:20:31 -04002123 ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1);
Chris Masoneb60cea2007-02-02 09:18:22 -05002124 if (ret == 0) {
Chris Masonf0930a32007-03-02 09:47:58 -05002125 return -EEXIST;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002126 }
Chris Masoned2ff2c2007-03-01 18:59:40 -05002127 if (ret < 0)
2128 goto out;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002129
Chris Mason62e27492007-03-15 12:56:47 -04002130 slot_orig = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002131 leaf = path->nodes[0];
Chris Mason74123bd2007-02-02 11:05:29 -05002132
Chris Mason5f39d392007-10-15 16:14:19 -04002133 nritems = btrfs_header_nritems(leaf);
Chris Mason123abc82007-03-14 14:14:43 -04002134 data_end = leaf_data_end(root, leaf);
Chris Masoneb60cea2007-02-02 09:18:22 -05002135
Chris Mason123abc82007-03-14 14:14:43 -04002136 if (btrfs_leaf_free_space(root, leaf) <
Chris Masond4dbff92007-04-04 14:08:15 -04002137 sizeof(struct btrfs_item) + data_size) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05002138 BUG();
Chris Masond4dbff92007-04-04 14:08:15 -04002139 }
Chris Mason5f39d392007-10-15 16:14:19 -04002140
Chris Mason62e27492007-03-15 12:56:47 -04002141 slot = path->slots[0];
Chris Masoneb60cea2007-02-02 09:18:22 -05002142 BUG_ON(slot < 0);
Chris Mason5f39d392007-10-15 16:14:19 -04002143
Chris Masonbe0e5c02007-01-26 15:51:26 -05002144 if (slot != nritems) {
2145 int i;
Chris Mason5f39d392007-10-15 16:14:19 -04002146 unsigned int old_data = btrfs_item_end_nr(leaf, slot);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002147
Chris Mason5f39d392007-10-15 16:14:19 -04002148 if (old_data < data_end) {
2149 btrfs_print_leaf(root, leaf);
2150 printk("slot %d old_data %d data_end %d\n",
2151 slot, old_data, data_end);
2152 BUG_ON(1);
2153 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002154 /*
2155 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2156 */
2157 /* first correct the data pointers */
Chris Masondb945352007-10-15 16:15:53 -04002158 WARN_ON(leaf->map_token);
Chris Mason0783fcf2007-03-12 20:12:07 -04002159 for (i = slot; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04002160 u32 ioff;
Chris Masondb945352007-10-15 16:15:53 -04002161
Chris Mason5f39d392007-10-15 16:14:19 -04002162 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04002163 if (!leaf->map_token) {
2164 map_extent_buffer(leaf, (unsigned long)item,
2165 sizeof(struct btrfs_item),
2166 &leaf->map_token, &leaf->kaddr,
2167 &leaf->map_start, &leaf->map_len,
2168 KM_USER1);
2169 }
2170
Chris Mason5f39d392007-10-15 16:14:19 -04002171 ioff = btrfs_item_offset(leaf, item);
2172 btrfs_set_item_offset(leaf, item, ioff - data_size);
Chris Mason0783fcf2007-03-12 20:12:07 -04002173 }
Chris Masondb945352007-10-15 16:15:53 -04002174 if (leaf->map_token) {
2175 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2176 leaf->map_token = NULL;
2177 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002178
2179 /* shift the items */
Chris Mason5f39d392007-10-15 16:14:19 -04002180 memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
2181 btrfs_item_nr_offset(slot),
Chris Masond6025572007-03-30 14:27:56 -04002182 (nritems - slot) * sizeof(struct btrfs_item));
Chris Masonbe0e5c02007-01-26 15:51:26 -05002183
2184 /* shift the data */
Chris Mason5f39d392007-10-15 16:14:19 -04002185 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
Chris Masond6025572007-03-30 14:27:56 -04002186 data_end - data_size, btrfs_leaf_data(leaf) +
2187 data_end, old_data - data_end);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002188 data_end = old_data;
2189 }
Chris Mason5f39d392007-10-15 16:14:19 -04002190
Chris Mason62e27492007-03-15 12:56:47 -04002191 /* setup the item for the new data */
Chris Mason5f39d392007-10-15 16:14:19 -04002192 btrfs_set_item_key(leaf, &disk_key, slot);
2193 item = btrfs_item_nr(leaf, slot);
2194 btrfs_set_item_offset(leaf, item, data_end - data_size);
2195 btrfs_set_item_size(leaf, item, data_size);
2196 btrfs_set_header_nritems(leaf, nritems + 1);
2197 btrfs_mark_buffer_dirty(leaf);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002198
2199 ret = 0;
Chris Mason8e19f2c2007-02-28 09:27:02 -05002200 if (slot == 0)
Chris Masone089f052007-03-16 16:20:31 -04002201 ret = fixup_low_keys(trans, root, path, &disk_key, 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002202
Chris Mason5f39d392007-10-15 16:14:19 -04002203 if (btrfs_leaf_free_space(root, leaf) < 0) {
2204 btrfs_print_leaf(root, leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002205 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002206 }
Chris Masoned2ff2c2007-03-01 18:59:40 -05002207out:
Chris Mason62e27492007-03-15 12:56:47 -04002208 return ret;
2209}
2210
2211/*
2212 * Given a key and some data, insert an item into the tree.
2213 * This does all the path init required, making room in the tree if needed.
2214 */
Chris Masone089f052007-03-16 16:20:31 -04002215int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2216 *root, struct btrfs_key *cpu_key, void *data, u32
2217 data_size)
Chris Mason62e27492007-03-15 12:56:47 -04002218{
2219 int ret = 0;
Chris Mason2c90e5d2007-04-02 10:50:19 -04002220 struct btrfs_path *path;
Chris Mason5f39d392007-10-15 16:14:19 -04002221 struct extent_buffer *leaf;
2222 unsigned long ptr;
Chris Mason62e27492007-03-15 12:56:47 -04002223
Chris Mason2c90e5d2007-04-02 10:50:19 -04002224 path = btrfs_alloc_path();
2225 BUG_ON(!path);
Chris Mason2c90e5d2007-04-02 10:50:19 -04002226 ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
Chris Mason62e27492007-03-15 12:56:47 -04002227 if (!ret) {
Chris Mason5f39d392007-10-15 16:14:19 -04002228 leaf = path->nodes[0];
2229 ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
2230 write_extent_buffer(leaf, data, ptr, data_size);
2231 btrfs_mark_buffer_dirty(leaf);
Chris Mason62e27492007-03-15 12:56:47 -04002232 }
Chris Mason2c90e5d2007-04-02 10:50:19 -04002233 btrfs_free_path(path);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002234 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002235}
2236
Chris Mason74123bd2007-02-02 11:05:29 -05002237/*
Chris Mason5de08d72007-02-24 06:24:44 -05002238 * delete the pointer from a given node.
Chris Mason74123bd2007-02-02 11:05:29 -05002239 *
2240 * If the delete empties a node, the node is removed from the tree,
2241 * continuing all the way the root if required. The root is converted into
2242 * a leaf if all the nodes are emptied.
2243 */
Chris Masone089f052007-03-16 16:20:31 -04002244static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2245 struct btrfs_path *path, int level, int slot)
Chris Masonbe0e5c02007-01-26 15:51:26 -05002246{
Chris Mason5f39d392007-10-15 16:14:19 -04002247 struct extent_buffer *parent = path->nodes[level];
Chris Mason7518a232007-03-12 12:01:18 -04002248 u32 nritems;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002249 int ret = 0;
Chris Masonbb803952007-03-01 12:04:21 -05002250 int wret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002251
Chris Mason5f39d392007-10-15 16:14:19 -04002252 nritems = btrfs_header_nritems(parent);
Chris Masonbb803952007-03-01 12:04:21 -05002253 if (slot != nritems -1) {
Chris Mason5f39d392007-10-15 16:14:19 -04002254 memmove_extent_buffer(parent,
2255 btrfs_node_key_ptr_offset(slot),
2256 btrfs_node_key_ptr_offset(slot + 1),
Chris Masond6025572007-03-30 14:27:56 -04002257 sizeof(struct btrfs_key_ptr) *
2258 (nritems - slot - 1));
Chris Masonbb803952007-03-01 12:04:21 -05002259 }
Chris Mason7518a232007-03-12 12:01:18 -04002260 nritems--;
Chris Mason5f39d392007-10-15 16:14:19 -04002261 btrfs_set_header_nritems(parent, nritems);
Chris Mason7518a232007-03-12 12:01:18 -04002262 if (nritems == 0 && parent == root->node) {
Chris Mason5f39d392007-10-15 16:14:19 -04002263 BUG_ON(btrfs_header_level(root->node) != 1);
Chris Masonbb803952007-03-01 12:04:21 -05002264 /* just turn the root into a leaf and break */
Chris Mason5f39d392007-10-15 16:14:19 -04002265 btrfs_set_header_level(root->node, 0);
Chris Masonbb803952007-03-01 12:04:21 -05002266 } else if (slot == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04002267 struct btrfs_disk_key disk_key;
2268
2269 btrfs_node_key(parent, &disk_key, 0);
2270 wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
Chris Mason0f70abe2007-02-28 16:46:22 -05002271 if (wret)
2272 ret = wret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002273 }
Chris Masond6025572007-03-30 14:27:56 -04002274 btrfs_mark_buffer_dirty(parent);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002275 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002276}
2277
Chris Mason74123bd2007-02-02 11:05:29 -05002278/*
2279 * delete the item at the leaf level in path. If that empties
2280 * the leaf, remove it from the tree
2281 */
Chris Masone089f052007-03-16 16:20:31 -04002282int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2283 struct btrfs_path *path)
Chris Masonbe0e5c02007-01-26 15:51:26 -05002284{
Chris Masonbe0e5c02007-01-26 15:51:26 -05002285 int slot;
Chris Mason5f39d392007-10-15 16:14:19 -04002286 struct extent_buffer *leaf;
2287 struct btrfs_item *item;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002288 int doff;
2289 int dsize;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002290 int ret = 0;
2291 int wret;
Chris Mason7518a232007-03-12 12:01:18 -04002292 u32 nritems;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002293
Chris Mason5f39d392007-10-15 16:14:19 -04002294 leaf = path->nodes[0];
Chris Mason4920c9a2007-01-26 16:38:42 -05002295 slot = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002296 doff = btrfs_item_offset_nr(leaf, slot);
2297 dsize = btrfs_item_size_nr(leaf, slot);
2298 nritems = btrfs_header_nritems(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002299
Chris Mason7518a232007-03-12 12:01:18 -04002300 if (slot != nritems - 1) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05002301 int i;
Chris Mason123abc82007-03-14 14:14:43 -04002302 int data_end = leaf_data_end(root, leaf);
Chris Mason5f39d392007-10-15 16:14:19 -04002303
2304 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
Chris Masond6025572007-03-30 14:27:56 -04002305 data_end + dsize,
2306 btrfs_leaf_data(leaf) + data_end,
2307 doff - data_end);
Chris Mason5f39d392007-10-15 16:14:19 -04002308
Chris Mason0783fcf2007-03-12 20:12:07 -04002309 for (i = slot + 1; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04002310 u32 ioff;
Chris Masondb945352007-10-15 16:15:53 -04002311
Chris Mason5f39d392007-10-15 16:14:19 -04002312 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04002313 if (!leaf->map_token) {
2314 map_extent_buffer(leaf, (unsigned long)item,
2315 sizeof(struct btrfs_item),
2316 &leaf->map_token, &leaf->kaddr,
2317 &leaf->map_start, &leaf->map_len,
2318 KM_USER1);
2319 }
Chris Mason5f39d392007-10-15 16:14:19 -04002320 ioff = btrfs_item_offset(leaf, item);
2321 btrfs_set_item_offset(leaf, item, ioff + dsize);
Chris Mason0783fcf2007-03-12 20:12:07 -04002322 }
Chris Masondb945352007-10-15 16:15:53 -04002323
2324 if (leaf->map_token) {
2325 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2326 leaf->map_token = NULL;
2327 }
2328
Chris Mason5f39d392007-10-15 16:14:19 -04002329 memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
2330 btrfs_item_nr_offset(slot + 1),
Chris Masond6025572007-03-30 14:27:56 -04002331 sizeof(struct btrfs_item) *
2332 (nritems - slot - 1));
Chris Masonbe0e5c02007-01-26 15:51:26 -05002333 }
Chris Mason5f39d392007-10-15 16:14:19 -04002334 btrfs_set_header_nritems(leaf, nritems - 1);
Chris Mason7518a232007-03-12 12:01:18 -04002335 nritems--;
Chris Mason5f39d392007-10-15 16:14:19 -04002336
Chris Mason74123bd2007-02-02 11:05:29 -05002337 /* delete the leaf if we've emptied it */
Chris Mason7518a232007-03-12 12:01:18 -04002338 if (nritems == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04002339 if (leaf == root->node) {
2340 btrfs_set_header_level(leaf, 0);
Chris Mason9a8dd152007-02-23 08:38:36 -05002341 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04002342 clean_tree_block(trans, root, leaf);
2343 wait_on_tree_block_writeback(root, leaf);
Chris Masone089f052007-03-16 16:20:31 -04002344 wret = del_ptr(trans, root, path, 1, path->slots[1]);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002345 if (wret)
2346 ret = wret;
Chris Masone089f052007-03-16 16:20:31 -04002347 wret = btrfs_free_extent(trans, root,
Chris Masondb945352007-10-15 16:15:53 -04002348 leaf->start, leaf->len, 1);
Chris Mason0f70abe2007-02-28 16:46:22 -05002349 if (wret)
2350 ret = wret;
Chris Mason9a8dd152007-02-23 08:38:36 -05002351 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002352 } else {
Chris Mason7518a232007-03-12 12:01:18 -04002353 int used = leaf_space_used(leaf, 0, nritems);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002354 if (slot == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04002355 struct btrfs_disk_key disk_key;
2356
2357 btrfs_item_key(leaf, &disk_key, 0);
Chris Masone089f052007-03-16 16:20:31 -04002358 wret = fixup_low_keys(trans, root, path,
Chris Mason5f39d392007-10-15 16:14:19 -04002359 &disk_key, 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002360 if (wret)
2361 ret = wret;
2362 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05002363
Chris Mason74123bd2007-02-02 11:05:29 -05002364 /* delete the leaf if it is mostly empty */
Chris Mason123abc82007-03-14 14:14:43 -04002365 if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05002366 /* push_leaf_left fixes the path.
2367 * make sure the path still points to our leaf
2368 * for possible call to del_ptr below
2369 */
Chris Mason4920c9a2007-01-26 16:38:42 -05002370 slot = path->slots[1];
Chris Mason5f39d392007-10-15 16:14:19 -04002371 extent_buffer_get(leaf);
2372
Chris Masone089f052007-03-16 16:20:31 -04002373 wret = push_leaf_left(trans, root, path, 1);
Chris Mason54aa1f42007-06-22 14:16:25 -04002374 if (wret < 0 && wret != -ENOSPC)
Chris Masonaa5d6be2007-02-28 16:35:06 -05002375 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04002376
2377 if (path->nodes[0] == leaf &&
2378 btrfs_header_nritems(leaf)) {
Chris Masone089f052007-03-16 16:20:31 -04002379 wret = push_leaf_right(trans, root, path, 1);
Chris Mason54aa1f42007-06-22 14:16:25 -04002380 if (wret < 0 && wret != -ENOSPC)
Chris Masonaa5d6be2007-02-28 16:35:06 -05002381 ret = wret;
2382 }
Chris Mason5f39d392007-10-15 16:14:19 -04002383
2384 if (btrfs_header_nritems(leaf) == 0) {
Chris Masondb945352007-10-15 16:15:53 -04002385 u64 bytenr = leaf->start;
2386 u32 blocksize = leaf->len;
Chris Mason5f39d392007-10-15 16:14:19 -04002387
2388 clean_tree_block(trans, root, leaf);
2389 wait_on_tree_block_writeback(root, leaf);
2390
Chris Masone089f052007-03-16 16:20:31 -04002391 wret = del_ptr(trans, root, path, 1, slot);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002392 if (wret)
2393 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04002394
2395 free_extent_buffer(leaf);
Chris Masondb945352007-10-15 16:15:53 -04002396 wret = btrfs_free_extent(trans, root, bytenr,
2397 blocksize, 1);
Chris Mason0f70abe2007-02-28 16:46:22 -05002398 if (wret)
2399 ret = wret;
Chris Mason5de08d72007-02-24 06:24:44 -05002400 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04002401 btrfs_mark_buffer_dirty(leaf);
2402 free_extent_buffer(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002403 }
Chris Masond5719762007-03-23 10:01:08 -04002404 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04002405 btrfs_mark_buffer_dirty(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002406 }
2407 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05002408 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002409}
2410
Chris Mason97571fd2007-02-24 13:39:08 -05002411/*
2412 * walk up the tree as far as required to find the next leaf.
Chris Mason0f70abe2007-02-28 16:46:22 -05002413 * returns 0 if it found something or 1 if there are no greater leaves.
2414 * returns < 0 on io errors.
Chris Mason97571fd2007-02-24 13:39:08 -05002415 */
Chris Mason234b63a2007-03-13 10:46:10 -04002416int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
Chris Masond97e63b2007-02-20 16:40:44 -05002417{
2418 int slot;
2419 int level = 1;
Chris Masondb945352007-10-15 16:15:53 -04002420 u64 bytenr;
Chris Mason5f39d392007-10-15 16:14:19 -04002421 struct extent_buffer *c;
2422 struct extent_buffer *next = NULL;
Chris Masond97e63b2007-02-20 16:40:44 -05002423
Chris Mason234b63a2007-03-13 10:46:10 -04002424 while(level < BTRFS_MAX_LEVEL) {
Chris Masond97e63b2007-02-20 16:40:44 -05002425 if (!path->nodes[level])
Chris Mason0f70abe2007-02-28 16:46:22 -05002426 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04002427
Chris Masond97e63b2007-02-20 16:40:44 -05002428 slot = path->slots[level] + 1;
2429 c = path->nodes[level];
Chris Mason5f39d392007-10-15 16:14:19 -04002430 if (slot >= btrfs_header_nritems(c)) {
Chris Masond97e63b2007-02-20 16:40:44 -05002431 level++;
2432 continue;
2433 }
Chris Mason5f39d392007-10-15 16:14:19 -04002434
Chris Masondb945352007-10-15 16:15:53 -04002435 bytenr = btrfs_node_blockptr(c, slot);
Chris Masoncfaa7292007-02-21 17:04:57 -05002436 if (next)
Chris Mason5f39d392007-10-15 16:14:19 -04002437 free_extent_buffer(next);
2438
Chris Mason6702ed42007-08-07 16:15:09 -04002439 if (path->reada)
2440 reada_for_search(root, path, level, slot);
Chris Mason5f39d392007-10-15 16:14:19 -04002441
Chris Masondb945352007-10-15 16:15:53 -04002442 next = read_tree_block(root, bytenr,
2443 btrfs_level_size(root, level -1));
Chris Masond97e63b2007-02-20 16:40:44 -05002444 break;
2445 }
2446 path->slots[level] = slot;
2447 while(1) {
2448 level--;
2449 c = path->nodes[level];
Chris Mason5f39d392007-10-15 16:14:19 -04002450 free_extent_buffer(c);
Chris Masond97e63b2007-02-20 16:40:44 -05002451 path->nodes[level] = next;
2452 path->slots[level] = 0;
2453 if (!level)
2454 break;
Chris Mason6702ed42007-08-07 16:15:09 -04002455 if (path->reada)
Yan32020612007-08-27 15:17:54 -04002456 reada_for_search(root, path, level, 0);
Chris Masondb945352007-10-15 16:15:53 -04002457 next = read_tree_block(root, btrfs_node_blockptr(next, 0),
2458 btrfs_level_size(root, level - 1));
Chris Masond97e63b2007-02-20 16:40:44 -05002459 }
2460 return 0;
2461}