Whamcloud - gitweb
Merge branch 'maint' into next
[tools/e2fsprogs.git] / misc / tune2fs.c
index d0d44c8..f52beab 100644 (file)
@@ -73,6 +73,7 @@ char *io_options;
 static int c_flag, C_flag, e_flag, f_flag, g_flag, i_flag, l_flag, L_flag;
 static int m_flag, M_flag, Q_flag, r_flag, s_flag = -1, u_flag, U_flag, T_flag;
 static int I_flag;
+static int clear_mmp;
 static time_t last_check_time;
 static int print_label;
 static int max_mount_count, mount_count, mount_flags;
@@ -90,6 +91,7 @@ static char *extended_cmd;
 static unsigned long new_inode_size;
 static char *ext_mount_opts;
 static int usrquota, grpquota;
+static int rewrite_checksums;
 
 int journal_size, journal_flags;
 char *journal_device;
@@ -104,6 +106,8 @@ struct blk_move {
 
 
 static const char *please_fsck = N_("Please run e2fsck on the filesystem.\n");
+static const char *please_dir_fsck =
+               N_("Please run e2fsck -D on the filesystem.\n");
 
 #ifdef CONFIG_BUILD_FINDFS
 void do_findfs(int argc, char **argv);
@@ -116,7 +120,7 @@ static void usage(void)
                  "[-g group]\n"
                  "\t[-i interval[d|m|w]] [-j] [-J journal_options] [-l]\n"
                  "\t[-m reserved_blocks_percent] "
-                 "[-o [^]mount_options[,...]] \n"
+                 "[-o [^]mount_options[,...]] [-p mmp_update_interval]\n"
                  "\t[-r reserved_blocks_count] [-u user] [-C mount_count] "
                  "[-L volume_label]\n"
                  "\t[-M last_mounted_dir] [-O [^]feature[,...]]\n"
@@ -132,7 +136,8 @@ static __u32 ok_features[3] = {
        /* Incompat */
        EXT2_FEATURE_INCOMPAT_FILETYPE |
                EXT3_FEATURE_INCOMPAT_EXTENTS |
-               EXT4_FEATURE_INCOMPAT_FLEX_BG,
+               EXT4_FEATURE_INCOMPAT_FLEX_BG |
+               EXT4_FEATURE_INCOMPAT_MMP,
        /* R/O compat */
        EXT2_FEATURE_RO_COMPAT_LARGE_FILE |
                EXT4_FEATURE_RO_COMPAT_HUGE_FILE|
@@ -140,7 +145,10 @@ static __u32 ok_features[3] = {
                EXT4_FEATURE_RO_COMPAT_EXTRA_ISIZE|
                EXT4_FEATURE_RO_COMPAT_GDT_CSUM |
                EXT2_FEATURE_RO_COMPAT_SPARSE_SUPER |
-               EXT4_FEATURE_RO_COMPAT_QUOTA
+#ifdef CONFIG_QUOTA
+               EXT4_FEATURE_RO_COMPAT_QUOTA |
+#endif
+               EXT4_FEATURE_RO_COMPAT_METADATA_CSUM
 };
 
 static __u32 clear_ok_features[3] = {
@@ -150,20 +158,24 @@ static __u32 clear_ok_features[3] = {
                EXT2_FEATURE_COMPAT_DIR_INDEX,
        /* Incompat */
        EXT2_FEATURE_INCOMPAT_FILETYPE |
-               EXT4_FEATURE_INCOMPAT_FLEX_BG,
+               EXT4_FEATURE_INCOMPAT_FLEX_BG |
+               EXT4_FEATURE_INCOMPAT_MMP,
        /* R/O compat */
        EXT2_FEATURE_RO_COMPAT_LARGE_FILE |
                EXT4_FEATURE_RO_COMPAT_HUGE_FILE|
                EXT4_FEATURE_RO_COMPAT_DIR_NLINK|
                EXT4_FEATURE_RO_COMPAT_EXTRA_ISIZE|
                EXT4_FEATURE_RO_COMPAT_GDT_CSUM |
-               EXT4_FEATURE_RO_COMPAT_QUOTA
+#ifdef CONFIG_QUOTA
+               EXT4_FEATURE_RO_COMPAT_QUOTA |
+#endif
+               EXT4_FEATURE_RO_COMPAT_METADATA_CSUM
 };
 
 /*
  * Remove an external journal from the filesystem
  */
-static void remove_journal_device(ext2_filsys fs)
+static int remove_journal_device(ext2_filsys fs)
 {
        char            *journal_path;
        ext2_filsys     jfs;
@@ -202,7 +214,8 @@ static void remove_journal_device(ext2_filsys fs)
                        _("while trying to open external journal"));
                goto no_valid_journal;
        }
-       if (!(jfs->super->s_feature_incompat & EXT3_FEATURE_INCOMPAT_JOURNAL_DEV)) {
+       if (!(jfs->super->s_feature_incompat &
+             EXT3_FEATURE_INCOMPAT_JOURNAL_DEV)) {
                fprintf(stderr, _("%s is not a journal device.\n"),
                        journal_path);
                goto no_valid_journal;
@@ -216,8 +229,8 @@ static void remove_journal_device(ext2_filsys fs)
        }
 
        jsb = (journal_superblock_t *) buf;
-       if ((jsb->s_header.h_magic != (unsigned) ntohl(JFS_MAGIC_NUMBER)) ||
-           (jsb->s_header.h_blocktype != (unsigned) ntohl(JFS_SUPERBLOCK_V2))) {
+       if ((jsb->s_header.h_magic != (unsigned)ntohl(JFS_MAGIC_NUMBER)) ||
+           (jsb->s_header.h_blocktype != (unsigned)ntohl(JFS_SUPERBLOCK_V2))) {
                fputs(_("Journal superblock not found!\n"), stderr);
                goto no_valid_journal;
        }
@@ -225,8 +238,7 @@ static void remove_journal_device(ext2_filsys fs)
        /* Find the filesystem UUID */
        nr_users = ntohl(jsb->s_nr_users);
        for (i = 0; i < nr_users; i++) {
-               if (memcmp(fs->super->s_uuid,
-                          &jsb->s_users[i*16], 16) == 0)
+               if (memcmp(fs->super->s_uuid, &jsb->s_users[i * 16], 16) == 0)
                        break;
        }
        if (i >= nr_users) {
@@ -237,7 +249,7 @@ static void remove_journal_device(ext2_filsys fs)
        }
        nr_users--;
        for (i = 0; i < nr_users; i++)
-               memcpy(&jsb->s_users[i*16], &jsb->s_users[(i+1)*16], 16);
+               memcpy(&jsb->s_users[i * 16], &jsb->s_users[(i + 1) * 16], 16);
        jsb->s_nr_users = htonl(nr_users);
 
        /* Write back the journal superblock */
@@ -254,13 +266,15 @@ no_valid_journal:
                fputs(_("Cannot locate journal device. It was NOT removed\n"
                        "Use -f option to remove missing journal device.\n"),
                      stderr);
-               exit(1);
+               return 1;
        }
        fs->super->s_journal_dev = 0;
        uuid_clear(fs->super->s_journal_uuid);
        ext2fs_mark_super_dirty(fs);
        fputs(_("Journal removed\n"), stdout);
        free(journal_path);
+
+       return 0;
 }
 
 /* Helper function for remove_journal_inode */
@@ -285,7 +299,7 @@ static int release_blocks_proc(ext2_filsys fs, blk64_t *blocknr,
 /*
  * Remove the journal inode from the filesystem
  */
-static void remove_journal_inode(ext2_filsys fs)
+static errcode_t remove_journal_inode(ext2_filsys fs)
 {
        struct ext2_inode       inode;
        errcode_t               retval;
@@ -295,14 +309,14 @@ static void remove_journal_inode(ext2_filsys fs)
        if (retval) {
                com_err(program_name, retval,
                        _("while reading journal inode"));
-               exit(1);
+               return retval;
        }
        if (ino == EXT2_JOURNAL_INO) {
                retval = ext2fs_read_bitmaps(fs);
                if (retval) {
                        com_err(program_name, retval,
                                _("while reading bitmaps"));
-                       exit(1);
+                       return retval;
                }
                retval = ext2fs_block_iterate3(fs, ino,
                                               BLOCK_FLAG_READ_ONLY, NULL,
@@ -310,7 +324,7 @@ static void remove_journal_inode(ext2_filsys fs)
                if (retval) {
                        com_err(program_name, retval,
                                _("while clearing journal inode"));
-                       exit(1);
+                       return retval;
                }
                memset(&inode, 0, sizeof(inode));
                ext2fs_mark_bb_dirty(fs);
@@ -321,25 +335,51 @@ static void remove_journal_inode(ext2_filsys fs)
        if (retval) {
                com_err(program_name, retval,
                        _("while writing journal inode"));
-               exit(1);
+               return retval;
        }
        fs->super->s_journal_inum = 0;
        ext2fs_mark_super_dirty(fs);
+
+       return 0;
 }
 
 /*
  * Update the default mount options
  */
-static void update_mntopts(ext2_filsys fs, char *mntopts)
+static int update_mntopts(ext2_filsys fs, char *mntopts)
 {
        struct ext2_super_block *sb = fs->super;
 
        if (e2p_edit_mntopts(mntopts, &sb->s_default_mount_opts, ~0)) {
                fprintf(stderr, _("Invalid mount option set: %s\n"),
                        mntopts);
-               exit(1);
+               return 1;
        }
        ext2fs_mark_super_dirty(fs);
+
+       return 0;
+}
+
+static int check_fsck_needed(ext2_filsys fs)
+{
+       if (fs->super->s_state & EXT2_VALID_FS)
+               return 0;
+       printf("\n%s\n", _(please_fsck));
+       if (mount_flags & EXT2_MF_READONLY)
+               printf(_("(and reboot afterwards!)\n"));
+       return 1;
+}
+
+static void request_dir_fsck_afterwards(ext2_filsys fs)
+{
+       static int requested;
+
+       if (requested++)
+               return;
+       fs->super->s_state &= ~EXT2_VALID_FS;
+       printf("\n%s\n", _(please_dir_fsck));
+       if (mount_flags & EXT2_MF_READONLY)
+               printf(_("(and reboot afterwards!)\n"));
 }
 
 static void request_fsck_afterwards(ext2_filsys fs)
@@ -354,15 +394,469 @@ static void request_fsck_afterwards(ext2_filsys fs)
                printf(_("(and reboot afterwards!)\n"));
 }
 
+/* Rewrite extents */
+static errcode_t rewrite_extents(ext2_filsys fs, ext2_ino_t ino,
+                                struct ext2_inode *inode)
+{
+       ext2_extent_handle_t    handle;
+       struct ext2fs_extent    extent;
+       int                     op = EXT2_EXTENT_ROOT;
+       errcode_t               errcode;
+
+       if (!(inode->i_flags & EXT4_EXTENTS_FL))
+               return 0;
+
+       errcode = ext2fs_extent_open(fs, ino, &handle);
+       if (errcode)
+               return errcode;
+
+       while (1) {
+               errcode = ext2fs_extent_get(handle, op, &extent);
+               if (errcode)
+                       break;
+
+               /* Root node is in the separately checksummed inode */
+               if (op == EXT2_EXTENT_ROOT) {
+                       op = EXT2_EXTENT_NEXT;
+                       continue;
+               }
+               op = EXT2_EXTENT_NEXT;
+
+               /* Only visit the first extent in each extent block */
+               if (extent.e_flags & EXT2_EXTENT_FLAGS_SECOND_VISIT)
+                       continue;
+               errcode = ext2fs_extent_replace(handle, 0, &extent);
+               if (errcode)
+                       break;
+       }
+
+       /* Ok if we run off the end */
+       if (errcode == EXT2_ET_EXTENT_NO_NEXT)
+               errcode = 0;
+       return errcode;
+}
+
+/*
+ * Rewrite directory blocks with checksums
+ */
+struct rewrite_dir_context {
+       char *buf;
+       errcode_t errcode;
+       ext2_ino_t dir;
+       int is_htree;
+};
+
+static int rewrite_dir_block(ext2_filsys fs,
+                            blk64_t    *blocknr,
+                            e2_blkcnt_t blockcnt EXT2FS_ATTR((unused)),
+                            blk64_t    ref_block EXT2FS_ATTR((unused)),
+                            int        ref_offset EXT2FS_ATTR((unused)),
+                            void       *priv_data)
+{
+       struct ext2_dx_countlimit *dcl = NULL;
+       struct rewrite_dir_context *ctx = priv_data;
+       int dcl_offset, changed = 0;
+
+       ctx->errcode = ext2fs_read_dir_block4(fs, *blocknr, ctx->buf, 0,
+                                             ctx->dir);
+       if (ctx->errcode)
+               return BLOCK_ABORT;
+
+       /* if htree node... */
+       if (ctx->is_htree)
+               ext2fs_get_dx_countlimit(fs, (struct ext2_dir_entry *)ctx->buf,
+                                        &dcl, &dcl_offset);
+       if (dcl) {
+               if (!EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
+                               EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)) {
+                       /* Ensure limit is the max size */
+                       int max_entries = (fs->blocksize - dcl_offset) /
+                                         sizeof(struct ext2_dx_entry);
+                       if (ext2fs_le16_to_cpu(dcl->limit) != max_entries) {
+                               changed = 1;
+                               dcl->limit = ext2fs_cpu_to_le16(max_entries);
+                       }
+               } else {
+                       /* If htree block is full then rebuild the dir */
+                       if (ext2fs_le16_to_cpu(dcl->count) ==
+                           ext2fs_le16_to_cpu(dcl->limit)) {
+                               request_dir_fsck_afterwards(fs);
+                               return 0;
+                       }
+                       /*
+                        * Ensure dcl->limit is small enough to leave room for
+                        * the checksum tail.
+                        */
+                       int max_entries = (fs->blocksize - (dcl_offset +
+                                               sizeof(struct ext2_dx_tail))) /
+                                         sizeof(struct ext2_dx_entry);
+                       if (ext2fs_le16_to_cpu(dcl->limit) != max_entries)
+                               dcl->limit = ext2fs_cpu_to_le16(max_entries);
+                       /* Always rewrite checksum */
+                       changed = 1;
+               }
+       } else {
+               unsigned int rec_len, name_size;
+               char *top = ctx->buf + fs->blocksize;
+               struct ext2_dir_entry *de = (struct ext2_dir_entry *)ctx->buf;
+               struct ext2_dir_entry *last_de = NULL, *penultimate_de = NULL;
+
+               /* Find last and penultimate dirent */
+               while ((char *)de < top) {
+                       penultimate_de = last_de;
+                       last_de = de;
+                       ctx->errcode = ext2fs_get_rec_len(fs, de, &rec_len);
+                       if (!ctx->errcode && !rec_len)
+                               ctx->errcode = EXT2_ET_DIR_CORRUPTED;
+                       if (ctx->errcode)
+                               return BLOCK_ABORT;
+                       de = (struct ext2_dir_entry *)(((char *)de) + rec_len);
+               }
+               ctx->errcode = ext2fs_get_rec_len(fs, last_de, &rec_len);
+               if (ctx->errcode)
+                       return BLOCK_ABORT;
+               name_size = ext2fs_dirent_name_len(last_de);
+
+               if (!EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
+                               EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)) {
+                       if (!penultimate_de)
+                               return 0;
+                       if (last_de->inode ||
+                           name_size ||
+                           rec_len != sizeof(struct ext2_dir_entry_tail))
+                               return 0;
+                       /*
+                        * The last dirent is unused and the right length to
+                        * have stored a checksum.  Erase it.
+                        */
+                       ctx->errcode = ext2fs_get_rec_len(fs, penultimate_de,
+                                                         &rec_len);
+                       if (!rec_len)
+                               ctx->errcode = EXT2_ET_DIR_CORRUPTED;
+                       if (ctx->errcode)
+                               return BLOCK_ABORT;
+                       ext2fs_set_rec_len(fs, rec_len +
+                                       sizeof(struct ext2_dir_entry_tail),
+                                       penultimate_de);
+                       changed = 1;
+               } else {
+                       unsigned csum_size = sizeof(struct ext2_dir_entry_tail);
+                       struct ext2_dir_entry_tail *t;
+
+                       /*
+                        * If the last dirent looks like the tail, just update
+                        * the checksum.
+                        */
+                       if (!last_de->inode &&
+                           rec_len == csum_size) {
+                               t = (struct ext2_dir_entry_tail *)last_de;
+                               t->det_reserved_name_len =
+                                               EXT2_DIR_NAME_LEN_CSUM;
+                               changed = 1;
+                               goto out;
+                       }
+                       if (name_size & 3)
+                               name_size = (name_size & ~3) + 4;
+                       /* If there's not enough space for the tail, e2fsck */
+                       if (rec_len <= (8 + name_size + csum_size)) {
+                               request_dir_fsck_afterwards(fs);
+                               return 0;
+                       }
+                       /* Shorten that last de and insert the tail */
+                       ext2fs_set_rec_len(fs, rec_len - csum_size, last_de);
+                       t = EXT2_DIRENT_TAIL(ctx->buf, fs->blocksize);
+                       ext2fs_initialize_dirent_tail(fs, t);
+
+                       /* Always update checksum */
+                       changed = 1;
+               }
+       }
+
+out:
+       if (!changed)
+               return 0;
+
+       ctx->errcode = ext2fs_write_dir_block4(fs, *blocknr, ctx->buf,
+                                              0, ctx->dir);
+       if (ctx->errcode)
+               return BLOCK_ABORT;
+
+       return 0;
+}
+
+static errcode_t rewrite_directory(ext2_filsys fs, ext2_ino_t dir,
+                                  struct ext2_inode *inode)
+{
+       errcode_t       retval;
+       struct rewrite_dir_context ctx;
+
+       retval = ext2fs_get_mem(fs->blocksize, &ctx.buf);
+       if (retval)
+               return retval;
+
+       ctx.is_htree = (inode->i_flags & EXT2_INDEX_FL);
+       ctx.dir = dir;
+       ctx.errcode = 0;
+       retval = ext2fs_block_iterate3(fs, dir, BLOCK_FLAG_READ_ONLY |
+                                               BLOCK_FLAG_DATA_ONLY,
+                                      0, rewrite_dir_block, &ctx);
+
+       ext2fs_free_mem(&ctx.buf);
+       if (retval)
+               return retval;
+
+       return ctx.errcode;
+}
+
+/*
+ * Forcibly set checksums in all inodes.
+ */
+static void rewrite_inodes(ext2_filsys fs)
+{
+       int length = EXT2_INODE_SIZE(fs->super);
+       struct ext2_inode *inode, *zero;
+       char            *ea_buf;
+       ext2_inode_scan scan;
+       errcode_t       retval;
+       ext2_ino_t      ino;
+       blk64_t         file_acl_block;
+       int             inode_dirty;
+
+       if (fs->super->s_creator_os != EXT2_OS_LINUX)
+               return;
+
+       retval = ext2fs_open_inode_scan(fs, 0, &scan);
+       if (retval) {
+               com_err("set_csum", retval, "while opening inode scan");
+               exit(1);
+       }
+
+       retval = ext2fs_get_mem(length, &inode);
+       if (retval) {
+               com_err("set_csum", retval, "while allocating memory");
+               exit(1);
+       }
+
+       retval = ext2fs_get_memzero(length, &zero);
+       if (retval) {
+               com_err("set_csum", retval, "while allocating memory");
+               exit(1);
+       }
+
+       retval = ext2fs_get_mem(fs->blocksize, &ea_buf);
+       if (retval) {
+               com_err("set_csum", retval, "while allocating memory");
+               exit(1);
+       }
+
+       do {
+               retval = ext2fs_get_next_inode_full(scan, &ino, inode, length);
+               if (retval) {
+                       com_err("set_csum", retval, "while getting next inode");
+                       exit(1);
+               }
+               if (!ino)
+                       break;
+               if (ext2fs_test_inode_bitmap2(fs->inode_map, ino)) {
+                       inode_dirty = 1;
+               } else {
+                       if (memcmp(inode, zero, length) != 0) {
+                               memset(inode, 0, length);
+                               inode_dirty = 1;
+                       } else {
+                               inode_dirty = 0;
+                       }
+               }
+
+               if (inode_dirty) {
+                       retval = ext2fs_write_inode_full(fs, ino, inode,
+                                                        length);
+                       if (retval) {
+                               com_err("set_csum", retval, "while writing "
+                                       "inode");
+                               exit(1);
+                       }
+               }
+
+               retval = rewrite_extents(fs, ino, inode);
+               if (retval) {
+                       com_err("rewrite_extents", retval,
+                               "while rewriting extents");
+                       exit(1);
+               }
+
+               if (LINUX_S_ISDIR(inode->i_mode)) {
+                       retval = rewrite_directory(fs, ino, inode);
+                       if (retval) {
+                               com_err("rewrite_directory", retval,
+                                       "while rewriting directories");
+                               exit(1);
+                       }
+               }
+
+               file_acl_block = ext2fs_file_acl_block(fs, inode);
+               if (!file_acl_block)
+                       continue;
+               retval = ext2fs_read_ext_attr3(fs, file_acl_block, ea_buf, ino);
+               if (retval) {
+                       com_err("rewrite_eablock", retval,
+                               "while rewriting extended attribute");
+                       exit(1);
+               }
+               retval = ext2fs_write_ext_attr3(fs, file_acl_block, ea_buf,
+                                               ino);
+               if (retval) {
+                       com_err("rewrite_eablock", retval,
+                               "while rewriting extended attribute");
+                       exit(1);
+               }
+       } while (ino);
+
+       ext2fs_free_mem(&zero);
+       ext2fs_free_mem(&inode);
+       ext2fs_free_mem(&ea_buf);
+       ext2fs_close_inode_scan(scan);
+}
+
+static void rewrite_metadata_checksums(ext2_filsys fs)
+{
+       dgrp_t i;
+
+       fs->flags |= EXT2_FLAG_IGNORE_CSUM_ERRORS;
+       ext2fs_init_csum_seed(fs);
+       for (i = 0; i < fs->group_desc_count; i++)
+               ext2fs_group_desc_csum_set(fs, i);
+       ext2fs_read_bitmaps(fs);
+       rewrite_inodes(fs);
+       ext2fs_mark_ib_dirty(fs);
+       ext2fs_mark_bb_dirty(fs);
+       ext2fs_mmp_update2(fs, 1);
+       fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
+       fs->flags &= ~EXT2_FLAG_IGNORE_CSUM_ERRORS;
+       if (EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
+                                      EXT4_FEATURE_RO_COMPAT_METADATA_CSUM))
+               fs->super->s_checksum_type = EXT2_CRC32C_CHKSUM;
+       else
+               fs->super->s_checksum_type = 0;
+       ext2fs_mark_super_dirty(fs);
+}
+
+static void enable_uninit_bg(ext2_filsys fs)
+{
+       struct ext2_group_desc *gd;
+       dgrp_t i;
+
+       for (i = 0; i < fs->group_desc_count; i++) {
+               gd = ext2fs_group_desc(fs, fs->group_desc, i);
+               gd->bg_itable_unused = 0;
+               gd->bg_flags = EXT2_BG_INODE_ZEROED;
+               ext2fs_group_desc_csum_set(fs, i);
+       }
+       fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
+}
+
+static errcode_t zero_empty_inodes(ext2_filsys fs)
+{
+       int length = EXT2_INODE_SIZE(fs->super);
+       struct ext2_inode *inode;
+       ext2_inode_scan scan;
+       errcode_t       retval;
+       ext2_ino_t      ino;
+
+       retval = ext2fs_open_inode_scan(fs, 0, &scan);
+       if (retval)
+               goto out;
+
+       retval = ext2fs_get_mem(length, &inode);
+       if (retval)
+               goto out;
+
+       do {
+               retval = ext2fs_get_next_inode_full(scan, &ino, inode, length);
+               if (retval)
+                       goto out;
+               if (!ino)
+                       break;
+               if (!ext2fs_test_inode_bitmap2(fs->inode_map, ino)) {
+                       memset(inode, 0, length);
+                       retval = ext2fs_write_inode_full(fs, ino, inode,
+                                                        length);
+                       if (retval)
+                               goto out;
+               }
+       } while (1);
+
+out:
+       ext2fs_free_mem(&inode);
+       ext2fs_close_inode_scan(scan);
+       return retval;
+}
+
+static void disable_uninit_bg(ext2_filsys fs, __u32 csum_feature_flag)
+{
+       struct ext2_group_desc *gd;
+       dgrp_t i;
+       errcode_t retval;
+       blk64_t b, c, d;
+       int has_super;
+
+       /* Load bitmaps to ensure that the uninit ones get written out */
+       fs->super->s_feature_ro_compat |= csum_feature_flag;
+       ext2fs_read_bitmaps(fs);
+       ext2fs_mark_ib_dirty(fs);
+       ext2fs_mark_bb_dirty(fs);
+       fs->super->s_feature_ro_compat &= ~csum_feature_flag;
+
+       /* If we're only turning off uninit_bg, zero the inodes */
+       if (csum_feature_flag == EXT4_FEATURE_RO_COMPAT_GDT_CSUM) {
+               retval = zero_empty_inodes(fs);
+               if (retval) {
+                       com_err("disable_uninit_bg", retval,
+                               "while zeroing unused inodes");
+                       request_fsck_afterwards(fs);
+               }
+       }
+
+       /* The bbitmap is zeroed; we must mark group metadata blocks in use */
+       for (i = 0; i < fs->group_desc_count; i++) {
+               b = ext2fs_block_bitmap_loc(fs, i);
+               ext2fs_mark_block_bitmap2(fs->block_map, EXT2FS_B2C(fs, b));
+               b = ext2fs_inode_bitmap_loc(fs, i);
+               ext2fs_mark_block_bitmap2(fs->block_map, EXT2FS_B2C(fs, b));
+
+               retval = ext2fs_super_and_bgd_loc2(fs, i, &b, &c, &d, NULL);
+               if (retval == 0 && b)
+                       ext2fs_mark_block_bitmap2(fs->block_map,
+                                                 EXT2FS_B2C(fs, b));
+               if (retval == 0 && c)
+                       ext2fs_mark_block_bitmap2(fs->block_map,
+                                                 EXT2FS_B2C(fs, c));
+               if (retval == 0 && d)
+                       ext2fs_mark_block_bitmap2(fs->block_map,
+                                                 EXT2FS_B2C(fs, d));
+               if (retval) {
+                       com_err("disable_uninit_bg", retval,
+                               "while initializing block bitmaps");
+                       request_fsck_afterwards(fs);
+               }
+
+               gd = ext2fs_group_desc(fs, fs->group_desc, i);
+               gd->bg_itable_unused = 0;
+               gd->bg_flags = 0;
+               ext2fs_group_desc_csum_set(fs, i);
+       }
+       fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
+       ext2fs_mark_super_dirty(fs);
+}
+
 /*
  * Update the feature set as provided by the user.
  */
-static void update_feature_set(ext2_filsys fs, char *features)
+static int update_feature_set(ext2_filsys fs, char *features)
 {
        struct ext2_super_block *sb = fs->super;
-       struct ext2_group_desc *gd;
        __u32           old_features[3];
-       int             i, type_err;
+       int             type_err;
        unsigned int    mask_err;
 
 #define FEATURE_ON(type, mask) (!(old_features[(type)] & (mask)) && \
@@ -393,7 +887,7 @@ static void update_feature_set(ext2_filsys fs, char *features)
                        fprintf(stderr, _("Setting filesystem feature '%s' "
                                          "not supported.\n"),
                                e2p_feature2string(type_err, mask_err));
-               exit(1);
+               return 1;
        }
 
        if (FEATURE_OFF(E2P_FEATURE_COMPAT, EXT3_FEATURE_COMPAT_HAS_JOURNAL)) {
@@ -403,22 +897,89 @@ static void update_feature_set(ext2_filsys fs, char *features)
                                "cleared when the filesystem is\n"
                                "unmounted or mounted "
                                "read-only.\n"), stderr);
