#
# 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, 2015 by Delphix. All rights reserved.
#

. $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.

#
# Synchronize all the data in pool
#
# $1 pool name
#
function sync_pool #pool
{
	typeset pool=$1

	log_must $SYNC
	log_must $SLEEP 2
	# Flush all the pool data.
	typeset -i ret
	$ZPOOL scrub $pool >/dev/null 2>&1
	ret=$?
	(( $ret != 0 )) && \
		log_fail "$ZPOOL scrub $pool failed."

	while ! is_pool_scrubbed $pool; do
		if is_pool_resilvered $pool ; then
			log_fail "$pool should not be resilver completed."
		fi
		log_must $SLEEP 2
	done
}

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

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

	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=$AFFECTED_DEVICE \
		seek=8 bs=$DD_BLOCK count=$(( DD_COUNT - 128 )) 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
}
