blob: 70b6ddfe15a18d172cb8e6e5a109cc1c550ab314 [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 Masona6b6e752007-10-15 16:22:39 -040019#include <linux/sched.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,
Chris Masoncc0c5532007-10-25 15:42:57 -040029 struct btrfs_path *path, int data_size, int extend);
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 Mason0b86a832008-03-24 15:01:56 -040073static void add_root_to_dirty_list(struct btrfs_root *root)
74{
75 if (root->track_dirty && list_empty(&root->dirty_list)) {
76 list_add(&root->dirty_list,
77 &root->fs_info->dirty_cowonly_roots);
78 }
79}
80
Chris Masonbe20aa92007-12-17 20:14:01 -050081int btrfs_copy_root(struct btrfs_trans_handle *trans,
82 struct btrfs_root *root,
83 struct extent_buffer *buf,
84 struct extent_buffer **cow_ret, u64 new_root_objectid)
85{
86 struct extent_buffer *cow;
87 u32 nritems;
88 int ret = 0;
89 int level;
90 struct btrfs_key first_key;
Chris Mason4aec2b52007-12-18 16:25:45 -050091 struct btrfs_root *new_root;
Chris Masonbe20aa92007-12-17 20:14:01 -050092
Chris Mason4aec2b52007-12-18 16:25:45 -050093 new_root = kmalloc(sizeof(*new_root), GFP_NOFS);
94 if (!new_root)
95 return -ENOMEM;
96
97 memcpy(new_root, root, sizeof(*new_root));
98 new_root->root_key.objectid = new_root_objectid;
Chris Masonbe20aa92007-12-17 20:14:01 -050099
100 WARN_ON(root->ref_cows && trans->transid !=
101 root->fs_info->running_transaction->transid);
102 WARN_ON(root->ref_cows && trans->transid != root->last_trans);
103
104 level = btrfs_header_level(buf);
105 nritems = btrfs_header_nritems(buf);
106 if (nritems) {
107 if (level == 0)
108 btrfs_item_key_to_cpu(buf, &first_key, 0);
109 else
110 btrfs_node_key_to_cpu(buf, &first_key, 0);
111 } else {
112 first_key.objectid = 0;
113 }
Chris Mason4aec2b52007-12-18 16:25:45 -0500114 cow = __btrfs_alloc_free_block(trans, new_root, buf->len,
Chris Masonbe20aa92007-12-17 20:14:01 -0500115 new_root_objectid,
116 trans->transid, first_key.objectid,
117 level, buf->start, 0);
Chris Mason4aec2b52007-12-18 16:25:45 -0500118 if (IS_ERR(cow)) {
119 kfree(new_root);
Chris Masonbe20aa92007-12-17 20:14:01 -0500120 return PTR_ERR(cow);
Chris Mason4aec2b52007-12-18 16:25:45 -0500121 }
Chris Masonbe20aa92007-12-17 20:14:01 -0500122
123 copy_extent_buffer(cow, buf, 0, 0, cow->len);
124 btrfs_set_header_bytenr(cow, cow->start);
125 btrfs_set_header_generation(cow, trans->transid);
126 btrfs_set_header_owner(cow, new_root_objectid);
127
128 WARN_ON(btrfs_header_generation(buf) > trans->transid);
Chris Mason4aec2b52007-12-18 16:25:45 -0500129 ret = btrfs_inc_ref(trans, new_root, buf);
130 kfree(new_root);
131
Chris Masonbe20aa92007-12-17 20:14:01 -0500132 if (ret)
133 return ret;
134
135 btrfs_mark_buffer_dirty(cow);
136 *cow_ret = cow;
137 return 0;
138}
139
140int __btrfs_cow_block(struct btrfs_trans_handle *trans,
Chris Mason5f39d392007-10-15 16:14:19 -0400141 struct btrfs_root *root,
142 struct extent_buffer *buf,
143 struct extent_buffer *parent, int parent_slot,
144 struct extent_buffer **cow_ret,
145 u64 search_start, u64 empty_size)
Chris Mason6702ed42007-08-07 16:15:09 -0400146{
Chris Mason7bb86312007-12-11 09:25:06 -0500147 u64 root_gen;
Chris Mason5f39d392007-10-15 16:14:19 -0400148 struct extent_buffer *cow;
Chris Mason7bb86312007-12-11 09:25:06 -0500149 u32 nritems;
Chris Mason6702ed42007-08-07 16:15:09 -0400150 int ret = 0;
151 int different_trans = 0;
Chris Mason7bb86312007-12-11 09:25:06 -0500152 int level;
153 struct btrfs_key first_key;
Chris Mason6702ed42007-08-07 16:15:09 -0400154
Chris Mason7bb86312007-12-11 09:25:06 -0500155 if (root->ref_cows) {
156 root_gen = trans->transid;
157 } else {
158 root_gen = 0;
159 }
160
161 WARN_ON(root->ref_cows && trans->transid !=
162 root->fs_info->running_transaction->transid);
Chris Mason6702ed42007-08-07 16:15:09 -0400163 WARN_ON(root->ref_cows && trans->transid != root->last_trans);
Chris Mason5f39d392007-10-15 16:14:19 -0400164
Chris Mason7bb86312007-12-11 09:25:06 -0500165 level = btrfs_header_level(buf);
166 nritems = btrfs_header_nritems(buf);
167 if (nritems) {
168 if (level == 0)
169 btrfs_item_key_to_cpu(buf, &first_key, 0);
170 else
171 btrfs_node_key_to_cpu(buf, &first_key, 0);
172 } else {
173 first_key.objectid = 0;
174 }
175 cow = __btrfs_alloc_free_block(trans, root, buf->len,
176 root->root_key.objectid,
177 root_gen, first_key.objectid, level,
Chris Masondb945352007-10-15 16:15:53 -0400178 search_start, empty_size);
Chris Mason6702ed42007-08-07 16:15:09 -0400179 if (IS_ERR(cow))
180 return PTR_ERR(cow);
181
Chris Mason5f39d392007-10-15 16:14:19 -0400182 copy_extent_buffer(cow, buf, 0, 0, cow->len);
Chris Masondb945352007-10-15 16:15:53 -0400183 btrfs_set_header_bytenr(cow, cow->start);
Chris Mason5f39d392007-10-15 16:14:19 -0400184 btrfs_set_header_generation(cow, trans->transid);
185 btrfs_set_header_owner(cow, root->root_key.objectid);
Chris Mason6702ed42007-08-07 16:15:09 -0400186
Chris Mason5f39d392007-10-15 16:14:19 -0400187 WARN_ON(btrfs_header_generation(buf) > trans->transid);
188 if (btrfs_header_generation(buf) != trans->transid) {
Chris Mason6702ed42007-08-07 16:15:09 -0400189 different_trans = 1;
190 ret = btrfs_inc_ref(trans, root, buf);
191 if (ret)
192 return ret;
193 } else {
Chris Mason6702ed42007-08-07 16:15:09 -0400194 clean_tree_block(trans, root, buf);
195 }
196
197 if (buf == root->node) {
Chris Mason7bb86312007-12-11 09:25:06 -0500198 root_gen = btrfs_header_generation(buf);
Chris Mason6702ed42007-08-07 16:15:09 -0400199 root->node = cow;
Chris Mason5f39d392007-10-15 16:14:19 -0400200 extent_buffer_get(cow);
Chris Mason6702ed42007-08-07 16:15:09 -0400201 if (buf != root->commit_root) {
Chris Masondb945352007-10-15 16:15:53 -0400202 btrfs_free_extent(trans, root, buf->start,
Chris Mason7bb86312007-12-11 09:25:06 -0500203 buf->len, root->root_key.objectid,
204 root_gen, 0, 0, 1);
Chris Mason6702ed42007-08-07 16:15:09 -0400205 }
Chris Mason5f39d392007-10-15 16:14:19 -0400206 free_extent_buffer(buf);
Chris Mason0b86a832008-03-24 15:01:56 -0400207 add_root_to_dirty_list(root);
Chris Mason6702ed42007-08-07 16:15:09 -0400208 } else {
Chris Mason7bb86312007-12-11 09:25:06 -0500209 root_gen = btrfs_header_generation(parent);
Chris Mason5f39d392007-10-15 16:14:19 -0400210 btrfs_set_node_blockptr(parent, parent_slot,
Chris Masondb945352007-10-15 16:15:53 -0400211 cow->start);
Chris Mason74493f72007-12-11 09:25:06 -0500212 WARN_ON(trans->transid == 0);
213 btrfs_set_node_ptr_generation(parent, parent_slot,
214 trans->transid);
Chris Mason6702ed42007-08-07 16:15:09 -0400215 btrfs_mark_buffer_dirty(parent);
Chris Mason5f39d392007-10-15 16:14:19 -0400216 WARN_ON(btrfs_header_generation(parent) != trans->transid);
Chris Mason7bb86312007-12-11 09:25:06 -0500217 btrfs_free_extent(trans, root, buf->start, buf->len,
218 btrfs_header_owner(parent), root_gen,
219 0, 0, 1);
Chris Mason6702ed42007-08-07 16:15:09 -0400220 }
Chris Mason5f39d392007-10-15 16:14:19 -0400221 free_extent_buffer(buf);
Chris Mason6702ed42007-08-07 16:15:09 -0400222 btrfs_mark_buffer_dirty(cow);
223 *cow_ret = cow;
224 return 0;
225}
226
Chris Mason5f39d392007-10-15 16:14:19 -0400227int btrfs_cow_block(struct btrfs_trans_handle *trans,
228 struct btrfs_root *root, struct extent_buffer *buf,
229 struct extent_buffer *parent, int parent_slot,
230 struct extent_buffer **cow_ret)
Chris Mason02217ed2007-03-02 16:08:05 -0500231{
Chris Mason6702ed42007-08-07 16:15:09 -0400232 u64 search_start;
Chris Masondc17ff82008-01-08 15:46:30 -0500233 u64 header_trans;
Chris Masonf510cfe2007-10-15 16:14:48 -0400234 int ret;
Chris Masondc17ff82008-01-08 15:46:30 -0500235
Chris Masonccd467d2007-06-28 15:57:36 -0400236 if (trans->transaction != root->fs_info->running_transaction) {
237 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
238 root->fs_info->running_transaction->transid);
239 WARN_ON(1);
240 }
241 if (trans->transid != root->fs_info->generation) {
242 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
243 root->fs_info->generation);
244 WARN_ON(1);
245 }
Chris Masondc17ff82008-01-08 15:46:30 -0500246
247 header_trans = btrfs_header_generation(buf);
248 if (header_trans == trans->transid) {
Chris Mason02217ed2007-03-02 16:08:05 -0500249 *cow_ret = buf;
250 return 0;
251 }
Chris Mason6702ed42007-08-07 16:15:09 -0400252
Chris Mason0b86a832008-03-24 15:01:56 -0400253 search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
Chris Masonf510cfe2007-10-15 16:14:48 -0400254 ret = __btrfs_cow_block(trans, root, buf, parent,
Chris Mason6702ed42007-08-07 16:15:09 -0400255 parent_slot, cow_ret, search_start, 0);
Chris Masonf510cfe2007-10-15 16:14:48 -0400256 return ret;
Chris Mason6702ed42007-08-07 16:15:09 -0400257}
258
Chris Mason6b800532007-10-15 16:17:34 -0400259static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
Chris Mason6702ed42007-08-07 16:15:09 -0400260{
Chris Mason6b800532007-10-15 16:17:34 -0400261 if (blocknr < other && other - (blocknr + blocksize) < 32768)
Chris Mason6702ed42007-08-07 16:15:09 -0400262 return 1;
Chris Mason6b800532007-10-15 16:17:34 -0400263 if (blocknr > other && blocknr - (other + blocksize) < 32768)
Chris Mason6702ed42007-08-07 16:15:09 -0400264 return 1;
Chris Mason02217ed2007-03-02 16:08:05 -0500265 return 0;
266}
267
Chris Mason081e9572007-11-06 10:26:24 -0500268/*
269 * compare two keys in a memcmp fashion
270 */
271static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
272{
273 struct btrfs_key k1;
274
275 btrfs_disk_key_to_cpu(&k1, disk);
276
277 if (k1.objectid > k2->objectid)
278 return 1;
279 if (k1.objectid < k2->objectid)
280 return -1;
281 if (k1.type > k2->type)
282 return 1;
283 if (k1.type < k2->type)
284 return -1;
285 if (k1.offset > k2->offset)
286 return 1;
287 if (k1.offset < k2->offset)
288 return -1;
289 return 0;
290}
291
292
Chris Mason6702ed42007-08-07 16:15:09 -0400293int btrfs_realloc_node(struct btrfs_trans_handle *trans,
Chris Mason5f39d392007-10-15 16:14:19 -0400294 struct btrfs_root *root, struct extent_buffer *parent,
Chris Masona6b6e752007-10-15 16:22:39 -0400295 int start_slot, int cache_only, u64 *last_ret,
296 struct btrfs_key *progress)
Chris Mason6702ed42007-08-07 16:15:09 -0400297{
Chris Mason6b800532007-10-15 16:17:34 -0400298 struct extent_buffer *cur;
299 struct extent_buffer *tmp;
Chris Mason6702ed42007-08-07 16:15:09 -0400300 u64 blocknr;
Chris Masone9d0b132007-08-10 14:06:19 -0400301 u64 search_start = *last_ret;
302 u64 last_block = 0;
Chris Mason6702ed42007-08-07 16:15:09 -0400303 u64 other;
304 u32 parent_nritems;
Chris Mason6702ed42007-08-07 16:15:09 -0400305 int end_slot;
306 int i;
307 int err = 0;
Chris Masonf2183bd2007-08-10 14:42:37 -0400308 int parent_level;
Chris Mason6b800532007-10-15 16:17:34 -0400309 int uptodate;
310 u32 blocksize;
Chris Mason081e9572007-11-06 10:26:24 -0500311 int progress_passed = 0;
312 struct btrfs_disk_key disk_key;
Chris Mason6702ed42007-08-07 16:15:09 -0400313
Chris Mason5708b952007-10-25 15:43:18 -0400314 parent_level = btrfs_header_level(parent);
315 if (cache_only && parent_level != 1)
316 return 0;
317
Chris Mason6702ed42007-08-07 16:15:09 -0400318 if (trans->transaction != root->fs_info->running_transaction) {
319 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
320 root->fs_info->running_transaction->transid);
321 WARN_ON(1);
322 }
323 if (trans->transid != root->fs_info->generation) {
324 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
325 root->fs_info->generation);
326 WARN_ON(1);
327 }
Chris Mason86479a02007-09-10 19:58:16 -0400328
Chris Mason6b800532007-10-15 16:17:34 -0400329 parent_nritems = btrfs_header_nritems(parent);
Chris Mason6b800532007-10-15 16:17:34 -0400330 blocksize = btrfs_level_size(root, parent_level - 1);
Chris Mason6702ed42007-08-07 16:15:09 -0400331 end_slot = parent_nritems;
332
333 if (parent_nritems == 1)
334 return 0;
335
336 for (i = start_slot; i < end_slot; i++) {
337 int close = 1;
Chris Masona6b6e752007-10-15 16:22:39 -0400338
Chris Mason5708b952007-10-25 15:43:18 -0400339 if (!parent->map_token) {
340 map_extent_buffer(parent,
341 btrfs_node_key_ptr_offset(i),
342 sizeof(struct btrfs_key_ptr),
343 &parent->map_token, &parent->kaddr,
344 &parent->map_start, &parent->map_len,
345 KM_USER1);
346 }
Chris Mason081e9572007-11-06 10:26:24 -0500347 btrfs_node_key(parent, &disk_key, i);
348 if (!progress_passed && comp_keys(&disk_key, progress) < 0)
349 continue;
350
351 progress_passed = 1;
Chris Mason6b800532007-10-15 16:17:34 -0400352 blocknr = btrfs_node_blockptr(parent, i);
Chris Masone9d0b132007-08-10 14:06:19 -0400353 if (last_block == 0)
354 last_block = blocknr;
Chris Mason5708b952007-10-25 15:43:18 -0400355
Chris Mason6702ed42007-08-07 16:15:09 -0400356 if (i > 0) {
Chris Mason6b800532007-10-15 16:17:34 -0400357 other = btrfs_node_blockptr(parent, i - 1);
358 close = close_blocks(blocknr, other, blocksize);
Chris Mason6702ed42007-08-07 16:15:09 -0400359 }
Chris Mason5708b952007-10-25 15:43:18 -0400360 if (close && i < end_slot - 2) {
Chris Mason6b800532007-10-15 16:17:34 -0400361 other = btrfs_node_blockptr(parent, i + 1);
362 close = close_blocks(blocknr, other, blocksize);
Chris Mason6702ed42007-08-07 16:15:09 -0400363 }
Chris Masone9d0b132007-08-10 14:06:19 -0400364 if (close) {
365 last_block = blocknr;
Chris Mason6702ed42007-08-07 16:15:09 -0400366 continue;
Chris Masone9d0b132007-08-10 14:06:19 -0400367 }
Chris Mason5708b952007-10-25 15:43:18 -0400368 if (parent->map_token) {
369 unmap_extent_buffer(parent, parent->map_token,
370 KM_USER1);
371 parent->map_token = NULL;
372 }
Chris Mason6702ed42007-08-07 16:15:09 -0400373
Chris Mason6b800532007-10-15 16:17:34 -0400374 cur = btrfs_find_tree_block(root, blocknr, blocksize);
375 if (cur)
376 uptodate = btrfs_buffer_uptodate(cur);
377 else
378 uptodate = 0;
Chris Mason5708b952007-10-25 15:43:18 -0400379 if (!cur || !uptodate) {
Chris Mason6702ed42007-08-07 16:15:09 -0400380 if (cache_only) {
Chris Mason6b800532007-10-15 16:17:34 -0400381 free_extent_buffer(cur);
Chris Mason6702ed42007-08-07 16:15:09 -0400382 continue;
383 }
Chris Mason6b800532007-10-15 16:17:34 -0400384 if (!cur) {
385 cur = read_tree_block(root, blocknr,
386 blocksize);
387 } else if (!uptodate) {
388 btrfs_read_buffer(cur);
Chris Masonf2183bd2007-08-10 14:42:37 -0400389 }
Chris Mason6702ed42007-08-07 16:15:09 -0400390 }
Chris Masone9d0b132007-08-10 14:06:19 -0400391 if (search_start == 0)
Chris Mason6b800532007-10-15 16:17:34 -0400392 search_start = last_block;
Chris Masone9d0b132007-08-10 14:06:19 -0400393
Chris Mason6b800532007-10-15 16:17:34 -0400394 err = __btrfs_cow_block(trans, root, cur, parent, i,
395 &tmp, search_start,
396 min(16 * blocksize,
397 (end_slot - i) * blocksize));
Yan252c38f2007-08-29 09:11:44 -0400398 if (err) {
Chris Mason6b800532007-10-15 16:17:34 -0400399 free_extent_buffer(cur);
Chris Mason6702ed42007-08-07 16:15:09 -0400400 break;
Yan252c38f2007-08-29 09:11:44 -0400401 }
Chris Mason6b800532007-10-15 16:17:34 -0400402 search_start = tmp->start;
Chris Mason5708b952007-10-25 15:43:18 -0400403 last_block = tmp->start;
Chris Masonf2183bd2007-08-10 14:42:37 -0400404 *last_ret = search_start;
405 if (parent_level == 1)
Chris Mason6b800532007-10-15 16:17:34 -0400406 btrfs_clear_buffer_defrag(tmp);
Chris Mason6b800532007-10-15 16:17:34 -0400407 free_extent_buffer(tmp);
Chris Mason6702ed42007-08-07 16:15:09 -0400408 }
Chris Mason5708b952007-10-25 15:43:18 -0400409 if (parent->map_token) {
410 unmap_extent_buffer(parent, parent->map_token,
411 KM_USER1);
412 parent->map_token = NULL;
413 }
Chris Mason6702ed42007-08-07 16:15:09 -0400414 return err;
415}
416
Chris Mason74123bd2007-02-02 11:05:29 -0500417/*
418 * The leaf data grows from end-to-front in the node.
419 * this returns the address of the start of the last item,
420 * which is the stop of the leaf data stack
421 */
Chris Mason123abc82007-03-14 14:14:43 -0400422static inline unsigned int leaf_data_end(struct btrfs_root *root,
Chris Mason5f39d392007-10-15 16:14:19 -0400423 struct extent_buffer *leaf)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500424{
Chris Mason5f39d392007-10-15 16:14:19 -0400425 u32 nr = btrfs_header_nritems(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500426 if (nr == 0)
Chris Mason123abc82007-03-14 14:14:43 -0400427 return BTRFS_LEAF_DATA_SIZE(root);
Chris Mason5f39d392007-10-15 16:14:19 -0400428 return btrfs_item_offset_nr(leaf, nr - 1);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500429}
430
Chris Mason123abc82007-03-14 14:14:43 -0400431static int check_node(struct btrfs_root *root, struct btrfs_path *path,
432 int level)
Chris Masonaa5d6be2007-02-28 16:35:06 -0500433{
Chris Mason5f39d392007-10-15 16:14:19 -0400434 struct extent_buffer *parent = NULL;
435 struct extent_buffer *node = path->nodes[level];
436 struct btrfs_disk_key parent_key;
437 struct btrfs_disk_key node_key;
Chris Masonaa5d6be2007-02-28 16:35:06 -0500438 int parent_slot;
Chris Mason8d7be552007-05-10 11:24:42 -0400439 int slot;
440 struct btrfs_key cpukey;
Chris Mason5f39d392007-10-15 16:14:19 -0400441 u32 nritems = btrfs_header_nritems(node);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500442
443 if (path->nodes[level + 1])
Chris Mason5f39d392007-10-15 16:14:19 -0400444 parent = path->nodes[level + 1];
Aneesha1f39632007-07-11 10:03:27 -0400445
Chris Mason8d7be552007-05-10 11:24:42 -0400446 slot = path->slots[level];
Chris Mason7518a232007-03-12 12:01:18 -0400447 BUG_ON(nritems == 0);
448 if (parent) {
Aneesha1f39632007-07-11 10:03:27 -0400449 parent_slot = path->slots[level + 1];
Chris Mason5f39d392007-10-15 16:14:19 -0400450 btrfs_node_key(parent, &parent_key, parent_slot);
451 btrfs_node_key(node, &node_key, 0);
452 BUG_ON(memcmp(&parent_key, &node_key,
Chris Masone2fa7222007-03-12 16:22:34 -0400453 sizeof(struct btrfs_disk_key)));
Chris Mason1d4f8a02007-03-13 09:28:32 -0400454 BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
Chris Masondb945352007-10-15 16:15:53 -0400455 btrfs_header_bytenr(node));
Chris Masonaa5d6be2007-02-28 16:35:06 -0500456 }
Chris Mason123abc82007-03-14 14:14:43 -0400457 BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
Chris Mason8d7be552007-05-10 11:24:42 -0400458 if (slot != 0) {
Chris Mason5f39d392007-10-15 16:14:19 -0400459 btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
460 btrfs_node_key(node, &node_key, slot);
461 BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
Chris Mason8d7be552007-05-10 11:24:42 -0400462 }
463 if (slot < nritems - 1) {
Chris Mason5f39d392007-10-15 16:14:19 -0400464 btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
465 btrfs_node_key(node, &node_key, slot);
466 BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500467 }
468 return 0;
469}
470
Chris Mason123abc82007-03-14 14:14:43 -0400471static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
472 int level)
Chris Masonaa5d6be2007-02-28 16:35:06 -0500473{
Chris Mason5f39d392007-10-15 16:14:19 -0400474 struct extent_buffer *leaf = path->nodes[level];
475 struct extent_buffer *parent = NULL;
Chris Masonaa5d6be2007-02-28 16:35:06 -0500476 int parent_slot;
Chris Mason8d7be552007-05-10 11:24:42 -0400477 struct btrfs_key cpukey;
Chris Mason5f39d392007-10-15 16:14:19 -0400478 struct btrfs_disk_key parent_key;
479 struct btrfs_disk_key leaf_key;
480 int slot = path->slots[0];
Chris Mason8d7be552007-05-10 11:24:42 -0400481
Chris Mason5f39d392007-10-15 16:14:19 -0400482 u32 nritems = btrfs_header_nritems(leaf);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500483
484 if (path->nodes[level + 1])
Chris Mason5f39d392007-10-15 16:14:19 -0400485 parent = path->nodes[level + 1];
Chris Mason7518a232007-03-12 12:01:18 -0400486
487 if (nritems == 0)
488 return 0;
489
490 if (parent) {
Aneesha1f39632007-07-11 10:03:27 -0400491 parent_slot = path->slots[level + 1];
Chris Mason5f39d392007-10-15 16:14:19 -0400492 btrfs_node_key(parent, &parent_key, parent_slot);
493 btrfs_item_key(leaf, &leaf_key, 0);
Chris Mason6702ed42007-08-07 16:15:09 -0400494
Chris Mason5f39d392007-10-15 16:14:19 -0400495 BUG_ON(memcmp(&parent_key, &leaf_key,
Chris Masone2fa7222007-03-12 16:22:34 -0400496 sizeof(struct btrfs_disk_key)));
Chris Mason1d4f8a02007-03-13 09:28:32 -0400497 BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
Chris Masondb945352007-10-15 16:15:53 -0400498 btrfs_header_bytenr(leaf));
Chris Masonaa5d6be2007-02-28 16:35:06 -0500499 }
Chris Mason5f39d392007-10-15 16:14:19 -0400500#if 0
501 for (i = 0; nritems > 1 && i < nritems - 2; i++) {
502 btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
503 btrfs_item_key(leaf, &leaf_key, i);
504 if (comp_keys(&leaf_key, &cpukey) >= 0) {
505 btrfs_print_leaf(root, leaf);
506 printk("slot %d offset bad key\n", i);
507 BUG_ON(1);
508 }
509 if (btrfs_item_offset_nr(leaf, i) !=
510 btrfs_item_end_nr(leaf, i + 1)) {
511 btrfs_print_leaf(root, leaf);
512 printk("slot %d offset bad\n", i);
513 BUG_ON(1);
514 }
515 if (i == 0) {
516 if (btrfs_item_offset_nr(leaf, i) +
517 btrfs_item_size_nr(leaf, i) !=
518 BTRFS_LEAF_DATA_SIZE(root)) {
519 btrfs_print_leaf(root, leaf);
520 printk("slot %d first offset bad\n", i);
521 BUG_ON(1);
522 }
523 }
524 }
525 if (nritems > 0) {
526 if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
527 btrfs_print_leaf(root, leaf);
528 printk("slot %d bad size \n", nritems - 1);
529 BUG_ON(1);
530 }
531 }
532#endif
533 if (slot != 0 && slot < nritems - 1) {
534 btrfs_item_key(leaf, &leaf_key, slot);
535 btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
536 if (comp_keys(&leaf_key, &cpukey) <= 0) {
537 btrfs_print_leaf(root, leaf);
538 printk("slot %d offset bad key\n", slot);
539 BUG_ON(1);
540 }
541 if (btrfs_item_offset_nr(leaf, slot - 1) !=
542 btrfs_item_end_nr(leaf, slot)) {
543 btrfs_print_leaf(root, leaf);
544 printk("slot %d offset bad\n", slot);
545 BUG_ON(1);
546 }
Chris Masonaa5d6be2007-02-28 16:35:06 -0500547 }
Chris Mason8d7be552007-05-10 11:24:42 -0400548 if (slot < nritems - 1) {
Chris Mason5f39d392007-10-15 16:14:19 -0400549 btrfs_item_key(leaf, &leaf_key, slot);
550 btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
551 BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
552 if (btrfs_item_offset_nr(leaf, slot) !=
553 btrfs_item_end_nr(leaf, slot + 1)) {
554 btrfs_print_leaf(root, leaf);
555 printk("slot %d offset bad\n", slot);
556 BUG_ON(1);
557 }
Chris Mason8d7be552007-05-10 11:24:42 -0400558 }
Chris Mason5f39d392007-10-15 16:14:19 -0400559 BUG_ON(btrfs_item_offset_nr(leaf, 0) +
560 btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
Chris Masonaa5d6be2007-02-28 16:35:06 -0500561 return 0;
562}
563
Chris Mason98ed5172008-01-03 10:01:48 -0500564static int noinline check_block(struct btrfs_root *root,
565 struct btrfs_path *path, int level)
Chris Masonaa5d6be2007-02-28 16:35:06 -0500566{
Chris Mason810191f2007-10-15 16:18:55 -0400567 return 0;
Chris Masondb945352007-10-15 16:15:53 -0400568#if 0
Chris Mason5f39d392007-10-15 16:14:19 -0400569 struct extent_buffer *buf = path->nodes[level];
Chris Mason5f39d392007-10-15 16:14:19 -0400570
Chris Mason479965d2007-10-15 16:14:27 -0400571 if (memcmp_extent_buffer(buf, root->fs_info->fsid,
572 (unsigned long)btrfs_header_fsid(buf),
573 BTRFS_FSID_SIZE)) {
Chris Mason5f39d392007-10-15 16:14:19 -0400574 printk("warning bad block %Lu\n", buf->start);
Chris Masondb945352007-10-15 16:15:53 -0400575 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -0400576 }
Chris Masondb945352007-10-15 16:15:53 -0400577#endif
Chris Masonaa5d6be2007-02-28 16:35:06 -0500578 if (level == 0)
Chris Mason123abc82007-03-14 14:14:43 -0400579 return check_leaf(root, path, level);
580 return check_node(root, path, level);
Chris Masonaa5d6be2007-02-28 16:35:06 -0500581}
582
Chris Mason74123bd2007-02-02 11:05:29 -0500583/*
Chris Mason5f39d392007-10-15 16:14:19 -0400584 * search for key in the extent_buffer. The items start at offset p,
585 * and they are item_size apart. There are 'max' items in p.
586 *
Chris Mason74123bd2007-02-02 11:05:29 -0500587 * the slot in the array is returned via slot, and it points to
588 * the place where you would insert key if it is not found in
589 * the array.
590 *
591 * slot may point to max if the key is bigger than all of the keys
592 */
Chris Mason5f39d392007-10-15 16:14:19 -0400593static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
594 int item_size, struct btrfs_key *key,
595 int max, int *slot)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500596{
597 int low = 0;
598 int high = max;
599 int mid;
600 int ret;
Chris Mason479965d2007-10-15 16:14:27 -0400601 struct btrfs_disk_key *tmp = NULL;
Chris Mason5f39d392007-10-15 16:14:19 -0400602 struct btrfs_disk_key unaligned;
603 unsigned long offset;
604 char *map_token = NULL;
605 char *kaddr = NULL;
606 unsigned long map_start = 0;
607 unsigned long map_len = 0;
Chris Mason479965d2007-10-15 16:14:27 -0400608 int err;
Chris Masonbe0e5c02007-01-26 15:51:26 -0500609
610 while(low < high) {
611 mid = (low + high) / 2;
Chris Mason5f39d392007-10-15 16:14:19 -0400612 offset = p + mid * item_size;
613
614 if (!map_token || offset < map_start ||
615 (offset + sizeof(struct btrfs_disk_key)) >
616 map_start + map_len) {
Chris Mason479965d2007-10-15 16:14:27 -0400617 if (map_token) {
Chris Mason5f39d392007-10-15 16:14:19 -0400618 unmap_extent_buffer(eb, map_token, KM_USER0);
Chris Mason479965d2007-10-15 16:14:27 -0400619 map_token = NULL;
620 }
621 err = map_extent_buffer(eb, offset,
622 sizeof(struct btrfs_disk_key),
623 &map_token, &kaddr,
624 &map_start, &map_len, KM_USER0);
Chris Mason5f39d392007-10-15 16:14:19 -0400625
Chris Mason479965d2007-10-15 16:14:27 -0400626 if (!err) {
627 tmp = (struct btrfs_disk_key *)(kaddr + offset -
628 map_start);
629 } else {
630 read_extent_buffer(eb, &unaligned,
631 offset, sizeof(unaligned));
632 tmp = &unaligned;
633 }
634
Chris Mason5f39d392007-10-15 16:14:19 -0400635 } else {
636 tmp = (struct btrfs_disk_key *)(kaddr + offset -
637 map_start);
638 }
Chris Masonbe0e5c02007-01-26 15:51:26 -0500639 ret = comp_keys(tmp, key);
640
641 if (ret < 0)
642 low = mid + 1;
643 else if (ret > 0)
644 high = mid;
645 else {
646 *slot = mid;
Chris Mason479965d2007-10-15 16:14:27 -0400647 if (map_token)
648 unmap_extent_buffer(eb, map_token, KM_USER0);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500649 return 0;
650 }
651 }
652 *slot = low;
Chris Mason5f39d392007-10-15 16:14:19 -0400653 if (map_token)
654 unmap_extent_buffer(eb, map_token, KM_USER0);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500655 return 1;
656}
657
Chris Mason97571fd2007-02-24 13:39:08 -0500658/*
659 * simple bin_search frontend that does the right thing for
660 * leaves vs nodes
661 */
Chris Mason5f39d392007-10-15 16:14:19 -0400662static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
663 int level, int *slot)
Chris Masonbe0e5c02007-01-26 15:51:26 -0500664{
Chris Mason5f39d392007-10-15 16:14:19 -0400665 if (level == 0) {
666 return generic_bin_search(eb,
667 offsetof(struct btrfs_leaf, items),
Chris Mason0783fcf2007-03-12 20:12:07 -0400668 sizeof(struct btrfs_item),
Chris Mason5f39d392007-10-15 16:14:19 -0400669 key, btrfs_header_nritems(eb),
Chris Mason7518a232007-03-12 12:01:18 -0400670 slot);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500671 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400672 return generic_bin_search(eb,
673 offsetof(struct btrfs_node, ptrs),
Chris Mason123abc82007-03-14 14:14:43 -0400674 sizeof(struct btrfs_key_ptr),
Chris Mason5f39d392007-10-15 16:14:19 -0400675 key, btrfs_header_nritems(eb),
Chris Mason7518a232007-03-12 12:01:18 -0400676 slot);
Chris Masonbe0e5c02007-01-26 15:51:26 -0500677 }
678 return -1;
679}
680
Chris Mason5f39d392007-10-15 16:14:19 -0400681static struct extent_buffer *read_node_slot(struct btrfs_root *root,
682 struct extent_buffer *parent, int slot)
Chris Masonbb803952007-03-01 12:04:21 -0500683{
Chris Masonbb803952007-03-01 12:04:21 -0500684 if (slot < 0)
685 return NULL;
Chris Mason5f39d392007-10-15 16:14:19 -0400686 if (slot >= btrfs_header_nritems(parent))
Chris Masonbb803952007-03-01 12:04:21 -0500687 return NULL;
Chris Masondb945352007-10-15 16:15:53 -0400688 return read_tree_block(root, btrfs_node_blockptr(parent, slot),
689 btrfs_level_size(root, btrfs_header_level(parent) - 1));
Chris Masonbb803952007-03-01 12:04:21 -0500690}
691
Chris Mason98ed5172008-01-03 10:01:48 -0500692static int balance_level(struct btrfs_trans_handle *trans,
693 struct btrfs_root *root,
694 struct btrfs_path *path, int level)
Chris Masonbb803952007-03-01 12:04:21 -0500695{
Chris Mason5f39d392007-10-15 16:14:19 -0400696 struct extent_buffer *right = NULL;
697 struct extent_buffer *mid;
698 struct extent_buffer *left = NULL;
699 struct extent_buffer *parent = NULL;
Chris Masonbb803952007-03-01 12:04:21 -0500700 int ret = 0;
701 int wret;
702 int pslot;
Chris Masonbb803952007-03-01 12:04:21 -0500703 int orig_slot = path->slots[level];
Chris Mason54aa1f42007-06-22 14:16:25 -0400704 int err_on_enospc = 0;
Chris Mason79f95c82007-03-01 15:16:26 -0500705 u64 orig_ptr;
Chris Masonbb803952007-03-01 12:04:21 -0500706
707 if (level == 0)
708 return 0;
709
Chris Mason5f39d392007-10-15 16:14:19 -0400710 mid = path->nodes[level];
Chris Mason7bb86312007-12-11 09:25:06 -0500711 WARN_ON(btrfs_header_generation(mid) != trans->transid);
712
Chris Mason1d4f8a02007-03-13 09:28:32 -0400713 orig_ptr = btrfs_node_blockptr(mid, orig_slot);
Chris Mason79f95c82007-03-01 15:16:26 -0500714
Chris Mason234b63a2007-03-13 10:46:10 -0400715 if (level < BTRFS_MAX_LEVEL - 1)
Chris Mason5f39d392007-10-15 16:14:19 -0400716 parent = path->nodes[level + 1];
Chris Masonbb803952007-03-01 12:04:21 -0500717 pslot = path->slots[level + 1];
718
Chris Mason40689472007-03-17 14:29:23 -0400719 /*
720 * deal with the case where there is only one pointer in the root
721 * by promoting the node below to a root
722 */
Chris Mason5f39d392007-10-15 16:14:19 -0400723 if (!parent) {
724 struct extent_buffer *child;
Chris Masonbb803952007-03-01 12:04:21 -0500725
Chris Mason5f39d392007-10-15 16:14:19 -0400726 if (btrfs_header_nritems(mid) != 1)
Chris Masonbb803952007-03-01 12:04:21 -0500727 return 0;
728
729 /* promote the child to a root */
Chris Mason5f39d392007-10-15 16:14:19 -0400730 child = read_node_slot(root, mid, 0);
Chris Masonbb803952007-03-01 12:04:21 -0500731 BUG_ON(!child);
Yan2f375ab2008-02-01 14:58:07 -0500732 ret = btrfs_cow_block(trans, root, child, mid, 0, &child);
733 BUG_ON(ret);
734
Chris Masonbb803952007-03-01 12:04:21 -0500735 root->node = child;
Chris Mason0b86a832008-03-24 15:01:56 -0400736 add_root_to_dirty_list(root);
Chris Masonbb803952007-03-01 12:04:21 -0500737 path->nodes[level] = NULL;
Chris Mason5f39d392007-10-15 16:14:19 -0400738 clean_tree_block(trans, root, mid);
739 wait_on_tree_block_writeback(root, mid);
Chris Masonbb803952007-03-01 12:04:21 -0500740 /* once for the path */
Chris Mason5f39d392007-10-15 16:14:19 -0400741 free_extent_buffer(mid);
Chris Mason7bb86312007-12-11 09:25:06 -0500742 ret = btrfs_free_extent(trans, root, mid->start, mid->len,
743 root->root_key.objectid,
744 btrfs_header_generation(mid), 0, 0, 1);
Chris Masonbb803952007-03-01 12:04:21 -0500745 /* once for the root ptr */
Chris Mason5f39d392007-10-15 16:14:19 -0400746 free_extent_buffer(mid);
Chris Masondb945352007-10-15 16:15:53 -0400747 return ret;
Chris Masonbb803952007-03-01 12:04:21 -0500748 }
Chris Mason5f39d392007-10-15 16:14:19 -0400749 if (btrfs_header_nritems(mid) >
Chris Mason123abc82007-03-14 14:14:43 -0400750 BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
Chris Masonbb803952007-03-01 12:04:21 -0500751 return 0;
752
Chris Mason5f39d392007-10-15 16:14:19 -0400753 if (btrfs_header_nritems(mid) < 2)
Chris Mason54aa1f42007-06-22 14:16:25 -0400754 err_on_enospc = 1;
755
Chris Mason5f39d392007-10-15 16:14:19 -0400756 left = read_node_slot(root, parent, pslot - 1);
757 if (left) {
758 wret = btrfs_cow_block(trans, root, left,
759 parent, pslot - 1, &left);
Chris Mason54aa1f42007-06-22 14:16:25 -0400760 if (wret) {
761 ret = wret;
762 goto enospc;
763 }
Chris Mason2cc58cf2007-08-27 16:49:44 -0400764 }
Chris Mason5f39d392007-10-15 16:14:19 -0400765 right = read_node_slot(root, parent, pslot + 1);
766 if (right) {
767 wret = btrfs_cow_block(trans, root, right,
768 parent, pslot + 1, &right);
Chris Mason2cc58cf2007-08-27 16:49:44 -0400769 if (wret) {
770 ret = wret;
771 goto enospc;
772 }
773 }
774
775 /* first, try to make some room in the middle buffer */
Chris Mason5f39d392007-10-15 16:14:19 -0400776 if (left) {
777 orig_slot += btrfs_header_nritems(left);
778 wret = push_node_left(trans, root, left, mid);
Chris Mason79f95c82007-03-01 15:16:26 -0500779 if (wret < 0)
780 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -0400781 if (btrfs_header_nritems(mid) < 2)
Chris Mason54aa1f42007-06-22 14:16:25 -0400782 err_on_enospc = 1;
Chris Masonbb803952007-03-01 12:04:21 -0500783 }
Chris Mason79f95c82007-03-01 15:16:26 -0500784
785 /*
786 * then try to empty the right most buffer into the middle
787 */
Chris Mason5f39d392007-10-15 16:14:19 -0400788 if (right) {
789 wret = push_node_left(trans, root, mid, right);
Chris Mason54aa1f42007-06-22 14:16:25 -0400790 if (wret < 0 && wret != -ENOSPC)
Chris Mason79f95c82007-03-01 15:16:26 -0500791 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -0400792 if (btrfs_header_nritems(right) == 0) {
Chris Masondb945352007-10-15 16:15:53 -0400793 u64 bytenr = right->start;
Chris Mason7bb86312007-12-11 09:25:06 -0500794 u64 generation = btrfs_header_generation(parent);
Chris Masondb945352007-10-15 16:15:53 -0400795 u32 blocksize = right->len;
796
Chris Mason5f39d392007-10-15 16:14:19 -0400797 clean_tree_block(trans, root, right);
798 wait_on_tree_block_writeback(root, right);
799 free_extent_buffer(right);
Chris Masonbb803952007-03-01 12:04:21 -0500800 right = NULL;
Chris Masone089f052007-03-16 16:20:31 -0400801 wret = del_ptr(trans, root, path, level + 1, pslot +
802 1);
Chris Masonbb803952007-03-01 12:04:21 -0500803 if (wret)
804 ret = wret;
Chris Masondb945352007-10-15 16:15:53 -0400805 wret = btrfs_free_extent(trans, root, bytenr,
Chris Mason7bb86312007-12-11 09:25:06 -0500806 blocksize,
807 btrfs_header_owner(parent),
808 generation, 0, 0, 1);
Chris Masonbb803952007-03-01 12:04:21 -0500809 if (wret)
810 ret = wret;
811 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400812 struct btrfs_disk_key right_key;
813 btrfs_node_key(right, &right_key, 0);
814 btrfs_set_node_key(parent, &right_key, pslot + 1);
815 btrfs_mark_buffer_dirty(parent);
Chris Masonbb803952007-03-01 12:04:21 -0500816 }
817 }
Chris Mason5f39d392007-10-15 16:14:19 -0400818 if (btrfs_header_nritems(mid) == 1) {
Chris Mason79f95c82007-03-01 15:16:26 -0500819 /*
820 * we're not allowed to leave a node with one item in the
821 * tree during a delete. A deletion from lower in the tree
822 * could try to delete the only pointer in this node.
823 * So, pull some keys from the left.
824 * There has to be a left pointer at this point because
825 * otherwise we would have pulled some pointers from the
826 * right
827 */
Chris Mason5f39d392007-10-15 16:14:19 -0400828 BUG_ON(!left);
829 wret = balance_node_right(trans, root, mid, left);
Chris Mason54aa1f42007-06-22 14:16:25 -0400830 if (wret < 0) {
Chris Mason79f95c82007-03-01 15:16:26 -0500831 ret = wret;
Chris Mason54aa1f42007-06-22 14:16:25 -0400832 goto enospc;
833 }
Chris Mason79f95c82007-03-01 15:16:26 -0500834 BUG_ON(wret == 1);
835 }
Chris Mason5f39d392007-10-15 16:14:19 -0400836 if (btrfs_header_nritems(mid) == 0) {
Chris Mason79f95c82007-03-01 15:16:26 -0500837 /* we've managed to empty the middle node, drop it */
Chris Mason7bb86312007-12-11 09:25:06 -0500838 u64 root_gen = btrfs_header_generation(parent);
Chris Masondb945352007-10-15 16:15:53 -0400839 u64 bytenr = mid->start;
840 u32 blocksize = mid->len;
Chris Mason5f39d392007-10-15 16:14:19 -0400841 clean_tree_block(trans, root, mid);
842 wait_on_tree_block_writeback(root, mid);
843 free_extent_buffer(mid);
Chris Masonbb803952007-03-01 12:04:21 -0500844 mid = NULL;
Chris Masone089f052007-03-16 16:20:31 -0400845 wret = del_ptr(trans, root, path, level + 1, pslot);
Chris Masonbb803952007-03-01 12:04:21 -0500846 if (wret)
847 ret = wret;
Chris Mason7bb86312007-12-11 09:25:06 -0500848 wret = btrfs_free_extent(trans, root, bytenr, blocksize,
849 btrfs_header_owner(parent),
850 root_gen, 0, 0, 1);
Chris Masonbb803952007-03-01 12:04:21 -0500851 if (wret)
852 ret = wret;
Chris Mason79f95c82007-03-01 15:16:26 -0500853 } else {
854 /* update the parent key to reflect our changes */
Chris Mason5f39d392007-10-15 16:14:19 -0400855 struct btrfs_disk_key mid_key;
856 btrfs_node_key(mid, &mid_key, 0);
857 btrfs_set_node_key(parent, &mid_key, pslot);
858 btrfs_mark_buffer_dirty(parent);
Chris Mason79f95c82007-03-01 15:16:26 -0500859 }
Chris Masonbb803952007-03-01 12:04:21 -0500860
Chris Mason79f95c82007-03-01 15:16:26 -0500861 /* update the path */
Chris Mason5f39d392007-10-15 16:14:19 -0400862 if (left) {
863 if (btrfs_header_nritems(left) > orig_slot) {
864 extent_buffer_get(left);
865 path->nodes[level] = left;
Chris Masonbb803952007-03-01 12:04:21 -0500866 path->slots[level + 1] -= 1;
867 path->slots[level] = orig_slot;
Chris Mason5f39d392007-10-15 16:14:19 -0400868 if (mid)
869 free_extent_buffer(mid);
Chris Masonbb803952007-03-01 12:04:21 -0500870 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400871 orig_slot -= btrfs_header_nritems(left);
Chris Masonbb803952007-03-01 12:04:21 -0500872 path->slots[level] = orig_slot;
873 }
874 }
Chris Mason79f95c82007-03-01 15:16:26 -0500875 /* double check we haven't messed things up */
Chris Mason123abc82007-03-14 14:14:43 -0400876 check_block(root, path, level);
Chris Masone20d96d2007-03-22 12:13:20 -0400877 if (orig_ptr !=
Chris Mason5f39d392007-10-15 16:14:19 -0400878 btrfs_node_blockptr(path->nodes[level], path->slots[level]))
Chris Mason79f95c82007-03-01 15:16:26 -0500879 BUG();
Chris Mason54aa1f42007-06-22 14:16:25 -0400880enospc:
Chris Mason5f39d392007-10-15 16:14:19 -0400881 if (right)
882 free_extent_buffer(right);
883 if (left)
884 free_extent_buffer(left);
Chris Masonbb803952007-03-01 12:04:21 -0500885 return ret;
886}
887
Chris Masone66f7092007-04-20 13:16:02 -0400888/* returns zero if the push worked, non-zero otherwise */
Chris Mason98ed5172008-01-03 10:01:48 -0500889static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans,
890 struct btrfs_root *root,
891 struct btrfs_path *path, int level)
Chris Masone66f7092007-04-20 13:16:02 -0400892{
Chris Mason5f39d392007-10-15 16:14:19 -0400893 struct extent_buffer *right = NULL;
894 struct extent_buffer *mid;
895 struct extent_buffer *left = NULL;
896 struct extent_buffer *parent = NULL;
Chris Masone66f7092007-04-20 13:16:02 -0400897 int ret = 0;
898 int wret;
899 int pslot;
900 int orig_slot = path->slots[level];
901 u64 orig_ptr;
902
903 if (level == 0)
904 return 1;
905
Chris Mason5f39d392007-10-15 16:14:19 -0400906 mid = path->nodes[level];
Chris Mason7bb86312007-12-11 09:25:06 -0500907 WARN_ON(btrfs_header_generation(mid) != trans->transid);
Chris Masone66f7092007-04-20 13:16:02 -0400908 orig_ptr = btrfs_node_blockptr(mid, orig_slot);
909
910 if (level < BTRFS_MAX_LEVEL - 1)
Chris Mason5f39d392007-10-15 16:14:19 -0400911 parent = path->nodes[level + 1];
Chris Masone66f7092007-04-20 13:16:02 -0400912 pslot = path->slots[level + 1];
913
Chris Mason5f39d392007-10-15 16:14:19 -0400914 if (!parent)
Chris Masone66f7092007-04-20 13:16:02 -0400915 return 1;
Chris Masone66f7092007-04-20 13:16:02 -0400916
Chris Mason5f39d392007-10-15 16:14:19 -0400917 left = read_node_slot(root, parent, pslot - 1);
Chris Masone66f7092007-04-20 13:16:02 -0400918
919 /* first, try to make some room in the middle buffer */
Chris Mason5f39d392007-10-15 16:14:19 -0400920 if (left) {
Chris Masone66f7092007-04-20 13:16:02 -0400921 u32 left_nr;
Chris Mason5f39d392007-10-15 16:14:19 -0400922 left_nr = btrfs_header_nritems(left);
Chris Mason33ade1f2007-04-20 13:48:57 -0400923 if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
924 wret = 1;
925 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400926 ret = btrfs_cow_block(trans, root, left, parent,
927 pslot - 1, &left);
Chris Mason54aa1f42007-06-22 14:16:25 -0400928 if (ret)
929 wret = 1;
930 else {
Chris Mason54aa1f42007-06-22 14:16:25 -0400931 wret = push_node_left(trans, root,
Chris Mason5f39d392007-10-15 16:14:19 -0400932 left, mid);
Chris Mason54aa1f42007-06-22 14:16:25 -0400933 }
Chris Mason33ade1f2007-04-20 13:48:57 -0400934 }
Chris Masone66f7092007-04-20 13:16:02 -0400935 if (wret < 0)
936 ret = wret;
937 if (wret == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -0400938 struct btrfs_disk_key disk_key;
Chris Masone66f7092007-04-20 13:16:02 -0400939 orig_slot += left_nr;
Chris Mason5f39d392007-10-15 16:14:19 -0400940 btrfs_node_key(mid, &disk_key, 0);
941 btrfs_set_node_key(parent, &disk_key, pslot);
942 btrfs_mark_buffer_dirty(parent);
943 if (btrfs_header_nritems(left) > orig_slot) {
944 path->nodes[level] = left;
Chris Masone66f7092007-04-20 13:16:02 -0400945 path->slots[level + 1] -= 1;
946 path->slots[level] = orig_slot;
Chris Mason5f39d392007-10-15 16:14:19 -0400947 free_extent_buffer(mid);
Chris Masone66f7092007-04-20 13:16:02 -0400948 } else {
949 orig_slot -=
Chris Mason5f39d392007-10-15 16:14:19 -0400950 btrfs_header_nritems(left);
Chris Masone66f7092007-04-20 13:16:02 -0400951 path->slots[level] = orig_slot;
Chris Mason5f39d392007-10-15 16:14:19 -0400952 free_extent_buffer(left);
Chris Masone66f7092007-04-20 13:16:02 -0400953 }
Chris Masone66f7092007-04-20 13:16:02 -0400954 return 0;
955 }
Chris Mason5f39d392007-10-15 16:14:19 -0400956 free_extent_buffer(left);
Chris Masone66f7092007-04-20 13:16:02 -0400957 }
Chris Mason5f39d392007-10-15 16:14:19 -0400958 right= read_node_slot(root, parent, pslot + 1);
Chris Masone66f7092007-04-20 13:16:02 -0400959
960 /*
961 * then try to empty the right most buffer into the middle
962 */
Chris Mason5f39d392007-10-15 16:14:19 -0400963 if (right) {
Chris Mason33ade1f2007-04-20 13:48:57 -0400964 u32 right_nr;
Chris Mason5f39d392007-10-15 16:14:19 -0400965 right_nr = btrfs_header_nritems(right);
Chris Mason33ade1f2007-04-20 13:48:57 -0400966 if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
967 wret = 1;
968 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400969 ret = btrfs_cow_block(trans, root, right,
970 parent, pslot + 1,
971 &right);
Chris Mason54aa1f42007-06-22 14:16:25 -0400972 if (ret)
973 wret = 1;
974 else {
Chris Mason54aa1f42007-06-22 14:16:25 -0400975 wret = balance_node_right(trans, root,
Chris Mason5f39d392007-10-15 16:14:19 -0400976 right, mid);
Chris Mason54aa1f42007-06-22 14:16:25 -0400977 }
Chris Mason33ade1f2007-04-20 13:48:57 -0400978 }
Chris Masone66f7092007-04-20 13:16:02 -0400979 if (wret < 0)
980 ret = wret;
981 if (wret == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -0400982 struct btrfs_disk_key disk_key;
983
984 btrfs_node_key(right, &disk_key, 0);
985 btrfs_set_node_key(parent, &disk_key, pslot + 1);
986 btrfs_mark_buffer_dirty(parent);
987
988 if (btrfs_header_nritems(mid) <= orig_slot) {
989 path->nodes[level] = right;
Chris Masone66f7092007-04-20 13:16:02 -0400990 path->slots[level + 1] += 1;
991 path->slots[level] = orig_slot -
Chris Mason5f39d392007-10-15 16:14:19 -0400992 btrfs_header_nritems(mid);
993 free_extent_buffer(mid);
Chris Masone66f7092007-04-20 13:16:02 -0400994 } else {
Chris Mason5f39d392007-10-15 16:14:19 -0400995 free_extent_buffer(right);
Chris Masone66f7092007-04-20 13:16:02 -0400996 }
Chris Masone66f7092007-04-20 13:16:02 -0400997 return 0;
998 }
Chris Mason5f39d392007-10-15 16:14:19 -0400999 free_extent_buffer(right);
Chris Masone66f7092007-04-20 13:16:02 -04001000 }
Chris Masone66f7092007-04-20 13:16:02 -04001001 return 1;
1002}
1003
Chris Mason74123bd2007-02-02 11:05:29 -05001004/*
Chris Mason3c69fae2007-08-07 15:52:22 -04001005 * readahead one full node of leaves
1006 */
1007static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
Chris Mason01f46652007-12-21 16:24:26 -05001008 int level, int slot, u64 objectid)
Chris Mason3c69fae2007-08-07 15:52:22 -04001009{
Chris Mason5f39d392007-10-15 16:14:19 -04001010 struct extent_buffer *node;
Chris Mason01f46652007-12-21 16:24:26 -05001011 struct btrfs_disk_key disk_key;
Chris Mason3c69fae2007-08-07 15:52:22 -04001012 u32 nritems;
Chris Mason3c69fae2007-08-07 15:52:22 -04001013 u64 search;
Chris Mason6b800532007-10-15 16:17:34 -04001014 u64 lowest_read;
1015 u64 highest_read;
1016 u64 nread = 0;
Chris Mason3c69fae2007-08-07 15:52:22 -04001017 int direction = path->reada;
Chris Mason5f39d392007-10-15 16:14:19 -04001018 struct extent_buffer *eb;
Chris Mason6b800532007-10-15 16:17:34 -04001019 u32 nr;
1020 u32 blocksize;
1021 u32 nscan = 0;
Chris Masondb945352007-10-15 16:15:53 -04001022
Chris Masona6b6e752007-10-15 16:22:39 -04001023 if (level != 1)
Chris Mason3c69fae2007-08-07 15:52:22 -04001024 return;
1025
Chris Mason6702ed42007-08-07 16:15:09 -04001026 if (!path->nodes[level])
1027 return;
1028
Chris Mason5f39d392007-10-15 16:14:19 -04001029 node = path->nodes[level];
Chris Mason3c69fae2007-08-07 15:52:22 -04001030 search = btrfs_node_blockptr(node, slot);
Chris Mason6b800532007-10-15 16:17:34 -04001031 blocksize = btrfs_level_size(root, level - 1);
1032 eb = btrfs_find_tree_block(root, search, blocksize);
Chris Mason5f39d392007-10-15 16:14:19 -04001033 if (eb) {
1034 free_extent_buffer(eb);
Chris Mason3c69fae2007-08-07 15:52:22 -04001035 return;
1036 }
1037
Chris Mason6b800532007-10-15 16:17:34 -04001038 highest_read = search;
1039 lowest_read = search;
1040
Chris Mason5f39d392007-10-15 16:14:19 -04001041 nritems = btrfs_header_nritems(node);
Chris Mason6b800532007-10-15 16:17:34 -04001042 nr = slot;
Chris Mason3c69fae2007-08-07 15:52:22 -04001043 while(1) {
Chris Mason6b800532007-10-15 16:17:34 -04001044 if (direction < 0) {
1045 if (nr == 0)
1046 break;
1047 nr--;
1048 } else if (direction > 0) {
1049 nr++;
1050 if (nr >= nritems)
1051 break;
Chris Mason3c69fae2007-08-07 15:52:22 -04001052 }
Chris Mason01f46652007-12-21 16:24:26 -05001053 if (path->reada < 0 && objectid) {
1054 btrfs_node_key(node, &disk_key, nr);
1055 if (btrfs_disk_key_objectid(&disk_key) != objectid)
1056 break;
1057 }
Chris Mason6b800532007-10-15 16:17:34 -04001058 search = btrfs_node_blockptr(node, nr);
1059 if ((search >= lowest_read && search <= highest_read) ||
1060 (search < lowest_read && lowest_read - search <= 32768) ||
1061 (search > highest_read && search - highest_read <= 32768)) {
1062 readahead_tree_block(root, search, blocksize);
1063 nread += blocksize;
1064 }
1065 nscan++;
1066 if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
1067 break;
1068 if(nread > (1024 * 1024) || nscan > 128)
1069 break;
1070
1071 if (search < lowest_read)
1072 lowest_read = search;
1073 if (search > highest_read)
1074 highest_read = search;
Chris Mason3c69fae2007-08-07 15:52:22 -04001075 }
1076}
1077/*
Chris Mason74123bd2007-02-02 11:05:29 -05001078 * look for key in the tree. path is filled in with nodes along the way
1079 * if key is found, we return zero and you can find the item in the leaf
1080 * level of the path (level 0)
1081 *
1082 * If the key isn't found, the path points to the slot where it should
Chris Masonaa5d6be2007-02-28 16:35:06 -05001083 * be inserted, and 1 is returned. If there are other errors during the
1084 * search a negative error number is returned.
Chris Mason97571fd2007-02-24 13:39:08 -05001085 *
1086 * if ins_len > 0, nodes and leaves will be split as we walk down the
1087 * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if
1088 * possible)
Chris Mason74123bd2007-02-02 11:05:29 -05001089 */
Chris Masone089f052007-03-16 16:20:31 -04001090int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
1091 *root, struct btrfs_key *key, struct btrfs_path *p, int
1092 ins_len, int cow)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001093{
Chris Mason5f39d392007-10-15 16:14:19 -04001094 struct extent_buffer *b;
Chris Masondb945352007-10-15 16:15:53 -04001095 u64 bytenr;
Chris Mason74493f72007-12-11 09:25:06 -05001096 u64 ptr_gen;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001097 int slot;
1098 int ret;
1099 int level;
Chris Mason3c69fae2007-08-07 15:52:22 -04001100 int should_reada = p->reada;
Chris Mason9f3a7422007-08-07 15:52:19 -04001101 u8 lowest_level = 0;
1102
Chris Mason6702ed42007-08-07 16:15:09 -04001103 lowest_level = p->lowest_level;
1104 WARN_ON(lowest_level && ins_len);
Chris Mason22b0ebd2007-03-30 08:47:31 -04001105 WARN_ON(p->nodes[0] != NULL);
1106 WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
Chris Masonbb803952007-03-01 12:04:21 -05001107again:
1108 b = root->node;
Chris Mason5f39d392007-10-15 16:14:19 -04001109 extent_buffer_get(b);
Chris Masoneb60cea2007-02-02 09:18:22 -05001110 while (b) {
Chris Mason5f39d392007-10-15 16:14:19 -04001111 level = btrfs_header_level(b);
Chris Mason02217ed2007-03-02 16:08:05 -05001112 if (cow) {
1113 int wret;
Chris Masone20d96d2007-03-22 12:13:20 -04001114 wret = btrfs_cow_block(trans, root, b,
1115 p->nodes[level + 1],
1116 p->slots[level + 1],
Yan252c38f2007-08-29 09:11:44 -04001117 &b);
Chris Mason54aa1f42007-06-22 14:16:25 -04001118 if (wret) {
Chris Mason5f39d392007-10-15 16:14:19 -04001119 free_extent_buffer(b);
Chris Mason54aa1f42007-06-22 14:16:25 -04001120 return wret;
1121 }
Chris Mason02217ed2007-03-02 16:08:05 -05001122 }
1123 BUG_ON(!cow && ins_len);
Chris Mason5f39d392007-10-15 16:14:19 -04001124 if (level != btrfs_header_level(b))
Chris Mason2c90e5d2007-04-02 10:50:19 -04001125 WARN_ON(1);
Chris Mason5f39d392007-10-15 16:14:19 -04001126 level = btrfs_header_level(b);
Chris Masoneb60cea2007-02-02 09:18:22 -05001127 p->nodes[level] = b;
Chris Mason123abc82007-03-14 14:14:43 -04001128 ret = check_block(root, p, level);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001129 if (ret)
1130 return -1;
Chris Mason5f39d392007-10-15 16:14:19 -04001131 ret = bin_search(b, key, level, &slot);
1132 if (level != 0) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05001133 if (ret && slot > 0)
1134 slot -= 1;
1135 p->slots[level] = slot;
Chris Mason5f39d392007-10-15 16:14:19 -04001136 if (ins_len > 0 && btrfs_header_nritems(b) >=
Chris Masond4dbff92007-04-04 14:08:15 -04001137 BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
Chris Masone089f052007-03-16 16:20:31 -04001138 int sret = split_node(trans, root, p, level);
Chris Mason5c680ed2007-02-22 11:39:13 -05001139 BUG_ON(sret > 0);
1140 if (sret)
1141 return sret;
1142 b = p->nodes[level];
Chris Mason5c680ed2007-02-22 11:39:13 -05001143 slot = p->slots[level];
Chris Masonbb803952007-03-01 12:04:21 -05001144 } else if (ins_len < 0) {
Chris Masone089f052007-03-16 16:20:31 -04001145 int sret = balance_level(trans, root, p,
1146 level);
Chris Masonbb803952007-03-01 12:04:21 -05001147 if (sret)
1148 return sret;
1149 b = p->nodes[level];
Chris Masonf510cfe2007-10-15 16:14:48 -04001150 if (!b) {
1151 btrfs_release_path(NULL, p);
Chris Masonbb803952007-03-01 12:04:21 -05001152 goto again;
Chris Masonf510cfe2007-10-15 16:14:48 -04001153 }
Chris Masonbb803952007-03-01 12:04:21 -05001154 slot = p->slots[level];
Chris Mason5f39d392007-10-15 16:14:19 -04001155 BUG_ON(btrfs_header_nritems(b) == 1);
Chris Mason5c680ed2007-02-22 11:39:13 -05001156 }
Chris Mason9f3a7422007-08-07 15:52:19 -04001157 /* this is only true while dropping a snapshot */
1158 if (level == lowest_level)
1159 break;
Chris Masondb945352007-10-15 16:15:53 -04001160 bytenr = btrfs_node_blockptr(b, slot);
Chris Mason74493f72007-12-11 09:25:06 -05001161 ptr_gen = btrfs_node_ptr_generation(b, slot);
Chris Mason6702ed42007-08-07 16:15:09 -04001162 if (should_reada)
Chris Mason01f46652007-12-21 16:24:26 -05001163 reada_for_search(root, p, level, slot,
1164 key->objectid);
Chris Masondb945352007-10-15 16:15:53 -04001165 b = read_tree_block(root, bytenr,
1166 btrfs_level_size(root, level - 1));
Chris Mason74493f72007-12-11 09:25:06 -05001167 if (ptr_gen != btrfs_header_generation(b)) {
1168 printk("block %llu bad gen wanted %llu "
1169 "found %llu\n",
1170 (unsigned long long)b->start,
1171 (unsigned long long)ptr_gen,
1172 (unsigned long long)btrfs_header_generation(b));
1173 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05001174 } else {
1175 p->slots[level] = slot;
Chris Mason5f39d392007-10-15 16:14:19 -04001176 if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
Chris Mason0783fcf2007-03-12 20:12:07 -04001177 sizeof(struct btrfs_item) + ins_len) {
Chris Masond4dbff92007-04-04 14:08:15 -04001178 int sret = split_leaf(trans, root, key,
Chris Masoncc0c5532007-10-25 15:42:57 -04001179 p, ins_len, ret == 0);
Chris Mason5c680ed2007-02-22 11:39:13 -05001180 BUG_ON(sret > 0);
1181 if (sret)
1182 return sret;
1183 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05001184 return ret;
1185 }
1186 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05001187 return 1;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001188}
1189
Chris Mason74123bd2007-02-02 11:05:29 -05001190/*
1191 * adjust the pointers going up the tree, starting at level
1192 * making sure the right key of each node is points to 'key'.
1193 * This is used after shifting pointers to the left, so it stops
1194 * fixing up pointers when a given leaf/node is not in slot 0 of the
1195 * higher levels
Chris Masonaa5d6be2007-02-28 16:35:06 -05001196 *
1197 * If this fails to write a tree block, it returns -1, but continues
1198 * fixing up the blocks in ram so the tree is consistent.
Chris Mason74123bd2007-02-02 11:05:29 -05001199 */
Chris Mason5f39d392007-10-15 16:14:19 -04001200static int fixup_low_keys(struct btrfs_trans_handle *trans,
1201 struct btrfs_root *root, struct btrfs_path *path,
1202 struct btrfs_disk_key *key, int level)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001203{
1204 int i;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001205 int ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04001206 struct extent_buffer *t;
1207
Chris Mason234b63a2007-03-13 10:46:10 -04001208 for (i = level; i < BTRFS_MAX_LEVEL; i++) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05001209 int tslot = path->slots[i];
Chris Masoneb60cea2007-02-02 09:18:22 -05001210 if (!path->nodes[i])
Chris Masonbe0e5c02007-01-26 15:51:26 -05001211 break;
Chris Mason5f39d392007-10-15 16:14:19 -04001212 t = path->nodes[i];
1213 btrfs_set_node_key(t, key, tslot);
Chris Masond6025572007-03-30 14:27:56 -04001214 btrfs_mark_buffer_dirty(path->nodes[i]);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001215 if (tslot != 0)
1216 break;
1217 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05001218 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001219}
1220
Chris Mason74123bd2007-02-02 11:05:29 -05001221/*
1222 * try to push data from one node into the next node left in the
Chris Mason79f95c82007-03-01 15:16:26 -05001223 * tree.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001224 *
1225 * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1226 * error, and > 0 if there was no room in the left hand block.
Chris Mason74123bd2007-02-02 11:05:29 -05001227 */
Chris Mason98ed5172008-01-03 10:01:48 -05001228static int push_node_left(struct btrfs_trans_handle *trans,
1229 struct btrfs_root *root, struct extent_buffer *dst,
Chris Mason5f39d392007-10-15 16:14:19 -04001230 struct extent_buffer *src)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001231{
Chris Masonbe0e5c02007-01-26 15:51:26 -05001232 int push_items = 0;
Chris Masonbb803952007-03-01 12:04:21 -05001233 int src_nritems;
1234 int dst_nritems;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001235 int ret = 0;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001236
Chris Mason5f39d392007-10-15 16:14:19 -04001237 src_nritems = btrfs_header_nritems(src);
1238 dst_nritems = btrfs_header_nritems(dst);
Chris Mason123abc82007-03-14 14:14:43 -04001239 push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
Chris Mason7bb86312007-12-11 09:25:06 -05001240 WARN_ON(btrfs_header_generation(src) != trans->transid);
1241 WARN_ON(btrfs_header_generation(dst) != trans->transid);
Chris Mason54aa1f42007-06-22 14:16:25 -04001242
Chris Masoneb60cea2007-02-02 09:18:22 -05001243 if (push_items <= 0) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05001244 return 1;
Chris Masoneb60cea2007-02-02 09:18:22 -05001245 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05001246
1247 if (src_nritems < push_items)
Chris Mason79f95c82007-03-01 15:16:26 -05001248 push_items = src_nritems;
1249
Chris Mason5f39d392007-10-15 16:14:19 -04001250 copy_extent_buffer(dst, src,
1251 btrfs_node_key_ptr_offset(dst_nritems),
1252 btrfs_node_key_ptr_offset(0),
1253 push_items * sizeof(struct btrfs_key_ptr));
1254
Chris Masonbb803952007-03-01 12:04:21 -05001255 if (push_items < src_nritems) {
Chris Mason5f39d392007-10-15 16:14:19 -04001256 memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
1257 btrfs_node_key_ptr_offset(push_items),
1258 (src_nritems - push_items) *
1259 sizeof(struct btrfs_key_ptr));
Chris Masonbb803952007-03-01 12:04:21 -05001260 }
Chris Mason5f39d392007-10-15 16:14:19 -04001261 btrfs_set_header_nritems(src, src_nritems - push_items);
1262 btrfs_set_header_nritems(dst, dst_nritems + push_items);
1263 btrfs_mark_buffer_dirty(src);
1264 btrfs_mark_buffer_dirty(dst);
Chris Masonbb803952007-03-01 12:04:21 -05001265 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001266}
1267
Chris Mason97571fd2007-02-24 13:39:08 -05001268/*
Chris Mason79f95c82007-03-01 15:16:26 -05001269 * try to push data from one node into the next node right in the
1270 * tree.
1271 *
1272 * returns 0 if some ptrs were pushed, < 0 if there was some horrible
1273 * error, and > 0 if there was no room in the right hand block.
1274 *
1275 * this will only push up to 1/2 the contents of the left node over
1276 */
Chris Mason5f39d392007-10-15 16:14:19 -04001277static int balance_node_right(struct btrfs_trans_handle *trans,
1278 struct btrfs_root *root,
1279 struct extent_buffer *dst,
1280 struct extent_buffer *src)
Chris Mason79f95c82007-03-01 15:16:26 -05001281{
Chris Mason79f95c82007-03-01 15:16:26 -05001282 int push_items = 0;
1283 int max_push;
1284 int src_nritems;
1285 int dst_nritems;
1286 int ret = 0;
Chris Mason79f95c82007-03-01 15:16:26 -05001287
Chris Mason7bb86312007-12-11 09:25:06 -05001288 WARN_ON(btrfs_header_generation(src) != trans->transid);
1289 WARN_ON(btrfs_header_generation(dst) != trans->transid);
1290
Chris Mason5f39d392007-10-15 16:14:19 -04001291 src_nritems = btrfs_header_nritems(src);
1292 dst_nritems = btrfs_header_nritems(dst);
Chris Mason123abc82007-03-14 14:14:43 -04001293 push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001294 if (push_items <= 0)
Chris Mason79f95c82007-03-01 15:16:26 -05001295 return 1;
Chris Mason79f95c82007-03-01 15:16:26 -05001296
1297 max_push = src_nritems / 2 + 1;
1298 /* don't try to empty the node */
Yan252c38f2007-08-29 09:11:44 -04001299 if (max_push >= src_nritems)
Chris Mason79f95c82007-03-01 15:16:26 -05001300 return 1;
Yan252c38f2007-08-29 09:11:44 -04001301
Chris Mason79f95c82007-03-01 15:16:26 -05001302 if (max_push < push_items)
1303 push_items = max_push;
1304
Chris Mason5f39d392007-10-15 16:14:19 -04001305 memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
1306 btrfs_node_key_ptr_offset(0),
1307 (dst_nritems) *
1308 sizeof(struct btrfs_key_ptr));
Chris Masond6025572007-03-30 14:27:56 -04001309
Chris Mason5f39d392007-10-15 16:14:19 -04001310 copy_extent_buffer(dst, src,
1311 btrfs_node_key_ptr_offset(0),
1312 btrfs_node_key_ptr_offset(src_nritems - push_items),
1313 push_items * sizeof(struct btrfs_key_ptr));
Chris Mason79f95c82007-03-01 15:16:26 -05001314
Chris Mason5f39d392007-10-15 16:14:19 -04001315 btrfs_set_header_nritems(src, src_nritems - push_items);
1316 btrfs_set_header_nritems(dst, dst_nritems + push_items);
Chris Mason79f95c82007-03-01 15:16:26 -05001317
Chris Mason5f39d392007-10-15 16:14:19 -04001318 btrfs_mark_buffer_dirty(src);
1319 btrfs_mark_buffer_dirty(dst);
Chris Mason79f95c82007-03-01 15:16:26 -05001320 return ret;
1321}
1322
1323/*
Chris Mason97571fd2007-02-24 13:39:08 -05001324 * helper function to insert a new root level in the tree.
1325 * A new node is allocated, and a single item is inserted to
1326 * point to the existing root
Chris Masonaa5d6be2007-02-28 16:35:06 -05001327 *
1328 * returns zero on success or < 0 on failure.
Chris Mason97571fd2007-02-24 13:39:08 -05001329 */
Chris Mason98ed5172008-01-03 10:01:48 -05001330static int noinline insert_new_root(struct btrfs_trans_handle *trans,
Chris Mason5f39d392007-10-15 16:14:19 -04001331 struct btrfs_root *root,
1332 struct btrfs_path *path, int level)
Chris Mason5c680ed2007-02-22 11:39:13 -05001333{
Chris Mason7bb86312007-12-11 09:25:06 -05001334 u64 root_gen;
1335 u64 lower_gen;
Chris Mason5f39d392007-10-15 16:14:19 -04001336 struct extent_buffer *lower;
1337 struct extent_buffer *c;
1338 struct btrfs_disk_key lower_key;
Chris Mason5c680ed2007-02-22 11:39:13 -05001339
1340 BUG_ON(path->nodes[level]);
1341 BUG_ON(path->nodes[level-1] != root->node);
1342
Chris Mason7bb86312007-12-11 09:25:06 -05001343 if (root->ref_cows)
1344 root_gen = trans->transid;
1345 else
1346 root_gen = 0;
1347
1348 lower = path->nodes[level-1];
1349 if (level == 1)
1350 btrfs_item_key(lower, &lower_key, 0);
1351 else
1352 btrfs_node_key(lower, &lower_key, 0);
1353
1354 c = __btrfs_alloc_free_block(trans, root, root->nodesize,
1355 root->root_key.objectid,
1356 root_gen, lower_key.objectid, level,
Chris Masondb945352007-10-15 16:15:53 -04001357 root->node->start, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04001358 if (IS_ERR(c))
1359 return PTR_ERR(c);
1360 memset_extent_buffer(c, 0, 0, root->nodesize);
1361 btrfs_set_header_nritems(c, 1);
1362 btrfs_set_header_level(c, level);
Chris Masondb945352007-10-15 16:15:53 -04001363 btrfs_set_header_bytenr(c, c->start);
Chris Mason5f39d392007-10-15 16:14:19 -04001364 btrfs_set_header_generation(c, trans->transid);
1365 btrfs_set_header_owner(c, root->root_key.objectid);
Chris Masond5719762007-03-23 10:01:08 -04001366
Chris Mason5f39d392007-10-15 16:14:19 -04001367 write_extent_buffer(c, root->fs_info->fsid,
1368 (unsigned long)btrfs_header_fsid(c),
1369 BTRFS_FSID_SIZE);
Chris Mason5f39d392007-10-15 16:14:19 -04001370 btrfs_set_node_key(c, &lower_key, 0);
Chris Masondb945352007-10-15 16:15:53 -04001371 btrfs_set_node_blockptr(c, 0, lower->start);
Chris Mason7bb86312007-12-11 09:25:06 -05001372 lower_gen = btrfs_header_generation(lower);
1373 WARN_ON(lower_gen == 0);
1374
1375 btrfs_set_node_ptr_generation(c, 0, lower_gen);
Chris Mason5f39d392007-10-15 16:14:19 -04001376
1377 btrfs_mark_buffer_dirty(c);
Chris Masond5719762007-03-23 10:01:08 -04001378
Chris Mason5c680ed2007-02-22 11:39:13 -05001379 /* the super has an extra ref to root->node */
Chris Mason5f39d392007-10-15 16:14:19 -04001380 free_extent_buffer(root->node);
1381 root->node = c;
Chris Mason0b86a832008-03-24 15:01:56 -04001382 add_root_to_dirty_list(root);
Chris Mason5f39d392007-10-15 16:14:19 -04001383 extent_buffer_get(c);
1384 path->nodes[level] = c;
Chris Mason5c680ed2007-02-22 11:39:13 -05001385 path->slots[level] = 0;
Chris Mason7bb86312007-12-11 09:25:06 -05001386
1387 if (root->ref_cows && lower_gen != trans->transid) {
1388 struct btrfs_path *back_path = btrfs_alloc_path();
1389 int ret;
1390 ret = btrfs_insert_extent_backref(trans,
1391 root->fs_info->extent_root,
1392 path, lower->start,
1393 root->root_key.objectid,
1394 trans->transid, 0, 0);
1395 BUG_ON(ret);
1396 btrfs_free_path(back_path);
1397 }
Chris Mason5c680ed2007-02-22 11:39:13 -05001398 return 0;
1399}
1400
Chris Mason74123bd2007-02-02 11:05:29 -05001401/*
1402 * worker function to insert a single pointer in a node.
1403 * the node should have enough room for the pointer already
Chris Mason97571fd2007-02-24 13:39:08 -05001404 *
Chris Mason74123bd2007-02-02 11:05:29 -05001405 * slot and level indicate where you want the key to go, and
1406 * blocknr is the block the key points to.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001407 *
1408 * returns zero on success and < 0 on any error
Chris Mason74123bd2007-02-02 11:05:29 -05001409 */
Chris Masone089f052007-03-16 16:20:31 -04001410static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1411 *root, struct btrfs_path *path, struct btrfs_disk_key
Chris Masondb945352007-10-15 16:15:53 -04001412 *key, u64 bytenr, int slot, int level)
Chris Mason74123bd2007-02-02 11:05:29 -05001413{
Chris Mason5f39d392007-10-15 16:14:19 -04001414 struct extent_buffer *lower;
Chris Mason74123bd2007-02-02 11:05:29 -05001415 int nritems;
Chris Mason5c680ed2007-02-22 11:39:13 -05001416
1417 BUG_ON(!path->nodes[level]);
Chris Mason5f39d392007-10-15 16:14:19 -04001418 lower = path->nodes[level];
1419 nritems = btrfs_header_nritems(lower);
Chris Mason74123bd2007-02-02 11:05:29 -05001420 if (slot > nritems)
1421 BUG();
Chris Mason123abc82007-03-14 14:14:43 -04001422 if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
Chris Mason74123bd2007-02-02 11:05:29 -05001423 BUG();
1424 if (slot != nritems) {
Chris Mason5f39d392007-10-15 16:14:19 -04001425 memmove_extent_buffer(lower,
1426 btrfs_node_key_ptr_offset(slot + 1),
1427 btrfs_node_key_ptr_offset(slot),
Chris Masond6025572007-03-30 14:27:56 -04001428 (nritems - slot) * sizeof(struct btrfs_key_ptr));
Chris Mason74123bd2007-02-02 11:05:29 -05001429 }
Chris Mason5f39d392007-10-15 16:14:19 -04001430 btrfs_set_node_key(lower, key, slot);
Chris Masondb945352007-10-15 16:15:53 -04001431 btrfs_set_node_blockptr(lower, slot, bytenr);
Chris Mason74493f72007-12-11 09:25:06 -05001432 WARN_ON(trans->transid == 0);
1433 btrfs_set_node_ptr_generation(lower, slot, trans->transid);
Chris Mason5f39d392007-10-15 16:14:19 -04001434 btrfs_set_header_nritems(lower, nritems + 1);
1435 btrfs_mark_buffer_dirty(lower);
Chris Mason74123bd2007-02-02 11:05:29 -05001436 return 0;
1437}
1438
Chris Mason97571fd2007-02-24 13:39:08 -05001439/*
1440 * split the node at the specified level in path in two.
1441 * The path is corrected to point to the appropriate node after the split
1442 *
1443 * Before splitting this tries to make some room in the node by pushing
1444 * left and right, if either one works, it returns right away.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001445 *
1446 * returns 0 on success and < 0 on failure
Chris Mason97571fd2007-02-24 13:39:08 -05001447 */
Chris Masone089f052007-03-16 16:20:31 -04001448static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1449 *root, struct btrfs_path *path, int level)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001450{
Chris Mason7bb86312007-12-11 09:25:06 -05001451 u64 root_gen;
Chris Mason5f39d392007-10-15 16:14:19 -04001452 struct extent_buffer *c;
1453 struct extent_buffer *split;
1454 struct btrfs_disk_key disk_key;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001455 int mid;
Chris Mason5c680ed2007-02-22 11:39:13 -05001456 int ret;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001457 int wret;
Chris Mason7518a232007-03-12 12:01:18 -04001458 u32 c_nritems;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001459
Chris Mason5f39d392007-10-15 16:14:19 -04001460 c = path->nodes[level];
Chris Mason7bb86312007-12-11 09:25:06 -05001461 WARN_ON(btrfs_header_generation(c) != trans->transid);
Chris Mason5f39d392007-10-15 16:14:19 -04001462 if (c == root->node) {
Chris Mason5c680ed2007-02-22 11:39:13 -05001463 /* trying to split the root, lets make a new one */
Chris Masone089f052007-03-16 16:20:31 -04001464 ret = insert_new_root(trans, root, path, level + 1);
Chris Mason5c680ed2007-02-22 11:39:13 -05001465 if (ret)
1466 return ret;
Chris Masone66f7092007-04-20 13:16:02 -04001467 } else {
1468 ret = push_nodes_for_insert(trans, root, path, level);
Chris Mason5f39d392007-10-15 16:14:19 -04001469 c = path->nodes[level];
1470 if (!ret && btrfs_header_nritems(c) <
Chris Masone66f7092007-04-20 13:16:02 -04001471 BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
1472 return 0;
Chris Mason54aa1f42007-06-22 14:16:25 -04001473 if (ret < 0)
1474 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001475 }
Chris Masone66f7092007-04-20 13:16:02 -04001476
Chris Mason5f39d392007-10-15 16:14:19 -04001477 c_nritems = btrfs_header_nritems(c);
Chris Mason7bb86312007-12-11 09:25:06 -05001478 if (root->ref_cows)
1479 root_gen = trans->transid;
1480 else
1481 root_gen = 0;
1482
1483 btrfs_node_key(c, &disk_key, 0);
1484 split = __btrfs_alloc_free_block(trans, root, root->nodesize,
1485 root->root_key.objectid,
1486 root_gen,
1487 btrfs_disk_key_objectid(&disk_key),
1488 level, c->start, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04001489 if (IS_ERR(split))
1490 return PTR_ERR(split);
Chris Mason54aa1f42007-06-22 14:16:25 -04001491
Chris Mason5f39d392007-10-15 16:14:19 -04001492 btrfs_set_header_flags(split, btrfs_header_flags(c));
1493 btrfs_set_header_level(split, btrfs_header_level(c));
Chris Masondb945352007-10-15 16:15:53 -04001494 btrfs_set_header_bytenr(split, split->start);
Chris Mason5f39d392007-10-15 16:14:19 -04001495 btrfs_set_header_generation(split, trans->transid);
1496 btrfs_set_header_owner(split, root->root_key.objectid);
1497 write_extent_buffer(split, root->fs_info->fsid,
1498 (unsigned long)btrfs_header_fsid(split),
1499 BTRFS_FSID_SIZE);
1500
Chris Mason7518a232007-03-12 12:01:18 -04001501 mid = (c_nritems + 1) / 2;
Chris Mason5f39d392007-10-15 16:14:19 -04001502
1503 copy_extent_buffer(split, c,
1504 btrfs_node_key_ptr_offset(0),
1505 btrfs_node_key_ptr_offset(mid),
1506 (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
1507 btrfs_set_header_nritems(split, c_nritems - mid);
1508 btrfs_set_header_nritems(c, mid);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001509 ret = 0;
1510
Chris Mason5f39d392007-10-15 16:14:19 -04001511 btrfs_mark_buffer_dirty(c);
1512 btrfs_mark_buffer_dirty(split);
1513
1514 btrfs_node_key(split, &disk_key, 0);
Chris Masondb945352007-10-15 16:15:53 -04001515 wret = insert_ptr(trans, root, path, &disk_key, split->start,
Chris Mason5f39d392007-10-15 16:14:19 -04001516 path->slots[level + 1] + 1,
Chris Mason123abc82007-03-14 14:14:43 -04001517 level + 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001518 if (wret)
1519 ret = wret;
1520
Chris Mason5de08d72007-02-24 06:24:44 -05001521 if (path->slots[level] >= mid) {
Chris Mason5c680ed2007-02-22 11:39:13 -05001522 path->slots[level] -= mid;
Chris Mason5f39d392007-10-15 16:14:19 -04001523 free_extent_buffer(c);
1524 path->nodes[level] = split;
Chris Mason5c680ed2007-02-22 11:39:13 -05001525 path->slots[level + 1] += 1;
1526 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04001527 free_extent_buffer(split);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001528 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05001529 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001530}
1531
Chris Mason74123bd2007-02-02 11:05:29 -05001532/*
1533 * how many bytes are required to store the items in a leaf. start
1534 * and nr indicate which items in the leaf to check. This totals up the
1535 * space used both by the item structs and the item data
1536 */
Chris Mason5f39d392007-10-15 16:14:19 -04001537static int leaf_space_used(struct extent_buffer *l, int start, int nr)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001538{
1539 int data_len;
Chris Mason5f39d392007-10-15 16:14:19 -04001540 int nritems = btrfs_header_nritems(l);
Chris Masond4dbff92007-04-04 14:08:15 -04001541 int end = min(nritems, start + nr) - 1;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001542
1543 if (!nr)
1544 return 0;
Chris Mason5f39d392007-10-15 16:14:19 -04001545 data_len = btrfs_item_end_nr(l, start);
1546 data_len = data_len - btrfs_item_offset_nr(l, end);
Chris Mason0783fcf2007-03-12 20:12:07 -04001547 data_len += sizeof(struct btrfs_item) * nr;
Chris Masond4dbff92007-04-04 14:08:15 -04001548 WARN_ON(data_len < 0);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001549 return data_len;
1550}
1551
Chris Mason74123bd2007-02-02 11:05:29 -05001552/*
Chris Masond4dbff92007-04-04 14:08:15 -04001553 * The space between the end of the leaf items and
1554 * the start of the leaf data. IOW, how much room
1555 * the leaf has left for both items and data
1556 */
Chris Mason5f39d392007-10-15 16:14:19 -04001557int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
Chris Masond4dbff92007-04-04 14:08:15 -04001558{
Chris Mason5f39d392007-10-15 16:14:19 -04001559 int nritems = btrfs_header_nritems(leaf);
1560 int ret;
1561 ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
1562 if (ret < 0) {
1563 printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
Jens Axboeae2f5412007-10-19 09:22:59 -04001564 ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
Chris Mason5f39d392007-10-15 16:14:19 -04001565 leaf_space_used(leaf, 0, nritems), nritems);
1566 }
1567 return ret;
Chris Masond4dbff92007-04-04 14:08:15 -04001568}
1569
1570/*
Chris Mason00ec4c52007-02-24 12:47:20 -05001571 * push some data in the path leaf to the right, trying to free up at
1572 * least data_size bytes. returns zero if the push worked, nonzero otherwise
Chris Masonaa5d6be2007-02-28 16:35:06 -05001573 *
1574 * returns 1 if the push failed because the other node didn't have enough
1575 * room, 0 if everything worked out and < 0 if there were major errors.
Chris Mason00ec4c52007-02-24 12:47:20 -05001576 */
Chris Masone089f052007-03-16 16:20:31 -04001577static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
Chris Mason34a38212007-11-07 13:31:03 -05001578 *root, struct btrfs_path *path, int data_size,
1579 int empty)
Chris Mason00ec4c52007-02-24 12:47:20 -05001580{
Chris Mason5f39d392007-10-15 16:14:19 -04001581 struct extent_buffer *left = path->nodes[0];
1582 struct extent_buffer *right;
1583 struct extent_buffer *upper;
1584 struct btrfs_disk_key disk_key;
Chris Mason00ec4c52007-02-24 12:47:20 -05001585 int slot;
Chris Mason34a38212007-11-07 13:31:03 -05001586 u32 i;
Chris Mason00ec4c52007-02-24 12:47:20 -05001587 int free_space;
1588 int push_space = 0;
1589 int push_items = 0;
Chris Mason0783fcf2007-03-12 20:12:07 -04001590 struct btrfs_item *item;
Chris Mason7518a232007-03-12 12:01:18 -04001591 u32 left_nritems;
Chris Mason34a38212007-11-07 13:31:03 -05001592 u32 nr;
Chris Mason7518a232007-03-12 12:01:18 -04001593 u32 right_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001594 u32 data_end;
Chris Masondb945352007-10-15 16:15:53 -04001595 u32 this_item_size;
Chris Mason54aa1f42007-06-22 14:16:25 -04001596 int ret;
Chris Mason00ec4c52007-02-24 12:47:20 -05001597
1598 slot = path->slots[1];
1599 if (!path->nodes[1]) {
1600 return 1;
1601 }
1602 upper = path->nodes[1];
Chris Mason5f39d392007-10-15 16:14:19 -04001603 if (slot >= btrfs_header_nritems(upper) - 1)
Chris Mason00ec4c52007-02-24 12:47:20 -05001604 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04001605
Chris Masondb945352007-10-15 16:15:53 -04001606 right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
1607 root->leafsize);
Chris Mason123abc82007-03-14 14:14:43 -04001608 free_space = btrfs_leaf_free_space(root, right);
Chris Mason0783fcf2007-03-12 20:12:07 -04001609 if (free_space < data_size + sizeof(struct btrfs_item)) {
Chris Mason5f39d392007-10-15 16:14:19 -04001610 free_extent_buffer(right);
Chris Mason02217ed2007-03-02 16:08:05 -05001611 return 1;
1612 }
1613
Chris Mason5f39d392007-10-15 16:14:19 -04001614 /* cow and double check */
1615 ret = btrfs_cow_block(trans, root, right, upper,
1616 slot + 1, &right);
1617 if (ret) {
1618 free_extent_buffer(right);
Chris Masona429e512007-04-18 16:15:28 -04001619 return 1;
1620 }
Chris Mason5f39d392007-10-15 16:14:19 -04001621 free_space = btrfs_leaf_free_space(root, right);
1622 if (free_space < data_size + sizeof(struct btrfs_item)) {
1623 free_extent_buffer(right);
1624 return 1;
1625 }
1626
1627 left_nritems = btrfs_header_nritems(left);
1628 if (left_nritems == 0) {
1629 free_extent_buffer(right);
1630 return 1;
1631 }
1632
Chris Mason34a38212007-11-07 13:31:03 -05001633 if (empty)
1634 nr = 0;
1635 else
1636 nr = 1;
1637
1638 i = left_nritems - 1;
1639 while (i >= nr) {
Chris Mason5f39d392007-10-15 16:14:19 -04001640 item = btrfs_item_nr(left, i);
Chris Masondb945352007-10-15 16:15:53 -04001641
Chris Mason00ec4c52007-02-24 12:47:20 -05001642 if (path->slots[0] == i)
1643 push_space += data_size + sizeof(*item);
Chris Masondb945352007-10-15 16:15:53 -04001644
1645 if (!left->map_token) {
1646 map_extent_buffer(left, (unsigned long)item,
1647 sizeof(struct btrfs_item),
1648 &left->map_token, &left->kaddr,
1649 &left->map_start, &left->map_len,
1650 KM_USER1);
1651 }
1652
1653 this_item_size = btrfs_item_size(left, item);
1654 if (this_item_size + sizeof(*item) + push_space > free_space)
Chris Mason00ec4c52007-02-24 12:47:20 -05001655 break;
1656 push_items++;
Chris Masondb945352007-10-15 16:15:53 -04001657 push_space += this_item_size + sizeof(*item);
Chris Mason34a38212007-11-07 13:31:03 -05001658 if (i == 0)
1659 break;
1660 i--;
Chris Masondb945352007-10-15 16:15:53 -04001661 }
1662 if (left->map_token) {
1663 unmap_extent_buffer(left, left->map_token, KM_USER1);
1664 left->map_token = NULL;
Chris Mason00ec4c52007-02-24 12:47:20 -05001665 }
Chris Mason5f39d392007-10-15 16:14:19 -04001666
Chris Mason00ec4c52007-02-24 12:47:20 -05001667 if (push_items == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04001668 free_extent_buffer(right);
Chris Mason00ec4c52007-02-24 12:47:20 -05001669 return 1;
1670 }
Chris Mason5f39d392007-10-15 16:14:19 -04001671
Chris Mason34a38212007-11-07 13:31:03 -05001672 if (!empty && push_items == left_nritems)
Chris Masona429e512007-04-18 16:15:28 -04001673 WARN_ON(1);
Chris Mason5f39d392007-10-15 16:14:19 -04001674
Chris Mason00ec4c52007-02-24 12:47:20 -05001675 /* push left to right */
Chris Mason5f39d392007-10-15 16:14:19 -04001676 right_nritems = btrfs_header_nritems(right);
Chris Mason34a38212007-11-07 13:31:03 -05001677
Chris Mason5f39d392007-10-15 16:14:19 -04001678 push_space = btrfs_item_end_nr(left, left_nritems - push_items);
Chris Mason123abc82007-03-14 14:14:43 -04001679 push_space -= leaf_data_end(root, left);
Chris Mason5f39d392007-10-15 16:14:19 -04001680
Chris Mason00ec4c52007-02-24 12:47:20 -05001681 /* make room in the right data area */
Chris Mason5f39d392007-10-15 16:14:19 -04001682 data_end = leaf_data_end(root, right);
1683 memmove_extent_buffer(right,
1684 btrfs_leaf_data(right) + data_end - push_space,
1685 btrfs_leaf_data(right) + data_end,
1686 BTRFS_LEAF_DATA_SIZE(root) - data_end);
1687
Chris Mason00ec4c52007-02-24 12:47:20 -05001688 /* copy from the left data area */
Chris Mason5f39d392007-10-15 16:14:19 -04001689 copy_extent_buffer(right, left, btrfs_leaf_data(right) +
Chris Masond6025572007-03-30 14:27:56 -04001690 BTRFS_LEAF_DATA_SIZE(root) - push_space,
1691 btrfs_leaf_data(left) + leaf_data_end(root, left),
1692 push_space);
Chris Mason5f39d392007-10-15 16:14:19 -04001693
1694 memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
1695 btrfs_item_nr_offset(0),
1696 right_nritems * sizeof(struct btrfs_item));
1697
Chris Mason00ec4c52007-02-24 12:47:20 -05001698 /* copy the items from left to right */
Chris Mason5f39d392007-10-15 16:14:19 -04001699 copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
1700 btrfs_item_nr_offset(left_nritems - push_items),
1701 push_items * sizeof(struct btrfs_item));
Chris Mason00ec4c52007-02-24 12:47:20 -05001702
1703 /* update the item pointers */
Chris Mason7518a232007-03-12 12:01:18 -04001704 right_nritems += push_items;
Chris Mason5f39d392007-10-15 16:14:19 -04001705 btrfs_set_header_nritems(right, right_nritems);
Chris Mason123abc82007-03-14 14:14:43 -04001706 push_space = BTRFS_LEAF_DATA_SIZE(root);
Chris Mason7518a232007-03-12 12:01:18 -04001707 for (i = 0; i < right_nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04001708 item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04001709 if (!right->map_token) {
1710 map_extent_buffer(right, (unsigned long)item,
1711 sizeof(struct btrfs_item),
1712 &right->map_token, &right->kaddr,
1713 &right->map_start, &right->map_len,
1714 KM_USER1);
1715 }
1716 push_space -= btrfs_item_size(right, item);
1717 btrfs_set_item_offset(right, item, push_space);
1718 }
1719
1720 if (right->map_token) {
1721 unmap_extent_buffer(right, right->map_token, KM_USER1);
1722 right->map_token = NULL;
Chris Mason00ec4c52007-02-24 12:47:20 -05001723 }
Chris Mason7518a232007-03-12 12:01:18 -04001724 left_nritems -= push_items;
Chris Mason5f39d392007-10-15 16:14:19 -04001725 btrfs_set_header_nritems(left, left_nritems);
Chris Mason00ec4c52007-02-24 12:47:20 -05001726
Chris Mason34a38212007-11-07 13:31:03 -05001727 if (left_nritems)
1728 btrfs_mark_buffer_dirty(left);
Chris Mason5f39d392007-10-15 16:14:19 -04001729 btrfs_mark_buffer_dirty(right);
Chris Masona429e512007-04-18 16:15:28 -04001730
Chris Mason5f39d392007-10-15 16:14:19 -04001731 btrfs_item_key(right, &disk_key, 0);
1732 btrfs_set_node_key(upper, &disk_key, slot + 1);
Chris Masond6025572007-03-30 14:27:56 -04001733 btrfs_mark_buffer_dirty(upper);
Chris Mason02217ed2007-03-02 16:08:05 -05001734
Chris Mason00ec4c52007-02-24 12:47:20 -05001735 /* then fixup the leaf pointer in the path */
Chris Mason7518a232007-03-12 12:01:18 -04001736 if (path->slots[0] >= left_nritems) {
1737 path->slots[0] -= left_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001738 free_extent_buffer(path->nodes[0]);
1739 path->nodes[0] = right;
Chris Mason00ec4c52007-02-24 12:47:20 -05001740 path->slots[1] += 1;
1741 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04001742 free_extent_buffer(right);
Chris Mason00ec4c52007-02-24 12:47:20 -05001743 }
1744 return 0;
1745}
1746/*
Chris Mason74123bd2007-02-02 11:05:29 -05001747 * push some data in the path leaf to the left, trying to free up at
1748 * least data_size bytes. returns zero if the push worked, nonzero otherwise
1749 */
Chris Masone089f052007-03-16 16:20:31 -04001750static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
Chris Mason34a38212007-11-07 13:31:03 -05001751 *root, struct btrfs_path *path, int data_size,
1752 int empty)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001753{
Chris Mason5f39d392007-10-15 16:14:19 -04001754 struct btrfs_disk_key disk_key;
1755 struct extent_buffer *right = path->nodes[0];
1756 struct extent_buffer *left;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001757 int slot;
1758 int i;
1759 int free_space;
1760 int push_space = 0;
1761 int push_items = 0;
Chris Mason0783fcf2007-03-12 20:12:07 -04001762 struct btrfs_item *item;
Chris Mason7518a232007-03-12 12:01:18 -04001763 u32 old_left_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001764 u32 right_nritems;
Chris Mason34a38212007-11-07 13:31:03 -05001765 u32 nr;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001766 int ret = 0;
1767 int wret;
Chris Masondb945352007-10-15 16:15:53 -04001768 u32 this_item_size;
1769 u32 old_left_item_size;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001770
1771 slot = path->slots[1];
Chris Mason5f39d392007-10-15 16:14:19 -04001772 if (slot == 0)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001773 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04001774 if (!path->nodes[1])
Chris Masonbe0e5c02007-01-26 15:51:26 -05001775 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04001776
Chris Mason3685f792007-10-19 09:23:27 -04001777 right_nritems = btrfs_header_nritems(right);
1778 if (right_nritems == 0) {
1779 return 1;
1780 }
1781
Chris Mason5f39d392007-10-15 16:14:19 -04001782 left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
Chris Masondb945352007-10-15 16:15:53 -04001783 slot - 1), root->leafsize);
Chris Mason123abc82007-03-14 14:14:43 -04001784 free_space = btrfs_leaf_free_space(root, left);
Chris Mason0783fcf2007-03-12 20:12:07 -04001785 if (free_space < data_size + sizeof(struct btrfs_item)) {
Chris Mason5f39d392007-10-15 16:14:19 -04001786 free_extent_buffer(left);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001787 return 1;
1788 }
Chris Mason02217ed2007-03-02 16:08:05 -05001789
1790 /* cow and double check */
Chris Mason5f39d392007-10-15 16:14:19 -04001791 ret = btrfs_cow_block(trans, root, left,
1792 path->nodes[1], slot - 1, &left);
Chris Mason54aa1f42007-06-22 14:16:25 -04001793 if (ret) {
1794 /* we hit -ENOSPC, but it isn't fatal here */
Chris Mason5f39d392007-10-15 16:14:19 -04001795 free_extent_buffer(left);
Chris Mason54aa1f42007-06-22 14:16:25 -04001796 return 1;
1797 }
Chris Mason3685f792007-10-19 09:23:27 -04001798
Chris Mason123abc82007-03-14 14:14:43 -04001799 free_space = btrfs_leaf_free_space(root, left);
Chris Mason0783fcf2007-03-12 20:12:07 -04001800 if (free_space < data_size + sizeof(struct btrfs_item)) {
Chris Mason5f39d392007-10-15 16:14:19 -04001801 free_extent_buffer(left);
Chris Mason02217ed2007-03-02 16:08:05 -05001802 return 1;
1803 }
1804
Chris Mason34a38212007-11-07 13:31:03 -05001805 if (empty)
1806 nr = right_nritems;
1807 else
1808 nr = right_nritems - 1;
1809
1810 for (i = 0; i < nr; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04001811 item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04001812 if (!right->map_token) {
1813 map_extent_buffer(right, (unsigned long)item,
1814 sizeof(struct btrfs_item),
1815 &right->map_token, &right->kaddr,
1816 &right->map_start, &right->map_len,
1817 KM_USER1);
1818 }
1819
Chris Masonbe0e5c02007-01-26 15:51:26 -05001820 if (path->slots[0] == i)
1821 push_space += data_size + sizeof(*item);
Chris Masondb945352007-10-15 16:15:53 -04001822
1823 this_item_size = btrfs_item_size(right, item);
1824 if (this_item_size + sizeof(*item) + push_space > free_space)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001825 break;
Chris Masondb945352007-10-15 16:15:53 -04001826
Chris Masonbe0e5c02007-01-26 15:51:26 -05001827 push_items++;
Chris Masondb945352007-10-15 16:15:53 -04001828 push_space += this_item_size + sizeof(*item);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001829 }
Chris Masondb945352007-10-15 16:15:53 -04001830
1831 if (right->map_token) {
1832 unmap_extent_buffer(right, right->map_token, KM_USER1);
1833 right->map_token = NULL;
1834 }
1835
Chris Masonbe0e5c02007-01-26 15:51:26 -05001836 if (push_items == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04001837 free_extent_buffer(left);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001838 return 1;
1839 }
Chris Mason34a38212007-11-07 13:31:03 -05001840 if (!empty && push_items == btrfs_header_nritems(right))
Chris Masona429e512007-04-18 16:15:28 -04001841 WARN_ON(1);
Chris Mason5f39d392007-10-15 16:14:19 -04001842
Chris Masonbe0e5c02007-01-26 15:51:26 -05001843 /* push data from right to left */
Chris Mason5f39d392007-10-15 16:14:19 -04001844 copy_extent_buffer(left, right,
1845 btrfs_item_nr_offset(btrfs_header_nritems(left)),
1846 btrfs_item_nr_offset(0),
1847 push_items * sizeof(struct btrfs_item));
1848
Chris Mason123abc82007-03-14 14:14:43 -04001849 push_space = BTRFS_LEAF_DATA_SIZE(root) -
Chris Mason5f39d392007-10-15 16:14:19 -04001850 btrfs_item_offset_nr(right, push_items -1);
1851
1852 copy_extent_buffer(left, right, btrfs_leaf_data(left) +
Chris Masond6025572007-03-30 14:27:56 -04001853 leaf_data_end(root, left) - push_space,
1854 btrfs_leaf_data(right) +
Chris Mason5f39d392007-10-15 16:14:19 -04001855 btrfs_item_offset_nr(right, push_items - 1),
Chris Masond6025572007-03-30 14:27:56 -04001856 push_space);
Chris Mason5f39d392007-10-15 16:14:19 -04001857 old_left_nritems = btrfs_header_nritems(left);
Chris Masoneb60cea2007-02-02 09:18:22 -05001858 BUG_ON(old_left_nritems < 0);
1859
Chris Masondb945352007-10-15 16:15:53 -04001860 old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
Chris Mason0783fcf2007-03-12 20:12:07 -04001861 for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04001862 u32 ioff;
Chris Masondb945352007-10-15 16:15:53 -04001863
Chris Mason5f39d392007-10-15 16:14:19 -04001864 item = btrfs_item_nr(left, i);
Chris Masondb945352007-10-15 16:15:53 -04001865 if (!left->map_token) {
1866 map_extent_buffer(left, (unsigned long)item,
1867 sizeof(struct btrfs_item),
1868 &left->map_token, &left->kaddr,
1869 &left->map_start, &left->map_len,
1870 KM_USER1);
1871 }
1872
Chris Mason5f39d392007-10-15 16:14:19 -04001873 ioff = btrfs_item_offset(left, item);
1874 btrfs_set_item_offset(left, item,
Chris Masondb945352007-10-15 16:15:53 -04001875 ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
Chris Masonbe0e5c02007-01-26 15:51:26 -05001876 }
Chris Mason5f39d392007-10-15 16:14:19 -04001877 btrfs_set_header_nritems(left, old_left_nritems + push_items);
Chris Masondb945352007-10-15 16:15:53 -04001878 if (left->map_token) {
1879 unmap_extent_buffer(left, left->map_token, KM_USER1);
1880 left->map_token = NULL;
1881 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05001882
1883 /* fixup right node */
Chris Mason34a38212007-11-07 13:31:03 -05001884 if (push_items > right_nritems) {
1885 printk("push items %d nr %u\n", push_items, right_nritems);
1886 WARN_ON(1);
1887 }
Chris Mason5f39d392007-10-15 16:14:19 -04001888
Chris Mason34a38212007-11-07 13:31:03 -05001889 if (push_items < right_nritems) {
1890 push_space = btrfs_item_offset_nr(right, push_items - 1) -
1891 leaf_data_end(root, right);
1892 memmove_extent_buffer(right, btrfs_leaf_data(right) +
1893 BTRFS_LEAF_DATA_SIZE(root) - push_space,
1894 btrfs_leaf_data(right) +
1895 leaf_data_end(root, right), push_space);
1896
1897 memmove_extent_buffer(right, btrfs_item_nr_offset(0),
Chris Mason5f39d392007-10-15 16:14:19 -04001898 btrfs_item_nr_offset(push_items),
1899 (btrfs_header_nritems(right) - push_items) *
1900 sizeof(struct btrfs_item));
Chris Mason34a38212007-11-07 13:31:03 -05001901 }
Yaneef1c492007-11-26 10:58:13 -05001902 right_nritems -= push_items;
1903 btrfs_set_header_nritems(right, right_nritems);
Chris Mason123abc82007-03-14 14:14:43 -04001904 push_space = BTRFS_LEAF_DATA_SIZE(root);
Chris Mason5f39d392007-10-15 16:14:19 -04001905 for (i = 0; i < right_nritems; i++) {
1906 item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04001907
1908 if (!right->map_token) {
1909 map_extent_buffer(right, (unsigned long)item,
1910 sizeof(struct btrfs_item),
1911 &right->map_token, &right->kaddr,
1912 &right->map_start, &right->map_len,
1913 KM_USER1);
1914 }
1915
1916 push_space = push_space - btrfs_item_size(right, item);
1917 btrfs_set_item_offset(right, item, push_space);
1918 }
1919 if (right->map_token) {
1920 unmap_extent_buffer(right, right->map_token, KM_USER1);
1921 right->map_token = NULL;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001922 }
Chris Masoneb60cea2007-02-02 09:18:22 -05001923
Chris Mason5f39d392007-10-15 16:14:19 -04001924 btrfs_mark_buffer_dirty(left);
Chris Mason34a38212007-11-07 13:31:03 -05001925 if (right_nritems)
1926 btrfs_mark_buffer_dirty(right);
Chris Mason098f59c2007-05-11 11:33:21 -04001927
Chris Mason5f39d392007-10-15 16:14:19 -04001928 btrfs_item_key(right, &disk_key, 0);
1929 wret = fixup_low_keys(trans, root, path, &disk_key, 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001930 if (wret)
1931 ret = wret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001932
1933 /* then fixup the leaf pointer in the path */
1934 if (path->slots[0] < push_items) {
1935 path->slots[0] += old_left_nritems;
Chris Mason5f39d392007-10-15 16:14:19 -04001936 free_extent_buffer(path->nodes[0]);
1937 path->nodes[0] = left;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001938 path->slots[1] -= 1;
1939 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04001940 free_extent_buffer(left);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001941 path->slots[0] -= push_items;
1942 }
Chris Masoneb60cea2007-02-02 09:18:22 -05001943 BUG_ON(path->slots[0] < 0);
Chris Masonaa5d6be2007-02-28 16:35:06 -05001944 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001945}
1946
Chris Mason74123bd2007-02-02 11:05:29 -05001947/*
1948 * split the path's leaf in two, making sure there is at least data_size
1949 * available for the resulting leaf level of the path.
Chris Masonaa5d6be2007-02-28 16:35:06 -05001950 *
1951 * returns 0 if all went well and < 0 on failure.
Chris Mason74123bd2007-02-02 11:05:29 -05001952 */
Chris Masone089f052007-03-16 16:20:31 -04001953static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
Chris Masond4dbff92007-04-04 14:08:15 -04001954 *root, struct btrfs_key *ins_key,
Chris Masoncc0c5532007-10-25 15:42:57 -04001955 struct btrfs_path *path, int data_size, int extend)
Chris Masonbe0e5c02007-01-26 15:51:26 -05001956{
Chris Mason7bb86312007-12-11 09:25:06 -05001957 u64 root_gen;
Chris Mason5f39d392007-10-15 16:14:19 -04001958 struct extent_buffer *l;
Chris Mason7518a232007-03-12 12:01:18 -04001959 u32 nritems;
Chris Masoneb60cea2007-02-02 09:18:22 -05001960 int mid;
1961 int slot;
Chris Mason5f39d392007-10-15 16:14:19 -04001962 struct extent_buffer *right;
Chris Mason0783fcf2007-03-12 20:12:07 -04001963 int space_needed = data_size + sizeof(struct btrfs_item);
Chris Masonbe0e5c02007-01-26 15:51:26 -05001964 int data_copy_size;
1965 int rt_data_off;
1966 int i;
Chris Masond4dbff92007-04-04 14:08:15 -04001967 int ret = 0;
Chris Masonaa5d6be2007-02-28 16:35:06 -05001968 int wret;
Chris Masoncc0c5532007-10-25 15:42:57 -04001969 int double_split;
1970 int num_doubles = 0;
Chris Masond4dbff92007-04-04 14:08:15 -04001971 struct btrfs_disk_key disk_key;
Chris Masonbe0e5c02007-01-26 15:51:26 -05001972
Chris Masoncc0c5532007-10-25 15:42:57 -04001973 if (extend)
1974 space_needed = data_size;
1975
Chris Mason7bb86312007-12-11 09:25:06 -05001976 if (root->ref_cows)
1977 root_gen = trans->transid;
1978 else
1979 root_gen = 0;
1980
Chris Mason40689472007-03-17 14:29:23 -04001981 /* first try to make some room by pushing left and right */
Chris Mason3685f792007-10-19 09:23:27 -04001982 if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
Chris Mason34a38212007-11-07 13:31:03 -05001983 wret = push_leaf_right(trans, root, path, data_size, 0);
Chris Mason3685f792007-10-19 09:23:27 -04001984 if (wret < 0) {
Chris Masoneaee50e2007-03-13 11:17:52 -04001985 return wret;
Chris Mason3685f792007-10-19 09:23:27 -04001986 }
1987 if (wret) {
Chris Mason34a38212007-11-07 13:31:03 -05001988 wret = push_leaf_left(trans, root, path, data_size, 0);
Chris Mason3685f792007-10-19 09:23:27 -04001989 if (wret < 0)
1990 return wret;
1991 }
1992 l = path->nodes[0];
Chris Masonaa5d6be2007-02-28 16:35:06 -05001993
Chris Mason3685f792007-10-19 09:23:27 -04001994 /* did the pushes work? */
Chris Masoncc0c5532007-10-25 15:42:57 -04001995 if (btrfs_leaf_free_space(root, l) >= space_needed)
Chris Mason3685f792007-10-19 09:23:27 -04001996 return 0;
Chris Mason3326d1b2007-10-15 16:18:25 -04001997 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05001998
Chris Mason5c680ed2007-02-22 11:39:13 -05001999 if (!path->nodes[1]) {
Chris Masone089f052007-03-16 16:20:31 -04002000 ret = insert_new_root(trans, root, path, 1);
Chris Mason5c680ed2007-02-22 11:39:13 -05002001 if (ret)
2002 return ret;
2003 }
Chris Masoncc0c5532007-10-25 15:42:57 -04002004again:
2005 double_split = 0;
2006 l = path->nodes[0];
Chris Masoneb60cea2007-02-02 09:18:22 -05002007 slot = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002008 nritems = btrfs_header_nritems(l);
Chris Masoneb60cea2007-02-02 09:18:22 -05002009 mid = (nritems + 1)/ 2;
Chris Mason54aa1f42007-06-22 14:16:25 -04002010
Chris Mason7bb86312007-12-11 09:25:06 -05002011 btrfs_item_key(l, &disk_key, 0);
2012
2013 right = __btrfs_alloc_free_block(trans, root, root->leafsize,
2014 root->root_key.objectid,
2015 root_gen, disk_key.objectid, 0,
2016 l->start, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04002017 if (IS_ERR(right))
2018 return PTR_ERR(right);
Chris Mason54aa1f42007-06-22 14:16:25 -04002019
Chris Mason5f39d392007-10-15 16:14:19 -04002020 memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
Chris Masondb945352007-10-15 16:15:53 -04002021 btrfs_set_header_bytenr(right, right->start);
Chris Mason5f39d392007-10-15 16:14:19 -04002022 btrfs_set_header_generation(right, trans->transid);
2023 btrfs_set_header_owner(right, root->root_key.objectid);
2024 btrfs_set_header_level(right, 0);
2025 write_extent_buffer(right, root->fs_info->fsid,
2026 (unsigned long)btrfs_header_fsid(right),
2027 BTRFS_FSID_SIZE);
Chris Masond4dbff92007-04-04 14:08:15 -04002028 if (mid <= slot) {
2029 if (nritems == 1 ||
2030 leaf_space_used(l, mid, nritems - mid) + space_needed >
2031 BTRFS_LEAF_DATA_SIZE(root)) {
2032 if (slot >= nritems) {
2033 btrfs_cpu_key_to_disk(&disk_key, ins_key);
Chris Mason5f39d392007-10-15 16:14:19 -04002034 btrfs_set_header_nritems(right, 0);
Chris Masond4dbff92007-04-04 14:08:15 -04002035 wret = insert_ptr(trans, root, path,
Chris Masondb945352007-10-15 16:15:53 -04002036 &disk_key, right->start,
Chris Masond4dbff92007-04-04 14:08:15 -04002037 path->slots[1] + 1, 1);
2038 if (wret)
2039 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04002040 free_extent_buffer(path->nodes[0]);
2041 path->nodes[0] = right;
Chris Masond4dbff92007-04-04 14:08:15 -04002042 path->slots[0] = 0;
2043 path->slots[1] += 1;
2044 return ret;
2045 }
2046 mid = slot;
Chris Mason3326d1b2007-10-15 16:18:25 -04002047 if (mid != nritems &&
2048 leaf_space_used(l, mid, nritems - mid) +
2049 space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2050 double_split = 1;
2051 }
Chris Masond4dbff92007-04-04 14:08:15 -04002052 }
2053 } else {
2054 if (leaf_space_used(l, 0, mid + 1) + space_needed >
2055 BTRFS_LEAF_DATA_SIZE(root)) {
Chris Masoncc0c5532007-10-25 15:42:57 -04002056 if (!extend && slot == 0) {
Chris Masond4dbff92007-04-04 14:08:15 -04002057 btrfs_cpu_key_to_disk(&disk_key, ins_key);
Chris Mason5f39d392007-10-15 16:14:19 -04002058 btrfs_set_header_nritems(right, 0);
Chris Masond4dbff92007-04-04 14:08:15 -04002059 wret = insert_ptr(trans, root, path,
2060 &disk_key,
Chris Masondb945352007-10-15 16:15:53 -04002061 right->start,
Chris Mason098f59c2007-05-11 11:33:21 -04002062 path->slots[1], 1);
Chris Masond4dbff92007-04-04 14:08:15 -04002063 if (wret)
2064 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04002065 free_extent_buffer(path->nodes[0]);
2066 path->nodes[0] = right;
Chris Masond4dbff92007-04-04 14:08:15 -04002067 path->slots[0] = 0;
Chris Masona429e512007-04-18 16:15:28 -04002068 if (path->slots[1] == 0) {
2069 wret = fixup_low_keys(trans, root,
2070 path, &disk_key, 1);
2071 if (wret)
2072 ret = wret;
2073 }
Chris Masond4dbff92007-04-04 14:08:15 -04002074 return ret;
Chris Masoncc0c5532007-10-25 15:42:57 -04002075 } else if (extend && slot == 0) {
2076 mid = 1;
2077 } else {
2078 mid = slot;
2079 if (mid != nritems &&
2080 leaf_space_used(l, mid, nritems - mid) +
2081 space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2082 double_split = 1;
2083 }
Chris Mason5ee78ac2007-10-19 14:01:21 -04002084 }
Chris Masond4dbff92007-04-04 14:08:15 -04002085 }
2086 }
Chris Mason5f39d392007-10-15 16:14:19 -04002087 nritems = nritems - mid;
2088 btrfs_set_header_nritems(right, nritems);
2089 data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
2090
2091 copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
2092 btrfs_item_nr_offset(mid),
2093 nritems * sizeof(struct btrfs_item));
2094
2095 copy_extent_buffer(right, l,
Chris Masond6025572007-03-30 14:27:56 -04002096 btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
2097 data_copy_size, btrfs_leaf_data(l) +
2098 leaf_data_end(root, l), data_copy_size);
Chris Mason74123bd2007-02-02 11:05:29 -05002099
Chris Mason5f39d392007-10-15 16:14:19 -04002100 rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
2101 btrfs_item_end_nr(l, mid);
2102
2103 for (i = 0; i < nritems; i++) {
2104 struct btrfs_item *item = btrfs_item_nr(right, i);
Chris Masondb945352007-10-15 16:15:53 -04002105 u32 ioff;
2106
2107 if (!right->map_token) {
2108 map_extent_buffer(right, (unsigned long)item,
2109 sizeof(struct btrfs_item),
2110 &right->map_token, &right->kaddr,
2111 &right->map_start, &right->map_len,
2112 KM_USER1);
2113 }
2114
2115 ioff = btrfs_item_offset(right, item);
Chris Mason5f39d392007-10-15 16:14:19 -04002116 btrfs_set_item_offset(right, item, ioff + rt_data_off);
Chris Mason0783fcf2007-03-12 20:12:07 -04002117 }
Chris Mason74123bd2007-02-02 11:05:29 -05002118
Chris Masondb945352007-10-15 16:15:53 -04002119 if (right->map_token) {
2120 unmap_extent_buffer(right, right->map_token, KM_USER1);
2121 right->map_token = NULL;
2122 }
2123
Chris Mason5f39d392007-10-15 16:14:19 -04002124 btrfs_set_header_nritems(l, mid);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002125 ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04002126 btrfs_item_key(right, &disk_key, 0);
Chris Masondb945352007-10-15 16:15:53 -04002127 wret = insert_ptr(trans, root, path, &disk_key, right->start,
2128 path->slots[1] + 1, 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002129 if (wret)
2130 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04002131
2132 btrfs_mark_buffer_dirty(right);
2133 btrfs_mark_buffer_dirty(l);
Chris Masoneb60cea2007-02-02 09:18:22 -05002134 BUG_ON(path->slots[0] != slot);
Chris Mason5f39d392007-10-15 16:14:19 -04002135
Chris Masonbe0e5c02007-01-26 15:51:26 -05002136 if (mid <= slot) {
Chris Mason5f39d392007-10-15 16:14:19 -04002137 free_extent_buffer(path->nodes[0]);
2138 path->nodes[0] = right;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002139 path->slots[0] -= mid;
2140 path->slots[1] += 1;
Chris Masoneb60cea2007-02-02 09:18:22 -05002141 } else
Chris Mason5f39d392007-10-15 16:14:19 -04002142 free_extent_buffer(right);
2143
Chris Masoneb60cea2007-02-02 09:18:22 -05002144 BUG_ON(path->slots[0] < 0);
Chris Masond4dbff92007-04-04 14:08:15 -04002145
Chris Masoncc0c5532007-10-25 15:42:57 -04002146 if (double_split) {
2147 BUG_ON(num_doubles != 0);
2148 num_doubles++;
2149 goto again;
Chris Mason3326d1b2007-10-15 16:18:25 -04002150 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002151 return ret;
2152}
2153
Chris Masonb18c6682007-04-17 13:26:50 -04002154int btrfs_truncate_item(struct btrfs_trans_handle *trans,
2155 struct btrfs_root *root,
2156 struct btrfs_path *path,
Chris Mason179e29e2007-11-01 11:28:41 -04002157 u32 new_size, int from_end)
Chris Masonb18c6682007-04-17 13:26:50 -04002158{
2159 int ret = 0;
2160 int slot;
2161 int slot_orig;
Chris Mason5f39d392007-10-15 16:14:19 -04002162 struct extent_buffer *leaf;
2163 struct btrfs_item *item;
Chris Masonb18c6682007-04-17 13:26:50 -04002164 u32 nritems;
2165 unsigned int data_end;
2166 unsigned int old_data_start;
2167 unsigned int old_size;
2168 unsigned int size_diff;
2169 int i;
2170
2171 slot_orig = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002172 leaf = path->nodes[0];
Chris Mason179e29e2007-11-01 11:28:41 -04002173 slot = path->slots[0];
2174
2175 old_size = btrfs_item_size_nr(leaf, slot);
2176 if (old_size == new_size)
2177 return 0;
Chris Masonb18c6682007-04-17 13:26:50 -04002178
Chris Mason5f39d392007-10-15 16:14:19 -04002179 nritems = btrfs_header_nritems(leaf);
Chris Masonb18c6682007-04-17 13:26:50 -04002180 data_end = leaf_data_end(root, leaf);
2181
Chris Mason5f39d392007-10-15 16:14:19 -04002182 old_data_start = btrfs_item_offset_nr(leaf, slot);
Chris Mason179e29e2007-11-01 11:28:41 -04002183
Chris Masonb18c6682007-04-17 13:26:50 -04002184 size_diff = old_size - new_size;
2185
2186 BUG_ON(slot < 0);
2187 BUG_ON(slot >= nritems);
2188
2189 /*
2190 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2191 */
2192 /* first correct the data pointers */
2193 for (i = slot; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04002194 u32 ioff;
2195 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04002196
2197 if (!leaf->map_token) {
2198 map_extent_buffer(leaf, (unsigned long)item,
2199 sizeof(struct btrfs_item),
2200 &leaf->map_token, &leaf->kaddr,
2201 &leaf->map_start, &leaf->map_len,
2202 KM_USER1);
2203 }
2204
Chris Mason5f39d392007-10-15 16:14:19 -04002205 ioff = btrfs_item_offset(leaf, item);
2206 btrfs_set_item_offset(leaf, item, ioff + size_diff);
Chris Masonb18c6682007-04-17 13:26:50 -04002207 }
Chris Masondb945352007-10-15 16:15:53 -04002208
2209 if (leaf->map_token) {
2210 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2211 leaf->map_token = NULL;
2212 }
2213
Chris Masonb18c6682007-04-17 13:26:50 -04002214 /* shift the data */
Chris Mason179e29e2007-11-01 11:28:41 -04002215 if (from_end) {
2216 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2217 data_end + size_diff, btrfs_leaf_data(leaf) +
2218 data_end, old_data_start + new_size - data_end);
2219 } else {
2220 struct btrfs_disk_key disk_key;
2221 u64 offset;
2222
2223 btrfs_item_key(leaf, &disk_key, slot);
2224
2225 if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
2226 unsigned long ptr;
2227 struct btrfs_file_extent_item *fi;
2228
2229 fi = btrfs_item_ptr(leaf, slot,
2230 struct btrfs_file_extent_item);
2231 fi = (struct btrfs_file_extent_item *)(
2232 (unsigned long)fi - size_diff);
2233
2234 if (btrfs_file_extent_type(leaf, fi) ==
2235 BTRFS_FILE_EXTENT_INLINE) {
2236 ptr = btrfs_item_ptr_offset(leaf, slot);
2237 memmove_extent_buffer(leaf, ptr,
2238 (unsigned long)fi,
2239 offsetof(struct btrfs_file_extent_item,
2240 disk_bytenr));
2241 }
2242 }
2243
2244 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2245 data_end + size_diff, btrfs_leaf_data(leaf) +
2246 data_end, old_data_start - data_end);
2247
2248 offset = btrfs_disk_key_offset(&disk_key);
2249 btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
2250 btrfs_set_item_key(leaf, &disk_key, slot);
2251 if (slot == 0)
2252 fixup_low_keys(trans, root, path, &disk_key, 1);
2253 }
Chris Mason5f39d392007-10-15 16:14:19 -04002254
2255 item = btrfs_item_nr(leaf, slot);
2256 btrfs_set_item_size(leaf, item, new_size);
2257 btrfs_mark_buffer_dirty(leaf);
Chris Masonb18c6682007-04-17 13:26:50 -04002258
2259 ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04002260 if (btrfs_leaf_free_space(root, leaf) < 0) {
2261 btrfs_print_leaf(root, leaf);
Chris Masonb18c6682007-04-17 13:26:50 -04002262 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002263 }
Chris Masonb18c6682007-04-17 13:26:50 -04002264 return ret;
2265}
2266
Chris Mason5f39d392007-10-15 16:14:19 -04002267int btrfs_extend_item(struct btrfs_trans_handle *trans,
2268 struct btrfs_root *root, struct btrfs_path *path,
2269 u32 data_size)
Chris Mason6567e832007-04-16 09:22:45 -04002270{
2271 int ret = 0;
2272 int slot;
2273 int slot_orig;
Chris Mason5f39d392007-10-15 16:14:19 -04002274 struct extent_buffer *leaf;
2275 struct btrfs_item *item;
Chris Mason6567e832007-04-16 09:22:45 -04002276 u32 nritems;
2277 unsigned int data_end;
2278 unsigned int old_data;
2279 unsigned int old_size;
2280 int i;
2281
2282 slot_orig = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002283 leaf = path->nodes[0];
Chris Mason6567e832007-04-16 09:22:45 -04002284
Chris Mason5f39d392007-10-15 16:14:19 -04002285 nritems = btrfs_header_nritems(leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002286 data_end = leaf_data_end(root, leaf);
2287
Chris Mason5f39d392007-10-15 16:14:19 -04002288 if (btrfs_leaf_free_space(root, leaf) < data_size) {
2289 btrfs_print_leaf(root, leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002290 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002291 }
Chris Mason6567e832007-04-16 09:22:45 -04002292 slot = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002293 old_data = btrfs_item_end_nr(leaf, slot);
Chris Mason6567e832007-04-16 09:22:45 -04002294
2295 BUG_ON(slot < 0);
Chris Mason3326d1b2007-10-15 16:18:25 -04002296 if (slot >= nritems) {
2297 btrfs_print_leaf(root, leaf);
2298 printk("slot %d too large, nritems %d\n", slot, nritems);
2299 BUG_ON(1);
2300 }
Chris Mason6567e832007-04-16 09:22:45 -04002301
2302 /*
2303 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2304 */
2305 /* first correct the data pointers */
2306 for (i = slot; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04002307 u32 ioff;
2308 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04002309
2310 if (!leaf->map_token) {
2311 map_extent_buffer(leaf, (unsigned long)item,
2312 sizeof(struct btrfs_item),
2313 &leaf->map_token, &leaf->kaddr,
2314 &leaf->map_start, &leaf->map_len,
2315 KM_USER1);
2316 }
Chris Mason5f39d392007-10-15 16:14:19 -04002317 ioff = btrfs_item_offset(leaf, item);
2318 btrfs_set_item_offset(leaf, item, ioff - data_size);
Chris Mason6567e832007-04-16 09:22:45 -04002319 }
Chris Mason5f39d392007-10-15 16:14:19 -04002320
Chris Masondb945352007-10-15 16:15:53 -04002321 if (leaf->map_token) {
2322 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2323 leaf->map_token = NULL;
2324 }
2325
Chris Mason6567e832007-04-16 09:22:45 -04002326 /* shift the data */
Chris Mason5f39d392007-10-15 16:14:19 -04002327 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
Chris Mason6567e832007-04-16 09:22:45 -04002328 data_end - data_size, btrfs_leaf_data(leaf) +
2329 data_end, old_data - data_end);
Chris Mason5f39d392007-10-15 16:14:19 -04002330
Chris Mason6567e832007-04-16 09:22:45 -04002331 data_end = old_data;
Chris Mason5f39d392007-10-15 16:14:19 -04002332 old_size = btrfs_item_size_nr(leaf, slot);
2333 item = btrfs_item_nr(leaf, slot);
2334 btrfs_set_item_size(leaf, item, old_size + data_size);
2335 btrfs_mark_buffer_dirty(leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002336
2337 ret = 0;
Chris Mason5f39d392007-10-15 16:14:19 -04002338 if (btrfs_leaf_free_space(root, leaf) < 0) {
2339 btrfs_print_leaf(root, leaf);
Chris Mason6567e832007-04-16 09:22:45 -04002340 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002341 }
Chris Mason6567e832007-04-16 09:22:45 -04002342 return ret;
2343}
2344
Chris Mason74123bd2007-02-02 11:05:29 -05002345/*
2346 * Given a key and some data, insert an item into the tree.
2347 * This does all the path init required, making room in the tree if needed.
2348 */
Chris Mason9c583092008-01-29 15:15:18 -05002349int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
Chris Mason5f39d392007-10-15 16:14:19 -04002350 struct btrfs_root *root,
2351 struct btrfs_path *path,
Chris Mason9c583092008-01-29 15:15:18 -05002352 struct btrfs_key *cpu_key, u32 *data_size,
2353 int nr)
Chris Masonbe0e5c02007-01-26 15:51:26 -05002354{
Chris Mason5f39d392007-10-15 16:14:19 -04002355 struct extent_buffer *leaf;
2356 struct btrfs_item *item;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002357 int ret = 0;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002358 int slot;
Chris Masoneb60cea2007-02-02 09:18:22 -05002359 int slot_orig;
Chris Mason9c583092008-01-29 15:15:18 -05002360 int i;
Chris Mason7518a232007-03-12 12:01:18 -04002361 u32 nritems;
Chris Mason9c583092008-01-29 15:15:18 -05002362 u32 total_size = 0;
2363 u32 total_data = 0;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002364 unsigned int data_end;
Chris Masone2fa7222007-03-12 16:22:34 -04002365 struct btrfs_disk_key disk_key;
2366
Chris Mason9c583092008-01-29 15:15:18 -05002367 for (i = 0; i < nr; i++) {
2368 total_data += data_size[i];
2369 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002370
Chris Mason74123bd2007-02-02 11:05:29 -05002371 /* create a root if there isn't one */
Chris Mason5c680ed2007-02-22 11:39:13 -05002372 if (!root->node)
Chris Masoncfaa7292007-02-21 17:04:57 -05002373 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002374
Chris Mason9c583092008-01-29 15:15:18 -05002375 total_size = total_data + (nr - 1) * sizeof(struct btrfs_item);
2376 ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
Chris Masoneb60cea2007-02-02 09:18:22 -05002377 if (ret == 0) {
Chris Masonf0930a32007-03-02 09:47:58 -05002378 return -EEXIST;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002379 }
Chris Masoned2ff2c2007-03-01 18:59:40 -05002380 if (ret < 0)
2381 goto out;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002382
Chris Mason62e27492007-03-15 12:56:47 -04002383 slot_orig = path->slots[0];
Chris Mason5f39d392007-10-15 16:14:19 -04002384 leaf = path->nodes[0];
Chris Mason74123bd2007-02-02 11:05:29 -05002385
Chris Mason5f39d392007-10-15 16:14:19 -04002386 nritems = btrfs_header_nritems(leaf);
Chris Mason123abc82007-03-14 14:14:43 -04002387 data_end = leaf_data_end(root, leaf);
Chris Masoneb60cea2007-02-02 09:18:22 -05002388
Chris Mason123abc82007-03-14 14:14:43 -04002389 if (btrfs_leaf_free_space(root, leaf) <
Chris Mason9c583092008-01-29 15:15:18 -05002390 sizeof(struct btrfs_item) + total_size) {
Chris Mason3326d1b2007-10-15 16:18:25 -04002391 btrfs_print_leaf(root, leaf);
2392 printk("not enough freespace need %u have %d\n",
Chris Mason9c583092008-01-29 15:15:18 -05002393 total_size, btrfs_leaf_free_space(root, leaf));
Chris Masonbe0e5c02007-01-26 15:51:26 -05002394 BUG();
Chris Masond4dbff92007-04-04 14:08:15 -04002395 }
Chris Mason5f39d392007-10-15 16:14:19 -04002396
Chris Mason62e27492007-03-15 12:56:47 -04002397 slot = path->slots[0];
Chris Masoneb60cea2007-02-02 09:18:22 -05002398 BUG_ON(slot < 0);
Chris Mason5f39d392007-10-15 16:14:19 -04002399
Chris Masonbe0e5c02007-01-26 15:51:26 -05002400 if (slot != nritems) {
2401 int i;
Chris Mason5f39d392007-10-15 16:14:19 -04002402 unsigned int old_data = btrfs_item_end_nr(leaf, slot);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002403
Chris Mason5f39d392007-10-15 16:14:19 -04002404 if (old_data < data_end) {
2405 btrfs_print_leaf(root, leaf);
2406 printk("slot %d old_data %d data_end %d\n",
2407 slot, old_data, data_end);
2408 BUG_ON(1);
2409 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002410 /*
2411 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2412 */
2413 /* first correct the data pointers */
Chris Masondb945352007-10-15 16:15:53 -04002414 WARN_ON(leaf->map_token);
Chris Mason0783fcf2007-03-12 20:12:07 -04002415 for (i = slot; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04002416 u32 ioff;
Chris Masondb945352007-10-15 16:15:53 -04002417
Chris Mason5f39d392007-10-15 16:14:19 -04002418 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04002419 if (!leaf->map_token) {
2420 map_extent_buffer(leaf, (unsigned long)item,
2421 sizeof(struct btrfs_item),
2422 &leaf->map_token, &leaf->kaddr,
2423 &leaf->map_start, &leaf->map_len,
2424 KM_USER1);
2425 }
2426
Chris Mason5f39d392007-10-15 16:14:19 -04002427 ioff = btrfs_item_offset(leaf, item);
Chris Mason9c583092008-01-29 15:15:18 -05002428 btrfs_set_item_offset(leaf, item, ioff - total_data);
Chris Mason0783fcf2007-03-12 20:12:07 -04002429 }
Chris Masondb945352007-10-15 16:15:53 -04002430 if (leaf->map_token) {
2431 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2432 leaf->map_token = NULL;
2433 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002434
2435 /* shift the items */
Chris Mason9c583092008-01-29 15:15:18 -05002436 memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
Chris Mason5f39d392007-10-15 16:14:19 -04002437 btrfs_item_nr_offset(slot),
Chris Masond6025572007-03-30 14:27:56 -04002438 (nritems - slot) * sizeof(struct btrfs_item));
Chris Masonbe0e5c02007-01-26 15:51:26 -05002439
2440 /* shift the data */
Chris Mason5f39d392007-10-15 16:14:19 -04002441 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
Chris Mason9c583092008-01-29 15:15:18 -05002442 data_end - total_data, btrfs_leaf_data(leaf) +
Chris Masond6025572007-03-30 14:27:56 -04002443 data_end, old_data - data_end);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002444 data_end = old_data;
2445 }
Chris Mason5f39d392007-10-15 16:14:19 -04002446
Chris Mason62e27492007-03-15 12:56:47 -04002447 /* setup the item for the new data */
Chris Mason9c583092008-01-29 15:15:18 -05002448 for (i = 0; i < nr; i++) {
2449 btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
2450 btrfs_set_item_key(leaf, &disk_key, slot + i);
2451 item = btrfs_item_nr(leaf, slot + i);
2452 btrfs_set_item_offset(leaf, item, data_end - data_size[i]);
2453 data_end -= data_size[i];
2454 btrfs_set_item_size(leaf, item, data_size[i]);
2455 }
2456 btrfs_set_header_nritems(leaf, nritems + nr);
Chris Mason5f39d392007-10-15 16:14:19 -04002457 btrfs_mark_buffer_dirty(leaf);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002458
2459 ret = 0;
Chris Mason5a01a2e2008-01-30 11:43:54 -05002460 if (slot == 0) {
2461 btrfs_cpu_key_to_disk(&disk_key, cpu_key);
Chris Masone089f052007-03-16 16:20:31 -04002462 ret = fixup_low_keys(trans, root, path, &disk_key, 1);
Chris Mason5a01a2e2008-01-30 11:43:54 -05002463 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05002464
Chris Mason5f39d392007-10-15 16:14:19 -04002465 if (btrfs_leaf_free_space(root, leaf) < 0) {
2466 btrfs_print_leaf(root, leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002467 BUG();
Chris Mason5f39d392007-10-15 16:14:19 -04002468 }
Chris Mason5a01a2e2008-01-30 11:43:54 -05002469
Chris Masoned2ff2c2007-03-01 18:59:40 -05002470out:
Chris Mason62e27492007-03-15 12:56:47 -04002471 return ret;
2472}
2473
2474/*
2475 * Given a key and some data, insert an item into the tree.
2476 * This does all the path init required, making room in the tree if needed.
2477 */
Chris Masone089f052007-03-16 16:20:31 -04002478int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2479 *root, struct btrfs_key *cpu_key, void *data, u32
2480 data_size)
Chris Mason62e27492007-03-15 12:56:47 -04002481{
2482 int ret = 0;
Chris Mason2c90e5d2007-04-02 10:50:19 -04002483 struct btrfs_path *path;
Chris Mason5f39d392007-10-15 16:14:19 -04002484 struct extent_buffer *leaf;
2485 unsigned long ptr;
Chris Mason62e27492007-03-15 12:56:47 -04002486
Chris Mason2c90e5d2007-04-02 10:50:19 -04002487 path = btrfs_alloc_path();
2488 BUG_ON(!path);
Chris Mason2c90e5d2007-04-02 10:50:19 -04002489 ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
Chris Mason62e27492007-03-15 12:56:47 -04002490 if (!ret) {
Chris Mason5f39d392007-10-15 16:14:19 -04002491 leaf = path->nodes[0];
2492 ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
2493 write_extent_buffer(leaf, data, ptr, data_size);
2494 btrfs_mark_buffer_dirty(leaf);
Chris Mason62e27492007-03-15 12:56:47 -04002495 }
Chris Mason2c90e5d2007-04-02 10:50:19 -04002496 btrfs_free_path(path);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002497 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002498}
2499
Chris Mason74123bd2007-02-02 11:05:29 -05002500/*
Chris Mason5de08d72007-02-24 06:24:44 -05002501 * delete the pointer from a given node.
Chris Mason74123bd2007-02-02 11:05:29 -05002502 *
2503 * If the delete empties a node, the node is removed from the tree,
2504 * continuing all the way the root if required. The root is converted into
2505 * a leaf if all the nodes are emptied.
2506 */
Chris Masone089f052007-03-16 16:20:31 -04002507static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2508 struct btrfs_path *path, int level, int slot)
Chris Masonbe0e5c02007-01-26 15:51:26 -05002509{
Chris Mason5f39d392007-10-15 16:14:19 -04002510 struct extent_buffer *parent = path->nodes[level];
Chris Mason7518a232007-03-12 12:01:18 -04002511 u32 nritems;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002512 int ret = 0;
Chris Masonbb803952007-03-01 12:04:21 -05002513 int wret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002514
Chris Mason5f39d392007-10-15 16:14:19 -04002515 nritems = btrfs_header_nritems(parent);
Chris Masonbb803952007-03-01 12:04:21 -05002516 if (slot != nritems -1) {
Chris Mason5f39d392007-10-15 16:14:19 -04002517 memmove_extent_buffer(parent,
2518 btrfs_node_key_ptr_offset(slot),
2519 btrfs_node_key_ptr_offset(slot + 1),
Chris Masond6025572007-03-30 14:27:56 -04002520 sizeof(struct btrfs_key_ptr) *
2521 (nritems - slot - 1));
Chris Masonbb803952007-03-01 12:04:21 -05002522 }
Chris Mason7518a232007-03-12 12:01:18 -04002523 nritems--;
Chris Mason5f39d392007-10-15 16:14:19 -04002524 btrfs_set_header_nritems(parent, nritems);
Chris Mason7518a232007-03-12 12:01:18 -04002525 if (nritems == 0 && parent == root->node) {
Chris Mason5f39d392007-10-15 16:14:19 -04002526 BUG_ON(btrfs_header_level(root->node) != 1);
Chris Masonbb803952007-03-01 12:04:21 -05002527 /* just turn the root into a leaf and break */
Chris Mason5f39d392007-10-15 16:14:19 -04002528 btrfs_set_header_level(root->node, 0);
Chris Masonbb803952007-03-01 12:04:21 -05002529 } else if (slot == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04002530 struct btrfs_disk_key disk_key;
2531
2532 btrfs_node_key(parent, &disk_key, 0);
2533 wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
Chris Mason0f70abe2007-02-28 16:46:22 -05002534 if (wret)
2535 ret = wret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002536 }
Chris Masond6025572007-03-30 14:27:56 -04002537 btrfs_mark_buffer_dirty(parent);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002538 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002539}
2540
Chris Mason74123bd2007-02-02 11:05:29 -05002541/*
2542 * delete the item at the leaf level in path. If that empties
2543 * the leaf, remove it from the tree
2544 */
Chris Mason85e21ba2008-01-29 15:11:36 -05002545int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2546 struct btrfs_path *path, int slot, int nr)
Chris Masonbe0e5c02007-01-26 15:51:26 -05002547{
Chris Mason5f39d392007-10-15 16:14:19 -04002548 struct extent_buffer *leaf;
2549 struct btrfs_item *item;
Chris Mason85e21ba2008-01-29 15:11:36 -05002550 int last_off;
2551 int dsize = 0;
Chris Masonaa5d6be2007-02-28 16:35:06 -05002552 int ret = 0;
2553 int wret;
Chris Mason85e21ba2008-01-29 15:11:36 -05002554 int i;
Chris Mason7518a232007-03-12 12:01:18 -04002555 u32 nritems;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002556
Chris Mason5f39d392007-10-15 16:14:19 -04002557 leaf = path->nodes[0];
Chris Mason85e21ba2008-01-29 15:11:36 -05002558 last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);
2559
2560 for (i = 0; i < nr; i++)
2561 dsize += btrfs_item_size_nr(leaf, slot + i);
2562
Chris Mason5f39d392007-10-15 16:14:19 -04002563 nritems = btrfs_header_nritems(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002564
Chris Mason85e21ba2008-01-29 15:11:36 -05002565 if (slot + nr != nritems) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05002566 int i;
Chris Mason123abc82007-03-14 14:14:43 -04002567 int data_end = leaf_data_end(root, leaf);
Chris Mason5f39d392007-10-15 16:14:19 -04002568
2569 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
Chris Masond6025572007-03-30 14:27:56 -04002570 data_end + dsize,
2571 btrfs_leaf_data(leaf) + data_end,
Chris Mason85e21ba2008-01-29 15:11:36 -05002572 last_off - data_end);
Chris Mason5f39d392007-10-15 16:14:19 -04002573
Chris Mason85e21ba2008-01-29 15:11:36 -05002574 for (i = slot + nr; i < nritems; i++) {
Chris Mason5f39d392007-10-15 16:14:19 -04002575 u32 ioff;
Chris Masondb945352007-10-15 16:15:53 -04002576
Chris Mason5f39d392007-10-15 16:14:19 -04002577 item = btrfs_item_nr(leaf, i);
Chris Masondb945352007-10-15 16:15:53 -04002578 if (!leaf->map_token) {
2579 map_extent_buffer(leaf, (unsigned long)item,
2580 sizeof(struct btrfs_item),
2581 &leaf->map_token, &leaf->kaddr,
2582 &leaf->map_start, &leaf->map_len,
2583 KM_USER1);
2584 }
Chris Mason5f39d392007-10-15 16:14:19 -04002585 ioff = btrfs_item_offset(leaf, item);
2586 btrfs_set_item_offset(leaf, item, ioff + dsize);
Chris Mason0783fcf2007-03-12 20:12:07 -04002587 }
Chris Masondb945352007-10-15 16:15:53 -04002588
2589 if (leaf->map_token) {
2590 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2591 leaf->map_token = NULL;
2592 }
2593
Chris Mason5f39d392007-10-15 16:14:19 -04002594 memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
Chris Mason85e21ba2008-01-29 15:11:36 -05002595 btrfs_item_nr_offset(slot + nr),
Chris Masond6025572007-03-30 14:27:56 -04002596 sizeof(struct btrfs_item) *
Chris Mason85e21ba2008-01-29 15:11:36 -05002597 (nritems - slot - nr));
Chris Masonbe0e5c02007-01-26 15:51:26 -05002598 }
Chris Mason85e21ba2008-01-29 15:11:36 -05002599 btrfs_set_header_nritems(leaf, nritems - nr);
2600 nritems -= nr;
Chris Mason5f39d392007-10-15 16:14:19 -04002601
Chris Mason74123bd2007-02-02 11:05:29 -05002602 /* delete the leaf if we've emptied it */
Chris Mason7518a232007-03-12 12:01:18 -04002603 if (nritems == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04002604 if (leaf == root->node) {
2605 btrfs_set_header_level(leaf, 0);
Chris Mason9a8dd152007-02-23 08:38:36 -05002606 } else {
Chris Mason7bb86312007-12-11 09:25:06 -05002607 u64 root_gen = btrfs_header_generation(path->nodes[1]);
Chris Mason5f39d392007-10-15 16:14:19 -04002608 clean_tree_block(trans, root, leaf);
2609 wait_on_tree_block_writeback(root, leaf);
Chris Masone089f052007-03-16 16:20:31 -04002610 wret = del_ptr(trans, root, path, 1, path->slots[1]);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002611 if (wret)
2612 ret = wret;
Chris Masone089f052007-03-16 16:20:31 -04002613 wret = btrfs_free_extent(trans, root,
Chris Mason7bb86312007-12-11 09:25:06 -05002614 leaf->start, leaf->len,
2615 btrfs_header_owner(path->nodes[1]),
2616 root_gen, 0, 0, 1);
Chris Mason0f70abe2007-02-28 16:46:22 -05002617 if (wret)
2618 ret = wret;
Chris Mason9a8dd152007-02-23 08:38:36 -05002619 }
Chris Masonbe0e5c02007-01-26 15:51:26 -05002620 } else {
Chris Mason7518a232007-03-12 12:01:18 -04002621 int used = leaf_space_used(leaf, 0, nritems);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002622 if (slot == 0) {
Chris Mason5f39d392007-10-15 16:14:19 -04002623 struct btrfs_disk_key disk_key;
2624
2625 btrfs_item_key(leaf, &disk_key, 0);
Chris Masone089f052007-03-16 16:20:31 -04002626 wret = fixup_low_keys(trans, root, path,
Chris Mason5f39d392007-10-15 16:14:19 -04002627 &disk_key, 1);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002628 if (wret)
2629 ret = wret;
2630 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05002631
Chris Mason74123bd2007-02-02 11:05:29 -05002632 /* delete the leaf if it is mostly empty */
Chris Mason85e21ba2008-01-29 15:11:36 -05002633 if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) {
Chris Masonbe0e5c02007-01-26 15:51:26 -05002634 /* push_leaf_left fixes the path.
2635 * make sure the path still points to our leaf
2636 * for possible call to del_ptr below
2637 */
Chris Mason4920c9a2007-01-26 16:38:42 -05002638 slot = path->slots[1];
Chris Mason5f39d392007-10-15 16:14:19 -04002639 extent_buffer_get(leaf);
2640
Chris Mason85e21ba2008-01-29 15:11:36 -05002641 wret = push_leaf_left(trans, root, path, 1, 1);
Chris Mason54aa1f42007-06-22 14:16:25 -04002642 if (wret < 0 && wret != -ENOSPC)
Chris Masonaa5d6be2007-02-28 16:35:06 -05002643 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04002644
2645 if (path->nodes[0] == leaf &&
2646 btrfs_header_nritems(leaf)) {
Chris Mason85e21ba2008-01-29 15:11:36 -05002647 wret = push_leaf_right(trans, root, path, 1, 1);
Chris Mason54aa1f42007-06-22 14:16:25 -04002648 if (wret < 0 && wret != -ENOSPC)
Chris Masonaa5d6be2007-02-28 16:35:06 -05002649 ret = wret;
2650 }
Chris Mason5f39d392007-10-15 16:14:19 -04002651
2652 if (btrfs_header_nritems(leaf) == 0) {
Chris Mason7bb86312007-12-11 09:25:06 -05002653 u64 root_gen;
Chris Masondb945352007-10-15 16:15:53 -04002654 u64 bytenr = leaf->start;
2655 u32 blocksize = leaf->len;
Chris Mason5f39d392007-10-15 16:14:19 -04002656
Chris Mason7bb86312007-12-11 09:25:06 -05002657 root_gen = btrfs_header_generation(
2658 path->nodes[1]);
2659
Chris Mason5f39d392007-10-15 16:14:19 -04002660 clean_tree_block(trans, root, leaf);
2661 wait_on_tree_block_writeback(root, leaf);
2662
Chris Masone089f052007-03-16 16:20:31 -04002663 wret = del_ptr(trans, root, path, 1, slot);
Chris Masonaa5d6be2007-02-28 16:35:06 -05002664 if (wret)
2665 ret = wret;
Chris Mason5f39d392007-10-15 16:14:19 -04002666
2667 free_extent_buffer(leaf);
Chris Masondb945352007-10-15 16:15:53 -04002668 wret = btrfs_free_extent(trans, root, bytenr,
Chris Mason7bb86312007-12-11 09:25:06 -05002669 blocksize,
2670 btrfs_header_owner(path->nodes[1]),
2671 root_gen, 0, 0, 1);
Chris Mason0f70abe2007-02-28 16:46:22 -05002672 if (wret)
2673 ret = wret;
Chris Mason5de08d72007-02-24 06:24:44 -05002674 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04002675 btrfs_mark_buffer_dirty(leaf);
2676 free_extent_buffer(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002677 }
Chris Masond5719762007-03-23 10:01:08 -04002678 } else {
Chris Mason5f39d392007-10-15 16:14:19 -04002679 btrfs_mark_buffer_dirty(leaf);
Chris Masonbe0e5c02007-01-26 15:51:26 -05002680 }
2681 }
Chris Masonaa5d6be2007-02-28 16:35:06 -05002682 return ret;
Chris Masonbe0e5c02007-01-26 15:51:26 -05002683}
2684
Chris Mason97571fd2007-02-24 13:39:08 -05002685/*
Chris Mason7bb86312007-12-11 09:25:06 -05002686 * walk up the tree as far as required to find the previous leaf.
2687 * returns 0 if it found something or 1 if there are no lesser leaves.
2688 * returns < 0 on io errors.
2689 */
2690int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
2691{
Chris Mason8f662a72008-01-02 10:01:11 -05002692 u64 bytenr;
Chris Mason7bb86312007-12-11 09:25:06 -05002693 int slot;
2694 int level = 1;
Chris Mason7bb86312007-12-11 09:25:06 -05002695 struct extent_buffer *c;
2696 struct extent_buffer *next = NULL;
2697
2698 while(level < BTRFS_MAX_LEVEL) {
2699 if (!path->nodes[level])
2700 return 1;
2701
2702 slot = path->slots[level];
2703 c = path->nodes[level];
2704 if (slot == 0) {
2705 level++;
2706 if (level == BTRFS_MAX_LEVEL)
2707 return 1;
2708 continue;
2709 }
2710 slot--;
2711
2712 bytenr = btrfs_node_blockptr(c, slot);
2713 if (next)
2714 free_extent_buffer(next);
2715
Chris Mason7bb86312007-12-11 09:25:06 -05002716 next = read_tree_block(root, bytenr,
2717 btrfs_level_size(root, level - 1));
2718 break;
2719 }
2720 path->slots[level] = slot;
2721 while(1) {
2722 level--;
2723 c = path->nodes[level];
2724 free_extent_buffer(c);
Chris Mason8f662a72008-01-02 10:01:11 -05002725 slot = btrfs_header_nritems(next);
2726 if (slot != 0)
2727 slot--;
Chris Mason7bb86312007-12-11 09:25:06 -05002728 path->nodes[level] = next;
Chris Mason8f662a72008-01-02 10:01:11 -05002729 path->slots[level] = slot;
Chris Mason7bb86312007-12-11 09:25:06 -05002730 if (!level)
2731 break;
Chris Mason8f662a72008-01-02 10:01:11 -05002732 next = read_tree_block(root, btrfs_node_blockptr(next, slot),
Chris Mason7bb86312007-12-11 09:25:06 -05002733 btrfs_level_size(root, level - 1));
2734 }
2735 return 0;
2736}
2737
2738/*
Chris Mason97571fd2007-02-24 13:39:08 -05002739 * walk up the tree as far as required to find the next leaf.
Chris Mason0f70abe2007-02-28 16:46:22 -05002740 * returns 0 if it found something or 1 if there are no greater leaves.
2741 * returns < 0 on io errors.
Chris Mason97571fd2007-02-24 13:39:08 -05002742 */
Chris Mason234b63a2007-03-13 10:46:10 -04002743int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
Chris Masond97e63b2007-02-20 16:40:44 -05002744{
2745 int slot;
2746 int level = 1;
Chris Masondb945352007-10-15 16:15:53 -04002747 u64 bytenr;
Chris Mason5f39d392007-10-15 16:14:19 -04002748 struct extent_buffer *c;
2749 struct extent_buffer *next = NULL;
Chris Masond97e63b2007-02-20 16:40:44 -05002750
Chris Mason234b63a2007-03-13 10:46:10 -04002751 while(level < BTRFS_MAX_LEVEL) {
Chris Masond97e63b2007-02-20 16:40:44 -05002752 if (!path->nodes[level])
Chris Mason0f70abe2007-02-28 16:46:22 -05002753 return 1;
Chris Mason5f39d392007-10-15 16:14:19 -04002754
Chris Masond97e63b2007-02-20 16:40:44 -05002755 slot = path->slots[level] + 1;
2756 c = path->nodes[level];
Chris Mason5f39d392007-10-15 16:14:19 -04002757 if (slot >= btrfs_header_nritems(c)) {
Chris Masond97e63b2007-02-20 16:40:44 -05002758 level++;
Chris Mason7bb86312007-12-11 09:25:06 -05002759 if (level == BTRFS_MAX_LEVEL)
2760 return 1;
Chris Masond97e63b2007-02-20 16:40:44 -05002761 continue;
2762 }
Chris Mason5f39d392007-10-15 16:14:19 -04002763
Chris Masondb945352007-10-15 16:15:53 -04002764 bytenr = btrfs_node_blockptr(c, slot);
Chris Masoncfaa7292007-02-21 17:04:57 -05002765 if (next)
Chris Mason5f39d392007-10-15 16:14:19 -04002766 free_extent_buffer(next);
2767
Chris Mason6702ed42007-08-07 16:15:09 -04002768 if (path->reada)
Chris Mason01f46652007-12-21 16:24:26 -05002769 reada_for_search(root, path, level, slot, 0);
Chris Mason5f39d392007-10-15 16:14:19 -04002770
Chris Masondb945352007-10-15 16:15:53 -04002771 next = read_tree_block(root, bytenr,
2772 btrfs_level_size(root, level -1));
Chris Masond97e63b2007-02-20 16:40:44 -05002773 break;
2774 }
2775 path->slots[level] = slot;
2776 while(1) {
2777 level--;
2778 c = path->nodes[level];
Chris Mason5f39d392007-10-15 16:14:19 -04002779 free_extent_buffer(c);
Chris Masond97e63b2007-02-20 16:40:44 -05002780 path->nodes[level] = next;
2781 path->slots[level] = 0;
2782 if (!level)
2783 break;
Chris Mason6702ed42007-08-07 16:15:09 -04002784 if (path->reada)
Chris Mason01f46652007-12-21 16:24:26 -05002785 reada_for_search(root, path, level, 0, 0);
Chris Masondb945352007-10-15 16:15:53 -04002786 next = read_tree_block(root, btrfs_node_blockptr(next, 0),
2787 btrfs_level_size(root, level - 1));
Chris Masond97e63b2007-02-20 16:40:44 -05002788 }
2789 return 0;
2790}
Chris Mason0b86a832008-03-24 15:01:56 -04002791
2792int btrfs_previous_item(struct btrfs_root *root,
2793 struct btrfs_path *path, u64 min_objectid,
2794 int type)
2795{
2796 struct btrfs_key found_key;
2797 struct extent_buffer *leaf;
2798 int ret;
2799
2800 while(1) {
2801 if (path->slots[0] == 0) {
2802 ret = btrfs_prev_leaf(root, path);
2803 if (ret != 0)
2804 return ret;
2805 } else {
2806 path->slots[0]--;
2807 }
2808 leaf = path->nodes[0];
2809 btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
2810 if (found_key.type == type)
2811 return 0;
2812 }
2813 return 1;
2814}
2815