1d583b39bSJohn Wren Kennedy#!/usr/bin/ksh
2d583b39bSJohn Wren Kennedy
3d583b39bSJohn Wren Kennedy#
4d583b39bSJohn Wren Kennedy# This file and its contents are supplied under the terms of the
5d583b39bSJohn Wren Kennedy# Common Development and Distribution License ("CDDL"), version 1.0.
6d583b39bSJohn Wren Kennedy# You may only use this file in accordance with the terms of version
7d583b39bSJohn Wren Kennedy# 1.0 of the CDDL.
8d583b39bSJohn Wren Kennedy#
9d583b39bSJohn Wren Kennedy# A full copy of the text of the CDDL should have accompanied this
10d583b39bSJohn Wren Kennedy# source.  A copy of the CDDL is also available via the Internet at
11d583b39bSJohn Wren Kennedy# http://www.illumos.org/license/CDDL.
12d583b39bSJohn Wren Kennedy#
13d583b39bSJohn Wren Kennedy
14d583b39bSJohn Wren Kennedy#
1555a13001SPavel Zakharov# Copyright (c) 2012, 2016 by Delphix. All rights reserved.
16cf988facSDan McDonald# Copyright 2014, OmniTI Computer Consulting, Inc. All rights reserved.
1766597161SYuri Pankov# Copyright 2021 Tintri by DDN, Inc. All rights reserved.
1878801af7SAndy Fiddaman# Copyright 2020 OmniOS Community Edition (OmniOSce) Association.
19d583b39bSJohn Wren Kennedy#
20d583b39bSJohn Wren Kennedy
211d32ba66SJohn Wren Kennedyexport PATH="/usr/bin"
22447b1e1fSJohn Wren Kennedyexport NOINUSE_CHECK=1
23d583b39bSJohn Wren Kennedyexport STF_SUITE="/opt/zfs-tests"
24b8052df9SRyan Moellerexport COMMON="$STF_SUITE/runfiles/common.run"
25d583b39bSJohn Wren Kennedyexport STF_TOOLS="/opt/test-runner/stf"
261d32ba66SJohn Wren Kennedyexport PATHDIR=""
27d583b39bSJohn Wren Kennedyrunner="/opt/test-runner/bin/run"
28d583b39bSJohn Wren Kennedyauto_detect=false
29d583b39bSJohn Wren Kennedy
3055a13001SPavel Zakharovif [[ -z "$TESTFAIL_CALLBACKS" ]] ; then
31ad42deeaSJohn Wren Kennedy	export TESTFAIL_CALLBACKS="$STF_SUITE/callbacks/zfs_dbgmsg"
3255a13001SPavel Zakharovfi
3355a13001SPavel Zakharov
34d583b39bSJohn Wren Kennedyfunction fail
35d583b39bSJohn Wren Kennedy{
36d583b39bSJohn Wren Kennedy	echo $1
37d583b39bSJohn Wren Kennedy	exit ${2:-1}
38d583b39bSJohn Wren Kennedy}
39d583b39bSJohn Wren Kennedy
40d583b39bSJohn Wren Kennedyfunction find_disks
41d583b39bSJohn Wren Kennedy{
421d32ba66SJohn Wren Kennedy	typeset all_disks=$(echo '' | sudo -k format | awk \
43d583b39bSJohn Wren Kennedy	    '/c[0-9]/ {print $2}')
441d32ba66SJohn Wren Kennedy	typeset used_disks=$(zpool status | awk \
4566597161SYuri Pankov	    '/c[0-9]+(t[0-9a-fA-F]+)?d[0-9]+/ {print $1}' | sed -E \
46ce326879SAttila Fülöp	    's/(s|p)[0-9]+//g')
47d583b39bSJohn Wren Kennedy
48d583b39bSJohn Wren Kennedy	typeset disk used avail_disks
49d583b39bSJohn Wren Kennedy	for disk in $all_disks; do
50d583b39bSJohn Wren Kennedy		for used in $used_disks; do
51d583b39bSJohn Wren Kennedy			[[ "$disk" = "$used" ]] && continue 2
52d583b39bSJohn Wren Kennedy		done
53d583b39bSJohn Wren Kennedy		[[ -n $avail_disks ]] && avail_disks="$avail_disks $disk"
54d583b39bSJohn Wren Kennedy		[[ -z $avail_disks ]] && avail_disks="$disk"
55d583b39bSJohn Wren Kennedy	done
56d583b39bSJohn Wren Kennedy
57d583b39bSJohn Wren Kennedy	echo $avail_disks
58d583b39bSJohn Wren Kennedy}
59d583b39bSJohn Wren Kennedy
60d583b39bSJohn Wren Kennedyfunction find_rpool
61d583b39bSJohn Wren Kennedy{
621d32ba66SJohn Wren Kennedy	typeset ds=$(mount | awk '/^\/ / {print $3}')
63d583b39bSJohn Wren Kennedy	echo ${ds%%/*}
64d583b39bSJohn Wren Kennedy}
65d583b39bSJohn Wren Kennedy
66d583b39bSJohn Wren Kennedyfunction find_runfile
67d583b39bSJohn Wren Kennedy{
68d583b39bSJohn Wren Kennedy	typeset distro=
69d583b39bSJohn Wren Kennedy	if [[ -d /opt/delphix && -h /etc/delphix/version ]]; then
70d583b39bSJohn Wren Kennedy		distro=delphix
71d583b39bSJohn Wren Kennedy	elif [[ 0 -ne $(grep -c OpenIndiana /etc/release 2>/dev/null) ]]; then
72d583b39bSJohn Wren Kennedy		distro=openindiana
73cf988facSDan McDonald	elif [[ 0 -ne $(grep -c OmniOS /etc/release 2>/dev/null) ]]; then
74cf988facSDan McDonald		distro=omnios
75d583b39bSJohn Wren Kennedy	fi
76d583b39bSJohn Wren Kennedy
77b8052df9SRyan Moeller	[[ -n $distro ]] && echo $COMMON,$STF_SUITE/runfiles/$distro.run
78d583b39bSJohn Wren Kennedy}
79d583b39bSJohn Wren Kennedy
80d583b39bSJohn Wren Kennedyfunction verify_id
81d583b39bSJohn Wren Kennedy{
82d583b39bSJohn Wren Kennedy	[[ $(id -u) = "0" ]] && fail "This script must not be run as root."
83d583b39bSJohn Wren Kennedy
84412db4e9SJohn Wren Kennedy	sudo -k -n id >/dev/null 2>&1
85d583b39bSJohn Wren Kennedy	[[ $? -eq 0 ]] || fail "User must be able to sudo without a password."
86d583b39bSJohn Wren Kennedy}
87d583b39bSJohn Wren Kennedy
88d583b39bSJohn Wren Kennedyfunction verify_disks
89d583b39bSJohn Wren Kennedy{
90d583b39bSJohn Wren Kennedy	typeset disk
910dbcca9bSGordon Ross	typeset path
92d583b39bSJohn Wren Kennedy	for disk in $DISKS; do
930dbcca9bSGordon Ross		case $disk in
940dbcca9bSGordon Ross		/*) path=$disk;;
950dbcca9bSGordon Ross		*) path=/dev/rdsk/${disk}s0
960dbcca9bSGordon Ross		esac
970dbcca9bSGordon Ross		sudo -k prtvtoc $path >/dev/null 2>&1
98d583b39bSJohn Wren Kennedy		[[ $? -eq 0 ]] || return 1
99d583b39bSJohn Wren Kennedy	done
100d583b39bSJohn Wren Kennedy	return 0
101d583b39bSJohn Wren Kennedy}
102d583b39bSJohn Wren Kennedy
1031d32ba66SJohn Wren Kennedyfunction create_links
1041d32ba66SJohn Wren Kennedy{
1051d32ba66SJohn Wren Kennedy	typeset dir=$1
1061d32ba66SJohn Wren Kennedy	typeset file_list=$2
1071d32ba66SJohn Wren Kennedy
1081d32ba66SJohn Wren Kennedy	[[ -n $PATHDIR ]] || fail "PATHDIR wasn't correctly set"
1091d32ba66SJohn Wren Kennedy
1101d32ba66SJohn Wren Kennedy	for i in $file_list; do
1111d32ba66SJohn Wren Kennedy		[[ ! -e $PATHDIR/$i ]] || fail "$i already exists"
1121d32ba66SJohn Wren Kennedy		ln -s $dir/$i $PATHDIR/$i || fail "Couldn't link $i"
1131d32ba66SJohn Wren Kennedy	done
1141d32ba66SJohn Wren Kennedy
1151d32ba66SJohn Wren Kennedy}
1161d32ba66SJohn Wren Kennedy
1171d32ba66SJohn Wren Kennedyfunction constrain_path
1181d32ba66SJohn Wren Kennedy{
1191d32ba66SJohn Wren Kennedy	. $STF_SUITE/include/commands.cfg
1201d32ba66SJohn Wren Kennedy
1211d32ba66SJohn Wren Kennedy	PATHDIR=$(/usr/bin/mktemp -d /var/tmp/constrained_path.XXXX)
1221d32ba66SJohn Wren Kennedy	chmod 755 $PATHDIR || fail "Couldn't chmod $PATHDIR"
1231d32ba66SJohn Wren Kennedy
1241d32ba66SJohn Wren Kennedy	create_links "/usr/bin" "$USR_BIN_FILES"
1251d32ba66SJohn Wren Kennedy	create_links "/usr/sbin" "$USR_SBIN_FILES"
1261d32ba66SJohn Wren Kennedy	create_links "/sbin" "$SBIN_FILES"
1271d32ba66SJohn Wren Kennedy	create_links "/opt/zfs-tests/bin" "$ZFSTEST_FILES"
1281d32ba66SJohn Wren Kennedy
1291d32ba66SJohn Wren Kennedy	# Special case links
1301d32ba66SJohn Wren Kennedy	ln -s /usr/gnu/bin/dd $PATHDIR/gnu_dd
1311d32ba66SJohn Wren Kennedy}
1321d32ba66SJohn Wren Kennedy
1331d32ba66SJohn Wren Kennedyconstrain_path
1341d32ba66SJohn Wren Kennedyexport PATH=$PATHDIR
135d583b39bSJohn Wren Kennedy
1361d32ba66SJohn Wren Kennedyverify_id
137b8052df9SRyan Moellerwhile getopts ac:l:qT: c; do
138d583b39bSJohn Wren Kennedy	case $c in
139d583b39bSJohn Wren Kennedy	'a')
140d583b39bSJohn Wren Kennedy		auto_detect=true
141d583b39bSJohn Wren Kennedy		;;
142d583b39bSJohn Wren Kennedy	'c')
143*8b426108SToomas Soome		runfile=$OPTARG
144*8b426108SToomas Soome		[[ -f $runfile ]] || fail "Cannot read file: $runfile"
145*8b426108SToomas Soome		if [[ -z $runfiles ]]; then
146*8b426108SToomas Soome			runfiles=$runfile
147*8b426108SToomas Soome		else
148*8b426108SToomas Soome			runfiles+=",$runfile"
149*8b426108SToomas Soome		fi
150d583b39bSJohn Wren Kennedy		;;
15178801af7SAndy Fiddaman	'l')
15278801af7SAndy Fiddaman		logfile=$OPTARG
15378801af7SAndy Fiddaman		[[ -f $logfile ]] || fail "Cannot read file: $logfile"
15478801af7SAndy Fiddaman		xargs+=" -l $logfile"
15578801af7SAndy Fiddaman		;;
156d583b39bSJohn Wren Kennedy	'q')
15778801af7SAndy Fiddaman		xargs+=" -q"
158d583b39bSJohn Wren Kennedy		;;
159b8052df9SRyan Moeller	'T')
160b8052df9SRyan Moeller		xargs+=" -T $OPTARG"
161b8052df9SRyan Moeller		;;
162d583b39bSJohn Wren Kennedy	esac
163d583b39bSJohn Wren Kennedydone
164d583b39bSJohn Wren Kennedyshift $((OPTIND - 1))
165d583b39bSJohn Wren Kennedy
166d583b39bSJohn Wren Kennedy# If the user specified -a, then use free disks, otherwise use those in $DISKS.
167d583b39bSJohn Wren Kennedyif $auto_detect; then
168d583b39bSJohn Wren Kennedy	export DISKS=$(find_disks)
169d583b39bSJohn Wren Kennedyelif [[ -z $DISKS ]]; then
170d583b39bSJohn Wren Kennedy	fail "\$DISKS not set in env, and -a not specified."
171d583b39bSJohn Wren Kennedyelse
172d583b39bSJohn Wren Kennedy	verify_disks || fail "Couldn't verify all the disks in \$DISKS"
173d583b39bSJohn Wren Kennedyfi
174d583b39bSJohn Wren Kennedy
1758808ac5dSYuri Pankov# Add the root pool to $KEEP according to its contents.
1768808ac5dSYuri Pankov# It's ok to list it twice.
177d583b39bSJohn Wren Kennedyif [[ -z $KEEP ]]; then
1788808ac5dSYuri Pankov	KEEP="$(find_rpool)"
179d583b39bSJohn Wren Kennedyelse
1808808ac5dSYuri Pankov	KEEP+=" $(find_rpool)"
181d583b39bSJohn Wren Kennedyfi
182d583b39bSJohn Wren Kennedy
1838808ac5dSYuri Pankovexport __ZFS_POOL_EXCLUDE="$KEEP"
1848808ac5dSYuri Pankovexport KEEP="^$(echo $KEEP | sed 's/ /$|^/g')\$"
1858808ac5dSYuri Pankov
186b8052df9SRyan Moeller[[ -z $runfiles ]] && runfiles=$(find_runfile)
187b8052df9SRyan Moeller[[ -z $runfiles ]] && fail "Couldn't determine distro"
188d583b39bSJohn Wren Kennedy
189d583b39bSJohn Wren Kennedy. $STF_SUITE/include/default.cfg
190d583b39bSJohn Wren Kennedy
191d583b39bSJohn Wren Kennedynum_disks=$(echo $DISKS | awk '{print NF}')
192d583b39bSJohn Wren Kennedy[[ $num_disks -lt 3 ]] && fail "Not enough disks to run ZFS Test Suite"
193d583b39bSJohn Wren Kennedy
194412db4e9SJohn Wren Kennedy# Ensure user has only basic privileges.
195b8052df9SRyan Moellerppriv -s EIP=basic -e $runner -c $runfiles $xargs
1961d32ba66SJohn Wren Kennedyret=$?
1971d32ba66SJohn Wren Kennedy
1981d32ba66SJohn Wren Kennedyrm -rf $PATHDIR || fail "Couldn't remove $PATHDIR"
199d583b39bSJohn Wren Kennedy
2001d32ba66SJohn Wren Kennedyexit $ret
201