p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Extract method _get_ssh_port

This commit is contained in:
Kekoa Kaaikala 2022-10-05 14:50:00 +00:00 committed by Ilija Lazoroski
parent 0a1901b9a1
commit 431d6ae775
1 changed files with 19 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
monkey/infection_monkey/exploit/sshexec.py
View File

@ -177,8 +177,15 @@ class SSHExploiter(HostExploiter):
raise FailedExploitationError raise FailedExploitationError
def _exploit_host(self) -> ExploiterResultData: def _exploit_host(self) -> ExploiterResultData:
port = self._get_ssh_port()
is_open, _ = check_tcp_port(self.host.ip_addr, port)
if not is_open:
self.exploit_result.error_message = f"SSH port is closed on {self.host}, skipping"
logger.info(self.exploit_result.error_message)
return self.exploit_result
try: try:
ssh = self._exploit() ssh = self._exploit(port)
except FailedExploitationError as err: except FailedExploitationError as err:
self.exploit_result.error_message = str(err) self.exploit_result.error_message = str(err)
logger.error(str(err)) logger.error(str(err))
@ -203,26 +210,7 @@ class SSHExploiter(HostExploiter):
) )
return self.exploit_result return self.exploit_result
def _exploit(self) -> paramiko.SSHClient: def _exploit(self, port) -> paramiko.SSHClient:
port = SSH_PORT
# if ssh banner found on different port, use that port.
for servkey, servdata in list(self.host.services.items()):
if servdata.get("name") == "ssh" and servkey.startswith("tcp-"):
port = int(servkey.replace("tcp-", ""))
is_open, _ = check_tcp_port(self.host.ip_addr, port)
if not is_open:
self.exploit_result.error_message = f"SSH port is closed on {self.host}, skipping"
self._publish_exploitation_event(
target=self.host.ip_addr,
exploitation_success=False,
error_message=self.exploit_result.error_message,
tags=(SSH_EXPLOITER_TAG,),
)
logger.info(self.exploit_result.error_message)
raise FailedExploitationError(self.exploit_result.error_message)
try: try:
ssh = self.exploit_with_ssh_keys(port) ssh = self.exploit_with_ssh_keys(port)
except FailedExploitationError: except FailedExploitationError:
@ -294,6 +282,16 @@ class SSHExploiter(HostExploiter):
except Exception as exc: except Exception as exc:
raise FailedExploitationError(f"Error running monkey on victim {self.host}: ({exc})") raise FailedExploitationError(f"Error running monkey on victim {self.host}: ({exc})")
def _get_ssh_port(self) -> int:
port = SSH_PORT
# if ssh banner found on different port, use that port.
for servkey, servdata in list(self.host.services.items()):
if servdata.get("name") == "ssh" and servkey.startswith("tcp-"):
port = int(servkey.replace("tcp-", ""))
return port
def _get_victim_os(self, ssh: paramiko.SSHClient) -> bool: def _get_victim_os(self, ssh: paramiko.SSHClient) -> bool:
try: try:
_, stdout, _ = ssh.exec_command("uname -o", timeout=SSH_EXEC_TIMEOUT) _, stdout, _ = ssh.exec_command("uname -o", timeout=SSH_EXEC_TIMEOUT)