Fix DUO116 warnings for:

- monkey/infection_monkey/dropper.py
- monkey/infection_monkey/system_info/windows_info_collector.py
- monkey/infection_monkey/utils/windows/users.py
- monkey/infection_monkey/windows_upgrader.py
This commit is contained in:
Shreya 2021-04-25 00:25:13 +05:30 committed by Mike Salvatore
parent 294e8fe56a
commit 410cbadbb3
4 changed files with 14 additions and 13 deletions

View File

@ -4,6 +4,7 @@ import filecmp
import logging import logging
import os import os
import pprint import pprint
import shlex
import shutil import shutil
import subprocess import subprocess
import sys import sys
@ -164,9 +165,10 @@ class MonkeyDrops(object):
"monkey_commandline": inner_monkey_cmdline, "monkey_commandline": inner_monkey_cmdline,
} }
monkey_cmdline_split = shlex.split(monkey_cmdline)
monkey_process = subprocess.Popen( monkey_process = subprocess.Popen(
monkey_cmdline, monkey_cmdline_split,
shell=True,
stdin=subprocess.PIPE, stdin=subprocess.PIPE,
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, stderr=subprocess.PIPE,

View File

@ -1,4 +1,5 @@
import logging import logging
import shlex
import subprocess import subprocess
import sys import sys
@ -51,10 +52,10 @@ class WindowsInfoCollector(InfoCollector):
def get_installed_packages(self): def get_installed_packages(self):
LOG.info("Getting installed packages") LOG.info("Getting installed packages")
packages = subprocess.check_output("dism /online /get-packages", shell=True) packages = subprocess.check_output(shlex.split("dism /online /get-packages"))
self.info["installed_packages"] = packages.decode("utf-8", errors="ignore") self.info["installed_packages"] = packages.decode("utf-8", errors="ignore")
features = subprocess.check_output("dism /online /get-features", shell=True) features = subprocess.check_output(shlex.split("dism /online /get-features"))
self.info["installed_features"] = features.decode("utf-8", errors="ignore") self.info["installed_features"] = features.decode("utf-8", errors="ignore")
LOG.debug("Got installed packages") LOG.debug("Got installed packages")

View File

@ -39,7 +39,7 @@ class AutoNewWindowsUser(AutoNewUser):
windows_cmds = get_windows_commands_to_add_user(self.username, self.password, True) 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))) logger.debug("Trying to add {} with commands {}".format(self.username, str(windows_cmds)))
_ = subprocess.check_output(windows_cmds, stderr=subprocess.STDOUT, shell=True) _ = subprocess.check_output(windows_cmds, stderr=subprocess.STDOUT)
def __enter__(self): def __enter__(self):
# Importing these only on windows, as they won't exist on linux. # Importing these only on windows, as they won't exist on linux.
@ -127,9 +127,7 @@ class AutoNewWindowsUser(AutoNewUser):
self.username, str(commands_to_deactivate_user) self.username, str(commands_to_deactivate_user)
) )
) )
_ = subprocess.check_output( _ = subprocess.check_output(commands_to_deactivate_user, stderr=subprocess.STDOUT)
commands_to_deactivate_user, stderr=subprocess.STDOUT, shell=True
)
except Exception as err: except Exception as err:
raise NewUserError("Can't deactivate user {}. Info: {}".format(self.username, err)) raise NewUserError("Can't deactivate user {}. Info: {}".format(self.username, err))
@ -141,8 +139,6 @@ class AutoNewWindowsUser(AutoNewUser):
self.username, str(commands_to_delete_user) self.username, str(commands_to_delete_user)
) )
) )
_ = subprocess.check_output( _ = subprocess.check_output(commands_to_delete_user, stderr=subprocess.STDOUT)
commands_to_delete_user, stderr=subprocess.STDOUT, shell=True
)
except Exception as err: except Exception as err:
raise NewUserError("Can't delete user {}. Info: {}".format(self.username, err)) raise NewUserError("Can't delete user {}. Info: {}".format(self.username, err))

View File

@ -1,4 +1,5 @@
import logging import logging
import shlex
import shutil import shutil
import subprocess import subprocess
import sys import sys
@ -50,9 +51,10 @@ class WindowsUpgrader(object):
+ monkey_options + monkey_options
) )
monkey_cmdline_split = shlex.split(monkey_cmdline)
monkey_process = subprocess.Popen( monkey_process = subprocess.Popen(
monkey_cmdline, monkey_cmdline_split,
shell=True,
stdin=None, stdin=None,
stdout=None, stdout=None,
stderr=None, stderr=None,