zebra: Add nexthop hash entry list to zebra interface info

Add a nexthop hash entry list to the local zebra
interface info for each interface. This will allow
us to modify nexthops on link events.

Signed-off-by: Stephen Worley <sworley@cumulusnetworks.com>

2 files changed, 99 insertions, 0 deletions

diff --git a/zebra/interface.c b/zebra/interface.c
index ef03cf87f..6b77f3c87 100644
--- a/zebra/interface.c
+++ b/zebra/interface.c
@@ -53,6 +53,7 @@
 #include "zebra/zebra_errors.h"
 
 DEFINE_MTYPE_STATIC(ZEBRA, ZINFO, "Zebra Interface Information")
+DEFINE_MTYPE_STATIC(ZEBRA, NHE_CONNECTED, "Nexthops Connected")
 
 #define ZEBRA_PTM_SUPPORT
 
@@ -120,6 +121,10 @@ static int if_zebra_new_hook(struct interface *ifp)
 
 	zebra_if->multicast = IF_ZEBRA_MULTICAST_UNSPEC;
 	zebra_if->shutdown = IF_ZEBRA_SHUTDOWN_OFF;
+
+	zebra_if->nhe_connected = list_new();
+	zebra_if->nhe_connected->del = (void (*)(void *))nhe_connected_free;
+
 	zebra_ptm_if_init(zebra_if);
 
 	ifp->ptm_enable = zebra_ptm_get_enable_state();
@@ -196,6 +201,8 @@ static int if_zebra_delete_hook(struct interface *ifp)
 		list_delete(&rtadv->AdvDNSSLList);
 #endif /* HAVE_RTADV */
 
+		list_delete(&zebra_if->nhe_connected);
+
 		XFREE(MTYPE_TMP, zebra_if->desc);
 		THREAD_OFF(zebra_if->speed_update);
 
@@ -925,6 +932,51 @@ static void if_down_del_nbr_connected(struct interface *ifp)
 	}
 }
 
+/**
+ * nhe_connected_add() - Add the nexthop entry to the interfaces connected list
+ *
+ * @ifp:	Interface to add to
+ * @nhe:	Nexthop hash entry to add
+ *
+ * Return:	nhe_connected struct created and added
+ */
+struct nhe_connected *nhe_connected_add(struct interface *ifp,
+					struct nhg_hash_entry *nhe)
+{
+	struct nhe_connected *if_nhec = NULL;
+	struct zebra_if *zif = (struct zebra_if *)ifp->info;
+
+	if_nhec = nhe_connected_new();
+	if_nhec->ifp = ifp;
+
+	/* Attach the nhe */
+	if_nhec->nhe = nhe;
+
+	/* Add connected nexthop to the interface */
+	listnode_add(zif->nhe_connected, if_nhec);
+	return if_nhec;
+}
+
+/**
+ * nhe_connected() - Allocate nhe connected structure
+ *
+ * Return:	Allocated nhe_connected structure
+ */
+struct nhe_connected *nhe_connected_new(void)
+{
+	return XCALLOC(MTYPE_NHE_CONNECTED, sizeof(struct nhe_connected));
+}
+
+/**
+ * nhe_connected_free() - Free nhe_connected structure
+ *
+ * @nhe_connected:	nhe_connected structure to free
+ */
+void nhe_connected_free(struct nhe_connected *connected)
+{
+	XFREE(MTYPE_NHE_CONNECTED, connected);
+}
+
 /* Interface is up. */
 void if_up(struct interface *ifp)
 {
@@ -1132,6 +1184,19 @@ static void nbr_connected_dump_vty(struct vty *vty,
 	vty_out(vty, "\n");
 }
 
+/**
+ * nhe_connected_dump() - Dump nexthops connected to this interface to vty
+ *
+ * @vty:		Vty output
+ * @nhe_connected:	List of connected nexthop hash entries
+ */
+static void nhe_connected_dump_vty(struct vty *vty,
+				   struct nhe_connected *connected)
+{
+	/* Just outputing ID for now. */
+	vty_out(vty, "  (%u)", connected->nhe->id);
+}
+
 static const char *zebra_ziftype_2str(zebra_iftype_t zif_type)
 {
 	switch (zif_type) {
@@ -1279,6 +1344,7 @@ static void if_dump_vty(struct vty *vty, struct interface *ifp)
 {
 	struct connected *connected;
 	struct nbr_connected *nbr_connected;
+	struct nhe_connected *nhe_connected;
 	struct listnode *node;
 	struct route_node *rn;
 	struct zebra_if *zebra_if;
@@ -1364,6 +1430,14 @@ static void if_dump_vty(struct vty *vty, struct interface *ifp)
 			connected_dump_vty(vty, connected);
 	}
 
+	if (listhead(zebra_if->nhe_connected)) {
+		vty_out(vty, "  Nexthop IDs connected:");
+		for (ALL_LIST_ELEMENTS_RO(zebra_if->nhe_connected, node,
+					  nhe_connected))
+			nhe_connected_dump_vty(vty, nhe_connected);
+		vty_out(vty, "\n");
+	}
+
 	vty_out(vty, "  Interface Type %s\n",
 		zebra_ziftype_2str(zebra_if->zif_type));
 	if (IS_ZEBRA_IF_BRIDGE(ifp)) {
diff --git a/zebra/interface.h b/zebra/interface.h
index e134b9b42..270cc868c 100644
--- a/zebra/interface.h
+++ b/zebra/interface.h
@@ -27,6 +27,7 @@
 #include "hook.h"
 
 #include "zebra/zebra_l2.h"
+#include "zebra/zebra_nhg.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -263,6 +264,15 @@ typedef enum {
 
 struct irdp_interface;
 
+/* Nexthop hash entry connected structure */
+struct nhe_connected {
+	/* Attached interface */
+	struct interface *ifp;
+
+	/* Connected nexthop hash entry */
+	struct nhg_hash_entry *nhe;
+};
+
 /* `zebra' daemon local interface structure. */
 struct zebra_if {
 	/* Shutdown configuration. */
@@ -277,6 +287,15 @@ struct zebra_if {
 	/* Installed addresses chains tree. */
 	struct route_table *ipv4_subnets;
 
+	/* Nexthops pointed to it list */
+	/**
+	 * Any nexthop that we get should have an
+	 * interface. When an interface goes down,
+	 * we will use this list to update the nexthops
+	 * pointing to it with that info.
+	 */
+	struct list *nhe_connected;
+
 	/* Information about up/down changes */
 	unsigned int up_count;
 	char up_last[QUAGGA_TIMESTAMP_LEN];
@@ -424,6 +443,12 @@ extern void zebra_if_update_link(struct interface *ifp, ifindex_t link_ifindex,
 extern void zebra_if_update_all_links(void);
 extern void zebra_if_set_protodown(struct interface *ifp, bool down);
 
+/* Nexthop connected list functions */
+struct nhe_connected *nhe_connected_add(struct interface *ifp,
+					struct nhg_hash_entry *nhe);
+struct nhe_connected *nhe_connected_new(void);
+void nhe_connected_free(struct nhe_connected *connected);
+
 extern void vrf_add_update(struct vrf *vrfp);
 
 #ifdef HAVE_PROC_NET_DEV