e2fsck/pass3.c

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