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