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

Agent: Add timeouts to utils/windows/users.py

This commit is contained in:
Shreya Malviya 2022-04-01 18:05:20 +05:30
parent 885a871be8
commit 88788d24d0
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
monkey/infection_monkey/utils/windows/users.py
View File

@ -1,6 +1,7 @@
import logging
import subprocess
from common.common_consts.timeouts import SHORT_REQUEST_TIMEOUT
from infection_monkey.utils.auto_new_user import AutoNewUser
from infection_monkey.utils.environment import is_windows_os
from infection_monkey.utils.new_user_error import NewUserError
@ -49,7 +50,9 @@ class AutoNewWindowsUser(AutoNewUser):
windows_cmds = get_windows_commands_to_add_user(self.username, self.password, True)
logger.debug("Trying to add {} with commands {}".format(self.username, str(windows_cmds)))
_ = subprocess.check_output(windows_cmds, stderr=subprocess.STDOUT)
_ = subprocess.check_output(
windows_cmds, stderr=subprocess.STDOUT, timeout=SHORT_REQUEST_TIMEOUT
)
def __enter__(self):
try:
@ -124,7 +127,9 @@ class AutoNewWindowsUser(AutoNewUser):
self.username, str(commands_to_deactivate_user)
)
)
_ = subprocess.check_output(commands_to_deactivate_user, stderr=subprocess.STDOUT)
_ = subprocess.check_output(
commands_to_deactivate_user, stderr=subprocess.STDOUT, timeout=SHORT_REQUEST_TIMEOUT
)
except Exception as err:
raise NewUserError("Can't deactivate user {}. Info: {}".format(self.username, err))
@ -136,6 +141,8 @@ class AutoNewWindowsUser(AutoNewUser):
self.username, str(commands_to_delete_user)
)
)
_ = subprocess.check_output(commands_to_delete_user, stderr=subprocess.STDOUT)
_ = subprocess.check_output(
commands_to_delete_user, stderr=subprocess.STDOUT, timeout=SHORT_REQUEST_TIMEOUT
)
except Exception as err:
raise NewUserError("Can't delete user {}. Info: {}".format(self.username, err))