Whamcloud - gitweb
Fix miscellaneous gcc -Wall warnings
[tools/e2fsprogs.git] / misc / badblocks.c
index 9a6acdf..709effe 100644 (file)
@@ -46,6 +46,7 @@ extern int optind;
 #include <unistd.h>
 #include <setjmp.h>
 #include <time.h>
+#include <limits.h>
 
 #include <sys/ioctl.h>
 #include <sys/types.h>
@@ -57,7 +58,7 @@ extern int optind;
 #include "nls-enable.h"
 
 const char * program_name = "badblocks";
-const char * done_string = N_("done                        \n");
+const char * done_string = N_("done                                \n");
 
 static int v_flag = 0;                 /* verbose */
 static int w_flag = 0;                 /* do r/w test: 0=no, 1=yes,
@@ -66,12 +67,13 @@ static int s_flag = 0;                      /* show progress of test */
 static int force = 0;                  /* force check of mounted device */
 static int t_flag = 0;                 /* number of test patterns */
 static int t_max = 0;                  /* allocated test patterns */
-static unsigned long *t_patts = NULL;  /* test patterns */
+static unsigned int *t_patts = NULL;   /* test patterns */
 static int current_O_DIRECT = 0;       /* Current status of O_DIRECT flag */
+static int exclusive_ok = 0;
 
 #define T_INC 32
 
-int sys_page_size = 4096;
+unsigned int sys_page_size = 4096;
 
 static void usage(void)
 {
@@ -80,8 +82,16 @@ static void usage(void)
        exit (1);
 }
 
-static unsigned long currently_testing = 0;
-static unsigned long num_blocks = 0;
+static void exclusive_usage(void)
+{
+       fprintf(stderr, 
+               _("%s: The -n and -w options are mutually exclusive.\n\n"), 
+               program_name);
+       exit(1);
+}
+
+static blk_t currently_testing = 0;
+static blk_t num_blocks = 0;
 static ext2_badblocks_list bb_list = NULL;
 static FILE *out;
 static blk_t next_bad = 0;
@@ -114,14 +124,14 @@ static void *allocate_buffer(size_t size)
  * This routine reports a new bad block.  If the bad block has already
  * been seen before, then it returns 0; otherwise it returns 1.
  */
