#!/bin/bash
# -----------------------------------------------------------------------------
# Copyright (c) 2010 - RZ Uni FR
# Copyright (c) 2010 - OpenSLX GmbH
#
# This program is free software distributed under the GPL version 2.
# See http://openslx.org/COPYING
#
# If you have any feedback please consult http://openslx.org/feedback and
# send your suggestions, praise, or complaints to feedback@openslx.org
#
# General information about OpenSLX can be found at http://openslx.org/
# -----------------------------------------------------------------------------
# vmgrid
#    - This is the generic wrapper for the several virtualization solutions...
################################################################################

. /etc/opt/openslx/openslx.conf

################################################################################
### Define default dirs / get configs
################################################################################

PLUGINCONFROOT=${OPENSLX_DEFAULT_CONFDIR}/plugins
PLUGINCONFVMGRID=${PLUGINCONFROOT}/vmgrid
RWSHARE=/var/opt/openslx/plugins/vmgrid/share
# include general configuration from vmgrid
[ -f ${PLUGINCONFVMGRID}/vmgrid.conf ] && \
  . ${PLUGINCONFVMGRID}/vmgrid.conf
# load general virtualization information
[ -f ${PLUGINCONFROOT}/virtualization/virtualization.conf ] && \
  . ${PLUGINCONFROOT}/virtualization/virtualization.conf
# get the vmchooser_active var
[ -f ${PLUGINCONFROOT}/vmchooser/vmchooser.conf ] && \
  . ${PLUGINCONFROOT}/vmchooser/vmchooser.conf

################################################################################
### Functions used throughout the script
################################################################################

# function to write to stdout and logfile
LOGFILE=${OPENSLX_DEFAULT_LOGDIR}/run-vmgrid.${USER}.$$.log
writelog () {
  # write to stdout
  echo -e "$1"
  # log into file
  echo -e "$1" >> ${LOGFILE}
  # log into share dir, so that log is available in vm as well
  echo -e "$1" >> ${vmgrid_rwmnt}/logs/run-vmgrid.${USER}.$$.log
}

# remove config dirs when exit
cleanexit () {
  if echo "${RMDIRS}" 2>/dev/null | grep -q ${vmgrid_virt}; then
    writelog "${vmgrid_virt} exited. Cleanning up... \c"
    rm -rf ${RMDIRS} >/dev/null 2>&1
    writelog "done"
  fi

  exit "$1"
}

################################################################################
### Get XML file and dir
################################################################################

# check if mem, nice or graphical mode set
headless=1
unice=19
while [ $# -gt 0 ]; do
  case "$1" in
    -h|--help)
      echo -e "Usage: run-vmgrid.sh [-g] [--mem <memory size in MB>] \c"
      echo -e "[--nice <nice level>] [/path/]filename[.xml]"
      exit
    ;;
    -g)
      headless=0
    ;;
    --mem)
      # get a result which can be divided through 4
      if echo $2 | grep -qE \
         '^[2-9][0-9][0-9]$|^[1-9][0-9][0-9][0-9]$|^1[0-9][0-9][0-9][0-9]$';
      then
        forcemem=$(expr $2 / 4 \* 4)
      else
        echo "memory $2 not supported (200-19999)!"
        exit 1
      fi
      shift
    ;;
    --nice)
      # set nice level
      if echo $2 | grep -qE '^-[1-9]$|-1[0-9]$|^-20$|^[0-9]$|^1[0-9]$'; then
        unice=$2
      else
        echo "nice level $2 does not exist!"
        exit 1
      fi
      shift
    ;;
    *)
      xmlfile=$@
      break
    ;;
  esac
  shift
done

# absolute or relative path?
if ls ${xmlfile} 2>/dev/null | grep '/' >/dev/null 2>&1; then
  xmlpath=$(dirname ${xmlfile})
  xmlfile=$(basename ${xmlfile})
else
  xmlpath=${vmgrid_xmlpath}
fi
# full path
xmlfile="${xmlpath}/${xmlfile%.xml}.xml"

################################################################################
### Sanity checks
################################################################################

