From 190e2681d441e26208156691d573b4db86e55ccd Mon Sep 17 00:00:00 2001
From: sever-sever <v.gletenko@vyos.io>
Date: Tue, 9 Mar 2021 17:41:36 +0000
Subject: syslog: T3396: Fix remote IPv6 host

---
 data/templates/syslog/rsyslog.conf.tmpl | 58 ++++++++++++++++-----------------
 python/vyos/template.py                 |  7 ++++
 2 files changed, 36 insertions(+), 29 deletions(-)

diff --git a/data/templates/syslog/rsyslog.conf.tmpl b/data/templates/syslog/rsyslog.conf.tmpl
index 10fbb9d3c..e25ef48d4 100644
--- a/data/templates/syslog/rsyslog.conf.tmpl
+++ b/data/templates/syslog/rsyslog.conf.tmpl
@@ -2,47 +2,47 @@
 ## file based logging
 {% if files['global']['marker'] %}
 $ModLoad immark
-{% if files['global']['marker-interval'] %}
+{%   if files['global']['marker-interval'] %}
 $MarkMessagePeriod  {{files['global']['marker-interval']}}
-{% endif %}
+{%   endif %}
 {% endif %}
 {% if files['global']['preserver_fqdn'] %}
 $PreserveFQDN on
 {% endif %}
-{% for file in files %}
-$outchannel {{file}},{{files[file]['log-file']}},{{files[file]['max-size']}},{{files[file]['action-on-max-size']}}
-{{files[file]['selectors']}} :omfile:${{file}}
+{% for file, file_options in files.items() %}
+$outchannel {{ file }},{{ file_options['log-file'] }},{{ file_options['max-size'] }},{{ file_options['action-on-max-size'] }}
+{{ file_options['selectors'] }} :omfile:${{ file }}
 {% endfor %}
-{% if console %}
+{% if console is defined and console is not none %}
 ## console logging
-{% for con in console %}
-{{console[con]['selectors']}} /dev/console
-{% endfor %}
+{%   for con, con_options in console.items() %}
+{{ con_options['selectors'] }} /dev/console
+{%   endfor %}
 {% endif %}
-{% if hosts %}
+{% if hosts is defined and hosts is not none %}
 ## remote logging
-{% for host in hosts %}
-{%   if hosts[host]['proto'] == 'tcp' %}
-{%     if hosts[host]['port'] %}
-{%       if hosts[host]['oct_count'] %}
-{{hosts[host]['selectors']}} @@(o){{host}}:{{hosts[host]['port']}};RSYSLOG_SyslogProtocol23Format
+{%   for host, host_options in hosts.items() %}
+{%     if host_options.proto == 'tcp' %}
+{%       if host_options.port is defined %}
+{%         if host_options.oct_count is defined %}
+{{ host_options.selectors }} @@(o){{ host }}:{{ host_options.port }};RSYSLOG_SyslogProtocol23Format
+{%         else %}
+{{ host_options.selectors }} @@{{ host }}:{{ host_options.port }}
+{%         endif %}
 {%       else %}
-{{hosts[host]['selectors']}} @@{{host}}:{{hosts[host]['port']}}
+{{ host_options.selectors }} @@{{ host }}
 {%       endif %}
 {%     else %}
-{{hosts[host]['selectors']}} @@{{host}}
-{%     endif %}
-{%   else %}
-{%     if hosts[host]['port'] %}
-{{hosts[host]['selectors']}} @{{host}}:{{hosts[host]['port']}}
-{%     else %}
-{{hosts[host]['selectors']}} @{{host}}
+{%       if host_options['port'] %}
+{{ host_options.selectors }} @{{ host | bracketize_ipv6 }}:{{ host_options.port }}
+{%       else %}
+{{ host_options.selectors }} @{{ host | bracketize_ipv6 }}
+{%       endif %}
 {%     endif %}
-{%   endif %}
-{% endfor %}
+{%   endfor %}
 {% endif %}
-{% if user %}
-{% for u in user %}
-{{user[u]['selectors']}} :omusrmsg:{{u}}
-{% endfor %}
+{% if user is defined and user is not none %}
+{%   for username, user_options in user.items() %}
+{{ user_options.selectors }} :omusrmsg:{{ username }}
+{%   endfor %}
 {% endif %}
diff --git a/python/vyos/template.py b/python/vyos/template.py
index 22883b103..85e4d12b3 100644
--- a/python/vyos/template.py
+++ b/python/vyos/template.py
@@ -131,6 +131,13 @@ def address_from_cidr(prefix):
     from ipaddress import ip_network
     return str(ip_network(prefix).network_address)
 
+@register_filter('bracketize_ipv6')
+def bracketize_ipv6(address):
+    """ Place a passed IPv6 address into [] brackets, do nothing for IPv4 """
+    if is_ipv6(address):
+        return f'[{address}]'
+    return address
+
 @register_filter('netmask_from_cidr')
 def netmask_from_cidr(prefix):
     """ Take CIDR prefix and convert the prefix length to a "subnet mask".
-- 
cgit v1.2.3