tests/zfs-tests/tests/functional/cli_root/cli_common.kshlib

   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 #
  32 function 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
  41 function 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 #
  73 function 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 }