| Dir : /opt/imunify360/venv/lib/python3.11/site-packages/im360/internals/core/rules/ |
| Server: Linux cloud.virginhosting.lk 4.18.0-477.27.2.lve.el8.x86_64 #1 SMP Wed Oct 11 12:32:56 UTC 2023 x86_64 IP: 128.140.68.198 |
| Dir : //opt/imunify360/venv/lib/python3.11/site-packages/im360/internals/core/rules/port.py |
"""Put redirect_port_rules here to avoid circular import."""
from typing import Callable, Dict, Iterator, Set
from im360.internals.core.firewall import FirewallRules
from .types_ import FirewallRule
def redirect_port_rules(
ipset_name: str,
dest_ports: Set[int],
redirect_map: Dict[int, int],
table: str,
redirector: Callable,
priority: int = FirewallRules.DEFAULT_PRIORITY,
) -> Iterator[FirewallRule]:
"""Generate common port redirection rules for captcha/splashscreen."""
for dest_port, target in redirect_map.items():
if dest_port not in dest_ports:
continue
yield FirewallRule(
rule=redirector(ipset_name, dest_port, target),
table=table,
priority=priority,
)