diff options
| author | sarthurdev <965089+sarthurdev@users.noreply.github.com> | 2023-09-24 14:38:12 +0200 |
|---|---|---|
| committer | sarthurdev <965089+sarthurdev@users.noreply.github.com> | 2023-09-24 16:44:32 +0200 |
| commit | 81dee963a9ca3224ddbd54767a36efae5851a001 (patch) | |
| tree | dd959ad5a29ff75a94e7b1e8d1738ad4a66847d1 /python/vyos/firewall.py | |
| parent | 734392fdff1276110d81c2315e6a4a29a1316f7d (diff) | |
| download | vyos-1x-81dee963a9ca3224ddbd54767a36efae5851a001.tar.gz vyos-1x-81dee963a9ca3224ddbd54767a36efae5851a001.zip | |
firewall: T5614: Add support for matching on conntrack helper
Diffstat (limited to 'python/vyos/firewall.py')
| -rw-r--r-- | python/vyos/firewall.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/python/vyos/firewall.py b/python/vyos/firewall.py index 3ca7a25b9..7e43b815a 100644 --- a/python/vyos/firewall.py +++ b/python/vyos/firewall.py @@ -102,6 +102,20 @@ def parse_rule(rule_conf, hook, fw_name, rule_id, ip_name): if states: output.append(f'ct state {{{states}}}') + if 'conntrack_helper' in rule_conf: + helper_map = {'h323': ['RAS', 'Q.931'], 'nfs': ['rpc'], 'sqlnet': ['tns']} + helper_out = [] + + for helper in rule_conf['conntrack_helper']: + if helper in helper_map: + helper_out.extend(helper_map[helper]) + else: + helper_out.append(helper) + + if helper_out: + helper_str = ','.join(f'"{s}"' for s in helper_out) + output.append(f'ct helper {{{helper_str}}}') + if 'connection_status' in rule_conf and rule_conf['connection_status']: status = rule_conf['connection_status'] if status['nat'] == 'destination': |