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