Whamcloud - gitweb
b=12599
[fs/lustre-release.git] / lustre / tests / test-framework.sh
1 #!/bin/bash
2 # vim:expandtab:shiftwidth=4:softtabstop=4:tabstop=4:
3
4 trap 'echo "test-framework exiting on error"' ERR
5 set -e
6 #set -x
7
8
9 export REFORMAT=${REFORMAT:-""}
10 export VERBOSE=false
11 export GMNALNID=${GMNALNID:-/usr/sbin/gmlndnid}
12 export CATASTROPHE=${CATASTROPHE:-/proc/sys/lnet/catastrophe}
13 #export PDSH="pdsh -S -Rssh -w"
14
15 # eg, assert_env LUSTRE MDSNODES OSTNODES CLIENTS
16 assert_env() {
17     local failed=""
18     for name in $@; do
19         if [ -z "${!name}" ]; then
20             echo "$0: $name must be set"
21             failed=1
22         fi
23     done
24     [ $failed ] && exit 1 || true
25 }
26
27 usage() {
28     echo "usage: $0 [-r] [-f cfgfile]"
29     echo "       -r: reformat"
30
31     exit
32 }
33
34 init_test_env() {
35     export LUSTRE=`absolute_path $LUSTRE`
36     export TESTSUITE=`basename $0 .sh`
37     export LTESTDIR=${LTESTDIR:-$LUSTRE/../ltest}
38
39     [ -d /r ] && export ROOT=${ROOT:-/r}
40     export TMP=${TMP:-$ROOT/tmp}
41     export TESTSUITELOG=${TMP}/${TESTSUITE}.log
42
43     export PATH=:$PATH:$LUSTRE/utils:$LUSTRE/utils/gss:$LUSTRE/tests
44     export LCTL=${LCTL:-"$LUSTRE/utils/lctl"}
45     export LFS=${LFS:-"$LUSTRE/utils/lfs"}
46     [ ! -f "$LCTL" ] && export LCTL=$(which lctl) 
47     export MKFS=${MKFS:-"$LUSTRE/utils/mkfs.lustre"}
48     [ ! -f "$MKFS" ] && export MKFS=$(which mkfs.lustre) 
49     export TUNEFS=${TUNEFS:-"$LUSTRE/utils/tunefs.lustre"}
50     [ ! -f "$TUNEFS" ] && export TUNEFS=$(which tunefs.lustre) 
51     export CHECKSTAT="${CHECKSTAT:-"checkstat -v"} "
52     export FSYTPE=${FSTYPE:-"ldiskfs"}
53     export NAME=${NAME:-local}
54     export LPROC=/proc/fs/lustre
55     export LGSSD=${LGSSD:-"$LUSTRE/utils/gss/lgssd"}
56     export LSVCGSSD=${LSVCGSSD:-"$LUSTRE/utils/gss/lsvcgssd"}
57     export KRB5DIR=${KRB5DIR:-"/usr/kerberos"}
58
59     if [ "$ACCEPTOR_PORT" ]; then
60         export PORT_OPT="--port $ACCEPTOR_PORT"
61     fi
62
63     case "x$SEC" in
64         xkrb5*)
65             echo "Using GSS/krb5 ptlrpc security flavor"
66             export USING_KRB5="y"
67             ;;
68     esac
69
70     # Paths on remote nodes, if different 
71     export RLUSTRE=${RLUSTRE:-$LUSTRE}
72     export RPWD=${RPWD:-$PWD}
73     export I_MOUNTED=${I_MOUNTED:-"no"}
74
75     # command line
76     
77     while getopts "rvf:" opt $*; do 
78         case $opt in
79             f) CONFIG=$OPTARG;;
80             r) REFORMAT=--reformat;;
81             v) VERBOSE=true;;
82             \?) usage;;
83         esac
84     done
85
86     shift $((OPTIND - 1))
87     ONLY=${ONLY:-$*}
88
89     [ "$TESTSUITELOG" ] && rm -f $TESTSUITELOG || true
90
91 }
92
93 load_module() {
94     EXT=".ko"
95     module=$1
96     shift
97     BASE=`basename $module $EXT`
98     lsmod | grep -q ${BASE} || \
99       if [ -f ${LUSTRE}/${module}${EXT} ]; then
100         insmod ${LUSTRE}/${module}${EXT} $@
101     else
102         # must be testing a "make install" or "rpm" installation
103         # note failed to load ptlrpc_gss is considered not fatal
104         if [ "$BASE" == "ptlrpc_gss" ]; then
105             modprobe $BASE $@ || echo "gss/krb5 is not supported"
106         else
107             modprobe $BASE $@
108         fi
109     fi
110 }
111
112 load_modules() {
113     if [ -n "$MODPROBE" ]; then
114         # use modprobe
115     return 0
116     fi
117     if [ "$HAVE_MODULES" = true ]; then
118     # we already loaded
119         return 0
120     fi
121     HAVE_MODULES=true
122
123     echo Loading modules from $LUSTRE
124     load_module ../lnet/libcfs/libcfs
125     [ -z "$LNETOPTS" ] && \
126         LNETOPTS=$(awk '/^options lnet/ { print $0}' /etc/modprobe.conf | sed 's/^options lnet //g')
127     echo "lnet options: '$LNETOPTS'"
128     # note that insmod will ignore anything in modprobe.conf
129     load_module ../lnet/lnet/lnet $LNETOPTS
130     LNETLND=${LNETLND:-"socklnd/ksocklnd"}
131     load_module ../lnet/klnds/$LNETLND
132     [ "$FSTYPE" = "ldiskfs" ] && load_module ../ldiskfs/ldiskfs/ldiskfs
133     load_module lvfs/lvfs
134     load_module obdclass/obdclass
135     load_module lvfs/fsfilt_$FSTYPE
136     load_module ptlrpc/ptlrpc
137     load_module ptlrpc/gss/ptlrpc_gss
138     # Now, some modules depend on lquota without USE_QUOTA check,
139     # will fix later. Disable check "$USE_QUOTA" = "yes" temporary.
140     #[ "$USE_QUOTA" = "yes" ] && load_module quota/lquota
141     load_module quota/lquota
142     load_module fid/fid
143     load_module fld/fld
144     load_module lmv/lmv
145     load_module mdc/mdc
146     load_module osc/osc
147     load_module lov/lov
148     load_module mds/mds
149     load_module mdd/mdd
150     load_module mdt/mdt
151     load_module cmm/cmm
152     load_module osd/osd
153     load_module ost/ost
154     load_module obdfilter/obdfilter
155     load_module llite/lustre
156     load_module mgc/mgc
157     load_module mgs/mgs
158     rm -f $TMP/ogdb-`hostname`
159     $LCTL modules > $TMP/ogdb-`hostname`
160     # 'mount' doesn't look in $PATH, just sbin
161     [ -f $LUSTRE/utils/mount.lustre ] && cp $LUSTRE/utils/mount.lustre /sbin/. || true
162 }
163
164 RMMOD=rmmod
165 if [ `uname -r | cut -c 3` -eq 4 ]; then
166     RMMOD="modprobe -r"
167 fi
168
169 wait_for_lnet() {
170     local UNLOADED=0
171     local WAIT=0
172     local MAX=60
173     MODULES=$($LCTL modules | awk '{ print $2 }')
174     while [ -n "$MODULES" ]; do
175     sleep 5
176     $RMMOD $MODULES > /dev/null 2>&1 || true
177     MODULES=$($LCTL modules | awk '{ print $2 }')
178         if [ -z "$MODULES" ]; then
179         return 0
180         else
181             WAIT=$((WAIT + 5))
182             echo "waiting, $((MAX - WAIT)) secs left"
183         fi
184         if [ $WAIT -eq $MAX ]; then
185             echo "LNET modules $MODULES will not unload"
186         lsmod
187             return 3
188         fi
189     done
190 }
191
192 unload_modules() {
193     lsmod | grep lnet > /dev/null && $LCTL dl && $LCTL dk $TMP/debug
194     local MODULES=$($LCTL modules | awk '{ print $2 }')
195     $RMMOD $MODULES > /dev/null 2>&1 || true
196      # do it again, in case we tried to unload ksocklnd too early
197     MODULES=$($LCTL modules | awk '{ print $2 }')
198     [ -n "$MODULES" ] && $RMMOD $MODULES > /dev/null 2>&1 || true
199     MODULES=$($LCTL modules | awk '{ print $2 }')
200     if [ -n "$MODULES" ]; then
201     echo "Modules still loaded: "
202     echo $MODULES 
203     if [ -e $LPROC ]; then
204         echo "Lustre still loaded"
205         cat $LPROC/devices || true
206         lsmod
207         return 2
208     else
209         echo "Lustre stopped but LNET is still loaded, waiting..."
210         wait_for_lnet || return 3
211     fi
212     fi
213     HAVE_MODULES=false
214
215     LEAK_LUSTRE=$(dmesg | tail -n 30 | grep "obd mem.*leaked" || true)
216     LEAK_PORTALS=$(dmesg | tail -n 20 | grep "Portals memory leaked" || true)
217     if [ "$LEAK_LUSTRE" -o "$LEAK_PORTALS" ]; then
218         echo "$LEAK_LUSTRE" 1>&2
219         echo "$LEAK_PORTALS" 1>&2
220         mv $TMP/debug $TMP/debug-leak.`date +%s` || true
221         echo "Memory leaks detected"
222         return 254
223     fi
224     echo "modules unloaded."
225     return 0
226 }
227
228 check_gss_daemon_facet() {
229     facet=$1
230     dname=$2
231
232     num=`do_facet $facet ps -o cmd -C $dname | grep $dname | wc -l`
233     if [ $num -ne 1 ]; then
234         echo "$num instance of $dname on $facet"
235         return 1
236     fi
237     return 0
238 }
239
240 send_sigint() {
241     local facet=$1
242     shift
243     do_facet $facet "killall -2 $@ 2>/dev/null || true"
244 }
245
246 start_gss_daemons() {
247     # starting on MDT
248     for num in `seq $MDSCOUNT`; do
249         do_facet mds$num "$LSVCGSSD -v"
250         do_facet mds$num "$LGSSD -v"
251     done
252     # starting on OSTs
253     for num in `seq $OSTCOUNT`; do
254         do_facet ost$num "$LSVCGSSD -v"
255     done
256     # starting on client
257     # FIXME: is "client" the right facet name?
258     do_facet client "$LGSSD -v"
259
260     # wait daemons entering "stable" status
261     sleep 5
262
263     #
264     # check daemons are running
265     #
266     for num in `seq $MDSCOUNT`; do
267         check_gss_daemon_facet mds$num lsvcgssd
268         check_gss_daemon_facet mds$num lgssd
269     done
270     for num in `seq $OSTCOUNT`; do
271         check_gss_daemon_facet ost$num lsvcgssd
272     done
273     check_gss_daemon_facet client lgssd
274 }
275
276 stop_gss_daemons() {
277     for num in `seq $MDSCOUNT`; do
278         send_sigint mds$num lsvcgssd lgssd
279     done
280     for num in `seq $OSTCOUNT`; do
281         send_sigint ost$num lsvcgssd
282     done
283     send_sigint client lgssd
284 }
285
286 init_krb5_env() {
287     if [ ! -z $SEC ]; then
288         MDS_MOUNT_OPTS=$MDS_MOUNT_OPTS,sec=$SEC
289         OST_MOUNT_OPTS=$OST_MOUNT_OPTS,sec=$SEC
290     fi
291
292     if [ ! -z $USING_KRB5 ]; then
293         start_gss_daemons
294     fi
295 }
296
297 cleanup_krb5_env() {
298     if [ ! -z $USING_KRB5 ]; then
299         stop_gss_daemons
300         # maybe cleanup credential cache?
301     fi
302 }
303
304 mdsdevlabel() {
305     local num=$1
306     local device=`mdsdevname $num`
307     local label=`do_facet mds$num "e2label ${device}" | grep -v "CMD: "`
308     echo -n $label
309 }
310
311 ostdevlabel() {
312     local num=$1
313     local device=`ostdevname $num`
314     local label=`do_facet ost$num "e2label ${device}" | grep -v "CMD: "`
315     echo -n $label
316 }
317
318 # Facet functions
319 # start facet device options 
320 start() {
321     facet=$1
322     shift
323     device=$1
324     shift
325     echo "Starting ${facet}: $@ ${device} ${MOUNT%/*}/${facet}"
326     do_facet ${facet} mkdir -p ${MOUNT%/*}/${facet}
327     do_facet ${facet} mount -t lustre $@ ${device} ${MOUNT%/*}/${facet} 
328     RC=${PIPESTATUS[0]}
329     if [ $RC -ne 0 ]; then
330         echo mount -t lustre $@ ${device} ${MOUNT%/*}/${facet} 
331         echo Start of ${device} on ${facet} failed ${RC}
332     else 
333         do_facet ${facet} sync
334         label=$(do_facet ${facet} "e2label ${device}")
335         [ -z "$label" ] && echo no label for ${device} && exit 1
336         eval export ${facet}_svc=${label}
337         eval export ${facet}_dev=${device}
338         eval export ${facet}_opt=\"$@\"
339         echo Started ${label}
340     fi
341     return $RC
342 }
343
344 stop() {
345     local running
346     facet=$1
347     shift
348     HOST=`facet_active_host $facet`
349     [ -z $HOST ] && echo stop: no host for $facet && return 0
350
351     running=$(do_facet ${facet} "grep -c ${MOUNT%/*}/${facet}' ' /proc/mounts") || true
352     if [ ${running} -ne 0 ]; then
353         echo "Stopping ${MOUNT%/*}/${facet} (opts:$@)"
354         do_facet ${facet} umount -d $@ ${MOUNT%/*}/${facet}
355     fi
356
357     # umount should block, but we should wait for unrelated obd's
358     # like the MGS or MGC to also stop.
359     local WAIT=0
360     local INTERVAL=1
361     # conf-sanity 31 takes a long time cleanup
362     while [ $WAIT -lt 300 ]; do
363         running=$(do_facet ${facet} "[ -e $LPROC ] && grep ST' ' $LPROC/devices") || true
364         if [ -z "${running}" ]; then
365                 return 0
366         fi
367         echo "waited $WAIT for${running}"
368         if [ $INTERVAL -lt 64 ]; then 
369             INTERVAL=$((INTERVAL + INTERVAL))
370         fi
371         sleep $INTERVAL
372         WAIT=$((WAIT + INTERVAL))
373     done
374     echo "service didn't stop after $WAIT seconds.  Still running:"
375     echo ${running}
376     exit 1
377 }
378
379 zconf_mount() {
380     local OPTIONS
381     local client=$1
382     local mnt=$2
383     # Only supply -o to mount if we have options
384     if [ -n "$MOUNTOPT" ]; then
385         OPTIONS="-o $MOUNTOPT"
386     fi
387     local device=$MGSNID:/$FSNAME
388     if [ -z "$mnt" -o -z "$FSNAME" ]; then
389         echo Bad zconf mount command: opt=$OPTIONS dev=$device mnt=$mnt
390         exit 1
391     fi
392
393     echo "Starting client: $OPTIONS $device $mnt" 
394     do_node $client mkdir -p $mnt
395     do_node $client mount -t lustre $OPTIONS $device $mnt || return 1
396
397     do_node $client "sysctl -w lnet.debug=$PTLDEBUG; sysctl -w lnet.subsystem_debug=${SUBSYSTEM# }"
398     [ -d /r ] && $LCTL modules > /r/tmp/ogdb-`hostname`
399     return 0
400 }
401
402 zconf_umount() {
403     client=$1
404     mnt=$2
405     [ "$3" ] && force=-f
406     local running=$(do_node $client "grep -c $mnt' ' /proc/mounts") || true
407     if [ $running -ne 0 ]; then
408         echo "Stopping client $mnt (opts:$force)"
409         do_node $client umount $force $mnt
410     fi
411 }
412
413 shutdown_facet() {
414     facet=$1
415     if [ "$FAILURE_MODE" = HARD ]; then
416         $POWER_DOWN `facet_active_host $facet`
417         sleep 2 
418     elif [ "$FAILURE_MODE" = SOFT ]; then
419         stop $facet
420     fi
421 }
422
423 reboot_facet() {
424     facet=$1
425     if [ "$FAILURE_MODE" = HARD ]; then
426         $POWER_UP `facet_active_host $facet`
427     else
428         sleep 10
429     fi
430 }
431
432 # verify that lustre actually cleaned up properly
433 cleanup_check() {
434     [ -f $CATASTROPHE ] && [ `cat $CATASTROPHE` -ne 0 ] && \
435         error "LBUG/LASSERT detected"
436     BUSY=`dmesg | grep -i destruct || true`
437     if [ "$BUSY" ]; then
438         echo "$BUSY" 1>&2
439         [ -e $TMP/debug ] && mv $TMP/debug $TMP/debug-busy.`date +%s`
440         exit 205
441     fi
442     LEAK_LUSTRE=`dmesg | tail -n 30 | grep "obd mem.*leaked" || true`
443     LEAK_PORTALS=`dmesg | tail -n 20 | grep "Portals memory leaked" || true`
444     if [ "$LEAK_LUSTRE" -o "$LEAK_PORTALS" ]; then
445         echo "$0: $LEAK_LUSTRE" 1>&2
446         echo "$0: $LEAK_PORTALS" 1>&2
447         echo "$0: Memory leak(s) detected..." 1>&2
448         mv $TMP/debug $TMP/debug-leak.`date +%s`
449         exit 204
450     fi
451
452     [ "`lctl dl 2> /dev/null | wc -l`" -gt 0 ] && lctl dl && \
453         echo "$0: lustre didn't clean up..." 1>&2 && return 202 || true
454
455     if [ "`/sbin/lsmod 2>&1 | egrep 'lnet|libcfs'`" ]; then
456         echo "$0: modules still loaded..." 1>&2
457         /sbin/lsmod 1>&2
458         return 203
459     fi
460     return 0
461 }
462
463 wait_for_host() {
464     HOST=$1
465     check_network "$HOST" 900
466     while ! do_node $HOST "ls -d $LUSTRE " > /dev/null; do sleep 5; done
467 }
468
469 wait_for() {
470     facet=$1
471     HOST=`facet_active_host $facet`
472     wait_for_host $HOST
473 }
474
475 client_df() {
476     # not every config has many clients
477     if [ ! -z "$CLIENTS" ]; then
478         $PDSH $CLIENTS "df $MOUNT" > /dev/null
479     fi
480 }
481
482 client_reconnect() {
483     uname -n >> $MOUNT/recon
484     if [ ! -z "$CLIENTS" ]; then
485         $PDSH $CLIENTS "df $MOUNT; uname -n >> $MOUNT/recon" > /dev/null
486     fi
487     echo Connected clients:
488     cat $MOUNT/recon
489     ls -l $MOUNT/recon > /dev/null
490     rm $MOUNT/recon
491 }
492
493 facet_failover() {
494     facet=$1
495     echo "Failing $facet on node `facet_active_host $facet`"
496     shutdown_facet $facet
497     reboot_facet $facet
498     client_df &
499     DFPID=$!
500     echo "df pid is $DFPID"
501     change_active $facet
502     TO=`facet_active_host $facet`
503     echo "Failover $facet to $TO"
504     wait_for $facet
505     local dev=${facet}_dev
506     local opt=${facet}_opt
507     start $facet ${!dev} ${!opt} || error "Restart of $facet failed"
508 }
509
510 obd_name() {
511     local facet=$1
512 }
513
514 replay_barrier() {
515     local facet=$1
516     do_facet $facet sync
517     df $MOUNT
518     local svc=${facet}_svc
519     do_facet $facet $LCTL --device %${!svc} readonly
520     do_facet $facet $LCTL --device %${!svc} notransno
521     do_facet $facet $LCTL mark "$facet REPLAY BARRIER on ${!svc}"
522     $LCTL mark "local REPLAY BARRIER on ${!svc}"
523 }
524
525 replay_barrier_nodf() {
526     local facet=$1    echo running=${running}
527     do_facet $facet sync
528     local svc=${facet}_svc
529     echo Replay barrier on ${!svc}
530     do_facet $facet $LCTL --device %${!svc} readonly
531     do_facet $facet $LCTL --device %${!svc} notransno
532     do_facet $facet $LCTL mark "$facet REPLAY BARRIER on ${!svc}"
533     $LCTL mark "local REPLAY BARRIER on ${!svc}"
534 }
535
536 mds_evict_client() {
537     UUID=`cat /proc/fs/lustre/mdc/${mds1_svc}-mdc-*/uuid`
538     do_facet mds1 "echo $UUID > /proc/fs/lustre/mdt/${mds1_svc}/evict_client"
539 }
540
541 ost_evict_client() {
542     UUID=`grep ${ost1_svc}-osc- $LPROC/devices | egrep -v 'MDT' | awk '{print $5}'`
543     do_facet ost1 "echo $UUID > /proc/fs/lustre/obdfilter/${ost1_svc}/evict_client"
544 }
545
546 fail() {
547     facet_failover $* || error "failover: $?"
548     df $MOUNT || error "post-failover df: $?"
549 }
550
551 fail_abort() {
552     local facet=$1
553     stop $facet
554     change_active $facet
555     local svc=${facet}_svc
556     local dev=${facet}_dev
557     local opt=${facet}_opt
558     start $facet ${!dev} ${!opt}
559     do_facet $facet lctl --device %${!svc} abort_recovery
560     df $MOUNT || echo "first df failed: $?"
561     sleep 1
562     df $MOUNT || error "post-failover df: $?"
563 }
564
565 do_lmc() {
566     echo There is no lmc.  This is mountconf, baby.
567     exit 1
568 }
569
570 h2gm () {
571     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
572         ID=`$PDSH $1 $GMNALNID -l | cut -d\  -f2`
573         echo $ID"@gm"
574     fi
575 }
576
577 h2name_or_ip() {
578     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
579         echo $1"@$2" 
580     fi
581 }
582
583 h2ptl() {
584    if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
585        ID=`xtprocadmin -n $1 2>/dev/null | egrep -v 'NID' | awk '{print $1}'`
586        if [ -z "$ID" ]; then
587            echo "Could not get a ptl id for $1..."
588            exit 1
589        fi
590        echo $ID"@ptl"
591    fi
592 }
593 declare -fx h2ptl
594
595 h2tcp() {
596     h2name_or_ip "$1" "tcp"
597 }
598 declare -fx h2tcp
599
600 h2elan() {
601     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
602         if type __h2elan >/dev/null 2>&1; then
603             ID=$(__h2elan $1)
604         else
605             ID=`echo $1 | sed 's/[^0-9]*//g'`
606         fi
607         echo $ID"@elan"
608     fi
609 }
610 declare -fx h2elan
611
612 h2openib() {
613     h2name_or_ip "$1" "openib"
614 }
615 declare -fx h2openib
616
617 h2o2ib() {
618     h2name_or_ip "$1" "o2ib"
619 }
620 declare -fx h2o2ib
621
622 facet_host() {
623     local facet=$1
624     varname=${facet}_HOST
625     if [ -z "${!varname}" ]; then
626         if [ "${facet:0:3}" == "ost" ]; then
627             eval ${facet}_HOST=${ost_HOST}
628         fi
629     fi
630     echo -n ${!varname}
631 }
632
633 facet_active() {
634     local facet=$1
635     local activevar=${facet}active
636
637     if [ -f ./${facet}active ] ; then
638         source ./${facet}active
639     fi
640
641     active=${!activevar}
642     if [ -z "$active" ] ; then 
643         echo -n ${facet}
644     else
645         echo -n ${active}
646     fi
647 }
648
649 facet_active_host() {
650     local facet=$1
651     local active=`facet_active $facet`
652     if [ "$facet" == client ]; then
653         hostname
654     else
655         echo `facet_host $active`
656     fi
657 }
658
659 change_active() {
660     local facet=$1
661     failover=${facet}failover 
662     host=`facet_host $failover`
663     [ -z "$host" ] && return
664     curactive=`facet_active $facet`
665     if [ -z "${curactive}" -o "$curactive" == "$failover" ] ; then
666         eval export ${facet}active=$facet
667     else
668         eval export ${facet}active=$failover
669     fi
670     # save the active host for this facet
671     activevar=${facet}active
672     echo "$activevar=${!activevar}" > ./$activevar
673 }
674
675 do_node() {
676     HOST=$1
677     shift
678     local myPDSH=$PDSH
679     if [ "$HOST" = "$(hostname)" ]; then
680         myPDSH="no_dsh"
681     fi
682     if $VERBOSE; then
683         echo "CMD: $HOST $@" >&2
684         $myPDSH $HOST $LCTL mark "$@" > /dev/null 2>&1 || :
685     fi
686     $myPDSH $HOST "(PATH=\$PATH:$RLUSTRE/utils:$RLUSTRE/tests:/sbin:/usr/sbin; cd $RPWD; sh -c \"$@\")" | sed "s/^${HOST}: //"
687     return ${PIPESTATUS[0]}
688 }
689
690 do_facet() {
691     facet=$1
692     shift
693     HOST=`facet_active_host $facet`
694     [ -z $HOST ] && echo No host defined for facet ${facet} && exit 1
695     do_node $HOST $@
696 }
697
698 add() {
699     local facet=$1
700     shift
701     # make sure its not already running
702     stop ${facet} -f
703     rm -f ${facet}active
704     do_facet ${facet} $MKFS $*
705 }
706
707 ostdevname() {
708     num=$1
709     DEVNAME=OSTDEV$num
710     #if $OSTDEVn isn't defined, default is $OSTDEVBASE + num
711     eval DEVPTR=${!DEVNAME:=${OSTDEVBASE}${num}}
712     echo -n $DEVPTR
713 }
714
715 mdsdevname() {
716     num=$1
717     DEVNAME=MDSDEV$num
718     #if $MDSDEVn isn't defined, default is $MDSDEVBASE + num
719     eval DEVPTR=${!DEVNAME:=${MDSDEVBASE}${num}}
720     echo -n $DEVPTR
721 }
722
723 ########
724 ## MountConf setup
725
726 stopall() {
727     # make sure we are using the primary server, so test-framework will
728     # be able to clean up properly.
729     activemds=`facet_active mds1`
730     if [ $activemds != "mds1" ]; then
731         fail mds1
732     fi
733     
734     # assume client mount is local 
735     grep " $MOUNT " /proc/mounts && zconf_umount `hostname` $MOUNT $*
736     grep " $MOUNT2 " /proc/mounts && zconf_umount `hostname` $MOUNT2 $*
737     for num in `seq $MDSCOUNT`; do
738         stop mds$num -f
739     done
740     for num in `seq $OSTCOUNT`; do
741         stop ost$num -f
742     done
743     return 0
744 }
745
746 cleanupall() {
747     stopall $*
748     unload_modules
749     cleanup_krb5_env
750 }
751
752 mdsmkfsopts()
753 {
754     local nr=$1
755     test $nr = 1 && echo -n $MDS_MKFS_OPTS || echo -n $MDSn_MKFS_OPTS
756 }
757
758 formatall() {
759     [ "$FSTYPE" ] && FSTYPE_OPT="--backfstype $FSTYPE"
760
761     stopall
762     # We need ldiskfs here, may as well load them all
763     load_modules
764     [ "$CLIENTONLY" ] && return
765     echo "Formatting mdts, osts"
766     for num in `seq $MDSCOUNT`; do
767         echo "Format mds$num: $(mdsdevname $num)"
768         if $VERBOSE; then
769             add mds$num `mdsmkfsopts $num` $FSTYPE_OPT --reformat `mdsdevname $num` || exit 9
770         else
771             add mds$num `mdsmkfsopts $num` $FSTYPE_OPT --reformat `mdsdevname $num` > /dev/null || exit 9
772         fi
773     done
774
775     for num in `seq $OSTCOUNT`; do
776         echo "Format ost$num: $(ostdevname $num)"
777         if $VERBOSE; then
778             add ost$num $OST_MKFS_OPTS --reformat `ostdevname $num` || exit 10
779         else
780             add ost$num $OST_MKFS_OPTS --reformat `ostdevname $num` > /dev/null || exit 10
781         fi
782     done
783 }
784
785 mount_client() {
786     grep " $1 " /proc/mounts || zconf_mount `hostname` $*
787 }
788
789 setupall() {
790     load_modules
791     init_krb5_env
792     if [ -z "$CLIENTONLY" ]; then
793         echo "Setup mdts, osts"
794         for num in `seq $MDSCOUNT`; do
795             DEVNAME=$(mdsdevname $num)
796             echo $REFORMAT | grep -q "reformat" \
797             || do_facet mds$num "$TUNEFS --writeconf $DEVNAME"
798             start mds$num $DEVNAME $MDS_MOUNT_OPTS
799         done
800         for num in `seq $OSTCOUNT`; do
801             DEVNAME=$(ostdevname $num)
802             start ost$num $DEVNAME $OST_MOUNT_OPTS
803         done
804     fi
805     [ "$DAEMONFILE" ] && $LCTL debug_daemon start $DAEMONFILE $DAEMONSIZE
806     mount_client $MOUNT
807     if [ "$MOUNT_2" ]; then
808         mount_client $MOUNT2
809     fi
810     sleep 5
811 }
812
813 mounted_lustre_filesystems() {
814         awk '($3 ~ "lustre" && $1 ~ ":") { print $2 }' /proc/mounts
815 }
816
817 check_and_setup_lustre() {
818     MOUNTED="`mounted_lustre_filesystems`"
819     if [ -z "$MOUNTED" ]; then
820         [ "$REFORMAT" ] && formatall
821         setupall
822         MOUNTED="`mounted_lustre_filesystems`"
823         [ -z "$MOUNTED" ] && error "NAME=$NAME not mounted"
824         export I_MOUNTED=yes
825     fi
826     if [ "$ONLY" == "setup" ]; then
827         exit 0
828     fi
829 }
830
831 cleanup_and_setup_lustre() {
832     if [ "$ONLY" == "cleanup" -o "`mount | grep $MOUNT`" ]; then
833         sysctl -w lnet.debug=0 || true
834         cleanupall
835         if [ "$ONLY" == "cleanup" ]; then 
836             exit 0
837         fi
838     fi
839     check_and_setup_lustre
840 }
841
842 check_and_cleanup_lustre() {
843     if [ "`mount | grep $MOUNT`" ]; then
844         rm -rf $DIR/[Rdfs][1-9]*
845     fi
846     if [ "$I_MOUNTED" = "yes" ]; then
847         cleanupall -f || error "cleanup failed"
848     fi
849     unset I_MOUNTED
850 }
851
852 ####### 
853 # General functions
854
855 check_network() {
856     local NETWORK=0
857     local WAIT=0
858     local MAX=$2
859     while [ $NETWORK -eq 0 ]; do
860         ping -c 1 -w 3 $1 > /dev/null
861         if [ $? -eq 0 ]; then
862             NETWORK=1
863         else
864             WAIT=$((WAIT + 5))
865             echo "waiting for $1, $((MAX - WAIT)) secs left"
866             sleep 5
867         fi
868         if [ $WAIT -gt $MAX ]; then
869             echo "Network not available"
870             exit 1
871         fi
872     done
873 }
874 check_port() {
875     while( !($DSH2 $1 "netstat -tna | grep -q $2") ) ; do
876         sleep 9
877     done
878 }
879
880 no_dsh() {
881     shift
882     eval $@
883 }
884
885 comma_list() {
886     # the sed converts spaces to commas, but leaves the last space
887     # alone, so the line doesn't end with a comma.
888     echo "$*" | tr -s " " "\n" | sort -b -u | tr "\n" " " | sed 's/ \([^$]\)/,\1/g'
889 }
890
891 absolute_path() {
892     (cd `dirname $1`; echo $PWD/`basename $1`)
893 }
894
895 ##################################
896 # OBD_FAIL funcs
897
898 drop_request() {
899 # OBD_FAIL_MDS_ALL_REQUEST_NET
900     RC=0
901     do_facet mds sysctl -w lustre.fail_loc=0x123
902     do_facet client "$1" || RC=$?
903     do_facet mds sysctl -w lustre.fail_loc=0
904     return $RC
905 }
906
907 drop_reply() {
908 # OBD_FAIL_MDS_ALL_REPLY_NET
909     RC=0
910     do_facet mds sysctl -w lustre.fail_loc=0x122
911     do_facet client "$@" || RC=$?
912     do_facet mds sysctl -w lustre.fail_loc=0
913     return $RC
914 }
915
916 drop_reint_reply() {
917 # OBD_FAIL_MDS_REINT_NET_REP
918     RC=0
919     do_facet mds sysctl -w lustre.fail_loc=0x119
920     do_facet client "$@" || RC=$?
921     do_facet mds sysctl -w lustre.fail_loc=0
922     return $RC
923 }
924
925 pause_bulk() {
926 #define OBD_FAIL_OST_BRW_PAUSE_BULK      0x214
927     RC=0
928     do_facet ost1 sysctl -w lustre.fail_loc=0x214
929     do_facet client "$1" || RC=$?
930     do_facet client "sync"
931     do_facet ost1 sysctl -w lustre.fail_loc=0
932     return $RC
933 }
934
935 drop_ldlm_cancel() {
936 #define OBD_FAIL_LDLM_CANCEL             0x304
937     RC=0
938     do_facet client sysctl -w lustre.fail_loc=0x304
939     do_facet client "$@" || RC=$?
940     do_facet client sysctl -w lustre.fail_loc=0
941     return $RC
942 }
943
944 drop_bl_callback() {
945 #define OBD_FAIL_LDLM_BL_CALLBACK        0x305
946     RC=0
947     do_facet client sysctl -w lustre.fail_loc=0x305
948     do_facet client "$@" || RC=$?
949     do_facet client sysctl -w lustre.fail_loc=0
950     return $RC
951 }
952
953 drop_ldlm_reply() {
954 #define OBD_FAIL_LDLM_REPLY              0x30c
955     RC=0
956     do_facet mds sysctl -w lustre.fail_loc=0x30c
957     do_facet client "$@" || RC=$?
958     do_facet mds sysctl -w lustre.fail_loc=0
959     return $RC
960 }
961
962 clear_failloc() {
963     facet=$1
964     pause=$2
965     sleep $pause
966     echo "clearing fail_loc on $facet"
967     do_facet $facet "sysctl -w lustre.fail_loc=0"
968 }
969
970 cancel_lru_locks() {
971     $LCTL mark "cancel_lru_locks $1 start"
972     for d in `find $LPROC/ldlm/namespaces | egrep -i $1`; do
973         [ -f $d/lru_size ] && echo clear > $d/lru_size
974         [ -f $d/lock_unused_count ] && grep [1-9] $d/lock_unused_count /dev/null
975     done
976     $LCTL mark "cancel_lru_locks $1 stop"
977 }
978
979
980 pgcache_empty() {
981     for a in /proc/fs/lustre/llite/*/dump_page_cache; do
982         if [ `wc -l $a | awk '{print $1}'` -gt 1 ]; then
983             echo there is still data in page cache $a ?
984             cat $a;
985             return 1;
986         fi
987     done
988     return 0
989 }
990
991 FAIL_ON_ERROR=true
992 ##################################
993 # Test interface 
994 error() {
995     local ERRLOG
996     sysctl -w lustre.fail_loc=0 2> /dev/null || true
997     log "${TESTSUITE} ${TESTNAME}: **** FAIL:" $@
998     ERRLOG=$TMP/lustre_${TESTSUITE}_${TESTNAME}.$(date +%s)
999     echo "Dumping lctl log to $ERRLOG"
1000     # We need to dump the logs on all nodes
1001     $LCTL dk $ERRLOG
1002     [ ! "$mds_HOST" = "$(hostname)" ] && do_node $mds_HOST $LCTL dk $ERRLOG
1003     [ ! "$ost_HOST" = "$(hostname)" -a ! "$ost_HOST" = "$mds_HOST" ] && do_node $ost_HOST $LCTL dk $ERRLOG
1004     debugrestore
1005     [ "$TESTSUITELOG" ] && echo "$0: FAIL: $TESTNAME $@" >> $TESTSUITELOG
1006     if $FAIL_ON_ERROR; then
1007         exit 1
1008     fi
1009 }
1010
1011 skip () {
1012         log " SKIP: ${TESTSUITE} ${TESTNAME} $@"
1013         [ "$TESTSUITELOG" ] && echo "${TESTSUITE}: SKIP: $TESTNAME $@" >> $TESTSUITELOG
1014 }
1015
1016 build_test_filter() {
1017     [ "$ONLY" ] && log "only running test `echo $ONLY`"
1018     for O in $ONLY; do
1019         eval ONLY_${O}=true
1020     done
1021     [ "$EXCEPT$ALWAYS_EXCEPT" ] && \
1022         log "skipping tests: `echo $EXCEPT $ALWAYS_EXCEPT`"
1023     for E in $EXCEPT $ALWAYS_EXCEPT; do
1024         eval EXCEPT_${E}=true
1025     done
1026     for G in $GRANT_CHECK_LIST; do
1027         eval GCHECK_ONLY_${G}=true
1028         done
1029 }
1030
1031 _basetest() {
1032     echo $*
1033 }
1034
1035 basetest() {
1036     IFS=abcdefghijklmnopqrstuvwxyz _basetest $1
1037 }
1038
1039 run_test() {
1040     export base=`basetest $1`
1041     if [ ! -z "$ONLY" ]; then
1042         testname=ONLY_$1
1043         if [ ${!testname}x != x ]; then
1044             run_one $1 "$2"
1045             return $?
1046         fi
1047         testname=ONLY_$base
1048         if [ ${!testname}x != x ]; then
1049             run_one $1 "$2"
1050             return $?
1051         fi
1052         echo -n "."
1053         return 0
1054     fi
1055     testname=EXCEPT_$1
1056     if [ ${!testname}x != x ]; then
1057         TESTNAME=test_$1 skip "skipping excluded test $1"
1058         return 0
1059     fi
1060     testname=EXCEPT_$base
1061     if [ ${!testname}x != x ]; then
1062         TESTNAME=test_$1 skip "skipping excluded test $1 (base $base)"
1063         return 0
1064     fi
1065     run_one $1 "$2"
1066     
1067     return $?
1068 }
1069
1070 EQUALS="======================================================================"
1071 equals_msg() {
1072     msg="$@"
1073
1074     local suffixlen=$((${#EQUALS} - ${#msg}))
1075     [ $suffixlen -lt 5 ] && suffixlen=5
1076     printf '===== %s %.*s\n' "$msg" $suffixlen $EQUALS
1077 }
1078
1079 log() {
1080     echo "$*"
1081     lsmod | grep lnet > /dev/null || load_modules
1082     $LCTL mark "$*" 2> /dev/null || true
1083 }
1084
1085 trace() {
1086         log "STARTING: $*"
1087         strace -o $TMP/$1.strace -ttt $*
1088         RC=$?
1089         log "FINISHED: $*: rc $RC"
1090         return 1
1091 }
1092
1093 pass() {
1094     echo PASS $@
1095 }
1096
1097 check_mds() {
1098     FFREE=`cat /proc/fs/lustre/mds/*/filesfree`
1099     FTOTAL=`cat /proc/fs/lustre/mds/*/filestotal`
1100     [ $FFREE -ge $FTOTAL ] && error "files free $FFREE > total $FTOTAL" || true
1101 }
1102
1103 run_one() {
1104     testnum=$1
1105     message=$2
1106     tfile=f${testnum}
1107     tdir=d${base}
1108
1109     BEFORE=`date +%s`
1110     log "== test $testnum: $message ============ `date +%H:%M:%S` ($BEFORE)"
1111     #check_mds
1112     export TESTNAME=test_$testnum
1113     test_${testnum} || error "test_$testnum failed with $?"
1114     #check_mds
1115     check_grant ${testnum} || error "check_grant $testnum failed with $?"
1116     [ -f $CATASTROPHE ] && [ `cat $CATASTROPHE` -ne 0 ] && \
1117         error "LBUG/LASSERT detected"
1118     pass "($((`date +%s` - $BEFORE))s)"
1119     unset TESTNAME
1120     cd $SAVE_PWD
1121     $CLEANUP
1122 }
1123
1124 canonical_path() {
1125     (cd `dirname $1`; echo $PWD/`basename $1`)
1126 }
1127
1128 sync_clients() {
1129     [ -d $DIR1 ] && cd $DIR1 && sync; sleep 1; sync 
1130     [ -d $DIR2 ] && cd $DIR2 && sync; sleep 1; sync 
1131         cd $SAVE_PWD
1132 }
1133
1134 check_grant() {
1135     export base=`basetest $1`
1136     [ "$CHECK_GRANT" == "no" ] && return 0
1137
1138         testname=GCHECK_ONLY_${base}
1139         [ ${!testname}x == x ] && return 0
1140
1141         echo -n "checking grant......"
1142         cd $SAVE_PWD
1143         # write some data to sync client lost_grant
1144         rm -f $DIR1/${tfile}_check_grant_* 2>&1
1145         for i in `seq $OSTCOUNT`; do
1146                 $LFS setstripe $DIR1/${tfile}_check_grant_$i 0 $(($i -1)) 1
1147                 dd if=/dev/zero of=$DIR1/${tfile}_check_grant_$i bs=4k \
1148                                               count=1 > /dev/null 2>&1 
1149         done
1150         # sync all the data and make sure no pending data on server
1151         sync_clients
1152         
1153         #get client grant and server grant 
1154         client_grant=0
1155     for d in ${LPROC}/osc/*/cur_grant_bytes; do 
1156                 client_grant=$((client_grant + `cat $d`))
1157         done
1158         server_grant=0
1159         for d in ${LPROC}/obdfilter/*/tot_granted; do
1160                 server_grant=$((server_grant + `cat $d`))
1161         done
1162
1163         # cleanup the check_grant file
1164         for i in `seq $OSTCOUNT`; do
1165                 rm $DIR1/${tfile}_check_grant_$i
1166         done
1167
1168         #check whether client grant == server grant 
1169         if [ $client_grant != $server_grant ]; then
1170                 echo "failed: client:${client_grant} server: ${server_grant}"
1171                 return 1
1172         else
1173                 echo "pass"
1174         fi
1175 }
1176
1177 ########################
1178 # helper functions
1179
1180 osc_to_ost()
1181 {
1182     osc=$1
1183     ost=`echo $1 | awk -F_ '{print $3}'`
1184     if [ -z $ost ]; then
1185         ost=`echo $1 | sed 's/-osc.*//'`
1186     fi
1187     echo $ost
1188 }
1189
1190 remote_mds ()
1191 {
1192     [ ! -e /proc/fs/lustre/mdt/*MDT* ]
1193 }
1194
1195 remote_ost ()
1196 {
1197     [ $(grep -c obdfilter $LPROC/devices) -eq 0 ]
1198 }