1# 2# This file and its contents are supplied under the terms of the 3# Common Development and Distribution License ("CDDL"), version 1.0. 4# You may only use this file in accordance with the terms of version 5# 1.0 of the CDDL. 6# 7# A full copy of the text of the CDDL should have accompanied this 8# source. A copy of the CDDL is also available via the Internet at 9# http://www.illumos.org/license/CDDL. 10# 11 12# 13# Copyright (c) 2012, 2016 by Delphix. All rights reserved. 14# 15 16# 17# Return 0 if the percentage difference between $a and $b is $percent or 18# greater. Return 1 if the percentage is lower or if we would divide by 19# zero. For use like this: 20# 21# Do $action if the calculated percentage is greater or equal to that passed in: 22# within_percent A B P && $action 23# Do $action if the calculated percentage is less than that passed in: 24# within_percent A B P || $action 25# 26function within_percent 27{ 28 typeset a=$1 29 typeset b=$1 30 typeset percent=$3 31 32 # Set $a or $b to $2 such that a >= b 33 [ 1 -eq $(echo "$2 > $a" | bc) ] && a=$2 || b=$2 34 35 # Prevent division by 0 36 [[ $a =~ [1-9] ]] || return 1 37 38 typeset p=$(echo "scale=2; $b * 100 / $a" | bc) 39 log_note "Comparing $a and $b given $percent% (calculated: $p%)" 40 [ 1 -eq $(echo "scale=2; $p >= $percent" | bc) ] 41} 42 43# 44# Return 0 if value is within +/-tolerance of target. 45# Return 1 if value exceeds our tolerance. 46# For use like this: 47# 48# Do $action if value is within the tolerance from target passed in: 49# within_tolerance VAL TAR TOL && $action 50# Do $action if value surpasses the tolerance from target passed in: 51# within_tolerance VAL TAR TOL || $action 52# 53function within_tolerance #value #target #tolerance 54{ 55 typeset val=$1 56 typeset target=$2 57 typeset tol=$3 58 59 typeset diff=$((abs(val - target))) 60 log_note "Checking if $val is within +/-$tol of $target (diff: $diff)" 61 ((diff <= tol)) 62} 63 64# 65# Return 0 if the human readable string of the form <value>[suffix] can 66# be converted to bytes. Allow suffixes are shown in the table below. 67# 68function to_bytes 69{ 70 typeset size=$1 71 typeset value=$(echo "$size" | grep -o '[0-9]\+') 72 73 case $size in 74 *PB|*pb|*P|*p) factor='1024^5' ;; 75 *TB|*tb|*T|*t) factor='1024^4' ;; 76 *GB|*gb|*G|*g) factor='1024^3' ;; 77 *MB|*mb|*M|*m) factor='1024^2' ;; 78 *KB|*kb|*K|*k) factor='1024^1' ;; 79 *B|*b) factor='1024^0' ;; 80 *[!0-9.]*) return 1 ;; 81 *) factor='1024^0' ;; 82 esac 83 84 echo "$value * ($factor)" | bc 85 86 return 0 87} 88 89# 90# Verify $a is equal to $b, otherwise raise an error specifying 91# the $type of values being compared 92# 93function verify_eq # <a> <b> <type> 94{ 95 typeset a=$1 96 typeset b=$2 97 typeset type=$3 98 99 if [[ $a -ne $b ]]; then 100 log_fail "Compared $type should be equal: $a != $b" 101 fi 102} 103 104# 105# Verify $a is not equal to $b, otherwise raise an error specifying 106# the $type of values being compared 107# 108function verify_ne # <a> <b> <type> 109{ 110 typeset a=$1 111 typeset b=$2 112 typeset type=$3 113 114 if [[ $a -eq $b ]]; then 115 log_fail "Compared $type should be not equal: $a == $b" 116 fi 117} 118 119# A simple function to get a random number between two bounds (inclusive) 120# 121# Note since we're using $RANDOM, $min+32767 is the largest number we 122# can accept as the upper bound. 123# 124# $1 lower bound 125# $2 upper bound 126# [$3 how many] 127function random_int_between 128{ 129 typeset -i min=$1 130 typeset -i max=$2 131 typeset -i count 132 typeset -i i 133 134 if [[ -z "$3" ]]; then 135 count=1 136 else 137 count=$3 138 fi 139 140 for (( i = 0; i < $count; i++ )); do 141 echo $(( (RANDOM % (max - min + 1)) + min )) 142 done 143} 144