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