1 /*
2 * main.c --- ext2 resizer main program
3 *
4 * Copyright (C) 1997, 1998 by Theodore Ts'o and
5 * PowerQuest, Inc.
6 *
7 * Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004 by Theodore Ts'o
8 *
9 * %Begin-Header%
10 * This file may be redistributed under the terms of the GNU Public
11 * License.
12 * %End-Header%
13 */
14
15 #define _LARGEFILE_SOURCE
16 #define _LARGEFILE64_SOURCE
17
18 #ifdef HAVE_GETOPT_H
19 #include <getopt.h>
20 #else
21 extern char *optarg;
22 extern int optind;
23 #endif
24 #include <unistd.h>
25 #ifdef HAVE_STDLIB_H
26 #include <stdlib.h>
27 #endif
28 #include <sys/types.h>
29 #include <sys/stat.h>
30 #include <fcntl.h>
31
32 #include "e2p/e2p.h"
33
34 #include "resize2fs.h"
35
36 #include "../version.h"
37
38 char *program_name, *device_name, *io_options;
39
usage(char * prog)40 static void usage (char *prog)
41 {
42 fprintf (stderr, _("Usage: %s [-d debug_flags] [-f] [-F] [-M] [-P] "
43 "[-p] device [new_size]\n\n"), prog);
44
45 exit (1);
46 }
47
resize_progress_func(ext2_resize_t rfs,int pass,unsigned long cur,unsigned long max)48 static errcode_t resize_progress_func(ext2_resize_t rfs, int pass,
49 unsigned long cur, unsigned long max)
50 {
51 ext2_sim_progmeter progress;
52 const char *label;
53 errcode_t retval;
54
55 progress = (ext2_sim_progmeter) rfs->prog_data;
56 if (max == 0)
57 return 0;
58 if (cur == 0) {
59 if (progress)
60 ext2fs_progress_close(progress);
61 progress = 0;
62 switch (pass) {
63 case E2_RSZ_EXTEND_ITABLE_PASS:
64 label = _("Extending the inode table");
65 break;
66 case E2_RSZ_BLOCK_RELOC_PASS:
67 label = _("Relocating blocks");
68 break;
69 case E2_RSZ_INODE_SCAN_PASS:
70 label = _("Scanning inode table");
71 break;
72 case E2_RSZ_INODE_REF_UPD_PASS:
73 label = _("Updating inode references");
74 break;
75 case E2_RSZ_MOVE_ITABLE_PASS:
76 label = _("Moving inode table");
77 break;
78 default:
79 label = _("Unknown pass?!?");
80 break;
81 }
82 printf(_("Begin pass %d (max = %lu)\n"), pass, max);
83 retval = ext2fs_progress_init(&progress, label, 30,
84 40, max, 0);
85 if (retval)
86 progress = 0;
87 rfs->prog_data = (void *) progress;
88 }
89 if (progress)
90 ext2fs_progress_update(progress, cur);
91 if (cur >= max) {
92 if (progress)
93 ext2fs_progress_close(progress);
94 progress = 0;
95 rfs->prog_data = 0;
96 }
97 return 0;
98 }
99
determine_fs_stride(ext2_filsys fs)100 static void determine_fs_stride(ext2_filsys fs)
101 {
102 unsigned int group;
103 unsigned long long sum;
104 unsigned int has_sb, prev_has_sb, num;
105 int i_stride, b_stride;
106
107 if (fs->stride)
108 return;
109 num = 0; sum = 0;
110 for (group = 0; group < fs->group_desc_count; group++) {
111 has_sb = ext2fs_bg_has_super(fs, group);
112 if (group == 0 || has_sb != prev_has_sb)
113 goto next;
114 b_stride = fs->group_desc[group].bg_block_bitmap -
115 fs->group_desc[group-1].bg_block_bitmap -
116 fs->super->s_blocks_per_group;
117 i_stride = fs->group_desc[group].bg_inode_bitmap -
118 fs->group_desc[group-1].bg_inode_bitmap -
119 fs->super->s_blocks_per_group;
120 if (b_stride != i_stride ||
121 b_stride < 0)
122 goto next;
123
124 /* printf("group %d has stride %d\n", group, b_stride); */
125 sum += b_stride;
126 num++;
127
128 next:
129 prev_has_sb = has_sb;
130 }
131
132 if (fs->group_desc_count > 12 && num < 3)
133 sum = 0;
134
135 if (num)
136 fs->stride = sum / num;
137 else
138 fs->stride = 0;
139
140 fs->super->s_raid_stride = fs->stride;
141 ext2fs_mark_super_dirty(fs);
142
143 #if 0
144 if (fs->stride)
145 printf("Using RAID stride of %d\n", fs->stride);
146 #endif
147 }
148
main(int argc,char ** argv)149 int main (int argc, char ** argv)
150 {
151 errcode_t retval;
152 ext2_filsys fs;
153 int c;
154 int flags = 0;
155 int flush = 0;
156 int force = 0;
157 int io_flags = 0;
158 int force_min_size = 0;
159 int print_min_size = 0;
160 int fd, ret;
161 blk_t new_size = 0;
162 blk_t max_size = 0;
163 blk_t min_size = 0;
164 io_manager io_ptr;
165 char *new_size_str = 0;
166 int use_stride = -1;
167 #ifdef HAVE_FSTAT64
168 struct stat64 st_buf;
169 #else
170 struct stat st_buf;
171 #endif
172 __s64 new_file_size;
173 unsigned int sys_page_size = 4096;
174 long sysval;
175 int len, mount_flags;
176 char *mtpt;
177
178 #ifdef ENABLE_NLS
179 setlocale(LC_MESSAGES, "");
180 setlocale(LC_CTYPE, "");
181 bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
182 textdomain(NLS_CAT_NAME);
183 #endif
184
185 add_error_table(&et_ext2_error_table);
186
187 fprintf (stderr, "resize2fs %s (%s)\n",
188 E2FSPROGS_VERSION, E2FSPROGS_DATE);
189 if (argc && *argv)
190 program_name = *argv;
191
192 while ((c = getopt (argc, argv, "d:fFhMPpS:")) != EOF) {
193 switch (c) {
194 case 'h':
195 usage(program_name);
196 break;
197 case 'f':
198 force = 1;
199 break;
200 case 'F':
201 flush = 1;
202 break;
203 case 'M':
204 force_min_size = 1;
205 break;
206 case 'P':
207 print_min_size = 1;
208 break;
209 case 'd':
210 flags |= atoi(optarg);
211 break;
212 case 'p':
213 flags |= RESIZE_PERCENT_COMPLETE;
214 break;
215 case 'S':
216 use_stride = atoi(optarg);
217 break;
218 default:
219 usage(program_name);
220 }
221 }
222 if (optind == argc)
223 usage(program_name);
224
225 device_name = argv[optind++];
226 if (optind < argc)
227 new_size_str = argv[optind++];
228 if (optind < argc)
229 usage(program_name);
230
231 io_options = strchr(device_name, '?');
232 if (io_options)
233 *io_options++ = 0;
234
235 /*
236 * Figure out whether or not the device is mounted, and if it is
237 * where it is mounted.
238 */
239 len=80;
240 while (1) {
241 mtpt = malloc(len);
242 if (!mtpt)
243 return ENOMEM;
244 mtpt[len-1] = 0;
245 retval = ext2fs_check_mount_point(device_name, &mount_flags,
246 mtpt, len);
247 if (retval) {
248 com_err("ext2fs_check_mount_point", retval,
249 _("while determining whether %s is mounted."),
250 device_name);
251 exit(1);
252 }
253 if (!(mount_flags & EXT2_MF_MOUNTED) || (mtpt[len-1] == 0))
254 break;
255 free(mtpt);
256 len = 2 * len;
257 }
258
259 #ifdef HAVE_OPEN64
260 fd = open64(device_name, O_RDWR);
261 #else
262 fd = open(device_name, O_RDWR);
263 #endif
264 if (fd < 0) {
265 com_err("open", errno, _("while opening %s"),
266 device_name);
267 exit(1);
268 }
269
270 #ifdef HAVE_FSTAT64
271 ret = fstat64(fd, &st_buf);
272 #else
273 ret = fstat(fd, &st_buf);
274 #endif
275 if (ret < 0) {
276 com_err("open", errno,
277 _("while getting stat information for %s"),
278 device_name);
279 exit(1);
280 }
281
282 if (flush) {
283 retval = ext2fs_sync_device(fd, 1);
284 if (retval) {
285 com_err(argv[0], retval,
286 _("while trying to flush %s"),
287 device_name);
288 exit(1);
289 }
290 }
291
292 if (!S_ISREG(st_buf.st_mode )) {
293 close(fd);
294 fd = -1;
295 }
296
297 #ifdef CONFIG_TESTIO_DEBUG
298 if (getenv("TEST_IO_FLAGS") || getenv("TEST_IO_BLOCK")) {
299 io_ptr = test_io_manager;
300 test_io_backing_manager = unix_io_manager;
301 } else
302 #endif
303 io_ptr = unix_io_manager;
304
305 if (!(mount_flags & EXT2_MF_MOUNTED))
306 io_flags = EXT2_FLAG_RW | EXT2_FLAG_EXCLUSIVE;
307 retval = ext2fs_open2(device_name, io_options, io_flags,
308 0, 0, io_ptr, &fs);
309 if (retval) {
310 com_err (program_name, retval, _("while trying to open %s"),
311 device_name);
312 printf (_("Couldn't find valid filesystem superblock.\n"));
313 exit (1);
314 }
315
316 /*
317 * Check for compatibility with the feature sets. We need to
318 * be more stringent than ext2fs_open().
319 */
320 if (fs->super->s_feature_compat & ~EXT2_LIB_FEATURE_COMPAT_SUPP) {
321 com_err(program_name, EXT2_ET_UNSUPP_FEATURE,
322 "(%s)", device_name);
323 exit(1);
324 }
325
326 /*
327 * XXXX The combination of flex_bg and !resize_inode causes
328 * major problems for resize2fs, since when the group descriptors
329 * grow in size this can potentially require multiple inode
330 * tables to be moved aside to make room, and resize2fs chokes
331 * rather badly in this scenario. It's a rare combination,
332 * except when a filesystem is expanded more than a certain
333 * size, so for now, we'll just prohibit that combination.
334 * This is something we should fix eventually, though.
335 */
336 if ((fs->super->s_feature_incompat & EXT4_FEATURE_INCOMPAT_FLEX_BG) &&
337 !(fs->super->s_feature_compat & EXT2_FEATURE_COMPAT_RESIZE_INODE)) {
338 com_err(program_name, 0, _("%s: The combination of flex_bg "
339 "and\n\t!resize_inode features "
340 "is not supported by resize2fs.\n"),
341 device_name);
342 exit(1);
343 }
344
345 min_size = calculate_minimum_resize_size(fs);
346
347 if (print_min_size) {
348 printf(_("Estimated minimum size of the filesystem: %u\n"),
349 min_size);
350 exit(0);
351 }
352
353 /* Determine the system page size if possible */
354 #ifdef HAVE_SYSCONF
355 #if (!defined(_SC_PAGESIZE) && defined(_SC_PAGE_SIZE))
356 #define _SC_PAGESIZE _SC_PAGE_SIZE
357 #endif
358 #ifdef _SC_PAGESIZE
359 sysval = sysconf(_SC_PAGESIZE);
360 if (sysval > 0)
361 sys_page_size = sysval;
362 #endif /* _SC_PAGESIZE */
363 #endif /* HAVE_SYSCONF */
364
365 /*
366 * Get the size of the containing partition, and use this for
367 * defaults and for making sure the new filesystem doesn't
368 * exceed the partition size.
369 */
370 retval = ext2fs_get_device_size(device_name, fs->blocksize,
371 &max_size);
372 if (retval) {
373 com_err(program_name, retval,
374 _("while trying to determine filesystem size"));
375 exit(1);
376 }
377 if (force_min_size)
378 new_size = min_size;
379 else if (new_size_str) {
380 new_size = parse_num_blocks(new_size_str,
381 fs->super->s_log_block_size);
382 if (new_size == 0) {
383 com_err(program_name, 0,
384 _("Invalid new size: %s\n"), new_size_str);
385 exit(1);
386 }
387 } else {
388 new_size = max_size;
389 /* Round down to an even multiple of a pagesize */
390 if (sys_page_size > fs->blocksize)
391 new_size &= ~((sys_page_size / fs->blocksize)-1);
392 }
393
394 if (!force && new_size < min_size) {
395 com_err(program_name, 0,
396 _("New size smaller than minimum (%u)\n"), min_size);
397 exit(1);
398 }
399 if (use_stride >= 0) {
400 if (use_stride >= (int) fs->super->s_blocks_per_group) {
401 com_err(program_name, 0,
402 _("Invalid stride length"));
403 exit(1);
404 }
405 fs->stride = fs->super->s_raid_stride = use_stride;
406 ext2fs_mark_super_dirty(fs);
407 } else
408 determine_fs_stride(fs);
409
410 /*
411 * If we are resizing a plain file, and it's not big enough,
412 * automatically extend it in a sparse fashion by writing the
413 * last requested block.
414 */
415 new_file_size = ((__u64) new_size) * fs->blocksize;
416 if ((__u64) new_file_size >
417 (((__u64) 1) << (sizeof(st_buf.st_size)*8 - 1)) - 1)
418 fd = -1;
419 if ((new_file_size > st_buf.st_size) &&
420 (fd > 0)) {
421 if ((ext2fs_llseek(fd, new_file_size-1, SEEK_SET) >= 0) &&
422 (write(fd, "0", 1) == 1))
423 max_size = new_size;
424 }
425 if (!force && (new_size > max_size)) {
426 fprintf(stderr, _("The containing partition (or device)"
427 " is only %u (%dk) blocks.\nYou requested a new size"
428 " of %u blocks.\n\n"), max_size,
429 fs->blocksize / 1024, new_size);
430 exit(1);
431 }
432 if (new_size == fs->super->s_blocks_count) {
433 fprintf(stderr, _("The filesystem is already %u blocks "
434 "long. Nothing to do!\n\n"), new_size);
435 exit(0);
436 }
437 if (mount_flags & EXT2_MF_MOUNTED) {
438 retval = online_resize_fs(fs, mtpt, &new_size, flags);
439 } else {
440 if (!force && ((fs->super->s_lastcheck < fs->super->s_mtime) ||
441 (fs->super->s_state & EXT2_ERROR_FS) ||
442 ((fs->super->s_state & EXT2_VALID_FS) == 0))) {
443 fprintf(stderr,
444 _("Please run 'e2fsck -f %s' first.\n\n"),
445 device_name);
446 exit(1);
447 }
448 printf(_("Resizing the filesystem on "
449 "%s to %u (%dk) blocks.\n"),
450 device_name, new_size, fs->blocksize / 1024);
451 retval = resize_fs(fs, &new_size, flags,
452 ((flags & RESIZE_PERCENT_COMPLETE) ?
453 resize_progress_func : 0));
454 }
455 free(mtpt);
456 if (retval) {
457 com_err(program_name, retval, _("while trying to resize %s"),
458 device_name);
459 fprintf(stderr,
460 _("Please run 'e2fsck -fy %s' to fix the filesystem\n"
461 "after the aborted resize operation.\n"),
462 device_name);
463 ext2fs_close(fs);
464 exit(1);
465 }
466 printf(_("The filesystem on %s is now %u blocks long.\n\n"),
467 device_name, new_size);
468
469 if ((st_buf.st_size > new_file_size) &&
470 (fd > 0)) {
471 #ifdef HAVE_FTRUNCATE64
472 retval = ftruncate64(fd, new_file_size);
473 #else
474 retval = 0;
475 /* Only truncate if new_file_size doesn't overflow off_t */
476 if (((off_t) new_file_size) == new_file_size)
477 retval = ftruncate(fd, (off_t) new_file_size);
478 #endif
479 if (retval)
480 com_err(program_name, retval,
481 _("while trying to truncate %s"),
482 device_name);
483 }
484 if (fd > 0)
485 close(fd);
486 remove_error_table(&et_ext2_error_table);
487 return (0);
488 }
489