208 lines
4.6 KiB
Go
208 lines
4.6 KiB
Go
|
package sflow
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestUDPHeader(t *testing.T) {
|
||
|
|
octets := bytes.NewBuffer([]byte{
|
||
|
|
0x00, 0x01, // src_port
|
||
|
|
0x00, 0x02, // dst_port
|
||
|
|
0x00, 0x03, // udp_length
|
||
|
|
0x00, 0x00, // checksum
|
||
|
|
})
|
||
|
|
|
||
|
|
dc := NewDecoder()
|
||
|
|
actual, err := dc.decodeUDPHeader(octets)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
expected := UDPHeader{
|
||
|
|
SourcePort: 1,
|
||
|
|
DestinationPort: 2,
|
||
|
|
UDPLength: 3,
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Equal(t, expected, actual)
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkUDPHeader(b *testing.B) {
|
||
|
|
octets := bytes.NewBuffer([]byte{
|
||
|
|
0x00, 0x01, // src_port
|
||
|
|
0x00, 0x02, // dst_port
|
||
|
|
0x00, 0x03, // udp_length
|
||
|
|
0x00, 0x00, // checksum
|
||
|
|
})
|
||
|
|
|
||
|
|
dc := NewDecoder()
|
||
|
|
|
||
|
|
b.ResetTimer()
|
||
|
|
for n := 0; n < b.N; n++ {
|
||
|
|
dc.decodeUDPHeader(octets)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIPv4Header(t *testing.T) {
|
||
|
|
octets := bytes.NewBuffer(
|
||
|
|
[]byte{
|
||
|
|
0x45, // version + IHL
|
||
|
|
0x00, // ip_dscp + ip_ecn
|
||
|
|
0x00, 0x00, // total length
|
||
|
|
0x00, 0x00, // identification
|
||
|
|
0x00, 0x00, // flags + frag offset
|
||
|
|
0x00, // ttl
|
||
|
|
0x11, // protocol; 0x11 = udp
|
||
|
|
0x00, 0x00, // header checksum
|
||
|
|
0x7f, 0x00, 0x00, 0x01, // src ip
|
||
|
|
0x7f, 0x00, 0x00, 0x02, // dst ip
|
||
|
|
0x00, 0x01, // src_port
|
||
|
|
0x00, 0x02, // dst_port
|
||
|
|
0x00, 0x03, // udp_length
|
||
|
|
0x00, 0x00, // checksum
|
||
|
|
},
|
||
|
|
)
|
||
|
|
dc := NewDecoder()
|
||
|
|
actual, err := dc.decodeIPv4Header(octets)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
expected := IPV4Header{
|
||
|
|
Version: 0x40,
|
||
|
|
InternetHeaderLength: 0x05,
|
||
|
|
DSCP: 0,
|
||
|
|
ECN: 0,
|
||
|
|
TotalLength: 0,
|
||
|
|
Identification: 0,
|
||
|
|
Flags: 0,
|
||
|
|
FragmentOffset: 0,
|
||
|
|
TTL: 0,
|
||
|
|
Protocol: 0x11,
|
||
|
|
HeaderChecksum: 0,
|
||
|
|
SourceIP: [4]byte{127, 0, 0, 1},
|
||
|
|
DestIP: [4]byte{127, 0, 0, 2},
|
||
|
|
ProtocolHeader: UDPHeader{
|
||
|
|
SourcePort: 1,
|
||
|
|
DestinationPort: 2,
|
||
|
|
UDPLength: 3,
|
||
|
|
Checksum: 0,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Equal(t, expected, actual)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Using the same Directive instance, prior paths through the parse tree should
|
||
|
|
// not affect the latest parse.
|
||
|
|
func TestIPv4HeaderSwitch(t *testing.T) {
|
||
|
|
octets := bytes.NewBuffer(
|
||
|
|
[]byte{
|
||
|
|
0x45, // version + IHL
|
||
|
|
0x00, // ip_dscp + ip_ecn
|
||
|
|
0x00, 0x00, // total length
|
||
|
|
0x00, 0x00, // identification
|
||
|
|
0x00, 0x00, // flags + frag offset
|
||
|
|
0x00, // ttl
|
||
|
|
0x11, // protocol; 0x11 = udp
|
||
|
|
0x00, 0x00, // header checksum
|
||
|
|
0x7f, 0x00, 0x00, 0x01, // src ip
|
||
|
|
0x7f, 0x00, 0x00, 0x02, // dst ip
|
||
|
|
0x00, 0x01, // src_port
|
||
|
|
0x00, 0x02, // dst_port
|
||
|
|
0x00, 0x03, // udp_length
|
||
|
|
0x00, 0x00, // checksum
|
||
|
|
},
|
||
|
|
)
|
||
|
|
dc := NewDecoder()
|
||
|
|
_, err := dc.decodeIPv4Header(octets)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
octets = bytes.NewBuffer(
|
||
|
|
[]byte{
|
||
|
|
0x45, // version + IHL
|
||
|
|
0x00, // ip_dscp + ip_ecn
|
||
|
|
0x00, 0x00, // total length
|
||
|
|
0x00, 0x00, // identification
|
||
|
|
0x00, 0x00, // flags + frag offset
|
||
|
|
0x00, // ttl
|
||
|
|
0x06, // protocol; 0x06 = tcp
|
||
|
|
0x00, 0x00, // header checksum
|
||
|
|
0x7f, 0x00, 0x00, 0x01, // src ip
|
||
|
|
0x7f, 0x00, 0x00, 0x02, // dst ip
|
||
|
|
0x00, 0x01, // src_port
|
||
|
|
0x00, 0x02, // dst_port
|
||
|
|
0x00, 0x00, 0x00, 0x00, // sequence
|
||
|
|
0x00, 0x00, 0x00, 0x00, // ack_number
|
||
|
|
0x00, 0x00, // tcp_header_length
|
||
|
|
0x00, 0x00, // tcp_window_size
|
||
|
|
0x00, 0x00, // checksum
|
||
|
|
0x00, 0x00, // tcp_urgent_pointer
|
||
|
|
},
|
||
|
|
)
|
||
|
|
dc = NewDecoder()
|
||
|
|
actual, err := dc.decodeIPv4Header(octets)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
expected := IPV4Header{
|
||
|
|
Version: 64,
|
||
|
|
InternetHeaderLength: 5,
|
||
|
|
Protocol: 6,
|
||
|
|
SourceIP: [4]byte{127, 0, 0, 1},
|
||
|
|
DestIP: [4]byte{127, 0, 0, 2},
|
||
|
|
ProtocolHeader: TCPHeader{
|
||
|
|
SourcePort: 1,
|
||
|
|
DestinationPort: 2,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Equal(t, expected, actual)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUnknownProtocol(t *testing.T) {
|
||
|
|
octets := bytes.NewBuffer(
|
||
|
|
[]byte{
|
||
|
|
0x45, // version + IHL
|
||
|
|
0x00, // ip_dscp + ip_ecn
|
||
|
|
0x00, 0x00, // total length
|
||
|
|
0x00, 0x00, // identification
|
||
|
|
0x00, 0x00, // flags + frag offset
|
||
|
|
0x00, // ttl
|
||
|
|
0x99, // protocol
|
||
|
|
0x00, 0x00, // header checksum
|
||
|
|
0x7f, 0x00, 0x00, 0x01, // src ip
|
||
|
|
0x7f, 0x00, 0x00, 0x02, // dst ip
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
0x00,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
dc := NewDecoder()
|
||
|
|
actual, err := dc.decodeIPv4Header(octets)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
expected := IPV4Header{
|
||
|
|
Version: 64,
|
||
|
|
InternetHeaderLength: 5,
|
||
|
|
Protocol: 153,
|
||
|
|
SourceIP: [4]byte{127, 0, 0, 1},
|
||
|
|
DestIP: [4]byte{127, 0, 0, 2},
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Equal(t, expected, actual)
|
||
|
|
}
|