Whamcloud - gitweb
b=13798
[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} mount -t lustre $@ ${device} ${MOUNT%/*}/${facet} 
347     RC=${PIPESTATUS[0]}
348     if [ $RC -ne 0 ]; then
349         echo mount -t lustre $@ ${device} ${MOUNT%/*}/${facet} 
350         echo Start of ${device} on ${facet} failed ${RC}
351     else 
352         do_facet ${facet} sync
353         label=$(do_facet ${facet} "e2label ${device}")
354         [ -z "$label" ] && echo no label for ${device} && exit 1
355         eval export ${facet}_svc=${label}
356         eval export ${facet}_dev=${device}
357         eval export ${facet}_opt=\"$@\"
358         echo Started ${label}
359     fi
360     return $RC
361 }
362
363 stop() {
364     local running
365     facet=$1
366     shift
367     HOST=`facet_active_host $facet`
368     [ -z $HOST ] && echo stop: no host for $facet && return 0
369
370     running=$(do_facet ${facet} "grep -c ${MOUNT%/*}/${facet}' ' /proc/mounts") || true
371     if [ ${running} -ne 0 ]; then
372         echo "Stopping ${MOUNT%/*}/${facet} (opts:$@)"
373         do_facet ${facet} umount -d $@ ${MOUNT%/*}/${facet}
374     fi
375
376     # umount should block, but we should wait for unrelated obd's
377     # like the MGS or MGC to also stop.
378     wait_exit_ST ${facet}
379 }
380
381 zconf_mount() {
382     local OPTIONS
383     local client=$1
384     local mnt=$2
385     # Only supply -o to mount if we have options
386     if [ -n "$MOUNTOPT" ]; then
387         OPTIONS="-o $MOUNTOPT"
388     fi
389     local device=$MGSNID:/$FSNAME
390     if [ -z "$mnt" -o -z "$FSNAME" ]; then
391         echo Bad zconf mount command: opt=$OPTIONS dev=$device mnt=$mnt
392         exit 1
393     fi
394
395     echo "Starting client: $OPTIONS $device $mnt" 
396     do_node $client mkdir -p $mnt
397     do_node $client mount -t lustre $OPTIONS $device $mnt || return 1
398
399     do_node $client "sysctl -w lnet.debug=$PTLDEBUG; sysctl -w lnet.subsystem_debug=${SUBSYSTEM# }"
400     [ -d /r ] && $LCTL modules > /r/tmp/ogdb-`hostname`
401     return 0
402 }
403
404 zconf_umount() {
405     client=$1
406     mnt=$2
407     [ "$3" ] && force=-f
408     local running=$(do_node $client "grep -c $mnt' ' /proc/mounts") || true
409     if [ $running -ne 0 ]; then
410         echo "Stopping client $mnt (opts:$force)"
411         do_node $client umount $force $mnt
412     fi
413 }
414
415 shutdown_facet() {
416     facet=$1
417     if [ "$FAILURE_MODE" = HARD ]; then
418         $POWER_DOWN `facet_active_host $facet`
419         sleep 2 
420     elif [ "$FAILURE_MODE" = SOFT ]; then
421         stop $facet
422     fi
423 }
424
425 reboot_facet() {
426     facet=$1
427     if [ "$FAILURE_MODE" = HARD ]; then
428         $POWER_UP `facet_active_host $facet`
429     else
430         sleep 10
431     fi
432 }
433
434 # verify that lustre actually cleaned up properly
435 cleanup_check() {
436     [ -f $CATASTROPHE ] && [ `cat $CATASTROPHE` -ne 0 ] && \
437         error "LBUG/LASSERT detected"
438     BUSY=`dmesg | grep -i destruct || true`
439     if [ "$BUSY" ]; then
440         echo "$BUSY" 1>&2
441         [ -e $TMP/debug ] && mv $TMP/debug $TMP/debug-busy.`date +%s`
442         exit 205
443     fi
444     LEAK_LUSTRE=`dmesg | tail -n 30 | grep "obd mem.*leaked" || true`
445     LEAK_PORTALS=`dmesg | tail -n 20 | grep "Portals memory leaked" || true`
446     if [ "$LEAK_LUSTRE" -o "$LEAK_PORTALS" ]; then
447         echo "$0: $LEAK_LUSTRE" 1>&2
448         echo "$0: $LEAK_PORTALS" 1>&2
449         echo "$0: Memory leak(s) detected..." 1>&2
450         mv $TMP/debug $TMP/debug-leak.`date +%s`
451         exit 204
452     fi
453
454     [ "`lctl dl 2> /dev/null | wc -l`" -gt 0 ] && lctl dl && \
455         echo "$0: lustre didn't clean up..." 1>&2 && return 202 || true
456
457     if [ "`/sbin/lsmod 2>&1 | egrep 'lnet|libcfs'`" ]; then
458         echo "$0: modules still loaded..." 1>&2
459         /sbin/lsmod 1>&2
460         return 203
461     fi
462     return 0
463 }
464
465 wait_delete_completed () {
466     local TOTALPREV=`awk 'BEGIN{total=0}; {total+=$1}; END{print total}' \
467             $LPROC/osc/*/kbytesavail`
468
469     local WAIT=0
470     local MAX_WAIT=20
471     while [ "$WAIT" -ne "$MAX_WAIT" ]; do
472         sleep 1
473         TOTAL=`awk 'BEGIN{total=0}; {total+=$1}; END{print total}' \
474             $LPROC/osc/*/kbytesavail`
475         [ "$TOTAL" -eq "$TOTALPREV" ] && break
476         echo "Waiting delete completed ... prev: $TOTALPREV current: $TOTAL "
477         TOTALPREV=$TOTAL
478         WAIT=$(( WAIT + 1))
479     done
480     echo "Delete completed."
481 }
482
483 wait_for_host() {
484     HOST=$1
485     check_network "$HOST" 900
486     while ! do_node $HOST "ls -d $LUSTRE " > /dev/null; do sleep 5; done
487 }
488
489 wait_for() {
490     facet=$1
491     HOST=`facet_active_host $facet`
492     wait_for_host $HOST
493 }
494
495 wait_mds_recovery_done () {
496     local timeout=`do_facet mds cat /proc/sys/lustre/timeout`
497 #define OBD_RECOVERY_TIMEOUT (obd_timeout * 5 / 2)
498 # as we are in process of changing obd_timeout in different ways
499 # let's set MAX longer than that
500     MAX=$(( timeout * 4 ))
501     WAIT=0
502     while [ $WAIT -lt $MAX ]; do
503         STATUS=`do_facet mds grep status /proc/fs/lustre/mdt/*-MDT*/recovery_status`
504         echo $STATUS | grep COMPLETE && return 0
505         sleep 5
506         WAIT=$((WAIT + 5))
507         echo "Waiting $(($MAX - $WAIT)) secs for MDS recovery done"
508     done
509     echo "MDS recovery not done in $MAX sec"
510     return 1            
511 }
512
513 wait_exit_ST () {
514     local facet=$1
515
516     local WAIT=0
517     local INTERVAL=1
518     # conf-sanity 31 takes a long time cleanup
519     while [ $WAIT -lt 300 ]; do
520         running=$(do_facet ${facet} "[ -e $LPROC ] && grep ST' ' $LPROC/devices") || true
521         [ -z "${running}" ] && return 0
522         echo "waited $WAIT for${running}"
523         [ $INTERVAL -lt 64 ] && INTERVAL=$((INTERVAL + INTERVAL))
524         sleep $INTERVAL
525         WAIT=$((WAIT + INTERVAL))
526     done
527     echo "service didn't stop after $WAIT seconds.  Still running:"
528     echo ${running}
529     return 1
530 }
531
532 client_df() {
533     # not every config has many clients
534     if [ ! -z "$CLIENTS" ]; then
535         $PDSH $CLIENTS "df $MOUNT" > /dev/null
536     fi
537 }
538
539 client_reconnect() {
540     uname -n >> $MOUNT/recon
541     if [ ! -z "$CLIENTS" ]; then
542         $PDSH $CLIENTS "df $MOUNT; uname -n >> $MOUNT/recon" > /dev/null
543     fi
544     echo Connected clients:
545     cat $MOUNT/recon
546     ls -l $MOUNT/recon > /dev/null
547     rm $MOUNT/recon
548 }
549
550 facet_failover() {
551     facet=$1
552     echo "Failing $facet on node `facet_active_host $facet`"
553     shutdown_facet $facet
554     reboot_facet $facet
555     client_df &
556     DFPID=$!
557     echo "df pid is $DFPID"
558     change_active $facet
559     TO=`facet_active_host $facet`
560     echo "Failover $facet to $TO"
561     wait_for $facet
562     local dev=${facet}_dev
563     local opt=${facet}_opt
564     start $facet ${!dev} ${!opt} || error "Restart of $facet failed"
565 }
566
567 obd_name() {
568     local facet=$1
569 }
570
571 replay_barrier() {
572     local facet=$1
573     do_facet $facet sync
574     df $MOUNT
575     local svc=${facet}_svc
576     do_facet $facet $LCTL --device %${!svc} readonly
577     do_facet $facet $LCTL --device %${!svc} notransno
578     do_facet $facet $LCTL mark "$facet REPLAY BARRIER on ${!svc}"
579     $LCTL mark "local REPLAY BARRIER on ${!svc}"
580 }
581
582 replay_barrier_nodf() {
583     local facet=$1    echo running=${running}
584     do_facet $facet sync
585     local svc=${facet}_svc
586     echo Replay barrier on ${!svc}
587     do_facet $facet $LCTL --device %${!svc} readonly
588     do_facet $facet $LCTL --device %${!svc} notransno
589     do_facet $facet $LCTL mark "$facet REPLAY BARRIER on ${!svc}"
590     $LCTL mark "local REPLAY BARRIER on ${!svc}"
591 }
592
593 mds_evict_client() {
594     UUID=`cat /proc/fs/lustre/mdc/${mds1_svc}-mdc-*/uuid`
595     do_facet mds1 "echo $UUID > /proc/fs/lustre/mdt/${mds1_svc}/evict_client"
596 }
597
598 ost_evict_client() {
599     UUID=`grep ${ost1_svc}-osc- $LPROC/devices | egrep -v 'MDT' | awk '{print $5}'`
600     do_facet ost1 "echo $UUID > /proc/fs/lustre/obdfilter/${ost1_svc}/evict_client"
601 }
602
603 fail() {
604     facet_failover $* || error "failover: $?"
605     df $MOUNT || error "post-failover df: $?"
606 }
607
608 fail_abort() {
609     local facet=$1
610     stop $facet
611     change_active $facet
612     local svc=${facet}_svc
613     local dev=${facet}_dev
614     local opt=${facet}_opt
615     start $facet ${!dev} ${!opt}
616     do_facet $facet lctl --device %${!svc} abort_recovery
617     df $MOUNT || echo "first df failed: $?"
618     sleep 1
619     df $MOUNT || error "post-failover df: $?"
620 }
621
622 do_lmc() {
623     echo There is no lmc.  This is mountconf, baby.
624     exit 1
625 }
626
627 h2gm () {
628     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
629         ID=`$PDSH $1 $GMNALNID -l | cut -d\  -f2`
630         echo $ID"@gm"
631     fi
632 }
633
634 h2name_or_ip() {
635     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
636         echo $1"@$2" 
637     fi
638 }
639
640 h2ptl() {
641    if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
642        ID=`xtprocadmin -n $1 2>/dev/null | egrep -v 'NID' | awk '{print $1}'`
643        if [ -z "$ID" ]; then
644            echo "Could not get a ptl id for $1..."
645            exit 1
646        fi
647        echo $ID"@ptl"
648    fi
649 }
650 declare -fx h2ptl
651
652 h2tcp() {
653     h2name_or_ip "$1" "tcp"
654 }
655 declare -fx h2tcp
656
657 h2elan() {
658     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
659         if type __h2elan >/dev/null 2>&1; then
660             ID=$(__h2elan $1)
661         else
662             ID=`echo $1 | sed 's/[^0-9]*//g'`
663         fi
664         echo $ID"@elan"
665     fi
666 }
667 declare -fx h2elan
668
669 h2openib() {
670     h2name_or_ip "$1" "openib"
671 }
672 declare -fx h2openib
673
674 h2o2ib() {
675     h2name_or_ip "$1" "o2ib"
676 }
677 declare -fx h2o2ib
678
679 facet_host() {
680     local facet=$1
681     varname=${facet}_HOST
682     if [ -z "${!varname}" ]; then
683         if [ "${facet:0:3}" == "ost" ]; then
684             eval ${facet}_HOST=${ost_HOST}
685         fi
686     fi
687     echo -n ${!varname}
688 }
689
690 facet_active() {
691     local facet=$1
692     local activevar=${facet}active
693
694     if [ -f ./${facet}active ] ; then
695         source ./${facet}active
696     fi
697
698     active=${!activevar}
699     if [ -z "$active" ] ; then 
700         echo -n ${facet}
701     else
702         echo -n ${active}
703     fi
704 }
705
706 facet_active_host() {
707     local facet=$1
708     local active=`facet_active $facet`
709     if [ "$facet" == client ]; then
710         hostname
711     else
712         echo `facet_host $active`
713     fi
714 }
715
716 change_active() {
717     local facet=$1
718     failover=${facet}failover 
719     host=`facet_host $failover`
720     [ -z "$host" ] && return
721     curactive=`facet_active $facet`
722     if [ -z "${curactive}" -o "$curactive" == "$failover" ] ; then
723         eval export ${facet}active=$facet
724     else
725         eval export ${facet}active=$failover
726     fi
727     # save the active host for this facet
728     activevar=${facet}active
729     echo "$activevar=${!activevar}" > ./$activevar
730 }
731
732 do_node() {
733     HOST=$1
734     shift
735     local myPDSH=$PDSH
736     if [ "$HOST" = "$(hostname)" ]; then
737         myPDSH="no_dsh"
738     fi
739     if $VERBOSE; then
740         echo "CMD: $HOST $@" >&2
741         $myPDSH $HOST $LCTL mark "$@" > /dev/null 2>&1 || :
742     fi
743     $myPDSH $HOST "(PATH=\$PATH:$RLUSTRE/utils:$RLUSTRE/tests:/sbin:/usr/sbin; cd $RPWD; sh -c \"$@\")" | sed "s/^${HOST}: //"
744     return ${PIPESTATUS[0]}
745 }
746
747 do_facet() {
748     facet=$1
749     shift
750     HOST=`facet_active_host $facet`
751     [ -z $HOST ] && echo No host defined for facet ${facet} && exit 1
752     do_node $HOST $@
753 }
754
755 add() {
756     local facet=$1
757     shift
758     # make sure its not already running
759     stop ${facet} -f
760     rm -f ${facet}active
761     do_facet ${facet} $MKFS $*
762 }
763
764 ostdevname() {
765     num=$1
766     DEVNAME=OSTDEV$num
767     #if $OSTDEVn isn't defined, default is $OSTDEVBASE + num
768     eval DEVPTR=${!DEVNAME:=${OSTDEVBASE}${num}}
769     echo -n $DEVPTR
770 }
771
772 mdsdevname() {
773     num=$1
774     DEVNAME=MDSDEV$num
775     #if $MDSDEVn isn't defined, default is $MDSDEVBASE + num
776     eval DEVPTR=${!DEVNAME:=${MDSDEVBASE}${num}}
777     echo -n $DEVPTR
778 }
779
780 ########
781 ## MountConf setup
782
783 stopall() {
784     # make sure we are using the primary server, so test-framework will
785     # be able to clean up properly.
786     activemds=`facet_active mds1`
787     if [ $activemds != "mds1" ]; then
788         fail mds1
789     fi
790     
791     # assume client mount is local 
792     grep " $MOUNT " /proc/mounts && zconf_umount `hostname` $MOUNT $*
793     grep " $MOUNT2 " /proc/mounts && zconf_umount `hostname` $MOUNT2 $*
794     for num in `seq $MDSCOUNT`; do
795         stop mds$num -f
796     done
797     for num in `seq $OSTCOUNT`; do
798         stop ost$num -f
799     done
800     return 0
801 }
802
803 cleanupall() {
804     stopall $*
805     unload_modules
806     cleanup_krb5_env
807 }
808
809 mdsmkfsopts()
810 {
811     local nr=$1
812     test $nr = 1 && echo -n $MDS_MKFS_OPTS || echo -n $MDSn_MKFS_OPTS
813 }
814
815 formatall() {
816     [ "$FSTYPE" ] && FSTYPE_OPT="--backfstype $FSTYPE"
817
818     stopall
819     # We need ldiskfs here, may as well load them all
820     load_modules
821     [ "$CLIENTONLY" ] && return
822     echo "Formatting mdts, osts"
823     for num in `seq $MDSCOUNT`; do
824         echo "Format mds$num: $(mdsdevname $num)"
825         if $VERBOSE; then
826             add mds$num `mdsmkfsopts $num` $FSTYPE_OPT --reformat `mdsdevname $num` || exit 9
827         else
828             add mds$num `mdsmkfsopts $num` $FSTYPE_OPT --reformat `mdsdevname $num` > /dev/null || exit 9
829         fi
830     done
831
832     for num in `seq $OSTCOUNT`; do
833         echo "Format ost$num: $(ostdevname $num)"
834         if $VERBOSE; then
835             add ost$num $OST_MKFS_OPTS --reformat `ostdevname $num` || exit 10
836         else
837             add ost$num $OST_MKFS_OPTS --reformat `ostdevname $num` > /dev/null || exit 10
838         fi
839     done
840 }
841
842 mount_client() {
843     grep " $1 " /proc/mounts || zconf_mount `hostname` $*
844 }
845
846 setupall() {
847     load_modules
848     init_krb5_env
849     if [ -z "$CLIENTONLY" ]; then
850         echo "Setup mdts, osts"
851         for num in `seq $MDSCOUNT`; do
852             DEVNAME=$(mdsdevname $num)
853             echo $REFORMAT | grep -q "reformat" \
854             || do_facet mds$num "$TUNEFS --writeconf $DEVNAME"
855             start mds$num $DEVNAME $MDS_MOUNT_OPTS
856         done
857         for num in `seq $OSTCOUNT`; do
858             DEVNAME=$(ostdevname $num)
859             start ost$num $DEVNAME $OST_MOUNT_OPTS
860         done
861     fi
862     [ "$DAEMONFILE" ] && $LCTL debug_daemon start $DAEMONFILE $DAEMONSIZE
863     mount_client $MOUNT
864     if [ "$MOUNT_2" ]; then
865         mount_client $MOUNT2
866     fi
867     sleep 5
868 }
869
870 mounted_lustre_filesystems() {
871         awk '($3 ~ "lustre" && $1 ~ ":") { print $2 }' /proc/mounts
872 }
873
874 check_and_setup_lustre() {
875     MOUNTED="`mounted_lustre_filesystems`"
876     if [ -z "$MOUNTED" ]; then
877         [ "$REFORMAT" ] && formatall
878         setupall
879         MOUNTED="`mounted_lustre_filesystems`"
880         [ -z "$MOUNTED" ] && error "NAME=$NAME not mounted"
881         export I_MOUNTED=yes
882     fi
883     if [ "$ONLY" == "setup" ]; then
884         exit 0
885     fi
886 }
887
888 cleanup_and_setup_lustre() {
889     if [ "$ONLY" == "cleanup" -o "`mount | grep $MOUNT`" ]; then
890         sysctl -w lnet.debug=0 || true
891         cleanupall
892         if [ "$ONLY" == "cleanup" ]; then 
893             exit 0
894         fi
895     fi
896     check_and_setup_lustre
897 }
898
899 check_and_cleanup_lustre() {
900     if [ "`mount | grep $MOUNT`" ]; then
901         rm -rf $DIR/[Rdfs][1-9]*
902         rm -f $DIR/${TESTSUITE}/[Rdfs][1-9]*
903     fi
904     if [ "$I_MOUNTED" = "yes" ]; then
905         cleanupall -f || error "cleanup failed"
906     fi
907     unset I_MOUNTED
908 }
909
910 ####### 
911 # General functions
912
913 check_network() {
914     local NETWORK=0
915     local WAIT=0
916     local MAX=$2
917     while [ $NETWORK -eq 0 ]; do
918         ping -c 1 -w 3 $1 > /dev/null
919         if [ $? -eq 0 ]; then
920             NETWORK=1
921         else
922             WAIT=$((WAIT + 5))
923             echo "waiting for $1, $((MAX - WAIT)) secs left"
924             sleep 5
925         fi
926         if [ $WAIT -gt $MAX ]; then
927             echo "Network not available"
928             exit 1
929         fi
930     done
931 }
932 check_port() {
933     while( !($DSH2 $1 "netstat -tna | grep -q $2") ) ; do
934         sleep 9
935     done
936 }
937
938 no_dsh() {
939     shift
940     eval $@
941 }
942
943 comma_list() {
944     # the sed converts spaces to commas, but leaves the last space
945     # alone, so the line doesn't end with a comma.
946     echo "$*" | tr -s " " "\n" | sort -b -u | tr "\n" " " | sed 's/ \([^$]\)/,\1/g'
947 }
948
949 absolute_path() {
950     (cd `dirname $1`; echo $PWD/`basename $1`)
951 }
952
953 ##################################
954 # OBD_FAIL funcs
955
956 drop_request() {
957 # OBD_FAIL_MDS_ALL_REQUEST_NET
958     RC=0
959     do_facet mds sysctl -w lustre.fail_loc=0x123
960     do_facet client "$1" || RC=$?
961     do_facet mds sysctl -w lustre.fail_loc=0
962     return $RC
963 }
964
965 drop_reply() {
966 # OBD_FAIL_MDS_ALL_REPLY_NET
967     RC=0
968     do_facet mds sysctl -w lustre.fail_loc=0x122
969     do_facet client "$@" || RC=$?
970     do_facet mds sysctl -w lustre.fail_loc=0
971     return $RC
972 }
973
974 drop_reint_reply() {
975 # OBD_FAIL_MDS_REINT_NET_REP
976     RC=0
977     do_facet mds sysctl -w lustre.fail_loc=0x119
978     do_facet client "$@" || RC=$?
979     do_facet mds sysctl -w lustre.fail_loc=0
980     return $RC
981 }
982
983 pause_bulk() {
984 #define OBD_FAIL_OST_BRW_PAUSE_BULK      0x214
985     RC=0
986     do_facet ost1 sysctl -w lustre.fail_loc=0x214
987     do_facet client "$1" || RC=$?
988     do_facet client "sync"
989     do_facet ost1 sysctl -w lustre.fail_loc=0
990     return $RC
991 }
992
993 drop_ldlm_cancel() {
994 #define OBD_FAIL_LDLM_CANCEL             0x304
995     RC=0
996     do_facet client sysctl -w lustre.fail_loc=0x304
997     do_facet client "$@" || RC=$?
998     do_facet client sysctl -w lustre.fail_loc=0
999     return $RC
1000 }
1001
1002 drop_bl_callback() {
1003 #define OBD_FAIL_LDLM_BL_CALLBACK        0x305
1004     RC=0
1005     do_facet client sysctl -w lustre.fail_loc=0x305
1006     do_facet client "$@" || RC=$?
1007     do_facet client sysctl -w lustre.fail_loc=0
1008     return $RC
1009 }
1010
1011 drop_ldlm_reply() {
1012 #define OBD_FAIL_LDLM_REPLY              0x30c
1013     RC=0
1014     do_facet mds sysctl -w lustre.fail_loc=0x30c
1015     do_facet client "$@" || RC=$?
1016     do_facet mds sysctl -w lustre.fail_loc=0
1017     return $RC
1018 }
1019
1020 clear_failloc() {
1021     facet=$1
1022     pause=$2
1023     sleep $pause
1024     echo "clearing fail_loc on $facet"
1025     do_facet $facet "sysctl -w lustre.fail_loc=0"
1026 }
1027
1028 cancel_lru_locks() {
1029     $LCTL mark "cancel_lru_locks $1 start"
1030     for d in `find $LPROC/ldlm/namespaces | egrep -i $1`; do
1031         [ -f $d/lru_size ] && echo clear > $d/lru_size
1032         [ -f $d/lock_unused_count ] && grep [1-9] $d/lock_unused_count /dev/null
1033     done
1034     $LCTL mark "cancel_lru_locks $1 stop"
1035 }
1036
1037
1038 pgcache_empty() {
1039     for a in /proc/fs/lustre/llite/*/dump_page_cache; do
1040         if [ `wc -l $a | awk '{print $1}'` -gt 1 ]; then
1041             echo there is still data in page cache $a ?
1042             cat $a;
1043             return 1;
1044         fi
1045     done
1046     return 0
1047 }
1048
1049 debugsave() {
1050     DEBUGSAVE="$(sysctl -n lnet.debug)"
1051 }
1052
1053 debugrestore() {
1054     [ -n "$DEBUGSAVE" ] && sysctl -w lnet.debug="${DEBUGSAVE}"
1055     DEBUGSAVE=""
1056 }
1057
1058 ##################################
1059 # Test interface 
1060 ##################################
1061
1062 error() {
1063     local FAIL_ON_ERROR=${FAIL_ON_ERROR:-true}
1064     local TYPE=${TYPE:-"FAIL"}
1065     local ERRLOG
1066     sysctl -w lustre.fail_loc=0 2> /dev/null || true
1067     log "${TESTSUITE} ${TESTNAME}: **** ${TYPE}:" $@
1068     ERRLOG=$TMP/lustre_${TESTSUITE}_${TESTNAME}.$(date +%s)
1069     echo "Dumping lctl log to $ERRLOG"
1070     # We need to dump the logs on all nodes
1071     $LCTL dk $ERRLOG
1072     [ ! "$mds_HOST" = "$(hostname)" ] && do_node $mds_HOST $LCTL dk $ERRLOG
1073     [ ! "$ost_HOST" = "$(hostname)" -a ! "$ost_HOST" = "$mds_HOST" ] && do_node $ost_HOST $LCTL dk $ERRLOG
1074     debugrestore
1075     [ "$TESTSUITELOG" ] && echo "$0: ${TYPE}: $TESTNAME $@" >> $TESTSUITELOG
1076     if $FAIL_ON_ERROR; then
1077         exit 1
1078     fi
1079 }
1080
1081 # use only if we are ignoring failures for this test, bugno required.
1082 # (like ALWAYS_EXCEPT, but run the test and ignore the results.)
1083 # e.g. error_ignore 5494 "your message"
1084 error_ignore() {
1085     FAIL_ON_ERROR=false TYPE="IGNORE (bz$1)" error $2
1086 }
1087
1088 skip () {
1089         log " SKIP: ${TESTSUITE} ${TESTNAME} $@"
1090         [ "$TESTSUITELOG" ] && echo "${TESTSUITE}: SKIP: $TESTNAME $@" >> $TESTSUITELOG
1091 }
1092
1093 build_test_filter() {
1094     [ "$ONLY" ] && log "only running test `echo $ONLY`"
1095     for O in $ONLY; do
1096         eval ONLY_${O}=true
1097     done
1098     [ "$EXCEPT$ALWAYS_EXCEPT" ] && \
1099         log "skipping tests: `echo $EXCEPT $ALWAYS_EXCEPT`"
1100     for E in $EXCEPT $ALWAYS_EXCEPT; do
1101         eval EXCEPT_${E}=true
1102     done
1103     for G in $GRANT_CHECK_LIST; do
1104         eval GCHECK_ONLY_${G}=true
1105         done
1106 }
1107
1108 _basetest() {
1109     echo $*
1110 }
1111
1112 basetest() {
1113     IFS=abcdefghijklmnopqrstuvwxyz _basetest $1
1114 }
1115
1116 run_test() {
1117     export base=`basetest $1`
1118     if [ ! -z "$ONLY" ]; then
1119         testname=ONLY_$1
1120         if [ ${!testname}x != x ]; then
1121             run_one $1 "$2"
1122             return $?
1123         fi
1124         testname=ONLY_$base
1125         if [ ${!testname}x != x ]; then
1126             run_one $1 "$2"
1127             return $?
1128         fi
1129         echo -n "."
1130         return 0
1131     fi
1132     testname=EXCEPT_$1
1133     if [ ${!testname}x != x ]; then
1134         TESTNAME=test_$1 skip "skipping excluded test $1"
1135         return 0
1136     fi
1137     testname=EXCEPT_$base
1138     if [ ${!testname}x != x ]; then
1139         TESTNAME=test_$1 skip "skipping excluded test $1 (base $base)"
1140         return 0
1141     fi
1142     run_one $1 "$2"
1143     
1144     return $?
1145 }
1146
1147 EQUALS="======================================================================"
1148 equals_msg() {
1149     msg="$@"
1150
1151     local suffixlen=$((${#EQUALS} - ${#msg}))
1152     [ $suffixlen -lt 5 ] && suffixlen=5
1153     log `echo $(printf '===== %s %.*s\n' "$msg" $suffixlen $EQUALS)`
1154 }
1155
1156 log() {
1157     echo "$*"
1158     lsmod | grep lnet > /dev/null || load_modules
1159     $LCTL mark "$*" 2> /dev/null || true
1160 }
1161
1162 trace() {
1163         log "STARTING: $*"
1164         strace -o $TMP/$1.strace -ttt $*
1165         RC=$?
1166         log "FINISHED: $*: rc $RC"
1167         return 1
1168 }
1169
1170 pass() {
1171     echo PASS $@
1172 }
1173
1174 check_mds() {
1175     FFREE=`cat /proc/fs/lustre/mds/*/filesfree`
1176     FTOTAL=`cat /proc/fs/lustre/mds/*/filestotal`
1177     [ $FFREE -ge $FTOTAL ] && error "files free $FFREE > total $FTOTAL" || true
1178 }
1179
1180 run_one() {
1181     testnum=$1
1182     message=$2
1183     tfile=f${testnum}
1184     export tdir=d${TESTSUITE}/d${base}
1185     local SAVE_UMASK=`umask`
1186     umask 0022
1187     mkdir -p $DIR/$tdir
1188
1189     BEFORE=`date +%s`
1190     log "== test $testnum: $message ============ `date +%H:%M:%S` ($BEFORE)"
1191     #check_mds
1192     export TESTNAME=test_$testnum
1193     test_${testnum} || error "test_$testnum failed with $?"
1194     #check_mds
1195     check_grant ${testnum} || error "check_grant $testnum failed with $?"
1196     [ -f $CATASTROPHE ] && [ `cat $CATASTROPHE` -ne 0 ] && \
1197         error "LBUG/LASSERT detected"
1198     pass "($((`date +%s` - $BEFORE))s)"
1199     rmdir ${DIR}/$tdir >/dev/null 2>&1 || true
1200     unset TESTNAME
1201     unset tdir
1202     umask $SAVE_UMASK
1203     cd $SAVE_PWD
1204     $CLEANUP
1205 }
1206
1207 canonical_path() {
1208     (cd `dirname $1`; echo $PWD/`basename $1`)
1209 }
1210
1211 sync_clients() {
1212     [ -d $DIR1 ] && cd $DIR1 && sync; sleep 1; sync 
1213     [ -d $DIR2 ] && cd $DIR2 && sync; sleep 1; sync 
1214         cd $SAVE_PWD
1215 }
1216
1217 check_grant() {
1218     export base=`basetest $1`
1219     [ "$CHECK_GRANT" == "no" ] && return 0
1220
1221         testname=GCHECK_ONLY_${base}
1222         [ ${!testname}x == x ] && return 0
1223
1224         echo -n "checking grant......"
1225         cd $SAVE_PWD
1226         # write some data to sync client lost_grant
1227         rm -f $DIR1/${tfile}_check_grant_* 2>&1
1228         for i in `seq $OSTCOUNT`; do
1229                 $LFS setstripe $DIR1/${tfile}_check_grant_$i 0 $(($i -1)) 1
1230                 dd if=/dev/zero of=$DIR1/${tfile}_check_grant_$i bs=4k \
1231                                               count=1 > /dev/null 2>&1 
1232         done
1233         # sync all the data and make sure no pending data on server
1234         sync_clients
1235         
1236         #get client grant and server grant 
1237         client_grant=0
1238     for d in ${LPROC}/osc/*/cur_grant_bytes; do 
1239                 client_grant=$((client_grant + `cat $d`))
1240         done
1241         server_grant=0
1242         for d in ${LPROC}/obdfilter/*/tot_granted; do
1243                 server_grant=$((server_grant + `cat $d`))
1244         done
1245
1246         # cleanup the check_grant file
1247         for i in `seq $OSTCOUNT`; do
1248                 rm $DIR1/${tfile}_check_grant_$i
1249         done
1250
1251         #check whether client grant == server grant 
1252         if [ $client_grant != $server_grant ]; then
1253                 echo "failed: client:${client_grant} server: ${server_grant}"
1254                 return 1
1255         else
1256                 echo "pass"
1257         fi
1258 }
1259
1260 ########################
1261 # helper functions
1262
1263 osc_to_ost()
1264 {
1265     osc=$1
1266     ost=`echo $1 | awk -F_ '{print $3}'`
1267     if [ -z $ost ]; then
1268         ost=`echo $1 | sed 's/-osc.*//'`
1269     fi
1270     echo $ost
1271 }
1272
1273 remote_mds ()
1274 {
1275     [ ! -e /proc/fs/lustre/mdt/*MDT* ]
1276 }
1277
1278 remote_ost ()
1279 {
1280     [ $(grep -c obdfilter $LPROC/devices) -eq 0 ]
1281 }
1282
1283 is_patchless ()
1284 {
1285     grep -q patchless $LPROC/version
1286 }