+#!/usr/bin/python
+
+import sys
+import cgi
+import os
+import string
+import subprocess
+
+print 'Content-Type: text/html\n'
+sys.stderr = sys.stdout
+sys.path.append('/home/ecprice/.local/lib/python2.5/site-packages')
+
+from Cheetah.Template import Template
+from sipb_xen_database import *
+import random
+
+# ... and stolen from xend/uuid.py
+def randomUUID():
+ """Generate a random UUID."""
+
+ return [ random.randint(0, 255) for _ in range(0, 16) ]
+
+def uuidToString(u):
+ return "-".join(["%02x" * 4, "%02x" * 2, "%02x" * 2, "%02x" * 2,
+ "%02x" * 6]) % tuple(u)
+
+
+def maxMemory(user):
+ return 256
+
+def haveAccess(user, machine):
+ return True
+
+
+def error(op, user, fields, errorMessage):
+ d = dict(op=op,
+ user=user,
+ errorMessage=errorMessage)
+ print Template(file='error.tmpl',
+ searchList=d);
+
+def validMachineName(name):
+ if not name:
+ return False
+ charset = string.ascii_letters + string.digits + '-'
+ if name[0] == '-' or len(name) > 22:
+ return False
+ return all(x in charset for x in name)
+
+def kinit():
+ keytab = '/etc/tabbott.keytab'
+ username = 'tabbott/extra'
+ p = subprocess.Popen(['kinit', "-k", "-t", keytab,
+ username])
+ p.wait()
+
+def checkKinit():
+ p = subprocess.Popen(['klist', '-s'])
+ if p.wait():
+ kinit()
+
+def remctl(*args):
+ checkKinit()
+ p = subprocess.Popen(['remctl', 'black-mesa.mit.edu']
+ + list(args),
+ stdout=subprocess.PIPE,
+ stderr=subprocess.PIPE)
+ if p.wait():
+ print >> sys.stderr, 'ERROR on remctl ', args
+ print >> sys.stderr, p.stderr.read()
+
+def makeDisks():
+ remctl('lvcreate','all')
+
+def bootMachine(machine, cdtype):
+ if cdtype is not None:
+ remctl('vmboot', 'cdrom', str(machine.name),
+ cdtype)
+ else:
+ remctl('vmboot', 'cdrom', str(machine.name))
+
+def createVm(user, name, memory, disk, is_hvm, cdrom):
+ # put stuff in the table
+ transaction = ctx.current.create_transaction()
+ try:
+ res = meta.engine.execute('select nextval(\'"machines_machine_id_seq"\')')
+ id = res.fetchone()[0]
+ machine = Machine()
+ machine.machine_id = id
+ machine.name = name
+ machine.memory = memory
+ machine.owner = user.username
+ machine.contact = user.email
+ machine.uuid = uuidToString(randomUUID())
+ machine.boot_off_cd = True
+ machine_type = Type.get_by(hvm=is_hvm)
+ machine.type_id = machine_type.type_id
+ ctx.current.save(machine)
+ disk = Disk(machine.machine_id,
+ 'hda', disk)
+ open = NIC.select_by(machine_id=None)
+ if not open: #No IPs left!
+ return "No IP addresses left! Contact sipb-xen-dev@mit.edu"
+ nic = open[0]
+ nic.machine_id = machine.machine_id
+ nic.hostname = name
+ ctx.current.save(nic)
+ ctx.current.save(disk)
+ transaction.commit()
+ except:
+ transaction.rollback()
+ raise
+ makeDisks()
+ # tell it to boot with cdrom
+ bootMachine(machine, cdrom)
+
+ return machine
+
+def create(user, fields):
+ name = fields.getfirst('name')
+ if not validMachineName(name):
+ return error('create', user, fields,
+ "Invalid name '%s'" % name)
+ name = name.lower()
+
+ if Machine.get_by(name=name):
+ return error('create', user, fields,
+ "A machine named '%s' already exists" % name)
+
+ memory = fields.getfirst('memory')
+ try:
+ memory = int(memory)
+ if memory <= 0:
+ raise ValueError
+ except ValueError:
+ return error('create', user, fields,
+ "Invalid memory amount")
+ if memory > maxMemory(user):
+ return error('create', user, fields,
+ "Too much memory requested")
+
+ disk = fields.getfirst('disk')
+ try:
+ disk = float(disk)
+ disk = int(disk * 1024)
+ if disk <= 0:
+ raise ValueError
+ except ValueError:
+ return error('create', user, fields,
+ "Invalid disk amount")
+
+ vm_type = fields.getfirst('vmtype')
+ if vm_type not in ('hvm', 'paravm'):
+ return error('create', user, fields,
+ "Invalid vm type '%s'" % vm_type)
+ is_hvm = (vm_type == 'hvm')
+
+ cdrom = fields.getfirst('cdrom')
+ if cdrom is not None and not CDROM.get(cdrom):
+ return error('create', user, fields,
+ "Invalid cdrom type '%s'" % cdrom)
+
+ machine = createVm(user, name, memory, disk, is_hvm, cdrom)
+ if isinstance(machine, basestring):
+ return error('create', user, fields,
+ machine)
+ d = dict(user=user,
+ machine=machine)
+ print Template(file='create.tmpl',
+ searchList=d);
+
+def listVms(user, fields):
+ machines = Machine.select()
+ d = dict(user=user,
+ machines=machines,
+ cdroms=CDROM.select())
+
+ print Template(file='list.tmpl', searchList=d)
+
+def testMachineId(user, machineId, exists=True):
+ if machineId is None:
+ error('vnc', user, fields,
+ "No machine ID specified")
+ return False
+ try:
+ machineId = int(machineId)
+ except ValueError:
+ error('vnc', user, fields,
+ "Invalid machine ID '%s'"
+ % machineId)
+ return False
+ machine = Machine.get(machineId)
+ if exists and machine is None:
+ error('vnc', user, fields,
+ "No such machine ID '%s'"
+ % machineId)
+ return False
+ if not haveAccess(user, machine):
+ error('vnc', user, fields,
+ "No access to machine ID '%s'"
+ % machineId)
+ return False
+ return machine
+
+def vnc(user, fields):
+ machine = testMachineId(user, fields.getfirst('machine_id'))
+ if machine is None: #gave error page already
+ return
+ token = 'quentin'
+ d = dict(user=user,
+ machine=machine,
+ hostname='localhost',
+ authtoken=token)
+ print Template(file='vnc.tmpl',
+ searchList=d)
+
+def info(user, fields):
+ machine = testMachineId(user, fields.getfirst('machine_id'))
+ if machine is None: #gave error page already
+ return
+
+ d = dict(user=user,
+ machine=machine)
+ print Template(file='info.tmpl',
+ searchList=d)
+
+mapping = dict(list=listVms,
+ vnc=vnc,
+ info=info,
+ create=create)
+
+if __name__ == '__main__':
+ fields = cgi.FieldStorage()
+ class C:
+ username = "moo"
+ email = 'moo@cow.com'
+ u = C()
+ connect('postgres://sipb-xen@sipb-xen-dev/sipb_xen')
+ operation = os.environ.get('PATH_INFO', '')
+ if operation.startswith('/'):
+ operation = operation[1:]
+ if not operation:
+ operation = 'list'
+
+ fun = mapping.get(operation,
+ lambda u, e:
+ error(operation, u, e,
+ "Invalid operation '%'" % operation))
+ fun(u, fields)
projects / invirt/packages/invirt-web.git / commitdiff
