Home | History | Annotate | Download | only in ext2fs
      1 /*
      2  * This testing program makes sure the badblocks implementation works.
      3  *
      4  * Copyright (C) 1996 by Theodore Ts'o.
      5  *
      6  * %Begin-Header%
      7  * This file may be redistributed under the terms of the GNU Library
      8  * General Public License, version 2.
      9  * %End-Header%
     10  */
     11 
     12 #include <stdio.h>
     13 #include <string.h>
     14 #if HAVE_UNISTD_H
     15 #include <unistd.h>
     16 #endif
     17 #include <fcntl.h>
     18 #include <time.h>
     19 #include <sys/stat.h>
     20 #include <sys/types.h>
     21 #if HAVE_ERRNO_H
     22 #include <errno.h>
     23 #endif
     24 
     25 #include "ext2_fs.h"
     26 #include "ext2fs.h"
     27 
     28 #define ADD_BLK	0x0001
     29 #define DEL_BLK	0x0002
     30 
     31 blk_t test1[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 0 };
     32 blk_t test2[] = { 11, 10, 9, 8, 7, 6, 5, 4, 3, 3, 2, 1 };
     33 blk_t test3[] = { 3, 1, 4, 5, 9, 2, 7, 10, 5, 6, 10, 8, 0 };
     34 blk_t test4[] = { 20, 50, 12, 17, 13, 2, 66, 23, 56, 0 };
     35 blk_t test4a[] = {
     36  	20, 1,
     37 	50, 1,
     38 	3, 0,
     39 	17, 1,
     40 	18, 0,
     41 	16, 0,
     42 	11, 0,
     43 	12, 1,
     44 	13, 1,
     45 	14, 0,
     46 	80, 0,
     47 	45, 0,
     48 	66, 1,
     49 	0 };
     50 blk_t test5[] = { 31, 20, 17, 51, 23, 1, 56, 57, 0 };
     51 blk_t test5a[] = {
     52 	50, ADD_BLK,
     53 	51, DEL_BLK,
     54 	57, DEL_BLK,
     55 	66, ADD_BLK,
     56 	31, DEL_BLK,
     57 	12, ADD_BLK,
     58 	2, ADD_BLK,
     59 	13, ADD_BLK,
     60 	1, DEL_BLK,
     61 	0
     62 	};
     63 
     64 
     65 static int test_fail = 0;
     66 static int test_expected_fail = 0;
     67 
     68 static errcode_t create_test_list(blk_t *vec, badblocks_list *ret)
     69 {
     70 	errcode_t	retval;
     71 	badblocks_list	bb;
     72 	int		i;
     73 
     74 	retval = ext2fs_badblocks_list_create(&bb, 5);
     75 	if (retval) {
     76 		com_err("create_test_list", retval, "while creating list");
     77 		return retval;
     78 	}
     79 	for (i=0; vec[i]; i++) {
     80 		retval = ext2fs_badblocks_list_add(bb, vec[i]);
     81 		if (retval) {
     82 			com_err("create_test_list", retval,
     83 				"while adding test vector %d", i);
     84 			ext2fs_badblocks_list_free(bb);
     85 			return retval;
     86 		}
     87 	}
     88 	*ret = bb;
     89 	return 0;
     90 }
     91 
     92 static void print_list(badblocks_list bb, int verify)
     93 {
     94 	errcode_t	retval;
     95 	badblocks_iterate	iter;
     96 	blk_t			blk;
     97 	int			i, ok;
     98 
     99 	retval = ext2fs_badblocks_list_iterate_begin(bb, &iter);
    100 	if (retval) {
    101 		com_err("print_list", retval, "while setting up iterator");
    102 		return;
    103 	}
    104 	ok = i = 1;
    105 	while (ext2fs_badblocks_list_iterate(iter, &blk)) {
    106 		printf("%u ", blk);
    107 		if (i++ != blk)
    108 			ok = 0;
    109 	}
    110 	ext2fs_badblocks_list_iterate_end(iter);
    111 	if (verify) {
    112 		if (ok)
    113 			printf("--- OK");
    114 		else {
    115 			printf("--- NOT OK");
    116 			test_fail++;
    117 		}
    118 	}
    119 }
    120 
    121 static void validate_test_seq(badblocks_list bb, blk_t *vec)
    122 {
    123 	int	i, match, ok;
    124 
    125 	for (i = 0; vec[i]; i += 2) {
    126 		match = ext2fs_badblocks_list_test(bb, vec[i]);
    127 		if (match == vec[i+1])
    128 			ok = 1;
    129 		else {
    130 			ok = 0;
    131 			test_fail++;
    132 		}
    133 		printf("\tblock %u is %s --- %s\n", vec[i],
    134 		       match ? "present" : "absent",
    135 		       ok ? "OK" : "NOT OK");
    136 	}
    137 }
    138 
    139 static void do_test_seq(badblocks_list bb, blk_t *vec)
    140 {
    141 	int	i, match;
    142 
    143 	for (i = 0; vec[i]; i += 2) {
    144 		switch (vec[i+1]) {
    145 		case ADD_BLK:
    146 			ext2fs_badblocks_list_add(bb, vec[i]);
    147 			match = ext2fs_badblocks_list_test(bb, vec[i]);
    148 			printf("Adding block %u --- now %s\n", vec[i],
    149 			       match ? "present" : "absent");
    150 			if (!match) {
    151 				printf("FAILURE!\n");
    152 				test_fail++;
    153 			}
    154 			break;
    155 		case DEL_BLK:
    156 			ext2fs_badblocks_list_del(bb, vec[i]);
    157 			match = ext2fs_badblocks_list_test(bb, vec[i]);
    158 			printf("Removing block %u --- now %s\n", vec[i],
    159 			       ext2fs_badblocks_list_test(bb, vec[i]) ?
    160 			       "present" : "absent");
    161 			if (match) {
    162 				printf("FAILURE!\n");
    163 				test_fail++;
    164 			}
    165 			break;
    166 		}
    167 	}
    168 }
    169 
    170 
    171 int file_test(badblocks_list bb)
    172 {
    173 	badblocks_list new_bb = 0;
    174 	errcode_t	retval;
    175 	FILE	*f;
    176 
    177 	f = tmpfile();
    178 	if (!f) {
    179 		fprintf(stderr, "Error opening temp file: %s\n",
    180 			error_message(errno));
    181 		return 1;
    182 	}
    183 	retval = ext2fs_write_bb_FILE(bb, 0, f);
    184 	if (retval) {
    185 		com_err("file_test", retval, "while writing bad blocks");
    186 		return 1;
    187 	}
    188 
    189 	rewind(f);
    190 	retval = ext2fs_read_bb_FILE2(0, f, &new_bb, 0, 0);
    191 	if (retval) {
    192 		com_err("file_test", retval, "while reading bad blocks");
    193 		return 1;
    194 	}
    195 	fclose(f);
    196 
    197 	if (ext2fs_badblocks_equal(bb, new_bb)) {
    198 		printf("Block bitmap matched after reading and writing.\n");
    199 	} else {
    200 		printf("Block bitmap NOT matched.\n");
    201 		test_fail++;
    202 	}
    203 	return 0;
    204 }
    205 
    206 static void invalid_proc(ext2_filsys fs, blk_t blk)
    207 {
    208 	if (blk == 34500) {
    209 		printf("Expected invalid block\n");
    210 		test_expected_fail++;
    211 	} else {
    212 		printf("Invalid block #: %u\n", blk);
    213 		test_fail++;
    214 	}
    215 }
    216 
    217 int file_test_invalid(badblocks_list bb)
    218 {
    219 	badblocks_list new_bb = 0;
    220 	errcode_t	retval;
    221 	ext2_filsys 	fs;
    222 	FILE	*f;
    223 
    224 	fs = malloc(sizeof(struct struct_ext2_filsys));
    225 	memset(fs, 0, sizeof(struct struct_ext2_filsys));
    226 	fs->magic = EXT2_ET_MAGIC_EXT2FS_FILSYS;
    227 	fs->super = malloc(SUPERBLOCK_SIZE);
    228 	memset(fs->super, 0, SUPERBLOCK_SIZE);
    229 	fs->super->s_first_data_block = 1;
    230 	fs->super->s_blocks_count = 100;
    231 
    232 	f = tmpfile();
    233 	if (!f) {
    234 		fprintf(stderr, "Error opening temp file: %s\n",
    235 			error_message(errno));
    236 		return 1;
    237 	}
    238 	retval = ext2fs_write_bb_FILE(bb, 0, f);
    239 	if (retval) {
    240 		com_err("file_test", retval, "while writing bad blocks");
    241 		return 1;
    242 	}
    243 	fprintf(f, "34500\n");
    244 
    245 	rewind(f);
    246 	test_expected_fail = 0;
    247 	retval = ext2fs_read_bb_FILE(fs, f, &new_bb, invalid_proc);
    248 	if (retval) {
    249 		com_err("file_test", retval, "while reading bad blocks");
    250 		return 1;
    251 	}
    252 	fclose(f);
    253 	if (!test_expected_fail) {
    254 		printf("Expected test failure didn't happen!\n");
    255 		test_fail++;
    256 	}
    257 
    258 
    259 	if (ext2fs_badblocks_equal(bb, new_bb)) {
    260 		printf("Block bitmap matched after reading and writing.\n");
    261 	} else {
    262 		printf("Block bitmap NOT matched.\n");
    263 		test_fail++;
    264 	}
    265 	return 0;
    266 }
    267 
    268 int main(int argc, char **argv)
    269 {
    270 	badblocks_list bb1, bb2, bb3, bb4, bb5;
    271 	int	equal;
    272 	errcode_t	retval;
    273 
    274 	add_error_table(&et_ext2_error_table);
    275 
    276 	bb1 = bb2 = bb3 = bb4 = bb5 = 0;
    277 
    278 	printf("test1: ");
    279 	retval = create_test_list(test1, &bb1);
    280 	if (retval == 0)
    281 		print_list(bb1, 1);
    282 	printf("\n");
    283 
    284 	printf("test2: ");
    285 	retval = create_test_list(test2, &bb2);
    286 	if (retval == 0)
    287 		print_list(bb2, 1);
    288 	printf("\n");
    289 
    290 	printf("test3: ");
    291 	retval = create_test_list(test3, &bb3);
    292 	if (retval == 0)
    293 		print_list(bb3, 1);
    294 	printf("\n");
    295 
    296 	printf("test4: ");
    297 	retval = create_test_list(test4, &bb4);
    298 	if (retval == 0) {
    299 		print_list(bb4, 0);
    300 		printf("\n");
    301 		validate_test_seq(bb4, test4a);
    302 	}
    303 	printf("\n");
    304 
    305 	printf("test5: ");
    306 	retval = create_test_list(test5, &bb5);
    307 	if (retval == 0) {
    308 		print_list(bb5, 0);
    309 		printf("\n");
    310 		do_test_seq(bb5, test5a);
    311 		printf("After test5 sequence: ");
    312 		print_list(bb5, 0);
    313 		printf("\n");
    314 	}
    315 	printf("\n");
    316 
    317 	if (bb1 && bb2 && bb3 && bb4 && bb5) {
    318 		printf("Comparison tests:\n");
    319 		equal = ext2fs_badblocks_equal(bb1, bb2);
    320 		printf("bb1 and bb2 are %sequal.\n", equal ? "" : "NOT ");
    321 		if (equal)
    322 			test_fail++;
    323 
    324 		equal = ext2fs_badblocks_equal(bb1, bb3);
    325 		printf("bb1 and bb3 are %sequal.\n", equal ? "" : "NOT ");
    326 		if (!equal)
    327 			test_fail++;
    328 
    329 		equal = ext2fs_badblocks_equal(bb1, bb4);
    330 		printf("bb1 and bb4 are %sequal.\n", equal ? "" : "NOT ");
    331 		if (equal)
    332 			test_fail++;
    333 
    334 		equal = ext2fs_badblocks_equal(bb4, bb5);
    335 		printf("bb4 and bb5 are %sequal.\n", equal ? "" : "NOT ");
    336 		if (!equal)
    337 			test_fail++;
    338 		printf("\n");
    339 	}
    340 
    341 	file_test(bb4);
    342 
    343 	file_test_invalid(bb4);
    344 
    345 	if (test_fail == 0)
    346 		printf("ext2fs library badblocks tests checks out OK!\n");
    347 
    348 	if (bb1)
    349 		ext2fs_badblocks_list_free(bb1);
    350 	if (bb2)
    351 		ext2fs_badblocks_list_free(bb2);
    352 	if (bb3)
    353 		ext2fs_badblocks_list_free(bb3);
    354 	if (bb4)
    355 		ext2fs_badblocks_list_free(bb4);
    356 
    357 	return test_fail;
    358 
    359 }
    360