server: resolve next-hop with FRR/Zebra for VRF routes

osrg · Jan 26, 2018 · e36879f · e36879f
1 parent 3356b9b
commit e36879f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/server/zclient.go b/server/zclient.go
@@ -506,7 +506,13 @@ func (z *zebraClient) loop() {
 				}
 			case *WatchEventUpdate:
 				if body, isWithdraw := newNexthopRegisterBody(msg.PathList, z.nhtManager); body != nil {
-					z.client.SendNexthopRegister(0, body, isWithdraw)
+					vrfId := uint16(0)
+					for _, vrf := range z.server.GetVrf() {
+						if vrf.Name == msg.Neighbor.Config.Vrf {
+							vrfId = uint16(vrf.Id)
+						}
+					}
+					z.client.SendNexthopRegister(vrfId, body, isWithdraw)
 				}
 			}
 		}