bgpd: write advertise <ipv4|ipv6> unicast under bgp vrf config

Signed-off-by: Mitesh Kanjariya <mitesh@cumulusnetworks.com>

diff --git a/bgpd/bgp_evpn_vty.c b/bgpd/bgp_evpn_vty.c
index 0130b33cf85f4683e039ac65aaf7c3d5cc73a41e..1cf4e773e17824234517dca5ee02a2a8c0a8bfd5 100644 (file)
--- a/bgpd/bgp_evpn_vty.c
+++ b/bgpd/bgp_evpn_vty.c
@@ -4027,6 +4027,12 @@ void bgp_config_write_evpn_info(struct vty *vty, struct bgp *bgp, afi_t afi,
 
        if (bgp->advertise_gw_macip)
                vty_out(vty, "  advertise-default-gw\n");
+
+       if (CHECK_FLAG(bgp->vrf_flags, BGP_VRF_ADVERTISE_IPV4_IN_EVPN))
+               vty_out(vty, "  advertise ipv4 unicast\n");
+
+       if (CHECK_FLAG(bgp->vrf_flags, BGP_VRF_ADVERTISE_IPV6_IN_EVPN))
+               vty_out(vty, "  advertise ipv6 unicast\n");
 }
 
 void bgp_ethernetvpn_init(void)