-"""
-Functions to perform remctls.
-"""
-
import validation
-from webcommon import CodeError, InvalidInput
+from invirt.common import CodeError, InvalidInput
import random
-import subprocess
import sys
import time
import re
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():
"%02x" * 6]) % tuple(u)
# end stolen code
-def kinit():
- """Kinit with a given username and keytab"""
- p = subprocess.Popen(['kinit', "-k", "-t", '/etc/invirt/keytab',
- 'daemon/'+config.web.hostname],
- 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', config.remote.hostname]
- + 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"""
def lvinstall(machine, autoinstall):
disksize = machine.disks[0].size
memsize = machine.memory
- imagesize = disksize - getswap(disksize, memsize)
+ swapsize = getswap(disksize, memsize)
+ imagesize = disksize - swapsize
ip = machine.nics[0].ip
- remctl('web', 'install', machine.name, autoinstall.distribution,
- autoinstall.mirror, str(imagesize), 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"""
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
session.save_or_update(machine)
disk = Disk(machine=machine,
guest_device_name='hda', size=disksize)
- nic = NIC.query().filter_by(machine_id=None).first()
+ 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 %s." % config.web.errormail)
session.save_or_update(nic)
session.save_or_update(disk)
cache_acls.refreshMachine(machine)
+ makeDisks(machine)
session.commit()
except:
session.rollback()
raise
- makeDisks(machine)
- if autoinstall:
- lvinstall(machine, autoinstall)
- # tell it to boot with cdrom
- bootMachine(machine, cdrom)
+ try:
+ if autoinstall:
+ lvinstall(machine, autoinstall)
+ else:
+ # tell it to boot with cdrom
+ bootMachine(machine, cdrom)
+ except CodeError, e:
+ deleteVM(machine)
+ raise
return machine
def getList():
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)
session.rollback()
raise
-def commandResult(username, state, fields):
+def commandResult(username, state, command_name, machine_id, fields):
start_time = 0
- 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):
+ machine = validation.Validate(username, state, machine_id=machine_id).machine
+ action = command_name
+ cdrom = fields.get('cdrom')
+ if not cdrom:
+ cdrom = None
+ 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'):
+ if action not in "reboot create destroy shutdown delete".split(" "):
raise CodeError("Invalid action '%s'" % action)
- if action == 'Reboot':
+ if action == 'reboot':
if cdrom is not None:
out, err = remctl('control', machine.name, 'reboot', cdrom,
err=True)
print >> sys.stderr, err
raise CodeError('ERROR on remctl')
- elif action == 'Power on':
+ elif action == 'create':
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.")
bootMachine(machine, cdrom)
- elif action == 'Power off':
+ elif action == 'destroy':
out, err = remctl('control', machine.name, 'destroy', err=True)
if err:
if re.match("machine '.*' is not on", err):
print >> sys.stderr, 'Error on power off:'
print >> sys.stderr, err
raise CodeError('ERROR on remctl')
- elif action == 'Shutdown':
+ elif action == 'shutdown':
out, err = remctl('control', machine.name, 'shutdown', err=True)
if err:
if re.match("machine '.*' is not on", err):
print >> sys.stderr, 'Error on Shutdown:'
print >> sys.stderr, err
raise CodeError('ERROR on remctl')
- elif action == 'Delete VM':
+ elif action == 'delete':
deleteVM(machine)
d = dict(user=username,
projects / invirt/packages/invirt-web.git / blobdiff