# test if the xml file is valid
if ! [ -r "${xmlfile}" ]; then
  writelog "${xmlfile} not a readable XML file!"
  run-vmgrid.sh -h &
  exit 1
fi

# start to log, create shrare log dir
mkdir -m 1777 -p ${vmgrid_rwmnt}/logs
echo "Starting to log at $(date)" \
  >${vmgrid_rwmnt}/logs/run-vmgrid.${USER}.$$.log

# test how many instances running
runningvms=$(ps aux | grep "run-vmgrid.sh " | grep -v grep | wc -l)
# 2 steps, /w only one is seems not to work
runningvms=$(expr ${runningvms} - 1)
# if Xen use different method
if [ "${vmgrid_virt}" = "xen" ]; then
  runningvms=$(xm list 2>/dev/null | grep -vE "Domain-0|Name.*ID" | wc -l)
  runningvms=$(expr ${runningvms} + 1)
fi
# check value
if [ ${runningvms} -le 0 ]; then
  writelog "Error in value: Running VMs: ${runningvms}. Exit!"
  exit 1
fi
# check if vmchooser plugin installed
if [ -n "${vmchooser_active}" ] && [ ${runningvms} -gt 1 ]; then
  # only allow one instance of vmgrid
  writelog "Already 1 VMs running and vmchooser plugin is active."
  writelog "Can't start ${xmlfile}, exiting!"
  exit 1
# else allow max. 4 instances
elif [ ${runningvms} -gt 4 ]; then
  writelog "Already 4 VMs running, exiting!"
  exit 1
fi
VM_ID="0${runningvms}"

# test if XML file
if ! grep '<?xml' "${xmlfile}" >/dev/null 2>&1; then
  writelog \
    "Submitted configuration file ${xmlfile} seems to have wrong XML format"
  exit 1
fi

################################################################################
###  Logo for console
################################################################################

cat <<EOL
                                                          __     __ 
         .----.--.--.-----.___.--.--.--------.-----.----.|__|.--|  |
         |   _|  |  |     |___|  |  |        |  _  |   _||  ||  _  |
         |__| |_____|__|__|    \___/|__|__|__|___  |__|  |__||_____|
                                             |_____|                
         OpenSLX virtual machine environment preparation script ...

EOL

################################################################################
### Read needed variables from XML file
################################################################################

writelog "Starting configuration..."
writelog "\tVM-ID:\t\t\t${VM_ID}"
writelog "\tLogfile:\t\t${OPENSLX_DEFAULT_LOGDIR}/run-vmgrid.${USER}.$$.log"
writelog "\t/tmp info:\t\t$(df -h | grep " /tmp$" | awk '{print $2}') \c"
writelog "$(grep "/tmp " /proc/mounts | awk '{print $1" "$2" "$3" "$4}')"
writelog "\tVM XML dir:\t\t$(dirname ${xmlfile})"
writelog "\tXML file:\t\t${xmlfile}"
writelog "VM config:"

# name of the virt image or dir
imgname=$(grep -io '<image_name param=.*"' ${xmlfile} \
          | sed -e "s/&.*;/; /g" | awk -F '"' '{ print $2 }')

# image mode
imgmode=$(grep -io '<image_mode param=.*"' ${xmlfile} \
          | sed -e "s/&.*;/; /g" | awk -F '"' '{ print $2 }')

# get boot attr
boot=$(grep -io 'boot param=.*"' ${xmlfile} | awk -F '"' '{ print $2 }')
diskless=0

# special rw image
if [ "${imgmode}" = "rwimg" ]; then
  # maybe you want to use an empty rw imgae?
  # define only for writelog, will be set in the plugin include again!
  vmpath="{special rw image}"
  # add new path for rwimg
  imgpath=${vmgrid_rwmnt}/specialrwimages
  unset $imgname
  mkdir -m 1777 -p ${imgpath}
# diskless boot
elif [ "${boot}" = "n" ] && [ -z "${imgname}" ]; then
  diskless=1
  vmpath="{diskless boot}"
# imagename /w full path
elif echo ${imgname} 2>/dev/null | grep -q '^/' >/dev/null 2>&1; then
  imgpath=$(dirname ${imgname})
  imgname=$(basename ${imgname})
  vmpath=${imgpath}/${imgname}
