Whamcloud - gitweb
LU-10626 test: create custom udev rule
[fs/lustre-release.git] / lustre / tests / test-framework.sh
1 #!/bin/bash
2
3 trap 'print_summary && print_stack_trace | tee $TF_FAIL && \
4     echo "$TESTSUITE: FAIL: test-framework exiting on error"' ERR
5 set -e
6
7 export LANG=en_US
8 export REFORMAT=${REFORMAT:-""}
9 export WRITECONF=${WRITECONF:-""}
10 export VERBOSE=${VERBOSE:-false}
11 export GSS=${GSS:-false}
12 export GSS_SK=${GSS_SK:-false}
13 export GSS_KRB5=false
14 export GSS_PIPEFS=false
15 export SHARED_KEY=${SHARED_KEY:-false}
16 export SK_PATH=${SK_PATH:-/tmp/test-framework-keys}
17 export SK_OM_PATH=$SK_PATH'/tmp-request-mount'
18 export SK_MOUNTED=${SK_MOUNTED:-false}
19 export SK_FLAVOR=${SK_FLAVOR:-ski}
20 export SK_NO_KEY=${SK_NO_KEY:-true}
21 export SK_UNIQUE_NM=${SK_UNIQUE_NM:-false}
22 export SK_S2S=${SK_S2S:-false}
23 export SK_S2SNM=${SK_S2SNM:-TestFrameNM}
24 export SK_S2SNMCLI=${SK_S2SNMCLI:-TestFrameNMCli}
25 export SK_SKIPFIRST=${SK_SKIPFIRST:-true}
26 export IDENTITY_UPCALL=default
27 export QUOTA_AUTO=1
28 export FLAKEY=${FLAKEY:-true}
29 # specify environment variable containing batch job name for server statistics
30 export JOBID_VAR=${JOBID_VAR:-"procname_uid"}  # or "existing" or "disable"
31
32 #export PDSH="pdsh -S -Rssh -w"
33 export MOUNT_CMD=${MOUNT_CMD:-"mount -t lustre"}
34 export UMOUNT=${UMOUNT:-"umount -d"}
35
36 export LSNAPSHOT_CONF="/etc/ldev.conf"
37 export LSNAPSHOT_LOG="/var/log/lsnapshot.log"
38
39 # sles12 umount has a issue with -d option
40 [ -e /etc/SuSE-release ] && grep -w VERSION /etc/SuSE-release | grep -wq 12 && {
41         export UMOUNT="umount"
42 }
43
44 # function used by scripts run on remote nodes
45 LUSTRE=${LUSTRE:-$(cd $(dirname $0)/..; echo $PWD)}
46 . $LUSTRE/tests/functions.sh
47 . $LUSTRE/tests/yaml.sh
48
49 export LD_LIBRARY_PATH=${LUSTRE}/utils/.libs:${LUSTRE}/utils:${LD_LIBRARY_PATH}
50
51 LUSTRE_TESTS_CFG_DIR=${LUSTRE_TESTS_CFG_DIR:-${LUSTRE}/tests/cfg}
52
53 EXCEPT_LIST_FILE=${EXCEPT_LIST_FILE:-${LUSTRE_TESTS_CFG_DIR}/tests-to-skip.sh}
54
55 if [ -f "$EXCEPT_LIST_FILE" ]; then
56     echo "Reading test skip list from $EXCEPT_LIST_FILE"
57     cat $EXCEPT_LIST_FILE
58     . $EXCEPT_LIST_FILE
59 fi
60
61 # check config files for options in decreasing order of preference
62 [ -z "$MODPROBECONF" -a -f /etc/modprobe.d/lustre.conf ] &&
63     MODPROBECONF=/etc/modprobe.d/lustre.conf
64 [ -z "$MODPROBECONF" -a -f /etc/modprobe.d/Lustre ] &&
65     MODPROBECONF=/etc/modprobe.d/Lustre
66 [ -z "$MODPROBECONF" -a -f /etc/modprobe.conf ] &&
67     MODPROBECONF=/etc/modprobe.conf
68
69 sanitize_parameters() {
70         for i in DIR DIR1 DIR2 MOUNT MOUNT1 MOUNT2
71         do
72                 local path=${!i}
73                 if [ -d "$path" ]; then
74                         eval export $i=$(echo $path | sed -r 's/\/+$//g')
75                 fi
76         done
77 }
78 assert_DIR () {
79         local failed=""
80         [[ $DIR/ = $MOUNT/* ]] ||
81                 { failed=1 && echo "DIR=$DIR not in $MOUNT. Aborting."; }
82         [[ $DIR1/ = $MOUNT1/* ]] ||
83                 { failed=1 && echo "DIR1=$DIR1 not in $MOUNT1. Aborting."; }
84         [[ $DIR2/ = $MOUNT2/* ]] ||
85                 { failed=1 && echo "DIR2=$DIR2 not in $MOUNT2. Aborting"; }
86
87         [ -n "$failed" ] && exit 99 || true
88 }
89
90 usage() {
91     echo "usage: $0 [-r] [-f cfgfile]"
92     echo "       -r: reformat"
93
94     exit
95 }
96
97 print_summary () {
98     trap 0
99         [ -z "$DEFAULT_SUITES"] && return 0
100     [ -n "$ONLY" ] && echo "WARNING: ONLY is set to $(echo $ONLY)"
101     local details
102     local form="%-13s %-17s %-9s %s %s\n"
103     printf "$form" "status" "script" "Total(sec)" "E(xcluded) S(low)"
104     echo "------------------------------------------------------------------------------------"
105     for O in $DEFAULT_SUITES; do
106         O=$(echo $O  | tr "-" "_" | tr "[:lower:]" "[:upper:]")
107         [ "${!O}" = "no" ] && continue || true
108         local o=$(echo $O  | tr "[:upper:]_" "[:lower:]-")
109         local log=${TMP}/${o}.log
110         if is_sanity_benchmark $o; then
111             log=${TMP}/sanity-benchmark.log
112         fi
113         local slow=
114         local skipped=
115         local total=
116         local status=Unfinished
117         if [ -f $log ]; then
118                 skipped=$(grep excluded $log | awk '{ printf " %s", $3 }' |
119                         sed 's/test_//g')
120                 slow=$(egrep "^PASS|^FAIL" $log | tr -d "("| sed s/s\)$//g |
121                         sort -nr -k 3  | head -n5 |  awk '{ print $2":"$3"s" }')
122                 total=$(grep duration $log | awk '{ print $2 }')
123                 if [ "${!O}" = "done" ]; then
124                         status=Done
125                 fi
126                 if $DDETAILS; then
127                         local durations=$(egrep "^PASS|^FAIL" $log |
128                                 tr -d "("| sed s/s\)$//g |
129                                 awk '{ print $2":"$3"|" }')
130                         details=$(printf "%s\n%s %s %s\n" "$details" \
131                                 "DDETAILS" "$O" "$(echo $durations)")
132                 fi
133         fi
134         printf "$form" $status "$O" "${total}" "E=$skipped"
135         printf "$form" "-" "-" "-" "S=$(echo $slow)"
136     done
137
138     for O in $DEFAULT_SUITES; do
139         O=$(echo $O  | tr "-" "_" | tr "[:lower:]" "[:upper:]")
140         if [ "${!O}" = "no" ]; then
141             printf "$form" "Skipped" "$O" ""
142         fi
143     done
144
145     # print the detailed tests durations if DDETAILS=true
146     if $DDETAILS; then
147         echo "$details"
148     fi
149 }
150
151 init_test_env() {
152         export LUSTRE=$(absolute_path $LUSTRE)
153         export TESTSUITE=$(basename $0 .sh)
154         export TEST_FAILED=false
155         export FAIL_ON_SKIP_ENV=${FAIL_ON_SKIP_ENV:-false}
156         export RPC_MODE=${RPC_MODE:-false}
157         export DO_CLEANUP=${DO_CLEANUP:-true}
158         export KEEP_ZPOOL=${KEEP_ZPOOL:-false}
159         export CLEANUP_DM_DEV=false
160         export PAGE_SIZE=$(get_page_size client)
161
162         export MKE2FS=$MKE2FS
163         if [ -z "$MKE2FS" ]; then
164                 if which mkfs.ldiskfs >/dev/null 2>&1; then
165                         export MKE2FS=mkfs.ldiskfs
166                 else
167                         export MKE2FS=mke2fs
168                 fi
169         fi
170
171         export DEBUGFS=$DEBUGFS
172         if [ -z "$DEBUGFS" ]; then
173                 if which debugfs.ldiskfs >/dev/null 2>&1; then
174                         export DEBUGFS=debugfs.ldiskfs
175                 else
176                         export DEBUGFS=debugfs
177                 fi
178         fi
179
180         export TUNE2FS=$TUNE2FS
181         if [ -z "$TUNE2FS" ]; then
182                 if which tunefs.ldiskfs >/dev/null 2>&1; then
183                         export TUNE2FS=tunefs.ldiskfs
184                 else
185                         export TUNE2FS=tune2fs
186                 fi
187         fi
188
189         export E2LABEL=$E2LABEL
190         if [ -z "$E2LABEL" ]; then
191                 if which label.ldiskfs >/dev/null 2>&1; then
192                         export E2LABEL=label.ldiskfs
193                 else
194                         export E2LABEL=e2label
195                 fi
196         fi
197
198         export DUMPE2FS=$DUMPE2FS
199         if [ -z "$DUMPE2FS" ]; then
200                 if which dumpfs.ldiskfs >/dev/null 2>&1; then
201                         export DUMPE2FS=dumpfs.ldiskfs
202                 else
203                         export DUMPE2FS=dumpe2fs
204                 fi
205         fi
206
207         export E2FSCK=$E2FSCK
208         if [ -z "$E2FSCK" ]; then
209                 if which fsck.ldiskfs >/dev/null 2>&1; then
210                         export E2FSCK=fsck.ldiskfs
211                 else
212                          export E2FSCK=e2fsck
213                 fi
214         fi
215
216         export RESIZE2FS=$RESIZE2FS
217         if [ -z "$RESIZE2FS" ]; then
218                 if which resizefs.ldiskfs >/dev/null 2>&1; then
219                         export RESIZE2FS=resizefs.ldiskfs
220                 else
221                         export RESIZE2FS=resize2fs
222                 fi
223         fi
224
225         export LFSCK_ALWAYS=${LFSCK_ALWAYS:-"no"} # check fs after test suite
226         export FSCK_MAX_ERR=4   # File system errors left uncorrected
227
228         export ZFS=${ZFS:-zfs}
229         export ZPOOL=${ZPOOL:-zpool}
230         export ZDB=${ZDB:-zdb}
231         export PARTPROBE=${PARTPROBE:-partprobe}
232
233         #[ -d /r ] && export ROOT=${ROOT:-/r}
234         export TMP=${TMP:-$ROOT/tmp}
235         export TESTSUITELOG=${TMP}/${TESTSUITE}.log
236         export LOGDIR=${LOGDIR:-${TMP}/test_logs/$(date +%s)}
237         export TESTLOG_PREFIX=$LOGDIR/$TESTSUITE
238
239         export HOSTNAME=${HOSTNAME:-$(hostname -s)}
240         if ! echo $PATH | grep -q $LUSTRE/utils; then
241                 export PATH=$LUSTRE/utils:$PATH
242         fi
243         if ! echo $PATH | grep -q $LUSTRE/utils/gss; then
244                 export PATH=$LUSTRE/utils/gss:$PATH
245         fi
246         if ! echo $PATH | grep -q $LUSTRE/tests; then
247                 export PATH=$LUSTRE/tests:$PATH
248         fi
249         if ! echo $PATH | grep -q $LUSTRE/../lustre-iokit/sgpdd-survey; then
250                 export PATH=$LUSTRE/../lustre-iokit/sgpdd-survey:$PATH
251         fi
252         export LST=${LST:-"$LUSTRE/../lnet/utils/lst"}
253         [ ! -f "$LST" ] && export LST=$(which lst)
254         export SGPDDSURVEY=${SGPDDSURVEY:-"$LUSTRE/../lustre-iokit/sgpdd-survey/sgpdd-survey")}
255         [ ! -f "$SGPDDSURVEY" ] && export SGPDDSURVEY=$(which sgpdd-survey)
256         export MCREATE=${MCREATE:-mcreate}
257         export MULTIOP=${MULTIOP:-multiop}
258         # Ubuntu, at least, has a truncate command in /usr/bin
259         # so fully path our truncate command.
260         export TRUNCATE=${TRUNCATE:-$LUSTRE/tests/truncate}
261         export FSX=${FSX:-$LUSTRE/tests/fsx}
262         export MDSRATE=${MDSRATE:-"$LUSTRE/tests/mpi/mdsrate"}
263         [ ! -f "$MDSRATE" ] && export MDSRATE=$(which mdsrate 2> /dev/null)
264         if ! echo $PATH | grep -q $LUSTRE/tests/racer; then
265                 export PATH=$LUSTRE/tests/racer:$PATH:
266         fi
267         if ! echo $PATH | grep -q $LUSTRE/tests/mpi; then
268                 export PATH=$LUSTRE/tests/mpi:$PATH
269         fi
270         export RSYNC_RSH=${RSYNC_RSH:-rsh}
271
272         export LCTL=${LCTL:-"$LUSTRE/utils/lctl"}
273         [ ! -f "$LCTL" ] && export LCTL=$(which lctl)
274         export LFS=${LFS:-"$LUSTRE/utils/lfs"}
275         [ ! -f "$LFS" ] && export LFS=$(which lfs)
276         SETSTRIPE=${SETSTRIPE:-"$LFS setstripe"}
277         GETSTRIPE=${GETSTRIPE:-"$LFS getstripe"}
278
279         export PERM_CMD=${PERM_CMD:-"$LCTL conf_param"}
280
281         export L_GETIDENTITY=${L_GETIDENTITY:-"$LUSTRE/utils/l_getidentity"}
282         if [ ! -f "$L_GETIDENTITY" ]; then
283                 if `which l_getidentity > /dev/null 2>&1`; then
284                         export L_GETIDENTITY=$(which l_getidentity)
285                 else
286                         export L_GETIDENTITY=NONE
287                 fi
288         fi
289         export LL_DECODE_FILTER_FID=${LL_DECODE_FILTER_FID:-"$LUSTRE/utils/ll_decode_filter_fid"}
290         [ ! -f "$LL_DECODE_FILTER_FID" ] && export LL_DECODE_FILTER_FID="ll_decode_filter_fid"
291         export LL_DECODE_LINKEA=${LL_DECODE_LINKEA:-"$LUSTRE/utils/ll_decode_linkea"}
292         [ ! -f "$LL_DECODE_LINKEA" ] && export LL_DECODE_LINKEA="ll_decode_linkea"
293         export MKFS=${MKFS:-"$LUSTRE/utils/mkfs.lustre"}
294         [ ! -f "$MKFS" ] && export MKFS="mkfs.lustre"
295         export TUNEFS=${TUNEFS:-"$LUSTRE/utils/tunefs.lustre"}
296         [ ! -f "$TUNEFS" ] && export TUNEFS="tunefs.lustre"
297         export CHECKSTAT="${CHECKSTAT:-"checkstat -v"} "
298         export LUSTRE_RMMOD=${LUSTRE_RMMOD:-$LUSTRE/scripts/lustre_rmmod}
299         [ ! -f "$LUSTRE_RMMOD" ] &&
300                 export LUSTRE_RMMOD=$(which lustre_rmmod 2> /dev/null)
301         export LUSTRE_ROUTES_CONVERSION=${LUSTRE_ROUTES_CONVERSION:-$LUSTRE/scripts/lustre_routes_conversion}
302         [ ! -f "$LUSTRE_ROUTES_CONVERSION" ] &&
303                 export LUSTRE_ROUTES_CONVERSION=$(which lustre_routes_conversion 2> /dev/null)
304         export LFS_MIGRATE=${LFS_MIGRATE:-$LUSTRE/scripts/lfs_migrate}
305         [ ! -f "$LFS_MIGRATE" ] &&
306                 export LFS_MIGRATE=$(which lfs_migrate 2> /dev/null)
307         export LR_READER=${LR_READER:-"$LUSTRE/utils/lr_reader"}
308         [ ! -f "$LR_READER" ] &&
309                 export LR_READER=$(which lr_reader 2> /dev/null)
310         [ -z "$LR_READER" ] && export LR_READER="/usr/sbin/lr_reader"
311         export LSOM_SYNC=${LSOM_SYNC:-"$LUSTRE/utils/llsom_sync"}
312         [ ! -f "$LSOM_SYNC" ] &&
313                 export LSOM_SYNC=$(which llsom_sync 2> /dev/null)
314         [ -z "$LSOM_SYNC" ] && export LSOM_SYNC="/usr/sbin/llsom_sync"
315         export NAME=${NAME:-local}
316         export LGSSD=${LGSSD:-"$LUSTRE/utils/gss/lgssd"}
317         [ "$GSS_PIPEFS" = "true" ] && [ ! -f "$LGSSD" ] &&
318                 export LGSSD=$(which lgssd)
319         export LSVCGSSD=${LSVCGSSD:-"$LUSTRE/utils/gss/lsvcgssd"}
320         [ ! -f "$LSVCGSSD" ] && export LSVCGSSD=$(which lsvcgssd 2> /dev/null)
321         export KRB5DIR=${KRB5DIR:-"/usr/kerberos"}
322         export DIR2
323         export SAVE_PWD=${SAVE_PWD:-$LUSTRE/tests}
324         export AT_MAX_PATH
325         export LDEV=${LDEV:-"$LUSTRE/scripts/ldev"}
326         [ ! -f "$LDEV" ] && export LDEV=$(which ldev 2> /dev/null)
327
328         export DMSETUP=${DMSETUP:-dmsetup}
329         export DM_DEV_PATH=${DM_DEV_PATH:-/dev/mapper}
330         export LOSETUP=${LOSETUP:-losetup}
331
332         if [ "$ACCEPTOR_PORT" ]; then
333                 export PORT_OPT="--port $ACCEPTOR_PORT"
334         fi
335
336         if $SHARED_KEY; then
337                 $RPC_MODE || echo "Using GSS shared-key feature"
338                 which lgss_sk > /dev/null 2>&1 ||
339                         error_exit "built with lgss_sk disabled! SEC=$SEC"
340                 GSS=true
341                 GSS_SK=true
342                 SEC=$SK_FLAVOR
343         fi
344
345         case "x$SEC" in
346                 xkrb5*)
347                 $RPC_MODE || echo "Using GSS/krb5 ptlrpc security flavor"
348                 which lgss_keyring > /dev/null 2>&1 ||
349                         error_exit "built with gss disabled! SEC=$SEC"
350                 GSS=true
351                 GSS_KRB5=true
352                 ;;
353         esac
354
355         case "x$IDUP" in
356                 xtrue)
357                         IDENTITY_UPCALL=true
358                         ;;
359                 xfalse)
360                         IDENTITY_UPCALL=false
361                         ;;
362         esac
363
364         export LOAD_MODULES_REMOTE=${LOAD_MODULES_REMOTE:-false}
365
366         # Paths on remote nodes, if different
367         export RLUSTRE=${RLUSTRE:-$LUSTRE}
368         export RPWD=${RPWD:-$PWD}
369         export I_MOUNTED=${I_MOUNTED:-"no"}
370         export AUSTER_CLEANUP=${AUSTER_CLEANUP:-false}
371         if [ ! -f /lib/modules/$(uname -r)/kernel/fs/lustre/mdt.ko -a \
372              ! -f /lib/modules/$(uname -r)/updates/kernel/fs/lustre/mdt.ko -a \
373              ! -f /lib/modules/$(uname -r)/extra/kernel/fs/lustre/mdt.ko -a \
374              ! -f $LUSTRE/mdt/mdt.ko ]; then
375             export CLIENTMODSONLY=yes
376         fi
377
378         export SHUTDOWN_ATTEMPTS=${SHUTDOWN_ATTEMPTS:-3}
379         export OSD_TRACK_DECLARES_LBUG=${OSD_TRACK_DECLARES_LBUG:-"yes"}
380
381         # command line
382
383         while getopts "rvwf:" opt $*; do
384                 case $opt in
385                         f) CONFIG=$OPTARG;;
386                         r) REFORMAT=yes;;
387                         v) VERBOSE=true;;
388                         w) WRITECONF=writeconf;;
389                         \?) usage;;
390                 esac
391         done
392
393         shift $((OPTIND - 1))
394         ONLY=${ONLY:-$*}
395
396         # print the durations of each test if "true"
397         DDETAILS=${DDETAILS:-false}
398         [ "$TESTSUITELOG" ] && rm -f $TESTSUITELOG || true
399         if ! $RPC_MODE; then
400                 rm -f $TMP/*active
401         fi
402
403         export TF_FAIL=${TF_FAIL:-$TMP/tf.fail}
404 }
405
406 check_cpt_number() {
407         local facet=$1
408         local ncpts
409
410         ncpts=$(do_facet $facet "lctl get_param -n " \
411                 "cpu_partition_table 2>/dev/null| wc -l" || echo 1)
412
413         if [ $ncpts -eq 0 ]; then
414                 echo "1"
415         else
416                 echo $ncpts
417         fi
418 }
419
420 # Return a numeric version code based on a version string.  The version
421 # code is useful for comparison two version strings to see which is newer.
422 version_code() {
423         # split arguments like "1.8.6-wc3" into "1", "8", "6", "wc3"
424         eval set -- $(tr "[:punct:]" " " <<< $*)
425
426         echo -n "$((($1 << 16) | ($2 << 8) | $3))"
427 }
428
429 export LINUX_VERSION=$(uname -r | sed -e "s/\([0-9]*\.[0-9]*\.[0-9]*\).*/\1/")
430 export LINUX_VERSION_CODE=$(version_code ${LINUX_VERSION//\./ })
431
432 # Report the Lustre build version string (e.g. 1.8.7.3 or 2.4.1).
433 #
434 # usage: lustre_build_version
435 #
436 # All Lustre versions support "lctl get_param" to report the version of the
437 # code running in the kernel (what our tests are interested in), but it
438 # doesn't work without modules loaded.  After 2.9.53 and in upstream kernels
439 # the "version" parameter doesn't include "lustre: " at the beginning.
440 # If that fails, call "lctl lustre_build_version" which prints either (or both)
441 # the userspace and kernel build versions, but until 2.8.55 required root
442 # access to get the Lustre kernel version.  If that also fails, fall back to
443 # using "lctl --version", which is easy to parse and works without the kernel
444 # modules, but was only added in 2.6.50 and only prints the lctl tool version,
445 # not the module version, though they are usually the same.
446 #
447 # Various commands and their output format for different Lustre versions:
448 # lctl get_param version:       2.9.55
449 # lctl get_param version:       lustre: 2.8.53
450 # lctl get_param version:       lustre: 2.6.52
451 #                               kernel: patchless_client
452 #                               build: v2_6_92_0-2.6.32-431.el6_lustre.x86_64
453 # lctl lustre_build_version:    Lustre version: 2.8.53_27_gae67fc01
454 # lctl lustre_build_version:    error: lustre_build_version: Permission denied
455 #       (as non-root user)      lctl   version: v2_6_92_0-2.6.32-431.el6.x86_64
456 # lctl lustre_build_version:    Lustre version: 2.5.3-2.6.32.26-175.fc12.x86_64
457 #                               lctl   version: 2.5.3-2.6.32..26-175fc12.x86_64
458 # lctl --version:               lctl 2.6.50
459 #
460 # output: prints version string to stdout in (up to 4) dotted-decimal values
461 lustre_build_version() {
462         local facet=${1:-client}
463         local ver=$(do_facet $facet "$LCTL get_param -n version 2>/dev/null ||
464                                 $LCTL lustre_build_version 2>/dev/null ||
465                                 $LCTL --version 2>/dev/null | cut -d' ' -f2")
466         local lver=$(egrep -i "lustre: |version: " <<<"$ver" | head -n 1)
467         [ -n "$lver" ] && ver="$lver"
468
469         sed -e 's/[^:]*: //' -e 's/^v//' -e 's/[ -].*//' -e 's/_/./g' <<<$ver |
470                 cut -d. -f1-4
471 }
472
473 # Report the Lustre numeric build version code for the supplied facet.
474 lustre_version_code() {
475         version_code $(lustre_build_version $1)
476 }
477
478 module_loaded () {
479         /sbin/lsmod | grep -q "^\<$1\>"
480 }
481
482 PRLFS=false
483 lustre_insmod() {
484         local module=$1
485         shift
486         local args="$@"
487         local msg
488         local rc=0
489
490         if ! $PRLFS; then
491                 msg="$(insmod $module $args 2>&1)" && return 0 || rc=$?
492         fi
493
494         # parallels can't load modules directly from prlfs, use /tmp instead
495         if $PRLFS || [[ "$(stat -f -c%t $module)" == "7c7c6673" ]]; then
496                 local target="$(mktemp)"
497
498                 cp "$module" "$target"
499                 insmod $target $args
500                 rc=$?
501                 [[ $rc == 0 ]] && PRLFS=true
502                 rm -f $target
503         else
504                 echo "$msg"
505         fi
506         return $rc
507 }
508
509 # Load a module on the system where this is running.
510 #
511 # usage: load_module module_name [module arguments for insmod/modprobe]
512 #
513 # If module arguments are not given but MODOPTS_<MODULE> is set, then its value
514 # will be used as the arguments.  Otherwise arguments will be obtained from
515 # /etc/modprobe.conf, from /etc/modprobe.d/Lustre, or else none will be used.
516 #
517 load_module() {
518         local module=$1 # '../libcfs/libcfs/libcfs', 'obdclass/obdclass', ...
519         shift
520         local ext=".ko"
521         local base=$(basename $module $ext)
522         local path
523         local -A module_is_loaded_aa
524         local optvar
525         local mod
526
527         for mod in $(lsmod | awk '{ print $1; }'); do
528                 module_is_loaded_aa[${mod//-/_}]=true
529         done
530
531         module_is_loaded() {
532                 ${module_is_loaded_aa[${1//-/_}]:-false}
533         }
534
535         if module_is_loaded $base; then
536                 return
537         fi
538
539         if [[ -f $LUSTRE/$module$ext ]]; then
540                 path=$LUSTRE/$module$ext
541         elif [[ "$base" == lnet_selftest ]] &&
542              [[ -f $LUSTRE/../lnet/selftest/$base$ext ]]; then
543                 path=$LUSTRE/../lnet/selftest/$base$ext
544         else
545                 path=''
546         fi
547
548         if [[ -n "$path" ]]; then
549                 # Try to load any non-Lustre modules that $module depends on.
550                 for mod in $(modinfo --field=depends $path | tr ',' ' '); do
551                         if ! module_is_loaded $mod; then
552                                 modprobe $mod
553                         fi
554                 done
555         fi
556
557         # If no module arguments were passed then get them from
558         # $MODOPTS_<MODULE>, otherwise from modprobe.conf.
559         if [ $# -eq 0 ]; then
560                 # $MODOPTS_<MODULE>; we could use associative arrays, but that's
561                 # not in Bash until 4.x, so we resort to eval.
562                 optvar="MODOPTS_$(basename $module | tr a-z A-Z)"
563                 eval set -- \$$optvar
564                 if [ $# -eq 0 -a -n "$MODPROBECONF" ]; then
565                         # Nothing in $MODOPTS_<MODULE>; try modprobe.conf
566                         local opt
567                         opt=$(awk -v var="^options $base" '$0 ~ var \
568                               {gsub("'"options $base"'",""); print}' \
569                                 $MODPROBECONF)
570                         set -- $(echo -n $opt)
571
572                         # Ensure we have accept=all for lnet
573                         if [[ "$base" == lnet ]]; then
574                                 # OK, this is a bit wordy...
575                                 local arg accept_all_present=false
576
577                                 for arg in "$@"; do
578                                         [[ "$arg" == accept=all ]] &&
579                                                 accept_all_present=true
580                                 done
581
582                                 $accept_all_present || set -- "$@" accept=all
583                         fi
584
585                         export $optvar="$*"
586                 fi
587         fi
588
589         [ $# -gt 0 ] && echo "${module} options: '$*'"
590
591         # Note that insmod will ignore anything in modprobe.conf, which is why
592         # we're passing options on the command-line. If $path does not exist
593         # then we must be testing a "make install" or"rpm" installation. Also
594         # note that failing to load ptlrpc_gss is not considered fatal.
595         if [[ -n "$path" ]]; then
596                 lustre_insmod $path "$@"
597         elif [[ "$base" == ptlrpc_gss ]]; then
598                 if ! modprobe $base "$@" 2>/dev/null; then
599                         echo "gss/krb5 is not supported"
600                 fi
601         else
602                 modprobe $base "$@"
603         fi
604 }
605
606 load_modules_local() {
607         if [ -n "$MODPROBE" ]; then
608                 # use modprobe
609                 echo "Using modprobe to load modules"
610                 return 0
611         fi
612
613         # Create special udev test rules on every node
614         if [ -f $LUSTRE/lustre/conf/99-lustre.rules ]; then {
615                 sed -e 's|/usr/sbin/lctl|$LCTL|g' $LUSTRE/lustre/conf/99-lustre.rules > /etc/udev/rules.d/99-lustre-test.rules
616         } else {
617                 echo "SUBSYSTEM==\"lustre\", ACTION==\"change\", ENV{PARAM}==\"?*\", RUN+=\"$LCTL set_param '\$env{PARAM}=\$env{SETTING}'\"" > /etc/udev/rules.d/99-lustre-test.rules
618         } fi
619         udevadm control --reload-rules
620         udevadm trigger
621
622         echo Loading modules from $LUSTRE
623
624         local ncpus
625
626         if [ -f /sys/devices/system/cpu/online ]; then
627                 ncpus=$(($(cut -d "-" -f 2 /sys/devices/system/cpu/online) + 1))
628                 echo "detected $ncpus online CPUs by sysfs"
629         else
630                 ncpus=$(getconf _NPROCESSORS_CONF 2>/dev/null)
631                 local rc=$?
632                 if [ $rc -eq 0 ]; then
633                         echo "detected $ncpus online CPUs by getconf"
634                 else
635                         echo "Can't detect number of CPUs"
636                         ncpus=1
637                 fi
638         fi
639
640         # if there is only one CPU core, libcfs can only create one partition
641         # if there is more than 4 CPU cores, libcfs should create multiple CPU
642         # partitions. So we just force libcfs to create 2 partitions for
643         # system with 2 or 4 cores
644         if [ $ncpus -le 4 ] && [ $ncpus -gt 1 ]; then
645                 # force to enable multiple CPU partitions
646                 echo "Force libcfs to create 2 CPU partitions"
647                 MODOPTS_LIBCFS="cpu_npartitions=2 $MODOPTS_LIBCFS"
648         else
649                 echo "libcfs will create CPU partition based on online CPUs"
650         fi
651
652         load_module ../libcfs/libcfs/libcfs
653         # Prevent local MODOPTS_LIBCFS being passed as part of environment
654         # variable to remote nodes
655         unset MODOPTS_LIBCFS
656
657         set_default_debug
658         load_module ../lnet/lnet/lnet
659
660         LNDPATH=${LNDPATH:-"../lnet/klnds"}
661         if [ -z "$LNETLND" ]; then
662                 case $NETTYPE in
663                 o2ib*)  LNETLND="o2iblnd/ko2iblnd" ;;
664                 tcp*)   LNETLND="socklnd/ksocklnd" ;;
665                 *)      local lnd="${NETTYPE%%[0-9]}lnd"
666                         [ -f "$LNDPATH/$lnd/k$lnd.ko" ] &&
667                                 LNETLND="$lnd/k$lnd" ||
668                                 LNETLND="socklnd/ksocklnd"
669                 esac
670         fi
671         load_module ../lnet/klnds/$LNETLND
672         load_module obdclass/obdclass
673         load_module ptlrpc/ptlrpc
674         load_module ptlrpc/gss/ptlrpc_gss
675         load_module fld/fld
676         load_module fid/fid
677         load_module lmv/lmv
678         load_module osc/osc
679         load_module mdc/mdc
680         load_module lov/lov
681         load_module mgc/mgc
682         load_module obdecho/obdecho
683         if ! client_only; then
684                 load_module lfsck/lfsck
685                 [ "$LQUOTA" != "no" ] &&
686                         load_module quota/lquota $LQUOTAOPTS
687                 if [[ $(node_fstypes $HOSTNAME) == *zfs* ]]; then
688                         load_module osd-zfs/osd_zfs
689                 elif [[ $(node_fstypes $HOSTNAME) == *ldiskfs* ]]; then
690                         load_module ../ldiskfs/ldiskfs
691                         load_module osd-ldiskfs/osd_ldiskfs
692                 fi
693                 load_module mgs/mgs
694                 load_module mdd/mdd
695                 load_module mdt/mdt
696                 load_module ost/ost
697                 load_module lod/lod
698                 load_module osp/osp
699                 load_module ofd/ofd
700                 load_module osp/osp
701         fi
702
703         load_module llite/lustre
704         [ -d /r ] && OGDB=${OGDB:-"/r/tmp"}
705         OGDB=${OGDB:-$TMP}
706         rm -f $OGDB/ogdb-$HOSTNAME
707         $LCTL modules > $OGDB/ogdb-$HOSTNAME
708
709         # 'mount' doesn't look in $PATH, just sbin
710         local mount_lustre=$LUSTRE/utils/mount.lustre
711         if [ -f $mount_lustre ]; then
712                 local sbin_mount=$(readlink -f /sbin)/mount.lustre
713                 if grep -qw "$sbin_mount" /proc/mounts; then
714                         cmp -s $mount_lustre $sbin_mount || umount $sbin_mount
715                 fi
716                 if ! grep -qw "$sbin_mount" /proc/mounts; then
717                         [ ! -f "$sbin_mount" ] && touch "$sbin_mount"
718                         if [ ! -s "$sbin_mount" -a -w "$sbin_mount" ]; then
719                                 cat <<- EOF > "$sbin_mount"
720                                 #!/bin/sh
721                                 #STUB MARK
722                                 echo "This $sbin_mount just a mountpoint." 1>&2
723                                 echo "It is never supposed to be run." 1>&2
724                                 logger -p emerg -- "using stub $sbin_mount $@"
725                                 exit 1
726                                 EOF
727                                 chmod a+x $sbin_mount
728                         fi
729                         mount --bind $mount_lustre $sbin_mount ||
730                                 error "can't bind $mount_lustre to $sbin_mount"
731                 fi
732         fi
733 }
734
735 load_modules () {
736         load_modules_local
737         # bug 19124
738         # load modules on remote nodes optionally
739         # lustre-tests have to be installed on these nodes
740         if $LOAD_MODULES_REMOTE; then
741                 local list=$(comma_list $(remote_nodes_list))
742                 if [ -n "$list" ]; then
743                         echo "loading modules on: '$list'"
744                         do_rpc_nodes "$list" load_modules_local
745                 fi
746         fi
747 }
748
749 check_mem_leak () {
750     LEAK_LUSTRE=$(dmesg | tail -n 30 | grep "obd_memory.*leaked" || true)
751     LEAK_PORTALS=$(dmesg | tail -n 20 | grep "Portals memory leaked" || true)
752     if [ "$LEAK_LUSTRE" -o "$LEAK_PORTALS" ]; then
753         echo "$LEAK_LUSTRE" 1>&2
754         echo "$LEAK_PORTALS" 1>&2
755         mv $TMP/debug $TMP/debug-leak.`date +%s` || true
756         echo "Memory leaks detected"
757         [ -n "$IGNORE_LEAK" ] && { echo "ignoring leaks" && return 0; } || true
758         return 1
759     fi
760 }
761
762 unload_modules() {
763         wait_exit_ST client # bug 12845
764
765         $LUSTRE_RMMOD ldiskfs || return 2
766
767         [ -f /etc/udev/rules.d/99-lustre-test.rules ] &&
768                 rm /etc/udev/rules.d/99-lustre-test.rules
769         udevadm control --reload-rules
770         udevadm trigger
771
772         if $LOAD_MODULES_REMOTE; then
773                 local list=$(comma_list $(remote_nodes_list))
774                 if [ -n "$list" ]; then
775                         echo "unloading modules on: '$list'"
776                         do_rpc_nodes "$list" $LUSTRE_RMMOD ldiskfs
777                         do_rpc_nodes "$list" check_mem_leak
778                         do_rpc_nodes "$list" "rm /etc/udev/rules.d/99-lustre-test.rules"
779                         do_rpc_nodes "$list" "udevadm control --reload-rules"
780                         do_rpc_nodes "$list" "udevadm trigger"
781                 fi
782         fi
783
784         local sbin_mount=$(readlink -f /sbin)/mount.lustre
785         if grep -qe "$sbin_mount " /proc/mounts; then
786                 umount $sbin_mount || true
787                 [ -s $sbin_mount ] && ! grep -q "STUB MARK" $sbin_mount ||
788                         rm -f $sbin_mount
789         fi
790
791         check_mem_leak || return 254
792
793         echo "modules unloaded."
794         return 0
795 }
796
797 fs_log_size() {
798         local facet=${1:-$SINGLEMDS}
799         local size=0
800
801         case $(facet_fstype $facet) in
802                 ldiskfs) size=50;; # largest seen is 44, leave some headroom
803                 # grant_block_size is in bytes, allow at least 2x max blocksize
804                 zfs)     size=$(lctl get_param osc.$FSNAME*.import |
805                                 awk '/grant_block_size:/ {print $2/512; exit;}')
806                           ;;
807         esac
808
809         echo -n $((size * MDSCOUNT))
810 }
811
812 fs_inode_ksize() {
813         local facet=${1:-$SINGLEMDS}
814         local fstype=$(facet_fstype $facet)
815         local size=0
816         case $fstype in
817                 ldiskfs) size=4;;  # ~4KB per inode
818                 zfs)     size=11;; # 10 to 11KB per inode
819         esac
820
821         echo -n $size
822 }
823
824 check_gss_daemon_nodes() {
825     local list=$1
826     dname=$2
827
828     do_nodesv $list "num=\\\$(ps -o cmd -C $dname | grep $dname | wc -l);
829 if [ \\\"\\\$num\\\" -ne 1 ]; then
830     echo \\\$num instance of $dname;
831     exit 1;
832 fi; "
833 }
834
835 check_gss_daemon_facet() {
836     facet=$1
837     dname=$2
838
839     num=`do_facet $facet ps -o cmd -C $dname | grep $dname | wc -l`
840     if [ $num -ne 1 ]; then
841         echo "$num instance of $dname on $facet"
842         return 1
843     fi
844     return 0
845 }
846
847 send_sigint() {
848     local list=$1
849     shift
850     echo Stopping $@ on $list
851     do_nodes $list "killall -2 $@ 2>/dev/null || true"
852 }
853
854 # start gss daemons on all nodes, or "daemon" on "nodes" if set
855 start_gss_daemons() {
856         local nodes=$1
857         local daemon=$2
858
859         if [ "$nodes" ] && [ "$daemon" ] ; then
860                 echo "Starting gss daemon on nodes: $nodes"
861                 do_nodes $nodes "$daemon" || return 8
862                 return 0
863         fi
864
865         nodes=$(comma_list $(mdts_nodes))
866         echo "Starting gss daemon on mds: $nodes"
867         if $GSS_SK; then
868                 # Start all versions, in case of switching
869                 do_nodes $nodes "$LSVCGSSD -vvv -s -m -o -z" || return 1
870         else
871                 do_nodes $nodes "$LSVCGSSD -v" || return 1
872         fi
873         if $GSS_PIPEFS; then
874                 do_nodes $nodes "$LGSSD -v" || return 2
875         fi
876
877         nodes=$(comma_list $(osts_nodes))
878         echo "Starting gss daemon on ost: $nodes"
879         if $GSS_SK; then
880                 # Start all versions, in case of switching
881                 do_nodes $nodes "$LSVCGSSD -vvv -s -m -o -z" || return 3
882         else
883                 do_nodes $nodes "$LSVCGSSD -v" || return 3
884         fi
885         # starting on clients
886
887         local clients=${CLIENTS:-$HOSTNAME}
888         if $GSS_PIPEFS; then
889                 echo "Starting $LGSSD on clients $clients "
890                 do_nodes $clients  "$LGSSD -v" || return 4
891         fi
892
893         # wait daemons entering "stable" status
894         sleep 5
895
896         #
897         # check daemons are running
898         #
899         nodes=$(comma_list $(mdts_nodes) $(osts_nodes))
900         check_gss_daemon_nodes $nodes lsvcgssd || return 5
901         if $GSS_PIPEFS; then
902                 nodes=$(comma_list $(mdts_nodes))
903                 check_gss_daemon_nodes $nodes lgssd || return 6
904         fi
905         if $GSS_PIPEFS; then
906                 check_gss_daemon_nodes $clients lgssd || return 7
907         fi
908 }
909
910 stop_gss_daemons() {
911         local nodes=$(comma_list $(mdts_nodes))
912
913         send_sigint $nodes lsvcgssd lgssd
914
915         nodes=$(comma_list $(osts_nodes))
916         send_sigint $nodes lsvcgssd
917
918         nodes=${CLIENTS:-$HOSTNAME}
919         send_sigint $nodes lgssd
920 }
921
922 add_sk_mntflag() {
923         # Add mount flags for shared key
924         local mt_opts=$@
925         if grep -q skpath <<< "$mt_opts" ; then
926                 mt_opts=$(echo $mt_opts |
927                         sed -e "s#skpath=[^ ,]*#skpath=$SK_PATH#")
928         else
929                 if [ -z "$mt_opts" ]; then
930                         mt_opts="-o skpath=$SK_PATH"
931                 else
932                         mt_opts="$mt_opts,skpath=$SK_PATH"
933                 fi
934         fi
935         echo -n $mt_opts
936 }
937
938 init_gss() {
939         if $SHARED_KEY; then
940                 GSS=true
941                 GSS_SK=true
942         fi
943
944         if ! $GSS; then
945                 return
946         fi
947
948         if ! module_loaded ptlrpc_gss; then
949                 load_module ptlrpc/gss/ptlrpc_gss
950                 module_loaded ptlrpc_gss ||
951                         error_exit "init_gss: GSS=$GSS, but gss/krb5 missing"
952         fi
953
954         if $GSS_KRB5 || $GSS_SK; then
955                 start_gss_daemons || error_exit "start gss daemon failed! rc=$?"
956         fi
957
958         if $GSS_SK && $SK_NO_KEY; then
959                 local numclients=${1:-$CLIENTCOUNT}
960                 local clients=${CLIENTS:-$HOSTNAME}
961
962                 # security ctx config for keyring
963                 SK_NO_KEY=false
964                 mkdir -p $SK_OM_PATH
965                 if grep -q request-key /proc/mounts > /dev/null; then
966                         echo "SSK: Request key already mounted."
967                 else
968                         mount -o bind $SK_OM_PATH /etc/request-key.d/
969                 fi
970                 local lgssc_conf_line='create lgssc * * '
971                 lgssc_conf_line+=$(which lgss_keyring)
972                 lgssc_conf_line+=' %o %k %t %d %c %u %g %T %P %S'
973
974                 local lgssc_conf_file="/etc/request-key.d/lgssc.conf"
975                 echo "$lgssc_conf_line" > $lgssc_conf_file
976                 [ -e $lgssc_conf_file ] ||
977                         error_exit "Could not find key options in $lgssc_conf_file"
978
979                 if ! local_mode; then
980                         do_nodes $(comma_list $(all_nodes)) "mkdir -p \
981                                 $SK_OM_PATH"
982                         do_nodes $(comma_list $(all_nodes)) "mount \
983                                 -o bind $SK_OM_PATH \
984                                 /etc/request-key.d/"
985                         do_nodes $(comma_list $(all_nodes)) "rsync -aqv \
986                                 $HOSTNAME:$lgssc_conf_file \
987                                 $lgssc_conf_file >/dev/null 2>&1"
988                 fi
989
990                 # create shared key on all nodes
991                 mkdir -p $SK_PATH/nodemap
992                 rm -f $SK_PATH/$FSNAME.key $SK_PATH/nodemap/c*.key \
993                         $SK_PATH/$FSNAME-*.key
994                 # for nodemap testing each client may need own key,
995                 # and S2S now requires keys as well, both for "client"
996                 # and for "server"
997                 if $SK_S2S; then
998                         lgss_sk -t server -f$FSNAME -n $SK_S2SNMCLI \
999                                 -w $SK_PATH/$FSNAME-nmclient.key \
1000                                 -d /dev/urandom >/dev/null 2>&1
1001                         lgss_sk -t mgs,server -f$FSNAME -n $SK_S2SNM \
1002                                 -w $SK_PATH/$FSNAME-s2s-server.key \
1003                                 -d /dev/urandom >/dev/null 2>&1
1004                 fi
1005                 # basic key create
1006                 lgss_sk -t server -f$FSNAME -w $SK_PATH/$FSNAME.key \
1007                         -d /dev/urandom >/dev/null 2>&1
1008                 # per-nodemap keys
1009                 for i in $(seq 0 $((numclients - 1))); do
1010                         lgss_sk -t server -f$FSNAME -n c$i \
1011                                 -w $SK_PATH/nodemap/c$i.key -d /dev/urandom \
1012                                 >/dev/null 2>&1
1013                 done
1014                 # Distribute keys
1015                 if ! local_mode; then
1016                         do_nodes $(comma_list $(all_nodes)) "rsync -av \
1017                                 $HOSTNAME:$SK_PATH/ $SK_PATH >/dev/null 2>&1"
1018                 fi
1019                 # Set client keys to client type to generate prime P
1020                 if local_mode; then
1021                         do_nodes $(all_nodes) "lgss_sk -t client,server -m \
1022                                 $SK_PATH/$FSNAME.key >/dev/null 2>&1"
1023                 else
1024                         do_nodes $clients "lgss_sk -t client -m \
1025                                 $SK_PATH/$FSNAME.key >/dev/null 2>&1"
1026                         do_nodes $clients "find $SK_PATH/nodemap -name \*.key | \
1027                                 xargs -IX lgss_sk -t client -m X >/dev/null 2>&1"
1028                 fi
1029                 # This is required for servers as well, if S2S in use
1030                 if $SK_S2S; then
1031                         do_nodes $(comma_list $(mdts_nodes)) \
1032                                 "cp $SK_PATH/$FSNAME-s2s-server.key \
1033                                 $SK_PATH/$FSNAME-s2s-client.key; lgss_sk \
1034                                 -t client -m $SK_PATH/$FSNAME-s2s-client.key \
1035                                 >/dev/null 2>&1"
1036                         do_nodes $(comma_list $(osts_nodes)) \
1037                                 "cp $SK_PATH/$FSNAME-s2s-server.key \
1038                                 $SK_PATH/$FSNAME-s2s-client.key; lgss_sk \
1039                                 -t client -m $SK_PATH/$FSNAME-s2s-client.key \
1040                                 >/dev/null 2>&1"
1041                         do_nodes $clients "lgss_sk -t client \
1042                                 -m $SK_PATH/$FSNAME-nmclient.key \
1043                                  >/dev/null 2>&1"
1044                 fi
1045         fi
1046         if $GSS_SK; then
1047                 # mount options for servers and clients
1048                 MGS_MOUNT_OPTS=$(add_sk_mntflag $MGS_MOUNT_OPTS)
1049                 MDS_MOUNT_OPTS=$(add_sk_mntflag $MDS_MOUNT_OPTS)
1050                 OST_MOUNT_OPTS=$(add_sk_mntflag $OST_MOUNT_OPTS)
1051                 MOUNT_OPTS=$(add_sk_mntflag $MOUNT_OPTS)
1052                 SEC=$SK_FLAVOR
1053         fi
1054
1055         if [ -n "$LGSS_KEYRING_DEBUG" ]; then
1056                 lctl set_param -n \
1057                         sptlrpc.gss.lgss_keyring.debug_level=$LGSS_KEYRING_DEBUG
1058         fi
1059 }
1060
1061 cleanup_gss() {
1062         if $GSS; then
1063                 stop_gss_daemons
1064                 # maybe cleanup credential cache?
1065         fi
1066 }
1067
1068 cleanup_sk() {
1069         if $GSS_SK; then
1070                 if $SK_S2S; then
1071                         do_node $(mgs_node) "$LCTL nodemap_del $SK_S2SNM"
1072                         do_node $(mgs_node) "$LCTL nodemap_del $SK_S2SNMCLI"
1073                         $RPC_MODE || echo "Sleeping for 10 sec for Nodemap.."
1074                         sleep 10
1075                 fi
1076                 stop_gss_daemons
1077                 $RPC_MODE || echo "Cleaning up Shared Key.."
1078                 do_nodes $(comma_list $(all_nodes)) "rm -f \
1079                         $SK_PATH/$FSNAME*.key $SK_PATH/nodemap/$FSNAME*.key"
1080                 do_nodes $(comma_list $(all_nodes)) "keyctl show | \
1081                   awk '/lustre/ { print \\\$1 }' | xargs -IX keyctl unlink X"
1082                 # Remove the mount and clean up the files we added to SK_PATH
1083                 do_nodes $(comma_list $(all_nodes)) "while grep -q \
1084                         request-key.d /proc/mounts; do umount \
1085                         /etc/request-key.d/; done"
1086                 do_nodes $(comma_list $(all_nodes)) "rm -f \
1087                         $SK_OM_PATH/lgssc.conf"
1088                 do_nodes $(comma_list $(all_nodes)) "rmdir $SK_OM_PATH"
1089                 SK_NO_KEY=true
1090         fi
1091 }
1092
1093 facet_svc() {
1094         local facet=$1
1095         local var=${facet}_svc
1096
1097         echo -n ${!var}
1098 }
1099
1100 facet_type() {
1101         local facet=$1
1102
1103         echo -n $facet | sed -e 's/^fs[0-9]\+//' -e 's/[0-9_]\+//' |
1104                 tr '[:lower:]' '[:upper:]'
1105 }
1106
1107 facet_number() {
1108         local facet=$1
1109
1110         if [ $facet == mgs ] || [ $facet == client ]; then
1111                 return 1
1112         fi
1113
1114         echo -n $facet | sed -e 's/^fs[0-9]\+//' | sed -e 's/^[a-z]\+//'
1115 }
1116
1117 facet_fstype() {
1118         local facet=$1
1119         local var
1120
1121         var=${facet}_FSTYPE
1122         if [ -n "${!var}" ]; then
1123                 echo -n ${!var}
1124                 return
1125         fi
1126
1127         var=$(facet_type $facet)FSTYPE
1128         if [ -n "${!var}" ]; then
1129                 echo -n ${!var}
1130                 return
1131         fi
1132
1133         if [ -n "$FSTYPE" ]; then
1134                 echo -n $FSTYPE
1135                 return
1136         fi
1137
1138         if [[ $facet == mgs ]] && combined_mgs_mds; then
1139                 facet_fstype mds1
1140                 return
1141         fi
1142
1143         return 1
1144 }
1145
1146 node_fstypes() {
1147         local node=$1
1148         local fstypes
1149         local fstype
1150         local facets=$(get_facets)
1151         local facet
1152
1153         for facet in ${facets//,/ }; do
1154                 if [ $node == $(facet_host $facet) ] ||
1155                    [ $node == "$(facet_failover_host $facet)" ]; then
1156                         fstype=$(facet_fstype $facet)
1157                         if [[ $fstypes != *$fstype* ]]; then
1158                                 fstypes+="${fstypes:+,}$fstype"
1159                         fi
1160                 fi
1161         done
1162         echo -n $fstypes
1163 }
1164
1165 facet_index() {
1166         local facet=$1
1167         local num=$(facet_number $facet)
1168         local index
1169
1170         if [[ $(facet_type $facet) = OST ]]; then
1171                 index=OSTINDEX${num}
1172                 if [[ -n "${!index}" ]]; then
1173                         echo -n ${!index}
1174                         return
1175                 fi
1176
1177                 index=${OST_INDICES[num - 1]}
1178         fi
1179
1180         [[ -n "$index" ]] || index=$((num - 1))
1181         echo -n $index
1182 }
1183
1184 devicelabel() {
1185         local facet=$1
1186         local dev=$2
1187         local label
1188         local fstype=$(facet_fstype $facet)
1189
1190         case $fstype in
1191         ldiskfs)
1192                 label=$(do_facet ${facet} "$E2LABEL ${dev} 2>/dev/null");;
1193         zfs)
1194                 label=$(do_facet ${facet} "$ZFS get -H -o value lustre:svname \
1195                                            ${dev} 2>/dev/null");;
1196         *)
1197                 error "unknown fstype!";;
1198         esac
1199
1200         echo -n $label
1201 }
1202
1203 mdsdevlabel() {
1204         local num=$1
1205         local device=$(mdsdevname $num)
1206         local label=$(devicelabel mds$num ${device} | grep -v "CMD: ")
1207         echo -n $label
1208 }
1209
1210 ostdevlabel() {
1211         local num=$1
1212         local device=$(ostdevname $num)
1213         local label=$(devicelabel ost$num ${device} | grep -v "CMD: ")
1214         echo -n $label
1215 }
1216
1217 #
1218 # Get the device of a facet.
1219 #
1220 facet_device() {
1221         local facet=$1
1222         local device
1223
1224         case $facet in
1225                 mgs) device=$(mgsdevname) ;;
1226                 mds*) device=$(mdsdevname $(facet_number $facet)) ;;
1227                 ost*) device=$(ostdevname $(facet_number $facet)) ;;
1228                 fs2mds) device=$(mdsdevname 1_2) ;;
1229                 fs2ost) device=$(ostdevname 1_2) ;;
1230                 fs3ost) device=$(ostdevname 2_2) ;;
1231                 *) ;;
1232         esac
1233
1234         echo -n $device
1235 }
1236
1237 #
1238 # Get the virtual device of a facet.
1239 #
1240 facet_vdevice() {
1241         local facet=$1
1242         local device
1243
1244         case $facet in
1245                 mgs) device=$(mgsvdevname) ;;
1246                 mds*) device=$(mdsvdevname $(facet_number $facet)) ;;
1247                 ost*) device=$(ostvdevname $(facet_number $facet)) ;;
1248                 fs2mds) device=$(mdsvdevname 1_2) ;;
1249                 fs2ost) device=$(ostvdevname 1_2) ;;
1250                 fs3ost) device=$(ostvdevname 2_2) ;;
1251                 *) ;;
1252         esac
1253
1254         echo -n $device
1255 }
1256
1257 running_in_vm() {
1258         local virt=$(virt-what 2> /dev/null)
1259
1260         [ $? -eq 0 ] && [ -n "$virt" ] && { echo $virt; return; }
1261
1262         virt=$(dmidecode -s system-product-name | awk '{print $1}')
1263
1264         case $virt in
1265                 VMware|KVM|VirtualBox|Parallels)
1266                         echo $virt | tr '[A-Z]' '[a-z]' ;;
1267                 *) ;;
1268         esac
1269 }
1270
1271 #
1272 # Re-read the partition table on failover partner host.
1273 # After a ZFS storage pool is created on a shared device, the partition table
1274 # on the device may change. However, the operating system on the failover
1275 # host may not notice the change automatically. Without the up-to-date partition
1276 # block devices, 'zpool import ..' cannot find the labels, whose positions are
1277 # relative to partition rather than disk beginnings.
1278 #
1279 # This function performs partprobe on the failover host to make it re-read the
1280 # partition table.
1281 #
1282 refresh_partition_table() {
1283         local facet=$1
1284         local device=$2
1285         local host
1286
1287         host=$(facet_passive_host $facet)
1288         if [[ -n "$host" ]]; then
1289                 do_node $host "$PARTPROBE $device"
1290         fi
1291 }
1292
1293 #
1294 # Get ZFS storage pool name.
1295 #
1296 zpool_name() {
1297         local facet=$1
1298         local device
1299         local poolname
1300
1301         device=$(facet_device $facet)
1302         # poolname is string before "/"
1303         poolname="${device%%/*}"
1304
1305         echo -n $poolname
1306 }
1307
1308 #
1309 #
1310 # Get ZFS local fsname.
1311 #
1312 zfs_local_fsname() {
1313         local facet=$1
1314         local lfsname=$(basename $(facet_device $facet))
1315
1316         echo -n $lfsname
1317 }
1318
1319 #
1320 # Create ZFS storage pool.
1321 #
1322 create_zpool() {
1323         local facet=$1
1324         local poolname=$2
1325         local vdev=$3
1326         shift 3
1327         local opts=${@:-"-o cachefile=none"}
1328
1329         do_facet $facet "lsmod | grep zfs >&/dev/null || modprobe zfs;
1330                 $ZPOOL list -H $poolname >/dev/null 2>&1 ||
1331                 $ZPOOL create -f $opts $poolname $vdev"
1332 }
1333
1334 #
1335 # Create ZFS file system.
1336 #
1337 create_zfs() {
1338         local facet=$1
1339         local dataset=$2
1340         shift 2
1341         local opts=${@:-"-o mountpoint=legacy"}
1342
1343         do_facet $facet "$ZFS list -H $dataset >/dev/null 2>&1 ||
1344                 $ZFS create $opts $dataset"
1345 }
1346
1347 #
1348 # Export ZFS storage pool.
1349 # Before exporting the pool, all datasets within the pool should be unmounted.
1350 #
1351 export_zpool() {
1352         local facet=$1
1353         shift
1354         local opts="$@"
1355         local poolname
1356
1357         poolname=$(zpool_name $facet)
1358
1359         if [[ -n "$poolname" ]]; then
1360                 do_facet $facet "! $ZPOOL list -H $poolname >/dev/null 2>&1 ||
1361                         grep -q ^$poolname/ /proc/mounts ||
1362                         $ZPOOL export $opts $poolname"
1363         fi
1364 }
1365
1366 #
1367 # Destroy ZFS storage pool.
1368 # Destroy the given pool and free up any devices for other use. This command
1369 # tries to unmount any active datasets before destroying the pool.
1370 # -f    Force any active datasets contained within the pool to be unmounted.
1371 #
1372 destroy_zpool() {
1373         local facet=$1
1374         local poolname=${2:-$(zpool_name $facet)}
1375
1376         if [[ -n "$poolname" ]]; then
1377                 do_facet $facet "! $ZPOOL list -H $poolname >/dev/null 2>&1 ||
1378                         $ZPOOL destroy -f $poolname"
1379         fi
1380 }
1381
1382 #
1383 # Import ZFS storage pool.
1384 # Force importing, even if the pool appears to be potentially active.
1385 #
1386 import_zpool() {
1387         local facet=$1
1388         shift
1389         local opts=${@:-"-o cachefile=none -o failmode=panic"}
1390         local poolname
1391
1392         poolname=$(zpool_name $facet)
1393
1394         if [[ -n "$poolname" ]]; then
1395                 opts+=" -d $(dirname $(facet_vdevice $facet))"
1396                 do_facet $facet "lsmod | grep zfs >&/dev/null || modprobe zfs;
1397                         $ZPOOL list -H $poolname >/dev/null 2>&1 ||
1398                         $ZPOOL import -f $opts $poolname"
1399         fi
1400 }
1401
1402 #
1403 # Reimport ZFS storage pool with new name
1404 #
1405 reimport_zpool() {
1406         local facet=$1
1407         local newpool=$2
1408         local opts="-o cachefile=none"
1409         local poolname=$(zpool_name $facet)
1410
1411         opts+=" -d $(dirname $(facet_vdevice $facet))"
1412         do_facet $facet "$ZPOOL export $poolname;
1413                          $ZPOOL import $opts $poolname $newpool"
1414 }
1415
1416 #
1417 # Set the "cachefile=none" property on ZFS storage pool so that the pool
1418 # is not automatically imported on system startup.
1419 #
1420 # In a failover environment, this will provide resource level fencing which
1421 # will ensure that the same ZFS storage pool will not be imported concurrently
1422 # on different nodes.
1423 #
1424 disable_zpool_cache() {
1425         local facet=$1
1426         local poolname
1427
1428         poolname=$(zpool_name $facet)
1429
1430         if [[ -n "$poolname" ]]; then
1431                 do_facet $facet "$ZPOOL set cachefile=none $poolname"
1432         fi
1433 }
1434
1435 #
1436 # This and set_osd_param() shall be used to access OSD parameters
1437 # once existed under "obdfilter":
1438 #
1439 #   mntdev
1440 #   stats
1441 #   read_cache_enable
1442 #   writethrough_cache_enable
1443 #
1444 get_osd_param() {
1445         local nodes=$1
1446         local device=${2:-$FSNAME-OST*}
1447         local name=$3
1448
1449         do_nodes $nodes "$LCTL get_param -n obdfilter.$device.$name \
1450                 osd-*.$device.$name 2>&1" | grep -v 'error:'
1451 }
1452
1453 set_osd_param() {
1454         local nodes=$1
1455         local device=${2:-$FSNAME-OST*}
1456         local name=$3
1457         local value=$4
1458
1459         do_nodes $nodes "$LCTL set_param -n obdfilter.$device.$name=$value \
1460                 osd-*.$device.$name=$value 2>&1" | grep -v 'error:'
1461 }
1462
1463 set_debug_size () {
1464     local dz=${1:-$DEBUG_SIZE}
1465
1466     if [ -f /sys/devices/system/cpu/possible ]; then
1467         local cpus=$(($(cut -d "-" -f 2 /sys/devices/system/cpu/possible)+1))
1468     else
1469         local cpus=$(getconf _NPROCESSORS_CONF 2>/dev/null)
1470     fi
1471
1472     # bug 19944, adjust size to be -gt num_possible_cpus()
1473     # promise 2MB for every cpu at least
1474     if [ -n "$cpus" ] && [ $((cpus * 2)) -gt $dz ]; then
1475         dz=$((cpus * 2))
1476     fi
1477     lctl set_param debug_mb=$dz
1478 }
1479
1480 set_default_debug () {
1481     local debug=${1:-"$PTLDEBUG"}
1482     local subsys=${2:-"$SUBSYSTEM"}
1483     local debug_size=${3:-$DEBUG_SIZE}
1484
1485     [ -n "$debug" ] && lctl set_param debug="$debug" >/dev/null
1486     [ -n "$subsys" ] && lctl set_param subsystem_debug="${subsys# }" >/dev/null
1487
1488     [ -n "$debug_size" ] && set_debug_size $debug_size > /dev/null
1489 }
1490
1491 set_default_debug_nodes () {
1492         local nodes="$1"
1493
1494         if [[ ,$nodes, = *,$HOSTNAME,* ]]; then
1495                 nodes=$(exclude_items_from_list "$nodes" "$HOSTNAME")
1496                 set_default_debug
1497         fi
1498
1499         do_rpc_nodes "$nodes" set_default_debug \
1500                 \\\"$PTLDEBUG\\\" \\\"$SUBSYSTEM\\\" $DEBUG_SIZE || true
1501 }
1502
1503 set_default_debug_facet () {
1504     local facet=$1
1505     local node=$(facet_active_host $facet)
1506     [ -z "$node" ] && echo "No host defined for facet $facet" && exit 1
1507
1508     set_default_debug_nodes $node
1509 }
1510
1511 set_hostid () {
1512     local hostid=${1:-$(hostid)}
1513
1514     if [ ! -s /etc/hostid ]; then
1515         printf $(echo -n $hostid |
1516             sed 's/\(..\)\(..\)\(..\)\(..\)/\\x\4\\x\3\\x\2\\x\1/') >/etc/hostid
1517     fi
1518 }
1519
1520 # Facet functions
1521 mount_facets () {
1522         local facets=${1:-$(get_facets)}
1523         local facet
1524
1525         for facet in ${facets//,/ }; do
1526                 mount_facet $facet
1527                 local RC=$?
1528                 [ $RC -eq 0 ] && continue
1529
1530                 if [ "$TESTSUITE.$TESTNAME" = "replay-dual.test_0a" ]; then
1531                         skip_noexit "Restart of $facet failed!." &&
1532                                 touch $LU482_FAILED
1533                 else
1534                         error "Restart of $facet failed!"
1535                 fi
1536                 return $RC
1537         done
1538 }
1539
1540 #
1541 # Add argument "arg" (e.g., "loop") to the comma-separated list
1542 # of arguments for option "opt" (e.g., "-o") on command
1543 # line "opts" (e.g., "-o flock").
1544 #
1545 csa_add() {
1546         local opts=$1
1547         local opt=$2
1548         local arg=$3
1549         local opt_pattern="\([[:space:]]\+\|^\)$opt"
1550
1551         if echo "$opts" | grep -q $opt_pattern; then
1552                 opts=$(echo "$opts" | sed -e \
1553                         "s/$opt_pattern[[:space:]]*[^[:space:]]\+/&,$arg/")
1554         else
1555                 opts+="${opts:+ }$opt $arg"
1556         fi
1557         echo -n "$opts"
1558 }
1559
1560 #
1561 # Associate loop device with a given regular file.
1562 # Return the loop device.
1563 #
1564 setup_loop_device() {
1565         local facet=$1
1566         local file=$2
1567
1568         do_facet $facet "loop_dev=\\\$($LOSETUP -j $file | cut -d : -f 1);
1569                          if [[ -z \\\$loop_dev ]]; then
1570                                 loop_dev=\\\$($LOSETUP -f);
1571                                 $LOSETUP \\\$loop_dev $file || loop_dev=;
1572                          fi;
1573                          echo -n \\\$loop_dev"
1574 }
1575
1576 #
1577 # Detach a loop device.
1578 #
1579 cleanup_loop_device() {
1580         local facet=$1
1581         local loop_dev=$2
1582
1583         do_facet $facet "! $LOSETUP $loop_dev >/dev/null 2>&1 ||
1584                          $LOSETUP -d $loop_dev"
1585 }
1586
1587 #
1588 # Check if a given device is a block device.
1589 #
1590 is_blkdev() {
1591         local facet=$1
1592         local dev=$2
1593         local size=${3:-""}
1594
1595         [[ -n "$dev" ]] || return 1
1596         do_facet $facet "test -b $dev" || return 1
1597         if [[ -n "$size" ]]; then
1598                 local in=$(do_facet $facet "dd if=$dev of=/dev/null bs=1k \
1599                                             count=1 skip=$size 2>&1" |
1600                                             awk '($3 == "in") { print $1 }')
1601                 [[ "$in" = "1+0" ]] || return 1
1602         fi
1603 }
1604
1605 #
1606 # Check if a given device is a device-mapper device.
1607 #
1608 is_dm_dev() {
1609         local facet=$1
1610         local dev=$2
1611
1612         [[ -n "$dev" ]] || return 1
1613         do_facet $facet "$DMSETUP status $dev >/dev/null 2>&1"
1614 }
1615
1616 #
1617 # Check if a given device is a device-mapper flakey device.
1618 #
1619 is_dm_flakey_dev() {
1620         local facet=$1
1621         local dev=$2
1622         local type
1623
1624         [[ -n "$dev" ]] || return 1
1625
1626         type=$(do_facet $facet "$DMSETUP status $dev 2>&1" |
1627                awk '{print $3}')
1628         [[ $type = flakey ]] && return 0 || return 1
1629 }
1630
1631 #
1632 # Check if device-mapper flakey device is supported by the kernel
1633 # of $facet node or not.
1634 #
1635 dm_flakey_supported() {
1636         local facet=$1
1637
1638         $FLAKEY || return 1
1639         do_facet $facet "modprobe dm-flakey;
1640                          $DMSETUP targets | grep -q flakey" &> /dev/null
1641 }
1642
1643 #
1644 # Get the device-mapper flakey device name of a given facet.
1645 #
1646 dm_facet_devname() {
1647         local facet=$1
1648         [[ $facet = mgs ]] && combined_mgs_mds && facet=mds1
1649
1650         echo -n ${facet}_flakey
1651 }
1652
1653 #
1654 # Get the device-mapper flakey device of a given facet.
1655 # A device created by dmsetup will appear as /dev/mapper/<device-name>.
1656 #
1657 dm_facet_devpath() {
1658         local facet=$1
1659
1660         echo -n $DM_DEV_PATH/$(dm_facet_devname $facet)
1661 }
1662
1663 #
1664 # Set a device-mapper device with a new table.
1665 #
1666 # The table has the following format:
1667 # <logical_start_sector> <num_sectors> <target_type> <target_args>
1668 #
1669 # flakey <target_args> includes:
1670 # <destination_device> <offset> <up_interval> <down_interval> \
1671 # [<num_features> [<feature_arguments>]]
1672 #
1673 # linear <target_args> includes:
1674 # <destination_device> <start_sector>
1675 #
1676 dm_set_dev_table() {
1677         local facet=$1
1678         local dm_dev=$2
1679         local target_type=$3
1680         local num_sectors
1681         local real_dev
1682         local tmp
1683         local table
1684
1685         read tmp num_sectors tmp real_dev tmp \
1686                 <<< $(do_facet $facet "$DMSETUP table $dm_dev")
1687
1688         case $target_type in
1689         flakey)
1690                 table="0 $num_sectors flakey $real_dev 0 0 1800 1 drop_writes"
1691                 ;;
1692         linear)
1693                 table="0 $num_sectors linear $real_dev 0"
1694                 ;;
1695         *) error "invalid target type $target_type" ;;
1696         esac
1697
1698         do_facet $facet "$DMSETUP suspend --nolockfs --noflush $dm_dev" ||
1699                 error "failed to suspend $dm_dev"
1700         do_facet $facet "$DMSETUP load $dm_dev --table \\\"$table\\\"" ||
1701                 error "failed to load $target_type table into $dm_dev"
1702         do_facet $facet "$DMSETUP resume $dm_dev" ||
1703                 error "failed to resume $dm_dev"
1704 }
1705
1706 #
1707 # Set a device-mapper flakey device as "read-only" by using the "drop_writes"
1708 # feature parameter.
1709 #
1710 # drop_writes:
1711 #       All write I/O is silently ignored.
1712 #       Read I/O is handled correctly.
1713 #
1714 dm_set_dev_readonly() {
1715         local facet=$1
1716         local dm_dev=${2:-$(dm_facet_devpath $facet)}
1717
1718         dm_set_dev_table $facet $dm_dev flakey
1719 }
1720
1721 #
1722 # Set a device-mapper device to traditional linear mapping mode.
1723 #
1724 dm_clear_dev_readonly() {
1725         local facet=$1
1726         local dm_dev=${2:-$(dm_facet_devpath $facet)}
1727
1728         dm_set_dev_table $facet $dm_dev linear
1729 }
1730
1731 #
1732 # Set the device of a given facet as "read-only".
1733 #
1734 set_dev_readonly() {
1735         local facet=$1
1736         local svc=${facet}_svc
1737
1738         if [[ $(facet_fstype $facet) = zfs ]] ||
1739            ! dm_flakey_supported $facet; then
1740                 do_facet $facet $LCTL --device ${!svc} readonly
1741         else
1742                 dm_set_dev_readonly $facet
1743         fi
1744 }
1745
1746 #
1747 # Get size in 512-byte sectors (BLKGETSIZE64 / 512) of a given device.
1748 #
1749 get_num_sectors() {
1750         local facet=$1
1751         local dev=$2
1752         local num_sectors
1753
1754         num_sectors=$(do_facet $facet "blockdev --getsz $dev 2>/dev/null")
1755         [[ ${PIPESTATUS[0]} = 0 && -n "$num_sectors" ]] || num_sectors=0
1756         echo -n $num_sectors
1757 }
1758
1759 #
1760 # Create a device-mapper device with a given block device or regular file (will
1761 # be associated with loop device).
1762 # Return the full path of the device-mapper device.
1763 #
1764 dm_create_dev() {
1765         local facet=$1
1766         local real_dev=$2                                  # destination device
1767         local dm_dev_name=${3:-$(dm_facet_devname $facet)} # device name
1768         local dm_dev=$DM_DEV_PATH/$dm_dev_name            # device-mapper device
1769
1770         # check if the device-mapper device to be created already exists
1771         if is_dm_dev $facet $dm_dev; then
1772                 # if the existing device was set to "read-only", then clear it
1773                 ! is_dm_flakey_dev $facet $dm_dev ||
1774                         dm_clear_dev_readonly $facet $dm_dev
1775
1776                 echo -n $dm_dev
1777                 return 0
1778         fi
1779
1780         # check if the destination device is a block device, and if not,
1781         # associate it with a loop device
1782         is_blkdev $facet $real_dev ||
1783                 real_dev=$(setup_loop_device $facet $real_dev)
1784         [[ -n "$real_dev" ]] || { echo -n $real_dev; return 2; }
1785
1786         # now create the device-mapper device
1787         local num_sectors=$(get_num_sectors $facet $real_dev)
1788         local table="0 $num_sectors linear $real_dev 0"
1789         local rc=0
1790
1791         do_facet $facet "$DMSETUP create $dm_dev_name --table \\\"$table\\\"" ||
1792                 { rc=${PIPESTATUS[0]}; dm_dev=; }
1793         do_facet $facet "$DMSETUP mknodes >/dev/null 2>&1"
1794
1795         echo -n $dm_dev
1796         return $rc
1797 }
1798
1799 #
1800 # Map the facet name to its device variable name.
1801 #
1802 facet_device_alias() {
1803         local facet=$1
1804         local dev_alias=$facet
1805
1806         case $facet in
1807                 fs2mds) dev_alias=mds1_2 ;;
1808                 fs2ost) dev_alias=ost1_2 ;;
1809                 fs3ost) dev_alias=ost2_2 ;;
1810                 *) ;;
1811         esac
1812
1813         echo -n $dev_alias
1814 }
1815
1816 #
1817 # Save the original value of the facet device and export the new value.
1818 #
1819 export_dm_dev() {
1820         local facet=$1
1821         local dm_dev=$2
1822
1823         local active_facet=$(facet_active $facet)
1824         local dev_alias=$(facet_device_alias $active_facet)
1825         local dev_name=${dev_alias}_dev
1826         local dev=${!dev_name}
1827
1828         if [[ $active_facet = $facet ]]; then
1829                 local failover_dev=${dev_alias}failover_dev
1830                 if [[ ${!failover_dev} = $dev ]]; then
1831                         eval export ${failover_dev}_saved=$dev
1832                         eval export ${failover_dev}=$dm_dev
1833                 fi
1834         else
1835                 dev_alias=$(facet_device_alias $facet)
1836                 local facet_dev=${dev_alias}_dev
1837                 if [[ ${!facet_dev} = $dev ]]; then
1838                         eval export ${facet_dev}_saved=$dev
1839                         eval export ${facet_dev}=$dm_dev
1840                 fi
1841         fi
1842
1843         eval export ${dev_name}_saved=$dev
1844         eval export ${dev_name}=$dm_dev
1845 }
1846
1847 #
1848 # Restore the saved value of the facet device.
1849 #
1850 unexport_dm_dev() {
1851         local facet=$1
1852
1853         [[ $facet = mgs ]] && combined_mgs_mds && facet=mds1
1854         local dev_alias=$(facet_device_alias $facet)
1855
1856         local saved_dev=${dev_alias}_dev_saved
1857         [[ -z ${!saved_dev} ]] ||
1858                 eval export ${dev_alias}_dev=${!saved_dev}
1859
1860         saved_dev=${dev_alias}failover_dev_saved
1861         [[ -z ${!saved_dev} ]] ||
1862                 eval export ${dev_alias}failover_dev=${!saved_dev}
1863 }
1864
1865 #
1866 # Remove a device-mapper device.
1867 # If the destination device is a loop device, then also detach it.
1868 #
1869 dm_cleanup_dev() {
1870         local facet=$1
1871         local dm_dev=${2:-$(dm_facet_devpath $facet)}
1872         local major
1873         local minor
1874
1875         is_dm_dev $facet $dm_dev || return 0
1876
1877         read major minor <<< $(do_facet $facet "$DMSETUP table $dm_dev" |
1878                 awk '{ print $4 }' | awk -F: '{ print $1" "$2 }')
1879
1880         do_facet $facet "$DMSETUP remove $dm_dev"
1881         do_facet $facet "$DMSETUP mknodes >/dev/null 2>&1"
1882
1883         unexport_dm_dev $facet
1884
1885         # detach a loop device
1886         [[ $major -ne 7 ]] || cleanup_loop_device $facet /dev/loop$minor
1887
1888         # unload dm-flakey module
1889         do_facet $facet "modprobe -r dm-flakey" || true
1890 }
1891
1892 mount_facet() {
1893         local facet=$1
1894         shift
1895         local active_facet=$(facet_active $facet)
1896         local dev_alias=$(facet_device_alias $active_facet)
1897         local dev=${dev_alias}_dev
1898         local opt=${facet}_opt
1899         local mntpt=$(facet_mntpt $facet)
1900         local opts="${!opt} $@"
1901         local fstype=$(facet_fstype $facet)
1902         local devicelabel
1903         local dm_dev=${!dev}
1904
1905         module_loaded lustre || load_modules
1906
1907         case $fstype in
1908         ldiskfs)
1909                 if dm_flakey_supported $facet; then
1910                         dm_dev=$(dm_create_dev $facet ${!dev})
1911                         [[ -n "$dm_dev" ]] || dm_dev=${!dev}
1912                 fi
1913
1914                 is_blkdev $facet $dm_dev || opts=$(csa_add "$opts" -o loop)
1915
1916                 devicelabel=$(do_facet ${facet} "$E2LABEL $dm_dev");;
1917         zfs)
1918                 # import ZFS storage pool
1919                 import_zpool $facet || return ${PIPESTATUS[0]}
1920
1921                 devicelabel=$(do_facet ${facet} "$ZFS get -H -o value \
1922                                                 lustre:svname $dm_dev");;
1923         *)
1924                 error "unknown fstype!";;
1925         esac
1926
1927         echo "Starting ${facet}: $opts $dm_dev $mntpt"
1928         # for testing LU-482 error handling in mount_facets() and test_0a()
1929         if [ -f $TMP/test-lu482-trigger ]; then
1930                 RC=2
1931         else
1932                 do_facet ${facet} \
1933                         "mkdir -p $mntpt; $MOUNT_CMD $opts $dm_dev $mntpt"
1934                 RC=${PIPESTATUS[0]}
1935         fi
1936
1937         if [ $RC -ne 0 ]; then
1938                 echo "Start of $dm_dev on ${facet} failed ${RC}"
1939                 return $RC
1940         fi
1941
1942         health=$(do_facet ${facet} "$LCTL get_param -n health_check")
1943         if [[ "$health" != "healthy" ]]; then
1944                 error "$facet is in a unhealthy state"
1945         fi
1946
1947         set_default_debug_facet $facet
1948
1949         if [[ $facet == mds* ]]; then
1950                 do_facet $facet \
1951                 lctl set_param -n mdt.${FSNAME}*.enable_remote_dir=1 2>/dev/null
1952         fi
1953
1954         if [[ $opts =~ .*nosvc.* ]]; then
1955                 echo "Start $dm_dev without service"
1956         else
1957
1958                 case $fstype in
1959                 ldiskfs)
1960                         wait_update_facet ${facet} "$E2LABEL $dm_dev \
1961                                 2>/dev/null | grep -E ':[a-zA-Z]{3}[0-9]{4}'" \
1962                                 "" || error "$dm_dev failed to initialize!";;
1963                 zfs)
1964                         wait_update_facet ${facet} "$ZFS get -H -o value \
1965                                 lustre:svname $dm_dev 2>/dev/null | \
1966                                 grep -E ':[a-zA-Z]{3}[0-9]{4}'" "" ||
1967                                 error "$dm_dev failed to initialize!";;
1968
1969                 *)
1970                         error "unknown fstype!";;
1971                 esac
1972         fi
1973
1974         # commit the device label change to disk
1975         if [[ $devicelabel =~ (:[a-zA-Z]{3}[0-9]{4}) ]]; then
1976                 echo "Commit the device label on ${!dev}"
1977                 do_facet $facet "sync; sleep 1; sync"
1978         fi
1979
1980
1981         label=$(devicelabel ${facet} $dm_dev)
1982         [ -z "$label" ] && echo no label for $dm_dev && exit 1
1983         eval export ${facet}_svc=${label}
1984         echo Started ${label}
1985
1986         export_dm_dev $facet $dm_dev
1987
1988         return $RC
1989 }
1990
1991 # start facet device options
1992 start() {
1993         local facet=$1
1994         shift
1995         local device=$1
1996         shift
1997         local dev_alias=$(facet_device_alias $facet)
1998
1999         eval export ${dev_alias}_dev=${device}
2000         eval export ${facet}_opt=\"$@\"
2001
2002         local varname=${dev_alias}failover_dev
2003         if [ -n "${!varname}" ] ; then
2004                 eval export ${dev_alias}failover_dev=${!varname}
2005         else
2006                 eval export ${dev_alias}failover_dev=$device
2007         fi
2008
2009         local mntpt=$(facet_mntpt $facet)
2010         do_facet ${facet} mkdir -p $mntpt
2011         eval export ${facet}_MOUNT=$mntpt
2012         mount_facet ${facet}
2013         RC=$?
2014
2015         if [[ $facet == mds* ]]; then
2016                 do_facet $facet \
2017                         lctl set_param -n mdt.${FSNAME}*.enable_remote_dir=1 \
2018                                 2>/dev/null
2019         fi
2020
2021         return $RC
2022 }
2023
2024 stop() {
2025         local running
2026         local facet=$1
2027         shift
2028         local HOST=$(facet_active_host $facet)
2029         [[ -z $HOST ]] && echo stop: no host for $facet && return 0
2030
2031         local mntpt=$(facet_mntpt $facet)
2032         running=$(do_facet ${facet} "grep -c $mntpt' ' /proc/mounts || true")
2033         if [ ${running} -ne 0 ]; then
2034                 echo "Stopping $mntpt (opts:$@) on $HOST"
2035                 do_facet ${facet} $UMOUNT $@ $mntpt
2036         fi
2037
2038         # umount should block, but we should wait for unrelated obd's
2039         # like the MGS or MGC to also stop.
2040         wait_exit_ST ${facet} || return ${PIPESTATUS[0]}
2041
2042         if [[ $(facet_fstype $facet) == zfs ]]; then
2043                 # export ZFS storage pool
2044                 [ "$KEEP_ZPOOL" = "true" ] || export_zpool $facet
2045         elif dm_flakey_supported $facet; then
2046                 local host=${facet}_HOST
2047                 local failover_host=${facet}failover_HOST
2048                 if [[ -n ${!failover_host} && ${!failover_host} != ${!host} ]]||
2049                         $CLEANUP_DM_DEV || [[ $facet = fs* ]]; then
2050                         dm_cleanup_dev $facet
2051                 fi
2052         fi
2053 }
2054
2055 # get mdt quota type
2056 mdt_quota_type() {
2057         local varsvc=${SINGLEMDS}_svc
2058         do_facet $SINGLEMDS $LCTL get_param -n \
2059                 osd-$(facet_fstype $SINGLEMDS).${!varsvc}.quota_slave.enabled
2060 }
2061
2062 # get ost quota type
2063 ost_quota_type() {
2064         # All OSTs should have same quota type
2065         local varsvc=ost1_svc
2066         do_facet ost1 $LCTL get_param -n \
2067                 osd-$(facet_fstype ost1).${!varsvc}.quota_slave.enabled
2068 }
2069
2070 # restore old quota type settings
2071 restore_quota() {
2072         if [ "$old_MDT_QUOTA_TYPE" ]; then
2073                 if [[ $PERM_CMD = *"set_param -P"* ]]; then
2074                         do_facet mgs $PERM_CMD \
2075                                 osd-*.$FSNAME-MDT*.quota_slave.enable = \
2076                                 $old_MDT_QUOTA_TYPE
2077                 else
2078                         do_facet mgs $PERM_CMD \
2079                                 $FSNAME.quota.mdt=$old_MDT_QUOTA_TYPE
2080                 fi
2081         fi
2082         if [ "$old_OST_QUOTA_TYPE" ]; then
2083                 if [[ $PERM_CMD = *"set_param -P"* ]]; then
2084                         do_facet mgs $PERM_CMD \
2085                                 osd-*.$FSNAME-OST*.quota_slave.enable = \
2086                                 $old_OST_QUOTA_TYPE
2087                 else
2088                         do_facet mgs $LCTL conf_param \
2089                                 $FSNAME.quota.ost=$old_OST_QUOTA_TYPE
2090                 fi
2091         fi
2092 }
2093
2094 # Handle the case when there is a space in the lfs df
2095 # "filesystem summary" line the same as when there is no space.
2096 # This will allow fixing the "lfs df" summary line in the future.
2097 lfs_df() {
2098         $LFS df $* | sed -e 's/filesystem /filesystem_/'
2099 }
2100
2101 # Get free inodes on the MDT specified by mdt index, free indoes on
2102 # the whole filesystem will be returned when index == -1.
2103 mdt_free_inodes() {
2104         local index=$1
2105         local free_inodes
2106         local mdt_uuid
2107
2108         if [ $index -eq -1 ]; then
2109                 mdt_uuid="summary"
2110         else
2111                 mdt_uuid=$(mdtuuid_from_index $index)
2112         fi
2113
2114         free_inodes=$(lfs_df -i $MOUNT | grep $mdt_uuid | awk '{print $4}')
2115         echo $free_inodes
2116 }
2117
2118 #
2119 # Get the OST device status from 'lfs df' with a given OST index.
2120 #
2121 ost_dev_status() {
2122         local ost_idx=$1
2123         local mnt_pnt=${2:-$MOUNT}
2124         local ost_uuid
2125
2126         ost_uuid=$(ostuuid_from_index $ost_idx $mnt_pnt)
2127         lfs_df $mnt_pnt | awk '/'$ost_uuid'/ { print $7 }'
2128 }
2129
2130 setup_quota(){
2131         local mntpt=$1
2132
2133         # save old quota type & set new quota type
2134         local mdt_qtype=$(mdt_quota_type)
2135         local ost_qtype=$(ost_quota_type)
2136
2137         echo "[HOST:$HOSTNAME] [old_mdt_qtype:$mdt_qtype]" \
2138                 "[old_ost_qtype:$ost_qtype] [new_qtype:$QUOTA_TYPE]"
2139
2140         export old_MDT_QUOTA_TYPE=$mdt_qtype
2141         export old_OST_QUOTA_TYPE=$ost_qtype
2142
2143         if [[ $PERM_CMD = *"set_param -P"* ]]; then
2144                 do_facet mgs $PERM_CMD \
2145                         osd-*.$FSNAME-MDT*.quota_slave.enable=$QUOTA_TYPE
2146                 do_facet mgs $PERM_CMD \
2147                         osd-*.$FSNAME-OST*.quota_slave.enable=$QUOTA_TYPE
2148         else
2149                 do_facet mgs $PERM_CMD $FSNAME.quota.mdt=$QUOTA_TYPE ||
2150                         error "set mdt quota type failed"
2151                 do_facet mgs $PERM_CMD $FSNAME.quota.ost=$QUOTA_TYPE ||
2152                         error "set ost quota type failed"
2153         fi
2154
2155         local quota_usrs=$QUOTA_USERS
2156
2157         # get_filesystem_size
2158         local disksz=$(lfs_df $mntpt | grep "summary" | awk '{print $2}')
2159         local blk_soft=$((disksz + 1024))
2160         local blk_hard=$((blk_soft + blk_soft / 20)) # Go 5% over
2161
2162         local inodes=$(lfs_df -i $mntpt | grep "summary" | awk '{print $2}')
2163         local i_soft=$inodes
2164         local i_hard=$((i_soft + i_soft / 20))
2165
2166         echo "Total disk size: $disksz  block-softlimit: $blk_soft" \
2167                 "block-hardlimit: $blk_hard inode-softlimit: $i_soft" \
2168                 "inode-hardlimit: $i_hard"
2169
2170         local cmd
2171         for usr in $quota_usrs; do
2172                 echo "Setting up quota on $HOSTNAME:$mntpt for $usr..."
2173                 for type in u g; do
2174                         cmd="$LFS setquota -$type $usr -b $blk_soft"
2175                         cmd="$cmd -B $blk_hard -i $i_soft -I $i_hard $mntpt"
2176                         echo "+ $cmd"
2177                         eval $cmd || error "$cmd FAILED!"
2178                 done
2179                 # display the quota status
2180                 echo "Quota settings for $usr : "
2181                 $LFS quota -v -u $usr $mntpt || true
2182         done
2183 }
2184
2185 zconf_mount() {
2186         local client=$1
2187         local mnt=$2
2188         local opts=${3:-$MOUNT_OPTS}
2189         opts=${opts:+-o $opts}
2190         local flags=${4:-$MOUNT_FLAGS}
2191
2192         local device=$MGSNID:/$FSNAME$FILESET
2193         if [ -z "$mnt" -o -z "$FSNAME" ]; then
2194                 echo "Bad mount command: opt=$flags $opts dev=$device " \
2195                      "mnt=$mnt"
2196                 exit 1
2197         fi
2198
2199         echo "Starting client: $client: $flags $opts $device $mnt"
2200         do_node $client mkdir -p $mnt
2201         if [ -n "$FILESET" -a -z "$SKIP_FILESET" ];then
2202                 do_node $client $MOUNT_CMD $flags $opts $MGSNID:/$FSNAME \
2203                         $mnt || return 1
2204                 #disable FILESET if not supported
2205                 do_nodes $client lctl get_param -n \
2206                         mdc.$FSNAME-MDT0000*.import | grep -q subtree ||
2207                                 device=$MGSNID:/$FSNAME
2208                 do_node $client mkdir -p $mnt/$FILESET
2209                 do_node $client "! grep -q $mnt' ' /proc/mounts ||
2210                         umount $mnt"
2211         fi
2212         if $GSS_SK && ($SK_UNIQUE_NM || $SK_S2S); then
2213                 # Mount using nodemap key
2214                 local mountkey=$SK_PATH/$FSNAME-nmclient.key
2215                 if $SK_UNIQUE_NM; then
2216                         mountkey=$SK_PATH/nodemap/c0.key
2217                 fi
2218                 local prunedopts=$(echo $opts |
2219                                 sed -e "s#skpath=[^,^ ]*#skpath=$mountkey#g")
2220                 do_node $client $MOUNT_CMD $flags $prunedopts $device $mnt ||
2221                                 return 1
2222         else
2223                 do_node $client $MOUNT_CMD $flags $opts $device $mnt ||
2224                                 return 1
2225         fi
2226
2227         set_default_debug_nodes $client
2228
2229         return 0
2230 }
2231
2232 zconf_umount() {
2233     local client=$1
2234     local mnt=$2
2235     local force
2236     local busy
2237     local need_kill
2238
2239     [ "$3" ] && force=-f
2240     local running=$(do_node $client "grep -c $mnt' ' /proc/mounts") || true
2241     if [ $running -ne 0 ]; then
2242         echo "Stopping client $client $mnt (opts:$force)"
2243         do_node $client lsof -t $mnt || need_kill=no
2244         if [ "x$force" != "x" -a "x$need_kill" != "xno" ]; then
2245             pids=$(do_node $client lsof -t $mnt | sort -u);
2246             if [ -n $pids ]; then
2247                 do_node $client kill -9 $pids || true
2248             fi
2249         fi
2250
2251         busy=$(do_node $client "umount $force $mnt 2>&1" | grep -c "busy") || true
2252         if [ $busy -ne 0 ] ; then
2253             echo "$mnt is still busy, wait one second" && sleep 1
2254             do_node $client umount $force $mnt
2255         fi
2256     fi
2257 }
2258
2259 # Mount the file system on the MGS
2260 mount_mgs_client() {
2261         do_facet mgs "mkdir -p $MOUNT"
2262         zconf_mount $mgs_HOST $MOUNT $MOUNT_OPTS ||
2263                 error "unable to mount $MOUNT on MGS"
2264 }
2265
2266 # Unmount the file system on the MGS
2267 umount_mgs_client() {
2268         zconf_umount $mgs_HOST $MOUNT
2269         do_facet mgs "rm -rf $MOUNT"
2270 }
2271
2272 # nodes is comma list
2273 sanity_mount_check_nodes () {
2274     local nodes=$1
2275     shift
2276     local mnts="$@"
2277     local mnt
2278
2279     # FIXME: assume that all cluster nodes run the same os
2280     [ "$(uname)" = Linux ] || return 0
2281
2282     local rc=0
2283     for mnt in $mnts ; do
2284         do_nodes $nodes "running=\\\$(grep -c $mnt' ' /proc/mounts);
2285 mpts=\\\$(mount | grep -c $mnt' ');
2286 if [ \\\$running -ne \\\$mpts ]; then
2287     echo \\\$(hostname) env are INSANE!;
2288     exit 1;
2289 fi"
2290     [ $? -eq 0 ] || rc=1
2291     done
2292     return $rc
2293 }
2294
2295 sanity_mount_check_servers () {
2296     [ -n "$CLIENTONLY" ] &&
2297         { echo "CLIENTONLY mode, skip mount_check_servers"; return 0; } || true
2298     echo Checking servers environments
2299
2300     # FIXME: modify get_facets to display all facets wo params
2301     local facets="$(get_facets OST),$(get_facets MDS),mgs"
2302     local node
2303     local mntpt
2304     local facet
2305     for facet in ${facets//,/ }; do
2306         node=$(facet_host ${facet})
2307         mntpt=$(facet_mntpt $facet)
2308         sanity_mount_check_nodes $node $mntpt ||
2309             { error "server $node environments are insane!"; return 1; }
2310     done
2311 }
2312
2313 sanity_mount_check_clients () {
2314     local clients=${1:-$CLIENTS}
2315     local mntpt=${2:-$MOUNT}
2316     local mntpt2=${3:-$MOUNT2}
2317
2318     [ -z $clients ] && clients=$(hostname)
2319     echo Checking clients $clients environments
2320
2321     sanity_mount_check_nodes $clients $mntpt $mntpt2 ||
2322        error "clients environments are insane!"
2323 }
2324
2325 sanity_mount_check () {
2326     sanity_mount_check_servers || return 1
2327     sanity_mount_check_clients || return 2
2328 }
2329
2330 # mount clients if not mouted
2331 zconf_mount_clients() {
2332         local clients=$1
2333         local mnt=$2
2334         local opts=${3:-$MOUNT_OPTS}
2335         opts=${opts:+-o $opts}
2336         local flags=${4:-$MOUNT_FLAGS}
2337         local device=$MGSNID:/$FSNAME$FILESET
2338         if [ -z "$mnt" -o -z "$FSNAME" ]; then
2339                 echo "Bad conf mount command: opt=$flags $opts dev=$device " \
2340                      "mnt=$mnt"
2341                 exit 1
2342         fi
2343
2344         echo "Starting client $clients: $flags $opts $device $mnt"
2345         if [ -n "$FILESET" -a ! -n "$SKIP_FILESET" ]; then
2346                 if $GSS_SK && ($SK_UNIQUE_NM || $SK_S2S); then
2347                         # Mount with own nodemap key
2348                         local i=0
2349                         # Mount all server nodes first with per-NM keys
2350                         for nmclient in ${clients//,/ }; do
2351 #                               do_nodes $(comma_list $(all_server_nodes)) "lgss_sk -t server -l $SK_PATH/nodemap/c$i.key -n c$i"
2352                                 do_nodes $(comma_list $(all_server_nodes)) "lgss_sk -t server -l $SK_PATH/nodemap/c$i.key"
2353                                 i=$((i + 1))
2354                         done
2355                         # set perms for per-nodemap keys else permission denied
2356                         do_nodes $(comma_list $(all_nodes)) \
2357                                 "keyctl show | grep lustre | cut -c1-11 |
2358                                 sed -e 's/ //g;' |
2359                                 xargs -IX keyctl setperm X 0x3f3f3f3f"
2360                         local mountkey=$SK_PATH/$FSNAME-nmclient.key
2361                         i=0
2362                         for nmclient in ${clients//,/ }; do
2363                                 if $SK_UNIQUE_NM; then
2364                                         mountkey=$SK_PATH/nodemap/c$i.key
2365                                 fi
2366                                 do_node $nmclient "! grep -q $mnt' ' \
2367                                         /proc/mounts || umount $mnt"
2368                                 local prunedopts=$(add_sk_mntflag $prunedopts);
2369                                 prunedopts=$(echo $prunedopts | sed -e \
2370                                         "s#skpath=[^ ^,]*#skpath=$mountkey#g")
2371                                 set -x
2372                                 do_nodes $(comma_list $(all_server_nodes)) \
2373                                         "keyctl show"
2374                                 set +x
2375                                 do_node $nmclient $MOUNT_CMD $flags \
2376                                         $prunedopts $MGSNID:/$FSNAME $mnt ||
2377                                         return 1
2378                                 i=$((i + 1))
2379                         done
2380                 else
2381                         do_nodes $clients "! grep -q $mnt' ' /proc/mounts ||
2382                                         umount $mnt"
2383                         do_nodes $clients $MOUNT_CMD $flags $opts \
2384                                         $MGSNID:/$FSNAME $mnt || return 1
2385                 fi
2386                 #disable FILESET if not supported
2387                 do_nodes $clients lctl get_param -n \
2388                         mdc.$FSNAME-MDT0000*.import | grep -q subtree ||
2389                                 device=$MGSNID:/$FSNAME
2390                 do_nodes $clients mkdir -p $mnt/$FILESET
2391                 do_nodes $clients "! grep -q $mnt' ' /proc/mounts ||
2392                         umount $mnt"
2393         fi
2394
2395         if $GSS_SK && ($SK_UNIQUE_NM || $SK_S2S); then
2396                 # Mount with nodemap key
2397                 local i=0
2398                 local mountkey=$SK_PATH/$FSNAME-nmclient.key
2399                 for nmclient in ${clients//,/ }; do
2400                         if $SK_UNIQUE_NM; then
2401                                 mountkey=$SK_PATH/nodemap/c$i.key
2402                         fi
2403                         local prunedopts=$(echo $opts | sed -e \
2404                                 "s#skpath=[^ ^,]*#skpath=$mountkey#g");
2405                         do_node $nmclient "! grep -q $mnt' ' /proc/mounts ||
2406                                 umount $mnt"
2407                         do_node $nmclient "
2408                 running=\\\$(mount | grep -c $mnt' ');
2409                 rc=0;
2410                 if [ \\\$running -eq 0 ] ; then
2411                         mkdir -p $mnt;
2412                         $MOUNT_CMD $flags $prunedopts $device $mnt;
2413                         rc=\\\$?;
2414                 else
2415                         lustre_mnt_count=\\\$(mount | grep $mnt' ' | \
2416                                 grep 'type lustre' | wc -l);
2417                         if [ \\\$running -ne \\\$lustre_mnt_count ] ; then
2418                                 echo zconf_mount_clients FAILED: \
2419                                         mount count \\\$running, not matching \
2420                                         with mount count of 'type lustre' \
2421                                         \\\$lustre_mnt_count;
2422                                 rc=1;
2423                         fi;
2424                 fi;
2425         exit \\\$rc" || return ${PIPESTATUS[0]}
2426
2427                         i=$((i + 1))
2428                 done
2429         else
2430
2431                 local tmpopts=$opts
2432                 if $SHARED_KEY; then
2433                         tmpopts=$(add_sk_mntflag $opts)
2434                 fi
2435                 do_nodes $clients "
2436 running=\\\$(mount | grep -c $mnt' ');
2437 rc=0;
2438 if [ \\\$running -eq 0 ] ; then
2439         mkdir -p $mnt;
2440         $MOUNT_CMD $flags $tmpopts $device $mnt;
2441         rc=\\\$?;
2442 fi;
2443 exit \\\$rc" || return ${PIPESTATUS[0]}
2444         fi
2445
2446         echo "Started clients $clients: "
2447         do_nodes $clients "mount | grep $mnt' '"
2448
2449         set_default_debug_nodes $clients
2450
2451         return 0
2452 }
2453
2454 zconf_umount_clients() {
2455     local clients=$1
2456     local mnt=$2
2457     local force
2458
2459     [ "$3" ] && force=-f
2460
2461     echo "Stopping clients: $clients $mnt (opts:$force)"
2462     do_nodes $clients "running=\\\$(grep -c $mnt' ' /proc/mounts);
2463 if [ \\\$running -ne 0 ] ; then
2464 echo Stopping client \\\$(hostname) $mnt opts:$force;
2465 lsof $mnt || need_kill=no;
2466 if [ "x$force" != "x" -a "x\\\$need_kill" != "xno" ]; then
2467     pids=\\\$(lsof -t $mnt | sort -u);
2468     if [ -n \\\"\\\$pids\\\" ]; then
2469              kill -9 \\\$pids;
2470     fi
2471 fi;
2472 while umount $force $mnt 2>&1 | grep -q "busy"; do
2473     echo "$mnt is still busy, wait one second" && sleep 1;
2474 done;
2475 fi"
2476 }
2477
2478 shutdown_node () {
2479     local node=$1
2480     echo + $POWER_DOWN $node
2481     $POWER_DOWN $node
2482 }
2483
2484 shutdown_node_hard () {
2485     local host=$1
2486     local attempts=$SHUTDOWN_ATTEMPTS
2487
2488     for i in $(seq $attempts) ; do
2489         shutdown_node $host
2490         sleep 1
2491         wait_for_function --quiet "! ping -w 3 -c 1 $host" 5 1 && return 0
2492         echo "waiting for $host to fail attempts=$attempts"
2493         [ $i -lt $attempts ] || \
2494             { echo "$host still pingable after power down! attempts=$attempts" && return 1; }
2495     done
2496 }
2497
2498 shutdown_client() {
2499     local client=$1
2500     local mnt=${2:-$MOUNT}
2501     local attempts=3
2502
2503     if [ "$FAILURE_MODE" = HARD ]; then
2504         shutdown_node_hard $client
2505     else
2506        zconf_umount_clients $client $mnt -f
2507     fi
2508 }
2509
2510 facets_on_host () {
2511     local host=$1
2512     local facets="$(get_facets OST),$(get_facets MDS)"
2513     local affected
2514
2515     combined_mgs_mds || facets="$facets,mgs"
2516
2517     for facet in ${facets//,/ }; do
2518         if [ $(facet_active_host $facet) == $host ]; then
2519            affected="$affected $facet"
2520         fi
2521     done
2522
2523     echo $(comma_list $affected)
2524 }
2525
2526 facet_up() {
2527         local facet=$1
2528         local host=${2:-$(facet_host $facet)}
2529
2530         local label=$(convert_facet2label $facet)
2531         do_node $host $LCTL dl | awk '{ print $4 }' | grep -q "^$label\$"
2532 }
2533
2534 facets_up_on_host () {
2535     local host=$1
2536     local facets=$(facets_on_host $host)
2537     local affected_up
2538
2539     for facet in ${facets//,/ }; do
2540         if $(facet_up $facet $host); then
2541             affected_up="$affected_up $facet"
2542         fi
2543     done
2544
2545     echo $(comma_list $affected_up)
2546 }
2547
2548 shutdown_facet() {
2549         local facet=$1
2550         local affected_facet
2551         local affected_facets
2552
2553         if [[ "$FAILURE_MODE" = HARD ]]; then
2554                 if [[ $(facet_fstype $facet) = ldiskfs ]] &&
2555                         dm_flakey_supported $facet; then
2556                         affected_facets=$(affected_facets $facet)
2557                         for affected_facet in ${affected_facets//,/ }; do
2558                                 unexport_dm_dev $affected_facet
2559                         done
2560                 fi
2561
2562                 shutdown_node_hard $(facet_active_host $facet)
2563         else
2564                 stop $facet
2565         fi
2566 }
2567
2568 reboot_node() {
2569     local node=$1
2570     echo + $POWER_UP $node
2571     $POWER_UP $node
2572 }
2573
2574 remount_facet() {
2575     local facet=$1
2576
2577     stop $facet
2578     mount_facet $facet
2579 }
2580
2581 reboot_facet() {
2582         local facet=$1
2583         if [ "$FAILURE_MODE" = HARD ]; then
2584                 reboot_node $(facet_active_host $facet)
2585         else
2586                 sleep 10
2587         fi
2588 }
2589
2590 boot_node() {
2591     local node=$1
2592     if [ "$FAILURE_MODE" = HARD ]; then
2593        reboot_node $node
2594        wait_for_host $node
2595     fi
2596 }
2597
2598 facets_hosts () {
2599     local facets=$1
2600     local hosts
2601
2602     for facet in ${facets//,/ }; do
2603         hosts=$(expand_list $hosts $(facet_host $facet) )
2604     done
2605
2606     echo $hosts
2607 }
2608
2609 _check_progs_installed () {
2610     local progs=$@
2611     local rc=0
2612
2613     for prog in $progs; do
2614         if ! [ "$(which $prog)"  -o  "${!prog}" ]; then
2615            echo $prog missing on $(hostname)
2616            rc=1
2617         fi
2618     done
2619     return $rc
2620 }
2621
2622 check_progs_installed () {
2623         local nodes=$1
2624         shift
2625
2626         do_rpc_nodes "$nodes" _check_progs_installed $@
2627 }
2628
2629 # recovery-scale functions
2630 node_var_name() {
2631     echo __$(echo $1 | tr '-' '_' | tr '.' '_')
2632 }
2633
2634 start_client_load() {
2635         local client=$1
2636         local load=$2
2637         local var=$(node_var_name $client)_load
2638         eval export ${var}=$load
2639
2640         do_node $client "PATH=$PATH MOUNT=$MOUNT ERRORS_OK=$ERRORS_OK \
2641                         BREAK_ON_ERROR=$BREAK_ON_ERROR \
2642                         END_RUN_FILE=$END_RUN_FILE \
2643                         LOAD_PID_FILE=$LOAD_PID_FILE \
2644                         TESTLOG_PREFIX=$TESTLOG_PREFIX \
2645                         TESTNAME=$TESTNAME \
2646                         DBENCH_LIB=$DBENCH_LIB \
2647                         DBENCH_SRC=$DBENCH_SRC \
2648                         CLIENT_COUNT=$((CLIENTCOUNT - 1)) \
2649                         LFS=$LFS \
2650                         LCTL=$LCTL \
2651                         FSNAME=$FSNAME \
2652                         run_${load}.sh" &
2653         local ppid=$!
2654         log "Started client load: ${load} on $client"
2655
2656         # get the children process IDs
2657         local pids=$(ps --ppid $ppid -o pid= | xargs)
2658         CLIENT_LOAD_PIDS="$CLIENT_LOAD_PIDS $ppid $pids"
2659         return 0
2660 }
2661
2662 start_client_loads () {
2663     local -a clients=(${1//,/ })
2664     local numloads=${#CLIENT_LOADS[@]}
2665     local testnum
2666
2667     for ((nodenum=0; nodenum < ${#clients[@]}; nodenum++ )); do
2668         testnum=$((nodenum % numloads))
2669         start_client_load ${clients[nodenum]} ${CLIENT_LOADS[testnum]}
2670     done
2671     # bug 22169: wait the background threads to start
2672     sleep 2
2673 }
2674
2675 # only for remote client
2676 check_client_load () {
2677         local client=$1
2678         local var=$(node_var_name $client)_load
2679         local testload=run_${!var}.sh
2680
2681         ps auxww | grep -v grep | grep $client | grep -q $testload || return 1
2682
2683         # bug 18914: try to connect several times not only when
2684         # check ps, but  while check_node_health also
2685
2686         local tries=3
2687         local RC=254
2688         while [ $RC = 254 -a $tries -gt 0 ]; do
2689                 let tries=$tries-1
2690                 # assume success
2691                 RC=0
2692                 if ! check_node_health $client; then
2693                         RC=${PIPESTATUS[0]}
2694                         if [ $RC -eq 254 ]; then
2695                                 # FIXME: not sure how long we shuold sleep here
2696                                 sleep 10
2697                                 continue
2698                         fi
2699                         echo "check node health failed: RC=$RC "
2700                         return $RC
2701                 fi
2702         done
2703         # We can continue try to connect if RC=254
2704         # Just print the warning about this
2705         if [ $RC = 254 ]; then
2706                 echo "got a return status of $RC from do_node while checking " \
2707                 "node health on $client"
2708         fi
2709
2710         # see if the load is still on the client
2711         tries=3
2712         RC=254
2713         while [ $RC = 254 -a $tries -gt 0 ]; do
2714                 let tries=$tries-1
2715                 # assume success
2716                 RC=0
2717                 if ! do_node $client \
2718                         "ps auxwww | grep -v grep | grep -q $testload"; then
2719                         RC=${PIPESTATUS[0]}
2720                         sleep 30
2721                 fi
2722         done
2723         if [ $RC = 254 ]; then
2724                 echo "got a return status of $RC from do_node while checking " \
2725                 "(node health and 'ps') the client load on $client"
2726                 # see if we can diagnose a bit why this is
2727         fi
2728
2729         return $RC
2730 }
2731 check_client_loads () {
2732    local clients=${1//,/ }
2733    local client=
2734    local rc=0
2735
2736    for client in $clients; do
2737       check_client_load $client
2738       rc=${PIPESTATUS[0]}
2739       if [ "$rc" != 0 ]; then
2740         log "Client load failed on node $client, rc=$rc"
2741         return $rc
2742       fi
2743    done
2744 }
2745
2746 restart_client_loads () {
2747     local clients=${1//,/ }
2748     local expectedfail=${2:-""}
2749     local client=
2750     local rc=0
2751
2752     for client in $clients; do
2753         check_client_load $client
2754         rc=${PIPESTATUS[0]}
2755         if [ "$rc" != 0 -a "$expectedfail" ]; then
2756             local var=$(node_var_name $client)_load
2757             start_client_load $client ${!var}
2758             echo "Restarted client load ${!var}: on $client. Checking ..."
2759             check_client_load $client
2760             rc=${PIPESTATUS[0]}
2761             if [ "$rc" != 0 ]; then
2762                 log "Client load failed to restart on node $client, rc=$rc"
2763                 # failure one client load means test fail
2764                 # we do not need to check other
2765                 return $rc
2766             fi
2767         else
2768             return $rc
2769         fi
2770     done
2771 }
2772
2773 # Start vmstat and save its process ID in a file.
2774 start_vmstat() {
2775     local nodes=$1
2776     local pid_file=$2
2777
2778     [ -z "$nodes" -o -z "$pid_file" ] && return 0
2779
2780     do_nodes $nodes \
2781         "vmstat 1 > $TESTLOG_PREFIX.$TESTNAME.vmstat.\\\$(hostname -s).log \
2782         2>/dev/null </dev/null & echo \\\$! > $pid_file"
2783 }
2784
2785 # Display the nodes on which client loads failed.
2786 print_end_run_file() {
2787     local file=$1
2788     local node
2789
2790     [ -s $file ] || return 0
2791
2792     echo "Found the END_RUN_FILE file: $file"
2793     cat $file
2794
2795     # A client load will stop if it finds the END_RUN_FILE file.
2796     # That does not mean the client load actually failed though.
2797     # The first node in END_RUN_FILE is the one we are interested in.
2798     read node < $file
2799
2800     if [ -n "$node" ]; then
2801         local var=$(node_var_name $node)_load
2802
2803         local prefix=$TESTLOG_PREFIX
2804         [ -n "$TESTNAME" ] && prefix=$prefix.$TESTNAME
2805         local stdout_log=$prefix.run_${!var}_stdout.$node.log
2806         local debug_log=$(echo $stdout_log | sed 's/\(.*\)stdout/\1debug/')
2807
2808         echo "Client load ${!var} failed on node $node:"
2809         echo "$stdout_log"
2810         echo "$debug_log"
2811     fi
2812 }
2813
2814 # Stop the process which had its PID saved in a file.
2815 stop_process() {
2816     local nodes=$1
2817     local pid_file=$2
2818
2819     [ -z "$nodes" -o -z "$pid_file" ] && return 0
2820
2821     do_nodes $nodes "test -f $pid_file &&
2822         { kill -s TERM \\\$(cat $pid_file); rm -f $pid_file; }" || true
2823 }
2824
2825 # Stop all client loads.
2826 stop_client_loads() {
2827     local nodes=${1:-$CLIENTS}
2828     local pid_file=$2
2829
2830     # stop the client loads
2831     stop_process $nodes $pid_file
2832
2833     # clean up the processes that started them
2834     [ -n "$CLIENT_LOAD_PIDS" ] && kill -9 $CLIENT_LOAD_PIDS 2>/dev/null || true
2835 }
2836 # End recovery-scale functions
2837
2838 # verify that lustre actually cleaned up properly
2839 cleanup_check() {
2840         VAR=$(lctl get_param -n catastrophe 2>&1)
2841         if [ $? = 0 ] ; then
2842                 if [ $VAR != 0 ]; then
2843                         error "LBUG/LASSERT detected"
2844                 fi
2845         fi
2846         BUSY=$(dmesg | grep -i destruct || true)
2847         if [ -n "$BUSY" ]; then
2848                 echo "$BUSY" 1>&2
2849                 [ -e $TMP/debug ] && mv $TMP/debug $TMP/debug-busy.$(date +%s)
2850                 exit 205
2851         fi
2852
2853         check_mem_leak || exit 204
2854
2855         [[ $($LCTL dl 2>/dev/null | wc -l) -gt 0 ]] && $LCTL dl &&
2856                 echo "$TESTSUITE: lustre didn't clean up..." 1>&2 &&
2857                 return 202 || true
2858
2859         if module_loaded lnet || module_loaded libcfs; then
2860                 echo "$TESTSUITE: modules still loaded..." 1>&2
2861                 /sbin/lsmod 1>&2
2862                 return 203
2863         fi
2864         return 0
2865 }
2866
2867 wait_update () {
2868         local verbose=false
2869         if [[ "$1" == "--verbose" ]]; then
2870                 shift
2871                 verbose=true
2872         fi
2873
2874         local node=$1
2875         local TEST=$2
2876         local FINAL=$3
2877         local MAX=${4:-90}
2878         local RESULT
2879         local PREV_RESULT
2880         local WAIT=0
2881         local sleep=1
2882         local print=10
2883
2884         PREV_RESULT=$(do_node $node "$TEST")
2885         while [ true ]; do
2886                 RESULT=$(do_node $node "$TEST")
2887                 if [[ "$RESULT" == "$FINAL" ]]; then
2888                         [[ -z "$RESULT" || $WAIT -le $sleep ]] ||
2889                                 echo "Updated after ${WAIT}s: wanted '$FINAL'"\
2890                                      "got '$RESULT'"
2891                         return 0
2892                 fi
2893                 if [[ $verbose && "$RESULT" != "$PREV_RESULT" ]]; then
2894                         echo "Changed after ${WAIT}s: from '$PREV_RESULT'"\
2895                              "to '$RESULT'"
2896                         PREV_RESULT=$RESULT
2897                 fi
2898                 [[ $WAIT -ge $MAX ]] && break
2899                 [[ $((WAIT % print)) -eq 0 ]] &&
2900                         echo "Waiting $((MAX - WAIT)) secs for update"
2901                 WAIT=$((WAIT + sleep))
2902                 sleep $sleep
2903         done
2904         echo "Update not seen after ${MAX}s: wanted '$FINAL' got '$RESULT'"
2905         return 3
2906 }
2907
2908 wait_update_facet() {
2909         local verbose=
2910         [ "$1" = "--verbose" ] && verbose="$1" && shift
2911
2912         local facet=$1
2913         shift
2914         wait_update $verbose $(facet_active_host $facet) "$@"
2915 }
2916
2917 sync_all_data() {
2918         do_nodes $(comma_list $(mdts_nodes)) \
2919             "lctl set_param -n os[cd]*.*MDT*.force_sync=1"
2920         do_nodes $(comma_list $(osts_nodes)) \
2921             "lctl set_param -n osd*.*OS*.force_sync=1" 2>&1 |
2922                 grep -v 'Found no match'
2923 }
2924
2925 wait_zfs_commit() {
2926         local zfs_wait=${2:-5}
2927
2928         # the occupied disk space will be released
2929         # only after TXGs are committed
2930         if [[ $(facet_fstype $1) == zfs ]]; then
2931                 echo "sleep $zfs_wait for ZFS $(facet_fstype $1)"
2932                 sleep $zfs_wait
2933         fi
2934 }
2935
2936 wait_delete_completed_mds() {
2937         local max_wait=${1:-20}
2938         local mds2sync=""
2939         local stime=$(date +%s)
2940         local etime
2941         local node
2942         local changes
2943
2944         # find MDS with pending deletions
2945         for node in $(mdts_nodes); do
2946                 changes=$(do_node $node "$LCTL get_param -n osc.*MDT*.sync_*" \
2947                         2>/dev/null | calc_sum)
2948                 if [[ $changes -eq 0 ]]; then
2949                         continue
2950                 fi
2951                 mds2sync="$mds2sync $node"
2952         done
2953         if [ -z "$mds2sync" ]; then
2954                 wait_zfs_commit $SINGLEMDS
2955                 return 0
2956         fi
2957         mds2sync=$(comma_list $mds2sync)
2958
2959         # sync MDS transactions
2960         do_nodes $mds2sync "$LCTL set_param -n os[cd]*.*MD*.force_sync 1"
2961
2962         # wait till all changes are sent and commmitted by OSTs
2963         # for ldiskfs space is released upon execution, but DMU
2964         # do this upon commit
2965
2966         local WAIT=0
2967         while [[ $WAIT -ne $max_wait ]]; do
2968                 changes=$(do_nodes $mds2sync \
2969                         "$LCTL get_param -n osc.*MDT*.sync_*" | calc_sum)
2970                 #echo "$node: $changes changes on all"
2971                 if [[ $changes -eq 0 ]]; then
2972                         wait_zfs_commit $SINGLEMDS
2973
2974                         # the occupied disk space will be released
2975                         # only after TXGs are committed
2976                         wait_zfs_commit ost1
2977                         return 0
2978                 fi
2979                 sleep 1
2980                 WAIT=$((WAIT + 1))
2981         done
2982
2983         etime=$(date +%s)
2984         echo "Delete is not completed in $((etime - stime)) seconds"
2985         do_nodes $mds2sync "$LCTL get_param osc.*MDT*.sync_*"
2986         return 1
2987 }
2988
2989 wait_for_host() {
2990     local hostlist=$1
2991
2992     # we can use "for" here because we are waiting the slowest
2993     for host in ${hostlist//,/ }; do
2994         check_network "$host" 900
2995     done
2996     while ! do_nodes $hostlist hostname  > /dev/null; do sleep 5; done
2997 }
2998
2999 wait_for_facet() {
3000     local facetlist=$1
3001     local hostlist
3002
3003     for facet in ${facetlist//,/ }; do
3004         hostlist=$(expand_list $hostlist $(facet_active_host $facet))
3005     done
3006     wait_for_host $hostlist
3007 }
3008
3009 _wait_recovery_complete () {
3010     local param=$1
3011
3012     # Use default policy if $2 is not passed by caller.
3013     local MAX=${2:-$(max_recovery_time)}
3014
3015     local WAIT=0
3016     local STATUS=
3017
3018     while [ $WAIT -lt $MAX ]; do
3019         STATUS=$(lctl get_param -n $param | grep status)
3020         echo $param $STATUS
3021         [[ $STATUS = "status: COMPLETE" || $STATUS = "status: INACTIVE" ]] && return 0
3022         sleep 5
3023         WAIT=$((WAIT + 5))
3024         echo "Waiting $((MAX - WAIT)) secs for $param recovery done. $STATUS"
3025     done
3026     echo "$param recovery not done in $MAX sec. $STATUS"
3027     return 1
3028 }
3029
3030 wait_recovery_complete () {
3031     local facet=$1
3032
3033     # with an assumption that at_max is the same on all nodes
3034     local MAX=${2:-$(max_recovery_time)}
3035
3036     local facets=$facet
3037     if [ "$FAILURE_MODE" = HARD ]; then
3038         facets=$(facets_on_host $(facet_active_host $facet))
3039     fi
3040     echo affected facets: $facets
3041
3042         # we can use "for" here because we are waiting the slowest
3043         for facet in ${facets//,/ }; do
3044                 local var_svc=${facet}_svc
3045                 local param="*.${!var_svc}.recovery_status"
3046
3047                 local host=$(facet_active_host $facet)
3048                 do_rpc_nodes "$host" _wait_recovery_complete $param $MAX
3049         done
3050 }
3051
3052 wait_mds_ost_sync () {
3053         # just because recovery is done doesn't mean we've finished
3054         # orphan cleanup. Wait for llogs to get synchronized.
3055         echo "Waiting for orphan cleanup..."
3056         # MAX value includes time needed for MDS-OST reconnection
3057         local MAX=$(( TIMEOUT * 2 ))
3058         local WAIT_TIMEOUT=${1:-$MAX}
3059         local WAIT=0
3060         local new_wait=true
3061         local list=$(comma_list $(mdts_nodes))
3062         local cmd="$LCTL get_param -n osp.*osc*.old_sync_processed"
3063         if ! do_facet $SINGLEMDS \
3064                 "$LCTL list_param osp.*osc*.old_sync_processed 2> /dev/null"
3065         then
3066                 # old way, use mds_sync
3067                 new_wait=false
3068                 list=$(comma_list $(osts_nodes))
3069                 cmd="$LCTL get_param -n obdfilter.*.mds_sync"
3070         fi
3071
3072         echo "wait $WAIT_TIMEOUT secs maximumly for $list mds-ost sync done."
3073         while [ $WAIT -lt $WAIT_TIMEOUT ]; do
3074                 local -a sync=($(do_nodes $list "$cmd"))
3075                 local con=1
3076                 local i
3077                 for ((i=0; i<${#sync[@]}; i++)); do
3078                         if $new_wait; then
3079                                 [ ${sync[$i]} -eq 1 ] && continue
3080                         else
3081                                 [ ${sync[$i]} -eq 0 ] && continue
3082                         fi
3083                         # there is a not finished MDS-OST synchronization
3084                         con=0
3085                         break;
3086                 done
3087                 sleep 2 # increase waiting time and cover statfs cache
3088                 [ ${con} -eq 1 ] && return 0
3089                 echo "Waiting $WAIT secs for $list $i mds-ost sync done."
3090                 WAIT=$((WAIT + 2))
3091         done
3092
3093         # show which nodes are not finished.
3094         cmd=$(echo $cmd | sed 's/-n//')
3095         do_nodes $list "$cmd"
3096         echo "$facet recovery node $i not done in $WAIT_TIMEOUT sec. $STATUS"
3097         return 1
3098 }
3099
3100 # Wait OSTs to be active on both client and MDT side.
3101 wait_osts_up() {
3102         local cmd="$LCTL get_param -n lov.$FSNAME-clilov-*.target_obd |
3103                 awk 'BEGIN {c = 0} /ACTIVE/{c += 1} END {printf \\\"%d\\\", c}'"
3104         wait_update $HOSTNAME "eval $cmd" $OSTCOUNT ||
3105                 error "wait_update OSTs up on client failed"
3106
3107         cmd="$LCTL get_param osp.$FSNAME-OST*-MDT0000.prealloc_last_id |
3108              awk '/=[1-9][0-9]/ { c += 1 } END { printf \\\"%d\\\", c }'"
3109         wait_update_facet $SINGLEMDS "eval $cmd" $OSTCOUNT ||
3110                 error "wait_update OSTs up on MDT0000 failed"
3111 }
3112
3113 wait_destroy_complete () {
3114         echo "Waiting for local destroys to complete"
3115         # MAX value shouldn't be big as this mean server responsiveness
3116         # never increase this just to make test pass but investigate
3117         # why it takes so long time
3118         local MAX=5
3119         local WAIT=0
3120         while [ $WAIT -lt $MAX ]; do
3121                 local -a RPCs=($($LCTL get_param -n osc.*.destroys_in_flight))
3122                 local con=1
3123                 local i
3124
3125                 for ((i=0; i<${#RPCs[@]}; i++)); do
3126                         [ ${RPCs[$i]} -eq 0 ] && continue
3127                         # there are still some destroy RPCs in flight
3128                         con=0
3129                         break;
3130                 done
3131                 sleep 1
3132                 [ ${con} -eq 1 ] && return 0 # done waiting
3133                 echo "Waiting ${WAIT}s for local destroys to complete"
3134                 WAIT=$((WAIT + 1))
3135         done
3136         echo "Local destroys weren't done in $MAX sec."
3137         return 1
3138 }
3139
3140 wait_delete_completed() {
3141         wait_delete_completed_mds $1 || return $?
3142         wait_destroy_complete || return $?
3143 }
3144
3145 wait_exit_ST () {
3146     local facet=$1
3147
3148     local WAIT=0
3149     local INTERVAL=1
3150     local running
3151     # conf-sanity 31 takes a long time cleanup
3152     while [ $WAIT -lt 300 ]; do
3153         running=$(do_facet ${facet} "lsmod | grep lnet > /dev/null &&
3154 lctl dl | grep ' ST ' || true")
3155         [ -z "${running}" ] && return 0
3156         echo "waited $WAIT for${running}"
3157         [ $INTERVAL -lt 64 ] && INTERVAL=$((INTERVAL + INTERVAL))
3158         sleep $INTERVAL
3159         WAIT=$((WAIT + INTERVAL))
3160     done
3161     echo "service didn't stop after $WAIT seconds.  Still running:"
3162     echo ${running}
3163     return 1
3164 }
3165
3166 wait_remote_prog () {
3167    local prog=$1
3168    local WAIT=0
3169    local INTERVAL=5
3170    local rc=0
3171
3172    [ "$PDSH" = "no_dsh" ] && return 0
3173
3174    while [ $WAIT -lt $2 ]; do
3175         running=$(ps uax | grep "$PDSH.*$prog.*$MOUNT" | grep -v grep) || true
3176         [ -z "${running}" ] && return 0 || true
3177         echo "waited $WAIT for: "
3178         echo "$running"
3179         [ $INTERVAL -lt 60 ] && INTERVAL=$((INTERVAL + INTERVAL))
3180         sleep $INTERVAL
3181         WAIT=$((WAIT + INTERVAL))
3182     done
3183     local pids=$(ps  uax | grep "$PDSH.*$prog.*$MOUNT" | grep -v grep | awk '{print $2}')
3184     [ -z "$pids" ] && return 0
3185     echo "$PDSH processes still exists after $WAIT seconds.  Still running: $pids"
3186     # FIXME: not portable
3187     for pid in $pids; do
3188         cat /proc/${pid}/status || true
3189         cat /proc/${pid}/wchan || true
3190         echo "Killing $pid"
3191         kill -9 $pid || true
3192         sleep 1
3193         ps -P $pid && rc=1
3194     done
3195
3196     return $rc
3197 }
3198
3199 lfs_df_check() {
3200         local clients=${1:-$CLIENTS}
3201
3202         if [ -z "$clients" ]; then
3203                 $LFS df $MOUNT
3204         else
3205                 $PDSH $clients "$LFS df $MOUNT" > /dev/null
3206         fi
3207 }
3208
3209 clients_up() {
3210         # not every config has many clients
3211         sleep 1
3212         lfs_df_check
3213 }
3214
3215 client_up() {
3216         # usually checked on particular client or locally
3217         sleep 1
3218         lfs_df_check $1
3219 }
3220
3221 client_evicted() {
3222     ! client_up $1
3223 }
3224
3225 client_reconnect_try() {
3226         local f=$MOUNT/recon
3227
3228         uname -n >> $f
3229         if [ -z "$CLIENTS" ]; then
3230                 $LFS df $MOUNT; uname -n >> $f
3231         else
3232                 do_nodes $CLIENTS "$LFS df $MOUNT; uname -n >> $f" > /dev/null
3233         fi
3234         echo "Connected clients: $(cat $f)"
3235         ls -l $f > /dev/null
3236         rm $f
3237 }
3238
3239 client_reconnect() {
3240         # one client_reconnect_try call does not always do the job...
3241         while true ; do
3242                 client_reconnect_try && break
3243                 sleep 1
3244         done
3245 }
3246
3247 affected_facets () {
3248     local facet=$1
3249
3250     local host=$(facet_active_host $facet)
3251     local affected=$facet
3252
3253     if [ "$FAILURE_MODE" = HARD ]; then
3254         affected=$(facets_up_on_host $host)
3255     fi
3256     echo $affected
3257 }
3258
3259 facet_failover() {
3260         local E2FSCK_ON_MDT0=false
3261         if [ "$1" == "--fsck" ]; then
3262                 shift
3263                 [ $(facet_fstype $SINGLEMDS) == ldiskfs ] &&
3264                         E2FSCK_ON_MDT0=true
3265         fi
3266
3267         local facets=$1
3268         local sleep_time=$2
3269         local -a affecteds
3270         local facet
3271         local total=0
3272         local index=0
3273         local skip
3274
3275         #Because it will only get up facets, we need get affected
3276         #facets before shutdown
3277         #For HARD Failure mode, it needs make sure facets on the same
3278         #HOST will only be shutdown and reboot once
3279         for facet in ${facets//,/ }; do
3280                 local affected_facet
3281                 skip=0
3282                 #check whether facet has been included in other affected facets
3283                 for ((index=0; index<$total; index++)); do
3284                         [[ *,$facet,* == ,${affecteds[index]}, ]] && skip=1
3285                 done
3286
3287                 if [ $skip -eq 0 ]; then
3288                         affecteds[$total]=$(affected_facets $facet)
3289                         total=$((total+1))
3290                 fi