-static int bb_output (unsigned long bad)
+static int bb_output (blk_t bad)
 {
        errcode_t errcode;
 
        if (ext2fs_badblocks_list_test(bb_list, bad))
                return 0;
 
-       fprintf(out, "%lu\n", bad);
+       fprintf(out, "%lu\n", (unsigned long) bad);
        fflush(out);
 
        errcode = ext2fs_badblocks_list_add (bb_list, bad);
@@ -141,7 +151,8 @@ static int bb_output (unsigned long bad)
 
 static void print_status(void)
 {
-       fprintf(stderr, "%15ld/%15ld", currently_testing, num_blocks);
+       fprintf(stderr, "%15lu/%15lu", (unsigned long) currently_testing, 
+               (unsigned long) num_blocks);
        fputs("\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b", stderr);
        fflush (stderr);
 }
@@ -187,7 +198,7 @@ static void uncapture_terminate(void)
 }
 
 static void set_o_direct(int dev, unsigned char *buffer, size_t size,
-                        unsigned long current_block)
+                        blk_t current_block)
 {
 #ifdef O_DIRECT
        int new_flag = O_DIRECT;
@@ -211,13 +222,13 @@ static void set_o_direct(int dev, unsigned char *buffer, size_t size,
 }
 
 
-static void pattern_fill(unsigned char *buffer, unsigned long pattern,
+static void pattern_fill(unsigned char *buffer, unsigned int pattern,
                         size_t n)
 {
        unsigned int    i, nb;
        unsigned char   bpattern[sizeof(pattern)], *ptr;
        
-       if (pattern == (unsigned long) ~0) {
+       if (pattern == (unsigned int) ~0) {
                for (ptr = buffer; ptr < buffer + n; ptr++) {
                        (*ptr) = random() % (1 << (8 * sizeof(char)));
                }
@@ -252,8 +263,8 @@ static void pattern_fill(unsigned char *buffer, unsigned long pattern,
  * Perform a read of a sequence of blocks; return the number of blocks
  *    successfully sequentially read.
  */
-static long do_read (int dev, unsigned char * buffer, int try, int block_size,
-                    unsigned long current_block)
+static int do_read (int dev, unsigned char * buffer, int try, int block_size,
+                   blk_t current_block)
 {
        long got;
 
@@ -281,8 +292,8 @@ static long do_read (int dev, unsigned char * buffer, int try, int block_size,
  * Perform a write of a sequence of blocks; return the number of blocks
  *    successfully sequentially written.
  */
-static long do_write (int dev, unsigned char * buffer, int try, int block_size,
-                    unsigned long current_block)
+static int do_write(int dev, unsigned char * buffer, int try, int block_size,
+                   unsigned long current_block)
 {
        long got;
 
@@ -317,13 +328,13 @@ static void flush_bufs(void)
                com_err(program_name, retval, _("during ext2fs_sync_device"));
 }
 
-static unsigned int test_ro (int dev, unsigned long last_block,
-                            int block_size, unsigned long from_count,
-                            unsigned long blocks_at_once)
+static unsigned int test_ro (int dev, blk_t last_block,
+                            int block_size, blk_t from_count,
+                            unsigned int blocks_at_once)
 {
        unsigned char * blkbuf;
        int try;
-       long got;
+       int got;
        unsigned int bb_count = 0;
        errcode_t errcode;
 
@@ -348,8 +359,9 @@ static unsigned int test_ro (int dev, unsigned long last_block,
                exit (1);
        }
        if (v_flag) {
-           fprintf (stderr, _("Checking blocks %lu to %lu\n"), from_count,
-                    last_block);
+               fprintf (stderr, _("Checking blocks %lu to %lu\n"), 
+                        (unsigned long) from_count, 
+                        (unsigned long) last_block - 1);
        }
        if (t_flag) {
                fputs(_("Checking for bad blocks in read-only mode\n"), stderr);
@@ -359,7 +371,7 @@ static unsigned int test_ro (int dev, unsigned long last_block,
        flush_bufs();
        try = blocks_at_once;
        currently_testing = from_count;
-       num_blocks = last_block;
+       num_blocks = last_block - 1;
        if (!t_flag && (s_flag || v_flag)) {
                fputs(_("Checking for bad blocks (read-only test): "), stderr);
                if (v_flag <= 1)
@@ -394,7 +406,7 @@ static unsigned int test_ro (int dev, unsigned long last_block,
                if (got == try) {
                        try = blocks_at_once;
                        /* recover page-aligned offset for O_DIRECT */
-                       if ( blocks_at_once >= (unsigned long) (sys_page_size >> 9)
+                       if ( (blocks_at_once >= sys_page_size >> 9)
                             && (currently_testing % (sys_page_size >> 9)!= 0))
                                try -= (sys_page_size >> 9)
                                        - (currently_testing 
@@ -420,13 +432,13 @@ static unsigned int test_ro (int dev, unsigned long last_block,
        return bb_count;
 }
 
-static unsigned int test_rw (int dev, unsigned long last_block,
-                            int block_size, unsigned long from_count,
-                            unsigned long blocks_at_once)
+static unsigned int test_rw (int dev, blk_t last_block,
+                            int block_size, blk_t from_count,
+                            unsigned int blocks_at_once)
 {
        unsigned char *buffer, *read_buffer;
-       const unsigned long patterns[] = {0xaa, 0x55, 0xff, 0x00};
-       const unsigned long *pattern;
+       const unsigned int patterns[] = {0xaa, 0x55, 0xff, 0x00};
+       const unsigned int *pattern;
        int i, try, got, nr_pattern, pat_idx;
        unsigned int bb_count = 0;
 
@@ -444,7 +456,8 @@ static unsigned int test_rw (int dev, unsigned long last_block,
                fputs(_("Checking for bad blocks in read-write mode\n"), 
                      stderr);
                fprintf(stderr, _("From block %lu to %lu\n"),
-                        from_count, last_block);
+                       (unsigned long) from_count, 
+                       (unsigned long) last_block);
        }
        if (t_flag) {
                pattern = t_patts;
@@ -456,7 +469,7 @@ static unsigned int test_rw (int dev, unsigned long last_block,
        for (pat_idx = 0; pat_idx < nr_pattern; pat_idx++) {
                pattern_fill(buffer, pattern[pat_idx],
                             blocks_at_once * block_size);
-               num_blocks = last_block;
+               num_blocks = last_block - 1;
                currently_testing = from_count;
                if (s_flag && v_flag <= 1)
                        alarm_intr(SIGALRM);
@@ -474,7 +487,7 @@ static unsigned int test_rw (int dev, unsigned long last_block,
                        if (got == try) {
                                try = blocks_at_once;
                                /* recover page-aligned offset for O_DIRECT */
-                               if ( blocks_at_once >= (unsigned long) (sys_page_size >> 9)
+                               if ( (blocks_at_once >= sys_page_size >> 9)
                                     && (currently_testing % 
                                         (sys_page_size >> 9)!= 0))
                                        try -= (sys_page_size >> 9)
@@ -518,7 +531,7 @@ static unsigned int test_rw (int dev, unsigned long last_block,
                        }
                        currently_testing += got;
                        /* recover page-aligned offset for O_DIRECT */
-                       if ( blocks_at_once >= (unsigned long) (sys_page_size >> 9)
+                       if ( (blocks_at_once >= sys_page_size >> 9)
                             && (currently_testing % (sys_page_size >> 9)!= 0))
                                try = blocks_at_once - (sys_page_size >> 9)
                                        - (currently_testing 
@@ -545,17 +558,18 @@ struct saved_blk_record {
        int     num;
 };
 
-static unsigned int test_nd (int dev, unsigned long last_block,
-                            int block_size, unsigned long from_count,
-                            unsigned long blocks_at_once)
+static unsigned int test_nd (int dev, blk_t last_block,
+                            int block_size, blk_t from_count,
+                            unsigned int blocks_at_once)
 {
        unsigned char *blkbuf, *save_ptr, *test_ptr, *read_ptr;
        unsigned char *test_base, *save_base, *read_base;
        int try, i;
-       const unsigned long patterns[] = { ~0 };
-       const unsigned long *pattern;
+       const unsigned int patterns[] = { ~0 };
+       const unsigned int *pattern;
        int nr_pattern, pat_idx;
-       long got, used2, written, save_currently_testing;
+       int got, used2, written;
+       blk_t save_currently_testing;
        struct saved_blk_record *test_record;
        /* This is static to prevent being clobbered by the longjmp */
        static int num_saved;
@@ -591,7 +605,8 @@ static unsigned int test_nd (int dev, unsigned long last_block,
        flush_bufs();
        if (v_flag) {
            fputs(_("Checking for bad blocks in non-destructive read-write mode\n"), stderr);
-           fprintf (stderr, _("From block %lu to %lu\n"), from_count, last_block);
+           fprintf (stderr, _("From block %lu to %lu\n"), 
+                    (unsigned long) from_count, (unsigned long) last_block);
        }
        if (s_flag || v_flag > 1) {
                fputs(_("Checking for bad blocks (non-destructive read-write test)\n"), stderr);
@@ -632,7 +647,7 @@ static unsigned int test_nd (int dev, unsigned long last_block,
                save_ptr = save_base;
                test_ptr = test_base;
                currently_testing = from_count;
-               num_blocks = last_block;
+               num_blocks = last_block - 1;
                if (s_flag && v_flag <= 1)
                        alarm_intr(SIGALRM);
 
@@ -672,7 +687,8 @@ static unsigned int test_nd (int dev, unsigned long last_block,
                        if (written != got)
                                com_err (program_name, errno,
                                         _("during test data write, block %lu"),
-                                        currently_testing + written);
+                                        (unsigned long) currently_testing + 
+                                        written);
 
                        buf_used += got;
                        save_ptr += got * block_size;
@@ -792,7 +808,7 @@ static void check_mount(char *device_name)
                exit(1);
        }
 
-       if (mount_flags & EXT2_MF_BUSY) {
+       if ((mount_flags & EXT2_MF_BUSY) && !exclusive_ok) {
                fprintf(stderr, _("%s is apparently in use by the system; "),
                        device_name);
                if (force)
@@ -803,27 +819,43 @@ static void check_mount(char *device_name)
 
 }
 
+/*
+ * This function will convert a string to an unsigned long, printing
+ * an error message if it fails, and returning success or failure in err.
+ */
+static unsigned int parse_uint(const char *str, const char *descr)
+{
+       char            *tmp;
+       unsigned long   ret;
+       
+       ret = strtoul(str, &tmp, 0);
+       if (*tmp || errno || (ret > UINT_MAX) ||
+           (ret == ULONG_MAX && errno == ERANGE)) {
+               com_err (program_name, 0, _("invalid %s - %s"), descr, str);
+               exit (1);
+       }
+       return ret;
+}
 
 int main (int argc, char ** argv)
 {
        int c;
-       char * tmp;
        char * device_name;
        char * host_device_name = NULL;
        char * input_file = NULL;
        char * output_file = NULL;
        FILE * in = NULL;
        int block_size = 1024;
-       unsigned long blocks_at_once = 64;
+       unsigned int blocks_at_once = 64;
        blk_t last_block, from_count;
        int num_passes = 0;
        int passes_clean = 0;
        int dev;
        errcode_t errcode;
-       unsigned long pattern;
-       unsigned int (*test_func)(int, unsigned long,
-                                 int, unsigned long,
-                                 unsigned long);
+       unsigned int pattern;
+       unsigned int (*test_func)(int, blk_t,
+                                 int, blk_t,
+                                 unsigned int);
        int open_flag = 0;
        long sysval;
 
@@ -852,11 +884,11 @@ int main (int argc, char ** argv)
        
        if (argc && *argv)
                program_name = *argv;
-       while ((c = getopt (argc, argv, "b:fi:o:svwnc:p:h:t:")) != EOF) {
+       while ((c = getopt (argc, argv, "b:fi:o:svwnc:p:h:t:X")) != EOF) {
                switch (c) {
                case 'b':
-                       block_size = strtoul (optarg, &tmp, 0);
-                       if (*tmp || block_size > 4096) {
+                       block_size = parse_uint(optarg, "block size");
+                       if (block_size > 4096) {
                                com_err (program_name, 0,
                                         _("bad block size - %s"), optarg);
                                exit (1);
@@ -879,38 +911,29 @@ int main (int argc, char ** argv)
                        break;
                case 'w':
                        if (w_flag)
-                               usage();
+                               exclusive_usage();
                        test_func = test_rw;
                        w_flag = 1;
                        break;
                case 'n':
                        if (w_flag)
-                               usage();
+                               exclusive_usage();
                        test_func = test_nd;
                        w_flag = 2;
                        break;
                case 'c':
-                       blocks_at_once = strtoul (optarg, &tmp, 0);
-                       if (*tmp) {
-                               com_err (program_name, 0,
-                                        "bad simultaneous block count - %s", optarg);
-                               exit (1);
-                       }
+                       blocks_at_once = parse_uint(optarg, "blocks at once");
                        break;
                case 'p':
-                       num_passes = strtoul (optarg, &tmp, 0);
-                       if (*tmp) {
-                               com_err (program_name, 0,
-                                   "bad number of clean passes - %s", optarg);
-                               exit (1);
-                       }
+                       num_passes = parse_uint(optarg, 
+                                               "number of clean passes");
                        break;
                case 'h':
                        host_device_name = optarg;
                        break;
                case 't':
                        if (t_flag + 1 > t_max) {
-                               unsigned long *t_patts_new;
+                               unsigned int *t_patts_new;
 
                                t_patts_new = realloc(t_patts, t_max + T_INC);
                                if (!t_patts_new) {
@@ -926,18 +949,15 @@ int main (int argc, char ** argv)
                        if (!strcmp(optarg, "r") || !strcmp(optarg,"random")) {
                                t_patts[t_flag++] = ~0;
                        } else {
-                               pattern = strtoul(optarg, &tmp, 0);
-                               if (*tmp) {
-                                       com_err(program_name, 0,
-                                       _("invalid test_pattern: %s\n"),
-                                               optarg);
-                                       exit(1);
-                               }
-                               if (pattern == (unsigned long) ~0)
+                               pattern = parse_uint(optarg, "test pattern");
+                               if (pattern == (unsigned int) ~0)
                                        pattern = 0xffff;
                                t_patts[t_flag++] = pattern;
                        }
                        break;
+               case 'X':
+                       exclusive_ok++;
+                       break;
                default:
                        usage();
                }
@@ -949,7 +969,7 @@ int main (int argc, char ** argv)
                          "in read-only mode"));
                        exit(1);
                }
-               if (t_patts && (t_patts[0] == (unsigned long) ~0)) {
+               if (t_patts && (t_patts[0] == (unsigned int) ~0)) {
                        com_err(program_name, 0,
                        _("Random test_pattern is not allowed "
                          "in read-only mode"));
@@ -975,24 +995,19 @@ int main (int argc, char ** argv)
                        exit(1);
                }
        } else {
-               last_block = strtoul (argv[optind], &tmp, 0);
-               if (*tmp) {
-                       com_err (program_name, 0, _("invalid blocks count - %s"),
-                                argv[optind]);
-                       exit (1);
-               }
+               errno = 0;
+               last_block = parse_uint(argv[optind], "last block");
+               printf("last_block = %d (%s)\n", last_block, argv[optind]);
+               last_block++;
                optind++;
        }
        if (optind <= argc-1) {
-               from_count = strtoul (argv[optind], &tmp, 0);
-               if (*tmp) {
-                       com_err (program_name, 0, _("invalid starting block - %s"),
-                                argv[optind]);
-                       exit (1);
-               }
+               errno = 0;
+               from_count = parse_uint(argv[optind], "start block");
+               printf("from_count = %d\n", from_count);
        } else from_count = 0;
        if (from_count >= last_block) {
-           com_err (program_name, 0, _("invalid blocks range: %lu-%lu"),
+           com_err (program_name, 0, _("invalid starting block (%lu): must be less than %lu"),
                     (unsigned long) from_count, (unsigned long) last_block);
            exit (1);
        }
@@ -1047,7 +1062,7 @@ int main (int argc, char ** argv)
        errcode = ext2fs_badblocks_list_create(&bb_list,0);
        if (errcode) {
                com_err (program_name, errcode,
-                        _("creating in-memory bad blocks list"));
+                        _("while creating in-memory bad blocks list"));
                exit (1);
        }
 
@@ -1062,7 +1077,7 @@ int main (int argc, char ** argv)
                                default:
                                        errcode = ext2fs_badblocks_list_add(bb_list,next_bad);
                                        if (errcode) {
-                                               com_err (program_name, errcode, _("adding to in-memory bad block list"));
+                                               com_err (program_name, errcode, _("while adding to in-memory bad block list"));
                                                exit (1);
                                        }
                                        continue;