X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-database.git/blobdiff_plain/7c4372ece5fad89044efe4d18e439e0fb54f750a..4ce4ff22c2a6d47c549b109c71bca31e40846ed6:/client/etc/xen/sipb-database

diff --git a/client/etc/xen/sipb-database b/client/etc/xen/sipb-database
index daccd93..723eb78 100644
--- a/client/etc/xen/sipb-database
+++ b/client/etc/xen/sipb-database
@@ -1,15 +1,30 @@
+# -*- mode: python; -*-
 import sipb_xen_database.models as models
 from sipb_xen_database import connect
+import re
+import tempfile
+from subprocess import call
 
 connect('postgres://sipb-xen@sipb-xen-dev.mit.edu/sipb_xen')
-# `machine_name' should get passed in from the xm create call
+prefix = "d_"
+
+# 'machine_name', and optionally 'cdrom_image', should get passed in
+# from the xm create call
+
+def check(b):
+    if not b:
+        import sys
+        sys.exit(1)
 
 machine = models.Machine.get_by(name=machine_name)
+check(machine is not None)
 machine_type = models.Type.get_by(type_id=machine.type_id)
 
 memory = machine.memory
 maxmem = memory
-name = machine.name
+check(re.match('^[A-Za-z0-9][A-Za-z0-9._-]*$', machine.name))
+name = prefix + machine.name
+check(re.match('^[0-9a-f-]+$', machine.uuid))
 uuid = machine.uuid
 
 vcpus = machine.cpus
@@ -19,12 +34,13 @@ viftype = ""
 
 if machine_type.hvm:
     ioemu = "ioemu:"
-    viftype = "type=ioemu"
+    viftype = "type=ioemu, "
 
     kernel = 'hvmloader'
     builder = 'hvm'
     vnc = 1
-    device_model = 'qemu-dm'
+    vncpasswd = 'moocow'
+    device_model = '/usr/sbin/qemu-dm-sipb'
 else:
     kernel  = '/boot/vmlinuz-2.6.18-4-xen-amd64'
     ramdisk = '/boot/initrd.img-2.6.18-4-xen-amd64'
@@ -37,20 +53,33 @@ apic = machine_type.apic
 vif = []
 
 for n in machine.nics:
-    d = '%s, mac=%s, ip=%s, bridge=xenbr0, script=vif-sipbroute' % (viftype, n.mac_addr, n.ip)
+    check(re.match('^[0-9a-fA-F:]+$', n.mac_addr) and re.match('^[0-9.]*$', n.ip))
+    d = '%smac=%s, ip=%s, bridge=xenbr0, script=vif-sipbroute' % (viftype, n.mac_addr, n.ip)
     vif.append(d)
 
 disk = []
 
 for d in machine.disks:
-    device = 'xenvg/' + machine.name + '_' + d.guest_device_name
+    check(re.match('^[A-Za-z0-9]+$', d.guest_device_name))
+    device = '/dev/xenvg/' + prefix + machine.name + '_' + d.guest_device_name
     dspec = 'phy:%s,%s%s,w' % (device, diskioemu, d.guest_device_name)
     disk.append(dspec)
 
-restart = 'destroy'
-if machine.autorestart:
-    restart = 'restart'
+if 'installer' in locals():
+    check(re.match('^[A-Za-z0-9][A-Za-z0-9_.-]*$', installer))
+    tmptree = tempfile.mkdtemp('', 'auto-install.', '/tmp')
+    call(['/usr/sbin/sipb-xen-make-iso', installer, tmptree]
+         + installer_options.split(' '))
+    disk.append('file:'+tmptree+'/install.iso,hdc:cdrom,r')
+    boot = 'd'
+    
+elif 'cdrom_image' in locals():
+    check(re.match('^[A-Za-z0-9][A-Za-z0-9_.-]*$', cdrom_image))
+    disk.append('file:/srv/images/' + cdrom_image + '.iso,hdc:cdrom,r')
+    boot = 'd'
 
 on_poweroff = 'destroy'
 on_reboot = 'restart'
-on_crash = restart
+on_crash = 'destroy'
+if machine.autorestart:
+    on_crash = 'restart'