1 files changed, 87 insertions, 0 deletions

diff --git a/zebra/zebra_ns.c b/zebra/zebra_ns.c
index c48a6f7bd8..1715881f7e 100644
--- a/zebra/zebra_ns.c
+++ b/zebra/zebra_ns.c
@@ -34,8 +34,25 @@
 
 DEFINE_MTYPE(ZEBRA, ZEBRA_NS, "Zebra Name Space")
 
+static inline int
+zebra_ns_table_entry_compare(const struct zebra_ns_table *e1,
+			     const struct zebra_ns_table *e2);
+
+RB_GENERATE(zebra_ns_table_head, zebra_ns_table, zebra_ns_table_entry,
+	    zebra_ns_table_entry_compare);
+
 static struct zebra_ns *dzns;
 
+static inline int
+zebra_ns_table_entry_compare(const struct zebra_ns_table *e1,
+			     const struct zebra_ns_table *e2)
+{
+	if (e1->tableid == e2->tableid)
+		return (e1->afi - e2->afi);
+
+	return e1->tableid - e2->tableid;
+}
+
 struct zebra_ns *zebra_ns_lookup(ns_id_t ns_id)
 {
 	return dzns;
@@ -57,10 +74,79 @@ int zebra_ns_enable(ns_id_t ns_id, void **info)
 	return 0;
 }
 
+struct route_table *zebra_ns_find_table(struct zebra_ns *zns,
+					uint32_t tableid, afi_t afi)
+{
+	struct zebra_ns_table finder;
+	struct zebra_ns_table *znst;
+
+	memset(&finder, 0, sizeof(finder));
+	finder.afi = afi;
+	finder.tableid = tableid;
+	znst = RB_FIND(zebra_ns_table_head, &zns->ns_tables, &finder);
+
+	if (znst)
+		return znst->table;
+	else
+		return NULL;
+}
+
+struct route_table *zebra_ns_get_table(struct zebra_ns *zns,
+				       struct zebra_vrf *zvrf, uint32_t tableid,
+				       afi_t afi)
+{
+	struct zebra_ns_table finder;
+	struct zebra_ns_table *znst;
+	rib_table_info_t *info;
+
+	memset(&finder, 0, sizeof(finder));
+	finder.afi = afi;
+	finder.tableid = tableid;
+	znst = RB_FIND(zebra_ns_table_head, &zns->ns_tables, &finder);
+
+	if (znst)
+		return znst->table;
+
+	znst = XCALLOC(MTYPE_ZEBRA_NS, sizeof(*znst));
+	znst->tableid = tableid;
+	znst->afi = afi;
+	znst->table =
+		(afi == AFI_IP6) ? srcdest_table_init() : route_table_init();
+
+	info = XCALLOC(MTYPE_RIB_TABLE_INFO, sizeof(*info));
+	info->zvrf = zvrf;
+	info->afi = afi;
+	info->safi = SAFI_UNICAST;
+	znst->table->info = info;
+	znst->table->cleanup = zebra_rtable_node_cleanup;
+
+	RB_INSERT(zebra_ns_table_head, &zns->ns_tables, znst);
+	return znst->table;
+}
+
+static void zebra_ns_free_table(struct zebra_ns_table *znst)
+{
+	void *table_info;
+
+	rib_close_table(znst->table);
+
+	table_info = znst->table->info;
+	route_table_finish(znst->table);
+	XFREE(MTYPE_RIB_TABLE_INFO, table_info);
+	XFREE(MTYPE_ZEBRA_NS, znst);
+}
+
 int zebra_ns_disable(ns_id_t ns_id, void **info)
 {
+	struct zebra_ns_table *znst;
 	struct zebra_ns *zns = (struct zebra_ns *)(*info);
 
+	while (!RB_EMPTY(zebra_ns_table_head, &zns->ns_tables)) {
+		znst = RB_ROOT(zebra_ns_table_head, &zns->ns_tables);
+
+		RB_REMOVE(zebra_ns_table_head, &zns->ns_tables, znst);
+		zebra_ns_free_table(znst);
+	}
 	route_table_finish(zns->if_table);
 	zebra_vxlan_ns_disable(zns);
 #if defined(HAVE_RTADV)
@@ -72,6 +158,7 @@ int zebra_ns_disable(ns_id_t ns_id, void **info)
 	return 0;
 }
 
+
 int zebra_ns_init(void)
 {
 	dzns = XCALLOC(MTYPE_ZEBRA_NS, sizeof(struct zebra_ns));