# get path from var
else
  imgpath=${vmgrid_imgpath}
  vmpath=${imgpath}/${imgname}
fi

# check if virtual machine container file exists, only if not rwimg
if ! [ -e "${vmpath}" ] && [ "${imgmode}" != "rwimg" ] && [ ${diskless} -eq 0 ];
then
  writelog "Virtual machine image or directory ${vmpath} not found!"
  exit 1
fi

# name of the virt machine, sed because of Windows formatting
vm_name=$(grep -o 'short_description param=.*"' ${xmlfile} \
          | sed -e "s/&.*;/; /g" | awk -F '"' '{print $2}')
# if ${vm_name} not defined use ${xmlfile}
vm_name=${vm_name:-${xmlfile%.xml}}

# define vm_shortname since vm_name can be very long
vm_shortname=$(basename ${xmlfile%.xml} | sed -e "s, ,-,g")

# vm_name = displayname, define for old scripts
displayname=${vm_name}

[ "${imgmode}" != "rwimg" ] && [ ${diskless} -eq 0 ] \
  && writelog "\tVM Image dir:\t\t${imgpath}"
writelog "\tVM file/dir:\t\t$vmpath"

# is there an additional configuration provided?
additional_config=$(grep -o 'additional_config param=.*"' ${xmlfile} \
                    | sed -e "s/&.*;/; /g" | awk -F '"' '{print $2}')

# image is for the following virtual machine 
xmlvirt=$(grep -o 'virtualmachine param=.*"' ${xmlfile} \
          | sed -e "s/&.*;/; /g" | awk -F '"' '{print $2}')

# make a guess from the filename extension if ${xmlvirt} is empty
# (not set within the XML file)
# TODO: implement possibility to submit own configuration files
if [ -z "${xmlvirt}" ] && [ -n "${additional_config}" ]; then
  writelog "No virtual machine parameter defined in ${xmlfile}"
  writelog "Trying to guess VM...\c"
  case "$(cat ${additional_config} | tr [A-Z] [a-z])" in
    *config.version*|*virtualhw.version*|*independent-nonpersistent*|*vmdk*)
      xmlvirt="vmware"
    ;;
    *innotek*|*virtualbox*)
      xmlvirt="virtualbox"
    ;;
    *qemu*|*kvm*)
      xmlvirt="qemukvm"
    ;;
    *)
      xmlvirt="none"
    ;;
  esac
elif [ -z "${xmlvirt}" ]; then
  case "$(echo ${imgname##*.} | tr [A-Z] [a-z])" in
    vmdk)
      xmlvirt="vmware"
    ;;
    vbox)
      xmlvirt="virtualbox"
    ;;
    img|qcow*)
      xmlvirt="qemukvm"
    ;;
    *)
      xmlvirt="none"
    ;;
  esac
  writelog "result:\t${xmlvirt}"
fi

# check for virt
if [ "${vmgrid_virt}" != "${xmlvirt}" ]; then
  writelog "The virtual machine specified in ${xmlfile}"
  writelog "does not match the virtualization used here (${vmgrid_virt})"
  exit 1
fi

# definition of the client system
vmostype=$(grep -io '<os param=.*"' ${xmlfile} | awk -F '"' '{ print $2 }' \
           | tr [A-Z] [a-z])

# definition of the networking the client system is connected to
network_kind=$(grep -io '<network param=.*"' ${xmlfile} \
               | awk -F '"' '{ print $2 }' | tr [A-Z] [a-z])
network_card=$(grep -io '<netcard param=.*"' ${xmlfile} \
               | awk -F '"' '{ print $2 }'| tr [A-Z] [a-z])

