Home | History | Annotate | Download | only in e2fsck
      1 /*
      2  * pass3.c -- pass #3 of e2fsck: Check for directory connectivity
      3  *
      4  * Copyright (C) 1993, 1994, 1995, 1996, 1997, 1998, 1999 Theodore Ts'o.
      5  *
      6  * %Begin-Header%
      7  * This file may be redistributed under the terms of the GNU Public
      8  * License.
      9  * %End-Header%
     10  *
     11  * Pass #3 assures that all directories are connected to the
     12  * filesystem tree, using the following algorithm:
     13  *
     14  * First, the root directory is checked to make sure it exists; if
     15  * not, e2fsck will offer to create a new one.  It is then marked as
     16  * "done".
     17  *
     18  * Then, pass3 interates over all directory inodes; for each directory
     19  * it attempts to trace up the filesystem tree, using dirinfo.parent
     20  * until it reaches a directory which has been marked "done".  If it
     21  * can not do so, then the directory must be disconnected, and e2fsck
     22  * will offer to reconnect it to /lost+found.  While it is chasing
     23  * parent pointers up the filesystem tree, if pass3 sees a directory
     24  * twice, then it has detected a filesystem loop, and it will again
     25  * offer to reconnect the directory to /lost+found in to break the
     26  * filesystem loop.
     27  *
     28  * Pass 3 also contains the subroutine, e2fsck_reconnect_file() to
     29  * reconnect inodes to /lost+found; this subroutine is also used by
     30  * pass 4.  e2fsck_reconnect_file() calls get_lost_and_found(), which
     31  * is responsible for creating /lost+found if it does not exist.
     32  *
     33  * Pass 3 frees the following data structures:
     34  *     	- The dirinfo directory information cache.
     35  */
     36 
     37 #ifdef HAVE_ERRNO_H
     38 #include <errno.h>
     39 #endif
     40 
     41 #include "e2fsck.h"
     42 #include "problem.h"
     43 
     44 static void check_root(e2fsck_t ctx);
     45 static int check_directory(e2fsck_t ctx, ext2_ino_t ino,
     46 			   struct problem_context *pctx);
     47 static void fix_dotdot(e2fsck_t ctx, ext2_ino_t ino, ext2_ino_t parent);
     48 
     49 static ext2fs_inode_bitmap inode_loop_detect = 0;
     50 static ext2fs_inode_bitmap inode_done_map = 0;
     51 
     52 void e2fsck_pass3(e2fsck_t ctx)
     53 {
     54 	ext2_filsys fs = ctx->fs;
     55 	struct dir_info_iter *iter = NULL;
     56 #ifdef RESOURCE_TRACK
     57 	struct resource_track	rtrack;
     58 #endif
     59 	struct problem_context	pctx;
     60 	struct dir_info	*dir;
     61 	unsigned long maxdirs, count;
     62 
     63 	init_resource_track(&rtrack, ctx->fs->io);
     64 	clear_problem_context(&pctx);
     65 
     66 #ifdef MTRACE
     67 	mtrace_print("Pass 3");
     68 #endif
     69 
     70 	if (!(ctx->options & E2F_OPT_PREEN))
     71 		fix_problem(ctx, PR_3_PASS_HEADER, &pctx);
     72 
     73 	/*
     74 	 * Allocate some bitmaps to do loop detection.
     75 	 */
     76 	pctx.errcode = e2fsck_allocate_inode_bitmap(fs, _("inode done bitmap"),
     77 					EXT2FS_BMAP64_AUTODIR,
     78 					"inode_done_map", &inode_done_map);
     79 	if (pctx.errcode) {
     80 		pctx.num = 2;
     81 		fix_problem(ctx, PR_3_ALLOCATE_IBITMAP_ERROR, &pctx);
     82 		ctx->flags |= E2F_FLAG_ABORT;
     83 		goto abort_exit;
     84 	}
     85 	print_resource_track(ctx, _("Peak memory"), &ctx->global_rtrack, NULL);
     86 
     87 	check_root(ctx);
     88 	if (ctx->flags & E2F_FLAG_SIGNAL_MASK)
     89 		goto abort_exit;
     90 
     91 	ext2fs_mark_inode_bitmap2(inode_done_map, EXT2_ROOT_INO);
     92 
     93 	maxdirs = e2fsck_get_num_dirinfo(ctx);
     94 	count = 1;
     95 
     96 	if (ctx->progress)
     97 		if ((ctx->progress)(ctx, 3, 0, maxdirs))
     98 			goto abort_exit;
     99 
    100 	iter = e2fsck_dir_info_iter_begin(ctx);
    101 	while ((dir = e2fsck_dir_info_iter(ctx, iter)) != 0) {
    102 		if (ctx->flags & E2F_FLAG_SIGNAL_MASK)
    103 			goto abort_exit;
    104 		if (ctx->progress && (ctx->progress)(ctx, 3, count++, maxdirs))
    105 			goto abort_exit;
    106 		if (ext2fs_test_inode_bitmap2(ctx->inode_dir_map, dir->ino))
    107 			if (check_directory(ctx, dir->ino, &pctx))
    108 				goto abort_exit;
    109 	}
    110 
    111 	/*
    112 	 * Force the creation of /lost+found if not present
    113 	 */
    114 	if ((ctx->options & E2F_OPT_READONLY) == 0)
    115 		e2fsck_get_lost_and_found(ctx, 1);
    116 
    117 	/*
    118 	 * If there are any directories that need to be indexed or
    119 	 * optimized, do it here.
    120 	 */
    121 	e2fsck_rehash_directories(ctx);
    122 
    123 abort_exit:
    124 	if (iter)
    125 		e2fsck_dir_info_iter_end(ctx, iter);
    126 	e2fsck_free_dir_info(ctx);
    127 	if (inode_loop_detect) {
    128 		ext2fs_free_inode_bitmap(inode_loop_detect);
    129 		inode_loop_detect = 0;
    130 	}
    131 	if (inode_done_map) {
    132 		ext2fs_free_inode_bitmap(inode_done_map);
    133 		inode_done_map = 0;
    134 	}
    135 
    136 	print_resource_track(ctx, _("Pass 3"), &rtrack, ctx->fs->io);
    137 }
    138 
    139 /*
    140  * This makes sure the root inode is present; if not, we ask if the
    141  * user wants us to create it.  Not creating it is a fatal error.
    142  */
    143 static void check_root(e2fsck_t ctx)
    144 {
    145 	ext2_filsys fs = ctx->fs;
    146 	blk64_t			blk;
    147 	struct ext2_inode	inode;
    148 	char *			block;
    149 	struct problem_context	pctx;
    150 
    151 	clear_problem_context(&pctx);
    152 
    153 	if (ext2fs_test_inode_bitmap2(ctx->inode_used_map, EXT2_ROOT_INO)) {
    154 		/*
    155 		 * If the root inode is not a directory, die here.  The
    156 		 * user must have answered 'no' in pass1 when we
    157 		 * offered to clear it.
    158 		 */
    159 		if (!(ext2fs_test_inode_bitmap2(ctx->inode_dir_map,
    160 					       EXT2_ROOT_INO))) {
    161 			fix_problem(ctx, PR_3_ROOT_NOT_DIR_ABORT, &pctx);
    162 			ctx->flags |= E2F_FLAG_ABORT;
    163 		}
    164 		return;
    165 	}
    166 
    167 	if (!fix_problem(ctx, PR_3_NO_ROOT_INODE, &pctx)) {
    168 		fix_problem(ctx, PR_3_NO_ROOT_INODE_ABORT, &pctx);
    169 		ctx->flags |= E2F_FLAG_ABORT;
    170 		return;
    171 	}
    172 
    173 	e2fsck_read_bitmaps(ctx);
    174 
    175 	/*
    176 	 * First, find a free block
    177 	 */
    178 	pctx.errcode = ext2fs_new_block2(fs, 0, ctx->block_found_map, &blk);
    179 	if (pctx.errcode) {
    180 		pctx.str = "ext2fs_new_block";
    181 		fix_problem(ctx, PR_3_CREATE_ROOT_ERROR, &pctx);
    182 		ctx->flags |= E2F_FLAG_ABORT;
    183 		return;
    184 	}
    185 	ext2fs_mark_block_bitmap2(ctx->block_found_map, blk);
    186 	ext2fs_mark_block_bitmap2(fs->block_map, blk);
    187 	ext2fs_mark_bb_dirty(fs);
    188 
    189 	/*
    190 	 * Now let's create the actual data block for the inode
    191 	 */
    192 	pctx.errcode = ext2fs_new_dir_block(fs, EXT2_ROOT_INO, EXT2_ROOT_INO,
    193 					    &block);
    194 	if (pctx.errcode) {
    195 		pctx.str = "ext2fs_new_dir_block";
    196 		fix_problem(ctx, PR_3_CREATE_ROOT_ERROR, &pctx);
    197 		ctx->flags |= E2F_FLAG_ABORT;
    198 		return;
    199 	}
    200 
    201 	pctx.errcode = ext2fs_write_dir_block3(fs, blk, block, 0);
    202 	if (pctx.errcode) {
    203 		pctx.str = "ext2fs_write_dir_block3";
    204 		fix_problem(ctx, PR_3_CREATE_ROOT_ERROR, &pctx);
    205 		ctx->flags |= E2F_FLAG_ABORT;
    206 		return;
    207 	}
    208 	ext2fs_free_mem(&block);
    209 
    210 	/*
    211 	 * Set up the inode structure
    212 	 */
    213 	memset(&inode, 0, sizeof(inode));
    214 	inode.i_mode = 040755;
    215 	inode.i_size = fs->blocksize;
    216 	inode.i_atime = inode.i_ctime = inode.i_mtime = ctx->now;
    217 	inode.i_links_count = 2;
    218 	ext2fs_iblk_set(fs, &inode, 1);
    219 	inode.i_block[0] = blk;
    220 
    221 	/*
    222 	 * Write out the inode.
    223 	 */
    224 	pctx.errcode = ext2fs_write_new_inode(fs, EXT2_ROOT_INO, &inode);
    225 	if (pctx.errcode) {
    226 		pctx.str = "ext2fs_write_inode";
    227 		fix_problem(ctx, PR_3_CREATE_ROOT_ERROR, &pctx);
    228 		ctx->flags |= E2F_FLAG_ABORT;
    229 		return;
    230 	}
    231 
    232 	/*
    233 	 * Miscellaneous bookkeeping...
    234 	 */
    235 	e2fsck_add_dir_info(ctx, EXT2_ROOT_INO, EXT2_ROOT_INO);
    236 	ext2fs_icount_store(ctx->inode_count, EXT2_ROOT_INO, 2);
    237 	ext2fs_icount_store(ctx->inode_link_info, EXT2_ROOT_INO, 2);
    238 
    239 	ext2fs_mark_inode_bitmap2(ctx->inode_used_map, EXT2_ROOT_INO);
    240 	ext2fs_mark_inode_bitmap2(ctx->inode_dir_map, EXT2_ROOT_INO);
    241 	ext2fs_mark_inode_bitmap2(fs->inode_map, EXT2_ROOT_INO);
    242 	ext2fs_mark_ib_dirty(fs);
    243 }
    244 
    245 /*
    246  * This subroutine is responsible for making sure that a particular
    247  * directory is connected to the root; if it isn't we trace it up as
    248  * far as we can go, and then offer to connect the resulting parent to
    249  * the lost+found.  We have to do loop detection; if we ever discover
    250  * a loop, we treat that as a disconnected directory and offer to
    251  * reparent it to lost+found.
    252  *
    253  * However, loop detection is expensive, because for very large
    254  * filesystems, the inode_loop_detect bitmap is huge, and clearing it
    255  * is non-trivial.  Loops in filesystems are also a rare error case,
    256  * and we shouldn't optimize for error cases.  So we try two passes of
    257  * the algorithm.  The first time, we ignore loop detection and merely
    258  * increment a counter; if the counter exceeds some extreme threshold,
    259  * then we try again with the loop detection bitmap enabled.
    260  */
    261 static int check_directory(e2fsck_t ctx, ext2_ino_t dir,
    262 			   struct problem_context *pctx)
    263 {
    264 	ext2_filsys 	fs = ctx->fs;
    265 	ext2_ino_t	ino = dir, parent;
    266 	int		loop_pass = 0, parent_count = 0;
    267 
    268 	while (1) {
    269 		/*
    270 		 * Mark this inode as being "done"; by the time we
    271 		 * return from this function, the inode we either be
    272 		 * verified as being connected to the directory tree,
    273 		 * or we will have offered to reconnect this to
    274 		 * lost+found.
    275 		 *
    276 		 * If it was marked done already, then we've reached a
    277 		 * parent we've already checked.
    278 		 */
    279 	  	if (ext2fs_mark_inode_bitmap2(inode_done_map, ino))
    280 			break;
    281 
    282 		if (e2fsck_dir_info_get_parent(ctx, ino, &parent)) {
    283 			fix_problem(ctx, PR_3_NO_DIRINFO, pctx);
    284 			return 0;
    285 		}
    286 
    287 		/*
    288 		 * If this directory doesn't have a parent, or we've
    289 		 * seen the parent once already, then offer to
    290 		 * reparent it to lost+found
    291 		 */
    292 		if (!parent ||
    293 		    (loop_pass &&
    294 		     (ext2fs_test_inode_bitmap2(inode_loop_detect,
    295 					       parent)))) {
    296 			pctx->ino = ino;
    297 			if (fix_problem(ctx, PR_3_UNCONNECTED_DIR, pctx)) {
    298 				if (e2fsck_reconnect_file(ctx, pctx->ino))
    299 					ext2fs_unmark_valid(fs);
    300 				else {
    301 					fix_dotdot(ctx, pctx->ino,
    302 						   ctx->lost_and_found);
    303 					parent = ctx->lost_and_found;
    304 				}
    305 			}
    306 			break;
    307 		}
    308 		ino = parent;
    309 		if (loop_pass) {
    310 			ext2fs_mark_inode_bitmap2(inode_loop_detect, ino);
    311 		} else if (parent_count++ > 2048) {
    312 			/*
    313 			 * If we've run into a path depth that's
    314 			 * greater than 2048, try again with the inode
    315 			 * loop bitmap turned on and start from the
    316 			 * top.
    317 			 */
    318 			loop_pass = 1;
    319 			if (inode_loop_detect)
    320 				ext2fs_clear_inode_bitmap(inode_loop_detect);
    321 			else {
    322 				pctx->errcode = e2fsck_allocate_inode_bitmap(fs, _("inode loop detection bitmap"), EXT2FS_BMAP64_AUTODIR, "inode_loop_detect", &inode_loop_detect);
    323 				if (pctx->errcode) {
    324 					pctx->num = 1;
    325 					fix_problem(ctx,
    326 				    PR_3_ALLOCATE_IBITMAP_ERROR, pctx);
    327 					ctx->flags |= E2F_FLAG_ABORT;
    328 					return -1;
    329 				}
    330 			}
    331 			ino = dir;
    332 		}
    333 	}
    334 
    335 	/*
    336 	 * Make sure that .. and the parent directory are the same;
    337 	 * offer to fix it if not.
    338 	 */
    339 	pctx->ino = dir;
    340 	if (e2fsck_dir_info_get_dotdot(ctx, dir, &pctx->ino2) ||
    341 	    e2fsck_dir_info_get_parent(ctx, dir, &pctx->dir)) {
    342 		fix_problem(ctx, PR_3_NO_DIRINFO, pctx);
    343 		return 0;
    344 	}
    345 	if (pctx->ino2 != pctx->dir) {
    346 		if (fix_problem(ctx, PR_3_BAD_DOT_DOT, pctx))
    347 			fix_dotdot(ctx, dir, pctx->dir);
    348 	}
    349 	return 0;
    350 }
    351 
    352 /*
    353  * This routine gets the lost_and_found inode, making it a directory
    354  * if necessary
    355  */
    356 ext2_ino_t e2fsck_get_lost_and_found(e2fsck_t ctx, int fix)
    357 {
    358 	ext2_filsys fs = ctx->fs;
    359 	ext2_ino_t			ino;
    360 	blk64_t			blk;
    361 	errcode_t		retval;
    362 	struct ext2_inode	inode;
    363 	char *			block;
    364 	static const char	name[] = "lost+found";
    365 	struct 	problem_context	pctx;
    366 
    367 	if (ctx->lost_and_found)
    368 		return ctx->lost_and_found;
    369 
    370 	clear_problem_context(&pctx);
    371 
    372 	retval = ext2fs_lookup(fs, EXT2_ROOT_INO, name,
    373 			       sizeof(name)-1, 0, &ino);
    374 	if (retval && !fix)
    375 		return 0;
    376 	if (!retval) {
    377 		if (ext2fs_test_inode_bitmap2(ctx->inode_dir_map, ino)) {
    378 			ctx->lost_and_found = ino;
    379 			return ino;
    380 		}
    381 
    382 		/* Lost+found isn't a directory! */
    383 		if (!fix)
    384 			return 0;
    385 		pctx.ino = ino;
    386 		if (!fix_problem(ctx, PR_3_LPF_NOTDIR, &pctx))
    387 			return 0;
    388 
    389 		/* OK, unlink the old /lost+found file. */
    390 		pctx.errcode = ext2fs_unlink(fs, EXT2_ROOT_INO, name, ino, 0);
    391 		if (pctx.errcode) {
    392 			pctx.str = "ext2fs_unlink";
    393 			fix_problem(ctx, PR_3_CREATE_LPF_ERROR, &pctx);
    394 			return 0;
    395 		}
    396 		(void) e2fsck_dir_info_set_parent(ctx, ino, 0);
    397 		e2fsck_adjust_inode_count(ctx, ino, -1);
    398 	} else if (retval != EXT2_ET_FILE_NOT_FOUND) {
    399 		pctx.errcode = retval;
    400 		fix_problem(ctx, PR_3_ERR_FIND_LPF, &pctx);
    401 	}
    402 	if (!fix_problem(ctx, PR_3_NO_LF_DIR, 0))
    403 		return 0;
    404 
    405 	/*
    406 	 * Read the inode and block bitmaps in; we'll be messing with
    407 	 * them.
    408 	 */
    409 	e2fsck_read_bitmaps(ctx);
    410 
    411 	/*
    412 	 * First, find a free block
    413 	 */
    414 	retval = ext2fs_new_block2(fs, 0, ctx->block_found_map, &blk);
    415 	if (retval) {
    416 		pctx.errcode = retval;
    417 		fix_problem(ctx, PR_3_ERR_LPF_NEW_BLOCK, &pctx);
    418 		return 0;
    419 	}
    420 	ext2fs_mark_block_bitmap2(ctx->block_found_map, blk);
    421 	ext2fs_block_alloc_stats2(fs, blk, +1);
    422 
    423 	/*
    424 	 * Next find a free inode.
    425 	 */
    426 	retval = ext2fs_new_inode(fs, EXT2_ROOT_INO, 040700,
    427 				  ctx->inode_used_map, &ino);
    428 	if (retval) {
    429 		pctx.errcode = retval;
    430 		fix_problem(ctx, PR_3_ERR_LPF_NEW_INODE, &pctx);
    431 		return 0;
    432 	}
    433 	ext2fs_mark_inode_bitmap2(ctx->inode_used_map, ino);
    434 	ext2fs_mark_inode_bitmap2(ctx->inode_dir_map, ino);
    435 	ext2fs_inode_alloc_stats2(fs, ino, +1, 1);
    436 
    437 	/*
    438 	 * Now let's create the actual data block for the inode
    439 	 */
    440 	retval = ext2fs_new_dir_block(fs, ino, EXT2_ROOT_INO, &block);
    441 	if (retval) {
    442 		pctx.errcode = retval;
    443 		fix_problem(ctx, PR_3_ERR_LPF_NEW_DIR_BLOCK, &pctx);
    444 		return 0;
    445 	}
    446 
    447 	retval = ext2fs_write_dir_block3(fs, blk, block, 0);
    448 	ext2fs_free_mem(&block);
    449 	if (retval) {
    450 		pctx.errcode = retval;
    451 		fix_problem(ctx, PR_3_ERR_LPF_WRITE_BLOCK, &pctx);
    452 		return 0;
    453 	}
    454 
    455 	/*
    456 	 * Set up the inode structure
    457 	 */
    458 	memset(&inode, 0, sizeof(inode));
    459 	inode.i_mode = 040700;
    460 	inode.i_size = fs->blocksize;
    461 	inode.i_atime = inode.i_ctime = inode.i_mtime = ctx->now;
    462 	inode.i_links_count = 2;
    463 	ext2fs_iblk_set(fs, &inode, 1);
    464 	inode.i_block[0] = blk;
    465 
    466 	/*
    467 	 * Next, write out the inode.
    468 	 */
    469 	pctx.errcode = ext2fs_write_new_inode(fs, ino, &inode);
    470 	if (pctx.errcode) {
    471 		pctx.str = "ext2fs_write_inode";
    472 		fix_problem(ctx, PR_3_CREATE_LPF_ERROR, &pctx);
    473 		return 0;
    474 	}
    475 	/*
    476 	 * Finally, create the directory link
    477 	 */
    478 	pctx.errcode = ext2fs_link(fs, EXT2_ROOT_INO, name, ino, EXT2_FT_DIR);
    479 	if (pctx.errcode) {
    480 		pctx.str = "ext2fs_link";
    481 		fix_problem(ctx, PR_3_CREATE_LPF_ERROR, &pctx);
    482 		return 0;
    483 	}
    484 
    485 	/*
    486 	 * Miscellaneous bookkeeping that needs to be kept straight.
    487 	 */
    488 	e2fsck_add_dir_info(ctx, ino, EXT2_ROOT_INO);
    489 	e2fsck_adjust_inode_count(ctx, EXT2_ROOT_INO, 1);
    490 	ext2fs_icount_store(ctx->inode_count, ino, 2);
    491 	ext2fs_icount_store(ctx->inode_link_info, ino, 2);
    492 	ctx->lost_and_found = ino;
    493 	quota_data_add(ctx->qctx, &inode, ino, fs->blocksize);
    494 	quota_data_inodes(ctx->qctx, &inode, ino, +1);
    495 #if 0
    496 	printf("/lost+found created; inode #%lu\n", ino);
    497 #endif
    498 	return ino;
    499 }
    500 
    501 /*
    502  * This routine will connect a file to lost+found
    503  */
    504 int e2fsck_reconnect_file(e2fsck_t ctx, ext2_ino_t ino)
    505 {
    506 	ext2_filsys fs = ctx->fs;
    507 	errcode_t	retval;
    508 	char		name[80];
    509 	struct problem_context	pctx;
    510 	struct ext2_inode 	inode;
    511 	int		file_type = 0;
    512 
    513 	clear_problem_context(&pctx);
    514 	pctx.ino = ino;
    515 
    516 	if (!ctx->bad_lost_and_found && !ctx->lost_and_found) {
    517 		if (e2fsck_get_lost_and_found(ctx, 1) == 0)
    518 			ctx->bad_lost_and_found++;
    519 	}
    520 	if (ctx->bad_lost_and_found) {
    521 		fix_problem(ctx, PR_3_NO_LPF, &pctx);
    522 		return 1;
    523 	}
    524 
    525 	sprintf(name, "#%u", ino);
    526 	if (ext2fs_read_inode(fs, ino, &inode) == 0)
    527 		file_type = ext2_file_type(inode.i_mode);
    528 	retval = ext2fs_link(fs, ctx->lost_and_found, name, ino, file_type);
    529 	if (retval == EXT2_ET_DIR_NO_SPACE) {
    530 		if (!fix_problem(ctx, PR_3_EXPAND_LF_DIR, &pctx))
    531 			return 1;
    532 		retval = e2fsck_expand_directory(ctx, ctx->lost_and_found,
    533 						 1, 0);
    534 		if (retval) {
    535 			pctx.errcode = retval;
    536 			fix_problem(ctx, PR_3_CANT_EXPAND_LPF, &pctx);
    537 			return 1;
    538 		}
    539 		retval = ext2fs_link(fs, ctx->lost_and_found, name,
    540 				     ino, file_type);
    541 	}
    542 	if (retval) {
    543 		pctx.errcode = retval;
    544 		fix_problem(ctx, PR_3_CANT_RECONNECT, &pctx);
    545 		return 1;
    546 	}
    547 	e2fsck_adjust_inode_count(ctx, ino, 1);
    548 
    549 	return 0;
    550 }
    551 
    552 /*
    553  * Utility routine to adjust the inode counts on an inode.
    554  */
    555 errcode_t e2fsck_adjust_inode_count(e2fsck_t ctx, ext2_ino_t ino, int adj)
    556 {
    557 	ext2_filsys fs = ctx->fs;
    558 	errcode_t		retval;
    559 	struct ext2_inode 	inode;
    560 
    561 	if (!ino)
    562 		return 0;
    563 
    564 	retval = ext2fs_read_inode(fs, ino, &inode);
    565 	if (retval)
    566 		return retval;
    567 
    568 #if 0
    569 	printf("Adjusting link count for inode %lu by %d (from %d)\n", ino, adj,
    570 	       inode.i_links_count);
    571 #endif
    572 
    573 	if (adj == 1) {
    574 		ext2fs_icount_increment(ctx->inode_count, ino, 0);
    575 		if (inode.i_links_count == (__u16) ~0)
    576 			return 0;
    577 		ext2fs_icount_increment(ctx->inode_link_info, ino, 0);
    578 		inode.i_links_count++;
    579 	} else if (adj == -1) {
    580 		ext2fs_icount_decrement(ctx->inode_count, ino, 0);
    581 		if (inode.i_links_count == 0)
    582 			return 0;
    583 		ext2fs_icount_decrement(ctx->inode_link_info, ino, 0);
    584 		inode.i_links_count--;
    585 	}
    586 
    587 	retval = ext2fs_write_inode(fs, ino, &inode);
    588 	if (retval)
    589 		return retval;
    590 
    591 	return 0;
    592 }
    593 
    594 /*
    595  * Fix parent --- this routine fixes up the parent of a directory.
    596  */
    597 struct fix_dotdot_struct {
    598 	ext2_filsys	fs;
    599 	ext2_ino_t	parent;
    600 	int		done;
    601 	e2fsck_t	ctx;
    602 };
    603 
    604 static int fix_dotdot_proc(struct ext2_dir_entry *dirent,
    605 			   int	offset EXT2FS_ATTR((unused)),
    606 			   int	blocksize EXT2FS_ATTR((unused)),
    607 			   char	*buf EXT2FS_ATTR((unused)),
    608 			   void	*priv_data)
    609 {
    610 	struct fix_dotdot_struct *fp = (struct fix_dotdot_struct *) priv_data;
    611 	errcode_t	retval;
    612 	struct problem_context pctx;
    613 
    614 	if ((dirent->name_len & 0xFF) != 2)
    615 		return 0;
    616 	if (strncmp(dirent->name, "..", 2))
    617 		return 0;
    618 
    619 	clear_problem_context(&pctx);
    620 
    621 	retval = e2fsck_adjust_inode_count(fp->ctx, dirent->inode, -1);
    622 	if (retval) {
    623 		pctx.errcode = retval;
    624 		fix_problem(fp->ctx, PR_3_ADJUST_INODE, &pctx);
    625 	}
    626 	retval = e2fsck_adjust_inode_count(fp->ctx, fp->parent, 1);
    627 	if (retval) {
    628 		pctx.errcode = retval;
    629 		fix_problem(fp->ctx, PR_3_ADJUST_INODE, &pctx);
    630 	}
    631 	dirent->inode = fp->parent;
    632 	if (fp->ctx->fs->super->s_feature_incompat &
    633 	    EXT2_FEATURE_INCOMPAT_FILETYPE)
    634 		dirent->name_len = (dirent->name_len & 0xFF) |
    635 			(EXT2_FT_DIR << 8);
    636 	else
    637 		dirent->name_len = dirent->name_len & 0xFF;
    638 
    639 	fp->done++;
    640 	return DIRENT_ABORT | DIRENT_CHANGED;
    641 }
    642 
    643 static void fix_dotdot(e2fsck_t ctx, ext2_ino_t ino, ext2_ino_t parent)
    644 {
    645 	ext2_filsys fs = ctx->fs;
    646 	errcode_t	retval;
    647 	struct fix_dotdot_struct fp;
    648 	struct problem_context pctx;
    649 
    650 	fp.fs = fs;
    651 	fp.parent = parent;
    652 	fp.done = 0;
    653 	fp.ctx = ctx;
    654 
    655 #if 0
    656 	printf("Fixing '..' of inode %lu to be %lu...\n", ino, parent);
    657 #endif
    658 
    659 	clear_problem_context(&pctx);
    660 	pctx.ino = ino;
    661 	retval = ext2fs_dir_iterate(fs, ino, DIRENT_FLAG_INCLUDE_EMPTY,
    662 				    0, fix_dotdot_proc, &fp);
    663 	if (retval || !fp.done) {
    664 		pctx.errcode = retval;
    665 		fix_problem(ctx, retval ? PR_3_FIX_PARENT_ERR :
    666 			    PR_3_FIX_PARENT_NOFIND, &pctx);
    667 		ext2fs_unmark_valid(fs);
    668 	}
    669 	(void) e2fsck_dir_info_set_dotdot(ctx, ino, parent);
    670 	if (e2fsck_dir_info_set_parent(ctx, ino, ctx->lost_and_found))
    671 		fix_problem(ctx, PR_3_NO_DIRINFO, &pctx);
    672 
    673 	return;
    674 }
    675 
    676 /*
    677  * These routines are responsible for expanding a /lost+found if it is
    678  * too small.
    679  */
    680 
    681 struct expand_dir_struct {
    682 	blk64_t			num;
    683 	e2_blkcnt_t		guaranteed_size;
    684 	blk64_t			newblocks;
    685 	blk64_t			last_block;
    686 	errcode_t		err;
    687 	e2fsck_t		ctx;
    688 };
    689 
    690 static int expand_dir_proc(ext2_filsys fs,
    691 			   blk64_t	*blocknr,
    692 			   e2_blkcnt_t	blockcnt,
    693 			   blk64_t ref_block EXT2FS_ATTR((unused)),
    694 			   int ref_offset EXT2FS_ATTR((unused)),
    695 			   void	*priv_data)
    696 {
    697 	struct expand_dir_struct *es = (struct expand_dir_struct *) priv_data;
    698 	blk64_t	new_blk;
    699 	static blk64_t	last_blk = 0;
    700 	char		*block;
    701 	errcode_t	retval;
    702 	e2fsck_t	ctx;
    703 
    704 	ctx = es->ctx;
    705 
    706 	if (es->guaranteed_size && blockcnt >= es->guaranteed_size)
    707 		return BLOCK_ABORT;
    708 
    709 	if (blockcnt > 0)
    710 		es->last_block = blockcnt;
    711 	if (*blocknr) {
    712 		last_blk = *blocknr;
    713 		return 0;
    714 	}
    715 
    716 	if (blockcnt &&
    717 	    (EXT2FS_B2C(fs, last_blk) == EXT2FS_B2C(fs, last_blk + 1)))
    718 		new_blk = last_blk + 1;
    719 	else {
    720 		last_blk &= ~EXT2FS_CLUSTER_MASK(fs);
    721 		retval = ext2fs_new_block2(fs, last_blk, ctx->block_found_map,
    722 					  &new_blk);
    723 		if (retval) {
    724 			es->err = retval;
    725 			return BLOCK_ABORT;
    726 		}
    727 		es->newblocks++;
    728 		ext2fs_block_alloc_stats2(fs, new_blk, +1);
    729 	}
    730 	last_blk = new_blk;
    731 
    732 	if (blockcnt > 0) {
    733 		retval = ext2fs_new_dir_block(fs, 0, 0, &block);
    734 		if (retval) {
    735 			es->err = retval;
    736 			return BLOCK_ABORT;
    737 		}
    738 		es->num--;
    739 		retval = ext2fs_write_dir_block3(fs, new_blk, block, 0);
    740 	} else {
    741 		retval = ext2fs_get_mem(fs->blocksize, &block);
    742 		if (retval) {
    743 			es->err = retval;
    744 			return BLOCK_ABORT;
    745 		}
    746 		memset(block, 0, fs->blocksize);
    747 		retval = io_channel_write_blk64(fs->io, new_blk, 1, block);
    748 	}
    749 	if (retval) {
    750 		es->err = retval;
    751 		return BLOCK_ABORT;
    752 	}
    753 	ext2fs_free_mem(&block);
    754 	*blocknr = new_blk;
    755 	ext2fs_mark_block_bitmap2(ctx->block_found_map, new_blk);
    756 
    757 	if (es->num == 0)
    758 		return (BLOCK_CHANGED | BLOCK_ABORT);
    759 	else
    760 		return BLOCK_CHANGED;
    761 }
    762 
    763 /*
    764  * Ensure that all blocks are marked in the block_found_map, since it's
    765  * possible that the library allocated an extent node block or a block map
    766  * block during the directory rebuilding; these new allocations are not
    767  * captured in block_found_map.  This is bad since we could later use
    768  * block_found_map to allocate more blocks.
    769  */
    770 static int find_new_blocks_proc(ext2_filsys fs,
    771 				blk64_t	*blocknr,
    772 				e2_blkcnt_t	blockcnt,
    773 				blk64_t ref_block EXT2FS_ATTR((unused)),
    774 				int ref_offset EXT2FS_ATTR((unused)),
    775 				void	*priv_data)
    776 {
    777 	struct expand_dir_struct *es = (struct expand_dir_struct *) priv_data;
    778 	e2fsck_t	ctx = es->ctx;
    779 
    780 	ext2fs_mark_block_bitmap2(ctx->block_found_map, *blocknr);
    781 	return 0;
    782 }
    783 
    784 errcode_t e2fsck_expand_directory(e2fsck_t ctx, ext2_ino_t dir,
    785 				  int num, int guaranteed_size)
    786 {
    787 	ext2_filsys fs = ctx->fs;
    788 	errcode_t	retval;
    789 	struct expand_dir_struct es;
    790 	struct ext2_inode	inode;
    791 	blk64_t		sz, before, after;
    792 
    793 	if (!(fs->flags & EXT2_FLAG_RW))
    794 		return EXT2_ET_RO_FILSYS;
    795 
    796 	/*
    797 	 * Read the inode and block bitmaps in; we'll be messing with
    798 	 * them.
    799 	 */
    800 	e2fsck_read_bitmaps(ctx);
    801 
    802 	retval = ext2fs_check_directory(fs, dir);
    803 	if (retval)
    804 		return retval;
    805 
    806 	es.num = num;
    807 	es.guaranteed_size = guaranteed_size;
    808 	es.last_block = 0;
    809 	es.err = 0;
    810 	es.newblocks = 0;
    811 	es.ctx = ctx;
    812 
    813 	before = ext2fs_free_blocks_count(fs->super);
    814 	retval = ext2fs_block_iterate3(fs, dir, BLOCK_FLAG_APPEND,
    815 				       0, expand_dir_proc, &es);
    816 
    817 	if (es.err)
    818 		return es.err;
    819 	after = ext2fs_free_blocks_count(fs->super);
    820 
    821 	/*
    822 	 * If the free block count has dropped by more than the blocks we
    823 	 * allocated ourselves, then we must've allocated some extent/map
    824 	 * blocks.  Therefore, we must iterate this dir's blocks again to
    825 	 * ensure that all newly allocated blocks are captured in
    826 	 * block_found_map.
    827 	 */
    828 	if ((before - after) > es.newblocks) {
    829 		retval = ext2fs_block_iterate3(fs, dir, BLOCK_FLAG_READ_ONLY,
    830 					       0, find_new_blocks_proc, &es);
    831 		if (es.err)
    832 			return es.err;
    833 	}
    834 
    835 	/*
    836 	 * Update the size and block count fields in the inode.
    837 	 */
    838 	retval = ext2fs_read_inode(fs, dir, &inode);
    839 	if (retval)
    840 		return retval;
    841 
    842 	sz = (es.last_block + 1) * fs->blocksize;
    843 	inode.i_size = sz;
    844 	inode.i_size_high = sz >> 32;
    845 	ext2fs_iblk_add_blocks(fs, &inode, es.newblocks);
    846 	quota_data_add(ctx->qctx, &inode, dir, es.newblocks * fs->blocksize);
    847 
    848 	e2fsck_write_inode(ctx, dir, &inode, "expand_directory");
    849 
    850 	return 0;
    851 }
    852 
    853