lustre/tests/functions.sh

   1 #!/bin/bash
   2 # -*- mode: Bash; tab-width: 4; indent-tabs-mode: t; -*-
   3 # vim:shiftwidth=4:softtabstop=4:tabstop=4:
   4
   5 # Simple function used by run_*.sh scripts
   6
   7 assert_env() {
   8     local failed=""
   9     for name in $@; do
  10         if [ -z "${!name}" ]; then
  11             echo "$0: $name must be set"
  12             failed=1
  13         fi
  14     done
  15     [ $failed ] && exit 1 || true
  16 }
  17
  18 # lrepl - Lustre test Read-Eval-Print Loop.
  19 #
  20 # This function implements a REPL for the Lustre test framework.  It
  21 # doesn't exec an actual shell because the user may want to inspect
  22 # variables and use functions from the test framework.
  23 lrepl() {
  24     local line
  25     local rawline
  26     local prompt
  27
  28     cat <<EOF
  29         This is an interactive read-eval-print loop interactive shell
  30         simulation that you can use to debug failing tests.  You can
  31         enter most bash command lines (see notes below).
  32
  33         Use this REPL to inspect variables, set them, call test
  34         framework shell functions, etcetera.
  35
  36         'exit' or EOF to exit this shell.
  37
  38         set \$retcode to 0 to cause the assertion failure that
  39         triggered this REPL to be ignored.
  40
  41         Examples:
  42             do_facet ost1 lctl get_param ost.*.ost.threads_*
  43             do_rpc_nodes \$OSTNODES unload_modules
  44
  45         NOTES:
  46             All but the last line of multi-line statements or blocks
  47             must end in a backslash.
  48
  49             "Here documents" are not supported.
  50
  51             History is not supported, but command-line editing is.
  52
  53 EOF
  54
  55     # Prompt escapes don't work in read -p, sadly.
  56     prompt=":test_${testnum:-UNKNOWN}:$(uname -n):$(basename $PWD)% "
  57
  58     # We use read -r to get close to a shell experience
  59     while read -e -r -p "$prompt" rawline; do
  60         line=
  61         case "$rawline" in
  62         # Don't want to exit-exit, just exit the REPL
  63         exit) break;;
  64         # We need to handle continuations, and read -r doesn't do
  65         # that for us.  Yet we need read -r.
  66         #
  67         # We also use case/esac to compare lines read to "*\\"
  68         # because [ "$line" = *\\ ] and variants of that don't work.
  69         *\\) line="$rawline"
  70             while read -e -r -p '> ' rawline
  71             do
  72                 line="$line"$'\n'"$rawline"
  73                 case "$rawline" in
  74                 # We could check for here documents by matching
  75                 # against *<<*, but who cares.
  76                 *\\) continue;;
  77                 *) break;;
  78                 esac
  79             done
  80             ;;
  81         *) line=$rawline
  82         esac
  83
  84         case "$line" in
  85         *\\) break;;
  86         esac
  87
  88         # Finally!  Time to eval.
  89         eval "$line"
  90     done
  91
  92     echo $'\n\tExiting interactive shell...\n'
  93     return 0
  94 }
  95
  96 # lassert - Lustre test framework assert
  97 #
  98 # Arguments: failure code, failure message, expression/statement
  99 #
 100 # lassert evaluates the expression given, and, if false, calls
 101 # error() to trigger test failure.  If REPL_ON_LASSERT is true then
 102 # lassert will call lrepl() to give the user an interactive shell.
 103 # If the REPL sets retcode=0 then the assertion failure will be
 104 # ignored.
 105 lassert() {
 106     local retcode=$1
 107     local msg=$2
 108     shift 2
 109
 110     echo "checking $* ($(eval echo \""$*"\"))..."
 111     eval "$@" && return 0;
 112
 113     if ${REPL_ON_LASSERT:-false}; then
 114         echo "Assertion $retcode failed: $* (expanded: $(eval echo \""$*"\"))
 115 $msg"
 116         lrepl
 117     fi
 118
 119     error "Assertion $retcode failed: $* (expanded: $(eval echo \""$*"\"))
 120 $msg"
 121     return $retcode
 122 }
 123
 124 # setmodopts- set module options for subsequent calls to load_modules
 125 #
 126 # Usage: setmodopts module_name new_value [var_in_which_to_save_old_value]
 127 #        setmodopts -a module_name new_value [var_in_which_to_save_old_value]
 128 #
 129 # In the second usage the new value is appended to the old.
 130 setmodopts() {
 131         local _append=false
 132
 133         if [ "$1" = -a ]; then
 134             _append=true
 135             shift
 136         fi
 137
 138         local _var=MODOPTS_$1
 139         local _newvalue=$2
 140         local _savevar=$3
 141         local _oldvalue
 142
 143         # Dynamic naming of variables is a pain in bash.  In ksh93 we could
 144         # write "nameref opts_var=${modname}_MODOPTS" then assign directly
 145         # to opts_var.  Associative arrays would also help, alternatively.
 146         # Alas, we're stuck with eval until all distros move to a more recent
 147         # version of bash.  Fortunately we don't need to eval unset and export.
 148
 149         if [ -z "$_newvalue" ]; then
 150             unset $_var
 151             return 0
 152         fi
 153
 154         _oldvalue=${!var}
 155         $_append && _newvalue="$_oldvalue $_newvalue"
 156         export $_var="$_newvalue"
 157         echo setmodopts: ${_var}=${_newvalue}
 158
 159         [ -n "$_savevar" ] && eval $_savevar=\""$_oldvalue"\"
 160 }
 161
 162 echoerr () { echo "$@" 1>&2 ; }
 163
 164 signaled() {
 165     echoerr "$(date +'%F %H:%M:%S'): client load was signaled to terminate"
 166
 167     local PGID=$(ps -eo "%c %p %r" | awk "/ $PPID / {print \$3}")
 168     kill -TERM -$PGID
 169     sleep 5
 170     kill -KILL -$PGID
 171 }
 172
 173 mpi_run () {
 174     local mpirun="$MPIRUN $MPIRUN_OPTIONS"
 175     local command="$mpirun $@"
 176     local mpilog=$TMP/mpi.log
 177     local rc
 178
 179     if [ -n "$MPI_USER" -a "$MPI_USER" != root -a -n "$mpirun" ]; then
 180         echo "+ chmod 0777 $MOUNT"
 181         chmod 0777 $MOUNT
 182         command="su $MPI_USER sh -c \"$command \""
 183     fi
 184
 185     ls -ald $MOUNT
 186     echo "+ $command"
 187     eval $command 2>&1 | tee $mpilog || true
 188
 189     rc=${PIPESTATUS[0]}
 190     if [ $rc -eq 0 ] && grep -q "p4_error:" $mpilog ; then
 191        rc=1
 192     fi
 193     return $rc
 194 }
 195
 196 nids_list () {
 197         local list
 198         local escape="$2"
 199         for i in ${1//,/ }; do
 200                 if [ "$list" = "" ]; then
 201                         list="$i@$NETTYPE"
 202                 else
 203                         list="$list$escape $i@$NETTYPE"
 204                 fi
 205         done
 206         echo $list
 207 }
 208
 209 # FIXME: all setup/cleanup can be done without rpc.sh
 210 lst_end_session () {
 211     local verbose=false
 212     [ x$1 = x--verbose ] && verbose=true
 213
 214     export LST_SESSION=`$LST show_session 2>/dev/null | awk -F " " '{print $5}'`
 215     [ "$LST_SESSION" == "" ] && return
 216
 217         $LST stop b
 218     if $verbose; then
 219         $LST show_error c s
 220     fi
 221     $LST end_session
 222 }
 223
 224 lst_session_cleanup_all () {
 225     local list=$(comma_list $(nodes_list))
 226     do_rpc_nodes $list lst_end_session
 227 }
 228
 229 lst_cleanup () {
 230     lsmod | grep -q lnet_selftest && \
 231         rmmod lnet_selftest > /dev/null 2>&1 || true
 232 }
 233
 234 lst_cleanup_all () {
 235    local list=$(comma_list $(nodes_list))
 236
 237    # lst end_session needs to be executed only locally
 238    # i.e. on node where lst new_session was called
 239    lst_end_session --verbose
 240    do_rpc_nodes $list lst_cleanup
 241 }
 242
 243 lst_setup () {
 244     load_module lnet_selftest
 245 }
 246
 247 lst_setup_all () {
 248     local list=$(comma_list $(nodes_list))
 249     do_rpc_nodes $list lst_setup
 250 }
 251
 252 ###
 253 # short_hostname
 254 #
 255 # Passed a single argument, strips everything off following
 256 # and includes the first period.
 257 # client-20.lab.whamcloud.com becomes client-20
 258 short_hostname() {
 259   echo $(sed 's/\..*//' <<< $1)
 260 }
 261
 262 ###
 263 # short_nodename
 264 #
 265 # Find remote nodename, stripped of any domain, etc.
 266 # 'hostname -s' is easy, but not implemented on all systems
 267 short_nodename() {
 268         local rname=$(do_node $1 "uname -n" || echo -1)
 269         if [[ "$rname" = "-1" ]]; then
 270                 rname=$1
 271         fi
 272         echo $(short_hostname $rname)
 273 }
 274
 275 print_opts () {
 276     local var
 277
 278     echo OPTIONS:
 279
 280     for i in $@; do
 281         var=$i
 282         echo "${var}=${!var}"
 283     done
 284     [ -e $MACHINEFILE ] && cat $MACHINEFILE
 285 }
 286
 287 run_compilebench() {
 288         # Space estimation:
 289         # compile dir kernel-0  ~1GB
 290         # required space        ~1GB * cbench_IDIRS
 291
 292         local dir=${1:-$DIR}
 293
 294     cbench_DIR=${cbench_DIR:-""}
 295     cbench_IDIRS=${cbench_IDIRS:-2}
 296     cbench_RUNS=${cbench_RUNS:-2}
 297
 298     print_opts cbench_DIR cbench_IDIRS cbench_RUNS
 299
 300     [ x$cbench_DIR = x ] &&
 301         { skip_env "compilebench not found" && return; }
 302
 303     [ -e $cbench_DIR/compilebench ] || \
 304         { skip_env "No compilebench build" && return; }
 305
 306         local space=$(df -P $dir | tail -n 1 | awk '{ print $4 }')
 307         if [[ $space -le $((1024 * 1024 * cbench_IDIRS)) ]]; then
 308                 cbench_IDIRS=$((space / 1024 / 1024))
 309                 [[ $cbench_IDIRS -eq 0 ]] &&
 310                         skip_env "Need free space at least 1GB, have $space" &&
 311                         return
 312
 313                 echo "free space=$space, reducing initial dirs to $cbench_IDIRS"
 314         fi
 315
 316     # FIXME:
 317     # t-f _base needs to be modifyed to set properly tdir
 318     # for new "test_foo" functions names
 319     # local testdir=$DIR/$tdir
 320     local testdir=$dir/d0.compilebench
 321     mkdir -p $testdir
 322
 323     local savePWD=$PWD
 324     cd $cbench_DIR
 325     local cmd="./compilebench -D $testdir -i $cbench_IDIRS \
 326         -r $cbench_RUNS --makej"
 327
 328     log "$cmd"
 329
 330     local rc=0
 331     eval $cmd
 332     rc=$?
 333
 334     cd $savePWD
 335     [ $rc = 0 ] || error "compilebench failed: $rc"
 336     rm -rf $testdir
 337 }
 338
 339 run_metabench() {
 340
 341     METABENCH=${METABENCH:-$(which metabench 2> /dev/null || true)}
 342     mbench_NFILES=${mbench_NFILES:-30400}
 343     # threads per client
 344     mbench_THREADS=${mbench_THREADS:-4}
 345         mbench_OPTIONS=${mbench_OPTIONS:-}
 346
 347     [ x$METABENCH = x ] &&
 348         { skip_env "metabench not found" && return; }
 349
 350     # FIXME
 351     # Need space estimation here.
 352
 353     print_opts METABENCH clients mbench_NFILES mbench_THREADS
 354
 355     local testdir=$DIR/d0.metabench
 356     mkdir -p $testdir
 357     # mpi_run uses mpiuser
 358     chmod 0777 $testdir
 359
 360     # -C             Run the file creation tests.
 361     # -S             Run the file stat tests.
 362     # -c nfile       Number of files to be used in each test.
 363     # -k             Cleanup.  Remove the test directories.
 364         local cmd="$METABENCH -w $testdir -c $mbench_NFILES -C -S -k $mbench_OPTIONS"
 365     echo "+ $cmd"
 366
 367         # find out if we need to use srun by checking $SRUN_PARTITION
 368         if [ "$SRUN_PARTITION" ]; then
 369                 $SRUN $SRUN_OPTIONS -D $testdir -w $clients -N $num_clients \
 370                         -n $((num_clients * mbench_THREADS)) \
 371                         -p $SRUN_PARTITION -- $cmd
 372         else
 373                 mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 374                         -np $((num_clients * $mbench_THREADS)) $cmd
 375         fi
 376
 377     local rc=$?
 378     if [ $rc != 0 ] ; then
 379         error "metabench failed! $rc"
 380     fi
 381     rm -rf $testdir
 382 }
 383
 384 run_simul() {
 385
 386     SIMUL=${SIMUL:=$(which simul 2> /dev/null || true)}
 387     # threads per client
 388     simul_THREADS=${simul_THREADS:-2}
 389     simul_REP=${simul_REP:-20}
 390
 391     if [ "$NFSCLIENT" ]; then
 392         skip "skipped for NFSCLIENT mode"
 393         return
 394     fi
 395
 396     [ x$SIMUL = x ] &&
 397         { skip_env "simul not found" && return; }
 398
 399     # FIXME
 400     # Need space estimation here.
 401
 402     print_opts SIMUL clients simul_REP simul_THREADS
 403
 404     local testdir=$DIR/d0.simul
 405     mkdir -p $testdir
 406     # mpi_run uses mpiuser
 407     chmod 0777 $testdir
 408
 409     # -n # : repeat each test # times
 410     # -N # : repeat the entire set of tests # times
 411
 412     local cmd="$SIMUL -d $testdir -n $simul_REP -N $simul_REP"
 413
 414         echo "+ $cmd"
 415         # find out if we need to use srun by checking $SRUN_PARTITION
 416         if [ "$SRUN_PARTITION" ]; then
 417                 $SRUN $SRUN_OPTIONS -D $testdir -w $clients -N $num_clients \
 418                         -n $((num_clients * simul_THREADS)) -p $SRUN_PARTITION \
 419                         -- $cmd
 420         else
 421                 mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 422                         -np $((num_clients * simul_THREADS)) $cmd
 423         fi
 424
 425     local rc=$?
 426     if [ $rc != 0 ] ; then
 427         error "simul failed! $rc"
 428     fi
 429     rm -rf $testdir
 430 }
 431
 432 run_mdtest() {
 433
 434     MDTEST=${MDTEST:=$(which mdtest 2> /dev/null || true)}
 435     # threads per client
 436     mdtest_THREADS=${mdtest_THREADS:-2}
 437     mdtest_nFiles=${mdtest_nFiles:-"100000"}
 438     # We devide the files by number of core
 439     mdtest_nFiles=$((mdtest_nFiles/mdtest_THREADS/num_clients))
 440     mdtest_iteration=${mdtest_iteration:-1}
 441
 442     local type=${1:-"ssf"}
 443
 444     if [ "$NFSCLIENT" ]; then
 445         skip "skipped for NFSCLIENT mode"
 446         return
 447     fi
 448
 449     [ x$MDTEST = x ] &&
 450         { skip_env "mdtest not found" && return; }
 451
 452     # FIXME
 453     # Need space estimation here.
 454
 455     print_opts MDTEST mdtest_iteration mdtest_THREADS mdtest_nFiles
 456
 457     local testdir=$DIR/d0.mdtest
 458     mkdir -p $testdir
 459     # mpi_run uses mpiuser
 460     chmod 0777 $testdir
 461
 462     # -i # : repeat each test # times
 463     # -d   : test dir
 464     # -n # : number of file/dir to create/stat/remove
 465     # -u   : each process create/stat/remove individually
 466
 467     local cmd="$MDTEST -d $testdir -i $mdtest_iteration -n $mdtest_nFiles"
 468     [ $type = "fpp" ] && cmd="$cmd -u"
 469
 470         echo "+ $cmd"
 471         # find out if we need to use srun by checking $SRUN_PARTITION
 472         if [ "$SRUN_PARTITION" ]; then
 473                 $SRUN $SRUN_OPTIONS -D $testdir -w $clients -N $num_clients \
 474                         -n $((num_clients * mdtest_THREADS)) \
 475                         -p $SRUN_PARTITION -- $cmd
 476         else
 477                 mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 478                         -np $((num_clients * mdtest_THREADS)) $cmd
 479         fi
 480
 481     local rc=$?
 482     if [ $rc != 0 ] ; then
 483         error "mdtest failed! $rc"
 484     fi
 485     rm -rf $testdir
 486 }
 487
 488 run_connectathon() {
 489
 490     cnt_DIR=${cnt_DIR:-""}
 491     cnt_NRUN=${cnt_NRUN:-10}
 492
 493     print_opts cnt_DIR cnt_NRUN
 494
 495     [ x$cnt_DIR = x ] &&
 496         { skip_env "connectathon dir not found" && return; }
 497
 498     [ -e $cnt_DIR/runtests ] || \
 499         { skip_env "No connectathon runtests found" && return; }
 500
 501     local testdir=$DIR/d0.connectathon
 502     mkdir -p $testdir
 503
 504     local savePWD=$PWD
 505     cd $cnt_DIR
 506
 507     #
 508     # cthon options (must be in this order)
 509     #
 510     # -N numpasses - will be passed to the runtests script.  This argument
 511     #         is optional.  It specifies the number of times to run
 512     #         through the tests.
 513     #
 514     # One of these test types
 515     #    -b  basic
 516     #    -g  general
 517     #    -s  special
 518     #    -l  lock
 519     #    -a  all of the above
 520     #
 521     # -f      a quick functionality test
 522     #
 523
 524     tests="-b -g -s"
 525     # Include lock tests unless we're running on nfsv4
 526     local fstype=$(df -TP $testdir | awk 'NR==2  {print $2}')
 527     echo "$testdir: $fstype"
 528     if [[ $fstype != "nfs4" ]]; then
 529         tests="$tests -l"
 530     fi
 531     echo "tests: $tests"
 532     for test in $tests; do
 533         local cmd="./runtests -N $cnt_NRUN $test -f $testdir"
 534         local rc=0
 535
 536         log "$cmd"
 537         eval $cmd
 538         rc=$?
 539         [ $rc = 0 ] || error "connectathon failed: $rc"
 540     done
 541
 542     cd $savePWD
 543     rm -rf $testdir
 544 }
 545
 546 run_ior() {
 547         local type=${1:="ssf"}
 548
 549         IOR=${IOR:-$(which IOR 2> /dev/null || true)}
 550         # threads per client
 551         ior_THREADS=${ior_THREADS:-2}
 552         ior_iteration=${ior_iteration:-1}
 553         ior_blockSize=${ior_blockSize:-6}
 554         ior_blockUnit=${ior_blockUnit:-M}   # K, M, G
 555         ior_xferSize=${ior_xferSize:-1M}
 556         ior_type=${ior_type:-POSIX}
 557         ior_DURATION=${ior_DURATION:-30}        # minutes
 558         local multiplier=1
 559         case ${ior_blockUnit} in
 560                 [G])
 561                         multiplier=$((1024 * 1024 * 1024))
 562                         ;;
 563                 [M])
 564                         multiplier=$((1024 * 1024))
 565                         ;;
 566                 [K])
 567                         multiplier=1024
 568                         ;;
 569                 *)      error "Incorrect block unit should be one of [KMG]"
 570                         ;;
 571         esac
 572
 573         [ x$IOR = x ] &&
 574         { skip_env "IOR not found" && return; }
 575
 576         # calculate the space in bytes
 577         local space=$(df -B 1 -P $DIR | tail -n 1 | awk '{ print $4 }')
 578         local total_threads=$((num_clients * ior_THREADS))
 579         echo "+ $ior_blockSize * $multiplier * $total_threads "
 580         if [ $((space / 2)) -le \
 581              $((ior_blockSize * multiplier * total_threads)) ]; then
 582                 ior_blockSize=$((space / 2 / multiplier / total_threads))
 583                 [ $ior_blockSize -eq 0 ] && \
 584                 skip_env "Need free space more than $((2 * total_threads)) \
 585                          ${ior_blockUnit}: have $((space / multiplier))" &&
 586                          return
 587
 588                 echo "(reduced blockSize to $ior_blockSize \
 589                      ${ior_blockUnit} bytes)"
 590         fi
 591
 592     print_opts IOR ior_THREADS ior_DURATION MACHINEFILE
 593
 594     local testdir=$DIR/d0.ior.$type
 595     mkdir -p $testdir
 596     # mpi_run uses mpiuser
 597     chmod 0777 $testdir
 598     if [ "$NFSCLIENT" ]; then
 599         setstripe_nfsserver $testdir -c -1 ||
 600             { error "setstripe on nfsserver failed" && return 1; }
 601     else
 602         $LFS setstripe $testdir -c -1 ||
 603             { error "setstripe failed" && return 2; }
 604     fi
 605         #
 606         # -b N  blockSize --
 607         #       contiguous bytes to write per task (e.g.: 8, 4K, 2M, 1G)"
 608         # -o S  testFileName
 609         # -t N  transferSize -- size of transfer in bytes (e.g.: 8, 4K, 2M, 1G)"
 610         # -w    writeFile -- write file"
 611         # -r    readFile -- read existing file"
 612         # -W    checkWrite -- check read after write"
 613         # -C    reorderTasks -- changes task ordering to n+1 ordering for readback
 614         # -T    maxTimeDuration -- max time in minutes to run tests"
 615         # -k    keepFile -- keep testFile(s) on program exit
 616
 617         local cmd="$IOR -a $ior_type -b ${ior_blockSize}${ior_blockUnit} \
 618                 -o $testdir/iorData -t $ior_xferSize -v -C -w -r -W \
 619                 -i $ior_iteration -T $ior_DURATION -k"
 620
 621         [ $type = "fpp" ] && cmd="$cmd -F"
 622
 623         echo "+ $cmd"
 624         # find out if we need to use srun by checking $SRUN_PARTITION
 625         if [ "$SRUN_PARTITION" ]; then
 626                 $SRUN $SRUN_OPTIONS -D $testdir -w $clients -N $num_clients \
 627                         -n $((num_clients * ior_THREADS)) -p $SRUN_PARTITION \
 628                         -- $cmd
 629         else
 630                 mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 631                         -np $((num_clients * $ior_THREADS)) $cmd
 632         fi
 633
 634     local rc=$?
 635     if [ $rc != 0 ] ; then
 636         error "ior failed! $rc"
 637     fi
 638     rm -rf $testdir
 639 }
 640
 641 run_mib() {
 642
 643     MIB=${MIB:=$(which mib 2> /dev/null || true)}
 644     # threads per client
 645     mib_THREADS=${mib_THREADS:-2}
 646     mib_xferSize=${mib_xferSize:-1m}
 647     mib_xferLimit=${mib_xferLimit:-5000}
 648     mib_timeLimit=${mib_timeLimit:-300}
 649
 650     if [ "$NFSCLIENT" ]; then
 651         skip "skipped for NFSCLIENT mode"
 652         return
 653     fi
 654
 655     [ x$MIB = x ] &&
 656         { skip_env "MIB not found" && return; }
 657
 658     print_opts MIB mib_THREADS mib_xferSize mib_xferLimit mib_timeLimit \
 659         MACHINEFILE
 660
 661     local testdir=$DIR/d0.mib
 662     mkdir -p $testdir
 663     # mpi_run uses mpiuser
 664     chmod 0777 $testdir
 665     $LFS setstripe $testdir -c -1 ||
 666         { error "setstripe failed" && return 2; }
 667     #
 668     # -I    Show intermediate values in output
 669     # -H    Show headers in output
 670     # -L    Do not issue new system calls after this many seconds
 671     # -s    Use system calls of this size
 672     # -t    test dir
 673     # -l    Issue no more than this many system calls
 674     local cmd="$MIB -t $testdir -s $mib_xferSize -l $mib_xferLimit \
 675         -L $mib_timeLimit -HI -p mib.$(date +%Y%m%d%H%M%S)"
 676
 677         echo "+ $cmd"
 678         # find out if we need to use srun by checking $SRUN_PARTITION
 679         if [ "$SRUN_PARTITION" ]; then
 680                 $SRUN $SRUN_OPTIONS -D $testdir -w $clients -N $num_clients \
 681                         -n $((num_clients * mib_THREADS)) -p $SRUN_PARTITION \
 682                         -- $cmd
 683         else
 684                 mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 685                         -np $((num_clients * mib_THREADS)) $cmd
 686         fi
 687
 688     local rc=$?
 689     if [ $rc != 0 ] ; then
 690         error "mib failed! $rc"
 691     fi
 692     rm -rf $testdir
 693 }
 694
 695 run_cascading_rw() {
 696
 697     CASC_RW=${CASC_RW:-$(which cascading_rw 2> /dev/null || true)}
 698     # threads per client
 699     casc_THREADS=${casc_THREADS:-2}
 700     casc_REP=${casc_REP:-300}
 701
 702     if [ "$NFSCLIENT" ]; then
 703         skip "skipped for NFSCLIENT mode"
 704         return
 705     fi
 706
 707     [ x$CASC_RW = x ] &&
 708         { skip_env "cascading_rw not found" && return; }
 709
 710     # FIXME
 711     # Need space estimation here.
 712
 713     print_opts CASC_RW clients casc_THREADS casc_REP MACHINEFILE
 714
 715     local testdir=$DIR/d0.cascading_rw
 716     mkdir -p $testdir
 717     # mpi_run uses mpiuser
 718     chmod 0777 $testdir
 719
 720     # -g: debug mode
 721     # -n: repeat test # times
 722
 723     local cmd="$CASC_RW -g -d $testdir -n $casc_REP"
 724
 725         echo "+ $cmd"
 726         mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 727                 -np $((num_clients * $casc_THREADS)) $cmd
 728
 729     local rc=$?
 730     if [ $rc != 0 ] ; then
 731         error "cascading_rw failed! $rc"
 732     fi
 733     rm -rf $testdir
 734 }
 735
 736 run_write_append_truncate() {
 737
 738     # threads per client
 739     write_THREADS=${write_THREADS:-8}
 740     write_REP=${write_REP:-10000}
 741
 742     if [ "$NFSCLIENT" ]; then
 743         skip "skipped for NFSCLIENT mode"
 744         return
 745     fi
 746
 747     # location is lustre/tests dir
 748     if ! which write_append_truncate > /dev/null 2>&1 ; then
 749         skip_env "write_append_truncate not found"
 750         return
 751     fi
 752
 753     # FIXME
 754     # Need space estimation here.
 755
 756     local testdir=$DIR/d0.write_append_truncate
 757     local file=$testdir/f0.wat
 758
 759     print_opts clients write_REP write_THREADS MACHINEFILE
 760
 761     mkdir -p $testdir
 762     # mpi_run uses mpiuser
 763     chmod 0777 $testdir
 764
 765     local cmd="write_append_truncate -n $write_REP $file"
 766
 767         echo "+ $cmd"
 768         mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 769                 -np $((num_clients * $write_THREADS)) $cmd
 770
 771     local rc=$?
 772     if [ $rc != 0 ] ; then
 773         error "write_append_truncate failed! $rc"
 774         return $rc
 775     fi
 776     rm -rf $testdir
 777 }
 778
 779 run_write_disjoint() {
 780
 781     WRITE_DISJOINT=${WRITE_DISJOINT:-$(which write_disjoint \
 782         2> /dev/null || true)}
 783     # threads per client
 784     wdisjoint_THREADS=${wdisjoint_THREADS:-4}
 785     wdisjoint_REP=${wdisjoint_REP:-10000}
 786
 787     if [ "$NFSCLIENT" ]; then
 788         skip "skipped for NFSCLIENT mode"
 789         return
 790     fi
 791
 792     [ x$WRITE_DISJOINT = x ] &&
 793         { skip_env "write_disjoint not found" && return; }
 794
 795     # FIXME
 796     # Need space estimation here.
 797
 798     print_opts WRITE_DISJOINT clients wdisjoint_THREADS wdisjoint_REP \
 799         MACHINEFILE
 800     local testdir=$DIR/d0.write_disjoint
 801     mkdir -p $testdir
 802     # mpi_run uses mpiuser
 803     chmod 0777 $testdir
 804
 805     local cmd="$WRITE_DISJOINT -f $testdir/file -n $wdisjoint_REP"
 806
 807         echo "+ $cmd"
 808         mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 809                 -np $((num_clients * $wdisjoint_THREADS)) $cmd
 810
 811     local rc=$?
 812     if [ $rc != 0 ] ; then
 813         error "write_disjoint failed! $rc"
 814     fi
 815     rm -rf $testdir
 816 }
 817
 818 run_parallel_grouplock() {
 819
 820     PARALLEL_GROUPLOCK=${PARALLEL_GROUPLOCK:-$(which parallel_grouplock \
 821         2> /dev/null || true)}
 822     parallel_grouplock_MINTASKS=${parallel_grouplock_MINTASKS:-5}
 823
 824     if [ "$NFSCLIENT" ]; then
 825         skip "skipped for NFSCLIENT mode"
 826         return
 827     fi
 828
 829     [ x$PARALLEL_GROUPLOCK = x ] &&
 830         { skip "PARALLEL_GROUPLOCK not found" && return; }
 831
 832     print_opts clients parallel_grouplock_MINTASKS MACHINEFILE
 833
 834     local testdir=$DIR/d0.parallel_grouplock
 835     mkdir -p $testdir
 836     # mpi_run uses mpiuser
 837     chmod 0777 $testdir
 838
 839     local cmd
 840     local status=0
 841     local subtest
 842         for i in $(seq 12); do
 843                 subtest="-t $i"
 844                 local cmd="$PARALLEL_GROUPLOCK -g -v -d $testdir $subtest"
 845                 echo "+ $cmd"
 846
 847                 mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 848                         -np $parallel_grouplock_MINTASKS $cmd
 849                 local rc=$?
 850                 if [ $rc != 0 ] ; then
 851                         error_noexit "parallel_grouplock subtests $subtest " \
 852                                      "failed! $rc"
 853                 else
 854                         echo "parallel_grouplock subtests $subtest PASS"
 855                 fi
 856                 let status=$((status + rc))
 857                 # clear debug to collect one log per one test
 858                 do_nodes $(comma_list $(nodes_list)) lctl clear
 859         done
 860         [ $status -eq 0 ] || error "parallel_grouplock status: $status"
 861         rm -rf $testdir
 862 }
 863
 864 cleanup_statahead () {
 865     trap 0
 866
 867     local clients=$1
 868     local mntpt_root=$2
 869     local num_mntpts=$3
 870
 871     for i in $(seq 0 $num_mntpts);do
 872         zconf_umount_clients $clients ${mntpt_root}$i ||
 873             error_exit "Failed to umount lustre on ${mntpt_root}$i"
 874     done
 875 }
 876
 877 run_statahead () {
 878
 879     statahead_NUMMNTPTS=${statahead_NUMMNTPTS:-5}
 880     statahead_NUMFILES=${statahead_NUMFILES:-500000}
 881
 882     if [[ -n $NFSCLIENT ]]; then
 883         skip "Statahead testing is not supported on NFS clients."
 884         return 0
 885     fi
 886
 887     [ x$MDSRATE = x ] &&
 888         { skip_env "mdsrate not found" && return; }
 889
 890     print_opts MDSRATE clients statahead_NUMMNTPTS statahead_NUMFILES
 891
 892     # create large dir
 893
 894     # do not use default "d[0-9]*" dir name
 895     # to avoid of rm $statahead_NUMFILES (500k) files in t-f cleanup
 896     local dir=dstatahead
 897     local testdir=$DIR/$dir
 898
 899     # cleanup only if dir exists
 900     # cleanup only $statahead_NUMFILES number of files
 901     # ignore the other files created by someone else
 902     [ -d $testdir ] &&
 903         mdsrate_cleanup $((num_clients * 32)) $MACHINEFILE \
 904             $statahead_NUMFILES $testdir 'f%%d' --ignore
 905
 906     mkdir -p $testdir
 907     # mpi_run uses mpiuser
 908     chmod 0777 $testdir
 909
 910     local num_files=$statahead_NUMFILES
 911
 912     local IFree=$(inodes_available)
 913     if [ $IFree -lt $num_files ]; then
 914       num_files=$IFree
 915     fi
 916
 917     cancel_lru_locks mdc
 918
 919     local cmd1="${MDSRATE} ${MDSRATE_DEBUG} --mknod --dir $testdir"
 920     local cmd2="--nfiles $num_files --filefmt 'f%%d'"
 921     local cmd="$cmd1 $cmd2"
 922     echo "+ $cmd"
 923
 924         mpi_run ${MACHINEFILE_OPTION} ${MACHINEFILE} \
 925                 -np $((num_clients * 32)) $cmd
 926
 927     local rc=$?
 928     if [ $rc != 0 ] ; then
 929         error "mdsrate failed to create $rc"
 930         return $rc
 931     fi
 932
 933     local num_mntpts=$statahead_NUMMNTPTS
 934     local mntpt_root=$TMP/mntpt/lustre
 935     local mntopts=$MNTOPTSTATAHEAD
 936
 937     echo "Mounting $num_mntpts lustre clients starts on $clients"
 938     trap "cleanup_statahead $clients $mntpt_root $num_mntpts" EXIT ERR
 939     for i in $(seq 0 $num_mntpts); do
 940         zconf_mount_clients $clients ${mntpt_root}$i "$mntopts" ||
 941             error_exit "Failed to mount lustre on ${mntpt_root}$i on $clients"
 942     done
 943
 944     do_rpc_nodes $clients cancel_lru_locks mdc
 945
 946     do_rpc_nodes $clients do_ls $mntpt_root $num_mntpts $dir
 947
 948     mdsrate_cleanup $((num_clients * 32)) $MACHINEFILE \
 949         $num_files $testdir 'f%%d' --ignore
 950
 951     # use rm instead of rmdir because of
 952     # testdir could contain the files created by someone else,
 953     # or by previous run where is num_files prev > num_files current
 954     rm -rf $testdir
 955     cleanup_statahead $clients $mntpt_root $num_mntpts
 956 }