From e32ce63b52eff0255d602b475d90cd97d3b282c1 Mon Sep 17 00:00:00 2001
From: Christian Poessinger <christian@poessinger.com>
Date: Fri, 6 Jan 2023 08:14:44 +0100
Subject: static: T4883: fix KeyError: 'table'

Commit dafb0da2 ("static: T4883: add a description field for routing tables")
added an iproute2 description table but lacked checking if the key exists.
This has been fixed and also converted to Jinja2 to keep the "common" style
inside the routing protocols. It might feel overengineered indeed.
---
 src/conf_mode/protocols_static.py | 17 +++++++----------
 src/conf_mode/vrf.py              |  6 ++----
 2 files changed, 9 insertions(+), 14 deletions(-)

(limited to 'src/conf_mode')

diff --git a/src/conf_mode/protocols_static.py b/src/conf_mode/protocols_static.py
index cbbc476a7..3e5ebb805 100755
--- a/src/conf_mode/protocols_static.py
+++ b/src/conf_mode/protocols_static.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 #
-# Copyright (C) 2021 VyOS maintainers and contributors
+# Copyright (C) 2021-2023 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
@@ -25,12 +25,15 @@ from vyos.configdict import get_dhcp_interfaces
 from vyos.configdict import get_pppoe_interfaces
 from vyos.configverify import verify_common_route_maps
 from vyos.configverify import verify_vrf
+from vyos.template import render
 from vyos.template import render_to_string
 from vyos import ConfigError
 from vyos import frr
 from vyos import airbag
 airbag.enable()
 
+config_file = '/etc/iproute2/rt_tables.d/vyos-static.conf'
+
 def get_config(config=None):
     if config:
         conf = config
@@ -94,19 +97,13 @@ def verify(static):
 def generate(static):
     if not static:
         return None
+
+    # Put routing table names in /etc/iproute2/rt_tables
+    render(config_file, 'iproute2/static.conf.j2', static)
     static['new_frr_config'] = render_to_string('frr/staticd.frr.j2', static)
     return None
 
 def apply(static):
-    ## Put routing table names in /etc/iproute2/rt_tables
-    with open("/etc/iproute2/rt_tables.d/vyos.conf", 'w') as f:
-        print("# Generated by VyOS (protocols_static.py), do not edit by hand", file=f)
-        for t in static['table']:
-            if 'description' in static['table'][t]:
-                print(f"{t}\t{static['table'][t]['description']}", file=f)
-
-    ## Inject routes into FRR
-
     static_daemon = 'staticd'
     zebra_daemon = 'zebra'
 
diff --git a/src/conf_mode/vrf.py b/src/conf_mode/vrf.py
index 1b4156895..c17cca3bd 100755
--- a/src/conf_mode/vrf.py
+++ b/src/conf_mode/vrf.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 #
-# Copyright (C) 2020-2022 VyOS maintainers and contributors
+# Copyright (C) 2020-2023 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
@@ -140,11 +140,9 @@ def verify(vrf):
 
 
 def generate(vrf):
-    render(config_file, 'vrf/vrf.conf.j2', vrf)
+    render(config_file, 'iproute2/vrf.conf.j2', vrf)
     # Render nftables zones config
-
     render(nft_vrf_config, 'firewall/nftables-vrf-zones.j2', vrf)
-
     return None
 
 
-- 
cgit v1.2.3