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