xref: /illumos-gate/usr/src/test/zfs-tests/tests/functional/cli_root/cli_common.kshlib (revision b6805bf78d2bbbeeaea8909a05623587b42d58b3)
1#
2# CDDL HEADER START
3#
4# The contents of this file are subject to the terms of the
5# Common Development and Distribution License (the "License").
6# You may not use this file except in compliance with the License.
7#
8# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9# or http://www.opensolaris.org/os/licensing.
10# See the License for the specific language governing permissions
11# and limitations under the License.
12#
13# When distributing Covered Code, include this CDDL HEADER in each
14# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15# If applicable, add the following below this CDDL HEADER, with the
16# fields enclosed by brackets "[]" replaced with your own identifying
17# information: Portions Copyright [yyyy] [name of copyright owner]
18#
19# CDDL HEADER END
20#
21
22#
23# Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
24# Use is subject to license terms.
25#
26
27. $STF_SUITE/include/libtest.shlib
28
29#
30# Get the checksum and size of the file.
31#
32function get_cksum # <file path>
33{
34	return $($CKSUM $1 | $AWK '{print $1 $2}')
35}
36
37#
38# Compare the check sum of target files with the original file
39#
40
41function compare_cksum #<orig_data> <target_data1>...<target_datan>
42{
43	typeset orig_data=$1
44	typeset orig_sum=$(get_cksum $orig_data)
45	typeset target_sum=""
46	typeset bad_data_list=""
47	typeset -i bad_count=0
48
49	shift
50	for data in $@; do
51		if [[ ! -e $data ]]; then
52			bad_data_list="$bad_data_list $data"
53			(( bad_count +=1 ))
54			continue
55		fi
56
57		target_sum=$(get_cksum $data)
58		if [[ $target_sum != $orig_sum ]]; then
59			bad_data_list="$bad_data_list $data"
60			(( bad_count +=1 ))
61		fi
62	done
63
64	[[ $bad_data_list != "" ]] && \
65		log_fail "Data corruptions appear during send->receive." \
66			"There are total $bad_count corruptions. They are:\n"\
67			"$bad_data_list"
68}
69
70#
71# Check the received dataset exists or not
72#
73function receive_check #<dataset1>...<datasetn>
74{
75	typeset bad_rst_tgts=""
76
77	for dataset in $@; do
78		! datasetexists $dataset && \
79			bad_rst_tgts="$bad_rst_tgts $dataset"
80	done
81
82	if [[ $bad_rst_tgts != "" ]]; then
83		log_fail "Restoring fails. The specified datasets"\
84			"$bad_rst_tgts are not being received."
85	fi
86}
87