misc/tune2fs.c

   1 /*
   2  * tune2fs.c - Change the file system parameters on an ext2 file system
   3  *
   4  * Copyright (C) 1992, 1993, 1994  Remy Card <card@masi.ibp.fr>
   5  *                                 Laboratoire MASI, Institut Blaise Pascal
   6  *                                 Universite Pierre et Marie Curie (Paris VI)
   7  *
   8  * Copyright 1995, 1996, 1997, 1998, 1999, 2000 by Theodore Ts'o.
   9  *
  10  * %Begin-Header%
  11  * This file may be redistributed under the terms of the GNU Public
  12  * License.
  13  * %End-Header%
  14  */
  15
  16 /*
  17  * History:
  18  * 93/06/01     - Creation
  19  * 93/10/31     - Added the -c option to change the maximal mount counts
  20  * 93/12/14     - Added -l flag to list contents of superblock
  21  *                M.J.E. Mol (marcel@duteca.et.tudelft.nl)
  22  *                F.W. ten Wolde (franky@duteca.et.tudelft.nl)
  23  * 93/12/29     - Added the -e option to change errors behavior
  24  * 94/02/27     - Ported to use the ext2fs library
  25  * 94/03/06     - Added the checks interval from Uwe Ohse (uwe@tirka.gun.de)
  26  */
  27
  28 #define _XOPEN_SOURCE 600 /* for inclusion of strptime() */
  29 #include "config.h"
  30 #include <fcntl.h>
  31 #include <grp.h>
  32 #ifdef HAVE_GETOPT_H
  33 #include <getopt.h>
  34 #else
  35 extern char *optarg;
  36 extern int optind;
  37 #endif
  38 #include <pwd.h>
  39 #include <stdio.h>
  40 #ifdef HAVE_STDLIB_H
  41 #include <stdlib.h>
  42 #endif
  43 #ifdef HAVE_STRINGS_H
  44 #include <strings.h>    /* for strcasecmp() */
  45 #else
  46 #define _BSD_SOURCE     /* for inclusion of strcasecmp() via <string.h> */
  47 #define _DEFAULT_SOURCE   /* since glibc 2.20 _BSD_SOURCE is deprecated */
  48 #endif
  49 #include <string.h>
  50 #include <time.h>
  51 #include <unistd.h>
  52 #include <sys/types.h>
  53 #include <libgen.h>
  54 #include <limits.h>
  55
  56 #include "ext2fs/ext2_fs.h"
  57 #include "ext2fs/ext2fs.h"
  58 #include "ext2fs/kernel-jbd.h"
  59 #include "et/com_err.h"
  60 #include "support/plausible.h"
  61 #include "support/quotaio.h"
  62 #include "uuid/uuid.h"
  63 #include "e2p/e2p.h"
  64 #include "util.h"
  65 #include "blkid/blkid.h"
  66
  67 #include "../version.h"
  68 #include "support/nls-enable.h"
  69
  70 #define QOPT_ENABLE     (1)
  71 #define QOPT_DISABLE    (-1)
  72
  73 extern int ask_yn(const char *string, int def);
  74
  75 const char *program_name = "tune2fs";
  76 char *device_name;
  77 char *new_label, *new_last_mounted, *new_UUID;
  78 char *io_options;
  79 static int c_flag, C_flag, e_flag, f_flag, g_flag, i_flag, l_flag, L_flag;
  80 static int m_flag, M_flag, Q_flag, r_flag, s_flag = -1, u_flag, U_flag, T_flag;
  81 static int I_flag;
  82 static int clear_mmp;
  83 static time_t last_check_time;
  84 static int print_label;
  85 static int max_mount_count, mount_count, mount_flags;
  86 static unsigned long interval;
  87 static blk64_t reserved_blocks;
  88 static double reserved_ratio;
  89 static unsigned long resgid, resuid;
  90 static unsigned short errors;
  91 static int open_flag;
  92 static char *features_cmd;
  93 static char *mntopts_cmd;
  94 static int stride, stripe_width;
  95 static int stride_set, stripe_width_set;
  96 static char *extended_cmd;
  97 static unsigned long new_inode_size;
  98 static char *ext_mount_opts;
  99 static int quota_enable[MAXQUOTAS];
 100 static int rewrite_checksums;
 101 static int feature_64bit;
 102 static int fsck_requested;
 103 static char *undo_file;
 104
 105 int journal_size, journal_flags;
 106 char *journal_device;
 107 static blk64_t journal_location = ~0LL;
 108
 109 static struct list_head blk_move_list;
 110
 111 struct blk_move {
 112         struct list_head list;
 113         blk64_t old_loc;
 114         blk64_t new_loc;
 115 };
 116
 117 errcode_t ext2fs_run_ext3_journal(ext2_filsys *fs);
 118
 119 static const char *fsck_explain = N_("\nThis operation requires a freshly checked filesystem.\n");
 120
 121 static const char *please_fsck = N_("Please run e2fsck -f on the filesystem.\n");
 122 static const char *please_dir_fsck =
 123                 N_("Please run e2fsck -fD on the filesystem.\n");
 124
 125 #ifdef CONFIG_BUILD_FINDFS
 126 void do_findfs(int argc, char **argv);
 127 #endif
 128
 129 #ifdef CONFIG_JBD_DEBUG         /* Enabled by configure --enable-jbd-debug */
 130 int journal_enable_debug = -1;
 131 #endif
 132
 133 static void usage(void)
 134 {
 135         fprintf(stderr,
 136                 _("Usage: %s [-c max_mounts_count] [-e errors_behavior] [-f] "
 137                   "[-g group]\n"
 138                   "\t[-i interval[d|m|w]] [-j] [-J journal_options] [-l]\n"
 139                   "\t[-m reserved_blocks_percent] [-o [^]mount_options[,...]]\n"
 140                   "\t[-r reserved_blocks_count] [-u user] [-C mount_count]\n"
 141                   "\t[-L volume_label] [-M last_mounted_dir]\n"
 142                   "\t[-O [^]feature[,...]] [-Q quota_options]\n"
 143                   "\t[-E extended-option[,...]] [-T last_check_time] "
 144                   "[-U UUID]\n\t[-I new_inode_size] [-z undo_file] device\n"),
 145                 program_name);
 146         exit(1);
 147 }
 148
 149 static __u32 ok_features[3] = {
 150         /* Compat */
 151         EXT3_FEATURE_COMPAT_HAS_JOURNAL |
 152                 EXT2_FEATURE_COMPAT_DIR_INDEX,
 153         /* Incompat */
 154         EXT2_FEATURE_INCOMPAT_FILETYPE |
 155                 EXT3_FEATURE_INCOMPAT_EXTENTS |
 156                 EXT4_FEATURE_INCOMPAT_FLEX_BG |
 157                 EXT4_FEATURE_INCOMPAT_EA_INODE|
 158                 EXT4_FEATURE_INCOMPAT_MMP |
 159                 EXT4_FEATURE_INCOMPAT_64BIT |
 160                 EXT4_FEATURE_INCOMPAT_ENCRYPT |
 161                 EXT4_FEATURE_INCOMPAT_CSUM_SEED |
 162                 EXT4_FEATURE_INCOMPAT_LARGEDIR,
 163         /* R/O compat */
 164         EXT2_FEATURE_RO_COMPAT_LARGE_FILE |
 165                 EXT4_FEATURE_RO_COMPAT_HUGE_FILE|
 166                 EXT4_FEATURE_RO_COMPAT_DIR_NLINK|
 167                 EXT4_FEATURE_RO_COMPAT_EXTRA_ISIZE|
 168                 EXT4_FEATURE_RO_COMPAT_GDT_CSUM |
 169                 EXT2_FEATURE_RO_COMPAT_SPARSE_SUPER |
 170                 EXT4_FEATURE_RO_COMPAT_QUOTA |
 171                 EXT4_FEATURE_RO_COMPAT_METADATA_CSUM |
 172                 EXT4_FEATURE_RO_COMPAT_READONLY |
 173                 EXT4_FEATURE_RO_COMPAT_PROJECT
 174 };
 175
 176 static __u32 clear_ok_features[3] = {
 177         /* Compat */
 178         EXT3_FEATURE_COMPAT_HAS_JOURNAL |
 179                 EXT2_FEATURE_COMPAT_RESIZE_INODE |
 180                 EXT2_FEATURE_COMPAT_DIR_INDEX,
 181         /* Incompat */
 182         EXT2_FEATURE_INCOMPAT_FILETYPE |
 183                 EXT4_FEATURE_INCOMPAT_FLEX_BG |
 184                 EXT4_FEATURE_INCOMPAT_MMP |
 185                 EXT4_FEATURE_INCOMPAT_64BIT |
 186                 EXT4_FEATURE_INCOMPAT_CSUM_SEED,
 187         /* R/O compat */
 188         EXT2_FEATURE_RO_COMPAT_LARGE_FILE |
 189                 EXT4_FEATURE_RO_COMPAT_HUGE_FILE|
 190                 EXT4_FEATURE_RO_COMPAT_DIR_NLINK|
 191                 EXT4_FEATURE_RO_COMPAT_EXTRA_ISIZE|
 192                 EXT4_FEATURE_RO_COMPAT_GDT_CSUM |
 193                 EXT4_FEATURE_RO_COMPAT_QUOTA |
 194                 EXT4_FEATURE_RO_COMPAT_PROJECT |
 195                 EXT4_FEATURE_RO_COMPAT_METADATA_CSUM |
 196                 EXT4_FEATURE_RO_COMPAT_READONLY
 197 };
 198
 199 /**
 200  * Try to get journal super block if any
 201  */
 202 static int get_journal_sb(ext2_filsys jfs, char buf[SUPERBLOCK_SIZE])
 203 {
 204         int retval;
 205         journal_superblock_t *jsb;
 206
 207         if (!ext2fs_has_feature_journal_dev(jfs->super)) {
 208                 return EXT2_ET_UNSUPP_FEATURE;
 209         }
 210
 211         /* Get the journal superblock */
 212         if ((retval = io_channel_read_blk64(jfs->io,
 213             ext2fs_journal_sb_start(jfs->blocksize), -SUPERBLOCK_SIZE, buf))) {
 214                 com_err(program_name, retval, "%s",
 215                 _("while reading journal superblock"));
 216                 return retval;
 217         }
 218
 219         jsb = (journal_superblock_t *) buf;
 220         if ((jsb->s_header.h_magic != (unsigned)ntohl(JFS_MAGIC_NUMBER)) ||
 221             (jsb->s_header.h_blocktype != (unsigned)ntohl(JFS_SUPERBLOCK_V2))) {
 222                 fputs(_("Journal superblock not found!\n"), stderr);
 223                 return EXT2_ET_BAD_MAGIC;
 224         }
 225
 226         return 0;
 227 }
 228
 229 static __u8 *journal_user(__u8 uuid[UUID_SIZE], __u8 s_users[JFS_USERS_SIZE],
 230                           int nr_users)
 231 {
 232         int i;
 233         for (i = 0; i < nr_users; i++) {
 234                 if (memcmp(uuid, &s_users[i * UUID_SIZE], UUID_SIZE) == 0)
 235                         return &s_users[i * UUID_SIZE];
 236         }
 237
 238         return NULL;
 239 }
 240
 241 /*
 242  * Remove an external journal from the filesystem
 243  */
 244 static int remove_journal_device(ext2_filsys fs)
 245 {
 246         char            *journal_path;
 247         ext2_filsys     jfs;
 248         char            buf[SUPERBLOCK_SIZE] __attribute__ ((aligned(8)));
 249         journal_superblock_t    *jsb;
 250         int             i, nr_users;
 251         errcode_t       retval;
 252         int             commit_remove_journal = 0;
 253         io_manager      io_ptr;
 254
 255         if (f_flag)
 256                 commit_remove_journal = 1; /* force removal even if error */
 257
 258         uuid_unparse(fs->super->s_journal_uuid, buf);
 259         journal_path = blkid_get_devname(NULL, "UUID", buf);
 260
 261         if (!journal_path) {
 262                 journal_path =
 263                         ext2fs_find_block_device(fs->super->s_journal_dev);
 264                 if (!journal_path)
 265                         goto no_valid_journal;
 266         }
 267
 268 #ifdef CONFIG_TESTIO_DEBUG
 269         if (getenv("TEST_IO_FLAGS") || getenv("TEST_IO_BLOCK")) {
 270                 io_ptr = test_io_manager;
 271                 test_io_backing_manager = unix_io_manager;
 272         } else
 273 #endif
 274                 io_ptr = unix_io_manager;
 275         retval = ext2fs_open(journal_path, EXT2_FLAG_RW|
 276                              EXT2_FLAG_JOURNAL_DEV_OK, 0,
 277                              fs->blocksize, io_ptr, &jfs);
 278         if (retval) {
 279                 com_err(program_name, retval, "%s",
 280                         _("while trying to open external journal"));
 281                 goto no_valid_journal;
 282         }
 283
 284         if ((retval = get_journal_sb(jfs, buf))) {
 285                 if (retval == EXT2_ET_UNSUPP_FEATURE)
 286                         fprintf(stderr, _("%s is not a journal device.\n"),
 287                                 journal_path);
 288                 goto no_valid_journal;
 289         }
 290
 291         jsb = (journal_superblock_t *) buf;
 292         /* Find the filesystem UUID */
 293         nr_users = ntohl(jsb->s_nr_users);
 294
 295         if (!journal_user(fs->super->s_uuid, jsb->s_users, nr_users)) {
 296                 fputs(_("Filesystem's UUID not found on journal device.\n"),
 297                       stderr);
 298                 commit_remove_journal = 1;
 299                 goto no_valid_journal;
 300         }
 301         nr_users--;
 302         for (i = 0; i < nr_users; i++)
 303                 memcpy(&jsb->s_users[i * 16], &jsb->s_users[(i + 1) * 16], 16);
 304         jsb->s_nr_users = htonl(nr_users);
 305
 306         /* Write back the journal superblock */
 307         retval = io_channel_write_blk64(jfs->io,
 308                                         ext2fs_journal_sb_start(fs->blocksize),
 309                                         -SUPERBLOCK_SIZE, buf);
 310         if (retval) {
 311                 com_err(program_name, retval,
 312                         "while writing journal superblock.");
 313                 goto no_valid_journal;
 314         }
 315
 316         commit_remove_journal = 1;
 317
 318 no_valid_journal:
 319         if (commit_remove_journal == 0) {
 320                 fputs(_("Cannot locate journal device. It was NOT removed\n"
 321                         "Use -f option to remove missing journal device.\n"),
 322                       stderr);
 323                 return 1;
 324         }
 325         fs->super->s_journal_dev = 0;
 326         memset(fs->super->s_jnl_blocks, 0, sizeof(fs->super->s_jnl_blocks));
 327         uuid_clear(fs->super->s_journal_uuid);
 328         ext2fs_mark_super_dirty(fs);
 329         fputs(_("Journal removed\n"), stdout);
 330         free(journal_path);
 331
 332         return 0;
 333 }
 334
 335 /* Helper function for remove_journal_inode */
 336 static int release_blocks_proc(ext2_filsys fs, blk64_t *blocknr,
 337                                e2_blkcnt_t blockcnt EXT2FS_ATTR((unused)),
 338                                blk64_t ref_block EXT2FS_ATTR((unused)),
 339                                int ref_offset EXT2FS_ATTR((unused)),
 340                                void *private EXT2FS_ATTR((unused)))
 341 {
 342         blk64_t block;
 343         int     group;
 344
 345         block = *blocknr;
 346         ext2fs_unmark_block_bitmap2(fs->block_map, block);
 347         group = ext2fs_group_of_blk2(fs, block);
 348         ext2fs_bg_free_blocks_count_set(fs, group, ext2fs_bg_free_blocks_count(fs, group) + 1);
 349         ext2fs_group_desc_csum_set(fs, group);
 350         ext2fs_free_blocks_count_add(fs->super, EXT2FS_CLUSTER_RATIO(fs));
 351         return 0;
 352 }
 353
 354 /*
 355  * Remove the journal inode from the filesystem
 356  */
 357 static errcode_t remove_journal_inode(ext2_filsys fs)
 358 {
 359         struct ext2_inode       inode;
 360         errcode_t               retval;
 361         ino_t                   ino = fs->super->s_journal_inum;
 362
 363         retval = ext2fs_read_inode(fs, ino,  &inode);
 364         if (retval) {
 365                 com_err(program_name, retval, "%s",
 366                         _("while reading journal inode"));
 367                 return retval;
 368         }
 369         if (ino == EXT2_JOURNAL_INO) {
 370                 retval = ext2fs_read_bitmaps(fs);
 371                 if (retval) {
 372                         com_err(program_name, retval, "%s",
 373                                 _("while reading bitmaps"));
 374                         return retval;
 375                 }
 376                 retval = ext2fs_block_iterate3(fs, ino,
 377                                                BLOCK_FLAG_READ_ONLY, NULL,
 378                                                release_blocks_proc, NULL);
 379                 if (retval) {
 380                         com_err(program_name, retval, "%s",
 381                                 _("while clearing journal inode"));
 382                         return retval;
 383                 }
 384                 memset(&inode, 0, sizeof(inode));
 385                 ext2fs_mark_bb_dirty(fs);
 386                 fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
 387         } else
 388                 inode.i_flags &= ~EXT2_IMMUTABLE_FL;
 389         retval = ext2fs_write_inode(fs, ino, &inode);
 390         if (retval) {
 391                 com_err(program_name, retval, "%s",
 392                         _("while writing journal inode"));
 393                 return retval;
 394         }
 395         fs->super->s_journal_inum = 0;
 396         memset(fs->super->s_jnl_blocks, 0, sizeof(fs->super->s_jnl_blocks));
 397         ext2fs_mark_super_dirty(fs);
 398
 399         return 0;
 400 }
 401
 402 /*
 403  * Update the default mount options
 404  */
 405 static int update_mntopts(ext2_filsys fs, char *mntopts)
 406 {
 407         struct ext2_super_block *sb = fs->super;
 408
 409         if (e2p_edit_mntopts(mntopts, &sb->s_default_mount_opts, ~0)) {
 410                 fprintf(stderr, _("Invalid mount option set: %s\n"),
 411                         mntopts);
 412                 return 1;
 413         }
 414         ext2fs_mark_super_dirty(fs);
 415
 416         return 0;
 417 }
 418
 419 static void check_fsck_needed(ext2_filsys fs, const char *prompt)
 420 {
 421         /* Refuse to modify anything but a freshly checked valid filesystem. */
 422         if (!(fs->super->s_state & EXT2_VALID_FS) ||
 423             (fs->super->s_state & EXT2_ERROR_FS) ||
 424             (fs->super->s_lastcheck < fs->super->s_mtime)) {
 425                 puts(_(fsck_explain));
 426                 puts(_(please_fsck));
 427                 if (mount_flags & EXT2_MF_READONLY)
 428                         printf("%s", _("(and reboot afterwards!)\n"));
 429                 exit(1);
 430         }
 431
 432         /* Give the admin a few seconds to bail out of a dangerous op. */
 433         if (!getenv("TUNE2FS_FORCE_PROMPT") && (!isatty(0) || !isatty(1)))
 434                 return;
 435
 436         puts(prompt);
 437         proceed_question(5);
 438 }
 439
 440 static void request_dir_fsck_afterwards(ext2_filsys fs)
 441 {
 442         static int requested;
 443
 444         if (requested++)
 445                 return;
 446         fsck_requested++;
 447         fs->super->s_state &= ~EXT2_VALID_FS;
 448         puts(_(fsck_explain));
 449         puts(_(please_dir_fsck));
 450         if (mount_flags & EXT2_MF_READONLY)
 451                 printf("%s", _("(and reboot afterwards!)\n"));
 452 }
 453
 454 static void request_fsck_afterwards(ext2_filsys fs)
 455 {
 456         static int requested = 0;
 457
 458         if (requested++)
 459                 return;
 460         fsck_requested++;
 461         fs->super->s_state &= ~EXT2_VALID_FS;
 462         printf("\n%s\n", _(please_fsck));
 463         if (mount_flags & EXT2_MF_READONLY)
 464                 printf("%s", _("(and reboot afterwards!)\n"));
 465 }
 466
 467 static void convert_64bit(ext2_filsys fs, int direction)
 468 {
 469         /*
 470          * Is resize2fs going to demand a fsck run? Might as well tell the
 471          * user now.
 472          */
 473         if (!fsck_requested &&
 474             ((fs->super->s_state & EXT2_ERROR_FS) ||
 475              !(fs->super->s_state & EXT2_VALID_FS) ||
 476              fs->super->s_lastcheck < fs->super->s_mtime))
 477                 request_fsck_afterwards(fs);
 478         if (fsck_requested)
 479                 fprintf(stderr, _("After running e2fsck, please run `resize2fs %s %s"),
 480                         direction > 0 ? "-b" : "-s", fs->device_name);
 481         else
 482                 fprintf(stderr, _("Please run `resize2fs %s %s"),
 483                         direction > 0 ? "-b" : "-s", fs->device_name);
 484
 485         if (undo_file)
 486                 fprintf(stderr, _(" -z \"%s\""), undo_file);
 487         if (direction > 0)
 488                 fprintf(stderr, _("' to enable 64-bit mode.\n"));
 489         else
 490                 fprintf(stderr, _("' to disable 64-bit mode.\n"));
 491 }
 492
 493 /* Rewrite extents */
 494 static errcode_t rewrite_extents(ext2_filsys fs, ext2_ino_t ino,
 495                                  struct ext2_inode *inode)
 496 {
 497         ext2_extent_handle_t    handle;
 498         struct ext2fs_extent    extent;
 499         errcode_t               errcode;
 500         struct ext2_extent_info info;
 501
 502         if (!(inode->i_flags & EXT4_EXTENTS_FL) ||
 503             !ext2fs_has_feature_metadata_csum(fs->super))
 504                 return 0;
 505
 506         errcode = ext2fs_extent_open(fs, ino, &handle);
 507         if (errcode)
 508                 return errcode;
 509
 510         errcode = ext2fs_extent_get(handle, EXT2_EXTENT_ROOT, &extent);
 511         if (errcode)
 512                 goto out;
 513
 514         do {
 515                 errcode = ext2fs_extent_get_info(handle, &info);
 516                 if (errcode)
 517                         break;
 518
 519                 /*
 520                  * If this is the first extent in an extent block that we
 521                  * haven't visited, rewrite the extent to force the ETB
 522                  * checksum to be rewritten.
 523                  */
 524                 if (info.curr_entry == 1 && info.curr_level != 0 &&
 525                     !(extent.e_flags & EXT2_EXTENT_FLAGS_SECOND_VISIT)) {
 526                         errcode = ext2fs_extent_replace(handle, 0, &extent);
 527                         if (errcode)
 528                                 break;
 529                 }
 530
 531                 /* Skip to the end of a block of leaf nodes */
 532                 if (extent.e_flags & EXT2_EXTENT_FLAGS_LEAF) {
 533                         errcode = ext2fs_extent_get(handle,
 534                                                     EXT2_EXTENT_LAST_SIB,
 535                                                     &extent);
 536                         if (errcode)
 537                                 break;
 538                 }
 539
 540                 errcode = ext2fs_extent_get(handle, EXT2_EXTENT_NEXT, &extent);
 541         } while (errcode == 0);
 542
 543 out:
 544         /* Ok if we run off the end */
 545         if (errcode == EXT2_ET_EXTENT_NO_NEXT)
 546                 errcode = 0;
 547         ext2fs_extent_free(handle);
 548         return errcode;
 549 }
 550
 551 /*
 552  * Rewrite directory blocks with checksums
 553  */
 554 struct rewrite_dir_context {
 555         char *buf;
 556         errcode_t errcode;
 557         ext2_ino_t dir;
 558         int is_htree;
 559 };
 560
 561 static int rewrite_dir_block(ext2_filsys fs,
 562                              blk64_t    *blocknr,
 563                              e2_blkcnt_t blockcnt EXT2FS_ATTR((unused)),
 564                              blk64_t    ref_block EXT2FS_ATTR((unused)),
 565                              int        ref_offset EXT2FS_ATTR((unused)),
 566                              void       *priv_data)
 567 {
 568         struct ext2_dx_countlimit *dcl = NULL;
 569         struct rewrite_dir_context *ctx = priv_data;
 570         int dcl_offset, changed = 0;
 571
 572         ctx->errcode = ext2fs_read_dir_block4(fs, *blocknr, ctx->buf, 0,
 573                                               ctx->dir);
 574         if (ctx->errcode)
 575                 return BLOCK_ABORT;
 576
 577         /* if htree node... */
 578         if (ctx->is_htree)
 579                 ext2fs_get_dx_countlimit(fs, (struct ext2_dir_entry *)ctx->buf,
 580                                          &dcl, &dcl_offset);
 581         if (dcl) {
 582                 if (!ext2fs_has_feature_metadata_csum(fs->super)) {
 583                         /* Ensure limit is the max size */
 584                         int max_entries = (fs->blocksize - dcl_offset) /
 585                                           sizeof(struct ext2_dx_entry);
 586                         if (ext2fs_le16_to_cpu(dcl->limit) != max_entries) {
 587                                 changed = 1;
 588                                 dcl->limit = ext2fs_cpu_to_le16(max_entries);
 589                         }
 590                 } else {
 591                         /* If htree block is full then rebuild the dir */
 592                         if (ext2fs_le16_to_cpu(dcl->count) ==
 593                             ext2fs_le16_to_cpu(dcl->limit)) {
 594                                 request_dir_fsck_afterwards(fs);
 595                                 return 0;
 596                         }
 597                         /*
 598                          * Ensure dcl->limit is small enough to leave room for
 599                          * the checksum tail.
 600                          */
 601                         int max_entries = (fs->blocksize - (dcl_offset +
 602                                                 sizeof(struct ext2_dx_tail))) /
 603                                           sizeof(struct ext2_dx_entry);
 604                         if (ext2fs_le16_to_cpu(dcl->limit) != max_entries)
 605                                 dcl->limit = ext2fs_cpu_to_le16(max_entries);
 606                         /* Always rewrite checksum */
 607                         changed = 1;
 608                 }
 609         } else {
 610                 unsigned int rec_len, name_size;
 611                 char *top = ctx->buf + fs->blocksize;
 612                 struct ext2_dir_entry *de = (struct ext2_dir_entry *)ctx->buf;
 613                 struct ext2_dir_entry *last_de = NULL, *penultimate_de = NULL;
 614
 615                 /* Find last and penultimate dirent */
 616                 while ((char *)de < top) {
 617                         penultimate_de = last_de;
 618                         last_de = de;
 619                         ctx->errcode = ext2fs_get_rec_len(fs, de, &rec_len);
 620                         if (!ctx->errcode && !rec_len)
 621                                 ctx->errcode = EXT2_ET_DIR_CORRUPTED;
 622                         if (ctx->errcode)
 623                                 return BLOCK_ABORT;
 624                         de = (struct ext2_dir_entry *)(((char *)de) + rec_len);
 625                 }
 626                 ctx->errcode = ext2fs_get_rec_len(fs, last_de, &rec_len);
 627                 if (ctx->errcode)
 628                         return BLOCK_ABORT;
 629                 name_size = ext2fs_dirent_name_len(last_de);
 630
 631                 if (!ext2fs_has_feature_metadata_csum(fs->super)) {
 632                         if (!penultimate_de)
 633                                 return 0;
 634                         if (last_de->inode ||
 635                             name_size ||
 636                             rec_len != sizeof(struct ext2_dir_entry_tail))
 637                                 return 0;
 638                         /*
 639                          * The last dirent is unused and the right length to
 640                          * have stored a checksum.  Erase it.
 641                          */
 642                         ctx->errcode = ext2fs_get_rec_len(fs, penultimate_de,
 643                                                           &rec_len);
 644                         if (!rec_len)
 645                                 ctx->errcode = EXT2_ET_DIR_CORRUPTED;
 646                         if (ctx->errcode)
 647                                 return BLOCK_ABORT;
 648                         ext2fs_set_rec_len(fs, rec_len +
 649                                         sizeof(struct ext2_dir_entry_tail),
 650                                         penultimate_de);
 651                         changed = 1;
 652                 } else {
 653                         unsigned csum_size = sizeof(struct ext2_dir_entry_tail);
 654                         struct ext2_dir_entry_tail *t;
 655
 656                         /*
 657                          * If the last dirent looks like the tail, just update
 658                          * the checksum.
 659                          */
 660                         if (!last_de->inode &&
 661                             rec_len == csum_size) {
 662                                 t = (struct ext2_dir_entry_tail *)last_de;
 663                                 t->det_reserved_name_len =
 664                                                 EXT2_DIR_NAME_LEN_CSUM;
 665                                 changed = 1;
 666                                 goto out;
 667                         }
 668                         if (name_size & 3)
 669                                 name_size = (name_size & ~3) + 4;
 670                         /* If there's not enough space for the tail, e2fsck */
 671                         if (rec_len <= (8 + name_size + csum_size)) {
 672                                 request_dir_fsck_afterwards(fs);
 673                                 return 0;
 674                         }
 675                         /* Shorten that last de and insert the tail */
 676                         ext2fs_set_rec_len(fs, rec_len - csum_size, last_de);
 677                         t = EXT2_DIRENT_TAIL(ctx->buf, fs->blocksize);
 678                         ext2fs_initialize_dirent_tail(fs, t);
 679
 680                         /* Always update checksum */
 681                         changed = 1;
 682                 }
 683         }
 684
 685 out:
 686         if (!changed)
 687                 return 0;
 688
 689         ctx->errcode = ext2fs_write_dir_block4(fs, *blocknr, ctx->buf,
 690                                                0, ctx->dir);
 691         if (ctx->errcode)
 692                 return BLOCK_ABORT;
 693
 694         return 0;
 695 }
 696
 697 static errcode_t rewrite_directory(ext2_filsys fs, ext2_ino_t dir,
 698                                    struct ext2_inode *inode)
 699 {
 700         errcode_t       retval;
 701         struct rewrite_dir_context ctx;
 702
 703         retval = ext2fs_get_mem(fs->blocksize, &ctx.buf);
 704         if (retval)
 705                 return retval;
 706
 707         ctx.is_htree = (inode->i_flags & EXT2_INDEX_FL);
 708         ctx.dir = dir;
 709         ctx.errcode = 0;
 710         retval = ext2fs_block_iterate3(fs, dir, BLOCK_FLAG_READ_ONLY |
 711                                                 BLOCK_FLAG_DATA_ONLY,
 712                                        0, rewrite_dir_block, &ctx);
 713
 714         ext2fs_free_mem(&ctx.buf);
 715         if (retval)
 716                 return retval;
 717
 718         return ctx.errcode;
 719 }
 720
 721 /*
 722  * Context information that does not change across rewrite_one_inode()
 723  * invocations.
 724  */
 725 struct rewrite_context {
 726         ext2_filsys fs;
 727         struct ext2_inode *zero_inode;
 728         char *ea_buf;
 729         int inode_size;
 730 };
 731
 732 #define fatal_err(code, args...)                \
 733         do {                                    \
 734                 com_err(__func__, code, args);  \
 735                 exit(1);                        \
 736         } while (0);
 737
 738 static void update_ea_inode_hash(struct rewrite_context *ctx, ext2_ino_t ino,
 739                                  struct ext2_inode *inode)
 740 {
 741         errcode_t retval;
 742         ext2_file_t file;
 743         __u32 hash;
 744
 745         retval = ext2fs_file_open(ctx->fs, ino, 0, &file);
 746         if (retval)
 747                 fatal_err(retval, "open ea_inode");
 748         retval = ext2fs_file_read(file, ctx->ea_buf, inode->i_size,
 749                                   NULL);
 750         if (retval)
 751                 fatal_err(retval, "read ea_inode");
 752         retval = ext2fs_file_close(file);
 753         if (retval)
 754                 fatal_err(retval, "close ea_inode");
 755
 756         hash = ext2fs_crc32c_le(ctx->fs->csum_seed,
 757                                 (unsigned char *) ctx->ea_buf, inode->i_size);
 758         ext2fs_set_ea_inode_hash(inode, hash);
 759 }
 760
 761 static int update_xattr_entry_hashes(ext2_filsys fs,
 762                                      struct ext2_ext_attr_entry *entry,
 763                                      struct ext2_ext_attr_entry *end)
 764 {
 765         int modified = 0;
 766         errcode_t retval;
 767
 768         while (entry < end && !EXT2_EXT_IS_LAST_ENTRY(entry)) {
 769                 if (entry->e_value_inum) {
 770                         retval = ext2fs_ext_attr_hash_entry2(fs, entry, NULL,
 771                                                              &entry->e_hash);
 772                         if (retval)
 773                                 fatal_err(retval, "hash ea_inode entry");
 774                         modified = 1;
 775                 }
 776                 entry = EXT2_EXT_ATTR_NEXT(entry);
 777         }
 778         return modified;
 779 }
 780
 781 static void update_inline_xattr_hashes(struct rewrite_context *ctx,
 782                                        struct ext2_inode_large *inode)
 783 {
 784         struct ext2_ext_attr_entry *start, *end;
 785         __u32 *ea_magic;
 786
 787         if (inode->i_extra_isize == 0)
 788                 return;
 789
 790         if (inode->i_extra_isize & 3 ||
 791             inode->i_extra_isize > ctx->inode_size - EXT2_GOOD_OLD_INODE_SIZE)
 792                 fatal_err(EXT2_ET_INODE_CORRUPTED, "bad i_extra_isize")
 793
 794         ea_magic = (__u32 *)((char *)inode + EXT2_GOOD_OLD_INODE_SIZE +
 795                                 inode->i_extra_isize);
 796         if (*ea_magic != EXT2_EXT_ATTR_MAGIC)
 797                 return;
 798
 799         start = (struct ext2_ext_attr_entry *)(ea_magic + 1);
 800         end = (struct ext2_ext_attr_entry *)((char *)inode + ctx->inode_size);
 801
 802         update_xattr_entry_hashes(ctx->fs, start, end);
 803 }
 804
 805 static void update_block_xattr_hashes(struct rewrite_context *ctx,
 806                                       char *block_buf)
 807 {
 808         struct ext2_ext_attr_header *header;
 809         struct ext2_ext_attr_entry *start, *end;
 810
 811         header = (struct ext2_ext_attr_header *)block_buf;
 812         if (header->h_magic != EXT2_EXT_ATTR_MAGIC)
 813                 return;
 814
 815         start = (struct ext2_ext_attr_entry *)(header+1);
 816         end = (struct ext2_ext_attr_entry *)(block_buf + ctx->fs->blocksize);
 817
 818         if (update_xattr_entry_hashes(ctx->fs, start, end))
 819                 ext2fs_ext_attr_block_rehash(header, end);
 820 }
 821
 822 static void rewrite_one_inode(struct rewrite_context *ctx, ext2_ino_t ino,
 823                               struct ext2_inode *inode)
 824 {
 825         blk64_t file_acl_block;
 826         errcode_t retval;
 827
 828         if (!ext2fs_test_inode_bitmap2(ctx->fs->inode_map, ino)) {
 829                 if (!memcmp(inode, ctx->zero_inode, ctx->inode_size))
 830                         return;
 831                 memset(inode, 0, ctx->inode_size);
 832         }
 833
 834         if (inode->i_flags & EXT4_EA_INODE_FL)
 835                 update_ea_inode_hash(ctx, ino, inode);
 836
 837         if (ctx->inode_size != EXT2_GOOD_OLD_INODE_SIZE)
 838                 update_inline_xattr_hashes(ctx,
 839                                            (struct ext2_inode_large *)inode);
 840
 841         retval = ext2fs_write_inode_full(ctx->fs, ino, inode, ctx->inode_size);
 842         if (retval)
 843                 fatal_err(retval, "while writing inode");
 844
 845         retval = rewrite_extents(ctx->fs, ino, inode);
 846         if (retval)
 847                 fatal_err(retval, "while rewriting extents");
 848
 849         if (LINUX_S_ISDIR(inode->i_mode) &&
 850             ext2fs_inode_has_valid_blocks2(ctx->fs, inode)) {
 851                 retval = rewrite_directory(ctx->fs, ino, inode);
 852                 if (retval)
 853                         fatal_err(retval, "while rewriting directories");
 854         }
 855
 856         file_acl_block = ext2fs_file_acl_block(ctx->fs, inode);
 857         if (!file_acl_block)
 858                 return;
 859
 860         retval = ext2fs_read_ext_attr3(ctx->fs, file_acl_block, ctx->ea_buf,
 861                                        ino);
 862         if (retval)
 863                 fatal_err(retval, "while rewriting extended attribute");
 864
 865         update_block_xattr_hashes(ctx, ctx->ea_buf);
 866         retval = ext2fs_write_ext_attr3(ctx->fs, file_acl_block, ctx->ea_buf,
 867                                         ino);
 868         if (retval)
 869                 fatal_err(retval, "while rewriting extended attribute");
 870 }
 871
 872 /*
 873  * Forcibly set checksums in all inodes.
 874  */
 875 static void rewrite_inodes(ext2_filsys fs)
 876 {
 877         ext2_inode_scan scan;
 878         errcode_t       retval;
 879         ext2_ino_t      ino;
 880         struct ext2_inode *inode;
 881         int pass;
 882         struct rewrite_context ctx = {
 883                 .fs = fs,
 884                 .inode_size = EXT2_INODE_SIZE(fs->super),
 885         };
 886
 887         if (fs->super->s_creator_os == EXT2_OS_HURD)
 888                 return;
 889
 890         retval = ext2fs_get_mem(ctx.inode_size, &inode);
 891         if (retval)
 892                 fatal_err(retval, "while allocating memory");
 893
 894         retval = ext2fs_get_memzero(ctx.inode_size, &ctx.zero_inode);
 895         if (retval)
 896                 fatal_err(retval, "while allocating memory");
 897
 898         retval = ext2fs_get_mem(64 * 1024, &ctx.ea_buf);
 899         if (retval)
 900                 fatal_err(retval, "while allocating memory");
 901
 902         /*
 903          * Extended attribute inodes have a lookup hash that needs to be
 904          * recalculated with the new csum_seed. Other inodes referencing xattr
 905          * inodes need this value to be up to date. That's why we do two passes:
 906          *
 907          * pass 1: update xattr inodes to update their lookup hash as well as
 908          *         other checksums.
 909          *
 910          * pass 2: go over other inodes to update their checksums.
 911          */
 912         if (ext2fs_has_feature_ea_inode(fs->super))
 913                 pass = 1;
 914         else
 915                 pass = 2;
 916         for (;pass <= 2; pass++) {
 917                 retval = ext2fs_open_inode_scan(fs, 0, &scan);
 918                 if (retval)
 919                         fatal_err(retval, "while opening inode scan");
 920
 921                 do {
 922                         retval = ext2fs_get_next_inode_full(scan, &ino, inode,
 923                                                             ctx.inode_size);
 924                         if (retval)
 925                                 fatal_err(retval, "while getting next inode");
 926                         if (!ino)
 927                                 break;
 928
 929                         if (((pass == 1) &&
 930                              (inode->i_flags & EXT4_EA_INODE_FL)) ||
 931                             ((pass == 2) &&
 932                              !(inode->i_flags & EXT4_EA_INODE_FL)))
 933                                 rewrite_one_inode(&ctx, ino, inode);
 934                 } while (ino);
 935
 936                 ext2fs_close_inode_scan(scan);
 937         }
 938
 939         ext2fs_free_mem(&ctx.zero_inode);
 940         ext2fs_free_mem(&ctx.ea_buf);
 941         ext2fs_free_mem(&inode);
 942 }
 943
 944 static void rewrite_metadata_checksums(ext2_filsys fs)
 945 {
 946         errcode_t retval;
 947         dgrp_t i;
 948
 949         fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
 950         ext2fs_init_csum_seed(fs);
 951         for (i = 0; i < fs->group_desc_count; i++)
 952                 ext2fs_group_desc_csum_set(fs, i);
 953         retval = ext2fs_read_bitmaps(fs);
 954         if (retval)
 955                 fatal_err(retval, "while reading bitmaps");
 956         rewrite_inodes(fs);
 957         ext2fs_mark_ib_dirty(fs);
 958         ext2fs_mark_bb_dirty(fs);
 959         ext2fs_mmp_update2(fs, 1);
 960         fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
 961         fs->flags &= ~EXT2_FLAG_IGNORE_CSUM_ERRORS;
 962         if (ext2fs_has_feature_metadata_csum(fs->super))
 963                 fs->super->s_checksum_type = EXT2_CRC32C_CHKSUM;
 964         else
 965                 fs->super->s_checksum_type = 0;
 966         ext2fs_mark_super_dirty(fs);
 967 }
 968
 969 static void enable_uninit_bg(ext2_filsys fs)
 970 {
 971         struct ext2_group_desc *gd;
 972         dgrp_t i;
 973
 974         for (i = 0; i < fs->group_desc_count; i++) {
 975                 gd = ext2fs_group_desc(fs, fs->group_desc, i);
 976                 gd->bg_itable_unused = 0;
 977                 gd->bg_flags = EXT2_BG_INODE_ZEROED;
 978                 ext2fs_group_desc_csum_set(fs, i);
 979         }
 980         fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
 981 }
 982
 983 static errcode_t zero_empty_inodes(ext2_filsys fs)
 984 {
 985         int length = EXT2_INODE_SIZE(fs->super);
 986         struct ext2_inode *inode = NULL;
 987         ext2_inode_scan scan;
 988         errcode_t       retval;
 989         ext2_ino_t      ino;
 990
 991         retval = ext2fs_open_inode_scan(fs, 0, &scan);
 992         if (retval)
 993                 goto out;
 994
 995         retval = ext2fs_get_mem(length, &inode);
 996         if (retval)
 997                 goto out;
 998
 999         do {
1000                 retval = ext2fs_get_next_inode_full(scan, &ino, inode, length);
1001                 if (retval)
1002                         goto out;
1003                 if (!ino)
1004                         break;
1005                 if (!ext2fs_test_inode_bitmap2(fs->inode_map, ino)) {
1006                         memset(inode, 0, length);
1007                         retval = ext2fs_write_inode_full(fs, ino, inode,
1008                                                          length);
1009                         if (retval)
1010                                 goto out;
1011                 }
1012         } while (1);
1013
1014 out:
1015         ext2fs_free_mem(&inode);
1016         ext2fs_close_inode_scan(scan);
1017         return retval;
1018 }
1019
1020 static errcode_t disable_uninit_bg(ext2_filsys fs, __u32 csum_feature_flag)
1021 {
1022         struct ext2_group_desc *gd;
1023         dgrp_t i;
1024         errcode_t retval;
1025         blk64_t b, c, d;
1026
1027         /* Load bitmaps to ensure that the uninit ones get written out */
1028         fs->super->s_feature_ro_compat |= csum_feature_flag;
1029         retval = ext2fs_read_bitmaps(fs);
1030         fs->super->s_feature_ro_compat &= ~csum_feature_flag;
1031         if (retval) {
1032                 com_err("disable_uninit_bg", retval,
1033                         "while reading bitmaps");
1034                 request_fsck_afterwards(fs);
1035                 return retval;
1036         }
1037         ext2fs_mark_ib_dirty(fs);
1038         ext2fs_mark_bb_dirty(fs);
1039
1040         /* If we're only turning off uninit_bg, zero the inodes */
1041         if (csum_feature_flag == EXT4_FEATURE_RO_COMPAT_GDT_CSUM) {
1042                 retval = zero_empty_inodes(fs);
1043                 if (retval) {
1044                         com_err("disable_uninit_bg", retval,
1045                                 "while zeroing unused inodes");
1046                         request_fsck_afterwards(fs);
1047                         return retval;
1048                 }
1049         }
1050
1051         /* The bbitmap is zeroed; we must mark group metadata blocks in use */
1052         for (i = 0; i < fs->group_desc_count; i++) {
1053                 b = ext2fs_block_bitmap_loc(fs, i);
1054                 ext2fs_mark_block_bitmap2(fs->block_map, b);
1055                 b = ext2fs_inode_bitmap_loc(fs, i);
1056                 ext2fs_mark_block_bitmap2(fs->block_map, b);
1057
1058                 retval = ext2fs_super_and_bgd_loc2(fs, i, &b, &c, &d, NULL);
1059                 if (retval == 0 && b)
1060                         ext2fs_mark_block_bitmap2(fs->block_map, b);
1061                 if (retval == 0 && c)
1062                         ext2fs_mark_block_bitmap2(fs->block_map, c);
1063                 if (retval == 0 && d)
1064                         ext2fs_mark_block_bitmap2(fs->block_map, d);
1065                 if (retval) {
1066                         com_err("disable_uninit_bg", retval,
1067                                 "while initializing block bitmaps");
1068                         request_fsck_afterwards(fs);
1069                 }
1070
1071                 gd = ext2fs_group_desc(fs, fs->group_desc, i);
1072                 gd->bg_itable_unused = 0;
1073                 gd->bg_flags = 0;
1074                 ext2fs_group_desc_csum_set(fs, i);
1075         }
1076         fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
1077         ext2fs_mark_super_dirty(fs);
1078
1079         return 0;
1080 }
1081
1082 static void
1083 try_confirm_csum_seed_support(void)
1084 {
1085         if (access("/sys/fs/ext4/features/metadata_csum_seed", R_OK))
1086                 fputs(_("WARNING: Could not confirm kernel support for "
1087                         "metadata_csum_seed.\n  This requires Linux >= "
1088                         "v4.4.\n"), stderr);
1089 }
1090
1091 /*
1092  * Update the feature set as provided by the user.
1093  */
1094 static int update_feature_set(ext2_filsys fs, char *features)
1095 {
1096         struct ext2_super_block *sb = fs->super;
1097         __u32           old_features[3];
1098         int             type_err;
1099         unsigned int    mask_err;
1100         errcode_t       err;
1101         enum quota_type qtype;
1102
1103 #define FEATURE_ON(type, mask) (!(old_features[(type)] & (mask)) && \
1104                                 ((&sb->s_feature_compat)[(type)] & (mask)))
1105 #define FEATURE_OFF(type, mask) ((old_features[(type)] & (mask)) && \
1106                                  !((&sb->s_feature_compat)[(type)] & (mask)))
1107 #define FEATURE_CHANGED(type, mask) ((mask) & \
1108                      (old_features[(type)] ^ (&sb->s_feature_compat)[(type)]))
1109
1110         old_features[E2P_FEATURE_COMPAT] = sb->s_feature_compat;
1111         old_features[E2P_FEATURE_INCOMPAT] = sb->s_feature_incompat;
1112         old_features[E2P_FEATURE_RO_INCOMPAT] = sb->s_feature_ro_compat;
1113
1114         if (e2p_edit_feature2(features, &sb->s_feature_compat,
1115                               ok_features, clear_ok_features,
1116                               &type_err, &mask_err)) {
1117                 if (!mask_err)
1118                         fprintf(stderr,
1119                                 _("Invalid filesystem option set: %s\n"),
1120                                 features);
1121                 else if (type_err & E2P_FEATURE_NEGATE_FLAG)
1122                         fprintf(stderr, _("Clearing filesystem feature '%s' "
1123                                           "not supported.\n"),
1124                                 e2p_feature2string(type_err &
1125                                                    E2P_FEATURE_TYPE_MASK,
1126                                                    mask_err));
1127                 else
1128                         fprintf(stderr, _("Setting filesystem feature '%s' "
1129                                           "not supported.\n"),
1130                                 e2p_feature2string(type_err, mask_err));
1131                 return 1;
1132         }
1133
1134         if (FEATURE_OFF(E2P_FEATURE_COMPAT, EXT3_FEATURE_COMPAT_HAS_JOURNAL)) {
1135                 if ((mount_flags & EXT2_MF_MOUNTED) &&
1136                     !(mount_flags & EXT2_MF_READONLY)) {
1137                         fputs(_("The has_journal feature may only be "
1138                                 "cleared when the filesystem is\n"
1139                                 "unmounted or mounted "
1140                                 "read-only.\n"), stderr);
1141                         return 1;
1142                 }
1143                 if (ext2fs_has_feature_journal_needs_recovery(sb) &&
1144                     f_flag < 2) {
1145                         fputs(_("The needs_recovery flag is set.  "
1146                                 "Please run e2fsck before clearing\n"
1147                                 "the has_journal flag.\n"), stderr);
1148                         return 1;
1149                 }
1150                 if (sb->s_journal_inum) {
1151                         if (remove_journal_inode(fs))
1152                                 return 1;
1153                 }
1154                 if (sb->s_journal_dev) {
1155                         if (remove_journal_device(fs))
1156                                 return 1;
1157                 }
1158         }
1159
1160         if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
1161                 EXT2_FEATURE_RO_COMPAT_SPARSE_SUPER)) {
1162                 if (ext2fs_has_feature_meta_bg(sb)) {
1163                         fputs(_("Setting filesystem feature 'sparse_super' "
1164                                 "not supported\nfor filesystems with "
1165                                 "the meta_bg feature enabled.\n"),
1166                                 stderr);
1167                         return 1;
1168                 }
1169         }
1170
1171         if (FEATURE_ON(E2P_FEATURE_INCOMPAT, EXT4_FEATURE_INCOMPAT_MMP)) {
1172                 int error;
1173
1174                 if ((mount_flags & EXT2_MF_MOUNTED) ||
1175                     (mount_flags & EXT2_MF_READONLY)) {
1176                         fputs(_("The multiple mount protection feature can't\n"
1177                                 "be set if the filesystem is mounted or\n"
1178                                 "read-only.\n"), stderr);
1179                         return 1;
1180                 }
1181
1182                 error = ext2fs_mmp_init(fs);
1183                 if (error) {
1184                         fputs(_("\nError while enabling multiple mount "
1185                                 "protection feature."), stderr);
1186                         return 1;
1187                 }
1188
1189                 /*
1190                  * We want to update group desc with the new free blocks count
1191                  */
1192                 fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
1193
1194                 printf(_("Multiple mount protection has been enabled "
1195                          "with update interval %ds.\n"),
1196                        sb->s_mmp_update_interval);
1197         }
1198
1199         if (FEATURE_OFF(E2P_FEATURE_INCOMPAT, EXT4_FEATURE_INCOMPAT_MMP)) {
1200                 int error;
1201
1202                 if (mount_flags & EXT2_MF_READONLY) {
1203                         fputs(_("The multiple mount protection feature cannot\n"
1204                                 "be disabled if the filesystem is readonly.\n"),
1205                                 stderr);
1206                         return 1;
1207                 }
1208
1209                 error = ext2fs_read_bitmaps(fs);
1210                 if (error) {
1211                         fputs(_("Error while reading bitmaps\n"), stderr);
1212                         return 1;
1213                 }
1214
1215                 error = ext2fs_mmp_read(fs, sb->s_mmp_block, NULL);
1216                 if (error) {
1217                         struct mmp_struct *mmp_cmp = fs->mmp_cmp;
1218
1219                         if (error == EXT2_ET_MMP_MAGIC_INVALID)
1220                                 printf(_("Magic number in MMP block does not "
1221                                          "match. expected: %x, actual: %x\n"),
1222                                          EXT4_MMP_MAGIC, mmp_cmp->mmp_magic);
1223                         else
1224                                 com_err(program_name, error, "%s",
1225                                         _("while reading MMP block."));
1226                         goto mmp_error;
1227                 }
1228
1229                 /* We need to force out the group descriptors as well */
1230                 fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
1231                 ext2fs_block_alloc_stats2(fs, sb->s_mmp_block, -1);
1232 mmp_error:
1233                 sb->s_mmp_block = 0;
1234                 sb->s_mmp_update_interval = 0;
1235         }
1236
1237         if (FEATURE_ON(E2P_FEATURE_COMPAT, EXT3_FEATURE_COMPAT_HAS_JOURNAL)) {
1238                 /*
1239                  * If adding a journal flag, let the create journal
1240                  * code below handle setting the flag and creating the
1241                  * journal.  We supply a default size if necessary.
1242                  */
1243                 if (!journal_size)
1244                         journal_size = -1;
1245                 ext2fs_clear_feature_journal(sb);
1246         }
1247
1248         if (FEATURE_ON(E2P_FEATURE_COMPAT, EXT2_FEATURE_COMPAT_DIR_INDEX)) {
1249                 if (!sb->s_def_hash_version)
1250                         sb->s_def_hash_version = EXT2_HASH_HALF_MD4;
1251                 if (uuid_is_null((unsigned char *) sb->s_hash_seed))
1252                         uuid_generate((unsigned char *) sb->s_hash_seed);
1253         }
1254
1255         if (FEATURE_OFF(E2P_FEATURE_INCOMPAT, EXT4_FEATURE_INCOMPAT_FLEX_BG)) {
1256                 if (ext2fs_check_desc(fs)) {
1257                         fputs(_("Clearing the flex_bg flag would "
1258                                 "cause the the filesystem to be\n"
1259                                 "inconsistent.\n"), stderr);
1260                         return 1;
1261                 }
1262         }
1263
1264         if (FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
1265                             EXT4_FEATURE_RO_COMPAT_HUGE_FILE)) {
1266                 if ((mount_flags & EXT2_MF_MOUNTED) &&
1267                     !(mount_flags & EXT2_MF_READONLY)) {
1268                         fputs(_("The huge_file feature may only be "
1269                                 "cleared when the filesystem is\n"
1270                                 "unmounted or mounted "
1271                                 "read-only.\n"), stderr);
1272                         return 1;
1273                 }
1274         }
1275
1276         if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
1277                        EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)) {
1278                 check_fsck_needed(fs,
1279                         _("Enabling checksums could take some time."));
1280                 if (mount_flags & EXT2_MF_MOUNTED) {
1281                         fputs(_("Cannot enable metadata_csum on a mounted "
1282                                 "filesystem!\n"), stderr);
1283                         exit(1);
1284                 }
1285                 if (!ext2fs_has_feature_extents(fs->super))
1286                         printf("%s",
1287                                _("Extents are not enabled.  The file extent "
1288                                  "tree can be checksummed, whereas block maps "
1289                                  "cannot.  Not enabling extents reduces the "
1290                                  "coverage of metadata checksumming.  "
1291                                  "Re-run with -O extent to rectify.\n"));
1292                 if (!ext2fs_has_feature_64bit(fs->super))
1293                         printf("%s",
1294                                _("64-bit filesystem support is not enabled.  "
1295                                  "The larger fields afforded by this feature "
1296                                  "enable full-strength checksumming.  "
1297                                  "Run resize2fs -b to rectify.\n"));
1298                 rewrite_checksums = 1;
1299                 /* metadata_csum supersedes uninit_bg */
1300                 ext2fs_clear_feature_gdt_csum(fs->super);
1301
1302                 /* if uninit_bg was previously off, rewrite group desc */
1303                 if (!(old_features[E2P_FEATURE_RO_INCOMPAT] &
1304                       EXT4_FEATURE_RO_COMPAT_GDT_CSUM))
1305                         enable_uninit_bg(fs);
1306
1307                 /*
1308                  * Since metadata_csum supersedes uninit_bg, pretend like
1309                  * uninit_bg has been off all along.
1310                  */
1311                 old_features[E2P_FEATURE_RO_INCOMPAT] &=
1312                         ~EXT4_FEATURE_RO_COMPAT_GDT_CSUM;
1313         }
1314
1315         if (FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
1316                         EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)) {
1317                 __u32   test_features[3];
1318
1319                 check_fsck_needed(fs,
1320                         _("Disabling checksums could take some time."));
1321                 if (mount_flags & EXT2_MF_MOUNTED) {
1322                         fputs(_("Cannot disable metadata_csum on a mounted "
1323                                 "filesystem!\n"), stderr);
1324                         exit(1);
1325                 }
1326                 rewrite_checksums = 1;
1327
1328                 /* Enable uninit_bg unless the user expressly turned it off */
1329                 memcpy(test_features, old_features, sizeof(test_features));
1330                 test_features[E2P_FEATURE_RO_INCOMPAT] |=
1331                                                 EXT4_FEATURE_RO_COMPAT_GDT_CSUM;
1332                 e2p_edit_feature2(features, test_features, ok_features,
1333                                   clear_ok_features, NULL, NULL);
1334                 if (test_features[E2P_FEATURE_RO_INCOMPAT] &
1335                                                 EXT4_FEATURE_RO_COMPAT_GDT_CSUM)
1336                         ext2fs_set_feature_gdt_csum(fs->super);
1337
1338                 /*
1339                  * If we're turning off metadata_csum and not turning on
1340                  * uninit_bg, rewrite group desc.
1341                  */
1342                 if (!ext2fs_has_feature_gdt_csum(fs->super)) {
1343                         err = disable_uninit_bg(fs,
1344                                         EXT4_FEATURE_RO_COMPAT_METADATA_CSUM);
1345                         if (err)
1346                                 return 1;
1347                 } else
1348                         /*
1349                          * metadata_csum previously provided uninit_bg, so if
1350                          * we're also setting the uninit_bg feature bit,
1351                          * pretend like it was previously enabled.  Checksums
1352                          * will be rewritten with crc16 later.
1353                          */
1354                         old_features[E2P_FEATURE_RO_INCOMPAT] |=
1355                                 EXT4_FEATURE_RO_COMPAT_GDT_CSUM;
1356                 fs->super->s_checksum_seed = 0;
1357                 ext2fs_clear_feature_csum_seed(fs->super);
1358         }
1359
1360         if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
1361                        EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) {
1362                 /* Do not enable uninit_bg when metadata_csum enabled */
1363                 if (ext2fs_has_feature_metadata_csum(fs->super))
1364                         ext2fs_clear_feature_gdt_csum(fs->super);
1365                 else
1366                         enable_uninit_bg(fs);
1367         }
1368
1369         if (FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
1370                         EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) {
1371                 err = disable_uninit_bg(fs,
1372                                 EXT4_FEATURE_RO_COMPAT_GDT_CSUM);
1373                 if (err)
1374                         return 1;
1375         }
1376
1377         /*
1378          * We don't actually toggle 64bit; resize2fs does that.  But this
1379          * must come after the metadata_csum feature_on so that it won't
1380          * complain about the lack of 64bit.
1381          */
1382         if (FEATURE_ON(E2P_FEATURE_INCOMPAT,
1383                        EXT4_FEATURE_INCOMPAT_64BIT)) {
1384                 if (mount_flags & EXT2_MF_MOUNTED) {
1385                         fprintf(stderr, _("Cannot enable 64-bit mode "
1386                                           "while mounted!\n"));
1387                         exit(1);
1388                 }
1389                 ext2fs_clear_feature_64bit(sb);
1390                 feature_64bit = 1;
1391         }
1392         if (FEATURE_OFF(E2P_FEATURE_INCOMPAT,
1393                         EXT4_FEATURE_INCOMPAT_64BIT)) {
1394                 if (mount_flags & EXT2_MF_MOUNTED) {
1395                         fprintf(stderr, _("Cannot disable 64-bit mode "
1396                                           "while mounted!\n"));
1397                         exit(1);
1398                 }
1399                 ext2fs_set_feature_64bit(sb);
1400                 feature_64bit = -1;
1401         }
1402
1403         if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
1404                                 EXT4_FEATURE_RO_COMPAT_QUOTA)) {
1405                 /*
1406                  * Set the Q_flag here and handle the quota options in the code
1407                  * below.
1408                  */
1409                 if (!Q_flag) {
1410                         Q_flag = 1;
1411                         /* Enable usr/grp quota by default */
1412                         for (qtype = 0; qtype < MAXQUOTAS; qtype++) {
1413                                 if (qtype != PRJQUOTA)
1414                                         quota_enable[qtype] = QOPT_ENABLE;
1415                                 else
1416                                         quota_enable[qtype] = QOPT_DISABLE;
1417                         }
1418                 }
1419                 ext2fs_clear_feature_quota(sb);
1420         }
1421
1422         if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
1423                        EXT4_FEATURE_RO_COMPAT_PROJECT)) {
1424                 if (fs->super->s_inode_size == EXT2_GOOD_OLD_INODE_SIZE) {
1425                         fprintf(stderr, _("Cannot enable project feature; "
1426                                           "inode size too small.\n"));
1427                         exit(1);
1428                 }
1429                 Q_flag = 1;
1430                 quota_enable[PRJQUOTA] = QOPT_ENABLE;
1431         }
1432
1433         if (FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
1434                         EXT4_FEATURE_RO_COMPAT_PROJECT)) {
1435                 Q_flag = 1;
1436                 quota_enable[PRJQUOTA] = QOPT_DISABLE;
1437         }
1438
1439         if (FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
1440                                 EXT4_FEATURE_RO_COMPAT_QUOTA)) {
1441                 /*
1442                  * Set the Q_flag here and handle the quota options in the code
1443                  * below.
1444                  */
1445                 if (Q_flag)
1446                         fputs(_("\nWarning: '^quota' option overrides '-Q'"
1447                                 "arguments.\n"), stderr);
1448                 Q_flag = 1;
1449                 /* Disable all quota by default */
1450                 for (qtype = 0; qtype < MAXQUOTAS; qtype++)
1451                         quota_enable[qtype] = QOPT_DISABLE;
1452         }
1453
1454         if (FEATURE_ON(E2P_FEATURE_INCOMPAT, EXT4_FEATURE_INCOMPAT_ENCRYPT)) {
1455                 fs->super->s_encrypt_algos[0] =
1456                         EXT4_ENCRYPTION_MODE_AES_256_XTS;
1457                 fs->super->s_encrypt_algos[1] =
1458                         EXT4_ENCRYPTION_MODE_AES_256_CTS;
1459         }
1460
1461         if (FEATURE_ON(E2P_FEATURE_INCOMPAT,
1462                 EXT4_FEATURE_INCOMPAT_CSUM_SEED)) {
1463                 if (!ext2fs_has_feature_metadata_csum(sb)) {
1464                         fputs(_("Setting feature 'metadata_csum_seed' "
1465                                 "is only supported\non filesystems with "
1466                                 "the metadata_csum feature enabled.\n"),
1467                                 stderr);
1468                         return 1;
1469                 }
1470                 try_confirm_csum_seed_support();
1471                 fs->super->s_checksum_seed = fs->csum_seed;
1472         }
1473
1474         if (FEATURE_OFF(E2P_FEATURE_INCOMPAT,
1475                 EXT4_FEATURE_INCOMPAT_CSUM_SEED)) {
1476                 __le32 uuid_seed;
1477
1478                 uuid_seed = ext2fs_crc32c_le(~0, fs->super->s_uuid,
1479                                         sizeof(fs->super->s_uuid));
1480                 if (fs->super->s_checksum_seed != uuid_seed) {
1481                         if (mount_flags & (EXT2_MF_BUSY|EXT2_MF_MOUNTED)) {
1482                                 fputs(_("UUID has changed since enabling "
1483                 "metadata_csum.  Filesystem must be unmounted "
1484                 "\nto safely rewrite all metadata to match the new UUID.\n"),
1485                                       stderr);
1486                                 return 1;
1487                         }
1488                         check_fsck_needed(fs, _("Recalculating checksums "
1489                                                 "could take some time."));
1490                         rewrite_checksums = 1;
1491                 }
1492         }
1493
1494         if (sb->s_rev_level == EXT2_GOOD_OLD_REV &&
1495             (sb->s_feature_compat || sb->s_feature_ro_compat ||
1496              sb->s_feature_incompat))
1497                 ext2fs_update_dynamic_rev(fs);
1498
1499         if (FEATURE_CHANGED(E2P_FEATURE_RO_INCOMPAT,
1500                             EXT2_FEATURE_RO_COMPAT_SPARSE_SUPER) ||
1501             FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
1502                         EXT4_FEATURE_RO_COMPAT_HUGE_FILE) ||
1503             FEATURE_CHANGED(E2P_FEATURE_INCOMPAT,
1504                             EXT2_FEATURE_INCOMPAT_FILETYPE) ||
1505             FEATURE_CHANGED(E2P_FEATURE_COMPAT,
1506                             EXT2_FEATURE_COMPAT_RESIZE_INODE) ||
1507             FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
1508                         EXT2_FEATURE_RO_COMPAT_LARGE_FILE))
1509                 request_fsck_afterwards(fs);
1510
1511         if ((old_features[E2P_FEATURE_COMPAT] != sb->s_feature_compat) ||
1512             (old_features[E2P_FEATURE_INCOMPAT] != sb->s_feature_incompat) ||
1513             (old_features[E2P_FEATURE_RO_INCOMPAT] != sb->s_feature_ro_compat))
1514                 ext2fs_mark_super_dirty(fs);
1515
1516         return 0;
1517 }
1518
1519 /*
1520  * Add a journal to the filesystem.
1521  */
1522 static int add_journal(ext2_filsys fs)
1523 {
1524         unsigned long journal_blocks;
1525         errcode_t       retval;
1526         ext2_filsys     jfs;
1527         io_manager      io_ptr;
1528
1529         if (ext2fs_has_feature_journal(fs->super)) {
1530                 fputs(_("The filesystem already has a journal.\n"), stderr);
1531                 goto err;
1532         }
1533         if (journal_device) {
1534                 if (!check_plausibility(journal_device, CHECK_BLOCK_DEV,
1535                                         NULL))
1536                         proceed_question(-1);
1537                 check_mount(journal_device, 0, _("journal"));
1538 #ifdef CONFIG_TESTIO_DEBUG
1539                 if (getenv("TEST_IO_FLAGS") || getenv("TEST_IO_BLOCK")) {
1540                         io_ptr = test_io_manager;
1541                         test_io_backing_manager = unix_io_manager;
1542                 } else
1543 #endif
1544                         io_ptr = unix_io_manager;
1545                 retval = ext2fs_open(journal_device, EXT2_FLAG_RW|
1546                                      EXT2_FLAG_JOURNAL_DEV_OK, 0,
1547                                      fs->blocksize, io_ptr, &jfs);
1548                 if (retval) {
1549                         com_err(program_name, retval,
1550                                 _("\n\twhile trying to open journal on %s\n"),
1551                                 journal_device);
1552                         goto err;
1553                 }
1554                 printf(_("Creating journal on device %s: "),
1555                        journal_device);
1556                 fflush(stdout);
1557
1558                 retval = ext2fs_add_journal_device(fs, jfs);
1559                 ext2fs_close_free(&jfs);
1560                 if (retval) {
1561                         com_err(program_name, retval,
1562                                 _("while adding filesystem to journal on %s"),
1563                                 journal_device);
1564                         goto err;
1565                 }
1566                 fputs(_("done\n"), stdout);
1567         } else if (journal_size) {
1568                 fputs(_("Creating journal inode: "), stdout);
1569                 fflush(stdout);
1570                 journal_blocks = figure_journal_size(journal_size, fs);
1571
1572                 if (journal_location_string)
1573                         journal_location =
1574                                 parse_num_blocks2(journal_location_string,
1575                                                   fs->super->s_log_block_size);
1576                 retval = ext2fs_add_journal_inode2(fs, journal_blocks,
1577                                                    journal_location,
1578                                                    journal_flags);
1579                 if (retval) {
1580                         fprintf(stderr, "\n");
1581                         com_err(program_name, retval, "%s",
1582                                 _("\n\twhile trying to create journal file"));
1583                         return retval;
1584                 } else
1585                         fputs(_("done\n"), stdout);
1586                 /*
1587                  * If the filesystem wasn't mounted, we need to force
1588                  * the block group descriptors out.
1589                  */
1590                 if ((mount_flags & EXT2_MF_MOUNTED) == 0)
1591                         fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
1592         }
1593         print_check_message(fs->super->s_max_mnt_count,
1594                             fs->super->s_checkinterval);
1595         return 0;
1596
1597 err:
1598         free(journal_device);
1599         return 1;
1600 }
1601
1602 static void handle_quota_options(ext2_filsys fs)
1603 {
1604         errcode_t retval;
1605         quota_ctx_t qctx;
1606         ext2_ino_t qf_ino;
1607         enum quota_type qtype;
1608         unsigned int qtype_bits = 0;
1609         int need_dirty = 0;
1610
1611         for (qtype = 0 ; qtype < MAXQUOTAS; qtype++)
1612                 if (quota_enable[qtype] != 0)
1613                         break;
1614         if (qtype == MAXQUOTAS)
1615                 /* Nothing to do. */
1616                 return;
1617
1618         if (quota_enable[PRJQUOTA] == QOPT_ENABLE &&
1619             fs->super->s_inode_size == EXT2_GOOD_OLD_INODE_SIZE) {
1620                 fprintf(stderr, _("Cannot enable project quota; "
1621                                   "inode size too small.\n"));
1622                 exit(1);
1623         }
1624
1625         for (qtype = 0; qtype < MAXQUOTAS; qtype++) {
1626                 if (quota_enable[qtype] == QOPT_ENABLE)
1627                         qtype_bits |= 1 << qtype;
1628         }
1629
1630         retval = quota_init_context(&qctx, fs, qtype_bits);
1631         if (retval) {
1632                 com_err(program_name, retval,
1633                         _("while initializing quota context in support library"));
1634                 exit(1);
1635         }
1636
1637         if (qtype_bits)
1638                 quota_compute_usage(qctx);
1639
1640         for (qtype = 0 ; qtype < MAXQUOTAS; qtype++) {
1641                 if (quota_enable[qtype] == QOPT_ENABLE &&
1642                     *quota_sb_inump(fs->super, qtype) == 0) {
1643                         if ((qf_ino = quota_file_exists(fs, qtype)) > 0) {
1644                                 retval = quota_update_limits(qctx, qf_ino,
1645                                                              qtype);
1646                                 if (retval) {
1647                                         com_err(program_name, retval,
1648                                                 _("while updating quota limits (%d)"),
1649                                                 qtype);
1650                                         exit(1);
1651                                 }
1652                         }
1653                         retval = quota_write_inode(qctx, 1 << qtype);
1654                         if (retval) {
1655                                 com_err(program_name, retval,
1656                                         _("while writing quota file (%d)"),
1657                                         qtype);
1658                                 exit(1);
1659                         }
1660                         /* Enable Quota feature if one of quota enabled */
1661                         if (!ext2fs_has_feature_quota(fs->super)) {
1662                                 ext2fs_set_feature_quota(fs->super);
1663                                 need_dirty = 1;
1664                         }
1665                         if (qtype == PRJQUOTA &&
1666                             !ext2fs_has_feature_project(fs->super)) {
1667                                 ext2fs_set_feature_project(fs->super);
1668                                 need_dirty = 1;
1669                         }
1670                 } else if (quota_enable[qtype] == QOPT_DISABLE) {
1671                         retval = quota_remove_inode(fs, qtype);
1672                         if (retval) {
1673                                 com_err(program_name, retval,
1674                                         _("while removing quota file (%d)"),
1675                                         qtype);
1676                                 exit(1);
1677                         }
1678                         if (qtype == PRJQUOTA) {
1679                                 ext2fs_clear_feature_project(fs->super);
1680                                 need_dirty = 1;
1681                         }
1682                 }
1683         }
1684
1685         quota_release_context(&qctx);
1686         /* Clear Quota feature if all quota types disabled. */
1687         if (!qtype_bits) {
1688                 for (qtype = 0 ; qtype < MAXQUOTAS; qtype++)
1689                         if (*quota_sb_inump(fs->super, qtype))
1690                                 break;
1691                 if (qtype == MAXQUOTAS) {
1692                         ext2fs_clear_feature_quota(fs->super);
1693                         need_dirty = 1;
1694                 }
1695
1696         }
1697         if (need_dirty)
1698                 ext2fs_mark_super_dirty(fs);
1699         return;
1700 }
1701
1702 static int option_handle_function(char *token)
1703 {
1704         if (strncmp(token, "usr", 3) == 0) {
1705                 quota_enable[USRQUOTA] = QOPT_ENABLE;
1706         } else if (strncmp(token, "^usr", 4) == 0) {
1707                 quota_enable[USRQUOTA] = QOPT_DISABLE;
1708         } else if (strncmp(token, "grp", 3) == 0) {
1709                 quota_enable[GRPQUOTA] = QOPT_ENABLE;
1710         } else if (strncmp(token, "^grp", 4) == 0) {
1711                 quota_enable[GRPQUOTA] = QOPT_DISABLE;
1712         } else if (strncmp(token, "prj", 3) == 0) {
1713                 quota_enable[PRJQUOTA] = QOPT_ENABLE;
1714         } else if (strncmp(token, "^prj", 4) == 0) {
1715                 quota_enable[PRJQUOTA] = QOPT_DISABLE;
1716         } else {
1717                 fputs(_("\nBad quota options specified.\n\n"
1718                         "Following valid quota options are available "
1719                         "(pass by separating with comma):\n"
1720                         "\t[^]usr[quota]\n"
1721                         "\t[^]grp[quota]\n"
1722                         "\t[^]prj[quota]\n"
1723                         "\n\n"), stderr);
1724                 return 1;
1725         }
1726         return 0;
1727 }
1728
1729 static void parse_e2label_options(int argc, char ** argv)
1730 {
1731         if ((argc < 2) || (argc > 3)) {
1732                 fputs(_("Usage: e2label device [newlabel]\n"), stderr);
1733                 exit(1);
1734         }
1735         io_options = strchr(argv[1], '?');
1736         if (io_options)
1737                 *io_options++ = 0;
1738         device_name = blkid_get_devname(NULL, argv[1], NULL);
1739         if (!device_name) {
1740                 com_err("e2label", 0, _("Unable to resolve '%s'"),
1741                         argv[1]);
1742                 exit(1);
1743         }
1744         open_flag = EXT2_FLAG_JOURNAL_DEV_OK;
1745         if (argc == 3) {
1746                 open_flag |= EXT2_FLAG_RW;
1747                 L_flag = 1;
1748                 new_label = argv[2];
1749         } else
1750                 print_label++;
1751 }
1752
1753 static time_t parse_time(char *str)
1754 {
1755         struct  tm      ts;
1756
1757         if (strcmp(str, "now") == 0) {
1758                 return (time(0));
1759         }
1760         memset(&ts, 0, sizeof(ts));
1761 #ifdef HAVE_STRPTIME
1762         strptime(str, "%Y%m%d%H%M%S", &ts);
1763 #else
1764         sscanf(str, "%4d%2d%2d%2d%2d%2d", &ts.tm_year, &ts.tm_mon,
1765                &ts.tm_mday, &ts.tm_hour, &ts.tm_min, &ts.tm_sec);
1766         ts.tm_year -= 1900;
1767         ts.tm_mon -= 1;
1768         if (ts.tm_year < 0 || ts.tm_mon < 0 || ts.tm_mon > 11 ||
1769             ts.tm_mday < 0 || ts.tm_mday > 31 || ts.tm_hour > 23 ||
1770             ts.tm_min > 59 || ts.tm_sec > 61)
1771                 ts.tm_mday = 0;
1772 #endif
1773         if (ts.tm_mday == 0) {
1774                 com_err(program_name, 0,
1775                         _("Couldn't parse date/time specifier: %s"),
1776                         str);
1777                 usage();
1778         }
1779         ts.tm_isdst = -1;
1780         return (mktime(&ts));
1781 }
1782
1783 static void parse_tune2fs_options(int argc, char **argv)
1784 {
1785         int c;
1786         char *tmp;
1787         struct group *gr;
1788         struct passwd *pw;
1789         int ret;
1790         char optstring[100] = "c:e:fg:i:jlm:o:r:s:u:C:E:I:J:L:M:O:T:U:z:Q:";
1791
1792         open_flag = 0;
1793         printf("tune2fs %s (%s)\n", E2FSPROGS_VERSION, E2FSPROGS_DATE);
1794         while ((c = getopt(argc, argv, optstring)) != EOF)
1795                 switch (c) {
1796                 case 'c':
1797                         max_mount_count = strtol(optarg, &tmp, 0);
1798                         if (*tmp || max_mount_count > 16000) {
1799                                 com_err(program_name, 0,
1800                                         _("bad mounts count - %s"),
1801                                         optarg);
1802                                 usage();
1803                         }
1804                         if (max_mount_count == 0)
1805                                 max_mount_count = -1;
1806                         c_flag = 1;
1807                         open_flag = EXT2_FLAG_RW;
1808                         break;
1809                 case 'C':
1810                         mount_count = strtoul(optarg, &tmp, 0);
1811                         if (*tmp || mount_count > 16000) {
1812                                 com_err(program_name, 0,
1813                                         _("bad mounts count - %s"),
1814                                         optarg);
1815                                 usage();
1816                         }
1817                         C_flag = 1;
1818                         open_flag = EXT2_FLAG_RW;
1819                         break;
1820                 case 'e':
1821                         if (strcmp(optarg, "continue") == 0)
1822                                 errors = EXT2_ERRORS_CONTINUE;
1823                         else if (strcmp(optarg, "remount-ro") == 0)
1824                                 errors = EXT2_ERRORS_RO;
1825                         else if (strcmp(optarg, "panic") == 0)
1826                                 errors = EXT2_ERRORS_PANIC;
1827                         else {
1828                                 com_err(program_name, 0,
1829                                         _("bad error behavior - %s"),
1830                                         optarg);
1831                                 usage();
1832                         }
1833                         e_flag = 1;
1834                         open_flag = EXT2_FLAG_RW;
1835                         break;
1836                 case 'E':
1837                         extended_cmd = optarg;
1838                         open_flag |= EXT2_FLAG_RW;
1839                         break;
1840                 case 'f': /* Force */
1841                         f_flag++;
1842                         break;
1843                 case 'g':
1844                         resgid = strtoul(optarg, &tmp, 0);
1845                         if (*tmp) {
1846                                 gr = getgrnam(optarg);
1847                                 if (gr == NULL)
1848                                         tmp = optarg;
1849                                 else {
1850                                         resgid = gr->gr_gid;
1851                                         *tmp = 0;
1852                                 }
1853                         }
1854                         if (*tmp) {
1855                                 com_err(program_name, 0,
1856                                         _("bad gid/group name - %s"),
1857                                         optarg);
1858                                 usage();
1859                         }
1860                         g_flag = 1;
1861                         open_flag = EXT2_FLAG_RW;
1862                         break;
1863                 case 'i':
1864                         interval = strtoul(optarg, &tmp, 0);
1865                         switch (*tmp) {
1866                         case 's':
1867                                 tmp++;
1868                                 break;
1869                         case '\0':
1870                         case 'd':
1871                         case 'D': /* days */
1872                                 interval *= 86400;
1873                                 if (*tmp != '\0')
1874                                         tmp++;
1875                                 break;
1876                         case 'm':
1877                         case 'M': /* months! */
1878                                 interval *= 86400 * 30;
1879                                 tmp++;
1880                                 break;
1881                         case 'w':
1882                         case 'W': /* weeks */
1883                                 interval *= 86400 * 7;
1884                                 tmp++;
1885                                 break;
1886                         }
1887                         if (*tmp) {
1888                                 com_err(program_name, 0,
1889                                         _("bad interval - %s"), optarg);
1890                                 usage();
1891                         }
1892                         i_flag = 1;
1893                         open_flag = EXT2_FLAG_RW;
1894                         break;
1895                 case 'j':
1896                         if (!journal_size)
1897                                 journal_size = -1;
1898                         open_flag = EXT2_FLAG_RW;
1899                         break;
1900                 case 'J':
1901                         parse_journal_opts(optarg);
1902                         open_flag = EXT2_FLAG_RW;
1903                         break;
1904                 case 'l':
1905                         l_flag = 1;
1906                         break;
1907                 case 'L':
1908                         new_label = optarg;
1909                         L_flag = 1;
1910                         open_flag |= EXT2_FLAG_RW |
1911                                 EXT2_FLAG_JOURNAL_DEV_OK;
1912                         break;
1913                 case 'm':
1914                         reserved_ratio = strtod(optarg, &tmp);
1915                         if (*tmp || reserved_ratio > 50 ||
1916                             reserved_ratio < 0) {
1917                                 com_err(program_name, 0,
1918                                         _("bad reserved block ratio - %s"),
1919                                         optarg);
1920                                 usage();
1921                         }
1922                         m_flag = 1;
1923                         open_flag = EXT2_FLAG_RW;
1924                         break;
1925                 case 'M':
1926                         new_last_mounted = optarg;
1927                         M_flag = 1;
1928                         open_flag = EXT2_FLAG_RW;
1929                         break;
1930                 case 'o':
1931                         if (mntopts_cmd) {
1932                                 com_err(program_name, 0, "%s",
1933                                         _("-o may only be specified once"));
1934                                 usage();
1935                         }
1936                         mntopts_cmd = optarg;
1937                         open_flag = EXT2_FLAG_RW;
1938                         break;
1939                 case 'O':
1940                         if (features_cmd) {
1941                                 com_err(program_name, 0, "%s",
1942                                         _("-O may only be specified once"));
1943                                 usage();
1944                         }
1945                         features_cmd = optarg;
1946                         open_flag = EXT2_FLAG_RW;
1947                         break;
1948                 case 'Q':
1949                         Q_flag = 1;
1950                         ret = parse_quota_opts(optarg, option_handle_function);
1951                         if (ret)
1952                                 exit(1);
1953                         open_flag = EXT2_FLAG_RW;
1954                         break;
1955                 case 'r':
1956                         reserved_blocks = strtoul(optarg, &tmp, 0);
1957                         if (*tmp) {
1958                                 com_err(program_name, 0,
1959                                         _("bad reserved blocks count - %s"),
1960                                         optarg);
1961                                 usage();
1962                         }
1963                         r_flag = 1;
1964                         open_flag = EXT2_FLAG_RW;
1965                         break;
1966                 case 's': /* Deprecated */
1967                         s_flag = atoi(optarg);
1968                         open_flag = EXT2_FLAG_RW;
1969                         break;
1970                 case 'T':
1971                         T_flag = 1;
1972                         last_check_time = parse_time(optarg);
1973                         open_flag = EXT2_FLAG_RW;
1974                         break;
1975                 case 'u':
1976                                 resuid = strtoul(optarg, &tmp, 0);
1977                                 if (*tmp) {
1978                                         pw = getpwnam(optarg);
1979                                         if (pw == NULL)
1980                                                 tmp = optarg;
1981                                         else {
1982                                                 resuid = pw->pw_uid;
1983                                                 *tmp = 0;
1984                                         }
1985                                 }
1986                                 if (*tmp) {
1987                                         com_err(program_name, 0,
1988                                                 _("bad uid/user name - %s"),
1989                                                 optarg);
1990                                         usage();
1991                                 }
1992                                 u_flag = 1;
1993                                 open_flag = EXT2_FLAG_RW;
1994                                 break;
1995                 case 'U':
1996                         new_UUID = optarg;
1997                         U_flag = 1;
1998                         open_flag = EXT2_FLAG_RW |
1999                                 EXT2_FLAG_JOURNAL_DEV_OK;
2000                         break;
2001                 case 'I':
2002                         new_inode_size = strtoul(optarg, &tmp, 0);
2003                         if (*tmp) {
2004                                 com_err(program_name, 0,
2005                                         _("bad inode size - %s"),
2006                                         optarg);
2007                                 usage();
2008                         }
2009                         if (!((new_inode_size &
2010                                (new_inode_size - 1)) == 0)) {
2011                                 com_err(program_name, 0,
2012                                         _("Inode size must be a "
2013                                           "power of two- %s"),
2014                                         optarg);
2015                                 usage();
2016                         }
2017                         open_flag = EXT2_FLAG_RW;
2018                         I_flag = 1;
2019                         break;
2020                 case 'z':
2021                         undo_file = optarg;
2022                         break;
2023                 default:
2024                         usage();
2025                 }
2026         if (optind < argc - 1 || optind == argc)
2027                 usage();
2028         if (!open_flag && !l_flag)
2029                 usage();
2030         io_options = strchr(argv[optind], '?');
2031         if (io_options)
2032                 *io_options++ = 0;
2033         device_name = blkid_get_devname(NULL, argv[optind], NULL);
2034         if (!device_name) {
2035                 com_err(program_name, 0, _("Unable to resolve '%s'"),
2036                         argv[optind]);
2037                 exit(1);
2038         }
2039 }
2040
2041 #ifdef CONFIG_BUILD_FINDFS
2042 void do_findfs(int argc, char **argv)
2043 {
2044         char    *dev;
2045
2046         if ((argc != 2) ||
2047             (strncmp(argv[1], "LABEL=", 6) && strncmp(argv[1], "UUID=", 5))) {
2048                 fprintf(stderr, "Usage: findfs LABEL=<label>|UUID=<uuid>\n");
2049                 exit(2);
2050         }
2051         dev = blkid_get_devname(NULL, argv[1], NULL);
2052         if (!dev) {
2053                 com_err("findfs", 0, _("Unable to resolve '%s'"),
2054                         argv[1]);
2055                 exit(1);
2056         }
2057         puts(dev);
2058         exit(0);
2059 }
2060 #endif
2061
2062 static int parse_extended_opts(ext2_filsys fs, const char *opts)
2063 {
2064         char    *buf, *token, *next, *p, *arg;
2065         int     len, hash_alg;
2066         int     r_usage = 0;
2067
2068         len = strlen(opts);
2069         buf = malloc(len+1);
2070         if (!buf) {
2071                 fprintf(stderr, "%s",
2072                         _("Couldn't allocate memory to parse options!\n"));
2073                 return 1;
2074         }
2075         strcpy(buf, opts);
2076         for (token = buf; token && *token; token = next) {
2077                 p = strchr(token, ',');
2078                 next = 0;
2079                 if (p) {
2080                         *p = 0;
2081                         next = p+1;
2082                 }
2083                 arg = strchr(token, '=');
2084                 if (arg) {
2085                         *arg = 0;
2086                         arg++;
2087                 }
2088                 if (strcmp(token, "clear-mmp") == 0 ||
2089                     strcmp(token, "clear_mmp") == 0) {
2090                         clear_mmp = 1;
2091                 } else if (strcmp(token, "mmp_update_interval") == 0) {
2092                         unsigned long intv;
2093                         if (!arg) {
2094                                 r_usage++;
2095                                 continue;
2096                         }
2097                         intv = strtoul(arg, &p, 0);
2098                         if (*p) {
2099                                 fprintf(stderr,
2100                                         _("Invalid mmp_update_interval: %s\n"),
2101                                         arg);
2102                                 r_usage++;
2103                                 continue;
2104                         }
2105                         if (intv == 0) {
2106                                 intv = EXT4_MMP_UPDATE_INTERVAL;
2107                         } else if (intv > EXT4_MMP_MAX_UPDATE_INTERVAL) {
2108                                 fprintf(stderr,
2109                                         _("mmp_update_interval too big: %lu\n"),
2110                                         intv);
2111                                 r_usage++;
2112                                 continue;
2113                         }
2114                         printf(P_("Setting multiple mount protection update "
2115                                   "interval to %lu second\n",
2116                                   "Setting multiple mount protection update "
2117                                   "interval to %lu seconds\n", intv),
2118                                intv);
2119                         fs->super->s_mmp_update_interval = intv;
2120                         ext2fs_mark_super_dirty(fs);
2121                 } else if (!strcmp(token, "force_fsck")) {
2122                         fs->super->s_state |= EXT2_ERROR_FS;
2123                         printf(_("Setting filesystem error flag to force fsck.\n"));
2124                         ext2fs_mark_super_dirty(fs);
2125                 } else if (!strcmp(token, "test_fs")) {
2126                         fs->super->s_flags |= EXT2_FLAGS_TEST_FILESYS;
2127                         printf("Setting test filesystem flag\n");
2128                         ext2fs_mark_super_dirty(fs);
2129                 } else if (!strcmp(token, "^test_fs")) {
2130                         fs->super->s_flags &= ~EXT2_FLAGS_TEST_FILESYS;
2131                         printf("Clearing test filesystem flag\n");
2132                         ext2fs_mark_super_dirty(fs);
2133                 } else if (strcmp(token, "stride") == 0) {
2134                         if (!arg) {
2135                                 r_usage++;
2136                                 continue;
2137                         }
2138                         stride = strtoul(arg, &p, 0);
2139                         if (*p) {
2140                                 fprintf(stderr,
2141                                         _("Invalid RAID stride: %s\n"),
2142                                         arg);
2143                                 r_usage++;
2144                                 continue;
2145                         }
2146                         stride_set = 1;
2147                 } else if (strcmp(token, "stripe-width") == 0 ||
2148                            strcmp(token, "stripe_width") == 0) {
2149                         if (!arg) {
2150                                 r_usage++;
2151                                 continue;
2152                         }
2153                         stripe_width = strtoul(arg, &p, 0);
2154                         if (*p) {
2155                                 fprintf(stderr,
2156                                         _("Invalid RAID stripe-width: %s\n"),
2157                                         arg);
2158                                 r_usage++;
2159                                 continue;
2160                         }
2161                         stripe_width_set = 1;
2162                 } else if (strcmp(token, "hash_alg") == 0 ||
2163                            strcmp(token, "hash-alg") == 0) {
2164                         if (!arg) {
2165                                 r_usage++;
2166                                 continue;
2167                         }
2168                         hash_alg = e2p_string2hash(arg);
2169                         if (hash_alg < 0) {
2170                                 fprintf(stderr,
2171                                         _("Invalid hash algorithm: %s\n"),
2172                                         arg);
2173                                 r_usage++;
2174                                 continue;
2175                         }
2176                         fs->super->s_def_hash_version = hash_alg;
2177                         printf(_("Setting default hash algorithm "
2178                                  "to %s (%d)\n"),
2179                                arg, hash_alg);
2180                         ext2fs_mark_super_dirty(fs);
2181                 } else if (!strcmp(token, "mount_opts")) {
2182                         if (!arg) {
2183                                 r_usage++;
2184                                 continue;
2185                         }
2186                         if (strlen(arg) >= sizeof(fs->super->s_mount_opts)) {
2187                                 fprintf(stderr,
2188                                         "Extended mount options too long\n");
2189                                 continue;
2190                         }
2191                         ext_mount_opts = strdup(arg);
2192                 } else
2193                         r_usage++;
2194         }
2195         if (r_usage) {
2196                 fprintf(stderr, "%s", _("\nBad options specified.\n\n"
2197                         "Extended options are separated by commas, "
2198                         "and may take an argument which\n"
2199                         "\tis set off by an equals ('=') sign.\n\n"
2200                         "Valid extended options are:\n"
2201                         "\tclear_mmp\n"
2202                         "\thash_alg=<hash algorithm>\n"
2203                         "\tmount_opts=<extended default mount options>\n"
2204                         "\tmmp_update_interval=<mmp update interval in seconds>\n"
2205                         "\tstride=<RAID per-disk chunk size in blocks>\n"
2206                         "\tstripe_width=<RAID stride*data disks in blocks>\n"
2207                         "\tforce_fsck\n"
2208                         "\ttest_fs\n"
2209                         "\t^test_fs\n"));
2210                 free(buf);
2211                 return 1;
2212         }
2213         free(buf);
2214
2215         return 0;
2216 }
2217
2218 /*
2219  * Fill in the block bitmap bmap with the information regarding the
2220  * blocks to be moved
2221  */
2222 static int get_move_bitmaps(ext2_filsys fs, int new_ino_blks_per_grp,
2223                             ext2fs_block_bitmap bmap)
2224 {
2225         dgrp_t i;
2226         int retval;
2227         ext2_badblocks_list bb_list = 0;
2228         blk64_t j, needed_blocks = 0;
2229         blk64_t start_blk, end_blk;
2230
2231         retval = ext2fs_read_bb_inode(fs, &bb_list);
2232         if (retval)
2233                 return retval;
2234
2235         for (i = 0; i < fs->group_desc_count; i++) {
2236                 start_blk = ext2fs_inode_table_loc(fs, i) +
2237                                         fs->inode_blocks_per_group;
2238
2239                 end_blk = ext2fs_inode_table_loc(fs, i) +
2240                                         new_ino_blks_per_grp;
2241
2242                 for (j = start_blk; j < end_blk; j++) {
2243                         if (ext2fs_test_block_bitmap2(fs->block_map, j)) {
2244                                 /*
2245                                  * IF the block is a bad block we fail
2246                                  */
2247                                 if (ext2fs_badblocks_list_test(bb_list, j)) {
2248                                         ext2fs_badblocks_list_free(bb_list);
2249                                         return ENOSPC;
2250                                 }
2251
2252                                 ext2fs_mark_block_bitmap2(bmap, j);
2253                         } else {
2254                                 /*
2255                                  * We are going to use this block for
2256                                  * inode table. So mark them used.
2257                                  */
2258                                 ext2fs_mark_block_bitmap2(fs->block_map, j);
2259                         }
2260                 }
2261                 needed_blocks += end_blk - start_blk;
2262         }
2263
2264         ext2fs_badblocks_list_free(bb_list);
2265         if (needed_blocks > ext2fs_free_blocks_count(fs->super))
2266                 return ENOSPC;
2267
2268         return 0;
2269 }
2270
2271 static int ext2fs_is_meta_block(ext2_filsys fs, blk64_t blk)
2272 {
2273         dgrp_t group;
2274         group = ext2fs_group_of_blk2(fs, blk);
2275         if (ext2fs_block_bitmap_loc(fs, group) == blk)
2276                 return 1;
2277         if (ext2fs_inode_bitmap_loc(fs, group) == blk)
2278                 return 1;
2279         return 0;
2280 }
2281
2282 static int ext2fs_is_block_in_group(ext2_filsys fs, dgrp_t group, blk64_t blk)
2283 {
2284         blk64_t start_blk, end_blk;
2285         start_blk = fs->super->s_first_data_block +
2286                         EXT2_GROUPS_TO_BLOCKS(fs->super, group);
2287         /*
2288          * We cannot get new block beyond end_blk for for the last block group
2289          * so we can check with EXT2_BLOCKS_PER_GROUP even for last block group
2290          */
2291         end_blk   = start_blk + EXT2_BLOCKS_PER_GROUP(fs->super);
2292         if (blk >= start_blk && blk <= end_blk)
2293                 return 1;
2294         return 0;
2295 }
2296
2297 static int move_block(ext2_filsys fs, ext2fs_block_bitmap bmap)
2298 {
2299
2300         char *buf;
2301         dgrp_t group = 0;
2302         errcode_t retval;
2303         int meta_data = 0;
2304         blk64_t blk, new_blk, goal;
2305         struct blk_move *bmv;
2306
2307         retval = ext2fs_get_mem(fs->blocksize, &buf);
2308         if (retval)
2309                 return retval;
2310
2311         for (new_blk = blk = fs->super->s_first_data_block;
2312              blk < ext2fs_blocks_count(fs->super); blk++) {
2313                 if (!ext2fs_test_block_bitmap2(bmap, blk))
2314                         continue;
2315
2316                 if (ext2fs_is_meta_block(fs, blk)) {
2317                         /*
2318                          * If the block is mapping a fs meta data block
2319                          * like group desc/block bitmap/inode bitmap. We
2320                          * should find a block in the same group and fix
2321                          * the respective fs metadata pointers. Otherwise
2322                          * fail
2323                          */
2324                         group = ext2fs_group_of_blk2(fs, blk);
2325                         goal = ext2fs_group_first_block2(fs, group);
2326                         meta_data = 1;
2327
2328                 } else {
2329                         goal = new_blk;
2330                 }
2331                 retval = ext2fs_new_block2(fs, goal, NULL, &new_blk);
2332                 if (retval)
2333                         goto err_out;
2334
2335                 /* new fs meta data block should be in the same group */
2336                 if (meta_data && !ext2fs_is_block_in_group(fs, group, new_blk)) {
2337                         retval = ENOSPC;
2338                         goto err_out;
2339                 }
2340
2341                 /* Mark this block as allocated */
2342                 ext2fs_mark_block_bitmap2(fs->block_map, new_blk);
2343
2344                 /* Add it to block move list */
2345                 retval = ext2fs_get_mem(sizeof(struct blk_move), &bmv);
2346                 if (retval)
2347                         goto err_out;
2348
2349                 bmv->old_loc = blk;
2350                 bmv->new_loc = new_blk;
2351
2352                 list_add(&(bmv->list), &blk_move_list);
2353
2354                 retval = io_channel_read_blk64(fs->io, blk, 1, buf);
2355                 if (retval)
2356                         goto err_out;
2357
2358                 retval = io_channel_write_blk64(fs->io, new_blk, 1, buf);
2359                 if (retval)
2360                         goto err_out;
2361         }
2362
2363 err_out:
2364         ext2fs_free_mem(&buf);
2365         return retval;
2366 }
2367
2368 static blk64_t translate_block(blk64_t blk)
2369 {
2370         struct list_head *entry;
2371         struct blk_move *bmv;
2372
2373         list_for_each(entry, &blk_move_list) {
2374                 bmv = list_entry(entry, struct blk_move, list);
2375                 if (bmv->old_loc == blk)
2376                         return bmv->new_loc;
2377         }
2378
2379         return 0;
2380 }
2381
2382 static int process_block(ext2_filsys fs EXT2FS_ATTR((unused)),
2383                          blk64_t *block_nr,
2384                          e2_blkcnt_t blockcnt EXT2FS_ATTR((unused)),
2385                          blk64_t ref_block EXT2FS_ATTR((unused)),
2386                          int ref_offset EXT2FS_ATTR((unused)),
2387                          void *priv_data)
2388 {
2389         int ret = 0;
2390         blk64_t new_blk;
2391         ext2fs_block_bitmap bmap = (ext2fs_block_bitmap) priv_data;
2392
2393         if (!ext2fs_test_block_bitmap2(bmap, *block_nr))
2394                 return 0;
2395         new_blk = translate_block(*block_nr);
2396         if (new_blk) {
2397                 *block_nr = new_blk;
2398                 /*
2399                  * This will force the ext2fs_write_inode in the iterator
2400                  */
2401                 ret |= BLOCK_CHANGED;
2402         }
2403
2404         return ret;
2405 }
2406
2407 static int inode_scan_and_fix(ext2_filsys fs, ext2fs_block_bitmap bmap)
2408 {
2409         errcode_t retval = 0;
2410         ext2_ino_t ino;
2411         blk64_t blk;
2412         char *block_buf = 0;
2413         struct ext2_inode inode;
2414         ext2_inode_scan scan = NULL;
2415
2416         retval = ext2fs_get_mem(fs->blocksize * 3, &block_buf);
2417         if (retval)
2418                 return retval;
2419
2420         retval = ext2fs_open_inode_scan(fs, 0, &scan);
2421         if (retval)
2422                 goto err_out;
2423
2424         while (1) {
2425                 retval = ext2fs_get_next_inode(scan, &ino, &inode);
2426                 if (retval)
2427                         goto err_out;
2428
2429                 if (!ino)
2430                         break;
2431
2432                 if (inode.i_links_count == 0)
2433                         continue; /* inode not in use */
2434
2435                 /* FIXME!!
2436                  * If we end up modifying the journal inode
2437                  * the sb->s_jnl_blocks will differ. But a
2438                  * subsequent e2fsck fixes that.
2439                  * Do we need to fix this ??
2440                  */
2441
2442                 if (ext2fs_file_acl_block(fs, &inode) &&
2443                     ext2fs_test_block_bitmap2(bmap,
2444                                         ext2fs_file_acl_block(fs, &inode))) {
2445                         blk = translate_block(ext2fs_file_acl_block(fs,
2446                                                                     &inode));
2447                         if (!blk)
2448                                 continue;
2449
2450                         ext2fs_file_acl_block_set(fs, &inode, blk);
2451
2452                         /*
2453                          * Write the inode to disk so that inode table
2454                          * resizing can work
2455                          */
2456                         retval = ext2fs_write_inode(fs, ino, &inode);
2457                         if (retval)
2458                                 goto err_out;
2459                 }
2460
2461                 if (!ext2fs_inode_has_valid_blocks2(fs, &inode))
2462                         continue;
2463
2464                 retval = ext2fs_block_iterate3(fs, ino, 0, block_buf,
2465                                                process_block, bmap);
2466                 if (retval)
2467                         goto err_out;
2468
2469         }
2470
2471 err_out:
2472         ext2fs_free_mem(&block_buf);
2473         ext2fs_close_inode_scan(scan);
2474
2475         return retval;
2476 }
2477
2478 /*
2479  * We need to scan for inode and block bitmaps that may need to be
2480  * moved.  This can take place if the filesystem was formatted for
2481  * RAID arrays using the mke2fs's extended option "stride".
2482  */
2483 static int group_desc_scan_and_fix(ext2_filsys fs, ext2fs_block_bitmap bmap)
2484 {
2485         dgrp_t i;
2486         blk64_t blk, new_blk;
2487
2488         for (i = 0; i < fs->group_desc_count; i++) {
2489                 blk = ext2fs_block_bitmap_loc(fs, i);
2490                 if (ext2fs_test_block_bitmap2(bmap, blk)) {
2491                         new_blk = translate_block(blk);
2492                         if (!new_blk)
2493                                 continue;
2494                         ext2fs_block_bitmap_loc_set(fs, i, new_blk);
2495                 }
2496
2497                 blk = ext2fs_inode_bitmap_loc(fs, i);
2498                 if (ext2fs_test_block_bitmap2(bmap, blk)) {
2499                         new_blk = translate_block(blk);
2500                         if (!new_blk)
2501                                 continue;
2502                         ext2fs_inode_bitmap_loc_set(fs, i, new_blk);
2503                 }
2504         }
2505         return 0;
2506 }
2507
2508 static int expand_inode_table(ext2_filsys fs, unsigned long new_ino_size)
2509 {
2510         dgrp_t i;
2511         blk64_t blk;
2512         errcode_t retval;
2513         int new_ino_blks_per_grp;
2514         unsigned int j;
2515         char *old_itable = NULL, *new_itable = NULL;
2516         char *tmp_old_itable = NULL, *tmp_new_itable = NULL;
2517         unsigned long old_ino_size;
2518         int old_itable_size, new_itable_size;
2519
2520         old_itable_size = fs->inode_blocks_per_group * fs->blocksize;
2521         old_ino_size = EXT2_INODE_SIZE(fs->super);
2522
2523         new_ino_blks_per_grp = ext2fs_div_ceil(
2524                                         EXT2_INODES_PER_GROUP(fs->super) *
2525                                         new_ino_size,
2526                                         fs->blocksize);
2527
2528         new_itable_size = new_ino_blks_per_grp * fs->blocksize;
2529
2530         retval = ext2fs_get_mem(old_itable_size, &old_itable);
2531         if (retval)
2532                 return retval;
2533
2534         retval = ext2fs_get_mem(new_itable_size, &new_itable);
2535         if (retval)
2536                 goto err_out;
2537
2538         tmp_old_itable = old_itable;
2539         tmp_new_itable = new_itable;
2540
2541         for (i = 0; i < fs->group_desc_count; i++) {
2542                 blk = ext2fs_inode_table_loc(fs, i);
2543                 retval = io_channel_read_blk64(fs->io, blk,
2544                                 fs->inode_blocks_per_group, old_itable);
2545                 if (retval)
2546                         goto err_out;
2547
2548                 for (j = 0; j < EXT2_INODES_PER_GROUP(fs->super); j++) {
2549                         memcpy(new_itable, old_itable, old_ino_size);
2550
2551                         memset(new_itable+old_ino_size, 0,
2552                                         new_ino_size - old_ino_size);
2553
2554                         new_itable += new_ino_size;
2555                         old_itable += old_ino_size;
2556                 }
2557
2558                 /* reset the pointer */
2559                 old_itable = tmp_old_itable;
2560                 new_itable = tmp_new_itable;
2561
2562                 retval = io_channel_write_blk64(fs->io, blk,
2563                                         new_ino_blks_per_grp, new_itable);
2564                 if (retval)
2565                         goto err_out;
2566         }
2567
2568         /* Update the meta data */
2569         fs->inode_blocks_per_group = new_ino_blks_per_grp;
2570         ext2fs_free_inode_cache(fs->icache);
2571         fs->icache = 0;
2572         fs->super->s_inode_size = new_ino_size;
2573
2574 err_out:
2575         if (old_itable)
2576                 ext2fs_free_mem(&old_itable);
2577
2578         if (new_itable)
2579                 ext2fs_free_mem(&new_itable);
2580
2581         return retval;
2582 }
2583
2584 static errcode_t ext2fs_calculate_summary_stats(ext2_filsys fs)
2585 {
2586         blk64_t         blk;
2587         ext2_ino_t      ino;
2588         unsigned int    group = 0;
2589         unsigned int    count = 0;
2590         int             total_free = 0;
2591         int             group_free = 0;
2592
2593         /*
2594          * First calculate the block statistics
2595          */
2596         for (blk = fs->super->s_first_data_block;
2597              blk < ext2fs_blocks_count(fs->super); blk++) {
2598                 if (!ext2fs_fast_test_block_bitmap2(fs->block_map, blk)) {
2599                         group_free++;
2600                         total_free++;
2601                 }
2602                 count++;
2603                 if ((count == fs->super->s_blocks_per_group) ||
2604                     (blk == ext2fs_blocks_count(fs->super)-1)) {
2605                         ext2fs_bg_free_blocks_count_set(fs, group++,
2606                                                         group_free);
2607                         count = 0;
2608                         group_free = 0;
2609                 }
2610         }
2611         total_free = EXT2FS_C2B(fs, total_free);
2612         ext2fs_free_blocks_count_set(fs->super, total_free);
2613
2614         /*
2615          * Next, calculate the inode statistics
2616          */
2617         group_free = 0;
2618         total_free = 0;
2619         count = 0;
2620         group = 0;
2621
2622         /* Protect loop from wrap-around if s_inodes_count maxed */
2623         for (ino = 1; ino <= fs->super->s_inodes_count && ino > 0; ino++) {
2624                 if (!ext2fs_fast_test_inode_bitmap2(fs->inode_map, ino)) {
2625                         group_free++;
2626                         total_free++;
2627                 }
2628                 count++;
2629                 if ((count == fs->super->s_inodes_per_group) ||
2630                     (ino == fs->super->s_inodes_count)) {
2631                         ext2fs_bg_free_inodes_count_set(fs, group++,
2632                                                         group_free);
2633                         count = 0;
2634                         group_free = 0;
2635                 }
2636         }
2637         fs->super->s_free_inodes_count = total_free;
2638         ext2fs_mark_super_dirty(fs);
2639         return 0;
2640 }
2641
2642 #define list_for_each_safe(pos, pnext, head) \
2643         for (pos = (head)->next, pnext = pos->next; pos != (head); \
2644              pos = pnext, pnext = pos->next)
2645
2646 static void free_blk_move_list(void)
2647 {
2648         struct list_head *entry, *tmp;
2649         struct blk_move *bmv;
2650
2651         list_for_each_safe(entry, tmp, &blk_move_list) {
2652                 bmv = list_entry(entry, struct blk_move, list);
2653                 list_del(entry);
2654                 ext2fs_free_mem(&bmv);
2655         }
2656         return;
2657 }
2658
2659 static int resize_inode(ext2_filsys fs, unsigned long new_size)
2660 {
2661         errcode_t retval;
2662         int new_ino_blks_per_grp;
2663         ext2fs_block_bitmap bmap;
2664
2665         retval = ext2fs_read_inode_bitmap(fs);
2666         if (retval) {
2667                 fputs(_("Failed to read inode bitmap\n"), stderr);
2668                 return retval;
2669         }
2670         retval = ext2fs_read_block_bitmap(fs);
2671         if (retval) {
2672                 fputs(_("Failed to read block bitmap\n"), stderr);
2673                 return retval;
2674         }
2675         INIT_LIST_HEAD(&blk_move_list);
2676
2677
2678         new_ino_blks_per_grp = ext2fs_div_ceil(
2679                                         EXT2_INODES_PER_GROUP(fs->super)*
2680                                         new_size,
2681                                         fs->blocksize);
2682
2683         /* We may change the file system.
2684          * Mark the file system as invalid so that
2685          * the user is prompted to run fsck.
2686          */
2687         fs->super->s_state &= ~EXT2_VALID_FS;
2688
2689         retval = ext2fs_allocate_block_bitmap(fs, _("blocks to be moved"),
2690                                                 &bmap);
2691         if (retval) {
2692                 fputs(_("Failed to allocate block bitmap when "
2693                                 "increasing inode size\n"), stderr);
2694                 return retval;
2695         }
2696         retval = get_move_bitmaps(fs, new_ino_blks_per_grp, bmap);
2697         if (retval) {
2698                 fputs(_("Not enough space to increase inode size \n"), stderr);
2699                 goto err_out;
2700         }
2701         retval = move_block(fs, bmap);
2702         if (retval) {
2703                 fputs(_("Failed to relocate blocks during inode resize \n"),
2704                       stderr);
2705                 goto err_out;
2706         }
2707         retval = inode_scan_and_fix(fs, bmap);
2708         if (retval)
2709                 goto err_out_undo;
2710
2711         retval = group_desc_scan_and_fix(fs, bmap);
2712         if (retval)
2713                 goto err_out_undo;
2714
2715         retval = expand_inode_table(fs, new_size);
2716         if (retval)
2717                 goto err_out_undo;
2718
2719         ext2fs_calculate_summary_stats(fs);
2720
2721         fs->super->s_state |= EXT2_VALID_FS;
2722         /* mark super block and block bitmap as dirty */
2723         ext2fs_mark_super_dirty(fs);
2724         ext2fs_mark_bb_dirty(fs);
2725
2726 err_out:
2727         free_blk_move_list();
2728         ext2fs_free_block_bitmap(bmap);
2729
2730         return retval;
2731
2732 err_out_undo:
2733         free_blk_move_list();
2734         ext2fs_free_block_bitmap(bmap);
2735         fputs(_("Error in resizing the inode size.\n"
2736                         "Run e2undo to undo the "
2737                         "file system changes. \n"), stderr);
2738
2739         return retval;
2740 }
2741
2742 static int tune2fs_setup_tdb(const char *name, io_manager *io_ptr)
2743 {
2744         errcode_t retval = 0;
2745         const char *tdb_dir;
2746         char *tdb_file = NULL;
2747         char *dev_name, *tmp_name;
2748
2749         /* (re)open a specific undo file */
2750         if (undo_file && undo_file[0] != 0) {
2751                 retval = set_undo_io_backing_manager(*io_ptr);
2752                 if (retval)
2753                         goto err;
2754                 *io_ptr = undo_io_manager;
2755                 retval = set_undo_io_backup_file(undo_file);
2756                 if (retval)
2757                         goto err;
2758                 printf(_("Overwriting existing filesystem; this can be undone "
2759                          "using the command:\n"
2760                          "    e2undo %s %s\n\n"),
2761                          undo_file, name);
2762                 return retval;
2763         }
2764
2765         /*
2766          * Configuration via a conf file would be
2767          * nice
2768          */
2769         tdb_dir = getenv("E2FSPROGS_UNDO_DIR");
2770         if (!tdb_dir)
2771                 tdb_dir = "/var/lib/e2fsprogs";
2772
2773         if (!strcmp(tdb_dir, "none") || (tdb_dir[0] == 0) ||
2774             access(tdb_dir, W_OK))
2775                 return 0;
2776
2777         tmp_name = strdup(name);
2778         if (!tmp_name)
2779                 goto errout;
2780         dev_name = basename(tmp_name);
2781         tdb_file = malloc(strlen(tdb_dir) + 9 + strlen(dev_name) + 7 + 1);
2782         if (!tdb_file) {
2783                 free(tmp_name);
2784                 goto errout;
2785         }
2786         sprintf(tdb_file, "%s/tune2fs-%s.e2undo", tdb_dir, dev_name);
2787         free(tmp_name);
2788
2789         if ((unlink(tdb_file) < 0) && (errno != ENOENT)) {
2790                 retval = errno;
2791                 com_err(program_name, retval,
2792                         _("while trying to delete %s"), tdb_file);
2793                 goto errout;
2794         }
2795
2796         retval = set_undo_io_backing_manager(*io_ptr);
2797         if (retval)
2798                 goto errout;
2799         *io_ptr = undo_io_manager;
2800         retval = set_undo_io_backup_file(tdb_file);
2801         if (retval)
2802                 goto errout;
2803         printf(_("Overwriting existing filesystem; this can be undone "
2804                  "using the command:\n"
2805                  "    e2undo %s %s\n\n"),
2806                  tdb_file, name);
2807
2808         free(tdb_file);
2809         return 0;
2810 errout:
2811         free(tdb_file);
2812 err:
2813         com_err("tune2fs", retval, "while trying to setup undo file\n");
2814         return retval;
2815 }
2816
2817 static int
2818 fs_update_journal_user(struct ext2_super_block *sb, __u8 old_uuid[UUID_SIZE])
2819 {
2820         int retval, nr_users, start;
2821         journal_superblock_t *jsb;
2822         ext2_filsys jfs;
2823         __u8 *j_uuid;
2824         char *journal_path;
2825         char uuid[UUID_STR_SIZE];
2826         char buf[SUPERBLOCK_SIZE] __attribute__ ((aligned(8)));
2827
2828         if (!ext2fs_has_feature_journal(sb) || uuid_is_null(sb->s_journal_uuid))
2829                 return 0;
2830
2831         uuid_unparse(sb->s_journal_uuid, uuid);
2832         journal_path = blkid_get_devname(NULL, "UUID", uuid);
2833         if (!journal_path)
2834                 return 0;
2835
2836         retval = ext2fs_open2(journal_path, io_options,
2837                               EXT2_FLAG_JOURNAL_DEV_OK | EXT2_FLAG_RW,
2838                               0, 0, unix_io_manager, &jfs);
2839         if (retval) {
2840                 com_err(program_name, retval,
2841                         _("while trying to open %s"),
2842                         journal_path);
2843                 return retval;
2844         }
2845
2846         retval = get_journal_sb(jfs, buf);
2847         if (retval != 0) {
2848                 if (retval == EXT2_ET_UNSUPP_FEATURE)
2849                         fprintf(stderr, _("%s is not a journal device.\n"),
2850                                 journal_path);
2851                 return retval;
2852         }
2853
2854         jsb = (journal_superblock_t *) buf;
2855         /* Find the filesystem UUID */
2856         nr_users = ntohl(jsb->s_nr_users);
2857
2858         j_uuid = journal_user(old_uuid, jsb->s_users, nr_users);
2859         if (j_uuid == NULL) {
2860                 fputs(_("Filesystem's UUID not found on journal device.\n"),
2861                       stderr);
2862                 return EXT2_ET_LOAD_EXT_JOURNAL;
2863         }
2864
2865         memcpy(j_uuid, sb->s_uuid, UUID_SIZE);
2866
2867         start = ext2fs_journal_sb_start(jfs->blocksize);
2868         /* Write back the journal superblock */
2869         retval = io_channel_write_blk64(jfs->io, start, -SUPERBLOCK_SIZE, buf);
2870         if (retval != 0) {
2871                 com_err(program_name, retval,
2872                         "while writing journal superblock.");
2873                 return retval;
2874         }
2875
2876         ext2fs_close(jfs);
2877
2878         return 0;
2879 }
2880
2881 #ifndef BUILD_AS_LIB
2882 int main(int argc, char **argv)
2883 #else
2884 int tune2fs_main(int argc, char **argv)
2885 #endif  /* BUILD_AS_LIB */
2886 {
2887         errcode_t retval;
2888         ext2_filsys fs;
2889         struct ext2_super_block *sb;
2890         io_manager io_ptr, io_ptr_orig = NULL;
2891         int rc = 0;
2892         char default_undo_file[1] = { 0 };
2893
2894 #ifdef ENABLE_NLS
2895         setlocale(LC_MESSAGES, "");
2896         setlocale(LC_CTYPE, "");
2897         bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
2898         textdomain(NLS_CAT_NAME);
2899         set_com_err_gettext(gettext);
2900 #endif
2901         if (argc && *argv)
2902                 program_name = *argv;
2903         add_error_table(&et_ext2_error_table);
2904
2905 #ifdef CONFIG_BUILD_FINDFS
2906         if (strcmp(get_progname(argv[0]), "findfs") == 0)
2907                 do_findfs(argc, argv);
2908 #endif
2909         if (strcmp(get_progname(argv[0]), "e2label") == 0)
2910                 parse_e2label_options(argc, argv);
2911         else
2912                 parse_tune2fs_options(argc, argv);
2913
2914 #ifdef CONFIG_TESTIO_DEBUG
2915         if (getenv("TEST_IO_FLAGS") || getenv("TEST_IO_DEBUG")) {
2916                 io_ptr = test_io_manager;
2917                 test_io_backing_manager = unix_io_manager;
2918         } else
2919 #endif
2920                 io_ptr = unix_io_manager;
2921
2922 retry_open:
2923         if ((open_flag & EXT2_FLAG_RW) == 0 || f_flag)
2924                 open_flag |= EXT2_FLAG_SKIP_MMP;
2925
2926         open_flag |= EXT2_FLAG_64BITS | EXT2_FLAG_JOURNAL_DEV_OK;
2927
2928         /* keep the filesystem struct around to dump MMP data */
2929         open_flag |= EXT2_FLAG_NOFREE_ON_ERROR;
2930
2931         retval = ext2fs_open2(device_name, io_options, open_flag,
2932                               0, 0, io_ptr, &fs);
2933         if (retval) {
2934                 com_err(program_name, retval,
2935                         _("while trying to open %s"),
2936                         device_name);
2937                 if (retval == EXT2_ET_MMP_FSCK_ON ||
2938                     retval == EXT2_ET_MMP_UNKNOWN_SEQ)
2939                         dump_mmp_msg(fs->mmp_buf,
2940                                      _("If you are sure the filesystem "
2941                                        "is not in use on any node, run:\n"
2942                                        "'tune2fs -f -E clear_mmp {device}'\n"));
2943                 else if (retval == EXT2_ET_MMP_FAILED)
2944                         dump_mmp_msg(fs->mmp_buf, NULL);
2945                 else if (retval == EXT2_ET_MMP_MAGIC_INVALID)
2946                         fprintf(stderr,
2947                                 _("MMP block magic is bad. Try to fix it by "
2948                                   "running:\n'e2fsck -f %s'\n"), device_name);
2949                 else if (retval == EXT2_ET_BAD_MAGIC)
2950                         check_plausibility(device_name, CHECK_FS_EXIST, NULL);
2951                 else if (retval != EXT2_ET_MMP_FAILED)
2952                         fprintf(stderr, "%s",
2953                              _("Couldn't find valid filesystem superblock.\n"));
2954
2955                 ext2fs_free(fs);
2956                 exit(1);
2957         }
2958         if (ext2fs_has_feature_journal_dev(fs->super)) {
2959                 fprintf(stderr, "%s", _("Cannot modify a journal device.\n"));
2960                 ext2fs_free(fs);
2961                 exit(1);
2962         }
2963         fs->default_bitmap_type = EXT2FS_BMAP64_RBTREE;
2964
2965         if (I_flag) {
2966                 /*
2967                  * Check the inode size is right so we can issue an
2968                  * error message and bail before setting up the tdb
2969                  * file.
2970                  */
2971                 if (new_inode_size == EXT2_INODE_SIZE(fs->super)) {
2972                         fprintf(stderr, _("The inode size is already %lu\n"),
2973                                 new_inode_size);
2974                         rc = 1;
2975                         goto closefs;
2976                 }
2977                 if (new_inode_size < EXT2_INODE_SIZE(fs->super)) {
2978                         fprintf(stderr, "%s",
2979                                 _("Shrinking inode size is not supported\n"));
2980                         rc = 1;
2981                         goto closefs;
2982                 }
2983                 if (new_inode_size > fs->blocksize) {
2984                         fprintf(stderr, _("Invalid inode size %lu (max %d)\n"),
2985                                 new_inode_size, fs->blocksize);
2986                         rc = 1;
2987                         goto closefs;
2988                 }
2989                 check_fsck_needed(fs,
2990                         _("Resizing inodes could take some time."));
2991                 /*
2992                  * If inode resize is requested use the
2993                  * Undo I/O manager
2994                  */
2995                 undo_file = default_undo_file;
2996         }
2997
2998         /* Set up an undo file */
2999         if (undo_file && io_ptr_orig == NULL) {
3000                 io_ptr_orig = io_ptr;
3001                 retval = tune2fs_setup_tdb(device_name, &io_ptr);
3002                 if (retval) {
3003                         rc = 1;
3004                         goto closefs;
3005                 }
3006                 if (io_ptr != io_ptr_orig) {
3007                         ext2fs_close_free(&fs);
3008                         goto retry_open;
3009                 }
3010         }
3011
3012         sb = fs->super;
3013         fs->flags &= ~EXT2_FLAG_MASTER_SB_ONLY;
3014
3015         if (print_label) {
3016                 /* For e2label emulation */
3017                 printf("%.*s\n", (int) sizeof(sb->s_volume_name),
3018                        sb->s_volume_name);
3019                 remove_error_table(&et_ext2_error_table);
3020                 goto closefs;
3021         }
3022
3023         retval = ext2fs_check_if_mounted(device_name, &mount_flags);
3024         if (retval) {
3025                 com_err("ext2fs_check_if_mount", retval,
3026                         _("while determining whether %s is mounted."),
3027                         device_name);
3028                 rc = 1;
3029                 goto closefs;
3030         }
3031
3032 #ifdef NO_RECOVERY
3033         /* Warn if file system needs recovery and it is opened for writing. */
3034         if ((open_flag & EXT2_FLAG_RW) && !(mount_flags & EXT2_MF_MOUNTED) &&
3035             (sb->s_feature_compat & EXT3_FEATURE_COMPAT_HAS_JOURNAL) &&
3036             (sb->s_feature_incompat & EXT3_FEATURE_INCOMPAT_RECOVER)) {
3037                 fprintf(stderr,
3038 _("Warning: The journal is dirty. You may wish to replay the journal like:\n\n"
3039   "\te2fsck -E journal_only %s\n\n"
3040   "then rerun this command.  Otherwise, any changes made may be overwritten\n"
3041   "by journal recovery.\n"), device_name);
3042         }
3043 #else
3044         /* Recover the journal if possible. */
3045         if ((open_flag & EXT2_FLAG_RW) && !(mount_flags & (EXT2_MF_BUSY | EXT2_MF_MOUNTED)) &&
3046             ext2fs_has_feature_journal_needs_recovery(fs->super)) {
3047                 errcode_t err;
3048
3049                 printf(_("Recovering journal.\n"));
3050                 err = ext2fs_run_ext3_journal(&fs);
3051                 if (err) {
3052                         com_err("tune2fs", err, "while recovering journal.\n");
3053                         printf(_("Please run e2fsck -fy %s.\n"), argv[1]);
3054                         if (fs)
3055                                 ext2fs_close_free(&fs);
3056                         exit(1);
3057                 }
3058         }
3059 #endif
3060
3061         /* Normally we only need to write out the superblock */
3062         fs->flags |= EXT2_FLAG_SUPER_ONLY;
3063
3064         if (c_flag) {
3065                 sb->s_max_mnt_count = max_mount_count;
3066                 ext2fs_mark_super_dirty(fs);
3067                 printf(_("Setting maximal mount count to %d\n"),
3068                        max_mount_count);
3069         }
3070         if (C_flag) {
3071                 sb->s_mnt_count = mount_count;
3072                 ext2fs_mark_super_dirty(fs);
3073                 printf(_("Setting current mount count to %d\n"), mount_count);
3074         }
3075         if (e_flag) {
3076                 sb->s_errors = errors;
3077                 ext2fs_mark_super_dirty(fs);
3078                 printf(_("Setting error behavior to %d\n"), errors);
3079         }
3080         if (g_flag) {
3081                 sb->s_def_resgid = resgid;
3082                 ext2fs_mark_super_dirty(fs);
3083                 printf(_("Setting reserved blocks gid to %lu\n"), resgid);
3084         }
3085         if (i_flag) {
3086                 if ((unsigned long long)interval >= (1ULL << 32)) {
3087                         com_err(program_name, 0,
3088                                 _("interval between checks is too big (%lu)"),
3089                                 interval);
3090                         rc = 1;
3091                         goto closefs;
3092                 }
3093                 sb->s_checkinterval = interval;
3094                 ext2fs_mark_super_dirty(fs);
3095                 printf(_("Setting interval between checks to %lu seconds\n"),
3096                        interval);
3097         }
3098         if (m_flag) {
3099                 ext2fs_r_blocks_count_set(sb, reserved_ratio *
3100                                           ext2fs_blocks_count(sb) / 100.0);
3101                 ext2fs_mark_super_dirty(fs);
3102                 printf (_("Setting reserved blocks percentage to %g%% (%llu blocks)\n"),
3103                         reserved_ratio, ext2fs_r_blocks_count(sb));
3104         }
3105         if (r_flag) {
3106                 if (reserved_blocks > ext2fs_blocks_count(sb)/2) {
3107                         com_err(program_name, 0,
3108                                 _("reserved blocks count is too big (%llu)"),
3109                                 reserved_blocks);
3110                         rc = 1;
3111                         goto closefs;
3112                 }
3113                 ext2fs_r_blocks_count_set(sb, reserved_blocks);
3114                 ext2fs_mark_super_dirty(fs);
3115                 printf(_("Setting reserved blocks count to %llu\n"),
3116                        reserved_blocks);
3117         }
3118         if (s_flag == 1) {
3119                 if (ext2fs_has_feature_sparse_super(sb)) {
3120                         fputs(_("\nThe filesystem already has sparse "
3121                                 "superblocks.\n"), stderr);
3122                 } else if (ext2fs_has_feature_meta_bg(sb)) {
3123                         fputs(_("\nSetting the sparse superblock flag not "
3124                                 "supported\nfor filesystems with "
3125                                 "the meta_bg feature enabled.\n"),
3126                                 stderr);
3127                         rc = 1;
3128                         goto closefs;
3129                 } else {
3130                         ext2fs_set_feature_sparse_super(sb);
3131                         sb->s_state &= ~EXT2_VALID_FS;
3132                         ext2fs_mark_super_dirty(fs);
3133                         printf(_("\nSparse superblock flag set.  %s"),
3134                                _(please_fsck));
3135                 }
3136         }
3137         if (s_flag == 0) {
3138                 fputs(_("\nClearing the sparse superblock flag not supported.\n"),
3139                       stderr);
3140                 rc = 1;
3141                 goto closefs;
3142         }
3143         if (T_flag) {
3144                 sb->s_lastcheck = last_check_time;
3145                 ext2fs_mark_super_dirty(fs);
3146                 printf(_("Setting time filesystem last checked to %s\n"),
3147                        ctime(&last_check_time));
3148         }
3149         if (u_flag) {
3150                 sb->s_def_resuid = resuid;
3151                 ext2fs_mark_super_dirty(fs);
3152                 printf(_("Setting reserved blocks uid to %lu\n"), resuid);
3153         }
3154         if (L_flag) {
3155                 if (strlen(new_label) > sizeof(sb->s_volume_name))
3156                         fputs(_("Warning: label too long, truncating.\n"),
3157                               stderr);
3158                 memset(sb->s_volume_name, 0, sizeof(sb->s_volume_name));
3159                 strncpy(sb->s_volume_name, new_label,
3160                         sizeof(sb->s_volume_name));
3161                 ext2fs_mark_super_dirty(fs);
3162         }
3163         if (M_flag) {
3164                 memset(sb->s_last_mounted, 0, sizeof(sb->s_last_mounted));
3165                 strncpy(sb->s_last_mounted, new_last_mounted,
3166                         sizeof(sb->s_last_mounted));
3167                 ext2fs_mark_super_dirty(fs);
3168         }
3169         if (mntopts_cmd) {
3170                 rc = update_mntopts(fs, mntopts_cmd);
3171                 if (rc)
3172                         goto closefs;
3173         }
3174         if (features_cmd) {
3175                 rc = update_feature_set(fs, features_cmd);
3176                 if (rc)
3177                         goto closefs;
3178         }
3179         if (extended_cmd) {
3180                 rc = parse_extended_opts(fs, extended_cmd);
3181                 if (rc)
3182                         goto closefs;
3183                 if (clear_mmp && !f_flag) {
3184                         fputs(_("Error in using clear_mmp. "
3185                                 "It must be used with -f\n"),
3186                               stderr);
3187                         goto closefs;
3188                 }
3189         }
3190         if (clear_mmp) {
3191                 rc = ext2fs_mmp_clear(fs);
3192                 goto closefs;
3193         }
3194         if (journal_size || journal_device) {
3195                 rc = add_journal(fs);
3196                 if (rc)
3197                         goto closefs;
3198         }
3199
3200         if (Q_flag) {
3201                 if (mount_flags & EXT2_MF_MOUNTED) {
3202                         fputs(_("The quota feature may only be changed when "
3203                                 "the filesystem is unmounted.\n"), stderr);
3204                         rc = 1;
3205                         goto closefs;
3206                 }
3207                 handle_quota_options(fs);
3208         }
3209
3210         if (U_flag) {
3211                 int set_csum = 0;
3212                 dgrp_t i;
3213                 char buf[SUPERBLOCK_SIZE] __attribute__ ((aligned(8)));
3214                 __u8 old_uuid[UUID_SIZE];
3215
3216                 if (ext2fs_has_group_desc_csum(fs)) {
3217                         /*
3218                          * Changing the UUID on a metadata_csum FS requires
3219                          * rewriting all metadata, which can race with a
3220                          * mounted fs.  Don't allow that unless we're saving
3221                          * the checksum seed.
3222                          */
3223                         if ((mount_flags & EXT2_MF_MOUNTED) &&
3224                             !ext2fs_has_feature_csum_seed(fs->super) &&
3225                             ext2fs_has_feature_metadata_csum(fs->super)) {
3226                                 fputs(_("The UUID may only be "
3227                                         "changed when the filesystem is "
3228                                         "unmounted.\n"), stderr);
3229                                 fputs(_("If you only use kernels newer than "
3230                                         "v4.4, run 'tune2fs -O "
3231                                         "metadata_csum_seed' and re-run this "
3232                                         "command.\n"), stderr);
3233                                 try_confirm_csum_seed_support();
3234                                 exit(1);
3235                         }
3236                         if (!ext2fs_has_feature_csum_seed(fs->super))
3237                                 check_fsck_needed(fs,
3238                                         _("Setting UUID on a checksummed "
3239                                           "filesystem could take some time."));
3240
3241                         /*
3242                          * Determine if the block group checksums are
3243                          * correct so we know whether or not to set
3244                          * them later on.
3245                          */
3246                         for (i = 0; i < fs->group_desc_count; i++)
3247                                 if (!ext2fs_group_desc_csum_verify(fs, i))
3248                                         break;
3249                         if (i >= fs->group_desc_count)
3250                                 set_csum = 1;
3251                 }
3252
3253                 memcpy(old_uuid, sb->s_uuid, UUID_SIZE);
3254                 if ((strcasecmp(new_UUID, "null") == 0) ||
3255                     (strcasecmp(new_UUID, "clear") == 0)) {
3256                         uuid_clear(sb->s_uuid);
3257                 } else if (strcasecmp(new_UUID, "time") == 0) {
3258                         uuid_generate_time(sb->s_uuid);
3259                 } else if (strcasecmp(new_UUID, "random") == 0) {
3260                         uuid_generate(sb->s_uuid);
3261                 } else if (uuid_parse(new_UUID, sb->s_uuid)) {
3262                         com_err(program_name, 0, "%s",
3263                                 _("Invalid UUID format\n"));
3264                         rc = 1;
3265                         goto closefs;
3266                 }
3267                 ext2fs_init_csum_seed(fs);
3268                 if (set_csum) {
3269                         for (i = 0; i < fs->group_desc_count; i++)
3270                                 ext2fs_group_desc_csum_set(fs, i);
3271                         fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
3272                 }
3273
3274                 /* If this is a journal dev, we need to copy UUID into jsb */
3275                 if (!(rc = get_journal_sb(fs, buf))) {
3276                         journal_superblock_t *jsb;
3277
3278                         jsb = (journal_superblock_t *) buf;
3279                         fputs(_("Need to update journal superblock.\n"), stdout);
3280                         memcpy(jsb->s_uuid, sb->s_uuid, sizeof(sb->s_uuid));
3281
3282                         /* Writeback the journal superblock */
3283                         if ((rc = io_channel_write_blk64(fs->io,
3284                                 ext2fs_journal_sb_start(fs->blocksize),
3285                                         -SUPERBLOCK_SIZE, buf)))
3286                                 goto closefs;
3287                 } else if (rc != EXT2_ET_UNSUPP_FEATURE)
3288                         goto closefs;
3289                 else {
3290                         rc = 0; /** Reset rc to avoid ext2fs_mmp_stop() */
3291
3292                         if ((rc = fs_update_journal_user(sb, old_uuid)))
3293                                 goto closefs;
3294                 }
3295
3296                 ext2fs_mark_super_dirty(fs);
3297                 if (!ext2fs_has_feature_csum_seed(fs->super) &&
3298                     (ext2fs_has_feature_metadata_csum(fs->super) ||
3299                      ext2fs_has_feature_ea_inode(fs->super)))
3300                         rewrite_checksums = 1;
3301         }
3302
3303         if (I_flag) {
3304                 if (mount_flags & EXT2_MF_MOUNTED) {
3305                         fputs(_("The inode size may only be "
3306                                 "changed when the filesystem is "
3307                                 "unmounted.\n"), stderr);
3308                         rc = 1;
3309                         goto closefs;
3310                 }
3311                 if (ext2fs_has_feature_flex_bg(fs->super)) {
3312                         fputs(_("Changing the inode size not supported for "
3313                                 "filesystems with the flex_bg\n"
3314                                 "feature enabled.\n"),
3315                               stderr);
3316                         rc = 1;
3317                         goto closefs;
3318                 }
3319                 /*
3320                  * We want to update group descriptor also
3321                  * with the new free inode count
3322                  */
3323                 if (rewrite_checksums)
3324                         fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
3325                 fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
3326                 retval = resize_inode(fs, new_inode_size);
3327                 if (rewrite_checksums)
3328                         fs->flags &= ~EXT2_FLAG_IGNORE_CSUM_ERRORS;
3329                 if (retval == 0) {
3330                         printf(_("Setting inode size %lu\n"),
3331                                                         new_inode_size);
3332                         rewrite_checksums = 1;
3333                 } else {
3334                         printf("%s", _("Failed to change inode size\n"));
3335                         rc = 1;
3336                         goto closefs;
3337                 }
3338         }
3339
3340         if (rewrite_checksums)
3341                 rewrite_metadata_checksums(fs);
3342
3343         if (l_flag)
3344                 list_super(sb);
3345         if (stride_set) {
3346                 sb->s_raid_stride = stride;
3347                 ext2fs_mark_super_dirty(fs);
3348                 printf(_("Setting stride size to %d\n"), stride);
3349         }
3350         if (stripe_width_set) {
3351                 sb->s_raid_stripe_width = stripe_width;
3352                 ext2fs_mark_super_dirty(fs);
3353                 printf(_("Setting stripe width to %d\n"), stripe_width);
3354         }
3355         if (ext_mount_opts) {
3356                 strncpy((char *)(fs->super->s_mount_opts), ext_mount_opts,
3357                         sizeof(fs->super->s_mount_opts));
3358                 fs->super->s_mount_opts[sizeof(fs->super->s_mount_opts)-1] = 0;
3359                 ext2fs_mark_super_dirty(fs);
3360                 printf(_("Setting extended default mount options to '%s'\n"),
3361                        ext_mount_opts);
3362                 free(ext_mount_opts);
3363         }
3364
3365         free(device_name);
3366         remove_error_table(&et_ext2_error_table);
3367
3368 closefs:
3369         if (rc) {
3370                 ext2fs_mmp_stop(fs);
3371 #ifndef BUILD_AS_LIB
3372                 exit(1);
3373 #endif
3374         }
3375
3376         if (feature_64bit)
3377                 convert_64bit(fs, feature_64bit);
3378         return (ext2fs_close_free(&fs) ? 1 : 0);
3379 }