| [101] | 1 | # -*- mode: python; -*- |
|---|
| [24] | 2 | import sipb_xen_database.models as models |
|---|
| [85] | 3 | from sipb_xen_database import connect |
|---|
| [109] | 4 | import re |
|---|
| [223] | 5 | import tempfile |
|---|
| 6 | from subprocess import call |
|---|
| [24] | 7 | |
|---|
| [99] | 8 | connect('postgres://sipb-xen@sipb-xen-dev.mit.edu/sipb_xen') |
|---|
| [146] | 9 | prefix = "d_" |
|---|
| [24] | 10 | |
|---|
| [109] | 11 | # 'machine_name', and optionally 'cdrom_image', should get passed in |
|---|
| 12 | # from the xm create call |
|---|
| 13 | |
|---|
| 14 | def check(b): |
|---|
| 15 | if not b: |
|---|
| [117] | 16 | import sys |
|---|
| 17 | sys.exit(1) |
|---|
| [109] | 18 | |
|---|
| [85] | 19 | machine = models.Machine.get_by(name=machine_name) |
|---|
| [109] | 20 | check(machine is not None) |
|---|
| [65] | 21 | machine_type = models.Type.get_by(type_id=machine.type_id) |
|---|
| [24] | 22 | |
|---|
| 23 | memory = machine.memory |
|---|
| 24 | maxmem = memory |
|---|
| [109] | 25 | check(re.match('^[A-Za-z0-9][A-Za-z0-9._-]*$', machine.name)) |
|---|
| [146] | 26 | name = prefix + machine.name |
|---|
| [109] | 27 | check(re.match('^[0-9a-f-]+$', machine.uuid)) |
|---|
| [24] | 28 | uuid = machine.uuid |
|---|
| 29 | |
|---|
| 30 | vcpus = machine.cpus |
|---|
| 31 | |
|---|
| [65] | 32 | diskioemu = "" |
|---|
| 33 | viftype = "" |
|---|
| [24] | 34 | |
|---|
| [65] | 35 | if machine_type.hvm: |
|---|
| 36 | ioemu = "ioemu:" |
|---|
| [101] | 37 | viftype = "type=ioemu, " |
|---|
| [65] | 38 | |
|---|
| 39 | kernel = 'hvmloader' |
|---|
| 40 | builder = 'hvm' |
|---|
| 41 | vnc = 1 |
|---|
| [104] | 42 | vncpasswd = 'moocow' |
|---|
| 43 | device_model = '/usr/sbin/qemu-dm-sipb' |
|---|
| [65] | 44 | else: |
|---|
| 45 | kernel = '/boot/vmlinuz-2.6.18-4-xen-amd64' |
|---|
| 46 | ramdisk = '/boot/initrd.img-2.6.18-4-xen-amd64' |
|---|
| 47 | builder = 'linux' |
|---|
| 48 | |
|---|
| 49 | pae = machine_type.pae |
|---|
| 50 | acpi = machine_type.acpi |
|---|
| 51 | apic = machine_type.apic |
|---|
| 52 | |
|---|
| 53 | vif = [] |
|---|
| 54 | |
|---|
| 55 | for n in machine.nics: |
|---|
| [109] | 56 | check(re.match('^[0-9a-fA-F:]+$', n.mac_addr) and re.match('^[0-9.]*$', n.ip)) |
|---|
| [101] | 57 | d = '%smac=%s, ip=%s, bridge=xenbr0, script=vif-sipbroute' % (viftype, n.mac_addr, n.ip) |
|---|
| [84] | 58 | vif.append(d) |
|---|
| [65] | 59 | |
|---|
| [24] | 60 | disk = [] |
|---|
| 61 | |
|---|
| 62 | for d in machine.disks: |
|---|
| [109] | 63 | check(re.match('^[A-Za-z0-9]+$', d.guest_device_name)) |
|---|
| [146] | 64 | device = '/dev/xenvg/' + prefix + machine.name + '_' + d.guest_device_name |
|---|
| [65] | 65 | dspec = 'phy:%s,%s%s,w' % (device, diskioemu, d.guest_device_name) |
|---|
| [24] | 66 | disk.append(dspec) |
|---|
| 67 | |
|---|
| [223] | 68 | if 'installer' in locals(): |
|---|
| 69 | check(re.match('^[A-Za-z0-9][A-Za-z0-9_.-]*$', installer)) |
|---|
| 70 | tmptree = tempfile.mkdtemp('', 'auto-install.', '/tmp') |
|---|
| 71 | call(['/usr/sbin/sipb-xen-make-iso', installer, tmptree] |
|---|
| 72 | + installer_options.split(' ')) |
|---|
| 73 | disk.append('file:'+tmptree+'/install.iso,hdc:cdrom,r') |
|---|
| 74 | boot = 'd' |
|---|
| 75 | |
|---|
| 76 | elif 'cdrom_image' in locals(): |
|---|
| [109] | 77 | check(re.match('^[A-Za-z0-9][A-Za-z0-9_.-]*$', cdrom_image)) |
|---|
| 78 | disk.append('file:/srv/images/' + cdrom_image + '.iso,hdc:cdrom,r') |
|---|
| 79 | boot = 'd' |
|---|
| [24] | 80 | |
|---|
| 81 | on_poweroff = 'destroy' |
|---|
| 82 | on_reboot = 'restart' |
|---|
| [109] | 83 | on_crash = 'destroy' |
|---|
| 84 | if machine.autorestart: |
|---|
| 85 | on_crash = 'restart' |
|---|
