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