forked from p15670423/monkey
Agent: Add timeouts in shell startup modification PBA's
This commit is contained in:
parent
581ece577d
commit
4ad07ae3ff
|
@ -1,5 +1,6 @@
|
||||||
import subprocess
|
import subprocess
|
||||||
|
|
||||||
|
from common.common_consts.timeouts import MEDIUM_REQUEST_TIMEOUT
|
||||||
from infection_monkey.utils.environment import is_windows_os
|
from infection_monkey.utils.environment import is_windows_os
|
||||||
|
|
||||||
|
|
||||||
|
@ -12,7 +13,9 @@ def get_linux_commands_to_modify_shell_startup_files():
|
||||||
# get list of usernames
|
# get list of usernames
|
||||||
USERS = (
|
USERS = (
|
||||||
subprocess.check_output( # noqa: DUO116
|
subprocess.check_output( # noqa: DUO116
|
||||||
"cut -d: -f1,3 /etc/passwd | egrep ':[0-9]{4}$' | cut -d: -f1", shell=True
|
"cut -d: -f1,3 /etc/passwd | egrep ':[0-9]{4}$' | cut -d: -f1",
|
||||||
|
shell=True,
|
||||||
|
timeout=MEDIUM_REQUEST_TIMEOUT,
|
||||||
)
|
)
|
||||||
.decode()
|
.decode()
|
||||||
.split("\n")[:-1]
|
.split("\n")[:-1]
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
import subprocess
|
import subprocess
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
from common.common_consts.timeouts import MEDIUM_REQUEST_TIMEOUT
|
||||||
from infection_monkey.utils.environment import is_windows_os
|
from infection_monkey.utils.environment import is_windows_os
|
||||||
|
|
||||||
MODIFY_POWERSHELL_STARTUP_SCRIPT = Path(__file__).parent / "modify_powershell_startup_file.ps1"
|
MODIFY_POWERSHELL_STARTUP_SCRIPT = Path(__file__).parent / "modify_powershell_startup_file.ps1"
|
||||||
|
@ -16,7 +17,9 @@ def get_windows_commands_to_modify_shell_startup_files():
|
||||||
|
|
||||||
# get list of usernames
|
# get list of usernames
|
||||||
USERS = (
|
USERS = (
|
||||||
subprocess.check_output("dir C:\\Users /b", shell=True) # noqa: DUO116
|
subprocess.check_output( # noqa: DUO116
|
||||||
|
"dir C:\\Users /b", shell=True, timeout=MEDIUM_REQUEST_TIMEOUT
|
||||||
|
)
|
||||||
.decode()
|
.decode()
|
||||||
.split("\r\n")[:-1]
|
.split("\r\n")[:-1]
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue