Merge branch 't1831-rtradv' of github.com:c-po/vyos-1x into current

* 't1831-rtradv' of github.com:c-po/vyos-1x:
  router-advert: T1831: add config migration script
  router-advert: T1831: new implementation using XML and Python

2 files changed, 318 insertions, 0 deletions

diff --git a/src/conf_mode/service-router-advert.py b/src/conf_mode/service-router-advert.py
new file mode 100755
index 000000000..87a4e9d92
--- /dev/null
+++ b/src/conf_mode/service-router-advert.py
@@ -0,0 +1,207 @@
+#!/usr/bin/env python3
+#
+# Copyright (C) 2018-2019 VyOS maintainers and contributors
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 or later as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import os
+import sys
+import jinja2
+
+from stat import S_IRUSR, S_IWUSR, S_IRGRP
+from vyos.config import Config
+from vyos import ConfigError
+
+config_file = r'/etc/radvd.conf'
+
+config_tmpl = """
+### Autogenerated by service-ipv6-ra.py ###
+
+{% for i in interfaces -%}
+interface {{ i.name }} {
+    IgnoreIfMissing on;
+    AdvDefaultPreference {{ i.default_preference }};
+    AdvManagedFlag {{ i.managed_flag }};
+    MaxRtrAdvInterval {{ i.interval_max }};
+{% if i.interval_min %}
+    MinRtrAdvInterval {{ i.interval_min }};
+{% endif %}
+    AdvReachableTime {{ i.reachable_time }};
+    AdvIntervalOpt {{ i.send_advert }};
+    AdvSendAdvert {{ i.send_advert }};
+{% if i.default_lifetime %}
+    AdvDefaultLifetime {{ i.default_lifetime }};
+{% endif %}
+    AdvLinkMTU {{ i.link_mtu }};
+    AdvOtherConfigFlag {{ i.other_config_flag }};
+    AdvRetransTimer {{ i.retrans_timer }};
+    AdvCurHopLimit {{ i.hop_limit }};
+{% for p in i.prefixes %}
+    prefix {{ p.prefix }} {
+        AdvAutonomous {{ p.autonomous_flag }};
+        AdvValidLifetime {{ p.valid_lifetime }};
+        AdvOnLink {{ p.on_link }};
+        AdvPreferredLifetime {{ p.preferred_lifetime }};
+    };
+{% endfor %}
+{% if i.name_server %}
+    RDNSS {{ i.name_server | join(" ") }} {
+    };
+{% endif %}
+};
+{% endfor -%}
+"""
+
+default_config_data = {
+    'interfaces': []
+}
+
+def get_config():
+    rtradv = default_config_data
+    conf = Config()
+    base_level = ['service', 'router-advert']
+
+    if not conf.exists(base_level):
+        return rtradv
+
+    for interface in conf.list_nodes(base_level + ['interface']):
+        intf = {
+            'name': interface,
+            'hop_limit' : '64',
+            'default_lifetime': '',
+            'default_preference': 'medium',
+            'dnssl': [],
+            'link_mtu': '0',
+            'managed_flag': 'off',
+            'interval_max': '600',
+            'interval_min': '',
+            'name_server': [],
+            'other_config_flag': 'off',
+            'prefixes' : [],
+            'reachable_time': '0',
+            'retrans_timer': '0',
+            'send_advert': 'on'
+        }
+
+        # set config level first to reduce boilerplate code
+        conf.set_level(base_level + ['interface', interface])
+
+        if conf.exists(['hop-limit']):
+            intf['hop_limit'] = conf.return_value(['hop-limit'])
+
+        if conf.exists(['default-lifetim']):
+            intf['default_lifetime'] = conf.return_value(['default-lifetim'])
+
+        if conf.exists(['default-preference']):
+            intf['default_preference'] = conf.return_value(['default-preference'])
+
+        if conf.exists(['dnssl']):
+            intf['dnssl'] = conf.return_values(['dnssl'])
+
+        if conf.exists(['link-mtu']):
+            intf['link_mtu'] = conf.return_value(['link-mtu'])
+
+        if conf.exists(['managed-flag']):
+            intf['managed_flag'] = 'on'
+
+        if conf.exists(['interval', 'max']):
+            intf['interval_max'] = conf.return_value(['interval', 'max'])
+
+        if conf.exists(['interval', 'min']):
+            intf['interval_min'] = conf.return_value(['interval', 'min'])
+
+        if conf.exists(['name-server']):
+            intf['name_server'] = conf.return_values(['name-server'])
+
+        if conf.exists(['other-config-flag']):
+            intf['other_config_flag'] = 'on'
+
+        if conf.exists(['reachable-time']):
+            intf['reachable_time'] = conf.return_value(['reachable-time'])
+
+        if conf.exists(['retrans-timer']):
+            intf['retrans_timer'] = conf.return_value(['retrans-timer'])
+
+        if conf.exists(['no-send-advert']):
+            intf['send_advert'] = 'off'
+
+        for prefix in conf.list_nodes(['prefix']):
+            tmp = {
+                'prefix' : prefix,
+                'autonomous_flag' : 'on',
+                'on_link' : 'on',
+                'preferred_lifetime': '14400',
+                'valid_lifetime' : '2592000'
+
+            }
+
+            # set config level first to reduce boilerplate code
+            conf.set_level(base_level + ['interface', interface, 'prefix', prefix])
+
+            if conf.exists(['no-autonomous-flag']):
+                tmp['autonomous_flag'] = 'off'
+
+            if conf.exists(['no-on-link-flag']):
+                tmp['on_link'] = 'off'
+
+            if conf.exists(['preferred-lifetime']):
+                tmp['preferred_lifetime'] = conf.return_value(['preferred-lifetime'])
+
+            if conf.exists(['valid-lifetime']):
+                tmp['valid_lifetime'] = conf.return_value(['valid-lifetime'])
+
+            intf['prefixes'].append(tmp)
+
+        rtradv['interfaces'].append(intf)
+
+    return rtradv
+
+def verify(rtradv):
+    return None
+
+def generate(rtradv):
+    if not rtradv['interfaces']:
+        return None
+
+    tmpl = jinja2.Template(config_tmpl, trim_blocks=True)
+    config_text = tmpl.render(rtradv)
+    with open(config_file, 'w') as f:
+        f.write(config_text)
+
+    # adjust file permissions of new configuration file
+    if os.path.exists(config_file):
+        os.chmod(config_file, S_IRUSR | S_IWUSR | S_IRGRP)
+
+    return None
+
+def apply(rtradv):
+    if not rtradv['interfaces']:
+        # bail out early - looks like removal from running config
+        os.system('sudo systemctl stop radvd.service')
+        if os.path.exists(config_file):
+            os.unlink(config_file)
+
+        return None
+
+    os.system('sudo systemctl restart radvd.service')
+    return None
+
+if __name__ == '__main__':
+    try:
+        c = get_config()
+        verify(c)
+        generate(c)
+        apply(c)
+    except ConfigError as e:
+        print(e)
+        sys.exit(1)
diff --git a/src/migration-scripts/interfaces/5-to-6 b/src/migration-scripts/interfaces/5-to-6
new file mode 100755
index 000000000..9dbfd30e1
--- /dev/null
+++ b/src/migration-scripts/interfaces/5-to-6
@@ -0,0 +1,111 @@
+#!/usr/bin/env python3
+#
+# Copyright (C) 2020 VyOS maintainers and contributors
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 or later as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+# Migrate IPv6 router advertisments from a nested interface configuration to
+# a denested "service router-advert"
+
+import sys
+from vyos.configtree import ConfigTree
+
+def copy_rtradv(c, old_base, interface):
+    base = ['service', 'router-advert', 'interface']
+
+    if c.exists(old_base):
+        if not c.exists(base):
+            c.set(base)
+            c.set_tag(base)
+
+        # take the old node as a whole and copy it to new new path,
+        # additional migrations will be done afterwards
+        new_base = base + [interface]
+        c.copy(old_base, new_base)
+        c.delete(old_base)
+
+        # cur-hop-limit has been renamed to hop-limit
+        if c.exists(new_base + ['cur-hop-limit']):
+            c.rename(new_base + ['cur-hop-limit'], 'hop-limit')
+
+        bool_cleanup = ['managed-flag', 'other-config-flag']
+        for bool in bool_cleanup:
+            if c.exists(new_base + [bool]):
+                tmp = c.return_value(new_base + [bool])
+                c.delete(new_base + [bool])
+                if tmp == 'true':
+                    c.set(new_base + [bool])
+
+        # max/min interval moved to subnode
+        intervals = ['max-interval', 'min-interval']
+        for interval in intervals:
+            if c.exists(new_base + [interval]):
+                tmp = c.return_value(new_base + [interval])
+                c.delete(new_base + [interval])
+                min_max = interval.split('-')[0]
+                c.set(new_base + ['interval', min_max], value=tmp)
+
+        # cleanup boolean nodes in individual prefix
+        prefix_base = new_base + ['prefix']
+        if c.exists(prefix_base):
+            for prefix in config.list_nodes(prefix_base):
+                bool_cleanup = ['autonomous-flag', 'on-link-flag']
+                for bool in bool_cleanup:
+                    if c.exists(prefix_base + [prefix, bool]):
+                        tmp = c.return_value(prefix_base + [prefix, bool])
+                        c.delete(prefix_base + [prefix, bool])
+                        if tmp == 'true':
+                            c.set(prefix_base + [prefix, bool])
+
+        # router advertisement can be individually disabled per interface
+        # the node has been renamed from send-advert {true | false} to no-send-advert
+        if c.exists(new_base + ['send-advert']):
+            tmp = c.return_value(new_base + ['send-advert'])
+            c.delete(new_base + ['send-advert'])
+            if tmp == 'false':
+                c.set(new_base + ['no-send-advert'])
+
+if __name__ == '__main__':
+    if (len(sys.argv) < 1):
+        print("Must specify file name!")
+        exit(1)
+
+    file_name = sys.argv[1]
+    with open(file_name, 'r') as f:
+        config_file = f.read()
+
+    config = ConfigTree(config_file)
+
+    # list all individual interface types like dummy, ethernet and so on
+    for if_type in config.list_nodes(['interfaces']):
+        base_if_type = ['interfaces', if_type]
+
+        # for every individual interface we need to check if there is an
+        # ipv6 ra configured ... and also for every VIF (VLAN) interface
+        for intf in config.list_nodes(base_if_type):
+            old_base = base_if_type + [intf, 'ipv6', 'router-advert']
+            copy_rtradv(config, old_base, intf)
+
+            vif_base = base_if_type + [intf, 'vif']
+            if config.exists(vif_base):
+                for vif in config.list_nodes(vif_base):
+                    old_base = vif_base + [vif, 'ipv6', 'router-advert']
+                    vlan_name = f'{intf}.{vif}'
+                    copy_rtradv(config, old_base, vlan_name)
+
+    try:
+        with open(file_name, 'w') as f:
+            f.write(config.to_string())
+    except OSError as e:
+        print("Failed to save the modified config: {}".format(e))
+        sys.exit(1)