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