johanels/akvorado
1
0
Fork 0
mirror of https://github.com/akvorado/akvorado.git synced 2025-12-11 22:14:02 +01:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
Files
0c3a1668d873f55f4b3ca0bd405cda97b92cd62e
akvorado/inlet/flow/decoder/netflow/root_test.go
Vincent Bernat 0c3a1668d8 inlet/flow: rename SrcNet/DstNet to SrcNetMask/DstNetMask
2022-11-26 14:40:02 +01:00

216 lines
8.9 KiB
Go
Raw Blame History

// SPDX-FileCopyrightText: 2022 Free Mobile
// SPDX-License-Identifier: AGPL-3.0-only
package netflow
import (
"net"
"path/filepath"
"testing"
"akvorado/common/helpers"
"akvorado/common/reporter"
"akvorado/inlet/flow/decoder"
)
func TestDecode(t *testing.T) {
r := reporter.NewMock(t)
nfdecoder := New(r)
// Send an option template
template := helpers.ReadPcapPayload(t, filepath.Join("testdata", "options-template-257.pcap"))
got := nfdecoder.Decode(decoder.RawFlow{Payload: template, Source: net.ParseIP("127.0.0.1")})
if got == nil {
t.Fatalf("Decode() error on options template")
}
if len(got) != 0 {
t.Fatalf("Decode() on options template got flows")
}
// Check metrics
gotMetrics := r.GetMetrics("akvorado_inlet_flow_decoder_netflow_")
expectedMetrics := map[string]string{
`count{exporter="127.0.0.1",version="9"}`: "1",
`flowset_records_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`templates_count{exporter="127.0.0.1",obs_domain_id="0",template_id="257",type="options_template",version="9"}`: "1",
}
if diff := helpers.Diff(gotMetrics, expectedMetrics); diff != "" {
t.Fatalf("Metrics after template (-got, +want):\n%s", diff)
}
// Send option data
data := helpers.ReadPcapPayload(t, filepath.Join("testdata", "options-data-257.pcap"))
got = nfdecoder.Decode(decoder.RawFlow{Payload: data, Source: net.ParseIP("127.0.0.1")})
if got == nil {
t.Fatalf("Decode() error on options data")
}
if len(got) != 0 {
t.Fatalf("Decode() on options data got flows")
}
// Check metrics
gotMetrics = r.GetMetrics("akvorado_inlet_flow_decoder_netflow_")
expectedMetrics = map[string]string{
`count{exporter="127.0.0.1",version="9"}`: "2",
`flowset_records_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`flowset_records_sum{exporter="127.0.0.1",type="OptionsDataFlowSet",version="9"}`: "4",
`flowset_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="OptionsDataFlowSet",version="9"}`: "1",
`templates_count{exporter="127.0.0.1",obs_domain_id="0",template_id="257",type="options_template",version="9"}`: "1",
}
if diff := helpers.Diff(gotMetrics, expectedMetrics); diff != "" {
t.Fatalf("Metrics after template (-got, +want):\n%s", diff)
}
// Send a regular template
template = helpers.ReadPcapPayload(t, filepath.Join("testdata", "template-260.pcap"))
got = nfdecoder.Decode(decoder.RawFlow{Payload: template, Source: net.ParseIP("127.0.0.1")})
if got == nil {
t.Fatalf("Decode() error on template")
}
if len(got) != 0 {
t.Fatalf("Decode() on template got flows")
}
// Check metrics
gotMetrics = r.GetMetrics("akvorado_inlet_flow_decoder_netflow_")
expectedMetrics = map[string]string{
`count{exporter="127.0.0.1",version="9"}`: "3",
`flowset_records_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`flowset_records_sum{exporter="127.0.0.1",type="OptionsDataFlowSet",version="9"}`: "4",
`flowset_records_sum{exporter="127.0.0.1",type="TemplateFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="OptionsDataFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="TemplateFlowSet",version="9"}`: "1",
`templates_count{exporter="127.0.0.1",obs_domain_id="0",template_id="257",type="options_template",version="9"}`: "1",
`templates_count{exporter="127.0.0.1",obs_domain_id="0",template_id="260",type="template",version="9"}`: "1",
}
if diff := helpers.Diff(gotMetrics, expectedMetrics); diff != "" {
t.Fatalf("Metrics after template (-got, +want):\n%s", diff)
}
// Send data
data = helpers.ReadPcapPayload(t, filepath.Join("testdata", "data-260.pcap"))
got = nfdecoder.Decode(decoder.RawFlow{Payload: data, Source: net.ParseIP("127.0.0.1")})
if got == nil {
t.Fatalf("Decode() error on data")
}
expectedFlows := []*decoder.FlowMessage{
{
SequenceNum: 44797001,
ExporterAddress: net.ParseIP("127.0.0.1").To16(),
SamplingRate: 30000,
TimeFlowStart: 1647285926,
TimeFlowEnd: 1647285926,
Bytes: 1500,
Packets: 1,
SrcAddr: net.ParseIP("198.38.121.178").To16(),
DstAddr: net.ParseIP("91.170.143.87").To16(),
SrcNetMask: 24,
DstNetMask: 14,
Etype: 0x800,
Proto: 6,
SrcPort: 443,
DstPort: 19624,
InIf: 335,
OutIf: 450,
ForwardingStatus: 64,
TCPFlags: 16,
NextHop: net.ParseIP("194.149.174.63").To16(),
}, {
SequenceNum: 44797001,
ExporterAddress: net.ParseIP("127.0.0.1").To16(),
SamplingRate: 30000,
TimeFlowStart: 1647285926,
TimeFlowEnd: 1647285926,
Bytes: 1500,
Packets: 1,
SrcAddr: net.ParseIP("198.38.121.219").To16(),
DstAddr: net.ParseIP("88.122.57.97").To16(),
SrcNetMask: 24,
DstNetMask: 14,
Etype: 0x800,
Proto: 6,
SrcPort: 443,
DstPort: 2444,
InIf: 335,
OutIf: 452,
ForwardingStatus: 64,
TCPFlags: 16,
NextHop: net.ParseIP("194.149.174.71").To16(),
}, {
SequenceNum: 44797001,
ExporterAddress: net.ParseIP("127.0.0.1").To16(),
SamplingRate: 30000,
TimeFlowStart: 1647285926,
TimeFlowEnd: 1647285926,
Bytes: 1400,
Packets: 1,
SrcAddr: net.ParseIP("173.194.190.106").To16(),
DstAddr: net.ParseIP("37.165.129.20").To16(),
SrcNetMask: 20,
DstNetMask: 18,
Etype: 0x800,
Proto: 6,
SrcPort: 443,
DstPort: 53697,
InIf: 461,
OutIf: 306,
ForwardingStatus: 64,
TCPFlags: 16,
NextHop: net.ParseIP("252.223.0.0").To16(),
}, {
SequenceNum: 44797001,
ExporterAddress: net.ParseIP("127.0.0.1").To16(),
SamplingRate: 30000,
TimeFlowStart: 1647285926,
TimeFlowEnd: 1647285926,
Bytes: 1448,
Packets: 1,
SrcAddr: net.ParseIP("74.125.100.234").To16(),
DstAddr: net.ParseIP("88.120.219.117").To16(),
SrcNetMask: 16,
DstNetMask: 14,
Etype: 0x800,
Proto: 6,
SrcPort: 443,
DstPort: 52300,
InIf: 461,
OutIf: 451,
ForwardingStatus: 64,
TCPFlags: 16,
NextHop: net.ParseIP("194.149.174.61").To16(),
},
}
for _, f := range got {
f.TimeReceived = 0
}
if diff := helpers.Diff(got, expectedFlows); diff != "" {
t.Fatalf("Decode() (-got, +want):\n%s", diff)
}
gotMetrics = r.GetMetrics(
"akvorado_inlet_flow_decoder_netflow_",
"count",
"flowset_",
"templates_",
)
expectedMetrics = map[string]string{
`count{exporter="127.0.0.1",version="9"}`: "4",
`flowset_records_sum{exporter="127.0.0.1",type="DataFlowSet",version="9"}`: "4",
`flowset_records_sum{exporter="127.0.0.1",type="OptionsDataFlowSet",version="9"}`: "4",
`flowset_records_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`flowset_records_sum{exporter="127.0.0.1",type="TemplateFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="DataFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="OptionsDataFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="OptionsTemplateFlowSet",version="9"}`: "1",
`flowset_sum{exporter="127.0.0.1",type="TemplateFlowSet",version="9"}`: "1",
`templates_count{exporter="127.0.0.1",obs_domain_id="0",template_id="257",type="options_template",version="9"}`: "1",
`templates_count{exporter="127.0.0.1",obs_domain_id="0",template_id="260",type="template",version="9"}`: "1",
}
if diff := helpers.Diff(gotMetrics, expectedMetrics); diff != "" {
t.Fatalf("Metrics after data (-got, +want):\n%s", diff)
}
}
Reference in New Issue View Git Blame Copy Permalink