powerstager.py 文件源码

def __init__(self, lhost, lport):
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.lhost = lhost
        self.lport = lport
        self.framework = Framework(self.send_command, self.check_command, self.get_response)
        self.command_definition = {}
        self.credentials = []
        self.fast_load = False
        self.prompt = ""
        self.os_target = None

        # Define command behavior
        self.command_definition["Local-Invoke"]                 = lambda x: self.psh_Local_Invoke(x[len("Local-Invoke"):].strip())
        self.command_definition["Local-Import-Module"]          = lambda x: self.psh_Local_Invoke(x[len("Local-Import-Module"):].strip(), True)
        self.command_definition["Local-Set-Width"]              = lambda x: self.psh_Local_Set_Width(x[len("Local-Set-Width"):].strip())
        self.command_definition["Local-Upload"]                 = lambda x: self.psh_Local_Upload(x[len("Local-Upload"):].strip())
        self.command_definition["Local-Download"]               = lambda x: self.psh_Local_Download(x[len("Local-Download"):].strip())
        self.command_definition["Local-Download-Commands"]      = lambda x: self.psh_Local_Download_Commands()      
        self.command_definition["Local-Enumerate-System"]       = lambda x: self.psh_Local_Enumerate_System()
        self.command_definition["Local-Check-Status"]           = lambda x: self.psh_Local_Check_Status()
        self.command_definition["Local-Spawn-Meterpreter"]      = lambda x: self.psh_Local_Spawn_Shell(conf_name.METERPRETER)
        self.command_definition["Local-Spawn-Reverse-Shell"]    = lambda x: self.psh_Local_Spawn_Shell(conf_name.REVERSE_SHELL)
        self.command_definition["Local-Credential-Create"]      = lambda x: self.psh_Local_Credential(True)
        self.command_definition["Local-Credential-List"]        = lambda x: self.psh_Local_Credential()
        self.command_definition["clear"]                        = lambda x: self.psh_Local_Clear()

        # Define autocomplete
        readline.parse_and_bind("tab: complete")
        readline.set_completer(self.cmd_complete)
        readline.set_completion_display_matches_hook(self.cmd_match_display_hook)
        readline.set_completer_delims("")