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