#!/bin/bash
#
# loop through all of our bad images and make sure fsck repairs them properly
-#
-# It's GPL, same as everything else in this tree.
-#
-here=`pwd`
+LANG=C
+SCRIPT_DIR=$(dirname $(readlink -f "$0"))
+TOP=$(readlink -f "$SCRIPT_DIR/../")
+TEST_DEV=${TEST_DEV:-}
+RESULTS="$TOP/tests/fsck-tests-results.txt"
+IMAGE="$TOP/tests/test.img"
-_fail()
-{
- echo "$*" | tee -a fsck-tests-results.txt
- exit 1
-}
+source "$TOP/tests/common"
-rm -f fsck-tests-results.txt
+export TOP
+export RESULTS
+export LANG
+export IMAGE
+export TEST_DEV
-for i in $(find $here/tests/fsck-tests -name '*.img')
-do
- echo "testing image $i" >> fsck-tests-results.txt
- $here/btrfs-image -r $i test.img >> fsck-tests-results.txt 2>&1 \
- || _fail "restore failed"
- $here/btrfsck test.img >> fsck-test-results.txt 2>&1
- [ $? -eq 0 ] && _fail "btrfsck should have detected corruption"
+rm -f "$RESULTS"
+
+# test rely on corrupting blocks tool
+check_prereq btrfs-corrupt-block
+check_prereq btrfs-image
+check_prereq btrfs
+check_prereq btrfstune
+check_kernel_support
- $here/btrfsck --repair test.img >> fsck-test-results.txt 2>&1 || \
- _fail "btrfsck should have repaired the image"
+run_one_test() {
+ local testname
- $here/btrfsck test.img >> fsck-test-results.txt 2>&1 || \
- _fail "btrfsck did not correct corruption"
+ testname="$1"
+ echo " [TEST/fsck] $(basename $testname)"
+ cd "$testname"
+ echo "=== Entering $testname" >> "$RESULTS"
+ if [ -x test.sh ]; then
+ # Type 2
+ ./test.sh
+ if [ $? -ne 0 ]; then
+ if [[ $TEST_LOG =~ dump ]]; then
+ cat "$RESULTS"
+ fi
+ _fail "test failed for case $(basename $testname)"
+ fi
+ else
+ # Type 1
+ check_all_images
+ fi
+ cd "$TOP"
+}
+
+# Each dir contains one type of error for btrfsck test.
+# Each dir must be one of the following 2 types:
+# 1) Only btrfs-image dump
+# Only contains one or several btrfs-image dumps (.img)
+# Each image will be tested by generic test routine
+# (btrfsck --repair and btrfsck).
+# This is for case that btree-healthy images.
+# 2) Custom test script
+# This dir contains test.sh which will do custom image
+# generation/check/verification.
+# This is for case btrfs-image can't dump or case needs extra
+# check/verify
+
+for i in $(find "$TOP/tests/fsck-tests" -maxdepth 1 -mindepth 1 -type d \
+ ${TEST:+-name "$TEST"} | sort)
+do
+ run_one_test "$i"
done