From 400ceaf0de0d68753e4bb7ef2cac56d3b9a3e1b0 Mon Sep 17 00:00:00 2001
From: Sebastian Neuner <neuner@belwue.de>
Date: Thu, 5 May 2022 14:04:11 +0200
Subject: [PATCH] enable AddPath for BGP neighbors by manually configuring the
 AFI/SAFI (IPv4 for IPv4 neighbors, and IPv6 for IPv6 neighbors)

---
 routeinfo/routeinfo.go | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/routeinfo/routeinfo.go b/routeinfo/routeinfo.go
index 3909245..696d38d 100644
--- a/routeinfo/routeinfo.go
+++ b/routeinfo/routeinfo.go
@@ -97,9 +97,18 @@ type Router struct {
 
 func (r *Router) Connect() {
 	for _, addr := range r.Neighbors {
+		// determine AFI
 		var parsed net.IP
-		if parsed = net.ParseIP(addr); parsed == nil {
+		var afi api.Family_Afi
+		if parsed = net.ParseIP(addr); parsed != nil {
+		} else {
 			log.Printf("[error] Invalid address: %s", addr)
+			continue
+		}
+		if addr4 := parsed.To4(); addr4 != nil {
+			afi = api.Family_AFI_IP
+		} else {
+			afi = api.Family_AFI_IP6
 		}
 
 		if err := r.GobgpServer.AddPeer(context.Background(), &api.AddPeerRequest{
@@ -108,6 +117,21 @@ func (r *Router) Connect() {
 					NeighborAddress: addr,
 					PeerAsn:         r.Asn,
 				},
+				// define the AFI manually to enable Add-Paths
+				AfiSafis: []*api.AfiSafi{&api.AfiSafi{
+					Config: &api.AfiSafiConfig{
+						Family: &api.Family{
+							Afi:  afi,
+							Safi: api.Family_SAFI_UNICAST,
+						},
+						Enabled: true,
+					},
+					AddPaths: &api.AddPaths{
+						Config: &api.AddPathsConfig{
+							Receive: true,
+						},
+					},
+				}},
 			},
 		}); err != nil {
 			log.Fatal(err)