pimd: Fix 'show ip pim state...' vrf changes

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

diff --git a/pimd/pim_cmd.c b/pimd/pim_cmd.c
index 2949d17cd9bb18beb83ba265d784bbcd1d0f6e03..4547bbb9fdc82f4623885be20928e4fd31f8cd23 100644 (file)
--- a/pimd/pim_cmd.c
+++ b/pimd/pim_cmd.c
@@ -3628,11 +3628,11 @@ DEFUN (show_ip_pim_state,
        if (uj)
                argc--;
 
-       if (argc == 6) {
-               src_or_group = argv[4]->arg;
-               group = argv[5]->arg;
-       } else if (argc == 5)
-               src_or_group = argv[4]->arg;
+       if (argv_find(argv, argc, "A.B.C.D", &idx)) {
+               src_or_group = argv[idx]->arg;
+               if (idx + 1 < argc)
+                       group = argv[idx + 1]->arg;
+       }
 
        pim_show_state(vty, vrf->info, src_or_group, group, uj);