bc: convert to "G trick" - this returns bc to zero bss increase
[oweals/busybox.git] / util-linux / fdisk.c
index 6b5e3880fc9baac7fc24c1923d351b62a8457a49..288b9235f8e2fcc7b420d73d99174695317dcd8e 100644 (file)
@@ -1,11 +1,90 @@
 /* vi: set sw=4 ts=4: */
-/* fdisk.c -- Partition table manipulator for Linux.
+/*
+ * fdisk.c -- Partition table manipulator for Linux.
  *
  * Copyright (C) 1992  A. V. Le Blanc (LeBlanc@mcc.ac.uk)
  * Copyright (C) 2001,2002 Vladimir Oleynik <dzo@simtreas.ru> (initial bb port)
  *
  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
+//config:config FDISK
+//config:      bool "fdisk (41 kb)"
+//config:      default y
+//config:      select PLATFORM_LINUX
+//config:      help
+//config:      The fdisk utility is used to divide hard disks into one or more
+//config:      logical disks, which are generally called partitions. This utility
+//config:      can be used to list and edit the set of partitions or BSD style
+//config:      'disk slices' that are defined on a hard drive.
+//config:
+//config:config FDISK_SUPPORT_LARGE_DISKS
+//config:      bool "Support over 4GB disks"
+//config:      default y
+//config:      depends on FDISK
+//config:      depends on !LFS   # with LFS no special code is needed
+//config:
+//config:config FEATURE_FDISK_WRITABLE
+//config:      bool "Write support"
+//config:      default y
+//config:      depends on FDISK
+//config:      help
+//config:      Enabling this option allows you to create or change a partition table
+//config:      and write those changes out to disk. If you leave this option
+//config:      disabled, you will only be able to view the partition table.
+//config:
+//config:config FEATURE_AIX_LABEL
+//config:      bool "Support AIX disklabels"
+//config:      default n
+//config:      depends on FDISK && FEATURE_FDISK_WRITABLE
+//config:      help
+//config:      Enabling this option allows you to create or change AIX disklabels.
+//config:      Most people can safely leave this option disabled.
+//config:
+//config:config FEATURE_SGI_LABEL
+//config:      bool "Support SGI disklabels"
+//config:      default n
+//config:      depends on FDISK && FEATURE_FDISK_WRITABLE
+//config:      help
+//config:      Enabling this option allows you to create or change SGI disklabels.
+//config:      Most people can safely leave this option disabled.
+//config:
+//config:config FEATURE_SUN_LABEL
+//config:      bool "Support SUN disklabels"
+//config:      default n
+//config:      depends on FDISK && FEATURE_FDISK_WRITABLE
+//config:      help
+//config:      Enabling this option allows you to create or change SUN disklabels.
+//config:      Most people can safely leave this option disabled.
+//config:
+//config:config FEATURE_OSF_LABEL
+//config:      bool "Support BSD disklabels"
+//config:      default n
+//config:      depends on FDISK && FEATURE_FDISK_WRITABLE
+//config:      help
+//config:      Enabling this option allows you to create or change BSD disklabels
+//config:      and define and edit BSD disk slices.
+//config:
+//config:config FEATURE_GPT_LABEL
+//config:      bool "Support GPT disklabels"
+//config:      default n
+//config:      depends on FDISK && FEATURE_FDISK_WRITABLE
+//config:      help
+//config:      Enabling this option allows you to view GUID Partition Table
+//config:      disklabels.
+//config:
+//config:config FEATURE_FDISK_ADVANCED
+//config:      bool "Support expert mode"
+//config:      default y
+//config:      depends on FDISK && FEATURE_FDISK_WRITABLE
+//config:      help
+//config:      Enabling this option allows you to do terribly unsafe things like
+//config:      define arbitrary drive geometry, move the beginning of data in a
+//config:      partition, and similarly evil things. Unless you have a very good
+//config:      reason you would be wise to leave this disabled.
+
+//applet:IF_FDISK(APPLET(fdisk, BB_DIR_SBIN, BB_SUID_DROP))
+
+//kbuild:lib-$(CONFIG_FDISK) += fdisk.o
 
 /* Looks like someone forgot to add this to config system */
 //usage:#ifndef ENABLE_FEATURE_FDISK_BLKSIZE
 //usage:       )
 //usage:     "\n       -b 2048         (for certain MO disks) use 2048-byte sectors"
 //usage:     "\n       -C CYLINDERS    Set number of cylinders/heads/sectors"
-//usage:     "\n       -H HEADS"
-//usage:     "\n       -S SECTORS"
+//usage:     "\n       -H HEADS        Typically 255"
+//usage:     "\n       -S SECTORS      Typically 63"
 
 #ifndef _LARGEFILE64_SOURCE
 /* For lseek64 */
 # define BLKGETSIZE64 _IOR(0x12,114,size_t)
 #endif
 #include "libbb.h"
+#include "unicode.h"
 
 #if BB_LITTLE_ENDIAN
 # define inline_if_little_endian ALWAYS_INLINE
