Added MQTT detection

2025-10-09 17:42:55 +02:00
parent 13dc5a5d50
commit 05686075c4
3 changed files with 237 additions and 0 deletions
--- a/protocol/detect_mqtt_test.go
+++ b/protocol/detect_mqtt_test.go
@@ -0,0 +1,86 @@
+package protocol
+
+import "testing"
+
+func TestDetectMQTT(t *testing.T) {
+	// A valid CONNECT packet
+	validSimplePacket := []byte{
+		0x10, 0x15, // Fixed Header: Type=CONNECT, Remaining Length=21
+		0x00, 0x04, 'M', 'Q', 'T', 'T', // Protocol Name: "MQTT"
+		0x04,       // Protocol Version: 4 (v3.1.1)
+		0x02,       // Connect Flags: Clean Session=true
+		0x00, 0x3C, // Keep Alive: 60 seconds
+		0x00, 0x07, 'm', 'a', 'z', 'e', '.', 'i', 'o', // Client ID: "maze.io"
+	}
+
+	// A valid packet with Will, Username, and Password
+	validFullPacket := []byte{
+		0x10, 0x3E, // Type=CONNECT, Remaining Length=62
+		0x00, 0x04, 'M', 'Q', 'T', 'T',
+		0x05,       // Version
+		0b11001110, // Flags: User, Pass, Will(QoS=1, Retain=false), Clean
+		0x00, 0x1E, // Keep Alive: 30s
+		0x00, 0x0B, 't', 'e', 's', 't', '-', 'c', 'l', 'i', 'e', 'n', 't', // Client ID
+		0x00, 0x0D, 's', 't', 'a', 't', 'u', 's', '/', 'd', 'e', 'v', 'i', 'c', 'e', // Will Topic
+		0x00, 0x07, 'o', 'f', 'f', 'l', 'i', 'n', 'e', // Will Message
+		0x00, 0x04, 'u', 's', 'e', 'r', // Username
+		0x00, 0x09, 's', 'e', 'c', 'r', 'e', 't', '1', '2', '3', // Password
+	}
+
+	// Invalid packet (not a CONNECT packet)
+	//notConnectPacket := []byte{0x20, 0x02, 0x00, 0x00} // A CONNACK packet
+
+	// Partial packet (bad length)
+	partialPacket := []byte{
+		0x10, 0x0B, // Remaining length is 11, but we provide more data
+		0x00, 0x04, 'M', 'Q', 'T', 'T', // Protocol Name: "MQ"
+	}
+
+	// Trailing garbage packet (bad remaining length)
+	trailingGarbagePacket := []byte{
+		0x10, 0x0B, // Remaining length is 11, but we provide more data
+		0x00, 0x04, 'M', 'Q', 'T', 'T', // Protocol Name: "MQTT"
+		0x04,       // Protocol Version: 4 (v3.1.1)
+		0b00000010, // Flags: Clean Session=true
+		0x00, 0x3C, // Keep Alive: 60 seconds
+		0x00, 0x07, 'm', 'a', 'z', 'e', '.', 'i', 'o', // Client ID: "maze.io"
+		0x00, 0x01, 0x02, 0x03, // garbage
+	}
+
+	tests := []*testCase{
+		{
+			Name:           "MQTT simple packet",
+			Direction:      Client,
+			Data:           validSimplePacket,
+			DstPort:        1883,
+			WantProto:      ProtocolMQTT,
+			WantConfidence: .99,
+		},
+		{
+			Name:           "MQTT full packet",
+			Direction:      Client,
+			Data:           validFullPacket,
+			DstPort:        1883,
+			WantProto:      ProtocolMQTT,
+			WantConfidence: .99,
+		},
+		{
+			Name:           "MQTT partial packet",
+			Direction:      Client,
+			Data:           partialPacket,
+			DstPort:        1883,
+			WantProto:      ProtocolMQTT,
+			WantConfidence: .5,
+		},
+		{
+			Name:           "MQTT trailing garbage packet",
+			Direction:      Client,
+			Data:           trailingGarbagePacket,
+			DstPort:        1883,
+			WantProto:      ProtocolMQTT,
+			WantConfidence: .75,
+		},
+	}
+
+	testRunner(t, tests)
+}