Look for CD images in /dev/xenvg/image_* and validate them against the

[invirt/packages/invirt-database.git] / client / etc / xen / sipb-database

diff --git a/client/etc/xen/sipb-database b/client/etc/xen/sipb-database
index d197d9b..0d4d466 100644 (file)
--- a/client/etc/xen/sipb-database
+++ b/client/etc/xen/sipb-database
@@ -1,15 +1,34 @@
-import sipb_xen_database.models as models
-from sipb_xen_database import connect
+# -*- mode: python; -*-
+from invirt.database import models, connect
+from invirt.config import structs as config
+import re
+import tempfile
+from subprocess import call

 
 

-connect('postgres://sipb-xen@sipb-vm-1/sipb_xen')
-# `machine_name' should get passed in from the xm create call
+connect()
+prefix = "d_"

 
 

-machine = models.Machine.get_by(name=machine_name)
-machine_type = models.Type.get_by(type_id=machine.type_id)
+# '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.query().filter_by(name=machine_name).one()
+check(machine is not None)
+machine_type = machine.type
+cdrom = None
+if 'cdrom_image' in locals():
+    cdrom = models.CDROM.query().filter_by(name=cdrom_image).one()
+    check(cdrom is not None)

 
 memory = machine.memory
 maxmem = memory
 
 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
 uuid = machine.uuid
 
 vcpus = machine.cpus

@@ -17,40 +36,70 @@ vcpus = machine.cpus
 diskioemu = ""
 viftype = ""
 
 diskioemu = ""
 viftype = ""
 

-if machine_type.hvm:
-    ioemu = "ioemu:"
-    viftype = "type=ioemu"
-
-    kernel = 'hvmloader'
-    builder = 'hvm'
-    vnc = 1
-    device_model = 'qemu-dm'
-else:
-    kernel  = '/boot/vmlinuz-2.6.18-4-xen-amd64'
-    ramdisk = '/boot/initrd.img-2.6.18-4-xen-amd64'
-    builder = 'linux'
-

 pae = machine_type.pae
 acpi = machine_type.acpi
 apic = machine_type.apic
 
 vif = []
 
 pae = machine_type.pae
 acpi = machine_type.acpi
 apic = machine_type.apic
 
 vif = []
 

+disk = []
+
+if machine_type.hvm:
+    codepath = 'hvm'
+else:
+    codepath = 'paravm'
+
+if 'mirror' in locals(): #Installer
+    disk.append('phy:/dev/xenvg/s_install_hda,hdb,r')
+
+    kernel = '/boot/vmlinuz-2.6.24-19-xen' #From hardy
+    ramdisk = '/boot/initrd.img-2.6.24-19-xen'
+
+    if not machine.nics:
+        raise RuntimeError('You must have a nic to autoinstall')
+    n = machine.nics[0]
+    extra = 'ro noresume'
+    extra += (' ip=%s::%s:%s:%s:eth0:off'
+         % (n.ip, config.dhcp.gateway, config.dhcp.netmask, machine.name))
+    extra += ' mirror=%s dist=%s' % (mirror, dist)
+    extra += ' imagesize=%s' % imagesize
+    if 'noinstall' in locals():
+        extra += ' noinstall'
+    root = '/dev/hdb1'
+    codepath = None
+elif cdrom is not None:
+    disk.append('phy:/etc/xenvg/image_' + cdrom.cdrom_id + ',hdc:cdrom,r')
+    boot = 'd'
+    codepath = 'hvm'
+
+if codepath == 'hvm':
+    ioemu = "ioemu:"
+    viftype = "type=ioemu, "
+    kernel = '/usr/lib/xen/boot/hvmloader'
+    builder = 'hvm'
+    vnc = 1
+    vncpasswd = 'moocow'
+    device_model = '/usr/sbin/qemu-dm-sipb'
+    serial = "pty"
+elif codepath == 'paravm':
+    bootloader = '/usr/lib/xen-default/bin/pygrub'
+
+

 for n in machine.nics:
 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)
 
     vif.append(d)
 

-disk = []
-

 for d in machine.disks:
 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)
 
     dspec = 'phy:%s,%s%s,w' % (device, diskioemu, d.guest_device_name)
     disk.append(dspec)
 

-restart = 'destroy'
-if machine.autorestart:
-    restart = 'restart'
+usbdevice = 'tablet'

 
 on_poweroff = 'destroy'
 on_reboot = 'restart'
 
 on_poweroff = 'destroy'
 on_reboot = 'restart'

-on_crash = restart
+on_crash = 'destroy'
+if machine.autorestart:
+    on_crash = 'restart'