83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
package protocol
|
|
|
|
import (
|
|
"errors"
|
|
"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}
|
|
|
|
t.Run("MySQL 8", func(t *testing.T) {
|
|
p, c, _ := Detect(Server, mysql8Banner, 3306, 0)
|
|
if p == nil {
|
|
t.Fatal("expected MySQL protocol, got nil")
|
|
}
|
|
t.Logf("detected %s version %s confidence %g%%", p.Name, p.Version, c*100)
|
|
})
|
|
|
|
t.Run("MariaDB", func(t *testing.T) {
|
|
p, c, _ := Detect(Server, mariaDBBanner, 3306, 0)
|
|
if p == nil {
|
|
t.Fatal("expected MySQL protocol, got nil")
|
|
}
|
|
t.Logf("detected %s version %s confidence %g%%", p.Name, p.Version, c*100)
|
|
})
|
|
|
|
t.Run("Invalid HTTP", func(t *testing.T) {
|
|
_, _, err := Detect(Server, httpBanner, 1234, 80)
|
|
if !errors.Is(err, ErrUnknown) {
|
|
t.Fatalf("expected unknown format, got error %T: %q", err, err)
|
|
} else {
|
|
t.Logf("error %q, as expected", err)
|
|
}
|
|
})
|
|
|
|
t.Run("Too short", func(t *testing.T) {
|
|
_, _, err := Detect(Server, shortSlice, 3306, 1234)
|
|
if !errors.Is(err, ErrUnknown) {
|
|
t.Fatalf("expected unknown format, got error %T: %q", err, err)
|
|
} else {
|
|
t.Logf("error %q, as expected", err)
|
|
}
|
|
})
|
|
|
|
t.Run("Malformed", func(t *testing.T) {
|
|
_, _, err := Detect(Server, malformedSlice, 3306, 1234)
|
|
if !errors.Is(err, ErrUnknown) {
|
|
t.Fatalf("expected unknown format, got error %T: %q", err, err)
|
|
} else {
|
|
t.Logf("error %q, as expected", err)
|
|
}
|
|
})
|
|
}
|