Merge tag 'arm64-fixes' of git://git.kernel.org/pub/scm/linux/kernel/git/arm64/linux
[linux-2.6-block.git] / fs / reiserfs / fix_node.c
CommitLineData
1da177e4
LT
1/*
2 * Copyright 2000 by Hans Reiser, licensing governed by reiserfs/README
3 */
4
1da177e4 5#include <linux/time.h>
5a0e3ad6 6#include <linux/slab.h>
1da177e4 7#include <linux/string.h>
f466c6fd 8#include "reiserfs.h"
1da177e4
LT
9#include <linux/buffer_head.h>
10
098297b2
JM
11/*
12 * To make any changes in the tree we find a node that contains item
13 * to be changed/deleted or position in the node we insert a new item
14 * to. We call this node S. To do balancing we need to decide what we
15 * will shift to left/right neighbor, or to a new node, where new item
16 * will be etc. To make this analysis simpler we build virtual
17 * node. Virtual node is an array of items, that will replace items of
18 * node S. (For instance if we are going to delete an item, virtual
19 * node does not contain it). Virtual node keeps information about
20 * item sizes and types, mergeability of first and last items, sizes
21 * of all entries in directory item. We use this array of items when
22 * calculating what we can shift to neighbors and how many nodes we
23 * have to have if we do not any shiftings, if we shift to left/right
24 * neighbor or to both.
25 */
26
27/*
28 * Takes item number in virtual node, returns number of item
29 * that it has in source buffer
30 */
bd4c625c 31static inline int old_item_num(int new_num, int affected_item_num, int mode)
1da177e4 32{
bd4c625c
LT
33 if (mode == M_PASTE || mode == M_CUT || new_num < affected_item_num)
34 return new_num;
1da177e4 35
bd4c625c 36 if (mode == M_INSERT) {
1da177e4 37
bd4c625c
LT
38 RFALSE(new_num == 0,
39 "vs-8005: for INSERT mode and item number of inserted item");
1da177e4 40
bd4c625c
LT
41 return new_num - 1;
42 }
1da177e4 43
bd4c625c
LT
44 RFALSE(mode != M_DELETE,
45 "vs-8010: old_item_num: mode must be M_DELETE (mode = \'%c\'",
46 mode);
47 /* delete mode */
48 return new_num + 1;
1da177e4
LT
49}
50
bd4c625c 51static void create_virtual_node(struct tree_balance *tb, int h)
1da177e4 52{
bd4c625c
LT
53 struct item_head *ih;
54 struct virtual_node *vn = tb->tb_vn;
55 int new_num;
56 struct buffer_head *Sh; /* this comes from tb->S[h] */
1da177e4 57
bd4c625c 58 Sh = PATH_H_PBUFFER(tb->tb_path, h);
1da177e4 59
bd4c625c
LT
60 /* size of changed node */
61 vn->vn_size =
62 MAX_CHILD_SIZE(Sh) - B_FREE_SPACE(Sh) + tb->insert_size[h];
1da177e4 63
bd4c625c
LT
64 /* for internal nodes array if virtual items is not created */
65 if (h) {
66 vn->vn_nr_item = (vn->vn_size - DC_SIZE) / (DC_SIZE + KEY_SIZE);
67 return;
1da177e4 68 }
1da177e4 69
bd4c625c
LT
70 /* number of items in virtual node */
71 vn->vn_nr_item =
72 B_NR_ITEMS(Sh) + ((vn->vn_mode == M_INSERT) ? 1 : 0) -
73 ((vn->vn_mode == M_DELETE) ? 1 : 0);
74
75 /* first virtual item */
76 vn->vn_vi = (struct virtual_item *)(tb->tb_vn + 1);
77 memset(vn->vn_vi, 0, vn->vn_nr_item * sizeof(struct virtual_item));
78 vn->vn_free_ptr += vn->vn_nr_item * sizeof(struct virtual_item);
79
80 /* first item in the node */
4cf5f7ad 81 ih = item_head(Sh, 0);
bd4c625c
LT
82
83 /* define the mergeability for 0-th item (if it is not being deleted) */
a228bf8f 84 if (op_is_left_mergeable(&ih->ih_key, Sh->b_size)
bd4c625c
LT
85 && (vn->vn_mode != M_DELETE || vn->vn_affected_item_num))
86 vn->vn_vi[0].vi_type |= VI_TYPE_LEFT_MERGEABLE;
87
098297b2
JM
88 /*
89 * go through all items that remain in the virtual
90 * node (except for the new (inserted) one)
91 */
bd4c625c
LT
92 for (new_num = 0; new_num < vn->vn_nr_item; new_num++) {
93 int j;
94 struct virtual_item *vi = vn->vn_vi + new_num;
95 int is_affected =
96 ((new_num != vn->vn_affected_item_num) ? 0 : 1);
97
98 if (is_affected && vn->vn_mode == M_INSERT)
99 continue;
100
101 /* get item number in source node */
102 j = old_item_num(new_num, vn->vn_affected_item_num,
103 vn->vn_mode);
104
105 vi->vi_item_len += ih_item_len(ih + j) + IH_SIZE;
106 vi->vi_ih = ih + j;
4cf5f7ad 107 vi->vi_item = ih_item_body(Sh, ih + j);
bd4c625c
LT
108 vi->vi_uarea = vn->vn_free_ptr;
109
098297b2
JM
110 /*
111 * FIXME: there is no check that item operation did not
112 * consume too much memory
113 */
bd4c625c
LT
114 vn->vn_free_ptr +=
115 op_create_vi(vn, vi, is_affected, tb->insert_size[0]);
116 if (tb->vn_buf + tb->vn_buf_size < vn->vn_free_ptr)
c3a9c210 117 reiserfs_panic(tb->tb_sb, "vs-8030",
bd4c625c
LT
118 "virtual node space consumed");
119
120 if (!is_affected)
121 /* this is not being changed */
122 continue;
123
124 if (vn->vn_mode == M_PASTE || vn->vn_mode == M_CUT) {
125 vn->vn_vi[new_num].vi_item_len += tb->insert_size[0];
098297b2
JM
126 /* pointer to data which is going to be pasted */
127 vi->vi_new_data = vn->vn_data;
bd4c625c 128 }
1da177e4 129 }
bd4c625c
LT
130
131 /* virtual inserted item is not defined yet */
132 if (vn->vn_mode == M_INSERT) {
133 struct virtual_item *vi = vn->vn_vi + vn->vn_affected_item_num;
134
9dce07f1 135 RFALSE(vn->vn_ins_ih == NULL,
bd4c625c
LT
136 "vs-8040: item header of inserted item is not specified");
137 vi->vi_item_len = tb->insert_size[0];
138 vi->vi_ih = vn->vn_ins_ih;
139 vi->vi_item = vn->vn_data;
140 vi->vi_uarea = vn->vn_free_ptr;
141
142 op_create_vi(vn, vi, 0 /*not pasted or cut */ ,
143 tb->insert_size[0]);
144 }
145
098297b2
JM
146 /*
147 * set right merge flag we take right delimiting key and
148 * check whether it is a mergeable item
149 */
bd4c625c
LT
150 if (tb->CFR[0]) {
151 struct reiserfs_key *key;
152
4cf5f7ad 153 key = internal_key(tb->CFR[0], tb->rkey[0]);
bd4c625c
LT
154 if (op_is_left_mergeable(key, Sh->b_size)
155 && (vn->vn_mode != M_DELETE
156 || vn->vn_affected_item_num != B_NR_ITEMS(Sh) - 1))
157 vn->vn_vi[vn->vn_nr_item - 1].vi_type |=
158 VI_TYPE_RIGHT_MERGEABLE;
159
160#ifdef CONFIG_REISERFS_CHECK
161 if (op_is_left_mergeable(key, Sh->b_size) &&
162 !(vn->vn_mode != M_DELETE
163 || vn->vn_affected_item_num != B_NR_ITEMS(Sh) - 1)) {
098297b2
JM
164 /*
165 * we delete last item and it could be merged
166 * with right neighbor's first item
167 */
bd4c625c
LT
168 if (!
169 (B_NR_ITEMS(Sh) == 1
4cf5f7ad
JM
170 && is_direntry_le_ih(item_head(Sh, 0))
171 && ih_entry_count(item_head(Sh, 0)) == 1)) {
098297b2
JM
172 /*
173 * node contains more than 1 item, or item
174 * is not directory item, or this item
175 * contains more than 1 entry
176 */
bd4c625c 177 print_block(Sh, 0, -1, -1);
c3a9c210
JM
178 reiserfs_panic(tb->tb_sb, "vs-8045",
179 "rdkey %k, affected item==%d "
180 "(mode==%c) Must be %c",
bd4c625c
LT
181 key, vn->vn_affected_item_num,
182 vn->vn_mode, M_DELETE);
cd02b966 183 }
bd4c625c 184 }
1da177e4 185#endif
1da177e4 186
bd4c625c
LT
187 }
188}
1da177e4 189
098297b2
JM
190/*
191 * Using virtual node check, how many items can be
192 * shifted to left neighbor
193 */
bd4c625c 194static void check_left(struct tree_balance *tb, int h, int cur_free)
1da177e4 195{
bd4c625c
LT
196 int i;
197 struct virtual_node *vn = tb->tb_vn;
198 struct virtual_item *vi;
199 int d_size, ih_size;
1da177e4 200
bd4c625c 201 RFALSE(cur_free < 0, "vs-8050: cur_free (%d) < 0", cur_free);
1da177e4 202
bd4c625c
LT
203 /* internal level */
204 if (h > 0) {
205 tb->lnum[h] = cur_free / (DC_SIZE + KEY_SIZE);
206 return;
207 }
1da177e4 208
bd4c625c 209 /* leaf level */
1da177e4 210
bd4c625c
LT
211 if (!cur_free || !vn->vn_nr_item) {
212 /* no free space or nothing to move */
213 tb->lnum[h] = 0;
214 tb->lbytes = -1;
215 return;
216 }
1da177e4 217
bd4c625c
LT
218 RFALSE(!PATH_H_PPARENT(tb->tb_path, 0),
219 "vs-8055: parent does not exist or invalid");
1da177e4 220
bd4c625c
LT
221 vi = vn->vn_vi;
222 if ((unsigned int)cur_free >=
223 (vn->vn_size -
224 ((vi->vi_type & VI_TYPE_LEFT_MERGEABLE) ? IH_SIZE : 0))) {
225 /* all contents of S[0] fits into L[0] */
1da177e4 226
bd4c625c
LT
227 RFALSE(vn->vn_mode == M_INSERT || vn->vn_mode == M_PASTE,
228 "vs-8055: invalid mode or balance condition failed");
1da177e4 229
bd4c625c
LT
230 tb->lnum[0] = vn->vn_nr_item;
231 tb->lbytes = -1;
232 return;
1da177e4 233 }
bd4c625c
LT
234
235 d_size = 0, ih_size = IH_SIZE;
236
237 /* first item may be merge with last item in left neighbor */
238 if (vi->vi_type & VI_TYPE_LEFT_MERGEABLE)
239 d_size = -((int)IH_SIZE), ih_size = 0;
240
241 tb->lnum[0] = 0;
242 for (i = 0; i < vn->vn_nr_item;
243 i++, ih_size = IH_SIZE, d_size = 0, vi++) {
244 d_size += vi->vi_item_len;
245 if (cur_free >= d_size) {
246 /* the item can be shifted entirely */
247 cur_free -= d_size;
248 tb->lnum[0]++;
249 continue;
250 }
251
252 /* the item cannot be shifted entirely, try to split it */
098297b2
JM
253 /*
254 * check whether L[0] can hold ih and at least one byte
255 * of the item body
256 */
257
258 /* cannot shift even a part of the current item */
bd4c625c 259 if (cur_free <= ih_size) {
bd4c625c
LT
260 tb->lbytes = -1;
261 return;
262 }
263 cur_free -= ih_size;
264
265 tb->lbytes = op_check_left(vi, cur_free, 0, 0);
266 if (tb->lbytes != -1)
267 /* count partially shifted item */
268 tb->lnum[0]++;
269
270 break;
1da177e4 271 }
1da177e4 272
bd4c625c
LT
273 return;
274}
1da177e4 275
098297b2
JM
276/*
277 * Using virtual node check, how many items can be
278 * shifted to right neighbor
279 */
bd4c625c 280static void check_right(struct tree_balance *tb, int h, int cur_free)
1da177e4 281{
bd4c625c
LT
282 int i;
283 struct virtual_node *vn = tb->tb_vn;
284 struct virtual_item *vi;
285 int d_size, ih_size;
286
287 RFALSE(cur_free < 0, "vs-8070: cur_free < 0");
288
289 /* internal level */
290 if (h > 0) {
291 tb->rnum[h] = cur_free / (DC_SIZE + KEY_SIZE);
292 return;
1da177e4 293 }
bd4c625c
LT
294
295 /* leaf level */
296
297 if (!cur_free || !vn->vn_nr_item) {
298 /* no free space */
299 tb->rnum[h] = 0;
300 tb->rbytes = -1;
301 return;
1da177e4 302 }
1da177e4 303
bd4c625c
LT
304 RFALSE(!PATH_H_PPARENT(tb->tb_path, 0),
305 "vs-8075: parent does not exist or invalid");
306
307 vi = vn->vn_vi + vn->vn_nr_item - 1;
308 if ((unsigned int)cur_free >=
309 (vn->vn_size -
310 ((vi->vi_type & VI_TYPE_RIGHT_MERGEABLE) ? IH_SIZE : 0))) {
311 /* all contents of S[0] fits into R[0] */
312
313 RFALSE(vn->vn_mode == M_INSERT || vn->vn_mode == M_PASTE,
314 "vs-8080: invalid mode or balance condition failed");
315
316 tb->rnum[h] = vn->vn_nr_item;
317 tb->rbytes = -1;
318 return;
319 }
320
321 d_size = 0, ih_size = IH_SIZE;
322
323 /* last item may be merge with first item in right neighbor */
324 if (vi->vi_type & VI_TYPE_RIGHT_MERGEABLE)
325 d_size = -(int)IH_SIZE, ih_size = 0;
326
327 tb->rnum[0] = 0;
328 for (i = vn->vn_nr_item - 1; i >= 0;
329 i--, d_size = 0, ih_size = IH_SIZE, vi--) {
330 d_size += vi->vi_item_len;
331 if (cur_free >= d_size) {
332 /* the item can be shifted entirely */
333 cur_free -= d_size;
334 tb->rnum[0]++;
335 continue;
336 }
337
098297b2
JM
338 /*
339 * check whether R[0] can hold ih and at least one
340 * byte of the item body
341 */
342
343 /* cannot shift even a part of the current item */
344 if (cur_free <= ih_size) {
bd4c625c
LT
345 tb->rbytes = -1;
346 return;
347 }
348
098297b2
JM
349 /*
350 * R[0] can hold the header of the item and at least
351 * one byte of its body
352 */
bd4c625c
LT
353 cur_free -= ih_size; /* cur_free is still > 0 */
354
355 tb->rbytes = op_check_right(vi, cur_free);
356 if (tb->rbytes != -1)
357 /* count partially shifted item */
358 tb->rnum[0]++;
359
360 break;
361 }
362
363 return;
364}
1da177e4
LT
365
366/*
367 * from - number of items, which are shifted to left neighbor entirely
368 * to - number of item, which are shifted to right neighbor entirely
098297b2
JM
369 * from_bytes - number of bytes of boundary item (or directory entries)
370 * which are shifted to left neighbor
371 * to_bytes - number of bytes of boundary item (or directory entries)
372 * which are shifted to right neighbor
373 */
bd4c625c
LT
374static int get_num_ver(int mode, struct tree_balance *tb, int h,
375 int from, int from_bytes,
376 int to, int to_bytes, short *snum012, int flow)
1da177e4 377{
bd4c625c
LT
378 int i;
379 int cur_free;
bd4c625c
LT
380 int units;
381 struct virtual_node *vn = tb->tb_vn;
bd4c625c
LT
382 int total_node_size, max_node_size, current_item_size;
383 int needed_nodes;
098297b2
JM
384
385 /* position of item we start filling node from */
386 int start_item;
387
388 /* position of item we finish filling node by */
389 int end_item;
390
391 /*
392 * number of first bytes (entries for directory) of start_item-th item
393 * we do not include into node that is being filled
394 */
395 int start_bytes;
396
397 /*
398 * number of last bytes (entries for directory) of end_item-th item
399 * we do node include into node that is being filled
400 */
401 int end_bytes;
402
403 /*
404 * these are positions in virtual item of items, that are split
405 * between S[0] and S1new and S1new and S2new
406 */
407 int split_item_positions[2];
bd4c625c
LT
408
409 split_item_positions[0] = -1;
410 split_item_positions[1] = -1;
411
098297b2
JM
412 /*
413 * We only create additional nodes if we are in insert or paste mode
414 * or we are in replace mode at the internal level. If h is 0 and
415 * the mode is M_REPLACE then in fix_nodes we change the mode to
416 * paste or insert before we get here in the code.
417 */
bd4c625c
LT
418 RFALSE(tb->insert_size[h] < 0 || (mode != M_INSERT && mode != M_PASTE),
419 "vs-8100: insert_size < 0 in overflow");
420
421 max_node_size = MAX_CHILD_SIZE(PATH_H_PBUFFER(tb->tb_path, h));
422
098297b2
JM
423 /*
424 * snum012 [0-2] - number of items, that lay
425 * to S[0], first new node and second new node
426 */
bd4c625c
LT
427 snum012[3] = -1; /* s1bytes */
428 snum012[4] = -1; /* s2bytes */
429
430 /* internal level */
431 if (h > 0) {
432 i = ((to - from) * (KEY_SIZE + DC_SIZE) + DC_SIZE);
433 if (i == max_node_size)
434 return 1;
435 return (i / max_node_size + 1);
1da177e4
LT
436 }
437
bd4c625c
LT
438 /* leaf level */
439 needed_nodes = 1;
440 total_node_size = 0;
441 cur_free = max_node_size;
442
098297b2 443 /* start from 'from'-th item */
bd4c625c 444 start_item = from;
098297b2 445 /* skip its first 'start_bytes' units */
bd4c625c
LT
446 start_bytes = ((from_bytes != -1) ? from_bytes : 0);
447
098297b2 448 /* last included item is the 'end_item'-th one */
bd4c625c 449 end_item = vn->vn_nr_item - to - 1;
098297b2 450 /* do not count last 'end_bytes' units of 'end_item'-th item */
bd4c625c
LT
451 end_bytes = (to_bytes != -1) ? to_bytes : 0;
452
098297b2
JM
453 /*
454 * go through all item beginning from the start_item-th item
455 * and ending by the end_item-th item. Do not count first
456 * 'start_bytes' units of 'start_item'-th item and last
457 * 'end_bytes' of 'end_item'-th item
458 */
bd4c625c
LT
459 for (i = start_item; i <= end_item; i++) {
460 struct virtual_item *vi = vn->vn_vi + i;
461 int skip_from_end = ((i == end_item) ? end_bytes : 0);
462
463 RFALSE(needed_nodes > 3, "vs-8105: too many nodes are needed");
464
465 /* get size of current item */
466 current_item_size = vi->vi_item_len;
467
098297b2
JM
468 /*
469 * do not take in calculation head part (from_bytes)
470 * of from-th item
471 */
bd4c625c
LT
472 current_item_size -=
473 op_part_size(vi, 0 /*from start */ , start_bytes);
474
475 /* do not take in calculation tail part of last item */
476 current_item_size -=
477 op_part_size(vi, 1 /*from end */ , skip_from_end);
478
479 /* if item fits into current node entierly */
480 if (total_node_size + current_item_size <= max_node_size) {
481 snum012[needed_nodes - 1]++;
482 total_node_size += current_item_size;
483 start_bytes = 0;
484 continue;
485 }
486
098297b2
JM
487 /*
488 * virtual item length is longer, than max size of item in
489 * a node. It is impossible for direct item
490 */
bd4c625c 491 if (current_item_size > max_node_size) {
bd4c625c
LT
492 RFALSE(is_direct_le_ih(vi->vi_ih),
493 "vs-8110: "
494 "direct item length is %d. It can not be longer than %d",
495 current_item_size, max_node_size);
496 /* we will try to split it */
497 flow = 1;
498 }
499
098297b2 500 /* as we do not split items, take new node and continue */
bd4c625c 501 if (!flow) {
bd4c625c
LT
502 needed_nodes++;
503 i--;
504 total_node_size = 0;
505 continue;
506 }
098297b2
JM
507
508 /*
509 * calculate number of item units which fit into node being
510 * filled
511 */
bd4c625c
LT
512 {
513 int free_space;
514
515 free_space = max_node_size - total_node_size - IH_SIZE;
516 units =
517 op_check_left(vi, free_space, start_bytes,
518 skip_from_end);
098297b2
JM
519 /*
520 * nothing fits into current node, take new
521 * node and continue
522 */
bd4c625c 523 if (units == -1) {
bd4c625c
LT
524 needed_nodes++, i--, total_node_size = 0;
525 continue;
526 }
527 }
528
529 /* something fits into the current node */
bd4c625c
LT
530 start_bytes += units;
531 snum012[needed_nodes - 1 + 3] = units;
532
533 if (needed_nodes > 2)
45b03d5e
JM
534 reiserfs_warning(tb->tb_sb, "vs-8111",
535 "split_item_position is out of range");
bd4c625c
LT
536 snum012[needed_nodes - 1]++;
537 split_item_positions[needed_nodes - 1] = i;
538 needed_nodes++;
539 /* continue from the same item with start_bytes != -1 */
540 start_item = i;
541 i--;
542 total_node_size = 0;
1da177e4
LT
543 }
544
098297b2
JM
545 /*
546 * sum012[4] (if it is not -1) contains number of units of which
547 * are to be in S1new, snum012[3] - to be in S0. They are supposed
548 * to be S1bytes and S2bytes correspondingly, so recalculate
549 */
bd4c625c
LT
550 if (snum012[4] > 0) {
551 int split_item_num;
552 int bytes_to_r, bytes_to_l;
553 int bytes_to_S1new;
554
555 split_item_num = split_item_positions[1];
556 bytes_to_l =
557 ((from == split_item_num
558 && from_bytes != -1) ? from_bytes : 0);
559 bytes_to_r =
560 ((end_item == split_item_num
561 && end_bytes != -1) ? end_bytes : 0);
562 bytes_to_S1new =
563 ((split_item_positions[0] ==
564 split_item_positions[1]) ? snum012[3] : 0);
565
098297b2 566 /* s2bytes */
bd4c625c
LT
567 snum012[4] =
568 op_unit_num(&vn->vn_vi[split_item_num]) - snum012[4] -
569 bytes_to_r - bytes_to_l - bytes_to_S1new;
570
571 if (vn->vn_vi[split_item_num].vi_index != TYPE_DIRENTRY &&
572 vn->vn_vi[split_item_num].vi_index != TYPE_INDIRECT)
45b03d5e
JM
573 reiserfs_warning(tb->tb_sb, "vs-8115",
574 "not directory or indirect item");
1da177e4
LT
575 }
576
bd4c625c
LT
577 /* now we know S2bytes, calculate S1bytes */
578 if (snum012[3] > 0) {
579 int split_item_num;
580 int bytes_to_r, bytes_to_l;
581 int bytes_to_S2new;
582
583 split_item_num = split_item_positions[0];
584 bytes_to_l =
585 ((from == split_item_num
586 && from_bytes != -1) ? from_bytes : 0);
587 bytes_to_r =
588 ((end_item == split_item_num
589 && end_bytes != -1) ? end_bytes : 0);
590 bytes_to_S2new =
591 ((split_item_positions[0] == split_item_positions[1]
592 && snum012[4] != -1) ? snum012[4] : 0);
593
098297b2 594 /* s1bytes */
bd4c625c
LT
595 snum012[3] =
596 op_unit_num(&vn->vn_vi[split_item_num]) - snum012[3] -
597 bytes_to_r - bytes_to_l - bytes_to_S2new;
1da177e4
LT
598 }
599
bd4c625c 600 return needed_nodes;
1da177e4
LT
601}
602
1da177e4 603
098297b2
JM
604/*
605 * Set parameters for balancing.
1da177e4 606 * Performs write of results of analysis of balancing into structure tb,
0222e657 607 * where it will later be used by the functions that actually do the balancing.
1da177e4
LT
608 * Parameters:
609 * tb tree_balance structure;
610 * h current level of the node;
611 * lnum number of items from S[h] that must be shifted to L[h];
612 * rnum number of items from S[h] that must be shifted to R[h];
613 * blk_num number of blocks that S[h] will be splitted into;
614 * s012 number of items that fall into splitted nodes.
098297b2
JM
615 * lbytes number of bytes which flow to the left neighbor from the
616 * item that is not not shifted entirely
617 * rbytes number of bytes which flow to the right neighbor from the
618 * item that is not not shifted entirely
619 * s1bytes number of bytes which flow to the first new node when
620 * S[0] splits (this number is contained in s012 array)
1da177e4
LT
621 */
622
bd4c625c
LT
623static void set_parameters(struct tree_balance *tb, int h, int lnum,
624 int rnum, int blk_num, short *s012, int lb, int rb)
1da177e4
LT
625{
626
bd4c625c
LT
627 tb->lnum[h] = lnum;
628 tb->rnum[h] = rnum;
629 tb->blknum[h] = blk_num;
1da177e4 630
098297b2
JM
631 /* only for leaf level */
632 if (h == 0) {
bd4c625c 633 if (s012 != NULL) {
b49fb112
JM
634 tb->s0num = *s012++;
635 tb->snum[0] = *s012++;
636 tb->snum[1] = *s012++;
637 tb->sbytes[0] = *s012++;
638 tb->sbytes[1] = *s012;
bd4c625c
LT
639 }
640 tb->lbytes = lb;
641 tb->rbytes = rb;
1da177e4 642 }
bd4c625c
LT
643 PROC_INFO_ADD(tb->tb_sb, lnum[h], lnum);
644 PROC_INFO_ADD(tb->tb_sb, rnum[h], rnum);
1da177e4 645
bd4c625c
LT
646 PROC_INFO_ADD(tb->tb_sb, lbytes[h], lb);
647 PROC_INFO_ADD(tb->tb_sb, rbytes[h], rb);
648}
1da177e4 649
098297b2
JM
650/*
651 * check if node disappears if we shift tb->lnum[0] items to left
652 * neighbor and tb->rnum[0] to the right one.
653 */
bd4c625c 654static int is_leaf_removable(struct tree_balance *tb)
1da177e4 655{
bd4c625c
LT
656 struct virtual_node *vn = tb->tb_vn;
657 int to_left, to_right;
658 int size;
659 int remain_items;
660
098297b2
JM
661 /*
662 * number of items that will be shifted to left (right) neighbor
663 * entirely
664 */
bd4c625c
LT
665 to_left = tb->lnum[0] - ((tb->lbytes != -1) ? 1 : 0);
666 to_right = tb->rnum[0] - ((tb->rbytes != -1) ? 1 : 0);
667 remain_items = vn->vn_nr_item;
668
669 /* how many items remain in S[0] after shiftings to neighbors */
670 remain_items -= (to_left + to_right);
671
098297b2 672 /* all content of node can be shifted to neighbors */
bd4c625c 673 if (remain_items < 1) {
bd4c625c
LT
674 set_parameters(tb, 0, to_left, vn->vn_nr_item - to_left, 0,
675 NULL, -1, -1);
676 return 1;
677 }
1da177e4 678
098297b2 679 /* S[0] is not removable */
bd4c625c 680 if (remain_items > 1 || tb->lbytes == -1 || tb->rbytes == -1)
bd4c625c
LT
681 return 0;
682
098297b2 683 /* check whether we can divide 1 remaining item between neighbors */
bd4c625c
LT
684
685 /* get size of remaining item (in item units) */
a228bf8f 686 size = op_unit_num(&vn->vn_vi[to_left]);
bd4c625c
LT
687
688 if (tb->lbytes + tb->rbytes >= size) {
689 set_parameters(tb, 0, to_left + 1, to_right + 1, 0, NULL,
690 tb->lbytes, -1);
691 return 1;
692 }
693
694 return 0;
695}
1da177e4
LT
696
697/* check whether L, S, R can be joined in one node */
bd4c625c 698static int are_leaves_removable(struct tree_balance *tb, int lfree, int rfree)
1da177e4 699{
bd4c625c
LT
700 struct virtual_node *vn = tb->tb_vn;
701 int ih_size;
702 struct buffer_head *S0;
703
704 S0 = PATH_H_PBUFFER(tb->tb_path, 0);
705
706 ih_size = 0;
707 if (vn->vn_nr_item) {
708 if (vn->vn_vi[0].vi_type & VI_TYPE_LEFT_MERGEABLE)
709 ih_size += IH_SIZE;
710
711 if (vn->vn_vi[vn->vn_nr_item - 1].
712 vi_type & VI_TYPE_RIGHT_MERGEABLE)
713 ih_size += IH_SIZE;
714 } else {
715 /* there was only one item and it will be deleted */
716 struct item_head *ih;
717
718 RFALSE(B_NR_ITEMS(S0) != 1,
719 "vs-8125: item number must be 1: it is %d",
720 B_NR_ITEMS(S0));
721
4cf5f7ad 722 ih = item_head(S0, 0);
bd4c625c 723 if (tb->CFR[0]
a228bf8f 724 && !comp_short_le_keys(&ih->ih_key,
4cf5f7ad 725 internal_key(tb->CFR[0],
bd4c625c 726 tb->rkey[0])))
098297b2
JM
727 /*
728 * Directory must be in correct state here: that is
729 * somewhere at the left side should exist first
730 * directory item. But the item being deleted can
731 * not be that first one because its right neighbor
732 * is item of the same directory. (But first item
733 * always gets deleted in last turn). So, neighbors
734 * of deleted item can be merged, so we can save
735 * ih_size
736 */
bd4c625c 737 if (is_direntry_le_ih(ih)) {
bd4c625c
LT
738 ih_size = IH_SIZE;
739
098297b2
JM
740 /*
741 * we might check that left neighbor exists
742 * and is of the same directory
743 */
bd4c625c
LT
744 RFALSE(le_ih_k_offset(ih) == DOT_OFFSET,
745 "vs-8130: first directory item can not be removed until directory is not empty");
746 }
1da177e4 747
bd4c625c
LT
748 }
749
750 if (MAX_CHILD_SIZE(S0) + vn->vn_size <= rfree + lfree + ih_size) {
751 set_parameters(tb, 0, -1, -1, -1, NULL, -1, -1);
752 PROC_INFO_INC(tb->tb_sb, leaves_removable);
753 return 1;
754 }
755 return 0;
1da177e4 756
bd4c625c 757}
1da177e4
LT
758
759/* when we do not split item, lnum and rnum are numbers of entire items */
760#define SET_PAR_SHIFT_LEFT \
761if (h)\
762{\
763 int to_l;\
764 \
765 to_l = (MAX_NR_KEY(Sh)+1 - lpar + vn->vn_nr_item + 1) / 2 -\
766 (MAX_NR_KEY(Sh) + 1 - lpar);\
767 \
768 set_parameters (tb, h, to_l, 0, lnver, NULL, -1, -1);\
769}\
770else \
771{\
772 if (lset==LEFT_SHIFT_FLOW)\
773 set_parameters (tb, h, lpar, 0, lnver, snum012+lset,\
774 tb->lbytes, -1);\
775 else\
776 set_parameters (tb, h, lpar - (tb->lbytes!=-1), 0, lnver, snum012+lset,\
777 -1, -1);\
778}
779
1da177e4
LT
780#define SET_PAR_SHIFT_RIGHT \
781if (h)\
782{\
783 int to_r;\
784 \
785 to_r = (MAX_NR_KEY(Sh)+1 - rpar + vn->vn_nr_item + 1) / 2 - (MAX_NR_KEY(Sh) + 1 - rpar);\
786 \
787 set_parameters (tb, h, 0, to_r, rnver, NULL, -1, -1);\
788}\
789else \
790{\
791 if (rset==RIGHT_SHIFT_FLOW)\
792 set_parameters (tb, h, 0, rpar, rnver, snum012+rset,\
793 -1, tb->rbytes);\
794 else\
795 set_parameters (tb, h, 0, rpar - (tb->rbytes!=-1), rnver, snum012+rset,\
796 -1, -1);\
797}
798
a063ae17 799static void free_buffers_in_tb(struct tree_balance *tb)
bd4c625c 800{
ee93961b 801 int i;
bd4c625c 802
a063ae17 803 pathrelse(tb->tb_path);
bd4c625c 804
ee93961b
JM
805 for (i = 0; i < MAX_HEIGHT; i++) {
806 brelse(tb->L[i]);
807 brelse(tb->R[i]);
808 brelse(tb->FL[i]);
809 brelse(tb->FR[i]);
810 brelse(tb->CFL[i]);
811 brelse(tb->CFR[i]);
812
813 tb->L[i] = NULL;
814 tb->R[i] = NULL;
815 tb->FL[i] = NULL;
816 tb->FR[i] = NULL;
817 tb->CFL[i] = NULL;
818 tb->CFR[i] = NULL;
bd4c625c 819 }
1da177e4
LT
820}
821
098297b2
JM
822/*
823 * Get new buffers for storing new nodes that are created while balancing.
1da177e4
LT
824 * Returns: SCHEDULE_OCCURRED - schedule occurred while the function worked;
825 * CARRY_ON - schedule didn't occur while the function worked;
826 * NO_DISK_SPACE - no disk space.
827 */
828/* The function is NOT SCHEDULE-SAFE! */
ee93961b 829static int get_empty_nodes(struct tree_balance *tb, int h)
bd4c625c 830{
098297b2 831 struct buffer_head *new_bh, *Sh = PATH_H_PBUFFER(tb->tb_path, h);
ee93961b 832 b_blocknr_t *blocknr, blocknrs[MAX_AMOUNT_NEEDED] = { 0, };
098297b2
JM
833 int counter, number_of_freeblk;
834 int amount_needed; /* number of needed empty blocks */
835 int retval = CARRY_ON;
a063ae17 836 struct super_block *sb = tb->tb_sb;
bd4c625c 837
098297b2
JM
838 /*
839 * number_of_freeblk is the number of empty blocks which have been
840 * acquired for use by the balancing algorithm minus the number of
841 * empty blocks used in the previous levels of the analysis,
842 * number_of_freeblk = tb->cur_blknum can be non-zero if a schedule
843 * occurs after empty blocks are acquired, and the balancing analysis
844 * is then restarted, amount_needed is the number needed by this
845 * level (h) of the balancing analysis.
846 *
847 * Note that for systems with many processes writing, it would be
848 * more layout optimal to calculate the total number needed by all
849 * levels and then to run reiserfs_new_blocks to get all of them at
850 * once.
851 */
852
853 /*
854 * Initiate number_of_freeblk to the amount acquired prior to the
855 * restart of the analysis or 0 if not restarted, then subtract the
856 * amount needed by all of the levels of the tree below h.
857 */
ee93961b
JM
858 /* blknum includes S[h], so we subtract 1 in this calculation */
859 for (counter = 0, number_of_freeblk = tb->cur_blknum;
860 counter < h; counter++)
861 number_of_freeblk -=
862 (tb->blknum[counter]) ? (tb->blknum[counter] -
bd4c625c
LT
863 1) : 0;
864
865 /* Allocate missing empty blocks. */
d68caa95 866 /* if Sh == 0 then we are getting a new root */
ee93961b 867 amount_needed = (Sh) ? (tb->blknum[h] - 1) : 1;
098297b2
JM
868 /*
869 * Amount_needed = the amount that we need more than the
870 * amount that we have.
871 */
ee93961b
JM
872 if (amount_needed > number_of_freeblk)
873 amount_needed -= number_of_freeblk;
098297b2 874 else /* If we have enough already then there is nothing to do. */
bd4c625c
LT
875 return CARRY_ON;
876
098297b2
JM
877 /*
878 * No need to check quota - is not allocated for blocks used
879 * for formatted nodes
880 */
ee93961b
JM
881 if (reiserfs_new_form_blocknrs(tb, blocknrs,
882 amount_needed) == NO_DISK_SPACE)
bd4c625c
LT
883 return NO_DISK_SPACE;
884
885 /* for each blocknumber we just got, get a buffer and stick it on FEB */
ee93961b
JM
886 for (blocknr = blocknrs, counter = 0;
887 counter < amount_needed; blocknr++, counter++) {
bd4c625c 888
d68caa95 889 RFALSE(!*blocknr,
bd4c625c
LT
890 "PAP-8135: reiserfs_new_blocknrs failed when got new blocks");
891
d68caa95
JM
892 new_bh = sb_getblk(sb, *blocknr);
893 RFALSE(buffer_dirty(new_bh) ||
894 buffer_journaled(new_bh) ||
895 buffer_journal_dirty(new_bh),
febe29d9 896 "PAP-8140: journaled or dirty buffer %b for the new block",
d68caa95 897 new_bh);
bd4c625c
LT
898
899 /* Put empty buffers into the array. */
a063ae17 900 RFALSE(tb->FEB[tb->cur_blknum],
bd4c625c
LT
901 "PAP-8141: busy slot for new buffer");
902
d68caa95
JM
903 set_buffer_journal_new(new_bh);
904 tb->FEB[tb->cur_blknum++] = new_bh;
bd4c625c
LT
905 }
906
ee93961b
JM
907 if (retval == CARRY_ON && FILESYSTEM_CHANGED_TB(tb))
908 retval = REPEAT_SEARCH;
1da177e4 909
ee93961b 910 return retval;
bd4c625c 911}
1da177e4 912
098297b2
JM
913/*
914 * Get free space of the left neighbor, which is stored in the parent
915 * node of the left neighbor.
916 */
bd4c625c 917static int get_lfree(struct tree_balance *tb, int h)
1da177e4 918{
bd4c625c
LT
919 struct buffer_head *l, *f;
920 int order;
1da177e4 921
9dce07f1
AV
922 if ((f = PATH_H_PPARENT(tb->tb_path, h)) == NULL ||
923 (l = tb->FL[h]) == NULL)
bd4c625c 924 return 0;
1da177e4 925
bd4c625c
LT
926 if (f == l)
927 order = PATH_H_B_ITEM_ORDER(tb->tb_path, h) - 1;
928 else {
929 order = B_NR_ITEMS(l);
930 f = l;
931 }
1da177e4 932
bd4c625c 933 return (MAX_CHILD_SIZE(f) - dc_size(B_N_CHILD(f, order)));
1da177e4
LT
934}
935
098297b2
JM
936/*
937 * Get free space of the right neighbor,
1da177e4
LT
938 * which is stored in the parent node of the right neighbor.
939 */
bd4c625c 940static int get_rfree(struct tree_balance *tb, int h)
1da177e4 941{
bd4c625c
LT
942 struct buffer_head *r, *f;
943 int order;
1da177e4 944
9dce07f1
AV
945 if ((f = PATH_H_PPARENT(tb->tb_path, h)) == NULL ||
946 (r = tb->FR[h]) == NULL)
bd4c625c 947 return 0;
1da177e4 948
bd4c625c
LT
949 if (f == r)
950 order = PATH_H_B_ITEM_ORDER(tb->tb_path, h) + 1;
951 else {
952 order = 0;
953 f = r;
954 }
1da177e4 955
bd4c625c 956 return (MAX_CHILD_SIZE(f) - dc_size(B_N_CHILD(f, order)));
1da177e4
LT
957
958}
959
1da177e4 960/* Check whether left neighbor is in memory. */
ee93961b 961static int is_left_neighbor_in_cache(struct tree_balance *tb, int h)
bd4c625c 962{
d68caa95 963 struct buffer_head *father, *left;
a063ae17 964 struct super_block *sb = tb->tb_sb;
ee93961b
JM
965 b_blocknr_t left_neighbor_blocknr;
966 int left_neighbor_position;
bd4c625c 967
a063ae17 968 /* Father of the left neighbor does not exist. */
ee93961b 969 if (!tb->FL[h])
bd4c625c
LT
970 return 0;
971
972 /* Calculate father of the node to be balanced. */
ee93961b 973 father = PATH_H_PBUFFER(tb->tb_path, h + 1);
bd4c625c 974
d68caa95
JM
975 RFALSE(!father ||
976 !B_IS_IN_TREE(father) ||
ee93961b 977 !B_IS_IN_TREE(tb->FL[h]) ||
d68caa95 978 !buffer_uptodate(father) ||
ee93961b 979 !buffer_uptodate(tb->FL[h]),
bd4c625c 980 "vs-8165: F[h] (%b) or FL[h] (%b) is invalid",
ee93961b 981 father, tb->FL[h]);
bd4c625c 982
098297b2
JM
983 /*
984 * Get position of the pointer to the left neighbor
985 * into the left father.
986 */
ee93961b
JM
987 left_neighbor_position = (father == tb->FL[h]) ?
988 tb->lkey[h] : B_NR_ITEMS(tb->FL[h]);
bd4c625c 989 /* Get left neighbor block number. */
ee93961b
JM
990 left_neighbor_blocknr =
991 B_N_CHILD_NUM(tb->FL[h], left_neighbor_position);
bd4c625c 992 /* Look for the left neighbor in the cache. */
ee93961b 993 if ((left = sb_find_get_block(sb, left_neighbor_blocknr))) {
bd4c625c
LT
994
995 RFALSE(buffer_uptodate(left) && !B_IS_IN_TREE(left),
996 "vs-8170: left neighbor (%b %z) is not in the tree",
997 left, left);
998 put_bh(left);
999 return 1;
1000 }
1da177e4 1001
bd4c625c
LT
1002 return 0;
1003}
1da177e4
LT
1004
1005#define LEFT_PARENTS 'l'
1006#define RIGHT_PARENTS 'r'
1007
d68caa95 1008static void decrement_key(struct cpu_key *key)
1da177e4 1009{
098297b2 1010 /* call item specific function for this key */
d68caa95 1011 item_ops[cpu_key_k_type(key)]->decrement_key(key);
1da177e4
LT
1012}
1013
098297b2
JM
1014/*
1015 * Calculate far left/right parent of the left/right neighbor of the
1016 * current node, that is calculate the left/right (FL[h]/FR[h]) neighbor
1017 * of the parent F[h].
1da177e4
LT
1018 * Calculate left/right common parent of the current node and L[h]/R[h].
1019 * Calculate left/right delimiting key position.
098297b2
JM
1020 * Returns: PATH_INCORRECT - path in the tree is not correct
1021 * SCHEDULE_OCCURRED - schedule occurred while the function worked
1022 * CARRY_ON - schedule didn't occur while the function
1023 * worked
1da177e4 1024 */
a063ae17 1025static int get_far_parent(struct tree_balance *tb,
ee93961b 1026 int h,
d68caa95
JM
1027 struct buffer_head **pfather,
1028 struct buffer_head **pcom_father, char c_lr_par)
1da177e4 1029{
d68caa95 1030 struct buffer_head *parent;
bd4c625c 1031 INITIALIZE_PATH(s_path_to_neighbor_father);
d68caa95 1032 struct treepath *path = tb->tb_path;
bd4c625c 1033 struct cpu_key s_lr_father_key;
ee93961b
JM
1034 int counter,
1035 position = INT_MAX,
1036 first_last_position = 0,
1037 path_offset = PATH_H_PATH_OFFSET(path, h);
bd4c625c 1038
098297b2
JM
1039 /*
1040 * Starting from F[h] go upwards in the tree, and look for the common
1041 * ancestor of F[h], and its neighbor l/r, that should be obtained.
1042 */
bd4c625c 1043
ee93961b 1044 counter = path_offset;
bd4c625c 1045
ee93961b 1046 RFALSE(counter < FIRST_PATH_ELEMENT_OFFSET,
bd4c625c
LT
1047 "PAP-8180: invalid path length");
1048
ee93961b 1049 for (; counter > FIRST_PATH_ELEMENT_OFFSET; counter--) {
098297b2
JM
1050 /*
1051 * Check whether parent of the current buffer in the path
1052 * is really parent in the tree.
1053 */
bd4c625c 1054 if (!B_IS_IN_TREE
ee93961b 1055 (parent = PATH_OFFSET_PBUFFER(path, counter - 1)))
bd4c625c 1056 return REPEAT_SEARCH;
098297b2 1057
bd4c625c 1058 /* Check whether position in the parent is correct. */
ee93961b 1059 if ((position =
d68caa95 1060 PATH_OFFSET_POSITION(path,
ee93961b 1061 counter - 1)) >
d68caa95 1062 B_NR_ITEMS(parent))
bd4c625c 1063 return REPEAT_SEARCH;
098297b2
JM
1064
1065 /*
1066 * Check whether parent at the path really points
1067 * to the child.
1068 */
ee93961b
JM
1069 if (B_N_CHILD_NUM(parent, position) !=
1070 PATH_OFFSET_PBUFFER(path, counter)->b_blocknr)
bd4c625c 1071 return REPEAT_SEARCH;
098297b2
JM
1072
1073 /*
1074 * Return delimiting key if position in the parent is not
1075 * equal to first/last one.
1076 */
bd4c625c 1077 if (c_lr_par == RIGHT_PARENTS)
ee93961b
JM
1078 first_last_position = B_NR_ITEMS(parent);
1079 if (position != first_last_position) {
d68caa95
JM
1080 *pcom_father = parent;
1081 get_bh(*pcom_father);
1082 /*(*pcom_father = parent)->b_count++; */
bd4c625c
LT
1083 break;
1084 }
1da177e4 1085 }
bd4c625c
LT
1086
1087 /* if we are in the root of the tree, then there is no common father */
ee93961b 1088 if (counter == FIRST_PATH_ELEMENT_OFFSET) {
098297b2
JM
1089 /*
1090 * Check whether first buffer in the path is the
1091 * root of the tree.
1092 */
bd4c625c 1093 if (PATH_OFFSET_PBUFFER
a063ae17 1094 (tb->tb_path,
bd4c625c 1095 FIRST_PATH_ELEMENT_OFFSET)->b_blocknr ==
a063ae17 1096 SB_ROOT_BLOCK(tb->tb_sb)) {
d68caa95 1097 *pfather = *pcom_father = NULL;
bd4c625c
LT
1098 return CARRY_ON;
1099 }
1100 return REPEAT_SEARCH;
1da177e4 1101 }
1da177e4 1102
d68caa95 1103 RFALSE(B_LEVEL(*pcom_father) <= DISK_LEAF_NODE_LEVEL,
bd4c625c 1104 "PAP-8185: (%b %z) level too small",
d68caa95 1105 *pcom_father, *pcom_father);
1da177e4 1106
bd4c625c 1107 /* Check whether the common parent is locked. */
1da177e4 1108
d68caa95 1109 if (buffer_locked(*pcom_father)) {
8ebc4232
FW
1110
1111 /* Release the write lock while the buffer is busy */
278f6679 1112 int depth = reiserfs_write_unlock_nested(tb->tb_sb);
d68caa95 1113 __wait_on_buffer(*pcom_father);
278f6679 1114 reiserfs_write_lock_nested(tb->tb_sb, depth);
a063ae17 1115 if (FILESYSTEM_CHANGED_TB(tb)) {
d68caa95 1116 brelse(*pcom_father);
bd4c625c
LT
1117 return REPEAT_SEARCH;
1118 }
1da177e4 1119 }
1da177e4 1120
098297b2
JM
1121 /*
1122 * So, we got common parent of the current node and its
1123 * left/right neighbor. Now we are getting the parent of the
1124 * left/right neighbor.
1125 */
1da177e4 1126
bd4c625c
LT
1127 /* Form key to get parent of the left/right neighbor. */
1128 le_key2cpu_key(&s_lr_father_key,
4cf5f7ad 1129 internal_key(*pcom_father,
bd4c625c 1130 (c_lr_par ==
ee93961b
JM
1131 LEFT_PARENTS) ? (tb->lkey[h - 1] =
1132 position -
1133 1) : (tb->rkey[h -
bd4c625c 1134 1] =
ee93961b 1135 position)));
1da177e4 1136
bd4c625c
LT
1137 if (c_lr_par == LEFT_PARENTS)
1138 decrement_key(&s_lr_father_key);
1da177e4 1139
bd4c625c 1140 if (search_by_key
a063ae17 1141 (tb->tb_sb, &s_lr_father_key, &s_path_to_neighbor_father,
ee93961b 1142 h + 1) == IO_ERROR)
098297b2 1143 /* path is released */
bd4c625c 1144 return IO_ERROR;
1da177e4 1145
a063ae17 1146 if (FILESYSTEM_CHANGED_TB(tb)) {
3cd6dbe6 1147 pathrelse(&s_path_to_neighbor_father);
d68caa95 1148 brelse(*pcom_father);
bd4c625c
LT
1149 return REPEAT_SEARCH;
1150 }
1da177e4 1151
d68caa95 1152 *pfather = PATH_PLAST_BUFFER(&s_path_to_neighbor_father);
1da177e4 1153
ee93961b 1154 RFALSE(B_LEVEL(*pfather) != h + 1,
d68caa95 1155 "PAP-8190: (%b %z) level too small", *pfather, *pfather);
bd4c625c
LT
1156 RFALSE(s_path_to_neighbor_father.path_length <
1157 FIRST_PATH_ELEMENT_OFFSET, "PAP-8192: path length is too small");
1da177e4 1158
bd4c625c 1159 s_path_to_neighbor_father.path_length--;
3cd6dbe6 1160 pathrelse(&s_path_to_neighbor_father);
bd4c625c 1161 return CARRY_ON;
1da177e4
LT
1162}
1163
098297b2
JM
1164/*
1165 * Get parents of neighbors of node in the path(S[path_offset]) and
1166 * common parents of S[path_offset] and L[path_offset]/R[path_offset]:
1167 * F[path_offset], FL[path_offset], FR[path_offset], CFL[path_offset],
1168 * CFR[path_offset].
1169 * Calculate numbers of left and right delimiting keys position:
1170 * lkey[path_offset], rkey[path_offset].
1171 * Returns: SCHEDULE_OCCURRED - schedule occurred while the function worked
1172 * CARRY_ON - schedule didn't occur while the function worked
1da177e4 1173 */
ee93961b 1174static int get_parents(struct tree_balance *tb, int h)
1da177e4 1175{
d68caa95 1176 struct treepath *path = tb->tb_path;
ee93961b
JM
1177 int position,
1178 ret,
1179 path_offset = PATH_H_PATH_OFFSET(tb->tb_path, h);
d68caa95 1180 struct buffer_head *curf, *curcf;
bd4c625c
LT
1181
1182 /* Current node is the root of the tree or will be root of the tree */
ee93961b 1183 if (path_offset <= FIRST_PATH_ELEMENT_OFFSET) {
098297b2
JM
1184 /*
1185 * The root can not have parents.
1186 * Release nodes which previously were obtained as
1187 * parents of the current node neighbors.
1188 */
ee93961b
JM
1189 brelse(tb->FL[h]);
1190 brelse(tb->CFL[h]);
1191 brelse(tb->FR[h]);
1192 brelse(tb->CFR[h]);
1193 tb->FL[h] = NULL;
1194 tb->CFL[h] = NULL;
1195 tb->FR[h] = NULL;
1196 tb->CFR[h] = NULL;
bd4c625c
LT
1197 return CARRY_ON;
1198 }
1199
ee93961b
JM
1200 /* Get parent FL[path_offset] of L[path_offset]. */
1201 position = PATH_OFFSET_POSITION(path, path_offset - 1);
1202 if (position) {
bd4c625c 1203 /* Current node is not the first child of its parent. */
ee93961b
JM
1204 curf = PATH_OFFSET_PBUFFER(path, path_offset - 1);
1205 curcf = PATH_OFFSET_PBUFFER(path, path_offset - 1);
d68caa95
JM
1206 get_bh(curf);
1207 get_bh(curf);
ee93961b 1208 tb->lkey[h] = position - 1;
bd4c625c 1209 } else {
098297b2
JM
1210 /*
1211 * Calculate current parent of L[path_offset], which is the
1212 * left neighbor of the current node. Calculate current
1213 * common parent of L[path_offset] and the current node.
1214 * Note that CFL[path_offset] not equal FL[path_offset] and
1215 * CFL[path_offset] not equal F[path_offset].
1216 * Calculate lkey[path_offset].
1217 */
ee93961b 1218 if ((ret = get_far_parent(tb, h + 1, &curf,
d68caa95 1219 &curcf,
bd4c625c 1220 LEFT_PARENTS)) != CARRY_ON)
ee93961b 1221 return ret;
bd4c625c
LT
1222 }
1223
ee93961b
JM
1224 brelse(tb->FL[h]);
1225 tb->FL[h] = curf; /* New initialization of FL[h]. */
1226 brelse(tb->CFL[h]);
1227 tb->CFL[h] = curcf; /* New initialization of CFL[h]. */
bd4c625c 1228
d68caa95
JM
1229 RFALSE((curf && !B_IS_IN_TREE(curf)) ||
1230 (curcf && !B_IS_IN_TREE(curcf)),
1231 "PAP-8195: FL (%b) or CFL (%b) is invalid", curf, curcf);
1da177e4 1232
098297b2 1233 /* Get parent FR[h] of R[h]. */
1da177e4 1234
098297b2 1235 /* Current node is the last child of F[h]. FR[h] != F[h]. */
ee93961b 1236 if (position == B_NR_ITEMS(PATH_H_PBUFFER(path, h + 1))) {
098297b2
JM
1237 /*
1238 * Calculate current parent of R[h], which is the right
1239 * neighbor of F[h]. Calculate current common parent of
1240 * R[h] and current node. Note that CFR[h] not equal
1241 * FR[path_offset] and CFR[h] not equal F[h].
1242 */
ee93961b
JM
1243 if ((ret =
1244 get_far_parent(tb, h + 1, &curf, &curcf,
bd4c625c 1245 RIGHT_PARENTS)) != CARRY_ON)
ee93961b 1246 return ret;
bd4c625c 1247 } else {
098297b2 1248 /* Current node is not the last child of its parent F[h]. */
ee93961b
JM
1249 curf = PATH_OFFSET_PBUFFER(path, path_offset - 1);
1250 curcf = PATH_OFFSET_PBUFFER(path, path_offset - 1);
d68caa95
JM
1251 get_bh(curf);
1252 get_bh(curf);
ee93961b 1253 tb->rkey[h] = position;
bd4c625c 1254 }
1da177e4 1255
ee93961b
JM
1256 brelse(tb->FR[h]);
1257 /* New initialization of FR[path_offset]. */
1258 tb->FR[h] = curf;
bd4c625c 1259
ee93961b
JM
1260 brelse(tb->CFR[h]);
1261 /* New initialization of CFR[path_offset]. */
1262 tb->CFR[h] = curcf;
bd4c625c 1263
d68caa95
JM
1264 RFALSE((curf && !B_IS_IN_TREE(curf)) ||
1265 (curcf && !B_IS_IN_TREE(curcf)),
1266 "PAP-8205: FR (%b) or CFR (%b) is invalid", curf, curcf);
bd4c625c
LT
1267
1268 return CARRY_ON;
1269}
1da177e4 1270
098297b2
JM
1271/*
1272 * it is possible to remove node as result of shiftings to
1273 * neighbors even when we insert or paste item.
1274 */
bd4c625c
LT
1275static inline int can_node_be_removed(int mode, int lfree, int sfree, int rfree,
1276 struct tree_balance *tb, int h)
1da177e4 1277{
bd4c625c
LT
1278 struct buffer_head *Sh = PATH_H_PBUFFER(tb->tb_path, h);
1279 int levbytes = tb->insert_size[h];
1280 struct item_head *ih;
1281 struct reiserfs_key *r_key = NULL;
1282
4cf5f7ad 1283 ih = item_head(Sh, 0);
bd4c625c 1284 if (tb->CFR[h])
4cf5f7ad 1285 r_key = internal_key(tb->CFR[h], tb->rkey[h]);
bd4c625c
LT
1286
1287 if (lfree + rfree + sfree < MAX_CHILD_SIZE(Sh) + levbytes
1288 /* shifting may merge items which might save space */
1289 -
1290 ((!h
a228bf8f 1291 && op_is_left_mergeable(&ih->ih_key, Sh->b_size)) ? IH_SIZE : 0)
bd4c625c
LT
1292 -
1293 ((!h && r_key
1294 && op_is_left_mergeable(r_key, Sh->b_size)) ? IH_SIZE : 0)
1295 + ((h) ? KEY_SIZE : 0)) {
1296 /* node can not be removed */
098297b2
JM
1297 if (sfree >= levbytes) {
1298 /* new item fits into node S[h] without any shifting */
bd4c625c
LT
1299 if (!h)
1300 tb->s0num =
1301 B_NR_ITEMS(Sh) +
1302 ((mode == M_INSERT) ? 1 : 0);
1303 set_parameters(tb, h, 0, 0, 1, NULL, -1, -1);
1304 return NO_BALANCING_NEEDED;
1305 }
1da177e4 1306 }
bd4c625c
LT
1307 PROC_INFO_INC(tb->tb_sb, can_node_be_removed[h]);
1308 return !NO_BALANCING_NEEDED;
1da177e4
LT
1309}
1310
098297b2
JM
1311/*
1312 * Check whether current node S[h] is balanced when increasing its size by
1da177e4
LT
1313 * Inserting or Pasting.
1314 * Calculate parameters for balancing for current level h.
1315 * Parameters:
1316 * tb tree_balance structure;
1317 * h current level of the node;
1318 * inum item number in S[h];
1319 * mode i - insert, p - paste;
0222e657 1320 * Returns: 1 - schedule occurred;
1da177e4
LT
1321 * 0 - balancing for higher levels needed;
1322 * -1 - no balancing for higher levels needed;
1323 * -2 - no disk space.
1324 */
1325/* ip means Inserting or Pasting */
bd4c625c 1326static int ip_check_balance(struct tree_balance *tb, int h)
1da177e4 1327{
bd4c625c 1328 struct virtual_node *vn = tb->tb_vn;
098297b2
JM
1329 /*
1330 * Number of bytes that must be inserted into (value is negative
1331 * if bytes are deleted) buffer which contains node being balanced.
1332 * The mnemonic is that the attempted change in node space used
1333 * level is levbytes bytes.
1334 */
1335 int levbytes;
1336 int ret;
bd4c625c
LT
1337
1338 int lfree, sfree, rfree /* free space in L, S and R */ ;
1339
098297b2
JM
1340 /*
1341 * nver is short for number of vertixes, and lnver is the number if
1342 * we shift to the left, rnver is the number if we shift to the
1343 * right, and lrnver is the number if we shift in both directions.
1344 * The goal is to minimize first the number of vertixes, and second,
1345 * the number of vertixes whose contents are changed by shifting,
1346 * and third the number of uncached vertixes whose contents are
1347 * changed by shifting and must be read from disk.
1348 */
bd4c625c
LT
1349 int nver, lnver, rnver, lrnver;
1350
098297b2
JM
1351 /*
1352 * used at leaf level only, S0 = S[0] is the node being balanced,
1353 * sInum [ I = 0,1,2 ] is the number of items that will
1354 * remain in node SI after balancing. S1 and S2 are new
1355 * nodes that might be created.
1356 */
bd4c625c 1357
098297b2
JM
1358 /*
1359 * we perform 8 calls to get_num_ver(). For each call we
1360 * calculate five parameters. where 4th parameter is s1bytes
1361 * and 5th - s2bytes
1362 *
1363 * s0num, s1num, s2num for 8 cases
1364 * 0,1 - do not shift and do not shift but bottle
1365 * 2 - shift only whole item to left
1366 * 3 - shift to left and bottle as much as possible
1367 * 4,5 - shift to right (whole items and as much as possible
1368 * 6,7 - shift to both directions (whole items and as much as possible)
bd4c625c 1369 */
098297b2 1370 short snum012[40] = { 0, };
bd4c625c
LT
1371
1372 /* Sh is the node whose balance is currently being checked */
1373 struct buffer_head *Sh;
1374
1375 Sh = PATH_H_PBUFFER(tb->tb_path, h);
1376 levbytes = tb->insert_size[h];
1377
1378 /* Calculate balance parameters for creating new root. */
1379 if (!Sh) {
1380 if (!h)
c3a9c210
JM
1381 reiserfs_panic(tb->tb_sb, "vs-8210",
1382 "S[0] can not be 0");
ee93961b 1383 switch (ret = get_empty_nodes(tb, h)) {
098297b2 1384 /* no balancing for higher levels needed */
bd4c625c
LT
1385 case CARRY_ON:
1386 set_parameters(tb, h, 0, 0, 1, NULL, -1, -1);
098297b2 1387 return NO_BALANCING_NEEDED;
bd4c625c
LT
1388
1389 case NO_DISK_SPACE:
1390 case REPEAT_SEARCH:
ee93961b 1391 return ret;
bd4c625c 1392 default:
c3a9c210
JM
1393 reiserfs_panic(tb->tb_sb, "vs-8215", "incorrect "
1394 "return value of get_empty_nodes");
bd4c625c 1395 }
1da177e4 1396 }
1da177e4 1397
098297b2
JM
1398 /* get parents of S[h] neighbors. */
1399 ret = get_parents(tb, h);
1400 if (ret != CARRY_ON)
ee93961b 1401 return ret;
1da177e4 1402
bd4c625c
LT
1403 sfree = B_FREE_SPACE(Sh);
1404
1405 /* get free space of neighbors */
1406 rfree = get_rfree(tb, h);
1407 lfree = get_lfree(tb, h);
1408
098297b2 1409 /* and new item fits into node S[h] without any shifting */
bd4c625c
LT
1410 if (can_node_be_removed(vn->vn_mode, lfree, sfree, rfree, tb, h) ==
1411 NO_BALANCING_NEEDED)
bd4c625c 1412 return NO_BALANCING_NEEDED;
1da177e4 1413
bd4c625c 1414 create_virtual_node(tb, h);
1da177e4 1415
0222e657 1416 /*
098297b2
JM
1417 * determine maximal number of items we can shift to the left
1418 * neighbor (in tb structure) and the maximal number of bytes
1419 * that can flow to the left neighbor from the left most liquid
1420 * item that cannot be shifted from S[0] entirely (returned value)
1da177e4 1421 */
bd4c625c 1422 check_left(tb, h, lfree);
1da177e4 1423
bd4c625c 1424 /*
098297b2
JM
1425 * determine maximal number of items we can shift to the right
1426 * neighbor (in tb structure) and the maximal number of bytes
1427 * that can flow to the right neighbor from the right most liquid
1428 * item that cannot be shifted from S[0] entirely (returned value)
bd4c625c
LT
1429 */
1430 check_right(tb, h, rfree);
1431
098297b2
JM
1432 /*
1433 * all contents of internal node S[h] can be moved into its
1434 * neighbors, S[h] will be removed after balancing
1435 */
bd4c625c
LT
1436 if (h && (tb->rnum[h] + tb->lnum[h] >= vn->vn_nr_item + 1)) {
1437 int to_r;
1438
098297b2
JM
1439 /*
1440 * Since we are working on internal nodes, and our internal
1441 * nodes have fixed size entries, then we can balance by the
1442 * number of items rather than the space they consume. In this
1443 * routine we set the left node equal to the right node,
1444 * allowing a difference of less than or equal to 1 child
1445 * pointer.
1446 */
bd4c625c
LT
1447 to_r =
1448 ((MAX_NR_KEY(Sh) << 1) + 2 - tb->lnum[h] - tb->rnum[h] +
1449 vn->vn_nr_item + 1) / 2 - (MAX_NR_KEY(Sh) + 1 -
1450 tb->rnum[h]);
1451 set_parameters(tb, h, vn->vn_nr_item + 1 - to_r, to_r, 0, NULL,
1452 -1, -1);
1453 return CARRY_ON;
1454 }
1455
098297b2
JM
1456 /*
1457 * this checks balance condition, that any two neighboring nodes
1458 * can not fit in one node
1459 */
bd4c625c
LT
1460 RFALSE(h &&
1461 (tb->lnum[h] >= vn->vn_nr_item + 1 ||
1462 tb->rnum[h] >= vn->vn_nr_item + 1),
1463 "vs-8220: tree is not balanced on internal level");
1464 RFALSE(!h && ((tb->lnum[h] >= vn->vn_nr_item && (tb->lbytes == -1)) ||
1465 (tb->rnum[h] >= vn->vn_nr_item && (tb->rbytes == -1))),
1466 "vs-8225: tree is not balanced on leaf level");
1467
098297b2
JM
1468 /*
1469 * all contents of S[0] can be moved into its neighbors
1470 * S[0] will be removed after balancing.
1471 */
bd4c625c
LT
1472 if (!h && is_leaf_removable(tb))
1473 return CARRY_ON;
1474
098297b2
JM
1475 /*
1476 * why do we perform this check here rather than earlier??
1477 * Answer: we can win 1 node in some cases above. Moreover we
1478 * checked it above, when we checked, that S[0] is not removable
1479 * in principle
1480 */
1481
1482 /* new item fits into node S[h] without any shifting */
1483 if (sfree >= levbytes) {
bd4c625c
LT
1484 if (!h)
1485 tb->s0num = vn->vn_nr_item;
1486 set_parameters(tb, h, 0, 0, 1, NULL, -1, -1);
1487 return NO_BALANCING_NEEDED;
1488 }
1489
1490 {
1491 int lpar, rpar, nset, lset, rset, lrset;
098297b2 1492 /* regular overflowing of the node */
bd4c625c 1493
098297b2
JM
1494 /*
1495 * get_num_ver works in 2 modes (FLOW & NO_FLOW)
1496 * lpar, rpar - number of items we can shift to left/right
1497 * neighbor (including splitting item)
1498 * nset, lset, rset, lrset - shows, whether flowing items
1499 * give better packing
bd4c625c 1500 */
1da177e4 1501#define FLOW 1
bd4c625c 1502#define NO_FLOW 0 /* do not any splitting */
1da177e4 1503
098297b2 1504 /* we choose one of the following */
1da177e4
LT
1505#define NOTHING_SHIFT_NO_FLOW 0
1506#define NOTHING_SHIFT_FLOW 5
1507#define LEFT_SHIFT_NO_FLOW 10
1508#define LEFT_SHIFT_FLOW 15
1509#define RIGHT_SHIFT_NO_FLOW 20
1510#define RIGHT_SHIFT_FLOW 25
1511#define LR_SHIFT_NO_FLOW 30
1512#define LR_SHIFT_FLOW 35
1513
bd4c625c
LT
1514 lpar = tb->lnum[h];
1515 rpar = tb->rnum[h];
1516
098297b2
JM
1517 /*
1518 * calculate number of blocks S[h] must be split into when
1519 * nothing is shifted to the neighbors, as well as number of
1520 * items in each part of the split node (s012 numbers),
1521 * and number of bytes (s1bytes) of the shared drop which
1522 * flow to S1 if any
1523 */
bd4c625c
LT
1524 nset = NOTHING_SHIFT_NO_FLOW;
1525 nver = get_num_ver(vn->vn_mode, tb, h,
1526 0, -1, h ? vn->vn_nr_item : 0, -1,
1527 snum012, NO_FLOW);
1528
1529 if (!h) {
1530 int nver1;
1531
098297b2
JM
1532 /*
1533 * note, that in this case we try to bottle
1534 * between S[0] and S1 (S1 - the first new node)
1535 */
bd4c625c
LT
1536 nver1 = get_num_ver(vn->vn_mode, tb, h,
1537 0, -1, 0, -1,
1538 snum012 + NOTHING_SHIFT_FLOW, FLOW);
1539 if (nver > nver1)
1540 nset = NOTHING_SHIFT_FLOW, nver = nver1;
1541 }
1da177e4 1542
098297b2
JM
1543 /*
1544 * calculate number of blocks S[h] must be split into when
1545 * l_shift_num first items and l_shift_bytes of the right
1546 * most liquid item to be shifted are shifted to the left
1547 * neighbor, as well as number of items in each part of the
1548 * splitted node (s012 numbers), and number of bytes
1549 * (s1bytes) of the shared drop which flow to S1 if any
bd4c625c
LT
1550 */
1551 lset = LEFT_SHIFT_NO_FLOW;
1552 lnver = get_num_ver(vn->vn_mode, tb, h,
1553 lpar - ((h || tb->lbytes == -1) ? 0 : 1),
1554 -1, h ? vn->vn_nr_item : 0, -1,
1555 snum012 + LEFT_SHIFT_NO_FLOW, NO_FLOW);
1556 if (!h) {
1557 int lnver1;
1558
1559 lnver1 = get_num_ver(vn->vn_mode, tb, h,
1560 lpar -
1561 ((tb->lbytes != -1) ? 1 : 0),
1562 tb->lbytes, 0, -1,
1563 snum012 + LEFT_SHIFT_FLOW, FLOW);
1564 if (lnver > lnver1)
1565 lset = LEFT_SHIFT_FLOW, lnver = lnver1;
1566 }
1da177e4 1567
098297b2
JM
1568 /*
1569 * calculate number of blocks S[h] must be split into when
1570 * r_shift_num first items and r_shift_bytes of the left most
1571 * liquid item to be shifted are shifted to the right neighbor,
1572 * as well as number of items in each part of the splitted
1573 * node (s012 numbers), and number of bytes (s1bytes) of the
1574 * shared drop which flow to S1 if any
bd4c625c
LT
1575 */
1576 rset = RIGHT_SHIFT_NO_FLOW;
1577 rnver = get_num_ver(vn->vn_mode, tb, h,
1578 0, -1,
1579 h ? (vn->vn_nr_item - rpar) : (rpar -
1580 ((tb->
1581 rbytes !=
1582 -1) ? 1 :
1583 0)), -1,
1584 snum012 + RIGHT_SHIFT_NO_FLOW, NO_FLOW);
1585 if (!h) {
1586 int rnver1;
1587
1588 rnver1 = get_num_ver(vn->vn_mode, tb, h,
1589 0, -1,
1590 (rpar -
1591 ((tb->rbytes != -1) ? 1 : 0)),
1592 tb->rbytes,
1593 snum012 + RIGHT_SHIFT_FLOW, FLOW);
1594
1595 if (rnver > rnver1)
1596 rset = RIGHT_SHIFT_FLOW, rnver = rnver1;
1597 }
1da177e4 1598
098297b2
JM
1599 /*
1600 * calculate number of blocks S[h] must be split into when
1601 * items are shifted in both directions, as well as number
1602 * of items in each part of the splitted node (s012 numbers),
1603 * and number of bytes (s1bytes) of the shared drop which
1604 * flow to S1 if any
bd4c625c
LT
1605 */
1606 lrset = LR_SHIFT_NO_FLOW;
1607 lrnver = get_num_ver(vn->vn_mode, tb, h,
1608 lpar - ((h || tb->lbytes == -1) ? 0 : 1),
1609 -1,
1610 h ? (vn->vn_nr_item - rpar) : (rpar -
1611 ((tb->
1612 rbytes !=
1613 -1) ? 1 :
1614 0)), -1,
1615 snum012 + LR_SHIFT_NO_FLOW, NO_FLOW);
1616 if (!h) {
1617 int lrnver1;
1618
1619 lrnver1 = get_num_ver(vn->vn_mode, tb, h,
1620 lpar -
1621 ((tb->lbytes != -1) ? 1 : 0),
1622 tb->lbytes,
1623 (rpar -
1624 ((tb->rbytes != -1) ? 1 : 0)),
1625 tb->rbytes,
1626 snum012 + LR_SHIFT_FLOW, FLOW);
1627 if (lrnver > lrnver1)
1628 lrset = LR_SHIFT_FLOW, lrnver = lrnver1;
1629 }
1da177e4 1630
098297b2
JM
1631 /*
1632 * Our general shifting strategy is:
1633 * 1) to minimized number of new nodes;
1634 * 2) to minimized number of neighbors involved in shifting;
1635 * 3) to minimized number of disk reads;
1636 */
bd4c625c
LT
1637
1638 /* we can win TWO or ONE nodes by shifting in both directions */
1639 if (lrnver < lnver && lrnver < rnver) {
1640 RFALSE(h &&
1641 (tb->lnum[h] != 1 ||
1642 tb->rnum[h] != 1 ||
1643 lrnver != 1 || rnver != 2 || lnver != 2
1644 || h != 1), "vs-8230: bad h");
1645 if (lrset == LR_SHIFT_FLOW)
1646 set_parameters(tb, h, tb->lnum[h], tb->rnum[h],
1647 lrnver, snum012 + lrset,
1648 tb->lbytes, tb->rbytes);
1649 else
1650 set_parameters(tb, h,
1651 tb->lnum[h] -
1652 ((tb->lbytes == -1) ? 0 : 1),
1653 tb->rnum[h] -
1654 ((tb->rbytes == -1) ? 0 : 1),
1655 lrnver, snum012 + lrset, -1, -1);
1656
1657 return CARRY_ON;
1658 }
1da177e4 1659
098297b2
JM
1660 /*
1661 * if shifting doesn't lead to better packing
1662 * then don't shift
1663 */
bd4c625c
LT
1664 if (nver == lrnver) {
1665 set_parameters(tb, h, 0, 0, nver, snum012 + nset, -1,
1666 -1);
1667 return CARRY_ON;
1668 }
1da177e4 1669
098297b2
JM
1670 /*
1671 * now we know that for better packing shifting in only one
1672 * direction either to the left or to the right is required
1673 */
1da177e4 1674
098297b2
JM
1675 /*
1676 * if shifting to the left is better than
1677 * shifting to the right
1678 */
bd4c625c
LT
1679 if (lnver < rnver) {
1680 SET_PAR_SHIFT_LEFT;
1681 return CARRY_ON;
1682 }
1da177e4 1683
098297b2
JM
1684 /*
1685 * if shifting to the right is better than
1686 * shifting to the left
1687 */
bd4c625c
LT
1688 if (lnver > rnver) {
1689 SET_PAR_SHIFT_RIGHT;
1690 return CARRY_ON;
1691 }
1da177e4 1692
098297b2
JM
1693 /*
1694 * now shifting in either direction gives the same number
1695 * of nodes and we can make use of the cached neighbors
1696 */
bd4c625c
LT
1697 if (is_left_neighbor_in_cache(tb, h)) {
1698 SET_PAR_SHIFT_LEFT;
1699 return CARRY_ON;
1700 }
1da177e4 1701
098297b2
JM
1702 /*
1703 * shift to the right independently on whether the
1704 * right neighbor in cache or not
1705 */
bd4c625c
LT
1706 SET_PAR_SHIFT_RIGHT;
1707 return CARRY_ON;
1da177e4 1708 }
1da177e4
LT
1709}
1710
098297b2
JM
1711/*
1712 * Check whether current node S[h] is balanced when Decreasing its size by
1da177e4
LT
1713 * Deleting or Cutting for INTERNAL node of S+tree.
1714 * Calculate parameters for balancing for current level h.
1715 * Parameters:
1716 * tb tree_balance structure;
1717 * h current level of the node;
1718 * inum item number in S[h];
1719 * mode i - insert, p - paste;
0222e657 1720 * Returns: 1 - schedule occurred;
1da177e4
LT
1721 * 0 - balancing for higher levels needed;
1722 * -1 - no balancing for higher levels needed;
1723 * -2 - no disk space.
1724 *
1725 * Note: Items of internal nodes have fixed size, so the balance condition for
1726 * the internal part of S+tree is as for the B-trees.
1727 */
bd4c625c 1728static int dc_check_balance_internal(struct tree_balance *tb, int h)
1da177e4 1729{
bd4c625c 1730 struct virtual_node *vn = tb->tb_vn;
1da177e4 1731
098297b2
JM
1732 /*
1733 * Sh is the node whose balance is currently being checked,
1734 * and Fh is its father.
1735 */
bd4c625c 1736 struct buffer_head *Sh, *Fh;
ee93961b 1737 int maxsize, ret;
bd4c625c 1738 int lfree, rfree /* free space in L and R */ ;
1da177e4 1739
bd4c625c
LT
1740 Sh = PATH_H_PBUFFER(tb->tb_path, h);
1741 Fh = PATH_H_PPARENT(tb->tb_path, h);
1da177e4 1742
bd4c625c 1743 maxsize = MAX_CHILD_SIZE(Sh);
1da177e4 1744
098297b2
JM
1745 /*
1746 * using tb->insert_size[h], which is negative in this case,
1747 * create_virtual_node calculates:
1748 * new_nr_item = number of items node would have if operation is
1749 * performed without balancing (new_nr_item);
1750 */
bd4c625c 1751 create_virtual_node(tb, h);
1da177e4 1752
bd4c625c 1753 if (!Fh) { /* S[h] is the root. */
098297b2 1754 /* no balancing for higher levels needed */
bd4c625c
LT
1755 if (vn->vn_nr_item > 0) {
1756 set_parameters(tb, h, 0, 0, 1, NULL, -1, -1);
098297b2 1757 return NO_BALANCING_NEEDED;
bd4c625c 1758 }
098297b2
JM
1759 /*
1760 * new_nr_item == 0.
bd4c625c 1761 * Current root will be deleted resulting in
098297b2
JM
1762 * decrementing the tree height.
1763 */
bd4c625c
LT
1764 set_parameters(tb, h, 0, 0, 0, NULL, -1, -1);
1765 return CARRY_ON;
1766 }
1767
ee93961b
JM
1768 if ((ret = get_parents(tb, h)) != CARRY_ON)
1769 return ret;
bd4c625c
LT
1770
1771 /* get free space of neighbors */
1772 rfree = get_rfree(tb, h);
1773 lfree = get_lfree(tb, h);
1774
1775 /* determine maximal number of items we can fit into neighbors */
1776 check_left(tb, h, lfree);
1777 check_right(tb, h, rfree);
1778
098297b2
JM
1779 /*
1780 * Balance condition for the internal node is valid.
1781 * In this case we balance only if it leads to better packing.
1782 */
1783 if (vn->vn_nr_item >= MIN_NR_KEY(Sh)) {
1784 /*
1785 * Here we join S[h] with one of its neighbors,
1786 * which is impossible with greater values of new_nr_item.
1787 */
1788 if (vn->vn_nr_item == MIN_NR_KEY(Sh)) {
1789 /* All contents of S[h] can be moved to L[h]. */
bd4c625c 1790 if (tb->lnum[h] >= vn->vn_nr_item + 1) {
bd4c625c
LT
1791 int n;
1792 int order_L;
1793
1794 order_L =
1795 ((n =
1796 PATH_H_B_ITEM_ORDER(tb->tb_path,
1797 h)) ==
1798 0) ? B_NR_ITEMS(tb->FL[h]) : n - 1;
1799 n = dc_size(B_N_CHILD(tb->FL[h], order_L)) /
1800 (DC_SIZE + KEY_SIZE);
1801 set_parameters(tb, h, -n - 1, 0, 0, NULL, -1,
1802 -1);
1803 return CARRY_ON;
1804 }
1805
098297b2 1806 /* All contents of S[h] can be moved to R[h]. */
bd4c625c 1807 if (tb->rnum[h] >= vn->vn_nr_item + 1) {
bd4c625c
LT
1808 int n;
1809 int order_R;
1810
1811 order_R =
1812 ((n =
1813 PATH_H_B_ITEM_ORDER(tb->tb_path,
1814 h)) ==
1815 B_NR_ITEMS(Fh)) ? 0 : n + 1;
1816 n = dc_size(B_N_CHILD(tb->FR[h], order_R)) /
1817 (DC_SIZE + KEY_SIZE);
1818 set_parameters(tb, h, 0, -n - 1, 0, NULL, -1,
1819 -1);
1820 return CARRY_ON;
1821 }
1822 }
1823
098297b2
JM
1824 /*
1825 * All contents of S[h] can be moved to the neighbors
1826 * (L[h] & R[h]).
1827 */
bd4c625c 1828 if (tb->rnum[h] + tb->lnum[h] >= vn->vn_nr_item + 1) {
bd4c625c
LT
1829 int to_r;
1830
1831 to_r =
1832 ((MAX_NR_KEY(Sh) << 1) + 2 - tb->lnum[h] -
1833 tb->rnum[h] + vn->vn_nr_item + 1) / 2 -
1834 (MAX_NR_KEY(Sh) + 1 - tb->rnum[h]);
1835 set_parameters(tb, h, vn->vn_nr_item + 1 - to_r, to_r,
1836 0, NULL, -1, -1);
1837 return CARRY_ON;
1838 }
1839
1840 /* Balancing does not lead to better packing. */
1841 set_parameters(tb, h, 0, 0, 1, NULL, -1, -1);
1842 return NO_BALANCING_NEEDED;
1da177e4 1843 }
bd4c625c 1844
098297b2
JM
1845 /*
1846 * Current node contain insufficient number of items.
1847 * Balancing is required.
1848 */
bd4c625c
LT
1849 /* Check whether we can merge S[h] with left neighbor. */
1850 if (tb->lnum[h] >= vn->vn_nr_item + 1)
1851 if (is_left_neighbor_in_cache(tb, h)
1852 || tb->rnum[h] < vn->vn_nr_item + 1 || !tb->FR[h]) {
1853 int n;
1854 int order_L;
1855
1856 order_L =
1857 ((n =
1858 PATH_H_B_ITEM_ORDER(tb->tb_path,
1859 h)) ==
1860 0) ? B_NR_ITEMS(tb->FL[h]) : n - 1;
1861 n = dc_size(B_N_CHILD(tb->FL[h], order_L)) / (DC_SIZE +
1862 KEY_SIZE);
1863 set_parameters(tb, h, -n - 1, 0, 0, NULL, -1, -1);
1864 return CARRY_ON;
1865 }
1866
1867 /* Check whether we can merge S[h] with right neighbor. */
1868 if (tb->rnum[h] >= vn->vn_nr_item + 1) {
1869 int n;
1870 int order_R;
1871
1872 order_R =
1873 ((n =
1874 PATH_H_B_ITEM_ORDER(tb->tb_path,
1875 h)) == B_NR_ITEMS(Fh)) ? 0 : (n + 1);
1876 n = dc_size(B_N_CHILD(tb->FR[h], order_R)) / (DC_SIZE +
1877 KEY_SIZE);
1878 set_parameters(tb, h, 0, -n - 1, 0, NULL, -1, -1);
1879 return CARRY_ON;
1da177e4
LT
1880 }
1881
bd4c625c
LT
1882 /* All contents of S[h] can be moved to the neighbors (L[h] & R[h]). */
1883 if (tb->rnum[h] + tb->lnum[h] >= vn->vn_nr_item + 1) {
1884 int to_r;
1885
1886 to_r =
1887 ((MAX_NR_KEY(Sh) << 1) + 2 - tb->lnum[h] - tb->rnum[h] +
1888 vn->vn_nr_item + 1) / 2 - (MAX_NR_KEY(Sh) + 1 -
1889 tb->rnum[h]);
1890 set_parameters(tb, h, vn->vn_nr_item + 1 - to_r, to_r, 0, NULL,
1891 -1, -1);
1892 return CARRY_ON;
1893 }
1da177e4 1894
bd4c625c
LT
1895 /* For internal nodes try to borrow item from a neighbor */
1896 RFALSE(!tb->FL[h] && !tb->FR[h], "vs-8235: trying to borrow for root");
1897
1898 /* Borrow one or two items from caching neighbor */
1899 if (is_left_neighbor_in_cache(tb, h) || !tb->FR[h]) {
1900 int from_l;
1901
1902 from_l =
1903 (MAX_NR_KEY(Sh) + 1 - tb->lnum[h] + vn->vn_nr_item +
1904 1) / 2 - (vn->vn_nr_item + 1);
1905 set_parameters(tb, h, -from_l, 0, 1, NULL, -1, -1);
1906 return CARRY_ON;
1da177e4
LT
1907 }
1908
bd4c625c
LT
1909 set_parameters(tb, h, 0,
1910 -((MAX_NR_KEY(Sh) + 1 - tb->rnum[h] + vn->vn_nr_item +
1911 1) / 2 - (vn->vn_nr_item + 1)), 1, NULL, -1, -1);
1da177e4 1912 return CARRY_ON;
1da177e4
LT
1913}
1914
098297b2
JM
1915/*
1916 * Check whether current node S[h] is balanced when Decreasing its size by
1da177e4
LT
1917 * Deleting or Truncating for LEAF node of S+tree.
1918 * Calculate parameters for balancing for current level h.
1919 * Parameters:
1920 * tb tree_balance structure;
1921 * h current level of the node;
1922 * inum item number in S[h];
1923 * mode i - insert, p - paste;
0222e657 1924 * Returns: 1 - schedule occurred;
1da177e4
LT
1925 * 0 - balancing for higher levels needed;
1926 * -1 - no balancing for higher levels needed;
1927 * -2 - no disk space.
1928 */
bd4c625c 1929static int dc_check_balance_leaf(struct tree_balance *tb, int h)
1da177e4 1930{
bd4c625c
LT
1931 struct virtual_node *vn = tb->tb_vn;
1932
098297b2
JM
1933 /*
1934 * Number of bytes that must be deleted from
1935 * (value is negative if bytes are deleted) buffer which
1936 * contains node being balanced. The mnemonic is that the
1937 * attempted change in node space used level is levbytes bytes.
1938 */
bd4c625c 1939 int levbytes;
098297b2 1940
bd4c625c 1941 /* the maximal item size */
ee93961b 1942 int maxsize, ret;
098297b2
JM
1943
1944 /*
1945 * S0 is the node whose balance is currently being checked,
1946 * and F0 is its father.
1947 */
bd4c625c
LT
1948 struct buffer_head *S0, *F0;
1949 int lfree, rfree /* free space in L and R */ ;
1950
1951 S0 = PATH_H_PBUFFER(tb->tb_path, 0);
1952 F0 = PATH_H_PPARENT(tb->tb_path, 0);
1da177e4 1953
bd4c625c 1954 levbytes = tb->insert_size[h];
1da177e4 1955
bd4c625c
LT
1956 maxsize = MAX_CHILD_SIZE(S0); /* maximal possible size of an item */
1957
1958 if (!F0) { /* S[0] is the root now. */
1959
1960 RFALSE(-levbytes >= maxsize - B_FREE_SPACE(S0),
1961 "vs-8240: attempt to create empty buffer tree");
1962
1963 set_parameters(tb, h, 0, 0, 1, NULL, -1, -1);
1964 return NO_BALANCING_NEEDED;
1965 }
1966
ee93961b
JM
1967 if ((ret = get_parents(tb, h)) != CARRY_ON)
1968 return ret;
bd4c625c
LT
1969
1970 /* get free space of neighbors */
1971 rfree = get_rfree(tb, h);
1972 lfree = get_lfree(tb, h);
1973
1974 create_virtual_node(tb, h);
1975
1976 /* if 3 leaves can be merge to one, set parameters and return */
1977 if (are_leaves_removable(tb, lfree, rfree))
1978 return CARRY_ON;
1979
098297b2
JM
1980 /*
1981 * determine maximal number of items we can shift to the left/right
1982 * neighbor and the maximal number of bytes that can flow to the
1983 * left/right neighbor from the left/right most liquid item that
1984 * cannot be shifted from S[0] entirely
bd4c625c
LT
1985 */
1986 check_left(tb, h, lfree);
1987 check_right(tb, h, rfree);
1988
1989 /* check whether we can merge S with left neighbor. */
1990 if (tb->lnum[0] >= vn->vn_nr_item && tb->lbytes == -1)
1991 if (is_left_neighbor_in_cache(tb, h) || ((tb->rnum[0] - ((tb->rbytes == -1) ? 0 : 1)) < vn->vn_nr_item) || /* S can not be merged with R */
1992 !tb->FR[h]) {
1993
1994 RFALSE(!tb->FL[h],
1995 "vs-8245: dc_check_balance_leaf: FL[h] must exist");
1996
1997 /* set parameter to merge S[0] with its left neighbor */
1998 set_parameters(tb, h, -1, 0, 0, NULL, -1, -1);
1999 return CARRY_ON;
2000 }
2001
2002 /* check whether we can merge S[0] with right neighbor. */
2003 if (tb->rnum[0] >= vn->vn_nr_item && tb->rbytes == -1) {
2004 set_parameters(tb, h, 0, -1, 0, NULL, -1, -1);
2005 return CARRY_ON;
2006 }
2007
098297b2
JM
2008 /*
2009 * All contents of S[0] can be moved to the neighbors (L[0] & R[0]).
2010 * Set parameters and return
2011 */
bd4c625c
LT
2012 if (is_leaf_removable(tb))
2013 return CARRY_ON;
2014
2015 /* Balancing is not required. */
2016 tb->s0num = vn->vn_nr_item;
2017 set_parameters(tb, h, 0, 0, 1, NULL, -1, -1);
2018 return NO_BALANCING_NEEDED;
2019}
1da177e4 2020
098297b2
JM
2021/*
2022 * Check whether current node S[h] is balanced when Decreasing its size by
1da177e4
LT
2023 * Deleting or Cutting.
2024 * Calculate parameters for balancing for current level h.
2025 * Parameters:
2026 * tb tree_balance structure;
2027 * h current level of the node;
2028 * inum item number in S[h];
2029 * mode d - delete, c - cut.
0222e657 2030 * Returns: 1 - schedule occurred;
1da177e4
LT
2031 * 0 - balancing for higher levels needed;
2032 * -1 - no balancing for higher levels needed;
2033 * -2 - no disk space.
2034 */
bd4c625c 2035static int dc_check_balance(struct tree_balance *tb, int h)
1da177e4 2036{
bd4c625c
LT
2037 RFALSE(!(PATH_H_PBUFFER(tb->tb_path, h)),
2038 "vs-8250: S is not initialized");
1da177e4 2039
bd4c625c
LT
2040 if (h)
2041 return dc_check_balance_internal(tb, h);
2042 else
2043 return dc_check_balance_leaf(tb, h);
1da177e4
LT
2044}
2045
098297b2
JM
2046/*
2047 * Check whether current node S[h] is balanced.
1da177e4
LT
2048 * Calculate parameters for balancing for current level h.
2049 * Parameters:
2050 *
2051 * tb tree_balance structure:
2052 *
098297b2
JM
2053 * tb is a large structure that must be read about in the header
2054 * file at the same time as this procedure if the reader is
2055 * to successfully understand this procedure
1da177e4
LT
2056 *
2057 * h current level of the node;
2058 * inum item number in S[h];
2059 * mode i - insert, p - paste, d - delete, c - cut.
0222e657 2060 * Returns: 1 - schedule occurred;
1da177e4
LT
2061 * 0 - balancing for higher levels needed;
2062 * -1 - no balancing for higher levels needed;
2063 * -2 - no disk space.
2064 */
bd4c625c
LT
2065static int check_balance(int mode,
2066 struct tree_balance *tb,
2067 int h,
2068 int inum,
2069 int pos_in_item,
2070 struct item_head *ins_ih, const void *data)
1da177e4 2071{
bd4c625c 2072 struct virtual_node *vn;
1da177e4 2073
bd4c625c
LT
2074 vn = tb->tb_vn = (struct virtual_node *)(tb->vn_buf);
2075 vn->vn_free_ptr = (char *)(tb->tb_vn + 1);
2076 vn->vn_mode = mode;
2077 vn->vn_affected_item_num = inum;
2078 vn->vn_pos_in_item = pos_in_item;
2079 vn->vn_ins_ih = ins_ih;
2080 vn->vn_data = data;
1da177e4 2081
bd4c625c
LT
2082 RFALSE(mode == M_INSERT && !vn->vn_ins_ih,
2083 "vs-8255: ins_ih can not be 0 in insert mode");
1da177e4 2084
098297b2 2085 /* Calculate balance parameters when size of node is increasing. */
bd4c625c 2086 if (tb->insert_size[h] > 0)
bd4c625c 2087 return ip_check_balance(tb, h);
1da177e4 2088
bd4c625c
LT
2089 /* Calculate balance parameters when size of node is decreasing. */
2090 return dc_check_balance(tb, h);
1da177e4
LT
2091}
2092
bd4c625c 2093/* Check whether parent at the path is the really parent of the current node.*/
ee93961b 2094static int get_direct_parent(struct tree_balance *tb, int h)
bd4c625c 2095{
ad31a4fc 2096 struct buffer_head *bh;
d68caa95 2097 struct treepath *path = tb->tb_path;
ee93961b
JM
2098 int position,
2099 path_offset = PATH_H_PATH_OFFSET(tb->tb_path, h);
bd4c625c
LT
2100
2101 /* We are in the root or in the new root. */
ee93961b 2102 if (path_offset <= FIRST_PATH_ELEMENT_OFFSET) {
bd4c625c 2103
ee93961b 2104 RFALSE(path_offset < FIRST_PATH_ELEMENT_OFFSET - 1,
bd4c625c
LT
2105 "PAP-8260: invalid offset in the path");
2106
d68caa95 2107 if (PATH_OFFSET_PBUFFER(path, FIRST_PATH_ELEMENT_OFFSET)->
a063ae17 2108 b_blocknr == SB_ROOT_BLOCK(tb->tb_sb)) {
bd4c625c 2109 /* Root is not changed. */
ee93961b
JM
2110 PATH_OFFSET_PBUFFER(path, path_offset - 1) = NULL;
2111 PATH_OFFSET_POSITION(path, path_offset - 1) = 0;
bd4c625c
LT
2112 return CARRY_ON;
2113 }
098297b2
JM
2114 /* Root is changed and we must recalculate the path. */
2115 return REPEAT_SEARCH;
bd4c625c
LT
2116 }
2117
098297b2 2118 /* Parent in the path is not in the tree. */
bd4c625c 2119 if (!B_IS_IN_TREE
ee93961b 2120 (bh = PATH_OFFSET_PBUFFER(path, path_offset - 1)))
098297b2 2121 return REPEAT_SEARCH;
1da177e4 2122
ee93961b 2123 if ((position =
d68caa95 2124 PATH_OFFSET_POSITION(path,
ee93961b 2125 path_offset - 1)) > B_NR_ITEMS(bh))
bd4c625c 2126 return REPEAT_SEARCH;
1da177e4 2127
098297b2 2128 /* Parent in the path is not parent of the current node in the tree. */
ee93961b
JM
2129 if (B_N_CHILD_NUM(bh, position) !=
2130 PATH_OFFSET_PBUFFER(path, path_offset)->b_blocknr)
bd4c625c
LT
2131 return REPEAT_SEARCH;
2132
ad31a4fc 2133 if (buffer_locked(bh)) {
278f6679 2134 int depth = reiserfs_write_unlock_nested(tb->tb_sb);
ad31a4fc 2135 __wait_on_buffer(bh);
278f6679 2136 reiserfs_write_lock_nested(tb->tb_sb, depth);
a063ae17 2137 if (FILESYSTEM_CHANGED_TB(tb))
bd4c625c 2138 return REPEAT_SEARCH;
1da177e4 2139 }
1da177e4 2140
098297b2
JM
2141 /*
2142 * Parent in the path is unlocked and really parent
2143 * of the current node.
2144 */
2145 return CARRY_ON;
bd4c625c 2146}
1da177e4 2147
098297b2
JM
2148/*
2149 * Using lnum[h] and rnum[h] we should determine what neighbors
ee93961b
JM
2150 * of S[h] we
2151 * need in order to balance S[h], and get them if necessary.
1da177e4
LT
2152 * Returns: SCHEDULE_OCCURRED - schedule occurred while the function worked;
2153 * CARRY_ON - schedule didn't occur while the function worked;
2154 */
ee93961b 2155static int get_neighbors(struct tree_balance *tb, int h)
bd4c625c 2156{
ee93961b
JM
2157 int child_position,
2158 path_offset = PATH_H_PATH_OFFSET(tb->tb_path, h + 1);
2159 unsigned long son_number;
a063ae17 2160 struct super_block *sb = tb->tb_sb;
ad31a4fc 2161 struct buffer_head *bh;
278f6679 2162 int depth;
bd4c625c 2163
ee93961b 2164 PROC_INFO_INC(sb, get_neighbors[h]);
bd4c625c 2165
ee93961b
JM
2166 if (tb->lnum[h]) {
2167 /* We need left neighbor to balance S[h]. */
2168 PROC_INFO_INC(sb, need_l_neighbor[h]);
2169 bh = PATH_OFFSET_PBUFFER(tb->tb_path, path_offset);
bd4c625c 2170
ee93961b
JM
2171 RFALSE(bh == tb->FL[h] &&
2172 !PATH_OFFSET_POSITION(tb->tb_path, path_offset),
bd4c625c
LT
2173 "PAP-8270: invalid position in the parent");
2174
ee93961b 2175 child_position =
ad31a4fc 2176 (bh ==
ee93961b
JM
2177 tb->FL[h]) ? tb->lkey[h] : B_NR_ITEMS(tb->
2178 FL[h]);
2179 son_number = B_N_CHILD_NUM(tb->FL[h], child_position);
278f6679 2180 depth = reiserfs_write_unlock_nested(tb->tb_sb);
ee93961b 2181 bh = sb_bread(sb, son_number);
278f6679 2182 reiserfs_write_lock_nested(tb->tb_sb, depth);
ad31a4fc 2183 if (!bh)
bd4c625c 2184 return IO_ERROR;
a063ae17 2185 if (FILESYSTEM_CHANGED_TB(tb)) {
ad31a4fc 2186 brelse(bh);
ee93961b 2187 PROC_INFO_INC(sb, get_neighbors_restart[h]);
bd4c625c
LT
2188 return REPEAT_SEARCH;
2189 }
2190
ee93961b
JM
2191 RFALSE(!B_IS_IN_TREE(tb->FL[h]) ||
2192 child_position > B_NR_ITEMS(tb->FL[h]) ||
2193 B_N_CHILD_NUM(tb->FL[h], child_position) !=
ad31a4fc
JM
2194 bh->b_blocknr, "PAP-8275: invalid parent");
2195 RFALSE(!B_IS_IN_TREE(bh), "PAP-8280: invalid child");
ee93961b 2196 RFALSE(!h &&
ad31a4fc
JM
2197 B_FREE_SPACE(bh) !=
2198 MAX_CHILD_SIZE(bh) -
ee93961b 2199 dc_size(B_N_CHILD(tb->FL[0], child_position)),
bd4c625c
LT
2200 "PAP-8290: invalid child size of left neighbor");
2201
ee93961b
JM
2202 brelse(tb->L[h]);
2203 tb->L[h] = bh;
1da177e4 2204 }
bd4c625c 2205
ee93961b 2206 /* We need right neighbor to balance S[path_offset]. */
098297b2 2207 if (tb->rnum[h]) {
ee93961b
JM
2208 PROC_INFO_INC(sb, need_r_neighbor[h]);
2209 bh = PATH_OFFSET_PBUFFER(tb->tb_path, path_offset);
bd4c625c 2210
ee93961b 2211 RFALSE(bh == tb->FR[h] &&
a063ae17 2212 PATH_OFFSET_POSITION(tb->tb_path,
ee93961b 2213 path_offset) >=
ad31a4fc 2214 B_NR_ITEMS(bh),
bd4c625c
LT
2215 "PAP-8295: invalid position in the parent");
2216
ee93961b
JM
2217 child_position =
2218 (bh == tb->FR[h]) ? tb->rkey[h] + 1 : 0;
2219 son_number = B_N_CHILD_NUM(tb->FR[h], child_position);
278f6679 2220 depth = reiserfs_write_unlock_nested(tb->tb_sb);
ee93961b 2221 bh = sb_bread(sb, son_number);
278f6679 2222 reiserfs_write_lock_nested(tb->tb_sb, depth);
ad31a4fc 2223 if (!bh)
bd4c625c 2224 return IO_ERROR;
a063ae17 2225 if (FILESYSTEM_CHANGED_TB(tb)) {
ad31a4fc 2226 brelse(bh);
ee93961b 2227 PROC_INFO_INC(sb, get_neighbors_restart[h]);
bd4c625c
LT
2228 return REPEAT_SEARCH;
2229 }
ee93961b
JM
2230 brelse(tb->R[h]);
2231 tb->R[h] = bh;
bd4c625c 2232
ee93961b 2233 RFALSE(!h
ad31a4fc
JM
2234 && B_FREE_SPACE(bh) !=
2235 MAX_CHILD_SIZE(bh) -
ee93961b 2236 dc_size(B_N_CHILD(tb->FR[0], child_position)),
bd4c625c 2237 "PAP-8300: invalid child size of right neighbor (%d != %d - %d)",
ad31a4fc 2238 B_FREE_SPACE(bh), MAX_CHILD_SIZE(bh),
ee93961b 2239 dc_size(B_N_CHILD(tb->FR[0], child_position)));
bd4c625c 2240
1da177e4 2241 }
bd4c625c 2242 return CARRY_ON;
1da177e4
LT
2243}
2244
bd4c625c 2245static int get_virtual_node_size(struct super_block *sb, struct buffer_head *bh)
1da177e4 2246{
bd4c625c
LT
2247 int max_num_of_items;
2248 int max_num_of_entries;
2249 unsigned long blocksize = sb->s_blocksize;
1da177e4
LT
2250
2251#define MIN_NAME_LEN 1
2252
bd4c625c
LT
2253 max_num_of_items = (blocksize - BLKH_SIZE) / (IH_SIZE + MIN_ITEM_LEN);
2254 max_num_of_entries = (blocksize - BLKH_SIZE - IH_SIZE) /
2255 (DEH_SIZE + MIN_NAME_LEN);
1da177e4 2256
bd4c625c
LT
2257 return sizeof(struct virtual_node) +
2258 max(max_num_of_items * sizeof(struct virtual_item),
2259 sizeof(struct virtual_item) + sizeof(struct direntry_uarea) +
2260 (max_num_of_entries - 1) * sizeof(__u16));
1da177e4
LT
2261}
2262
098297b2
JM
2263/*
2264 * maybe we should fail balancing we are going to perform when kmalloc
2265 * fails several times. But now it will loop until kmalloc gets
2266 * required memory
2267 */
bd4c625c 2268static int get_mem_for_virtual_node(struct tree_balance *tb)
1da177e4 2269{
bd4c625c
LT
2270 int check_fs = 0;
2271 int size;
2272 char *buf;
2273
2274 size = get_virtual_node_size(tb->tb_sb, PATH_PLAST_BUFFER(tb->tb_path));
2275
098297b2 2276 /* we have to allocate more memory for virtual node */
bd4c625c 2277 if (size > tb->vn_buf_size) {
bd4c625c
LT
2278 if (tb->vn_buf) {
2279 /* free memory allocated before */
d739b42b 2280 kfree(tb->vn_buf);
bd4c625c
LT
2281 /* this is not needed if kfree is atomic */
2282 check_fs = 1;
2283 }
1da177e4 2284
bd4c625c
LT
2285 /* virtual node requires now more memory */
2286 tb->vn_buf_size = size;
2287
2288 /* get memory for virtual item */
d739b42b 2289 buf = kmalloc(size, GFP_ATOMIC | __GFP_NOWARN);
bd4c625c 2290 if (!buf) {
098297b2
JM
2291 /*
2292 * getting memory with GFP_KERNEL priority may involve
2293 * balancing now (due to indirect_to_direct conversion
2294 * on dcache shrinking). So, release path and collected
2295 * resources here
2296 */
bd4c625c 2297 free_buffers_in_tb(tb);
d739b42b 2298 buf = kmalloc(size, GFP_NOFS);
bd4c625c 2299 if (!buf) {
bd4c625c
LT
2300 tb->vn_buf_size = 0;
2301 }
2302 tb->vn_buf = buf;
2303 schedule();
2304 return REPEAT_SEARCH;
2305 }
1da177e4 2306
bd4c625c
LT
2307 tb->vn_buf = buf;
2308 }
1da177e4 2309
bd4c625c
LT
2310 if (check_fs && FILESYSTEM_CHANGED_TB(tb))
2311 return REPEAT_SEARCH;
1da177e4 2312
bd4c625c 2313 return CARRY_ON;
1da177e4
LT
2314}
2315
1da177e4 2316#ifdef CONFIG_REISERFS_CHECK
a9dd3643 2317static void tb_buffer_sanity_check(struct super_block *sb,
ad31a4fc 2318 struct buffer_head *bh,
bd4c625c 2319 const char *descr, int level)
1da177e4 2320{
ad31a4fc
JM
2321 if (bh) {
2322 if (atomic_read(&(bh->b_count)) <= 0)
1da177e4 2323
a9dd3643 2324 reiserfs_panic(sb, "jmacd-1", "negative or zero "
c3a9c210 2325 "reference counter for buffer %s[%d] "
ad31a4fc 2326 "(%b)", descr, level, bh);
1da177e4 2327
ad31a4fc 2328 if (!buffer_uptodate(bh))
a9dd3643 2329 reiserfs_panic(sb, "jmacd-2", "buffer is not up "
c3a9c210 2330 "to date %s[%d] (%b)",
ad31a4fc 2331 descr, level, bh);
1da177e4 2332
ad31a4fc 2333 if (!B_IS_IN_TREE(bh))
a9dd3643 2334 reiserfs_panic(sb, "jmacd-3", "buffer is not "
c3a9c210 2335 "in tree %s[%d] (%b)",
ad31a4fc 2336 descr, level, bh);
1da177e4 2337
ad31a4fc 2338 if (bh->b_bdev != sb->s_bdev)
a9dd3643 2339 reiserfs_panic(sb, "jmacd-4", "buffer has wrong "
c3a9c210 2340 "device %s[%d] (%b)",
ad31a4fc 2341 descr, level, bh);
1da177e4 2342
ad31a4fc 2343 if (bh->b_size != sb->s_blocksize)
a9dd3643 2344 reiserfs_panic(sb, "jmacd-5", "buffer has wrong "
c3a9c210 2345 "blocksize %s[%d] (%b)",
ad31a4fc 2346 descr, level, bh);
1da177e4 2347
ad31a4fc 2348 if (bh->b_blocknr > SB_BLOCK_COUNT(sb))
a9dd3643 2349 reiserfs_panic(sb, "jmacd-6", "buffer block "
c3a9c210 2350 "number too high %s[%d] (%b)",
ad31a4fc 2351 descr, level, bh);
bd4c625c
LT
2352 }
2353}
2354#else
a9dd3643 2355static void tb_buffer_sanity_check(struct super_block *sb,
ad31a4fc 2356 struct buffer_head *bh,
bd4c625c
LT
2357 const char *descr, int level)
2358{;
2359}
2360#endif
1da177e4 2361
bd4c625c
LT
2362static int clear_all_dirty_bits(struct super_block *s, struct buffer_head *bh)
2363{
2364 return reiserfs_prepare_for_journal(s, bh, 0);
2365}
1da177e4 2366
a063ae17 2367static int wait_tb_buffers_until_unlocked(struct tree_balance *tb)
bd4c625c
LT
2368{
2369 struct buffer_head *locked;
2370#ifdef CONFIG_REISERFS_CHECK
2371 int repeat_counter = 0;
2372#endif
2373 int i;
1da177e4 2374
bd4c625c 2375 do {
1da177e4 2376
bd4c625c
LT
2377 locked = NULL;
2378
a063ae17 2379 for (i = tb->tb_path->path_length;
bd4c625c 2380 !locked && i > ILLEGAL_PATH_ELEMENT_OFFSET; i--) {
a063ae17 2381 if (PATH_OFFSET_PBUFFER(tb->tb_path, i)) {
098297b2
JM
2382 /*
2383 * if I understand correctly, we can only
2384 * be sure the last buffer in the path is
2385 * in the tree --clm
bd4c625c
LT
2386 */
2387#ifdef CONFIG_REISERFS_CHECK
a063ae17
JM
2388 if (PATH_PLAST_BUFFER(tb->tb_path) ==
2389 PATH_OFFSET_PBUFFER(tb->tb_path, i))
2390 tb_buffer_sanity_check(tb->tb_sb,
bd4c625c 2391 PATH_OFFSET_PBUFFER
a063ae17 2392 (tb->tb_path,
bd4c625c 2393 i), "S",
a063ae17 2394 tb->tb_path->
bd4c625c 2395 path_length - i);
bd4c625c 2396#endif
a063ae17 2397 if (!clear_all_dirty_bits(tb->tb_sb,
bd4c625c 2398 PATH_OFFSET_PBUFFER
a063ae17 2399 (tb->tb_path,
bd4c625c
LT
2400 i))) {
2401 locked =
a063ae17 2402 PATH_OFFSET_PBUFFER(tb->tb_path,
bd4c625c
LT
2403 i);
2404 }
2405 }
1da177e4
LT
2406 }
2407
a063ae17 2408 for (i = 0; !locked && i < MAX_HEIGHT && tb->insert_size[i];
bd4c625c
LT
2409 i++) {
2410
a063ae17 2411 if (tb->lnum[i]) {
bd4c625c 2412
a063ae17
JM
2413 if (tb->L[i]) {
2414 tb_buffer_sanity_check(tb->tb_sb,
2415 tb->L[i],
bd4c625c
LT
2416 "L", i);
2417 if (!clear_all_dirty_bits
a063ae17
JM
2418 (tb->tb_sb, tb->L[i]))
2419 locked = tb->L[i];
bd4c625c
LT
2420 }
2421
a063ae17
JM
2422 if (!locked && tb->FL[i]) {
2423 tb_buffer_sanity_check(tb->tb_sb,
2424 tb->FL[i],
bd4c625c
LT
2425 "FL", i);
2426 if (!clear_all_dirty_bits
a063ae17
JM
2427 (tb->tb_sb, tb->FL[i]))
2428 locked = tb->FL[i];
bd4c625c
LT
2429 }
2430
a063ae17
JM
2431 if (!locked && tb->CFL[i]) {
2432 tb_buffer_sanity_check(tb->tb_sb,
2433 tb->CFL[i],
bd4c625c
LT
2434 "CFL", i);
2435 if (!clear_all_dirty_bits
a063ae17
JM
2436 (tb->tb_sb, tb->CFL[i]))
2437 locked = tb->CFL[i];
bd4c625c
LT
2438 }
2439
2440 }
2441
a063ae17 2442 if (!locked && (tb->rnum[i])) {
bd4c625c 2443
a063ae17
JM
2444 if (tb->R[i]) {
2445 tb_buffer_sanity_check(tb->tb_sb,
2446 tb->R[i],
bd4c625c
LT
2447 "R", i);
2448 if (!clear_all_dirty_bits
a063ae17
JM
2449 (tb->tb_sb, tb->R[i]))
2450 locked = tb->R[i];
bd4c625c
LT
2451 }
2452
a063ae17
JM
2453 if (!locked && tb->FR[i]) {
2454 tb_buffer_sanity_check(tb->tb_sb,
2455 tb->FR[i],
bd4c625c
LT
2456 "FR", i);
2457 if (!clear_all_dirty_bits
a063ae17
JM
2458 (tb->tb_sb, tb->FR[i]))
2459 locked = tb->FR[i];
bd4c625c
LT
2460 }
2461
a063ae17
JM
2462 if (!locked && tb->CFR[i]) {
2463 tb_buffer_sanity_check(tb->tb_sb,
2464 tb->CFR[i],
bd4c625c
LT
2465 "CFR", i);
2466 if (!clear_all_dirty_bits
a063ae17
JM
2467 (tb->tb_sb, tb->CFR[i]))
2468 locked = tb->CFR[i];
bd4c625c
LT
2469 }
2470 }
2471 }
098297b2
JM
2472
2473 /*
2474 * as far as I can tell, this is not required. The FEB list
2475 * seems to be full of newly allocated nodes, which will
2476 * never be locked, dirty, or anything else.
2477 * To be safe, I'm putting in the checks and waits in.
2478 * For the moment, they are needed to keep the code in
2479 * journal.c from complaining about the buffer.
2480 * That code is inside CONFIG_REISERFS_CHECK as well. --clm
bd4c625c
LT
2481 */
2482 for (i = 0; !locked && i < MAX_FEB_SIZE; i++) {
a063ae17 2483 if (tb->FEB[i]) {
bd4c625c 2484 if (!clear_all_dirty_bits
a063ae17
JM
2485 (tb->tb_sb, tb->FEB[i]))
2486 locked = tb->FEB[i];
bd4c625c 2487 }
1da177e4 2488 }
1da177e4 2489
bd4c625c 2490 if (locked) {
278f6679 2491 int depth;
1da177e4 2492#ifdef CONFIG_REISERFS_CHECK
bd4c625c
LT
2493 repeat_counter++;
2494 if ((repeat_counter % 10000) == 0) {
a063ae17 2495 reiserfs_warning(tb->tb_sb, "reiserfs-8200",
45b03d5e
JM
2496 "too many iterations waiting "
2497 "for buffer to unlock "
bd4c625c
LT
2498 "(%b)", locked);
2499
2500 /* Don't loop forever. Try to recover from possible error. */
2501
a063ae17 2502 return (FILESYSTEM_CHANGED_TB(tb)) ?
bd4c625c
LT
2503 REPEAT_SEARCH : CARRY_ON;
2504 }
1da177e4 2505#endif
278f6679 2506 depth = reiserfs_write_unlock_nested(tb->tb_sb);
bd4c625c 2507 __wait_on_buffer(locked);
278f6679 2508 reiserfs_write_lock_nested(tb->tb_sb, depth);
a063ae17 2509 if (FILESYSTEM_CHANGED_TB(tb))
bd4c625c 2510 return REPEAT_SEARCH;
bd4c625c 2511 }
1da177e4 2512
bd4c625c 2513 } while (locked);
1da177e4 2514
bd4c625c 2515 return CARRY_ON;
1da177e4
LT
2516}
2517
098297b2
JM
2518/*
2519 * Prepare for balancing, that is
1da177e4
LT
2520 * get all necessary parents, and neighbors;
2521 * analyze what and where should be moved;
2522 * get sufficient number of new nodes;
2523 * Balancing will start only after all resources will be collected at a time.
0222e657 2524 *
1da177e4
LT
2525 * When ported to SMP kernels, only at the last moment after all needed nodes
2526 * are collected in cache, will the resources be locked using the usual
2527 * textbook ordered lock acquisition algorithms. Note that ensuring that
098297b2
JM
2528 * this code neither write locks what it does not need to write lock nor locks
2529 * out of order will be a pain in the butt that could have been avoided.
2530 * Grumble grumble. -Hans
0222e657 2531 *
1da177e4 2532 * fix is meant in the sense of render unchanging
0222e657 2533 *
098297b2
JM
2534 * Latency might be improved by first gathering a list of what buffers
2535 * are needed and then getting as many of them in parallel as possible? -Hans
1da177e4
LT
2536 *
2537 * Parameters:
2538 * op_mode i - insert, d - delete, c - cut (truncate), p - paste (append)
2539 * tb tree_balance structure;
2540 * inum item number in S[h];
2541 * pos_in_item - comment this if you can
a063ae17
JM
2542 * ins_ih item head of item being inserted
2543 * data inserted item or data to be pasted
1da177e4
LT
2544 * Returns: 1 - schedule occurred while the function worked;
2545 * 0 - schedule didn't occur while the function worked;
0222e657 2546 * -1 - if no_disk_space
1da177e4
LT
2547 */
2548
ee93961b 2549int fix_nodes(int op_mode, struct tree_balance *tb,
d68caa95 2550 struct item_head *ins_ih, const void *data)
bd4c625c 2551{
ee93961b
JM
2552 int ret, h, item_num = PATH_LAST_POSITION(tb->tb_path);
2553 int pos_in_item;
1da177e4 2554
098297b2
JM
2555 /*
2556 * we set wait_tb_buffers_run when we have to restore any dirty
2557 * bits cleared during wait_tb_buffers_run
bd4c625c
LT
2558 */
2559 int wait_tb_buffers_run = 0;
a063ae17 2560 struct buffer_head *tbS0 = PATH_PLAST_BUFFER(tb->tb_path);
1da177e4 2561
a063ae17 2562 ++REISERFS_SB(tb->tb_sb)->s_fix_nodes;
bd4c625c 2563
ee93961b 2564 pos_in_item = tb->tb_path->pos_in_item;
bd4c625c 2565
a063ae17 2566 tb->fs_gen = get_generation(tb->tb_sb);
1da177e4 2567
098297b2
JM
2568 /*
2569 * we prepare and log the super here so it will already be in the
2570 * transaction when do_balance needs to change it.
2571 * This way do_balance won't have to schedule when trying to prepare
2572 * the super for logging
bd4c625c 2573 */
a063ae17
JM
2574 reiserfs_prepare_for_journal(tb->tb_sb,
2575 SB_BUFFER_WITH_SB(tb->tb_sb), 1);
09f1b80b 2576 journal_mark_dirty(tb->transaction_handle,
a063ae17
JM
2577 SB_BUFFER_WITH_SB(tb->tb_sb));
2578 if (FILESYSTEM_CHANGED_TB(tb))
bd4c625c 2579 return REPEAT_SEARCH;
1da177e4 2580
bd4c625c 2581 /* if it possible in indirect_to_direct conversion */
a063ae17 2582 if (buffer_locked(tbS0)) {
278f6679 2583 int depth = reiserfs_write_unlock_nested(tb->tb_sb);
a063ae17 2584 __wait_on_buffer(tbS0);
278f6679 2585 reiserfs_write_lock_nested(tb->tb_sb, depth);
a063ae17 2586 if (FILESYSTEM_CHANGED_TB(tb))
bd4c625c
LT
2587 return REPEAT_SEARCH;
2588 }
2589#ifdef CONFIG_REISERFS_CHECK
08f14fc8 2590 if (REISERFS_SB(tb->tb_sb)->cur_tb) {
bd4c625c 2591 print_cur_tb("fix_nodes");
a063ae17 2592 reiserfs_panic(tb->tb_sb, "PAP-8305",
c3a9c210 2593 "there is pending do_balance");
bd4c625c 2594 }
1da177e4 2595
a063ae17
JM
2596 if (!buffer_uptodate(tbS0) || !B_IS_IN_TREE(tbS0))
2597 reiserfs_panic(tb->tb_sb, "PAP-8320", "S[0] (%b %z) is "
c3a9c210
JM
2598 "not uptodate at the beginning of fix_nodes "
2599 "or not in tree (mode %c)",
ee93961b 2600 tbS0, tbS0, op_mode);
1da177e4 2601
bd4c625c 2602 /* Check parameters. */
ee93961b 2603 switch (op_mode) {
bd4c625c 2604 case M_INSERT:
ee93961b 2605 if (item_num <= 0 || item_num > B_NR_ITEMS(tbS0))
a063ae17 2606 reiserfs_panic(tb->tb_sb, "PAP-8330", "Incorrect "
c3a9c210 2607 "item number %d (in S0 - %d) in case "
ee93961b 2608 "of insert", item_num,
a063ae17 2609 B_NR_ITEMS(tbS0));
bd4c625c
LT
2610 break;
2611 case M_PASTE:
2612 case M_DELETE:
2613 case M_CUT:
ee93961b 2614 if (item_num < 0 || item_num >= B_NR_ITEMS(tbS0)) {
a063ae17
JM
2615 print_block(tbS0, 0, -1, -1);
2616 reiserfs_panic(tb->tb_sb, "PAP-8335", "Incorrect "
c3a9c210
JM
2617 "item number(%d); mode = %c "
2618 "insert_size = %d",
ee93961b 2619 item_num, op_mode,
a063ae17 2620 tb->insert_size[0]);
1da177e4 2621 }
1da177e4 2622 break;
bd4c625c 2623 default:
a063ae17 2624 reiserfs_panic(tb->tb_sb, "PAP-8340", "Incorrect mode "
c3a9c210 2625 "of operation");
1da177e4 2626 }
bd4c625c 2627#endif
1da177e4 2628
a063ae17 2629 if (get_mem_for_virtual_node(tb) == REPEAT_SEARCH)
098297b2 2630 /* FIXME: maybe -ENOMEM when tb->vn_buf == 0? Now just repeat */
bd4c625c 2631 return REPEAT_SEARCH;
1da177e4 2632
ee93961b
JM
2633 /* Starting from the leaf level; for all levels h of the tree. */
2634 for (h = 0; h < MAX_HEIGHT && tb->insert_size[h]; h++) {
2635 ret = get_direct_parent(tb, h);
2636 if (ret != CARRY_ON)
bd4c625c 2637 goto repeat;
1da177e4 2638
ee93961b
JM
2639 ret = check_balance(op_mode, tb, h, item_num,
2640 pos_in_item, ins_ih, data);
2641 if (ret != CARRY_ON) {
2642 if (ret == NO_BALANCING_NEEDED) {
bd4c625c 2643 /* No balancing for higher levels needed. */
ee93961b
JM
2644 ret = get_neighbors(tb, h);
2645 if (ret != CARRY_ON)
bd4c625c 2646 goto repeat;
ee93961b
JM
2647 if (h != MAX_HEIGHT - 1)
2648 tb->insert_size[h + 1] = 0;
098297b2
JM
2649 /*
2650 * ok, analysis and resource gathering
2651 * are complete
2652 */
bd4c625c
LT
2653 break;
2654 }
2655 goto repeat;
2656 }
1da177e4 2657
ee93961b
JM
2658 ret = get_neighbors(tb, h);
2659 if (ret != CARRY_ON)
bd4c625c 2660 goto repeat;
bd4c625c 2661
098297b2
JM
2662 /*
2663 * No disk space, or schedule occurred and analysis may be
2664 * invalid and needs to be redone.
2665 */
ee93961b
JM
2666 ret = get_empty_nodes(tb, h);
2667 if (ret != CARRY_ON)
a063ae17 2668 goto repeat;
bd4c625c 2669
098297b2
JM
2670 /*
2671 * We have a positive insert size but no nodes exist on this
2672 * level, this means that we are creating a new root.
2673 */
ee93961b 2674 if (!PATH_H_PBUFFER(tb->tb_path, h)) {
bd4c625c 2675
ee93961b 2676 RFALSE(tb->blknum[h] != 1,
bd4c625c
LT
2677 "PAP-8350: creating new empty root");
2678
ee93961b
JM
2679 if (h < MAX_HEIGHT - 1)
2680 tb->insert_size[h + 1] = 0;
2681 } else if (!PATH_H_PBUFFER(tb->tb_path, h + 1)) {
098297b2
JM
2682 /*
2683 * The tree needs to be grown, so this node S[h]
2684 * which is the root node is split into two nodes,
2685 * and a new node (S[h+1]) will be created to
2686 * become the root node.
2687 */
ee93961b 2688 if (tb->blknum[h] > 1) {
bd4c625c 2689
ee93961b 2690 RFALSE(h == MAX_HEIGHT - 1,
bd4c625c
LT
2691 "PAP-8355: attempt to create too high of a tree");
2692
ee93961b 2693 tb->insert_size[h + 1] =
bd4c625c 2694 (DC_SIZE +
ee93961b 2695 KEY_SIZE) * (tb->blknum[h] - 1) +
bd4c625c 2696 DC_SIZE;
ee93961b
JM
2697 } else if (h < MAX_HEIGHT - 1)
2698 tb->insert_size[h + 1] = 0;
bd4c625c 2699 } else
ee93961b
JM
2700 tb->insert_size[h + 1] =
2701 (DC_SIZE + KEY_SIZE) * (tb->blknum[h] - 1);
1da177e4 2702 }
1da177e4 2703
ee93961b
JM
2704 ret = wait_tb_buffers_until_unlocked(tb);
2705 if (ret == CARRY_ON) {
a063ae17 2706 if (FILESYSTEM_CHANGED_TB(tb)) {
bd4c625c 2707 wait_tb_buffers_run = 1;
ee93961b 2708 ret = REPEAT_SEARCH;
bd4c625c
LT
2709 goto repeat;
2710 } else {
2711 return CARRY_ON;
2712 }
1da177e4 2713 } else {
bd4c625c
LT
2714 wait_tb_buffers_run = 1;
2715 goto repeat;
1da177e4
LT
2716 }
2717
cf776a7a 2718repeat:
098297b2
JM
2719 /*
2720 * fix_nodes was unable to perform its calculation due to
2721 * filesystem got changed under us, lack of free disk space or i/o
2722 * failure. If the first is the case - the search will be
2723 * repeated. For now - free all resources acquired so far except
2724 * for the new allocated nodes
2725 */
bd4c625c
LT
2726 {
2727 int i;
2728
2729 /* Release path buffers. */
2730 if (wait_tb_buffers_run) {
a063ae17 2731 pathrelse_and_restore(tb->tb_sb, tb->tb_path);
bd4c625c 2732 } else {
a063ae17 2733 pathrelse(tb->tb_path);
bd4c625c
LT
2734 }
2735 /* brelse all resources collected for balancing */
2736 for (i = 0; i < MAX_HEIGHT; i++) {
2737 if (wait_tb_buffers_run) {
a063ae17
JM
2738 reiserfs_restore_prepared_buffer(tb->tb_sb,
2739 tb->L[i]);
2740 reiserfs_restore_prepared_buffer(tb->tb_sb,
2741 tb->R[i]);
2742 reiserfs_restore_prepared_buffer(tb->tb_sb,
2743 tb->FL[i]);
2744 reiserfs_restore_prepared_buffer(tb->tb_sb,
2745 tb->FR[i]);
2746 reiserfs_restore_prepared_buffer(tb->tb_sb,
2747 tb->
bd4c625c 2748 CFL[i]);
a063ae17
JM
2749 reiserfs_restore_prepared_buffer(tb->tb_sb,
2750 tb->
bd4c625c
LT
2751 CFR[i]);
2752 }
2753
a063ae17
JM
2754 brelse(tb->L[i]);
2755 brelse(tb->R[i]);
2756 brelse(tb->FL[i]);
2757 brelse(tb->FR[i]);
2758 brelse(tb->CFL[i]);
2759 brelse(tb->CFR[i]);
2760
2761 tb->L[i] = NULL;
2762 tb->R[i] = NULL;
2763 tb->FL[i] = NULL;
2764 tb->FR[i] = NULL;
2765 tb->CFL[i] = NULL;
2766 tb->CFR[i] = NULL;
bd4c625c
LT
2767 }
2768
2769 if (wait_tb_buffers_run) {
2770 for (i = 0; i < MAX_FEB_SIZE; i++) {
a063ae17 2771 if (tb->FEB[i])
bd4c625c 2772 reiserfs_restore_prepared_buffer
a063ae17 2773 (tb->tb_sb, tb->FEB[i]);
bd4c625c 2774 }
1da177e4 2775 }
ee93961b 2776 return ret;
1da177e4 2777 }
1da177e4
LT
2778
2779}
2780
bd4c625c 2781void unfix_nodes(struct tree_balance *tb)
1da177e4 2782{
bd4c625c 2783 int i;
1da177e4 2784
bd4c625c
LT
2785 /* Release path buffers. */
2786 pathrelse_and_restore(tb->tb_sb, tb->tb_path);
1da177e4 2787
bd4c625c
LT
2788 /* brelse all resources collected for balancing */
2789 for (i = 0; i < MAX_HEIGHT; i++) {
2790 reiserfs_restore_prepared_buffer(tb->tb_sb, tb->L[i]);
2791 reiserfs_restore_prepared_buffer(tb->tb_sb, tb->R[i]);
2792 reiserfs_restore_prepared_buffer(tb->tb_sb, tb->FL[i]);
2793 reiserfs_restore_prepared_buffer(tb->tb_sb, tb->FR[i]);
2794 reiserfs_restore_prepared_buffer(tb->tb_sb, tb->CFL[i]);
2795 reiserfs_restore_prepared_buffer(tb->tb_sb, tb->CFR[i]);
2796
2797 brelse(tb->L[i]);
2798 brelse(tb->R[i]);
2799 brelse(tb->FL[i]);
2800 brelse(tb->FR[i]);
2801 brelse(tb->CFL[i]);
2802 brelse(tb->CFR[i]);
2803 }
1da177e4 2804
bd4c625c
LT
2805 /* deal with list of allocated (used and unused) nodes */
2806 for (i = 0; i < MAX_FEB_SIZE; i++) {
2807 if (tb->FEB[i]) {
2808 b_blocknr_t blocknr = tb->FEB[i]->b_blocknr;
098297b2
JM
2809 /*
2810 * de-allocated block which was not used by
2811 * balancing and bforget about buffer for it
2812 */
bd4c625c
LT
2813 brelse(tb->FEB[i]);
2814 reiserfs_free_block(tb->transaction_handle, NULL,
2815 blocknr, 0);
2816 }
2817 if (tb->used[i]) {
2818 /* release used as new nodes including a new root */
2819 brelse(tb->used[i]);
2820 }
2821 }
1da177e4 2822
d739b42b 2823 kfree(tb->vn_buf);
1da177e4 2824
bd4c625c 2825}