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