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 elif [[ 0 -ne $(grep -c Nexenta /etc/release 2>/dev/null) ]]; then
74 distro=nexenta
75 fi
76
77 [[ -n $distro ]] && echo $STF_SUITE/runfiles/$distro.run
78 }
79
80 function verify_id
81 {
82 [[ $(id -u) = "0" ]] && fail "This script must not be run as root."
83
84 sudo -k -n id >/dev/null 2>&1
85 [[ $? -eq 0 ]] || fail "User must be able to sudo without a password."
86 }
87
88 function verify_disks
89 {
90 typeset disk
91 typeset path
92 for disk in $DISKS; do
93 case $disk in
94 /*) path=$disk;;
95 *) path=/dev/rdsk/${disk}s0
96 esac
97 sudo -k prtvtoc $path >/dev/null 2>&1
98 [[ $? -eq 0 ]] || return 1
99 done
100 return 0
101 }
102
103 function create_links
104 {
105 typeset dir=$1
106 typeset file_list=$2
107
108 [[ -n $PATHDIR ]] || fail "PATHDIR wasn't correctly set"
109
110 for i in $file_list; do
111 [[ ! -e $PATHDIR/$i ]] || fail "$i already exists"
112 ln -s $dir/$i $PATHDIR/$i || fail "Couldn't link $i"
113 done
114
115 }
116
117 function constrain_path
118 {
119 . $STF_SUITE/include/commands.cfg
120
121 PATHDIR=$(/usr/bin/mktemp -d /var/tmp/constrained_path.XXXX)
122 chmod 755 $PATHDIR || fail "Couldn't chmod $PATHDIR"
123
124 create_links "/usr/bin" "$USR_BIN_FILES"
125 create_links "/usr/sbin" "$USR_SBIN_FILES"
126 create_links "/sbin" "$SBIN_FILES"
127 create_links "/opt/zfs-tests/bin" "$ZFSTEST_FILES"
128
129 # Special case links
130 ln -s /usr/gnu/bin/dd $PATHDIR/gnu_dd
131 }
132
133 constrain_path
134 export PATH=$PATHDIR
135
136 verify_id
137 while getopts ac:q c; do
138 case $c in
139 'a')
140 auto_detect=true
141 ;;
142 'c')
143 runfile=$OPTARG
144 [[ -f $runfile ]] || fail "Cannot read file: $runfile"
145 ;;
146 'q')
147 quiet='-q'
148 ;;
149 esac
150 done
151 shift $((OPTIND - 1))
152
153 # If the user specified -a, then use free disks, otherwise use those in $DISKS.
154 if $auto_detect; then
155 export DISKS=$(find_disks)
156 elif [[ -z $DISKS ]]; then
157 fail "\$DISKS not set in env, and -a not specified."
158 else
159 verify_disks || fail "Couldn't verify all the disks in \$DISKS"
160 fi
161
162 # Add the root pool to $KEEP according to its contents.
163 # It's ok to list it twice.
164 if [[ -z $KEEP ]]; then
165 KEEP="$(find_rpool)"
166 else
167 KEEP+=" $(find_rpool)"
168 fi
169
170 export __ZFS_POOL_EXCLUDE="$KEEP"
171 export KEEP="^$(echo $KEEP | sed 's/ /$|^/g')\$"
172
173 [[ -z $runfile ]] && runfile=$(find_runfile)
174 [[ -z $runfile ]] && fail "Couldn't determine distro"
175
176 . $STF_SUITE/include/default.cfg
177
178 num_disks=$(echo $DISKS | awk '{print NF}')
179 [[ $num_disks -lt 3 ]] && fail "Not enough disks to run ZFS Test Suite"
180
181 # Ensure user has only basic privileges.
182 ppriv -s EIP=basic -e $runner $quiet -c $runfile
183 ret=$?
184
185 rm -rf $PATHDIR || fail "Couldn't remove $PATHDIR"
186
187 exit $ret