-                       exit(1);
+                       return 1;
                }
                if (sb->s_feature_incompat &
                    EXT3_FEATURE_INCOMPAT_RECOVER) {
                        fputs(_("The needs_recovery flag is set.  "
                                "Please run e2fsck before clearing\n"
                                "the has_journal flag.\n"), stderr);
-                       exit(1);
+                       return 1;
                }
                if (sb->s_journal_inum) {
-                       remove_journal_inode(fs);
+                       if (remove_journal_inode(fs))
+                               return 1;
                }
                if (sb->s_journal_dev) {
-                       remove_journal_device(fs);
+                       if (remove_journal_device(fs))
+                               return 1;
                }
        }
+       if (FEATURE_ON(E2P_FEATURE_INCOMPAT, EXT4_FEATURE_INCOMPAT_MMP)) {
+               int error;
+
+               if ((mount_flags & EXT2_MF_MOUNTED) ||
+                   (mount_flags & EXT2_MF_READONLY)) {
+                       fputs(_("The multiple mount protection feature can't\n"
+                               "be set if the filesystem is mounted or\n"
+                               "read-only.\n"), stderr);
+                       return 1;
+               }
+
+               error = ext2fs_mmp_init(fs);
+               if (error) {
+                       fputs(_("\nError while enabling multiple mount "
+                               "protection feature."), stderr);
+                       return 1;
+               }
+
+               /*
+                * We want to update group desc with the new free blocks count
+                */
+               fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
+
+               printf(_("Multiple mount protection has been enabled "
+                        "with update interval %ds.\n"),
+                      sb->s_mmp_update_interval);
+       }
+
+       if (FEATURE_OFF(E2P_FEATURE_INCOMPAT, EXT4_FEATURE_INCOMPAT_MMP)) {
+               int error;
+
+               if (mount_flags & EXT2_MF_READONLY) {
+                       fputs(_("The multiple mount protection feature cannot\n"
+                               "be disabled if the filesystem is readonly.\n"),
+                               stderr);
+                       return 1;
+               }
+
+               error = ext2fs_read_bitmaps(fs);
+               if (error) {
+                       fputs(_("Error while reading bitmaps\n"), stderr);
+                       return 1;
+               }
+
+               error = ext2fs_mmp_read(fs, sb->s_mmp_block, NULL);
+               if (error) {
+                       struct mmp_struct *mmp_cmp = fs->mmp_cmp;
+
+                       if (error == EXT2_ET_MMP_MAGIC_INVALID)
+                               printf(_("Magic number in MMP block does not "
+                                        "match. expected: %x, actual: %x\n"),
+                                        EXT4_MMP_MAGIC, mmp_cmp->mmp_magic);
+                       else
+                               com_err(program_name, error,
+                                       _("while reading MMP block."));
+                       goto mmp_error;
+               }
+
+               /* We need to force out the group descriptors as well */
+               fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
+               ext2fs_block_alloc_stats2(fs, sb->s_mmp_block, -1);
+mmp_error:
+               sb->s_mmp_block = 0;
+               sb->s_mmp_update_interval = 0;
+       }
 
        if (FEATURE_ON(E2P_FEATURE_COMPAT, EXT3_FEATURE_COMPAT_HAS_JOURNAL)) {
                /*
@@ -443,7 +1004,7 @@ static void update_feature_set(ext2_filsys fs, char *features)
                        fputs(_("Clearing the flex_bg flag would "
                                "cause the the filesystem to be\n"
                                "inconsistent.\n"), stderr);
-                       exit(1);
+                       return 1;
                }
        }
 
@@ -455,41 +1016,73 @@ static void update_feature_set(ext2_filsys fs, char *features)
                                "cleared when the filesystem is\n"
                                "unmounted or mounted "
                                "read-only.\n"), stderr);
-                       exit(1);
+                       return 1;
                }
        }
 
        if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
-                      EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) {
-               for (i = 0; i < fs->group_desc_count; i++) {
-                       gd = ext2fs_group_desc(fs, fs->group_desc, i);
-                       gd->bg_itable_unused = 0;
-                       gd->bg_flags = EXT2_BG_INODE_ZEROED;
-                       ext2fs_group_desc_csum_set(fs, i);
-               }
-               fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
+                      EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)) {
+               if (check_fsck_needed(fs))
+                       exit(1);
+               rewrite_checksums = 1;
+               /* metadata_csum supersedes uninit_bg */
+               fs->super->s_feature_ro_compat &=
+                       ~EXT4_FEATURE_RO_COMPAT_GDT_CSUM;
+
+               /* if uninit_bg was previously off, rewrite group desc */
+               if (!(old_features[E2P_FEATURE_RO_INCOMPAT] &
+                     EXT4_FEATURE_RO_COMPAT_GDT_CSUM))
+                       enable_uninit_bg(fs);
+
+               /*
+                * Since metadata_csum supersedes uninit_bg, pretend like
+                * uninit_bg has been off all along.
+                */
+               old_features[E2P_FEATURE_RO_INCOMPAT] &=
+                       ~EXT4_FEATURE_RO_COMPAT_GDT_CSUM;
        }
 
        if (FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
-                       EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) {
-               for (i = 0; i < fs->group_desc_count; i++) {
-                       gd = ext2fs_group_desc(fs, fs->group_desc, i);
-                       if ((gd->bg_flags & EXT2_BG_INODE_ZEROED) == 0) {
-                               /* 
-                                * XXX what we really should do is zap
-                                * uninitialized inode tables instead.
-                                */
-                               request_fsck_afterwards(fs);
-                               break;
-                       }
-                       gd->bg_itable_unused = 0;
-                       gd->bg_flags = 0;
-                       gd->bg_checksum = 0;
-               }
-               fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
+                       EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)) {
+               if (check_fsck_needed(fs))
+                       exit(1);
+               rewrite_checksums = 1;
+               /*
+                * If we're turning off metadata_csum and not turning on
+                * uninit_bg, rewrite group desc.
+                */
+               if (!(fs->super->s_feature_ro_compat &
+                     EXT4_FEATURE_RO_COMPAT_GDT_CSUM))
+                       disable_uninit_bg(fs,
+                               EXT4_FEATURE_RO_COMPAT_METADATA_CSUM);
+               else
+                       /*
+                        * metadata_csum previously provided uninit_bg, so if
+                        * we're also setting the uninit_bg feature bit,
+                        * pretend like it was previously enabled.  Checksums
+                        * will be rewritten with crc16 later.
+                        */
+                       old_features[E2P_FEATURE_RO_INCOMPAT] |=
+                               EXT4_FEATURE_RO_COMPAT_GDT_CSUM;
        }
 
        if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
