Manuel Amador (Rudd-O) 05c020e1f6 Source code committed
2010-08-11 09:13:29 -07:00

305 lines
6.7 KiB
Bash
Executable File

#!/usr/bin/env bash
# $Id: runvm.sh 9132 2010-06-04 20:17:43Z manuel $ $HeadURL: svn://svn.lab.vmops.com/repos/vmdev/java/scripts/storage/zfs/nfs/runvm.sh $
# runvm.sh -- start a vm from a directory containing the kernel and os images
#
#
usage() {
printf "Usage: %s: -v <vnet-id> -i <ip-address> -m <memory in MB> -a <mac address> -g <gateway ip addr> -c <vnc console #> -w <vnc password>-l <vm label> -n <# cores> -u <compute unites in pct> <image-dir>\n" $(basename $0) >&2
}
check_vnetd_running() {
/sbin/lsmod | grep vnet_module > /dev/null
local module_loaded=$?
if [[ ( -z $(pidof vnetd) ) && ( $module_loaded -ne 0 ) ]]
then
printf 'vnet: Neither userspace daemon running nor kernel module loaded!, not starting vm '"$vmname\n" >&2
exit 2
fi
}
is_xen_3_3 () {
local a;
local b;
a=$(xm info | grep xen_minor | cut -d":" -f2);
b=$(echo $a)
if [ $b -lt 3 ]
then
return 1
fi
return 0
}
get_dom0_ip () {
if ifconfig eth0 &> /dev/null;
then
eval "$1=$(ifconfig eth0 | awk '/inet addr/ {split ($2,A,":"); print A[2]}')"
return 0
elif ifconfig eth1 &> /dev/null;
then
eval "$1=$(ifconfig eth1 | awk '/inet addr/ {split ($2,A,":"); print A[2]}')"
return 0;
fi
return 1;
}
create_vnet () {
local vnetid=$1
local bridgeid="vnbr""$vnetid"
local vnifid="vnif""$vnetid"
local longvnetid="0000:0000:0000:0000:0000:0000:0000:""$vnetid"
eval "$2=$bridgeid"
# Create the vnet even if it already exists. /usr/sbin/vn operation is
# idempotent
vn vnet-create -b ${bridgeid} ${longvnetid} &> /dev/null
#echo $bridgeid $vnifid $longvnetid
return 0
}
# check if gateway domain is up and running
check_gw() {
ping -c 1 -n -q $1 > /dev/null
if [ $? -gt 0 ]
then
sleep 1
ping -c 1 -n -q $1 > /dev/null
fi
return $?;
}
#Append an entry into dhcp hosts file and knock the dhcp server on its head
add_dhcp_entry() {
local gw=$1
local mac=$2
local ip=$3
local vm=$4
ssh -o StrictHostKeyChecking=no -i ./id_rsa root@$gw "/root/edithosts.sh $mac $ip $vm" >/dev/null
if [ $? -gt 0 ]
then
$5=1
return 1
fi
return 0
}
#set -x
vflag=
iflag=
mflag=
aflag=
gflag=
lflag=
cflag=
vcpus=0
cpucap=0
vncpwd="password"
while getopts 'v:i:m:a:g:l:c:n:u:w:' OPTION
do
case $OPTION in
v) vflag=1
vnetid="$OPTARG"
;;
i) iflag=1
ipaddr="$OPTARG"
;;
m) mflag=1
ram="$OPTARG"
;;
a) aflag=1
macaddr="$OPTARG"
;;
g) gflag=1
gateway="$OPTARG"
;;
l) lflag=1
vmname="$OPTARG"
;;
c) cflag=1
vncconsole="$OPTARG"
;;
w) wflag=1
vncpwd="$OPTARG"
;;
n) nflag=1
vcpus="$OPTARG"
;;
u) uflag=1
cpucap="$OPTARG"
;;
?) usage
exit 2
;;
esac
done
if [ "$vflag$mflag$iflag$aflag$gflag$lflag$cflag" != "1111111" ]
then
usage
exit 2
fi
wincpuopts=""
if [ "$vcpus" == 0 -a "$cpucap" == 0 ]
then
#Windows doesn't like it
wincpuopts=""
elif [ "$vcpus" == 0 ]
then
wincpuopts="cpu_cap=$cpucap"
elif [ "$cpucap" == 0 ]
then
wincpuopts="vcpus=$vcpus"
else
wincpuopts="vcpus=$vcpus cpu_cap=$cpucap"
fi
shift $(($OPTIND - 1))
imagedir=$1
if [ -z $imagedir ]
then
usage
exit 2
fi
if [ "$vmname" == gateway ]
then
printf "Error: $vmname is illegal (this is the hostname of domR)\n" >&2
exit 2
fi
if xm list $vmname &>/dev/null
then
printf "Error: vm $vmname already exists\n" >&2
exit 2
fi
if [ "$ipaddr" == "10.1.1.1" ]
then
printf "Error: 10.1.1.1 is the gateway and cannot be assigned to a vm\n" >&2
exit 2
fi
#make sure vnetd is running
check_vnetd_running
# grab the kernel, bootloader and disk images
kernel="$imagedir/"$(ls $imagedir | grep vmlinuz)
ramdisk="$imagedir/"$(ls $imagedir | grep initrd)
rootdisk="$imagedir/"$(ls $imagedir | grep vmi-root)
swapdisk="$imagedir/"$(ls $imagedir | grep swap)
datadisk1="$imagedir/"$(ls $imagedir | grep vmi-data1)
datadisk2="$imagedir/"$(ls $imagedir | grep vmi-data2)
#set -x
if [ "$rootdisk" == "$imagedir/" ]
then
printf "Error: No root disk found\nVM $vmname not started\n" >&2
exit 2
fi
if [ "$datadisk1" == "$imagedir/" ]
then
#look for subdirs called 'datadisk'
i=0
datadisks=( )
for diskfs in $(find $imagedir -type d | grep datadisk)
do
datadisks[$i]=$(find $diskfs | grep vmi-data | head -1) #expect only 1 disk
let i=i+1
done
datadisk1=${datadisks[0]};
if [ -z "$datadisk1" ]
then
printf "Error: No data disk found\nVM $vmname not started\n" >&2
exit 2
fi
fi
linux=0
windows=0
pygr=0
if [ "$kernel" != "$imagedir/" ] && [ "$ramdisk" != "$imagedir/" ]
then
#Linux kernel
linux=1
builder="linux"
device_model=""
elif [ -f "$imagedir/pygrub" ]
then
#pygrub linux image
pygr=1
else
kernel="/usr/lib/xen/boot/hvmloader"
windows=1
builder="hvm"
device_model="/usr/lib64/xen/bin/qemu-dm"
fi
#get dom 0 ip to figure out which ip to bind vnc to
dom0ip=""
get_dom0_ip dom0ip
# check if gateway domain is up and running
if ! check_gw "$gateway"
then
printf "Unable to ping the gateway domain, exiting\n" >&2
exit 3
fi
#Append an entry into dhcp hosts file and knock the dhcp server on its head
added=0
if ! add_dhcp_entry $gateway $macaddr $ipaddr $vmname $added
then
printf "Unable add dhcp entry on gateway (reason=$added), exiting\n" >&2
exit 4
fi
# Create the vnet locally if not already created
bridge=""
if ! create_vnet "$vnetid" bridge ; then
printf "Failed to create vnet, exiting\n" >&2
exit 5
fi
#hvm disk prefix for 3.1 is file:, for 3.3 it is tap:aio
hvmdisk="tap:aio"
if ! is_xen_3_3
then
hvmdisk="file"
fi
# create the vm (linux)
if [ $linux -eq 1 ]; then
xm new /dev/null kernel="$kernel" name="$vmname" disk="tap:aio:$rootdisk,xvda1,w" disk="tap:aio:$swapdisk,xvda2,w" disk="tap:aio:$datadisk1,xvda3,w" ramdisk="$ramdisk" memory=$ram vif="mac=$macaddr,bridge=$bridge" dhcp="dhcp" root="/dev/xvda1 ro" vnc=1 vnclisten="$dom0ip" vfb="type=vnc,vncdisplay=$vncconsole,vncpasswd=$vncpwd" ip="$ipaddr" extra="fastboot" vcpus=$vcpus cpu_cap=$cpucap
elif [ $pygr -eq 1 ]; then
xm new /dev/null bootloader="/usr/bin/pygrub" name="$vmname" disk="tap:aio:$rootdisk,sda,w" disk="tap:aio:$datadisk1,sdb,w" memory=$ram vif="mac=$macaddr,bridge=$bridge" dhcp="dhcp" root="/dev/sda ro" vnc=1 vnclisten="$dom0ip" vfb="type=vnc,vncdisplay=$vncconsole,vncpasswd=$vncpwd" ip="$ipaddr" extra="fastboot" vcpus=$vcpus cpu_cap=$cpucap
else
#create the vm (windows/HVM)
xm new /dev/null kernel="$kernel" name="$vmname" disk="$hvmdisk:$rootdisk,hda,w" disk="$hvmdisk:$datadisk1,hdb,w" builder="$builder" device_model="$device_model" memory=$ram vif="mac=$macaddr,bridge=$bridge,type=ioemu" dhcp="dhcp" vnc=1 vnclisten="$dom0ip" vncdisplay="$vncconsole" vncpasswd="$vncpwd" usbdevice="tablet" localtime="yes" $wincpuopts
fi
if [ $? -gt 0 ]; then
exit 10
fi
#Kick off the vm
xm start $vmname
if [ $? -gt 0 ]; then
exit 20
fi
exit 0