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 by Delphix. All rights reserved.
  16 # Copyright 2014, OmniTI Computer Consulting, Inc. All rights reserved.
  17 #
  18 
  19 export STF_SUITE="/opt/zfs-tests"
  20 export STF_TOOLS="/opt/test-runner/stf"
  21 runner="/opt/test-runner/bin/run"
  22 auto_detect=false
  23 
  24 function fail
  25 {
  26         echo $1
  27         exit ${2:-1}
  28 }
  29 
  30 function find_disks
  31 {
  32         typeset all_disks=$(echo '' | sudo /usr/sbin/format | awk \
  33             '/c[0-9]/ {print $2}')
  34         typeset used_disks=$(/sbin/zpool status | awk \
  35             '/c[0-9]*t[0-9a-f]*d[0-9]/ {print $1}' | sed 's/s[0-9]//g')
  36 
  37         typeset disk used avail_disks
  38         for disk in $all_disks; do
  39                 for used in $used_disks; do
  40                         [[ "$disk" = "$used" ]] && continue 2
  41                 done
  42                 [[ -n $avail_disks ]] && avail_disks="$avail_disks $disk"
  43                 [[ -z $avail_disks ]] && avail_disks="$disk"
  44         done
  45 
  46         echo $avail_disks
  47 }
  48 
  49 function find_rpool
  50 {
  51         typeset ds=$(/usr/sbin/mount | awk '/^\/ / {print $3}')
  52         echo ${ds%%/*}
  53 }
  54 
  55 function find_runfile
  56 {
  57         typeset distro=
  58         if [[ -d /opt/delphix && -h /etc/delphix/version ]]; then
  59                 distro=delphix
  60         elif [[ 0 -ne $(grep -c OpenIndiana /etc/release 2>/dev/null) ]]; then
  61                 distro=openindiana
  62         elif [[ 0 -ne $(grep -c OmniOS /etc/release 2>/dev/null) ]]; then
  63                 distro=omnios
  64         fi
  65 
  66         [[ -n $distro ]] && echo $STF_SUITE/runfiles/$distro.run
  67 }
  68 
  69 function verify_id
  70 {
  71         [[ $(id -u) = "0" ]] && fail "This script must not be run as root."
  72 
  73         sudo -n id >/dev/null 2>&1
  74         [[ $? -eq 0 ]] || fail "User must be able to sudo without a password."
  75 
  76         typeset -i priv_cnt=$(ppriv $$ | egrep -v \
  77             ": basic$|  L:| <none>|$$:" | wc -l)
  78         [[ $priv_cnt -ne 0 ]] && fail "User must only have basic privileges."
  79 }
  80 
  81 function verify_disks
  82 {
  83         typeset disk
  84         for disk in $DISKS; do
  85                 sudo /usr/sbin/prtvtoc /dev/rdsk/${disk}s0 >/dev/null 2>&1
  86                 [[ $? -eq 0 ]] || return 1
  87         done
  88         return 0
  89 }
  90 
  91 verify_id
  92 
  93 while getopts ac:q c; do
  94         case $c in
  95         'a')
  96                 auto_detect=true
  97                 ;;
  98         'c')
  99                 runfile=$OPTARG
 100                 [[ -f $runfile ]] || fail "Cannot read file: $runfile"
 101                 ;;
 102         'q')
 103                 quiet='-q'
 104                 ;;
 105         esac
 106 done
 107 shift $((OPTIND - 1))
 108 
 109 # If the user specified -a, then use free disks, otherwise use those in $DISKS.
 110 if $auto_detect; then
 111         export DISKS=$(find_disks)
 112 elif [[ -z $DISKS ]]; then
 113         fail "\$DISKS not set in env, and -a not specified."
 114 else
 115         verify_disks || fail "Couldn't verify all the disks in \$DISKS"
 116 fi
 117 
 118 # Add the rpool to $KEEP according to its contents. It's ok to list it twice.
 119 if [[ -z $KEEP ]]; then
 120         export KEEP="^$(find_rpool)\$"
 121 else
 122         export KEEP="^$(echo $KEEP | sed 's/ /|$/')\$"
 123         KEEP+="|^$(find_rpool)\$"
 124 fi
 125 
 126 [[ -z $runfile ]] && runfile=$(find_runfile)
 127 [[ -z $runfile ]] && fail "Couldn't determine distro"
 128 
 129 . $STF_SUITE/include/default.cfg
 130 
 131 num_disks=$(echo $DISKS | awk '{print NF}')
 132 [[ $num_disks -lt 3 ]] && fail "Not enough disks to run ZFS Test Suite"
 133 
 134 $runner $quiet -c $runfile
 135 
 136 exit $?