class ConsoleUI(VmxUI):
"""
"""
vmrun = None
def __init__(self, prompt, intro, config='.config', debug=False):
"""
Arguments:
- `prompt`:
- `intro`:
"""
VmxUI.__init__(self, config)
self.prompt = self.make_prompt(prompt)
self.intro = intro
self.doc_header = "...oooOOO iConsole Command OOOooo..." \
"\n (for help, type: help <command>)"
self.undoc_header = ""
self.misc_header = ""
self.ruler = " "
self.debug = debug
# FIXME awkward assert...
def assert_vmrun(self):
"""Assert vmrun is properly initialized.
Arguments:
- `self`:
"""
if self.vmrun is None:
print "type 'use' first..."
return 0
else:
return 1
def do_use(self, args):
"""use specified vmx file, and initialize vmrun.
Usage:
use vmx-name
"""
VmxUI.do_use(self, args)
if ( self.vmx is not None and
self.vmx_admin is not None and
self.vmx_pass is not None ):
self.vmrun = Vmrun( self.vmx, self.vmx_admin, self.vmx_pass, debug=self.debug )
def do_vmstart(self, args):
"""
Start vm
Usage:
vmstart
"""
return self.assert_vmrun() and "".join( self.vmrun.start() )
do_start = do_vmstart
def do_vmsuspend(self, args):
"""
Suspend vm
Usage:
vmsuspend
"""
return self.assert_vmrun() and "".join( self.vmrun.suspend( "hard" ) )
do_suspend = do_vmsuspend
def do_vmstop(self, args):
"""
Suspend vm
Usage:
vmstop
"""
return self.assert_vmrun() and "".join( self.vmrun.stop() )
do_stop = do_vmstop
def do_vmcopy(self, args):
"""
Copy file to vm
Usage:
vmcopy from to
"""
argv = self.checkargs(args, 2)
if argv == None: return
# TODO change path
return self.assert_vmrun() and \
"".join( self.vmrun.copyFileFromHostToGuest( argv[0], "\"%s\\%s\"" % (self.cwd_guest, argv[1]) ) )
do_cp = do_vmcopy
def do_vmget(self, args):
"""
Get file from vm
Usage:
vmget file_of_vm as_file_host
"""
argv = self.checkargs(args, 2)
if argv == None: return
return self.assert_vmrun() and \
"".join( self.vmrun.copyFileFromGuestToHost( "\"%s\\%s\"" % (self.cwd_guest, argv[0]),
"%s%s%s" % (self.cwd_host, os.sep, argv[1]) ) )
do_get = do_vmget
def do_vmsnap(self, args):
"""Snapshot related commands
Usage:
[vm]snap list/create/delete/revert snap-name
"""
def vmsnap(argv):
"""
TODO - fix issue of quoted arguments
Arguments:
- `*argv`:
"""
if ( argv[0] == "list" or argv[0] == "l" ):
print "".join( self.vmrun.listSnapshots() )
elif ( argv[0] == "create" or
argv[0] == "c" ) and argv[1] != "":
print "".join( self.vmrun.snapshot( argv[1] ) )
elif ( argv[0] == "delete" or
argv[0] == "d" or
argv[0] == "del" ) and argv[1] != "":
print "".join( self.vmrun.deleteSnapshot( argv[1] ) )
elif ( argv[0] == "revert" or
argv[0] == "r" ) and argv[1] != "":
print "".join( self.vmrun.revertToSnapshot( argv[1] ) )
argv = self.checkargs(args)
if argv == []: print "type 'help [vm]snap' ..."; return
return self.assert_vmrun() and vmsnap(argv)
do_snap = do_vmsnap
def do_vmexec(self, args):
"""Execute program in the guest
Arguments:
- `self`:
- `args`:
"""
def vmexec(argv):
"""
Arguments:
- `argv`:
"""
try:
pname = argv.pop(0)
param = " ".join(argv) # FIXME
# TODO n/a/i
print "".join( self.vmrun.runProgramInGuest( "\"%s\"" % self.cfg.get( self.section, pname ), "n", "\"%s\"" % param) )
except Exception, e:
print "[-] error : %s" % str(e)
argv = self.checkargs(args)
if argv == []: print "type 'help [vm]exec' ..."; return
return self.assert_vmrun() and vmexec(argv)
def suspend(self):
vm = Vmrun(self.vmx)
vm.suspend()
puts(colored.green("Suspended", vm))
def suspend(self):
vm = Vmrun(self.vmx)
vm.suspend()
puts(colored.green("Suspended", vm))
