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