diff options
| -rw-r--r-- | util-linux/fdisk.c | 90 | ||||
| -rw-r--r-- | util-linux/fdisk_sgi.c | 2 | ||||
| -rw-r--r-- | util-linux/fdisk_sun.c | 2 |
3 files changed, 46 insertions, 48 deletions
diff --git a/util-linux/fdisk.c b/util-linux/fdisk.c index e47bc7324..296716575 100644 --- a/util-linux/fdisk.c +++ b/util-linux/fdisk.c @@ -426,7 +426,7 @@ struct globals { unsigned sector_offset; // = 1; unsigned g_heads, g_sectors, g_cylinders; smallint /* enum label_type */ current_label_type; - smallint display_in_cyl_units; // = 1; + smallint display_in_cyl_units; #if ENABLE_FEATURE_OSF_LABEL smallint possibly_osf_label; #endif @@ -488,7 +488,6 @@ struct globals { sector_size = DEFAULT_SECTOR_SIZE; \ sector_offset = 1; \ g_partitions = 4; \ - display_in_cyl_units = 1; \ units_per_sector = 1; \ dos_compatible_flag = 1; \ } while (0) @@ -639,25 +638,6 @@ seek_sector(sector_t secno) } #if ENABLE_FEATURE_FDISK_WRITABLE -/* Read line; return 0 or first printable char */ -static int -read_line(const char *prompt) -{ - int sz; - - sz = read_line_input(NULL, prompt, line_buffer, sizeof(line_buffer)); - if (sz <= 0) - exit(EXIT_SUCCESS); /* Ctrl-D or Ctrl-C */ - - if (line_buffer[sz-1] == '\n') - line_buffer[--sz] = '\0'; - - line_ptr = line_buffer; - while (*line_ptr != '\0' && (unsigned char)*line_ptr <= ' ') - line_ptr++; - return *line_ptr; -} - static void set_all_unchanged(void) { @@ -680,6 +660,25 @@ write_part_table_flag(char *b) b[511] = 0xaa; } +/* Read line; return 0 or first printable non-space char */ +static int +read_line(const char *prompt) +{ + int sz; + + sz = read_line_input(NULL, prompt, line_buffer, sizeof(line_buffer)); + if (sz <= 0) + exit(EXIT_SUCCESS); /* Ctrl-D or Ctrl-C */ + + if (line_buffer[sz-1] == '\n') + line_buffer[--sz] = '\0'; + + line_ptr = line_buffer; + while (*line_ptr != '\0' && (unsigned char)*line_ptr <= ' ') + line_ptr++; + return *line_ptr; +} + static char read_nonempty(const char *mesg) { @@ -1614,7 +1613,7 @@ read_int(sector_t low, sector_t dflt, sector_t high, sector_t base, const char * if (*line_ptr == '+' || *line_ptr == '-') { int minus = (*line_ptr == '-'); - int absolute = 0; + unsigned scale_shift; if (sizeof(value) <= sizeof(long)) value = strtoul(line_ptr + 1, NULL, 10); @@ -1622,48 +1621,46 @@ read_int(sector_t low, sector_t dflt, sector_t high, sector_t base, const char * value = strtoull(line_ptr + 1, NULL, 10); /* (1) if 2nd char is digit, use_default = 0. - * (2) move line_ptr to first non-digit. */ + * (2) move line_ptr to first non-digit. + */ while (isdigit(*++line_ptr)) use_default = 0; - switch (*line_ptr) { - case 'c': - case 'C': - if (!display_in_cyl_units) - value *= g_heads * g_sectors; - break; - case 'K': - absolute = 1024; - break; + scale_shift = 0; + switch (*line_ptr | 0x20) { case 'k': - absolute = 1000; + scale_shift = 10; /* 1024 */ break; case 'm': - case 'M': - absolute = 1000000; + scale_shift = 20; /* 1024*1024 */ break; case 'g': - case 'G': - absolute = 1000000000; + scale_shift = 30; /* 1024*1024*1024 */ + break; + case 't': + scale_shift = 40; /* 1024*1024*1024*1024 */ break; default: break; } - if (absolute) { + if (scale_shift) { ullong bytes; unsigned long unit; - bytes = (ullong) value * absolute; + bytes = (ullong) value << scale_shift; unit = sector_size * units_per_sector; bytes += unit/2; /* round */ bytes /= unit; - value = bytes; + value = (bytes != 0 ? bytes - 1 : 0); } if (minus) value = -value; value += base; } else { - value = atoi(line_ptr); + if (sizeof(value) <= sizeof(long)) + value = strtoul(line_ptr, NULL, 10); + else + value = strtoull(line_ptr, NULL, 10); while (isdigit(*line_ptr)) { line_ptr++; use_default = 0; @@ -2529,8 +2526,9 @@ add_partition(int n, int sys) stop = limit; } else { snprintf(mesg, sizeof(mesg), - "Last %s or +size or +sizeM or +sizeK", - str_units(SINGULAR)); + "Last %s or +size{,K,M,G,T}", + str_units(SINGULAR) + ); stop = read_int(cround(start), cround(limit), cround(limit), cround(start), mesg); if (display_in_cyl_units) { stop = stop * units_per_sector - 1; @@ -2614,9 +2612,9 @@ new_partition(void) } else { char c, line[80]; snprintf(line, sizeof(line), - "Command action\n" - " %s\n" - " p primary partition (1-4)\n", + "Partition type\n" + " p primary partition (1-4)\n" + " %s\n", (extended_offset ? "l logical (5 or over)" : "e extended")); while (1) { diff --git a/util-linux/fdisk_sgi.c b/util-linux/fdisk_sgi.c index 1cf0af5cc..0e5491a19 100644 --- a/util-linux/fdisk_sgi.c +++ b/util-linux/fdisk_sgi.c @@ -623,7 +623,7 @@ sgi_change_sysid(int i, int sys) "retrieve from its directory standalone tools like sash and fx.\n" "Only the \"SGI volume\" entire disk section may violate this.\n" "Type YES if you are sure about tagging this partition differently.\n"); - if (strcmp(line_ptr, "YES\n") != 0) + if (strcmp(line_ptr, "YES") != 0) return; } sgilabel->partitions[i].id = SGI_SSWAP32(sys); diff --git a/util-linux/fdisk_sun.c b/util-linux/fdisk_sun.c index f62a53ac6..3697a69b9 100644 --- a/util-linux/fdisk_sun.c +++ b/util-linux/fdisk_sun.c @@ -606,7 +606,7 @@ sun_change_sysid(int i, int sys) "there may destroy your partition table and bootblock.\n" "Type YES if you're very sure you would like that partition\n" "tagged with 82 (Linux swap): "); - if (strcmp(line_ptr, "YES\n")) + if (strcmp(line_ptr, "YES") != 0) return; } switch (sys) { |