p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Add a query timeout to pymssql.connect()

This commit is contained in:
Mike Salvatore 2022-03-17 14:26:07 -04:00
parent 33f2bac275
commit 0ffe023a9f
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/infection_monkey/exploit/mssqlexec.py
View File

@ -23,6 +23,7 @@ class MSSQLExploiter(HostExploiter):
_EXPLOITED_SERVICE = "MSSQL" _EXPLOITED_SERVICE = "MSSQL"
_TARGET_OS_TYPE = ["windows"] _TARGET_OS_TYPE = ["windows"]
LOGIN_TIMEOUT = 15 LOGIN_TIMEOUT = 15
QUERY_TIMEOUT = LOGIN_TIMEOUT
# Time in seconds to wait between MSSQL queries. # Time in seconds to wait between MSSQL queries.
QUERY_BUFFER = 0.5 QUERY_BUFFER = 0.5
SQL_DEFAULT_TCP_PORT = "1433" SQL_DEFAULT_TCP_PORT = "1433"
@ -213,7 +214,12 @@ class MSSQLExploiter(HostExploiter):
# Core steps # Core steps
# Trying to connect # Trying to connect
conn = pymssql.connect( conn = pymssql.connect(
host, user, password, port=port, login_timeout=self.LOGIN_TIMEOUT host,
user,
password,
port=port,
login_timeout=self.LOGIN_TIMEOUT,
timeout=self.QUERY_TIMEOUT,
) )
logger.info( logger.info(
f"Successfully connected to host: {host} using user: {user} and password" f"Successfully connected to host: {host} using user: {user} and password"