1 files changed, 152 insertions, 0 deletions
diff --git a/staticd/static_vrf.c b/staticd/static_vrf.c
new file mode 100644
index 0000000..a67dce2
--- /dev/null
+++ b/staticd/static_vrf.c
@@ -0,0 +1,152 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * STATICd - vrf code
+ * Copyright (C) 2018 Cumulus Networks, Inc.
+ *               Donald Sharp
+ */
+#include <zebra.h>
+
+#include "vrf.h"
+#include "nexthop.h"
+#include "table.h"
+#include "srcdest_table.h"
+#include "northbound_cli.h"
+
+#include "static_vrf.h"
+#include "static_routes.h"
+#include "static_zebra.h"
+
+DEFINE_MTYPE_STATIC(STATIC, STATIC_RTABLE_INFO, "Static Route Table Info");
+
+static struct static_vrf *static_vrf_alloc(void)
+{
+	struct route_table *table;
+	struct static_vrf *svrf;
+	struct stable_info *info;
+	safi_t safi;
+	afi_t afi;
+
+	svrf = XCALLOC(MTYPE_STATIC_RTABLE_INFO, sizeof(struct static_vrf));
+
+	for (afi = AFI_IP; afi <= AFI_IP6; afi++) {
+		for (safi = SAFI_UNICAST; safi <= SAFI_MULTICAST; safi++) {
+			if (afi == AFI_IP6)
+				table = srcdest_table_init();
+			else
+				table = route_table_init();
+
+			info = XCALLOC(MTYPE_STATIC_RTABLE_INFO,
+				       sizeof(struct stable_info));
+			info->svrf = svrf;
+			info->afi = afi;
+			info->safi = safi;
+			route_table_set_info(table, info);
+
+			table->cleanup = zebra_stable_node_cleanup;
+			svrf->stable[afi][safi] = table;
+		}
+	}
+	return svrf;
+}
+
+static int static_vrf_new(struct vrf *vrf)
+{
+	struct static_vrf *svrf;
+
+	svrf = static_vrf_alloc();
+	vrf->info = svrf;
+	svrf->vrf = vrf;
+
+	return 0;
+}
+
+static int static_vrf_enable(struct vrf *vrf)
+{
+	static_zebra_vrf_register(vrf);
+
+	static_fixup_vrf_ids(vrf->info);
+
+	return 0;
+}
+
+static int static_vrf_disable(struct vrf *vrf)
+{
+	static_zebra_vrf_unregister(vrf);
+	return 0;
+}
+
+static int static_vrf_delete(struct vrf *vrf)
+{
+	struct route_table *table;
+	struct static_vrf *svrf;
+	safi_t safi;
+	afi_t afi;
+	void *info;
+
+	svrf = vrf->info;
+	for (afi = AFI_IP; afi <= AFI_IP6; afi++) {
+		for (safi = SAFI_UNICAST; safi <= SAFI_MULTICAST; safi++) {
+			table = svrf->stable[afi][safi];
+			info = route_table_get_info(table);
+			route_table_finish(table);
+			XFREE(MTYPE_STATIC_RTABLE_INFO, info);
+			svrf->stable[afi][safi] = NULL;
+		}
+	}
+	XFREE(MTYPE_STATIC_RTABLE_INFO, svrf);
+	return 0;
+}
+
+/* Lookup the static routing table in a VRF. */
+struct route_table *static_vrf_static_table(afi_t afi, safi_t safi,
+					    struct static_vrf *svrf)
+{
+	if (!svrf)
+		return NULL;
+
+	if (afi >= AFI_MAX || safi >= SAFI_MAX)
+		return NULL;
+
+	return svrf->stable[afi][safi];
+}
+
+struct static_vrf *static_vrf_lookup_by_name(const char *name)
+{
+	struct vrf *vrf;
+
+	if (!name)
+		name = VRF_DEFAULT_NAME;
+
+	vrf = vrf_lookup_by_name(name);
+	if (vrf)
+		return ((struct static_vrf *)vrf->info);
+
+	return NULL;
+}
+
+static int static_vrf_config_write(struct vty *vty)
+{
+	struct lyd_node *dnode;
+	int written = 0;
+
+	dnode = yang_dnode_get(running_config->dnode, "/frr-routing:routing");
+	if (dnode) {
+		nb_cli_show_dnode_cmds(vty, dnode, false);
+		written = 1;
+	}
+
+	return written;
+}
+
+void static_vrf_init(void)
+{
+	vrf_init(static_vrf_new, static_vrf_enable, static_vrf_disable,
+		 static_vrf_delete);
+
+	vrf_cmd_init(static_vrf_config_write);
+}
+
+void static_vrf_terminate(void)
+{
+	vrf_terminate();
+}