text: /*
   * 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.
   * %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;
  	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, &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;
  
  }
  