| /* |
| * 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 Library |
| * General Public License, version 2. |
| * %End-Header% |
| */ |
| |
| #include <stdio.h> |
| #include <string.h> |
| #if HAVE_UNISTD_H |
| #include <unistd.h> |
| #endif |
| #include <fcntl.h> |
| #include <time.h> |
| #include <sys/stat.h> |
| #include <sys/types.h> |
| #if HAVE_ERRNO_H |
| #include <errno.h> |
| #endif |
| |
| #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[] = { 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, |
| 50, 1, |
| 3, 0, |
| 17, 1, |
| 18, 0, |
| 16, 0, |
| 11, 0, |
| 12, 1, |
| 13, 1, |
| 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 = 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 = ext2fs_badblocks_list_add(bb, vec[i]); |
| if (retval) { |
| com_err("create_test_list", retval, |
| "while adding test vector %d", i); |
| ext2fs_badblocks_list_free(bb); |
| return retval; |
| } |
| } |
| *ret = bb; |
| return 0; |
| } |
| |
| static void print_list(badblocks_list bb, int verify) |
| { |
| errcode_t retval; |
| badblocks_iterate iter; |
| blk_t blk; |
| int i, ok; |
| |
| retval = ext2fs_badblocks_list_iterate_begin(bb, &iter); |
| if (retval) { |
| com_err("print_list", retval, "while setting up iterator"); |
| return; |
| } |
| ok = i = 1; |
| while (ext2fs_badblocks_list_iterate(iter, &blk)) { |
| printf("%u ", blk); |
| if (i++ != blk) |
| ok = 0; |
| } |
| ext2fs_badblocks_list_iterate_end(iter); |
| if (verify) { |
| if (ok) |
| printf("--- OK"); |
| else { |
| printf("--- NOT OK"); |
| test_fail++; |
| } |
| } |
| } |
| |
| static void validate_test_seq(badblocks_list bb, blk_t *vec) |
| { |
| int i, match, ok; |
| |
| for (i = 0; vec[i]; i += 2) { |
| match = ext2fs_badblocks_list_test(bb, vec[i]); |
| if (match == vec[i+1]) |
| ok = 1; |
| else { |
| ok = 0; |
| test_fail++; |
| } |
| printf("\tblock %u is %s --- %s\n", vec[i], |
| match ? "present" : "absent", |
| ok ? "OK" : "NOT OK"); |
| } |
| } |
| |
| 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 new_bb = 0; |
| errcode_t retval; |
| FILE *f; |
| |
| 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; |
| ext2fs_blocks_count_set(fs->super, 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; |
| |
| add_error_table(&et_ext2_error_table); |
| |
| 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, &bb2); |
| if (retval == 0) |
| print_list(bb2, 1); |
| printf("\n"); |
| |
| printf("test3: "); |
| 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(bb4, 0); |
| printf("\n"); |
| validate_test_seq(bb4, test4a); |
| } |
| printf("\n"); |
| |
| printf("test5: "); |
| retval = create_test_list(test5, &bb5); |
| if (retval == 0) { |
| print_list(bb5, 0); |
| printf("\n"); |
| do_test_seq(bb5, test5a); |
| printf("After test5 sequence: "); |
| print_list(bb5, 0); |
| printf("\n"); |
| } |
| 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; |
| |
| } |