hck.sh

#!/bin/bash

# Copyright (c) 2013, Daynix Computing LTD (www.daynix.com)
# All rights reserved.
#
# Maintained by oss@daynix.com
#
# This file is a part of VirtHCK, please see the wiki page
# on https://github.com/daynix/VirtHCK/wiki for more.
#
# This code is licensed under standard 3-clause BSD license.
# See file LICENSE supplied with this package for the full license text.

VERSION=0.1.2
VM_START_TIMEOUT=10
ARGS_CFG=args.cfg

if [ "$1" = "-h" ] || [ "$1" = "--help" ]
then
  echo Usage:
  echo $0" [-c <CONFIG_FILE>] [st] [c1] [c2]"
  echo
  echo This script starts VirtHCK Studio with two Clients
  echo if executed without parameters, or you can specify
  echo which VMs to run with the following parameters:
  echo st - Start HCK Studio VM
  echo c1 - Start HCK Client 1 VM
  echo c2 - Start HCK Client 2 VM
  echo
  echo The default configuration file is \"hck_setup.cfg\" in the same directory
  echo that the script is run from. A different file can be specified using the
  echo \"-c\" option.
  exit
fi

if [ "$1" = "-v" ] || [ "$1" = "--version" ]
then
  echo VirtHCK Version: $VERSION
  exit
fi

if [ "$USER" != root ]
then
  echo This script must be run as superuser
  exit 1
fi

> $ARGS_CFG

while [ $# -gt 0 ]
do
key="$1"
case $key in
    -c)
    CONFIG_FILE="$2"
    shift
    ;;
    -id)
    echo UNIQUE_ID=$2 >> $ARGS_CFG
    shift
    ;;
    -device_type)
    echo TEST_DEV_TYPE=\"$2\" >> $ARGS_CFG
    shift
    ;;
    -device_name)
    echo TEST_DEV_NAME=\"$2\" >> $ARGS_CFG
    shift
    ;;
    -device_extra)
    echo TEST_DEV_EXTRA_PARAMS=\"$2\" >> $ARGS_CFG
    shift
    ;;
    -st_image)
    echo STUDIO_IMAGE=$2 >> $ARGS_CFG
    shift
    ;;
    -c1_image)
    echo CLIENT1_IMAGE=$2 >> $ARGS_CFG
    shift
    ;;
    -c1_cpus)
    echo CLIENT1_CPUS=$2 >> $ARGS_CFG
    shift
    ;;
    -c1_memory)
    echo CLIENT1_MEMORY=$2 >> $ARGS_CFG
    shift
    ;;
    -c2_image)
    echo CLIENT2_IMAGE=$2 >> $ARGS_CFG
    shift
    ;;
    -c2_cpus)
    echo CLIENT2_CPUS=$2 >> $ARGS_CFG
    shift
    ;;
    -c2_memory)
    echo CLIENT2_MEMORY=$2 >> $ARGS_CFG
    shift
    ;;
    -world_bridge)
    echo WORLD_BR_NAME=$2 >> $ARGS_CFG
    shift
    ;;
    -world_net_device)
    echo WORLD_NET_DEVICE=$2 >> $ARGS_CFG
    shift
    ;;
    -ctrl_net_device)
    echo CTRL_NET_DEVICE=$2 >> $ARGS_CFG
    shift
    ;;
    -file_transfer_device)
    echo FILE_TRANSFER_DEVICE=$2 >> $ARGS_CFG
    shift
    ;;
    -qemu_bin)
    echo QEMU_BIN=$2 >> $ARGS_CFG
    shift
    ;;
    -vhost_state)
    echo VHOST_STATE=$2 >> $ARGS_CFG
    shift
    ;;
    -enlightenments_state)
    echo ENLIGHTENMENTS_STATE=$2 >> $ARGS_CFG
    shift
    ;;
    -s3)
    echo ENABLE_S3=$2 >> $ARGS_CFG
    shift
    ;;
    -s4)
    echo ENABLE_S4=$2 >> $ARGS_CFG
    shift
    ;;
    -machine_type)
    echo MACHINE_TYPE=$2 >> $ARGS_CFG
    shift
    ;;
    -ivshmem_server_bin)
    echo IVSHMEM_SERVER_BIN=$2 >> $ARGS_CFG
    shift
    ;;
    -fs_deamon_bin)
    echo FS_DEAMON_BIN=$2 >> $ARGS_CFG
    shift
    ;;
    -fs_deamon_shared_dir)
    echo VIOFSD_SHARE=$2 >> $ARGS_CFG
    shift
    ;;
    -filesystem_tests_image)
    echo FILESYSTEM_TESTS_IMAGE=$2 >> $ARGS_CFG
    shift
    ;;
    -pidfile)
    echo PID_FILE=$2 >> $ARGS_CFG
    shift
    ;;
    -viommu)
    echo vIOMMU=$2 >> $ARGS_CFG
    shift
    ;;
    -st_extra)
    echo "STUDIO_EXTRA='$2'" >> $ARGS_CFG
    shift
    ;;
    -c1_extra)
    echo "CLIENT1_EXTRA='$2'" >> $ARGS_CFG
    shift
    ;;
    -c2_extra)
    echo "CLIENT2_EXTRA='$2'" >> $ARGS_CFG
    shift
    ;;
    st)
    RUN_STUDIO=true
    ;;
    c1)
    RUN_CLIENT1=true
    ;;
    c2)
    RUN_CLIENT2=true
    ;;
    ci_mode)
    CI_MODE=true
    ;;
    end)
    END=true
    ;;
    *)
    echo "Unknown option: ${key}"
    exit 1
    ;;
