1 files changed, 31 insertions, 0 deletions
diff --git a/internal/web/web_test.go b/internal/web/web_test.go
index b1609cd..dfc9629 100644
--- a/internal/web/web_test.go
+++ b/internal/web/web_test.go
@@ -1049,3 +1049,34 @@ func TestCrossSiteForm(t *testing.T) {
 		t.Errorf("cross-site login: got %d, want 403", rec.Code)
 	}
 }
+
+// Every response says nosniff and a referrer policy; only the management pages refuse to be framed.
+func TestSecurityHeaders(t *testing.T) {
+	s := NewServer(&config.Config{BaseDomain: "example.com", JWTSecret: []byte("x")}, nil)
+	get := func(host, path string) http.Header {
+		rec := httptest.NewRecorder()
+		req := httptest.NewRequest("GET", path, nil)
+		req.Host = host
+		s.ServeHTTP(rec, req)
+		return rec.Header()
+	}
+	h := get("example.com", "/webadmin")
+	if h.Get("X-Content-Type-Options") != "nosniff" || h.Get("Referrer-Policy") == "" {
+		t.Errorf("login page headers: %v", h)
+	}
+	if h.Get("X-Frame-Options") != "DENY" || !strings.Contains(h.Get("Content-Security-Policy"), "frame-ancestors 'none'") {
+		t.Errorf("login page not frame-protected: %v", h)
+	}
+	if h.Get("Strict-Transport-Security") != "" {
+		t.Error("HSTS sent without HTTPS")
+	}
+	h = get("evil.com", "/") // even the unknown-host 404 carries the common ones
+	if h.Get("X-Content-Type-Options") != "nosniff" {
+		t.Errorf("404 headers: %v", h)
+	}
+	for _, p := range []string{"/static/blog.css", "/webadmin"} { // the blog host may be framed (these need no DB)
+		if h := get("a.example.com", p); h.Get("X-Frame-Options") != "" || h.Get("Content-Security-Policy") != "" {
+			t.Errorf("%s on a blog host is frame-protected", p)
+		}
+	}
+}