1 #!/usr/bin/ksh
   2 
   3 #
   4 # This file and its contents are supplied under the terms of the
   5 # Common Development and Distribution License ("CDDL"), version 1.0.
   6 # You may only use this file in accordance with the terms of version
   7 # 1.0 of the CDDL.
   8 #
   9 # A full copy of the text of the CDDL should have accompanied this
  10 # source.  A copy of the CDDL is also available via the Internet at
  11 # http://www.illumos.org/license/CDDL.
  12 #
  13 
  14 #
  15 # Copyright (c) 2012, 2016 by Delphix. All rights reserved.
  16 # Copyright 2014, OmniTI Computer Consulting, Inc. All rights reserved.
  17 # Copyright 2016 Nexenta Systems, Inc.
  18 #
  19 
  20 export PATH="/usr/bin"
  21 export NOINUSE_CHECK=1
  22 export STF_SUITE="/opt/zfs-tests"
  23 export STF_TOOLS="/opt/test-runner/stf"
  24 export PATHDIR=""
  25 runner="/opt/test-runner/bin/run"
  26 auto_detect=false
  27 
  28 if [[ -z "$TESTFAIL_CALLBACKS" ]] ; then
  29         export TESTFAIL_CALLBACKS="$STF_SUITE/callbacks/zfs_dbgmsg"
  30 fi
  31 
  32 function fail
  33 {
  34         echo $1
  35         exit ${2:-1}
  36 }
  37 
  38 function find_disks
  39 {
  40         typeset all_disks=$(echo '' | sudo -k format | awk \
  41             '/c[0-9]/ {print $2}')
  42         typeset used_disks=$(zpool status | awk \
  43             '/c[0-9]+(t[0-9a-f]+)?d[0-9]+/ {print $1}' | sed -E \
  44             's/(s|p)[0-9]+//g')
  45 
  46         typeset disk used avail_disks
  47         for disk in $all_disks; do
  48                 for used in $used_disks; do
  49                         [[ "$disk" = "$used" ]] && continue 2
  50                 done
  51                 [[ -n $avail_disks ]] && avail_disks="$avail_disks $disk"
  52                 [[ -z $avail_disks ]] && avail_disks="$disk"
  53         done
  54 
  55         echo $avail_disks
  56 }
  57 
  58 function find_rpool
  59 {
  60         typeset ds=$(mount | awk '/^\/ / {print $3}')
  61         echo ${ds%%/*}
  62 }
  63 
  64 function find_runfile
  65 {
  66         typeset distro=
  67         if [[ -d /opt/delphix && -h /etc/delphix/version ]]; then
  68                 distro=delphix
  69         elif [[ 0 -ne $(grep -c OpenIndiana /etc/release 2>/dev/null) ]]; then
  70                 distro=openindiana
  71         elif [[ 0 -ne $(grep -c OmniOS /etc/release 2>/dev/null) ]]; then
  72                 distro=omnios
  73         fi
  74 
  75         [[ -n $distro ]] && echo $STF_SUITE/runfiles/$distro.run
  76 }
  77 
  78 function verify_id
  79 {
  80         [[ $(id -u) = "0" ]] && fail "This script must not be run as root."
  81 
  82         sudo -k -n id >/dev/null 2>&1
  83         [[ $? -eq 0 ]] || fail "User must be able to sudo without a password."
  84 }
  85 
  86 function verify_disks
  87 {
  88         typeset disk
  89         typeset path
  90         for disk in $DISKS; do
  91                 case $disk in
  92                 /*) path=$disk;;
  93                 *) path=/dev/rdsk/${disk}s0
  94                 esac
  95                 sudo -k prtvtoc $path >/dev/null 2>&1
  96                 [[ $? -eq 0 ]] || return 1
  97         done
  98         return 0
  99 }
 100 
 101 function create_links
 102 {
 103         typeset dir=$1
 104         typeset file_list=$2
 105 
 106         [[ -n $PATHDIR ]] || fail "PATHDIR wasn't correctly set"
 107 
 108         for i in $file_list; do
 109                 [[ ! -e $PATHDIR/$i ]] || fail "$i already exists"
 110                 ln -s $dir/$i $PATHDIR/$i || fail "Couldn't link $i"
 111         done
 112 
 113 }
 114 
 115 function constrain_path
 116 {
 117         . $STF_SUITE/include/commands.cfg
 118 
 119         PATHDIR=$(/usr/bin/mktemp -d /var/tmp/constrained_path.XXXX)
 120         chmod 755 $PATHDIR || fail "Couldn't chmod $PATHDIR"
 121 
 122         create_links "/usr/bin" "$USR_BIN_FILES"
 123         create_links "/usr/sbin" "$USR_SBIN_FILES"
 124         create_links "/sbin" "$SBIN_FILES"
 125         create_links "/opt/zfs-tests/bin" "$ZFSTEST_FILES"
 126 
 127         # Special case links
 128         ln -s /usr/gnu/bin/dd $PATHDIR/gnu_dd
 129 }
 130 
 131 constrain_path
 132 export PATH=$PATHDIR
 133 
 134 verify_id
 135 while getopts ac:q c; do
 136         case $c in
 137         'a')
 138                 auto_detect=true
 139                 ;;
 140         'c')
 141                 runfile=$OPTARG
 142                 [[ -f $runfile ]] || fail "Cannot read file: $runfile"
 143                 ;;
 144         'q')
 145                 quiet='-q'
 146                 ;;
 147         esac
 148 done
 149 shift $((OPTIND - 1))
 150 
 151 # If the user specified -a, then use free disks, otherwise use those in $DISKS.
 152 if $auto_detect; then
 153         export DISKS=$(find_disks)
 154 elif [[ -z $DISKS ]]; then
 155         fail "\$DISKS not set in env, and -a not specified."
 156 else
 157         verify_disks || fail "Couldn't verify all the disks in \$DISKS"
 158 fi
 159 
 160 # Add the root pool to $KEEP according to its contents.
 161 # It's ok to list it twice.
 162 if [[ -z $KEEP ]]; then
 163         KEEP="$(find_rpool)"
 164 else
 165         KEEP+=" $(find_rpool)"
 166 fi
 167 
 168 export __ZFS_POOL_EXCLUDE="$KEEP"
 169 export KEEP="^$(echo $KEEP | sed 's/ /$|^/g')\$"
 170 
 171 [[ -z $runfile ]] && runfile=$(find_runfile)
 172 [[ -z $runfile ]] && fail "Couldn't determine distro"
 173 
 174 . $STF_SUITE/include/default.cfg
 175 
 176 num_disks=$(echo $DISKS | awk '{print NF}')
 177 [[ $num_disks -lt 3 ]] && fail "Not enough disks to run ZFS Test Suite"
 178 
 179 # Ensure user has only basic privileges.
 180 ppriv -s EIP=basic -e $runner $quiet -c $runfile
 181 ret=$?
 182 
 183 rm -rf $PATHDIR || fail "Couldn't remove $PATHDIR"
 184 
 185 exit $ret