/*
* 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;
}