Initial import

2025-10-08 20:53:56 +02:00
commit 2081d684ed
25 changed files with 3486 additions and 0 deletions
--- a/protocol/detect_http_test.go
+++ b/protocol/detect_http_test.go
@@ -0,0 +1,154 @@
+package protocol
+
+import (
+	"errors"
+	"testing"
+)
+
+func TestDetectHTTPRequest(t *testing.T) {
+	atomicFormats.Store([]format{{Client, "", detectHTTPRequest}})
+
+	// A valid HTTP/1.0 GET request
+	http10Request := []byte("GET /old-page.html HTTP/1.0\r\nUser-Agent: NCSA_Mosaic/1.0\r\n\r\n")
+
+	// A valid HTTP/1.1 GET request
+	getRequest := []byte("GET /resource/item?id=123 HTTP/1.1\r\nHost: example.com\r\n\r\n")
+
+	// An invalid HTTP request
+	sshBanner := []byte("SSH-2.0-OpenSSH_8.9p1 Ubuntu-3ubuntu0.4\r\n")
+
+	defer func() { Strict = false }()
+	for _, strict := range []bool{false, true} {
+		Strict = strict
+
+		name := "loose"
+		if strict {
+			name = "strict"
+		}
+
+		t.Run(name, func(t *testing.T) {
+			t.Run("HTTP/1.0 GET", func(t *testing.T) {
+				p, err := Detect(Client, http10Request)
+				if err != nil {
+					t.Fatal(err)
+					return
+				}
+				t.Logf("detected %s version %s", p.Name, p.Version)
+				if p.Name != ProtocolHTTP {
+					t.Fatalf("expected http protocol, got %s", p.Name)
+					return
+				}
+			})
+
+			t.Run("HTTP/1.1 GET", func(t *testing.T) {
+				p, err := Detect(Client, getRequest)
+				if err != nil {
+					t.Fatal(err)
+					return
+				}
+				t.Logf("detected %s version %s", p.Name, p.Version)
+				if p.Name != ProtocolHTTP {
+					t.Fatalf("expected http protocol, got %s", p.Name)
+					return
+				}
+			})
+
+			t.Run("Invalid SSH", func(t *testing.T) {
+				_, err := Detect(Server, sshBanner)
+				if !errors.Is(err, ErrUnknown) {
+					t.Fatalf("expected unknown format, got error %T: %q", err, err)
+				} else {
+					t.Logf("error %q, as expected", err)
+				}
+			})
+		})
+	}
+}
+
+func TestDetectHTTPResponse(t *testing.T) {
+	atomicFormats.Store([]format{{Server, "HTTP/?.? ", detectHTTPResponse}})
+
+	// A valid HTTP/1.0 403 Forbidden response
+	http10Response := []byte("HTTP/1.0 403 Forbidden\r\nServer: CERN/3.0\r\n\r\n")
+
+	// A valid HTTP/1.1 200 OK response
+	responseOK := []byte("HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n<html>...</html>")
+
+	// A valid HTTP/1.1 404 Not Found response
+	responseNotFound := []byte("HTTP/1.1 404 Not Found\r\nContent-Length: 0\r\n\r\n")
+
+	// An invalid HTTP GET request
+	getRequest := []byte("GET /resource/item?id=123 HTTP/1.1\r\nHost: example.com\r\n\r\n")
+
+	// An invalid banner (SSH)
+	sshBanner := []byte("SSH-2.0-OpenSSH_8.9p1 Ubuntu-3ubuntu0.4\r\n")
+
+	defer func() { Strict = false }()
+	for _, strict := range []bool{false, true} {
+		Strict = strict
+
+		name := "loose"
+		if strict {
+			name = "strict"
+		}
+
+		t.Run(name, func(t *testing.T) {
+			t.Run("HTTP/1.0 403", func(t *testing.T) {
+				p, err := Detect(Server, http10Response)
+				if err != nil {
+					t.Fatal(err)
+					return
+				}
+				t.Logf("detected %s version %s", p.Name, p.Version)
+				if p.Name != ProtocolHTTP {
+					t.Fatalf("expected http protocol, got %s", p.Name)
+					return
+				}
+			})
+
+			t.Run("HTTP/1.1 200", func(t *testing.T) {
+				p, err := Detect(Server, responseOK)
+				if err != nil {
+					t.Fatal(err)
+					return
+				}
+				t.Logf("detected %s version %s", p.Name, p.Version)
+				if p.Name != ProtocolHTTP {
+					t.Fatalf("expected http protocol, got %s", p.Name)
+					return
+				}
+			})
+
+			t.Run("HTTP/1.1 404", func(t *testing.T) {
+				p, err := Detect(Server, responseNotFound)
+				if err != nil {
+					t.Fatal(err)
+					return
+				}
+				t.Logf("detected %s version %s", p.Name, p.Version)
+				if p.Name != ProtocolHTTP {
+					t.Fatalf("expected http protocol, got %s", p.Name)
+					return
+				}
+			})
+
+			t.Run("Invalid HTTP/1.1 GET", func(t *testing.T) {
+				_, err := Detect(Server, getRequest)
+				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("Invalid SSH", func(t *testing.T) {
+				_, err := Detect(Server, sshBanner)
+				if !errors.Is(err, ErrUnknown) {
+					t.Fatalf("expected unknown format, got error %T: %q", err, err)
+				} else {
+					t.Logf("error %q, as expected", err)
+				}
+			})
+		})
+	}
+}