1 files changed, 31 insertions, 35 deletions

diff --git a/zebra/redistribute.c b/zebra/redistribute.c
index c7ed209d3c..e1ec22463a 100644
--- a/zebra/redistribute.c
+++ b/zebra/redistribute.c
@@ -107,45 +107,41 @@ zebra_redistribute_default (struct zserv *client, vrf_id_t vrf_id)
 
 /* Redistribute routes. */
 static void
-zebra_redistribute (struct zserv *client, int type, u_short instance, vrf_id_t vrf_id)
+zebra_redistribute (struct zserv *client, int type, u_short instance, vrf_id_t vrf_id, int afi)
 {
   struct rib *newrib;
   struct route_table *table;
   struct route_node *rn;
-  int afi;
 
-  for (afi = AFI_IP; afi <= AFI_IP6; afi++)
-    {
-      table = zebra_vrf_table (afi, SAFI_UNICAST, vrf_id);
-      if (! table)
-	continue;
+  table = zebra_vrf_table (afi, SAFI_UNICAST, vrf_id);
+  if (! table)
+    return;
 
-      for (rn = route_top (table); rn; rn = route_next (rn))
-	RNODE_FOREACH_RIB (rn, newrib)
-	  {
-	    struct prefix *dst_p, *src_p;
-	    srcdest_rnode_prefixes(rn, &dst_p, &src_p);
-
-	    if (IS_ZEBRA_DEBUG_EVENT)
-	      zlog_debug("%s: checking: selected=%d, type=%d, distance=%d, "
-			 "zebra_check_addr=%d", __func__,
-			 CHECK_FLAG (newrib->flags, ZEBRA_FLAG_SELECTED),
-			 newrib->type, newrib->distance,
-			 zebra_check_addr (dst_p));
-
-	    if (! CHECK_FLAG (newrib->flags, ZEBRA_FLAG_SELECTED))
-	      continue;
-	    if ((type != ZEBRA_ROUTE_ALL &&
-		 (newrib->type != type || newrib->instance != instance)))
-	      continue;
-	    if (newrib->distance == DISTANCE_INFINITY)
-	      continue;
-	    if (! zebra_check_addr (dst_p))
-	      continue;
-
-	    zsend_redistribute_route (1, client, dst_p, src_p, newrib);
-	  }
-    }
+  for (rn = route_top (table); rn; rn = route_next (rn))
+    RNODE_FOREACH_RIB (rn, newrib)
+      {
+        struct prefix *dst_p, *src_p;
+        srcdest_rnode_prefixes(rn, &dst_p, &src_p);
+
+        if (IS_ZEBRA_DEBUG_EVENT)
+          zlog_debug("%s: checking: selected=%d, type=%d, distance=%d, "
+                 "zebra_check_addr=%d", __func__,
+                 CHECK_FLAG (newrib->flags, ZEBRA_FLAG_SELECTED),
+                 newrib->type, newrib->distance,
+                 zebra_check_addr (dst_p));
+
+        if (! CHECK_FLAG (newrib->flags, ZEBRA_FLAG_SELECTED))
+          continue;
+        if ((type != ZEBRA_ROUTE_ALL &&
+         (newrib->type != type || newrib->instance != instance)))
+          continue;
+        if (newrib->distance == DISTANCE_INFINITY)
+          continue;
+        if (! zebra_check_addr (dst_p))
+          continue;
+
+        zsend_redistribute_route (1, client, dst_p, src_p, newrib);
+      }
 }
 
 /* Either advertise a route for redistribution to registered clients or */
@@ -265,13 +261,13 @@ zebra_redistribute_add (int command, struct zserv *client, int length,
       if (! redist_check_instance (&client->mi_redist[afi][type], instance))
 	{
 	  redist_add_instance (&client->mi_redist[afi][type], instance);
-	  zebra_redistribute (client, type, instance, zvrf_id (zvrf));
+	  zebra_redistribute (client, type, instance, zvrf_id (zvrf), afi);
 	}
     } else {
 	if (! vrf_bitmap_check (client->redist[afi][type], zvrf_id (zvrf)))
 	  {
 	    vrf_bitmap_set (client->redist[afi][type], zvrf_id (zvrf));
-	    zebra_redistribute (client, type, 0, zvrf_id (zvrf));
+	    zebra_redistribute (client, type, 0, zvrf_id (zvrf), afi);
 	  }
     }
 }