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