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