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