Drop privileges in VNC proxy if requested

[invirt/packages/invirt-web.git] / code / main.py

diff --git a/code/main.py b/code/main.py
index a8926f2..dec53d6 100755 (executable)
--- a/code/main.py
+++ b/code/main.py
@@ -372,6 +372,31 @@ console will suffer artifacts.
             return d
 
         @cherrypy.expose
+        @cherrypy.tools.mako(filename="/novnc.mako")
+        def novnc(self, machine_id):
+            """NoVNC page functionality.
+
+            You need to set config.vnc.novnc_port and run the NoVNC proxy on
+            the web server.
+            """
+            machine = validation.Validate(cherrypy.request.login,
+                                          cherrypy.request.state,
+                                          machine_id=machine_id).machine
+            token = controls.vnctoken(machine)
+            host = controls.listHost(machine)
+
+            status = controls.statusInfo(machine)
+            has_vnc = hasVnc(status)
+
+            d = dict(on=status,
+                     has_vnc=has_vnc,
+                     machine=machine,
+                     hostname=cherrypy.request.local.name,
+                     vnc_host=host,
+                     authtoken=token)
+            return d
+
+        @cherrypy.expose
         @cherrypy.tools.mako(filename="/command.mako")
         @cherrypy.tools.require_POST()
         def command(self, command_name, machine_id, **kwargs):