Whamcloud - gitweb
Remove trailing whitespace for the entire source tree
[tools/e2fsprogs.git] / lib / ext2fs / tst_badblocks.c
index d071406..358da20 100644 (file)
@@ -2,7 +2,7 @@
  * This testing program makes sure the badblocks implementation works.
  *
  * Copyright (C) 1996 by Theodore Ts'o.
- * 
+ *
  * %Begin-Header%
  * This file may be redistributed under the terms of the GNU Public
  * License.
@@ -11,8 +11,9 @@
 
 #include <stdio.h>
 #include <string.h>
+#if HAVE_UNISTD_H
 #include <unistd.h>
-#include <stdlib.h>
+#endif
 #include <fcntl.h>
 #include <time.h>
 #include <sys/stat.h>
 #include <errno.h>
 #endif
 
-#include <linux/ext2_fs.h>
-
+#include "ext2_fs.h"
 #include "ext2fs.h"
 
+#define ADD_BLK        0x0001
+#define DEL_BLK        0x0002
+
 blk_t test1[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 0 };
-blk_t test2[] = { 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0 };
-blk_t test3[] = { 3, 1, 4, 5, 9, 2, 7, 10, 6, 8, 0 };
+blk_t test2[] = { 11, 10, 9, 8, 7, 6, 5, 4, 3, 3, 2, 1 };
+blk_t test3[] = { 3, 1, 4, 5, 9, 2, 7, 10, 5, 6, 10, 8, 0 };
 blk_t test4[] = { 20, 50, 12, 17, 13, 2, 66, 23, 56, 0 };
 blk_t test4a[] = {
        20, 1,
@@ -39,31 +42,46 @@ blk_t test4a[] = {
        11, 0,
        12, 1,
        13, 1,
-       14, 0, 
+       14, 0,
        80, 0,
        45, 0,
        66, 1,
        0 };
+blk_t test5[] = { 31, 20, 17, 51, 23, 1, 56, 57, 0 };
+blk_t test5a[] = {
+       50, ADD_BLK,
+       51, DEL_BLK,
+       57, DEL_BLK,
+       66, ADD_BLK,
+       31, DEL_BLK,
+       12, ADD_BLK,
+       2, ADD_BLK,
+       13, ADD_BLK,
+       1, DEL_BLK,
+       0
+       };
+
 
 static int test_fail = 0;
+static int test_expected_fail = 0;
 
 static errcode_t create_test_list(blk_t *vec, badblocks_list *ret)
 {
        errcode_t       retval;
        badblocks_list  bb;
        int             i;
-       
-       retval = badblocks_list_create(&bb, 5);
+
+       retval = ext2fs_badblocks_list_create(&bb, 5);
        if (retval) {
                com_err("create_test_list", retval, "while creating list");
                return retval;
        }
        for (i=0; vec[i]; i++) {
-               retval = badblocks_list_add(bb, vec[i]);
+               retval = ext2fs_badblocks_list_add(bb, vec[i]);
                if (retval) {
                        com_err("create_test_list", retval,
                                "while adding test vector %d", i);
-                       badblocks_list_free(bb);
+                       ext2fs_badblocks_list_free(bb);
                        return retval;
                }
        }
@@ -77,19 +95,19 @@ static void print_list(badblocks_list bb, int verify)
        badblocks_iterate       iter;
        blk_t                   blk;
        int                     i, ok;
-       
-       retval = badblocks_list_iterate_begin(bb, &iter);
+
+       retval = ext2fs_badblocks_list_iterate_begin(bb, &iter);
        if (retval) {
                com_err("print_list", retval, "while setting up iterator");
                return;
        }
        ok = i = 1;
-       while (badblocks_list_iterate(iter, &blk)) {
-               printf("%d ", blk);
+       while (ext2fs_badblocks_list_iterate(iter, &blk)) {
+               printf("%u ", blk);
                if (i++ != blk)
                        ok = 0;
        }
-       badblocks_list_iterate_end(iter);
+       ext2fs_badblocks_list_iterate_end(iter);
        if (verify) {
                if (ok)
                        printf("--- OK");
@@ -105,58 +123,235 @@ static void validate_test_seq(badblocks_list bb, blk_t *vec)
        int     i, match, ok;
 
        for (i = 0; vec[i]; i += 2) {
-               match = badblocks_list_test(bb, vec[i]);
+               match = ext2fs_badblocks_list_test(bb, vec[i]);
                if (match == vec[i+1])
                        ok = 1;
                else {
                        ok = 0;
                        test_fail++;
                }
-               printf("\tblock %d is %s --- %s\n", vec[i],
+               printf("\tblock %u is %s --- %s\n", vec[i],
                       match ? "present" : "absent",
                       ok ? "OK" : "NOT OK");
        }
 }
 
-int main(int argc, char *argv)
+static void do_test_seq(badblocks_list bb, blk_t *vec)
+{
+       int     i, match;
+
+       for (i = 0; vec[i]; i += 2) {
+               switch (vec[i+1]) {
+               case ADD_BLK:
+                       ext2fs_badblocks_list_add(bb, vec[i]);
+                       match = ext2fs_badblocks_list_test(bb, vec[i]);
+                       printf("Adding block %u --- now %s\n", vec[i],
+                              match ? "present" : "absent");
+                       if (!match) {
+                               printf("FAILURE!\n");
+                               test_fail++;
+                       }
+                       break;
+               case DEL_BLK:
+                       ext2fs_badblocks_list_del(bb, vec[i]);
+                       match = ext2fs_badblocks_list_test(bb, vec[i]);
+                       printf("Removing block %u --- now %s\n", vec[i],
+                              ext2fs_badblocks_list_test(bb, vec[i]) ?
+                              "present" : "absent");
+                       if (match) {
+                               printf("FAILURE!\n");
+                               test_fail++;
+                       }
+                       break;
+               }
+       }
+}
+
+
+int file_test(badblocks_list bb)
 {
-       badblocks_list bb;
+       badblocks_list new_bb = 0;
        errcode_t       retval;
+       FILE    *f;
 
-       printf("test1: ");
-       retval = create_test_list(test1, &bb);
-       if (retval == 0) {
-               print_list(bb, 1);
-               badblocks_list_free(bb);
+       f = tmpfile();
+       if (!f) {
+               fprintf(stderr, "Error opening temp file: %s\n",
+                       error_message(errno));
+               return 1;
+       }
+       retval = ext2fs_write_bb_FILE(bb, 0, f);
+       if (retval) {
+               com_err("file_test", retval, "while writing bad blocks");
+               return 1;
+       }
+
+       rewind(f);
+       retval = ext2fs_read_bb_FILE2(0, f, &new_bb, 0, 0);
+       if (retval) {
+               com_err("file_test", retval, "while reading bad blocks");
+               return 1;
+       }
+       fclose(f);
+
+       if (ext2fs_badblocks_equal(bb, new_bb)) {
+               printf("Block bitmap matched after reading and writing.\n");
+       } else {
+               printf("Block bitmap NOT matched.\n");
+               test_fail++;
+       }
+       return 0;
+}
+
+static void invalid_proc(ext2_filsys fs, blk_t blk)
+{
+       if (blk == 34500) {
+               printf("Expected invalid block\n");
+               test_expected_fail++;
+       } else {
+               printf("Invalid block #: %u\n", blk);
+               test_fail++;
+       }
+}
+
+int file_test_invalid(badblocks_list bb)
+{
+       badblocks_list new_bb = 0;
+       errcode_t       retval;
+       ext2_filsys     fs;
+       FILE    *f;
+
+       fs = malloc(sizeof(struct struct_ext2_filsys));
+       memset(fs, 0, sizeof(struct struct_ext2_filsys));
+       fs->magic = EXT2_ET_MAGIC_EXT2FS_FILSYS;
+       fs->super = malloc(SUPERBLOCK_SIZE);
+       memset(fs->super, 0, SUPERBLOCK_SIZE);
+       fs->super->s_first_data_block = 1;
+       fs->super->s_blocks_count = 100;
+
+       f = tmpfile();
+       if (!f) {
+               fprintf(stderr, "Error opening temp file: %s\n",
+                       error_message(errno));
+               return 1;
+       }
+       retval = ext2fs_write_bb_FILE(bb, 0, f);
+       if (retval) {
+               com_err("file_test", retval, "while writing bad blocks");
+               return 1;
        }
+       fprintf(f, "34500\n");
+
+       rewind(f);
+       test_expected_fail = 0;
+       retval = ext2fs_read_bb_FILE(fs, f, &new_bb, invalid_proc);
+       if (retval) {
+               com_err("file_test", retval, "while reading bad blocks");
+               return 1;
+       }
+       fclose(f);
+       if (!test_expected_fail) {
+               printf("Expected test failure didn't happen!\n");
+               test_fail++;
+       }
+
+
+       if (ext2fs_badblocks_equal(bb, new_bb)) {
+               printf("Block bitmap matched after reading and writing.\n");
+       } else {
+               printf("Block bitmap NOT matched.\n");
+               test_fail++;
+       }
+       return 0;
+}
+
+int main(int argc, char **argv)
+{
+       badblocks_list bb1, bb2, bb3, bb4, bb5;
+       int     equal;
+       errcode_t       retval;
+
+       bb1 = bb2 = bb3 = bb4 = bb5 = 0;
+
+       printf("test1: ");
+       retval = create_test_list(test1, &bb1);
+       if (retval == 0)
+               print_list(bb1, 1);
        printf("\n");
-       
+
        printf("test2: ");
-       retval = create_test_list(test2, &bb);
-       if (retval == 0) {
-               print_list(bb, 1);
-               badblocks_list_free(bb);
-       }
+       retval = create_test_list(test2, &bb2);
+       if (retval == 0)
+               print_list(bb2, 1);
        printf("\n");
 
        printf("test3: ");
-       retval = create_test_list(test3, &bb);
+       retval = create_test_list(test3, &bb3);
+       if (retval == 0)
+               print_list(bb3, 1);
+       printf("\n");
+
+       printf("test4: ");
+       retval = create_test_list(test4, &bb4);
        if (retval == 0) {
-               print_list(bb, 1);
-               badblocks_list_free(bb);
+               print_list(bb4, 0);
+               printf("\n");
+               validate_test_seq(bb4, test4a);
        }
        printf("\n");
-       
-       printf("test4: ");
-       retval = create_test_list(test4, &bb);
+
+       printf("test5: ");
+       retval = create_test_list(test5, &bb5);
        if (retval == 0) {
-               print_list(bb, 0);
+               print_list(bb5, 0);
+               printf("\n");
+               do_test_seq(bb5, test5a);
+               printf("After test5 sequence: ");
+               print_list(bb5, 0);
                printf("\n");
-               validate_test_seq(bb, test4a);
-               badblocks_list_free(bb);
        }
        printf("\n");
+
+       if (bb1 && bb2 && bb3 && bb4 && bb5) {
+               printf("Comparison tests:\n");
+               equal = ext2fs_badblocks_equal(bb1, bb2);
+               printf("bb1 and bb2 are %sequal.\n", equal ? "" : "NOT ");
+               if (equal)
+                       test_fail++;
+
+               equal = ext2fs_badblocks_equal(bb1, bb3);
+               printf("bb1 and bb3 are %sequal.\n", equal ? "" : "NOT ");
+               if (!equal)
+                       test_fail++;
+
+               equal = ext2fs_badblocks_equal(bb1, bb4);
+               printf("bb1 and bb4 are %sequal.\n", equal ? "" : "NOT ");
+               if (equal)
+                       test_fail++;
+
+               equal = ext2fs_badblocks_equal(bb4, bb5);
+               printf("bb4 and bb5 are %sequal.\n", equal ? "" : "NOT ");
+               if (!equal)
+                       test_fail++;
+               printf("\n");
+       }
+
+       file_test(bb4);
+
+       file_test_invalid(bb4);
+
        if (test_fail == 0)
                printf("ext2fs library badblocks tests checks out OK!\n");
+
+       if (bb1)
+               ext2fs_badblocks_list_free(bb1);
+       if (bb2)
+               ext2fs_badblocks_list_free(bb2);
+       if (bb3)
+               ext2fs_badblocks_list_free(bb3);
+       if (bb4)
+               ext2fs_badblocks_list_free(bb4);
+
        return test_fail;
+
 }