Refactored detection logic to include ports and a confidence score
This commit is contained in:
@@ -179,7 +179,7 @@ func TestDetectTLS(t *testing.T) {
|
||||
t.Run(name, func(t *testing.T) {
|
||||
|
||||
t.Run("SSLv3 Client Hello", func(t *testing.T) {
|
||||
p, err := Detect(Client, sslV3ClientHello)
|
||||
p, _, err := Detect(Client, sslV3ClientHello, 0, 0)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
@@ -192,7 +192,7 @@ func TestDetectTLS(t *testing.T) {
|
||||
})
|
||||
|
||||
t.Run("TLS 1.1 Client Hello", func(t *testing.T) {
|
||||
p, err := Detect(Client, tls11ClientHello)
|
||||
p, _, err := Detect(Client, tls11ClientHello, 0, 0)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
@@ -205,7 +205,7 @@ func TestDetectTLS(t *testing.T) {
|
||||
})
|
||||
|
||||
t.Run("TLS 1.1 partial Client Hello", func(t *testing.T) {
|
||||
p, err := Detect(Client, tls11ClientHelloPartial)
|
||||
p, _, err := Detect(Client, tls11ClientHelloPartial, 0, 0)
|
||||
if strict {
|
||||
if !errors.Is(err, ErrUnknown) {
|
||||
t.Fatalf("expected unknown format, got error %T: %q", err, err)
|
||||
@@ -226,7 +226,7 @@ func TestDetectTLS(t *testing.T) {
|
||||
})
|
||||
|
||||
t.Run("TLS 1.2 Client Hello", func(t *testing.T) {
|
||||
p, err := Detect(Client, tls12ClientHello)
|
||||
p, _, err := Detect(Client, tls12ClientHello, 0, 0)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
@@ -239,7 +239,7 @@ func TestDetectTLS(t *testing.T) {
|
||||
})
|
||||
|
||||
t.Run("TLS 1.3 Client Hello", func(t *testing.T) {
|
||||
p, err := Detect(Client, tls13ClientHello)
|
||||
p, _, err := Detect(Client, tls13ClientHello, 0, 0)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
@@ -252,7 +252,7 @@ func TestDetectTLS(t *testing.T) {
|
||||
})
|
||||
|
||||
t.Run("Invalid PostgreSQL", func(t *testing.T) {
|
||||
_, err := Detect(Server, pgClientStartup)
|
||||
_, _, err := Detect(Server, pgClientStartup, 0, 0)
|
||||
if !errors.Is(err, ErrUnknown) {
|
||||
t.Fatalf("expected unknown format, got error %T: %q", err, err)
|
||||
} else {
|
||||
|
Reference in New Issue
Block a user