#! /bin/bash
# SPDX-License-Identifier: GPL-2.0
# Copyright (C) 2018 SUSE Linux Products GmbH. All Rights Reserved.
#
# FS QA Test No. 169
#
# Test that an incremental send operation produces correct results if a file
# that has a prealloc (unwritten) extent beyond its EOF gets a hole punched
# in a section of that prealloc extent.
#
seq=`basename $0`
seqres=$RESULT_DIR/$seq
echo "QA output created by $seq"
tmp=/tmp/$$
status=1	# failure is the default!
trap "_cleanup; exit \$status" 0 1 2 3 15

_cleanup()
{
	cd /
	rm -f $tmp.*
	rm -fr $send_files_dir
}

# get standard environment, filters and checks
. ./common/rc
. ./common/filter

# real QA test starts here
_supported_fs btrfs
_require_test
_require_scratch
_require_xfs_io_command "fpunch"
_require_xfs_io_command "falloc" "-k"

send_files_dir=$TEST_DIR/btrfs-test-$seq

rm -f $seqres.full
rm -fr $send_files_dir
mkdir $send_files_dir

_scratch_mkfs >>$seqres.full 2>&1
_scratch_mount

# Create our test file with a prealloc extent of 4Mb starting at offset 0,
# then write 1Mb of data into offset 0.
$XFS_IO_PROG -f -c "falloc -k 0 4M" \
	     -c "pwrite -S 0xea 0 1M" \
	     $SCRATCH_MNT/foobar | _filter_xfs_io

$BTRFS_UTIL_PROG subvolume snapshot -r $SCRATCH_MNT $SCRATCH_MNT/snap1 2>&1 \
	| _filter_scratch
$BTRFS_UTIL_PROG send -f $send_files_dir/1.snap $SCRATCH_MNT/snap1 2>&1 \
    | _filter_scratch

# Now punch a hole starting at an offset that corresponds to the file's current
# size (1Mb) and ends at an offset smaller then the end offset of the prealloc
# extent we allocated earlier (3Mb < 4Mb).
$XFS_IO_PROG -c "fpunch 1M 2M" $SCRATCH_MNT/foobar

$BTRFS_UTIL_PROG subvolume snapshot -r $SCRATCH_MNT $SCRATCH_MNT/snap2 2>&1 \
	| _filter_scratch
$BTRFS_UTIL_PROG send -p $SCRATCH_MNT/snap1 -f $send_files_dir/2.snap \
		 $SCRATCH_MNT/snap2 2>&1 | _filter_scratch

echo "File digest in the original filesystem:"
md5sum $SCRATCH_MNT/snap2/foobar | _filter_scratch

# Now recreate the filesystem by receiving both send streams and verify we get
# the same file content that the original filesystem had.
_scratch_unmount
_scratch_mkfs >>$seqres.full 2>&1
_scratch_mount

$BTRFS_UTIL_PROG receive -f $send_files_dir/1.snap $SCRATCH_MNT
$BTRFS_UTIL_PROG receive -f $send_files_dir/2.snap $SCRATCH_MNT

echo "File digest in the new filesystem:"
md5sum $SCRATCH_MNT/snap2/foobar | _filter_scratch

status=0
exit