+ action = fields.getfirst('action')
+ cdrom = fields.getfirst('cdrom')
+ print >> sys.stderr, time.time()-start_time
+ if cdrom is not None and not CDROM.get(cdrom):
+ raise CodeError("Invalid cdrom type '%s'" % cdrom)
+ if action not in ('Reboot', 'Power on', 'Power off', 'Shutdown', 'Delete VM'):
+ raise CodeError("Invalid action '%s'" % action)
+ if action == 'Reboot':
+ if cdrom is not None:
+ remctl('reboot', machine.name, cdrom)
+ else:
+ remctl('reboot', machine.name)
+ elif action == 'Power on':
+ if maxMemory(user) < 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':
+ remctl('destroy', machine.name)
+ elif action == 'Shutdown':
+ remctl('shutdown', machine.name)
+ elif action == 'Delete VM':
+ deleteVM(machine)
+ print >> sys.stderr, time.time()-start_time
+
+ d = dict(user=user,
+ command=action,
+ machine=machine)
+ return Template(file="command.tmpl", searchList=[d, global_dict])
+
+def testOwner(user, owner, machine=None):
+ if owner == machine.owner: #XXX What do we do when you lose access to the locker?
+ return owner
+ value = getafsgroups.checkLockerOwner(user.username, owner, verbose=True)
+ if value == True:
+ return owner
+ raise InvalidInput('owner', owner, value)
+
+def testContact(user, contact, machine=None):
+ if not re.match("^[A-Z0-9._%+-]+@[A-Z0-9.-]+\.[A-Z]{2,4}$", contact, re.I):
+ raise InvalidInput('contact', contact, "Not a valid email")
+ return contact
+
+def testDisk(user, disksize, machine=None):
+ return disksize
+
+def testName(user, name, machine=None):
+ if name is None:
+ return None
+ if not Machine.select_by(name=name):
+ return name
+ if name == machine.name:
+ return name
+ raise InvalidInput('name', name, "Already taken")
+
+def testHostname(user, hostname, machine):
+ for nic in machine.nics:
+ if hostname == nic.hostname:
+ return hostname
+ # check if doesn't already exist
+ if NIC.select_by(hostname=hostname):
+ raise InvalidInput('hostname', hostname,
+ "Already exists")
+ if not re.match("^[A-Z0-9-]{1,22}$", hostname, re.I):
+ raise InvalidInput('hostname', hostname, "Not a valid hostname; must only use number, letters, and dashes.")
+ return hostname
+
+def modify(user, fields):
+ """Handler for modifying attributes of a machine."""
+
+ olddisk = {}
+ transaction = ctx.current.create_transaction()
+ try:
+ machine = testMachineId(user, fields.getfirst('machine_id'))
+ owner = testOwner(user, fields.getfirst('owner'), machine)
+ contact = testContact(user, fields.getfirst('contact'))
+ hostname = testHostname(owner, fields.getfirst('hostname'),
+ machine)
+ name = testName(user, fields.getfirst('name'), machine)
+ oldname = machine.name
+ command="modify"
+
+ memory = fields.getfirst('memory')
+ if memory is not None:
+ memory = validMemory(user, memory, machine, on=False)
+ machine.memory = memory
+
+ disksize = testDisk(user, fields.getfirst('disk'))
+ if disksize is not None:
+ disksize = validDisk(user, disksize, machine)
+ disk = machine.disks[0]
+ if disk.size != disksize:
+ olddisk[disk.guest_device_name] = disksize
+ disk.size = disksize
+ ctx.current.save(disk)
+
+ # XXX first NIC gets hostname on change? Interface doesn't support more.
+ for nic in machine.nics[:1]:
+ nic.hostname = hostname
+ ctx.current.save(nic)
+
+ if owner is not None and owner != machine.owner:
+ machine.owner = owner
+ if name is not None and name != machine.name:
+ machine.name = name
+
+ ctx.current.save(machine)
+ transaction.commit()
+ except:
+ transaction.rollback()
+ raise
+ for diskname in olddisk:
+ remctl("web", "lvresize", oldname, diskname, str(olddisk[diskname]))
+ if name is not None and name != oldname:
+ for disk in machine.disks:
+ if oldname != name:
+ remctl("web", "lvrename", oldname, disk.guest_device_name, name)
+ remctl("web", "moveregister", oldname, name)