1 files changed, 88 insertions, 0 deletions
diff --git a/tests/test_fsck.sh b/tests/test_fsck.sh
new file mode 100755
index 0000000..678b331
--- /dev/null
+++ b/tests/test_fsck.sh
@@ -0,0 +1,88 @@
+#!/usr/bin/env bash
+
+TESTCASE_DIR=$1
+NEED_LOOPDEV=$2
+IMAGE_FILE=exfat.img
+FSCK_PROG=${FSCK1:-"fsck.exfat"}
+FSCK_PROG_2=${FSCK2:-"fsck.exfat"}
+FSCK_OPTS="-y -s"
+PASS_COUNT=0
+
+cleanup() {
+	echo ""
+	echo "Passed ${PASS_COUNT} of ${TEST_COUNT}"
+	if [ ${PASS_COUNT} -ne ${TEST_COUNT} ]; then
+		exit 1
+	else
+		exit 0
+	fi
+}
+
+if [ $# -eq 0 ]; then
+	TESTCASE_DIRS=$(find . -mindepth 1 -maxdepth 1 -type d)
+	TEST_COUNT=$(find . -mindepth 1 -maxdepth 1 -type d | wc -l)
+else
+	TESTCASE_DIRS=$@
+	TEST_COUNT=$#
+fi
+
+for TESTCASE_DIR in $TESTCASE_DIRS; do
+	if [ ! -e "${TESTCASE_DIR}/${IMAGE_FILE}.tar.xz" ]; then
+		TEST_COUNT=$((TEST_COUNT - 1))
+		continue
+	fi
+
+	echo "Running ${TESTCASE_DIR}"
+	echo "-----------------------------------"
+
+	# Set up image file as loop device
+	tar -C . -xf "${TESTCASE_DIR}/${IMAGE_FILE}.tar.xz"
+	if [ $NEED_LOOPDEV ]; then
+		DEV_FILE=$(losetup -f "${IMAGE_FILE}" --show)
+	else
+		DEV_FILE=$IMAGE_FILE
+	fi
+
+	# Run fsck to detect corruptions
+	$FSCK_PROG "$DEV_FILE" | grep -q "ERROR:\|corrupted"
+	if [ $? -ne 0 ]; then
+		echo ""
+		echo "Failed to detect corruption for ${TESTCASE_DIR}"
+		if [ $NEED_LOOPDEV ]; then
+			losetup -d "${DEV_FILE}"
+		fi
+		cleanup
+	fi
+
+	# Run fsck for repair
+	$FSCK_PROG $FSCK_OPTS "$DEV_FILE"
+	if [ $? -ne 1 ] && [ $? -ne 0 ]; then
+		echo ""
+		echo "Failed to repair ${TESTCASE_DIR}"
+		if [ $NEED_LOOPDEV ]; then
+			losetup -d "${DEV_FILE}"
+		fi
+		cleanup
+	fi
+
+	echo ""
+	# Run fsck again
+	$FSCK_PROG_2 "$DEV_FILE"
+	if [ $? -ne 0 ]; then
+		echo ""
+		echo "Failed, corrupted ${TESTCASE_DIR}"
+		if [ $NEED_LOOPDEV ]; then
+			losetup -d "${DEV_FILE}"
+		fi
+		cleanup
+	fi
+
+	echo ""
+	echo "Passed ${TESTCASE_DIR}"
+	PASS_COUNT=$((PASS_COUNT + 1))
+
+	if [ $NEED_LOOPDEV ]; then
+		losetup -d "${DEV_FILE}"
+	fi
+done
+cleanup