#!/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/>. # - Rename SSTP ppp-settings node to ppp-options to make use of a common # Jinja Template to render Accel-PPP services from vyos.configtree import ConfigTree from sys import argv from sys import exit if len(argv) < 2: print("Must specify file name!") exit(1) file_name = argv[1] with open(file_name, 'r') as f: config_file = f.read() config = ConfigTree(config_file) base_path = ['vpn', 'sstp'] if not config.exists(base_path): # Nothing to do exit(0) else: if config.exists(base_path + ['ppp-settings']): config.rename(base_path + ['ppp-settings'], 'ppp-options') config_ns = base_path + ['network-settings', 'name-server'] if config.exists(config_ns): config.copy(config_ns, base_path + ['name-server']) config.delete(config_ns) config_mtu = base_path + ['network-settings', 'mtu'] if config.exists(config_mtu): config.copy(config_mtu, base_path + ['mtu']) config.delete(config_mtu) config_gw = base_path + ['network-settings', 'client-ip-settings', 'gateway-address'] if config.exists(config_gw): config.copy(config_gw, base_path + ['gateway-address']) config.delete(config_gw) config_client_ip = base_path + ['network-settings', 'client-ip-settings'] if config.exists(config_client_ip): config.copy(config_client_ip, base_path + ['client-ip-pool']) config.delete(config_client_ip) config_client_ipv6 = base_path + ['network-settings', 'client-ipv6-pool'] if config.exists(config_client_ipv6): config.copy(config_client_ipv6, base_path + ['client-ipv6-pool']) config.delete(config_client_ipv6) # all nodes now have been migrated out of network-settings - delete node config_nw_settings = base_path + ['network-settings'] if config.exists(config_nw_settings): config.delete(config_nw_settings) 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)) exit(1)