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