lustre/tests/rundbench

   1 #!/bin/bash
   2
   3 LUSTRE=${LUSTRE:-$(cd $(dirname $0)/..; echo $PWD)}
   4 . $LUSTRE/tests/test-framework.sh
   5
   6 usage() {
   7     echo "-C use chroot instead of cd"
   8     echo "-D DIR - use 'DIR' as work directory"
   9     echo
  10     exit;
  11 }
  12
  13 PATH=${DBENCH_LIB}:${PATH}
  14 MOUNT=${MOUNT:-/mnt/lustre}
  15 DIR=${DIR:-$MOUNT/`hostname`}
  16
  17 PREFIX="on"
  18 while getopts "CD:" opt $*; do
  19         case $opt in
  20                 D) [ -d $OPTARG ] && DIR=$OPTARG ;;
  21                 C) CHROOT="yes" ;;
  22                 \?) usage ;;
  23         esac
  24 done
  25
  26 mkdir -p $DIR
  27 TGT=$DIR/client.txt
  28 CLIENT_PREFIX="${DBENCH_LIB} /usr/share/dbench /usr/local/share /usr/lib/dbench"
  29 CLIENT_FILE="client.txt client_plain.txt dbench_client"
  30 if ! which dbench > /dev/null 2>&1 ; then
  31     [ "$MISSING_DBENCH_OK" ] || { error "$0 : $(hostname) dbench is not installed !" && exit 3; }
  32     skip_env "$0 : $(hostname) dbench is not installed"
  33     exit 0
  34 fi
  35 CLIENT=""
  36
  37 for prefix in $CLIENT_PREFIX; do
  38         for file in $CLIENT_FILE; do
  39                 if [ -s "${prefix}/${file}" ]; then
  40                         CLIENT="${prefix}/${file}";
  41                         break;
  42                 fi
  43         done
  44         [ "x$CLIENT" != "x" ] && break;
  45 done
  46
  47 if [ -n "$DBENCH_SRC" -a -s "$DBENCH_SRC" ]; then
  48     CLIENT=$DBENCH_SRC
  49 fi
  50
  51 [ ! -s "$CLIENT" ] &&
  52     skip_env "$0: no client file found for dbench on $(hostname): "\
  53              "DBENCH_LIB=$DBENCH_LIB DBENCH_SRC=$DBENCH_SRC" && exit 0
  54
  55 [ ! -s "$TGT" ] && echo "copying $CLIENT to $TGT" && cp $CLIENT $TGT
  56 [ ! -s "$TGT" ] && \
  57     echo "$0 : $(hostname) $TGT file doesn't exist after cp $CLIENT $TGT" && exit 1
  58
  59 if [ "x$CHROOT" == "xyes" ]; then
  60         echo "copying necessary libs to $DIR"
  61         cp `which dbench` $DIR
  62         LIBS71=$(ldd $DIR/dbench|sed -e 's/\t*//' -e 's/.*=> //' -e 's/ .*//' -e 's/^\///')
  63         tar -C / -chf - $LIBS71 | tar -C $DIR -xvf -
  64         if [ "${PIPESTATUS[*]}" != "0 0" ]; then
  65                 echo "can't copy libs $LIBS71 to $DIR"
  66                 rm -f $DIR/dbench
  67                 exit 1
  68         fi
  69         RUN="chroot $DIR"
  70         PREFIX="in"
  71         PATH=.:/:$PATH
  72 fi
  73
  74 shift $((OPTIND - 1))
  75
  76 trap '
  77 echo kill dbench main pid=$DBENCHPID
  78 kill $DBENCHPID
  79 rm -rf dbench $LIBS71 client.txt
  80 exit 0
  81 ' TERM
  82
  83 cd $DIR
  84 echo "running 'dbench $@' $PREFIX $PWD at `date`"
  85
  86 $RUN dbench -c client.txt $@ &
  87 DBENCHPID=$!
  88 echo "dbench PID=$DBENCHPID"
  89 wait $DBENCHPID
  90 RC=$?
  91 [ $RC -ne 0 ] && killall -9 dbench
  92
  93 rm -rf dbench $LIBS71 client.txt
  94 exit $RC