bgpd: Fix crash when entering 'address-family ipv4'

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

diff --git a/bgpd/bgp_vty.c b/bgpd/bgp_vty.c
index d6a36169f809438f2bc47bcf1273564cad639a34..153fb3a1a4ed09aa9d113c8c4acb59c94362edaf 100644 (file)
--- a/bgpd/bgp_vty.c
+++ b/bgpd/bgp_vty.c
@@ -5517,7 +5517,7 @@ DEFUN (address_family_ipv4_safi,
        BGP_SAFI_HELP_STR)
 {
   int idx_safi = 2;
-  if (argc == idx_safi)
+  if (argc == (idx_safi + 1))
     {
       switch (bgp_vty_safi_from_arg(argv[idx_safi]->arg))
         {
@@ -5550,7 +5550,7 @@ DEFUN (address_family_ipv6_safi,
        BGP_SAFI_HELP_STR)
 {
   int idx_safi = 2;
-  if (argc == idx_safi)
+  if (argc == (idx_safi + 1))
     {
       switch (bgp_vty_safi_from_arg(argv[idx_safi]->arg))
         {