add BGP capability support for vpn

wobcom · Mar 14, 2024 · 6e577b5 · 6e577b5
1 parent bf7d66c
commit 6e577b5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/bgp_collector.rs b/src/bgp_collector.rs
@@ -23,13 +23,17 @@ pub async fn run_peer(
     let mut caps = vec![
         BgpCapability::SafiIPv4u,
         BgpCapability::SafiIPv6u,
+        BgpCapability::SafiVPNv4u,
+        BgpCapability::SafiVPNv6u,
         BgpCapability::CapRR,
         BgpCapability::CapASN32(cfg.asn),
     ];
     if cfg.add_path {
         caps.push(BgpCapability::CapAddPath(vec![
             BgpCapAddPath::new_from_cap(BgpCapability::SafiIPv4u, true, true).unwrap(),
             BgpCapAddPath::new_from_cap(BgpCapability::SafiIPv6u, true, true).unwrap(),
+            BgpCapAddPath::new_from_cap(BgpCapability::SafiVPNv4u, true, true).unwrap(),
+            BgpCapAddPath::new_from_cap(BgpCapability::SafiVPNv6u, true, true).unwrap(),
         ]));
     }