def send_shell_commands_for_console(client):
module_prompt_default_template = "\001\033[4m\002SHELL\001\033[0m\002 (\001\033[91m\002{hostname}\001\033[0m\002) > "
while True:
cmd = None
try:
address = client.address[0]
cmd = input(module_prompt_default_template.format(hostname=address))
if not cmd:
continue
elif cmd.lower() == "clear":
clear_history()
data_to_stdout("[i] history cleared\n")
elif cmd.lower() in ("x", "q", "exit", "quit", "bye"):
break
client.conn.send(str.encode(cmd + '\n'))
resp = poll_cmd_execute(client)
data_to_stdout(resp)
except Exception as ex:
logger.error(str(ex))
data_to_stdout("Connection Lost\n")
break
return True
def send_shell_commands(client):
auto_completion(AUTOCOMPLETE_TYPE.OS, OS.LINUX)
while True:
cmd = None
try:
address = client.address[0]
cmd = input("{0}>: ".format(address))
if not cmd:
continue
elif cmd.lower() == "clear":
clear_history()
data_to_stdout("[i] history cleared\n")
save_history(AUTOCOMPLETE_TYPE.POCSUITE)
elif cmd.lower() in ("x", "q", "exit", "quit", "bye"):
break
client.conn.send(str.encode(cmd + '\n'))
resp = poll_cmd_execute(client)
data_to_stdout(resp)
except Exception as ex:
logger.error(str(ex))
data_to_stdout("Connection Lost\n")
break
def handle_listener_connection():
_ = ["list", "select", "exit", "quit", "clear"]
auto_completion(AUTOCOMPLETE_TYPE.POCSUITE, commands=_)
while True:
cmd = None
cmd = input('shell>: ').strip()
if not cmd:
continue
elif cmd.lower() in ("?", "help"):
print_cmd_help()
elif cmd.lower() == "clear":
clear_history()
data_to_stdout("[i] history cleared\n")
save_history(AUTOCOMPLETE_TYPE.POCSUITE)
elif cmd.lower() in ("x", "q", "exit", "quit"):
raise PocsuiteShellQuitException
elif cmd == "list":
list_clients()
elif "select" in cmd:
client = get_client(cmd)
if client is not None:
send_shell_commands(client)
else:
save_history(AUTOCOMPLETE_TYPE.POCSUITE)
load_history(AUTOCOMPLETE_TYPE.POCSUITE)
data_to_stdout("Command Not Found... type ? for help.")
