mirror/netsampler-goflow2
1
0
Fork 0
mirror of https://github.com/netsampler/goflow2.git synced 2024-05-06 15:54:52 +00:00
Code Issues Projects Releases Wiki Activity

decoder: ipfix have max size (#238)

Browse Source
This commit is contained in:
Louis
2023-12-01 09:40:04 +01:00
committed by GitHub
parent 9287e05528
commit ac5e849a3b
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
decoders/netflow/netflow.go
+4 -1
View File
@@ -374,8 +374,10 @@ func DecodeMessageContext(ctx context.Context, payload *bytes.Buffer, templateKe
} else {
return nil, fmt.Errorf("NetFlow/IPFIX version error: %d", version)
}
read := 16
startSize := payload.Len()
for i := 0; ((i < int(size) && version == 9) || version == 10) && payload.Len() > 0; i++ {
for i := 0; ((i < int(size) && version == 9) || (uint16(read) < size && version == 10)) && payload.Len() > 0; i++ {
fsheader := FlowSetHeader{}
if err := utils.BinaryDecoder(payload, &fsheader.Id, &fsheader.Length); err != nil {
return returnItem, fmt.Errorf("Error decoding FlowSet header: %v", err)
@@ -522,6 +524,7 @@ func DecodeMessageContext(ctx context.Context, payload *bytes.Buffer, templateKe
} else if version == 10 && flowSet != nil {
packetIPFIX.FlowSets = append(packetIPFIX.FlowSets, flowSet)
}
read = startSize - payload.Len() + 16
}
if version == 9 {