@@ -89,9 +169,9 @@ typedef unsigned long long ullong;
  * do not support more than 2^32 sectors
  */
 typedef uint32_t sector_t;
-#if UINT_MAX == 4294967295
+#if UINT_MAX == 0xffffffff
 # define SECT_FMT ""
-#elif ULONG_MAX == 4294967295
+#elif ULONG_MAX == 0xffffffff
 # define SECT_FMT "l"
 #else
 # error Cant detect sizeof(uint32_t)
@@ -106,6 +186,8 @@ struct hd_geometry {
 
 #define HDIO_GETGEO     0x0301  /* get device geometry */
 
+/* TODO: #if ENABLE_FEATURE_FDISK_WRITABLE */
+/* (currently fdisk_sun/sgi.c do not have proper WRITABLE #ifs) */
 static const char msg_building_new_label[] ALIGN1 =
 "Building a new %s. Changes will remain in memory only,\n"
 "until you decide to write them. After that the previous content\n"
@@ -113,6 +195,7 @@ static const char msg_building_new_label[] ALIGN1 =
 
 static const char msg_part_already_defined[] ALIGN1 =
 "Partition %u is already defined, delete it before re-adding\n";
+/* #endif */
 
 
 struct partition {
@@ -343,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
@@ -405,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)
@@ -473,6 +555,42 @@ close_dev_fd(void)
        xmove_fd(xopen(bb_dev_null, O_RDONLY), dev_fd);
 }
 
+/* Return partition name */
+static const char *
+partname(const char *dev, int pno, int lth)
+{
+       const char *p;
+       int w, wp;
+       int bufsiz;
+       char *bufp;
+
+       bufp = auto_string(xzalloc(80));
+       bufsiz = 80;
+
+       w = strlen(dev);
+       p = "";
+
+       if (isdigit(dev[w-1]))
+               p = "p";
+
+       /* devfs kludge - note: fdisk partition names are not supposed
+          to equal kernel names, so there is no reason to do this */
+       if (strcmp(dev + w - 4, "disc") == 0) {
+               w -= 4;
+               p = "part";
+       }
+
+       wp = strlen(p);
+
+       if (lth) {
+               snprintf(bufp, bufsiz, "%*.*s%s%-2u",
+                       lth-wp-2, w, dev, p, pno);
+       } else {
+               snprintf(bufp, bufsiz, "%.*s%s%-2u", w, dev, p, pno);
+       }
+       return bufp;
+}
+
 static ALWAYS_INLINE struct partition *
 get_part_table(int i)
 {
@@ -520,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), /*timeout*/ -1);
-       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)
 {
@@ -561,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)
 {
@@ -1495,53 +1613,74 @@ 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;
 
-                       value = atoi(line_ptr + 1);
+                       if (sizeof(value) <= sizeof(long))
+                               value = strtoul(line_ptr + 1, NULL, 10);
+                       else
+                               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;
+/*
+ * fdisk from util-linux 2.31 seems to round '+NNNk' and '+NNNK' to megabytes,
+ * (512-byte) sector count of the partition does not equal NNN*2:
+ *
+ * Last sector, +sectors or +size{K,M,G,T,P} (1953792-1000215215, default 1000215215): +9727k
+ *   Device     Boot   Start     End Sectors  Size Id Type
+ *   /dev/sdaN       1953792 1972223   18432    9M 83 Linux   <-- size exactly 9*1024*1024 bytes
+ *
+ * Last sector, +sectors or +size{K,M,G,T,P} (1953792-1000215215, default 1000215215): +9728k
+ *   /dev/sdaN       1953792 1974271   20480   10M 83 Linux   <-- size exactly 10*1024*1024 bytes
+ *
+ * If 'k' means 1000 bytes (not 1024), then 9728k = 9728*1000 = 9500*1024,
+ * exactly halfway from 9000 to 10000, which explains why it jumps to next mbyte
+ * at this value.
+ *
+ * 'm' does not seem to behave this way: it means 1024*1024 bytes.
+ *
+ * Not sure we want to copy this. If user says he wants 1234kbyte partition,
+ * we do _exactly that_: 1234kbytes = 2468 sectors.
+ */
                        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;
@@ -1606,8 +1745,9 @@ get_existing_partition(int warn, unsigned max)
 }
 
 static int
-get_nonexisting_partition(int warn, unsigned max)
+get_nonexisting_partition(void)
 {
+       const int max = 4;
        int pno = -1;
        unsigned i;
 
@@ -1629,7 +1769,7 @@ get_nonexisting_partition(int warn, unsigned max)
        return -1;
 
  not_unique:
-       return get_partition(warn, max);
+       return get_partition(/*warn*/ 0, max);
 }
 
 
@@ -1885,34 +2025,29 @@ check_consistency(const struct partition *p, int partition)
                printf("     phys=(%u,%u,%u) ", pec, peh, pes);
                printf("logical=(%u,%u,%u)\n", lec, leh, les);
        }
