package protocol import ( "testing" ) func TestDetectMySQL(t *testing.T) { atomicFormats.Store([]format{{Server, "\x0a", detectMySQL}}) // 1. A valid MySQL 8.0 banner mysql8Banner := []byte{ 0x0a, 0x38, 0x2e, 0x30, 0x2e, 0x33, 0x32, 0x00, 0x0d, 0x00, 0x00, 0x00, 0x04, 0x5a, 0x56, 0x5f, 0x3e, 0x6e, 0x76, 0x27, 0x00, 0xff, 0xff, 0xff, 0x02, 0x00, 0xff, 0xc7, 0x15, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x63, 0x61, 0x63, 0x68, 0x69, 0x6e, 0x67, 0x5f, 0x73, 0x68, 0x61, 0x32, 0x5f, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x00, } // 2. A valid MariaDB banner (protocol-compatible) mariaDBBanner := []byte{ 0x0a, 0x35, 0x2e, 0x35, 0x2e, 0x35, 0x2d, 0x31, 0x30, 0x2e, 0x36, 0x2e, 0x35, 0x2d, 0x4d, 0x61, 0x72, 0x69, 0x61, 0x44, 0x42, 0x2d, 0x6c, 0x6f, 0x67, 0x00, 0x1a, 0x00, 0x00, 0x00, 0x4e, 0x5c, 0x32, 0x7b, 0x45, 0x3b, 0x40, 0x60, 0x00, 0xff, 0xf7, 0x08, 0x02, 0x00, 0xff, 0x81, 0x15, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x6d, 0x79, 0x73, 0x71, 0x6c, 0x5f, 0x6e, 0x61, 0x74, 0x69, 0x76, 0x65, 0x5f, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x00, } // 3. An invalid banner (e.g., an HTTP request) httpBanner := []byte("GET / HTTP/1.1\r\nHost: example.com\r\n\r\n") // 4. A short, invalid slice shortSlice := []byte{0x0a, 0x31, 0x32, 0x33} // 5. A slice that starts correctly but is malformed (no null terminator) malformedSlice := []byte{0x0a, 0x38, 0x2e, 0x30, 0x2e, 0x30, 0x01, 0x02, 0x03, 0x04, 0x05} testRunner(t, []*testCase{ { Name: "MySQL server", Direction: Server, Data: mysql8Banner, SrcPort: 3306, WantType: TypeMySQL, WantConfidence: .85, }, { Name: "MariaDB server", Direction: Server, Data: mariaDBBanner, SrcPort: 3306, WantType: TypeMySQL, WantConfidence: .85, }, { Name: "Invalid HTTP", Direction: Server, Data: httpBanner, SrcPort: 80, WantError: ErrUnknown, }, { Name: "Invalid too short", Direction: Server, Data: shortSlice, SrcPort: 3306, WantError: ErrUnknown, }, { Name: "Invalid malformed", Direction: Server, Data: malformedSlice, SrcPort: 3306, WantError: ErrUnknown, }, }) }