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)