misc/tune2fs.c

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