Changeset 24581 in vbox for trunk/src/VBox/Frontends

Timestamp:

Nov 11, 2009 2:35:18 PM (15 years ago)

Author:

vboxsync

Message:

Python shell: teleportation support

File:

• trunk/src/VBox/Frontends/VBoxShell/vboxshell.py (modified) (6 diffs)

trunk/src/VBox/Frontends/VBoxShell/vboxshell.py

@@ -219 +219 @@
 
 
+def reportError(ctx,session,rc):
+    if not ctx['remote']:
+            print session.QueryErrorObject(rc)
+
+
 def createVm(ctx,name,kind,base):
     mgr = ctx['mgr']
@@ -274 +279 @@
         session.close()
     else:
-        # Not yet implemented error string query API for remote API
-        if not ctx['remote']:
-            print session.QueryErrorObject(rc)
+       reportError(ctx,session,rc)
 
 def getMachines(ctx, invalidate = False):
@@ -353 +356 @@
     data = display.takeScreenShotSlow(w,h)
     size = (w,h)
-    mode = "RGBA"    
+    mode = "RGBA"
     im = Image.frombuffer(mode, size, data, "raw", mode, 0, 1)
     im.save(f, "PNG")
 
+
+def teleport(ctx,session,console,args):
+    if args[0].find(":") == -1:
+        print "Use host:port format for teleport target"
+        return
+    (host,port) = args[0].split(":")
+    if len(args) > 1:
+        passwd = args[1]
+    else:
+        passwd = ""
+
+    port = int(port)
+    print "Teleporting to %s:%d..." %(host,port)
+    progress = console.teleport(host, port, passwd)
+    progressBar(ctx, progress, 100)
+    completed = progress.completed
+    rc = int(progress.resultCode)
+    if rc == 0:
+        print "Success!"
+    else:
+        reportError(ctx,session,rc)
 
 def cmdExistingVm(ctx,mach,cmd,args):
@@ -387 +411 @@
          'monitorGuest':    lambda: monitorGuest(ctx, mach, console, args),
          'save':            lambda: progressBar(ctx,console.saveState()),
-         'screenshot':      lambda: takeScreenshot(ctx,console,args)
+         'screenshot':      lambda: takeScreenshot(ctx,console,args),
+         'teleport':        lambda: teleport(ctx,session,console,args)
          }
     try:
@@ -650 +675 @@
     cmdExistingVm(ctx, mach, 'screenshot', args[2:])
     return 0
+
+def teleportCmd(ctx, args):
+    if (len(args) < 3):
+        print "usage: teleport name host:port <password>"
+        return 0
+    mach = argsToMach(ctx,args)
+    if mach == None:
+        return 0
+    cmdExistingVm(ctx, mach, 'teleport', args[2:])
+    return 0
+
+def makeportalCmd(ctx, args):
+    if (len(args) < 3):
+        print "usage: makeportal name port <password>"
+        return 0
+    mach = argsToMach(ctx,args)
+    if mach == None:
+        return 0
+    port = int(args[2])
+    if (len(args) > 3):
+        passwd = args[3]
+    else:
+        passwd = ""
+    if not mach.teleporterEnabled or mach.teleporterPort != port:
+        session = ctx['global'].openMachineSession(mach.id)
+        mach1 = session.machine
+        mach1.teleporterEnabled = True
+        mach1.teleporterPort = port
+        mach1.saveSettings()
+        session.close()
+    startVm(ctx, mach, "gui")
+    return 0
+
+def closeportalCmd(ctx, args):
+    if (len(args) < 2):
+        print "usage: closeportal name"
+        return 0
+    mach = argsToMach(ctx,args)
+    if mach == None:
+        return 0
+    if mach.teleporterEnabled:
+        session = ctx['global'].openMachineSession(mach.id)
+        mach1 = session.machine
+        mach1.teleporterEnabled = False
+        mach1.saveSettings()
+        session.close()
+    return 0
+
 
 def setvarCmd(ctx, args):
@@ -1007 +1080 @@
             'shell':['Execute external shell command: shell "ls /etc/rc*"', shellCmd, 0],
             'exportVm':['Export VM in OVF format: export Win /tmp/win.ovf', exportVMCmd, 0],
-            'screenshot':['Take VM screenshot to a file: screenshot Win /tmp/win.png 1024 768', screenshotCmd, 0]
+            'screenshot':['Take VM screenshot to a file: screenshot Win /tmp/win.png 1024 768', screenshotCmd, 0],
+            'teleport':['Teleport VM to another box (see makeportal): teleport Win anotherhost:8000 <passwd>', teleportCmd, 0],
+            'makeportal':['Make portal for teleportation of VM from another box (see teleport): makeportal Win 8000 <passwd>', makeportalCmd, 0],
+            'closeportal':['Close portal for teleportation of VM from another box (see teleport): closeportal Win', closeportalCmd, 0]
             }