def shutdown_network(*, teams: Optional[List[int]], **_kwargs):
if teams is None:
logger.error('Specify all required parameters: teams')
exit(1)
remove_drop_rules()
all_rules = OPEN_NETWORK_RULES + INIT_RULES + get_team2vuln_rules(teams)
remove_rules(all_rules)
def close_network(*_args, **_kwargs):
remove_rules(OPEN_NETWORK_RULES)
def remove_drop_rules(*_args, **_kwargs):
remove_rules(DROP_RULES)
remove_rules(ALLOW_SSH_RULES)
def deisolate_team(args):
helpers.remove_rules(get_isolation_rules(args.team))
def unban_team(args):
helpers.remove_rules(get_ban_rules(args.team))
def shutdown_network(args):
helpers.parse_arguments_teams(args)
helpers.remove_rules(INIT_RULES)
isolation_rules = sum((get_isolation_rules(team) for team in args.teams),
[])
helpers.remove_rules(isolation_rules)
ban_rules = sum((get_ban_rules(team) for team in args.teams), [])
helpers.remove_rules(ban_rules)
helpers.set_chain_policy('INPUT', 'ACCEPT')
helpers.set_chain_policy('FORWARD', 'DROP')
helpers.remove_rules(ALLOW_SSH_RULES)
helpers.remove_rules(CLOSED_NETWORK_FORWARDING)
helpers.remove_rules(OPEN_NETWORK_FORWARDING)
for chain in CUSTOM_CHAINS:
helpers.remove_chain(chain)
for s in SETS:
helpers.remove_set(s)
def close_network(_args):
helpers.remove_rules(OPEN_NETWORK_FORWARDING)
helpers.add_rules(CLOSED_NETWORK_FORWARDING)
def shutdown_network(**_kwargs):
remove_drop_rules()
all_rules = INIT_RULES
remove_rules(all_rules)