-
-/* Ending on cylinder boundary? */
-       if (peh != (g_heads - 1) || pes != g_sectors) {
-               printf("Partition %u does not end on cylinder boundary\n",
-                       partition + 1);
-       }
 }
 
 static void
 list_disk_geometry(void)
 {
        ullong bytes = ((ullong)total_number_of_sectors << 9);
-       long megabytes = bytes / 1000000;
-
-       if (megabytes < 10000)
-               printf("\nDisk %s: %lu MB, %llu bytes\n",
-                       disk_device, megabytes, bytes);
-       else
-               printf("\nDisk %s: %lu.%lu GB, %llu bytes\n",
-                       disk_device, megabytes/1000, (megabytes/100)%10, bytes);
-       printf("%u heads, %u sectors/track, %u cylinders",
-                  g_heads, g_sectors, g_cylinders);
-       if (units_per_sector == 1)
-               printf(", total %"SECT_FMT"u sectors",
-                       total_number_of_sectors / (sector_size/512));
-       printf("\nUnits = %s of %u * %u = %u bytes\n\n",
+       ullong xbytes = bytes / (1024*1024);
+       char x = 'M';
+
+       if (xbytes >= 10000) {
+               xbytes += 512; /* fdisk util-linux 2.28 does this */
+               xbytes /= 1024;
+               x = 'G';
+       }
+       printf("Disk %s: %llu %cB, %llu bytes, %"SECT_FMT"u sectors\n"
+               "%u cylinders, %u heads, %u sectors/track\n"
+               "Units: %s of %u * %u = %u bytes\n\n",
+               disk_device, xbytes, x,
+               bytes, total_number_of_sectors,
+               g_cylinders, g_heads, g_sectors,
                str_units(PLURAL),
-               units_per_sector, sector_size, units_per_sector * sector_size);
+               units_per_sector, sector_size, units_per_sector * sector_size
+       );
 }
 
 /*
@@ -2056,42 +2191,6 @@ fix_partition_table_order(void)
 }
 #endif
 
-/* Return partition name */
-static const char *
-partname(const char *dev, int pno, int lth)
-{
-       const char *p;
-       int w, wp;
-       int bufsiz;
-       char *bufp;
-
-       bufp = auto_string(xzalloc(80));
-       bufsiz = 80;
-
-       w = strlen(dev);
-       p = "";
-
-       if (isdigit(dev[w-1]))
-               p = "p";
-
-       /* devfs kludge - note: fdisk partition names are not supposed
-          to equal kernel names, so there is no reason to do this */
-       if (strcmp(dev + w - 4, "disc") == 0) {
-               w -= 4;
-               p = "part";
-       }
-
-       wp = strlen(p);
-
-       if (lth) {
-               snprintf(bufp, bufsiz, "%*.*s%s%-2u",
-                       lth-wp-2, w, dev, p, pno);
-       } else {
-               snprintf(bufp, bufsiz, "%.*s%s%-2u", w, dev, p, pno);
-       }
-       return bufp;
-}
-
 static const char *
 chs_string11(unsigned cyl, unsigned head, unsigned sect)
 {
@@ -2277,6 +2376,7 @@ verify(void)
 {
        int i, j;
        sector_t total = 1;
+       sector_t chs_size;
        sector_t first[g_partitions], last[g_partitions];
        struct partition *p;
 
@@ -2338,11 +2438,14 @@ verify(void)
                }
        }
 
-       if (total > g_heads * g_sectors * g_cylinders)
-               printf("Total allocated sectors %u greater than the maximum "
-                       "%u\n", total, g_heads * g_sectors * g_cylinders);
+       chs_size = (sector_t)g_heads * g_sectors * g_cylinders;
+       if (total > chs_size)
+               printf("Total allocated sectors %u"
+                       " greater than CHS size %"SECT_FMT"u\n",
+                       total, chs_size
+               );
        else {
-               total = g_heads * g_sectors * g_cylinders - total;
+               total = chs_size - total;
                if (total != 0)
                        printf("%"SECT_FMT"u unallocated sectors\n", total);
        }
@@ -2444,8 +2547,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;
@@ -2529,15 +2633,16 @@ 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) {
                        c = read_nonempty(line);
-                       if ((c | 0x20) == 'p') {
-                               i = get_nonexisting_partition(0, 4);
+                       c |= 0x20; /* lowercase */
+                       if (c == 'p') {
+                               i = get_nonexisting_partition();
                                if (i >= 0)
                                        add_partition(i, LINUX_NATIVE);
                                return;
@@ -2547,7 +2652,7 @@ new_partition(void)
                                return;
                        }
                        if (c == 'e' && !extended_offset) {
-                               i = get_nonexisting_partition(0, 4);
+                               i = get_nonexisting_partition();
                                if (i >= 0)
                                        add_partition(i, EXTENDED);
                                return;
@@ -2761,7 +2866,7 @@ xselect(void)
                        if (dos_compatible_flag) {
                                sector_offset = g_sectors;
                                puts("Warning: setting sector offset for DOS "
-                                       "compatiblity");
+                                       "compatibility");
                        }
                        update_units();
                        break;