hackedteam/test-av

import subprocess
import sys
import os

class Command:
    def __init__(self, path, host=None, user=None, passwd=None):
        if not host and not user and not passwd:
            self.path = path
        else:
            self.path = path
            self.host = host
            self.user = user
            self.passwd = passwd
    
    def startup(self, vmx):
        sys.stdout.write("\r\nStartup %s!\r\n" % vmx)
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "start", vmx])
    
    def shutdown(self, vmx):
        sys.stdout.write("\r\nShutdown %s!\r\n" % vmx)
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "stop", vmx])

    def reboot(self, vmx):
        sys.stdout.write("\r\nRebooting %s!\r\n" % vmx)
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "reset", vmx, "soft"])

    def suspend(self, vmx):
        sys.stdout.write("\r\nSuspending %s!\r\n" % vmx)
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "suspend", vmx, "soft"])
        
    def executeCmd(self, vmx, cmd, script=None):
        sys.stdout.write("Executing %s %s.\n" % (cmd, script))
        if script is not None:
            subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "-gu", "avtest", "-gp", "avtest",
                        "runProgramInGuest", vmx, cmd, script])            
        else:
            subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "-gu", "avtest", "-gp", "avtest",
                        "runProgramInGuest", vmx, cmd])
    
    
    def update(self, vmx):
        cscriptPath="c:/windows/system32/cscript.exe"
        scriptPath="z:/WUA_SearchDownloadInstall.vbs"    
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "-gu", "avtest", "-gp", "avtest",
                        "runProgramInGuest", vmx, "c:/windows/system32/cscript.exe","c:/Users/avtest/WUA_SearchDownloadInstall.vbs"])
                        
    def refreshSnapshot(self, vmx, snapshot):
        sys.stdout.write("Deleting current snapshot.\n")
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "deleteSnapshot", vmx, snapshot])
        sys.stdout.write("Creating new current snapshot.\n")
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "snapshot", vmx, snapshot])
                        
    def revertSnapshot(self, vmx, snapshot):
        sys.stdout.write("Reverting to current snapshot.\n")
        subprocess.call([self.path,
                        "-h", self.host,
                        "-u", self.user, "-p", self.passwd,
                        "revertToSnapshot", vmx, snapshot])