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