blob: 0543bf9f80baae1296c201403957da86e8d1d490 [file] [log] [blame]
Koji Sato17c76b02009-04-06 19:01:24 -07001/*
2 * btree.c - NILFS B-tree.
3 *
4 * Copyright (C) 2005-2008 Nippon Telegraph and Telephone Corporation.
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19 *
20 * Written by Koji Sato <koji@osrg.net>.
21 */
22
23#include <linux/slab.h>
24#include <linux/string.h>
25#include <linux/errno.h>
26#include <linux/pagevec.h>
27#include "nilfs.h"
28#include "page.h"
29#include "btnode.h"
30#include "btree.h"
31#include "alloc.h"
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +090032#include "dat.h"
Koji Sato17c76b02009-04-06 19:01:24 -070033
Li Hongf9054402010-04-02 17:36:34 +080034static struct nilfs_btree_path *nilfs_btree_alloc_path(void)
Koji Sato17c76b02009-04-06 19:01:24 -070035{
Li Hongf9054402010-04-02 17:36:34 +080036 struct nilfs_btree_path *path;
37 int level = NILFS_BTREE_LEVEL_DATA;
Koji Sato17c76b02009-04-06 19:01:24 -070038
Li Hongf9054402010-04-02 17:36:34 +080039 path = kmem_cache_alloc(nilfs_btree_path_cache, GFP_NOFS);
40 if (path == NULL)
41 goto out;
Koji Sato17c76b02009-04-06 19:01:24 -070042
Li Hongf9054402010-04-02 17:36:34 +080043 for (; level < NILFS_BTREE_LEVEL_MAX; level++) {
Koji Sato17c76b02009-04-06 19:01:24 -070044 path[level].bp_bh = NULL;
45 path[level].bp_sib_bh = NULL;
46 path[level].bp_index = 0;
47 path[level].bp_oldreq.bpr_ptr = NILFS_BMAP_INVALID_PTR;
48 path[level].bp_newreq.bpr_ptr = NILFS_BMAP_INVALID_PTR;
49 path[level].bp_op = NULL;
50 }
Li Hongf9054402010-04-02 17:36:34 +080051
52out:
53 return path;
54}
55
Li Hong73bb4882010-04-02 18:35:00 +080056static void nilfs_btree_free_path(struct nilfs_btree_path *path)
Li Hongf9054402010-04-02 17:36:34 +080057{
Li Hong73bb4882010-04-02 18:35:00 +080058 int level = NILFS_BTREE_LEVEL_DATA;
Koji Sato17c76b02009-04-06 19:01:24 -070059
Li Hong73bb4882010-04-02 18:35:00 +080060 for (; level < NILFS_BTREE_LEVEL_MAX; level++)
Ryusuke Konishi32189292009-08-15 01:54:59 +090061 brelse(path[level].bp_bh);
Li Hong73bb4882010-04-02 18:35:00 +080062
63 kmem_cache_free(nilfs_btree_path_cache, path);
Koji Sato17c76b02009-04-06 19:01:24 -070064}
65
Koji Sato17c76b02009-04-06 19:01:24 -070066/*
67 * B-tree node operations
68 */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +090069static int nilfs_btree_get_block(const struct nilfs_bmap *btree, __u64 ptr,
Ryusuke Konishif198dbb2009-05-22 01:07:13 +090070 struct buffer_head **bhp)
71{
Ryusuke Konishie7c274f2010-07-10 19:09:49 +090072 struct address_space *btnc = &NILFS_BMAP_I(btree)->i_btnode_cache;
Ryusuke Konishi1d5385b2010-07-16 23:52:40 +090073 struct buffer_head *bh;
Ryusuke Konishi1376e932009-11-13 16:49:09 +090074 int err;
75
76 err = nilfs_btnode_submit_block(btnc, ptr, 0, bhp);
77 if (err)
78 return err == -EEXIST ? 0 : err;
79
Ryusuke Konishi1d5385b2010-07-16 23:52:40 +090080 bh = *bhp;
81 wait_on_buffer(bh);
82 if (!buffer_uptodate(bh)) {
83 brelse(bh);
Ryusuke Konishi1376e932009-11-13 16:49:09 +090084 return -EIO;
85 }
Ryusuke Konishi1d5385b2010-07-16 23:52:40 +090086 if (nilfs_btree_broken_node_block(bh)) {
87 clear_buffer_uptodate(bh);
88 brelse(bh);
89 return -EINVAL;
90 }
Ryusuke Konishi1376e932009-11-13 16:49:09 +090091 return 0;
Ryusuke Konishif198dbb2009-05-22 01:07:13 +090092}
93
Ryusuke Konishie7c274f2010-07-10 19:09:49 +090094static int nilfs_btree_get_new_block(const struct nilfs_bmap *btree,
Ryusuke Konishif198dbb2009-05-22 01:07:13 +090095 __u64 ptr, struct buffer_head **bhp)
96{
Ryusuke Konishie7c274f2010-07-10 19:09:49 +090097 struct address_space *btnc = &NILFS_BMAP_I(btree)->i_btnode_cache;
Ryusuke Konishi45f49102009-11-13 16:25:19 +090098 struct buffer_head *bh;
Ryusuke Konishif198dbb2009-05-22 01:07:13 +090099
Ryusuke Konishi45f49102009-11-13 16:25:19 +0900100 bh = nilfs_btnode_create_block(btnc, ptr);
101 if (!bh)
102 return -ENOMEM;
103
104 set_buffer_nilfs_volatile(bh);
105 *bhp = bh;
106 return 0;
Ryusuke Konishif198dbb2009-05-22 01:07:13 +0900107}
Koji Sato17c76b02009-04-06 19:01:24 -0700108
109static inline int
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900110nilfs_btree_node_get_flags(const struct nilfs_btree_node *node)
Koji Sato17c76b02009-04-06 19:01:24 -0700111{
112 return node->bn_flags;
113}
114
115static inline void
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900116nilfs_btree_node_set_flags(struct nilfs_btree_node *node, int flags)
Koji Sato17c76b02009-04-06 19:01:24 -0700117{
118 node->bn_flags = flags;
119}
120
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900121static inline int nilfs_btree_node_root(const struct nilfs_btree_node *node)
Koji Sato17c76b02009-04-06 19:01:24 -0700122{
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900123 return nilfs_btree_node_get_flags(node) & NILFS_BTREE_NODE_ROOT;
Koji Sato17c76b02009-04-06 19:01:24 -0700124}
125
126static inline int
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900127nilfs_btree_node_get_level(const struct nilfs_btree_node *node)
Koji Sato17c76b02009-04-06 19:01:24 -0700128{
129 return node->bn_level;
130}
131
132static inline void
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900133nilfs_btree_node_set_level(struct nilfs_btree_node *node, int level)
Koji Sato17c76b02009-04-06 19:01:24 -0700134{
135 node->bn_level = level;
136}
137
138static inline int
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900139nilfs_btree_node_get_nchildren(const struct nilfs_btree_node *node)
Koji Sato17c76b02009-04-06 19:01:24 -0700140{
141 return le16_to_cpu(node->bn_nchildren);
142}
143
144static inline void
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900145nilfs_btree_node_set_nchildren(struct nilfs_btree_node *node, int nchildren)
Koji Sato17c76b02009-04-06 19:01:24 -0700146{
147 node->bn_nchildren = cpu_to_le16(nchildren);
148}
149
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900150static inline int nilfs_btree_node_size(const struct nilfs_bmap *btree)
Koji Sato17c76b02009-04-06 19:01:24 -0700151{
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900152 return 1 << btree->b_inode->i_blkbits;
Koji Sato17c76b02009-04-06 19:01:24 -0700153}
154
155static inline int
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900156nilfs_btree_node_nchildren_min(const struct nilfs_btree_node *node,
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900157 const struct nilfs_bmap *btree)
Koji Sato17c76b02009-04-06 19:01:24 -0700158{
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900159 return nilfs_btree_node_root(node) ?
Koji Sato17c76b02009-04-06 19:01:24 -0700160 NILFS_BTREE_ROOT_NCHILDREN_MIN :
161 NILFS_BTREE_NODE_NCHILDREN_MIN(nilfs_btree_node_size(btree));
162}
163
164static inline int
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900165nilfs_btree_node_nchildren_max(const struct nilfs_btree_node *node,
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900166 const struct nilfs_bmap *btree)
Koji Sato17c76b02009-04-06 19:01:24 -0700167{
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900168 return nilfs_btree_node_root(node) ?
Koji Sato17c76b02009-04-06 19:01:24 -0700169 NILFS_BTREE_ROOT_NCHILDREN_MAX :
170 NILFS_BTREE_NODE_NCHILDREN_MAX(nilfs_btree_node_size(btree));
171}
172
173static inline __le64 *
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900174nilfs_btree_node_dkeys(const struct nilfs_btree_node *node)
Koji Sato17c76b02009-04-06 19:01:24 -0700175{
176 return (__le64 *)((char *)(node + 1) +
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900177 (nilfs_btree_node_root(node) ?
Koji Sato17c76b02009-04-06 19:01:24 -0700178 0 : NILFS_BTREE_NODE_EXTRA_PAD_SIZE));
179}
180
181static inline __le64 *
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900182nilfs_btree_node_dptrs(const struct nilfs_btree_node *node,
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900183 const struct nilfs_bmap *btree)
Koji Sato17c76b02009-04-06 19:01:24 -0700184{
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900185 return (__le64 *)(nilfs_btree_node_dkeys(node) +
186 nilfs_btree_node_nchildren_max(node, btree));
Koji Sato17c76b02009-04-06 19:01:24 -0700187}
188
189static inline __u64
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900190nilfs_btree_node_get_key(const struct nilfs_btree_node *node, int index)
Koji Sato17c76b02009-04-06 19:01:24 -0700191{
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +0900192 return le64_to_cpu(*(nilfs_btree_node_dkeys(node) + index));
Koji Sato17c76b02009-04-06 19:01:24 -0700193}
194
195static inline void
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900196nilfs_btree_node_set_key(struct nilfs_btree_node *node, int index, __u64 key)
Koji Sato17c76b02009-04-06 19:01:24 -0700197{
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +0900198 *(nilfs_btree_node_dkeys(node) + index) = cpu_to_le64(key);
Koji Sato17c76b02009-04-06 19:01:24 -0700199}
200
201static inline __u64
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900202nilfs_btree_node_get_ptr(const struct nilfs_bmap *btree,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900203 const struct nilfs_btree_node *node, int index)
Koji Sato17c76b02009-04-06 19:01:24 -0700204{
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +0900205 return le64_to_cpu(*(nilfs_btree_node_dptrs(node, btree) + index));
Koji Sato17c76b02009-04-06 19:01:24 -0700206}
207
208static inline void
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900209nilfs_btree_node_set_ptr(struct nilfs_bmap *btree,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900210 struct nilfs_btree_node *node, int index, __u64 ptr)
Koji Sato17c76b02009-04-06 19:01:24 -0700211{
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +0900212 *(nilfs_btree_node_dptrs(node, btree) + index) = cpu_to_le64(ptr);
Koji Sato17c76b02009-04-06 19:01:24 -0700213}
214
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900215static void nilfs_btree_node_init(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700216 struct nilfs_btree_node *node,
217 int flags, int level, int nchildren,
218 const __u64 *keys, const __u64 *ptrs)
219{
220 __le64 *dkeys;
221 __le64 *dptrs;
222 int i;
223
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900224 nilfs_btree_node_set_flags(node, flags);
225 nilfs_btree_node_set_level(node, level);
226 nilfs_btree_node_set_nchildren(node, nchildren);
Koji Sato17c76b02009-04-06 19:01:24 -0700227
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900228 dkeys = nilfs_btree_node_dkeys(node);
229 dptrs = nilfs_btree_node_dptrs(node, btree);
Koji Sato17c76b02009-04-06 19:01:24 -0700230 for (i = 0; i < nchildren; i++) {
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +0900231 dkeys[i] = cpu_to_le64(keys[i]);
232 dptrs[i] = cpu_to_le64(ptrs[i]);
Koji Sato17c76b02009-04-06 19:01:24 -0700233 }
234}
235
236/* Assume the buffer heads corresponding to left and right are locked. */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900237static void nilfs_btree_node_move_left(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700238 struct nilfs_btree_node *left,
239 struct nilfs_btree_node *right,
240 int n)
241{
242 __le64 *ldkeys, *rdkeys;
243 __le64 *ldptrs, *rdptrs;
244 int lnchildren, rnchildren;
245
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900246 ldkeys = nilfs_btree_node_dkeys(left);
247 ldptrs = nilfs_btree_node_dptrs(left, btree);
248 lnchildren = nilfs_btree_node_get_nchildren(left);
Koji Sato17c76b02009-04-06 19:01:24 -0700249
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900250 rdkeys = nilfs_btree_node_dkeys(right);
251 rdptrs = nilfs_btree_node_dptrs(right, btree);
252 rnchildren = nilfs_btree_node_get_nchildren(right);
Koji Sato17c76b02009-04-06 19:01:24 -0700253
254 memcpy(ldkeys + lnchildren, rdkeys, n * sizeof(*rdkeys));
255 memcpy(ldptrs + lnchildren, rdptrs, n * sizeof(*rdptrs));
256 memmove(rdkeys, rdkeys + n, (rnchildren - n) * sizeof(*rdkeys));
257 memmove(rdptrs, rdptrs + n, (rnchildren - n) * sizeof(*rdptrs));
258
259 lnchildren += n;
260 rnchildren -= n;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900261 nilfs_btree_node_set_nchildren(left, lnchildren);
262 nilfs_btree_node_set_nchildren(right, rnchildren);
Koji Sato17c76b02009-04-06 19:01:24 -0700263}
264
265/* Assume that the buffer heads corresponding to left and right are locked. */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900266static void nilfs_btree_node_move_right(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700267 struct nilfs_btree_node *left,
268 struct nilfs_btree_node *right,
269 int n)
270{
271 __le64 *ldkeys, *rdkeys;
272 __le64 *ldptrs, *rdptrs;
273 int lnchildren, rnchildren;
274
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900275 ldkeys = nilfs_btree_node_dkeys(left);
276 ldptrs = nilfs_btree_node_dptrs(left, btree);
277 lnchildren = nilfs_btree_node_get_nchildren(left);
Koji Sato17c76b02009-04-06 19:01:24 -0700278
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900279 rdkeys = nilfs_btree_node_dkeys(right);
280 rdptrs = nilfs_btree_node_dptrs(right, btree);
281 rnchildren = nilfs_btree_node_get_nchildren(right);
Koji Sato17c76b02009-04-06 19:01:24 -0700282
283 memmove(rdkeys + n, rdkeys, rnchildren * sizeof(*rdkeys));
284 memmove(rdptrs + n, rdptrs, rnchildren * sizeof(*rdptrs));
285 memcpy(rdkeys, ldkeys + lnchildren - n, n * sizeof(*rdkeys));
286 memcpy(rdptrs, ldptrs + lnchildren - n, n * sizeof(*rdptrs));
287
288 lnchildren -= n;
289 rnchildren += n;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900290 nilfs_btree_node_set_nchildren(left, lnchildren);
291 nilfs_btree_node_set_nchildren(right, rnchildren);
Koji Sato17c76b02009-04-06 19:01:24 -0700292}
293
294/* Assume that the buffer head corresponding to node is locked. */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900295static void nilfs_btree_node_insert(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700296 struct nilfs_btree_node *node,
297 __u64 key, __u64 ptr, int index)
298{
299 __le64 *dkeys;
300 __le64 *dptrs;
301 int nchildren;
302
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900303 dkeys = nilfs_btree_node_dkeys(node);
304 dptrs = nilfs_btree_node_dptrs(node, btree);
305 nchildren = nilfs_btree_node_get_nchildren(node);
Koji Sato17c76b02009-04-06 19:01:24 -0700306 if (index < nchildren) {
307 memmove(dkeys + index + 1, dkeys + index,
308 (nchildren - index) * sizeof(*dkeys));
309 memmove(dptrs + index + 1, dptrs + index,
310 (nchildren - index) * sizeof(*dptrs));
311 }
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +0900312 dkeys[index] = cpu_to_le64(key);
313 dptrs[index] = cpu_to_le64(ptr);
Koji Sato17c76b02009-04-06 19:01:24 -0700314 nchildren++;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900315 nilfs_btree_node_set_nchildren(node, nchildren);
Koji Sato17c76b02009-04-06 19:01:24 -0700316}
317
318/* Assume that the buffer head corresponding to node is locked. */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900319static void nilfs_btree_node_delete(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700320 struct nilfs_btree_node *node,
321 __u64 *keyp, __u64 *ptrp, int index)
322{
323 __u64 key;
324 __u64 ptr;
325 __le64 *dkeys;
326 __le64 *dptrs;
327 int nchildren;
328
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900329 dkeys = nilfs_btree_node_dkeys(node);
330 dptrs = nilfs_btree_node_dptrs(node, btree);
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +0900331 key = le64_to_cpu(dkeys[index]);
332 ptr = le64_to_cpu(dptrs[index]);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900333 nchildren = nilfs_btree_node_get_nchildren(node);
Koji Sato17c76b02009-04-06 19:01:24 -0700334 if (keyp != NULL)
335 *keyp = key;
336 if (ptrp != NULL)
337 *ptrp = ptr;
338
339 if (index < nchildren - 1) {
340 memmove(dkeys + index, dkeys + index + 1,
341 (nchildren - index - 1) * sizeof(*dkeys));
342 memmove(dptrs + index, dptrs + index + 1,
343 (nchildren - index - 1) * sizeof(*dptrs));
344 }
345 nchildren--;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900346 nilfs_btree_node_set_nchildren(node, nchildren);
Koji Sato17c76b02009-04-06 19:01:24 -0700347}
348
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900349static int nilfs_btree_node_lookup(const struct nilfs_btree_node *node,
Koji Sato17c76b02009-04-06 19:01:24 -0700350 __u64 key, int *indexp)
351{
352 __u64 nkey;
353 int index, low, high, s;
354
355 /* binary search */
356 low = 0;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900357 high = nilfs_btree_node_get_nchildren(node) - 1;
Koji Sato17c76b02009-04-06 19:01:24 -0700358 index = 0;
359 s = 0;
360 while (low <= high) {
361 index = (low + high) / 2;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900362 nkey = nilfs_btree_node_get_key(node, index);
Koji Sato17c76b02009-04-06 19:01:24 -0700363 if (nkey == key) {
364 s = 0;
365 goto out;
366 } else if (nkey < key) {
367 low = index + 1;
368 s = -1;
369 } else {
370 high = index - 1;
371 s = 1;
372 }
373 }
374
375 /* adjust index */
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900376 if (nilfs_btree_node_get_level(node) > NILFS_BTREE_LEVEL_NODE_MIN) {
377 if (s > 0 && index > 0)
Koji Sato17c76b02009-04-06 19:01:24 -0700378 index--;
379 } else if (s < 0)
380 index++;
381
382 out:
Koji Sato17c76b02009-04-06 19:01:24 -0700383 *indexp = index;
384
385 return s == 0;
386}
387
Ryusuke Konishi1d5385b2010-07-16 23:52:40 +0900388/**
389 * nilfs_btree_node_broken - verify consistency of btree node
390 * @node: btree node block to be examined
391 * @size: node size (in bytes)
392 * @blocknr: block number
393 *
394 * Return Value: If node is broken, 1 is returned. Otherwise, 0 is returned.
395 */
396static int nilfs_btree_node_broken(const struct nilfs_btree_node *node,
397 size_t size, sector_t blocknr)
398{
399 int level, flags, nchildren;
400 int ret = 0;
401
402 level = nilfs_btree_node_get_level(node);
403 flags = nilfs_btree_node_get_flags(node);
404 nchildren = nilfs_btree_node_get_nchildren(node);
405
406 if (unlikely(level < NILFS_BTREE_LEVEL_NODE_MIN ||
407 level >= NILFS_BTREE_LEVEL_MAX ||
408 (flags & NILFS_BTREE_NODE_ROOT) ||
409 nchildren < 0 ||
410 nchildren > NILFS_BTREE_NODE_NCHILDREN_MAX(size))) {
411 printk(KERN_CRIT "NILFS: bad btree node (blocknr=%llu): "
412 "level = %d, flags = 0x%x, nchildren = %d\n",
413 (unsigned long long)blocknr, level, flags, nchildren);
414 ret = 1;
415 }
416 return ret;
417}
418
419int nilfs_btree_broken_node_block(struct buffer_head *bh)
420{
421 return nilfs_btree_node_broken((struct nilfs_btree_node *)bh->b_data,
422 bh->b_size, bh->b_blocknr);
423}
424
Koji Sato17c76b02009-04-06 19:01:24 -0700425static inline struct nilfs_btree_node *
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900426nilfs_btree_get_root(const struct nilfs_bmap *btree)
Koji Sato17c76b02009-04-06 19:01:24 -0700427{
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900428 return (struct nilfs_btree_node *)btree->b_u.u_data;
Koji Sato17c76b02009-04-06 19:01:24 -0700429}
430
431static inline struct nilfs_btree_node *
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900432nilfs_btree_get_nonroot_node(const struct nilfs_btree_path *path, int level)
Koji Sato17c76b02009-04-06 19:01:24 -0700433{
434 return (struct nilfs_btree_node *)path[level].bp_bh->b_data;
435}
436
437static inline struct nilfs_btree_node *
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900438nilfs_btree_get_sib_node(const struct nilfs_btree_path *path, int level)
Koji Sato17c76b02009-04-06 19:01:24 -0700439{
440 return (struct nilfs_btree_node *)path[level].bp_sib_bh->b_data;
441}
442
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900443static inline int nilfs_btree_height(const struct nilfs_bmap *btree)
Koji Sato17c76b02009-04-06 19:01:24 -0700444{
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900445 return nilfs_btree_node_get_level(nilfs_btree_get_root(btree)) + 1;
Koji Sato17c76b02009-04-06 19:01:24 -0700446}
447
448static inline struct nilfs_btree_node *
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900449nilfs_btree_get_node(const struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700450 const struct nilfs_btree_path *path,
451 int level)
452{
453 return (level == nilfs_btree_height(btree) - 1) ?
454 nilfs_btree_get_root(btree) :
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900455 nilfs_btree_get_nonroot_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -0700456}
457
Ryusuke Konishi9b945d52009-10-10 22:58:10 +0900458static inline int
459nilfs_btree_bad_node(struct nilfs_btree_node *node, int level)
460{
461 if (unlikely(nilfs_btree_node_get_level(node) != level)) {
462 dump_stack();
463 printk(KERN_CRIT "NILFS: btree level mismatch: %d != %d\n",
464 nilfs_btree_node_get_level(node), level);
465 return 1;
466 }
467 return 0;
468}
469
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900470static int nilfs_btree_do_lookup(const struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700471 struct nilfs_btree_path *path,
472 __u64 key, __u64 *ptrp, int minlevel)
473{
474 struct nilfs_btree_node *node;
475 __u64 ptr;
476 int level, index, found, ret;
477
Koji Sato17c76b02009-04-06 19:01:24 -0700478 node = nilfs_btree_get_root(btree);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900479 level = nilfs_btree_node_get_level(node);
480 if (level < minlevel || nilfs_btree_node_get_nchildren(node) <= 0)
Koji Sato17c76b02009-04-06 19:01:24 -0700481 return -ENOENT;
482
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900483 found = nilfs_btree_node_lookup(node, key, &index);
Koji Sato17c76b02009-04-06 19:01:24 -0700484 ptr = nilfs_btree_node_get_ptr(btree, node, index);
485 path[level].bp_bh = NULL;
486 path[level].bp_index = index;
487
488 for (level--; level >= minlevel; level--) {
Ryusuke Konishif198dbb2009-05-22 01:07:13 +0900489 ret = nilfs_btree_get_block(btree, ptr, &path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700490 if (ret < 0)
491 return ret;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900492 node = nilfs_btree_get_nonroot_node(path, level);
Ryusuke Konishi9b945d52009-10-10 22:58:10 +0900493 if (nilfs_btree_bad_node(node, level))
494 return -EINVAL;
Koji Sato17c76b02009-04-06 19:01:24 -0700495 if (!found)
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900496 found = nilfs_btree_node_lookup(node, key, &index);
Koji Sato17c76b02009-04-06 19:01:24 -0700497 else
498 index = 0;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900499 if (index < nilfs_btree_node_nchildren_max(node, btree))
Koji Sato17c76b02009-04-06 19:01:24 -0700500 ptr = nilfs_btree_node_get_ptr(btree, node, index);
501 else {
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -0700502 WARN_ON(found || level != NILFS_BTREE_LEVEL_NODE_MIN);
Koji Sato17c76b02009-04-06 19:01:24 -0700503 /* insert */
504 ptr = NILFS_BMAP_INVALID_PTR;
505 }
506 path[level].bp_index = index;
507 }
508 if (!found)
509 return -ENOENT;
510
511 if (ptrp != NULL)
512 *ptrp = ptr;
513
514 return 0;
515}
516
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900517static int nilfs_btree_do_lookup_last(const struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700518 struct nilfs_btree_path *path,
519 __u64 *keyp, __u64 *ptrp)
520{
521 struct nilfs_btree_node *node;
522 __u64 ptr;
523 int index, level, ret;
524
525 node = nilfs_btree_get_root(btree);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900526 index = nilfs_btree_node_get_nchildren(node) - 1;
Koji Sato17c76b02009-04-06 19:01:24 -0700527 if (index < 0)
528 return -ENOENT;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900529 level = nilfs_btree_node_get_level(node);
Koji Sato17c76b02009-04-06 19:01:24 -0700530 ptr = nilfs_btree_node_get_ptr(btree, node, index);
531 path[level].bp_bh = NULL;
532 path[level].bp_index = index;
533
534 for (level--; level > 0; level--) {
Ryusuke Konishif198dbb2009-05-22 01:07:13 +0900535 ret = nilfs_btree_get_block(btree, ptr, &path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700536 if (ret < 0)
537 return ret;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900538 node = nilfs_btree_get_nonroot_node(path, level);
Ryusuke Konishi9b945d52009-10-10 22:58:10 +0900539 if (nilfs_btree_bad_node(node, level))
540 return -EINVAL;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900541 index = nilfs_btree_node_get_nchildren(node) - 1;
Koji Sato17c76b02009-04-06 19:01:24 -0700542 ptr = nilfs_btree_node_get_ptr(btree, node, index);
543 path[level].bp_index = index;
544 }
545
546 if (keyp != NULL)
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900547 *keyp = nilfs_btree_node_get_key(node, index);
Koji Sato17c76b02009-04-06 19:01:24 -0700548 if (ptrp != NULL)
549 *ptrp = ptr;
550
551 return 0;
552}
553
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900554static int nilfs_btree_lookup(const struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700555 __u64 key, int level, __u64 *ptrp)
556{
Koji Sato17c76b02009-04-06 19:01:24 -0700557 struct nilfs_btree_path *path;
558 __u64 ptr;
559 int ret;
560
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900561 path = nilfs_btree_alloc_path();
Koji Sato17c76b02009-04-06 19:01:24 -0700562 if (path == NULL)
563 return -ENOMEM;
Koji Sato17c76b02009-04-06 19:01:24 -0700564
565 ret = nilfs_btree_do_lookup(btree, path, key, &ptr, level);
566
567 if (ptrp != NULL)
568 *ptrp = ptr;
569
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900570 nilfs_btree_free_path(path);
Koji Sato17c76b02009-04-06 19:01:24 -0700571
572 return ret;
573}
574
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900575static int nilfs_btree_lookup_contig(const struct nilfs_bmap *btree,
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900576 __u64 key, __u64 *ptrp, unsigned maxblocks)
577{
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900578 struct nilfs_btree_path *path;
579 struct nilfs_btree_node *node;
580 struct inode *dat = NULL;
581 __u64 ptr, ptr2;
582 sector_t blocknr;
583 int level = NILFS_BTREE_LEVEL_NODE_MIN;
584 int ret, cnt, index, maxlevel;
585
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900586 path = nilfs_btree_alloc_path();
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900587 if (path == NULL)
588 return -ENOMEM;
Li Hongf9054402010-04-02 17:36:34 +0800589
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900590 ret = nilfs_btree_do_lookup(btree, path, key, &ptr, level);
591 if (ret < 0)
592 goto out;
593
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900594 if (NILFS_BMAP_USE_VBN(btree)) {
595 dat = nilfs_bmap_get_dat(btree);
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900596 ret = nilfs_dat_translate(dat, ptr, &blocknr);
597 if (ret < 0)
598 goto out;
599 ptr = blocknr;
600 }
601 cnt = 1;
602 if (cnt == maxblocks)
603 goto end;
604
605 maxlevel = nilfs_btree_height(btree) - 1;
606 node = nilfs_btree_get_node(btree, path, level);
607 index = path[level].bp_index + 1;
608 for (;;) {
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900609 while (index < nilfs_btree_node_get_nchildren(node)) {
610 if (nilfs_btree_node_get_key(node, index) !=
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900611 key + cnt)
612 goto end;
613 ptr2 = nilfs_btree_node_get_ptr(btree, node, index);
614 if (dat) {
615 ret = nilfs_dat_translate(dat, ptr2, &blocknr);
616 if (ret < 0)
617 goto out;
618 ptr2 = blocknr;
619 }
620 if (ptr2 != ptr + cnt || ++cnt == maxblocks)
621 goto end;
622 index++;
623 continue;
624 }
625 if (level == maxlevel)
626 break;
627
628 /* look-up right sibling node */
629 node = nilfs_btree_get_node(btree, path, level + 1);
630 index = path[level + 1].bp_index + 1;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900631 if (index >= nilfs_btree_node_get_nchildren(node) ||
632 nilfs_btree_node_get_key(node, index) != key + cnt)
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900633 break;
634 ptr2 = nilfs_btree_node_get_ptr(btree, node, index);
635 path[level + 1].bp_index = index;
636
637 brelse(path[level].bp_bh);
638 path[level].bp_bh = NULL;
639 ret = nilfs_btree_get_block(btree, ptr2, &path[level].bp_bh);
640 if (ret < 0)
641 goto out;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900642 node = nilfs_btree_get_nonroot_node(path, level);
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900643 index = 0;
644 path[level].bp_index = index;
645 }
646 end:
647 *ptrp = ptr;
648 ret = cnt;
649 out:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900650 nilfs_btree_free_path(path);
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +0900651 return ret;
652}
653
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900654static void nilfs_btree_promote_key(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700655 struct nilfs_btree_path *path,
656 int level, __u64 key)
657{
658 if (level < nilfs_btree_height(btree) - 1) {
659 do {
Koji Sato17c76b02009-04-06 19:01:24 -0700660 nilfs_btree_node_set_key(
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900661 nilfs_btree_get_nonroot_node(path, level),
Koji Sato17c76b02009-04-06 19:01:24 -0700662 path[level].bp_index, key);
663 if (!buffer_dirty(path[level].bp_bh))
664 nilfs_btnode_mark_dirty(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700665 } while ((path[level].bp_index == 0) &&
666 (++level < nilfs_btree_height(btree) - 1));
667 }
668
669 /* root */
670 if (level == nilfs_btree_height(btree) - 1) {
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900671 nilfs_btree_node_set_key(nilfs_btree_get_root(btree),
Koji Sato17c76b02009-04-06 19:01:24 -0700672 path[level].bp_index, key);
673 }
674}
675
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900676static void nilfs_btree_do_insert(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700677 struct nilfs_btree_path *path,
678 int level, __u64 *keyp, __u64 *ptrp)
679{
680 struct nilfs_btree_node *node;
681
682 if (level < nilfs_btree_height(btree) - 1) {
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900683 node = nilfs_btree_get_nonroot_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -0700684 nilfs_btree_node_insert(btree, node, *keyp, *ptrp,
685 path[level].bp_index);
686 if (!buffer_dirty(path[level].bp_bh))
687 nilfs_btnode_mark_dirty(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700688
689 if (path[level].bp_index == 0)
690 nilfs_btree_promote_key(btree, path, level + 1,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900691 nilfs_btree_node_get_key(node,
692 0));
Koji Sato17c76b02009-04-06 19:01:24 -0700693 } else {
694 node = nilfs_btree_get_root(btree);
695 nilfs_btree_node_insert(btree, node, *keyp, *ptrp,
696 path[level].bp_index);
697 }
698}
699
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900700static void nilfs_btree_carry_left(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700701 struct nilfs_btree_path *path,
702 int level, __u64 *keyp, __u64 *ptrp)
703{
704 struct nilfs_btree_node *node, *left;
705 int nchildren, lnchildren, n, move;
706
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900707 node = nilfs_btree_get_nonroot_node(path, level);
708 left = nilfs_btree_get_sib_node(path, level);
709 nchildren = nilfs_btree_node_get_nchildren(node);
710 lnchildren = nilfs_btree_node_get_nchildren(left);
Koji Sato17c76b02009-04-06 19:01:24 -0700711 move = 0;
712
713 n = (nchildren + lnchildren + 1) / 2 - lnchildren;
714 if (n > path[level].bp_index) {
715 /* move insert point */
716 n--;
717 move = 1;
718 }
719
720 nilfs_btree_node_move_left(btree, left, node, n);
721
722 if (!buffer_dirty(path[level].bp_bh))
723 nilfs_btnode_mark_dirty(path[level].bp_bh);
724 if (!buffer_dirty(path[level].bp_sib_bh))
725 nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
726
Koji Sato17c76b02009-04-06 19:01:24 -0700727 nilfs_btree_promote_key(btree, path, level + 1,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900728 nilfs_btree_node_get_key(node, 0));
Koji Sato17c76b02009-04-06 19:01:24 -0700729
730 if (move) {
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900731 brelse(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700732 path[level].bp_bh = path[level].bp_sib_bh;
733 path[level].bp_sib_bh = NULL;
734 path[level].bp_index += lnchildren;
735 path[level + 1].bp_index--;
736 } else {
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900737 brelse(path[level].bp_sib_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700738 path[level].bp_sib_bh = NULL;
739 path[level].bp_index -= n;
740 }
741
742 nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
743}
744
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900745static void nilfs_btree_carry_right(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700746 struct nilfs_btree_path *path,
747 int level, __u64 *keyp, __u64 *ptrp)
748{
749 struct nilfs_btree_node *node, *right;
750 int nchildren, rnchildren, n, move;
751
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900752 node = nilfs_btree_get_nonroot_node(path, level);
753 right = nilfs_btree_get_sib_node(path, level);
754 nchildren = nilfs_btree_node_get_nchildren(node);
755 rnchildren = nilfs_btree_node_get_nchildren(right);
Koji Sato17c76b02009-04-06 19:01:24 -0700756 move = 0;
757
758 n = (nchildren + rnchildren + 1) / 2 - rnchildren;
759 if (n > nchildren - path[level].bp_index) {
760 /* move insert point */
761 n--;
762 move = 1;
763 }
764
765 nilfs_btree_node_move_right(btree, node, right, n);
766
767 if (!buffer_dirty(path[level].bp_bh))
768 nilfs_btnode_mark_dirty(path[level].bp_bh);
769 if (!buffer_dirty(path[level].bp_sib_bh))
770 nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
771
Koji Sato17c76b02009-04-06 19:01:24 -0700772 path[level + 1].bp_index++;
773 nilfs_btree_promote_key(btree, path, level + 1,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900774 nilfs_btree_node_get_key(right, 0));
Koji Sato17c76b02009-04-06 19:01:24 -0700775 path[level + 1].bp_index--;
776
777 if (move) {
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900778 brelse(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700779 path[level].bp_bh = path[level].bp_sib_bh;
780 path[level].bp_sib_bh = NULL;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900781 path[level].bp_index -= nilfs_btree_node_get_nchildren(node);
Koji Sato17c76b02009-04-06 19:01:24 -0700782 path[level + 1].bp_index++;
783 } else {
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900784 brelse(path[level].bp_sib_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700785 path[level].bp_sib_bh = NULL;
786 }
787
788 nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
789}
790
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900791static void nilfs_btree_split(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700792 struct nilfs_btree_path *path,
793 int level, __u64 *keyp, __u64 *ptrp)
794{
795 struct nilfs_btree_node *node, *right;
796 __u64 newkey;
797 __u64 newptr;
798 int nchildren, n, move;
799
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900800 node = nilfs_btree_get_nonroot_node(path, level);
801 right = nilfs_btree_get_sib_node(path, level);
802 nchildren = nilfs_btree_node_get_nchildren(node);
Koji Sato17c76b02009-04-06 19:01:24 -0700803 move = 0;
804
805 n = (nchildren + 1) / 2;
806 if (n > nchildren - path[level].bp_index) {
807 n--;
808 move = 1;
809 }
810
811 nilfs_btree_node_move_right(btree, node, right, n);
812
813 if (!buffer_dirty(path[level].bp_bh))
814 nilfs_btnode_mark_dirty(path[level].bp_bh);
815 if (!buffer_dirty(path[level].bp_sib_bh))
816 nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
817
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900818 newkey = nilfs_btree_node_get_key(right, 0);
Koji Sato17c76b02009-04-06 19:01:24 -0700819 newptr = path[level].bp_newreq.bpr_ptr;
820
821 if (move) {
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900822 path[level].bp_index -= nilfs_btree_node_get_nchildren(node);
Koji Sato17c76b02009-04-06 19:01:24 -0700823 nilfs_btree_node_insert(btree, right, *keyp, *ptrp,
824 path[level].bp_index);
825
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900826 *keyp = nilfs_btree_node_get_key(right, 0);
Koji Sato17c76b02009-04-06 19:01:24 -0700827 *ptrp = path[level].bp_newreq.bpr_ptr;
828
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900829 brelse(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700830 path[level].bp_bh = path[level].bp_sib_bh;
831 path[level].bp_sib_bh = NULL;
832 } else {
833 nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
834
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900835 *keyp = nilfs_btree_node_get_key(right, 0);
Koji Sato17c76b02009-04-06 19:01:24 -0700836 *ptrp = path[level].bp_newreq.bpr_ptr;
837
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900838 brelse(path[level].bp_sib_bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700839 path[level].bp_sib_bh = NULL;
840 }
841
842 path[level + 1].bp_index++;
843}
844
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900845static void nilfs_btree_grow(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700846 struct nilfs_btree_path *path,
847 int level, __u64 *keyp, __u64 *ptrp)
848{
849 struct nilfs_btree_node *root, *child;
850 int n;
851
Koji Sato17c76b02009-04-06 19:01:24 -0700852 root = nilfs_btree_get_root(btree);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900853 child = nilfs_btree_get_sib_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -0700854
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900855 n = nilfs_btree_node_get_nchildren(root);
Koji Sato17c76b02009-04-06 19:01:24 -0700856
857 nilfs_btree_node_move_right(btree, root, child, n);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900858 nilfs_btree_node_set_level(root, level + 1);
Koji Sato17c76b02009-04-06 19:01:24 -0700859
860 if (!buffer_dirty(path[level].bp_sib_bh))
861 nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
862
Koji Sato17c76b02009-04-06 19:01:24 -0700863 path[level].bp_bh = path[level].bp_sib_bh;
864 path[level].bp_sib_bh = NULL;
865
866 nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
867
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900868 *keyp = nilfs_btree_node_get_key(child, 0);
Koji Sato17c76b02009-04-06 19:01:24 -0700869 *ptrp = path[level].bp_newreq.bpr_ptr;
870}
871
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900872static __u64 nilfs_btree_find_near(const struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700873 const struct nilfs_btree_path *path)
874{
875 struct nilfs_btree_node *node;
876 int level;
877
878 if (path == NULL)
879 return NILFS_BMAP_INVALID_PTR;
880
881 /* left sibling */
882 level = NILFS_BTREE_LEVEL_NODE_MIN;
883 if (path[level].bp_index > 0) {
884 node = nilfs_btree_get_node(btree, path, level);
885 return nilfs_btree_node_get_ptr(btree, node,
886 path[level].bp_index - 1);
887 }
888
889 /* parent */
890 level = NILFS_BTREE_LEVEL_NODE_MIN + 1;
891 if (level <= nilfs_btree_height(btree) - 1) {
892 node = nilfs_btree_get_node(btree, path, level);
893 return nilfs_btree_node_get_ptr(btree, node,
894 path[level].bp_index);
895 }
896
897 return NILFS_BMAP_INVALID_PTR;
898}
899
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900900static __u64 nilfs_btree_find_target_v(const struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700901 const struct nilfs_btree_path *path,
902 __u64 key)
903{
904 __u64 ptr;
905
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900906 ptr = nilfs_bmap_find_target_seq(btree, key);
Koji Sato17c76b02009-04-06 19:01:24 -0700907 if (ptr != NILFS_BMAP_INVALID_PTR)
908 /* sequential access */
909 return ptr;
910 else {
911 ptr = nilfs_btree_find_near(btree, path);
912 if (ptr != NILFS_BMAP_INVALID_PTR)
913 /* near */
914 return ptr;
915 }
916 /* block group */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900917 return nilfs_bmap_find_target_in_group(btree);
Koji Sato17c76b02009-04-06 19:01:24 -0700918}
919
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900920static int nilfs_btree_prepare_insert(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -0700921 struct nilfs_btree_path *path,
922 int *levelp, __u64 key, __u64 ptr,
923 struct nilfs_bmap_stats *stats)
924{
925 struct buffer_head *bh;
926 struct nilfs_btree_node *node, *parent, *sib;
927 __u64 sibptr;
928 int pindex, level, ret;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +0900929 struct inode *dat = NULL;
Koji Sato17c76b02009-04-06 19:01:24 -0700930
931 stats->bs_nblocks = 0;
932 level = NILFS_BTREE_LEVEL_DATA;
933
934 /* allocate a new ptr for data block */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900935 if (NILFS_BMAP_USE_VBN(btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -0700936 path[level].bp_newreq.bpr_ptr =
Ryusuke Konishi7cde31d2009-05-24 18:07:59 +0900937 nilfs_btree_find_target_v(btree, path, key);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900938 dat = nilfs_bmap_get_dat(btree);
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +0900939 }
Koji Sato17c76b02009-04-06 19:01:24 -0700940
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900941 ret = nilfs_bmap_prepare_alloc_ptr(btree, &path[level].bp_newreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -0700942 if (ret < 0)
943 goto err_out_data;
944
945 for (level = NILFS_BTREE_LEVEL_NODE_MIN;
946 level < nilfs_btree_height(btree) - 1;
947 level++) {
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900948 node = nilfs_btree_get_nonroot_node(path, level);
949 if (nilfs_btree_node_get_nchildren(node) <
950 nilfs_btree_node_nchildren_max(node, btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -0700951 path[level].bp_op = nilfs_btree_do_insert;
952 stats->bs_nblocks++;
953 goto out;
954 }
955
956 parent = nilfs_btree_get_node(btree, path, level + 1);
957 pindex = path[level + 1].bp_index;
958
959 /* left sibling */
960 if (pindex > 0) {
961 sibptr = nilfs_btree_node_get_ptr(btree, parent,
962 pindex - 1);
Ryusuke Konishif198dbb2009-05-22 01:07:13 +0900963 ret = nilfs_btree_get_block(btree, sibptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700964 if (ret < 0)
965 goto err_out_child_node;
966 sib = (struct nilfs_btree_node *)bh->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900967 if (nilfs_btree_node_get_nchildren(sib) <
968 nilfs_btree_node_nchildren_max(sib, btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -0700969 path[level].bp_sib_bh = bh;
970 path[level].bp_op = nilfs_btree_carry_left;
971 stats->bs_nblocks++;
972 goto out;
973 } else
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900974 brelse(bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700975 }
976
977 /* right sibling */
978 if (pindex <
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900979 nilfs_btree_node_get_nchildren(parent) - 1) {
Koji Sato17c76b02009-04-06 19:01:24 -0700980 sibptr = nilfs_btree_node_get_ptr(btree, parent,
981 pindex + 1);
Ryusuke Konishif198dbb2009-05-22 01:07:13 +0900982 ret = nilfs_btree_get_block(btree, sibptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700983 if (ret < 0)
984 goto err_out_child_node;
985 sib = (struct nilfs_btree_node *)bh->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +0900986 if (nilfs_btree_node_get_nchildren(sib) <
987 nilfs_btree_node_nchildren_max(sib, btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -0700988 path[level].bp_sib_bh = bh;
989 path[level].bp_op = nilfs_btree_carry_right;
990 stats->bs_nblocks++;
991 goto out;
992 } else
Ryusuke Konishi087d01b2009-05-22 00:33:13 +0900993 brelse(bh);
Koji Sato17c76b02009-04-06 19:01:24 -0700994 }
995
996 /* split */
997 path[level].bp_newreq.bpr_ptr =
998 path[level - 1].bp_newreq.bpr_ptr + 1;
Ryusuke Konishie7c274f2010-07-10 19:09:49 +0900999 ret = nilfs_bmap_prepare_alloc_ptr(btree,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001000 &path[level].bp_newreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001001 if (ret < 0)
1002 goto err_out_child_node;
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09001003 ret = nilfs_btree_get_new_block(btree,
1004 path[level].bp_newreq.bpr_ptr,
1005 &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001006 if (ret < 0)
1007 goto err_out_curr_node;
1008
1009 stats->bs_nblocks++;
1010
Koji Sato17c76b02009-04-06 19:01:24 -07001011 nilfs_btree_node_init(btree,
1012 (struct nilfs_btree_node *)bh->b_data,
1013 0, level, 0, NULL, NULL);
Koji Sato17c76b02009-04-06 19:01:24 -07001014 path[level].bp_sib_bh = bh;
1015 path[level].bp_op = nilfs_btree_split;
1016 }
1017
1018 /* root */
1019 node = nilfs_btree_get_root(btree);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001020 if (nilfs_btree_node_get_nchildren(node) <
1021 nilfs_btree_node_nchildren_max(node, btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -07001022 path[level].bp_op = nilfs_btree_do_insert;
1023 stats->bs_nblocks++;
1024 goto out;
1025 }
1026
1027 /* grow */
1028 path[level].bp_newreq.bpr_ptr = path[level - 1].bp_newreq.bpr_ptr + 1;
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001029 ret = nilfs_bmap_prepare_alloc_ptr(btree, &path[level].bp_newreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001030 if (ret < 0)
1031 goto err_out_child_node;
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09001032 ret = nilfs_btree_get_new_block(btree, path[level].bp_newreq.bpr_ptr,
1033 &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001034 if (ret < 0)
1035 goto err_out_curr_node;
1036
Koji Sato17c76b02009-04-06 19:01:24 -07001037 nilfs_btree_node_init(btree, (struct nilfs_btree_node *)bh->b_data,
1038 0, level, 0, NULL, NULL);
Koji Sato17c76b02009-04-06 19:01:24 -07001039 path[level].bp_sib_bh = bh;
1040 path[level].bp_op = nilfs_btree_grow;
1041
1042 level++;
1043 path[level].bp_op = nilfs_btree_do_insert;
1044
1045 /* a newly-created node block and a data block are added */
1046 stats->bs_nblocks += 2;
1047
1048 /* success */
1049 out:
1050 *levelp = level;
1051 return ret;
1052
1053 /* error */
1054 err_out_curr_node:
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001055 nilfs_bmap_abort_alloc_ptr(btree, &path[level].bp_newreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001056 err_out_child_node:
1057 for (level--; level > NILFS_BTREE_LEVEL_DATA; level--) {
Ryusuke Konishi9f098902009-05-22 00:38:56 +09001058 nilfs_btnode_delete(path[level].bp_sib_bh);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001059 nilfs_bmap_abort_alloc_ptr(btree, &path[level].bp_newreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001060
1061 }
1062
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001063 nilfs_bmap_abort_alloc_ptr(btree, &path[level].bp_newreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001064 err_out_data:
1065 *levelp = level;
1066 stats->bs_nblocks = 0;
1067 return ret;
1068}
1069
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001070static void nilfs_btree_commit_insert(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001071 struct nilfs_btree_path *path,
1072 int maxlevel, __u64 key, __u64 ptr)
1073{
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001074 struct inode *dat = NULL;
Koji Sato17c76b02009-04-06 19:01:24 -07001075 int level;
1076
1077 set_buffer_nilfs_volatile((struct buffer_head *)((unsigned long)ptr));
1078 ptr = path[NILFS_BTREE_LEVEL_DATA].bp_newreq.bpr_ptr;
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001079 if (NILFS_BMAP_USE_VBN(btree)) {
Ryusuke Konishidc935be2010-07-10 22:21:54 +09001080 nilfs_bmap_set_target_v(btree, key, ptr);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001081 dat = nilfs_bmap_get_dat(btree);
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001082 }
Koji Sato17c76b02009-04-06 19:01:24 -07001083
1084 for (level = NILFS_BTREE_LEVEL_NODE_MIN; level <= maxlevel; level++) {
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001085 nilfs_bmap_commit_alloc_ptr(btree,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001086 &path[level - 1].bp_newreq, dat);
Pekka Enberg8acfbf02009-04-06 19:01:49 -07001087 path[level].bp_op(btree, path, level, &key, &ptr);
Koji Sato17c76b02009-04-06 19:01:24 -07001088 }
1089
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001090 if (!nilfs_bmap_dirty(btree))
1091 nilfs_bmap_set_dirty(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001092}
1093
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001094static int nilfs_btree_insert(struct nilfs_bmap *btree, __u64 key, __u64 ptr)
Koji Sato17c76b02009-04-06 19:01:24 -07001095{
Koji Sato17c76b02009-04-06 19:01:24 -07001096 struct nilfs_btree_path *path;
1097 struct nilfs_bmap_stats stats;
1098 int level, ret;
1099
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001100 path = nilfs_btree_alloc_path();
Koji Sato17c76b02009-04-06 19:01:24 -07001101 if (path == NULL)
1102 return -ENOMEM;
Koji Sato17c76b02009-04-06 19:01:24 -07001103
1104 ret = nilfs_btree_do_lookup(btree, path, key, NULL,
1105 NILFS_BTREE_LEVEL_NODE_MIN);
1106 if (ret != -ENOENT) {
1107 if (ret == 0)
1108 ret = -EEXIST;
1109 goto out;
1110 }
1111
1112 ret = nilfs_btree_prepare_insert(btree, path, &level, key, ptr, &stats);
1113 if (ret < 0)
1114 goto out;
1115 nilfs_btree_commit_insert(btree, path, level, key, ptr);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001116 nilfs_bmap_add_blocks(btree, stats.bs_nblocks);
Koji Sato17c76b02009-04-06 19:01:24 -07001117
1118 out:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001119 nilfs_btree_free_path(path);
Koji Sato17c76b02009-04-06 19:01:24 -07001120 return ret;
1121}
1122
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001123static void nilfs_btree_do_delete(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001124 struct nilfs_btree_path *path,
1125 int level, __u64 *keyp, __u64 *ptrp)
1126{
1127 struct nilfs_btree_node *node;
1128
1129 if (level < nilfs_btree_height(btree) - 1) {
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001130 node = nilfs_btree_get_nonroot_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -07001131 nilfs_btree_node_delete(btree, node, keyp, ptrp,
1132 path[level].bp_index);
1133 if (!buffer_dirty(path[level].bp_bh))
1134 nilfs_btnode_mark_dirty(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001135 if (path[level].bp_index == 0)
1136 nilfs_btree_promote_key(btree, path, level + 1,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001137 nilfs_btree_node_get_key(node, 0));
Koji Sato17c76b02009-04-06 19:01:24 -07001138 } else {
1139 node = nilfs_btree_get_root(btree);
1140 nilfs_btree_node_delete(btree, node, keyp, ptrp,
1141 path[level].bp_index);
1142 }
1143}
1144
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001145static void nilfs_btree_borrow_left(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001146 struct nilfs_btree_path *path,
1147 int level, __u64 *keyp, __u64 *ptrp)
1148{
1149 struct nilfs_btree_node *node, *left;
1150 int nchildren, lnchildren, n;
1151
1152 nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
1153
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001154 node = nilfs_btree_get_nonroot_node(path, level);
1155 left = nilfs_btree_get_sib_node(path, level);
1156 nchildren = nilfs_btree_node_get_nchildren(node);
1157 lnchildren = nilfs_btree_node_get_nchildren(left);
Koji Sato17c76b02009-04-06 19:01:24 -07001158
1159 n = (nchildren + lnchildren) / 2 - nchildren;
1160
1161 nilfs_btree_node_move_right(btree, left, node, n);
1162
1163 if (!buffer_dirty(path[level].bp_bh))
1164 nilfs_btnode_mark_dirty(path[level].bp_bh);
1165 if (!buffer_dirty(path[level].bp_sib_bh))
1166 nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
1167
Koji Sato17c76b02009-04-06 19:01:24 -07001168 nilfs_btree_promote_key(btree, path, level + 1,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001169 nilfs_btree_node_get_key(node, 0));
Koji Sato17c76b02009-04-06 19:01:24 -07001170
Ryusuke Konishi087d01b2009-05-22 00:33:13 +09001171 brelse(path[level].bp_sib_bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001172 path[level].bp_sib_bh = NULL;
1173 path[level].bp_index += n;
1174}
1175
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001176static void nilfs_btree_borrow_right(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001177 struct nilfs_btree_path *path,
1178 int level, __u64 *keyp, __u64 *ptrp)
1179{
1180 struct nilfs_btree_node *node, *right;
1181 int nchildren, rnchildren, n;
1182
1183 nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
1184
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001185 node = nilfs_btree_get_nonroot_node(path, level);
1186 right = nilfs_btree_get_sib_node(path, level);
1187 nchildren = nilfs_btree_node_get_nchildren(node);
1188 rnchildren = nilfs_btree_node_get_nchildren(right);
Koji Sato17c76b02009-04-06 19:01:24 -07001189
1190 n = (nchildren + rnchildren) / 2 - nchildren;
1191
1192 nilfs_btree_node_move_left(btree, node, right, n);
1193
1194 if (!buffer_dirty(path[level].bp_bh))
1195 nilfs_btnode_mark_dirty(path[level].bp_bh);
1196 if (!buffer_dirty(path[level].bp_sib_bh))
1197 nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
1198
Koji Sato17c76b02009-04-06 19:01:24 -07001199 path[level + 1].bp_index++;
1200 nilfs_btree_promote_key(btree, path, level + 1,
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001201 nilfs_btree_node_get_key(right, 0));
Koji Sato17c76b02009-04-06 19:01:24 -07001202 path[level + 1].bp_index--;
1203
Ryusuke Konishi087d01b2009-05-22 00:33:13 +09001204 brelse(path[level].bp_sib_bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001205 path[level].bp_sib_bh = NULL;
1206}
1207
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001208static void nilfs_btree_concat_left(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001209 struct nilfs_btree_path *path,
1210 int level, __u64 *keyp, __u64 *ptrp)
1211{
1212 struct nilfs_btree_node *node, *left;
1213 int n;
1214
1215 nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
1216
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001217 node = nilfs_btree_get_nonroot_node(path, level);
1218 left = nilfs_btree_get_sib_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -07001219
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001220 n = nilfs_btree_node_get_nchildren(node);
Koji Sato17c76b02009-04-06 19:01:24 -07001221
1222 nilfs_btree_node_move_left(btree, left, node, n);
1223
1224 if (!buffer_dirty(path[level].bp_sib_bh))
1225 nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
1226
Ryusuke Konishi9f098902009-05-22 00:38:56 +09001227 nilfs_btnode_delete(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001228 path[level].bp_bh = path[level].bp_sib_bh;
1229 path[level].bp_sib_bh = NULL;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001230 path[level].bp_index += nilfs_btree_node_get_nchildren(left);
Koji Sato17c76b02009-04-06 19:01:24 -07001231}
1232
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001233static void nilfs_btree_concat_right(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001234 struct nilfs_btree_path *path,
1235 int level, __u64 *keyp, __u64 *ptrp)
1236{
1237 struct nilfs_btree_node *node, *right;
1238 int n;
1239
1240 nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
1241
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001242 node = nilfs_btree_get_nonroot_node(path, level);
1243 right = nilfs_btree_get_sib_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -07001244
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001245 n = nilfs_btree_node_get_nchildren(right);
Koji Sato17c76b02009-04-06 19:01:24 -07001246
1247 nilfs_btree_node_move_left(btree, node, right, n);
1248
1249 if (!buffer_dirty(path[level].bp_bh))
1250 nilfs_btnode_mark_dirty(path[level].bp_bh);
1251
Ryusuke Konishi9f098902009-05-22 00:38:56 +09001252 nilfs_btnode_delete(path[level].bp_sib_bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001253 path[level].bp_sib_bh = NULL;
1254 path[level + 1].bp_index++;
1255}
1256
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001257static void nilfs_btree_shrink(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001258 struct nilfs_btree_path *path,
1259 int level, __u64 *keyp, __u64 *ptrp)
1260{
1261 struct nilfs_btree_node *root, *child;
1262 int n;
1263
1264 nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
1265
Koji Sato17c76b02009-04-06 19:01:24 -07001266 root = nilfs_btree_get_root(btree);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001267 child = nilfs_btree_get_nonroot_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -07001268
1269 nilfs_btree_node_delete(btree, root, NULL, NULL, 0);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001270 nilfs_btree_node_set_level(root, level);
1271 n = nilfs_btree_node_get_nchildren(child);
Koji Sato17c76b02009-04-06 19:01:24 -07001272 nilfs_btree_node_move_left(btree, root, child, n);
Koji Sato17c76b02009-04-06 19:01:24 -07001273
Ryusuke Konishi9f098902009-05-22 00:38:56 +09001274 nilfs_btnode_delete(path[level].bp_bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001275 path[level].bp_bh = NULL;
1276}
1277
1278
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001279static int nilfs_btree_prepare_delete(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001280 struct nilfs_btree_path *path,
1281 int *levelp,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001282 struct nilfs_bmap_stats *stats,
1283 struct inode *dat)
Koji Sato17c76b02009-04-06 19:01:24 -07001284{
1285 struct buffer_head *bh;
1286 struct nilfs_btree_node *node, *parent, *sib;
1287 __u64 sibptr;
1288 int pindex, level, ret;
1289
1290 ret = 0;
1291 stats->bs_nblocks = 0;
1292 for (level = NILFS_BTREE_LEVEL_NODE_MIN;
1293 level < nilfs_btree_height(btree) - 1;
1294 level++) {
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001295 node = nilfs_btree_get_nonroot_node(path, level);
Koji Sato17c76b02009-04-06 19:01:24 -07001296 path[level].bp_oldreq.bpr_ptr =
1297 nilfs_btree_node_get_ptr(btree, node,
1298 path[level].bp_index);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001299 ret = nilfs_bmap_prepare_end_ptr(btree,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001300 &path[level].bp_oldreq, dat);
Ryusuke Konishid4b96152009-05-24 03:25:44 +09001301 if (ret < 0)
1302 goto err_out_child_node;
Koji Sato17c76b02009-04-06 19:01:24 -07001303
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001304 if (nilfs_btree_node_get_nchildren(node) >
1305 nilfs_btree_node_nchildren_min(node, btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -07001306 path[level].bp_op = nilfs_btree_do_delete;
1307 stats->bs_nblocks++;
1308 goto out;
1309 }
1310
1311 parent = nilfs_btree_get_node(btree, path, level + 1);
1312 pindex = path[level + 1].bp_index;
1313
1314 if (pindex > 0) {
1315 /* left sibling */
1316 sibptr = nilfs_btree_node_get_ptr(btree, parent,
1317 pindex - 1);
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09001318 ret = nilfs_btree_get_block(btree, sibptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001319 if (ret < 0)
1320 goto err_out_curr_node;
1321 sib = (struct nilfs_btree_node *)bh->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001322 if (nilfs_btree_node_get_nchildren(sib) >
1323 nilfs_btree_node_nchildren_min(sib, btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -07001324 path[level].bp_sib_bh = bh;
1325 path[level].bp_op = nilfs_btree_borrow_left;
1326 stats->bs_nblocks++;
1327 goto out;
1328 } else {
1329 path[level].bp_sib_bh = bh;
1330 path[level].bp_op = nilfs_btree_concat_left;
1331 stats->bs_nblocks++;
1332 /* continue; */
1333 }
1334 } else if (pindex <
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001335 nilfs_btree_node_get_nchildren(parent) - 1) {
Koji Sato17c76b02009-04-06 19:01:24 -07001336 /* right sibling */
1337 sibptr = nilfs_btree_node_get_ptr(btree, parent,
1338 pindex + 1);
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09001339 ret = nilfs_btree_get_block(btree, sibptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001340 if (ret < 0)
1341 goto err_out_curr_node;
1342 sib = (struct nilfs_btree_node *)bh->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001343 if (nilfs_btree_node_get_nchildren(sib) >
1344 nilfs_btree_node_nchildren_min(sib, btree)) {
Koji Sato17c76b02009-04-06 19:01:24 -07001345 path[level].bp_sib_bh = bh;
1346 path[level].bp_op = nilfs_btree_borrow_right;
1347 stats->bs_nblocks++;
1348 goto out;
1349 } else {
1350 path[level].bp_sib_bh = bh;
1351 path[level].bp_op = nilfs_btree_concat_right;
1352 stats->bs_nblocks++;
1353 /* continue; */
1354 }
1355 } else {
1356 /* no siblings */
1357 /* the only child of the root node */
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07001358 WARN_ON(level != nilfs_btree_height(btree) - 2);
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001359 if (nilfs_btree_node_get_nchildren(node) - 1 <=
Koji Sato17c76b02009-04-06 19:01:24 -07001360 NILFS_BTREE_ROOT_NCHILDREN_MAX) {
1361 path[level].bp_op = nilfs_btree_shrink;
1362 stats->bs_nblocks += 2;
1363 } else {
1364 path[level].bp_op = nilfs_btree_do_delete;
1365 stats->bs_nblocks++;
1366 }
1367
1368 goto out;
1369
1370 }
1371 }
1372
1373 node = nilfs_btree_get_root(btree);
1374 path[level].bp_oldreq.bpr_ptr =
1375 nilfs_btree_node_get_ptr(btree, node, path[level].bp_index);
Ryusuke Konishid4b96152009-05-24 03:25:44 +09001376
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001377 ret = nilfs_bmap_prepare_end_ptr(btree, &path[level].bp_oldreq, dat);
Ryusuke Konishid4b96152009-05-24 03:25:44 +09001378 if (ret < 0)
1379 goto err_out_child_node;
1380
Koji Sato17c76b02009-04-06 19:01:24 -07001381 /* child of the root node is deleted */
1382 path[level].bp_op = nilfs_btree_do_delete;
1383 stats->bs_nblocks++;
1384
1385 /* success */
1386 out:
1387 *levelp = level;
1388 return ret;
1389
1390 /* error */
1391 err_out_curr_node:
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001392 nilfs_bmap_abort_end_ptr(btree, &path[level].bp_oldreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001393 err_out_child_node:
1394 for (level--; level >= NILFS_BTREE_LEVEL_NODE_MIN; level--) {
Ryusuke Konishi087d01b2009-05-22 00:33:13 +09001395 brelse(path[level].bp_sib_bh);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001396 nilfs_bmap_abort_end_ptr(btree, &path[level].bp_oldreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001397 }
1398 *levelp = level;
1399 stats->bs_nblocks = 0;
1400 return ret;
1401}
1402
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001403static void nilfs_btree_commit_delete(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001404 struct nilfs_btree_path *path,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001405 int maxlevel, struct inode *dat)
Koji Sato17c76b02009-04-06 19:01:24 -07001406{
1407 int level;
1408
1409 for (level = NILFS_BTREE_LEVEL_NODE_MIN; level <= maxlevel; level++) {
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001410 nilfs_bmap_commit_end_ptr(btree, &path[level].bp_oldreq, dat);
Pekka Enberg8acfbf02009-04-06 19:01:49 -07001411 path[level].bp_op(btree, path, level, NULL, NULL);
Koji Sato17c76b02009-04-06 19:01:24 -07001412 }
1413
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001414 if (!nilfs_bmap_dirty(btree))
1415 nilfs_bmap_set_dirty(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001416}
1417
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001418static int nilfs_btree_delete(struct nilfs_bmap *btree, __u64 key)
Koji Sato17c76b02009-04-06 19:01:24 -07001419
1420{
Koji Sato17c76b02009-04-06 19:01:24 -07001421 struct nilfs_btree_path *path;
1422 struct nilfs_bmap_stats stats;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001423 struct inode *dat;
Koji Sato17c76b02009-04-06 19:01:24 -07001424 int level, ret;
1425
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001426 path = nilfs_btree_alloc_path();
Koji Sato17c76b02009-04-06 19:01:24 -07001427 if (path == NULL)
1428 return -ENOMEM;
Li Hongf9054402010-04-02 17:36:34 +08001429
Koji Sato17c76b02009-04-06 19:01:24 -07001430 ret = nilfs_btree_do_lookup(btree, path, key, NULL,
1431 NILFS_BTREE_LEVEL_NODE_MIN);
1432 if (ret < 0)
1433 goto out;
1434
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001435
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001436 dat = NILFS_BMAP_USE_VBN(btree) ? nilfs_bmap_get_dat(btree) : NULL;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001437
1438 ret = nilfs_btree_prepare_delete(btree, path, &level, &stats, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001439 if (ret < 0)
1440 goto out;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001441 nilfs_btree_commit_delete(btree, path, level, dat);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001442 nilfs_bmap_sub_blocks(btree, stats.bs_nblocks);
Koji Sato17c76b02009-04-06 19:01:24 -07001443
1444out:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001445 nilfs_btree_free_path(path);
Koji Sato17c76b02009-04-06 19:01:24 -07001446 return ret;
1447}
1448
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001449static int nilfs_btree_last_key(const struct nilfs_bmap *btree, __u64 *keyp)
Koji Sato17c76b02009-04-06 19:01:24 -07001450{
Koji Sato17c76b02009-04-06 19:01:24 -07001451 struct nilfs_btree_path *path;
1452 int ret;
1453
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001454 path = nilfs_btree_alloc_path();
Koji Sato17c76b02009-04-06 19:01:24 -07001455 if (path == NULL)
1456 return -ENOMEM;
Koji Sato17c76b02009-04-06 19:01:24 -07001457
1458 ret = nilfs_btree_do_lookup_last(btree, path, keyp, NULL);
1459
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001460 nilfs_btree_free_path(path);
Koji Sato17c76b02009-04-06 19:01:24 -07001461
1462 return ret;
1463}
1464
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001465static int nilfs_btree_check_delete(struct nilfs_bmap *btree, __u64 key)
Koji Sato17c76b02009-04-06 19:01:24 -07001466{
1467 struct buffer_head *bh;
Koji Sato17c76b02009-04-06 19:01:24 -07001468 struct nilfs_btree_node *root, *node;
1469 __u64 maxkey, nextmaxkey;
1470 __u64 ptr;
1471 int nchildren, ret;
1472
Koji Sato17c76b02009-04-06 19:01:24 -07001473 root = nilfs_btree_get_root(btree);
1474 switch (nilfs_btree_height(btree)) {
1475 case 2:
1476 bh = NULL;
1477 node = root;
1478 break;
1479 case 3:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001480 nchildren = nilfs_btree_node_get_nchildren(root);
Koji Sato17c76b02009-04-06 19:01:24 -07001481 if (nchildren > 1)
1482 return 0;
1483 ptr = nilfs_btree_node_get_ptr(btree, root, nchildren - 1);
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09001484 ret = nilfs_btree_get_block(btree, ptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001485 if (ret < 0)
1486 return ret;
1487 node = (struct nilfs_btree_node *)bh->b_data;
1488 break;
1489 default:
1490 return 0;
1491 }
1492
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001493 nchildren = nilfs_btree_node_get_nchildren(node);
1494 maxkey = nilfs_btree_node_get_key(node, nchildren - 1);
Koji Sato17c76b02009-04-06 19:01:24 -07001495 nextmaxkey = (nchildren > 1) ?
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001496 nilfs_btree_node_get_key(node, nchildren - 2) : 0;
Koji Sato17c76b02009-04-06 19:01:24 -07001497 if (bh != NULL)
Ryusuke Konishi087d01b2009-05-22 00:33:13 +09001498 brelse(bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001499
Ryusuke Konishi30333422009-05-24 00:09:44 +09001500 return (maxkey == key) && (nextmaxkey < NILFS_BMAP_LARGE_LOW);
Koji Sato17c76b02009-04-06 19:01:24 -07001501}
1502
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001503static int nilfs_btree_gather_data(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001504 __u64 *keys, __u64 *ptrs, int nitems)
1505{
1506 struct buffer_head *bh;
Koji Sato17c76b02009-04-06 19:01:24 -07001507 struct nilfs_btree_node *node, *root;
1508 __le64 *dkeys;
1509 __le64 *dptrs;
1510 __u64 ptr;
1511 int nchildren, i, ret;
1512
Koji Sato17c76b02009-04-06 19:01:24 -07001513 root = nilfs_btree_get_root(btree);
1514 switch (nilfs_btree_height(btree)) {
1515 case 2:
1516 bh = NULL;
1517 node = root;
1518 break;
1519 case 3:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001520 nchildren = nilfs_btree_node_get_nchildren(root);
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07001521 WARN_ON(nchildren > 1);
Koji Sato17c76b02009-04-06 19:01:24 -07001522 ptr = nilfs_btree_node_get_ptr(btree, root, nchildren - 1);
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09001523 ret = nilfs_btree_get_block(btree, ptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001524 if (ret < 0)
1525 return ret;
1526 node = (struct nilfs_btree_node *)bh->b_data;
1527 break;
1528 default:
1529 node = NULL;
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07001530 return -EINVAL;
Koji Sato17c76b02009-04-06 19:01:24 -07001531 }
1532
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001533 nchildren = nilfs_btree_node_get_nchildren(node);
Koji Sato17c76b02009-04-06 19:01:24 -07001534 if (nchildren < nitems)
1535 nitems = nchildren;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001536 dkeys = nilfs_btree_node_dkeys(node);
1537 dptrs = nilfs_btree_node_dptrs(node, btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001538 for (i = 0; i < nitems; i++) {
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +09001539 keys[i] = le64_to_cpu(dkeys[i]);
1540 ptrs[i] = le64_to_cpu(dptrs[i]);
Koji Sato17c76b02009-04-06 19:01:24 -07001541 }
1542
1543 if (bh != NULL)
Ryusuke Konishi087d01b2009-05-22 00:33:13 +09001544 brelse(bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001545
1546 return nitems;
1547}
1548
1549static int
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001550nilfs_btree_prepare_convert_and_insert(struct nilfs_bmap *btree, __u64 key,
Koji Sato17c76b02009-04-06 19:01:24 -07001551 union nilfs_bmap_ptr_req *dreq,
1552 union nilfs_bmap_ptr_req *nreq,
1553 struct buffer_head **bhp,
1554 struct nilfs_bmap_stats *stats)
1555{
1556 struct buffer_head *bh;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001557 struct inode *dat = NULL;
Koji Sato17c76b02009-04-06 19:01:24 -07001558 int ret;
1559
Koji Sato17c76b02009-04-06 19:01:24 -07001560 stats->bs_nblocks = 0;
1561
1562 /* for data */
1563 /* cannot find near ptr */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001564 if (NILFS_BMAP_USE_VBN(btree)) {
Ryusuke Konishi7cde31d2009-05-24 18:07:59 +09001565 dreq->bpr_ptr = nilfs_btree_find_target_v(btree, NULL, key);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001566 dat = nilfs_bmap_get_dat(btree);
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001567 }
Ryusuke Konishi7cde31d2009-05-24 18:07:59 +09001568
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001569 ret = nilfs_bmap_prepare_alloc_ptr(btree, dreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001570 if (ret < 0)
1571 return ret;
1572
1573 *bhp = NULL;
1574 stats->bs_nblocks++;
1575 if (nreq != NULL) {
1576 nreq->bpr_ptr = dreq->bpr_ptr + 1;
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001577 ret = nilfs_bmap_prepare_alloc_ptr(btree, nreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001578 if (ret < 0)
1579 goto err_out_dreq;
1580
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09001581 ret = nilfs_btree_get_new_block(btree, nreq->bpr_ptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001582 if (ret < 0)
1583 goto err_out_nreq;
1584
1585 *bhp = bh;
1586 stats->bs_nblocks++;
1587 }
1588
1589 /* success */
1590 return 0;
1591
1592 /* error */
1593 err_out_nreq:
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001594 nilfs_bmap_abort_alloc_ptr(btree, nreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001595 err_out_dreq:
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001596 nilfs_bmap_abort_alloc_ptr(btree, dreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001597 stats->bs_nblocks = 0;
1598 return ret;
1599
1600}
1601
1602static void
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001603nilfs_btree_commit_convert_and_insert(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001604 __u64 key, __u64 ptr,
1605 const __u64 *keys, const __u64 *ptrs,
Ryusuke Konishi30333422009-05-24 00:09:44 +09001606 int n,
Koji Sato17c76b02009-04-06 19:01:24 -07001607 union nilfs_bmap_ptr_req *dreq,
1608 union nilfs_bmap_ptr_req *nreq,
1609 struct buffer_head *bh)
1610{
Koji Sato17c76b02009-04-06 19:01:24 -07001611 struct nilfs_btree_node *node;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001612 struct inode *dat;
Koji Sato17c76b02009-04-06 19:01:24 -07001613 __u64 tmpptr;
1614
1615 /* free resources */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001616 if (btree->b_ops->bop_clear != NULL)
1617 btree->b_ops->bop_clear(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001618
1619 /* ptr must be a pointer to a buffer head. */
1620 set_buffer_nilfs_volatile((struct buffer_head *)((unsigned long)ptr));
1621
1622 /* convert and insert */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001623 dat = NILFS_BMAP_USE_VBN(btree) ? nilfs_bmap_get_dat(btree) : NULL;
1624 nilfs_btree_init(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001625 if (nreq != NULL) {
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001626 nilfs_bmap_commit_alloc_ptr(btree, dreq, dat);
1627 nilfs_bmap_commit_alloc_ptr(btree, nreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001628
1629 /* create child node at level 1 */
Koji Sato17c76b02009-04-06 19:01:24 -07001630 node = (struct nilfs_btree_node *)bh->b_data;
1631 nilfs_btree_node_init(btree, node, 0, 1, n, keys, ptrs);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001632 nilfs_btree_node_insert(btree, node, key, dreq->bpr_ptr, n);
Koji Sato17c76b02009-04-06 19:01:24 -07001633 if (!buffer_dirty(bh))
1634 nilfs_btnode_mark_dirty(bh);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001635 if (!nilfs_bmap_dirty(btree))
1636 nilfs_bmap_set_dirty(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001637
Ryusuke Konishi087d01b2009-05-22 00:33:13 +09001638 brelse(bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001639
1640 /* create root node at level 2 */
1641 node = nilfs_btree_get_root(btree);
1642 tmpptr = nreq->bpr_ptr;
1643 nilfs_btree_node_init(btree, node, NILFS_BTREE_NODE_ROOT,
1644 2, 1, &keys[0], &tmpptr);
1645 } else {
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001646 nilfs_bmap_commit_alloc_ptr(btree, dreq, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001647
1648 /* create root node at level 1 */
1649 node = nilfs_btree_get_root(btree);
1650 nilfs_btree_node_init(btree, node, NILFS_BTREE_NODE_ROOT,
1651 1, n, keys, ptrs);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001652 nilfs_btree_node_insert(btree, node, key, dreq->bpr_ptr, n);
1653 if (!nilfs_bmap_dirty(btree))
1654 nilfs_bmap_set_dirty(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001655 }
1656
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001657 if (NILFS_BMAP_USE_VBN(btree))
Ryusuke Konishidc935be2010-07-10 22:21:54 +09001658 nilfs_bmap_set_target_v(btree, key, dreq->bpr_ptr);
Koji Sato17c76b02009-04-06 19:01:24 -07001659}
1660
1661/**
1662 * nilfs_btree_convert_and_insert -
1663 * @bmap:
1664 * @key:
1665 * @ptr:
1666 * @keys:
1667 * @ptrs:
1668 * @n:
Koji Sato17c76b02009-04-06 19:01:24 -07001669 */
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001670int nilfs_btree_convert_and_insert(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001671 __u64 key, __u64 ptr,
Ryusuke Konishi30333422009-05-24 00:09:44 +09001672 const __u64 *keys, const __u64 *ptrs, int n)
Koji Sato17c76b02009-04-06 19:01:24 -07001673{
1674 struct buffer_head *bh;
1675 union nilfs_bmap_ptr_req dreq, nreq, *di, *ni;
1676 struct nilfs_bmap_stats stats;
1677 int ret;
1678
1679 if (n + 1 <= NILFS_BTREE_ROOT_NCHILDREN_MAX) {
1680 di = &dreq;
1681 ni = NULL;
1682 } else if ((n + 1) <= NILFS_BTREE_NODE_NCHILDREN_MAX(
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001683 1 << btree->b_inode->i_blkbits)) {
Koji Sato17c76b02009-04-06 19:01:24 -07001684 di = &dreq;
1685 ni = &nreq;
1686 } else {
1687 di = NULL;
1688 ni = NULL;
1689 BUG();
1690 }
1691
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001692 ret = nilfs_btree_prepare_convert_and_insert(btree, key, di, ni, &bh,
Koji Sato17c76b02009-04-06 19:01:24 -07001693 &stats);
1694 if (ret < 0)
1695 return ret;
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001696 nilfs_btree_commit_convert_and_insert(btree, key, ptr, keys, ptrs, n,
Ryusuke Konishi30333422009-05-24 00:09:44 +09001697 di, ni, bh);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001698 nilfs_bmap_add_blocks(btree, stats.bs_nblocks);
Koji Sato17c76b02009-04-06 19:01:24 -07001699 return 0;
1700}
1701
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001702static int nilfs_btree_propagate_p(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001703 struct nilfs_btree_path *path,
1704 int level,
1705 struct buffer_head *bh)
1706{
1707 while ((++level < nilfs_btree_height(btree) - 1) &&
1708 !buffer_dirty(path[level].bp_bh))
1709 nilfs_btnode_mark_dirty(path[level].bp_bh);
1710
1711 return 0;
1712}
1713
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001714static int nilfs_btree_prepare_update_v(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001715 struct nilfs_btree_path *path,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001716 int level, struct inode *dat)
Koji Sato17c76b02009-04-06 19:01:24 -07001717{
1718 struct nilfs_btree_node *parent;
1719 int ret;
1720
1721 parent = nilfs_btree_get_node(btree, path, level + 1);
1722 path[level].bp_oldreq.bpr_ptr =
1723 nilfs_btree_node_get_ptr(btree, parent,
1724 path[level + 1].bp_index);
1725 path[level].bp_newreq.bpr_ptr = path[level].bp_oldreq.bpr_ptr + 1;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001726 ret = nilfs_dat_prepare_update(dat, &path[level].bp_oldreq.bpr_req,
1727 &path[level].bp_newreq.bpr_req);
Koji Sato17c76b02009-04-06 19:01:24 -07001728 if (ret < 0)
1729 return ret;
1730
1731 if (buffer_nilfs_node(path[level].bp_bh)) {
1732 path[level].bp_ctxt.oldkey = path[level].bp_oldreq.bpr_ptr;
1733 path[level].bp_ctxt.newkey = path[level].bp_newreq.bpr_ptr;
1734 path[level].bp_ctxt.bh = path[level].bp_bh;
1735 ret = nilfs_btnode_prepare_change_key(
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001736 &NILFS_BMAP_I(btree)->i_btnode_cache,
Koji Sato17c76b02009-04-06 19:01:24 -07001737 &path[level].bp_ctxt);
1738 if (ret < 0) {
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001739 nilfs_dat_abort_update(dat,
1740 &path[level].bp_oldreq.bpr_req,
1741 &path[level].bp_newreq.bpr_req);
Koji Sato17c76b02009-04-06 19:01:24 -07001742 return ret;
1743 }
1744 }
1745
1746 return 0;
1747}
1748
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001749static void nilfs_btree_commit_update_v(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001750 struct nilfs_btree_path *path,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001751 int level, struct inode *dat)
Koji Sato17c76b02009-04-06 19:01:24 -07001752{
1753 struct nilfs_btree_node *parent;
1754
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001755 nilfs_dat_commit_update(dat, &path[level].bp_oldreq.bpr_req,
1756 &path[level].bp_newreq.bpr_req,
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001757 btree->b_ptr_type == NILFS_BMAP_PTR_VS);
Koji Sato17c76b02009-04-06 19:01:24 -07001758
1759 if (buffer_nilfs_node(path[level].bp_bh)) {
1760 nilfs_btnode_commit_change_key(
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001761 &NILFS_BMAP_I(btree)->i_btnode_cache,
Koji Sato17c76b02009-04-06 19:01:24 -07001762 &path[level].bp_ctxt);
1763 path[level].bp_bh = path[level].bp_ctxt.bh;
1764 }
1765 set_buffer_nilfs_volatile(path[level].bp_bh);
1766
1767 parent = nilfs_btree_get_node(btree, path, level + 1);
1768 nilfs_btree_node_set_ptr(btree, parent, path[level + 1].bp_index,
1769 path[level].bp_newreq.bpr_ptr);
1770}
1771
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001772static void nilfs_btree_abort_update_v(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001773 struct nilfs_btree_path *path,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001774 int level, struct inode *dat)
Koji Sato17c76b02009-04-06 19:01:24 -07001775{
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001776 nilfs_dat_abort_update(dat, &path[level].bp_oldreq.bpr_req,
1777 &path[level].bp_newreq.bpr_req);
Koji Sato17c76b02009-04-06 19:01:24 -07001778 if (buffer_nilfs_node(path[level].bp_bh))
1779 nilfs_btnode_abort_change_key(
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001780 &NILFS_BMAP_I(btree)->i_btnode_cache,
Koji Sato17c76b02009-04-06 19:01:24 -07001781 &path[level].bp_ctxt);
1782}
1783
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001784static int nilfs_btree_prepare_propagate_v(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001785 struct nilfs_btree_path *path,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001786 int minlevel, int *maxlevelp,
1787 struct inode *dat)
Koji Sato17c76b02009-04-06 19:01:24 -07001788{
1789 int level, ret;
1790
1791 level = minlevel;
1792 if (!buffer_nilfs_volatile(path[level].bp_bh)) {
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001793 ret = nilfs_btree_prepare_update_v(btree, path, level, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001794 if (ret < 0)
1795 return ret;
1796 }
1797 while ((++level < nilfs_btree_height(btree) - 1) &&
1798 !buffer_dirty(path[level].bp_bh)) {
1799
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07001800 WARN_ON(buffer_nilfs_volatile(path[level].bp_bh));
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001801 ret = nilfs_btree_prepare_update_v(btree, path, level, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001802 if (ret < 0)
1803 goto out;
1804 }
1805
1806 /* success */
Koji Sato17c76b02009-04-06 19:01:24 -07001807 *maxlevelp = level - 1;
1808 return 0;
1809
1810 /* error */
1811 out:
1812 while (--level > minlevel)
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001813 nilfs_btree_abort_update_v(btree, path, level, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001814 if (!buffer_nilfs_volatile(path[level].bp_bh))
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001815 nilfs_btree_abort_update_v(btree, path, level, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001816 return ret;
1817}
1818
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001819static void nilfs_btree_commit_propagate_v(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001820 struct nilfs_btree_path *path,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001821 int minlevel, int maxlevel,
1822 struct buffer_head *bh,
1823 struct inode *dat)
Koji Sato17c76b02009-04-06 19:01:24 -07001824{
1825 int level;
1826
1827 if (!buffer_nilfs_volatile(path[minlevel].bp_bh))
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001828 nilfs_btree_commit_update_v(btree, path, minlevel, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001829
1830 for (level = minlevel + 1; level <= maxlevel; level++)
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001831 nilfs_btree_commit_update_v(btree, path, level, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001832}
1833
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001834static int nilfs_btree_propagate_v(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001835 struct nilfs_btree_path *path,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001836 int level, struct buffer_head *bh)
Koji Sato17c76b02009-04-06 19:01:24 -07001837{
Li Hong308f4412010-04-02 18:40:39 +08001838 int maxlevel = 0, ret;
Koji Sato17c76b02009-04-06 19:01:24 -07001839 struct nilfs_btree_node *parent;
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001840 struct inode *dat = nilfs_bmap_get_dat(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07001841 __u64 ptr;
1842
1843 get_bh(bh);
1844 path[level].bp_bh = bh;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001845 ret = nilfs_btree_prepare_propagate_v(btree, path, level, &maxlevel,
1846 dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001847 if (ret < 0)
1848 goto out;
1849
1850 if (buffer_nilfs_volatile(path[level].bp_bh)) {
1851 parent = nilfs_btree_get_node(btree, path, level + 1);
1852 ptr = nilfs_btree_node_get_ptr(btree, parent,
1853 path[level + 1].bp_index);
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001854 ret = nilfs_dat_mark_dirty(dat, ptr);
Koji Sato17c76b02009-04-06 19:01:24 -07001855 if (ret < 0)
1856 goto out;
1857 }
1858
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09001859 nilfs_btree_commit_propagate_v(btree, path, level, maxlevel, bh, dat);
Koji Sato17c76b02009-04-06 19:01:24 -07001860
1861 out:
1862 brelse(path[level].bp_bh);
1863 path[level].bp_bh = NULL;
1864 return ret;
1865}
1866
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001867static int nilfs_btree_propagate(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001868 struct buffer_head *bh)
1869{
Koji Sato17c76b02009-04-06 19:01:24 -07001870 struct nilfs_btree_path *path;
1871 struct nilfs_btree_node *node;
1872 __u64 key;
1873 int level, ret;
1874
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07001875 WARN_ON(!buffer_dirty(bh));
Koji Sato17c76b02009-04-06 19:01:24 -07001876
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001877 path = nilfs_btree_alloc_path();
Koji Sato17c76b02009-04-06 19:01:24 -07001878 if (path == NULL)
1879 return -ENOMEM;
Koji Sato17c76b02009-04-06 19:01:24 -07001880
1881 if (buffer_nilfs_node(bh)) {
1882 node = (struct nilfs_btree_node *)bh->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001883 key = nilfs_btree_node_get_key(node, 0);
1884 level = nilfs_btree_node_get_level(node);
Koji Sato17c76b02009-04-06 19:01:24 -07001885 } else {
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001886 key = nilfs_bmap_data_get_key(btree, bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001887 level = NILFS_BTREE_LEVEL_DATA;
1888 }
1889
1890 ret = nilfs_btree_do_lookup(btree, path, key, NULL, level + 1);
1891 if (ret < 0) {
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07001892 if (unlikely(ret == -ENOENT))
Koji Sato17c76b02009-04-06 19:01:24 -07001893 printk(KERN_CRIT "%s: key = %llu, level == %d\n",
1894 __func__, (unsigned long long)key, level);
Koji Sato17c76b02009-04-06 19:01:24 -07001895 goto out;
1896 }
1897
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001898 ret = NILFS_BMAP_USE_VBN(btree) ?
Ryusuke Konishi7cde31d2009-05-24 18:07:59 +09001899 nilfs_btree_propagate_v(btree, path, level, bh) :
1900 nilfs_btree_propagate_p(btree, path, level, bh);
Koji Sato17c76b02009-04-06 19:01:24 -07001901
1902 out:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001903 nilfs_btree_free_path(path);
Koji Sato17c76b02009-04-06 19:01:24 -07001904
1905 return ret;
1906}
1907
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001908static int nilfs_btree_propagate_gc(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001909 struct buffer_head *bh)
1910{
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001911 return nilfs_dat_mark_dirty(nilfs_bmap_get_dat(btree), bh->b_blocknr);
Koji Sato17c76b02009-04-06 19:01:24 -07001912}
1913
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001914static void nilfs_btree_add_dirty_buffer(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001915 struct list_head *lists,
1916 struct buffer_head *bh)
1917{
1918 struct list_head *head;
1919 struct buffer_head *cbh;
1920 struct nilfs_btree_node *node, *cnode;
1921 __u64 key, ckey;
1922 int level;
1923
1924 get_bh(bh);
1925 node = (struct nilfs_btree_node *)bh->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001926 key = nilfs_btree_node_get_key(node, 0);
1927 level = nilfs_btree_node_get_level(node);
Ryusuke Konishicfa913a2010-07-07 17:19:54 +09001928 if (level < NILFS_BTREE_LEVEL_NODE_MIN ||
1929 level >= NILFS_BTREE_LEVEL_MAX) {
1930 dump_stack();
1931 printk(KERN_WARNING
1932 "%s: invalid btree level: %d (key=%llu, ino=%lu, "
1933 "blocknr=%llu)\n",
1934 __func__, level, (unsigned long long)key,
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001935 NILFS_BMAP_I(btree)->vfs_inode.i_ino,
Ryusuke Konishicfa913a2010-07-07 17:19:54 +09001936 (unsigned long long)bh->b_blocknr);
1937 return;
1938 }
1939
Koji Sato17c76b02009-04-06 19:01:24 -07001940 list_for_each(head, &lists[level]) {
1941 cbh = list_entry(head, struct buffer_head, b_assoc_buffers);
1942 cnode = (struct nilfs_btree_node *)cbh->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09001943 ckey = nilfs_btree_node_get_key(cnode, 0);
Koji Sato17c76b02009-04-06 19:01:24 -07001944 if (key < ckey)
1945 break;
1946 }
1947 list_add_tail(&bh->b_assoc_buffers, head);
1948}
1949
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001950static void nilfs_btree_lookup_dirty_buffers(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001951 struct list_head *listp)
1952{
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001953 struct address_space *btcache = &NILFS_BMAP_I(btree)->i_btnode_cache;
Koji Sato17c76b02009-04-06 19:01:24 -07001954 struct list_head lists[NILFS_BTREE_LEVEL_MAX];
1955 struct pagevec pvec;
1956 struct buffer_head *bh, *head;
1957 pgoff_t index = 0;
1958 int level, i;
1959
1960 for (level = NILFS_BTREE_LEVEL_NODE_MIN;
1961 level < NILFS_BTREE_LEVEL_MAX;
1962 level++)
1963 INIT_LIST_HEAD(&lists[level]);
1964
1965 pagevec_init(&pvec, 0);
1966
1967 while (pagevec_lookup_tag(&pvec, btcache, &index, PAGECACHE_TAG_DIRTY,
1968 PAGEVEC_SIZE)) {
1969 for (i = 0; i < pagevec_count(&pvec); i++) {
1970 bh = head = page_buffers(pvec.pages[i]);
1971 do {
1972 if (buffer_dirty(bh))
1973 nilfs_btree_add_dirty_buffer(btree,
1974 lists, bh);
1975 } while ((bh = bh->b_this_page) != head);
1976 }
1977 pagevec_release(&pvec);
1978 cond_resched();
1979 }
1980
1981 for (level = NILFS_BTREE_LEVEL_NODE_MIN;
1982 level < NILFS_BTREE_LEVEL_MAX;
1983 level++)
Ryusuke Konishi0935db72009-11-29 02:39:11 +09001984 list_splice_tail(&lists[level], listp);
Koji Sato17c76b02009-04-06 19:01:24 -07001985}
1986
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09001987static int nilfs_btree_assign_p(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07001988 struct nilfs_btree_path *path,
1989 int level,
1990 struct buffer_head **bh,
1991 sector_t blocknr,
1992 union nilfs_binfo *binfo)
1993{
1994 struct nilfs_btree_node *parent;
1995 __u64 key;
1996 __u64 ptr;
1997 int ret;
1998
1999 parent = nilfs_btree_get_node(btree, path, level + 1);
2000 ptr = nilfs_btree_node_get_ptr(btree, parent,
2001 path[level + 1].bp_index);
2002 if (buffer_nilfs_node(*bh)) {
2003 path[level].bp_ctxt.oldkey = ptr;
2004 path[level].bp_ctxt.newkey = blocknr;
2005 path[level].bp_ctxt.bh = *bh;
2006 ret = nilfs_btnode_prepare_change_key(
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002007 &NILFS_BMAP_I(btree)->i_btnode_cache,
Koji Sato17c76b02009-04-06 19:01:24 -07002008 &path[level].bp_ctxt);
2009 if (ret < 0)
2010 return ret;
2011 nilfs_btnode_commit_change_key(
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002012 &NILFS_BMAP_I(btree)->i_btnode_cache,
Koji Sato17c76b02009-04-06 19:01:24 -07002013 &path[level].bp_ctxt);
2014 *bh = path[level].bp_ctxt.bh;
2015 }
2016
2017 nilfs_btree_node_set_ptr(btree, parent,
2018 path[level + 1].bp_index, blocknr);
2019
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002020 key = nilfs_btree_node_get_key(parent, path[level + 1].bp_index);
Koji Sato17c76b02009-04-06 19:01:24 -07002021 /* on-disk format */
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +09002022 binfo->bi_dat.bi_blkoff = cpu_to_le64(key);
Koji Sato17c76b02009-04-06 19:01:24 -07002023 binfo->bi_dat.bi_level = level;
2024
2025 return 0;
2026}
2027
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002028static int nilfs_btree_assign_v(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07002029 struct nilfs_btree_path *path,
2030 int level,
2031 struct buffer_head **bh,
2032 sector_t blocknr,
2033 union nilfs_binfo *binfo)
2034{
2035 struct nilfs_btree_node *parent;
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002036 struct inode *dat = nilfs_bmap_get_dat(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07002037 __u64 key;
2038 __u64 ptr;
2039 union nilfs_bmap_ptr_req req;
2040 int ret;
2041
2042 parent = nilfs_btree_get_node(btree, path, level + 1);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002043 ptr = nilfs_btree_node_get_ptr(btree, parent, path[level + 1].bp_index);
Koji Sato17c76b02009-04-06 19:01:24 -07002044 req.bpr_ptr = ptr;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09002045 ret = nilfs_dat_prepare_start(dat, &req.bpr_req);
2046 if (ret < 0)
Koji Sato17c76b02009-04-06 19:01:24 -07002047 return ret;
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09002048 nilfs_dat_commit_start(dat, &req.bpr_req, blocknr);
Koji Sato17c76b02009-04-06 19:01:24 -07002049
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002050 key = nilfs_btree_node_get_key(parent, path[level + 1].bp_index);
Koji Sato17c76b02009-04-06 19:01:24 -07002051 /* on-disk format */
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +09002052 binfo->bi_v.bi_vblocknr = cpu_to_le64(ptr);
2053 binfo->bi_v.bi_blkoff = cpu_to_le64(key);
Koji Sato17c76b02009-04-06 19:01:24 -07002054
2055 return 0;
2056}
2057
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002058static int nilfs_btree_assign(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07002059 struct buffer_head **bh,
2060 sector_t blocknr,
2061 union nilfs_binfo *binfo)
2062{
Koji Sato17c76b02009-04-06 19:01:24 -07002063 struct nilfs_btree_path *path;
2064 struct nilfs_btree_node *node;
2065 __u64 key;
2066 int level, ret;
2067
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002068 path = nilfs_btree_alloc_path();
Koji Sato17c76b02009-04-06 19:01:24 -07002069 if (path == NULL)
2070 return -ENOMEM;
Koji Sato17c76b02009-04-06 19:01:24 -07002071
2072 if (buffer_nilfs_node(*bh)) {
2073 node = (struct nilfs_btree_node *)(*bh)->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002074 key = nilfs_btree_node_get_key(node, 0);
2075 level = nilfs_btree_node_get_level(node);
Koji Sato17c76b02009-04-06 19:01:24 -07002076 } else {
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002077 key = nilfs_bmap_data_get_key(btree, *bh);
Koji Sato17c76b02009-04-06 19:01:24 -07002078 level = NILFS_BTREE_LEVEL_DATA;
2079 }
2080
2081 ret = nilfs_btree_do_lookup(btree, path, key, NULL, level + 1);
2082 if (ret < 0) {
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07002083 WARN_ON(ret == -ENOENT);
Koji Sato17c76b02009-04-06 19:01:24 -07002084 goto out;
2085 }
2086
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002087 ret = NILFS_BMAP_USE_VBN(btree) ?
Ryusuke Konishi7cde31d2009-05-24 18:07:59 +09002088 nilfs_btree_assign_v(btree, path, level, bh, blocknr, binfo) :
2089 nilfs_btree_assign_p(btree, path, level, bh, blocknr, binfo);
Koji Sato17c76b02009-04-06 19:01:24 -07002090
2091 out:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002092 nilfs_btree_free_path(path);
Koji Sato17c76b02009-04-06 19:01:24 -07002093
2094 return ret;
2095}
2096
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002097static int nilfs_btree_assign_gc(struct nilfs_bmap *btree,
Koji Sato17c76b02009-04-06 19:01:24 -07002098 struct buffer_head **bh,
2099 sector_t blocknr,
2100 union nilfs_binfo *binfo)
2101{
Koji Sato17c76b02009-04-06 19:01:24 -07002102 struct nilfs_btree_node *node;
2103 __u64 key;
2104 int ret;
2105
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002106 ret = nilfs_dat_move(nilfs_bmap_get_dat(btree), (*bh)->b_blocknr,
Ryusuke Konishi2e0c2c72009-08-15 15:34:33 +09002107 blocknr);
Koji Sato17c76b02009-04-06 19:01:24 -07002108 if (ret < 0)
2109 return ret;
2110
2111 if (buffer_nilfs_node(*bh)) {
2112 node = (struct nilfs_btree_node *)(*bh)->b_data;
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002113 key = nilfs_btree_node_get_key(node, 0);
Koji Sato17c76b02009-04-06 19:01:24 -07002114 } else
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002115 key = nilfs_bmap_data_get_key(btree, *bh);
Koji Sato17c76b02009-04-06 19:01:24 -07002116
2117 /* on-disk format */
2118 binfo->bi_v.bi_vblocknr = cpu_to_le64((*bh)->b_blocknr);
Ryusuke Konishi25b8d7d2010-07-10 16:50:41 +09002119 binfo->bi_v.bi_blkoff = cpu_to_le64(key);
Koji Sato17c76b02009-04-06 19:01:24 -07002120
2121 return 0;
2122}
2123
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002124static int nilfs_btree_mark(struct nilfs_bmap *btree, __u64 key, int level)
Koji Sato17c76b02009-04-06 19:01:24 -07002125{
2126 struct buffer_head *bh;
Koji Sato17c76b02009-04-06 19:01:24 -07002127 struct nilfs_btree_path *path;
2128 __u64 ptr;
2129 int ret;
2130
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002131 path = nilfs_btree_alloc_path();
Koji Sato17c76b02009-04-06 19:01:24 -07002132 if (path == NULL)
2133 return -ENOMEM;
Koji Sato17c76b02009-04-06 19:01:24 -07002134
2135 ret = nilfs_btree_do_lookup(btree, path, key, &ptr, level + 1);
2136 if (ret < 0) {
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07002137 WARN_ON(ret == -ENOENT);
Koji Sato17c76b02009-04-06 19:01:24 -07002138 goto out;
2139 }
Ryusuke Konishif198dbb2009-05-22 01:07:13 +09002140 ret = nilfs_btree_get_block(btree, ptr, &bh);
Koji Sato17c76b02009-04-06 19:01:24 -07002141 if (ret < 0) {
Ryusuke Konishi1f5abe72009-04-06 19:01:55 -07002142 WARN_ON(ret == -ENOENT);
Koji Sato17c76b02009-04-06 19:01:24 -07002143 goto out;
2144 }
2145
2146 if (!buffer_dirty(bh))
2147 nilfs_btnode_mark_dirty(bh);
Ryusuke Konishi087d01b2009-05-22 00:33:13 +09002148 brelse(bh);
Ryusuke Konishie7c274f2010-07-10 19:09:49 +09002149 if (!nilfs_bmap_dirty(btree))
2150 nilfs_bmap_set_dirty(btree);
Koji Sato17c76b02009-04-06 19:01:24 -07002151
2152 out:
Ryusuke Konishi6d28f7e2009-08-15 01:14:10 +09002153 nilfs_btree_free_path(path);
Koji Sato17c76b02009-04-06 19:01:24 -07002154 return ret;
2155}
2156
2157static const struct nilfs_bmap_operations nilfs_btree_ops = {
2158 .bop_lookup = nilfs_btree_lookup,
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +09002159 .bop_lookup_contig = nilfs_btree_lookup_contig,
Koji Sato17c76b02009-04-06 19:01:24 -07002160 .bop_insert = nilfs_btree_insert,
2161 .bop_delete = nilfs_btree_delete,
2162 .bop_clear = NULL,
2163
2164 .bop_propagate = nilfs_btree_propagate,
2165
2166 .bop_lookup_dirty_buffers = nilfs_btree_lookup_dirty_buffers,
2167
2168 .bop_assign = nilfs_btree_assign,
2169 .bop_mark = nilfs_btree_mark,
2170
2171 .bop_last_key = nilfs_btree_last_key,
2172 .bop_check_insert = NULL,
2173 .bop_check_delete = nilfs_btree_check_delete,
2174 .bop_gather_data = nilfs_btree_gather_data,
2175};
2176
2177static const struct nilfs_bmap_operations nilfs_btree_ops_gc = {
2178 .bop_lookup = NULL,
Ryusuke Konishic3a7abf2009-05-25 02:47:14 +09002179 .bop_lookup_contig = NULL,
Koji Sato17c76b02009-04-06 19:01:24 -07002180 .bop_insert = NULL,
2181 .bop_delete = NULL,
2182 .bop_clear = NULL,
2183
2184 .bop_propagate = nilfs_btree_propagate_gc,
2185
2186 .bop_lookup_dirty_buffers = nilfs_btree_lookup_dirty_buffers,
2187
2188 .bop_assign = nilfs_btree_assign_gc,
2189 .bop_mark = NULL,
2190
2191 .bop_last_key = NULL,
2192 .bop_check_insert = NULL,
2193 .bop_check_delete = NULL,
2194 .bop_gather_data = NULL,
2195};
2196
Ryusuke Konishi30333422009-05-24 00:09:44 +09002197int nilfs_btree_init(struct nilfs_bmap *bmap)
Koji Sato17c76b02009-04-06 19:01:24 -07002198{
Koji Sato17c76b02009-04-06 19:01:24 -07002199 bmap->b_ops = &nilfs_btree_ops;
Koji Sato17c76b02009-04-06 19:01:24 -07002200 return 0;
2201}
2202
2203void nilfs_btree_init_gc(struct nilfs_bmap *bmap)
2204{
Koji Sato17c76b02009-04-06 19:01:24 -07002205 bmap->b_ops = &nilfs_btree_ops_gc;
2206}