| /* |
| * main.c --- ext2 resizer main program |
| * |
| * Copyright (C) 1997, 1998 by Theodore Ts'o and |
| * PowerQuest, Inc. |
| * |
| * Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004 by Theodore Ts'o |
| * |
| * %Begin-Header% |
| * This file may be redistributed under the terms of the GNU Public |
| * License. |
| * %End-Header% |
| */ |
| |
| #define _LARGEFILE_SOURCE |
| #define _LARGEFILE64_SOURCE |
| |
| #ifdef HAVE_GETOPT_H |
| #include <getopt.h> |
| #else |
| extern char *optarg; |
| extern int optind; |
| #endif |
| #include <unistd.h> |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <fcntl.h> |
| |
| #include "e2p/e2p.h" |
| |
| #include "resize2fs.h" |
| |
| #include "../version.h" |
| |
| char *program_name, *device_name, *io_options; |
| |
| static void usage (char *prog) |
| { |
| fprintf (stderr, _("Usage: %s [-d debug_flags] [-f] [-F] [-p] " |
| "device [new_size]\n\n"), prog); |
| |
| exit (1); |
| } |
| |
| static errcode_t resize_progress_func(ext2_resize_t rfs, int pass, |
| unsigned long cur, unsigned long max) |
| { |
| ext2_sim_progmeter progress; |
| const char *label; |
| errcode_t retval; |
| |
| progress = (ext2_sim_progmeter) rfs->prog_data; |
| if (max == 0) |
| return 0; |
| if (cur == 0) { |
| if (progress) |
| ext2fs_progress_close(progress); |
| progress = 0; |
| switch (pass) { |
| case E2_RSZ_EXTEND_ITABLE_PASS: |
| label = _("Extending the inode table"); |
| break; |
| case E2_RSZ_BLOCK_RELOC_PASS: |
| label = _("Relocating blocks"); |
| break; |
| case E2_RSZ_INODE_SCAN_PASS: |
| label = _("Scanning inode table"); |
| break; |
| case E2_RSZ_INODE_REF_UPD_PASS: |
| label = _("Updating inode references"); |
| break; |
| case E2_RSZ_MOVE_ITABLE_PASS: |
| label = _("Moving inode table"); |
| break; |
| default: |
| label = _("Unknown pass?!?"); |
| break; |
| } |
| printf(_("Begin pass %d (max = %lu)\n"), pass, max); |
| retval = ext2fs_progress_init(&progress, label, 30, |
| 40, max, 0); |
| if (retval) |
| progress = 0; |
| rfs->prog_data = (void *) progress; |
| } |
| if (progress) |
| ext2fs_progress_update(progress, cur); |
| if (cur >= max) { |
| if (progress) |
| ext2fs_progress_close(progress); |
| progress = 0; |
| rfs->prog_data = 0; |
| } |
| return 0; |
| } |
| |
| static void determine_fs_stride(ext2_filsys fs) |
| { |
| unsigned int group; |
| unsigned long long sum; |
| unsigned int has_sb, prev_has_sb, num; |
| int i_stride, b_stride; |
| |
| if (fs->stride) |
| return; |
| num = 0; sum = 0; |
| for (group = 0; group < fs->group_desc_count; group++) { |
| has_sb = ext2fs_bg_has_super(fs, group); |
| if (group == 0 || has_sb != prev_has_sb) |
| goto next; |
| b_stride = fs->group_desc[group].bg_block_bitmap - |
| fs->group_desc[group-1].bg_block_bitmap - |
| fs->super->s_blocks_per_group; |
| i_stride = fs->group_desc[group].bg_inode_bitmap - |
| fs->group_desc[group-1].bg_inode_bitmap - |
| fs->super->s_blocks_per_group; |
| if (b_stride != i_stride || |
| b_stride < 0) |
| goto next; |
| |
| /* printf("group %d has stride %d\n", group, b_stride); */ |
| sum += b_stride; |
| num++; |
| |
| next: |
| prev_has_sb = has_sb; |
| } |
| |
| if (fs->group_desc_count > 12 && num < 3) |
| sum = 0; |
| |
| if (num) |
| fs->stride = sum / num; |
| else |
| fs->stride = 0; |
| |
| fs->super->s_raid_stride = fs->stride; |
| ext2fs_mark_super_dirty(fs); |
| |
| #if 0 |
| if (fs->stride) |
| printf("Using RAID stride of %d\n", fs->stride); |
| #endif |
| } |
| |
| int main (int argc, char ** argv) |
| { |
| errcode_t retval; |
| ext2_filsys fs; |
| int c; |
| int flags = 0; |
| int flush = 0; |
| int force = 0; |
| int io_flags = 0; |
| int fd, ret; |
| blk_t new_size = 0; |
| blk_t max_size = 0; |
| io_manager io_ptr; |
| char *new_size_str = 0; |
| int use_stride = -1; |
| #ifdef HAVE_FSTAT64 |
| struct stat64 st_buf; |
| #else |
| struct stat st_buf; |
| #endif |
| __s64 new_file_size; |
| unsigned int sys_page_size = 4096; |
| long sysval; |
| int len, mount_flags; |
| char *mtpt; |
| |
| #ifdef ENABLE_NLS |
| setlocale(LC_MESSAGES, ""); |
| setlocale(LC_CTYPE, ""); |
| bindtextdomain(NLS_CAT_NAME, LOCALEDIR); |
| textdomain(NLS_CAT_NAME); |
| #endif |
| |
| add_error_table(&et_ext2_error_table); |
| |
| fprintf (stderr, "resize2fs %s (%s)\n", |
| E2FSPROGS_VERSION, E2FSPROGS_DATE); |
| if (argc && *argv) |
| program_name = *argv; |
| |
| while ((c = getopt (argc, argv, "d:fFhpS:")) != EOF) { |
| switch (c) { |
| case 'h': |
| usage(program_name); |
| break; |
| case 'f': |
| force = 1; |
| break; |
| case 'F': |
| flush = 1; |
| break; |
| case 'd': |
| flags |= atoi(optarg); |
| break; |
| case 'p': |
| flags |= RESIZE_PERCENT_COMPLETE; |
| break; |
| case 'S': |
| use_stride = atoi(optarg); |
| break; |
| default: |
| usage(program_name); |
| } |
| } |
| if (optind == argc) |
| usage(program_name); |
| |
| device_name = argv[optind++]; |
| if (optind < argc) |
| new_size_str = argv[optind++]; |
| if (optind < argc) |
| usage(program_name); |
| |
| io_options = strchr(device_name, '?'); |
| if (io_options) |
| *io_options++ = 0; |
| |
| /* |
| * Figure out whether or not the device is mounted, and if it is |
| * where it is mounted. |
| */ |
| len=80; |
| while (1) { |
| mtpt = malloc(len); |
| if (!mtpt) |
| return ENOMEM; |
| mtpt[len-1] = 0; |
| retval = ext2fs_check_mount_point(device_name, &mount_flags, |
| mtpt, len); |
| if (retval) { |
| com_err("ext2fs_check_mount_point", retval, |
| _("while determining whether %s is mounted."), |
| device_name); |
| exit(1); |
| } |
| if (!(mount_flags & EXT2_MF_MOUNTED) || (mtpt[len-1] == 0)) |
| break; |
| free(mtpt); |
| len = 2 * len; |
| } |
| |
| #ifdef HAVE_OPEN64 |
| fd = open64(device_name, O_RDWR); |
| #else |
| fd = open(device_name, O_RDWR); |
| #endif |
| if (fd < 0) { |
| com_err("open", errno, _("while opening %s"), |
| device_name); |
| exit(1); |
| } |
| |
| #ifdef HAVE_FSTAT64 |
| ret = fstat64(fd, &st_buf); |
| #else |
| ret = fstat(fd, &st_buf); |
| #endif |
| if (ret < 0) { |
| com_err("open", errno, |
| _("while getting stat information for %s"), |
| device_name); |
| exit(1); |
| } |
| |
| if (flush) { |
| retval = ext2fs_sync_device(fd, 1); |
| if (retval) { |
| com_err(argv[0], retval, |
| _("while trying to flush %s"), |
| device_name); |
| exit(1); |
| } |
| } |
| |
| if (!S_ISREG(st_buf.st_mode )) { |
| close(fd); |
| fd = -1; |
| } |
| |
| if (flags & RESIZE_DEBUG_IO) { |
| io_ptr = test_io_manager; |
| test_io_backing_manager = unix_io_manager; |
| } else |
| io_ptr = unix_io_manager; |
| |
| if (!(mount_flags & EXT2_MF_MOUNTED)) |
| io_flags = EXT2_FLAG_RW | EXT2_FLAG_EXCLUSIVE; |
| retval = ext2fs_open2(device_name, io_options, io_flags, |
| 0, 0, io_ptr, &fs); |
| if (retval) { |
| com_err (program_name, retval, _("while trying to open %s"), |
| device_name); |
| printf (_("Couldn't find valid filesystem superblock.\n")); |
| exit (1); |
| } |
| /* |
| * Check for compatibility with the feature sets. We need to |
| * be more stringent than ext2fs_open(). |
| */ |
| if (fs->super->s_feature_compat & ~EXT2_LIB_FEATURE_COMPAT_SUPP) { |
| com_err(program_name, EXT2_ET_UNSUPP_FEATURE, |
| "(%s)", device_name); |
| exit(1); |
| } |
| |
| /* Determine the system page size if possible */ |
| #ifdef HAVE_SYSCONF |
| #if (!defined(_SC_PAGESIZE) && defined(_SC_PAGE_SIZE)) |
| #define _SC_PAGESIZE _SC_PAGE_SIZE |
| #endif |
| #ifdef _SC_PAGESIZE |
| sysval = sysconf(_SC_PAGESIZE); |
| if (sysval > 0) |
| sys_page_size = sysval; |
| #endif /* _SC_PAGESIZE */ |
| #endif /* HAVE_SYSCONF */ |
| |
| /* |
| * Get the size of the containing partition, and use this for |
| * defaults and for making sure the new filesystem doesn't |
| * exceed the partition size. |
| */ |
| retval = ext2fs_get_device_size(device_name, fs->blocksize, |
| &max_size); |
| if (retval) { |
| com_err(program_name, retval, |
| _("while trying to determine filesystem size")); |
| exit(1); |
| } |
| if (new_size_str) { |
| new_size = parse_num_blocks(new_size_str, |
| fs->super->s_log_block_size); |
| if (!new_size) { |
| com_err(program_name, 0, _("bad filesystem size - %s"), |
| new_size_str); |
| exit(1); |
| } |
| } else { |
| new_size = max_size; |
| /* Round down to an even multiple of a pagesize */ |
| if (sys_page_size > fs->blocksize) |
| new_size &= ~((sys_page_size / fs->blocksize)-1); |
| } |
| |
| if (use_stride >= 0) { |
| if (use_stride >= (int) fs->super->s_blocks_per_group) { |
| com_err(program_name, 0, |
| _("Invalid stride length")); |
| exit(1); |
| } |
| fs->stride = fs->super->s_raid_stride = use_stride; |
| ext2fs_mark_super_dirty(fs); |
| } else |
| determine_fs_stride(fs); |
| |
| /* |
| * If we are resizing a plain file, and it's not big enough, |
| * automatically extend it in a sparse fashion by writing the |
| * last requested block. |
| */ |
| new_file_size = ((__u64) new_size) * fs->blocksize; |
| if ((__u64) new_file_size > |
| (((__u64) 1) << (sizeof(st_buf.st_size)*8 - 1)) - 1) |
| fd = -1; |
| if ((new_file_size > st_buf.st_size) && |
| (fd > 0)) { |
| if ((ext2fs_llseek(fd, new_file_size-1, SEEK_SET) >= 0) && |
| (write(fd, "0", 1) == 1)) |
| max_size = new_size; |
| } |
| if (!force && (new_size > max_size)) { |
| fprintf(stderr, _("The containing partition (or device)" |
| " is only %u (%dk) blocks.\nYou requested a new size" |
| " of %u blocks.\n\n"), max_size, |
| fs->blocksize / 1024, new_size); |
| exit(1); |
| } |
| if (new_size == fs->super->s_blocks_count) { |
| fprintf(stderr, _("The filesystem is already %u blocks " |
| "long. Nothing to do!\n\n"), new_size); |
| exit(0); |
| } |
| if (mount_flags & EXT2_MF_MOUNTED) { |
| retval = online_resize_fs(fs, mtpt, &new_size, flags); |
| } else { |
| if (!force && ((fs->super->s_lastcheck < fs->super->s_mtime) || |
| (fs->super->s_state & EXT2_ERROR_FS) || |
| ((fs->super->s_state & EXT2_VALID_FS) == 0))) { |
| fprintf(stderr, |
| _("Please run 'e2fsck -f %s' first.\n\n"), |
| device_name); |
| exit(1); |
| } |
| printf("Resizing the filesystem on %s to %u (%dk) blocks.\n", |
| device_name, new_size, fs->blocksize / 1024); |
| retval = resize_fs(fs, &new_size, flags, |
| ((flags & RESIZE_PERCENT_COMPLETE) ? |
| resize_progress_func : 0)); |
| } |
| if (retval) { |
| com_err(program_name, retval, _("while trying to resize %s"), |
| device_name); |
| ext2fs_close (fs); |
| exit(1); |
| } |
| printf(_("The filesystem on %s is now %u blocks long.\n\n"), |
| device_name, new_size); |
| |
| if ((st_buf.st_size > new_file_size) && |
| (fd > 0)) { |
| #ifdef HAVE_FTRUNCATE64 |
| ftruncate64(fd, new_file_size); |
| #else |
| /* Only truncate if new_file_size doesn't overflow off_t */ |
| if (((off_t) new_file_size) == new_file_size) |
| ftruncate(fd, (off_t) new_file_size); |
| #endif |
| } |
| if (fd > 0) |
| close(fd); |
| remove_error_table(&et_ext2_error_table); |
| return (0); |
| } |