From 6ee8363caa02411cbc6851e5f8ecfa407643e8f0 Mon Sep 17 00:00:00 2001
From: Michael Janczyk
Date: Wed, 28 Apr 2010 15:33:15 +0200
Subject: clean exit added + port redirects now via xml file

---
 .../plugins/vmgrid/OpenSLX/OSPlugin/vmgrid.pm      |  4 +-
 os-plugins/plugins/vmgrid/files/run-vmgrid.sh      | 46 ++++++++++++++++------
 .../init-hooks/80-after-plugins/virtualization.sh  |  2 +-
 3 files changed, 37 insertions(+), 15 deletions(-)

(limited to 'os-plugins/plugins/vmgrid')

diff --git a/os-plugins/plugins/vmgrid/OpenSLX/OSPlugin/vmgrid.pm b/os-plugins/plugins/vmgrid/OpenSLX/OSPlugin/vmgrid.pm
index 401a2151..c3542274 100644
--- a/os-plugins/plugins/vmgrid/OpenSLX/OSPlugin/vmgrid.pm
+++ b/os-plugins/plugins/vmgrid/OpenSLX/OSPlugin/vmgrid.pm
@@ -72,9 +72,9 @@ sub getAttrInfo
             description => unshiftHereDoc(<<'            End-of-Here'),
                 which virtualization technique schould be used?
             End-of-Here
-            content_regex => qr{^(vmware|virtualbox|qemukvm|xen)$},
+            content_regex => qr{^(virtualbox|qemukvm|xen)$},
             #openvz|vserver
-            content_descr => 'vmware, virtualbox, qemukvm, xen',
+            content_descr => 'virtualbox, qemukvm, xen',
             #openvz, vserver
             default => undef,
         },
diff --git a/os-plugins/plugins/vmgrid/files/run-vmgrid.sh b/os-plugins/plugins/vmgrid/files/run-vmgrid.sh
index a684310b..9b9adb44 100644
--- a/os-plugins/plugins/vmgrid/files/run-vmgrid.sh
+++ b/os-plugins/plugins/vmgrid/files/run-vmgrid.sh
@@ -49,6 +49,17 @@ writelog () {
   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
 ################################################################################
@@ -289,11 +300,28 @@ 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} \
+network_kind=$(grep -io '<network param=.*"' ${xmlfile} \
                | awk -F '"' '{ print $2 }' | tr [A-Z] [a-z])
-network_card=$(grep -io 'netcard param=.*"' ${xmlfile} \
+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"
@@ -332,6 +360,7 @@ ide="TRUE"
 scsi="FALSE"
 hddrv="ide"
 audio="false"
+remotedesktopport="590${VM_ID}"
 
 # add rw share
 sharepath="${vmgrid_rwmnt}/folders/${vm_shortname}"
@@ -367,7 +396,7 @@ fi
 if [ ${headless} -eq 0 ]; then
   if [ -z "${VIRTCMD}" ]; then
     writelog "Grapical mode not available, exiting!"
-    exit 1
+    cleanexit 1
   fi
   writelog "Starting ${vmgrid_virt} in graphical mode..."
   ${VIRTCMD} ${VIRTCMDOPTS} 2>/dev/null
@@ -377,7 +406,7 @@ elif [ -n "${VIRTCMDHL}" ]; then
   ${VIRTCMDHL} ${VIRTCMDOPTSHL} 2>/dev/null
 else
   writelog "${vmgrid_virt}: No headless mode defined, exiting!"
-  exit 1
+  cleanexit 1
 fi
 
 # postrun for scripts after virtualization finishes
@@ -385,12 +414,5 @@ if [ -n "${POSTRUN}" ]; then
   eval ${POSTRUN}  >/dev/null 2>&1
 fi
 
-# remove config dirs when finished
-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
-
-echo -e "Bye."
+cleanexit 0
 exit 0
diff --git a/os-plugins/plugins/vmgrid/init-hooks/80-after-plugins/virtualization.sh b/os-plugins/plugins/vmgrid/init-hooks/80-after-plugins/virtualization.sh
index 400075cd..a1da2e63 100644
--- a/os-plugins/plugins/vmgrid/init-hooks/80-after-plugins/virtualization.sh
+++ b/os-plugins/plugins/vmgrid/init-hooks/80-after-plugins/virtualization.sh
@@ -30,7 +30,7 @@ active=0
 [ ${vmgrid_active} -ne 0 ] && active=1
 
 [ $DEBUGLEVEL -gt 0 ] \
-  && echo -e "Executing the virtual 'virtualization' plugin ... \c"
+  && echo -e "Executing the 'virtualization' plugin ... \c"
 
 if [ ${active} -ne 0 ] ; then
 
-- 
cgit v1.2.3-55-g7522