From e842c45e42f76c8753100edfcadfe62d83ee582d Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Fri, 2 Sep 2022 16:19:21 +0200 Subject: [PATCH] Agent: Add list of servers to dropper --- monkey/infection_monkey/dropper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/dropper.py b/monkey/infection_monkey/dropper.py index 6f92feaea..f443f08ad 100644 --- a/monkey/infection_monkey/dropper.py +++ b/monkey/infection_monkey/dropper.py @@ -44,7 +44,7 @@ class MonkeyDrops(object): def __init__(self, args): arg_parser = argparse.ArgumentParser() arg_parser.add_argument("-p", "--parent") - arg_parser.add_argument("-s", "--server") + arg_parser.add_argument("-s", "--servers", type=lambda arg: arg.strip().split(",")) arg_parser.add_argument("-d", "--depth", type=positive_int, default=0) arg_parser.add_argument("-l", "--location") arg_parser.add_argument("-vp", "--vulnerable-port") @@ -131,7 +131,7 @@ class MonkeyDrops(object): monkey_options = build_monkey_commandline_explicitly( parent=self.opts.parent, - server=self.opts.server, + servers=self.opts.servers, depth=self.opts.depth, location=None, )