#
# CDDL HEADER START
#
# The contents of this file are subject to the terms of the
# Common Development and Distribution License (the "License").
# You may not use this file except in compliance with the License.
#
# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
# or http://www.opensolaris.org/os/licensing.
# See the License for the specific language governing permissions
# and limitations under the License.
#
# When distributing Covered Code, include this CDDL HEADER in each
# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
# If applicable, add the following below this CDDL HEADER, with the
# fields enclosed by brackets "[]" replaced with your own identifying
# information: Portions Copyright [yyyy] [name of copyright owner]
#
# CDDL HEADER END
#

#
# Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
# Use is subject to license terms.
#

#
# Copyright (c) 2013, 2016 by Delphix. All rights reserved.
# Copyright (c) 2017 Datto Inc.
# Copyright 2019 Joyent, Inc.
#

. $STF_SUITE/include/libtest.shlib
. $STF_SUITE/tests/functional/clean_mirror/default.cfg

# Most of the code related to the clearing of mirrors is duplicated in all
# the test cases below this directory, barring a few minor changes
# involving the device to be affected and the 'object' to use to mangle
# the contents of the mirror.
# This code is sourced into each of these test cases.

function overwrite_verify_mirror
{
	typeset AFFECTED_DEVICE=$1
	typeset OVERWRITING_DEVICE=$2

	typeset atfile=0
	set -A files
	set -A cksums
	set -A newcksums

	scrub_pool # Make sure we have a clean pool before starting the test.

	while (( atfile < FILE_COUNT )); do
		files[$atfile]=$TESTDIR/file.$atfile
		log_must file_write -o create -f $TESTDIR/file.$atfile \
			-b $FILE_SIZE -c 1
		cksums[$atfile]=$(cksum ${files[$atfile]})
		(( atfile = atfile + 1 ))
	done

	# dd the primary side of the mirror
	log_must dd if=$OVERWRITING_DEVICE of=/dev/dsk/$AFFECTED_DEVICE \
		seek=8 bs=$DD_BLOCK count=$(( DD_COUNT - 8 )) conv=notrunc

	atfile=0

	#
	# Flush out the cache so that we ensure we're reading from disk.
	#
	log_must zpool export $TESTPOOL
	log_must zpool import $TESTPOOL

	typeset -i failedcount=0
	while (( atfile < FILE_COUNT )); do
		files[$atfile]=$TESTDIR/file.$atfile
		newcksum=$(cksum ${files[$atfile]})
		if [[ $newcksum != ${cksums[$atfile]} ]]; then
			(( failedcount = failedcount + 1 ))
		fi
		rm -f ${files[$atfile]}
		(( atfile = atfile + 1 ))
	done

	if (( $failedcount > 0 )); then
		log_fail "of the $FILE_COUNT files $failedcount did not " \
		    "have the same checksum before and after."
	fi

	sync_pool $TESTPOOL
}

# Scrub the pool and wait for completion.
function scrub_pool
{
        typeset pool=$TESTPOOL
        log_note "Scrubbing $TESTPOOL"

        # Flush all the pool data.
        log_must zpool scrub $pool

        while ! is_pool_scrubbed $pool; do
                if is_pool_resilvered $pool ; then
                        log_fail "$pool should not have been resilvered"
                fi
                log_must sleep 2
        done
}