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