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