pimd: Fix the test to see if a upstream is (S,G,rpt)

Signed-off-by: Donald Sharp <sharpd@cumulusnetworks.com>

diff --git a/pimd/pim_upstream.c b/pimd/pim_upstream.c
index fe6e81ef63f68f2fff357d20a39d26ba8d4b1bb5..8552966782450e62f29fd8043fef803e9ecce0a0 100644 (file)
--- a/pimd/pim_upstream.c
+++ b/pimd/pim_upstream.c
@@ -991,8 +991,15 @@ pim_upstream_switch_to_spt_desired (struct prefix_sg *sg)
 int
 pim_upstream_is_sg_rpt (struct pim_upstream *up)
 {
-  if (up->sptbit == PIM_UPSTREAM_SPTBIT_TRUE)
-    return 1;
+  struct listnode *chnode;
+  struct pim_ifchannel *ch;
+
+  for (ALL_LIST_ELEMENTS_RO(pim_ifchannel_list, chnode, ch))
+    {
+      if ((ch->upstream == up) &&
+         (PIM_IF_FLAG_TEST_S_G_RPT(ch->flags)))
+       return 1;
+    }
 
   return 0;
 }