source: package_branches/invirt-web/cherrypy/code/main.py @ 2525

Last change on this file since 2525 was 2525, checked in by price, 15 years ago

fix super-long lines

  • Property svn:executable set to *
File size: 28.5 KB
RevLine 
[113]1#!/usr/bin/python
[205]2"""Main CGI script for web interface"""
[113]3
[2449]4from __future__ import with_statement
5
[205]6import base64
7import cPickle
[113]8import cgi
[205]9import datetime
10import hmac
[770]11import random
[205]12import sha
13import sys
[2449]14import threading
[118]15import time
[447]16import urllib
[2186]17import socket
[2389]18import cherrypy
[2485]19from cherrypy import _cperror
[205]20from StringIO import StringIO
[2449]21
[205]22def printError():
23    """Revert stderr to stdout, and print the contents of stderr"""
24    if isinstance(sys.stderr, StringIO):
25        print revertStandardError()
26
27if __name__ == '__main__':
28    import atexit
29    atexit.register(printError)
30
[209]31import validation
[446]32import cache_acls
[1612]33from webcommon import State
[209]34import controls
[632]35from getafsgroups import getAfsGroupMembers
[865]36from invirt import database
[1001]37from invirt.database import Machine, CDROM, session, connect, MachineAccess, Type, Autoinstall
[863]38from invirt.config import structs as config
[1612]39from invirt.common import InvalidInput, CodeError
[2481]40import invirt.remctl
[113]41
[2485]42from view import View, revertStandardError
[2432]43import ajaxterm
[2390]44
[2484]45class InvirtUnauthWeb(View):
46    @cherrypy.expose
47    @cherrypy.tools.mako(filename="/unauth.mako")
48    def index(self):
49        return {'simple': True}
50
[2390]51class InvirtWeb(View):
52    def __init__(self):
53        super(self.__class__,self).__init__()
54        connect()
[2391]55        self._cp_config['tools.require_login.on'] = True
[2485]56        self._cp_config['tools.catch_stderr.on'] = True
[2403]57        self._cp_config['tools.mako.imports'] = ['from invirt.config import structs as config',
58                                                 'from invirt import database']
[2485]59        self._cp_config['request.error_response'] = self.handle_error
[2390]60
[2485]61    @cherrypy.expose
62    @cherrypy.tools.mako(filename="/invalid.mako")
63    def invalidInput(self):
64        """Print an error page when an InvalidInput exception occurs"""
65        err = cherrypy.request.prev.params["err"]
66        emsg = cherrypy.request.prev.params["emsg"]
67        d = dict(err_field=err.err_field,
68                 err_value=str(err.err_value), stderr=emsg,
69                 errorMessage=str(err))
70        return d
71
72    @cherrypy.expose
73    @cherrypy.tools.mako(filename="/error.mako")
74    def error(self):
75        """Print an error page when an exception occurs"""
76        op = cherrypy.request.prev.path_info
77        username = cherrypy.request.login
78        err = cherrypy.request.prev.params["err"]
79        emsg = cherrypy.request.prev.params["emsg"]
80        traceback = cherrypy.request.prev.params["traceback"]
[2523]81        d = dict(op=op, user=username, fields=cherrypy.request.prev.params,
[2485]82                 errorMessage=str(err), stderr=emsg, traceback=traceback)
[2486]83        error_raw = cherrypy.request.lookup.get_template("/error_raw.mako")
[2485]84        details = error_raw.render(**d)
85        exclude = config.web.errormail_exclude
86        if username not in exclude and '*' not in exclude:
87            send_error_mail('xvm error on %s for %s: %s' % (op, cherrypy.request.login, err),
88                            details)
89        d['details'] = details
90        return d
91
[2482]92    def __getattr__(self, name):
93        if name in ("admin", "overlord"):
94            if not cherrypy.request.login in getAfsGroupMembers(config.adminacl, config.authz[0].cell):
95                raise InvalidInput('username', cherrypy.request.login,
96                                   'Not in admin group %s.' % config.adminacl)
97            cherrypy.request.state = State(cherrypy.request.login, isadmin=True)
98            return self
99        else:
100            return super(InvirtWeb, self).__getattr__(name)
[2403]101
[2485]102    def handle_error(self):
103        err = sys.exc_info()[1]
104        if isinstance(err, InvalidInput):
105            cherrypy.request.params['err'] = err
[2524]106            cherrypy.request.params['emsg'] = revertStandardError()
[2485]107            raise cherrypy.InternalRedirect('/invalidInput')
108        if not cherrypy.request.prev or 'err' not in cherrypy.request.prev.params:
109            cherrypy.request.params['err'] = err
[2524]110            cherrypy.request.params['emsg'] = revertStandardError()
[2485]111            cherrypy.request.params['traceback'] = _cperror.format_exc()
112            raise cherrypy.InternalRedirect('/error')
113        # fall back to cherrypy default error page
114        cherrypy.HTTPError(500).set_response()
115
[2390]116    @cherrypy.expose
[2391]117    @cherrypy.tools.mako(filename="/list.mako")
[2418]118    def list(self, result=None):
[2391]119        """Handler for list requests."""
120        checkpoint.checkpoint('Getting list dict')
[2396]121        d = getListDict(cherrypy.request.login, cherrypy.request.state)
[2418]122        if result is not None:
123            d['result'] = result
[2391]124        checkpoint.checkpoint('Got list dict')
[2395]125        return d
[2391]126    index=list
127
128    @cherrypy.expose
[2409]129    @cherrypy.tools.mako(filename="/help.mako")
130    def help(self, subject=None, simple=False):
131        """Handler for help messages."""
132
133        help_mapping = {
134            'Autoinstalls': """
135The autoinstaller builds a minimal Debian or Ubuntu system to run as a
136ParaVM.  You can access the resulting system by logging into the <a
137href="help?simple=true&subject=ParaVM+Console">serial console server</a>
138with your Kerberos tickets; there is no root password so sshd will
139refuse login.</p>
140
141<p>Under the covers, the autoinstaller uses our own patched version of
142xen-create-image, which is a tool based on debootstrap.  If you log
143into the serial console while the install is running, you can watch
144it.
145""",
146            'ParaVM Console': """
147ParaVM machines do not support local console access over VNC.  To
148access the serial console of these machines, you can SSH with Kerberos
149to %s, using the name of the machine as your
150username.""" % config.console.hostname,
151            'HVM/ParaVM': """
152HVM machines use the virtualization features of the processor, while
153ParaVM machines rely on a modified kernel to communicate directly with
154the hypervisor.  HVMs support boot CDs of any operating system, and
155the VNC console applet.  The three-minute autoinstaller produces
156ParaVMs.  ParaVMs typically are more efficient, and always support the
157<a href="help?subject=ParaVM+Console">console server</a>.</p>
158
159<p>More details are <a
160href="https://xvm.scripts.mit.edu/wiki/Paravirtualization">on the
161wiki</a>, including steps to prepare an HVM guest to boot as a ParaVM
162(which you can skip by using the autoinstaller to begin with.)</p>
163
164<p>We recommend using a ParaVM when possible and an HVM when necessary.
165""",
166            'CPU Weight': """
167Don't ask us!  We're as mystified as you are.""",
168            'Owner': """
169The owner field is used to determine <a
170href="help?subject=Quotas">quotas</a>.  It must be the name of a
171locker that you are an AFS administrator of.  In particular, you or an
172AFS group you are a member of must have AFS rlidwka bits on the
173locker.  You can check who administers the LOCKER locker using the
174commands 'attach LOCKER; fs la /mit/LOCKER' on Athena.)  See also <a
175href="help?subject=Administrator">administrator</a>.""",
176            'Administrator': """
177The administrator field determines who can access the console and
178power on and off the machine.  This can be either a user or a moira
179group.""",
180            'Quotas': """
181Quotas are determined on a per-locker basis.  Each locker may have a
182maximum of 512 mebibytes of active ram, 50 gibibytes of disk, and 4
183active machines.""",
184            'Console': """
185<strong>Framebuffer:</strong> At a Linux boot prompt in your VM, try
186setting <tt>fb=false</tt> to disable the framebuffer.  If you don't,
187your machine will run just fine, but the applet's display of the
188console will suffer artifacts.
189""",
190            'Windows': """
191<strong>Windows Vista:</strong> The Vista image is licensed for all MIT students and will automatically activate off the network; see <a href="/static/msca-email.txt">the licensing confirmation e-mail</a> for details. The installer requires 512 MiB RAM and at least 7.5 GiB disk space (15 GiB or more recommended).<br>
[2423]192<strong>Windows XP:</strong> This is the volume license CD image. You will need your own volume license key to complete the install. We do not have these available for the general MIT community; ask your department if they have one, or visit <a href="http://msca.mit.edu/">http://msca.mit.edu/</a> if you are staff/faculty to request one.
[2409]193"""
194            }
195
196        if not subject:
197            subject = sorted(help_mapping.keys())
198        if not isinstance(subject, list):
199            subject = [subject]
200
[2410]201        return dict(simple=simple,
[2409]202                    subjects=subject,
203                    mapping=help_mapping)
204    help._cp_config['tools.require_login.on'] = False
205
[2422]206    def parseCreate(self, fields):
[2525]207        kws = dict([(kw, fields.get(kw)) for kw in
208         'name description owner memory disksize vmtype cdrom autoinstall'.split()
209                    if fields.get(kw)])
210        validate = validation.Validate(cherrypy.request.login,
211                                       cherrypy.request.state,
212                                       strict=True, **kws)
213        return dict(contact=cherrypy.request.login, name=validate.name,
214                    description=validate.description, memory=validate.memory,
215                    disksize=validate.disksize, owner=validate.owner,
216                    machine_type=getattr(validate, 'vmtype', Defaults.type),
[2422]217                    cdrom=getattr(validate, 'cdrom', None),
218                    autoinstall=getattr(validate, 'autoinstall', None))
219
[2409]220    @cherrypy.expose
[2422]221    @cherrypy.tools.mako(filename="/list.mako")
222    @cherrypy.tools.require_POST()
223    def create(self, **fields):
224        """Handler for create requests."""
225        try:
226            parsed_fields = self.parseCreate(fields)
[2525]227            machine = controls.createVm(cherrypy.request.login,
228                                        cherrypy.request.state, **parsed_fields)
[2422]229        except InvalidInput, err:
230            pass
231        else:
232            err = None
233        cherrypy.request.state.clear() #Changed global state
234        d = getListDict(cherrypy.request.login, cherrypy.request.state)
235        d['err'] = err
236        if err:
237            for field in fields.keys():
238                setattr(d['defaults'], field, fields.get(field))
239        else:
240            d['new_machine'] = parsed_fields['name']
241        return d
242
243    @cherrypy.expose
[2391]244    @cherrypy.tools.mako(filename="/helloworld.mako")
[2408]245    def helloworld(self, **kwargs):
246        return {'request': cherrypy.request, 'kwargs': kwargs}
[2391]247    helloworld._cp_config['tools.require_login.on'] = False
[2390]248
[2428]249    @cherrypy.expose
250    def errortest(self):
251        """Throw an error, to test the error-tracing mechanisms."""
[2485]252        print >>sys.stderr, "look ma, it's a stderr"
[2428]253        raise RuntimeError("test of the emergency broadcast system")
254
[2413]255    class MachineView(View):
256        # This is hairy. Fix when CherryPy 3.2 is out. (rename to
257        # _cp_dispatch, and parse the argument as a list instead of
258        # string
259
260        def __getattr__(self, name):
261            try:
262                machine_id = int(name)
263                cherrypy.request.params['machine_id'] = machine_id
264                return self
265            except ValueError:
266                return None
267
268        @cherrypy.expose
269        @cherrypy.tools.mako(filename="/info.mako")
270        def info(self, machine_id):
271            """Handler for info on a single VM."""
[2525]272            machine = validation.Validate(cherrypy.request.login,
273                                          cherrypy.request.state,
274                                          machine_id=machine_id).machine
[2413]275            d = infoDict(cherrypy.request.login, cherrypy.request.state, machine)
276            checkpoint.checkpoint('Got infodict')
277            return d
278        index = info
279
[2414]280        @cherrypy.expose
[2483]281        @cherrypy.tools.mako(filename="/info.mako")
282        @cherrypy.tools.require_POST()
283        def modify(self, machine_id, **fields):
284            """Handler for modifying attributes of a machine."""
285            try:
[2525]286                modify_dict = modifyDict(cherrypy.request.login,
287                                         cherrypy.request.state,
288                                         machine_id, fields)
[2483]289            except InvalidInput, err:
290                result = None
[2525]291                machine = validation.Validate(cherrypy.request.login,
292                                              cherrypy.request.state,
293                                              machine_id=machine_id).machine
[2483]294            else:
295                machine = modify_dict['machine']
296                result = 'Success!'
297                err = None
[2525]298            info_dict = infoDict(cherrypy.request.login,
299                                 cherrypy.request.state, machine)
[2483]300            info_dict['err'] = err
301            if err:
302                for field in fields.keys():
303                    setattr(info_dict['defaults'], field, fields.get(field))
304            info_dict['result'] = result
305            return info_dict
306
307        @cherrypy.expose
[2414]308        @cherrypy.tools.mako(filename="/vnc.mako")
309        def vnc(self, machine_id):
310            """VNC applet page.
311
312            Note that due to same-domain restrictions, the applet connects to
313            the webserver, which needs to forward those requests to the xen
314            server.  The Xen server runs another proxy that (1) authenticates
315            and (2) finds the correct port for the VM.
316
317            You might want iptables like:
318
319            -t nat -A PREROUTING -s ! 18.181.0.60 -i eth1 -p tcp -m tcp \
320            --dport 10003 -j DNAT --to-destination 18.181.0.60:10003
321            -t nat -A POSTROUTING -d 18.181.0.60 -o eth1 -p tcp -m tcp \
322            --dport 10003 -j SNAT --to-source 18.187.7.142
323            -A FORWARD -d 18.181.0.60 -i eth1 -o eth1 -p tcp -m tcp \
324            --dport 10003 -j ACCEPT
325
326            Remember to enable iptables!
327            echo 1 > /proc/sys/net/ipv4/ip_forward
328            """
[2525]329            machine = validation.Validate(cherrypy.request.login,
330                                          cherrypy.request.state,
331                                          machine_id=machine_id).machine
[2414]332            token = controls.vnctoken(machine)
333            host = controls.listHost(machine)
334            if host:
335                port = 10003 + [h.hostname for h in config.hosts].index(host)
336            else:
337                port = 5900 # dummy
338
339            status = controls.statusInfo(machine)
340            has_vnc = hasVnc(status)
341
342            d = dict(on=status,
343                     has_vnc=has_vnc,
344                     machine=machine,
345                     hostname=cherrypy.request.local.name,
346                     port=port,
347                     authtoken=token)
348            return d
[2418]349        @cherrypy.expose
350        @cherrypy.tools.mako(filename="/command.mako")
[2422]351        @cherrypy.tools.require_POST()
[2418]352        def command(self, command_name, machine_id, **kwargs):
353            """Handler for running commands like boot and delete on a VM."""
354            back = kwargs.get('back', None)
355            try:
[2525]356                d = controls.commandResult(cherrypy.request.login,
357                                           cherrypy.request.state,
358                                           command_name, machine_id, kwargs)
[2418]359                if d['command'] == 'Delete VM':
360                    back = 'list'
361            except InvalidInput, err:
362                if not back:
363                    raise
364                print >> sys.stderr, err
[2485]365                result = str(err)
[2418]366            else:
367                result = 'Success!'
368                if not back:
369                    return d
370            if back == 'list':
371                cherrypy.request.state.clear() #Changed global state
[2525]372                raise cherrypy.InternalRedirect('/list?result=%s'
373                                                % urllib.quote(result))
[2418]374            elif back == 'info':
[2525]375                raise cherrypy.HTTPRedirect(cherrypy.request.base
376                                            + '/machine/%d/' % machine_id,
377                                            status=303)
[2418]378            else:
379                raise InvalidInput('back', back, 'Not a known back page.')
[2414]380
[2432]381        atmulti = ajaxterm.Multiplex()
382        atsessions = {}
[2449]383        atsessions_lock = threading.Lock()
[2432]384
385        @cherrypy.expose
386        @cherrypy.tools.mako(filename="/terminal.mako")
387        def terminal(self, machine_id):
388            machine = validation.Validate(cherrypy.request.login, cherrypy.request.state, machine_id=machine_id).machine
389
390            status = controls.statusInfo(machine)
391            has_vnc = hasVnc(status)
392
393            d = dict(on=status,
394                     has_vnc=has_vnc,
395                     machine=machine,
396                     hostname=cherrypy.request.local.name)
397            return d
398
[2433]399        @cherrypy.expose
[2452]400        @cherrypy.tools.require_POST()
[2440]401        @cherrypy.tools.gzip()
[2454]402        def at(self, machine_id, k=None, c=0, h=None):
[2433]403            machine = validation.Validate(cherrypy.request.login, cherrypy.request.state, machine_id=machine_id).machine
[2449]404            with self.atsessions_lock:
405                if machine_id in self.atsessions:
406                    term = self.atsessions[machine_id]
407                else:
408                    print >>sys.stderr, "spawning new session for terminal to ",machine_id
[2481]409                    invirt.remctl.checkKinit(principal='daemon/'+config.web.hostname)
[2450]410                    term = self.atmulti.create(
[2449]411                        ["ssh", "-e","none", "-l", machine.name, config.console.hostname]
412                        )
[2450]413                    # Clear out old sessions when fd is reused
414                    for key in self.atsessions:
415                        if self.atsessions[key] == term:
416                            del self.atsessions[key]
417                    self.atsessions[machine_id] = term
[2449]418                if k:
419                    self.atmulti.proc_write(term,k)
420                time.sleep(0.002)
[2454]421                dump=self.atmulti.dump(term,c,h)
[2449]422                cherrypy.response.headers['Content-Type']='text/xml'
423                if isinstance(dump,str):
424                    return dump
425                else:
[2481]426                    print "Removing session for", machine_id,"because we received",repr(dump)
[2449]427                    del self.atsessions[machine_id]
428                    return '<?xml version="1.0"?><idem></idem>'
[2433]429
[2413]430    machine = MachineView()
431
[235]432class Checkpoint:
433    def __init__(self):
434        self.start_time = time.time()
435        self.checkpoints = []
436
437    def checkpoint(self, s):
438        self.checkpoints.append((s, time.time()))
439
440    def __str__(self):
441        return ('Timing info:\n%s\n' %
442                '\n'.join(['%s: %s' % (d, t - self.start_time) for
443                           (d, t) in self.checkpoints]))
444
445checkpoint = Checkpoint()
446
[205]447class Defaults:
448    """Class to store default values for fields."""
449    memory = 256
450    disk = 4.0
451    cdrom = ''
[443]452    autoinstall = ''
[205]453    name = ''
[609]454    description = ''
[2483]455    administrator = ''
[515]456    type = 'linux-hvm'
457
[205]458    def __init__(self, max_memory=None, max_disk=None, **kws):
459        if max_memory is not None:
460            self.memory = min(self.memory, max_memory)
461        if max_disk is not None:
[1964]462            self.disk = min(self.disk, max_disk)
[205]463        for key in kws:
464            setattr(self, key, kws[key])
465
[119]466def hasVnc(status):
[133]467    """Does the machine with a given status list support VNC?"""
[119]468    if status is None:
469        return False
470    for l in status:
471        if l[0] == 'device' and l[1][0] == 'vfb':
472            d = dict(l[1][1:])
473            return 'location' in d
474    return False
475
[134]476
[572]477def getListDict(username, state):
[438]478    """Gets the list of local variables used by list.tmpl."""
[535]479    checkpoint.checkpoint('Starting')
[572]480    machines = state.machines
[235]481    checkpoint.checkpoint('Got my machines')
[133]482    on = {}
[119]483    has_vnc = {}
[2424]484    installing = {}
[572]485    xmlist = state.xmlist
[235]486    checkpoint.checkpoint('Got uptimes')
[136]487    for m in machines:
[535]488        if m not in xmlist:
[144]489            has_vnc[m] = 'Off'
[535]490            m.uptime = None
[136]491        else:
[535]492            m.uptime = xmlist[m]['uptime']
493            if xmlist[m]['console']:
494                has_vnc[m] = True
495            elif m.type.hvm:
496                has_vnc[m] = "WTF?"
497            else:
[2412]498                has_vnc[m] = "ParaVM"
[2424]499            if xmlist[m].get('autoinstall'):
500                installing[m] = True
501            else:
502                installing[m] = False
[572]503    max_memory = validation.maxMemory(username, state)
504    max_disk = validation.maxDisk(username)
[235]505    checkpoint.checkpoint('Got max mem/disk')
[205]506    defaults = Defaults(max_memory=max_memory,
507                        max_disk=max_disk,
[1739]508                        owner=username)
[235]509    checkpoint.checkpoint('Got defaults')
[424]510    def sortkey(machine):
[572]511        return (machine.owner != username, machine.owner, machine.name)
[424]512    machines = sorted(machines, key=sortkey)
[572]513    d = dict(user=username,
514             cant_add_vm=validation.cantAddVm(username, state),
[205]515             max_memory=max_memory,
[144]516             max_disk=max_disk,
[205]517             defaults=defaults,
[113]518             machines=machines,
[540]519             has_vnc=has_vnc,
[2424]520             installing=installing)
[205]521    return d
[113]522
[252]523def getHostname(nic):
[438]524    """Find the hostname associated with a NIC.
525
526    XXX this should be merged with the similar logic in DNS and DHCP.
527    """
[1976]528    if nic.hostname:
529        hostname = nic.hostname
[252]530    elif nic.machine:
[1976]531        hostname = nic.machine.name
[252]532    else:
533        return None
[1976]534    if '.' in hostname:
535        return hostname
536    else:
537        return hostname + '.' + config.dns.domains[0]
[252]538
[133]539def getNicInfo(data_dict, machine):
[145]540    """Helper function for info, get data on nics for a machine.
541
542    Modifies data_dict to include the relevant data, and returns a list
543    of (key, name) pairs to display "name: data_dict[key]" to the user.
544    """
[133]545    data_dict['num_nics'] = len(machine.nics)
[227]546    nic_fields_template = [('nic%s_hostname', 'NIC %s Hostname'),
[133]547                           ('nic%s_mac', 'NIC %s MAC Addr'),
548                           ('nic%s_ip', 'NIC %s IP'),
549                           ]
550    nic_fields = []
551    for i in range(len(machine.nics)):
552        nic_fields.extend([(x % i, y % i) for x, y in nic_fields_template])
[1976]553        data_dict['nic%s_hostname' % i] = getHostname(machine.nics[i])
[133]554        data_dict['nic%s_mac' % i] = machine.nics[i].mac_addr
555        data_dict['nic%s_ip' % i] = machine.nics[i].ip
556    if len(machine.nics) == 1:
557        nic_fields = [(x, y.replace('NIC 0 ', '')) for x, y in nic_fields]
558    return nic_fields
559
560def getDiskInfo(data_dict, machine):
[145]561    """Helper function for info, get data on disks for a machine.
562
563    Modifies data_dict to include the relevant data, and returns a list
564    of (key, name) pairs to display "name: data_dict[key]" to the user.
565    """
[133]566    data_dict['num_disks'] = len(machine.disks)
567    disk_fields_template = [('%s_size', '%s size')]
568    disk_fields = []
569    for disk in machine.disks:
570        name = disk.guest_device_name
[438]571        disk_fields.extend([(x % name, y % name) for x, y in
[205]572                            disk_fields_template])
[211]573        data_dict['%s_size' % name] = "%0.1f GiB" % (disk.size / 1024.)
[133]574    return disk_fields
575
[2483]576def modifyDict(username, state, machine_id, fields):
[438]577    """Modify a machine as specified by CGI arguments.
578
[2483]579    Return a dict containing the machine that was modified.
[438]580    """
[177]581    olddisk = {}
[1013]582    session.begin()
[161]583    try:
[2525]584        kws = dict([(kw, fields.get(kw)) for kw in
585         'owner admin contact name description memory vmtype disksize'.split()
586                    if fields.get(kw)])
[2483]587        kws['machine_id'] = machine_id
[572]588        validate = validation.Validate(username, state, **kws)
589        machine = validate.machine
[161]590        oldname = machine.name
[153]591
[572]592        if hasattr(validate, 'memory'):
593            machine.memory = validate.memory
[438]594
[572]595        if hasattr(validate, 'vmtype'):
596            machine.type = validate.vmtype
[440]597
[572]598        if hasattr(validate, 'disksize'):
599            disksize = validate.disksize
[177]600            disk = machine.disks[0]
601            if disk.size != disksize:
602                olddisk[disk.guest_device_name] = disksize
603                disk.size = disksize
[1013]604                session.save_or_update(disk)
[438]605
[446]606        update_acl = False
[572]607        if hasattr(validate, 'owner') and validate.owner != machine.owner:
608            machine.owner = validate.owner
[446]609            update_acl = True
[572]610        if hasattr(validate, 'name'):
[586]611            machine.name = validate.name
[1977]612            for n in machine.nics:
613                if n.hostname == oldname:
614                    n.hostname = validate.name
[609]615        if hasattr(validate, 'description'):
616            machine.description = validate.description
[572]617        if hasattr(validate, 'admin') and validate.admin != machine.administrator:
618            machine.administrator = validate.admin
[446]619            update_acl = True
[572]620        if hasattr(validate, 'contact'):
621            machine.contact = validate.contact
[438]622
[1013]623        session.save_or_update(machine)
[446]624        if update_acl:
625            cache_acls.refreshMachine(machine)
[1013]626        session.commit()
[161]627    except:
[1013]628        session.rollback()
[163]629        raise
[177]630    for diskname in olddisk:
[209]631        controls.resizeDisk(oldname, diskname, str(olddisk[diskname]))
[572]632    if hasattr(validate, 'name'):
633        controls.renameMachine(machine, oldname, validate.name)
[2483]634    return dict(machine=machine)
[438]635
[579]636def infoDict(username, state, machine):
[438]637    """Get the variables used by info.tmpl."""
[209]638    status = controls.statusInfo(machine)
[235]639    checkpoint.checkpoint('Getting status info')
[133]640    has_vnc = hasVnc(status)
641    if status is None:
642        main_status = dict(name=machine.name,
643                           memory=str(machine.memory))
[205]644        uptime = None
645        cputime = None
[133]646    else:
647        main_status = dict(status[1:])
[662]648        main_status['host'] = controls.listHost(machine)
[167]649        start_time = float(main_status.get('start_time', 0))
650        uptime = datetime.timedelta(seconds=int(time.time()-start_time))
651        cpu_time_float = float(main_status.get('cpu_time', 0))
652        cputime = datetime.timedelta(seconds=int(cpu_time_float))
[235]653    checkpoint.checkpoint('Status')
[133]654    display_fields = [('name', 'Name'),
[609]655                      ('description', 'Description'),
[133]656                      ('owner', 'Owner'),
[187]657                      ('administrator', 'Administrator'),
[133]658                      ('contact', 'Contact'),
[136]659                      ('type', 'Type'),
[133]660                      'NIC_INFO',
661                      ('uptime', 'uptime'),
662                      ('cputime', 'CPU usage'),
[662]663                      ('host', 'Hosted on'),
[133]664                      ('memory', 'RAM'),
665                      'DISK_INFO',
666                      ('state', 'state (xen format)'),
667                      ]
668    fields = []
669    machine_info = {}
[147]670    machine_info['name'] = machine.name
[609]671    machine_info['description'] = machine.description
[136]672    machine_info['type'] = machine.type.hvm and 'HVM' or 'ParaVM'
[133]673    machine_info['owner'] = machine.owner
[187]674    machine_info['administrator'] = machine.administrator
[133]675    machine_info['contact'] = machine.contact
676
677    nic_fields = getNicInfo(machine_info, machine)
678    nic_point = display_fields.index('NIC_INFO')
[438]679    display_fields = (display_fields[:nic_point] + nic_fields +
[205]680                      display_fields[nic_point+1:])
[133]681
682    disk_fields = getDiskInfo(machine_info, machine)
683    disk_point = display_fields.index('DISK_INFO')
[438]684    display_fields = (display_fields[:disk_point] + disk_fields +
[205]685                      display_fields[disk_point+1:])
[438]686
[211]687    main_status['memory'] += ' MiB'
[133]688    for field, disp in display_fields:
[167]689        if field in ('uptime', 'cputime') and locals()[field] is not None:
[133]690            fields.append((disp, locals()[field]))
[147]691        elif field in machine_info:
692            fields.append((disp, machine_info[field]))
[133]693        elif field in main_status:
694            fields.append((disp, main_status[field]))
695        else:
696            pass
697            #fields.append((disp, None))
[235]698
699    checkpoint.checkpoint('Got fields')
700
701
[572]702    max_mem = validation.maxMemory(machine.owner, state, machine, False)
[235]703    checkpoint.checkpoint('Got mem')
[566]704    max_disk = validation.maxDisk(machine.owner, machine)
[209]705    defaults = Defaults()
[609]706    for name in 'machine_id name description administrator owner memory contact'.split():
[2483]707        if getattr(machine, name):
708            setattr(defaults, name, getattr(machine, name))
[516]709    defaults.type = machine.type.type_id
[205]710    defaults.disk = "%0.2f" % (machine.disks[0].size/1024.)
[235]711    checkpoint.checkpoint('Got defaults')
[572]712    d = dict(user=username,
[133]713             on=status is not None,
714             machine=machine,
[205]715             defaults=defaults,
[133]716             has_vnc=has_vnc,
717             uptime=str(uptime),
718             ram=machine.memory,
[144]719             max_mem=max_mem,
720             max_disk=max_disk,
[133]721             fields = fields)
[205]722    return d
[113]723
[598]724def send_error_mail(subject, body):
725    import subprocess
[205]726
[863]727    to = config.web.errormail
[598]728    mail = """To: %s
[863]729From: root@%s
[598]730Subject: %s
731
732%s
[863]733""" % (to, config.web.hostname, subject, body)
[1718]734    p = subprocess.Popen(['/usr/sbin/sendmail', '-f', to, to],
735                         stdin=subprocess.PIPE)
[598]736    p.stdin.write(mail)
737    p.stdin.close()
738    p.wait()
739
[2485]740random.seed()
Note: See TracBrowser for help on using the repository browser.