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