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