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