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 client_df() {
 491     # not every config has many clients
 492     if [ ! -z "$CLIENTS" ]; then
 493         $PDSH $CLIENTS "df $MOUNT" > /dev/null
 494     fi
 495 }
 496
 497 client_reconnect() {
 498     uname -n >> $MOUNT/recon
 499     if [ ! -z "$CLIENTS" ]; then
 500         $PDSH $CLIENTS "df $MOUNT; uname -n >> $MOUNT/recon" > /dev/null
 501     fi
 502     echo Connected clients:
 503     cat $MOUNT/recon
 504     ls -l $MOUNT/recon > /dev/null
 505     rm $MOUNT/recon
 506 }
 507
 508 facet_failover() {
 509     facet=$1
 510     echo "Failing $facet on node `facet_active_host $facet`"
 511     shutdown_facet $facet
 512     reboot_facet $facet
 513     client_df &
 514     DFPID=$!
 515     echo "df pid is $DFPID"
 516     change_active $facet
 517     TO=`facet_active_host $facet`
 518     echo "Failover $facet to $TO"
 519     wait_for $facet
 520     local dev=${facet}_dev
 521     local opt=${facet}_opt
 522     start $facet ${!dev} ${!opt} || error "Restart of $facet failed"
 523 }
 524
 525 obd_name() {
 526     local facet=$1
 527 }
 528
 529 replay_barrier() {
 530     local facet=$1
 531     do_facet $facet sync
 532     df $MOUNT
 533     local svc=${facet}_svc
 534     do_facet $facet $LCTL --device %${!svc} readonly
 535     do_facet $facet $LCTL --device %${!svc} notransno
 536     do_facet $facet $LCTL mark "$facet REPLAY BARRIER on ${!svc}"
 537     $LCTL mark "local REPLAY BARRIER on ${!svc}"
 538 }
 539
 540 replay_barrier_nodf() {
 541     local facet=$1    echo running=${running}
 542     do_facet $facet sync
 543     local svc=${facet}_svc
 544     echo Replay barrier on ${!svc}
 545     do_facet $facet $LCTL --device %${!svc} readonly
 546     do_facet $facet $LCTL --device %${!svc} notransno
 547     do_facet $facet $LCTL mark "$facet REPLAY BARRIER on ${!svc}"
 548     $LCTL mark "local REPLAY BARRIER on ${!svc}"
 549 }
 550
 551 mds_evict_client() {
 552     UUID=`cat /proc/fs/lustre/mdc/${mds1_svc}-mdc-*/uuid`
 553     do_facet mds1 "echo $UUID > /proc/fs/lustre/mdt/${mds1_svc}/evict_client"
 554 }
 555
 556 ost_evict_client() {
 557     UUID=`grep ${ost1_svc}-osc- $LPROC/devices | egrep -v 'MDT' | awk '{print $5}'`
 558     do_facet ost1 "echo $UUID > /proc/fs/lustre/obdfilter/${ost1_svc}/evict_client"
 559 }
 560
 561 fail() {
 562     facet_failover $* || error "failover: $?"
 563     df $MOUNT || error "post-failover df: $?"
 564 }
 565
 566 fail_abort() {
 567     local facet=$1
 568     stop $facet
 569     change_active $facet
 570     local svc=${facet}_svc
 571     local dev=${facet}_dev
 572     local opt=${facet}_opt
 573     start $facet ${!dev} ${!opt}
 574     do_facet $facet lctl --device %${!svc} abort_recovery
 575     df $MOUNT || echo "first df failed: $?"
 576     sleep 1
 577     df $MOUNT || error "post-failover df: $?"
 578 }
 579
 580 do_lmc() {
 581     echo There is no lmc.  This is mountconf, baby.
 582     exit 1
 583 }
 584
 585 h2gm () {
 586     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
 587         ID=`$PDSH $1 $GMNALNID -l | cut -d\  -f2`
 588         echo $ID"@gm"
 589     fi
 590 }
 591
 592 h2name_or_ip() {
 593     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
 594         echo $1"@$2"
 595     fi
 596 }
 597
 598 h2ptl() {
 599    if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
 600        ID=`xtprocadmin -n $1 2>/dev/null | egrep -v 'NID' | awk '{print $1}'`
 601        if [ -z "$ID" ]; then
 602            echo "Could not get a ptl id for $1..."
 603            exit 1
 604        fi
 605        echo $ID"@ptl"
 606    fi
 607 }
 608 declare -fx h2ptl
 609
 610 h2tcp() {
 611     h2name_or_ip "$1" "tcp"
 612 }
 613 declare -fx h2tcp
 614
 615 h2elan() {
 616     if [ "$1" = "client" -o "$1" = "'*'" ]; then echo \'*\'; else
 617         if type __h2elan >/dev/null 2>&1; then
 618             ID=$(__h2elan $1)
 619         else
 620             ID=`echo $1 | sed 's/[^0-9]*//g'`
 621         fi
 622         echo $ID"@elan"
 623     fi
 624 }
 625 declare -fx h2elan
 626
 627 h2openib() {
 628     h2name_or_ip "$1" "openib"
 629 }
 630 declare -fx h2openib
 631
 632 h2o2ib() {
 633     h2name_or_ip "$1" "o2ib"
 634 }
 635 declare -fx h2o2ib
 636
 637 facet_host() {
 638     local facet=$1
 639     varname=${facet}_HOST
 640     if [ -z "${!varname}" ]; then
 641         if [ "${facet:0:3}" == "ost" ]; then
 642             eval ${facet}_HOST=${ost_HOST}
 643         fi
 644     fi
 645     echo -n ${!varname}
 646 }
 647
 648 facet_active() {
 649     local facet=$1
 650     local activevar=${facet}active
 651
 652     if [ -f ./${facet}active ] ; then
 653         source ./${facet}active
 654     fi
 655
 656     active=${!activevar}
 657     if [ -z "$active" ] ; then
 658         echo -n ${facet}
 659     else
 660         echo -n ${active}
 661     fi
 662 }
 663
 664 facet_active_host() {
 665     local facet=$1
 666     local active=`facet_active $facet`
 667     if [ "$facet" == client ]; then
 668         hostname
 669     else
 670         echo `facet_host $active`
 671     fi
 672 }
 673
 674 change_active() {
 675     local facet=$1
 676     failover=${facet}failover
 677     host=`facet_host $failover`
 678     [ -z "$host" ] && return
 679     curactive=`facet_active $facet`
 680     if [ -z "${curactive}" -o "$curactive" == "$failover" ] ; then
 681         eval export ${facet}active=$facet
 682     else
 683         eval export ${facet}active=$failover
 684     fi
 685     # save the active host for this facet
 686     activevar=${facet}active
 687     echo "$activevar=${!activevar}" > ./$activevar
 688 }
 689
 690 do_node() {
 691     HOST=$1
 692     shift
 693     local myPDSH=$PDSH
 694     if [ "$HOST" = "$(hostname)" ]; then
 695         myPDSH="no_dsh"
 696     fi
 697     if $VERBOSE; then
 698         echo "CMD: $HOST $@" >&2
 699         $myPDSH $HOST $LCTL mark "$@" > /dev/null 2>&1 || :
 700     fi
 701     $myPDSH $HOST "(PATH=\$PATH:$RLUSTRE/utils:$RLUSTRE/tests:/sbin:/usr/sbin; cd $RPWD; sh -c \"$@\")" | sed "s/^${HOST}: //"
 702     return ${PIPESTATUS[0]}
 703 }
 704
 705 do_facet() {
 706     facet=$1
 707     shift
 708     HOST=`facet_active_host $facet`
 709     [ -z $HOST ] && echo No host defined for facet ${facet} && exit 1
 710     do_node $HOST $@
 711 }
 712
 713 add() {
 714     local facet=$1
 715     shift
 716     # make sure its not already running
 717     stop ${facet} -f
 718     rm -f ${facet}active
 719     do_facet ${facet} $MKFS $*
 720 }
 721
 722 ostdevname() {
 723     num=$1
 724     DEVNAME=OSTDEV$num
 725     #if $OSTDEVn isn't defined, default is $OSTDEVBASE + num
 726     eval DEVPTR=${!DEVNAME:=${OSTDEVBASE}${num}}
 727     echo -n $DEVPTR
 728 }
 729
 730 mdsdevname() {
 731     num=$1
 732     DEVNAME=MDSDEV$num
 733     #if $MDSDEVn isn't defined, default is $MDSDEVBASE + num
 734     eval DEVPTR=${!DEVNAME:=${MDSDEVBASE}${num}}
 735     echo -n $DEVPTR
 736 }
 737
 738 ########
 739 ## MountConf setup
 740
 741 stopall() {
 742     # make sure we are using the primary server, so test-framework will
 743     # be able to clean up properly.
 744     activemds=`facet_active mds1`
 745     if [ $activemds != "mds1" ]; then
 746         fail mds1
 747     fi
 748
 749     # assume client mount is local
 750     grep " $MOUNT " /proc/mounts && zconf_umount `hostname` $MOUNT $*
 751     grep " $MOUNT2 " /proc/mounts && zconf_umount `hostname` $MOUNT2 $*
 752     for num in `seq $MDSCOUNT`; do
 753         stop mds$num -f
 754     done
 755     for num in `seq $OSTCOUNT`; do
 756         stop ost$num -f
 757     done
 758     return 0
 759 }
 760
 761 cleanupall() {
 762     stopall $*
 763     unload_modules
 764     cleanup_krb5_env
 765 }
 766
 767 mdsmkfsopts()
 768 {
 769     local nr=$1
 770     test $nr = 1 && echo -n $MDS_MKFS_OPTS || echo -n $MDSn_MKFS_OPTS
 771 }
 772
 773 formatall() {
 774     [ "$FSTYPE" ] && FSTYPE_OPT="--backfstype $FSTYPE"
 775
 776     stopall
 777     # We need ldiskfs here, may as well load them all
 778     load_modules
 779     [ "$CLIENTONLY" ] && return
 780     echo "Formatting mdts, osts"
 781     for num in `seq $MDSCOUNT`; do
 782         echo "Format mds$num: $(mdsdevname $num)"
 783         if $VERBOSE; then
 784             add mds$num `mdsmkfsopts $num` $FSTYPE_OPT --reformat `mdsdevname $num` || exit 9
 785         else
 786             add mds$num `mdsmkfsopts $num` $FSTYPE_OPT --reformat `mdsdevname $num` > /dev/null || exit 9
 787         fi
 788     done
 789
 790     for num in `seq $OSTCOUNT`; do
 791         echo "Format ost$num: $(ostdevname $num)"
 792         if $VERBOSE; then
 793             add ost$num $OST_MKFS_OPTS --reformat `ostdevname $num` || exit 10
 794         else
 795             add ost$num $OST_MKFS_OPTS --reformat `ostdevname $num` > /dev/null || exit 10
 796         fi
 797     done
 798 }
 799
 800 mount_client() {
 801     grep " $1 " /proc/mounts || zconf_mount `hostname` $*
 802 }
 803
 804 setupall() {
 805     load_modules
 806     init_krb5_env
 807     if [ -z "$CLIENTONLY" ]; then
 808         echo "Setup mdts, osts"
 809         for num in `seq $MDSCOUNT`; do
 810             DEVNAME=$(mdsdevname $num)
 811             echo $REFORMAT | grep -q "reformat" \
 812             || do_facet mds$num "$TUNEFS --writeconf $DEVNAME"
 813             start mds$num $DEVNAME $MDS_MOUNT_OPTS
 814         done
 815         for num in `seq $OSTCOUNT`; do
 816             DEVNAME=$(ostdevname $num)
 817             start ost$num $DEVNAME $OST_MOUNT_OPTS
 818         done
 819     fi
 820     [ "$DAEMONFILE" ] && $LCTL debug_daemon start $DAEMONFILE $DAEMONSIZE
 821     mount_client $MOUNT
 822     if [ "$MOUNT_2" ]; then
 823         mount_client $MOUNT2
 824     fi
 825     sleep 5
 826 }
 827
 828 mounted_lustre_filesystems() {
 829         awk '($3 ~ "lustre" && $1 ~ ":") { print $2 }' /proc/mounts
 830 }
 831
 832 check_and_setup_lustre() {
 833     MOUNTED="`mounted_lustre_filesystems`"
 834     if [ -z "$MOUNTED" ]; then
 835         [ "$REFORMAT" ] && formatall
 836         setupall
 837         MOUNTED="`mounted_lustre_filesystems`"
 838         [ -z "$MOUNTED" ] && error "NAME=$NAME not mounted"
 839         export I_MOUNTED=yes
 840     fi
 841     if [ "$ONLY" == "setup" ]; then
 842         exit 0
 843     fi
 844 }
 845
 846 cleanup_and_setup_lustre() {
 847     if [ "$ONLY" == "cleanup" -o "`mount | grep $MOUNT`" ]; then
 848         sysctl -w lnet.debug=0 || true
 849         cleanupall
 850         if [ "$ONLY" == "cleanup" ]; then
 851             exit 0
 852         fi
 853     fi
 854     check_and_setup_lustre
 855 }
 856
 857 check_and_cleanup_lustre() {
 858     if [ "`mount | grep $MOUNT`" ]; then
 859         rm -rf $DIR/[Rdfs][1-9]*
 860     fi
 861     if [ "$I_MOUNTED" = "yes" ]; then
 862         cleanupall -f || error "cleanup failed"
 863     fi
 864     unset I_MOUNTED
 865 }
 866
 867 #######
 868 # General functions
 869
 870 check_network() {
 871     local NETWORK=0
 872     local WAIT=0
 873     local MAX=$2
 874     while [ $NETWORK -eq 0 ]; do
 875         ping -c 1 -w 3 $1 > /dev/null
 876         if [ $? -eq 0 ]; then
 877             NETWORK=1
 878         else
 879             WAIT=$((WAIT + 5))
 880             echo "waiting for $1, $((MAX - WAIT)) secs left"
 881             sleep 5
 882         fi
 883         if [ $WAIT -gt $MAX ]; then
 884             echo "Network not available"
 885             exit 1
 886         fi
 887     done
 888 }
 889 check_port() {
 890     while( !($DSH2 $1 "netstat -tna | grep -q $2") ) ; do
 891         sleep 9
 892     done
 893 }
 894
 895 no_dsh() {
 896     shift
 897     eval $@
 898 }
 899
 900 comma_list() {
 901     # the sed converts spaces to commas, but leaves the last space
 902     # alone, so the line doesn't end with a comma.
 903     echo "$*" | tr -s " " "\n" | sort -b -u | tr "\n" " " | sed 's/ \([^$]\)/,\1/g'
 904 }
 905
 906 absolute_path() {
 907     (cd `dirname $1`; echo $PWD/`basename $1`)
 908 }
 909
 910 ##################################
 911 # OBD_FAIL funcs
 912
 913 drop_request() {
 914 # OBD_FAIL_MDS_ALL_REQUEST_NET
 915     RC=0
 916     do_facet mds sysctl -w lustre.fail_loc=0x123
 917     do_facet client "$1" || RC=$?
 918     do_facet mds sysctl -w lustre.fail_loc=0
 919     return $RC
 920 }
 921
 922 drop_reply() {
 923 # OBD_FAIL_MDS_ALL_REPLY_NET
 924     RC=0
 925     do_facet mds sysctl -w lustre.fail_loc=0x122
 926     do_facet client "$@" || RC=$?
 927     do_facet mds sysctl -w lustre.fail_loc=0
 928     return $RC
 929 }
 930
 931 drop_reint_reply() {
 932 # OBD_FAIL_MDS_REINT_NET_REP
 933     RC=0
 934     do_facet mds sysctl -w lustre.fail_loc=0x119
 935     do_facet client "$@" || RC=$?
 936     do_facet mds sysctl -w lustre.fail_loc=0
 937     return $RC
 938 }
 939
 940 pause_bulk() {
 941 #define OBD_FAIL_OST_BRW_PAUSE_BULK      0x214
 942     RC=0
 943     do_facet ost1 sysctl -w lustre.fail_loc=0x214
 944     do_facet client "$1" || RC=$?
 945     do_facet client "sync"
 946     do_facet ost1 sysctl -w lustre.fail_loc=0
 947     return $RC
 948 }
 949
 950 drop_ldlm_cancel() {
 951 #define OBD_FAIL_LDLM_CANCEL             0x304
 952     RC=0
 953     do_facet client sysctl -w lustre.fail_loc=0x304
 954     do_facet client "$@" || RC=$?
 955     do_facet client sysctl -w lustre.fail_loc=0
 956     return $RC
 957 }
 958
 959 drop_bl_callback() {
 960 #define OBD_FAIL_LDLM_BL_CALLBACK        0x305
 961     RC=0
 962     do_facet client sysctl -w lustre.fail_loc=0x305
 963     do_facet client "$@" || RC=$?
 964     do_facet client sysctl -w lustre.fail_loc=0
 965     return $RC
 966 }
 967
 968 drop_ldlm_reply() {
 969 #define OBD_FAIL_LDLM_REPLY              0x30c
 970     RC=0
 971     do_facet mds sysctl -w lustre.fail_loc=0x30c
 972     do_facet client "$@" || RC=$?
 973     do_facet mds sysctl -w lustre.fail_loc=0
 974     return $RC
 975 }
 976
 977 clear_failloc() {
 978     facet=$1
 979     pause=$2
 980     sleep $pause
 981     echo "clearing fail_loc on $facet"
 982     do_facet $facet "sysctl -w lustre.fail_loc=0"
 983 }
 984
 985 cancel_lru_locks() {
 986     $LCTL mark "cancel_lru_locks $1 start"
 987     for d in `find $LPROC/ldlm/namespaces | egrep -i $1`; do
 988         [ -f $d/lru_size ] && echo clear > $d/lru_size
 989         [ -f $d/lock_unused_count ] && grep [1-9] $d/lock_unused_count /dev/null
 990     done
 991     $LCTL mark "cancel_lru_locks $1 stop"
 992 }
 993
 994
 995 pgcache_empty() {
 996     for a in /proc/fs/lustre/llite/*/dump_page_cache; do
 997         if [ `wc -l $a | awk '{print $1}'` -gt 1 ]; then
 998             echo there is still data in page cache $a ?
 999             cat $a;
1000             return 1;
1001         fi
1002     done
1003     return 0
1004 }
1005
1006 debugsave() {
1007     DEBUGSAVE="$(sysctl -n lnet.debug)"
1008 }
1009
1010 debugrestore() {
1011     [ -n "$DEBUGSAVE" ] && sysctl -w lnet.debug="${DEBUGSAVE}"
1012     DEBUGSAVE=""
1013 }
1014
1015 FAIL_ON_ERROR=true
1016 ##################################
1017 # Test interface
1018 error() {
1019     local ERRLOG
1020     sysctl -w lustre.fail_loc=0 2> /dev/null || true
1021     log "${TESTSUITE} ${TESTNAME}: **** FAIL:" $@
1022     ERRLOG=$TMP/lustre_${TESTSUITE}_${TESTNAME}.$(date +%s)
1023     echo "Dumping lctl log to $ERRLOG"
1024     # We need to dump the logs on all nodes
1025     $LCTL dk $ERRLOG
1026     [ ! "$mds_HOST" = "$(hostname)" ] && do_node $mds_HOST $LCTL dk $ERRLOG
1027     [ ! "$ost_HOST" = "$(hostname)" -a ! "$ost_HOST" = "$mds_HOST" ] && do_node $ost_HOST $LCTL dk $ERRLOG
1028     debugrestore
1029     [ "$TESTSUITELOG" ] && echo "$0: FAIL: $TESTNAME $@" >> $TESTSUITELOG
1030     if $FAIL_ON_ERROR; then
1031         exit 1
1032     fi
1033 }
1034
1035 skip () {
1036         log " SKIP: ${TESTSUITE} ${TESTNAME} $@"
1037         [ "$TESTSUITELOG" ] && echo "${TESTSUITE}: SKIP: $TESTNAME $@" >> $TESTSUITELOG
1038 }
1039
1040 build_test_filter() {
1041     [ "$ONLY" ] && log "only running test `echo $ONLY`"
1042     for O in $ONLY; do
1043         eval ONLY_${O}=true
1044     done
1045     [ "$EXCEPT$ALWAYS_EXCEPT" ] && \
1046         log "skipping tests: `echo $EXCEPT $ALWAYS_EXCEPT`"
1047     for E in $EXCEPT $ALWAYS_EXCEPT; do
1048         eval EXCEPT_${E}=true
1049     done
1050     for G in $GRANT_CHECK_LIST; do
1051         eval GCHECK_ONLY_${G}=true
1052         done
1053 }
1054
1055 _basetest() {
1056     echo $*
1057 }
1058
1059 basetest() {
1060     IFS=abcdefghijklmnopqrstuvwxyz _basetest $1
1061 }
1062
1063 run_test() {
1064     export base=`basetest $1`
1065     if [ ! -z "$ONLY" ]; then
1066         testname=ONLY_$1
1067         if [ ${!testname}x != x ]; then
1068             run_one $1 "$2"
1069             return $?
1070         fi
1071         testname=ONLY_$base
1072         if [ ${!testname}x != x ]; then
1073             run_one $1 "$2"
1074             return $?
1075         fi
1076         echo -n "."
1077         return 0
1078     fi
1079     testname=EXCEPT_$1
1080     if [ ${!testname}x != x ]; then
1081         TESTNAME=test_$1 skip "skipping excluded test $1"
1082         return 0
1083     fi
1084     testname=EXCEPT_$base
1085     if [ ${!testname}x != x ]; then
1086         TESTNAME=test_$1 skip "skipping excluded test $1 (base $base)"
1087         return 0
1088     fi
1089     run_one $1 "$2"
1090
1091     return $?
1092 }
1093
1094 EQUALS="======================================================================"
1095 equals_msg() {
1096     msg="$@"
1097
1098     local suffixlen=$((${#EQUALS} - ${#msg}))
1099     [ $suffixlen -lt 5 ] && suffixlen=5
1100     printf '===== %s %.*s\n' "$msg" $suffixlen $EQUALS
1101 }
1102
1103 log() {
1104     echo "$*"
1105     lsmod | grep lnet > /dev/null || load_modules
1106     $LCTL mark "$*" 2> /dev/null || true
1107 }
1108
1109 trace() {
1110         log "STARTING: $*"
1111         strace -o $TMP/$1.strace -ttt $*
1112         RC=$?
1113         log "FINISHED: $*: rc $RC"
1114         return 1
1115 }
1116
1117 pass() {
1118     echo PASS $@
1119 }
1120
1121 check_mds() {
1122     FFREE=`cat /proc/fs/lustre/mds/*/filesfree`
1123     FTOTAL=`cat /proc/fs/lustre/mds/*/filestotal`
1124     [ $FFREE -ge $FTOTAL ] && error "files free $FFREE > total $FTOTAL" || true
1125 }
1126
1127 run_one() {
1128     testnum=$1
1129     message=$2
1130     tfile=f${testnum}
1131     export tdir=d${base}
1132
1133     BEFORE=`date +%s`
1134     log "== test $testnum: $message ============ `date +%H:%M:%S` ($BEFORE)"
1135     #check_mds
1136     export TESTNAME=test_$testnum
1137     test_${testnum} || error "test_$testnum failed with $?"
1138     #check_mds
1139     check_grant ${testnum} || error "check_grant $testnum failed with $?"
1140     [ -f $CATASTROPHE ] && [ `cat $CATASTROPHE` -ne 0 ] && \
1141         error "LBUG/LASSERT detected"
1142     pass "($((`date +%s` - $BEFORE))s)"
1143     unset TESTNAME
1144     unset tdir
1145     cd $SAVE_PWD
1146     $CLEANUP
1147 }
1148
1149 canonical_path() {
1150     (cd `dirname $1`; echo $PWD/`basename $1`)
1151 }
1152
1153 sync_clients() {
1154     [ -d $DIR1 ] && cd $DIR1 && sync; sleep 1; sync
1155     [ -d $DIR2 ] && cd $DIR2 && sync; sleep 1; sync
1156         cd $SAVE_PWD
1157 }
1158
1159 check_grant() {
1160     export base=`basetest $1`
1161     [ "$CHECK_GRANT" == "no" ] && return 0
1162
1163         testname=GCHECK_ONLY_${base}
1164         [ ${!testname}x == x ] && return 0
1165
1166         echo -n "checking grant......"
1167         cd $SAVE_PWD
1168         # write some data to sync client lost_grant
1169         rm -f $DIR1/${tfile}_check_grant_* 2>&1
1170         for i in `seq $OSTCOUNT`; do
1171                 $LFS setstripe $DIR1/${tfile}_check_grant_$i 0 $(($i -1)) 1
1172                 dd if=/dev/zero of=$DIR1/${tfile}_check_grant_$i bs=4k \
1173                                               count=1 > /dev/null 2>&1
1174         done
1175         # sync all the data and make sure no pending data on server
1176         sync_clients
1177
1178         #get client grant and server grant
1179         client_grant=0
1180     for d in ${LPROC}/osc/*/cur_grant_bytes; do
1181                 client_grant=$((client_grant + `cat $d`))
1182         done
1183         server_grant=0
1184         for d in ${LPROC}/obdfilter/*/tot_granted; do
1185                 server_grant=$((server_grant + `cat $d`))
1186         done
1187
1188         # cleanup the check_grant file
1189         for i in `seq $OSTCOUNT`; do
1190                 rm $DIR1/${tfile}_check_grant_$i
1191         done
1192
1193         #check whether client grant == server grant
1194         if [ $client_grant != $server_grant ]; then
1195                 echo "failed: client:${client_grant} server: ${server_grant}"
1196                 return 1
1197         else
1198                 echo "pass"
1199         fi
1200 }
1201
1202 ########################
1203 # helper functions
1204
1205 osc_to_ost()
1206 {
1207     osc=$1
1208     ost=`echo $1 | awk -F_ '{print $3}'`
1209     if [ -z $ost ]; then
1210         ost=`echo $1 | sed 's/-osc.*//'`
1211     fi
1212     echo $ost
1213 }
1214
1215 remote_mds ()
1216 {
1217     [ ! -e /proc/fs/lustre/mdt/*MDT* ]
1218 }
1219
1220 remote_ost ()
1221 {
1222     [ $(grep -c obdfilter $LPROC/devices) -eq 0 ]
1223 }
1224
1225 is_patchless ()
1226 {
1227     grep -q patchless $LPROC/version
1228 }