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

Agent: Rename unused '_' parameter to architecture in get_agent_binary

This commit is contained in:
Mike Salvatore 2022-03-15 13:13:07 -04:00
parent 048817d60a
commit 7facf302a4
2 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/common/utils/exceptions.py
View File

@ -18,10 +18,6 @@ class IncorrectCredentialsError(Exception):
""" Raise to indicate that authentication failed """ """ Raise to indicate that authentication failed """
class InvalidAWSKeys(Exception):
""" Raise to indicate that AWS API keys are invalid"""
class NoInternetError(Exception): class NoInternetError(Exception):
""" Raise to indicate problems caused when no internet connection is present""" """ Raise to indicate problems caused when no internet connection is present"""

2
monkey/infection_monkey/exploit/caching_agent_repository.py
View File

@ -22,7 +22,7 @@ class CachingAgentRepository(IAgentRepository):
self._proxies = proxies self._proxies = proxies
self._lock = threading.Lock() self._lock = threading.Lock()
def get_agent_binary(self, os: str, _: str = None) -> io.BytesIO: def get_agent_binary(self, os: str, architecture: str = None) -> io.BytesIO:
# If multiple calls to get_agent_binary() are made simultaneously before the result of # If multiple calls to get_agent_binary() are made simultaneously before the result of
# _download_binary_from_island() is cached, then multiple requests will be sent to the # _download_binary_from_island() is cached, then multiple requests will be sent to the
# island. Add a mutex in front of the call to _download_agent_binary_from_island() so # island. Add a mutex in front of the call to _download_agent_binary_from_island() so