Fix SrcAS/DstAS for sFlow (cloudflare#125)

tgragnato · Oct 8, 2022 · 57fad2e · 57fad2e
1 parent f2a5297
commit 57fad2e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/producer/producer_sf.go b/producer/producer_sf.go
@@ -296,16 +296,19 @@ func SearchSFlowSamplesConfig(samples []interface{}, config *SFlowMapper) []*flo
 			case sflow.ExtendedGateway:
 				ipNh = recordData.NextHop
 				flowMessage.BgpNextHop = ipNh
-				flowMessage.SrcAS = recordData.SrcAS
 				flowMessage.BgpCommunities = recordData.Communities
 				flowMessage.AsPath = recordData.ASPath
 				if len(recordData.ASPath) > 0 {
 					flowMessage.DstAS = recordData.ASPath[len(recordData.ASPath)-1]
 					flowMessage.NextHopAS = recordData.ASPath[0]
-					flowMessage.SrcAS = recordData.AS
 				} else {
 					flowMessage.DstAS = recordData.AS
 				}
+				if recordData.SrcAS > 0 {
+					flowMessage.SrcAS = recordData.SrcAS
+				} else {
+					flowMessage.SrcAS = recordData.AS
+				}
 			case sflow.ExtendedSwitch:
 				flowMessage.SrcVlan = recordData.SrcVlan
 				flowMessage.DstVlan = recordData.DstVlan