diff --git a/monkey/infection_monkey/exploit/tools/http_tools.py b/monkey/infection_monkey/exploit/tools/http_tools.py index 43d62862f..cbe0f8f66 100644 --- a/monkey/infection_monkey/exploit/tools/http_tools.py +++ b/monkey/infection_monkey/exploit/tools/http_tools.py @@ -16,9 +16,11 @@ logger = logging.getLogger(__name__) class HTTPTools(object): @staticmethod - def try_create_locked_transfer(host, src_path, local_ip=None, local_port=None): + def try_create_locked_transfer( + host, src_path, agent_repository, local_ip=None, local_port=None + ): http_path, http_thread = HTTPTools.create_locked_transfer( - host, src_path, local_ip, local_port + host, src_path, agent_repository, local_ip, local_port ) if not http_path: raise Exception("Http transfer creation failed.") @@ -33,6 +35,7 @@ class HTTPTools(object): Create http server for file transfer with a lock :param host: Variable with target's information :param src_path: Monkey's path on current system + :param agent_repository: Repository to download Monkey agents :param local_ip: IP where to host server :param local_port: Port at which to host monkey's download :return: Server address in http://%s:%s/%s format and LockedHTTPServer handler