esac
shift
done

if [ -z ${RUN_STUDIO} ] && [ -z ${RUN_CLIENT1} ] && [ -z ${RUN_CLIENT2} ]; then
    RUN_ALL=true
fi

SCRIPTS_DIR=`dirname $0`
if [ -z ${CONFIG_FILE+x} ]; then
    if [ -f "${SCRIPTS_DIR}/hck_setup.cfg" ]; then
        CONFIG_FILE="${SCRIPTS_DIR}/hck_setup.cfg"
    else
        echo "A configuration file is not present or specified."
        echo "Please refer to help ($0 -h) for details."
        exit 1
    fi
elif [ ! -f "${CONFIG_FILE}" ]; then
    echo "The file ${CONFIG_FILE} does not exist."
    exit 1
fi

. ${CONFIG_FILE}

kill_jobs() {
  jobs -p > /tmp/.jobs_$$
  kill `cat /tmp/.jobs_$$ | tr '\n' ' '`
  rm -f /tmp/.jobs_$$
}

make_bridge_script() {
SCRIPTFILE="${HCK_ROOT}"/$1
cat <<EOF > ${SCRIPTFILE}
#!/bin/bash
. ${CONFIG_FILE}
$2 $3 \$1
EOF
chmod 755 ${SCRIPTFILE}
chown ${REAL_ME}:`id -g -n ${REAL_ME}` ${SCRIPTFILE}
}

make_bridge_script "hck_ctrl_bridge_ifup_${UNIQUE_ID}.sh" "enslave_iface" '${CTRL_BR_NAME}'
make_bridge_script "hck_world_bridge_ifup_${UNIQUE_ID}.sh" "enslave_iface" '${WORLD_BR_NAME}'
make_bridge_script "hck_test_bridge_ifup_${UNIQUE_ID}.sh" "enslave_test_iface" '${TEST_BR_NAME}'


trap "kill_jobs; loop_run_reset; remove_bridges; remove_bridge_scripts; exit 0" INT

if [ ! "$END" = true ] ; then
  echo
  dump_config
  echo
fi

if [ "$END" = true ] ; then
  remove_bridges
  remove_bridge_scripts
  kill_ivshmem_server
elif  [ "$CI_MODE" = true ] ; then
  if [ x"${RUN_STUDIO}" = xtrue ] || [ x"${RUN_ALL}" = xtrue ]; then
    echo Creating bridges...
    disable_bridge_nf
    create_bridges
    run_ivshmem_server
    ${SCRIPTS_DIR}/run_hck_studio.sh ${CONFIG_FILE} &
  fi
  if [ x"${RUN_CLIENT1}" = xtrue ] || [ x"${RUN_ALL}" = xtrue ]; then
    ${SCRIPTS_DIR}/run_hck_client.sh ${CONFIG_FILE} 1 &
  fi
  if [ x"${RUN_CLIENT2}" = xtrue ] || [ x"${RUN_ALL}" = xtrue ]; then
    ${SCRIPTS_DIR}/run_hck_client.sh ${CONFIG_FILE} 2 &
  fi
else

  disable_bridge_nf

  echo Creating bridges...
  create_bridges

  run_ivshmem_server

  loop_run_reset
  if [ x"${RUN_STUDIO}" = xtrue ] || [ x"${RUN_ALL}" = xtrue ]; then
    loop_run_vm ${SCRIPTS_DIR}/run_hck_studio.sh ${CONFIG_FILE} &
    sleep $VM_START_TIMEOUT
  fi
  if [ x"${RUN_CLIENT1}" = xtrue ] || [ x"${RUN_ALL}" = xtrue ]; then
    loop_run_vm ${SCRIPTS_DIR}/run_hck_client.sh ${CONFIG_FILE} 1 &
    sleep $VM_START_TIMEOUT
  fi
  if [ x"${RUN_CLIENT2}" = xtrue ] || [ x"${RUN_ALL}" = xtrue ]; then
    loop_run_vm ${SCRIPTS_DIR}/run_hck_client.sh ${CONFIG_FILE} 2 &
  fi
  sleep 2

  read -p "Press ENTER to disable VMs respawn..." NOT_NEEDED_VAR
  loop_run_stop
  echo VMs won\'t respawn anymore.
  wait

  sleep 2
  remove_bridges
  remove_bridge_scripts
  loop_run_reset
  kill_ivshmem_server
fi