# define redirects
redirects=$(grep -ic '<redirect name=.*"' ${xmlfile})
[ -z "${redirects}" ] && redirects=0
(( i=1 ))
while [ ${i} -le ${redirects} ]; do
  # get only the $i-th line
  redirect_name[$i]=$(grep -m ${i} -io '<redirect.*' ${xmlfile} | tail -n 1 \
                      | grep -o "name=\".*" | awk -F '"' '{ print $2 }')
  redirect_proto[$i]=$(grep -m ${i} -io '<redirect.*' ${xmlfile} | tail -n 1 \
                       | grep -o "proto=\".*" | awk -F '"' '{ print $2 }')
  redirect_hport[$i]=$(grep -m ${i} -io '<redirect.*' ${xmlfile} | tail -n 1 \
                       | grep -o "hostport=\".*" | awk -F '"' '{ print $2 }')
  redirect_gport[$i]=$(grep -m ${i} -io '<redirect.*' ${xmlfile} | tail -n 1 \
                       | grep -o "guestport=\".*" | awk -F '"' '{ print $2 }')
  (( i=$i+1 ))
done

writelog "\tVirtualization:\t\t$xmlvirt"
writelog "\tVM name:\t\t$vm_name"
writelog "\tVM short name:\t\t$vm_shortname"

###############################################################################
### Declaration of default variables
###############################################################################

# make sure cpu_cores is not empty
cpu_cores=${cpu_cores:-"1"}
#vmgrid::maxvcpus
#vmgrid::minvcpus
# check for CPU virtualization flags
#vmgrid::vt

# total amount of memory defined in stage 3
permem=60
# get a result which can be divided through 4
mem=$(expr ${totalmem} / 100 \* ${permem} / 4 \* 4)
#vmgrid::maxmem
#vmgrid::minmem

# configuring ethernet mac address: first 3 bytes are fixed (00:50:56) 
# 4th byte is the VM-ID
# last two bytes are taken from the bridge of the host
# define one MAC per guest
macguestpart="00:50:56:${VM_ID}"
machostpart=$(echo ${hostmacaddr} | awk -F ":" '{print $(NF-1)":"$NF}')
macaddr=$(echo "${macguestpart}:${machostpart}" | tr [a-z] [A-Z])

# ide is expected default, test for the virtual disk image type should
# be done while creating the runscripts ...
# TODO: cdrom / floppy?
ide="TRUE"
scsi="FALSE"
hddrv="ide"
cdrom0="FALSE"
audio="false"
remotedesktopport="590${VM_ID}"

# add rw share
sharepath="${vmgrid_rwmnt}/folders/${vm_shortname}"
mkdir -p ${sharepath}
chmod -f 1777 "${vmgrid_rwmnt}/folders"
sharename="share"

# set hostname: using original hostname and adding string
hostname="vmgrid${VM_ID}-$(hostname)"

writelog "\tVM Hostname:\t\t$hostname"

################################################################################
### Setup the rest of the environment and run the configured vm
################################################################################

# Copy guest configuration (with added information) config.xml to be accessed
# via virtual floppy
# TODO: virt-floppy needed?

# Get all virtual machine specific stuff from the respective include file
if [ -e ${PLUGINCONFROOT}/${vmgrid_virt}/run-virt.include ] ; then
  writelog "Starting ${vmgrid_virt} specific part ..."
  self=${vmgrid_virt}
  . ${PLUGINCONFROOT}/${vmgrid_virt}/run-virt.include
else
  writelog "Failed because of missing ${vmgrid_virt} plugin"
  exit 1
fi

# start graphical mode
if [ ${headless} -eq 0 ]; then
  if [ -z "${VIRTCMD}" ]; then
    writelog "Grapical mode not available, exiting!"
    cleanexit 1
  fi
  writelog "Starting ${vmgrid_virt} in graphical mode /w nice level ${unice}..."
  # start /w low nice level
  nice -n ${unice} ${VIRTCMD} ${VIRTCMDOPTS} 2>/dev/null
# start headless
elif [ -n "${VIRTCMDHL}" ]; then
  writelog "Starting ${vmgrid_virt} in headless mode /w nice level ${unice}..."
  # start /w low nice level
  nice -n ${unice} ${VIRTCMDHL} ${VIRTCMDOPTSHL} 2>/dev/null
else
  writelog "${vmgrid_virt}: No headless mode defined, exiting!"
  cleanexit 1
fi

# postrun for scripts after virtualization finishes
if [ -n "${POSTRUN}" ]; then
  eval ${POSTRUN}  >/dev/null 2>&1
fi

cleanexit 0
exit 0