-"""
-Functions to perform remctls.
-"""
-
-from sipb_xen_database import Machine, Disk, Type, NIC, CDROM, ctx, meta
import validation
-from webcommon import CodeError, InvalidInput
+from invirt.common import CodeError, InvalidInput
import random
-import subprocess
import sys
import time
import re
import cache_acls
import yaml
+from invirt.config import structs as config
+from invirt.database import Machine, Disk, Type, NIC, CDROM, session, meta
+from invirt.remctl import remctl as gen_remctl
+
# ... and stolen from xend/uuid.py
def randomUUID():
"""Generate a random UUID."""
"%02x" * 6]) % tuple(u)
# end stolen code
-def kinit(username = 'daemon/sipb-xen.mit.edu', keytab = '/etc/sipb-xen.keytab'):
- """Kinit with a given username and keytab"""
-
- p = subprocess.Popen(['kinit', "-k", "-t", keytab, username],
- stderr=subprocess.PIPE)
- e = p.wait()
- if e:
- raise CodeError("Error %s in kinit: %s" % (e, p.stderr.read()))
-
-def checkKinit():
- """If we lack tickets, kinit."""
- p = subprocess.Popen(['klist', '-s'])
- if p.wait():
- kinit()
-
-def remctl(*args, **kws):
- """Perform a remctl and return the output.
-
- kinits if necessary, and outputs errors to stderr.
- """
- checkKinit()
- p = subprocess.Popen(['remctl', 'remote.mit.edu']
- + list(args),
- stdout=subprocess.PIPE,
- stderr=subprocess.PIPE)
- v = p.wait()
- if kws.get('err'):
- return p.stdout.read(), p.stderr.read()
- if v:
- print >> sys.stderr, 'Error', v, 'on remctl', args, ':'
- print >> sys.stderr, p.stderr.read()
- raise CodeError('ERROR on remctl')
- return p.stdout.read()
+def remctl(*args, **kwargs):
+ return gen_remctl(config.remote.hostname,
+ principal='daemon/'+config.web.hostname,
+ *args, **kwargs)
def lvcreate(machine, disk):
"""Create a single disk for a machine"""
for disk in machine.disks:
lvcreate(machine, disk)
+def getswap(disksize, memsize):
+ """Returns the recommended swap partition size."""
+ return int(min(disksize / 4, memsize * 1.5))
+
+def lvinstall(machine, autoinstall):
+ disksize = machine.disks[0].size
+ memsize = machine.memory
+ swapsize = getswap(disksize, memsize)
+ imagesize = disksize - swapsize
+ ip = machine.nics[0].ip
+ remctl('control', machine.name, 'install',
+ 'dist=%s' % autoinstall.distribution,
+ 'mirror=%s' % autoinstall.mirror,
+ 'arch=%s' % autoinstall.arch,
+ 'imagesize=%s' % imagesize)
+
def lvcopy(machine_orig_name, machine, rootpw):
"""Copy a golden image onto a machine's disk"""
remctl('web', 'lvcopy', machine_orig_name, machine.name, rootpw)
else:
out, err = remctl('control', machine.name, 'create',
err=True)
- if 'already exists' in out:
+ if 'already running' in err:
raise InvalidInput('action', 'create',
'VM %s is already on' % machine.name)
elif err:
raise CodeError('"%s" on "control %s create %s'
% (err, machine.name, cdtype))
-def createVm(owner, contact, name, memory, disk_size, machine_type, cdrom, clone_from):
+def createVm(username, state, owner, contact, name, description, memory, disksize, machine_type, cdrom, autoinstall):
"""Create a VM and put it in the database"""
# put stuff in the table
- transaction = ctx.current.create_transaction()
+ session.begin()
try:
- validation.validMemory(owner, memory)
- validation.validDisk(owner, disk_size * 1. / 1024)
- validation.validAddVm(owner)
- res = meta.engine.execute('select nextval('
- '\'"machines_machine_id_seq"\')')
- id = res.fetchone()[0]
+ validation.Validate(username, state, name=name, description=description, owner=owner, memory=memory, disksize=disksize/1024.)
machine = Machine()
- machine.machine_id = id
machine.name = name
+ machine.description = description
machine.memory = memory
machine.owner = owner
- machine.administrator = owner
+ machine.administrator = None
machine.contact = contact
machine.uuid = uuidToString(randomUUID())
machine.boot_off_cd = True
- machine.type_id = machine_type.type_id
- ctx.current.save(machine)
- disk = Disk(machine_id=machine.machine_id,
- guest_device_name='hda', size=disk_size)
- open_nics = NIC.select_by(machine_id=None)
- if not open_nics: #No IPs left!
+ machine.type = machine_type
+ session.save_or_update(machine)
+ disk = Disk(machine=machine,
+ guest_device_name='hda', size=disksize)
+ nic = NIC.query().filter_by(machine_id=None).filter_by(reusable=True).first()
+ if not nic: #No IPs left!
raise CodeError("No IP addresses left! "
- "Contact xvm@mit.edu.")
- nic = open_nics[0]
- nic.machine_id = machine.machine_id
+ "Contact %s." % config.web.errormail)
+ nic.machine = machine
nic.hostname = name
- ctx.current.save(nic)
- ctx.current.save(disk)
+ session.save_or_update(nic)
+ session.save_or_update(disk)
cache_acls.refreshMachine(machine)
- transaction.commit()
+ makeDisks(machine)
+ session.commit()
except:
- transaction.rollback()
+ session.rollback()
+ raise
+ try:
+ if autoinstall:
+ lvinstall(machine, autoinstall)
+ else:
+ # tell it to boot with cdrom
+ bootMachine(machine, cdrom)
+ except CodeError, e:
+ deleteVM(machine)
raise
- makeDisks(machine)
- if clone_from:
- lvcopy(clone_from, machine, 'password')
- # tell it to boot with cdrom
- bootMachine(machine, cdrom)
return machine
def getList():
if 'Unknown command' in err_string:
raise CodeError("ERROR in remctl list-long %s is not registered" %
(machine.name,))
- elif 'does not exist' in err_string:
+ elif 'is not on' in err_string:
return None
elif err_string:
raise CodeError("ERROR in remctl list-long %s: %s" %
status = parseStatus(value_string)
return status
+def listHost(machine):
+ """Return the host a machine is running on"""
+ out, err = remctl('control', machine.name, 'listhost', err=True)
+ if err:
+ return None
+ return out.strip()
+
+def vnctoken(machine):
+ """Return a time-stamped VNC token"""
+ out, err = remctl('control', machine.name, 'vnctoken', err=True)
+ if err:
+ return None
+ return out.strip()
+
def deleteVM(machine):
"""Delete a VM."""
remctl('control', machine.name, 'destroy', err=True)
- transaction = ctx.current.create_transaction()
+ session.begin()
delete_disk_pairs = [(machine.name, d.guest_device_name)
for d in machine.disks]
try:
+ for mname, dname in delete_disk_pairs:
+ remctl('web', 'lvremove', mname, dname)
for nic in machine.nics:
nic.machine_id = None
nic.hostname = None
- ctx.current.save(nic)
+ session.save_or_update(nic)
for disk in machine.disks:
- ctx.current.delete(disk)
- for access in machine.acl:
- ctx.current.delete(access)
- ctx.current.delete(machine)
- transaction.commit()
+ session.delete(disk)
+ session.delete(machine)
+ session.commit()
except:
- transaction.rollback()
+ session.rollback()
raise
- for mname, dname in delete_disk_pairs:
- remctl('web', 'lvremove', mname, dname)
-def commandResult(user, fields):
+def commandResult(username, state, fields):
start_time = 0
- machine = validation.testMachineId(user, fields.getfirst('machine_id'))
+ machine = validation.Validate(username, state, machine_id=fields.getfirst('machine_id')).machine
action = fields.getfirst('action')
cdrom = fields.getfirst('cdrom')
- if cdrom is not None and not CDROM.get(cdrom):
+ if cdrom is not None and not CDROM.query().filter_by(cdrom_id=cdrom).one():
raise CodeError("Invalid cdrom type '%s'" % cdrom)
if action not in ('Reboot', 'Power on', 'Power off', 'Shutdown',
'Delete VM'):
out, err = remctl('control', machine.name, 'reboot',
err=True)
if err:
- if re.match("Error: Domain '.*' does not exist.", err):
+ if re.match("machine '.*' is not on", err):
raise InvalidInput("action", "reboot",
"Machine is not on")
else:
raise CodeError('ERROR on remctl')
elif action == 'Power on':
- if validation.maxMemory(user, machine) < machine.memory:
+ if validation.maxMemory(username, state, machine) < machine.memory:
raise InvalidInput('action', 'Power on',
"You don't have enough free RAM quota "
"to turn on this machine.")
elif action == 'Power off':
out, err = remctl('control', machine.name, 'destroy', err=True)
if err:
- if re.match("Error: Domain '.*' does not exist.", err):
+ if re.match("machine '.*' is not on", err):
raise InvalidInput("action", "Power off",
"Machine is not on.")
else:
elif action == 'Shutdown':
out, err = remctl('control', machine.name, 'shutdown', err=True)
if err:
- if re.match("Error: Domain '.*' does not exist.", err):
+ if re.match("machine '.*' is not on", err):
raise InvalidInput("action", "Shutdown",
"Machine is not on.")
else:
elif action == 'Delete VM':
deleteVM(machine)
- d = dict(user=user,
+ d = dict(user=username,
command=action,
machine=machine)
return d
projects / invirt/packages/invirt-web.git / blobdiff