shadowy-pycoder
diff --git a/‎layers/tcp_test.go
Lines changed: 48 additions & 0 deletions b/‎layers/tcp_test.go
Lines changed: 48 additions & 0 deletions
diff --git a/‎layers/testdata/tcp.bin
40 Bytes b/‎layers/testdata/tcp.bin
40 Bytes
diff --git a/‎layers/testdata/udp.bin
52 Bytes b/‎layers/testdata/udp.bin
52 Bytes
diff --git a/‎layers/udp_test.go
Lines changed: 41 additions & 0 deletions b/‎layers/udp_test.go
Lines changed: 41 additions & 0 deletions
@@ -0,0 +1,48 @@
+package layers
+
+import (
+	"fmt"
+	"io"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func BenchmarkParseTCP(b *testing.B) {
+	packet, close := testPacketBench(b, "tcp")
+	defer close()
+	b.ResetTimer()
+	tcp := &TCPSegment{}
+	for i := 0; i < b.N; i++ {
+		_ = tcp.Parse(packet)
+		fmt.Fprint(io.Discard, tcp.String())
+	}
+}
+
+func TestParseTCP(t *testing.T) {
+	expected := &TCPSegment{
+		SrcPort:    42776,
+		DstPort:    443,
+		SeqNumber:  2726715057,
+		AckNumber:  0,
+		DataOffset: 10,
+		Reserved:   0,
+		Flags: &TCPFlags{
+			Raw: 2,
+			SYN: 1,
+		},
+		WindowSize: 64240,
+		Checksum:   30263,
+		Options: []byte{
+			0x02, 0x04, 0x05, 0xb4, 0x04, 0x02, 0x08, 0x0a, 0xac, 0xf8,
+			0x48, 0x3e, 0x00, 0x00, 0x00, 0x00, 0x01, 0x03, 0x03, 0x07},
+		payload: []byte{},
+	}
+	tcp := &TCPSegment{}
+	packet, close := testPacket(t, "tcp")
+	defer close()
+	if err := tcp.Parse(packet); err != nil {
+		t.Fatal(err)
+	}
+	require.Equal(t, expected, tcp)
+}
@@ -0,0 +1,41 @@
+package layers
+
+import (
+	"fmt"
+	"io"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func BenchmarkParseUDP(b *testing.B) {
+	packet, close := testPacketBench(b, "udp")
+	defer close()
+	b.ResetTimer()
+	udp := &UDPSegment{}
+	for i := 0; i < b.N; i++ {
+		_ = udp.Parse(packet)
+		fmt.Fprint(io.Discard, udp.String())
+	}
+}
+
+func TestParseUDP(t *testing.T) {
+	expected := &UDPSegment{
+		SrcPort:   33168,
+		DstPort:   53,
+		UDPLength: 52,
+		Checksum:  3985,
+		payload: []byte{
+			0xa8, 0x24, 0x01, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00,
+			0x01, 0x03, 0x77, 0x77, 0x77, 0x07, 0x67, 0x73, 0x74, 0x61, 0x74,
+			0x69, 0x63, 0x03, 0x63, 0x6f, 0x6d, 0x00, 0x00, 0x41, 0x00, 0x01,
+			0x00, 0x00, 0x29, 0x05, 0xc0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00},
+	}
+	udp := &UDPSegment{}
+	packet, close := testPacket(t, "udp")
+	defer close()
+	if err := udp.Parse(packet); err != nil {
+		t.Fatal(err)
+	}
+	require.Equal(t, expected, udp)
+}