xref: /openbmc/linux/fs/xfs/libxfs/xfs_da_btree.c (revision 842b6b16)
1 /*
2  * Copyright (c) 2000-2005 Silicon Graphics, Inc.
3  * Copyright (c) 2013 Red Hat, Inc.
4  * All Rights Reserved.
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License as
8  * published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it would be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write the Free Software Foundation,
17  * Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18  */
19 #include "xfs.h"
20 #include "xfs_fs.h"
21 #include "xfs_shared.h"
22 #include "xfs_format.h"
23 #include "xfs_log_format.h"
24 #include "xfs_trans_resv.h"
25 #include "xfs_bit.h"
26 #include "xfs_mount.h"
27 #include "xfs_da_format.h"
28 #include "xfs_da_btree.h"
29 #include "xfs_dir2.h"
30 #include "xfs_dir2_priv.h"
31 #include "xfs_inode.h"
32 #include "xfs_trans.h"
33 #include "xfs_inode_item.h"
34 #include "xfs_alloc.h"
35 #include "xfs_bmap.h"
36 #include "xfs_attr.h"
37 #include "xfs_attr_leaf.h"
38 #include "xfs_error.h"
39 #include "xfs_trace.h"
40 #include "xfs_cksum.h"
41 #include "xfs_buf_item.h"
42 #include "xfs_log.h"
43 
44 /*
45  * xfs_da_btree.c
46  *
47  * Routines to implement directories as Btrees of hashed names.
48  */
49 
50 /*========================================================================
51  * Function prototypes for the kernel.
52  *========================================================================*/
53 
54 /*
55  * Routines used for growing the Btree.
56  */
57 STATIC int xfs_da3_root_split(xfs_da_state_t *state,
58 					    xfs_da_state_blk_t *existing_root,
59 					    xfs_da_state_blk_t *new_child);
60 STATIC int xfs_da3_node_split(xfs_da_state_t *state,
61 					    xfs_da_state_blk_t *existing_blk,
62 					    xfs_da_state_blk_t *split_blk,
63 					    xfs_da_state_blk_t *blk_to_add,
64 					    int treelevel,
65 					    int *result);
66 STATIC void xfs_da3_node_rebalance(xfs_da_state_t *state,
67 					 xfs_da_state_blk_t *node_blk_1,
68 					 xfs_da_state_blk_t *node_blk_2);
69 STATIC void xfs_da3_node_add(xfs_da_state_t *state,
70 				   xfs_da_state_blk_t *old_node_blk,
71 				   xfs_da_state_blk_t *new_node_blk);
72 
73 /*
74  * Routines used for shrinking the Btree.
75  */
76 STATIC int xfs_da3_root_join(xfs_da_state_t *state,
77 					   xfs_da_state_blk_t *root_blk);
78 STATIC int xfs_da3_node_toosmall(xfs_da_state_t *state, int *retval);
79 STATIC void xfs_da3_node_remove(xfs_da_state_t *state,
80 					      xfs_da_state_blk_t *drop_blk);
81 STATIC void xfs_da3_node_unbalance(xfs_da_state_t *state,
82 					 xfs_da_state_blk_t *src_node_blk,
83 					 xfs_da_state_blk_t *dst_node_blk);
84 
85 /*
86  * Utility routines.
87  */
88 STATIC int	xfs_da3_blk_unlink(xfs_da_state_t *state,
89 				  xfs_da_state_blk_t *drop_blk,
90 				  xfs_da_state_blk_t *save_blk);
91 
92 
93 kmem_zone_t *xfs_da_state_zone;	/* anchor for state struct zone */
94 
95 /*
96  * Allocate a dir-state structure.
97  * We don't put them on the stack since they're large.
98  */
99 xfs_da_state_t *
100 xfs_da_state_alloc(void)
101 {
102 	return kmem_zone_zalloc(xfs_da_state_zone, KM_NOFS);
103 }
104 
105 /*
106  * Kill the altpath contents of a da-state structure.
107  */
108 STATIC void
109 xfs_da_state_kill_altpath(xfs_da_state_t *state)
110 {
111 	int	i;
112 
113 	for (i = 0; i < state->altpath.active; i++)
114 		state->altpath.blk[i].bp = NULL;
115 	state->altpath.active = 0;
116 }
117 
118 /*
119  * Free a da-state structure.
120  */
121 void
122 xfs_da_state_free(xfs_da_state_t *state)
123 {
124 	xfs_da_state_kill_altpath(state);
125 #ifdef DEBUG
126 	memset((char *)state, 0, sizeof(*state));
127 #endif /* DEBUG */
128 	kmem_zone_free(xfs_da_state_zone, state);
129 }
130 
131 static bool
132 xfs_da3_node_verify(
133 	struct xfs_buf		*bp)
134 {
135 	struct xfs_mount	*mp = bp->b_target->bt_mount;
136 	struct xfs_da_intnode	*hdr = bp->b_addr;
137 	struct xfs_da3_icnode_hdr ichdr;
138 	const struct xfs_dir_ops *ops;
139 
140 	ops = xfs_dir_get_ops(mp, NULL);
141 
142 	ops->node_hdr_from_disk(&ichdr, hdr);
143 
144 	if (xfs_sb_version_hascrc(&mp->m_sb)) {
145 		struct xfs_da3_node_hdr *hdr3 = bp->b_addr;
146 
147 		if (ichdr.magic != XFS_DA3_NODE_MAGIC)
148 			return false;
149 
150 		if (!uuid_equal(&hdr3->info.uuid, &mp->m_sb.sb_meta_uuid))
151 			return false;
152 		if (be64_to_cpu(hdr3->info.blkno) != bp->b_bn)
153 			return false;
154 		if (!xfs_log_check_lsn(mp, be64_to_cpu(hdr3->info.lsn)))
155 			return false;
156 	} else {
157 		if (ichdr.magic != XFS_DA_NODE_MAGIC)
158 			return false;
159 	}
160 	if (ichdr.level == 0)
161 		return false;
162 	if (ichdr.level > XFS_DA_NODE_MAXDEPTH)
163 		return false;
164 	if (ichdr.count == 0)
165 		return false;
166 
167 	/*
168 	 * we don't know if the node is for and attribute or directory tree,
169 	 * so only fail if the count is outside both bounds
170 	 */
171 	if (ichdr.count > mp->m_dir_geo->node_ents &&
172 	    ichdr.count > mp->m_attr_geo->node_ents)
173 		return false;
174 
175 	/* XXX: hash order check? */
176 
177 	return true;
178 }
179 
180 static void
181 xfs_da3_node_write_verify(
182 	struct xfs_buf	*bp)
183 {
184 	struct xfs_mount	*mp = bp->b_target->bt_mount;
185 	struct xfs_buf_log_item	*bip = bp->b_fspriv;
186 	struct xfs_da3_node_hdr *hdr3 = bp->b_addr;
187 
188 	if (!xfs_da3_node_verify(bp)) {
189 		xfs_buf_ioerror(bp, -EFSCORRUPTED);
190 		xfs_verifier_error(bp);
191 		return;
192 	}
193 
194 	if (!xfs_sb_version_hascrc(&mp->m_sb))
195 		return;
196 
197 	if (bip)
198 		hdr3->info.lsn = cpu_to_be64(bip->bli_item.li_lsn);
199 
200 	xfs_buf_update_cksum(bp, XFS_DA3_NODE_CRC_OFF);
201 }
202 
203 /*
204  * leaf/node format detection on trees is sketchy, so a node read can be done on
205  * leaf level blocks when detection identifies the tree as a node format tree
206  * incorrectly. In this case, we need to swap the verifier to match the correct
207  * format of the block being read.
208  */
209 static void
210 xfs_da3_node_read_verify(
211 	struct xfs_buf		*bp)
212 {
213 	struct xfs_da_blkinfo	*info = bp->b_addr;
214 
215 	switch (be16_to_cpu(info->magic)) {
216 		case XFS_DA3_NODE_MAGIC:
217 			if (!xfs_buf_verify_cksum(bp, XFS_DA3_NODE_CRC_OFF)) {
218 				xfs_buf_ioerror(bp, -EFSBADCRC);
219 				break;
220 			}
221 			/* fall through */
222 		case XFS_DA_NODE_MAGIC:
223 			if (!xfs_da3_node_verify(bp)) {
224 				xfs_buf_ioerror(bp, -EFSCORRUPTED);
225 				break;
226 			}
227 			return;
228 		case XFS_ATTR_LEAF_MAGIC:
229 		case XFS_ATTR3_LEAF_MAGIC:
230 			bp->b_ops = &xfs_attr3_leaf_buf_ops;
231 			bp->b_ops->verify_read(bp);
232 			return;
233 		case XFS_DIR2_LEAFN_MAGIC:
234 		case XFS_DIR3_LEAFN_MAGIC:
235 			bp->b_ops = &xfs_dir3_leafn_buf_ops;
236 			bp->b_ops->verify_read(bp);
237 			return;
238 		default:
239 			xfs_buf_ioerror(bp, -EFSCORRUPTED);
240 			break;
241 	}
242 
243 	/* corrupt block */
244 	xfs_verifier_error(bp);
245 }
246 
247 const struct xfs_buf_ops xfs_da3_node_buf_ops = {
248 	.verify_read = xfs_da3_node_read_verify,
249 	.verify_write = xfs_da3_node_write_verify,
250 };
251 
252 int
253 xfs_da3_node_read(
254 	struct xfs_trans	*tp,
255 	struct xfs_inode	*dp,
256 	xfs_dablk_t		bno,
257 	xfs_daddr_t		mappedbno,
258 	struct xfs_buf		**bpp,
259 	int			which_fork)
260 {
261 	int			err;
262 
263 	err = xfs_da_read_buf(tp, dp, bno, mappedbno, bpp,
264 					which_fork, &xfs_da3_node_buf_ops);
265 	if (!err && tp) {
266 		struct xfs_da_blkinfo	*info = (*bpp)->b_addr;
267 		int			type;
268 
269 		switch (be16_to_cpu(info->magic)) {
270 		case XFS_DA_NODE_MAGIC:
271 		case XFS_DA3_NODE_MAGIC:
272 			type = XFS_BLFT_DA_NODE_BUF;
273 			break;
274 		case XFS_ATTR_LEAF_MAGIC:
275 		case XFS_ATTR3_LEAF_MAGIC:
276 			type = XFS_BLFT_ATTR_LEAF_BUF;
277 			break;
278 		case XFS_DIR2_LEAFN_MAGIC:
279 		case XFS_DIR3_LEAFN_MAGIC:
280 			type = XFS_BLFT_DIR_LEAFN_BUF;
281 			break;
282 		default:
283 			type = 0;
284 			ASSERT(0);
285 			break;
286 		}
287 		xfs_trans_buf_set_type(tp, *bpp, type);
288 	}
289 	return err;
290 }
291 
292 /*========================================================================
293  * Routines used for growing the Btree.
294  *========================================================================*/
295 
296 /*
297  * Create the initial contents of an intermediate node.
298  */
299 int
300 xfs_da3_node_create(
301 	struct xfs_da_args	*args,
302 	xfs_dablk_t		blkno,
303 	int			level,
304 	struct xfs_buf		**bpp,
305 	int			whichfork)
306 {
307 	struct xfs_da_intnode	*node;
308 	struct xfs_trans	*tp = args->trans;
309 	struct xfs_mount	*mp = tp->t_mountp;
310 	struct xfs_da3_icnode_hdr ichdr = {0};
311 	struct xfs_buf		*bp;
312 	int			error;
313 	struct xfs_inode	*dp = args->dp;
314 
315 	trace_xfs_da_node_create(args);
316 	ASSERT(level <= XFS_DA_NODE_MAXDEPTH);
317 
318 	error = xfs_da_get_buf(tp, dp, blkno, -1, &bp, whichfork);
319 	if (error)
320 		return error;
321 	bp->b_ops = &xfs_da3_node_buf_ops;
322 	xfs_trans_buf_set_type(tp, bp, XFS_BLFT_DA_NODE_BUF);
323 	node = bp->b_addr;
324 
325 	if (xfs_sb_version_hascrc(&mp->m_sb)) {
326 		struct xfs_da3_node_hdr *hdr3 = bp->b_addr;
327 
328 		memset(hdr3, 0, sizeof(struct xfs_da3_node_hdr));
329 		ichdr.magic = XFS_DA3_NODE_MAGIC;
330 		hdr3->info.blkno = cpu_to_be64(bp->b_bn);
331 		hdr3->info.owner = cpu_to_be64(args->dp->i_ino);
332 		uuid_copy(&hdr3->info.uuid, &mp->m_sb.sb_meta_uuid);
333 	} else {
334 		ichdr.magic = XFS_DA_NODE_MAGIC;
335 	}
336 	ichdr.level = level;
337 
338 	dp->d_ops->node_hdr_to_disk(node, &ichdr);
339 	xfs_trans_log_buf(tp, bp,
340 		XFS_DA_LOGRANGE(node, &node->hdr, dp->d_ops->node_hdr_size));
341 
342 	*bpp = bp;
343 	return 0;
344 }
345 
346 /*
347  * Split a leaf node, rebalance, then possibly split
348  * intermediate nodes, rebalance, etc.
349  */
350 int							/* error */
351 xfs_da3_split(
352 	struct xfs_da_state	*state)
353 {
354 	struct xfs_da_state_blk	*oldblk;
355 	struct xfs_da_state_blk	*newblk;
356 	struct xfs_da_state_blk	*addblk;
357 	struct xfs_da_intnode	*node;
358 	struct xfs_buf		*bp;
359 	int			max;
360 	int			action = 0;
361 	int			error;
362 	int			i;
363 
364 	trace_xfs_da_split(state->args);
365 
366 	/*
367 	 * Walk back up the tree splitting/inserting/adjusting as necessary.
368 	 * If we need to insert and there isn't room, split the node, then
369 	 * decide which fragment to insert the new block from below into.
370 	 * Note that we may split the root this way, but we need more fixup.
371 	 */
372 	max = state->path.active - 1;
373 	ASSERT((max >= 0) && (max < XFS_DA_NODE_MAXDEPTH));
374 	ASSERT(state->path.blk[max].magic == XFS_ATTR_LEAF_MAGIC ||
375 	       state->path.blk[max].magic == XFS_DIR2_LEAFN_MAGIC);
376 
377 	addblk = &state->path.blk[max];		/* initial dummy value */
378 	for (i = max; (i >= 0) && addblk; state->path.active--, i--) {
379 		oldblk = &state->path.blk[i];
380 		newblk = &state->altpath.blk[i];
381 
382 		/*
383 		 * If a leaf node then
384 		 *     Allocate a new leaf node, then rebalance across them.
385 		 * else if an intermediate node then
386 		 *     We split on the last layer, must we split the node?
387 		 */
388 		switch (oldblk->magic) {
389 		case XFS_ATTR_LEAF_MAGIC:
390 			error = xfs_attr3_leaf_split(state, oldblk, newblk);
391 			if ((error != 0) && (error != -ENOSPC)) {
392 				return error;	/* GROT: attr is inconsistent */
393 			}
394 			if (!error) {
395 				addblk = newblk;
396 				break;
397 			}
398 			/*
399 			 * Entry wouldn't fit, split the leaf again.
400 			 */
401 			state->extravalid = 1;
402 			if (state->inleaf) {
403 				state->extraafter = 0;	/* before newblk */
404 				trace_xfs_attr_leaf_split_before(state->args);
405 				error = xfs_attr3_leaf_split(state, oldblk,
406 							    &state->extrablk);
407 			} else {
408 				state->extraafter = 1;	/* after newblk */
409 				trace_xfs_attr_leaf_split_after(state->args);
410 				error = xfs_attr3_leaf_split(state, newblk,
411 							    &state->extrablk);
412 			}
413 			if (error)
414 				return error;	/* GROT: attr inconsistent */
415 			addblk = newblk;
416 			break;
417 		case XFS_DIR2_LEAFN_MAGIC:
418 			error = xfs_dir2_leafn_split(state, oldblk, newblk);
419 			if (error)
420 				return error;
421 			addblk = newblk;
422 			break;
423 		case XFS_DA_NODE_MAGIC:
424 			error = xfs_da3_node_split(state, oldblk, newblk, addblk,
425 							 max - i, &action);
426 			addblk->bp = NULL;
427 			if (error)
428 				return error;	/* GROT: dir is inconsistent */
429 			/*
430 			 * Record the newly split block for the next time thru?
431 			 */
432 			if (action)
433 				addblk = newblk;
434 			else
435 				addblk = NULL;
436 			break;
437 		}
438 
439 		/*
440 		 * Update the btree to show the new hashval for this child.
441 		 */
442 		xfs_da3_fixhashpath(state, &state->path);
443 	}
444 	if (!addblk)
445 		return 0;
446 
447 	/*
448 	 * Split the root node.
449 	 */
450 	ASSERT(state->path.active == 0);
451 	oldblk = &state->path.blk[0];
452 	error = xfs_da3_root_split(state, oldblk, addblk);
453 	if (error) {
454 		addblk->bp = NULL;
455 		return error;	/* GROT: dir is inconsistent */
456 	}
457 
458 	/*
459 	 * Update pointers to the node which used to be block 0 and
460 	 * just got bumped because of the addition of a new root node.
461 	 * There might be three blocks involved if a double split occurred,
462 	 * and the original block 0 could be at any position in the list.
463 	 *
464 	 * Note: the magic numbers and sibling pointers are in the same
465 	 * physical place for both v2 and v3 headers (by design). Hence it
466 	 * doesn't matter which version of the xfs_da_intnode structure we use
467 	 * here as the result will be the same using either structure.
468 	 */
469 	node = oldblk->bp->b_addr;
470 	if (node->hdr.info.forw) {
471 		if (be32_to_cpu(node->hdr.info.forw) == addblk->blkno) {
472 			bp = addblk->bp;
473 		} else {
474 			ASSERT(state->extravalid);
475 			bp = state->extrablk.bp;
476 		}
477 		node = bp->b_addr;
478 		node->hdr.info.back = cpu_to_be32(oldblk->blkno);
479 		xfs_trans_log_buf(state->args->trans, bp,
480 		    XFS_DA_LOGRANGE(node, &node->hdr.info,
481 		    sizeof(node->hdr.info)));
482 	}
483 	node = oldblk->bp->b_addr;
484 	if (node->hdr.info.back) {
485 		if (be32_to_cpu(node->hdr.info.back) == addblk->blkno) {
486 			bp = addblk->bp;
487 		} else {
488 			ASSERT(state->extravalid);
489 			bp = state->extrablk.bp;
490 		}
491 		node = bp->b_addr;
492 		node->hdr.info.forw = cpu_to_be32(oldblk->blkno);
493 		xfs_trans_log_buf(state->args->trans, bp,
494 		    XFS_DA_LOGRANGE(node, &node->hdr.info,
495 		    sizeof(node->hdr.info)));
496 	}
497 	addblk->bp = NULL;
498 	return 0;
499 }
500 
501 /*
502  * Split the root.  We have to create a new root and point to the two
503  * parts (the split old root) that we just created.  Copy block zero to
504  * the EOF, extending the inode in process.
505  */
506 STATIC int						/* error */
507 xfs_da3_root_split(
508 	struct xfs_da_state	*state,
509 	struct xfs_da_state_blk	*blk1,
510 	struct xfs_da_state_blk	*blk2)
511 {
512 	struct xfs_da_intnode	*node;
513 	struct xfs_da_intnode	*oldroot;
514 	struct xfs_da_node_entry *btree;
515 	struct xfs_da3_icnode_hdr nodehdr;
516 	struct xfs_da_args	*args;
517 	struct xfs_buf		*bp;
518 	struct xfs_inode	*dp;
519 	struct xfs_trans	*tp;
520 	struct xfs_dir2_leaf	*leaf;
521 	xfs_dablk_t		blkno;
522 	int			level;
523 	int			error;
524 	int			size;
525 
526 	trace_xfs_da_root_split(state->args);
527 
528 	/*
529 	 * Copy the existing (incorrect) block from the root node position
530 	 * to a free space somewhere.
531 	 */
532 	args = state->args;
533 	error = xfs_da_grow_inode(args, &blkno);
534 	if (error)
535 		return error;
536 
537 	dp = args->dp;
538 	tp = args->trans;
539 	error = xfs_da_get_buf(tp, dp, blkno, -1, &bp, args->whichfork);
540 	if (error)
541 		return error;
542 	node = bp->b_addr;
543 	oldroot = blk1->bp->b_addr;
544 	if (oldroot->hdr.info.magic == cpu_to_be16(XFS_DA_NODE_MAGIC) ||
545 	    oldroot->hdr.info.magic == cpu_to_be16(XFS_DA3_NODE_MAGIC)) {
546 		struct xfs_da3_icnode_hdr icnodehdr;
547 
548 		dp->d_ops->node_hdr_from_disk(&icnodehdr, oldroot);
549 		btree = dp->d_ops->node_tree_p(oldroot);
550 		size = (int)((char *)&btree[icnodehdr.count] - (char *)oldroot);
551 		level = icnodehdr.level;
552 
553 		/*
554 		 * we are about to copy oldroot to bp, so set up the type
555 		 * of bp while we know exactly what it will be.
556 		 */
557 		xfs_trans_buf_set_type(tp, bp, XFS_BLFT_DA_NODE_BUF);
558 	} else {
559 		struct xfs_dir3_icleaf_hdr leafhdr;
560 		struct xfs_dir2_leaf_entry *ents;
561 
562 		leaf = (xfs_dir2_leaf_t *)oldroot;
563 		dp->d_ops->leaf_hdr_from_disk(&leafhdr, leaf);
564 		ents = dp->d_ops->leaf_ents_p(leaf);
565 
566 		ASSERT(leafhdr.magic == XFS_DIR2_LEAFN_MAGIC ||
567 		       leafhdr.magic == XFS_DIR3_LEAFN_MAGIC);
568 		size = (int)((char *)&ents[leafhdr.count] - (char *)leaf);
569 		level = 0;
570 
571 		/*
572 		 * we are about to copy oldroot to bp, so set up the type
573 		 * of bp while we know exactly what it will be.
574 		 */
575 		xfs_trans_buf_set_type(tp, bp, XFS_BLFT_DIR_LEAFN_BUF);
576 	}
577 
578 	/*
579 	 * we can copy most of the information in the node from one block to
580 	 * another, but for CRC enabled headers we have to make sure that the
581 	 * block specific identifiers are kept intact. We update the buffer
582 	 * directly for this.
583 	 */
584 	memcpy(node, oldroot, size);
585 	if (oldroot->hdr.info.magic == cpu_to_be16(XFS_DA3_NODE_MAGIC) ||
586 	    oldroot->hdr.info.magic == cpu_to_be16(XFS_DIR3_LEAFN_MAGIC)) {
587 		struct xfs_da3_intnode *node3 = (struct xfs_da3_intnode *)node;
588 
589 		node3->hdr.info.blkno = cpu_to_be64(bp->b_bn);
590 	}
591 	xfs_trans_log_buf(tp, bp, 0, size - 1);
592 
593 	bp->b_ops = blk1->bp->b_ops;
594 	xfs_trans_buf_copy_type(bp, blk1->bp);
595 	blk1->bp = bp;
596 	blk1->blkno = blkno;
597 
598 	/*
599 	 * Set up the new root node.
600 	 */
601 	error = xfs_da3_node_create(args,
602 		(args->whichfork == XFS_DATA_FORK) ? args->geo->leafblk : 0,
603 		level + 1, &bp, args->whichfork);
604 	if (error)
605 		return error;
606 
607 	node = bp->b_addr;
608 	dp->d_ops->node_hdr_from_disk(&nodehdr, node);
609 	btree = dp->d_ops->node_tree_p(node);
610 	btree[0].hashval = cpu_to_be32(blk1->hashval);
611 	btree[0].before = cpu_to_be32(blk1->blkno);
612 	btree[1].hashval = cpu_to_be32(blk2->hashval);
613 	btree[1].before = cpu_to_be32(blk2->blkno);
614 	nodehdr.count = 2;
615 	dp->d_ops->node_hdr_to_disk(node, &nodehdr);
616 
617 #ifdef DEBUG
618 	if (oldroot->hdr.info.magic == cpu_to_be16(XFS_DIR2_LEAFN_MAGIC) ||
619 	    oldroot->hdr.info.magic == cpu_to_be16(XFS_DIR3_LEAFN_MAGIC)) {
620 		ASSERT(blk1->blkno >= args->geo->leafblk &&
621 		       blk1->blkno < args->geo->freeblk);
622 		ASSERT(blk2->blkno >= args->geo->leafblk &&
623 		       blk2->blkno < args->geo->freeblk);
624 	}
625 #endif
626 
627 	/* Header is already logged by xfs_da_node_create */
628 	xfs_trans_log_buf(tp, bp,
629 		XFS_DA_LOGRANGE(node, btree, sizeof(xfs_da_node_entry_t) * 2));
630 
631 	return 0;
632 }
633 
634 /*
635  * Split the node, rebalance, then add the new entry.
636  */
637 STATIC int						/* error */
638 xfs_da3_node_split(
639 	struct xfs_da_state	*state,
640 	struct xfs_da_state_blk	*oldblk,
641 	struct xfs_da_state_blk	*newblk,
642 	struct xfs_da_state_blk	*addblk,
643 	int			treelevel,
644 	int			*result)
645 {
646 	struct xfs_da_intnode	*node;
647 	struct xfs_da3_icnode_hdr nodehdr;
648 	xfs_dablk_t		blkno;
649 	int			newcount;
650 	int			error;
651 	int			useextra;
652 	struct xfs_inode	*dp = state->args->dp;
653 
654 	trace_xfs_da_node_split(state->args);
655 
656 	node = oldblk->bp->b_addr;
657 	dp->d_ops->node_hdr_from_disk(&nodehdr, node);
658 
659 	/*
660 	 * With V2 dirs the extra block is data or freespace.
661 	 */
662 	useextra = state->extravalid && state->args->whichfork == XFS_ATTR_FORK;
663 	newcount = 1 + useextra;
664 	/*
665 	 * Do we have to split the node?
666 	 */
667 	if (nodehdr.count + newcount > state->args->geo->node_ents) {
668 		/*
669 		 * Allocate a new node, add to the doubly linked chain of
670 		 * nodes, then move some of our excess entries into it.
671 		 */
672 		error = xfs_da_grow_inode(state->args, &blkno);
673 		if (error)
674 			return error;	/* GROT: dir is inconsistent */
675 
676 		error = xfs_da3_node_create(state->args, blkno, treelevel,
677 					   &newblk->bp, state->args->whichfork);
678 		if (error)
679 			return error;	/* GROT: dir is inconsistent */
680 		newblk->blkno = blkno;
681 		newblk->magic = XFS_DA_NODE_MAGIC;
682 		xfs_da3_node_rebalance(state, oldblk, newblk);
683 		error = xfs_da3_blk_link(state, oldblk, newblk);
684 		if (error)
685 			return error;
686 		*result = 1;
687 	} else {
688 		*result = 0;
689 	}
690 
691 	/*
692 	 * Insert the new entry(s) into the correct block
693 	 * (updating last hashval in the process).
694 	 *
695 	 * xfs_da3_node_add() inserts BEFORE the given index,
696 	 * and as a result of using node_lookup_int() we always
697 	 * point to a valid entry (not after one), but a split
698 	 * operation always results in a new block whose hashvals
699 	 * FOLLOW the current block.
700 	 *
701 	 * If we had double-split op below us, then add the extra block too.
702 	 */
703 	node = oldblk->bp->b_addr;
704 	dp->d_ops->node_hdr_from_disk(&nodehdr, node);
705 	if (oldblk->index <= nodehdr.count) {
706 		oldblk->index++;
707 		xfs_da3_node_add(state, oldblk, addblk);
708 		if (useextra) {
709 			if (state->extraafter)
710 				oldblk->index++;
711 			xfs_da3_node_add(state, oldblk, &state->extrablk);
712 			state->extravalid = 0;
713 		}
714 	} else {
715 		newblk->index++;
716 		xfs_da3_node_add(state, newblk, addblk);
717 		if (useextra) {
718 			if (state->extraafter)
719 				newblk->index++;
720 			xfs_da3_node_add(state, newblk, &state->extrablk);
721 			state->extravalid = 0;
722 		}
723 	}
724 
725 	return 0;
726 }
727 
728 /*
729  * Balance the btree elements between two intermediate nodes,
730  * usually one full and one empty.
731  *
732  * NOTE: if blk2 is empty, then it will get the upper half of blk1.
733  */
734 STATIC void
735 xfs_da3_node_rebalance(
736 	struct xfs_da_state	*state,
737 	struct xfs_da_state_blk	*blk1,
738 	struct xfs_da_state_blk	*blk2)
739 {
740 	struct xfs_da_intnode	*node1;
741 	struct xfs_da_intnode	*node2;
742 	struct xfs_da_intnode	*tmpnode;
743 	struct xfs_da_node_entry *btree1;
744 	struct xfs_da_node_entry *btree2;
745 	struct xfs_da_node_entry *btree_s;
746 	struct xfs_da_node_entry *btree_d;
747 	struct xfs_da3_icnode_hdr nodehdr1;
748 	struct xfs_da3_icnode_hdr nodehdr2;
749 	struct xfs_trans	*tp;
750 	int			count;
751 	int			tmp;
752 	int			swap = 0;
753 	struct xfs_inode	*dp = state->args->dp;
754 
755 	trace_xfs_da_node_rebalance(state->args);
756 
757 	node1 = blk1->bp->b_addr;
758 	node2 = blk2->bp->b_addr;
759 	dp->d_ops->node_hdr_from_disk(&nodehdr1, node1);
760 	dp->d_ops->node_hdr_from_disk(&nodehdr2, node2);
761 	btree1 = dp->d_ops->node_tree_p(node1);
762 	btree2 = dp->d_ops->node_tree_p(node2);
763 
764 	/*
765 	 * Figure out how many entries need to move, and in which direction.
766 	 * Swap the nodes around if that makes it simpler.
767 	 */
768 	if (nodehdr1.count > 0 && nodehdr2.count > 0 &&
769 	    ((be32_to_cpu(btree2[0].hashval) < be32_to_cpu(btree1[0].hashval)) ||
770 	     (be32_to_cpu(btree2[nodehdr2.count - 1].hashval) <
771 			be32_to_cpu(btree1[nodehdr1.count - 1].hashval)))) {
772 		tmpnode = node1;
773 		node1 = node2;
774 		node2 = tmpnode;
775 		dp->d_ops->node_hdr_from_disk(&nodehdr1, node1);
776 		dp->d_ops->node_hdr_from_disk(&nodehdr2, node2);
777 		btree1 = dp->d_ops->node_tree_p(node1);
778 		btree2 = dp->d_ops->node_tree_p(node2);
779 		swap = 1;
780 	}
781 
782 	count = (nodehdr1.count - nodehdr2.count) / 2;
783 	if (count == 0)
784 		return;
785 	tp = state->args->trans;
786 	/*
787 	 * Two cases: high-to-low and low-to-high.
788 	 */
789 	if (count > 0) {
790 		/*
791 		 * Move elements in node2 up to make a hole.
792 		 */
793 		tmp = nodehdr2.count;
794 		if (tmp > 0) {
795 			tmp *= (uint)sizeof(xfs_da_node_entry_t);
796 			btree_s = &btree2[0];
797 			btree_d = &btree2[count];
798 			memmove(btree_d, btree_s, tmp);
799 		}
800 
801 		/*
802 		 * Move the req'd B-tree elements from high in node1 to
803 		 * low in node2.
804 		 */
805 		nodehdr2.count += count;
806 		tmp = count * (uint)sizeof(xfs_da_node_entry_t);
807 		btree_s = &btree1[nodehdr1.count - count];
808 		btree_d = &btree2[0];
809 		memcpy(btree_d, btree_s, tmp);
810 		nodehdr1.count -= count;
811 	} else {
812 		/*
813 		 * Move the req'd B-tree elements from low in node2 to
814 		 * high in node1.
815 		 */
816 		count = -count;
817 		tmp = count * (uint)sizeof(xfs_da_node_entry_t);
818 		btree_s = &btree2[0];
819 		btree_d = &btree1[nodehdr1.count];
820 		memcpy(btree_d, btree_s, tmp);
821 		nodehdr1.count += count;
822 
823 		xfs_trans_log_buf(tp, blk1->bp,
824 			XFS_DA_LOGRANGE(node1, btree_d, tmp));
825 
826 		/*
827 		 * Move elements in node2 down to fill the hole.
828 		 */
829 		tmp  = nodehdr2.count - count;
830 		tmp *= (uint)sizeof(xfs_da_node_entry_t);
831 		btree_s = &btree2[count];
832 		btree_d = &btree2[0];
833 		memmove(btree_d, btree_s, tmp);
834 		nodehdr2.count -= count;
835 	}
836 
837 	/*
838 	 * Log header of node 1 and all current bits of node 2.
839 	 */
840 	dp->d_ops->node_hdr_to_disk(node1, &nodehdr1);
841 	xfs_trans_log_buf(tp, blk1->bp,
842 		XFS_DA_LOGRANGE(node1, &node1->hdr, dp->d_ops->node_hdr_size));
843 
844 	dp->d_ops->node_hdr_to_disk(node2, &nodehdr2);
845 	xfs_trans_log_buf(tp, blk2->bp,
846 		XFS_DA_LOGRANGE(node2, &node2->hdr,
847 				dp->d_ops->node_hdr_size +
848 				(sizeof(btree2[0]) * nodehdr2.count)));
849 
850 	/*
851 	 * Record the last hashval from each block for upward propagation.
852 	 * (note: don't use the swapped node pointers)
853 	 */
854 	if (swap) {
855 		node1 = blk1->bp->b_addr;
856 		node2 = blk2->bp->b_addr;
857 		dp->d_ops->node_hdr_from_disk(&nodehdr1, node1);
858 		dp->d_ops->node_hdr_from_disk(&nodehdr2, node2);
859 		btree1 = dp->d_ops->node_tree_p(node1);
860 		btree2 = dp->d_ops->node_tree_p(node2);
861 	}
862 	blk1->hashval = be32_to_cpu(btree1[nodehdr1.count - 1].hashval);
863 	blk2->hashval = be32_to_cpu(btree2[nodehdr2.count - 1].hashval);
864 
865 	/*
866 	 * Adjust the expected index for insertion.
867 	 */
868 	if (blk1->index >= nodehdr1.count) {
869 		blk2->index = blk1->index - nodehdr1.count;
870 		blk1->index = nodehdr1.count + 1;	/* make it invalid */
871 	}
872 }
873 
874 /*
875  * Add a new entry to an intermediate node.
876  */
877 STATIC void
878 xfs_da3_node_add(
879 	struct xfs_da_state	*state,
880 	struct xfs_da_state_blk	*oldblk,
881 	struct xfs_da_state_blk	*newblk)
882 {
883 	struct xfs_da_intnode	*node;
884 	struct xfs_da3_icnode_hdr nodehdr;
885 	struct xfs_da_node_entry *btree;
886 	int			tmp;
887 	struct xfs_inode	*dp = state->args->dp;
888 
889 	trace_xfs_da_node_add(state->args);
890 
891 	node = oldblk->bp->b_addr;
892 	dp->d_ops->node_hdr_from_disk(&nodehdr, node);
893 	btree = dp->d_ops->node_tree_p(node);
894 
895 	ASSERT(oldblk->index >= 0 && oldblk->index <= nodehdr.count);
896 	ASSERT(newblk->blkno != 0);
897 	if (state->args->whichfork == XFS_DATA_FORK)
898 		ASSERT(newblk->blkno >= state->args->geo->leafblk &&
899 		       newblk->blkno < state->args->geo->freeblk);
900 
901 	/*
902 	 * We may need to make some room before we insert the new node.
903 	 */
904 	tmp = 0;
905 	if (oldblk->index < nodehdr.count) {
906 		tmp = (nodehdr.count - oldblk->index) * (uint)sizeof(*btree);
907 		memmove(&btree[oldblk->index + 1], &btree[oldblk->index], tmp);
908 	}
909 	btree[oldblk->index].hashval = cpu_to_be32(newblk->hashval);
910 	btree[oldblk->index].before = cpu_to_be32(newblk->blkno);
911 	xfs_trans_log_buf(state->args->trans, oldblk->bp,
912 		XFS_DA_LOGRANGE(node, &btree[oldblk->index],
913 				tmp + sizeof(*btree)));
914 
915 	nodehdr.count += 1;
916 	dp->d_ops->node_hdr_to_disk(node, &nodehdr);
917 	xfs_trans_log_buf(state->args->trans, oldblk->bp,
918 		XFS_DA_LOGRANGE(node, &node->hdr, dp->d_ops->node_hdr_size));
919 
920 	/*
921 	 * Copy the last hash value from the oldblk to propagate upwards.
922 	 */
923 	oldblk->hashval = be32_to_cpu(btree[nodehdr.count - 1].hashval);
924 }
925 
926 /*========================================================================
927  * Routines used for shrinking the Btree.
928  *========================================================================*/
929 
930 /*
931  * Deallocate an empty leaf node, remove it from its parent,
932  * possibly deallocating that block, etc...
933  */
934 int
935 xfs_da3_join(
936 	struct xfs_da_state	*state)
937 {
938 	struct xfs_da_state_blk	*drop_blk;
939 	struct xfs_da_state_blk	*save_blk;
940 	int			action = 0;
941 	int			error;
942 
943 	trace_xfs_da_join(state->args);
944 
945 	drop_blk = &state->path.blk[ state->path.active-1 ];
946 	save_blk = &state->altpath.blk[ state->path.active-1 ];
947 	ASSERT(state->path.blk[0].magic == XFS_DA_NODE_MAGIC);
948 	ASSERT(drop_blk->magic == XFS_ATTR_LEAF_MAGIC ||
949 	       drop_blk->magic == XFS_DIR2_LEAFN_MAGIC);
950 
951 	/*
952 	 * Walk back up the tree joining/deallocating as necessary.
953 	 * When we stop dropping blocks, break out.
954 	 */
955 	for (  ; state->path.active >= 2; drop_blk--, save_blk--,
956 		 state->path.active--) {
957 		/*
958 		 * See if we can combine the block with a neighbor.
959 		 *   (action == 0) => no options, just leave
960 		 *   (action == 1) => coalesce, then unlink
961 		 *   (action == 2) => block empty, unlink it
962 		 */
963 		switch (drop_blk->magic) {
964 		case XFS_ATTR_LEAF_MAGIC:
965 			error = xfs_attr3_leaf_toosmall(state, &action);
966 			if (error)
967 				return error;
968 			if (action == 0)
969 				return 0;
970 			xfs_attr3_leaf_unbalance(state, drop_blk, save_blk);
971 			break;
972 		case XFS_DIR2_LEAFN_MAGIC:
973 			error = xfs_dir2_leafn_toosmall(state, &action);
974 			if (error)
975 				return error;
976 			if (action == 0)
977 				return 0;
978 			xfs_dir2_leafn_unbalance(state, drop_blk, save_blk);
979 			break;
980 		case XFS_DA_NODE_MAGIC:
981 			/*
982 			 * Remove the offending node, fixup hashvals,
983 			 * check for a toosmall neighbor.
984 			 */
985 			xfs_da3_node_remove(state, drop_blk);
986 			xfs_da3_fixhashpath(state, &state->path);
987 			error = xfs_da3_node_toosmall(state, &action);
988 			if (error)
989 				return error;
990 			if (action == 0)
991 				return 0;
992 			xfs_da3_node_unbalance(state, drop_blk, save_blk);
993 			break;
994 		}
995 		xfs_da3_fixhashpath(state, &state->altpath);
996 		error = xfs_da3_blk_unlink(state, drop_blk, save_blk);
997 		xfs_da_state_kill_altpath(state);
998 		if (error)
999 			return error;
1000 		error = xfs_da_shrink_inode(state->args, drop_blk->blkno,
1001 							 drop_blk->bp);
1002 		drop_blk->bp = NULL;
1003 		if (error)
1004 			return error;
1005 	}
1006 	/*
1007 	 * We joined all the way to the top.  If it turns out that
1008 	 * we only have one entry in the root, make the child block
1009 	 * the new root.
1010 	 */
1011 	xfs_da3_node_remove(state, drop_blk);
1012 	xfs_da3_fixhashpath(state, &state->path);
1013 	error = xfs_da3_root_join(state, &state->path.blk[0]);
1014 	return error;
1015 }
1016 
1017 #ifdef	DEBUG
1018 static void
1019 xfs_da_blkinfo_onlychild_validate(struct xfs_da_blkinfo *blkinfo, __u16 level)
1020 {
1021 	__be16	magic = blkinfo->magic;
1022 
1023 	if (level == 1) {
1024 		ASSERT(magic == cpu_to_be16(XFS_DIR2_LEAFN_MAGIC) ||
1025 		       magic == cpu_to_be16(XFS_DIR3_LEAFN_MAGIC) ||
1026 		       magic == cpu_to_be16(XFS_ATTR_LEAF_MAGIC) ||
1027 		       magic == cpu_to_be16(XFS_ATTR3_LEAF_MAGIC));
1028 	} else {
1029 		ASSERT(magic == cpu_to_be16(XFS_DA_NODE_MAGIC) ||
1030 		       magic == cpu_to_be16(XFS_DA3_NODE_MAGIC));
1031 	}
1032 	ASSERT(!blkinfo->forw);
1033 	ASSERT(!blkinfo->back);
1034 }
1035 #else	/* !DEBUG */
1036 #define	xfs_da_blkinfo_onlychild_validate(blkinfo, level)
1037 #endif	/* !DEBUG */
1038 
1039 /*
1040  * We have only one entry in the root.  Copy the only remaining child of
1041  * the old root to block 0 as the new root node.
1042  */
1043 STATIC int
1044 xfs_da3_root_join(
1045 	struct xfs_da_state	*state,
1046 	struct xfs_da_state_blk	*root_blk)
1047 {
1048 	struct xfs_da_intnode	*oldroot;
1049 	struct xfs_da_args	*args;
1050 	xfs_dablk_t		child;
1051 	struct xfs_buf		*bp;
1052 	struct xfs_da3_icnode_hdr oldroothdr;
1053 	struct xfs_da_node_entry *btree;
1054 	int			error;
1055 	struct xfs_inode	*dp = state->args->dp;
1056 
1057 	trace_xfs_da_root_join(state->args);
1058 
1059 	ASSERT(root_blk->magic == XFS_DA_NODE_MAGIC);
1060 
1061 	args = state->args;
1062 	oldroot = root_blk->bp->b_addr;
1063 	dp->d_ops->node_hdr_from_disk(&oldroothdr, oldroot);
1064 	ASSERT(oldroothdr.forw == 0);
1065 	ASSERT(oldroothdr.back == 0);
1066 
1067 	/*
1068 	 * If the root has more than one child, then don't do anything.
1069 	 */
1070 	if (oldroothdr.count > 1)
1071 		return 0;
1072 
1073 	/*
1074 	 * Read in the (only) child block, then copy those bytes into
1075 	 * the root block's buffer and free the original child block.
1076 	 */
1077 	btree = dp->d_ops->node_tree_p(oldroot);
1078 	child = be32_to_cpu(btree[0].before);
1079 	ASSERT(child != 0);
1080 	error = xfs_da3_node_read(args->trans, dp, child, -1, &bp,
1081 					     args->whichfork);
1082 	if (error)
1083 		return error;
1084 	xfs_da_blkinfo_onlychild_validate(bp->b_addr, oldroothdr.level);
1085 
1086 	/*
1087 	 * This could be copying a leaf back into the root block in the case of
1088 	 * there only being a single leaf block left in the tree. Hence we have
1089 	 * to update the b_ops pointer as well to match the buffer type change
1090 	 * that could occur. For dir3 blocks we also need to update the block
1091 	 * number in the buffer header.
1092 	 */
1093 	memcpy(root_blk->bp->b_addr, bp->b_addr, args->geo->blksize);
1094 	root_blk->bp->b_ops = bp->b_ops;
1095 	xfs_trans_buf_copy_type(root_blk->bp, bp);
1096 	if (oldroothdr.magic == XFS_DA3_NODE_MAGIC) {
1097 		struct xfs_da3_blkinfo *da3 = root_blk->bp->b_addr;
1098 		da3->blkno = cpu_to_be64(root_blk->bp->b_bn);
1099 	}
1100 	xfs_trans_log_buf(args->trans, root_blk->bp, 0,
1101 			  args->geo->blksize - 1);
1102 	error = xfs_da_shrink_inode(args, child, bp);
1103 	return error;
1104 }
1105 
1106 /*
1107  * Check a node block and its neighbors to see if the block should be
1108  * collapsed into one or the other neighbor.  Always keep the block
1109  * with the smaller block number.
1110  * If the current block is over 50% full, don't try to join it, return 0.
1111  * If the block is empty, fill in the state structure and return 2.
1112  * If it can be collapsed, fill in the state structure and return 1.
1113  * If nothing can be done, return 0.
1114  */
1115 STATIC int
1116 xfs_da3_node_toosmall(
1117 	struct xfs_da_state	*state,
1118 	int			*action)
1119 {
1120 	struct xfs_da_intnode	*node;
1121 	struct xfs_da_state_blk	*blk;
1122 	struct xfs_da_blkinfo	*info;
1123 	xfs_dablk_t		blkno;
1124 	struct xfs_buf		*bp;
1125 	struct xfs_da3_icnode_hdr nodehdr;
1126 	int			count;
1127 	int			forward;
1128 	int			error;
1129 	int			retval;
1130 	int			i;
1131 	struct xfs_inode	*dp = state->args->dp;
1132 
1133 	trace_xfs_da_node_toosmall(state->args);
1134 
1135 	/*
1136 	 * Check for the degenerate case of the block being over 50% full.
1137 	 * If so, it's not worth even looking to see if we might be able
1138 	 * to coalesce with a sibling.
1139 	 */
1140 	blk = &state->path.blk[ state->path.active-1 ];
1141 	info = blk->bp->b_addr;
1142 	node = (xfs_da_intnode_t *)info;
1143 	dp->d_ops->node_hdr_from_disk(&nodehdr, node);
1144 	if (nodehdr.count > (state->args->geo->node_ents >> 1)) {
1145 		*action = 0;	/* blk over 50%, don't try to join */
1146 		return 0;	/* blk over 50%, don't try to join */
1147 	}
1148 
1149 	/*
1150 	 * Check for the degenerate case of the block being empty.
1151 	 * If the block is empty, we'll simply delete it, no need to
1152 	 * coalesce it with a sibling block.  We choose (arbitrarily)
1153 	 * to merge with the forward block unless it is NULL.
1154 	 */
1155 	if (nodehdr.count == 0) {
1156 		/*
1157 		 * Make altpath point to the block we want to keep and
1158 		 * path point to the block we want to drop (this one).
1159 		 */
1160 		forward = (info->forw != 0);
1161 		memcpy(&state->altpath, &state->path, sizeof(state->path));
1162 		error = xfs_da3_path_shift(state, &state->altpath, forward,
1163 						 0, &retval);
1164 		if (error)
1165 			return error;
1166 		if (retval) {
1167 			*action = 0;
1168 		} else {
1169 			*action = 2;
1170 		}
1171 		return 0;
1172 	}
1173 
1174 	/*
1175 	 * Examine each sibling block to see if we can coalesce with
1176 	 * at least 25% free space to spare.  We need to figure out
1177 	 * whether to merge with the forward or the backward block.
1178 	 * We prefer coalescing with the lower numbered sibling so as
1179 	 * to shrink a directory over time.
1180 	 */
1181 	count  = state->args->geo->node_ents;
1182 	count -= state->args->geo->node_ents >> 2;
1183 	count -= nodehdr.count;
1184 
1185 	/* start with smaller blk num */
1186 	forward = nodehdr.forw < nodehdr.back;
1187 	for (i = 0; i < 2; forward = !forward, i++) {
1188 		struct xfs_da3_icnode_hdr thdr;
1189 		if (forward)
1190 			blkno = nodehdr.forw;
1191 		else
1192 			blkno = nodehdr.back;
1193 		if (blkno == 0)
1194 			continue;
1195 		error = xfs_da3_node_read(state->args->trans, dp,
1196 					blkno, -1, &bp, state->args->whichfork);
1197 		if (error)
1198 			return error;
1199 
1200 		node = bp->b_addr;
1201 		dp->d_ops->node_hdr_from_disk(&thdr, node);
1202 		xfs_trans_brelse(state->args->trans, bp);
1203 
1204 		if (count - thdr.count >= 0)
1205 			break;	/* fits with at least 25% to spare */
1206 	}
1207 	if (i >= 2) {
1208 		*action = 0;
1209 		return 0;
1210 	}
1211 
1212 	/*
1213 	 * Make altpath point to the block we want to keep (the lower
1214 	 * numbered block) and path point to the block we want to drop.
1215 	 */
1216 	memcpy(&state->altpath, &state->path, sizeof(state->path));
1217 	if (blkno < blk->blkno) {
1218 		error = xfs_da3_path_shift(state, &state->altpath, forward,
1219 						 0, &retval);
1220 	} else {
1221 		error = xfs_da3_path_shift(state, &state->path, forward,
1222 						 0, &retval);
1223 	}
1224 	if (error)
1225 		return error;
1226 	if (retval) {
1227 		*action = 0;
1228 		return 0;
1229 	}
1230 	*action = 1;
1231 	return 0;
1232 }
1233 
1234 /*
1235  * Pick up the last hashvalue from an intermediate node.
1236  */
1237 STATIC uint
1238 xfs_da3_node_lasthash(
1239 	struct xfs_inode	*dp,
1240 	struct xfs_buf		*bp,
1241 	int			*count)
1242 {
1243 	struct xfs_da_intnode	 *node;
1244 	struct xfs_da_node_entry *btree;
1245 	struct xfs_da3_icnode_hdr nodehdr;
1246 
1247 	node = bp->b_addr;
1248 	dp->d_ops->node_hdr_from_disk(&nodehdr, node);
1249 	if (count)
1250 		*count = nodehdr.count;
1251 	if (!nodehdr.count)
1252 		return 0;
1253 	btree = dp->d_ops->node_tree_p(node);
1254 	return be32_to_cpu(btree[nodehdr.count - 1].hashval);
1255 }
1256 
1257 /*
1258  * Walk back up the tree adjusting hash values as necessary,
1259  * when we stop making changes, return.
1260  */
1261 void
1262 xfs_da3_fixhashpath(
1263 	struct xfs_da_state	*state,
1264 	struct xfs_da_state_path *path)
1265 {
1266 	struct xfs_da_state_blk	*blk;
1267 	struct xfs_da_intnode	*node;
1268 	struct xfs_da_node_entry *btree;
1269 	xfs_dahash_t		lasthash=0;
1270 	int			level;
1271 	int			count;
1272 	struct xfs_inode	*dp = state->args->dp;
1273 
1274 	trace_xfs_da_fixhashpath(state->args);
1275 
1276 	level = path->active-1;
1277 	blk = &path->blk[ level ];
1278 	switch (blk->magic) {
1279 	case XFS_ATTR_LEAF_MAGIC:
1280 		lasthash = xfs_attr_leaf_lasthash(blk->bp, &count);
1281 		if (count == 0)
1282 			return;
1283 		break;
1284 	case XFS_DIR2_LEAFN_MAGIC:
1285 		lasthash = xfs_dir2_leafn_lasthash(dp, blk->bp, &count);
1286 		if (count == 0)
1287 			return;
1288 		break;
1289 	case XFS_DA_NODE_MAGIC:
1290 		lasthash = xfs_da3_node_lasthash(dp, blk->bp, &count);
1291 		if (count == 0)
1292 			return;
1293 		break;
1294 	}
1295 	for (blk--, level--; level >= 0; blk--, level--) {
1296 		struct xfs_da3_icnode_hdr nodehdr;
1297 
1298 		node = blk->bp->b_addr;
1299 		dp->d_ops->node_hdr_from_disk(&nodehdr, node);
1300 		btree = dp->d_ops->node_tree_p(node);
1301 		if (be32_to_cpu(btree[blk->index].hashval) == lasthash)
1302 			break;
1303 		blk->hashval = lasthash;
1304 		btree[blk->index].hashval = cpu_to_be32(lasthash);
1305 		xfs_trans_log_buf(state->args->trans, blk->bp,
1306 				  XFS_DA_LOGRANGE(node, &btree[blk->index],
1307 						  sizeof(*btree)));
1308 
1309 		lasthash = be32_to_cpu(btree[nodehdr.count - 1].hashval);
1310 	}
1311 }
1312 
1313 /*
1314  * Remove an entry from an intermediate node.
1315  */
1316 STATIC void
1317 xfs_da3_node_remove(
1318 	struct xfs_da_state	*state,
1319 	struct xfs_da_state_blk	*drop_blk)
1320 {
1321 	struct xfs_da_intnode	*node;
1322 	struct xfs_da3_icnode_hdr nodehdr;
1323 	struct xfs_da_node_entry *btree;
1324 	int			index;
1325 	int			tmp;
1326 	struct xfs_inode	*dp = state->args->dp;
1327 
1328 	trace_xfs_da_node_remove(state->args);
1329 
1330 	node = drop_blk->bp->b_addr;
1331 	dp->d_ops->node_hdr_from_disk(&nodehdr, node);
1332 	ASSERT(drop_blk->index < nodehdr.count);
1333 	ASSERT(drop_blk->index >= 0);
1334 
1335 	/*
1336 	 * Copy over the offending entry, or just zero it out.
1337 	 */
1338 	index = drop_blk->index;
1339 	btree = dp->d_ops->node_tree_p(node);
1340 	if (index < nodehdr.count - 1) {
1341 		tmp  = nodehdr.count - index - 1;
1342 		tmp *= (uint)sizeof(xfs_da_node_entry_t);
1343 		memmove(&btree[index], &btree[index + 1], tmp);
1344 		xfs_trans_log_buf(state->args->trans, drop_blk->bp,
1345 		    XFS_DA_LOGRANGE(node, &btree[index], tmp));
1346 		index = nodehdr.count - 1;
1347 	}
1348 	memset(&btree[index], 0, sizeof(xfs_da_node_entry_t));
1349 	xfs_trans_log_buf(state->args->trans, drop_blk->bp,
1350 	    XFS_DA_LOGRANGE(node, &btree[index], sizeof(btree[index])));
1351 	nodehdr.count -= 1;
1352 	dp->d_ops->node_hdr_to_disk(node, &nodehdr);
1353 	xfs_trans_log_buf(state->args->trans, drop_blk->bp,
1354 	    XFS_DA_LOGRANGE(node, &node->hdr, dp->d_ops->node_hdr_size));
1355 
1356 	/*
1357 	 * Copy the last hash value from the block to propagate upwards.
1358 	 */
1359 	drop_blk->hashval = be32_to_cpu(btree[index - 1].hashval);
1360 }
1361 
1362 /*
1363  * Unbalance the elements between two intermediate nodes,
1364  * move all Btree elements from one node into another.
1365  */
1366 STATIC void
1367 xfs_da3_node_unbalance(
1368 	struct xfs_da_state	*state,
1369 	struct xfs_da_state_blk	*drop_blk,
1370 	struct xfs_da_state_blk	*save_blk)
1371 {
1372 	struct xfs_da_intnode	*drop_node;
1373 	struct xfs_da_intnode	*save_node;
1374 	struct xfs_da_node_entry *drop_btree;
1375 	struct xfs_da_node_entry *save_btree;
1376 	struct xfs_da3_icnode_hdr drop_hdr;
1377 	struct xfs_da3_icnode_hdr save_hdr;
1378 	struct xfs_trans	*tp;
1379 	int			sindex;
1380 	int			tmp;
1381 	struct xfs_inode	*dp = state->args->dp;
1382 
1383 	trace_xfs_da_node_unbalance(state->args);
1384 
1385 	drop_node = drop_blk->bp->b_addr;
1386 	save_node = save_blk->bp->b_addr;
1387 	dp->d_ops->node_hdr_from_disk(&drop_hdr, drop_node);
1388 	dp->d_ops->node_hdr_from_disk(&save_hdr, save_node);
1389 	drop_btree = dp->d_ops->node_tree_p(drop_node);
1390 	save_btree = dp->d_ops->node_tree_p(save_node);
1391 	tp = state->args->trans;
1392 
1393 	/*
1394 	 * If the dying block has lower hashvals, then move all the
1395 	 * elements in the remaining block up to make a hole.
1396 	 */
1397 	if ((be32_to_cpu(drop_btree[0].hashval) <
1398 			be32_to_cpu(save_btree[0].hashval)) ||
1399 	    (be32_to_cpu(drop_btree[drop_hdr.count - 1].hashval) <
1400 			be32_to_cpu(save_btree[save_hdr.count - 1].hashval))) {
1401 		/* XXX: check this - is memmove dst correct? */
1402 		tmp = save_hdr.count * sizeof(xfs_da_node_entry_t);
1403 		memmove(&save_btree[drop_hdr.count], &save_btree[0], tmp);
1404 
1405 		sindex = 0;
1406 		xfs_trans_log_buf(tp, save_blk->bp,
1407 			XFS_DA_LOGRANGE(save_node, &save_btree[0],
1408 				(save_hdr.count + drop_hdr.count) *
1409 						sizeof(xfs_da_node_entry_t)));
1410 	} else {
1411 		sindex = save_hdr.count;
1412 		xfs_trans_log_buf(tp, save_blk->bp,
1413 			XFS_DA_LOGRANGE(save_node, &save_btree[sindex],
1414 				drop_hdr.count * sizeof(xfs_da_node_entry_t)));
1415 	}
1416 
1417 	/*
1418 	 * Move all the B-tree elements from drop_blk to save_blk.
1419 	 */
1420 	tmp = drop_hdr.count * (uint)sizeof(xfs_da_node_entry_t);
1421 	memcpy(&save_btree[sindex], &drop_btree[0], tmp);
1422 	save_hdr.count += drop_hdr.count;
1423 
1424 	dp->d_ops->node_hdr_to_disk(save_node, &save_hdr);
1425 	xfs_trans_log_buf(tp, save_blk->bp,
1426 		XFS_DA_LOGRANGE(save_node, &save_node->hdr,
1427 				dp->d_ops->node_hdr_size));
1428 
1429 	/*
1430 	 * Save the last hashval in the remaining block for upward propagation.
1431 	 */
1432 	save_blk->hashval = be32_to_cpu(save_btree[save_hdr.count - 1].hashval);
1433 }
1434 
1435 /*========================================================================
1436  * Routines used for finding things in the Btree.
1437  *========================================================================*/
1438 
1439 /*
1440  * Walk down the Btree looking for a particular filename, filling
1441  * in the state structure as we go.
1442  *
1443  * We will set the state structure to point to each of the elements
1444  * in each of the nodes where either the hashval is or should be.
1445  *
1446  * We support duplicate hashval's so for each entry in the current
1447  * node that could contain the desired hashval, descend.  This is a
1448  * pruned depth-first tree search.
1449  */
1450 int							/* error */
1451 xfs_da3_node_lookup_int(
1452 	struct xfs_da_state	*state,
1453 	int			*result)
1454 {
1455 	struct xfs_da_state_blk	*blk;
1456 	struct xfs_da_blkinfo	*curr;
1457 	struct xfs_da_intnode	*node;
1458 	struct xfs_da_node_entry *btree;
1459 	struct xfs_da3_icnode_hdr nodehdr;
1460 	struct xfs_da_args	*args;
1461 	xfs_dablk_t		blkno;
1462 	xfs_dahash_t		hashval;
1463 	xfs_dahash_t		btreehashval;
1464 	int			probe;
1465 	int			span;
1466 	int			max;
1467 	int			error;
1468 	int			retval;
1469 	struct xfs_inode	*dp = state->args->dp;
1470 
1471 	args = state->args;
1472 
1473 	/*
1474 	 * Descend thru the B-tree searching each level for the right
1475 	 * node to use, until the right hashval is found.
1476 	 */
1477 	blkno = (args->whichfork == XFS_DATA_FORK)? args->geo->leafblk : 0;
1478 	for (blk = &state->path.blk[0], state->path.active = 1;
1479 			 state->path.active <= XFS_DA_NODE_MAXDEPTH;
1480 			 blk++, state->path.active++) {
1481 		/*
1482 		 * Read the next node down in the tree.
1483 		 */
1484 		blk->blkno = blkno;
1485 		error = xfs_da3_node_read(args->trans, args->dp, blkno,
1486 					-1, &blk->bp, args->whichfork);
1487 		if (error) {
1488 			blk->blkno = 0;
1489 			state->path.active--;
1490 			return error;
1491 		}
1492 		curr = blk->bp->b_addr;
1493 		blk->magic = be16_to_cpu(curr->magic);
1494 
1495 		if (blk->magic == XFS_ATTR_LEAF_MAGIC ||
1496 		    blk->magic == XFS_ATTR3_LEAF_MAGIC) {
1497 			blk->magic = XFS_ATTR_LEAF_MAGIC;
1498 			blk->hashval = xfs_attr_leaf_lasthash(blk->bp, NULL);
1499 			break;
1500 		}
1501 
1502 		if (blk->magic == XFS_DIR2_LEAFN_MAGIC ||
1503 		    blk->magic == XFS_DIR3_LEAFN_MAGIC) {
1504 			blk->magic = XFS_DIR2_LEAFN_MAGIC;
1505 			blk->hashval = xfs_dir2_leafn_lasthash(args->dp,
1506 							       blk->bp, NULL);
1507 			break;
1508 		}
1509 
1510 		blk->magic = XFS_DA_NODE_MAGIC;
1511 
1512 
1513 		/*
1514 		 * Search an intermediate node for a match.
1515 		 */
1516 		node = blk->bp->b_addr;
1517 		dp->d_ops->node_hdr_from_disk(&nodehdr, node);
1518 		btree = dp->d_ops->node_tree_p(node);
1519 
1520 		max = nodehdr.count;
1521 		blk->hashval = be32_to_cpu(btree[max - 1].hashval);
1522 
1523 		/*
1524 		 * Binary search.  (note: small blocks will skip loop)
1525 		 */
1526 		probe = span = max / 2;
1527 		hashval = args->hashval;
1528 		while (span > 4) {
1529 			span /= 2;
1530 			btreehashval = be32_to_cpu(btree[probe].hashval);
1531 			if (btreehashval < hashval)
1532 				probe += span;
1533 			else if (btreehashval > hashval)
1534 				probe -= span;
1535 			else
1536 				break;
1537 		}
1538 		ASSERT((probe >= 0) && (probe < max));
1539 		ASSERT((span <= 4) ||
1540 			(be32_to_cpu(btree[probe].hashval) == hashval));
1541 
1542 		/*
1543 		 * Since we may have duplicate hashval's, find the first
1544 		 * matching hashval in the node.
1545 		 */
1546 		while (probe > 0 &&
1547 		       be32_to_cpu(btree[probe].hashval) >= hashval) {
1548 			probe--;
1549 		}
1550 		while (probe < max &&
1551 		       be32_to_cpu(btree[probe].hashval) < hashval) {
1552 			probe++;
1553 		}
1554 
1555 		/*
1556 		 * Pick the right block to descend on.
1557 		 */
1558 		if (probe == max) {
1559 			blk->index = max - 1;
1560 			blkno = be32_to_cpu(btree[max - 1].before);
1561 		} else {
1562 			blk->index = probe;
1563 			blkno = be32_to_cpu(btree[probe].before);
1564 		}
1565 	}
1566 
1567 	/*
1568 	 * A leaf block that ends in the hashval that we are interested in
1569 	 * (final hashval == search hashval) means that the next block may
1570 	 * contain more entries with the same hashval, shift upward to the
1571 	 * next leaf and keep searching.
1572 	 */
1573 	for (;;) {
1574 		if (blk->magic == XFS_DIR2_LEAFN_MAGIC) {
1575 			retval = xfs_dir2_leafn_lookup_int(blk->bp, args,
1576 							&blk->index, state);
1577 		} else if (blk->magic == XFS_ATTR_LEAF_MAGIC) {
1578 			retval = xfs_attr3_leaf_lookup_int(blk->bp, args);
1579 			blk->index = args->index;
1580 			args->blkno = blk->blkno;
1581 		} else {
1582 			ASSERT(0);
1583 			return -EFSCORRUPTED;
1584 		}
1585 		if (((retval == -ENOENT) || (retval == -ENOATTR)) &&
1586 		    (blk->hashval == args->hashval)) {
1587 			error = xfs_da3_path_shift(state, &state->path, 1, 1,
1588 							 &retval);
1589 			if (error)
1590 				return error;
1591 			if (retval == 0) {
1592 				continue;
1593 			} else if (blk->magic == XFS_ATTR_LEAF_MAGIC) {
1594 				/* path_shift() gives ENOENT */
1595 				retval = -ENOATTR;
1596 			}
1597 		}
1598 		break;
1599 	}
1600 	*result = retval;
1601 	return 0;
1602 }
1603 
1604 /*========================================================================
1605  * Utility routines.
1606  *========================================================================*/
1607 
1608 /*
1609  * Compare two intermediate nodes for "order".
1610  */
1611 STATIC int
1612 xfs_da3_node_order(
1613 	struct xfs_inode *dp,
1614 	struct xfs_buf	*node1_bp,
1615 	struct xfs_buf	*node2_bp)
1616 {
1617 	struct xfs_da_intnode	*node1;
1618 	struct xfs_da_intnode	*node2;
1619 	struct xfs_da_node_entry *btree1;
1620 	struct xfs_da_node_entry *btree2;
1621 	struct xfs_da3_icnode_hdr node1hdr;
1622 	struct xfs_da3_icnode_hdr node2hdr;
1623 
1624 	node1 = node1_bp->b_addr;
1625 	node2 = node2_bp->b_addr;
1626 	dp->d_ops->node_hdr_from_disk(&node1hdr, node1);
1627 	dp->d_ops->node_hdr_from_disk(&node2hdr, node2);
1628 	btree1 = dp->d_ops->node_tree_p(node1);
1629 	btree2 = dp->d_ops->node_tree_p(node2);
1630 
1631 	if (node1hdr.count > 0 && node2hdr.count > 0 &&
1632 	    ((be32_to_cpu(btree2[0].hashval) < be32_to_cpu(btree1[0].hashval)) ||
1633 	     (be32_to_cpu(btree2[node2hdr.count - 1].hashval) <
1634 	      be32_to_cpu(btree1[node1hdr.count - 1].hashval)))) {
1635 		return 1;
1636 	}
1637 	return 0;
1638 }
1639 
1640 /*
1641  * Link a new block into a doubly linked list of blocks (of whatever type).
1642  */
1643 int							/* error */
1644 xfs_da3_blk_link(
1645 	struct xfs_da_state	*state,
1646 	struct xfs_da_state_blk	*old_blk,
1647 	struct xfs_da_state_blk	*new_blk)
1648 {
1649 	struct xfs_da_blkinfo	*old_info;
1650 	struct xfs_da_blkinfo	*new_info;
1651 	struct xfs_da_blkinfo	*tmp_info;
1652 	struct xfs_da_args	*args;
1653 	struct xfs_buf		*bp;
1654 	int			before = 0;
1655 	int			error;
1656 	struct xfs_inode	*dp = state->args->dp;
1657 
1658 	/*
1659 	 * Set up environment.
1660 	 */
1661 	args = state->args;
1662 	ASSERT(args != NULL);
1663 	old_info = old_blk->bp->b_addr;
1664 	new_info = new_blk->bp->b_addr;
1665 	ASSERT(old_blk->magic == XFS_DA_NODE_MAGIC ||
1666 	       old_blk->magic == XFS_DIR2_LEAFN_MAGIC ||
1667 	       old_blk->magic == XFS_ATTR_LEAF_MAGIC);
1668 
1669 	switch (old_blk->magic) {
1670 	case XFS_ATTR_LEAF_MAGIC:
1671 		before = xfs_attr_leaf_order(old_blk->bp, new_blk->bp);
1672 		break;
1673 	case XFS_DIR2_LEAFN_MAGIC:
1674 		before = xfs_dir2_leafn_order(dp, old_blk->bp, new_blk->bp);
1675 		break;
1676 	case XFS_DA_NODE_MAGIC:
1677 		before = xfs_da3_node_order(dp, old_blk->bp, new_blk->bp);
1678 		break;
1679 	}
1680 
1681 	/*
1682 	 * Link blocks in appropriate order.
1683 	 */
1684 	if (before) {
1685 		/*
1686 		 * Link new block in before existing block.
1687 		 */
1688 		trace_xfs_da_link_before(args);
1689 		new_info->forw = cpu_to_be32(old_blk->blkno);
1690 		new_info->back = old_info->back;
1691 		if (old_info->back) {
1692 			error = xfs_da3_node_read(args->trans, dp,
1693 						be32_to_cpu(old_info->back),
1694 						-1, &bp, args->whichfork);
1695 			if (error)
1696 				return error;
1697 			ASSERT(bp != NULL);
1698 			tmp_info = bp->b_addr;
1699 			ASSERT(tmp_info->magic == old_info->magic);
1700 			ASSERT(be32_to_cpu(tmp_info->forw) == old_blk->blkno);
1701 			tmp_info->forw = cpu_to_be32(new_blk->blkno);
1702 			xfs_trans_log_buf(args->trans, bp, 0, sizeof(*tmp_info)-1);
1703 		}
1704 		old_info->back = cpu_to_be32(new_blk->blkno);
1705 	} else {
1706 		/*
1707 		 * Link new block in after existing block.
1708 		 */
1709 		trace_xfs_da_link_after(args);
1710 		new_info->forw = old_info->forw;
1711 		new_info->back = cpu_to_be32(old_blk->blkno);
1712 		if (old_info->forw) {
1713 			error = xfs_da3_node_read(args->trans, dp,
1714 						be32_to_cpu(old_info->forw),
1715 						-1, &bp, args->whichfork);
1716 			if (error)
1717 				return error;
1718 			ASSERT(bp != NULL);
1719 			tmp_info = bp->b_addr;
1720 			ASSERT(tmp_info->magic == old_info->magic);
1721 			ASSERT(be32_to_cpu(tmp_info->back) == old_blk->blkno);
1722 			tmp_info->back = cpu_to_be32(new_blk->blkno);
1723 			xfs_trans_log_buf(args->trans, bp, 0, sizeof(*tmp_info)-1);
1724 		}
1725 		old_info->forw = cpu_to_be32(new_blk->blkno);
1726 	}
1727 
1728 	xfs_trans_log_buf(args->trans, old_blk->bp, 0, sizeof(*tmp_info) - 1);
1729 	xfs_trans_log_buf(args->trans, new_blk->bp, 0, sizeof(*tmp_info) - 1);
1730 	return 0;
1731 }
1732 
1733 /*
1734  * Unlink a block from a doubly linked list of blocks.
1735  */
1736 STATIC int						/* error */
1737 xfs_da3_blk_unlink(
1738 	struct xfs_da_state	*state,
1739 	struct xfs_da_state_blk	*drop_blk,
1740 	struct xfs_da_state_blk	*save_blk)
1741 {
1742 	struct xfs_da_blkinfo	*drop_info;
1743 	struct xfs_da_blkinfo	*save_info;
1744 	struct xfs_da_blkinfo	*tmp_info;
1745 	struct xfs_da_args	*args;
1746 	struct xfs_buf		*bp;
1747 	int			error;
1748 
1749 	/*
1750 	 * Set up environment.
1751 	 */
1752 	args = state->args;
1753 	ASSERT(args != NULL);
1754 	save_info = save_blk->bp->b_addr;
1755 	drop_info = drop_blk->bp->b_addr;
1756 	ASSERT(save_blk->magic == XFS_DA_NODE_MAGIC ||
1757 	       save_blk->magic == XFS_DIR2_LEAFN_MAGIC ||
1758 	       save_blk->magic == XFS_ATTR_LEAF_MAGIC);
1759 	ASSERT(save_blk->magic == drop_blk->magic);
1760 	ASSERT((be32_to_cpu(save_info->forw) == drop_blk->blkno) ||
1761 	       (be32_to_cpu(save_info->back) == drop_blk->blkno));
1762 	ASSERT((be32_to_cpu(drop_info->forw) == save_blk->blkno) ||
1763 	       (be32_to_cpu(drop_info->back) == save_blk->blkno));
1764 
1765 	/*
1766 	 * Unlink the leaf block from the doubly linked chain of leaves.
1767 	 */
1768 	if (be32_to_cpu(save_info->back) == drop_blk->blkno) {
1769 		trace_xfs_da_unlink_back(args);
1770 		save_info->back = drop_info->back;
1771 		if (drop_info->back) {
1772 			error = xfs_da3_node_read(args->trans, args->dp,
1773 						be32_to_cpu(drop_info->back),
1774 						-1, &bp, args->whichfork);
1775 			if (error)
1776 				return error;
1777 			ASSERT(bp != NULL);
1778 			tmp_info = bp->b_addr;
1779 			ASSERT(tmp_info->magic == save_info->magic);
1780 			ASSERT(be32_to_cpu(tmp_info->forw) == drop_blk->blkno);
1781 			tmp_info->forw = cpu_to_be32(save_blk->blkno);
1782 			xfs_trans_log_buf(args->trans, bp, 0,
1783 						    sizeof(*tmp_info) - 1);
1784 		}
1785 	} else {
1786 		trace_xfs_da_unlink_forward(args);
1787 		save_info->forw = drop_info->forw;
1788 		if (drop_info->forw) {
1789 			error = xfs_da3_node_read(args->trans, args->dp,
1790 						be32_to_cpu(drop_info->forw),
1791 						-1, &bp, args->whichfork);
1792 			if (error)
1793 				return error;
1794 			ASSERT(bp != NULL);
1795 			tmp_info = bp->b_addr;
1796 			ASSERT(tmp_info->magic == save_info->magic);
1797 			ASSERT(be32_to_cpu(tmp_info->back) == drop_blk->blkno);
1798 			tmp_info->back = cpu_to_be32(save_blk->blkno);
1799 			xfs_trans_log_buf(args->trans, bp, 0,
1800 						    sizeof(*tmp_info) - 1);
1801 		}
1802 	}
1803 
1804 	xfs_trans_log_buf(args->trans, save_blk->bp, 0, sizeof(*save_info) - 1);
1805 	return 0;
1806 }
1807 
1808 /*
1809  * Move a path "forward" or "!forward" one block at the current level.
1810  *
1811  * This routine will adjust a "path" to point to the next block
1812  * "forward" (higher hashvalues) or "!forward" (lower hashvals) in the
1813  * Btree, including updating pointers to the intermediate nodes between
1814  * the new bottom and the root.
1815  */
1816 int							/* error */
1817 xfs_da3_path_shift(
1818 	struct xfs_da_state	*state,
1819 	struct xfs_da_state_path *path,
1820 	int			forward,
1821 	int			release,
1822 	int			*result)
1823 {
1824 	struct xfs_da_state_blk	*blk;
1825 	struct xfs_da_blkinfo	*info;
1826 	struct xfs_da_intnode	*node;
1827 	struct xfs_da_args	*args;
1828 	struct xfs_da_node_entry *btree;
1829 	struct xfs_da3_icnode_hdr nodehdr;
1830 	struct xfs_buf		*bp;
1831 	xfs_dablk_t		blkno = 0;
1832 	int			level;
1833 	int			error;
1834 	struct xfs_inode	*dp = state->args->dp;
1835 
1836 	trace_xfs_da_path_shift(state->args);
1837 
1838 	/*
1839 	 * Roll up the Btree looking for the first block where our
1840 	 * current index is not at the edge of the block.  Note that
1841 	 * we skip the bottom layer because we want the sibling block.
1842 	 */
1843 	args = state->args;
1844 	ASSERT(args != NULL);
1845 	ASSERT(path != NULL);
1846 	ASSERT((path->active > 0) && (path->active < XFS_DA_NODE_MAXDEPTH));
1847 	level = (path->active-1) - 1;	/* skip bottom layer in path */
1848 	for (blk = &path->blk[level]; level >= 0; blk--, level--) {
1849 		node = blk->bp->b_addr;
1850 		dp->d_ops->node_hdr_from_disk(&nodehdr, node);
1851 		btree = dp->d_ops->node_tree_p(node);
1852 
1853 		if (forward && (blk->index < nodehdr.count - 1)) {
1854 			blk->index++;
1855 			blkno = be32_to_cpu(btree[blk->index].before);
1856 			break;
1857 		} else if (!forward && (blk->index > 0)) {
1858 			blk->index--;
1859 			blkno = be32_to_cpu(btree[blk->index].before);
1860 			break;
1861 		}
1862 	}
1863 	if (level < 0) {
1864 		*result = -ENOENT;	/* we're out of our tree */
1865 		ASSERT(args->op_flags & XFS_DA_OP_OKNOENT);
1866 		return 0;
1867 	}
1868 
1869 	/*
1870 	 * Roll down the edge of the subtree until we reach the
1871 	 * same depth we were at originally.
1872 	 */
1873 	for (blk++, level++; level < path->active; blk++, level++) {
1874 		/*
1875 		 * Read the next child block into a local buffer.
1876 		 */
1877 		error = xfs_da3_node_read(args->trans, dp, blkno, -1, &bp,
1878 					  args->whichfork);
1879 		if (error)
1880 			return error;
1881 
1882 		/*
1883 		 * Release the old block (if it's dirty, the trans doesn't
1884 		 * actually let go) and swap the local buffer into the path
1885 		 * structure. This ensures failure of the above read doesn't set
1886 		 * a NULL buffer in an active slot in the path.
1887 		 */
1888 		if (release)
1889 			xfs_trans_brelse(args->trans, blk->bp);
1890 		blk->blkno = blkno;
1891 		blk->bp = bp;
1892 
1893 		info = blk->bp->b_addr;
1894 		ASSERT(info->magic == cpu_to_be16(XFS_DA_NODE_MAGIC) ||
1895 		       info->magic == cpu_to_be16(XFS_DA3_NODE_MAGIC) ||
1896 		       info->magic == cpu_to_be16(XFS_DIR2_LEAFN_MAGIC) ||
1897 		       info->magic == cpu_to_be16(XFS_DIR3_LEAFN_MAGIC) ||
1898 		       info->magic == cpu_to_be16(XFS_ATTR_LEAF_MAGIC) ||
1899 		       info->magic == cpu_to_be16(XFS_ATTR3_LEAF_MAGIC));
1900 
1901 
1902 		/*
1903 		 * Note: we flatten the magic number to a single type so we
1904 		 * don't have to compare against crc/non-crc types elsewhere.
1905 		 */
1906 		switch (be16_to_cpu(info->magic)) {
1907 		case XFS_DA_NODE_MAGIC:
1908 		case XFS_DA3_NODE_MAGIC:
1909 			blk->magic = XFS_DA_NODE_MAGIC;
1910 			node = (xfs_da_intnode_t *)info;
1911 			dp->d_ops->node_hdr_from_disk(&nodehdr, node);
1912 			btree = dp->d_ops->node_tree_p(node);
1913 			blk->hashval = be32_to_cpu(btree[nodehdr.count - 1].hashval);
1914 			if (forward)
1915 				blk->index = 0;
1916 			else
1917 				blk->index = nodehdr.count - 1;
1918 			blkno = be32_to_cpu(btree[blk->index].before);
1919 			break;
1920 		case XFS_ATTR_LEAF_MAGIC:
1921 		case XFS_ATTR3_LEAF_MAGIC:
1922 			blk->magic = XFS_ATTR_LEAF_MAGIC;
1923 			ASSERT(level == path->active-1);
1924 			blk->index = 0;
1925 			blk->hashval = xfs_attr_leaf_lasthash(blk->bp, NULL);
1926 			break;
1927 		case XFS_DIR2_LEAFN_MAGIC:
1928 		case XFS_DIR3_LEAFN_MAGIC:
1929 			blk->magic = XFS_DIR2_LEAFN_MAGIC;
1930 			ASSERT(level == path->active-1);
1931 			blk->index = 0;
1932 			blk->hashval = xfs_dir2_leafn_lasthash(args->dp,
1933 							       blk->bp, NULL);
1934 			break;
1935 		default:
1936 			ASSERT(0);
1937 			break;
1938 		}
1939 	}
1940 	*result = 0;
1941 	return 0;
1942 }
1943 
1944 
1945 /*========================================================================
1946  * Utility routines.
1947  *========================================================================*/
1948 
1949 /*
1950  * Implement a simple hash on a character string.
1951  * Rotate the hash value by 7 bits, then XOR each character in.
1952  * This is implemented with some source-level loop unrolling.
1953  */
1954 xfs_dahash_t
1955 xfs_da_hashname(const __uint8_t *name, int namelen)
1956 {
1957 	xfs_dahash_t hash;
1958 
1959 	/*
1960 	 * Do four characters at a time as long as we can.
1961 	 */
1962 	for (hash = 0; namelen >= 4; namelen -= 4, name += 4)
1963 		hash = (name[0] << 21) ^ (name[1] << 14) ^ (name[2] << 7) ^
1964 		       (name[3] << 0) ^ rol32(hash, 7 * 4);
1965 
1966 	/*
1967 	 * Now do the rest of the characters.
1968 	 */
1969 	switch (namelen) {
1970 	case 3:
1971 		return (name[0] << 14) ^ (name[1] << 7) ^ (name[2] << 0) ^
1972 		       rol32(hash, 7 * 3);
1973 	case 2:
1974 		return (name[0] << 7) ^ (name[1] << 0) ^ rol32(hash, 7 * 2);
1975 	case 1:
1976 		return (name[0] << 0) ^ rol32(hash, 7 * 1);
1977 	default: /* case 0: */
1978 		return hash;
1979 	}
1980 }
1981 
1982 enum xfs_dacmp
1983 xfs_da_compname(
1984 	struct xfs_da_args *args,
1985 	const unsigned char *name,
1986 	int		len)
1987 {
1988 	return (args->namelen == len && memcmp(args->name, name, len) == 0) ?
1989 					XFS_CMP_EXACT : XFS_CMP_DIFFERENT;
1990 }
1991 
1992 static xfs_dahash_t
1993 xfs_default_hashname(
1994 	struct xfs_name	*name)
1995 {
1996 	return xfs_da_hashname(name->name, name->len);
1997 }
1998 
1999 const struct xfs_nameops xfs_default_nameops = {
2000 	.hashname	= xfs_default_hashname,
2001 	.compname	= xfs_da_compname
2002 };
2003 
2004 int
2005 xfs_da_grow_inode_int(
2006 	struct xfs_da_args	*args,
2007 	xfs_fileoff_t		*bno,
2008 	int			count)
2009 {
2010 	struct xfs_trans	*tp = args->trans;
2011 	struct xfs_inode	*dp = args->dp;
2012 	int			w = args->whichfork;
2013 	xfs_rfsblock_t		nblks = dp->i_d.di_nblocks;
2014 	struct xfs_bmbt_irec	map, *mapp;
2015 	int			nmap, error, got, i, mapi;
2016 
2017 	/*
2018 	 * Find a spot in the file space to put the new block.
2019 	 */
2020 	error = xfs_bmap_first_unused(tp, dp, count, bno, w);
2021 	if (error)
2022 		return error;
2023 
2024 	/*
2025 	 * Try mapping it in one filesystem block.
2026 	 */
2027 	nmap = 1;
2028 	ASSERT(args->firstblock != NULL);
2029 	error = xfs_bmapi_write(tp, dp, *bno, count,
2030 			xfs_bmapi_aflag(w)|XFS_BMAPI_METADATA|XFS_BMAPI_CONTIG,
2031 			args->firstblock, args->total, &map, &nmap,
2032 			args->flist);
2033 	if (error)
2034 		return error;
2035 
2036 	ASSERT(nmap <= 1);
2037 	if (nmap == 1) {
2038 		mapp = &map;
2039 		mapi = 1;
2040 	} else if (nmap == 0 && count > 1) {
2041 		xfs_fileoff_t		b;
2042 		int			c;
2043 
2044 		/*
2045 		 * If we didn't get it and the block might work if fragmented,
2046 		 * try without the CONTIG flag.  Loop until we get it all.
2047 		 */
2048 		mapp = kmem_alloc(sizeof(*mapp) * count, KM_SLEEP);
2049 		for (b = *bno, mapi = 0; b < *bno + count; ) {
2050 			nmap = MIN(XFS_BMAP_MAX_NMAP, count);
2051 			c = (int)(*bno + count - b);
2052 			error = xfs_bmapi_write(tp, dp, b, c,
2053 					xfs_bmapi_aflag(w)|XFS_BMAPI_METADATA,
2054 					args->firstblock, args->total,
2055 					&mapp[mapi], &nmap, args->flist);
2056 			if (error)
2057 				goto out_free_map;
2058 			if (nmap < 1)
2059 				break;
2060 			mapi += nmap;
2061 			b = mapp[mapi - 1].br_startoff +
2062 			    mapp[mapi - 1].br_blockcount;
2063 		}
2064 	} else {
2065 		mapi = 0;
2066 		mapp = NULL;
2067 	}
2068 
2069 	/*
2070 	 * Count the blocks we got, make sure it matches the total.
2071 	 */
2072 	for (i = 0, got = 0; i < mapi; i++)
2073 		got += mapp[i].br_blockcount;
2074 	if (got != count || mapp[0].br_startoff != *bno ||
2075 	    mapp[mapi - 1].br_startoff + mapp[mapi - 1].br_blockcount !=
2076 	    *bno + count) {
2077 		error = -ENOSPC;
2078 		goto out_free_map;
2079 	}
2080 
2081 	/* account for newly allocated blocks in reserved blocks total */
2082 	args->total -= dp->i_d.di_nblocks - nblks;
2083 
2084 out_free_map:
2085 	if (mapp != &map)
2086 		kmem_free(mapp);
2087 	return error;
2088 }
2089 
2090 /*
2091  * Add a block to the btree ahead of the file.
2092  * Return the new block number to the caller.
2093  */
2094 int
2095 xfs_da_grow_inode(
2096 	struct xfs_da_args	*args,
2097 	xfs_dablk_t		*new_blkno)
2098 {
2099 	xfs_fileoff_t		bno;
2100 	int			error;
2101 
2102 	trace_xfs_da_grow_inode(args);
2103 
2104 	bno = args->geo->leafblk;
2105 	error = xfs_da_grow_inode_int(args, &bno, args->geo->fsbcount);
2106 	if (!error)
2107 		*new_blkno = (xfs_dablk_t)bno;
2108 	return error;
2109 }
2110 
2111 /*
2112  * Ick.  We need to always be able to remove a btree block, even
2113  * if there's no space reservation because the filesystem is full.
2114  * This is called if xfs_bunmapi on a btree block fails due to ENOSPC.
2115  * It swaps the target block with the last block in the file.  The
2116  * last block in the file can always be removed since it can't cause
2117  * a bmap btree split to do that.
2118  */
2119 STATIC int
2120 xfs_da3_swap_lastblock(
2121 	struct xfs_da_args	*args,
2122 	xfs_dablk_t		*dead_blknop,
2123 	struct xfs_buf		**dead_bufp)
2124 {
2125 	struct xfs_da_blkinfo	*dead_info;
2126 	struct xfs_da_blkinfo	*sib_info;
2127 	struct xfs_da_intnode	*par_node;
2128 	struct xfs_da_intnode	*dead_node;
2129 	struct xfs_dir2_leaf	*dead_leaf2;
2130 	struct xfs_da_node_entry *btree;
2131 	struct xfs_da3_icnode_hdr par_hdr;
2132 	struct xfs_inode	*dp;
2133 	struct xfs_trans	*tp;
2134 	struct xfs_mount	*mp;
2135 	struct xfs_buf		*dead_buf;
2136 	struct xfs_buf		*last_buf;
2137 	struct xfs_buf		*sib_buf;
2138 	struct xfs_buf		*par_buf;
2139 	xfs_dahash_t		dead_hash;
2140 	xfs_fileoff_t		lastoff;
2141 	xfs_dablk_t		dead_blkno;
2142 	xfs_dablk_t		last_blkno;
2143 	xfs_dablk_t		sib_blkno;
2144 	xfs_dablk_t		par_blkno;
2145 	int			error;
2146 	int			w;
2147 	int			entno;
2148 	int			level;
2149 	int			dead_level;
2150 
2151 	trace_xfs_da_swap_lastblock(args);
2152 
2153 	dead_buf = *dead_bufp;
2154 	dead_blkno = *dead_blknop;
2155 	tp = args->trans;
2156 	dp = args->dp;
2157 	w = args->whichfork;
2158 	ASSERT(w == XFS_DATA_FORK);
2159 	mp = dp->i_mount;
2160 	lastoff = args->geo->freeblk;
2161 	error = xfs_bmap_last_before(tp, dp, &lastoff, w);
2162 	if (error)
2163 		return error;
2164 	if (unlikely(lastoff == 0)) {
2165 		XFS_ERROR_REPORT("xfs_da_swap_lastblock(1)", XFS_ERRLEVEL_LOW,
2166 				 mp);
2167 		return -EFSCORRUPTED;
2168 	}
2169 	/*
2170 	 * Read the last block in the btree space.
2171 	 */
2172 	last_blkno = (xfs_dablk_t)lastoff - args->geo->fsbcount;
2173 	error = xfs_da3_node_read(tp, dp, last_blkno, -1, &last_buf, w);
2174 	if (error)
2175 		return error;
2176 	/*
2177 	 * Copy the last block into the dead buffer and log it.
2178 	 */
2179 	memcpy(dead_buf->b_addr, last_buf->b_addr, args->geo->blksize);
2180 	xfs_trans_log_buf(tp, dead_buf, 0, args->geo->blksize - 1);
2181 	dead_info = dead_buf->b_addr;
2182 	/*
2183 	 * Get values from the moved block.
2184 	 */
2185 	if (dead_info->magic == cpu_to_be16(XFS_DIR2_LEAFN_MAGIC) ||
2186 	    dead_info->magic == cpu_to_be16(XFS_DIR3_LEAFN_MAGIC)) {
2187 		struct xfs_dir3_icleaf_hdr leafhdr;
2188 		struct xfs_dir2_leaf_entry *ents;
2189 
2190 		dead_leaf2 = (xfs_dir2_leaf_t *)dead_info;
2191 		dp->d_ops->leaf_hdr_from_disk(&leafhdr, dead_leaf2);
2192 		ents = dp->d_ops->leaf_ents_p(dead_leaf2);
2193 		dead_level = 0;
2194 		dead_hash = be32_to_cpu(ents[leafhdr.count - 1].hashval);
2195 	} else {
2196 		struct xfs_da3_icnode_hdr deadhdr;
2197 
2198 		dead_node = (xfs_da_intnode_t *)dead_info;
2199 		dp->d_ops->node_hdr_from_disk(&deadhdr, dead_node);
2200 		btree = dp->d_ops->node_tree_p(dead_node);
2201 		dead_level = deadhdr.level;
2202 		dead_hash = be32_to_cpu(btree[deadhdr.count - 1].hashval);
2203 	}
2204 	sib_buf = par_buf = NULL;
2205 	/*
2206 	 * If the moved block has a left sibling, fix up the pointers.
2207 	 */
2208 	if ((sib_blkno = be32_to_cpu(dead_info->back))) {
2209 		error = xfs_da3_node_read(tp, dp, sib_blkno, -1, &sib_buf, w);
2210 		if (error)
2211 			goto done;
2212 		sib_info = sib_buf->b_addr;
2213 		if (unlikely(
2214 		    be32_to_cpu(sib_info->forw) != last_blkno ||
2215 		    sib_info->magic != dead_info->magic)) {
2216 			XFS_ERROR_REPORT("xfs_da_swap_lastblock(2)",
2217 					 XFS_ERRLEVEL_LOW, mp);
2218 			error = -EFSCORRUPTED;
2219 			goto done;
2220 		}
2221 		sib_info->forw = cpu_to_be32(dead_blkno);
2222 		xfs_trans_log_buf(tp, sib_buf,
2223 			XFS_DA_LOGRANGE(sib_info, &sib_info->forw,
2224 					sizeof(sib_info->forw)));
2225 		sib_buf = NULL;
2226 	}
2227 	/*
2228 	 * If the moved block has a right sibling, fix up the pointers.
2229 	 */
2230 	if ((sib_blkno = be32_to_cpu(dead_info->forw))) {
2231 		error = xfs_da3_node_read(tp, dp, sib_blkno, -1, &sib_buf, w);
2232 		if (error)
2233 			goto done;
2234 		sib_info = sib_buf->b_addr;
2235 		if (unlikely(
2236 		       be32_to_cpu(sib_info->back) != last_blkno ||
2237 		       sib_info->magic != dead_info->magic)) {
2238 			XFS_ERROR_REPORT("xfs_da_swap_lastblock(3)",
2239 					 XFS_ERRLEVEL_LOW, mp);
2240 			error = -EFSCORRUPTED;
2241 			goto done;
2242 		}
2243 		sib_info->back = cpu_to_be32(dead_blkno);
2244 		xfs_trans_log_buf(tp, sib_buf,
2245 			XFS_DA_LOGRANGE(sib_info, &sib_info->back,
2246 					sizeof(sib_info->back)));
2247 		sib_buf = NULL;
2248 	}
2249 	par_blkno = args->geo->leafblk;
2250 	level = -1;
2251 	/*
2252 	 * Walk down the tree looking for the parent of the moved block.
2253 	 */
2254 	for (;;) {
2255 		error = xfs_da3_node_read(tp, dp, par_blkno, -1, &par_buf, w);
2256 		if (error)
2257 			goto done;
2258 		par_node = par_buf->b_addr;
2259 		dp->d_ops->node_hdr_from_disk(&par_hdr, par_node);
2260 		if (level >= 0 && level != par_hdr.level + 1) {
2261 			XFS_ERROR_REPORT("xfs_da_swap_lastblock(4)",
2262 					 XFS_ERRLEVEL_LOW, mp);
2263 			error = -EFSCORRUPTED;
2264 			goto done;
2265 		}
2266 		level = par_hdr.level;
2267 		btree = dp->d_ops->node_tree_p(par_node);
2268 		for (entno = 0;
2269 		     entno < par_hdr.count &&
2270 		     be32_to_cpu(btree[entno].hashval) < dead_hash;
2271 		     entno++)
2272 			continue;
2273 		if (entno == par_hdr.count) {
2274 			XFS_ERROR_REPORT("xfs_da_swap_lastblock(5)",
2275 					 XFS_ERRLEVEL_LOW, mp);
2276 			error = -EFSCORRUPTED;
2277 			goto done;
2278 		}
2279 		par_blkno = be32_to_cpu(btree[entno].before);
2280 		if (level == dead_level + 1)
2281 			break;
2282 		xfs_trans_brelse(tp, par_buf);
2283 		par_buf = NULL;
2284 	}
2285 	/*
2286 	 * We're in the right parent block.
2287 	 * Look for the right entry.
2288 	 */
2289 	for (;;) {
2290 		for (;
2291 		     entno < par_hdr.count &&
2292 		     be32_to_cpu(btree[entno].before) != last_blkno;
2293 		     entno++)
2294 			continue;
2295 		if (entno < par_hdr.count)
2296 			break;
2297 		par_blkno = par_hdr.forw;
2298 		xfs_trans_brelse(tp, par_buf);
2299 		par_buf = NULL;
2300 		if (unlikely(par_blkno == 0)) {
2301 			XFS_ERROR_REPORT("xfs_da_swap_lastblock(6)",
2302 					 XFS_ERRLEVEL_LOW, mp);
2303 			error = -EFSCORRUPTED;
2304 			goto done;
2305 		}
2306 		error = xfs_da3_node_read(tp, dp, par_blkno, -1, &par_buf, w);
2307 		if (error)
2308 			goto done;
2309 		par_node = par_buf->b_addr;
2310 		dp->d_ops->node_hdr_from_disk(&par_hdr, par_node);
2311 		if (par_hdr.level != level) {
2312 			XFS_ERROR_REPORT("xfs_da_swap_lastblock(7)",
2313 					 XFS_ERRLEVEL_LOW, mp);
2314 			error = -EFSCORRUPTED;
2315 			goto done;
2316 		}
2317 		btree = dp->d_ops->node_tree_p(par_node);
2318 		entno = 0;
2319 	}
2320 	/*
2321 	 * Update the parent entry pointing to the moved block.
2322 	 */
2323 	btree[entno].before = cpu_to_be32(dead_blkno);
2324 	xfs_trans_log_buf(tp, par_buf,
2325 		XFS_DA_LOGRANGE(par_node, &btree[entno].before,
2326 				sizeof(btree[entno].before)));
2327 	*dead_blknop = last_blkno;
2328 	*dead_bufp = last_buf;
2329 	return 0;
2330 done:
2331 	if (par_buf)
2332 		xfs_trans_brelse(tp, par_buf);
2333 	if (sib_buf)
2334 		xfs_trans_brelse(tp, sib_buf);
2335 	xfs_trans_brelse(tp, last_buf);
2336 	return error;
2337 }
2338 
2339 /*
2340  * Remove a btree block from a directory or attribute.
2341  */
2342 int
2343 xfs_da_shrink_inode(
2344 	xfs_da_args_t	*args,
2345 	xfs_dablk_t	dead_blkno,
2346 	struct xfs_buf	*dead_buf)
2347 {
2348 	xfs_inode_t *dp;
2349 	int done, error, w, count;
2350 	xfs_trans_t *tp;
2351 
2352 	trace_xfs_da_shrink_inode(args);
2353 
2354 	dp = args->dp;
2355 	w = args->whichfork;
2356 	tp = args->trans;
2357 	count = args->geo->fsbcount;
2358 	for (;;) {
2359 		/*
2360 		 * Remove extents.  If we get ENOSPC for a dir we have to move
2361 		 * the last block to the place we want to kill.
2362 		 */
2363 		error = xfs_bunmapi(tp, dp, dead_blkno, count,
2364 				    xfs_bmapi_aflag(w), 0, args->firstblock,
2365 				    args->flist, &done);
2366 		if (error == -ENOSPC) {
2367 			if (w != XFS_DATA_FORK)
2368 				break;
2369 			error = xfs_da3_swap_lastblock(args, &dead_blkno,
2370 						      &dead_buf);
2371 			if (error)
2372 				break;
2373 		} else {
2374 			break;
2375 		}
2376 	}
2377 	xfs_trans_binval(tp, dead_buf);
2378 	return error;
2379 }
2380 
2381 /*
2382  * See if the mapping(s) for this btree block are valid, i.e.
2383  * don't contain holes, are logically contiguous, and cover the whole range.
2384  */
2385 STATIC int
2386 xfs_da_map_covers_blocks(
2387 	int		nmap,
2388 	xfs_bmbt_irec_t	*mapp,
2389 	xfs_dablk_t	bno,
2390 	int		count)
2391 {
2392 	int		i;
2393 	xfs_fileoff_t	off;
2394 
2395 	for (i = 0, off = bno; i < nmap; i++) {
2396 		if (mapp[i].br_startblock == HOLESTARTBLOCK ||
2397 		    mapp[i].br_startblock == DELAYSTARTBLOCK) {
2398 			return 0;
2399 		}
2400 		if (off != mapp[i].br_startoff) {
2401 			return 0;
2402 		}
2403 		off += mapp[i].br_blockcount;
2404 	}
2405 	return off == bno + count;
2406 }
2407 
2408 /*
2409  * Convert a struct xfs_bmbt_irec to a struct xfs_buf_map.
2410  *
2411  * For the single map case, it is assumed that the caller has provided a pointer
2412  * to a valid xfs_buf_map.  For the multiple map case, this function will
2413  * allocate the xfs_buf_map to hold all the maps and replace the caller's single
2414  * map pointer with the allocated map.
2415  */
2416 static int
2417 xfs_buf_map_from_irec(
2418 	struct xfs_mount	*mp,
2419 	struct xfs_buf_map	**mapp,
2420 	int			*nmaps,
2421 	struct xfs_bmbt_irec	*irecs,
2422 	int			nirecs)
2423 {
2424 	struct xfs_buf_map	*map;
2425 	int			i;
2426 
2427 	ASSERT(*nmaps == 1);
2428 	ASSERT(nirecs >= 1);
2429 
2430 	if (nirecs > 1) {
2431 		map = kmem_zalloc(nirecs * sizeof(struct xfs_buf_map),
2432 				  KM_SLEEP | KM_NOFS);
2433 		if (!map)
2434 			return -ENOMEM;
2435 		*mapp = map;
2436 	}
2437 
2438 	*nmaps = nirecs;
2439 	map = *mapp;
2440 	for (i = 0; i < *nmaps; i++) {
2441 		ASSERT(irecs[i].br_startblock != DELAYSTARTBLOCK &&
2442 		       irecs[i].br_startblock != HOLESTARTBLOCK);
2443 		map[i].bm_bn = XFS_FSB_TO_DADDR(mp, irecs[i].br_startblock);
2444 		map[i].bm_len = XFS_FSB_TO_BB(mp, irecs[i].br_blockcount);
2445 	}
2446 	return 0;
2447 }
2448 
2449 /*
2450  * Map the block we are given ready for reading. There are three possible return
2451  * values:
2452  *	-1 - will be returned if we land in a hole and mappedbno == -2 so the
2453  *	     caller knows not to execute a subsequent read.
2454  *	 0 - if we mapped the block successfully
2455  *	>0 - positive error number if there was an error.
2456  */
2457 static int
2458 xfs_dabuf_map(
2459 	struct xfs_inode	*dp,
2460 	xfs_dablk_t		bno,
2461 	xfs_daddr_t		mappedbno,
2462 	int			whichfork,
2463 	struct xfs_buf_map	**map,
2464 	int			*nmaps)
2465 {
2466 	struct xfs_mount	*mp = dp->i_mount;
2467 	int			nfsb;
2468 	int			error = 0;
2469 	struct xfs_bmbt_irec	irec;
2470 	struct xfs_bmbt_irec	*irecs = &irec;
2471 	int			nirecs;
2472 
2473 	ASSERT(map && *map);
2474 	ASSERT(*nmaps == 1);
2475 
2476 	if (whichfork == XFS_DATA_FORK)
2477 		nfsb = mp->m_dir_geo->fsbcount;
2478 	else
2479 		nfsb = mp->m_attr_geo->fsbcount;
2480 
2481 	/*
2482 	 * Caller doesn't have a mapping.  -2 means don't complain
2483 	 * if we land in a hole.
2484 	 */
2485 	if (mappedbno == -1 || mappedbno == -2) {
2486 		/*
2487 		 * Optimize the one-block case.
2488 		 */
2489 		if (nfsb != 1)
2490 			irecs = kmem_zalloc(sizeof(irec) * nfsb,
2491 					    KM_SLEEP | KM_NOFS);
2492 
2493 		nirecs = nfsb;
2494 		error = xfs_bmapi_read(dp, (xfs_fileoff_t)bno, nfsb, irecs,
2495 				       &nirecs, xfs_bmapi_aflag(whichfork));
2496 		if (error)
2497 			goto out;
2498 	} else {
2499 		irecs->br_startblock = XFS_DADDR_TO_FSB(mp, mappedbno);
2500 		irecs->br_startoff = (xfs_fileoff_t)bno;
2501 		irecs->br_blockcount = nfsb;
2502 		irecs->br_state = 0;
2503 		nirecs = 1;
2504 	}
2505 
2506 	if (!xfs_da_map_covers_blocks(nirecs, irecs, bno, nfsb)) {
2507 		error = mappedbno == -2 ? -1 : -EFSCORRUPTED;
2508 		if (unlikely(error == -EFSCORRUPTED)) {
2509 			if (xfs_error_level >= XFS_ERRLEVEL_LOW) {
2510 				int i;
2511 				xfs_alert(mp, "%s: bno %lld dir: inode %lld",
2512 					__func__, (long long)bno,
2513 					(long long)dp->i_ino);
2514 				for (i = 0; i < *nmaps; i++) {
2515 					xfs_alert(mp,
2516 "[%02d] br_startoff %lld br_startblock %lld br_blockcount %lld br_state %d",
2517 						i,
2518 						(long long)irecs[i].br_startoff,
2519 						(long long)irecs[i].br_startblock,
2520 						(long long)irecs[i].br_blockcount,
2521 						irecs[i].br_state);
2522 				}
2523 			}
2524 			XFS_ERROR_REPORT("xfs_da_do_buf(1)",
2525 					 XFS_ERRLEVEL_LOW, mp);
2526 		}
2527 		goto out;
2528 	}
2529 	error = xfs_buf_map_from_irec(mp, map, nmaps, irecs, nirecs);
2530 out:
2531 	if (irecs != &irec)
2532 		kmem_free(irecs);
2533 	return error;
2534 }
2535 
2536 /*
2537  * Get a buffer for the dir/attr block.
2538  */
2539 int
2540 xfs_da_get_buf(
2541 	struct xfs_trans	*trans,
2542 	struct xfs_inode	*dp,
2543 	xfs_dablk_t		bno,
2544 	xfs_daddr_t		mappedbno,
2545 	struct xfs_buf		**bpp,
2546 	int			whichfork)
2547 {
2548 	struct xfs_buf		*bp;
2549 	struct xfs_buf_map	map;
2550 	struct xfs_buf_map	*mapp;
2551 	int			nmap;
2552 	int			error;
2553 
2554 	*bpp = NULL;
2555 	mapp = &map;
2556 	nmap = 1;
2557 	error = xfs_dabuf_map(dp, bno, mappedbno, whichfork,
2558 				&mapp, &nmap);
2559 	if (error) {
2560 		/* mapping a hole is not an error, but we don't continue */
2561 		if (error == -1)
2562 			error = 0;
2563 		goto out_free;
2564 	}
2565 
2566 	bp = xfs_trans_get_buf_map(trans, dp->i_mount->m_ddev_targp,
2567 				    mapp, nmap, 0);
2568 	error = bp ? bp->b_error : -EIO;
2569 	if (error) {
2570 		if (bp)
2571 			xfs_trans_brelse(trans, bp);
2572 		goto out_free;
2573 	}
2574 
2575 	*bpp = bp;
2576 
2577 out_free:
2578 	if (mapp != &map)
2579 		kmem_free(mapp);
2580 
2581 	return error;
2582 }
2583 
2584 /*
2585  * Get a buffer for the dir/attr block, fill in the contents.
2586  */
2587 int
2588 xfs_da_read_buf(
2589 	struct xfs_trans	*trans,
2590 	struct xfs_inode	*dp,
2591 	xfs_dablk_t		bno,
2592 	xfs_daddr_t		mappedbno,
2593 	struct xfs_buf		**bpp,
2594 	int			whichfork,
2595 	const struct xfs_buf_ops *ops)
2596 {
2597 	struct xfs_buf		*bp;
2598 	struct xfs_buf_map	map;
2599 	struct xfs_buf_map	*mapp;
2600 	int			nmap;
2601 	int			error;
2602 
2603 	*bpp = NULL;
2604 	mapp = &map;
2605 	nmap = 1;
2606 	error = xfs_dabuf_map(dp, bno, mappedbno, whichfork,
2607 				&mapp, &nmap);
2608 	if (error) {
2609 		/* mapping a hole is not an error, but we don't continue */
2610 		if (error == -1)
2611 			error = 0;
2612 		goto out_free;
2613 	}
2614 
2615 	error = xfs_trans_read_buf_map(dp->i_mount, trans,
2616 					dp->i_mount->m_ddev_targp,
2617 					mapp, nmap, 0, &bp, ops);
2618 	if (error)
2619 		goto out_free;
2620 
2621 	if (whichfork == XFS_ATTR_FORK)
2622 		xfs_buf_set_ref(bp, XFS_ATTR_BTREE_REF);
2623 	else
2624 		xfs_buf_set_ref(bp, XFS_DIR_BTREE_REF);
2625 	*bpp = bp;
2626 out_free:
2627 	if (mapp != &map)
2628 		kmem_free(mapp);
2629 
2630 	return error;
2631 }
2632 
2633 /*
2634  * Readahead the dir/attr block.
2635  */
2636 xfs_daddr_t
2637 xfs_da_reada_buf(
2638 	struct xfs_inode	*dp,
2639 	xfs_dablk_t		bno,
2640 	xfs_daddr_t		mappedbno,
2641 	int			whichfork,
2642 	const struct xfs_buf_ops *ops)
2643 {
2644 	struct xfs_buf_map	map;
2645 	struct xfs_buf_map	*mapp;
2646 	int			nmap;
2647 	int			error;
2648 
2649 	mapp = &map;
2650 	nmap = 1;
2651 	error = xfs_dabuf_map(dp, bno, mappedbno, whichfork,
2652 				&mapp, &nmap);
2653 	if (error) {
2654 		/* mapping a hole is not an error, but we don't continue */
2655 		if (error == -1)
2656 			error = 0;
2657 		goto out_free;
2658 	}
2659 
2660 	mappedbno = mapp[0].bm_bn;
2661 	xfs_buf_readahead_map(dp->i_mount->m_ddev_targp, mapp, nmap, ops);
2662 
2663 out_free:
2664 	if (mapp != &map)
2665 		kmem_free(mapp);
2666 
2667 	if (error)
2668 		return -1;
2669 	return mappedbno;
2670 }
2671