Initial import

2025-10-08 20:53:56 +02:00
commit 2081d684ed
25 changed files with 3486 additions and 0 deletions
--- a/protocol/detect_ssh_test.go
+++ b/protocol/detect_ssh_test.go
@@ -0,0 +1,103 @@
+package protocol
+
+import (
+	"errors"
+	"testing"
+)
+
+func TestDetectSSH(t *testing.T) {
+	atomicFormats.Store([]format{{Both, "", detectSSH}})
+
+	// 1. A standard OpenSSH banner
+	openSSHBanner := []byte("SSH-2.0-OpenSSH_8.9p1 Ubuntu-3ubuntu0.4\r\n")
+
+	// 2. An SSH banner with a pre-banner legal notice
+	preBannerSSH := []byte(
+		"*******************************************************************\r\n" +
+			"* W A R N I N G                          *\r\n" +
+			"* This system is for the use of authorized users only.          *\r\n" +
+			"*******************************************************************\r\n" +
+			"SSH-2.0-OpenSSH_7.6p1\r\n",
+	)
+
+	// 3. A different SSH implementation (Dropbear)
+	dropbearBanner := []byte("SSH-2.0-dropbear_2020.81\r\n")
+
+	// 4. An invalid banner (e.g., the MySQL banner from the previous example)
+	mysqlBanner := []byte{
+		0x0a, 0x38, 0x2e, 0x30, 0x2e, 0x33, 0x32, 0x00, 0x0d, 0x00, 0x00, 0x00,
+	}
+
+	// 5. A simple HTTP request
+	httpBanner := []byte("GET / HTTP/1.1\r\nHost: example.com\r\n\r\n")
+
+	t.Run("OpenSSH client", func(t *testing.T) {
+		p, err := Detect(Server, openSSHBanner)
+		if err != nil {
+			t.Fatal(err)
+			return
+		}
+		t.Logf("detected %s version %s", p.Name, p.Version)
+		if p.Name != ProtocolSSH {
+			t.Fatalf("expected ssh protocol, got %s", p.Name)
+			return
+		}
+	})
+
+	t.Run("OpenSSH server", func(t *testing.T) {
+		p, err := Detect(Server, openSSHBanner)
+		if err != nil {
+			t.Fatal(err)
+			return
+		}
+		t.Logf("detected %s version %s", p.Name, p.Version)
+		if p.Name != ProtocolSSH {
+			t.Fatalf("expected ssh protocol, got %s", p.Name)
+			return
+		}
+	})
+
+	t.Run("OpenSSH server with banner", func(t *testing.T) {
+		p, err := Detect(Server, preBannerSSH)
+		if err != nil {
+			t.Fatal(err)
+			return
+		}
+		t.Logf("detected %s version %s", p.Name, p.Version)
+		if p.Name != ProtocolSSH {
+			t.Fatalf("expected ssh protocol, got %s", p.Name)
+			return
+		}
+	})
+
+	t.Run("Dropbear server", func(t *testing.T) {
+		p, err := Detect(Server, dropbearBanner)
+		if err != nil {
+			t.Fatal(err)
+			return
+		}
+		t.Logf("detected %s version %s", p.Name, p.Version)
+		if p.Name != ProtocolSSH {
+			t.Fatalf("expected ssh protocol, got %s", p.Name)
+			return
+		}
+	})
+
+	t.Run("Invalid MySQL banner", func(t *testing.T) {
+		_, err := Detect(Server, mysqlBanner)
+		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 HTTP banner", func(t *testing.T) {
+		_, err := Detect(Server, httpBanner)
+		if !errors.Is(err, ErrUnknown) {
+			t.Fatalf("expected unknown format, got error %T: %q", err, err)
+		} else {
+			t.Logf("error %q, as expected", err)
+		}
+	})
+}