cmd/scripts/zfstest.ksh

	zfstest.ksh revision 447b1e1fca22e4de5e04623965fbb1460857930c
2N/A#!/usr/bin/ksh
2N/A
2N/A#
2N/A# This file and its contents are supplied under the terms of the
2N/A# Common Development and Distribution License ("CDDL"), version 1.0.
2N/A# You may only use this file in accordance with the terms of version
2N/A# 1.0 of the CDDL.
2N/A#
2N/A# A full copy of the text of the CDDL should have accompanied this
2N/A# source.  A copy of the CDDL is also available via the Internet at
2N/A# http://www.illumos.org/license/CDDL.
2N/A#
2N/A
2N/A#
2N/A# Copyright (c) 2012, 2015 by Delphix. All rights reserved.
2N/A# Copyright 2014, OmniTI Computer Consulting, Inc. All rights reserved.
2N/A# Copyright 2016 Nexenta Systems, Inc.
2N/A#
2N/A
2N/Aexport NOINUSE_CHECK=1
2N/Aexport STF_SUITE="/opt/zfs-tests"
2N/Aexport STF_TOOLS="/opt/test-runner/stf"
2N/Arunner="/opt/test-runner/bin/run"
2N/Aauto_detect=false
2N/A
2N/Afunction fail
2N/A{
2N/A    echo $1
2N/A    exit ${2:-1}
2N/A}
2N/A
2N/Afunction find_disks
2N/A{
2N/A    typeset all_disks=$(echo '' | sudo -k /usr/sbin/format | awk \
2N/A        '/c[0-9]/ {print $2}')
2N/A    typeset used_disks=$(/sbin/zpool status | awk \
2N/A        '/c[0-9]*t[0-9a-f]*d[0-9]/ {print $1}' | sed 's/s[0-9]//g')
2N/A
2N/A    typeset disk used avail_disks
2N/A    for disk in $all_disks; do
2N/A        for used in $used_disks; do
2N/A            [[ "$disk" = "$used" ]] && continue 2
2N/A        done
2N/A        [[ -n $avail_disks ]] && avail_disks="$avail_disks $disk"
2N/A        [[ -z $avail_disks ]] && avail_disks="$disk"
2N/A    done
2N/A
    echo $avail_disks
}

function find_rpool
{
    typeset ds=$(/usr/sbin/mount | awk '/^\/ / {print $3}')
    echo ${ds%%/*}
}

function find_runfile
{
    typeset distro=
    if [[ -d /opt/delphix && -h /etc/delphix/version ]]; then
        distro=delphix
    elif [[ 0 -ne $(grep -c OpenIndiana /etc/release 2>/dev/null) ]]; then
        distro=openindiana
    elif [[ 0 -ne $(grep -c OmniOS /etc/release 2>/dev/null) ]]; then
        distro=omnios
    fi

    [[ -n $distro ]] && echo $STF_SUITE/runfiles/$distro.run
}

function verify_id
{
    [[ $(id -u) = "0" ]] && fail "This script must not be run as root."

    sudo -k -n id >/dev/null 2>&1
    [[ $? -eq 0 ]] || fail "User must be able to sudo without a password."
}

function verify_disks
{
    typeset disk
    for disk in $DISKS; do
        sudo -k /usr/sbin/prtvtoc /dev/rdsk/${disk}s0 >/dev/null 2>&1
        [[ $? -eq 0 ]] || return 1
    done
    return 0
}

verify_id

while getopts ac:q c; do
    case $c in
    'a')
        auto_detect=true
        ;;
    'c')
        runfile=$OPTARG
        [[ -f $runfile ]] || fail "Cannot read file: $runfile"
        ;;
    'q')
        quiet='-q'
        ;;
    esac
done
shift $((OPTIND - 1))

# If the user specified -a, then use free disks, otherwise use those in $DISKS.
if $auto_detect; then
    export DISKS=$(find_disks)
elif [[ -z $DISKS ]]; then
    fail "\$DISKS not set in env, and -a not specified."
else
    verify_disks || fail "Couldn't verify all the disks in \$DISKS"
fi

# Add the root pool to $KEEP according to its contents.
# It's ok to list it twice.
if [[ -z $KEEP ]]; then
    KEEP="$(find_rpool)"
else
    KEEP+=" $(find_rpool)"
fi

export __ZFS_POOL_EXCLUDE="$KEEP"
export KEEP="^$(echo $KEEP | sed 's/ /$|^/g')\$"

[[ -z $runfile ]] && runfile=$(find_runfile)
[[ -z $runfile ]] && fail "Couldn't determine distro"

. $STF_SUITE/include/default.cfg

num_disks=$(echo $DISKS | awk '{print NF}')
[[ $num_disks -lt 3 ]] && fail "Not enough disks to run ZFS Test Suite"

# Ensure user has only basic privileges.
/usr/bin/ppriv -s EIP=basic -e $runner $quiet -c $runfile

exit $?