+                      EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) {
+               /* Do not enable uninit_bg when metadata_csum enabled */
+               if (fs->super->s_feature_ro_compat &
+                   EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)
+                       fs->super->s_feature_ro_compat &=
+                               ~EXT4_FEATURE_RO_COMPAT_GDT_CSUM;
+               else
+                       enable_uninit_bg(fs);
+       }
+
+       if (FEATURE_OFF(E2P_FEATURE_RO_INCOMPAT,
+                       EXT4_FEATURE_RO_COMPAT_GDT_CSUM))
+               disable_uninit_bg(fs,
+                               EXT4_FEATURE_RO_COMPAT_GDT_CSUM);
+
+       if (FEATURE_ON(E2P_FEATURE_RO_INCOMPAT,
                                EXT4_FEATURE_RO_COMPAT_QUOTA)) {
                /*
                 * Set the Q_flag here and handle the quota options in the code
@@ -540,12 +1133,14 @@ static void update_feature_set(ext2_filsys fs, char *features)
            (old_features[E2P_FEATURE_INCOMPAT] != sb->s_feature_incompat) ||
            (old_features[E2P_FEATURE_RO_INCOMPAT] != sb->s_feature_ro_compat))
                ext2fs_mark_super_dirty(fs);
+
+       return 0;
 }
 
 /*
  * Add a journal to the filesystem.
  */
-static void add_journal(ext2_filsys fs)
+static int add_journal(ext2_filsys fs)
 {
        unsigned long journal_blocks;
        errcode_t       retval;
@@ -600,7 +1195,7 @@ static void add_journal(ext2_filsys fs)
                        fprintf(stderr, "\n");
                        com_err(program_name, retval,
                                _("\n\twhile trying to create journal file"));
-                       exit(1);
+                       return retval;
                } else
                        fputs(_("done\n"), stdout);
                /*
@@ -612,49 +1207,56 @@ static void add_journal(ext2_filsys fs)
        }
        print_check_message(fs->super->s_max_mnt_count,
                            fs->super->s_checkinterval);
-       return;
+       return 0;
 
 err:
        free(journal_device);
-       exit(1);
+       return 1;
 }
 
-void handle_quota_options(ext2_filsys fs)
+static void handle_quota_options(ext2_filsys fs)
 {
        quota_ctx_t qctx;
-       errcode_t retval;
        ext2_ino_t qf_ino;
 
        if (!usrquota && !grpquota)
                /* Nothing to do. */
                return;
 
-       init_quota_context(&qctx, fs, -1);
+       quota_init_context(&qctx, fs, -1);
+
+       if (usrquota == QOPT_ENABLE || grpquota == QOPT_ENABLE)
+               quota_compute_usage(qctx);
 
        if (usrquota == QOPT_ENABLE && !fs->super->s_usr_quota_inum) {
-               if ((qf_ino = quota_file_exists(fs, USRQUOTA, QFMT_VFS_V1)) > 0)
-                       set_sb_quota_inum(fs, qf_ino, USRQUOTA);
-               else
-                       write_quota_inode(qctx, USRQUOTA);
+               if ((qf_ino = quota_file_exists(fs, USRQUOTA,
+                                               QFMT_VFS_V1)) > 0)
+                       quota_update_limits(qctx, qf_ino, USRQUOTA);
+               quota_write_inode(qctx, USRQUOTA);
        } else if (usrquota == QOPT_DISABLE) {
-               remove_quota_inode(fs, USRQUOTA);
+               quota_remove_inode(fs, USRQUOTA);
        }
 
        if (grpquota == QOPT_ENABLE && !fs->super->s_grp_quota_inum) {
-               if ((qf_ino = quota_file_exists(fs, GRPQUOTA, QFMT_VFS_V1)) > 0)
-                       set_sb_quota_inum(fs, qf_ino, GRPQUOTA);
-               else
-                       write_quota_inode(qctx, GRPQUOTA);
+               if ((qf_ino = quota_file_exists(fs, GRPQUOTA,
+                                               QFMT_VFS_V1)) > 0)
+                       quota_update_limits(qctx, qf_ino, GRPQUOTA);
+               quota_write_inode(qctx, GRPQUOTA);
        } else if (grpquota == QOPT_DISABLE) {
-               remove_quota_inode(fs, GRPQUOTA);
+               quota_remove_inode(fs, GRPQUOTA);
        }
 
-       release_quota_context(&qctx);
+       quota_release_context(&qctx);
 
        if ((usrquota == QOPT_ENABLE) || (grpquota == QOPT_ENABLE)) {
+               fprintf(stderr, _("\nWarning: the quota feature is still "
+                                 "under development\n"
+                                 "See https://ext4.wiki.kernel.org/"
+                                 "index.php/Quota for more information\n\n"));
                fs->super->s_feature_ro_compat |= EXT4_FEATURE_RO_COMPAT_QUOTA;
                ext2fs_mark_super_dirty(fs);
-       } else if ((usrquota == QOPT_DISABLE) && (grpquota == QOPT_DISABLE)) {
+       } else if (!fs->super->s_usr_quota_inum &&
+                  !fs->super->s_grp_quota_inum) {
                fs->super->s_feature_ro_compat &= ~EXT4_FEATURE_RO_COMPAT_QUOTA;
                ext2fs_mark_super_dirty(fs);
        }
@@ -662,9 +1264,9 @@ void handle_quota_options(ext2_filsys fs)
        return;
 }
 
