Changeset 1411 for trunk/packages/invirt-remote-server/files

Timestamp:

Oct 28, 2008, 11:17:24 PM (16 years ago)

Author:

broder

Message:

Don't proxy requests for the VNC cert if the requested hostname isn't
an Invirt host

File:

• trunk/packages/invirt-remote-server/files/usr/sbin/invirt-remote-vnccert (modified) (1 diff)

trunk/packages/invirt-remote-server/files/usr/sbin/invirt-remote-vnccert

@@ -1 @@
-#!/bin/bash
-# Get the VNC cert for the host passed as the first argument
+#!/usr/bin/python
 
-klist -s || kinit -k
+"""
+Retrieves the VNC certificate from an Invirt host
+"""
 
-exec remctl "$1" remote web vnccert
+from invirt.config import structs as config
+from subprocess import Popen, call, PIPE
+import sys
+
+
+def main(argv):
+    if len(argv) < 2:
+        print >> sys.stderr, "usage: invirt-remote-vnccert <host>"
+        return 2
+    
+    host = argv[1]
+    
+    if host not in list(i.hostname for i in config.hosts):
+        print >> sys.stderr, "Invalid hostname specified"
+        return 1
+    
+    return call(['remctl', host, 'remote', 'web', 'vnccert'])
+
+if __name__ == '__main__':
+    sys.exit(main(sys.argv))