generic: test fs-verity elide and patch extensions

Signed-off-by: Eric Biggers <ebiggers@google.com>

3 files changed, 288 insertions, 0 deletions

diff --git a/tests/generic/905 b/tests/generic/905
new file mode 100755
index 0000000000..122f46d30f
--- /dev/null
+++ b/tests/generic/905
@@ -0,0 +1,195 @@
+#! /bin/bash
+# FS QA Test generic/905
+#
+# Test fs-verity elide and patch extensions
+#
+#-----------------------------------------------------------------------
+# Copyright (c) 2018 Google, Inc.  All Rights Reserved.
+#
+# Author: Eric Biggers <ebiggers@google.com>
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it would be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write the Free Software Foundation,
+# Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#-----------------------------------------------------------------------
+
+seq=`basename $0`
+seqres=$RESULT_DIR/$seq
+echo "QA output created by $seq"
+
+here=`pwd`
+tmp=/tmp/$$
+status=1	# failure is the default!
+trap "_cleanup; exit \$status" 0 1 2 3 15
+
+_cleanup()
+{
+	cd /
+	rm -f $tmp.*
+}
+
+# get standard environment, filters and checks
+. ./common/rc
+. ./common/filter
+. ./common/verity
+
+# remove previous $seqres.full before test
+rm -f $seqres.full
+
+# real QA test starts here
+_supported_fs generic
+_supported_os Linux
+_require_scratch_verity
+
+_scratch_mkfs_verity &>> $seqres.full
+_scratch_mount
+fsv_orig_file=$SCRATCH_MNT/file
+fsv_file=$SCRATCH_MNT/file.fsv
+
+extension_test()
+{
+	local file_size=$1
+	local -n extensions=$2
+	local offsets=()
+	local lengths=()
+	local i
+	local args=()
+
+	for i in ${!extensions[@]}; do
+		local ext=${extensions[$i]}
+		local type=$(echo $ext | cut -d, -f1)
+		local offset=$(echo $ext | cut -d, -f2)
+		local length=$(echo $ext | cut -d, -f3)
+		case $type in
+		elide)
+			args+=("--elide=$offset,$length")
+			;;
+		patch)
+			head -c $length /dev/zero | tr '\0' A > $tmp.patch$i
+			args+=("--patch=$offset,$tmp.patch$i")
+			;;
+		*)
+			_fail "Unknown extension type: $type"
+		esac
+		offsets+=($offset)
+		lengths+=($length)
+	done
+
+	head -c $file_size /dev/zero > $fsv_orig_file
+	local measurement=$(_fsv_setup_file "${args[@]}" \
+			    $fsv_orig_file $fsv_file)
+	$FSVERITY_PROG enable $fsv_file
+	$FSVERITY_PROG set_measurement $fsv_file $measurement
+	cmp $fsv_orig_file $fsv_file && echo "Files matched"
+
+	echo "Modifying elided/patched region(s)..."
+	for i in ${!extensions[@]}; do
+		local offset=${offsets[$i]}
+		local length=${lengths[$i]}
+		head -c $length /dev/zero | tr '\0' 'X' \
+			| _fsv_corrupt_bytes $fsv_file $offset
+	done
+
+	echo "Comparing modified data..."
+	local cur_pos=0
+	echo -n > $tmp.data
+	for i in ${!extensions[@]}; do
+		local offset=${offsets[$i]}
+		local length=${lengths[$i]}
+		head -c $((offset - cur_pos)) /dev/zero >> $tmp.data
+		head -c $length /dev/zero | tr '\0' 'X' >> $tmp.data
+		cur_pos=$((offset + length))
+	done
+	head -c $((file_size - cur_pos)) /dev/zero >> $tmp.data
+	$FSVERITY_PROG set_measurement $fsv_file $measurement
+	cmp $tmp.data $fsv_file && echo "Files matched"
+
+	echo "Modifying unelided/unpatched regions..."
+	local cur_pos=0
+	for i in ${!extensions[@]}; do
+		local offset=${offsets[$i]}
+		local length=${lengths[$i]}
+
+		if (( cur_pos != offset )); then
+			echo -n X | _fsv_corrupt_bytes $fsv_file $((offset - 1))
+		fi
+		cur_pos=$((offset + length))
+	done
+	if (( cur_pos != file_size )); then
+		echo -n X | _fsv_corrupt_bytes $fsv_file $((cur_pos + 1))
+	fi
+
+	echo "Checking for I/O errors when reading from unelided/unpatched regions..."
+	$FSVERITY_PROG set_measurement $fsv_file $measurement
+	local cur_pos=0
+	for i in ${!extensions[@]}; do
+		local offset=${offsets[$i]}
+		local length=${lengths[$i]}
+
+		if (( cur_pos != offset )); then
+			dd if=$fsv_file bs=1 skip=$cur_pos \
+				count=$((offset - cur_pos)) status=none \
+				2>&1 >/dev/null | _filter_scratch
+		fi
+		cur_pos=$((offset + length))
+	done
+	if (( cur_pos != file_size )); then
+		dd if=$fsv_file bs=1 skip=$cur_pos \
+			count=$((file_size - cur_pos)) status=none \
+			2>&1 >/dev/null | _filter_scratch
+	fi
+}
+
+_fsv_begin_subtest "Test elision"
+EXTENSIONS=("elide,$FSV_BLOCK_SIZE,$((2 * FSV_BLOCK_SIZE))")
+extension_test $((4 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test elision (from start of file)"
+EXTENSIONS=("elide,0,$((2 * FSV_BLOCK_SIZE))")
+extension_test $((4 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test elision (to EOF)"
+EXTENSIONS=("elide,$((2 * FSV_BLOCK_SIZE)),$((2 * FSV_BLOCK_SIZE))")
+extension_test $((4 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test patch"
+EXTENSIONS=("patch,$((2 * FSV_BLOCK_SIZE - 7)),13")
+extension_test $((4 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test patch (from start of file)"
+EXTENSIONS=("patch,0,100")
+extension_test $((4 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test patch (to EOF)"
+EXTENSIONS=("patch,$((4 * FSV_BLOCK_SIZE - 100)),100")
+extension_test $((4 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test multiple elisions"
+EXTENSIONS=("elide,$((2 * FSV_BLOCK_SIZE)),$FSV_BLOCK_SIZE"
+	    "elide,$((4 * FSV_BLOCK_SIZE)),$((2 * FSV_BLOCK_SIZE))")
+extension_test $((7 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test multiple patches"
+EXTENSIONS=("patch,$((FSV_BLOCK_SIZE - 1000)),13"
+	    "patch,$((3 * FSV_BLOCK_SIZE - 20)),40")
+extension_test $((4 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+_fsv_begin_subtest "Test multiple elisions and patches"
+EXTENSIONS=("elide,$FSV_BLOCK_SIZE,$FSV_BLOCK_SIZE"
+	    "patch,$((2 * FSV_BLOCK_SIZE + 999)),33"
+	    "elide,$((4 * FSV_BLOCK_SIZE)),$((4 * FSV_BLOCK_SIZE))"
+	    "patch,$((9 * FSV_BLOCK_SIZE)),18")
+extension_test $((11 * FSV_BLOCK_SIZE)) EXTENSIONS
+
+# success, all done
+status=0
+exit
diff --git a/tests/generic/905.out b/tests/generic/905.out
new file mode 100644
index 0000000000..72bc558d7c
--- /dev/null
+++ b/tests/generic/905.out
@@ -0,0 +1,92 @@
+QA output created by 905
+
+# Test elision
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test elision (from start of file)
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test elision (to EOF)
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test patch
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test patch (from start of file)
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test patch (to EOF)
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test multiple elisions
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test multiple patches
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+
+# Test multiple elisions and patches
+Files matched
+Modifying elided/patched region(s)...
+Comparing modified data...
+Files matched
+Modifying unelided/unpatched regions...
+Checking for I/O errors when reading from unelided/unpatched regions...
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
+dd: error reading 'SCRATCH_MNT/file.fsv': Input/output error
diff --git a/tests/generic/group b/tests/generic/group
index 09f32e8f22..5269be32f0 100644
--- a/tests/generic/group
+++ b/tests/generic/group
@@ -491,3 +491,4 @@
 902 auto quick verity
 903 auto quick verity
 904 auto quick verity
+905 auto quick verity