-void parse_quota_opts(const char *opts)
+static void parse_quota_opts(const char *opts)
 {
-       char    *buf, *token, *next, *p, *arg;
+       char    *buf, *token, *next, *p;
        int     len;
 
        len = strlen(opts);
@@ -916,7 +1518,6 @@ static void parse_tune2fs_options(int argc, char **argv)
                        mntopts_cmd = optarg;
                        open_flag = EXT2_FLAG_RW;
                        break;
-                       
                case 'O':
                        if (features_cmd) {
                                com_err(program_name, 0,
@@ -1008,7 +1609,7 @@ static void parse_tune2fs_options(int argc, char **argv)
                *io_options++ = 0;
        device_name = blkid_get_devname(NULL, argv[optind], NULL);
        if (!device_name) {
-               com_err("tune2fs", 0, _("Unable to resolve '%s'"),
+               com_err(program_name, 0, _("Unable to resolve '%s'"),
                        argv[optind]);
                exit(1);
        }
@@ -1035,7 +1636,7 @@ void do_findfs(int argc, char **argv)
 }
 #endif
 
-static void parse_extended_opts(ext2_filsys fs, const char *opts)
+static int parse_extended_opts(ext2_filsys fs, const char *opts)
 {
        char    *buf, *token, *next, *p, *arg;
        int     len, hash_alg;
@@ -1046,7 +1647,7 @@ static void parse_extended_opts(ext2_filsys fs, const char *opts)
        if (!buf) {
                fprintf(stderr,
                        _("Couldn't allocate memory to parse options!\n"));
-               exit(1);
+               return 1;
        }
        strcpy(buf, opts);
        for (token = buf; token && *token; token = next) {
@@ -1061,7 +1662,40 @@ static void parse_extended_opts(ext2_filsys fs, const char *opts)
                        *arg = 0;
                        arg++;
                }
-               if (!strcmp(token, "test_fs")) {
+               if (strcmp(token, "clear-mmp") == 0 ||
+                   strcmp(token, "clear_mmp") == 0) {
+                       clear_mmp = 1;
+               } else if (strcmp(token, "mmp_update_interval") == 0) {
+                       unsigned long intv;
+                       if (!arg) {
+                               r_usage++;
+                               continue;
+                       }
+                       intv = strtoul(arg, &p, 0);
+                       if (*p) {
+                               fprintf(stderr,
+                                       _("Invalid mmp_update_interval: %s\n"),
+                                       arg);
+                               r_usage++;
+                               continue;
+                       }
+                       if (intv == 0) {
+                               intv = EXT4_MMP_UPDATE_INTERVAL;
+                       } else if (intv > EXT4_MMP_MAX_UPDATE_INTERVAL) {
+                               fprintf(stderr,
+                                       _("mmp_update_interval too big: %lu\n"),
+                                       intv);
+                               r_usage++;
+                               continue;
+                       }
+                       printf(P_("Setting multiple mount protection update "
+                                 "interval to %lu second\n",
+                                 "Setting multiple mount protection update "
+                                 "interval to %lu seconds\n", intv),
+                              intv);
+                       fs->super->s_mmp_update_interval = intv;
+                       ext2fs_mark_super_dirty(fs);
+               } else if (!strcmp(token, "test_fs")) {
                        fs->super->s_flags |= EXT2_FLAGS_TEST_FILESYS;
                        printf("Setting test filesystem flag\n");
                        ext2fs_mark_super_dirty(fs);
@@ -1077,7 +1711,7 @@ static void parse_extended_opts(ext2_filsys fs, const char *opts)
                        stride = strtoul(arg, &p, 0);
                        if (*p) {
                                fprintf(stderr,
-                                      _("Invalid RAID stride: %s\n"),
+                                       _("Invalid RAID stride: %s\n"),
                                        arg);
                                r_usage++;
                                continue;
@@ -1137,6 +1771,7 @@ static void parse_extended_opts(ext2_filsys fs, const char *opts)
                        "and may take an argument which\n"
                        "\tis set off by an equals ('=') sign.\n\n"
                        "Valid extended options are:\n"
+                       "\tclear_mmp\n"
                        "\thash_alg=<hash algorithm>\n"
                        "\tmount_opts=<extended default mount options>\n"
                        "\tstride=<RAID per-disk chunk size in blocks>\n"
@@ -1144,9 +1779,11 @@ static void parse_extended_opts(ext2_filsys fs, const char *opts)
                        "\ttest_fs\n"
                        "\t^test_fs\n"));
                free(buf);
-               exit(1);
+               return 1;
        }
        free(buf);
+
+       return 0;
 }
 
 /*
@@ -1202,10 +1839,10 @@ static int get_move_bitmaps(ext2_filsys fs, int new_ino_blks_per_grp,
        return 0;
 }
 
-static int ext2fs_is_meta_block(ext2_filsys fs, blk_t blk)
+static int ext2fs_is_meta_block(ext2_filsys fs, blk64_t blk)
 {
        dgrp_t group;
-       group = ext2fs_group_of_blk(fs, blk);
+       group = ext2fs_group_of_blk2(fs, blk);
        if (ext2fs_block_bitmap_loc(fs, group) == blk)
                return 1;
        if (ext2fs_inode_bitmap_loc(fs, group) == blk)
@@ -1213,9 +1850,9 @@ static int ext2fs_is_meta_block(ext2_filsys fs, blk_t blk)
        return 0;
 }
 
-static int ext2fs_is_block_in_group(ext2_filsys fs, dgrp_t group, blk_t blk)
+static int ext2fs_is_block_in_group(ext2_filsys fs, dgrp_t group, blk64_t blk)
 {
-       blk_t start_blk, end_blk;
+       blk64_t start_blk, end_blk;
        start_blk = fs->super->s_first_data_block +
                        EXT2_BLOCKS_PER_GROUP(fs->super) * group;
        /*
@@ -1255,7 +1892,7 @@ static int move_block(ext2_filsys fs, ext2fs_block_bitmap bmap)
                         * the respective fs metadata pointers. Otherwise
                         * fail
                         */
-                       group = ext2fs_group_of_blk(fs, blk);
+                       group = ext2fs_group_of_blk2(fs, blk);
                        goal = ext2fs_group_first_block2(fs, group);
                        meta_data = 1;
 
@@ -1373,14 +2010,15 @@ static int inode_scan_and_fix(ext2_filsys fs, ext2fs_block_bitmap bmap)
                 * Do we need to fix this ??
                 */
 
-               if (ext2fs_file_acl_block(&inode) &&
+               if (ext2fs_file_acl_block(fs, &inode) &&
                    ext2fs_test_block_bitmap2(bmap,
-                                             ext2fs_file_acl_block(&inode))) {
-                       blk = translate_block(ext2fs_file_acl_block(&inode));
+                                       ext2fs_file_acl_block(fs, &inode))) {
+                       blk = translate_block(ext2fs_file_acl_block(fs,
+                                                                   &inode));
                        if (!blk)
                                continue;
 
-                       ext2fs_file_acl_block_set(&inode, blk);
+                       ext2fs_file_acl_block_set(fs, &inode, blk);
 
                        /*
                         * Write the inode to disk so that inode table
@@ -1391,7 +2029,7 @@ static int inode_scan_and_fix(ext2_filsys fs, ext2fs_block_bitmap bmap)
                                goto err_out;
                }
 
-               if (!ext2fs_inode_has_valid_blocks(&inode))
+               if (!ext2fs_inode_has_valid_blocks2(fs, &inode))
                        continue;
 
                retval = ext2fs_block_iterate3(fs, ino, 0, block_buf,
@@ -1415,7 +2053,7 @@ err_out:
 static int group_desc_scan_and_fix(ext2_filsys fs, ext2fs_block_bitmap bmap)
 {
        dgrp_t i;
-       blk_t blk, new_blk;
+       blk64_t blk, new_blk;
 
        for (i = 0; i < fs->group_desc_count; i++) {
                blk = ext2fs_block_bitmap_loc(fs, i);
@@ -1599,7 +2237,7 @@ static int resize_inode(ext2_filsys fs, unsigned long new_size)
        }
        retval = ext2fs_read_block_bitmap(fs);
        if (retval) {
-               fputs(_("Failed to read blockbitmap\n"), stderr);
+               fputs(_("Failed to read block bitmap\n"), stderr);
                return retval;
        }
        INIT_LIST_HEAD(&blk_move_list);
@@ -1735,12 +2373,14 @@ int main(int argc, char **argv)
        ext2_filsys fs;
        struct ext2_super_block *sb;
        io_manager io_ptr, io_ptr_orig = NULL;
+       int rc = 0;
 
 #ifdef ENABLE_NLS
        setlocale(LC_MESSAGES, "");
        setlocale(LC_CTYPE, "");
        bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
        textdomain(NLS_CAT_NAME);
+       set_com_err_gettext(gettext);
 #endif
        if (argc && *argv)
                program_name = *argv;
@@ -1764,16 +2404,40 @@ int main(int argc, char **argv)
                io_ptr = unix_io_manager;
 
 retry_open:
+       if ((open_flag & EXT2_FLAG_RW) == 0 || f_flag)
+               open_flag |= EXT2_FLAG_SKIP_MMP;
+
+       open_flag |= EXT2_FLAG_64BITS;
+
+       /* keep the filesystem struct around to dump MMP data */
+       open_flag |= EXT2_FLAG_NOFREE_ON_ERROR;
+
        retval = ext2fs_open2(device_name, io_options, open_flag,
                              0, 0, io_ptr, &fs);
        if (retval) {
-                       com_err(program_name, retval,
-                               _("while trying to open %s"),
+               com_err(program_name, retval,
+                       _("while trying to open %s"),
                        device_name);
-               fprintf(stderr,
-                       _("Couldn't find valid filesystem superblock.\n"));
+               if (retval == EXT2_ET_MMP_FSCK_ON ||
+                   retval == EXT2_ET_MMP_UNKNOWN_SEQ)
+                       dump_mmp_msg(fs->mmp_buf,
+                                    _("If you are sure the filesystem "
+                                      "is not in use on any node, run:\n"
+                                      "'tune2fs -f -E clear_mmp {device}'\n"));
+               else if (retval == EXT2_ET_MMP_FAILED)
+                       dump_mmp_msg(fs->mmp_buf, NULL);
+               else if (retval == EXT2_ET_MMP_MAGIC_INVALID)
+                       fprintf(stderr,
+                               _("MMP block magic is bad. Try to fix it by "
+                                 "running:\n'e2fsck -f %s'\n"), device_name);
+               else if (retval != EXT2_ET_MMP_FAILED)
+                       fprintf(stderr,
+                            _("Couldn't find valid filesystem superblock.\n"));
+
+               ext2fs_free(fs);
                exit(1);
        }
+       fs->default_bitmap_type = EXT2FS_BMAP64_RBTREE;
 
        if (I_flag && !io_ptr_orig) {
                /*
@@ -1784,12 +2448,20 @@ retry_open:
                if (new_inode_size == EXT2_INODE_SIZE(fs->super)) {
                        fprintf(stderr, _("The inode size is already %lu\n"),
                                new_inode_size);
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
                if (new_inode_size < EXT2_INODE_SIZE(fs->super)) {
                        fprintf(stderr, _("Shrinking the inode size is "
                                          "not supported\n"));
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
+               }
+               if (new_inode_size > fs->blocksize) {
+                       fprintf(stderr, _("Invalid inode size %lu (max %d)\n"),
+                               new_inode_size, fs->blocksize);
+                       rc = 1;
+                       goto closefs;
                }
 
                /*
@@ -1798,8 +2470,10 @@ retry_open:
                 */
                io_ptr_orig = io_ptr;
                retval = tune2fs_setup_tdb(device_name, &io_ptr);
-               if (retval)
-                       exit(1);
+               if (retval) {
+                       rc = 1;
+                       goto closefs;
+               }
                if (io_ptr != io_ptr_orig) {
                        ext2fs_close(fs);
                        goto retry_open;
@@ -1814,7 +2488,7 @@ retry_open:
                printf("%.*s\n", (int) sizeof(sb->s_volume_name),
                       sb->s_volume_name);
                remove_error_table(&et_ext2_error_table);
-               exit(0);
+               goto closefs;
        }
 
        retval = ext2fs_check_if_mounted(device_name, &mount_flags);
@@ -1822,7 +2496,8 @@ retry_open:
                com_err("ext2fs_check_if_mount", retval,
                        _("while determining whether %s is mounted."),
                        device_name);
-               exit(1);
+               rc = 1;
+               goto closefs;
        }
        /* Normally we only need to write out the superblock */
        fs->flags |= EXT2_FLAG_SUPER_ONLY;
@@ -1853,7 +2528,8 @@ retry_open:
                        com_err(program_name, 0,
                                _("interval between checks is too big (%lu)"),
                                interval);
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
                sb->s_checkinterval = interval;
                ext2fs_mark_super_dirty(fs);
@@ -1872,7 +2548,8 @@ retry_open:
                        com_err(program_name, 0,
                                _("reserved blocks count is too big (%llu)"),
                                reserved_blocks);
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
                ext2fs_r_blocks_count_set(sb, reserved_blocks);
                ext2fs_mark_super_dirty(fs);
@@ -1896,7 +2573,8 @@ retry_open:
        if (s_flag == 0) {
                fputs(_("\nClearing the sparse superflag not supported.\n"),
                      stderr);
-               exit(1);
+               rc = 1;
+               goto closefs;
        }
        if (T_flag) {
                sb->s_lastcheck = last_check_time;
@@ -1924,20 +2602,43 @@ retry_open:
                        sizeof(sb->s_last_mounted));
                ext2fs_mark_super_dirty(fs);
        }
-       if (mntopts_cmd)
-               update_mntopts(fs, mntopts_cmd);
-       if (features_cmd)
-               update_feature_set(fs, features_cmd);
-       if (extended_cmd)
-               parse_extended_opts(fs, extended_cmd);
-       if (journal_size || journal_device)
-               add_journal(fs);
+       if (mntopts_cmd) {
+               rc = update_mntopts(fs, mntopts_cmd);
+               if (rc)
+                       goto closefs;
+       }
+       if (features_cmd) {
+               rc = update_feature_set(fs, features_cmd);
+               if (rc)
+                       goto closefs;
+       }
+       if (extended_cmd) {
+               rc = parse_extended_opts(fs, extended_cmd);
+               if (rc)
+                       goto closefs;
+               if (clear_mmp && !f_flag) {
+                       fputs(_("Error in using clear_mmp. "
+                               "It must be used with -f\n"),
+                             stderr);
+                       goto closefs;
+               }
+       }
+       if (clear_mmp) {
+               rc = ext2fs_mmp_clear(fs);
+               goto closefs;
+       }
+       if (journal_size || journal_device) {
+               rc = add_journal(fs);
+               if (rc)
+                       goto closefs;
+       }
 
        if (Q_flag) {
                if (mount_flags & EXT2_MF_MOUNTED) {
                        fputs(_("The quota feature may only be changed when "
                                "the filesystem is unmounted.\n"), stderr);
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
                handle_quota_options(fs);
        }
@@ -1946,8 +2647,24 @@ retry_open:
                int set_csum = 0;
                dgrp_t i;
 
-               if (sb->s_feature_ro_compat &
-                   EXT4_FEATURE_RO_COMPAT_GDT_CSUM) {
+               if (EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
+                               EXT4_FEATURE_RO_COMPAT_METADATA_CSUM)) {
+                       /*
+                        * Changing the UUID requires rewriting all metadata,
+                        * which can race with a mounted fs.  Don't allow that.
+                        */
+                       if (mount_flags & EXT2_MF_MOUNTED) {
+                               fputs(_("The UUID may only be "
+                                       "changed when the filesystem is "
+                                       "unmounted.\n"), stderr);
+                               exit(1);
+                       }
+
+                       if (check_fsck_needed(fs))
+                               exit(1);
+               }
+
+               if (ext2fs_has_group_desc_csum(fs)) {
                        /*
                         * Determine if the block group checksums are
                         * correct so we know whether or not to set
@@ -1968,21 +2685,29 @@ retry_open:
                        uuid_generate(sb->s_uuid);
                } else if (uuid_parse(new_UUID, sb->s_uuid)) {
                        com_err(program_name, 0, _("Invalid UUID format\n"));
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
+               ext2fs_init_csum_seed(fs);
                if (set_csum) {
                        for (i = 0; i < fs->group_desc_count; i++)
                                ext2fs_group_desc_csum_set(fs, i);
                        fs->flags &= ~EXT2_FLAG_SUPER_ONLY;
                }
                ext2fs_mark_super_dirty(fs);
+               if (EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
+                               EXT4_FEATURE_RO_COMPAT_METADATA_CSUM))
+                       rewrite_checksums = 1;
        }
+       if (rewrite_checksums)
+               rewrite_metadata_checksums(fs);
        if (I_flag) {
                if (mount_flags & EXT2_MF_MOUNTED) {
                        fputs(_("The inode size may only be "
                                "changed when the filesystem is "
                                "unmounted.\n"), stderr);
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
                if (fs->super->s_feature_incompat &
                    EXT4_FEATURE_INCOMPAT_FLEX_BG) {
@@ -1990,7 +2715,8 @@ retry_open:
                                "filesystems with the flex_bg\n"
                                "feature enabled.\n"),
                              stderr);
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
                /*
                 * We want to update group descriptor also
@@ -2002,7 +2728,8 @@ retry_open:
                                                        new_inode_size);
                } else {
                        printf(_("Failed to change inode size\n"));
-                       exit(1);
+                       rc = 1;
+                       goto closefs;
                }
        }
 
@@ -2029,5 +2756,12 @@ retry_open:
        }
        free(device_name);
        remove_error_table(&et_ext2_error_table);
+
+closefs:
+       if (rc) {
+               ext2fs_mmp_stop(fs);
+               exit(1);
+       }
+
        return (ext2fs_close(fs) ? 1 : 0);
 }