From 6412aada527e5cf17291fc8b653529576483da05 Mon Sep 17 00:00:00 2001 From: Kkevsterrr Date: Fri, 17 Jan 2020 11:03:14 -0500 Subject: [PATCH] Added server side flag --- engine.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/engine.py b/engine.py index f76cdc3..5271aaa 100644 --- a/engine.py +++ b/engine.py @@ -30,7 +30,7 @@ BASEPATH = os.path.dirname(os.path.abspath(__file__)) class Engine(): - def __init__(self, server_port, string_strategy, environment_id=None, output_directory="trials", log_level="info"): + def __init__(self, server_port, string_strategy, server_side=False, environment_id=None, output_directory="trials", log_level="info"): self.server_port = server_port self.seen_packets = [] # Set up the directory and ID for logging @@ -47,6 +47,7 @@ class Engine(): environment_id, log_level=log_level) self.output_directory = output_directory + self.server_side = server_side # Used for conditional context manager usage self.strategy = actions.utils.parse(string_strategy, self.logger) @@ -124,6 +125,8 @@ class Engine(): self.logger.debug("Configuring iptables rules") port1, port2 = "dport", "sport" + if self.server_side: + port1, port2 = "sport", "dport" out_chain = "OUTPUT" in_chain = "INPUT"