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