path: root/os-plugins/plugins/vmgrid/files/run-vmgrid.sh

#!/bin/bash
# -----------------------------------------------------------------------------
# Copyright (c) 2007..2010 - RZ Uni FR
# Copyright (c) 2007..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

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

# function to write to stdout and logfile
writelog () {
  # write to stdout
  echo -e "$1"
  # log into file
  echo -e "$1" >> ${OPENSLX_DEFAULT_LOGDIR}/run-vmgrid.${USER}.$$.log
}

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

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

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

# check if forcemem set
# TODO: klappt noch nicht immer :(
if [ "$1" = "--forcemem" ]; then
  forcemem=$2
  xmlfile=${3##*/}
else
  xmlfile=${1##*/}
fi

# absolute or relative path?
if ls ${1} 2>/dev/null | grep '/' >/dev/null 2>&1; then
  xmlpath=$(dirname ${1})
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
  echo -e "Usage: run-vmgrid.sh [--forcemem <memorysize in MB>] \c"
  echo -e "[/path/]filename[.xml]"
  exit 1
fi

# test how many instances running
runningvms=$(ps aux | grep "run-vmgrid.sh " | grep -v grep | wc -l)
runningvms=$(expr ${runningvms} - 1)
# check value
if [ $runningvms -le 0 ]; then
  writelog "Error in value: Running VMs: ${runningvms}. Exit!"
  exit 1
fi
# check if vmchooser plugin installed
if [ ${vmchooser_active} -eq 1 2>/dev/null ] && [ ${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$(df -h | grep " /tmp$" | awk '{print $2}') \c"
writelog "/tmp info:\t\t$(grep "/tmp " /proc/mounts | sed -e "s,/tmp ,,")"
writelog "\tVM XML dir:\t\t${vmgrid_xmlpath}"
writelog "\tVM Image dir:\t\t${vmgrid_imgpath}"
writelog "VM config:"

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

writelog "\tXML file:\t\t$xmlfile"

imgpath=${vmgrid_imgpath}

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

# check if virtual machine container file exists
if ! [ -e $vmpath ]; then
  writelog "\tVirtual machine image or directory ${vmpath} not found!"
  exit 1
fi

writelog "\tVM file/dir:\t\t$vmpath"

# 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}')

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

# if ${vm_name} not defined use ${imgname}
vm_name=${vm_name:-"${imgname%.*}"}
displayname=${vm_name}
# remove blanks

# 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)
if [ -z "${xmlvirt}" ] && [ -n "${additional_config}" ]; then
  writelog "\tNo virtual machine parameter defined in ${xmlfile}"
  writelog "\tTrying 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

# TODO: confs von openvz lesen?
if [ "${vmgrid_virt}" != "${xmlvirt}" ]; then
  writelog "\tThe virtual machine specified in ${xmlfile}"
  writelog "\tdoes 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 }')

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

writelog "\tVirtualization:\t\t$xmlvirt"
writelog "\tVM Name:\t\t$vm_name"
writelog "\tVM Displayname:\t\t$displayname"

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

# TODO: cpu cores
#vmgrid::maxvcpus
#vmgrid::minvcpus
#cpu_cores="${cpu_cores}"

# check for CPU virtualization flags
#vmgrid::hvm

# get total amount of memory installed in your machine
totalmem=$(expr $(grep -i "memtotal" /proc/meminfo | awk '{print $2}') / 1024)
permem=60
mem=$(expr ${totalmem} / 100 \* ${permem} / 4 \* 4)
#vmgrid::maxram
#vmgrid::minram

# 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=$($(which ifconfig) br0 | grep br0 | sed -e "s/ //g" | \
  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"

# TODO: vncclient IP:59001 oder Plugin?

# add rw share
rwshare="/var/opt/openslx/plugins/vmgrid/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
# TODO: rename to run-vuirt.include
if [ -e ${PLUGINCONFROOT}/${vmgrid_virt}/run-virt.include ] ; then
  writelog "Strating ${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 headless
if [ -n "${VIRTCMDHL}" ]; then
  writelog "Starting ${vmgrid_virt} head-less..."
  ${VIRTCMDHL} ${VIRTCMDOPTSHL}
else
  writelog "${vmgrid_virt}: No headless mode defined, exiting!"
  exit 1
fi

writelog "${vmgrid_virt} exited. Cleanning up...\c"
if echo "${RMDIRS}" 2>/dev/null | grep ${vmgrid_virt} >/dev/null 2>&1; then
  rm -rf ${RMDIRS} >/dev/null 2>&1
fi
writelog "done"
writelog "Bye."

exit 0