Gamercat
/
geneva
mirror of https://github.com/Kkevsterrr/geneva
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Added server side flag

This commit is contained in:
Kkevsterrr 2020-01-17 11:03:14 -05:00
parent 59cfb74928
commit 6412aada52
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
engine.py
View File

@ -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"