ceph/src/spdk/test/vhost/fiotest/vm_run.sh

   1 #!/usr/bin/env bash
   2
   3 BASE_DIR=$(readlink -f $(dirname $0))
   4 [[ -z "$TEST_DIR" ]] && TEST_DIR="$(cd $BASE_DIR/../../../../ && pwd)"
   5
   6 function usage()
   7 {
   8         [[ ! -z $2 ]] && ( echo "$2"; echo ""; )
   9         echo "Shortcut script for enabling VMs"
  10         echo "Usage: $(basename $1) [OPTIONS] VM..."
  11         echo
  12         echo "-h, --help                print help and exit"
  13         echo "    --work-dir=WORK_DIR   Where to find build file. Must exist. [default: ./..]"
  14         echo "-a                        Run all VMs in WORK_DIR"
  15         echo "-x                        set -x for script debug"
  16         exit 0
  17 }
  18 run_all=false
  19 while getopts 'xah-:' optchar; do
  20         case "$optchar" in
  21                 -)
  22                 case "$OPTARG" in
  23                         help) usage $0 ;;
  24                         work-dir=*) TEST_DIR="${OPTARG#*=}" ;;
  25                         *) usage $0 "Invalid argument '$OPTARG'" ;;
  26                 esac
  27                 ;;
  28         h) usage $0 ;;
  29         a) run_all=true ;;
  30         x) set -x ;;
  31         *) usage $0 "Invalid argument '$OPTARG'"
  32         esac
  33 done
  34
  35 . $BASE_DIR/common.sh
  36
  37 if [[ $EUID -ne 0 ]]; then
  38         echo "Go away user come back as root"
  39         exit 1
  40 fi
  41
  42 if $run_all; then
  43         vm_run -a
  44 else
  45         shift $((OPTIND-1))
  46         echo "INFO: running VMs: $@"
  47         vm_run "$@"
  48 fi